Return to buf.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / tftpd / src

Annotation of tftpd/src/buf.c, revision 1.3.2.1

1.3       misho       1: /*************************************************************************
                      2: * (C) 2014 AITNET ltd - Sofia/Bulgaria - <misho@aitnet.org>
                      3: *  by Michael Pounov <misho@elwix.org>
                      4: *
                      5: * $Author: misho $
1.3.2.1 ! misho       6: * $Id: buf.c,v 1.3 2014/02/24 22:28:54 misho Exp $
1.3       misho       7: *
                      8: **************************************************************************
                      9: The ELWIX and AITNET software is distributed under the following
                     10: terms:
                     11: 
                     12: All of the documentation and software included in the ELWIX and AITNET
                     13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
                     14: 
1.3.2.1 ! misho      15: Copyright 2004 - 2017
1.3       misho      16:        by Michael Pounov <misho@elwix.org>.  All rights reserved.
                     17: 
                     18: Redistribution and use in source and binary forms, with or without
                     19: modification, are permitted provided that the following conditions
                     20: are met:
                     21: 1. Redistributions of source code must retain the above copyright
                     22:    notice, this list of conditions and the following disclaimer.
                     23: 2. Redistributions in binary form must reproduce the above copyright
                     24:    notice, this list of conditions and the following disclaimer in the
                     25:    documentation and/or other materials provided with the distribution.
                     26: 3. All advertising materials mentioning features or use of this software
                     27:    must display the following acknowledgement:
                     28: This product includes software developed by Michael Pounov <misho@elwix.org>
                     29: ELWIX - Embedded LightWeight unIX and its contributors.
                     30: 4. Neither the name of AITNET nor the names of its contributors
                     31:    may be used to endorse or promote products derived from this software
                     32:    without specific prior written permission.
                     33: 
                     34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
                     35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     37: ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     44: SUCH DAMAGE.
                     45: */
1.2       misho      46: #include "global.h"
                     47: #include "buf.h"
                     48: 
                     49: 
                     50: int
                     51: initBuffer(int siz)
                     52: {
                     53:        ETRACE();
                     54: 
                     55:        bf = rpack_create(NULL, 0);
                     56:        if (!bf)
                     57:                return -1;
                     58:        if (rpack_attach(bf, siz) == -1) {
                     59:                rpack_destroy(&bf);
                     60:                return -1;
                     61:        }
                     62: 
                     63:        return 0;
                     64: }
                     65: 
                     66: void
                     67: endBuffer()
                     68: {
                     69:        flushBuffer(cli.fd);
                     70:        rpack_detach(bf);
                     71:        rpack_destroy(&bf);
                     72: }
                     73: 
                     74: int
                     75: flushBuffer(int fd)
                     76: {
                     77:        int ret = 0;
                     78:        const char *m;
                     79: 
                     80:        if (!bf || !cli.addr.sa.sa_len || cli.opc != TFTP_OPC_WRQ)
                     81:                return 0;
                     82: 
                     83:        m = cfg_getAttribute(&cfg, "tftpd", "buf_pad");
1.3       misho      84:        if (m && RPACK_OFF(bf) && RPACK_REMAIN(bf)) {
1.2       misho      85:                memset(RPACK_NEXT(bf), (u_char) strtol(m, NULL, 0), RPACK_REMAIN(bf));
1.3.2.1 ! misho      86:                EVERBOSE(4, "Padding buffer with '%c' len=%lu", *m, (u_long) RPACK_REMAIN(bf));
1.2       misho      87:                rpack_rnext(bf, RPACK_REMAIN(bf));
                     88:        }
                     89: 
                     90:        ret = write(fd, RPACK_BUF(bf), RPACK_OFF(bf));
                     91:        if (ret == -1)
                     92:                ESYSERR(0);
                     93:        else
                     94:                RPACK_REWIND(bf);
                     95: 
                     96:        EVERBOSE(3, "Flush buffer ret=%d", ret);
                     97:        return ret;
                     98: }
                     99: 
                    100: int
                    101: bfwrite(int fd, void *buf, int buflen)
                    102: {
                    103:        void *m;
                    104: 
                    105:        if (!buf)
                    106:                return -1;
                    107: 
                    108:        /* flush */
                    109:        if (RPACK_REMAIN(bf) < buflen)
                    110:                flushBuffer(fd);
                    111: 
                    112:        m = rpack_rdata(bf, buf, buflen);
                    113:        if (!m)
                    114:                return -1;
                    115: 
                    116:        e_free(m);
                    117:        return buflen;
                    118: }