Andrew Boyson / net

A stack which works with or without an Mbed os library. Provides IPv4 or IPv6 with a full 1500 byte buffer.

Dependents:   oldheating gps motorhome heating

udp/tftp/tftp.c@128:79052cb4a41c, 2019-03-11 (annotated)

Committer:
andrewboyson
Date:
Mon Mar 11 16:42:45 2019 +0000
Revision:
128:79052cb4a41c
Parent:
116:60521b29e4c9
Child:
132:db2174b36a6d
Tidied up the DNS label module and removed some declarations that had not left room for the terminating null.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
andrewboyson 115:5c003909bcf3 1 #include <stdint.h>
andrewboyson 115:5c003909bcf3 2 #include <stdbool.h>
andrewboyson 115:5c003909bcf3 3
andrewboyson 115:5c003909bcf3 4 #include "log.h"
andrewboyson 115:5c003909bcf3 5 #include "net.h"
andrewboyson 115:5c003909bcf3 6 #include "action.h"
andrewboyson 115:5c003909bcf3 7 #include "udp.h"
andrewboyson 115:5c003909bcf3 8 #include "dns.h"
andrewboyson 115:5c003909bcf3 9 #include "tftp.h"
andrewboyson 115:5c003909bcf3 10 #include "mstimer.h"
andrewboyson 116:60521b29e4c9 11 #include "resolve.h"
andrewboyson 128:79052cb4a41c 12 #include "dnslabel.h"
andrewboyson 115:5c003909bcf3 13
andrewboyson 115:5c003909bcf3 14 bool TftpTrace = false;
andrewboyson 115:5c003909bcf3 15
andrewboyson 116:60521b29e4c9 16 #define WRITE_TIMEOUT_MS 6000 //Make this longer than the resolve time which is up to 3 seconds
andrewboyson 115:5c003909bcf3 17
andrewboyson 115:5c003909bcf3 18 #define TFTP_RRQ 1
andrewboyson 115:5c003909bcf3 19 #define TFTP_WRQ 2
andrewboyson 115:5c003909bcf3 20 #define TFTP_DATA 3
andrewboyson 115:5c003909bcf3 21 #define TFTP_ACK 4
andrewboyson 115:5c003909bcf3 22 #define TFTP_ERROR 5
andrewboyson 115:5c003909bcf3 23
andrewboyson 115:5c003909bcf3 24 static void logOp(char* p)
andrewboyson 115:5c003909bcf3 25 {
andrewboyson 115:5c003909bcf3 26 if (*p)
andrewboyson 115:5c003909bcf3 27 {
andrewboyson 115:5c003909bcf3 28 LogF("Unknown op code %02x%02x", *p, *++p);
andrewboyson 115:5c003909bcf3 29 return;
andrewboyson 115:5c003909bcf3 30 }
andrewboyson 115:5c003909bcf3 31 p++;
andrewboyson 115:5c003909bcf3 32 switch (*p)
andrewboyson 115:5c003909bcf3 33 {
andrewboyson 115:5c003909bcf3 34 case TFTP_RRQ: Log ("RRQ") ; break;
andrewboyson 115:5c003909bcf3 35 case TFTP_WRQ: Log ("WRQ") ; break;
andrewboyson 115:5c003909bcf3 36 case TFTP_DATA: Log ("DATA") ; break;
andrewboyson 115:5c003909bcf3 37 case TFTP_ACK: Log ("ACK") ; break;
andrewboyson 115:5c003909bcf3 38 case TFTP_ERROR: Log ("ERROR") ; break;
andrewboyson 115:5c003909bcf3 39 default: LogF("Unknown op code 00%02x", *p); break;
andrewboyson 115:5c003909bcf3 40 }
andrewboyson 115:5c003909bcf3 41 }
andrewboyson 115:5c003909bcf3 42
andrewboyson 115:5c003909bcf3 43 static void logError(char* p)
andrewboyson 115:5c003909bcf3 44 {
andrewboyson 115:5c003909bcf3 45 if (*p)
andrewboyson 115:5c003909bcf3 46 {
andrewboyson 115:5c003909bcf3 47 LogF("Unknown error code %02x%02x", *p, *++p);
andrewboyson 115:5c003909bcf3 48 return;
andrewboyson 115:5c003909bcf3 49 }
andrewboyson 115:5c003909bcf3 50 p++;
andrewboyson 115:5c003909bcf3 51 switch (*p)
andrewboyson 115:5c003909bcf3 52 {
andrewboyson 115:5c003909bcf3 53 case 0: Log ("Not defined, see error message." ); break;
andrewboyson 115:5c003909bcf3 54 case 1: Log ("File not found." ); break;
andrewboyson 115:5c003909bcf3 55 case 2: Log ("Access violation." ); break;
andrewboyson 115:5c003909bcf3 56 case 3: Log ("Disk full or allocation exceeded."); break;
andrewboyson 115:5c003909bcf3 57 case 4: Log ("Illegal TFTP operation." ); break;
andrewboyson 115:5c003909bcf3 58 case 5: Log ("Unknown transfer ID." ); break;
andrewboyson 115:5c003909bcf3 59 case 6: Log ("File already exists." ); break;
andrewboyson 115:5c003909bcf3 60 case 7: Log ("No such user." ); break;
andrewboyson 115:5c003909bcf3 61 default: LogF("Unknown error code 00%02x", *p ); break;
andrewboyson 115:5c003909bcf3 62 }
andrewboyson 115:5c003909bcf3 63 }
andrewboyson 115:5c003909bcf3 64 bool TftpSendRequestsViaIp4 = false;
andrewboyson 115:5c003909bcf3 65 uint32_t TftpServerIp4;
andrewboyson 115:5c003909bcf3 66 char TftpServerIp6[16];
andrewboyson 115:5c003909bcf3 67 int TftpWriteStatus = TFTP_WRITE_STATUS_NONE;
andrewboyson 115:5c003909bcf3 68 char TftpServerName[DNS_MAX_LABEL_LENGTH+1];
andrewboyson 115:5c003909bcf3 69 char TftpFileName [DNS_MAX_LABEL_LENGTH+1];
andrewboyson 115:5c003909bcf3 70
andrewboyson 115:5c003909bcf3 71 int (*TftpGetNextByteFunction)();
andrewboyson 115:5c003909bcf3 72
andrewboyson 115:5c003909bcf3 73 static void logHeader(int size, char* p)
andrewboyson 115:5c003909bcf3 74 {
andrewboyson 115:5c003909bcf3 75 if (NetTraceVerbose)
andrewboyson 115:5c003909bcf3 76 {
andrewboyson 115:5c003909bcf3 77 Log ("TFTP header\r\n");
andrewboyson 115:5c003909bcf3 78 Log (" Op code "); logOp(p); Log("\r\n");
andrewboyson 115:5c003909bcf3 79 Log (" Size "); LogF("%d", size); Log("\r\n");
andrewboyson 115:5c003909bcf3 80 }
andrewboyson 115:5c003909bcf3 81 else
andrewboyson 115:5c003909bcf3 82 {
andrewboyson 115:5c003909bcf3 83 Log ("TFTP header");
andrewboyson 115:5c003909bcf3 84 Log (": Op "); logOp(p);
andrewboyson 115:5c003909bcf3 85 LogF(", %d bytes", size);
andrewboyson 115:5c003909bcf3 86 Log ("\r\n");
andrewboyson 115:5c003909bcf3 87 }
andrewboyson 115:5c003909bcf3 88 }
andrewboyson 115:5c003909bcf3 89 static int sendBlock(uint16_t block, int* pSize, char* pHeader)
andrewboyson 115:5c003909bcf3 90 {
andrewboyson 115:5c003909bcf3 91 /*2 bytes 2 bytes n bytes
andrewboyson 115:5c003909bcf3 92 ----------------------------------
andrewboyson 115:5c003909bcf3 93 | Opcode | Block # | Data |
andrewboyson 115:5c003909bcf3 94 ---------------------------------- */
andrewboyson 115:5c003909bcf3 95 char* p = pHeader;
andrewboyson 115:5c003909bcf3 96 *p++ = 0;
andrewboyson 115:5c003909bcf3 97 *p++ = TFTP_DATA;
andrewboyson 115:5c003909bcf3 98 *p++ = block >> 8;
andrewboyson 115:5c003909bcf3 99 *p++ = block & 0xFF;
andrewboyson 115:5c003909bcf3 100
andrewboyson 115:5c003909bcf3 101 int len = 0;
andrewboyson 115:5c003909bcf3 102 while (len < 512)
andrewboyson 115:5c003909bcf3 103 {
andrewboyson 115:5c003909bcf3 104 int c = TftpGetNextByteFunction();
andrewboyson 115:5c003909bcf3 105 if (c == -1) break;
andrewboyson 115:5c003909bcf3 106 *p++ = c;
andrewboyson 115:5c003909bcf3 107 len++;
andrewboyson 115:5c003909bcf3 108 }
andrewboyson 115:5c003909bcf3 109 if (len < 512) TftpWriteStatus = TFTP_WRITE_STATUS_NONE;
andrewboyson 115:5c003909bcf3 110 *pSize = p - pHeader;
andrewboyson 115:5c003909bcf3 111 return UNICAST_TFTP;
andrewboyson 115:5c003909bcf3 112 }
andrewboyson 115:5c003909bcf3 113 static void handleError(char* p)
andrewboyson 115:5c003909bcf3 114 {
andrewboyson 115:5c003909bcf3 115 /*2 bytes 2 bytes string 1 byte
andrewboyson 115:5c003909bcf3 116 -----------------------------------------
andrewboyson 115:5c003909bcf3 117 | ERROR | ErrorCode | ErrMsg | 0 |
andrewboyson 115:5c003909bcf3 118 ----------------------------------------- */
andrewboyson 115:5c003909bcf3 119 p += 2; //Skip the op code which we already know
andrewboyson 115:5c003909bcf3 120 if (NetTraceNewLine) Log("\r\n");
andrewboyson 115:5c003909bcf3 121 LogTime("TFTP error - ");
andrewboyson 115:5c003909bcf3 122 logError(p); p += 2;
andrewboyson 115:5c003909bcf3 123 Log(p);
andrewboyson 115:5c003909bcf3 124 Log("\r\n");
andrewboyson 115:5c003909bcf3 125 TftpWriteStatus = TFTP_WRITE_STATUS_NONE;
andrewboyson 115:5c003909bcf3 126 }
andrewboyson 115:5c003909bcf3 127 static int handleAck(char* pHeaderRx, int* pSizeTx, char* pHeaderTx)
andrewboyson 115:5c003909bcf3 128 {
andrewboyson 115:5c003909bcf3 129 /* 2 bytes 2 bytes
andrewboyson 115:5c003909bcf3 130 -----------------------
andrewboyson 115:5c003909bcf3 131 | ACK | Block # |
andrewboyson 115:5c003909bcf3 132 ----------------------- */
andrewboyson 115:5c003909bcf3 133 char* p = pHeaderRx;
andrewboyson 115:5c003909bcf3 134 p += 2; //Skip the op code which we already know
andrewboyson 115:5c003909bcf3 135 uint16_t block = *p++;
andrewboyson 115:5c003909bcf3 136 block <<= 8;
andrewboyson 115:5c003909bcf3 137 block += *p++;
andrewboyson 115:5c003909bcf3 138
andrewboyson 115:5c003909bcf3 139 return sendBlock(block + 1, pSizeTx, pHeaderTx);
andrewboyson 115:5c003909bcf3 140 }
andrewboyson 115:5c003909bcf3 141 static int sendRequest(int* pSize, char* pHeader)
andrewboyson 115:5c003909bcf3 142 {
andrewboyson 115:5c003909bcf3 143 /*2 bytes string 1 byte string 1 byte
andrewboyson 115:5c003909bcf3 144 -----------------------------------------------
andrewboyson 115:5c003909bcf3 145 | WRQ | Filename | 0 | Mode | 0 |
andrewboyson 115:5c003909bcf3 146 ----------------------------------------------- */
andrewboyson 115:5c003909bcf3 147 char* p = pHeader;
andrewboyson 115:5c003909bcf3 148 *p++ = 0;
andrewboyson 115:5c003909bcf3 149 *p++ = TFTP_WRQ;
andrewboyson 115:5c003909bcf3 150 char* pName = TftpFileName;
andrewboyson 115:5c003909bcf3 151 while (*pName) *p++ = *pName++;
andrewboyson 115:5c003909bcf3 152 *p++ = 0;
andrewboyson 115:5c003909bcf3 153 const char* pMode = "octet";
andrewboyson 115:5c003909bcf3 154 while (*pMode) *p++ = *pMode++;
andrewboyson 115:5c003909bcf3 155 *p++ = 0;
andrewboyson 115:5c003909bcf3 156 *pSize = p - pHeader;
andrewboyson 115:5c003909bcf3 157 return UNICAST_TFTP;
andrewboyson 115:5c003909bcf3 158 }
andrewboyson 115:5c003909bcf3 159 int TftpHandlePacketReceived(void (*traceback)(void), int sizeRx, void * pPacketRx, int* pSizeTx, void* pPacketTx)
andrewboyson 115:5c003909bcf3 160 {
andrewboyson 115:5c003909bcf3 161 char* pHeaderRx = (char*)pPacketRx;
andrewboyson 115:5c003909bcf3 162 char* pHeaderTx = (char*)pPacketTx;
andrewboyson 115:5c003909bcf3 163
andrewboyson 115:5c003909bcf3 164 char* p = pHeaderRx;
andrewboyson 115:5c003909bcf3 165
andrewboyson 115:5c003909bcf3 166 if (*p)
andrewboyson 115:5c003909bcf3 167 {
andrewboyson 115:5c003909bcf3 168 LogTimeF("Expected high byte of op code to be zero not %u\r\n", *p);
andrewboyson 115:5c003909bcf3 169 return DO_NOTHING;
andrewboyson 115:5c003909bcf3 170 }
andrewboyson 115:5c003909bcf3 171 if (TftpTrace)
andrewboyson 115:5c003909bcf3 172 {
andrewboyson 115:5c003909bcf3 173 if (NetTraceNewLine) Log("\r\n");
andrewboyson 115:5c003909bcf3 174 LogTimeF("TFTP received packet\r\n");
andrewboyson 115:5c003909bcf3 175 if (NetTraceStack) traceback();
andrewboyson 115:5c003909bcf3 176 logHeader(sizeRx, pHeaderRx);
andrewboyson 115:5c003909bcf3 177 }
andrewboyson 115:5c003909bcf3 178 p++;
andrewboyson 115:5c003909bcf3 179 int dest = DO_NOTHING;
andrewboyson 115:5c003909bcf3 180 switch (*p)
andrewboyson 115:5c003909bcf3 181 {
andrewboyson 115:5c003909bcf3 182 case TFTP_ACK:
andrewboyson 115:5c003909bcf3 183 if (TftpWriteStatus == TFTP_WRITE_STATUS_NONE) return DO_NOTHING;
andrewboyson 115:5c003909bcf3 184 dest = handleAck(pHeaderRx, pSizeTx, pHeaderTx);
andrewboyson 115:5c003909bcf3 185 break;
andrewboyson 115:5c003909bcf3 186 case TFTP_ERROR:
andrewboyson 115:5c003909bcf3 187 handleError(pHeaderRx);
andrewboyson 115:5c003909bcf3 188 return DO_NOTHING;
andrewboyson 115:5c003909bcf3 189 default:
andrewboyson 115:5c003909bcf3 190 LogTimeF("\r\nTFTP packet unknown mode %d\r\n", *p);
andrewboyson 115:5c003909bcf3 191 return DO_NOTHING;
andrewboyson 115:5c003909bcf3 192 }
andrewboyson 115:5c003909bcf3 193
andrewboyson 115:5c003909bcf3 194 if (TftpTrace) logHeader(*pSizeTx, pHeaderTx);
andrewboyson 115:5c003909bcf3 195
andrewboyson 115:5c003909bcf3 196 return ActionMakeFromDestAndTrace(dest, TftpTrace && NetTraceStack);
andrewboyson 115:5c003909bcf3 197 }
andrewboyson 115:5c003909bcf3 198 static bool isTimedOut()
andrewboyson 115:5c003909bcf3 199 {
andrewboyson 115:5c003909bcf3 200 static uint32_t writeStartMs = 0;
andrewboyson 115:5c003909bcf3 201
andrewboyson 115:5c003909bcf3 202 if (TftpWriteStatus == TFTP_WRITE_STATUS_NONE) writeStartMs = MsTimerCount;
andrewboyson 115:5c003909bcf3 203
andrewboyson 115:5c003909bcf3 204 if (MsTimerHasElapsed(writeStartMs, WRITE_TIMEOUT_MS))
andrewboyson 115:5c003909bcf3 205 {
andrewboyson 115:5c003909bcf3 206 LogTime("TFTP - write operation timed out so reset\r\n");
andrewboyson 115:5c003909bcf3 207 writeStartMs = MsTimerCount;
andrewboyson 115:5c003909bcf3 208 return true;
andrewboyson 115:5c003909bcf3 209 }
andrewboyson 115:5c003909bcf3 210 return false;
andrewboyson 115:5c003909bcf3 211 }
andrewboyson 115:5c003909bcf3 212 static bool isOkToGo()
andrewboyson 115:5c003909bcf3 213 {
andrewboyson 115:5c003909bcf3 214 if (!TftpServerName[0])
andrewboyson 115:5c003909bcf3 215 {
andrewboyson 115:5c003909bcf3 216 LogTimeF("TftpPollForRequestToSend - A request to send a client message has been made but no server name has been specified\r\n");
andrewboyson 115:5c003909bcf3 217 return false;
andrewboyson 115:5c003909bcf3 218 }
andrewboyson 115:5c003909bcf3 219
andrewboyson 115:5c003909bcf3 220 if (!TftpGetNextByteFunction)
andrewboyson 115:5c003909bcf3 221 {
andrewboyson 115:5c003909bcf3 222 LogTimeF("TftpPollForRequestToSend - A request to send a client message has been made but TFTP has not been plumbed into a file stream\r\n");
andrewboyson 115:5c003909bcf3 223 return false;
andrewboyson 115:5c003909bcf3 224 }
andrewboyson 115:5c003909bcf3 225 return true;
andrewboyson 115:5c003909bcf3 226 }
andrewboyson 115:5c003909bcf3 227 int TftpPollForPacketToSend(int type, void* pPacket, int* pSize)
andrewboyson 115:5c003909bcf3 228 {
andrewboyson 116:60521b29e4c9 229 if (isTimedOut() ) { TftpWriteStatus = TFTP_WRITE_STATUS_NONE; return DO_NOTHING; }
andrewboyson 116:60521b29e4c9 230 if (TftpWriteStatus != TFTP_WRITE_STATUS_REQUEST ) { return DO_NOTHING; }
andrewboyson 116:60521b29e4c9 231 if (!isOkToGo() ) { TftpWriteStatus = TFTP_WRITE_STATUS_NONE; return DO_NOTHING; }
andrewboyson 116:60521b29e4c9 232 if (!Resolve(TftpServerName, type, &TftpServerIp4, TftpServerIp6)) { return DO_NOTHING; }
andrewboyson 115:5c003909bcf3 233
andrewboyson 115:5c003909bcf3 234 //Have IP and MAC so send request
andrewboyson 115:5c003909bcf3 235 TftpWriteStatus = TFTP_WRITE_STATUS_IN_PROGRESS;
andrewboyson 115:5c003909bcf3 236 int dest = sendRequest(pSize, (char*)pPacket);
andrewboyson 115:5c003909bcf3 237 if (TftpTrace)
andrewboyson 115:5c003909bcf3 238 {
andrewboyson 115:5c003909bcf3 239 if (NetTraceNewLine) Log("\r\n");
andrewboyson 115:5c003909bcf3 240 LogTimeF("TFTP Sending request\r\n");
andrewboyson 115:5c003909bcf3 241 logHeader(*pSize, (char*)pPacket);
andrewboyson 115:5c003909bcf3 242 }
andrewboyson 115:5c003909bcf3 243 return ActionMakeFromDestAndTrace(dest, TftpTrace && NetTraceStack);
andrewboyson 115:5c003909bcf3 244 }