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