|
|
| version 1.1.1.1, 2014/02/14 15:38:37 | version 1.3, 2014/02/18 12:46:39 |
|---|---|
| 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(const char *opts, int rlen) | |
| { | |
| if (!opts) | |
| return -1; | |
| return 0; | |
| } | |
| static void * | static void * |
| RQ(sched_task_t *task) | RQ(sched_task_t *task) |
| { | { |
| rpack_t *pkt = TASK_DATA(task); | rpack_t *pkt = TASK_DATA(task); |
| struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt); | struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt); |
| int len, rlen = TASK_DATLEN(task) - 2; | int len, rlen = TASK_DATLEN(task) - 2; |
| char *str; | char *str, *opts; |
| u_short code = 0; | u_short code = 0; |
| ETRACE(); | ETRACE(); |
| Line 130 RQ(sched_task_t *task) | Line 144 RQ(sched_task_t *task) |
| rlen -= len; | rlen -= 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, &opts); |
| if (len == -1) | if (len == -1) |
| goto end; | goto end; |
| else { | else { |
| Line 149 RQ(sched_task_t *task) | Line 163 RQ(sched_task_t *task) |
| } | } |
| } | } |
| /* tftp extended options */ | |
| if (!RPACK_ISEND(pkt) && !getOpts(opts, rlen)) | |
| cli.opts = 42; /* we have options */ | |
| cli.opc = ntohs(tftp->tftp_opc); | cli.opc = ntohs(tftp->tftp_opc); |
| switch (cli.opc) { | switch (cli.opc) { |
| case TFTP_OPC_RRQ: | case TFTP_OPC_RRQ: |
| Line 178 RQ(sched_task_t *task) | Line 196 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 254 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"); |