[BACK]Return to srv.c CVS log [TXT] [DIR] Up to [ELWIX - Embedded LightWeight unIX -] / tftpd / src

Diff for /tftpd/src/srv.c between versions 1.1.1.1 and 1.2.2.1

version 1.1.1.1, 2014/02/14 15:38:37 version 1.2.2.1, 2014/02/20 00:44:49
Line 1 Line 1
 #include "global.h"  #include "global.h"
   #include "exec.h"
   #include "srv.h"
   
   
 static void *  static void *
Line 30  txPkt(sched_task_t *task) Line 32  txPkt(sched_task_t *task)
         else if (wlen != TASK_DATLEN(task)) {          else if (wlen != TASK_DATLEN(task)) {
                 EERROR(EIO, "Sended %d bytes != packet %d bytes",                   EERROR(EIO, "Sended %d bytes != packet %d bytes", 
                                 wlen, TASK_DATLEN(task));                                  wlen, TASK_DATLEN(task));
                   schedEvent(TASK_ROOT(task), execProg, "error", 0, NULL, TFTP_OPC_ERROR);
                 schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL,                   schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, 
                                 timeoutSession, NULL);                                  timeoutSession, NULL);
                memset(&cli, 0, sizeof(sockaddr_t));                schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, NULL, 0);
         } else          } else
                 EVERBOSE(2, "Sended %d bytes", wlen);                  EVERBOSE(2, "Sended %d bytes", wlen);
         /* on error or argument, drop session */          /* on error or argument, drop session */
         if (TASK_ARG(task) == (void*) -1 || ntohs(tftp->tftp_opc) == TFTP_OPC_ERROR) {          if (TASK_ARG(task) == (void*) -1 || ntohs(tftp->tftp_opc) == TFTP_OPC_ERROR) {
                   schedEvent(TASK_ROOT(task), execProg, "error", 0, NULL, TFTP_OPC_ERROR);
                 schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL,                   schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, 
                                 timeoutSession, NULL);                                  timeoutSession, NULL);
                memset(&cli, 0, sizeof cli);                schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, NULL, 0);
         }          }
   
         taskExit(task, NULL);          taskExit(task, NULL);
Line 104  txAck(sched_task_t *task) Line 108  txAck(sched_task_t *task)
                         TASK_DATA(task), RPACK_OFF(pkt));                          TASK_DATA(task), RPACK_OFF(pkt));
   
         if (cli.close) {          if (cli.close) {
                   schedEvent(TASK_ROOT(task), execProg, "complete", 0, NULL, TFTP_OPC_WRQ);
                 schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);                  schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);
                 schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, NULL, 0);                  schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, NULL, 0);
                 EVERBOSE(2, "Finish WRQ request");                  EVERBOSE(2, "Finish WRQ request");
Line 111  txAck(sched_task_t *task) Line 116  txAck(sched_task_t *task)
         taskExit(task, NULL);          taskExit(task, NULL);
 }  }
   
   static int
   getOpts(rpack_t * __restrict pkt)
   {
           return 0;
   }
   
 static void *  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)  RQ(sched_task_t *task)
 {  {
         rpack_t *pkt = TASK_DATA(task);          rpack_t *pkt = TASK_DATA(task);
Line 163  RQ(sched_task_t *task) Line 215  RQ(sched_task_t *task)
                         EVERBOSE(2, "WRQ:: file=%s mode=%s\n", cli.file, cli.mode);                          EVERBOSE(2, "WRQ:: file=%s mode=%s\n", cli.file, cli.mode);
                         break;                          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);          cli.fd = open(cli.file, code, 0644);
         if (cli.fd == -1) {          if (cli.fd == -1) {
                 if (errno == EACCES)                  if (errno == EACCES)
Line 178  RQ(sched_task_t *task) Line 235  RQ(sched_task_t *task)
         } else          } else
                 cli.seq = 0;                  cli.seq = 0;
   
           schedEvent(TASK_ROOT(task), execProg, "request", 0, NULL, cli.opc);
   
         if (cli.opc == TFTP_OPC_WRQ) {          if (cli.opc == TFTP_OPC_WRQ) {
                 /* ack */                  /* ack */
                 tftp->tftp_opc = htons(TFTP_OPC_ACK);                  tftp->tftp_opc = htons(TFTP_OPC_ACK);
Line 234  ACK(sched_task_t *task) Line 293  ACK(sched_task_t *task)
                 schedEvent(TASK_ROOT(task), txData, NULL, TASK_FD(task),                   schedEvent(TASK_ROOT(task), txData, NULL, TASK_FD(task), 
                                 TASK_DATA(task), 0);                                  TASK_DATA(task), 0);
         else {          else {
                   schedEvent(TASK_ROOT(task), execProg, "complete", 0, NULL, TFTP_OPC_RRQ);
                 schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);                  schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);
                 schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, NULL, 0);                  schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, NULL, 0);
                 EVERBOSE(2, "Finish RRQ request");                  EVERBOSE(2, "Finish RRQ request");
Removed from v.1.1.1.1  
changed lines
  Added in v.1.2.2.1

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>