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 }