--- tftpd/src/srv.c 2014/02/24 15:23:39 1.7 +++ tftpd/src/srv.c 2014/02/24 22:28:54 1.8 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: srv.c,v 1.7 2014/02/24 15:23:39 misho Exp $ +* $Id: srv.c,v 1.8 2014/02/24 22:28:54 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -171,6 +171,7 @@ txAck(sched_task_t *task) schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL); schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, TASK_DATA(task), 0); EVERBOSE(2, "Finish WRQ request"); + schedResumeby(TASK_ROOT(task), CRITERIA_ID, 0); } taskExit(task, NULL); } @@ -285,6 +286,10 @@ RQ(sched_task_t *task) cli.siz = TFTP_LOAD_MAX; cli.opc = ntohs(rpack_uint16(pkt, NULL, 0)); + if (!RW && cli.opc == TFTP_OPC_WRQ) { + code = htons(2); + goto end; + } len = str_getString(RPACK_NEXT(pkt), rlen, &str); if (len == -1) goto end; @@ -473,18 +478,19 @@ DATA(sched_task_t *task) EVERBOSE(3, "DATA:: seq=%hu; len=%d", cli.seq, len); - if (!bf) - len = pwrite(cli.fd, RPACK_NEXT(pkt), len, (cli.seq - 1) * cli.siz); - else - len = bfwrite(cli.fd, RPACK_NEXT(pkt), len); - if (len == -1) { - ESYSERR(0); - code = htons(3); - goto end; - } else { - rpack_rnext(pkt, len); - EVERBOSE(3, "Written to file %s %d bytes", cli.file, len); + if (len > 0) { + if (!bf) + len = pwrite(cli.fd, RPACK_NEXT(pkt), len, (cli.seq - 1) * cli.siz); + else + len = bfwrite(cli.fd, RPACK_NEXT(pkt), len); + if (len == -1) { + ESYSERR(0); + code = htons(3); + goto end; + } else + rpack_rnext(pkt, len); } + EVERBOSE(3, "Written to file %s %d bytes", cli.file, len); schedEvent(TASK_ROOT(task), txAck, NULL, TASK_FD(task), TASK_DATA(task), 0); taskExit(task, NULL);