diff --git a/fs/tftp.c b/fs/tftp.c index cc30c5e..025edbf 100644 --- a/fs/tftp.c +++ b/fs/tftp.c @@ -16,6 +16,9 @@ * GNU General Public License for more details. * */ + +#define pr_fmt(fmt) "tftp: " fmt + #include #include #include @@ -54,6 +57,7 @@ #define TFTP_ERROR 5 #define TFTP_OACK 6 +#define STATE_INVALID 0 #define STATE_RRQ 1 #define STATE_WRQ 2 #define STATE_RDATA 3 @@ -94,6 +98,18 @@ return 0; } +static char *tftp_states[] = { + [STATE_INVALID] = "INVALID", + [STATE_RRQ] = "RRQ", + [STATE_WRQ] = "WRQ", + [STATE_RDATA] = "RDATA", + [STATE_WDATA] = "WDATA", + [STATE_OACK] = "OACK", + [STATE_WAITACK] = "WAITACK", + [STATE_LAST] = "LAST", + [STATE_DONE] = "DONE", +}; + static int tftp_send(struct file_priv *priv) { unsigned char *xp; @@ -102,7 +118,7 @@ unsigned char *pkt = net_udp_get_payload(priv->tftp_con); int ret; - debug("%s: state %d\n", __func__, priv->state); + pr_vdebug("%s: state %s\n", __func__, tftp_states[priv->state]); switch (priv->state) { case STATE_RRQ: @@ -206,7 +222,7 @@ pkt[len - 1] = 0; - debug("got OACK\n"); + pr_debug("got OACK\n"); #ifdef DEBUG memory_display(pkt, 0, len, 1, 0); #endif @@ -222,7 +238,7 @@ priv->filesize = simple_strtoul(val, NULL, 10); if (!strcmp(opt, "blksize")) priv->blocksize = simple_strtoul(val, NULL, 10); - debug("OACK opt: %s val: %s\n", opt, val); + pr_debug("OACK opt: %s val: %s\n", opt, val); s = val + strlen(val) + 1; } } @@ -247,7 +263,7 @@ len -= 2; pkt += 2; - debug("%s: opcode 0x%04x\n", __func__, opcode); + pr_vdebug("%s: opcode 0x%04x\n", __func__, opcode); switch (opcode) { case TFTP_RRQ: @@ -260,7 +276,7 @@ priv->block = ntohs(*(uint16_t *)pkt); if (priv->block != priv->last_block) { - debug("ack %d != %d\n", priv->block, priv->last_block); + pr_vdebug("ack %d != %d\n", priv->block, priv->last_block); break; } @@ -303,7 +319,7 @@ priv->last_block = 0; if (priv->block != 1) { /* Assertion */ - printf("error: First block is not block 1 (%d)\n", + pr_err("error: First block is not block 1 (%d)\n", priv->block); priv->err = -EINVAL; priv->state = STATE_DONE; @@ -330,8 +346,7 @@ break; case TFTP_ERROR: - debug("\nTFTP error: '%s' (%d)\n", - pkt + 2, ntohs(*(uint16_t *)pkt)); + pr_err("error: '%s' (%d)\n", pkt + 2, ntohs(*(uint16_t *)pkt)); switch (ntohs(*(uint16_t *)pkt)) { case 1: priv->err = -ENOENT; @@ -512,7 +527,7 @@ size_t size, now; int ret; - debug("%s: %zu\n", __func__, insize); + pr_vdebug("%s: %zu\n", __func__, insize); size = insize; @@ -547,7 +562,7 @@ size_t outsize = 0, now; int ret; - debug("%s %zu\n", __func__, insize); + pr_vdebug("%s %zu\n", __func__, insize); while (insize) { now = kfifo_get(priv->fifo, buf, insize);