]> git.ipfire.org Git - people/ms/u-boot.git/blob - net/nfs.c
net: nfs: Share the file handle buffer for v2 / v3
[people/ms/u-boot.git] / net / nfs.c
1 /*
2 * NFS support driver - based on etherboot and U-BOOT's tftp.c
3 *
4 * Masami Komiya <mkomiya@sonare.it> 2004
5 *
6 */
7
8 /* NOTE: the NFS code is heavily inspired by the NetBSD netboot code (read:
9 * large portions are copied verbatim) as distributed in OSKit 0.97. A few
10 * changes were necessary to adapt the code to Etherboot and to fix several
11 * inconsistencies. Also the RPC message preparation is done "by hand" to
12 * avoid adding netsprintf() which I find hard to understand and use. */
13
14 /* NOTE 2: Etherboot does not care about things beyond the kernel image, so
15 * it loads the kernel image off the boot server (ARP_SERVER) and does not
16 * access the client root disk (root-path in dhcpd.conf), which would use
17 * ARP_ROOTSERVER. The root disk is something the operating system we are
18 * about to load needs to use. This is different from the OSKit 0.97 logic. */
19
20 /* NOTE 3: Symlink handling introduced by Anselm M Hoffmeister, 2003-July-14
21 * If a symlink is encountered, it is followed as far as possible (recursion
22 * possible, maximum 16 steps). There is no clearing of ".."'s inside the
23 * path, so please DON'T DO THAT. thx. */
24
25 /* NOTE 4: NFSv3 support added by Guillaume GARDET, 2016-June-20.
26 * NFSv2 is still used by default. But if server does not support NFSv2, then
27 * NFSv3 is used, if available on NFS server. */
28
29 #include <common.h>
30 #include <command.h>
31 #include <net.h>
32 #include <malloc.h>
33 #include <mapmem.h>
34 #include "nfs.h"
35 #include "bootp.h"
36
37 #define HASHES_PER_LINE 65 /* Number of "loading" hashes per line */
38 #define NFS_RETRY_COUNT 30
39 #ifndef CONFIG_NFS_TIMEOUT
40 # define NFS_TIMEOUT 2000UL
41 #else
42 # define NFS_TIMEOUT CONFIG_NFS_TIMEOUT
43 #endif
44
45 #define NFS_RPC_ERR 1
46 #define NFS_RPC_DROP 124
47
48 static int fs_mounted;
49 static unsigned long rpc_id;
50 static int nfs_offset = -1;
51 static int nfs_len;
52 static ulong nfs_timeout = NFS_TIMEOUT;
53
54 static char dirfh[NFS_FHSIZE]; /* NFSv2 / NFSv3 file handle of directory */
55 static char filefh[NFS3_FHSIZE]; /* NFSv2 / NFSv3 file handle */
56 static int filefh3_length; /* (variable) length of filefh when NFSv3 */
57
58 static enum net_loop_state nfs_download_state;
59 static struct in_addr nfs_server_ip;
60 static int nfs_server_mount_port;
61 static int nfs_server_port;
62 static int nfs_our_port;
63 static int nfs_timeout_count;
64 static int nfs_state;
65 #define STATE_PRCLOOKUP_PROG_MOUNT_REQ 1
66 #define STATE_PRCLOOKUP_PROG_NFS_REQ 2
67 #define STATE_MOUNT_REQ 3
68 #define STATE_UMOUNT_REQ 4
69 #define STATE_LOOKUP_REQ 5
70 #define STATE_READ_REQ 6
71 #define STATE_READLINK_REQ 7
72
73 static char *nfs_filename;
74 static char *nfs_path;
75 static char nfs_path_buff[2048];
76
77 #define NFSV2_FLAG 1
78 #define NFSV3_FLAG 1 << 1
79 static char supported_nfs_versions = NFSV2_FLAG | NFSV3_FLAG;
80
81 static inline int store_block(uchar *src, unsigned offset, unsigned len)
82 {
83 ulong newsize = offset + len;
84 #ifdef CONFIG_SYS_DIRECT_FLASH_NFS
85 int i, rc = 0;
86
87 for (i = 0; i < CONFIG_SYS_MAX_FLASH_BANKS; i++) {
88 /* start address in flash? */
89 if (load_addr + offset >= flash_info[i].start[0]) {
90 rc = 1;
91 break;
92 }
93 }
94
95 if (rc) { /* Flash is destination for this packet */
96 rc = flash_write((uchar *)src, (ulong)(load_addr+offset), len);
97 if (rc) {
98 flash_perror(rc);
99 return -1;
100 }
101 } else
102 #endif /* CONFIG_SYS_DIRECT_FLASH_NFS */
103 {
104 void *ptr = map_sysmem(load_addr + offset, len);
105
106 memcpy(ptr, src, len);
107 unmap_sysmem(ptr);
108 }
109
110 if (net_boot_file_size < (offset + len))
111 net_boot_file_size = newsize;
112 return 0;
113 }
114
115 static char *basename(char *path)
116 {
117 char *fname;
118
119 fname = path + strlen(path) - 1;
120 while (fname >= path) {
121 if (*fname == '/') {
122 fname++;
123 break;
124 }
125 fname--;
126 }
127 return fname;
128 }
129
130 static char *dirname(char *path)
131 {
132 char *fname;
133
134 fname = basename(path);
135 --fname;
136 *fname = '\0';
137 return path;
138 }
139
140 /**************************************************************************
141 RPC_ADD_CREDENTIALS - Add RPC authentication/verifier entries
142 **************************************************************************/
143 static uint32_t *rpc_add_credentials(uint32_t *p)
144 {
145 int hl;
146 int hostnamelen;
147 char hostname[256];
148
149 strcpy(hostname, "");
150 hostnamelen = strlen(hostname);
151
152 /* Here's the executive summary on authentication requirements of the
153 * various NFS server implementations: Linux accepts both AUTH_NONE
154 * and AUTH_UNIX authentication (also accepts an empty hostname field
155 * in the AUTH_UNIX scheme). *BSD refuses AUTH_NONE, but accepts
156 * AUTH_UNIX (also accepts an empty hostname field in the AUTH_UNIX
157 * scheme). To be safe, use AUTH_UNIX and pass the hostname if we have
158 * it (if the BOOTP/DHCP reply didn't give one, just use an empty
159 * hostname). */
160
161 hl = (hostnamelen + 3) & ~3;
162
163 /* Provide an AUTH_UNIX credential. */
164 *p++ = htonl(1); /* AUTH_UNIX */
165 *p++ = htonl(hl+20); /* auth length */
166 *p++ = htonl(0); /* stamp */
167 *p++ = htonl(hostnamelen); /* hostname string */
168 if (hostnamelen & 3)
169 *(p + hostnamelen / 4) = 0; /* add zero padding */
170 memcpy(p, hostname, hostnamelen);
171 p += hl / 4;
172 *p++ = 0; /* uid */
173 *p++ = 0; /* gid */
174 *p++ = 0; /* auxiliary gid list */
175
176 /* Provide an AUTH_NONE verifier. */
177 *p++ = 0; /* AUTH_NONE */
178 *p++ = 0; /* auth length */
179
180 return p;
181 }
182
183 /**************************************************************************
184 RPC_LOOKUP - Lookup RPC Port numbers
185 **************************************************************************/
186 static void rpc_req(int rpc_prog, int rpc_proc, uint32_t *data, int datalen)
187 {
188 struct rpc_t pkt;
189 unsigned long id;
190 uint32_t *p;
191 int pktlen;
192 int sport;
193
194 id = ++rpc_id;
195 pkt.u.call.id = htonl(id);
196 pkt.u.call.type = htonl(MSG_CALL);
197 pkt.u.call.rpcvers = htonl(2); /* use RPC version 2 */
198 pkt.u.call.prog = htonl(rpc_prog);
199 switch (rpc_prog) {
200 case PROG_NFS:
201 if (supported_nfs_versions & NFSV2_FLAG)
202 pkt.u.call.vers = htonl(2); /* NFS v2 */
203 else /* NFSV3_FLAG */
204 pkt.u.call.vers = htonl(3); /* NFS v3 */
205 break;
206 case PROG_PORTMAP:
207 case PROG_MOUNT:
208 default:
209 pkt.u.call.vers = htonl(2); /* portmapper is version 2 */
210 }
211 pkt.u.call.proc = htonl(rpc_proc);
212 p = (uint32_t *)&(pkt.u.call.data);
213
214 if (datalen)
215 memcpy((char *)p, (char *)data, datalen*sizeof(uint32_t));
216
217 pktlen = (char *)p + datalen*sizeof(uint32_t) - (char *)&pkt;
218
219 memcpy((char *)net_tx_packet + net_eth_hdr_size() + IP_UDP_HDR_SIZE,
220 (char *)&pkt, pktlen);
221
222 if (rpc_prog == PROG_PORTMAP)
223 sport = SUNRPC_PORT;
224 else if (rpc_prog == PROG_MOUNT)
225 sport = nfs_server_mount_port;
226 else
227 sport = nfs_server_port;
228
229 net_send_udp_packet(net_server_ethaddr, nfs_server_ip, sport,
230 nfs_our_port, pktlen);
231 }
232
233 /**************************************************************************
234 RPC_LOOKUP - Lookup RPC Port numbers
235 **************************************************************************/
236 static void rpc_lookup_req(int prog, int ver)
237 {
238 uint32_t data[16];
239
240 data[0] = 0; data[1] = 0; /* auth credential */
241 data[2] = 0; data[3] = 0; /* auth verifier */
242 data[4] = htonl(prog);
243 data[5] = htonl(ver);
244 data[6] = htonl(17); /* IP_UDP */
245 data[7] = 0;
246 rpc_req(PROG_PORTMAP, PORTMAP_GETPORT, data, 8);
247 }
248
249 /**************************************************************************
250 NFS_MOUNT - Mount an NFS Filesystem
251 **************************************************************************/
252 static void nfs_mount_req(char *path)
253 {
254 uint32_t data[1024];
255 uint32_t *p;
256 int len;
257 int pathlen;
258
259 pathlen = strlen(path);
260
261 p = &(data[0]);
262 p = rpc_add_credentials(p);
263
264 *p++ = htonl(pathlen);
265 if (pathlen & 3)
266 *(p + pathlen / 4) = 0;
267 memcpy(p, path, pathlen);
268 p += (pathlen + 3) / 4;
269
270 len = (uint32_t *)p - (uint32_t *)&(data[0]);
271
272 rpc_req(PROG_MOUNT, MOUNT_ADDENTRY, data, len);
273 }
274
275 /**************************************************************************
276 NFS_UMOUNTALL - Unmount all our NFS Filesystems on the Server
277 **************************************************************************/
278 static void nfs_umountall_req(void)
279 {
280 uint32_t data[1024];
281 uint32_t *p;
282 int len;
283
284 if ((nfs_server_mount_port == -1) || (!fs_mounted))
285 /* Nothing mounted, nothing to umount */
286 return;
287
288 p = &(data[0]);
289 p = rpc_add_credentials(p);
290
291 len = (uint32_t *)p - (uint32_t *)&(data[0]);
292
293 rpc_req(PROG_MOUNT, MOUNT_UMOUNTALL, data, len);
294 }
295
296 /***************************************************************************
297 * NFS_READLINK (AH 2003-07-14)
298 * This procedure is called when read of the first block fails -
299 * this probably happens when it's a directory or a symlink
300 * In case of successful readlink(), the dirname is manipulated,
301 * so that inside the nfs() function a recursion can be done.
302 **************************************************************************/
303 static void nfs_readlink_req(void)
304 {
305 uint32_t data[1024];
306 uint32_t *p;
307 int len;
308
309 p = &(data[0]);
310 p = rpc_add_credentials(p);
311
312 if (supported_nfs_versions & NFSV2_FLAG) {
313 memcpy(p, filefh, NFS_FHSIZE);
314 p += (NFS_FHSIZE / 4);
315 } else { /* NFSV3_FLAG */
316 *p++ = htonl(filefh3_length);
317 memcpy(p, filefh, filefh3_length);
318 p += (filefh3_length / 4);
319 }
320
321 len = (uint32_t *)p - (uint32_t *)&(data[0]);
322
323 rpc_req(PROG_NFS, NFS_READLINK, data, len);
324 }
325
326 /**************************************************************************
327 NFS_LOOKUP - Lookup Pathname
328 **************************************************************************/
329 static void nfs_lookup_req(char *fname)
330 {
331 uint32_t data[1024];
332 uint32_t *p;
333 int len;
334 int fnamelen;
335
336 fnamelen = strlen(fname);
337
338 p = &(data[0]);
339 p = rpc_add_credentials(p);
340
341 if (supported_nfs_versions & NFSV2_FLAG) {
342 memcpy(p, dirfh, NFS_FHSIZE);
343 p += (NFS_FHSIZE / 4);
344 *p++ = htonl(fnamelen);
345 if (fnamelen & 3)
346 *(p + fnamelen / 4) = 0;
347 memcpy(p, fname, fnamelen);
348 p += (fnamelen + 3) / 4;
349
350 len = (uint32_t *)p - (uint32_t *)&(data[0]);
351
352 rpc_req(PROG_NFS, NFS_LOOKUP, data, len);
353 } else { /* NFSV3_FLAG */
354 *p++ = htonl(NFS_FHSIZE); /* Dir handle length */
355 memcpy(p, dirfh, NFS_FHSIZE);
356 p += (NFS_FHSIZE / 4);
357 *p++ = htonl(fnamelen);
358 if (fnamelen & 3)
359 *(p + fnamelen / 4) = 0;
360 memcpy(p, fname, fnamelen);
361 p += (fnamelen + 3) / 4;
362
363 len = (uint32_t *)p - (uint32_t *)&(data[0]);
364
365 rpc_req(PROG_NFS, NFS3PROC_LOOKUP, data, len);
366 }
367 }
368
369 /**************************************************************************
370 NFS_READ - Read File on NFS Server
371 **************************************************************************/
372 static void nfs_read_req(int offset, int readlen)
373 {
374 uint32_t data[1024];
375 uint32_t *p;
376 int len;
377
378 p = &(data[0]);
379 p = rpc_add_credentials(p);
380
381 if (supported_nfs_versions & NFSV2_FLAG) {
382 memcpy(p, filefh, NFS_FHSIZE);
383 p += (NFS_FHSIZE / 4);
384 *p++ = htonl(offset);
385 *p++ = htonl(readlen);
386 *p++ = 0;
387 } else { /* NFSV3_FLAG */
388 *p++ = htonl(filefh3_length);
389 memcpy(p, filefh, filefh3_length);
390 p += (filefh3_length / 4);
391 *p++ = htonl(0); /* offset is 64-bit long, so fill with 0 */
392 *p++ = htonl(offset);
393 *p++ = htonl(readlen);
394 *p++ = 0;
395 }
396
397 len = (uint32_t *)p - (uint32_t *)&(data[0]);
398
399 rpc_req(PROG_NFS, NFS_READ, data, len);
400 }
401
402 /**************************************************************************
403 RPC request dispatcher
404 **************************************************************************/
405 static void nfs_send(void)
406 {
407 debug("%s\n", __func__);
408
409 switch (nfs_state) {
410 case STATE_PRCLOOKUP_PROG_MOUNT_REQ:
411 if (supported_nfs_versions & NFSV2_FLAG)
412 rpc_lookup_req(PROG_MOUNT, 1);
413 else /* NFSV3_FLAG */
414 rpc_lookup_req(PROG_MOUNT, 3);
415 break;
416 case STATE_PRCLOOKUP_PROG_NFS_REQ:
417 if (supported_nfs_versions & NFSV2_FLAG)
418 rpc_lookup_req(PROG_NFS, 2);
419 else /* NFSV3_FLAG */
420 rpc_lookup_req(PROG_NFS, 3);
421 break;
422 case STATE_MOUNT_REQ:
423 nfs_mount_req(nfs_path);
424 break;
425 case STATE_UMOUNT_REQ:
426 nfs_umountall_req();
427 break;
428 case STATE_LOOKUP_REQ:
429 nfs_lookup_req(nfs_filename);
430 break;
431 case STATE_READ_REQ:
432 nfs_read_req(nfs_offset, nfs_len);
433 break;
434 case STATE_READLINK_REQ:
435 nfs_readlink_req();
436 break;
437 }
438 }
439
440 /**************************************************************************
441 Handlers for the reply from server
442 **************************************************************************/
443
444 static int rpc_lookup_reply(int prog, uchar *pkt, unsigned len)
445 {
446 struct rpc_t rpc_pkt;
447
448 memcpy((unsigned char *)&rpc_pkt, pkt, len);
449
450 debug("%s\n", __func__);
451
452 if (ntohl(rpc_pkt.u.reply.id) > rpc_id)
453 return -NFS_RPC_ERR;
454 else if (ntohl(rpc_pkt.u.reply.id) < rpc_id)
455 return -NFS_RPC_DROP;
456
457 if (rpc_pkt.u.reply.rstatus ||
458 rpc_pkt.u.reply.verifier ||
459 rpc_pkt.u.reply.astatus)
460 return -1;
461
462 switch (prog) {
463 case PROG_MOUNT:
464 nfs_server_mount_port = ntohl(rpc_pkt.u.reply.data[0]);
465 break;
466 case PROG_NFS:
467 nfs_server_port = ntohl(rpc_pkt.u.reply.data[0]);
468 break;
469 }
470
471 return 0;
472 }
473
474 static int nfs_mount_reply(uchar *pkt, unsigned len)
475 {
476 struct rpc_t rpc_pkt;
477
478 debug("%s\n", __func__);
479
480 memcpy((unsigned char *)&rpc_pkt, pkt, len);
481
482 if (ntohl(rpc_pkt.u.reply.id) > rpc_id)
483 return -NFS_RPC_ERR;
484 else if (ntohl(rpc_pkt.u.reply.id) < rpc_id)
485 return -NFS_RPC_DROP;
486
487 if (rpc_pkt.u.reply.rstatus ||
488 rpc_pkt.u.reply.verifier ||
489 rpc_pkt.u.reply.astatus ||
490 rpc_pkt.u.reply.data[0])
491 return -1;
492
493 fs_mounted = 1;
494 /* NFSv2 and NFSv3 use same structure */
495 memcpy(dirfh, rpc_pkt.u.reply.data + 1, NFS_FHSIZE);
496
497 return 0;
498 }
499
500 static int nfs_umountall_reply(uchar *pkt, unsigned len)
501 {
502 struct rpc_t rpc_pkt;
503
504 debug("%s\n", __func__);
505
506 memcpy((unsigned char *)&rpc_pkt, pkt, len);
507
508 if (ntohl(rpc_pkt.u.reply.id) > rpc_id)
509 return -NFS_RPC_ERR;
510 else if (ntohl(rpc_pkt.u.reply.id) < rpc_id)
511 return -NFS_RPC_DROP;
512
513 if (rpc_pkt.u.reply.rstatus ||
514 rpc_pkt.u.reply.verifier ||
515 rpc_pkt.u.reply.astatus)
516 return -1;
517
518 fs_mounted = 0;
519 memset(dirfh, 0, sizeof(dirfh));
520
521 return 0;
522 }
523
524 static int nfs_lookup_reply(uchar *pkt, unsigned len)
525 {
526 struct rpc_t rpc_pkt;
527
528 debug("%s\n", __func__);
529
530 memcpy((unsigned char *)&rpc_pkt, pkt, len);
531
532 if (ntohl(rpc_pkt.u.reply.id) > rpc_id)
533 return -NFS_RPC_ERR;
534 else if (ntohl(rpc_pkt.u.reply.id) < rpc_id)
535 return -NFS_RPC_DROP;
536
537 if (rpc_pkt.u.reply.rstatus ||
538 rpc_pkt.u.reply.verifier ||
539 rpc_pkt.u.reply.astatus ||
540 rpc_pkt.u.reply.data[0]) {
541 switch (ntohl(rpc_pkt.u.reply.astatus)) {
542 case NFS_RPC_SUCCESS: /* Not an error */
543 break;
544 case NFS_RPC_PROG_MISMATCH:
545 /* Remote can't support NFS version */
546 switch (ntohl(rpc_pkt.u.reply.data[0])) {
547 /* Minimal supported NFS version */
548 case 3:
549 debug("*** Waring: NFS version not supported: Requested: V%d, accepted: min V%d - max V%d\n",
550 (supported_nfs_versions & NFSV2_FLAG) ? 2 : 3,
551 ntohl(rpc_pkt.u.reply.data[0]),
552 ntohl(rpc_pkt.u.reply.data[1]));
553 debug("Will retry with NFSv3\n");
554 /* Clear NFSV2_FLAG from supported versions */
555 supported_nfs_versions &= ~NFSV2_FLAG;
556 return -NFS_RPC_PROG_MISMATCH;
557 case 4:
558 default:
559 printf("*** ERROR: NFS version not supported: Requested: V%d, accepted: min V%d - max V%d\n",
560 (supported_nfs_versions & NFSV2_FLAG) ? 2 : 3,
561 ntohl(rpc_pkt.u.reply.data[0]),
562 ntohl(rpc_pkt.u.reply.data[1]));
563 }
564 break;
565 case NFS_RPC_PROG_UNAVAIL:
566 case NFS_RPC_PROC_UNAVAIL:
567 case NFS_RPC_GARBAGE_ARGS:
568 case NFS_RPC_SYSTEM_ERR:
569 default: /* Unknown error on 'accept state' flag */
570 printf("*** ERROR: accept state error (%d)\n",
571 ntohl(rpc_pkt.u.reply.astatus));
572 break;
573 }
574 return -1;
575 }
576
577 if (supported_nfs_versions & NFSV2_FLAG) {
578 memcpy(filefh, rpc_pkt.u.reply.data + 1, NFS_FHSIZE);
579 } else { /* NFSV3_FLAG */
580 filefh3_length = ntohl(rpc_pkt.u.reply.data[1]);
581 if (filefh3_length > NFS3_FHSIZE)
582 filefh3_length = NFS3_FHSIZE;
583 memcpy(filefh, rpc_pkt.u.reply.data + 2, filefh3_length);
584 }
585
586 return 0;
587 }
588
589 static int nfs_readlink_reply(uchar *pkt, unsigned len)
590 {
591 struct rpc_t rpc_pkt;
592 int rlen;
593
594 debug("%s\n", __func__);
595
596 memcpy((unsigned char *)&rpc_pkt, pkt, len);
597
598 if (ntohl(rpc_pkt.u.reply.id) > rpc_id)
599 return -NFS_RPC_ERR;
600 else if (ntohl(rpc_pkt.u.reply.id) < rpc_id)
601 return -NFS_RPC_DROP;
602
603 if (rpc_pkt.u.reply.rstatus ||
604 rpc_pkt.u.reply.verifier ||
605 rpc_pkt.u.reply.astatus ||
606 rpc_pkt.u.reply.data[0])
607 return -1;
608
609 if (supported_nfs_versions & NFSV2_FLAG) {
610
611 rlen = ntohl(rpc_pkt.u.reply.data[1]); /* new path length */
612
613 if (*((char *)&(rpc_pkt.u.reply.data[2])) != '/') {
614 int pathlen;
615 strcat(nfs_path, "/");
616 pathlen = strlen(nfs_path);
617 memcpy(nfs_path + pathlen,
618 (uchar *)&(rpc_pkt.u.reply.data[2]),
619 rlen);
620 nfs_path[pathlen + rlen] = 0;
621 } else {
622 memcpy(nfs_path,
623 (uchar *)&(rpc_pkt.u.reply.data[2]),
624 rlen);
625 nfs_path[rlen] = 0;
626 }
627 } else { /* NFSV3_FLAG */
628 int nfsv3_data_offset = 0;
629 if (ntohl(rpc_pkt.u.reply.data[1]) != 0) {
630 /* 'attributes_follow' flag is TRUE,
631 * so we have attributes on 21 bytes */
632 /* Skip unused values :
633 type; 32 bits value,
634 mode; 32 bits value,
635 nlink; 32 bits value,
636 uid; 32 bits value,
637 gid; 32 bits value,
638 size; 64 bits value,
639 used; 64 bits value,
640 rdev; 64 bits value,
641 fsid; 64 bits value,
642 fileid; 64 bits value,
643 atime; 64 bits value,
644 mtime; 64 bits value,
645 ctime; 64 bits value,
646 */
647 nfsv3_data_offset = 22;
648 } else {
649 /* 'attributes_follow' flag is FALSE,
650 * so we don't have any attributes */
651 nfsv3_data_offset = 1;
652 }
653
654 /* new path length */
655 rlen = ntohl(rpc_pkt.u.reply.data[1+nfsv3_data_offset]);
656
657 if (*((char *)&(rpc_pkt.u.reply.data[2+nfsv3_data_offset])) != '/') {
658 int pathlen;
659 strcat(nfs_path, "/");
660 pathlen = strlen(nfs_path);
661 memcpy(nfs_path + pathlen,
662 (uchar *)&(rpc_pkt.u.reply.data[2+nfsv3_data_offset]),
663 rlen);
664 nfs_path[pathlen + rlen] = 0;
665 } else {
666 memcpy(nfs_path,
667 (uchar *)&(rpc_pkt.u.reply.data[2+nfsv3_data_offset]),
668 rlen);
669 nfs_path[rlen] = 0;
670 }
671 }
672 return 0;
673 }
674
675 static int nfs_read_reply(uchar *pkt, unsigned len)
676 {
677 struct rpc_t rpc_pkt;
678 int rlen;
679 uchar *data_ptr;
680
681 debug("%s\n", __func__);
682
683 memcpy((uchar *)&rpc_pkt, pkt, sizeof(rpc_pkt.u.reply));
684
685 if (ntohl(rpc_pkt.u.reply.id) > rpc_id)
686 return -NFS_RPC_ERR;
687 else if (ntohl(rpc_pkt.u.reply.id) < rpc_id)
688 return -NFS_RPC_DROP;
689
690 if (rpc_pkt.u.reply.rstatus ||
691 rpc_pkt.u.reply.verifier ||
692 rpc_pkt.u.reply.astatus ||
693 rpc_pkt.u.reply.data[0]) {
694 if (rpc_pkt.u.reply.rstatus)
695 return -9999;
696 if (rpc_pkt.u.reply.astatus)
697 return -9999;
698 return -ntohl(rpc_pkt.u.reply.data[0]);
699 }
700
701 if ((nfs_offset != 0) && !((nfs_offset) %
702 (NFS_READ_SIZE / 2 * 10 * HASHES_PER_LINE)))
703 puts("\n\t ");
704 if (!(nfs_offset % ((NFS_READ_SIZE / 2) * 10)))
705 putc('#');
706
707 if (supported_nfs_versions & NFSV2_FLAG) {
708 rlen = ntohl(rpc_pkt.u.reply.data[18]);
709 data_ptr = (uchar *)&(rpc_pkt.u.reply.data[19]);
710 } else { /* NFSV3_FLAG */
711 if (ntohl(rpc_pkt.u.reply.data[1]) != 0) {
712 /* 'attributes_follow' is TRUE,
713 * so we have attributes on 21 bytes */
714 /* Skip unused values :
715 type; 32 bits value,
716 mode; 32 bits value,
717 nlink; 32 bits value,
718 uid; 32 bits value,
719 gid; 32 bits value,
720 size; 64 bits value,
721 used; 64 bits value,
722 rdev; 64 bits value,
723 fsid; 64 bits value,
724 fileid; 64 bits value,
725 atime; 64 bits value,
726 mtime; 64 bits value,
727 ctime; 64 bits value,
728 */
729 rlen = ntohl(rpc_pkt.u.reply.data[23]); /* count value */
730 /* Skip unused values :
731 EOF: 32 bits value,
732 data_size: 32 bits value,
733 */
734 data_ptr = (uchar *)&(rpc_pkt.u.reply.data[26]);
735 } else {
736 /* attributes_follow is FALSE, so we don't have any attributes */
737 rlen = ntohl(rpc_pkt.u.reply.data[2]); /* count value */
738 /* Skip unused values :
739 EOF: 32 bits value,
740 data_size: 32 bits value,
741 */
742 data_ptr = (uchar *)&(rpc_pkt.u.reply.data[5]);
743 }
744 }
745
746 if (store_block(data_ptr, nfs_offset, rlen))
747 return -9999;
748
749 return rlen;
750 }
751
752 /**************************************************************************
753 Interfaces of U-BOOT
754 **************************************************************************/
755 static void nfs_timeout_handler(void)
756 {
757 if (++nfs_timeout_count > NFS_RETRY_COUNT) {
758 puts("\nRetry count exceeded; starting again\n");
759 net_start_again();
760 } else {
761 puts("T ");
762 net_set_timeout_handler(nfs_timeout +
763 NFS_TIMEOUT * nfs_timeout_count,
764 nfs_timeout_handler);
765 nfs_send();
766 }
767 }
768
769 static void nfs_handler(uchar *pkt, unsigned dest, struct in_addr sip,
770 unsigned src, unsigned len)
771 {
772 int rlen;
773 int reply;
774
775 debug("%s\n", __func__);
776
777 if (dest != nfs_our_port)
778 return;
779
780 switch (nfs_state) {
781 case STATE_PRCLOOKUP_PROG_MOUNT_REQ:
782 if (rpc_lookup_reply(PROG_MOUNT, pkt, len) == -NFS_RPC_DROP)
783 break;
784 nfs_state = STATE_PRCLOOKUP_PROG_NFS_REQ;
785 nfs_send();
786 break;
787
788 case STATE_PRCLOOKUP_PROG_NFS_REQ:
789 if (rpc_lookup_reply(PROG_NFS, pkt, len) == -NFS_RPC_DROP)
790 break;
791 nfs_state = STATE_MOUNT_REQ;
792 nfs_send();
793 break;
794
795 case STATE_MOUNT_REQ:
796 reply = nfs_mount_reply(pkt, len);
797 if (reply == -NFS_RPC_DROP) {
798 break;
799 } else if (reply == -NFS_RPC_ERR) {
800 puts("*** ERROR: Cannot mount\n");
801 /* just to be sure... */
802 nfs_state = STATE_UMOUNT_REQ;
803 nfs_send();
804 } else {
805 nfs_state = STATE_LOOKUP_REQ;
806 nfs_send();
807 }
808 break;
809
810 case STATE_UMOUNT_REQ:
811 reply = nfs_umountall_reply(pkt, len);
812 if (reply == -NFS_RPC_DROP) {
813 break;
814 } else if (reply == -NFS_RPC_ERR) {
815 puts("*** ERROR: Cannot umount\n");
816 net_set_state(NETLOOP_FAIL);
817 } else {
818 puts("\ndone\n");
819 net_set_state(nfs_download_state);
820 }
821 break;
822
823 case STATE_LOOKUP_REQ:
824 reply = nfs_lookup_reply(pkt, len);
825 if (reply == -NFS_RPC_DROP) {
826 break;
827 } else if (reply == -NFS_RPC_ERR) {
828 puts("*** ERROR: File lookup fail\n");
829 nfs_state = STATE_UMOUNT_REQ;
830 nfs_send();
831 } else if (reply == -NFS_RPC_PROG_MISMATCH && supported_nfs_versions != 0) {
832 /* umount */
833 nfs_state = STATE_UMOUNT_REQ;
834 nfs_send();
835 /* And retry with another supported version */
836 nfs_state = STATE_PRCLOOKUP_PROG_MOUNT_REQ;
837 nfs_send();
838 } else {
839 nfs_state = STATE_READ_REQ;
840 nfs_offset = 0;
841 nfs_len = NFS_READ_SIZE;
842 nfs_send();
843 }
844 break;
845
846 case STATE_READLINK_REQ:
847 reply = nfs_readlink_reply(pkt, len);
848 if (reply == -NFS_RPC_DROP) {
849 break;
850 } else if (reply == -NFS_RPC_ERR) {
851 puts("*** ERROR: Symlink fail\n");
852 nfs_state = STATE_UMOUNT_REQ;
853 nfs_send();
854 } else {
855 debug("Symlink --> %s\n", nfs_path);
856 nfs_filename = basename(nfs_path);
857 nfs_path = dirname(nfs_path);
858
859 nfs_state = STATE_MOUNT_REQ;
860 nfs_send();
861 }
862 break;
863
864 case STATE_READ_REQ:
865 rlen = nfs_read_reply(pkt, len);
866 net_set_timeout_handler(nfs_timeout, nfs_timeout_handler);
867 if (rlen > 0) {
868 nfs_offset += rlen;
869 nfs_send();
870 } else if ((rlen == -NFSERR_ISDIR) || (rlen == -NFSERR_INVAL)) {
871 /* symbolic link */
872 nfs_state = STATE_READLINK_REQ;
873 nfs_send();
874 } else {
875 if (!rlen)
876 nfs_download_state = NETLOOP_SUCCESS;
877 if (rlen < 0)
878 printf("NFS READ error (%d)\n", rlen);
879 nfs_state = STATE_UMOUNT_REQ;
880 nfs_send();
881 }
882 break;
883 }
884 }
885
886
887 void nfs_start(void)
888 {
889 debug("%s\n", __func__);
890 nfs_download_state = NETLOOP_FAIL;
891
892 nfs_server_ip = net_server_ip;
893 nfs_path = (char *)nfs_path_buff;
894
895 if (nfs_path == NULL) {
896 net_set_state(NETLOOP_FAIL);
897 puts("*** ERROR: Fail allocate memory\n");
898 return;
899 }
900
901 if (net_boot_file_name[0] == '\0') {
902 sprintf(nfs_path, "/nfsroot/%02X%02X%02X%02X.img",
903 net_ip.s_addr & 0xFF,
904 (net_ip.s_addr >> 8) & 0xFF,
905 (net_ip.s_addr >> 16) & 0xFF,
906 (net_ip.s_addr >> 24) & 0xFF);
907
908 printf("*** Warning: no boot file name; using '%s'\n",
909 nfs_path);
910 } else {
911 char *p = net_boot_file_name;
912
913 p = strchr(p, ':');
914
915 if (p != NULL) {
916 nfs_server_ip = string_to_ip(net_boot_file_name);
917 ++p;
918 strcpy(nfs_path, p);
919 } else {
920 strcpy(nfs_path, net_boot_file_name);
921 }
922 }
923
924 nfs_filename = basename(nfs_path);
925 nfs_path = dirname(nfs_path);
926
927 printf("Using %s device\n", eth_get_name());
928
929 printf("File transfer via NFS from server %pI4; our IP address is %pI4",
930 &nfs_server_ip, &net_ip);
931
932 /* Check if we need to send across this subnet */
933 if (net_gateway.s_addr && net_netmask.s_addr) {
934 struct in_addr our_net;
935 struct in_addr server_net;
936
937 our_net.s_addr = net_ip.s_addr & net_netmask.s_addr;
938 server_net.s_addr = net_server_ip.s_addr & net_netmask.s_addr;
939 if (our_net.s_addr != server_net.s_addr)
940 printf("; sending through gateway %pI4",
941 &net_gateway);
942 }
943 printf("\nFilename '%s/%s'.", nfs_path, nfs_filename);
944
945 if (net_boot_file_expected_size_in_blocks) {
946 printf(" Size is 0x%x Bytes = ",
947 net_boot_file_expected_size_in_blocks << 9);
948 print_size(net_boot_file_expected_size_in_blocks << 9, "");
949 }
950 printf("\nLoad address: 0x%lx\n"
951 "Loading: *\b", load_addr);
952
953 net_set_timeout_handler(nfs_timeout, nfs_timeout_handler);
954 net_set_udp_handler(nfs_handler);
955
956 nfs_timeout_count = 0;
957 nfs_state = STATE_PRCLOOKUP_PROG_MOUNT_REQ;
958
959 /*nfs_our_port = 4096 + (get_ticks() % 3072);*/
960 /*FIX ME !!!*/
961 nfs_our_port = 1000;
962
963 /* zero out server ether in case the server ip has changed */
964 memset(net_server_ethaddr, 0, 6);
965
966 nfs_send();
967 }