Annotation of tftpd/src/srv.c, revision 1.1.1.1.2.1
1.1 misho 1: #include "global.h"
1.1.1.1.2.1! misho 2: #include "exec.h"
! 3: #include "srv.h"
1.1 misho 4:
5:
6: static void *
7: timeoutSession(sched_task_t *task)
8: {
9: ETRACE();
10:
11: /* drop session */
12: if (cli.fd > 2)
13: close(cli.fd);
14: memset(&cli, 0, sizeof cli);
15:
16: taskExit(task, NULL);
17: }
18:
19: static void *
20: txPkt(sched_task_t *task)
21: {
22: int wlen;
23: rpack_t *pkt = TASK_DATA(task);
24: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
25:
26: ETRACE();
27:
28: wlen = sendto(TASK_FD(task), RPACK_BUF(pkt), TASK_DATLEN(task),
29: 0, &cli.addr.sa, cli.addr.sa.sa_len);
30: if (wlen == -1)
31: ESYSERR(0);
32: else if (wlen != TASK_DATLEN(task)) {
33: EERROR(EIO, "Sended %d bytes != packet %d bytes",
34: wlen, TASK_DATLEN(task));
35: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL,
36: timeoutSession, NULL);
37: memset(&cli, 0, sizeof(sockaddr_t));
38: } else
39: EVERBOSE(2, "Sended %d bytes", wlen);
40: /* on error or argument, drop session */
41: if (TASK_ARG(task) == (void*) -1 || ntohs(tftp->tftp_opc) == TFTP_OPC_ERROR) {
42: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL,
43: timeoutSession, NULL);
44: memset(&cli, 0, sizeof cli);
45: }
46:
47: taskExit(task, NULL);
48: }
49:
50: static void *
51: txData(sched_task_t *task)
52: {
53: rpack_t *pkt = TASK_DATA(task);
54: u_short code = 0, n = htons(TFTP_OPC_DATA);
55: int len;
56:
57: ETRACE();
58:
59: RPACK_REWIND(pkt);
60: rpack_uint16(pkt, &n, 0);
61: n = htons(cli.seq);
62: rpack_uint16(pkt, &n, 0);
63:
64: len = pread(cli.fd, RPACK_NEXT(pkt), cli.siz, (cli.seq - 1) * cli.siz);
65: if (len == -1) {
66: ESYSERR(0);
67: code = htole16(3);
68: goto end;
69: } else {
70: rpack_rnext(pkt, len);
71: EVERBOSE(3, "Read from file %s %d bytes", cli.file, len);
72: }
73:
74: if (cli.siz - len)
75: cli.close = 42; /* last sended packet, should be close! */
76:
77: schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task),
78: TASK_DATA(task), RPACK_OFF(pkt));
79: taskExit(task, NULL);
80: end:
81: RPACK_REWIND(pkt);
82: n = htons(TFTP_OPC_ERROR);
83: rpack_uint16(pkt, &n, 0);
84: rpack_uint16(pkt, &code, 0);
85: rpack_rdata(pkt, errs[le16toh(code)].err_msg, strlen(errs[le16toh(code)].err_msg) + 1);
86:
87: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
88: TASK_DATA(task), RPACK_OFF(pkt));
89: taskExit(task, NULL);
90: }
91:
92: static void *
93: txAck(sched_task_t *task)
94: {
95: rpack_t *pkt = TASK_DATA(task);
96: u_short n = htons(TFTP_OPC_ACK);
97:
98: ETRACE();
99:
100: RPACK_REWIND(pkt);
101: rpack_uint16(pkt, &n, 0);
102: n = htons(cli.seq);
103: rpack_uint16(pkt, &n, 0);
104:
105: schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task),
106: TASK_DATA(task), RPACK_OFF(pkt));
107:
108: if (cli.close) {
109: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);
110: schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, NULL, 0);
111: EVERBOSE(2, "Finish WRQ request");
112: }
113: taskExit(task, NULL);
114: }
115:
116: static void *
117: RQ(sched_task_t *task)
118: {
119: rpack_t *pkt = TASK_DATA(task);
120: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
121: int len, rlen = TASK_DATLEN(task) - 2;
122: char *str;
123: u_short code = 0;
124:
125: ETRACE();
126:
127: cli.siz = TFTP_LOAD_MAX;
128: len = str_getString(tftp->tftp_data, rlen, &str);
129: if (len == -1)
130: goto end;
131: else {
132: rlen -= len;
133: strlcpy(cli.file, (char*) tftp->tftp_data, sizeof cli.file);
134: }
135: len = str_getString((const u_char*) str, rlen, NULL);
136: if (len == -1)
137: goto end;
138: else {
139: rlen -= len;
140: if (!strcasecmp(str, TFTP_MODE_ASCII))
141: strlcpy(cli.mode, TFTP_MODE_ASCII, sizeof cli.mode);
142: else if (!strcasecmp(str, TFTP_MODE_OCTET))
143: strlcpy(cli.mode, TFTP_MODE_OCTET, sizeof cli.mode);
144: else if (!strcasecmp(str, TFTP_MODE_MAIL)) {
145: strlcpy(cli.mode, TFTP_MODE_MAIL, sizeof cli.mode);
146: code = htole16(4);
147: goto end;
148: } else {
149: code = htole16(1);
150: goto end;
151: }
152: }
153:
154: cli.opc = ntohs(tftp->tftp_opc);
155: switch (cli.opc) {
156: case TFTP_OPC_RRQ:
157: code = O_RDONLY;
158: EVERBOSE(2, "RRQ:: file=%s mode=%s\n", cli.file, cli.mode);
159: break;
160: case TFTP_OPC_WRQ:
161: code = O_WRONLY | O_CREAT;
162: str = (char*) cfg_getAttribute(&cfg, "tftpd", "override");
163: if (!str || tolower(*str) != 'y')
164: code |= O_EXCL;
165: EVERBOSE(2, "WRQ:: file=%s mode=%s\n", cli.file, cli.mode);
166: break;
167: }
168: cli.fd = open(cli.file, code, 0644);
169: if (cli.fd == -1) {
170: if (errno == EACCES)
171: code = htole16(2);
172: else if (errno == ENFILE)
173: code = htole16(3);
174: else if (errno == EEXIST)
175: code = htole16(6);
176: else
177: code = htole16(0);
178: ESYSERR(0);
179: goto end;
180: } else
181: cli.seq = 0;
182:
183: if (cli.opc == TFTP_OPC_WRQ) {
184: /* ack */
185: tftp->tftp_opc = htons(TFTP_OPC_ACK);
186: RPACK_REWIND(pkt);
187: rpack_uint16(pkt, NULL, 0);
188: rpack_uint16(pkt, &cli.seq, 0);
189:
190: schedEvent(TASK_ROOT(task), txPkt, NULL, TASK_FD(task),
191: TASK_DATA(task), RPACK_OFF(pkt));
192: } else
193: schedEvent(TASK_ROOT(task), txData, NULL, TASK_FD(task),
194: TASK_DATA(task), 0);
195:
196: cli.seq = 1; /* 1st ack */
197: taskExit(task, NULL);
198: end:
199: tftp->tftp_opc = htons(TFTP_OPC_ERROR);
200: RPACK_REWIND(pkt);
201: rpack_uint16(pkt, NULL, 0);
202: rpack_uint16(pkt, &code, 0);
203: rpack_rdata(pkt, errs[le16toh(code)].err_msg, strlen(errs[le16toh(code)].err_msg) + 1);
204:
205: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
206: TASK_DATA(task), RPACK_OFF(pkt));
207: taskExit(task, NULL);
208: }
209:
210: static void *
211: ACK(sched_task_t *task)
212: {
213: rpack_t *pkt = TASK_DATA(task);
214: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
215: u_short code;
216:
217: ETRACE();
218:
219: RPACK_REWIND(pkt);
220: code = rpack_uint16(pkt, NULL, 0);
221: if (ntohs(code) != TFTP_OPC_ACK) {
222: code = htole16(5);
223: goto end;
224: }
225:
226: code = rpack_uint16(pkt, NULL, 0);
227: if (ntohs(code) > cli.seq || (ntohs(code) < (cli.seq - 1))) {
228: code = htole16(5);
229: goto end;
230: } else if (ntohs(code) == cli.seq)
231: cli.seq++;
232:
233: EVERBOSE(3, "ACK:: seq=%hu; my new seq=%hu;", ntohs(code), cli.seq);
234:
235: if (!cli.close)
236: schedEvent(TASK_ROOT(task), txData, NULL, TASK_FD(task),
237: TASK_DATA(task), 0);
238: else {
239: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);
240: schedEvent(TASK_ROOT(task), timeoutSession, NULL, 0, NULL, 0);
241: EVERBOSE(2, "Finish RRQ request");
242: }
243: taskExit(task, NULL);
244: end:
245: tftp->tftp_opc = htons(TFTP_OPC_ERROR);
246: RPACK_REWIND(pkt);
247: rpack_uint16(pkt, NULL, 0);
248: rpack_uint16(pkt, &code, 0);
249: rpack_rdata(pkt, errs[le16toh(code)].err_msg, strlen(errs[le16toh(code)].err_msg) + 1);
250:
251: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
252: TASK_DATA(task), RPACK_OFF(pkt));
253: taskExit(task, NULL);
254: }
255:
256: static void *
257: DATA(sched_task_t *task)
258: {
259: rpack_t *pkt = TASK_DATA(task);
260: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
261: u_short code;
262: int len;
263:
264: ETRACE();
265:
266: RPACK_REWIND(pkt);
267: code = rpack_uint16(pkt, NULL, 0);
268: if (ntohs(code) != TFTP_OPC_DATA) {
269: code = htole16(5);
270: goto end;
271: }
272:
273: code = rpack_uint16(pkt, NULL, 0);
274: if (ntohs(code) < cli.seq || ntohs(code) > cli.seq + 1) {
275: code = htole16(5);
276: goto end;
277: } else
278: cli.seq = ntohs(code);
279:
280: len = TASK_DATLEN(task) - RPACK_OFF(pkt);
281: if (len < cli.siz)
282: cli.close = 42; /* last received packet, should be close! */
283:
284: EVERBOSE(3, "DATA:: seq=%hu; len=%d", cli.seq, len);
285:
286: len = pwrite(cli.fd, RPACK_NEXT(pkt), len, (cli.seq - 1) * cli.siz);
287: if (len == -1) {
288: ESYSERR(0);
289: code = htole16(3);
290: goto end;
291: } else {
292: rpack_rnext(pkt, len);
293: EVERBOSE(3, "Written to file %s %d bytes", cli.file, len);
294: }
295:
296: schedEvent(TASK_ROOT(task), txAck, NULL, TASK_FD(task), TASK_DATA(task), 0);
297: taskExit(task, NULL);
298: end:
299: tftp->tftp_opc = htons(TFTP_OPC_ERROR);
300: RPACK_REWIND(pkt);
301: rpack_uint16(pkt, NULL, 0);
302: rpack_uint16(pkt, &code, 0);
303: rpack_rdata(pkt, errs[le16toh(code)].err_msg, strlen(errs[le16toh(code)].err_msg) + 1);
304:
305: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
306: TASK_DATA(task), RPACK_OFF(pkt));
307: taskExit(task, NULL);
308: }
309:
310: void *
311: rxPkt(sched_task_t *task)
312: {
313: sockaddr_t sa;
314: socklen_t salen = sizeof sa;
315: int rlen;
316: rpack_t *pkt = TASK_DATA(task);
317: struct tftp_hdr *tftp = (struct tftp_hdr*) RPACK_BUF(pkt);
318: u_short code;
319:
320: ETRACE();
321:
322: memset(RPACK_BUF(pkt), 0, RPACK_LEN(pkt));
323: rlen = recvfrom(TASK_FD(task), RPACK_BUF(pkt), RPACK_LEN(pkt), 0, &sa.sa, &salen);
324: if (rlen == -1) {
325: ESYSERR(0);
326: goto end;
327: } else if (!cli.addr.sa.sa_len) {
328: cli.addr = sa;
329: switch (ntohs(tftp->tftp_opc)) {
330: case TFTP_OPC_RRQ:
331: case TFTP_OPC_WRQ:
332: schedEvent(TASK_ROOT(task), RQ, NULL, TASK_FD(task),
333: TASK_DATA(task), rlen);
334: break;
335: case TFTP_OPC_ERROR:
336: default:
337: RPACK_REWIND(pkt);
338: code = htons(TFTP_OPC_ERROR);
339: rpack_uint16(pkt, &code, 0);
340: code = htole16(4);
341: rpack_uint16(pkt, &code, 0);
342: rpack_rdata(pkt, errs[4].err_msg, strlen(errs[4].err_msg) + 1);
343:
344: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
345: TASK_DATA(task), RPACK_OFF(pkt));
346: goto end;
347: }
348: } else if (memcmp(&cli.addr, &sa, salen)) {
349: EERROR(LOG_WARNING, "Packet dropped!!!\n"
350: "Get frame from different address for this session");
351: goto end;
352: } else
353: switch (ntohs(tftp->tftp_opc)) {
354: case TFTP_OPC_ACK:
355: schedEvent(TASK_ROOT(task), ACK, NULL, TASK_FD(task),
356: TASK_DATA(task), rlen);
357: break;
358: case TFTP_OPC_DATA:
359: schedEvent(TASK_ROOT(task), DATA, NULL, TASK_FD(task),
360: TASK_DATA(task), rlen);
361: break;
362: case TFTP_OPC_OACK:
363: ELOGGER(LOG_WARNING, "oack");
364: /*
365: schedEvent(TASK_ROOT(task), OACK, NULL, TASK_FD(task),
366: TASK_DATA(task), rlen);
367: */
368: break;
369: case TFTP_OPC_ERROR:
370: default:
371: RPACK_REWIND(pkt);
372: code = htons(TFTP_OPC_ERROR);
373: rpack_uint16(pkt, &code, 0);
374: code = htole16(4);
375: rpack_uint16(pkt, &code, 0);
376: rpack_rdata(pkt, errs[4].err_msg, strlen(errs[4].err_msg) + 1);
377:
378: schedEvent(TASK_ROOT(task), txPkt, (void*) -1, TASK_FD(task),
379: TASK_DATA(task), RPACK_OFF(pkt));
380: goto end;
381: }
382:
383: schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_CALL, timeoutSession, NULL);
384: schedTimer(TASK_ROOT(task), timeoutSession, NULL, timeout, NULL, 0);
385: end:
386: schedReadSelf(task);
387: taskExit(task, NULL);
388: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>