]>
git.ipfire.org Git - people/ms/u-boot.git/blob - net/tftp.c
2 * Copyright 1994, 1995, 2000 Neil Russell.
4 * Copyright 2000, 2001 DENX Software Engineering, Wolfgang Denk, wd@denx.de
15 #if (CONFIG_COMMANDS & CFG_CMD_NET)
17 #define WELL_KNOWN_PORT 69 /* Well known TFTP port # */
18 #define TIMEOUT 2 /* Seconds to timeout for a lost pkt */
19 #ifndef CONFIG_NET_RETRY_COUNT
20 # define TIMEOUT_COUNT 10 /* # of timeouts before giving up */
22 # define TIMEOUT_COUNT (CONFIG_NET_RETRY_COUNT * 2)
24 /* (for checking the image size) */
25 #define HASHES_PER_LINE 65 /* Number of "loading" hashes per line */
37 static int TftpServerPort
; /* The UDP port at their end */
38 static int TftpOurPort
; /* The UDP port at our end */
39 static int TftpTimeoutCount
;
40 static unsigned TftpBlock
;
41 static unsigned TftpLastBlock
;
45 #define STATE_TOO_LARGE 3
46 #define STATE_BAD_MAGIC 4
48 #define DEFAULT_NAME_LEN (8 + 4 + 1)
49 static char default_filename
[DEFAULT_NAME_LEN
];
50 static char *tftp_filename
;
52 #ifdef CFG_DIRECT_FLASH_TFTP
53 extern flash_info_t flash_info
[CFG_MAX_FLASH_BANKS
];
56 static __inline__
void
57 store_block (unsigned block
, uchar
* src
, unsigned len
)
59 ulong offset
= block
* 512, newsize
= offset
+ len
;
60 #ifdef CFG_DIRECT_FLASH_TFTP
63 for (i
=0; i
<CFG_MAX_FLASH_BANKS
; i
++) {
64 /* start address in flash? */
65 if (load_addr
+ offset
>= flash_info
[i
].start
[0]) {
71 if (rc
) { /* Flash is destination for this packet */
72 rc
= flash_write ((uchar
*)src
, (ulong
)(load_addr
+offset
), len
);
75 NetState
= NETLOOP_FAIL
;
80 #endif /* CFG_DIRECT_FLASH_TFTP */
82 (void)memcpy((void *)(load_addr
+ offset
), src
, len
);
85 if (NetBootFileXferSize
< newsize
)
86 NetBootFileXferSize
= newsize
;
89 static void TftpSend (void);
90 static void TftpTimeout (void);
92 /**********************************************************************/
102 * We will always be sending some sort of packet, so
103 * cobble together the packet headers now.
105 pkt
= NetTxPacket
+ ETHER_HDR_SIZE
+ IP_HDR_SIZE
;
111 *((ushort
*)pkt
)++ = htons(TFTP_RRQ
);
112 strcpy ((char *)pkt
, tftp_filename
);
113 pkt
+= strlen(tftp_filename
) + 1;
114 strcpy ((char *)pkt
, "octet");
115 pkt
+= 5 /*strlen("octet")*/ + 1;
121 *((ushort
*)pkt
)++ = htons(TFTP_ACK
);
122 *((ushort
*)pkt
)++ = htons(TftpBlock
);
126 case STATE_TOO_LARGE
:
128 *((ushort
*)pkt
)++ = htons(TFTP_ERROR
);
129 *((ushort
*)pkt
)++ = htons(3);
130 strcpy ((char *)pkt
, "File too large");
131 pkt
+= 14 /*strlen("File too large")*/ + 1;
135 case STATE_BAD_MAGIC
:
137 *((ushort
*)pkt
)++ = htons(TFTP_ERROR
);
138 *((ushort
*)pkt
)++ = htons(2);
139 strcpy ((char *)pkt
, "File has bad magic");
140 pkt
+= 18 /*strlen("File has bad magic")*/ + 1;
145 NetSetEther (NetTxPacket
, NetServerEther
, PROT_IP
);
146 NetSetIP (NetTxPacket
+ ETHER_HDR_SIZE
, NetServerIP
,
147 TftpServerPort
, TftpOurPort
, len
);
148 NetSendPacket (NetTxPacket
, ETHER_HDR_SIZE
+ IP_HDR_SIZE
+ len
);
153 TftpHandler (uchar
* pkt
, unsigned dest
, unsigned src
, unsigned len
)
157 if (dest
!= TftpOurPort
) {
160 if (TftpState
!= STATE_RRQ
&& src
!= TftpServerPort
) {
168 /* warning: don't use increment (++) in ntohs() macros!! */
169 proto
= *((ushort
*)pkt
)++;
170 switch (ntohs(proto
)) {
183 TftpBlock
= ntohs(*(ushort
*)pkt
);
184 if (((TftpBlock
- 1) % 10) == 0) {
186 } else if ((TftpBlock
% (10 * HASHES_PER_LINE
)) == 0) {
190 if (TftpState
== STATE_RRQ
) {
191 TftpState
= STATE_DATA
;
192 TftpServerPort
= src
;
195 if (TftpBlock
!= 1) { /* Assertion */
196 printf ("\nTFTP error: "
197 "First block is not block 1 (%d)\n"
198 "Starting again\n\n",
205 if (TftpBlock
== TftpLastBlock
) {
207 * Same block again; ignore it.
212 TftpLastBlock
= TftpBlock
;
213 NetSetTimeout (TIMEOUT
* CFG_HZ
, TftpTimeout
);
215 store_block (TftpBlock
- 1, pkt
+ 2, len
);
218 * Acknoledge the block just received, which will prompt
219 * the server for the next one.
225 * We received the whole thing. Try to
229 NetState
= NETLOOP_SUCCESS
;
234 printf ("\nTFTP error: '%s' (%d)\n",
235 pkt
+ 2, ntohs(*(ushort
*)pkt
));
236 puts ("Starting again\n\n");
246 if (++TftpTimeoutCount
>= TIMEOUT_COUNT
) {
247 puts ("\nRetry count exceeded; starting again\n");
251 NetSetTimeout (TIMEOUT
* CFG_HZ
, TftpTimeout
);
261 printf ("\nServer ethernet address %02x:%02x:%02x:%02x:%02x:%02x\n",
271 if (BootFile
[0] == '\0') {
272 IPaddr_t OurIP
= ntohl(NetOurIP
);
274 sprintf(default_filename
, "%02lX%02lX%02lX%02lX.img",
277 (OurIP
>> 16) & 0xFF,
278 (OurIP
>> 24) & 0xFF );
279 tftp_filename
= default_filename
;
281 printf ("*** Warning: no boot file name; using '%s'\n",
284 tftp_filename
= BootFile
;
287 puts ("TFTP from server "); print_IPaddr (NetServerIP
);
288 puts ("; our IP address is "); print_IPaddr (NetOurIP
);
290 /* Check if we need to send across this subnet */
291 if (NetOurGatewayIP
&& NetOurSubnetMask
) {
292 IPaddr_t OurNet
= NetOurIP
& NetOurSubnetMask
;
293 IPaddr_t ServerNet
= NetServerIP
& NetOurSubnetMask
;
295 if (OurNet
!= ServerNet
) {
296 puts ("; sending through gateway ");
297 print_IPaddr (NetOurGatewayIP
) ;
302 printf ("Filename '%s'.", tftp_filename
);
304 if (NetBootFileSize
) {
305 printf (" Size is 0x%x Bytes = ", NetBootFileSize
<<9);
306 print_size (NetBootFileSize
<<9, "");
311 printf ("Load address: 0x%lx\n", load_addr
);
313 puts ("Loading: *\b");
315 NetSetTimeout (TIMEOUT
* CFG_HZ
, TftpTimeout
);
316 NetSetHandler (TftpHandler
);
318 TftpServerPort
= WELL_KNOWN_PORT
;
319 TftpTimeoutCount
= 0;
320 TftpState
= STATE_RRQ
;
321 TftpOurPort
= 1024 + (get_timer(0) % 3072);
326 #endif /* CFG_CMD_NET */