7dc13291 |
1 | /** @file\r |
2 | This is a simple TFTP server application\r |
3 | \r |
f6e5cdd5 |
4 | Copyright (c) 2011, 2012, Intel Corporation\r |
5 | All rights reserved. This program and the accompanying materials\r |
6 | are licensed and made available under the terms and conditions of the BSD License\r |
7 | which accompanies this distribution. The full text of the license may be found at\r |
8 | http://opensource.org/licenses/bsd-license.php\r |
9 | \r |
10 | THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r |
11 | WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r |
12 | \r |
7dc13291 |
13 | **/\r |
14 | \r |
15 | #include <TftpServer.h>\r |
16 | \r |
f6e5cdd5 |
17 | TSDT_TFTP_SERVER mTftpServer; ///< TFTP server's control structure\r |
18 | volatile BOOLEAN mbTftpServerExit; ///< Set TRUE to cause TFTP server to exit\r |
7dc13291 |
19 | \r |
20 | \r |
21 | /**\r |
f6e5cdd5 |
22 | Read file data into a buffer\r |
7dc13291 |
23 | \r |
f6e5cdd5 |
24 | @param [in] pContext Connection context structure address\r |
7dc13291 |
25 | \r |
f6e5cdd5 |
26 | @retval TRUE if a read error occurred\r |
7dc13291 |
27 | \r |
28 | **/\r |
f6e5cdd5 |
29 | BOOLEAN\r |
30 | BufferFill (\r |
31 | IN TSDT_CONNECTION_CONTEXT * pContext\r |
7dc13291 |
32 | )\r |
33 | {\r |
f6e5cdd5 |
34 | BOOLEAN bReadError;\r |
35 | size_t BytesRead;\r |
36 | UINT64 LengthInBytes;\r |
7dc13291 |
37 | \r |
38 | DBG_ENTER ( );\r |
39 | \r |
40 | //\r |
f6e5cdd5 |
41 | // Use break instead of goto\r |
7dc13291 |
42 | //\r |
f6e5cdd5 |
43 | bReadError = FALSE;\r |
7dc13291 |
44 | for ( ; ; ) {\r |
45 | //\r |
f6e5cdd5 |
46 | // Determine if there is any work to do\r |
47 | //\r |
48 | LengthInBytes = DIM ( pContext->FileData ) >> 1;\r |
49 | if (( pContext->ValidBytes > LengthInBytes )\r |
50 | || ( 0 == pContext->BytesRemaining )) {\r |
51 | break;\r |
52 | }\r |
53 | \r |
54 | //\r |
55 | // Determine the number of bytes to read\r |
56 | //\r |
57 | if ( LengthInBytes > pContext->BytesRemaining ) {\r |
58 | LengthInBytes = pContext->BytesRemaining;\r |
59 | }\r |
60 | \r |
61 | //\r |
62 | // Read in the next portion of the file\r |
63 | //\r |
64 | BytesRead = fread ( pContext->pFill,\r |
65 | 1,\r |
66 | (size_t)LengthInBytes,\r |
67 | pContext->File );\r |
68 | if ( -1 == BytesRead ) {\r |
69 | bReadError = TRUE;\r |
7dc13291 |
70 | break;\r |
71 | }\r |
72 | \r |
f6e5cdd5 |
73 | //\r |
74 | // Account for the file data read\r |
75 | //\r |
76 | pContext->BytesRemaining -= BytesRead;\r |
77 | pContext->ValidBytes += BytesRead;\r |
78 | DEBUG (( DEBUG_FILE_BUFFER,\r |
79 | "0x%08x: Buffer filled with %Ld bytes, %Ld bytes ramaining\r\n",\r |
80 | pContext->pFill,\r |
81 | BytesRead,\r |
82 | pContext->BytesRemaining ));\r |
83 | \r |
84 | //\r |
85 | // Set the next buffer location\r |
86 | //\r |
87 | pContext->pFill += BytesRead;\r |
88 | if ( pContext->pEnd <= pContext->pFill ) {\r |
89 | pContext->pFill = &pContext->FileData[ 0 ];\r |
90 | }\r |
91 | \r |
92 | //\r |
93 | // Verify that the end of the buffer is reached\r |
94 | //\r |
95 | ASSERT ( 0 == ( DIM ( pContext->FileData ) & 1 ));\r |
96 | break;\r |
97 | }\r |
98 | \r |
99 | //\r |
100 | // Return the read status\r |
101 | //\r |
102 | DBG_EXIT ( );\r |
103 | return bReadError;\r |
104 | }\r |
105 | \r |
106 | \r |
107 | /**\r |
108 | Add a connection context to the list of connection contexts.\r |
109 | \r |
110 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
111 | @param [in] SocketFd Socket file descriptor\r |
112 | \r |
113 | @retval Context structure address, NULL if allocation fails\r |
114 | \r |
115 | **/\r |
116 | TSDT_CONNECTION_CONTEXT *\r |
117 | ContextAdd (\r |
118 | IN TSDT_TFTP_SERVER * pTftpServer,\r |
119 | IN int SocketFd\r |
120 | )\r |
121 | {\r |
122 | TSDT_CONNECTION_CONTEXT * pContext;\r |
123 | TFTP_PACKET * pEnd;\r |
124 | TFTP_PACKET * pPacket;\r |
125 | \r |
126 | DBG_ENTER ( );\r |
127 | \r |
128 | //\r |
129 | // Allocate a new context\r |
130 | //\r |
131 | pContext = (TSDT_CONNECTION_CONTEXT *)AllocateZeroPool ( sizeof ( *pContext ));\r |
132 | if ( NULL != pContext ) {\r |
7dc13291 |
133 | //\r |
134 | // Initialize the context\r |
135 | //\r |
f6e5cdd5 |
136 | pContext->SocketFd = SocketFd;\r |
7dc13291 |
137 | CopyMem ( &pContext->RemoteAddress,\r |
138 | &pTftpServer->RemoteAddress,\r |
139 | sizeof ( pContext->RemoteAddress ));\r |
f6e5cdd5 |
140 | pContext->BlockSize = 512;\r |
141 | \r |
142 | //\r |
143 | // Buffer management\r |
144 | //\r |
145 | pContext->pFill = &pContext->FileData[ 0 ];\r |
146 | pContext->pEnd = &pContext->FileData[ sizeof ( pContext->FileData )];\r |
147 | pContext->pBuffer = pContext->pFill;\r |
148 | \r |
149 | //\r |
150 | // Window management\r |
151 | //\r |
152 | pContext->MaxTimeout = MultU64x32 ( PcdGet32 ( Tftp_MaxTimeoutInSec ),\r |
153 | 2 * 1000 * 1000 * 1000 );\r |
154 | pContext->Rtt2x = pContext->MaxTimeout;\r |
155 | pContext->WindowSize = MAX_PACKETS;\r |
156 | WindowTimeout ( pContext );\r |
157 | \r |
158 | //\r |
159 | // Place the packets on the free list\r |
160 | //\r |
161 | pPacket = &pContext->Tx[ 0 ];\r |
162 | pEnd = &pPacket[ DIM ( pContext->Tx )];\r |
163 | while ( pEnd > pPacket ) {\r |
164 | PacketFree ( pContext, pPacket );\r |
165 | pPacket += 1;\r |
166 | }\r |
7dc13291 |
167 | \r |
168 | //\r |
169 | // Display the new context\r |
170 | //\r |
f6e5cdd5 |
171 | if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {\r |
172 | DEBUG (( DEBUG_PORT_WORK,\r |
173 | "0x%08x: Context for %d.%d.%d.%d:%d\r\n",\r |
174 | pContext,\r |
175 | (UINT8)pTftpServer->RemoteAddress.v4.sin_addr.s_addr,\r |
176 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ),\r |
177 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ),\r |
178 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ),\r |
179 | htons ( pTftpServer->RemoteAddress.v4.sin_port )));\r |
180 | }\r |
181 | else {\r |
182 | DEBUG (( DEBUG_PORT_WORK,\r |
183 | "0x%08x: Context for [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",\r |
184 | pContext,\r |
185 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],\r |
186 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],\r |
187 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],\r |
188 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],\r |
189 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],\r |
190 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],\r |
191 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],\r |
192 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],\r |
193 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],\r |
194 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],\r |
195 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],\r |
196 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],\r |
197 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],\r |
198 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],\r |
199 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],\r |
200 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],\r |
201 | htons ( pTftpServer->RemoteAddress.v6.sin6_port )));\r |
202 | }\r |
7dc13291 |
203 | \r |
204 | //\r |
205 | // Add the context to the context list\r |
206 | //\r |
207 | pContext->pNext = pTftpServer->pContextList;\r |
208 | pTftpServer->pContextList = pContext;\r |
7dc13291 |
209 | }\r |
210 | \r |
211 | //\r |
212 | // Return the connection context\r |
213 | //\r |
214 | DBG_EXIT_STATUS ( pContext );\r |
215 | return pContext;\r |
216 | }\r |
217 | \r |
218 | \r |
219 | /**\r |
220 | Locate a remote connection context.\r |
221 | \r |
f6e5cdd5 |
222 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
7dc13291 |
223 | @param [in] pIpAddress The start of the remote IP address in network order\r |
7dc13291 |
224 | @param [in] Port The remote port number\r |
225 | \r |
226 | @retval Context structure address, NULL if not found\r |
227 | \r |
228 | **/\r |
229 | TSDT_CONNECTION_CONTEXT *\r |
230 | ContextFind (\r |
231 | IN TSDT_TFTP_SERVER * pTftpServer\r |
232 | )\r |
233 | {\r |
234 | TSDT_CONNECTION_CONTEXT * pContext;\r |
235 | \r |
236 | DBG_ENTER ( );\r |
237 | \r |
238 | //\r |
239 | // Walk the list of connection contexts\r |
240 | //\r |
241 | pContext = pTftpServer->pContextList;\r |
242 | while ( NULL != pContext ) {\r |
243 | //\r |
244 | // Attempt to locate the remote network connection\r |
245 | //\r |
f6e5cdd5 |
246 | if ( 0 == memcmp ( &pTftpServer->RemoteAddress,\r |
247 | &pContext->RemoteAddress,\r |
248 | pTftpServer->RemoteAddress.v6.sin6_len )) {\r |
7dc13291 |
249 | //\r |
250 | // The connection was found\r |
251 | //\r |
252 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
253 | "0x%08x: pContext found\r\n",\r |
254 | pContext ));\r |
255 | break;\r |
256 | }\r |
257 | \r |
258 | //\r |
259 | // Set the next context\r |
260 | //\r |
261 | pContext = pContext->pNext;\r |
262 | }\r |
263 | \r |
264 | //\r |
265 | // Return the connection context structure address\r |
266 | //\r |
267 | DBG_EXIT_HEX ( pContext );\r |
268 | return pContext;\r |
269 | }\r |
270 | \r |
271 | \r |
272 | /**\r |
273 | Remove a context from the list.\r |
274 | \r |
f6e5cdd5 |
275 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
276 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
7dc13291 |
277 | \r |
278 | **/\r |
279 | VOID\r |
280 | ContextRemove (\r |
281 | IN TSDT_TFTP_SERVER * pTftpServer,\r |
282 | IN TSDT_CONNECTION_CONTEXT * pContext\r |
283 | )\r |
284 | {\r |
285 | TSDT_CONNECTION_CONTEXT * pNextContext;\r |
286 | TSDT_CONNECTION_CONTEXT * pPreviousContext;\r |
287 | \r |
288 | DBG_ENTER ( );\r |
289 | \r |
290 | //\r |
291 | // Attempt to locate the context in the list\r |
292 | //\r |
293 | pPreviousContext = NULL;\r |
294 | pNextContext = pTftpServer->pContextList;\r |
295 | while ( NULL != pNextContext ) {\r |
296 | //\r |
297 | // Determine if the context was found\r |
298 | //\r |
299 | if ( pNextContext == pContext ) {\r |
300 | //\r |
301 | // Remove the context from the list\r |
302 | //\r |
303 | if ( NULL == pPreviousContext ) {\r |
304 | pTftpServer->pContextList = pContext->pNext;\r |
305 | }\r |
306 | else {\r |
307 | pPreviousContext->pNext = pContext->pNext;\r |
308 | }\r |
309 | break;\r |
310 | }\r |
311 | \r |
312 | //\r |
313 | // Set the next context\r |
314 | //\r |
315 | pPreviousContext = pNextContext;\r |
316 | pNextContext = pNextContext->pNext;\r |
317 | }\r |
318 | \r |
319 | //\r |
320 | // Determine if the context was found\r |
321 | //\r |
322 | if ( NULL != pContext ) {\r |
323 | //\r |
324 | // Return the resources\r |
325 | //\r |
326 | gBS->FreePool ( pContext );\r |
327 | }\r |
328 | \r |
329 | DBG_EXIT ( );\r |
330 | }\r |
331 | \r |
332 | \r |
333 | /**\r |
f6e5cdd5 |
334 | Queue data packets for transmission\r |
335 | \r |
336 | @param [in] pContext Connection context structure address\r |
7dc13291 |
337 | \r |
f6e5cdd5 |
338 | @retval TRUE if a read error occurred\r |
7dc13291 |
339 | \r |
340 | **/\r |
f6e5cdd5 |
341 | BOOLEAN\r |
342 | PacketFill (\r |
343 | IN TSDT_CONNECTION_CONTEXT * pContext\r |
7dc13291 |
344 | )\r |
345 | {\r |
f6e5cdd5 |
346 | BOOLEAN bReadError;\r |
347 | UINT64 LengthInBytes;\r |
348 | UINT8 * pBuffer;\r |
349 | TFTP_PACKET * pPacket;\r |
7dc13291 |
350 | \r |
351 | DBG_ENTER ( );\r |
352 | \r |
353 | //\r |
f6e5cdd5 |
354 | // Use break instead of goto\r |
7dc13291 |
355 | //\r |
f6e5cdd5 |
356 | bReadError = FALSE;\r |
357 | for ( ; ; ) {\r |
7dc13291 |
358 | //\r |
f6e5cdd5 |
359 | // Fill the buffer if necessary\r |
7dc13291 |
360 | //\r |
f6e5cdd5 |
361 | bReadError = BufferFill ( pContext );\r |
362 | if ( bReadError ) {\r |
7dc13291 |
363 | //\r |
f6e5cdd5 |
364 | // File access mode not supported\r |
7dc13291 |
365 | //\r |
f6e5cdd5 |
366 | DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,\r |
367 | "ERROR - File read failure!\r\n" ));\r |
7dc13291 |
368 | \r |
369 | //\r |
f6e5cdd5 |
370 | // Tell the client of the error\r |
7dc13291 |
371 | //\r |
f6e5cdd5 |
372 | SendError ( pContext,\r |
373 | TFTP_ERROR_SEE_MSG,\r |
374 | (UINT8 *)"Read failure" );\r |
375 | break;\r |
7dc13291 |
376 | }\r |
f6e5cdd5 |
377 | \r |
378 | //\r |
379 | // Determine if any packets can be filled\r |
380 | //\r |
381 | if ( pContext->bEofSent\r |
382 | || ( NULL == pContext->pFreeList )) {\r |
7dc13291 |
383 | //\r |
f6e5cdd5 |
384 | // All of the packets are filled\r |
7dc13291 |
385 | //\r |
f6e5cdd5 |
386 | break;\r |
7dc13291 |
387 | }\r |
7dc13291 |
388 | \r |
7dc13291 |
389 | //\r |
f6e5cdd5 |
390 | // Set the TFTP opcode and block number\r |
391 | //\r |
392 | pPacket = PacketGet ( pContext );\r |
393 | pBuffer = &pPacket->TxBuffer[ 0 ];\r |
394 | *pBuffer++ = 0;\r |
395 | *pBuffer++ = TFTP_OP_DATA;\r |
396 | *pBuffer++ = (UINT8)( pContext->BlockNumber >> 8 );\r |
397 | *pBuffer++ = (UINT8)pContext->BlockNumber;\r |
398 | \r |
399 | //\r |
400 | // Determine how much data needs to be sent\r |
401 | //\r |
402 | LengthInBytes = pContext->BlockSize;\r |
403 | if (( pContext->BytesToSend < TFTP_MAX_BLOCK_SIZE )\r |
404 | && ( LengthInBytes > pContext->BytesToSend )) {\r |
405 | LengthInBytes = pContext->BytesToSend;\r |
406 | pContext->bEofSent = TRUE;\r |
407 | }\r |
408 | DEBUG (( DEBUG_TX_PACKET,\r |
409 | "0x%08x: Packet, Block %d filled with %d bytes\r\n",\r |
410 | pPacket,\r |
411 | pContext->BlockNumber,\r |
412 | (UINT32)LengthInBytes ));\r |
413 | \r |
414 | //\r |
415 | // Copy the file data into the packet\r |
416 | //\r |
417 | pPacket->TxBytes = (ssize_t)( 2 + 2 + LengthInBytes );\r |
418 | if ( 0 < LengthInBytes ) {\r |
419 | CopyMem ( pBuffer,\r |
420 | pContext->pBuffer,\r |
421 | (UINTN)LengthInBytes );\r |
422 | DEBUG (( DEBUG_FILE_BUFFER,\r |
423 | "0x%08x: Buffer consumed %d bytes of file data\r\n",\r |
424 | pContext->pBuffer,\r |
425 | LengthInBytes ));\r |
426 | \r |
427 | //\r |
428 | // Account for the file data consumed\r |
429 | //\r |
430 | pContext->ValidBytes -= LengthInBytes;\r |
431 | pContext->BytesToSend -= LengthInBytes;\r |
432 | pContext->pBuffer += LengthInBytes;\r |
433 | if ( pContext->pEnd <= pContext->pBuffer ) {\r |
434 | pContext->pBuffer = &pContext->FileData[ 0 ];\r |
435 | }\r |
436 | }\r |
437 | \r |
438 | //\r |
439 | // Queue the packet for transmission\r |
7dc13291 |
440 | //\r |
f6e5cdd5 |
441 | PacketQueue ( pContext, pPacket );\r |
7dc13291 |
442 | }\r |
443 | \r |
f6e5cdd5 |
444 | //\r |
445 | // Return the read status\r |
446 | //\r |
7dc13291 |
447 | DBG_EXIT ( );\r |
f6e5cdd5 |
448 | return bReadError;\r |
7dc13291 |
449 | }\r |
450 | \r |
451 | \r |
452 | /**\r |
f6e5cdd5 |
453 | Free the packet\r |
7dc13291 |
454 | \r |
f6e5cdd5 |
455 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
456 | @param [in] pPacket Address of a ::TFTP_PACKET structure\r |
7dc13291 |
457 | \r |
458 | **/\r |
459 | VOID\r |
f6e5cdd5 |
460 | PacketFree(\r |
461 | IN TSDT_CONNECTION_CONTEXT * pContext,\r |
462 | IN TFTP_PACKET * pPacket\r |
7dc13291 |
463 | )\r |
464 | {\r |
f6e5cdd5 |
465 | DBG_ENTER ( );\r |
7dc13291 |
466 | \r |
467 | //\r |
f6e5cdd5 |
468 | // Don't free the error packet\r |
7dc13291 |
469 | //\r |
f6e5cdd5 |
470 | if ( pPacket != &pContext->ErrorPacket ) {\r |
7dc13291 |
471 | //\r |
f6e5cdd5 |
472 | // Place the packet on the free list\r |
7dc13291 |
473 | //\r |
f6e5cdd5 |
474 | pPacket->pNext = pContext->pFreeList;\r |
475 | pContext->pFreeList = pPacket;\r |
476 | DEBUG (( DEBUG_TX_PACKET,\r |
477 | "0x%08x: Packet queued to free list\r\n",\r |
478 | pPacket ));\r |
7dc13291 |
479 | }\r |
480 | \r |
f6e5cdd5 |
481 | DBG_EXIT ( );\r |
7dc13291 |
482 | }\r |
483 | \r |
484 | \r |
485 | /**\r |
f6e5cdd5 |
486 | Get a packet from the free list for transmission\r |
7dc13291 |
487 | \r |
f6e5cdd5 |
488 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
7dc13291 |
489 | \r |
f6e5cdd5 |
490 | @retval Address of a ::TFTP_PACKET structure\r |
7dc13291 |
491 | \r |
492 | **/\r |
f6e5cdd5 |
493 | TFTP_PACKET *\r |
494 | PacketGet (\r |
495 | IN TSDT_CONNECTION_CONTEXT * pContext\r |
7dc13291 |
496 | )\r |
497 | {\r |
f6e5cdd5 |
498 | TFTP_PACKET * pPacket;\r |
499 | \r |
500 | DBG_ENTER ( );\r |
501 | \r |
7dc13291 |
502 | //\r |
f6e5cdd5 |
503 | // Get the next packet from the free list\r |
7dc13291 |
504 | //\r |
f6e5cdd5 |
505 | pPacket = pContext->pFreeList;\r |
506 | if ( NULL != pPacket ) {\r |
507 | pContext->pFreeList = pPacket->pNext;\r |
508 | pPacket->RetryCount = 0;\r |
509 | DEBUG (( DEBUG_TX_PACKET,\r |
510 | "0x%08x: Packet removed from free list\r\n",\r |
511 | pPacket ));\r |
7dc13291 |
512 | }\r |
513 | \r |
514 | //\r |
f6e5cdd5 |
515 | // Return the packet\r |
7dc13291 |
516 | //\r |
f6e5cdd5 |
517 | DBG_EXIT_HEX ( pPacket );\r |
518 | return pPacket;\r |
7dc13291 |
519 | }\r |
520 | \r |
521 | \r |
522 | /**\r |
f6e5cdd5 |
523 | Queue the packet for transmission\r |
7dc13291 |
524 | \r |
f6e5cdd5 |
525 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
526 | @param [in] pPacket Address of a ::TFTP_PACKET structure\r |
7dc13291 |
527 | \r |
f6e5cdd5 |
528 | @retval TRUE if a transmission error has occurred\r |
7dc13291 |
529 | \r |
530 | **/\r |
f6e5cdd5 |
531 | BOOLEAN\r |
532 | PacketQueue (\r |
533 | IN TSDT_CONNECTION_CONTEXT * pContext,\r |
534 | IN TFTP_PACKET * pPacket\r |
7dc13291 |
535 | )\r |
536 | {\r |
f6e5cdd5 |
537 | BOOLEAN bTransmitError;\r |
538 | TFTP_PACKET * pTail;\r |
539 | EFI_STATUS Status;\r |
540 | \r |
541 | DBG_ENTER ( );\r |
7dc13291 |
542 | \r |
543 | //\r |
f6e5cdd5 |
544 | // Account for this data block\r |
7dc13291 |
545 | //\r |
f6e5cdd5 |
546 | pPacket->BlockNumber = pContext->BlockNumber;\r |
547 | pContext->BlockNumber += 1;\r |
7dc13291 |
548 | \r |
f6e5cdd5 |
549 | //\r |
550 | // Queue the packet for transmission\r |
551 | //\r |
552 | pTail = pContext->pTxTail;\r |
553 | if ( NULL == pTail ) {\r |
554 | pContext->pTxHead = pPacket;\r |
555 | }\r |
556 | else {\r |
557 | pTail->pNext = pPacket;\r |
558 | }\r |
559 | pContext->pTxTail = pPacket;\r |
560 | pPacket->pNext = NULL;\r |
561 | DEBUG (( DEBUG_TX_PACKET,\r |
562 | "0x%08x: Packet queued to TX list\r\n",\r |
563 | pPacket ));\r |
7dc13291 |
564 | \r |
f6e5cdd5 |
565 | //\r |
566 | // Start the transmission if necessary\r |
567 | //\r |
568 | bTransmitError = FALSE;\r |
569 | if ( pContext->PacketsInWindow < pContext->WindowSize ) {\r |
570 | Status = PacketTx ( pContext, pPacket );\r |
571 | bTransmitError = (BOOLEAN)( EFI_ERROR ( Status ));\r |
572 | }\r |
573 | \r |
574 | //\r |
575 | // Return the transmit status\r |
576 | //\r |
577 | DBG_EXIT_TF ( bTransmitError );\r |
578 | return bTransmitError;\r |
579 | }\r |
580 | \r |
581 | \r |
582 | /**\r |
583 | Remove a packet from the transmit queue\r |
584 | \r |
585 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
586 | \r |
587 | **/\r |
588 | TFTP_PACKET *\r |
589 | PacketRemove(\r |
590 | IN TSDT_CONNECTION_CONTEXT * pContext\r |
591 | )\r |
592 | {\r |
593 | TFTP_PACKET * pNext;\r |
594 | TFTP_PACKET * pPacket;\r |
595 | \r |
596 | DBG_ENTER ( );\r |
597 | \r |
598 | //\r |
599 | // Remove a packet from the transmit queue\r |
600 | //\r |
601 | //\r |
602 | pPacket = pContext->pTxHead;\r |
603 | if ( NULL != pPacket ) {\r |
604 | pNext = pPacket->pNext;\r |
605 | pContext->pTxHead = pNext;\r |
606 | if ( NULL == pNext ) {\r |
607 | pContext->pTxTail = NULL;\r |
7dc13291 |
608 | }\r |
f6e5cdd5 |
609 | DEBUG (( DEBUG_TX_PACKET,\r |
610 | "0x%08x: Packet removed from TX list\r\n",\r |
611 | pPacket ));\r |
7dc13291 |
612 | \r |
613 | //\r |
f6e5cdd5 |
614 | // Remove this packet from the window\r |
7dc13291 |
615 | //\r |
f6e5cdd5 |
616 | pContext->PacketsInWindow -= 1;\r |
617 | }\r |
7dc13291 |
618 | \r |
619 | //\r |
f6e5cdd5 |
620 | // Return the packet\r |
7dc13291 |
621 | //\r |
f6e5cdd5 |
622 | DBG_EXIT_HEX ( pPacket );\r |
623 | return pPacket;\r |
7dc13291 |
624 | }\r |
625 | \r |
626 | \r |
627 | /**\r |
f6e5cdd5 |
628 | Transmit the packet\r |
7dc13291 |
629 | \r |
f6e5cdd5 |
630 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
631 | @param [in] pPacket Address of a ::TFTP_PACKET structure\r |
7dc13291 |
632 | \r |
633 | @retval EFI_SUCCESS Message processed successfully\r |
634 | \r |
635 | **/\r |
636 | EFI_STATUS\r |
f6e5cdd5 |
637 | PacketTx (\r |
638 | IN TSDT_CONNECTION_CONTEXT * pContext,\r |
639 | IN TFTP_PACKET * pPacket\r |
7dc13291 |
640 | )\r |
641 | {\r |
f6e5cdd5 |
642 | ssize_t LengthInBytes;\r |
7dc13291 |
643 | EFI_STATUS Status;\r |
644 | \r |
f6e5cdd5 |
645 | DBG_ENTER ( );\r |
646 | \r |
7dc13291 |
647 | //\r |
f6e5cdd5 |
648 | // Assume success\r |
7dc13291 |
649 | //\r |
f6e5cdd5 |
650 | Status = EFI_SUCCESS;\r |
651 | \r |
652 | //\r |
653 | // Determine if this packet should be transmitted\r |
654 | //\r |
655 | if ( PcdGet32 ( Tftp_MaxRetry ) >= pPacket->RetryCount ) {\r |
656 | pPacket->RetryCount += 1;\r |
657 | \r |
7dc13291 |
658 | //\r |
f6e5cdd5 |
659 | // Display the operation\r |
7dc13291 |
660 | //\r |
f6e5cdd5 |
661 | DEBUG (( DEBUG_TX_PACKET,\r |
662 | "0x%08x: Packet transmiting\r\n",\r |
663 | pPacket ));\r |
664 | DEBUG (( DEBUG_TX,\r |
665 | "0x%08x: pContext sending 0x%08x bytes\r\n",\r |
666 | pContext,\r |
667 | pPacket->TxBytes ));\r |
668 | \r |
7dc13291 |
669 | //\r |
f6e5cdd5 |
670 | // Keep track of when the packet was transmitted\r |
7dc13291 |
671 | //\r |
f6e5cdd5 |
672 | if ( PcdGetBool ( Tftp_HighSpeed )) {\r |
673 | pPacket->TxTime = GetPerformanceCounter ( );\r |
674 | }\r |
7dc13291 |
675 | \r |
676 | //\r |
f6e5cdd5 |
677 | // Send the TFTP packet\r |
7dc13291 |
678 | //\r |
f6e5cdd5 |
679 | pContext->PacketsInWindow += 1;\r |
680 | LengthInBytes = sendto ( pContext->SocketFd,\r |
681 | &pPacket->TxBuffer[ 0 ],\r |
682 | pPacket->TxBytes,\r |
683 | 0,\r |
684 | (struct sockaddr *)&pContext->RemoteAddress,\r |
685 | pContext->RemoteAddress.sin6_len );\r |
686 | if ( -1 == LengthInBytes ) {\r |
687 | DEBUG (( DEBUG_ERROR | DEBUG_TX,\r |
688 | "ERROR - Transmit failure, errno: 0x%08x\r\n",\r |
689 | errno ));\r |
690 | pContext->PacketsInWindow -= 1;\r |
691 | Status = EFI_DEVICE_ERROR;\r |
692 | }\r |
693 | }\r |
694 | else {\r |
695 | //\r |
696 | // Too many retries\r |
697 | //\r |
698 | Status = EFI_NO_RESPONSE;\r |
699 | DEBUG (( DEBUG_WARN | DEBUG_WINDOW,\r |
700 | "WARNING - No response from TFTP client\r\n" ));\r |
7dc13291 |
701 | }\r |
7dc13291 |
702 | \r |
703 | //\r |
704 | // Return the operation status\r |
705 | //\r |
f6e5cdd5 |
706 | DBG_EXIT_STATUS ( Status );\r |
7dc13291 |
707 | return Status;\r |
708 | }\r |
709 | \r |
710 | \r |
711 | /**\r |
f6e5cdd5 |
712 | Process the work for the sockets.\r |
7dc13291 |
713 | \r |
f6e5cdd5 |
714 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
715 | @param [in] pIndex Address of an index into the pollfd array\r |
7dc13291 |
716 | \r |
717 | **/\r |
f6e5cdd5 |
718 | VOID\r |
719 | PortWork (\r |
720 | IN TSDT_TFTP_SERVER * pTftpServer,\r |
721 | IN int * pIndex\r |
7dc13291 |
722 | )\r |
723 | {\r |
f6e5cdd5 |
724 | int Index;\r |
725 | TSDT_CONNECTION_CONTEXT * pContext;\r |
726 | struct pollfd * pTftpPort;\r |
727 | socklen_t RemoteAddressLength;\r |
728 | int revents;\r |
7dc13291 |
729 | \r |
f6e5cdd5 |
730 | DBG_ENTER ( );\r |
7dc13291 |
731 | \r |
732 | //\r |
f6e5cdd5 |
733 | // Locate the port\r |
734 | //\r |
735 | Index = *pIndex;\r |
736 | if ( -1 != Index ) {\r |
737 | pTftpPort = &pTftpServer->TftpPort[ *pIndex ];\r |
738 | \r |
739 | //\r |
740 | // Handle input events\r |
741 | //\r |
742 | revents = pTftpPort->revents;\r |
743 | pTftpPort->revents = 0;\r |
744 | if ( 0 != ( revents & POLLRDNORM )) {\r |
745 | //\r |
746 | // Receive the message from the remote system\r |
747 | //\r |
748 | RemoteAddressLength = sizeof ( pTftpServer->RemoteAddress );\r |
749 | pTftpServer->RxBytes = recvfrom ( pTftpPort->fd,\r |
750 | &pTftpServer->RxBuffer[ 0 ],\r |
751 | sizeof ( pTftpServer->RxBuffer ),\r |
752 | 0,\r |
753 | (struct sockaddr *) &pTftpServer->RemoteAddress,\r |
754 | &RemoteAddressLength );\r |
755 | if ( -1 != pTftpServer->RxBytes ) {\r |
756 | if ( PcdGetBool ( Tftp_HighSpeed )) {\r |
757 | pTftpServer->RxTime = GetPerformanceCounter ( );\r |
758 | }\r |
759 | if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {\r |
760 | DEBUG (( DEBUG_TFTP_PORT,\r |
761 | "Received %d bytes from %d.%d.%d.%d:%d\r\n",\r |
762 | pTftpServer->RxBytes,\r |
763 | pTftpServer->RemoteAddress.v4.sin_addr.s_addr & 0xff,\r |
764 | ( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ) & 0xff,\r |
765 | ( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ) & 0xff,\r |
766 | ( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ) & 0xff,\r |
767 | htons ( pTftpServer->RemoteAddress.v4.sin_port )));\r |
768 | }\r |
769 | else {\r |
770 | DEBUG (( DEBUG_TFTP_PORT,\r |
771 | "Received %d bytes from [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",\r |
772 | pTftpServer->RxBytes,\r |
773 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],\r |
774 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],\r |
775 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],\r |
776 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],\r |
777 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],\r |
778 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],\r |
779 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],\r |
780 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],\r |
781 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],\r |
782 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],\r |
783 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],\r |
784 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],\r |
785 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],\r |
786 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],\r |
787 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],\r |
788 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],\r |
789 | htons ( pTftpServer->RemoteAddress.v6.sin6_port )));\r |
790 | }\r |
791 | \r |
792 | //\r |
793 | // Lookup connection context using the remote system address and port\r |
794 | // to determine if an existing connection to this remote\r |
795 | // system exists\r |
796 | //\r |
797 | pContext = ContextFind ( pTftpServer );\r |
798 | \r |
799 | //\r |
800 | // Process the received message\r |
801 | //\r |
802 | TftpProcessRequest ( pTftpServer, pContext, pTftpPort->fd );\r |
803 | }\r |
804 | else {\r |
805 | //\r |
806 | // Receive error on the TFTP server port\r |
807 | // Close the server socket\r |
808 | //\r |
809 | DEBUG (( DEBUG_ERROR,\r |
810 | "ERROR - Failed receive on TFTP server port, errno: 0x%08x\r\n",\r |
811 | errno ));\r |
812 | revents |= POLLHUP;\r |
813 | }\r |
814 | }\r |
815 | \r |
816 | //\r |
817 | // Handle the close event\r |
818 | //\r |
819 | if ( 0 != ( revents & POLLHUP )) {\r |
820 | //\r |
821 | // Close the port\r |
822 | //\r |
823 | close ( pTftpPort->fd );\r |
824 | pTftpPort->fd = -1;\r |
825 | *pIndex = -1;\r |
826 | pTftpServer->Entries -= 1;\r |
827 | ASSERT ( 0 <= pTftpServer->Entries );\r |
828 | }\r |
829 | }\r |
830 | \r |
831 | DBG_EXIT ( );\r |
832 | }\r |
833 | \r |
834 | \r |
835 | /**\r |
836 | Build and send an error packet\r |
837 | \r |
838 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
839 | @param [in] Error Error number for the packet\r |
840 | @param [in] pError Zero terminated error string address\r |
841 | \r |
842 | @retval EFI_SUCCESS Message processed successfully\r |
843 | \r |
844 | **/\r |
845 | EFI_STATUS\r |
846 | SendError (\r |
847 | IN TSDT_CONNECTION_CONTEXT * pContext,\r |
848 | IN UINT16 Error,\r |
849 | IN UINT8 * pError\r |
850 | )\r |
851 | {\r |
852 | UINT8 Character;\r |
853 | UINT8 * pBuffer;\r |
854 | TFTP_PACKET * pPacket;\r |
855 | EFI_STATUS Status;\r |
856 | \r |
857 | DBG_ENTER ( );\r |
858 | \r |
859 | //\r |
860 | // Build the error packet\r |
861 | //\r |
862 | pPacket = &pContext->ErrorPacket;\r |
863 | pBuffer = &pPacket->TxBuffer[ 0 ];\r |
864 | pBuffer[ 0 ] = 0;\r |
865 | pBuffer[ 1 ] = TFTP_OP_ERROR;\r |
866 | pBuffer[ 2 ] = (UINT8)( Error >> 8 );\r |
867 | pBuffer[ 3 ] = (UINT8)Error;\r |
868 | \r |
869 | //\r |
870 | // Copy the zero terminated string into the buffer\r |
871 | //\r |
872 | pBuffer += 4;\r |
873 | do {\r |
874 | Character = *pError++;\r |
875 | *pBuffer++ = Character;\r |
876 | } while ( 0 != Character );\r |
877 | \r |
878 | //\r |
879 | // Send the error message\r |
880 | //\r |
881 | pPacket->TxBytes = pBuffer - &pPacket->TxBuffer[ 0 ];\r |
882 | Status = PacketTx ( pContext, pPacket );\r |
883 | \r |
884 | //\r |
885 | // Return the operation status\r |
886 | //\r |
887 | DBG_EXIT_STATUS ( Status );\r |
888 | return Status;\r |
889 | }\r |
890 | \r |
891 | \r |
892 | /**\r |
893 | Scan the list of sockets and process any pending work\r |
894 | \r |
895 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
896 | \r |
897 | **/\r |
898 | VOID\r |
899 | SocketPoll (\r |
900 | IN TSDT_TFTP_SERVER * pTftpServer\r |
901 | )\r |
902 | {\r |
903 | int FDCount;\r |
904 | \r |
905 | DEBUG (( DEBUG_SOCKET_POLL, "Entering SocketPoll\r\n" ));\r |
906 | \r |
907 | //\r |
908 | // Determine if any ports are active\r |
909 | //\r |
910 | if ( 0 != pTftpServer->Entries ) {\r |
911 | FDCount = poll ( &pTftpServer->TftpPort[ 0 ],\r |
912 | pTftpServer->Entries,\r |
913 | CLIENT_POLL_DELAY );\r |
914 | if ( 0 < FDCount ) {\r |
915 | //\r |
916 | // Process this port\r |
917 | //\r |
918 | PortWork ( pTftpServer, &pTftpServer->Udpv4Index );\r |
919 | PortWork ( pTftpServer, &pTftpServer->Udpv6Index );\r |
920 | }\r |
921 | }\r |
922 | \r |
923 | DEBUG (( DEBUG_SOCKET_POLL, "Exiting SocketPoll\r\n" ));\r |
924 | }\r |
925 | \r |
926 | \r |
927 | /**\r |
928 | Process the ACK\r |
929 | \r |
930 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
931 | @param [in] pContext Connection context structure address\r |
932 | \r |
933 | @retval TRUE if the context should be closed\r |
934 | \r |
935 | **/\r |
936 | BOOLEAN\r |
937 | TftpAck (\r |
938 | IN TSDT_TFTP_SERVER * pTftpServer,\r |
939 | IN TSDT_CONNECTION_CONTEXT * pContext\r |
940 | )\r |
941 | {\r |
942 | INTN AckNumber;\r |
943 | BOOLEAN bCloseContext;\r |
944 | UINT16 BlockNumber;\r |
945 | UINT8 * pBuffer;\r |
946 | TFTP_PACKET * pPacket;\r |
947 | EFI_STATUS Status;\r |
948 | \r |
949 | DBG_ENTER ( );\r |
950 | \r |
951 | //\r |
952 | // Use break instead of goto\r |
953 | //\r |
954 | bCloseContext = FALSE;\r |
955 | for ( ; ; ) {\r |
956 | //\r |
957 | // Validate the parameters\r |
958 | //\r |
959 | if ( NULL == pContext ) {\r |
960 | if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {\r |
961 | DEBUG (( DEBUG_ERROR,\r |
962 | "ERROR - File not open for %d.%d.%d.%d:%d\r\n",\r |
963 | (UINT8)pTftpServer->RemoteAddress.v4.sin_addr.s_addr,\r |
964 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ),\r |
965 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ),\r |
966 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ),\r |
967 | htons ( pTftpServer->RemoteAddress.v4.sin_port )));\r |
968 | }\r |
969 | else {\r |
970 | DEBUG (( DEBUG_ERROR,\r |
971 | "ERROR - File not open for [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",\r |
972 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],\r |
973 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],\r |
974 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],\r |
975 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],\r |
976 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],\r |
977 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],\r |
978 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],\r |
979 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],\r |
980 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],\r |
981 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],\r |
982 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],\r |
983 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],\r |
984 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],\r |
985 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],\r |
986 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],\r |
987 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],\r |
988 | htons ( pTftpServer->RemoteAddress.v6.sin6_port )));\r |
989 | }\r |
990 | break;\r |
991 | }\r |
992 | \r |
993 | //\r |
994 | // Verify that the ACK was expected\r |
995 | //\r |
996 | pPacket = pContext->pTxHead;\r |
997 | if ( NULL == pPacket ) {\r |
998 | //\r |
999 | // ACK not expected!\r |
1000 | //\r |
1001 | DEBUG (( DEBUG_ERROR,\r |
1002 | "ERROR - Expecting data not ACKs for pContext 0x%08x\r\n",\r |
1003 | pContext ));\r |
1004 | break;\r |
1005 | }\r |
1006 | \r |
1007 | //\r |
1008 | // Get the ACKed block number\r |
1009 | //\r |
1010 | pBuffer = &pTftpServer->RxBuffer[ 0 ];\r |
1011 | BlockNumber = HTONS ( *(UINT16 *)&pBuffer[ 2 ]);\r |
1012 | \r |
1013 | //\r |
1014 | // Determine if this is the correct ACK\r |
1015 | //\r |
1016 | DEBUG (( DEBUG_TFTP_ACK,\r |
1017 | "ACK for block 0x%04x received\r\n",\r |
1018 | BlockNumber ));\r |
1019 | AckNumber = BlockNumber - pPacket->BlockNumber;\r |
1020 | if (( 0 > AckNumber ) || ( AckNumber >= (INTN)pContext->PacketsInWindow )){\r |
1021 | DEBUG (( DEBUG_WARN | DEBUG_TFTP_ACK,\r |
1022 | "WARNING - Expecting ACK 0x%0x4 not received ACK 0x%08x\r\n",\r |
1023 | pPacket->BlockNumber,\r |
1024 | BlockNumber ));\r |
1025 | break;\r |
1026 | }\r |
1027 | \r |
1028 | //\r |
1029 | // Release the ACKed packets\r |
1030 | //\r |
1031 | do {\r |
1032 | //\r |
1033 | // Remove the packet from the transmit list and window\r |
1034 | //\r |
1035 | pPacket = PacketRemove ( pContext );\r |
1036 | \r |
1037 | //\r |
1038 | // Get the block number of this packet\r |
1039 | //\r |
1040 | AckNumber = pPacket->BlockNumber;\r |
1041 | \r |
1042 | //\r |
1043 | // Increase the size of the transmit window\r |
1044 | //\r |
1045 | if ( PcdGetBool ( Tftp_HighSpeed )\r |
1046 | && ( AckNumber == BlockNumber )) {\r |
1047 | WindowAck ( pTftpServer, pContext, pPacket );\r |
1048 | }\r |
1049 | \r |
1050 | //\r |
1051 | // Free this packet\r |
1052 | //\r |
1053 | PacketFree ( pContext, pPacket );\r |
1054 | } while (( NULL != pContext->pTxHead ) && ( AckNumber != BlockNumber ));\r |
1055 | \r |
1056 | //\r |
1057 | // Fill the window with packets\r |
1058 | //\r |
1059 | pPacket = pContext->pTxHead;\r |
1060 | while (( NULL != pPacket )\r |
1061 | && ( pContext->PacketsInWindow < pContext->WindowSize )\r |
1062 | && ( !bCloseContext )) {\r |
1063 | Status = PacketTx ( pContext, pPacket );\r |
1064 | bCloseContext = (BOOLEAN)( EFI_ERROR ( Status ));\r |
1065 | pPacket = pPacket->pNext;\r |
1066 | }\r |
1067 | \r |
1068 | //\r |
1069 | // Get more packets ready for transmission\r |
1070 | //\r |
1071 | PacketFill ( pContext );\r |
1072 | \r |
1073 | //\r |
1074 | // Close the context when the last packet is ACKed\r |
1075 | //\r |
1076 | if ( 0 == pContext->PacketsInWindow ) {\r |
1077 | bCloseContext = TRUE;\r |
1078 | \r |
1079 | //\r |
1080 | // Display the bandwidth\r |
1081 | //\r |
1082 | if ( PcdGetBool ( Tftp_Bandwidth )) {\r |
1083 | UINT64 Bandwidth;\r |
1084 | UINT64 DeltaTime;\r |
1085 | UINT64 NanoSeconds;\r |
1086 | UINT32 Value;\r |
1087 | \r |
1088 | //\r |
1089 | // Compute the download time\r |
1090 | //\r |
1091 | DeltaTime = GetPerformanceCounter ( );\r |
1092 | if ( pTftpServer->Time2 > pTftpServer->Time1 ) {\r |
1093 | DeltaTime = DeltaTime - pContext->TimeStart;\r |
1094 | }\r |
1095 | else {\r |
1096 | DeltaTime = pContext->TimeStart - DeltaTime;\r |
1097 | }\r |
1098 | NanoSeconds = GetTimeInNanoSecond ( DeltaTime );\r |
1099 | Bandwidth = pContext->LengthInBytes;\r |
1100 | DEBUG (( DEBUG_WINDOW,\r |
1101 | "File Length %Ld, Transfer Time: %d.%03d Sec\r\n",\r |
1102 | Bandwidth,\r |
1103 | DivU64x32 ( NanoSeconds, 1000 * 1000 * 1000 ),\r |
1104 | ((UINT32)DivU64x32 ( NanoSeconds, 1000 * 1000 )) % 1000 ));\r |
1105 | \r |
1106 | //\r |
1107 | // Display the round trip time\r |
1108 | //\r |
1109 | Bandwidth = MultU64x32 ( Bandwidth, 8 * 1000 * 1000 );\r |
1110 | Bandwidth /= NanoSeconds;\r |
1111 | if ( 1000 > Bandwidth ) {\r |
1112 | Value = (UINT32)Bandwidth;\r |
1113 | Print ( L"Bandwidth: %d Kbits/Sec\r\n",\r |
1114 | Value );\r |
1115 | }\r |
1116 | else if (( 1000 * 1000 ) > Bandwidth ) {\r |
1117 | Value = (UINT32)Bandwidth;\r |
1118 | Print ( L"Bandwidth: %d.%03d Mbits/Sec\r\n",\r |
1119 | Value / 1000,\r |
1120 | Value % 1000 );\r |
1121 | }\r |
1122 | else {\r |
1123 | Value = (UINT32)DivU64x32 ( Bandwidth, 1000 );\r |
1124 | Print ( L"Bandwidth: %d.%03d Gbits/Sec\r\n",\r |
1125 | Value / 1000,\r |
1126 | Value % 1000 );\r |
1127 | }\r |
1128 | }\r |
1129 | }\r |
1130 | break;\r |
1131 | }\r |
1132 | \r |
1133 | //\r |
1134 | // Return the operation status\r |
1135 | //\r |
1136 | DBG_EXIT ( );\r |
1137 | return bCloseContext;\r |
1138 | }\r |
1139 | \r |
1140 | \r |
1141 | /**\r |
1142 | Get the next TFTP option\r |
1143 | \r |
1144 | @param [in] pOption Address of a zero terminated option string\r |
1145 | @param [in] pEnd End of buffer address\r |
1146 | @param [in] ppNextOption Address to receive the address of the next\r |
1147 | zero terminated option string\r |
1148 | \r |
1149 | @retval EFI_SUCCESS Message processed successfully\r |
1150 | \r |
1151 | **/\r |
1152 | EFI_STATUS\r |
1153 | TftpOptionGet (\r |
1154 | IN UINT8 * pOption,\r |
1155 | IN UINT8 * pEnd,\r |
1156 | IN UINT8 ** ppNextOption\r |
1157 | )\r |
1158 | {\r |
1159 | UINT8 * pNextOption;\r |
1160 | EFI_STATUS Status;\r |
1161 | \r |
1162 | //\r |
1163 | // Locate the end of the option\r |
1164 | //\r |
1165 | pNextOption = pOption;\r |
1166 | while (( pEnd > pNextOption ) && ( 0 != *pNextOption )) {\r |
1167 | pNextOption += 1;\r |
1168 | }\r |
1169 | if ( pEnd <= pNextOption ) {\r |
1170 | //\r |
1171 | // Error - end of buffer reached\r |
1172 | //\r |
1173 | DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,\r |
1174 | "ERROR - Option without zero termination received!\r\n" ));\r |
1175 | Status = EFI_INVALID_PARAMETER;\r |
1176 | }\r |
1177 | else {\r |
1178 | //\r |
1179 | // Zero terminated option found\r |
1180 | //\r |
1181 | pNextOption += 1;\r |
1182 | \r |
1183 | //\r |
1184 | // Display the zero terminated ASCII option string\r |
1185 | //\r |
1186 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
1187 | "Option: %a\r\n",\r |
1188 | pOption ));\r |
1189 | Status = EFI_SUCCESS;\r |
1190 | }\r |
1191 | \r |
1192 | //\r |
1193 | // Return the next option address\r |
1194 | //\r |
1195 | *ppNextOption = pNextOption;\r |
1196 | \r |
1197 | //\r |
1198 | // Return the operation status\r |
1199 | //\r |
1200 | return Status;\r |
1201 | }\r |
1202 | \r |
1203 | \r |
1204 | /**\r |
1205 | Place an option value into the option acknowledgement\r |
1206 | \r |
1207 | @param [in] pOack Option acknowledgement address\r |
1208 | @param [in] Value Value to translate into ASCII decimal\r |
1209 | \r |
1210 | @return Option acknowledgement address\r |
1211 | \r |
1212 | **/\r |
1213 | UINT8 *\r |
1214 | TftpOptionSet (\r |
1215 | IN UINT8 * pOack,\r |
1216 | IN UINT64 Value\r |
1217 | )\r |
1218 | {\r |
1219 | UINT64 NextValue;\r |
1220 | \r |
1221 | //\r |
1222 | // Determine the next value\r |
1223 | //\r |
1224 | NextValue = Value / 10;\r |
1225 | \r |
1226 | //\r |
1227 | // Supress leading zeros\r |
7dc13291 |
1228 | //\r |
1229 | if ( 0 != NextValue ) {\r |
1230 | pOack = TftpOptionSet ( pOack, NextValue );\r |
1231 | }\r |
1232 | \r |
1233 | //\r |
1234 | // Output this digit\r |
1235 | //\r |
1236 | *pOack++ = (UINT8)( Value - ( NextValue * 10 ) + '0' );\r |
1237 | \r |
1238 | //\r |
1239 | // Return the next option acknowledgement location\r |
1240 | //\r |
1241 | return pOack;\r |
1242 | }\r |
1243 | \r |
1244 | \r |
1245 | /**\r |
1246 | Process the TFTP request\r |
1247 | \r |
f6e5cdd5 |
1248 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
7dc13291 |
1249 | @param [in] pOption Address of the first zero terminated option string\r |
1250 | @param [in] pEnd End of buffer address\r |
1251 | \r |
1252 | **/\r |
1253 | VOID\r |
1254 | TftpOptions (\r |
1255 | IN TSDT_CONNECTION_CONTEXT * pContext,\r |
1256 | IN UINT8 * pOption,\r |
1257 | IN UINT8 * pEnd\r |
1258 | )\r |
1259 | {\r |
1260 | UINT8 * pNextOption;\r |
1261 | UINT8 * pOack;\r |
f6e5cdd5 |
1262 | TFTP_PACKET * pPacket;\r |
7dc13291 |
1263 | UINT8 * pTemp;\r |
1264 | UINT8 * pValue;\r |
1265 | EFI_STATUS Status;\r |
1266 | INT32 Value;\r |
1267 | \r |
f6e5cdd5 |
1268 | //\r |
1269 | // Get a packet\r |
1270 | //\r |
1271 | pPacket = PacketGet ( pContext );\r |
1272 | \r |
7dc13291 |
1273 | //\r |
1274 | // Start the OACK packet\r |
1275 | // Let the OACK handle the parsing errors\r |
1276 | // See http://tools.ietf.org/html/rfc2347\r |
1277 | //\r |
f6e5cdd5 |
1278 | pOack = &pPacket->TxBuffer[ 0 ];\r |
7dc13291 |
1279 | *pOack++ = 0;\r |
1280 | *pOack++ = TFTP_OP_OACK;\r |
f6e5cdd5 |
1281 | pPacket->TxBytes = 2;\r |
1282 | pPacket->BlockNumber = 0;\r |
7dc13291 |
1283 | \r |
1284 | //\r |
1285 | // Walk the list of options\r |
1286 | //\r |
1287 | do {\r |
1288 | //\r |
1289 | // Get the next option, skip junk at end of message\r |
1290 | //\r |
1291 | Status = TftpOptionGet ( pOption, pEnd, &pNextOption );\r |
1292 | if ( !EFI_ERROR ( Status )) {\r |
1293 | //\r |
1294 | // Process the option\r |
1295 | //\r |
1296 | \r |
1297 | //\r |
1298 | // blksize - See http://tools.ietf.org/html/rfc2348\r |
1299 | //\r |
1300 | pValue = pNextOption;\r |
f6e5cdd5 |
1301 | if ( 0 == strcasecmp ((char *)pOption, "blksize" )) {\r |
7dc13291 |
1302 | //\r |
1303 | // Get the value\r |
1304 | //\r |
1305 | Status = TftpOptionGet ( pValue, pEnd, &pNextOption );\r |
1306 | if ( !EFI_ERROR ( Status )) {\r |
1307 | //\r |
1308 | // Validate the block size, skip non-numeric block sizes\r |
1309 | //\r |
1310 | Status = TftpOptionValue ( pValue, &Value );\r |
1311 | if ( !EFI_ERROR ( Status )) {\r |
1312 | //\r |
1313 | // Propose a smaller block size if necessary\r |
1314 | //\r |
1315 | if ( Value > TFTP_MAX_BLOCK_SIZE ) {\r |
1316 | Value = TFTP_MAX_BLOCK_SIZE;\r |
1317 | }\r |
1318 | \r |
1319 | //\r |
1320 | // Set the new block size\r |
1321 | //\r |
1322 | pContext->BlockSize = Value;\r |
1323 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
1324 | "Using block size of %d bytes\r\n",\r |
1325 | pContext->BlockSize ));\r |
1326 | \r |
1327 | //\r |
1328 | // Update the OACK\r |
1329 | //\r |
1330 | pTemp = pOack;\r |
1331 | *pOack++ = 'b';\r |
1332 | *pOack++ = 'l';\r |
1333 | *pOack++ = 'k';\r |
1334 | *pOack++ = 's';\r |
1335 | *pOack++ = 'i';\r |
1336 | *pOack++ = 'z';\r |
1337 | *pOack++ = 'e';\r |
1338 | *pOack++ = 0;\r |
1339 | pOack = TftpOptionSet ( pOack, pContext->BlockSize );\r |
1340 | *pOack++ = 0;\r |
f6e5cdd5 |
1341 | pPacket->TxBytes += pOack - pTemp;\r |
7dc13291 |
1342 | }\r |
1343 | }\r |
1344 | }\r |
1345 | \r |
1346 | //\r |
1347 | // timeout - See http://tools.ietf.org/html/rfc2349\r |
1348 | //\r |
f6e5cdd5 |
1349 | else if ( 0 == strcasecmp ((char *)pOption, "timeout" )) {\r |
7dc13291 |
1350 | //\r |
1351 | // Get the value\r |
1352 | //\r |
1353 | Status = TftpOptionGet ( pValue, pEnd, &pNextOption );\r |
1354 | if ( !EFI_ERROR ( Status )) {\r |
1355 | Status = TftpOptionValue ( pValue, &Value );\r |
1356 | if ( !EFI_ERROR ( Status )) {\r |
1357 | //\r |
1358 | // Set the timeout value\r |
1359 | //\r |
f6e5cdd5 |
1360 | pContext->MaxTimeout = Value;\r |
7dc13291 |
1361 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
1362 | "Using timeout of %d seconds\r\n",\r |
f6e5cdd5 |
1363 | pContext->MaxTimeout ));\r |
7dc13291 |
1364 | \r |
1365 | //\r |
1366 | // Update the OACK\r |
1367 | //\r |
1368 | pTemp = pOack;\r |
1369 | *pOack++ = 't';\r |
1370 | *pOack++ = 'i';\r |
1371 | *pOack++ = 'm';\r |
1372 | *pOack++ = 'e';\r |
1373 | *pOack++ = 'o';\r |
1374 | *pOack++ = 'u';\r |
1375 | *pOack++ = 't';\r |
1376 | *pOack++ = 0;\r |
f6e5cdd5 |
1377 | pOack = TftpOptionSet ( pOack, pContext->MaxTimeout );\r |
7dc13291 |
1378 | *pOack++ = 0;\r |
f6e5cdd5 |
1379 | pPacket->TxBytes += pOack - pTemp;\r |
7dc13291 |
1380 | }\r |
1381 | }\r |
1382 | }\r |
1383 | \r |
1384 | //\r |
1385 | // tsize - See http://tools.ietf.org/html/rfc2349\r |
1386 | //\r |
f6e5cdd5 |
1387 | else if ( 0 == strcasecmp ((char *)pOption, "tsize" )) {\r |
7dc13291 |
1388 | //\r |
1389 | // Get the value\r |
1390 | //\r |
1391 | Status = TftpOptionGet ( pValue, pEnd, &pNextOption );\r |
1392 | if ( !EFI_ERROR ( Status )) {\r |
1393 | Status = TftpOptionValue ( pValue, &Value );\r |
1394 | if ( !EFI_ERROR ( Status )) {\r |
1395 | //\r |
1396 | // Return the file size\r |
1397 | //\r |
1398 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
1399 | "Returning file size of %Ld bytes\r\n",\r |
1400 | pContext->LengthInBytes ));\r |
1401 | \r |
1402 | //\r |
1403 | // Update the OACK\r |
1404 | //\r |
1405 | pTemp = pOack;\r |
1406 | *pOack++ = 't';\r |
1407 | *pOack++ = 's';\r |
1408 | *pOack++ = 'i';\r |
1409 | *pOack++ = 'z';\r |
1410 | *pOack++ = 'e';\r |
1411 | *pOack++ = 0;\r |
1412 | pOack = TftpOptionSet ( pOack, pContext->LengthInBytes );\r |
1413 | *pOack++ = 0;\r |
f6e5cdd5 |
1414 | pPacket->TxBytes += pOack - pTemp;\r |
7dc13291 |
1415 | }\r |
1416 | }\r |
1417 | }\r |
1418 | else {\r |
1419 | //\r |
1420 | // Unknown option - Ignore it\r |
1421 | //\r |
1422 | DEBUG (( DEBUG_WARN | DEBUG_TFTP_REQUEST,\r |
1423 | "WARNING - Skipping unknown option: %a\r\n",\r |
1424 | pOption ));\r |
1425 | }\r |
1426 | }\r |
1427 | \r |
1428 | //\r |
1429 | // Set the next option\r |
1430 | //\r |
1431 | pOption = pNextOption;\r |
1432 | } while ( pEnd > pOption );\r |
f6e5cdd5 |
1433 | \r |
1434 | //\r |
1435 | // Transmit the OACK if necessary\r |
1436 | //\r |
1437 | if ( 2 < pPacket->TxBytes ) {\r |
1438 | PacketQueue ( pContext, pPacket );\r |
1439 | }\r |
1440 | else {\r |
1441 | PacketFree ( pContext, pPacket );\r |
1442 | }\r |
7dc13291 |
1443 | }\r |
1444 | \r |
1445 | \r |
1446 | /**\r |
1447 | Process the TFTP request\r |
1448 | \r |
1449 | @param [in] pOption Address of the first zero terminated option string\r |
1450 | @param [in] pValue Address to receive the value\r |
1451 | \r |
1452 | @retval EFI_SUCCESS Option translated into a value\r |
1453 | \r |
1454 | **/\r |
1455 | EFI_STATUS\r |
1456 | TftpOptionValue (\r |
1457 | IN UINT8 * pOption,\r |
1458 | IN INT32 * pValue\r |
1459 | )\r |
1460 | {\r |
1461 | UINT8 Digit;\r |
1462 | EFI_STATUS Status;\r |
1463 | INT32 Value;\r |
1464 | \r |
1465 | //\r |
1466 | // Assume success\r |
1467 | //\r |
1468 | Status = EFI_SUCCESS;\r |
1469 | \r |
1470 | //\r |
1471 | // Walk the characters in the option\r |
1472 | //\r |
1473 | Value = 0;\r |
1474 | while ( 0 != *pOption ) {\r |
1475 | //\r |
1476 | // Convert the next digit to binary\r |
1477 | //\r |
1478 | Digit = *pOption++;\r |
1479 | if (( '0' <= Digit ) && ( '9' >= Digit )) {\r |
1480 | Value *= 10;\r |
1481 | Value += Digit - '0';\r |
1482 | }\r |
1483 | else {\r |
1484 | DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,\r |
1485 | "ERROR - Invalid character '0x%02x' in the value\r\n",\r |
1486 | Digit ));\r |
1487 | Status = EFI_INVALID_PARAMETER;\r |
1488 | break;\r |
1489 | }\r |
1490 | }\r |
1491 | \r |
1492 | //\r |
1493 | // Return the value\r |
1494 | //\r |
1495 | *pValue = Value;\r |
1496 | \r |
1497 | //\r |
1498 | // Return the conversion status\r |
1499 | //\r |
1500 | return Status;\r |
1501 | }\r |
1502 | \r |
1503 | \r |
1504 | /**\r |
1505 | Process the TFTP request\r |
1506 | \r |
f6e5cdd5 |
1507 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
1508 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
1509 | @param [in] SocketFd Socket file descriptor\r |
7dc13291 |
1510 | \r |
1511 | **/\r |
1512 | VOID\r |
1513 | TftpProcessRequest (\r |
1514 | IN TSDT_TFTP_SERVER * pTftpServer,\r |
f6e5cdd5 |
1515 | IN TSDT_CONNECTION_CONTEXT * pContext,\r |
1516 | IN int SocketFd\r |
7dc13291 |
1517 | )\r |
1518 | {\r |
1519 | BOOLEAN bCloseContext;\r |
7dc13291 |
1520 | UINT16 Opcode;\r |
7dc13291 |
1521 | \r |
1522 | DBG_ENTER ( );\r |
1523 | \r |
1524 | //\r |
1525 | // Get the opcode\r |
1526 | //\r |
f6e5cdd5 |
1527 | Opcode = HTONS ( *(UINT16 *)&pTftpServer->RxBuffer[ 0 ]);\r |
1528 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
1529 | "TFTP Opcode: 0x%08x\r\n",\r |
1530 | Opcode ));\r |
7dc13291 |
1531 | \r |
1532 | //\r |
1533 | // Validate the parameters\r |
1534 | //\r |
1535 | bCloseContext = FALSE;\r |
7dc13291 |
1536 | switch ( Opcode ) {\r |
1537 | default:\r |
1538 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
1539 | "ERROR - Unknown TFTP opcode: %d\r\n",\r |
1540 | Opcode ));\r |
f6e5cdd5 |
1541 | break;\r |
1542 | \r |
1543 | case TFTP_OP_ACK:\r |
1544 | bCloseContext = TftpAck ( pTftpServer, pContext );\r |
7dc13291 |
1545 | break;\r |
1546 | \r |
1547 | case TFTP_OP_READ_REQUEST:\r |
f6e5cdd5 |
1548 | bCloseContext = TftpRead ( pTftpServer, pContext, SocketFd );\r |
7dc13291 |
1549 | break;\r |
1550 | \r |
f6e5cdd5 |
1551 | \r |
1552 | \r |
1553 | \r |
7dc13291 |
1554 | case TFTP_OP_DATA:\r |
1555 | if ( NULL == pContext ) {\r |
f6e5cdd5 |
1556 | if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {\r |
1557 | DEBUG (( DEBUG_ERROR,\r |
1558 | "ERROR - File not open for %d.%d.%d.%d:%d\r\n",\r |
1559 | (UINT8)pTftpServer->RemoteAddress.v4.sin_addr.s_addr,\r |
1560 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ),\r |
1561 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ),\r |
1562 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ),\r |
1563 | htons ( pTftpServer->RemoteAddress.v4.sin_port )));\r |
1564 | }\r |
1565 | else {\r |
1566 | DEBUG (( DEBUG_ERROR,\r |
1567 | "ERROR - File not open for [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",\r |
1568 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],\r |
1569 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],\r |
1570 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],\r |
1571 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],\r |
1572 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],\r |
1573 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],\r |
1574 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],\r |
1575 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],\r |
1576 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],\r |
1577 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],\r |
1578 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],\r |
1579 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],\r |
1580 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],\r |
1581 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],\r |
1582 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],\r |
1583 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],\r |
1584 | htons ( pTftpServer->RemoteAddress.v6.sin6_port )));\r |
1585 | }\r |
7dc13291 |
1586 | break;\r |
1587 | }\r |
f6e5cdd5 |
1588 | if ( 0 != pContext->PacketsInWindow ) {\r |
7dc13291 |
1589 | DEBUG (( DEBUG_ERROR,\r |
1590 | "ERROR - Expecting ACKs not data for pContext 0x%08x\r\n",\r |
1591 | pContext ));\r |
7dc13291 |
1592 | break;\r |
1593 | }\r |
59bc0593 |
1594 | if ( pTftpServer->RxBytes > (ssize_t)( pContext->BlockSize + 2 + 2 )) {\r |
7dc13291 |
1595 | DEBUG (( DEBUG_ERROR,\r |
1596 | "ERROR - Receive data length of %d > %d bytes (maximum block size) for pContext 0x%08x\r\n",\r |
1597 | pTftpServer->RxBytes - 2 - 2,\r |
1598 | pContext->BlockSize,\r |
1599 | pContext ));\r |
7dc13291 |
1600 | break;\r |
1601 | }\r |
1602 | break;\r |
1603 | \r |
1604 | case TFTP_OP_ERROR:\r |
1605 | if ( NULL == pContext ) {\r |
f6e5cdd5 |
1606 | if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {\r |
1607 | DEBUG (( DEBUG_ERROR,\r |
1608 | "ERROR - File not open for %d.%d.%d.%d:%d\r\n",\r |
1609 | (UINT8)pTftpServer->RemoteAddress.v4.sin_addr.s_addr,\r |
1610 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ),\r |
1611 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ),\r |
1612 | (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ),\r |
1613 | htons ( pTftpServer->RemoteAddress.v4.sin_port )));\r |
7dc13291 |
1614 | }\r |
1615 | else {\r |
f6e5cdd5 |
1616 | DEBUG (( DEBUG_ERROR,\r |
1617 | "ERROR - File not open for [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",\r |
1618 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],\r |
1619 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],\r |
1620 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],\r |
1621 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],\r |
1622 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],\r |
1623 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],\r |
1624 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],\r |
1625 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],\r |
1626 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],\r |
1627 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],\r |
1628 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],\r |
1629 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],\r |
1630 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],\r |
1631 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],\r |
1632 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],\r |
1633 | pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],\r |
1634 | htons ( pTftpServer->RemoteAddress.v6.sin6_port )));\r |
7dc13291 |
1635 | }\r |
7dc13291 |
1636 | }\r |
f6e5cdd5 |
1637 | break;\r |
7dc13291 |
1638 | }\r |
1639 | \r |
1640 | //\r |
1641 | // Determine if the context should be closed\r |
1642 | //\r |
1643 | if ( bCloseContext ) {\r |
1644 | ContextRemove ( pTftpServer, pContext );\r |
1645 | }\r |
1646 | \r |
1647 | DBG_EXIT ( );\r |
1648 | }\r |
1649 | \r |
1650 | \r |
1651 | /**\r |
f6e5cdd5 |
1652 | Process the read request\r |
7dc13291 |
1653 | \r |
f6e5cdd5 |
1654 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
1655 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
1656 | @param [in] SocketFd Socket file descriptor\r |
7dc13291 |
1657 | \r |
f6e5cdd5 |
1658 | @retval TRUE if the context should be closed\r |
7dc13291 |
1659 | \r |
1660 | **/\r |
f6e5cdd5 |
1661 | BOOLEAN\r |
1662 | TftpRead (\r |
7dc13291 |
1663 | IN TSDT_TFTP_SERVER * pTftpServer,\r |
1664 | IN TSDT_CONNECTION_CONTEXT * pContext,\r |
f6e5cdd5 |
1665 | IN int SocketFd\r |
7dc13291 |
1666 | )\r |
1667 | {\r |
f6e5cdd5 |
1668 | BOOLEAN bCloseContext;\r |
1669 | struct stat FileStatus;\r |
7dc13291 |
1670 | UINT8 * pBuffer;\r |
f6e5cdd5 |
1671 | UINT8 * pEnd;\r |
1672 | UINT8 * pFileName;\r |
1673 | UINT8 * pMode;\r |
1674 | UINT8 * pOption;\r |
1675 | CHAR8 * pReadMode;\r |
1676 | UINT64 TimeStart;\r |
7dc13291 |
1677 | \r |
1678 | DBG_ENTER ( );\r |
1679 | \r |
1680 | //\r |
f6e5cdd5 |
1681 | // Log the receive time\r |
7dc13291 |
1682 | //\r |
f6e5cdd5 |
1683 | TimeStart = 0;\r |
1684 | if ( PcdGetBool ( Tftp_Bandwidth )) {\r |
1685 | TimeStart = GetPerformanceCounter ( );\r |
1686 | }\r |
7dc13291 |
1687 | \r |
1688 | //\r |
f6e5cdd5 |
1689 | // Close the context if necessary\r |
7dc13291 |
1690 | //\r |
f6e5cdd5 |
1691 | bCloseContext = FALSE;\r |
1692 | if ( NULL != pContext ) {\r |
1693 | ContextRemove ( pTftpServer, pContext );\r |
1694 | }\r |
7dc13291 |
1695 | \r |
1696 | //\r |
f6e5cdd5 |
1697 | // Use break instead of goto\r |
7dc13291 |
1698 | //\r |
f6e5cdd5 |
1699 | for ( ; ; ) {\r |
1700 | //\r |
1701 | // Create the connection context\r |
1702 | //\r |
1703 | pContext = ContextAdd ( pTftpServer, SocketFd );\r |
1704 | if ( NULL == pContext ) {\r |
1705 | break;\r |
1706 | }\r |
7dc13291 |
1707 | \r |
f6e5cdd5 |
1708 | //\r |
1709 | // Set the start time\r |
1710 | //\r |
1711 | if ( PcdGetBool ( Tftp_Bandwidth )) {\r |
1712 | pContext->TimeStart = TimeStart;\r |
1713 | }\r |
7dc13291 |
1714 | \r |
f6e5cdd5 |
1715 | //\r |
1716 | // Locate the mode\r |
1717 | //\r |
1718 | pBuffer = &pTftpServer->RxBuffer[ 0 ];\r |
1719 | pEnd = &pBuffer[ pTftpServer->RxBytes ];\r |
1720 | pFileName = &pBuffer[ 2 ];\r |
1721 | pMode = pFileName;\r |
1722 | while (( pEnd > pMode ) && ( 0 != *pMode )) {\r |
1723 | pMode += 1;\r |
1724 | }\r |
1725 | if ( pEnd <= pMode ) {\r |
1726 | //\r |
1727 | // Mode not found\r |
1728 | //\r |
1729 | DEBUG (( DEBUG_ERROR | DEBUG_RX,\r |
1730 | "ERROR - File mode not found\r\n" ));\r |
1731 | //\r |
1732 | // Tell the client of the error\r |
1733 | //\r |
1734 | SendError ( pContext,\r |
1735 | TFTP_ERROR_SEE_MSG,\r |
1736 | (UINT8 *)"File open mode not found" );\r |
1737 | break;\r |
1738 | }\r |
1739 | pMode += 1;\r |
1740 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
1741 | "TFTP - FileName: %a\r\n",\r |
1742 | pFileName ));\r |
7dc13291 |
1743 | \r |
f6e5cdd5 |
1744 | //\r |
1745 | // Locate the options\r |
1746 | //\r |
1747 | pOption = pMode;\r |
1748 | while (( pEnd > pOption ) && ( 0 != *pOption )) {\r |
1749 | pOption += 1;\r |
1750 | }\r |
1751 | if ( pEnd <= pOption ) {\r |
1752 | //\r |
1753 | // End of mode not found\r |
1754 | //\r |
1755 | DEBUG (( DEBUG_ERROR | DEBUG_RX,\r |
1756 | "ERROR - File mode not valid\r\n" ));\r |
1757 | //\r |
1758 | // Tell the client of the error\r |
1759 | //\r |
1760 | SendError ( pContext,\r |
1761 | TFTP_ERROR_SEE_MSG,\r |
1762 | (UINT8 *)"File open mode not valid" );\r |
1763 | break;\r |
1764 | }\r |
1765 | pOption += 1;\r |
1766 | DEBUG (( DEBUG_TFTP_REQUEST,\r |
1767 | "TFTP - Mode: %a\r\n",\r |
1768 | pMode ));\r |
7dc13291 |
1769 | \r |
f6e5cdd5 |
1770 | //\r |
1771 | // Verify the mode is supported\r |
1772 | //\r |
1773 | pReadMode = "r";\r |
1774 | if ( 0 == strcasecmp ((char *)pMode, "octet" )) {\r |
1775 | //\r |
1776 | // Read the file as binary input\r |
1777 | //\r |
1778 | pReadMode = "rb";\r |
1779 | }\r |
7dc13291 |
1780 | \r |
f6e5cdd5 |
1781 | //\r |
1782 | // Determine the file length\r |
1783 | //\r |
1784 | pContext->File = fopen ( pFileName, pReadMode );\r |
1785 | if (( NULL == pContext->File )\r |
1786 | || ( -1 == stat ( pFileName, &FileStatus ))) {\r |
1787 | //\r |
1788 | // File not found\r |
1789 | //\r |
1790 | DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,\r |
1791 | ( NULL == pContext->File )\r |
1792 | ? "ERROR - File not found!\r\n"\r |
1793 | : "ERROR - Unable to determine file %a size!\r\n",\r |
1794 | pFileName ));\r |
7dc13291 |
1795 | \r |
f6e5cdd5 |
1796 | //\r |
1797 | // Tell the client of the error\r |
1798 | //\r |
1799 | SendError ( pContext,\r |
1800 | TFTP_ERROR_NOT_FOUND,\r |
1801 | (UINT8 *)"File not found" );\r |
1802 | break;\r |
1803 | }\r |
1804 | pContext->LengthInBytes = FileStatus.st_size;\r |
1805 | pContext->BytesRemaining = pContext->LengthInBytes;\r |
1806 | pContext->BytesToSend = pContext->LengthInBytes;\r |
7dc13291 |
1807 | \r |
f6e5cdd5 |
1808 | //\r |
1809 | // Display the file size\r |
1810 | //\r |
1811 | DEBUG_CODE_BEGIN ( );\r |
1812 | UINT32 Value;\r |
1813 | \r |
1814 | if ( 1024 > pContext->LengthInBytes ) {\r |
1815 | Value = (UINT32)pContext->LengthInBytes;\r |
1816 | DEBUG (( DEBUG_FILE_BUFFER,\r |
1817 | "%a size: %d Bytes\r\n",\r |
1818 | pFileName,\r |
1819 | Value ));\r |
1820 | }\r |
1821 | else if (( 1024 * 1024 ) > pContext->LengthInBytes ) {\r |
1822 | Value = (UINT32)pContext->LengthInBytes;\r |
1823 | DEBUG (( DEBUG_FILE_BUFFER,\r |
1824 | "%a size: %d.%03d KiBytes (%Ld Bytes)\r\n",\r |
1825 | pFileName,\r |
1826 | Value / 1024,\r |
1827 | (( Value % 1024 ) * 1000 ) / 1024,\r |
1828 | pContext->LengthInBytes ));\r |
1829 | }\r |
1830 | else if (( 1024 * 1024 * 1024 ) > pContext->LengthInBytes ) {\r |
1831 | Value = (UINT32)DivU64x32 ( pContext->LengthInBytes, 1024 );\r |
1832 | DEBUG (( DEBUG_FILE_BUFFER,\r |
1833 | "%a size: %d.%03d MiBytes (%Ld Bytes)\r\n",\r |
1834 | pFileName,\r |
1835 | Value / 1024,\r |
1836 | (( Value % 1024 ) * 1000 ) / 1024,\r |
1837 | pContext->LengthInBytes ));\r |
1838 | }\r |
1839 | else {\r |
1840 | Value = (UINT32)DivU64x32 ( pContext->LengthInBytes, 1024 * 1024 );\r |
1841 | DEBUG (( DEBUG_FILE_BUFFER,\r |
1842 | "%a size: %d.%03d GiBytes (%Ld Bytes)\r\n",\r |
1843 | pFileName,\r |
1844 | Value / 1024,\r |
1845 | (( Value % 1024 ) * 1000 ) / 1024,\r |
1846 | pContext->LengthInBytes ));\r |
1847 | }\r |
1848 | DEBUG_CODE_END ( );\r |
7dc13291 |
1849 | \r |
f6e5cdd5 |
1850 | //\r |
1851 | // Process the options\r |
1852 | //\r |
1853 | if ( pEnd > pOption ) {\r |
1854 | TftpOptions ( pContext, pOption, pEnd );\r |
1855 | }\r |
1856 | else {\r |
1857 | //\r |
1858 | // Skip the open ACK\r |
1859 | //\r |
1860 | pContext->BlockNumber = 1;\r |
1861 | }\r |
7dc13291 |
1862 | \r |
f6e5cdd5 |
1863 | //\r |
1864 | // Send the first packet (OACK or data block)\r |
1865 | //\r |
1866 | bCloseContext = PacketFill ( pContext );\r |
1867 | break;\r |
7dc13291 |
1868 | }\r |
1869 | \r |
1870 | //\r |
f6e5cdd5 |
1871 | // Return the close status\r |
7dc13291 |
1872 | //\r |
f6e5cdd5 |
1873 | DBG_EXIT ( );\r |
1874 | return bCloseContext;\r |
7dc13291 |
1875 | }\r |
1876 | \r |
1877 | \r |
1878 | /**\r |
1879 | Create the port for the TFTP server\r |
1880 | \r |
1881 | This routine polls the network layer to create the TFTP port for the\r |
1882 | TFTP server. More than one attempt may be necessary since it may take\r |
1883 | some time to get the IP address and initialize the upper layers of\r |
1884 | the network stack.\r |
1885 | \r |
f6e5cdd5 |
1886 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
1887 | @param [in] AddressFamily The address family to use for the conection.\r |
1888 | @param [in] pIndex Address of the index into the port array\r |
7dc13291 |
1889 | \r |
1890 | **/\r |
1891 | VOID\r |
f6e5cdd5 |
1892 | TftpServerSocket (\r |
1893 | IN TSDT_TFTP_SERVER * pTftpServer,\r |
1894 | IN sa_family_t AddressFamily,\r |
1895 | IN int * pIndex\r |
7dc13291 |
1896 | )\r |
1897 | {\r |
7dc13291 |
1898 | int SocketStatus;\r |
f6e5cdd5 |
1899 | struct pollfd * pTftpPort;\r |
1900 | UINT16 TftpPort;\r |
1901 | union {\r |
1902 | struct sockaddr_in v4;\r |
1903 | struct sockaddr_in6 v6;\r |
1904 | } TftpServerAddress;\r |
7dc13291 |
1905 | \r |
f6e5cdd5 |
1906 | DEBUG (( DEBUG_SERVER_TIMER, "Entering TftpServerListen\r\n" ));\r |
7dc13291 |
1907 | \r |
1908 | //\r |
f6e5cdd5 |
1909 | // Determine if the socket is already initialized\r |
7dc13291 |
1910 | //\r |
f6e5cdd5 |
1911 | if ( -1 == *pIndex ) {\r |
7dc13291 |
1912 | //\r |
1913 | // Attempt to create the socket for the TFTP server\r |
1914 | //\r |
f6e5cdd5 |
1915 | pTftpPort = &pTftpServer->TftpPort[ pTftpServer->Entries ];\r |
1916 | pTftpPort->fd = socket ( AddressFamily,\r |
1917 | SOCK_DGRAM,\r |
1918 | IPPROTO_UDP );\r |
1919 | if ( -1 != pTftpPort->fd ) {\r |
1920 | //\r |
1921 | // Initialize the poll structure\r |
1922 | //\r |
1923 | pTftpPort->events = POLLRDNORM | POLLHUP;\r |
1924 | pTftpPort->revents = 0;\r |
1925 | \r |
7dc13291 |
1926 | //\r |
1927 | // Set the socket address\r |
1928 | //\r |
7dc13291 |
1929 | TftpPort = 69;\r |
f6e5cdd5 |
1930 | ZeroMem ( &TftpServerAddress, sizeof ( TftpServerAddress ));\r |
1931 | TftpServerAddress.v4.sin_port = htons ( TftpPort );\r |
1932 | if ( AF_INET == AddressFamily ) {\r |
1933 | TftpServerAddress.v4.sin_len = sizeof ( TftpServerAddress.v4 );\r |
1934 | TftpServerAddress.v4.sin_family = AF_INET;\r |
1935 | }\r |
1936 | else {\r |
1937 | TftpServerAddress.v6.sin6_len = sizeof ( TftpServerAddress.v6 );\r |
1938 | TftpServerAddress.v6.sin6_family = AF_INET6;\r |
1939 | }\r |
7dc13291 |
1940 | \r |
1941 | //\r |
1942 | // Bind the socket to the TFTP port\r |
1943 | //\r |
f6e5cdd5 |
1944 | SocketStatus = bind ( pTftpPort->fd,\r |
1945 | (struct sockaddr *) &TftpServerAddress,\r |
1946 | TftpServerAddress.v6.sin6_len );\r |
7dc13291 |
1947 | if ( -1 != SocketStatus ) {\r |
1948 | DEBUG (( DEBUG_TFTP_PORT,\r |
1949 | "0x%08x: Socket bound to port %d\r\n",\r |
f6e5cdd5 |
1950 | pTftpPort->fd,\r |
7dc13291 |
1951 | TftpPort ));\r |
f6e5cdd5 |
1952 | \r |
1953 | //\r |
1954 | // Account for this connection\r |
1955 | //\r |
1956 | *pIndex = pTftpServer->Entries;\r |
1957 | pTftpServer->Entries += 1;\r |
1958 | ASSERT ( DIM ( pTftpServer->TftpPort ) >= pTftpServer->Entries );\r |
7dc13291 |
1959 | }\r |
1960 | \r |
1961 | //\r |
1962 | // Release the socket if necessary\r |
1963 | //\r |
1964 | if ( -1 == SocketStatus ) {\r |
f6e5cdd5 |
1965 | close ( pTftpPort->fd );\r |
1966 | pTftpPort->fd = -1;\r |
7dc13291 |
1967 | }\r |
1968 | }\r |
f6e5cdd5 |
1969 | }\r |
7dc13291 |
1970 | \r |
f6e5cdd5 |
1971 | DEBUG (( DEBUG_SERVER_TIMER, "Exiting TftpServerListen\r\n" ));\r |
7dc13291 |
1972 | }\r |
1973 | \r |
1974 | \r |
1975 | /**\r |
f6e5cdd5 |
1976 | Update the window due to the ACK\r |
7dc13291 |
1977 | \r |
f6e5cdd5 |
1978 | @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure\r |
1979 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
1980 | @param [in] pPacket Address of a ::TFTP_PACKET structure\r |
7dc13291 |
1981 | \r |
1982 | **/\r |
f6e5cdd5 |
1983 | VOID\r |
1984 | WindowAck (\r |
1985 | IN TSDT_TFTP_SERVER * pTftpServer,\r |
1986 | IN TSDT_CONNECTION_CONTEXT * pContext,\r |
1987 | IN TFTP_PACKET * pPacket\r |
7dc13291 |
1988 | )\r |
1989 | {\r |
f6e5cdd5 |
1990 | if ( PcdGetBool ( Tftp_HighSpeed )) {\r |
1991 | UINT64 DeltaTime;\r |
1992 | UINT64 NanoSeconds;\r |
7dc13291 |
1993 | \r |
f6e5cdd5 |
1994 | DBG_ENTER ( );\r |
7dc13291 |
1995 | \r |
7dc13291 |
1996 | //\r |
f6e5cdd5 |
1997 | // Compute the round trip time\r |
7dc13291 |
1998 | //\r |
f6e5cdd5 |
1999 | if ( pTftpServer->Time2 > pTftpServer->Time1 ) {\r |
2000 | DeltaTime = pTftpServer->RxTime - pPacket->TxTime;\r |
7dc13291 |
2001 | }\r |
2002 | else {\r |
f6e5cdd5 |
2003 | DeltaTime = pPacket->TxTime - pTftpServer->RxTime;\r |
7dc13291 |
2004 | }\r |
7dc13291 |
2005 | \r |
f6e5cdd5 |
2006 | //\r |
2007 | // Adjust the round trip time\r |
2008 | //\r |
2009 | NanoSeconds = GetTimeInNanoSecond ( DeltaTime );\r |
2010 | DeltaTime = RShiftU64 ( pContext->Rtt2x, ACK_SHIFT );\r |
2011 | pContext->Rtt2x += NanoSeconds + NanoSeconds - DeltaTime;\r |
2012 | if ( pContext->Rtt2x > pContext->MaxTimeout ) {\r |
2013 | pContext->Rtt2x = pContext->MaxTimeout;\r |
2014 | }\r |
2015 | \r |
2016 | //\r |
2017 | // Account for the ACK\r |
2018 | //\r |
2019 | if ( pContext->WindowSize < MAX_PACKETS ) {\r |
2020 | pContext->AckCount -= 1;\r |
2021 | if ( 0 == pContext->AckCount ) {\r |
2022 | //\r |
2023 | // Increase the window\r |
2024 | //\r |
2025 | pContext->WindowSize += 1;\r |
2026 | \r |
2027 | //\r |
2028 | // Set the ACK count\r |
2029 | //\r |
2030 | if ( pContext->WindowSize < pContext->Threshold ) {\r |
2031 | pContext->AckCount = pContext->WindowSize * PcdGet32 ( Tftp_AckMultiplier );\r |
2032 | }\r |
2033 | else {\r |
2034 | pContext->AckCount = PcdGet32 ( Tftp_AckLogBase ) << pContext->WindowSize;\r |
2035 | }\r |
2036 | \r |
2037 | //\r |
2038 | // Display the round trip time\r |
2039 | //\r |
2040 | DEBUG_CODE_BEGIN ( );\r |
2041 | UINT32 Value;\r |
2042 | \r |
2043 | DeltaTime = RShiftU64 ( pContext->Rtt2x, 1 );\r |
2044 | if ( 1000 > DeltaTime ) {\r |
2045 | DEBUG (( DEBUG_WINDOW,\r |
2046 | "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %Ld nSec\r\n",\r |
2047 | pContext->WindowSize,\r |
2048 | pContext->Threshold,\r |
2049 | pContext->AckCount,\r |
2050 | DeltaTime ));\r |
2051 | }\r |
2052 | else if (( 1000 * 1000 ) > DeltaTime ) {\r |
2053 | Value = (UINT32)DeltaTime;\r |
2054 | DEBUG (( DEBUG_WINDOW,\r |
2055 | "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d uSec\r\n",\r |
2056 | pContext->WindowSize,\r |
2057 | pContext->Threshold,\r |
2058 | pContext->AckCount,\r |
2059 | Value / 1000,\r |
2060 | Value % 1000 ));\r |
2061 | }\r |
2062 | else if (( 1000 * 1000 * 1000 ) > DeltaTime ) {\r |
2063 | Value = (UINT32)DivU64x32 ( DeltaTime, 1000 );\r |
2064 | DEBUG (( DEBUG_WINDOW,\r |
2065 | "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d mSec\r\n",\r |
2066 | pContext->WindowSize,\r |
2067 | pContext->Threshold,\r |
2068 | pContext->AckCount,\r |
2069 | Value / 1000,\r |
2070 | Value % 1000 ));\r |
2071 | }\r |
2072 | else {\r |
2073 | Value = (UINT32)DivU64x32 ( DeltaTime, 1000 * 1000 );\r |
2074 | DEBUG (( DEBUG_WINDOW,\r |
2075 | "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d Sec\r\n",\r |
2076 | pContext->WindowSize,\r |
2077 | pContext->Threshold,\r |
2078 | pContext->AckCount,\r |
2079 | Value / 1000,\r |
2080 | Value % 1000 ));\r |
2081 | }\r |
2082 | DEBUG_CODE_END ( );\r |
2083 | }\r |
2084 | }\r |
2085 | \r |
2086 | DBG_EXIT ( );\r |
2087 | }\r |
7dc13291 |
2088 | }\r |
2089 | \r |
2090 | \r |
2091 | /**\r |
f6e5cdd5 |
2092 | A timeout has occurred, close the window\r |
7dc13291 |
2093 | \r |
f6e5cdd5 |
2094 | @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure\r |
7dc13291 |
2095 | \r |
2096 | **/\r |
f6e5cdd5 |
2097 | VOID\r |
2098 | WindowTimeout (\r |
2099 | IN TSDT_CONNECTION_CONTEXT * pContext\r |
7dc13291 |
2100 | )\r |
2101 | {\r |
f6e5cdd5 |
2102 | if ( PcdGetBool ( Tftp_HighSpeed )) {\r |
2103 | TFTP_PACKET * pPacket;\r |
7dc13291 |
2104 | \r |
f6e5cdd5 |
2105 | DBG_ENTER ( );\r |
7dc13291 |
2106 | \r |
7dc13291 |
2107 | //\r |
f6e5cdd5 |
2108 | // Set the threshold at half the previous window size\r |
7dc13291 |
2109 | //\r |
f6e5cdd5 |
2110 | pContext->Threshold = ( pContext->WindowSize + 1 ) >> 1;\r |
7dc13291 |
2111 | \r |
f6e5cdd5 |
2112 | //\r |
2113 | // Close the transmit window\r |
2114 | //\r |
2115 | pContext->WindowSize = 1;\r |
2116 | pContext->PacketsInWindow = 0;\r |
2117 | \r |
2118 | //\r |
2119 | // Double the round trip time\r |
2120 | //\r |
2121 | pContext->Rtt2x = LShiftU64 ( pContext->Rtt2x, 1 );\r |
2122 | if ( pContext->Rtt2x > pContext->MaxTimeout ) {\r |
2123 | pContext->Rtt2x = pContext->MaxTimeout;\r |
2124 | }\r |
2125 | \r |
2126 | //\r |
2127 | // Set the ACK count\r |
2128 | //\r |
2129 | if ( pContext->WindowSize < pContext->Threshold ) {\r |
2130 | pContext->AckCount = pContext->WindowSize * PcdGet32 ( Tftp_AckMultiplier );\r |
7dc13291 |
2131 | }\r |
2132 | else {\r |
f6e5cdd5 |
2133 | pContext->AckCount = PcdGet32 ( Tftp_AckLogBase ) << pContext->WindowSize;\r |
7dc13291 |
2134 | }\r |
7dc13291 |
2135 | \r |
f6e5cdd5 |
2136 | //\r |
2137 | // Display the round trip time\r |
2138 | //\r |
2139 | DEBUG_CODE_BEGIN ( );\r |
2140 | UINT64 DeltaTime;\r |
2141 | UINT32 Value;\r |
2142 | \r |
2143 | DeltaTime = RShiftU64 ( pContext->Rtt2x, 1 );\r |
2144 | if ( 1000 > DeltaTime ) {\r |
2145 | DEBUG (( DEBUG_WINDOW,\r |
2146 | "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %Ld nSec\r\n",\r |
2147 | pContext->WindowSize,\r |
2148 | pContext->Threshold,\r |
2149 | pContext->AckCount,\r |
2150 | DeltaTime ));\r |
2151 | }\r |
2152 | else if (( 1000 * 1000 ) > DeltaTime ) {\r |
2153 | Value = (UINT32)DeltaTime;\r |
2154 | DEBUG (( DEBUG_WINDOW,\r |
2155 | "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d uSec\r\n",\r |
2156 | pContext->WindowSize,\r |
2157 | pContext->Threshold,\r |
2158 | pContext->AckCount,\r |
2159 | Value / 1000,\r |
2160 | Value % 1000 ));\r |
2161 | }\r |
2162 | else if (( 1000 * 1000 * 1000 ) > DeltaTime ) {\r |
2163 | Value = (UINT32)DivU64x32 ( DeltaTime, 1000 );\r |
2164 | DEBUG (( DEBUG_WINDOW,\r |
2165 | "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d mSec\r\n",\r |
2166 | pContext->WindowSize,\r |
2167 | pContext->Threshold,\r |
2168 | pContext->AckCount,\r |
2169 | Value / 1000,\r |
2170 | Value % 1000 ));\r |
2171 | }\r |
2172 | else {\r |
2173 | Value = (UINT32)DivU64x32 ( DeltaTime, 1000 * 1000 );\r |
2174 | DEBUG (( DEBUG_WINDOW,\r |
2175 | "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d Sec\r\n",\r |
2176 | pContext->WindowSize,\r |
2177 | pContext->Threshold,\r |
2178 | pContext->AckCount,\r |
2179 | Value / 1000,\r |
2180 | Value % 1000 ));\r |
2181 | }\r |
2182 | DEBUG_CODE_END ( );\r |
7dc13291 |
2183 | \r |
f6e5cdd5 |
2184 | //\r |
2185 | // Retransmit the first packet in the window\r |
2186 | //\r |
2187 | pPacket = pContext->pTxHead;\r |
2188 | if ( NULL != pPacket ) {\r |
2189 | PacketTx ( pContext, pPacket );\r |
2190 | }\r |
2191 | \r |
2192 | DBG_EXIT ( );\r |
7dc13291 |
2193 | }\r |
7dc13291 |
2194 | }\r |
2195 | \r |
2196 | \r |
2197 | /**\r |
2198 | Entry point for the TFTP server application.\r |
2199 | \r |
2200 | @param [in] Argc The number of arguments\r |
2201 | @param [in] Argv The argument value array\r |
2202 | \r |
2203 | @retval 0 The application exited normally.\r |
2204 | @retval Other An error occurred.\r |
2205 | **/\r |
2206 | int\r |
2207 | main (\r |
2208 | IN int Argc,\r |
2209 | IN char **Argv\r |
2210 | )\r |
2211 | {\r |
f6e5cdd5 |
2212 | UINTN Index;\r |
7dc13291 |
2213 | TSDT_TFTP_SERVER * pTftpServer;\r |
2214 | EFI_STATUS Status;\r |
f6e5cdd5 |
2215 | UINT64 TriggerTime;\r |
7dc13291 |
2216 | \r |
2217 | //\r |
f6e5cdd5 |
2218 | // Get the performance counter characteristics\r |
7dc13291 |
2219 | //\r |
2220 | pTftpServer = &mTftpServer;\r |
f6e5cdd5 |
2221 | if ( PcdGetBool ( Tftp_HighSpeed )\r |
2222 | || PcdGetBool ( Tftp_Bandwidth )) {\r |
2223 | pTftpServer->ClockFrequency = GetPerformanceCounterProperties ( &pTftpServer->Time1,\r |
2224 | &pTftpServer->Time2 );\r |
2225 | }\r |
2226 | \r |
2227 | //\r |
2228 | // Create a timer event to start TFTP port\r |
2229 | //\r |
7dc13291 |
2230 | Status = gBS->CreateEvent ( EVT_TIMER,\r |
2231 | TPL_TFTP_SERVER,\r |
2232 | NULL,\r |
2233 | NULL,\r |
2234 | &pTftpServer->TimerEvent );\r |
2235 | if ( !EFI_ERROR ( Status )) {\r |
f6e5cdd5 |
2236 | //\r |
2237 | // Compute the poll interval\r |
2238 | //\r |
2239 | TriggerTime = TFTP_PORT_POLL_DELAY * ( 1000 * 10 );\r |
2240 | Status = gBS->SetTimer ( pTftpServer->TimerEvent,\r |
2241 | TimerPeriodic,\r |
2242 | TriggerTime );\r |
7dc13291 |
2243 | if ( !EFI_ERROR ( Status )) {\r |
f6e5cdd5 |
2244 | DEBUG (( DEBUG_TFTP_PORT, "TFTP port timer started\r\n" ));\r |
2245 | \r |
7dc13291 |
2246 | //\r |
2247 | // Run the TFTP server forever\r |
2248 | //\r |
f6e5cdd5 |
2249 | pTftpServer->Udpv4Index = -1;\r |
2250 | pTftpServer->Udpv6Index = -1;\r |
2251 | do {\r |
7dc13291 |
2252 | //\r |
2253 | // Poll the network layer to create the TFTP port\r |
2254 | // for the tftp server. More than one attempt may\r |
2255 | // be necessary since it may take some time to get\r |
2256 | // the IP address and initialize the upper layers\r |
2257 | // of the network stack.\r |
2258 | //\r |
f6e5cdd5 |
2259 | if ( DIM ( pTftpServer->TftpPort ) != pTftpServer->Entries ) {\r |
2260 | do {\r |
2261 | //\r |
2262 | // Wait a while before polling for a connection\r |
2263 | //\r |
2264 | if ( EFI_SUCCESS != gBS->CheckEvent ( pTftpServer->TimerEvent )) {\r |
2265 | if ( 0 == pTftpServer->Entries ) {\r |
2266 | break;\r |
2267 | }\r |
2268 | gBS->WaitForEvent ( 1, &pTftpServer->TimerEvent, &Index );\r |
2269 | }\r |
2270 | \r |
2271 | //\r |
2272 | // Poll for a network connection\r |
2273 | //\r |
2274 | TftpServerSocket ( pTftpServer,\r |
2275 | AF_INET,\r |
2276 | &pTftpServer->Udpv4Index );\r |
2277 | TftpServerSocket ( pTftpServer,\r |
2278 | AF_INET6,\r |
2279 | &pTftpServer->Udpv6Index );\r |
2280 | } while ( 0 == pTftpServer->Entries );\r |
2281 | }\r |
7dc13291 |
2282 | \r |
2283 | //\r |
2284 | // Poll the socket for activity\r |
2285 | //\r |
2286 | do {\r |
2287 | SocketPoll ( pTftpServer );\r |
7dc13291 |
2288 | \r |
f6e5cdd5 |
2289 | //\r |
2290 | // Normal TFTP lets the client request the retransmit by\r |
2291 | // sending another ACK for the previous packet\r |
2292 | //\r |
2293 | if ( PcdGetBool ( Tftp_HighSpeed )) {\r |
2294 | UINT64 CurrentTime;\r |
2295 | UINT64 ElapsedTime;\r |
2296 | TSDT_CONNECTION_CONTEXT * pContext;\r |
2297 | TFTP_PACKET * pPacket;\r |
2298 | \r |
2299 | //\r |
2300 | // High speed TFTP uses an agressive retransmit to\r |
2301 | // get the TFTP client moving again when the ACK or\r |
2302 | // previous data packet was lost.\r |
2303 | //\r |
2304 | // Get the current time\r |
2305 | //\r |
2306 | CurrentTime = GetPerformanceCounter ( );\r |
2307 | \r |
2308 | //\r |
2309 | // Walk the list of contexts\r |
2310 | //\r |
2311 | pContext = pTftpServer->pContextList;\r |
2312 | while ( NULL != pContext )\r |
2313 | {\r |
2314 | //\r |
2315 | // Check for a transmit timeout\r |
2316 | //\r |
2317 | pPacket = pContext->pTxHead;\r |
2318 | if ( NULL != pPacket ) {\r |
2319 | //\r |
2320 | // Compute the elapsed time\r |
2321 | //\r |
2322 | if ( pTftpServer->Time2 > pTftpServer->Time1 ) {\r |
2323 | ElapsedTime = CurrentTime - pPacket->TxTime;\r |
2324 | }\r |
2325 | else {\r |
2326 | ElapsedTime = pPacket->TxTime - CurrentTime;\r |
2327 | }\r |
2328 | ElapsedTime = GetTimeInNanoSecond ( ElapsedTime );\r |
2329 | \r |
2330 | //\r |
2331 | // Determine if a retransmission is necessary\r |
2332 | //\r |
2333 | if ( ElapsedTime >= pContext->Rtt2x ) {\r |
2334 | DEBUG (( DEBUG_WINDOW,\r |
2335 | "0x%08x: Context TX timeout for packet 0x%08x, Window: %d\r\n",\r |
2336 | pContext,\r |
2337 | pPacket,\r |
2338 | pContext->WindowSize ));\r |
2339 | WindowTimeout ( pContext );\r |
2340 | }\r |
2341 | }\r |
2342 | \r |
2343 | //\r |
2344 | // Set the next context\r |
2345 | //\r |
2346 | pContext = pContext->pNext;\r |
2347 | }\r |
2348 | }\r |
2349 | } while ( DIM ( pTftpServer->TftpPort ) == pTftpServer->Entries );\r |
2350 | } while ( !mbTftpServerExit );\r |
7dc13291 |
2351 | \r |
2352 | //\r |
2353 | // Done with the timer event\r |
2354 | //\r |
f6e5cdd5 |
2355 | gBS->SetTimer ( pTftpServer->TimerEvent,\r |
2356 | TimerCancel,\r |
2357 | 0 );\r |
7dc13291 |
2358 | }\r |
f6e5cdd5 |
2359 | gBS->CloseEvent ( pTftpServer->TimerEvent );\r |
7dc13291 |
2360 | }\r |
2361 | \r |
2362 | //\r |
2363 | // Return the final status\r |
2364 | //\r |
2365 | DBG_EXIT_STATUS ( Status );\r |
2366 | return Status;\r |
2367 | }\r |