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