Annotation of tftpd/src/buf.c, revision 1.2.2.2
1.2.2.2 ! 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: exec.h,v 1.3 2014/02/21 09:09:15 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.2.2.1 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: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>