Return to buf.c CVS log

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

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

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 $
        !             6: * $Id: buf.c,v 1.2.2.2 2014/02/24 21:55:35 misho Exp $
        !             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: 
        !            15: Copyright 2004 - 2014
        !            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));
                     86:                EVERBOSE(4, "Padding buffer with '%c' len=%d", *m, RPACK_REMAIN(bf));
                     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: }