Annotation of tftpd/src/srv.c, revision 1.2.2.8
1.1 misho 1: #include "global.h"
1.2 misho 2: #include "exec.h"
3: #include "srv.h"
1.1 misho 4:
5:
6: static void *
7: timeoutSession(sched_task_t *task)
8: {
1.2.2.8 ! misho 9: rpack_t *pkt = TASK_DATA(task);
! 10:
1.1 misho 11: ETRACE();
12:
13: /* drop session */
14: if (cli.fd > 2)
15: close(cli.fd);
1.2.2.8 ! misho 16: rpack_resize(pkt, TFTP_PKT_MAX);
1.1 misho 17: memset(&cli, 0, sizeof cli);
18:
19: taskExit(task, NULL);
20: }
21:
22: static void *
23: txPkt(sched_task_t *task)
24: {
25: int wlen;
26: rpack_t *pkt = TASK_DATA(task);
27: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
28:
29: ETRACE();
30:
31: wlen = sendto(TASK_FD(task), RPACK_BUF(pkt), TASK_DATLEN(task),
32: 0, &cli.addr.sa, cli.addr.sa.sa_len);
33: if (wlen == -1)
34: ESYSERR(0);
35: else if (wlen != TASK_DATLEN(task)) {
36: EERROR(EIO, "Sended %d bytes != packet %d bytes",
37: wlen, TASK_DATLEN(task));
1.2 misho 38: schedEvent(TASK_ROOT(task), execProg, "error", 0, NULL, TFTP_OPC_ERROR);
1.1 misho 39: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL,
40: timeoutSession, NULL);
1.2.2.8 ! misho 41: schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, TASK_DATA(task), 0);
1.1 misho 42: } else
43: EVERBOSE(2, "Sended %d bytes", wlen);
44: /* on error or argument, drop session */
45: if (TASK_ARG(task) == (void*) -1 || ntohs(tftp->tftp_opc) == TFTP_OPC_ERROR) {
1.2 misho 46: schedEvent(TASK_ROOT(task), execProg, "error", 0, NULL, TFTP_OPC_ERROR);
1.1 misho 47: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL,
48: timeoutSession, NULL);
1.2.2.8 ! misho 49: schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, TASK_DATA(task), 0);
1.1 misho 50: }
51:
52: taskExit(task, NULL);
53: }
54:
55: static void *
56: txData(sched_task_t *task)
57: {
58: rpack_t *pkt = TASK_DATA(task);
59: u_short code = 0, n = htons(TFTP_OPC_DATA);
60: int len;
61:
62: ETRACE();
63:
64: RPACK_REWIND(pkt);
65: rpack_uint16(pkt, &n, 0);
66: n = htons(cli.seq);
67: rpack_uint16(pkt, &n, 0);
68:
1.2.2.4 misho 69: /* max file size check */
1.2.2.6 misho 70: if (cli.tsiz && cli.tsiz < cli.seq * cli.siz) {
71: len = cli.tsiz - (cli.seq - 1) * cli.siz;
1.2.2.4 misho 72: cli.close = 42; /* last sended packet, should be close! */
73: } else
74: len = cli.siz;
75:
76: len = pread(cli.fd, RPACK_NEXT(pkt), len, (cli.seq - 1) * cli.siz);
1.1 misho 77: if (len == -1) {
78: ESYSERR(0);
79: code = htole16(3);
80: goto end;
81: } else {
82: rpack_rnext(pkt, len);
83: EVERBOSE(3, "Read from file %s %d bytes", cli.file, len);
84: }
85:
86: if (cli.siz - len)
87: cli.close = 42; /* last sended packet, should be close! */
88:
89: schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task),
90: TASK_DATA(task), RPACK_OFF(pkt));
91: taskExit(task, NULL);
92: end:
93: RPACK_REWIND(pkt);
94: n = htons(TFTP_OPC_ERROR);
95: rpack_uint16(pkt, &n, 0);
96: rpack_uint16(pkt, &code, 0);
97: rpack_rdata(pkt, errs[le16toh(code)].err_msg, strlen(errs[le16toh(code)].err_msg) + 1);
98:
99: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
100: TASK_DATA(task), RPACK_OFF(pkt));
101: taskExit(task, NULL);
102: }
103:
104: static void *
105: txAck(sched_task_t *task)
106: {
107: rpack_t *pkt = TASK_DATA(task);
108: u_short n = htons(TFTP_OPC_ACK);
109:
110: ETRACE();
111:
112: RPACK_REWIND(pkt);
113: rpack_uint16(pkt, &n, 0);
114: n = htons(cli.seq);
115: rpack_uint16(pkt, &n, 0);
116:
117: schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task),
118: TASK_DATA(task), RPACK_OFF(pkt));
119:
120: if (cli.close) {
1.2 misho 121: schedEvent(TASK_ROOT(task), execProg, "complete", 0, NULL, TFTP_OPC_WRQ);
1.1 misho 122: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);
1.2.2.8 ! misho 123: schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, TASK_DATA(task), 0);
1.1 misho 124: EVERBOSE(2, "Finish WRQ request");
125: }
126: taskExit(task, NULL);
127: }
128:
1.2.2.1 misho 129: static int
1.2.2.6 misho 130: getOpts(rpack_t * __restrict pkt, int rlen)
1.2.2.1 misho 131: {
1.2.2.2 misho 132: char *opt, *val;
133: int len;
134:
1.2.2.7 misho 135: if (!rlen)
136: return -1;
137:
1.2.2.2 misho 138: do {
139: /* option */
140: len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL);
141: if (len == -1)
142: return -1;
1.2.2.6 misho 143: else
144: rlen -= len;
1.2.2.2 misho 145: opt = (char*) rpack_rnext(pkt, len);
146: if (!opt)
147: return -1;
1.2.2.7 misho 148: EVERBOSE(7, "opt=%s rlen=%d", opt, rlen);
1.2.2.2 misho 149: /* value */
150: len = str_getString(RPACK_NEXT(pkt), RPACK_REMAIN(pkt), NULL);
151: if (len == -1)
152: return -1;
1.2.2.6 misho 153: else
154: rlen -= len;
1.2.2.2 misho 155: val = (char*) rpack_rnext(pkt, len);
156: if (!val)
157: return -1;
1.2.2.7 misho 158: EVERBOSE(7, "val=%s rlen=%d", val, rlen);
1.2.2.2 misho 159:
1.2.2.8 ! misho 160: if (!strcasecmp(opt, TFTP_OPT_BLKSIZE))
1.2.2.7 misho 161: cli.tmp = strtol(val, NULL, 10);
1.2.2.8 ! misho 162: else if (!strcasecmp(opt, TFTP_OPT_TSIZE))
1.2.2.2 misho 163: cli.tsiz = strtoll(val, NULL, 10);
164: else if (!strcasecmp(opt, TFTP_OPT_TIMEOUT))
1.2.2.8 ! misho 165: cli.tout.tv_sec = strtol(val, NULL, 10);
1.2.2.6 misho 166: else if (!strcasecmp(opt, TFTP_OPT_ROLLOVER))
167: cli.roll = strtol(val, NULL, 10);
168: } while (rlen > 0);
1.2.2.2 misho 169:
1.2.2.6 misho 170: EVERBOSE(4, "blksize=%u tsize=%llu timeout=%u rollover=%u",
1.2.2.8 ! misho 171: cli.siz, cli.tsiz, cli.tout.tv_sec, cli.roll);
1.2.2.1 misho 172: return 0;
173: }
174:
175: static void *
176: txOack(sched_task_t *task)
177: {
178: rpack_t *pkt = TASK_DATA(task);
179: u_short n = htons(TFTP_OPC_OACK);
180: struct stat sb;
181: char szStr[STRSIZ];
182:
183: ETRACE();
184:
185: RPACK_REWIND(pkt);
186: rpack_uint16(pkt, &n, 0);
187:
188: /* if opcode is RRQ and tsize is 0 then we must return file size to client */
189: if (cli.opc == TFTP_OPC_RRQ && !cli.tsiz && stat(cli.file, &sb) != -1)
190: cli.tsiz = sb.st_size;
191:
192: if (cli.siz > TFTP_LOAD_MAX) {
193: memset(szStr, 0, sizeof szStr);
194: snprintf(szStr, sizeof szStr, "%u", cli.siz);
195: rpack_rdata(pkt, TFTP_OPT_BLKSIZE, strlen(TFTP_OPT_BLKSIZE) + 1);
196: rpack_rdata(pkt, szStr, strlen(szStr) + 1);
197: }
198: if (cli.tsiz) {
199: memset(szStr, 0, sizeof szStr);
200: snprintf(szStr, sizeof szStr, "%llu", cli.tsiz);
201: rpack_rdata(pkt, TFTP_OPT_TSIZE, strlen(TFTP_OPT_TSIZE) + 1);
202: rpack_rdata(pkt, szStr, strlen(szStr) + 1);
203: }
1.2.2.8 ! misho 204: if (cli.tout.tv_sec) {
1.2.2.1 misho 205: memset(szStr, 0, sizeof szStr);
1.2.2.8 ! misho 206: snprintf(szStr, sizeof szStr, "%u", cli.tout.tv_sec);
1.2.2.1 misho 207: rpack_rdata(pkt, TFTP_OPT_TIMEOUT, strlen(TFTP_OPT_TIMEOUT) + 1);
208: rpack_rdata(pkt, szStr, strlen(szStr) + 1);
209: }
1.2.2.6 misho 210: if (cli.roll) {
211: memset(szStr, 0, sizeof szStr);
212: snprintf(szStr, sizeof szStr, "%u", cli.roll);
213: rpack_rdata(pkt, TFTP_OPT_ROLLOVER, strlen(TFTP_OPT_ROLLOVER) + 1);
214: rpack_rdata(pkt, szStr, strlen(szStr) + 1);
215: }
1.2.2.1 misho 216:
1.2.2.6 misho 217: EVERBOSE(4, "blksize=%u tsize=%llu timeout=%u rollover=%u",
1.2.2.8 ! misho 218: cli.siz, cli.tsiz, cli.tout.tv_sec, cli.roll);
1.2.2.1 misho 219: schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task),
220: TASK_DATA(task), RPACK_OFF(pkt));
221: taskExit(task, NULL);
222: }
223:
1.1 misho 224: static void *
225: RQ(sched_task_t *task)
226: {
227: rpack_t *pkt = TASK_DATA(task);
228: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
229: int len, rlen = TASK_DATLEN(task) - 2;
230: char *str;
231: u_short code = 0;
232:
233: ETRACE();
234:
235: cli.siz = TFTP_LOAD_MAX;
1.2.2.6 misho 236: cli.opc = ntohs(rpack_uint16(pkt, NULL, 0));
1.1 misho 237: len = str_getString(tftp->tftp_data, rlen, &str);
238: if (len == -1)
239: goto end;
240: else {
241: rlen -= len;
1.2.2.6 misho 242: rpack_rnext(pkt, len);
1.1 misho 243: strlcpy(cli.file, (char*) tftp->tftp_data, sizeof cli.file);
244: }
245: len = str_getString((const u_char*) str, rlen, NULL);
246: if (len == -1)
247: goto end;
248: else {
249: rlen -= len;
1.2.2.6 misho 250: rpack_rnext(pkt, len);
1.1 misho 251: if (!strcasecmp(str, TFTP_MODE_ASCII))
252: strlcpy(cli.mode, TFTP_MODE_ASCII, sizeof cli.mode);
253: else if (!strcasecmp(str, TFTP_MODE_OCTET))
254: strlcpy(cli.mode, TFTP_MODE_OCTET, sizeof cli.mode);
255: else if (!strcasecmp(str, TFTP_MODE_MAIL)) {
256: strlcpy(cli.mode, TFTP_MODE_MAIL, sizeof cli.mode);
257: code = htole16(4);
258: goto end;
259: } else {
260: code = htole16(1);
261: goto end;
262: }
263: }
264:
265: switch (cli.opc) {
266: case TFTP_OPC_RRQ:
267: code = O_RDONLY;
268: EVERBOSE(2, "RRQ:: file=%s mode=%s\n", cli.file, cli.mode);
269: break;
270: case TFTP_OPC_WRQ:
271: code = O_WRONLY | O_CREAT;
272: str = (char*) cfg_getAttribute(&cfg, "tftpd", "override");
273: if (!str || tolower(*str) != 'y')
274: code |= O_EXCL;
275: EVERBOSE(2, "WRQ:: file=%s mode=%s\n", cli.file, cli.mode);
276: break;
277: }
1.2.2.1 misho 278:
1.1 misho 279: cli.fd = open(cli.file, code, 0644);
280: if (cli.fd == -1) {
281: if (errno == EACCES)
282: code = htole16(2);
283: else if (errno == ENFILE)
284: code = htole16(3);
285: else if (errno == EEXIST)
286: code = htole16(6);
287: else
288: code = htole16(0);
289: ESYSERR(0);
290: goto end;
291: } else
292: cli.seq = 0;
293:
1.2 misho 294: schedEvent(TASK_ROOT(task), execProg, "request", 0, NULL, cli.opc);
295:
1.2.2.7 misho 296: if (!getOpts(pkt, rlen)) {
297: if (cli.tmp > TFTP_LOAD_MAX) {
298: if (rpack_resize(pkt, cli.tmp + 4))
299: ELIBERR(elwix);
300: else
301: cli.siz = cli.tmp;
302: }
1.2.2.8 ! misho 303: if (cli.tout.tv_sec) {
! 304: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL,
! 305: timeoutSession, NULL);
! 306: schedTimer(TASK_ROOT(task), timeoutSession, NULL,
! 307: cli.tout, TASK_DATA(task), 0);
! 308: }
1.2.2.6 misho 309: schedEvent(TASK_ROOT(task), txOack, NULL, TASK_FD(task),
310: TASK_DATA(task), 0);
1.2.2.7 misho 311: } else if (cli.opc == TFTP_OPC_WRQ) {
1.1 misho 312: /* ack */
313: tftp->tftp_opc = htons(TFTP_OPC_ACK);
314: RPACK_REWIND(pkt);
315: rpack_uint16(pkt, NULL, 0);
316: rpack_uint16(pkt, &cli.seq, 0);
317:
318: schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task),
319: TASK_DATA(task), RPACK_OFF(pkt));
320: } else
321: schedEvent(TASK_ROOT(task), txData, NULL, TASK_FD(task),
322: TASK_DATA(task), 0);
323:
324: cli.seq = 1; /* 1st ack */
325: taskExit(task, NULL);
326: end:
327: tftp->tftp_opc = htons(TFTP_OPC_ERROR);
328: RPACK_REWIND(pkt);
329: rpack_uint16(pkt, NULL, 0);
330: rpack_uint16(pkt, &code, 0);
331: rpack_rdata(pkt, errs[le16toh(code)].err_msg, strlen(errs[le16toh(code)].err_msg) + 1);
332:
333: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
334: TASK_DATA(task), RPACK_OFF(pkt));
335: taskExit(task, NULL);
336: }
337:
338: static void *
339: ACK(sched_task_t *task)
340: {
341: rpack_t *pkt = TASK_DATA(task);
342: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
343: u_short code;
344:
345: ETRACE();
346:
347: RPACK_REWIND(pkt);
348: code = rpack_uint16(pkt, NULL, 0);
349: if (ntohs(code) != TFTP_OPC_ACK) {
350: code = htole16(5);
351: goto end;
352: }
353:
354: code = rpack_uint16(pkt, NULL, 0);
355: if (ntohs(code) > cli.seq || (ntohs(code) < (cli.seq - 1))) {
356: code = htole16(5);
357: goto end;
358: } else if (ntohs(code) == cli.seq)
359: cli.seq++;
360:
361: EVERBOSE(3, "ACK:: seq=%hu; my new seq=%hu;", ntohs(code), cli.seq);
362:
363: if (!cli.close)
364: schedEvent(TASK_ROOT(task), txData, NULL, TASK_FD(task),
365: TASK_DATA(task), 0);
366: else {
1.2 misho 367: schedEvent(TASK_ROOT(task), execProg, "complete", 0, NULL, TFTP_OPC_RRQ);
1.1 misho 368: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);
1.2.2.8 ! misho 369: schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, TASK_DATA(task), 0);
1.1 misho 370: EVERBOSE(2, "Finish RRQ request");
371: }
372: taskExit(task, NULL);
373: end:
374: tftp->tftp_opc = htons(TFTP_OPC_ERROR);
375: RPACK_REWIND(pkt);
376: rpack_uint16(pkt, NULL, 0);
377: rpack_uint16(pkt, &code, 0);
378: rpack_rdata(pkt, errs[le16toh(code)].err_msg, strlen(errs[le16toh(code)].err_msg) + 1);
379:
380: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
381: TASK_DATA(task), RPACK_OFF(pkt));
382: taskExit(task, NULL);
383: }
384:
385: static void *
386: DATA(sched_task_t *task)
387: {
388: rpack_t *pkt = TASK_DATA(task);
389: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
390: u_short code;
391: int len;
392:
393: ETRACE();
394:
395: RPACK_REWIND(pkt);
396: code = rpack_uint16(pkt, NULL, 0);
397: if (ntohs(code) != TFTP_OPC_DATA) {
398: code = htole16(5);
399: goto end;
400: }
401:
402: code = rpack_uint16(pkt, NULL, 0);
403: if (ntohs(code) < cli.seq || ntohs(code) > cli.seq + 1) {
404: code = htole16(5);
405: goto end;
406: } else
407: cli.seq = ntohs(code);
408:
1.2.2.4 misho 409: /* max file size check */
1.1 misho 410: len = TASK_DATLEN(task) - RPACK_OFF(pkt);
1.2.2.6 misho 411: if (cli.tsiz && cli.tsiz < cli.seq * cli.siz)
1.2.2.4 misho 412: len = MIN(len, cli.tsiz - (cli.seq - 1) * cli.siz);
1.1 misho 413: if (len < cli.siz)
414: cli.close = 42; /* last received packet, should be close! */
415:
416: EVERBOSE(3, "DATA:: seq=%hu; len=%d", cli.seq, len);
417:
418: len = pwrite(cli.fd, RPACK_NEXT(pkt), len, (cli.seq - 1) * cli.siz);
419: if (len == -1) {
420: ESYSERR(0);
421: code = htole16(3);
422: goto end;
423: } else {
424: rpack_rnext(pkt, len);
425: EVERBOSE(3, "Written to file %s %d bytes", cli.file, len);
426: }
427:
428: schedEvent(TASK_ROOT(task), txAck, NULL, TASK_FD(task), TASK_DATA(task), 0);
429: taskExit(task, NULL);
430: end:
431: tftp->tftp_opc = htons(TFTP_OPC_ERROR);
432: RPACK_REWIND(pkt);
433: rpack_uint16(pkt, NULL, 0);
434: rpack_uint16(pkt, &code, 0);
435: rpack_rdata(pkt, errs[le16toh(code)].err_msg, strlen(errs[le16toh(code)].err_msg) + 1);
436:
437: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
438: TASK_DATA(task), RPACK_OFF(pkt));
439: taskExit(task, NULL);
440: }
441:
442: void *
443: rxPkt(sched_task_t *task)
444: {
445: sockaddr_t sa;
446: socklen_t salen = sizeof sa;
447: int rlen;
448: rpack_t *pkt = TASK_DATA(task);
449: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
450: u_short code;
451:
452: ETRACE();
453:
454: memset(RPACK_BUF(pkt), 0, RPACK_LEN(pkt));
455: rlen = recvfrom(TASK_FD(task), RPACK_BUF(pkt), RPACK_LEN(pkt), 0, &sa.sa, &salen);
456: if (rlen == -1) {
457: ESYSERR(0);
458: goto end;
459: } else if (!cli.addr.sa.sa_len) {
460: cli.addr = sa;
461: switch (ntohs(tftp->tftp_opc)) {
462: case TFTP_OPC_RRQ:
463: case TFTP_OPC_WRQ:
464: schedEvent(TASK_ROOT(task), RQ, NULL, TASK_FD(task),
465: TASK_DATA(task), rlen);
466: break;
467: case TFTP_OPC_ERROR:
468: default:
469: RPACK_REWIND(pkt);
470: code = htons(TFTP_OPC_ERROR);
471: rpack_uint16(pkt, &code, 0);
472: code = htole16(4);
473: rpack_uint16(pkt, &code, 0);
474: rpack_rdata(pkt, errs[4].err_msg, strlen(errs[4].err_msg) + 1);
475:
476: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
477: TASK_DATA(task), RPACK_OFF(pkt));
478: goto end;
479: }
480: } else if (memcmp(&cli.addr, &sa, salen)) {
481: EERROR(LOG_WARNING, "Packet dropped!!!\n"
482: "Get frame from different address for this session");
483: goto end;
484: } else
485: switch (ntohs(tftp->tftp_opc)) {
486: case TFTP_OPC_ACK:
487: schedEvent(TASK_ROOT(task), ACK, NULL, TASK_FD(task),
488: TASK_DATA(task), rlen);
489: break;
490: case TFTP_OPC_DATA:
491: schedEvent(TASK_ROOT(task), DATA, NULL, TASK_FD(task),
492: TASK_DATA(task), rlen);
493: break;
494: case TFTP_OPC_OACK:
495: case TFTP_OPC_ERROR:
496: default:
497: RPACK_REWIND(pkt);
498: code = htons(TFTP_OPC_ERROR);
499: rpack_uint16(pkt, &code, 0);
500: code = htole16(4);
501: rpack_uint16(pkt, &code, 0);
502: rpack_rdata(pkt, errs[4].err_msg, strlen(errs[4].err_msg) + 1);
503:
504: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
505: TASK_DATA(task), RPACK_OFF(pkt));
506: goto end;
507: }
508:
509: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);
1.2.2.8 ! misho 510: schedTimer(TASK_ROOT(task), timeoutSession, NULL,
! 511: cli.tout.tv_sec ? cli.tout : timeout, TASK_DATA(task), 0);
1.1 misho 512: end:
513: schedReadSelf(task);
514: taskExit(task, NULL);
515: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>