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