|
version 1.2.2.1, 2014/02/20 00:44:49
|
version 1.2.2.7, 2014/02/20 16:04:03
|
|
Line 63 txData(sched_task_t *task)
|
Line 63 txData(sched_task_t *task)
|
| n = htons(cli.seq); |
n = htons(cli.seq); |
| rpack_uint16(pkt, &n, 0); |
rpack_uint16(pkt, &n, 0); |
| |
|
| len = pread(cli.fd, RPACK_NEXT(pkt), cli.siz, (cli.seq - 1) * cli.siz); | /* max file size check */ |
| | if (cli.tsiz && cli.tsiz < cli.seq * cli.siz) { |
| | len = cli.tsiz - (cli.seq - 1) * cli.siz; |
| | cli.close = 42; /* last sended packet, should be close! */ |
| | } else |
| | len = cli.siz; |
| | |
| | len = pread(cli.fd, RPACK_NEXT(pkt), len, (cli.seq - 1) * cli.siz); |
| if (len == -1) { |
if (len == -1) { |
| ESYSERR(0); |
ESYSERR(0); |
| code = htole16(3); |
code = htole16(3); |
|
Line 117 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; |
| |
int len; |
| |
|
| |
if (!rlen) |
| |
return -1; |
| |
|
| |
do { |
| |
/* option */ |
| |
len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL); |
| |
if (len == -1) |
| |
return -1; |
| |
else |
| |
rlen -= len; |
| |
opt = (char*) rpack_rnext(pkt, len); |
| |
if (!opt) |
| |
return -1; |
| |
EVERBOSE(7, "opt=%s rlen=%d", opt, rlen); |
| |
/* value */ |
| |
len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL); |
| |
if (len == -1) |
| |
return -1; |
| |
else |
| |
rlen -= len; |
| |
val = (char*) rpack_rnext(pkt, len); |
| |
if (!val) |
| |
return -1; |
| |
EVERBOSE(7, "val=%s rlen=%d", val, rlen); |
| |
|
| |
if (!strcasecmp(opt, TFTP_OPT_BLKSIZE)) { |
| |
cli.tmp = strtol(val, NULL, 10); |
| |
} 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 if (!strcasecmp(opt, TFTP_OPT_ROLLOVER)) |
| |
cli.roll = strtol(val, NULL, 10); |
| |
} while (rlen > 0); |
| |
|
| |
EVERBOSE(4, "blksize=%u tsize=%llu timeout=%u rollover=%u", |
| |
cli.siz, cli.tsiz, cli.tout, cli.roll); |
| return 0; |
return 0; |
| } |
} |
| |
|
|
Line 157 txOack(sched_task_t *task)
|
Line 204 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 175 RQ(sched_task_t *task)
|
Line 230 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 187 RQ(sched_task_t *task)
|
Line 244 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 201 RQ(sched_task_t *task)
|
Line 259 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 216 RQ(sched_task_t *task)
|
Line 273 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 237 RQ(sched_task_t *task)
|
Line 290 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 (!getOpts(pkt, rlen)) { |
| | if (cli.tmp > TFTP_LOAD_MAX) { |
| | if (rpack_resize(pkt, cli.tmp + 4)) |
| | ELIBERR(elwix); |
| | else |
| | cli.siz = cli.tmp; |
| | } |
| | 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 335 DATA(sched_task_t *task)
|
Line 397 DATA(sched_task_t *task)
|
| } else |
} else |
| cli.seq = ntohs(code); |
cli.seq = ntohs(code); |
| |
|
| |
/* 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) |
| |
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! */ |
| |
|
|
Line 418 rxPkt(sched_task_t *task)
|
Line 483 rxPkt(sched_task_t *task)
|
| TASK_DATA(task), rlen); |
TASK_DATA(task), rlen); |
| break; |
break; |
| case TFTP_OPC_OACK: |
case TFTP_OPC_OACK: |
| ELOGGER(LOG_WARNING, "oack"); |
|
| /* |
|
| schedEvent(TASK_ROOT(task), OACK, NULL, TASK_FD(task), |
|
| TASK_DATA(task), rlen); |
|
| */ |
|
| break; |
|
| case TFTP_OPC_ERROR: |
case TFTP_OPC_ERROR: |
| default: |
default: |
| RPACK_REWIND(pkt); |
RPACK_REWIND(pkt); |