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) ? 551 2 : 3, 552 ntohl(rpc_pkt.u.reply.data[0]), 553 ntohl(rpc_pkt.u.reply.data[1])); 554 debug("Will retry with NFSv3\n"); 555 /* Clear NFSV2_FLAG from supported versions */ 556 supported_nfs_versions &= ~NFSV2_FLAG; 557 return -NFS_RPC_PROG_MISMATCH; 558 case 4: 559 default: 560 printf("*** ERROR: NFS version not supported: Requested: V%d, accepted: min V%d - max V%d\n", 561 (supported_nfs_versions & NFSV2_FLAG) ? 562 2 : 3, 563 ntohl(rpc_pkt.u.reply.data[0]), 564 ntohl(rpc_pkt.u.reply.data[1])); 565 } 566 break; 567 case NFS_RPC_PROG_UNAVAIL: 568 case NFS_RPC_PROC_UNAVAIL: 569 case NFS_RPC_GARBAGE_ARGS: 570 case NFS_RPC_SYSTEM_ERR: 571 default: /* Unknown error on 'accept state' flag */ 572 printf("*** ERROR: accept state error (%d)\n", 573 ntohl(rpc_pkt.u.reply.astatus)); 574 break; 575 } 576 return -1; 577 } 578 579 if (supported_nfs_versions & NFSV2_FLAG) { 580 memcpy(filefh, rpc_pkt.u.reply.data + 1, NFS_FHSIZE); 581 } else { /* NFSV3_FLAG */ 582 filefh3_length = ntohl(rpc_pkt.u.reply.data[1]); 583 if (filefh3_length > NFS3_FHSIZE) 584 filefh3_length = NFS3_FHSIZE; 585 memcpy(filefh, rpc_pkt.u.reply.data + 2, filefh3_length); 586 } 587 588 return 0; 589 } 590 591 static int nfs_readlink_reply(uchar *pkt, unsigned len) 592 { 593 struct rpc_t rpc_pkt; 594 int rlen; 595 596 debug("%s\n", __func__); 597 598 memcpy((unsigned char *)&rpc_pkt, pkt, len); 599 600 if (ntohl(rpc_pkt.u.reply.id) > rpc_id) 601 return -NFS_RPC_ERR; 602 else if (ntohl(rpc_pkt.u.reply.id) < rpc_id) 603 return -NFS_RPC_DROP; 604 605 if (rpc_pkt.u.reply.rstatus || 606 rpc_pkt.u.reply.verifier || 607 rpc_pkt.u.reply.astatus || 608 rpc_pkt.u.reply.data[0]) 609 return -1; 610 611 if (supported_nfs_versions & NFSV2_FLAG) { 612 613 rlen = ntohl(rpc_pkt.u.reply.data[1]); /* new path length */ 614 615 if (*((char *)&(rpc_pkt.u.reply.data[2])) != '/') { 616 int pathlen; 617 strcat(nfs_path, "/"); 618 pathlen = strlen(nfs_path); 619 memcpy(nfs_path + pathlen, 620 (uchar *)&(rpc_pkt.u.reply.data[2]), 621 rlen); 622 nfs_path[pathlen + rlen] = 0; 623 } else { 624 memcpy(nfs_path, 625 (uchar *)&(rpc_pkt.u.reply.data[2]), 626 rlen); 627 nfs_path[rlen] = 0; 628 } 629 } else { /* NFSV3_FLAG */ 630 int nfsv3_data_offset = 0; 631 if (ntohl(rpc_pkt.u.reply.data[1]) != 0) { 632 /* 'attributes_follow' flag is TRUE, 633 * so we have attributes on 21 bytes */ 634 /* Skip unused values : 635 type; 32 bits value, 636 mode; 32 bits value, 637 nlink; 32 bits value, 638 uid; 32 bits value, 639 gid; 32 bits value, 640 size; 64 bits value, 641 used; 64 bits value, 642 rdev; 64 bits value, 643 fsid; 64 bits value, 644 fileid; 64 bits value, 645 atime; 64 bits value, 646 mtime; 64 bits value, 647 ctime; 64 bits value, 648 */ 649 nfsv3_data_offset = 22; 650 } else { 651 /* 'attributes_follow' flag is FALSE, 652 * so we don't have any attributes */ 653 nfsv3_data_offset = 1; 654 } 655 656 /* new path length */ 657 rlen = ntohl(rpc_pkt.u.reply.data[1+nfsv3_data_offset]); 658 659 if (*((char *)&(rpc_pkt.u.reply.data[2+nfsv3_data_offset])) != '/') { 660 int pathlen; 661 strcat(nfs_path, "/"); 662 pathlen = strlen(nfs_path); 663 memcpy(nfs_path + pathlen, 664 (uchar *)&(rpc_pkt.u.reply.data[2+nfsv3_data_offset]), 665 rlen); 666 nfs_path[pathlen + rlen] = 0; 667 } else { 668 memcpy(nfs_path, 669 (uchar *)&(rpc_pkt.u.reply.data[2+nfsv3_data_offset]), 670 rlen); 671 nfs_path[rlen] = 0; 672 } 673 } 674 return 0; 675 } 676 677 static int nfs_read_reply(uchar *pkt, unsigned len) 678 { 679 struct rpc_t rpc_pkt; 680 int rlen; 681 uchar *data_ptr; 682 683 debug("%s\n", __func__); 684 685 memcpy((uchar *)&rpc_pkt, pkt, sizeof(rpc_pkt.u.reply)); 686 687 if (ntohl(rpc_pkt.u.reply.id) > rpc_id) 688 return -NFS_RPC_ERR; 689 else if (ntohl(rpc_pkt.u.reply.id) < rpc_id) 690 return -NFS_RPC_DROP; 691 692 if (rpc_pkt.u.reply.rstatus || 693 rpc_pkt.u.reply.verifier || 694 rpc_pkt.u.reply.astatus || 695 rpc_pkt.u.reply.data[0]) { 696 if (rpc_pkt.u.reply.rstatus) 697 return -9999; 698 if (rpc_pkt.u.reply.astatus) 699 return -9999; 700 return -ntohl(rpc_pkt.u.reply.data[0]); 701 } 702 703 if ((nfs_offset != 0) && !((nfs_offset) % 704 (NFS_READ_SIZE / 2 * 10 * HASHES_PER_LINE))) 705 puts("\n\t "); 706 if (!(nfs_offset % ((NFS_READ_SIZE / 2) * 10))) 707 putc('#'); 708 709 if (supported_nfs_versions & NFSV2_FLAG) { 710 rlen = ntohl(rpc_pkt.u.reply.data[18]); 711 data_ptr = (uchar *)&(rpc_pkt.u.reply.data[19]); 712 } else { /* NFSV3_FLAG */ 713 if (ntohl(rpc_pkt.u.reply.data[1]) != 0) { 714 /* 'attributes_follow' is TRUE, 715 * so we have attributes on 21 bytes */ 716 /* Skip unused values : 717 type; 32 bits value, 718 mode; 32 bits value, 719 nlink; 32 bits value, 720 uid; 32 bits value, 721 gid; 32 bits value, 722 size; 64 bits value, 723 used; 64 bits value, 724 rdev; 64 bits value, 725 fsid; 64 bits value, 726 fileid; 64 bits value, 727 atime; 64 bits value, 728 mtime; 64 bits value, 729 ctime; 64 bits value, 730 */ 731 rlen = ntohl(rpc_pkt.u.reply.data[23]); /* count value */ 732 /* Skip unused values : 733 EOF: 32 bits value, 734 data_size: 32 bits value, 735 */ 736 data_ptr = (uchar *)&(rpc_pkt.u.reply.data[26]); 737 } else { 738 /* attributes_follow is FALSE, so we don't have any attributes */ 739 rlen = ntohl(rpc_pkt.u.reply.data[2]); /* count value */ 740 /* Skip unused values : 741 EOF: 32 bits value, 742 data_size: 32 bits value, 743 */ 744 data_ptr = (uchar *)&(rpc_pkt.u.reply.data[5]); 745 } 746 } 747 748 if (store_block(data_ptr, nfs_offset, rlen)) 749 return -9999; 750 751 return rlen; 752 } 753 754 /************************************************************************** 755 Interfaces of U-BOOT 756 **************************************************************************/ 757 static void nfs_timeout_handler(void) 758 { 759 if (++nfs_timeout_count > NFS_RETRY_COUNT) { 760 puts("\nRetry count exceeded; starting again\n"); 761 net_start_again(); 762 } else { 763 puts("T "); 764 net_set_timeout_handler(nfs_timeout + 765 NFS_TIMEOUT * nfs_timeout_count, 766 nfs_timeout_handler); 767 nfs_send(); 768 } 769 } 770 771 static void nfs_handler(uchar *pkt, unsigned dest, struct in_addr sip, 772 unsigned src, unsigned len) 773 { 774 int rlen; 775 int reply; 776 777 debug("%s\n", __func__); 778 779 if (dest != nfs_our_port) 780 return; 781 782 switch (nfs_state) { 783 case STATE_PRCLOOKUP_PROG_MOUNT_REQ: 784 if (rpc_lookup_reply(PROG_MOUNT, pkt, len) == -NFS_RPC_DROP) 785 break; 786 nfs_state = STATE_PRCLOOKUP_PROG_NFS_REQ; 787 nfs_send(); 788 break; 789 790 case STATE_PRCLOOKUP_PROG_NFS_REQ: 791 if (rpc_lookup_reply(PROG_NFS, pkt, len) == -NFS_RPC_DROP) 792 break; 793 nfs_state = STATE_MOUNT_REQ; 794 nfs_send(); 795 break; 796 797 case STATE_MOUNT_REQ: 798 reply = nfs_mount_reply(pkt, len); 799 if (reply == -NFS_RPC_DROP) { 800 break; 801 } else if (reply == -NFS_RPC_ERR) { 802 puts("*** ERROR: Cannot mount\n"); 803 /* just to be sure... */ 804 nfs_state = STATE_UMOUNT_REQ; 805 nfs_send(); 806 } else { 807 nfs_state = STATE_LOOKUP_REQ; 808 nfs_send(); 809 } 810 break; 811 812 case STATE_UMOUNT_REQ: 813 reply = nfs_umountall_reply(pkt, len); 814 if (reply == -NFS_RPC_DROP) { 815 break; 816 } else if (reply == -NFS_RPC_ERR) { 817 puts("*** ERROR: Cannot umount\n"); 818 net_set_state(NETLOOP_FAIL); 819 } else { 820 puts("\ndone\n"); 821 net_set_state(nfs_download_state); 822 } 823 break; 824 825 case STATE_LOOKUP_REQ: 826 reply = nfs_lookup_reply(pkt, len); 827 if (reply == -NFS_RPC_DROP) { 828 break; 829 } else if (reply == -NFS_RPC_ERR) { 830 puts("*** ERROR: File lookup fail\n"); 831 nfs_state = STATE_UMOUNT_REQ; 832 nfs_send(); 833 } else if (reply == -NFS_RPC_PROG_MISMATCH && 834 supported_nfs_versions != 0) { 835 /* umount */ 836 nfs_state = STATE_UMOUNT_REQ; 837 nfs_send(); 838 /* And retry with another supported version */ 839 nfs_state = STATE_PRCLOOKUP_PROG_MOUNT_REQ; 840 nfs_send(); 841 } else { 842 nfs_state = STATE_READ_REQ; 843 nfs_offset = 0; 844 nfs_len = NFS_READ_SIZE; 845 nfs_send(); 846 } 847 break; 848 849 case STATE_READLINK_REQ: 850 reply = nfs_readlink_reply(pkt, len); 851 if (reply == -NFS_RPC_DROP) { 852 break; 853 } else if (reply == -NFS_RPC_ERR) { 854 puts("*** ERROR: Symlink fail\n"); 855 nfs_state = STATE_UMOUNT_REQ; 856 nfs_send(); 857 } else { 858 debug("Symlink --> %s\n", nfs_path); 859 nfs_filename = basename(nfs_path); 860 nfs_path = dirname(nfs_path); 861 862 nfs_state = STATE_MOUNT_REQ; 863 nfs_send(); 864 } 865 break; 866 867 case STATE_READ_REQ: 868 rlen = nfs_read_reply(pkt, len); 869 net_set_timeout_handler(nfs_timeout, nfs_timeout_handler); 870 if (rlen > 0) { 871 nfs_offset += rlen; 872 nfs_send(); 873 } else if ((rlen == -NFSERR_ISDIR) || (rlen == -NFSERR_INVAL)) { 874 /* symbolic link */ 875 nfs_state = STATE_READLINK_REQ; 876 nfs_send(); 877 } else { 878 if (!rlen) 879 nfs_download_state = NETLOOP_SUCCESS; 880 if (rlen < 0) 881 printf("NFS READ error (%d)\n", rlen); 882 nfs_state = STATE_UMOUNT_REQ; 883 nfs_send(); 884 } 885 break; 886 } 887 } 888 889 890 void nfs_start(void) 891 { 892 debug("%s\n", __func__); 893 nfs_download_state = NETLOOP_FAIL; 894 895 nfs_server_ip = net_server_ip; 896 nfs_path = (char *)nfs_path_buff; 897 898 if (nfs_path == NULL) { 899 net_set_state(NETLOOP_FAIL); 900 puts("*** ERROR: Fail allocate memory\n"); 901 return; 902 } 903 904 if (net_boot_file_name[0] == '\0') { 905 sprintf(nfs_path, "/nfsroot/%02X%02X%02X%02X.img", 906 net_ip.s_addr & 0xFF, 907 (net_ip.s_addr >> 8) & 0xFF, 908 (net_ip.s_addr >> 16) & 0xFF, 909 (net_ip.s_addr >> 24) & 0xFF); 910 911 printf("*** Warning: no boot file name; using '%s'\n", 912 nfs_path); 913 } else { 914 char *p = net_boot_file_name; 915 916 p = strchr(p, ':'); 917 918 if (p != NULL) { 919 nfs_server_ip = string_to_ip(net_boot_file_name); 920 ++p; 921 strcpy(nfs_path, p); 922 } else { 923 strcpy(nfs_path, net_boot_file_name); 924 } 925 } 926 927 nfs_filename = basename(nfs_path); 928 nfs_path = dirname(nfs_path); 929 930 printf("Using %s device\n", eth_get_name()); 931 932 printf("File transfer via NFS from server %pI4; our IP address is %pI4", 933 &nfs_server_ip, &net_ip); 934 935 /* Check if we need to send across this subnet */ 936 if (net_gateway.s_addr && net_netmask.s_addr) { 937 struct in_addr our_net; 938 struct in_addr server_net; 939 940 our_net.s_addr = net_ip.s_addr & net_netmask.s_addr; 941 server_net.s_addr = net_server_ip.s_addr & net_netmask.s_addr; 942 if (our_net.s_addr != server_net.s_addr) 943 printf("; sending through gateway %pI4", 944 &net_gateway); 945 } 946 printf("\nFilename '%s/%s'.", nfs_path, nfs_filename); 947 948 if (net_boot_file_expected_size_in_blocks) { 949 printf(" Size is 0x%x Bytes = ", 950 net_boot_file_expected_size_in_blocks << 9); 951 print_size(net_boot_file_expected_size_in_blocks << 9, ""); 952 } 953 printf("\nLoad address: 0x%lx\n" 954 "Loading: *\b", load_addr); 955 956 net_set_timeout_handler(nfs_timeout, nfs_timeout_handler); 957 net_set_udp_handler(nfs_handler); 958 959 nfs_timeout_count = 0; 960 nfs_state = STATE_PRCLOOKUP_PROG_MOUNT_REQ; 961 962 /*nfs_our_port = 4096 + (get_ticks() % 3072);*/ 963 /*FIX ME !!!*/ 964 nfs_our_port = 1000; 965 966 /* zero out server ether in case the server ip has changed */ 967 memset(net_server_ethaddr, 0, 6); 968 969 nfs_send(); 970 } 971