--- tftpd/src/srv.c 2014/02/17 14:28:29 1.2 +++ tftpd/src/srv.c 2014/02/20 01:15:45 1.2.2.2 @@ -116,7 +116,85 @@ txAck(sched_task_t *task) taskExit(task, NULL); } +static int +getOpts(rpack_t * __restrict pkt) +{ + char *opt, *val; + int len; + + do { + /* option */ + len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL); + if (len == -1) + return -1; + opt = (char*) rpack_rnext(pkt, len); + if (!opt) + return -1; + /* value */ + len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL); + if (len == -1) + return -1; + val = (char*) rpack_rnext(pkt, len); + if (!val) + return -1; + + if (!strcasecmp(opt, TFTP_OPT_BLKSIZE)) { + len = strtol(val, NULL, 10); + if (len > TFTP_LOAD_MAX) + cli.siz = len; + } else if (!strcasecmp(opt, TFTP_OPT_TSIZE)) + cli.tsiz = strtoll(val, NULL, 10); + else if (!strcasecmp(opt, TFTP_OPT_TIMEOUT)) + cli.tout = strtol(val, NULL, 10); + else + return -1; + } while (!RPACK_ISEND(pkt)); + + return 0; +} + static void * +txOack(sched_task_t *task) +{ + rpack_t *pkt = TASK_DATA(task); + u_short n = htons(TFTP_OPC_OACK); + struct stat sb; + char szStr[STRSIZ]; + + ETRACE(); + + RPACK_REWIND(pkt); + rpack_uint16(pkt, &n, 0); + + /* if opcode is RRQ and tsize is 0 then we must return file size to client */ + if (cli.opc == TFTP_OPC_RRQ && !cli.tsiz && stat(cli.file, &sb) != -1) + cli.tsiz = sb.st_size; + + if (cli.siz > TFTP_LOAD_MAX) { + memset(szStr, 0, sizeof szStr); + snprintf(szStr, sizeof szStr, "%u", cli.siz); + rpack_rdata(pkt, TFTP_OPT_BLKSIZE, strlen(TFTP_OPT_BLKSIZE) + 1); + rpack_rdata(pkt, szStr, strlen(szStr) + 1); + } + if (cli.tsiz) { + memset(szStr, 0, sizeof szStr); + snprintf(szStr, sizeof szStr, "%llu", cli.tsiz); + rpack_rdata(pkt, TFTP_OPT_TSIZE, strlen(TFTP_OPT_TSIZE) + 1); + rpack_rdata(pkt, szStr, strlen(szStr) + 1); + } + if (cli.tout) { + memset(szStr, 0, sizeof szStr); + snprintf(szStr, sizeof szStr, "%u", cli.tout); + rpack_rdata(pkt, TFTP_OPT_TIMEOUT, strlen(TFTP_OPT_TIMEOUT) + 1); + rpack_rdata(pkt, szStr, strlen(szStr) + 1); + } + + schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task), + TASK_DATA(task), RPACK_OFF(pkt)); + taskExit(task, NULL); +} + +static void * RQ(sched_task_t *task) { rpack_t *pkt = TASK_DATA(task); @@ -168,6 +246,11 @@ RQ(sched_task_t *task) EVERBOSE(2, "WRQ:: file=%s mode=%s\n", cli.file, cli.mode); break; } + + if (!RPACK_ISEND(pkt) && !getOpts(pkt)) + schedEvent(TASK_ROOT(task), txOack, NULL, TASK_FD(task), + TASK_DATA(task), 0); + cli.fd = open(cli.file, code, 0644); if (cli.fd == -1) { if (errno == EACCES)