| /* |
| * NFS support driver - based on etherboot and U-BOOT's tftp.c |
| * |
| * Masami Komiya <mkomiya@sonare.it> 2004 |
| * |
| */ |
| |
| /* NOTE: the NFS code is heavily inspired by the NetBSD netboot code (read: |
| * large portions are copied verbatim) as distributed in OSKit 0.97. A few |
| * changes were necessary to adapt the code to Etherboot and to fix several |
| * inconsistencies. Also the RPC message preparation is done "by hand" to |
| * avoid adding netsprintf() which I find hard to understand and use. */ |
| |
| /* NOTE 2: Etherboot does not care about things beyond the kernel image, so |
| * it loads the kernel image off the boot server (ARP_SERVER) and does not |
| * access the client root disk (root-path in dhcpd.conf), which would use |
| * ARP_ROOTSERVER. The root disk is something the operating system we are |
| * about to load needs to use. This is different from the OSKit 0.97 logic. */ |
| |
| /* NOTE 3: Symlink handling introduced by Anselm M Hoffmeister, 2003-July-14 |
| * If a symlink is encountered, it is followed as far as possible (recursion |
| * possible, maximum 16 steps). There is no clearing of ".."'s inside the |
| * path, so please DON'T DO THAT. thx. */ |
| |
| /* NOTE 4: NFSv3 support added by Guillaume GARDET, 2016-June-20. |
| * NFSv2 is still used by default. But if server does not support NFSv2, then |
| * NFSv3 is used, if available on NFS server. */ |
| |
| /* NOTE 5: NFSv1 support added by Christian Gmeiner, Thomas Rienoessl, |
| * September 27, 2018. As of now, NFSv3 is the default choice. If the server |
| * does not support NFSv3, we fall back to versions 2 or 1. */ |
| |
| #include <command.h> |
| #include <display_options.h> |
| #ifdef CONFIG_SYS_DIRECT_FLASH_NFS |
| #include <flash.h> |
| #endif |
| #include <image.h> |
| #include <log.h> |
| #include <net.h> |
| #include <malloc.h> |
| #include <mapmem.h> |
| #include "nfs.h" |
| #include "bootp.h" |
| #include <time.h> |
| |
| #define HASHES_PER_LINE 65 /* Number of "loading" hashes per line */ |
| #define NFS_RETRY_COUNT 30 |
| |
| #define NFS_RPC_ERR 1 |
| #define NFS_RPC_DROP 124 |
| |
| static int fs_mounted; |
| static unsigned long rpc_id; |
| static int nfs_offset = -1; |
| static int nfs_len; |
| static const ulong nfs_timeout = CONFIG_NFS_TIMEOUT; |
| |
| static char dirfh[NFS3_FHSIZE]; /* NFSv2 / NFSv3 file handle of directory */ |
| static unsigned int dirfh3_length; /* (variable) length of dirfh when NFSv3 */ |
| static char filefh[NFS3_FHSIZE]; /* NFSv2 / NFSv3 file handle */ |
| static unsigned int filefh3_length; /* (variable) length of filefh when NFSv3 */ |
| |
| static enum net_loop_state nfs_download_state; |
| static struct in_addr nfs_server_ip; |
| static int nfs_server_mount_port; |
| static int nfs_server_port; |
| static int nfs_our_port; |
| static int nfs_timeout_count; |
| static int nfs_state; |
| #define STATE_PRCLOOKUP_PROG_MOUNT_REQ 1 |
| #define STATE_PRCLOOKUP_PROG_NFS_REQ 2 |
| #define STATE_MOUNT_REQ 3 |
| #define STATE_UMOUNT_REQ 4 |
| #define STATE_LOOKUP_REQ 5 |
| #define STATE_READ_REQ 6 |
| #define STATE_READLINK_REQ 7 |
| |
| static char *nfs_filename; |
| static char *nfs_path; |
| static char nfs_path_buff[2048]; |
| |
| enum nfs_version { |
| NFS_UNKOWN = 0, |
| NFS_V1 = 1, |
| NFS_V2 = 2, |
| NFS_V3 = 3, |
| }; |
| |
| static enum nfs_version choosen_nfs_version = NFS_V3; |
| static inline int store_block(uchar *src, unsigned offset, unsigned len) |
| { |
| ulong newsize = offset + len; |
| #ifdef CONFIG_SYS_DIRECT_FLASH_NFS |
| int i, rc = 0; |
| |
| for (i = 0; i < CONFIG_SYS_MAX_FLASH_BANKS; i++) { |
| /* start address in flash? */ |
| if (image_load_addr + offset >= flash_info[i].start[0]) { |
| rc = 1; |
| break; |
| } |
| } |
| |
| if (rc) { /* Flash is destination for this packet */ |
| rc = flash_write((uchar *)src, (ulong)image_load_addr + offset, |
| len); |
| if (rc) { |
| flash_perror(rc); |
| return -1; |
| } |
| } else |
| #endif /* CONFIG_SYS_DIRECT_FLASH_NFS */ |
| { |
| void *ptr = map_sysmem(image_load_addr + offset, len); |
| |
| memcpy(ptr, src, len); |
| unmap_sysmem(ptr); |
| } |
| |
| if (net_boot_file_size < (offset + len)) |
| net_boot_file_size = newsize; |
| return 0; |
| } |
| |
| static char *basename(char *path) |
| { |
| char *fname; |
| |
| fname = path + strlen(path) - 1; |
| while (fname >= path) { |
| if (*fname == '/') { |
| fname++; |
| break; |
| } |
| fname--; |
| } |
| return fname; |
| } |
| |
| static char *dirname(char *path) |
| { |
| char *fname; |
| |
| fname = basename(path); |
| --fname; |
| *fname = '\0'; |
| return path; |
| } |
| |
| /************************************************************************** |
| RPC_ADD_CREDENTIALS - Add RPC authentication/verifier entries |
| **************************************************************************/ |
| static uint32_t *rpc_add_credentials(uint32_t *p) |
| { |
| /* Here's the executive summary on authentication requirements of the |
| * various NFS server implementations: Linux accepts both AUTH_NONE |
| * and AUTH_UNIX authentication (also accepts an empty hostname field |
| * in the AUTH_UNIX scheme). *BSD refuses AUTH_NONE, but accepts |
| * AUTH_UNIX (also accepts an empty hostname field in the AUTH_UNIX |
| * scheme). To be safe, use AUTH_UNIX and pass the hostname if we have |
| * it (if the BOOTP/DHCP reply didn't give one, just use an empty |
| * hostname). */ |
| |
| /* Provide an AUTH_UNIX credential. */ |
| *p++ = htonl(1); /* AUTH_UNIX */ |
| *p++ = htonl(20); /* auth length */ |
| *p++ = 0; /* stamp */ |
| *p++ = 0; /* hostname string */ |
| *p++ = 0; /* uid */ |
| *p++ = 0; /* gid */ |
| *p++ = 0; /* auxiliary gid list */ |
| |
| /* Provide an AUTH_NONE verifier. */ |
| *p++ = 0; /* AUTH_NONE */ |
| *p++ = 0; /* auth length */ |
| |
| return p; |
| } |
| |
| /************************************************************************** |
| RPC_LOOKUP - Lookup RPC Port numbers |
| **************************************************************************/ |
| static void rpc_req(int rpc_prog, int rpc_proc, uint32_t *data, int datalen) |
| { |
| struct rpc_t rpc_pkt; |
| unsigned long id; |
| uint32_t *p; |
| int pktlen; |
| int sport; |
| |
| id = ++rpc_id; |
| rpc_pkt.u.call.id = htonl(id); |
| rpc_pkt.u.call.type = htonl(MSG_CALL); |
| rpc_pkt.u.call.rpcvers = htonl(2); /* use RPC version 2 */ |
| rpc_pkt.u.call.prog = htonl(rpc_prog); |
| switch (rpc_prog) { |
| case PROG_NFS: |
| switch (choosen_nfs_version) { |
| case NFS_V1: |
| case NFS_V2: |
| rpc_pkt.u.call.vers = htonl(2); |
| break; |
| |
| case NFS_V3: |
| rpc_pkt.u.call.vers = htonl(3); |
| break; |
| |
| case NFS_UNKOWN: |
| /* nothing to do */ |
| break; |
| } |
| break; |
| case PROG_MOUNT: |
| switch (choosen_nfs_version) { |
| case NFS_V1: |
| rpc_pkt.u.call.vers = htonl(1); |
| break; |
| |
| case NFS_V2: |
| rpc_pkt.u.call.vers = htonl(2); |
| break; |
| |
| case NFS_V3: |
| rpc_pkt.u.call.vers = htonl(3); |
| break; |
| |
| case NFS_UNKOWN: |
| /* nothing to do */ |
| break; |
| } |
| break; |
| case PROG_PORTMAP: |
| default: |
| rpc_pkt.u.call.vers = htonl(2); /* portmapper is version 2 */ |
| } |
| rpc_pkt.u.call.proc = htonl(rpc_proc); |
| p = rpc_pkt.u.call.data; |
| |
| if (datalen) |
| memcpy(p, data, datalen * sizeof(uint32_t)); |
| |
| pktlen = (char *)p + datalen * sizeof(uint32_t) - (char *)&rpc_pkt; |
| |
| memcpy((char *)net_tx_packet + net_eth_hdr_size() + IP_UDP_HDR_SIZE, |
| &rpc_pkt.u.data[0], pktlen); |
| |
| if (rpc_prog == PROG_PORTMAP) |
| sport = SUNRPC_PORT; |
| else if (rpc_prog == PROG_MOUNT) |
| sport = nfs_server_mount_port; |
| else |
| sport = nfs_server_port; |
| |
| net_send_udp_packet(net_server_ethaddr, nfs_server_ip, sport, |
| nfs_our_port, pktlen); |
| } |
| |
| /************************************************************************** |
| RPC_LOOKUP - Lookup RPC Port numbers |
| **************************************************************************/ |
| static void rpc_lookup_req(int prog, int ver) |
| { |
| uint32_t data[16]; |
| |
| data[0] = 0; data[1] = 0; /* auth credential */ |
| data[2] = 0; data[3] = 0; /* auth verifier */ |
| data[4] = htonl(prog); |
| data[5] = htonl(ver); |
| data[6] = htonl(17); /* IP_UDP */ |
| data[7] = 0; |
| rpc_req(PROG_PORTMAP, PORTMAP_GETPORT, data, 8); |
| } |
| |
| /************************************************************************** |
| NFS_MOUNT - Mount an NFS Filesystem |
| **************************************************************************/ |
| static void nfs_mount_req(char *path) |
| { |
| uint32_t data[1024]; |
| uint32_t *p; |
| int len; |
| int pathlen; |
| |
| pathlen = strlen(path); |
| |
| p = &(data[0]); |
| p = rpc_add_credentials(p); |
| |
| *p++ = htonl(pathlen); |
| if (pathlen & 3) |
| *(p + pathlen / 4) = 0; |
| memcpy(p, path, pathlen); |
| p += (pathlen + 3) / 4; |
| |
| len = (uint32_t *)p - (uint32_t *)&(data[0]); |
| |
| rpc_req(PROG_MOUNT, MOUNT_ADDENTRY, data, len); |
| } |
| |
| /************************************************************************** |
| NFS_UMOUNTALL - Unmount all our NFS Filesystems on the Server |
| **************************************************************************/ |
| static void nfs_umountall_req(void) |
| { |
| uint32_t data[1024]; |
| uint32_t *p; |
| int len; |
| |
| if ((nfs_server_mount_port == -1) || (!fs_mounted)) |
| /* Nothing mounted, nothing to umount */ |
| return; |
| |
| p = &(data[0]); |
| p = rpc_add_credentials(p); |
| |
| len = (uint32_t *)p - (uint32_t *)&(data[0]); |
| |
| rpc_req(PROG_MOUNT, MOUNT_UMOUNTALL, data, len); |
| } |
| |
| /*************************************************************************** |
| * NFS_READLINK (AH 2003-07-14) |
| * This procedure is called when read of the first block fails - |
| * this probably happens when it's a directory or a symlink |
| * In case of successful readlink(), the dirname is manipulated, |
| * so that inside the nfs() function a recursion can be done. |
| **************************************************************************/ |
| static void nfs_readlink_req(void) |
| { |
| uint32_t data[1024]; |
| uint32_t *p; |
| int len; |
| |
| p = &(data[0]); |
| p = rpc_add_credentials(p); |
| |
| if (choosen_nfs_version != NFS_V3) { |
| memcpy(p, filefh, NFS_FHSIZE); |
| p += (NFS_FHSIZE / 4); |
| } else { /* NFS_V3 */ |
| *p++ = htonl(filefh3_length); |
| memcpy(p, filefh, filefh3_length); |
| p += (filefh3_length / 4); |
| } |
| |
| len = (uint32_t *)p - (uint32_t *)&(data[0]); |
| |
| rpc_req(PROG_NFS, NFS_READLINK, data, len); |
| } |
| |
| /************************************************************************** |
| NFS_LOOKUP - Lookup Pathname |
| **************************************************************************/ |
| static void nfs_lookup_req(char *fname) |
| { |
| uint32_t data[1024]; |
| uint32_t *p; |
| int len; |
| int fnamelen; |
| |
| fnamelen = strlen(fname); |
| |
| p = &(data[0]); |
| p = rpc_add_credentials(p); |
| |
| if (choosen_nfs_version != NFS_V3) { |
| memcpy(p, dirfh, NFS_FHSIZE); |
| p += (NFS_FHSIZE / 4); |
| *p++ = htonl(fnamelen); |
| if (fnamelen & 3) |
| *(p + fnamelen / 4) = 0; |
| memcpy(p, fname, fnamelen); |
| p += (fnamelen + 3) / 4; |
| |
| len = (uint32_t *)p - (uint32_t *)&(data[0]); |
| |
| rpc_req(PROG_NFS, NFS_LOOKUP, data, len); |
| } else { /* NFS_V3 */ |
| *p++ = htonl(dirfh3_length); /* Dir handle length */ |
| memcpy(p, dirfh, dirfh3_length); |
| p += (dirfh3_length / 4); |
| *p++ = htonl(fnamelen); |
| if (fnamelen & 3) |
| *(p + fnamelen / 4) = 0; |
| memcpy(p, fname, fnamelen); |
| p += (fnamelen + 3) / 4; |
| |
| len = (uint32_t *)p - (uint32_t *)&(data[0]); |
| |
| rpc_req(PROG_NFS, NFS3PROC_LOOKUP, data, len); |
| } |
| } |
| |
| /************************************************************************** |
| NFS_READ - Read File on NFS Server |
| **************************************************************************/ |
| static void nfs_read_req(int offset, int readlen) |
| { |
| uint32_t data[1024]; |
| uint32_t *p; |
| int len; |
| |
| p = &(data[0]); |
| p = rpc_add_credentials(p); |
| |
| if (choosen_nfs_version != NFS_V3) { |
| memcpy(p, filefh, NFS_FHSIZE); |
| p += (NFS_FHSIZE / 4); |
| *p++ = htonl(offset); |
| *p++ = htonl(readlen); |
| *p++ = 0; |
| } else { /* NFS_V3 */ |
| *p++ = htonl(filefh3_length); |
| memcpy(p, filefh, filefh3_length); |
| p += (filefh3_length / 4); |
| *p++ = htonl(0); /* offset is 64-bit long, so fill with 0 */ |
| *p++ = htonl(offset); |
| *p++ = htonl(readlen); |
| *p++ = 0; |
| } |
| |
| len = (uint32_t *)p - (uint32_t *)&(data[0]); |
| |
| rpc_req(PROG_NFS, NFS_READ, data, len); |
| } |
| |
| /************************************************************************** |
| RPC request dispatcher |
| **************************************************************************/ |
| static void nfs_send(void) |
| { |
| debug("%s\n", __func__); |
| |
| switch (nfs_state) { |
| case STATE_PRCLOOKUP_PROG_MOUNT_REQ: |
| if (choosen_nfs_version != NFS_V3) |
| rpc_lookup_req(PROG_MOUNT, 1); |
| else /* NFS_V3 */ |
| rpc_lookup_req(PROG_MOUNT, 3); |
| break; |
| case STATE_PRCLOOKUP_PROG_NFS_REQ: |
| if (choosen_nfs_version != NFS_V3) |
| rpc_lookup_req(PROG_NFS, 2); |
| else /* NFS_V3 */ |
| rpc_lookup_req(PROG_NFS, 3); |
| break; |
| case STATE_MOUNT_REQ: |
| nfs_mount_req(nfs_path); |
| break; |
| case STATE_UMOUNT_REQ: |
| nfs_umountall_req(); |
| break; |
| case STATE_LOOKUP_REQ: |
| nfs_lookup_req(nfs_filename); |
| break; |
| case STATE_READ_REQ: |
| nfs_read_req(nfs_offset, nfs_len); |
| break; |
| case STATE_READLINK_REQ: |
| nfs_readlink_req(); |
| break; |
| } |
| } |
| |
| /************************************************************************** |
| Handlers for the reply from server |
| **************************************************************************/ |
| |
| static int rpc_handle_error(struct rpc_t *rpc_pkt) |
| { |
| if (rpc_pkt->u.reply.rstatus || |
| rpc_pkt->u.reply.verifier || |
| rpc_pkt->u.reply.astatus || |
| rpc_pkt->u.reply.data[0]) { |
| switch (ntohl(rpc_pkt->u.reply.astatus)) { |
| case NFS_RPC_SUCCESS: /* Not an error */ |
| break; |
| case NFS_RPC_PROG_MISMATCH: { |
| /* Remote can't support NFS version */ |
| const int min = ntohl(rpc_pkt->u.reply.data[0]); |
| const int max = ntohl(rpc_pkt->u.reply.data[1]); |
| |
| if (max < NFS_V1 || max > NFS_V3 || min > NFS_V3) { |
| puts("*** ERROR: NFS version not supported"); |
| debug(": Requested: V%d, accepted: min V%d - max V%d\n", |
| choosen_nfs_version, |
| ntohl(rpc_pkt->u.reply.data[0]), |
| ntohl(rpc_pkt->u.reply.data[1])); |
| puts("\n"); |
| choosen_nfs_version = NFS_UNKOWN; |
| break; |
| } |
| |
| debug("*** Warning: NFS version not supported: Requested: V%d, accepted: min V%d - max V%d\n", |
| choosen_nfs_version, |
| ntohl(rpc_pkt->u.reply.data[0]), |
| ntohl(rpc_pkt->u.reply.data[1])); |
| debug("Will retry with NFSv%d\n", min); |
| choosen_nfs_version = min; |
| return -NFS_RPC_PROG_MISMATCH; |
| } |
| case NFS_RPC_PROG_UNAVAIL: |
| case NFS_RPC_PROC_UNAVAIL: |
| case NFS_RPC_GARBAGE_ARGS: |
| case NFS_RPC_SYSTEM_ERR: |
| default: /* Unknown error on 'accept state' flag */ |
| debug("*** ERROR: accept state error (%d)\n", |
| ntohl(rpc_pkt->u.reply.astatus)); |
| break; |
| } |
| return -1; |
| } |
| |
| return 0; |
| } |
| |
| static int rpc_lookup_reply(int prog, uchar *pkt, unsigned len) |
| { |
| struct rpc_t rpc_pkt; |
| |
| memcpy(&rpc_pkt.u.data[0], pkt, len); |
| |
| debug("%s\n", __func__); |
| |
| if (ntohl(rpc_pkt.u.reply.id) > rpc_id) |
| return -NFS_RPC_ERR; |
| else if (ntohl(rpc_pkt.u.reply.id) < rpc_id) |
| return -NFS_RPC_DROP; |
| |
| if (rpc_pkt.u.reply.rstatus || |
| rpc_pkt.u.reply.verifier || |
| rpc_pkt.u.reply.astatus) |
| return -1; |
| |
| switch (prog) { |
| case PROG_MOUNT: |
| nfs_server_mount_port = ntohl(rpc_pkt.u.reply.data[0]); |
| break; |
| case PROG_NFS: |
| nfs_server_port = ntohl(rpc_pkt.u.reply.data[0]); |
| break; |
| } |
| |
| return 0; |
| } |
| |
| static int nfs_mount_reply(uchar *pkt, unsigned len) |
| { |
| struct rpc_t rpc_pkt; |
| int ret; |
| |
| debug("%s\n", __func__); |
| |
| memcpy(&rpc_pkt.u.data[0], pkt, len); |
| |
| if (ntohl(rpc_pkt.u.reply.id) > rpc_id) |
| return -NFS_RPC_ERR; |
| else if (ntohl(rpc_pkt.u.reply.id) < rpc_id) |
| return -NFS_RPC_DROP; |
| |
| ret = rpc_handle_error(&rpc_pkt); |
| if (ret) |
| return ret; |
| |
| fs_mounted = 1; |
| /* NFSv2 and NFSv3 use same structure */ |
| if (choosen_nfs_version != NFS_V3) { |
| memcpy(dirfh, rpc_pkt.u.reply.data + 1, NFS_FHSIZE); |
| } else { |
| dirfh3_length = ntohl(rpc_pkt.u.reply.data[1]); |
| if (dirfh3_length > NFS3_FHSIZE) |
| dirfh3_length = NFS3_FHSIZE; |
| memcpy(dirfh, rpc_pkt.u.reply.data + 2, dirfh3_length); |
| } |
| |
| return 0; |
| } |
| |
| static int nfs_umountall_reply(uchar *pkt, unsigned len) |
| { |
| struct rpc_t rpc_pkt; |
| |
| debug("%s\n", __func__); |
| |
| memcpy(&rpc_pkt.u.data[0], pkt, len); |
| |
| if (ntohl(rpc_pkt.u.reply.id) > rpc_id) |
| return -NFS_RPC_ERR; |
| else if (ntohl(rpc_pkt.u.reply.id) < rpc_id) |
| return -NFS_RPC_DROP; |
| |
| if (rpc_pkt.u.reply.rstatus || |
| rpc_pkt.u.reply.verifier || |
| rpc_pkt.u.reply.astatus) |
| return -1; |
| |
| fs_mounted = 0; |
| memset(dirfh, 0, sizeof(dirfh)); |
| |
| return 0; |
| } |
| |
| static int nfs_lookup_reply(uchar *pkt, unsigned len) |
| { |
| struct rpc_t rpc_pkt; |
| int ret; |
| |
| debug("%s\n", __func__); |
| |
| memcpy(&rpc_pkt.u.data[0], pkt, len); |
| |
| if (ntohl(rpc_pkt.u.reply.id) > rpc_id) |
| return -NFS_RPC_ERR; |
| else if (ntohl(rpc_pkt.u.reply.id) < rpc_id) |
| return -NFS_RPC_DROP; |
| |
| ret = rpc_handle_error(&rpc_pkt); |
| if (ret) |
| return ret; |
| |
| if (choosen_nfs_version != NFS_V3) { |
| if (((uchar *)&(rpc_pkt.u.reply.data[0]) - (uchar *)(&rpc_pkt) + NFS_FHSIZE) > len) |
| return -NFS_RPC_DROP; |
| memcpy(filefh, rpc_pkt.u.reply.data + 1, NFS_FHSIZE); |
| } else { /* NFS_V3 */ |
| filefh3_length = ntohl(rpc_pkt.u.reply.data[1]); |
| if (filefh3_length > NFS3_FHSIZE) |
| filefh3_length = NFS3_FHSIZE; |
| memcpy(filefh, rpc_pkt.u.reply.data + 2, filefh3_length); |
| } |
| |
| return 0; |
| } |
| |
| static int nfs3_get_attributes_offset(uint32_t *data) |
| { |
| if (data[1]) { |
| /* 'attributes_follow' flag is TRUE, |
| * so we have attributes on 21 dwords */ |
| /* Skip unused values : |
| type; 32 bits value, |
| mode; 32 bits value, |
| nlink; 32 bits value, |
| uid; 32 bits value, |
| gid; 32 bits value, |
| size; 64 bits value, |
| used; 64 bits value, |
| rdev; 64 bits value, |
| fsid; 64 bits value, |
| fileid; 64 bits value, |
| atime; 64 bits value, |
| mtime; 64 bits value, |
| ctime; 64 bits value, |
| */ |
| return 22; |
| } else { |
| /* 'attributes_follow' flag is FALSE, |
| * so we don't have any attributes */ |
| return 1; |
| } |
| } |
| |
| static int nfs_readlink_reply(uchar *pkt, unsigned len) |
| { |
| struct rpc_t rpc_pkt; |
| int rlen; |
| int nfsv3_data_offset = 0; |
| |
| debug("%s\n", __func__); |
| |
| memcpy((unsigned char *)&rpc_pkt, pkt, len); |
| |
| if (ntohl(rpc_pkt.u.reply.id) > rpc_id) |
| return -NFS_RPC_ERR; |
| else if (ntohl(rpc_pkt.u.reply.id) < rpc_id) |
| return -NFS_RPC_DROP; |
| |
| if (rpc_pkt.u.reply.rstatus || |
| rpc_pkt.u.reply.verifier || |
| rpc_pkt.u.reply.astatus || |
| rpc_pkt.u.reply.data[0]) |
| return -1; |
| |
| if (choosen_nfs_version == NFS_V3) { |
| nfsv3_data_offset = |
| nfs3_get_attributes_offset(rpc_pkt.u.reply.data); |
| } |
| |
| /* new path length */ |
| rlen = ntohl(rpc_pkt.u.reply.data[1 + nfsv3_data_offset]); |
| |
| if (((uchar *)&(rpc_pkt.u.reply.data[0]) - (uchar *)(&rpc_pkt) + rlen) > len) |
| return -NFS_RPC_DROP; |
| |
| if (*((char *)&(rpc_pkt.u.reply.data[2 + nfsv3_data_offset])) != '/') { |
| int pathlen; |
| |
| strcat(nfs_path, "/"); |
| pathlen = strlen(nfs_path); |
| memcpy(nfs_path + pathlen, |
| (uchar *)&(rpc_pkt.u.reply.data[2 + nfsv3_data_offset]), |
| rlen); |
| nfs_path[pathlen + rlen] = 0; |
| } else { |
| memcpy(nfs_path, |
| (uchar *)&(rpc_pkt.u.reply.data[2 + nfsv3_data_offset]), |
| rlen); |
| nfs_path[rlen] = 0; |
| } |
| return 0; |
| } |
| |
| static int nfs_read_reply(uchar *pkt, unsigned len) |
| { |
| struct rpc_t rpc_pkt; |
| int rlen; |
| uchar *data_ptr; |
| |
| debug("%s\n", __func__); |
| |
| memcpy(&rpc_pkt.u.data[0], pkt, sizeof(rpc_pkt.u.reply)); |
| |
| if (ntohl(rpc_pkt.u.reply.id) > rpc_id) |
| return -NFS_RPC_ERR; |
| else if (ntohl(rpc_pkt.u.reply.id) < rpc_id) |
| return -NFS_RPC_DROP; |
| |
| if (rpc_pkt.u.reply.rstatus || |
| rpc_pkt.u.reply.verifier || |
| rpc_pkt.u.reply.astatus || |
| rpc_pkt.u.reply.data[0]) { |
| if (rpc_pkt.u.reply.rstatus) |
| return -9999; |
| if (rpc_pkt.u.reply.astatus) |
| return -9999; |
| return -ntohl(rpc_pkt.u.reply.data[0]); |
| } |
| |
| if ((nfs_offset != 0) && !((nfs_offset) % |
| (NFS_READ_SIZE / 2 * 10 * HASHES_PER_LINE))) |
| puts("\n\t "); |
| if (!(nfs_offset % ((NFS_READ_SIZE / 2) * 10))) |
| putc('#'); |
| |
| if (choosen_nfs_version != NFS_V3) { |
| rlen = ntohl(rpc_pkt.u.reply.data[18]); |
| data_ptr = (uchar *)&(rpc_pkt.u.reply.data[19]); |
| } else { /* NFS_V3 */ |
| int nfsv3_data_offset = |
| nfs3_get_attributes_offset(rpc_pkt.u.reply.data); |
| |
| /* count value */ |
| rlen = ntohl(rpc_pkt.u.reply.data[1 + nfsv3_data_offset]); |
| /* Skip unused values : |
| EOF: 32 bits value, |
| data_size: 32 bits value, |
| */ |
| data_ptr = (uchar *) |
| &(rpc_pkt.u.reply.data[4 + nfsv3_data_offset]); |
| } |
| |
| if (((uchar *)&(rpc_pkt.u.reply.data[0]) - (uchar *)(&rpc_pkt) + rlen) > len) |
| return -9999; |
| |
| if (store_block(data_ptr, nfs_offset, rlen)) |
| return -9999; |
| |
| return rlen; |
| } |
| |
| /************************************************************************** |
| Interfaces of U-BOOT |
| **************************************************************************/ |
| static void nfs_timeout_handler(void) |
| { |
| if (++nfs_timeout_count > NFS_RETRY_COUNT) { |
| puts("\nRetry count exceeded; starting again\n"); |
| net_start_again(); |
| } else { |
| puts("T "); |
| net_set_timeout_handler(nfs_timeout + |
| nfs_timeout * nfs_timeout_count, |
| nfs_timeout_handler); |
| nfs_send(); |
| } |
| } |
| |
| static void nfs_handler(uchar *pkt, unsigned dest, struct in_addr sip, |
| unsigned src, unsigned len) |
| { |
| int rlen; |
| int reply; |
| |
| debug("%s\n", __func__); |
| |
| if (len > sizeof(struct rpc_t)) |
| return; |
| |
| if (dest != nfs_our_port) |
| return; |
| |
| switch (nfs_state) { |
| case STATE_PRCLOOKUP_PROG_MOUNT_REQ: |
| if (rpc_lookup_reply(PROG_MOUNT, pkt, len) == -NFS_RPC_DROP) |
| break; |
| nfs_state = STATE_PRCLOOKUP_PROG_NFS_REQ; |
| nfs_send(); |
| break; |
| |
| case STATE_PRCLOOKUP_PROG_NFS_REQ: |
| if (rpc_lookup_reply(PROG_NFS, pkt, len) == -NFS_RPC_DROP) |
| break; |
| nfs_state = STATE_MOUNT_REQ; |
| nfs_send(); |
| break; |
| |
| case STATE_MOUNT_REQ: |
| reply = nfs_mount_reply(pkt, len); |
| if (reply == -NFS_RPC_DROP) { |
| break; |
| } else if (reply == -NFS_RPC_ERR) { |
| puts("*** ERROR: Cannot mount\n"); |
| /* just to be sure... */ |
| nfs_state = STATE_UMOUNT_REQ; |
| nfs_send(); |
| } else if (reply == -NFS_RPC_PROG_MISMATCH && |
| choosen_nfs_version != NFS_UNKOWN) { |
| nfs_state = STATE_MOUNT_REQ; |
| nfs_send(); |
| } else { |
| nfs_state = STATE_LOOKUP_REQ; |
| nfs_send(); |
| } |
| break; |
| |
| case STATE_UMOUNT_REQ: |
| reply = nfs_umountall_reply(pkt, len); |
| if (reply == -NFS_RPC_DROP) { |
| break; |
| } else if (reply == -NFS_RPC_ERR) { |
| debug("*** ERROR: Cannot umount\n"); |
| net_set_state(NETLOOP_FAIL); |
| } else { |
| puts("\ndone\n"); |
| net_set_state(nfs_download_state); |
| } |
| break; |
| |
| case STATE_LOOKUP_REQ: |
| reply = nfs_lookup_reply(pkt, len); |
| if (reply == -NFS_RPC_DROP) { |
| break; |
| } else if (reply == -NFS_RPC_ERR) { |
| puts("*** ERROR: File lookup fail\n"); |
| nfs_state = STATE_UMOUNT_REQ; |
| nfs_send(); |
| } else if (reply == -NFS_RPC_PROG_MISMATCH && |
| choosen_nfs_version != NFS_UNKOWN) { |
| /* umount */ |
| nfs_state = STATE_UMOUNT_REQ; |
| nfs_send(); |
| /* And retry with another supported version */ |
| nfs_state = STATE_PRCLOOKUP_PROG_MOUNT_REQ; |
| nfs_send(); |
| } else { |
| nfs_state = STATE_READ_REQ; |
| nfs_offset = 0; |
| nfs_len = NFS_READ_SIZE; |
| nfs_send(); |
| } |
| break; |
| |
| case STATE_READLINK_REQ: |
| reply = nfs_readlink_reply(pkt, len); |
| if (reply == -NFS_RPC_DROP) { |
| break; |
| } else if (reply == -NFS_RPC_ERR) { |
| puts("*** ERROR: Symlink fail\n"); |
| nfs_state = STATE_UMOUNT_REQ; |
| nfs_send(); |
| } else { |
| debug("Symlink --> %s\n", nfs_path); |
| nfs_filename = basename(nfs_path); |
| nfs_path = dirname(nfs_path); |
| |
| nfs_state = STATE_MOUNT_REQ; |
| nfs_send(); |
| } |
| break; |
| |
| case STATE_READ_REQ: |
| rlen = nfs_read_reply(pkt, len); |
| if (rlen == -NFS_RPC_DROP) |
| break; |
| net_set_timeout_handler(nfs_timeout, nfs_timeout_handler); |
| if (rlen > 0) { |
| nfs_offset += rlen; |
| nfs_send(); |
| } else if ((rlen == -NFSERR_ISDIR) || (rlen == -NFSERR_INVAL)) { |
| /* symbolic link */ |
| nfs_state = STATE_READLINK_REQ; |
| nfs_send(); |
| } else { |
| if (!rlen) |
| nfs_download_state = NETLOOP_SUCCESS; |
| if (rlen < 0) |
| debug("NFS READ error (%d)\n", rlen); |
| nfs_state = STATE_UMOUNT_REQ; |
| nfs_send(); |
| } |
| break; |
| } |
| } |
| |
| |
| void nfs_start(void) |
| { |
| debug("%s\n", __func__); |
| nfs_download_state = NETLOOP_FAIL; |
| |
| nfs_server_ip = net_server_ip; |
| nfs_path = (char *)nfs_path_buff; |
| |
| if (nfs_path == NULL) { |
| net_set_state(NETLOOP_FAIL); |
| printf("*** ERROR: Fail allocate memory\n"); |
| return; |
| } |
| |
| if (!net_parse_bootfile(&nfs_server_ip, nfs_path, |
| sizeof(nfs_path_buff))) { |
| sprintf(nfs_path, "/nfsroot/%02X%02X%02X%02X.img", |
| net_ip.s_addr & 0xFF, |
| (net_ip.s_addr >> 8) & 0xFF, |
| (net_ip.s_addr >> 16) & 0xFF, |
| (net_ip.s_addr >> 24) & 0xFF); |
| |
| printf("*** Warning: no boot file name; using '%s'\n", |
| nfs_path); |
| } |
| |
| nfs_filename = basename(nfs_path); |
| nfs_path = dirname(nfs_path); |
| |
| printf("Using %s device\n", eth_get_name()); |
| |
| printf("File transfer via NFS from server %pI4; our IP address is %pI4", |
| &nfs_server_ip, &net_ip); |
| |
| /* Check if we need to send across this subnet */ |
| if (net_gateway.s_addr && net_netmask.s_addr) { |
| struct in_addr our_net; |
| struct in_addr server_net; |
| |
| our_net.s_addr = net_ip.s_addr & net_netmask.s_addr; |
| server_net.s_addr = nfs_server_ip.s_addr & net_netmask.s_addr; |
| if (our_net.s_addr != server_net.s_addr) |
| printf("; sending through gateway %pI4", |
| &net_gateway); |
| } |
| printf("\nFilename '%s/%s'.", nfs_path, nfs_filename); |
| |
| if (net_boot_file_expected_size_in_blocks) { |
| printf(" Size is 0x%x Bytes = ", |
| net_boot_file_expected_size_in_blocks << 9); |
| print_size(net_boot_file_expected_size_in_blocks << 9, ""); |
| } |
| printf("\nLoad address: 0x%lx\nLoading: *\b", image_load_addr); |
| |
| net_set_timeout_handler(nfs_timeout, nfs_timeout_handler); |
| net_set_udp_handler(nfs_handler); |
| |
| nfs_timeout_count = 0; |
| nfs_state = STATE_PRCLOOKUP_PROG_MOUNT_REQ; |
| |
| /*nfs_our_port = 4096 + (get_ticks() % 3072);*/ |
| /*FIX ME !!!*/ |
| nfs_our_port = 1000; |
| |
| /* zero out server ether in case the server ip has changed */ |
| memset(net_server_ethaddr, 0, 6); |
| |
| nfs_send(); |
| } |