Return to buf.c CVS log

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

version 1.2

    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: }