--- tftpd/src/tftpd.c 2014/02/21 09:09:15 1.2 +++ tftpd/src/tftpd.c 2014/02/24 15:23:39 1.3 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: tftpd.c,v 1.2 2014/02/21 09:09:15 misho Exp $ +* $Id: tftpd.c,v 1.3 2014/02/24 15:23:39 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -45,12 +45,14 @@ SUCH DAMAGE. */ #include "global.h" #include "srv.h" +#include "buf.h" intptr_t Kill; struct tagCli cli; cfg_root_t cfg; sched_root_task_t *root; +rpack_t *bf; char szCfgName[PATH_MAX] = DEFAULT_CFGNAME; struct timespec timeout = { DEFAULT_TIMEOUT, 0 }; extern char compiled[], compiledby[], compilehost[]; @@ -76,6 +78,7 @@ Usage() "\t-c \tConfig file [default=/etc/tftpd.conf]\n" "\t-w\t\tSwitch to read-write mode [default=read-only]\n" "\t-b\t\tRun into batch mode (default is daemon mode)\n" + "\t-d\t\tDebug program\n" "\t-v\t\tVerbose (more -v, more verbosity ...)\n" "\t-h\t\tThis help screen!\n" "\n", compiled, compiledby, compilehost); @@ -99,6 +102,11 @@ sigHandler(sched_task_t *task) str = cfg_getAttribute(&cfg, "tftpd", "timeout"); if (str) timeout.tv_sec = strtol(str, NULL, 10); + + endBuffer(); + str = cfg_getAttribute(&cfg, "tftpd", "buf_io"); + if (str) + initBuffer(strtol(str, NULL, 0)); break; case SIGINT: case SIGTERM: @@ -128,7 +136,7 @@ main(int argc, char **argv) sockaddr_t sa; rpack_t *pkt = NULL; - while ((ch = getopt(argc, argv, "hvbwc:")) != -1) + while ((ch = getopt(argc, argv, "hvdbwc:")) != -1) switch (ch) { case 'c': strlcpy(szCfgName, optarg, sizeof szCfgName); @@ -139,6 +147,9 @@ main(int argc, char **argv) case 'v': e_incVerbose; break; + case 'd': + elwix_Debug |= ELWIX_DEBUG_TRACE; + break; case 'b': b = 42; break; @@ -235,6 +246,10 @@ main(int argc, char **argv) chdir(str); } + str = cfg_getAttribute(&cfg, "tftpd", "buf_io"); + if (str) + initBuffer(strtol(str, NULL, 0)); + if (!(pkt = rpack_create(NULL, 0))) { ELIBERR(elwix); close(fd); @@ -261,6 +276,7 @@ end: schedEnd(&root); rpack_detach(pkt); rpack_destroy(&pkt); + endBuffer(); cfgUnloadConfig(&cfg); closelog(); return ret;