[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.2.2.5 and 1.2.2.6

version 1.2.2.5, 2014/02/20 14:20:42 version 1.2.2.6, 2014/02/20 15:39:15
Line 64  txData(sched_task_t *task) Line 64  txData(sched_task_t *task)
         rpack_uint16(pkt, &n, 0);          rpack_uint16(pkt, &n, 0);
   
         /* max file size check */          /* max file size check */
        if (cli.tsiz && cli.tsiz <= cli.seq * cli.siz) {        if (cli.tsiz && cli.tsiz < cli.seq * cli.siz) {
                len = MAX(0, cli.tsiz - (cli.seq - 1) * cli.siz);                len = cli.tsiz - (cli.seq - 1) * cli.siz;
                 cli.close = 42; /* last sended packet, should be close! */                  cli.close = 42; /* last sended packet, should be close! */
         } else          } else
                 len = cli.siz;                  len = cli.siz;
Line 124  txAck(sched_task_t *task) Line 124  txAck(sched_task_t *task)
 }  }
   
 static int  static int
getOpts(rpack_t * __restrict pkt)getOpts(rpack_t * __restrict pkt, int rlen)
 {  {
         char *opt, *val;          char *opt, *val;
         int len;          int len;
Line 134  getOpts(rpack_t * __restrict pkt) Line 134  getOpts(rpack_t * __restrict pkt)
                 len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL);                  len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL);
                 if (len == -1)                  if (len == -1)
                         return -1;                          return -1;
                   else
                           rlen -= len;
                 opt = (char*) rpack_rnext(pkt, len);                  opt = (char*) rpack_rnext(pkt, len);
                 if (!opt)                  if (!opt)
                         return -1;                          return -1;
Line 141  getOpts(rpack_t * __restrict pkt) Line 143  getOpts(rpack_t * __restrict pkt)
                 len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL);                  len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL);
                 if (len == -1)                  if (len == -1)
                         return -1;                          return -1;
                   else
                           rlen -= len;
                 val = (char*) rpack_rnext(pkt, len);                  val = (char*) rpack_rnext(pkt, len);
                 if (!val)                  if (!val)
                         return -1;                          return -1;
   
                 if (!strcasecmp(opt, TFTP_OPT_BLKSIZE)) {                  if (!strcasecmp(opt, TFTP_OPT_BLKSIZE)) {
                         len = strtol(val, NULL, 10);                          len = strtol(val, NULL, 10);
                        if (len > TFTP_LOAD_MAX)                        if (len > TFTP_LOAD_MAX) {
                                 cli.siz = len;                                  cli.siz = len;
                                   if (rpack_resize(pkt, cli.siz + 4))
                                           cli.siz = TFTP_PKT_MAX;
                           }
                 } else if (!strcasecmp(opt, TFTP_OPT_TSIZE))                  } else if (!strcasecmp(opt, TFTP_OPT_TSIZE))
                         cli.tsiz = strtoll(val, NULL, 10);                          cli.tsiz = strtoll(val, NULL, 10);
                 else if (!strcasecmp(opt, TFTP_OPT_TIMEOUT))                  else if (!strcasecmp(opt, TFTP_OPT_TIMEOUT))
                         cli.tout = strtol(val, NULL, 10);                          cli.tout = strtol(val, NULL, 10);
                   else if (!strcasecmp(opt, TFTP_OPT_ROLLOVER))
                           cli.roll = strtol(val, NULL, 10);
                 else                  else
                         return -1;                          return -1;
        } while (!RPACK_ISEND(pkt));        } while (rlen > 0);
   
        EVERBOSE(4, "blksize=%u tsize=%llu timeout=%u", cli.siz, cli.tsiz, cli.tout);        EVERBOSE(4, "blksize=%u tsize=%llu timeout=%u rollover=%u", 
                         cli.siz, cli.tsiz, cli.tout, cli.roll);
         return 0;          return 0;
 }  }
   
Line 196  txOack(sched_task_t *task) Line 206  txOack(sched_task_t *task)
                 rpack_rdata(pkt, TFTP_OPT_TIMEOUT, strlen(TFTP_OPT_TIMEOUT) + 1);                  rpack_rdata(pkt, TFTP_OPT_TIMEOUT, strlen(TFTP_OPT_TIMEOUT) + 1);
                 rpack_rdata(pkt, szStr, strlen(szStr) + 1);                  rpack_rdata(pkt, szStr, strlen(szStr) + 1);
         }          }
           if (cli.roll) {
                   memset(szStr, 0, sizeof szStr);
                   snprintf(szStr, sizeof szStr, "%u", cli.roll);
                   rpack_rdata(pkt, TFTP_OPT_ROLLOVER, strlen(TFTP_OPT_ROLLOVER) + 1);
                   rpack_rdata(pkt, szStr, strlen(szStr) + 1);
           }
   
           EVERBOSE(4, "blksize=%u tsize=%llu timeout=%u rollover=%u", 
                           cli.siz, cli.tsiz, cli.tout, cli.roll);
         schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task),           schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task), 
                         TASK_DATA(task), RPACK_OFF(pkt));                          TASK_DATA(task), RPACK_OFF(pkt));
         taskExit(task, NULL);          taskExit(task, NULL);
Line 214  RQ(sched_task_t *task) Line 232  RQ(sched_task_t *task)
         ETRACE();          ETRACE();
   
         cli.siz = TFTP_LOAD_MAX;          cli.siz = TFTP_LOAD_MAX;
           cli.opc = ntohs(rpack_uint16(pkt, NULL, 0));
         len = str_getString(tftp->tftp_data, rlen, &str);          len = str_getString(tftp->tftp_data, rlen, &str);
         if (len == -1)          if (len == -1)
                 goto end;                  goto end;
         else {          else {
                 rlen -= len;                  rlen -= len;
                   rpack_rnext(pkt, len);
                 strlcpy(cli.file, (char*) tftp->tftp_data, sizeof cli.file);                  strlcpy(cli.file, (char*) tftp->tftp_data, sizeof cli.file);
         }          }
         len = str_getString((const u_char*) str, rlen, NULL);          len = str_getString((const u_char*) str, rlen, NULL);
Line 226  RQ(sched_task_t *task) Line 246  RQ(sched_task_t *task)
                 goto end;                  goto end;
         else {          else {
                 rlen -= len;                  rlen -= len;
                   rpack_rnext(pkt, len);
                 if (!strcasecmp(str, TFTP_MODE_ASCII))                  if (!strcasecmp(str, TFTP_MODE_ASCII))
                         strlcpy(cli.mode, TFTP_MODE_ASCII, sizeof cli.mode);                          strlcpy(cli.mode, TFTP_MODE_ASCII, sizeof cli.mode);
                 else if (!strcasecmp(str, TFTP_MODE_OCTET))                  else if (!strcasecmp(str, TFTP_MODE_OCTET))
Line 240  RQ(sched_task_t *task) Line 261  RQ(sched_task_t *task)
                 }                  }
         }          }
   
         cli.opc = ntohs(tftp->tftp_opc);  
         switch (cli.opc) {          switch (cli.opc) {
                 case TFTP_OPC_RRQ:                  case TFTP_OPC_RRQ:
                         code = O_RDONLY;                          code = O_RDONLY;
Line 255  RQ(sched_task_t *task) Line 275  RQ(sched_task_t *task)
                         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 276  RQ(sched_task_t *task) Line 292  RQ(sched_task_t *task)
   
         schedEvent(TASK_ROOT(task), execProg, "request", 0, NULL, cli.opc);          schedEvent(TASK_ROOT(task), execProg, "request", 0, NULL, cli.opc);
   
        if (cli.opc == TFTP_OPC_WRQ) {        if (!RPACK_ISEND(pkt) && !getOpts(pkt, rlen))
                 schedEvent(TASK_ROOT(task), txOack, NULL, TASK_FD(task), 
                                 TASK_DATA(task), 0);
         else if (cli.opc == TFTP_OPC_WRQ) {
                 /* ack */                  /* ack */
                 tftp->tftp_opc = htons(TFTP_OPC_ACK);                  tftp->tftp_opc = htons(TFTP_OPC_ACK);
                 RPACK_REWIND(pkt);                  RPACK_REWIND(pkt);
Line 376  DATA(sched_task_t *task) Line 395  DATA(sched_task_t *task)
   
         /* max file size check */          /* max file size check */
         len = TASK_DATLEN(task) - RPACK_OFF(pkt);          len = TASK_DATLEN(task) - RPACK_OFF(pkt);
        if (cli.tsiz && cli.tsiz <= cli.seq * cli.siz)        if (cli.tsiz && cli.tsiz < cli.seq * cli.siz)
                 len = MIN(len, cli.tsiz - (cli.seq - 1) * cli.siz);                  len = MIN(len, cli.tsiz - (cli.seq - 1) * cli.siz);
         if (len < cli.siz)          if (len < cli.siz)
                 cli.close = 42; /* last received packet, should be close! */                  cli.close = 42; /* last received packet, should be close! */
Removed from v.1.2.2.5  
changed lines
  Added in v.1.2.2.6

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