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.3 2014/02/24 22:28:54 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: */
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");
84: if (m && RPACK_OFF(bf) && RPACK_REMAIN(bf)) {
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>