926 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			926 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C
		
	
	
	
/*
 | 
						|
 * 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. */
 | 
						|
 | 
						|
#include <common.h>
 | 
						|
#include <command.h>
 | 
						|
#include <net.h>
 | 
						|
#include <malloc.h>
 | 
						|
#include <mapmem.h>
 | 
						|
#include "nfs.h"
 | 
						|
#include "bootp.h"
 | 
						|
 | 
						|
#define HASHES_PER_LINE 65	/* Number of "loading" hashes per line	*/
 | 
						|
#define NFS_RETRY_COUNT 30
 | 
						|
#ifndef CONFIG_NFS_TIMEOUT
 | 
						|
# define NFS_TIMEOUT 2000UL
 | 
						|
#else
 | 
						|
# define NFS_TIMEOUT CONFIG_NFS_TIMEOUT
 | 
						|
#endif
 | 
						|
 | 
						|
#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 ulong nfs_timeout = NFS_TIMEOUT;
 | 
						|
 | 
						|
static char dirfh[NFS_FHSIZE];	/* NFSv2 / NFSv3 file handle of directory */
 | 
						|
static char filefh[NFS3_FHSIZE]; /* NFSv2 / NFSv3 file handle */
 | 
						|
static 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];
 | 
						|
 | 
						|
#define NFSV2_FLAG 1
 | 
						|
#define NFSV3_FLAG 1 << 1
 | 
						|
static char supported_nfs_versions = NFSV2_FLAG | NFSV3_FLAG;
 | 
						|
 | 
						|
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 (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)(load_addr+offset), len);
 | 
						|
		if (rc) {
 | 
						|
			flash_perror(rc);
 | 
						|
			return -1;
 | 
						|
		}
 | 
						|
	} else
 | 
						|
#endif /* CONFIG_SYS_DIRECT_FLASH_NFS */
 | 
						|
	{
 | 
						|
		void *ptr = map_sysmem(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:
 | 
						|
		if (supported_nfs_versions & NFSV2_FLAG)
 | 
						|
			rpc_pkt.u.call.vers = htonl(2);	/* NFS v2 */
 | 
						|
		else /* NFSV3_FLAG */
 | 
						|
			rpc_pkt.u.call.vers = htonl(3);	/* NFS v3 */
 | 
						|
		break;
 | 
						|
	case PROG_PORTMAP:
 | 
						|
	case PROG_MOUNT:
 | 
						|
	default:
 | 
						|
		rpc_pkt.u.call.vers = htonl(2);	/* portmapper is version 2 */
 | 
						|
	}
 | 
						|
	rpc_pkt.u.call.proc = htonl(rpc_proc);
 | 
						|
	p = (uint32_t *)&(rpc_pkt.u.call.data);
 | 
						|
 | 
						|
	if (datalen)
 | 
						|
		memcpy((char *)p, (char *)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 (supported_nfs_versions & NFSV2_FLAG) {
 | 
						|
		memcpy(p, filefh, NFS_FHSIZE);
 | 
						|
		p += (NFS_FHSIZE / 4);
 | 
						|
	} else { /* NFSV3_FLAG */
 | 
						|
		*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 (supported_nfs_versions & NFSV2_FLAG) {
 | 
						|
		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 {  /* NFSV3_FLAG */
 | 
						|
		*p++ = htonl(NFS_FHSIZE);	/* Dir handle length */
 | 
						|
		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, 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 (supported_nfs_versions & NFSV2_FLAG) {
 | 
						|
		memcpy(p, filefh, NFS_FHSIZE);
 | 
						|
		p += (NFS_FHSIZE / 4);
 | 
						|
		*p++ = htonl(offset);
 | 
						|
		*p++ = htonl(readlen);
 | 
						|
		*p++ = 0;
 | 
						|
	} else { /* NFSV3_FLAG */
 | 
						|
		*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 (supported_nfs_versions & NFSV2_FLAG)
 | 
						|
			rpc_lookup_req(PROG_MOUNT, 1);
 | 
						|
		else  /* NFSV3_FLAG */
 | 
						|
			rpc_lookup_req(PROG_MOUNT, 3);
 | 
						|
		break;
 | 
						|
	case STATE_PRCLOOKUP_PROG_NFS_REQ:
 | 
						|
		if (supported_nfs_versions & NFSV2_FLAG)
 | 
						|
			rpc_lookup_req(PROG_NFS, 2);
 | 
						|
		else  /* NFSV3_FLAG */
 | 
						|
			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_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;
 | 
						|
 | 
						|
	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  ||
 | 
						|
	    rpc_pkt.u.reply.data[0])
 | 
						|
		return -1;
 | 
						|
 | 
						|
	fs_mounted = 1;
 | 
						|
	/*  NFSv2 and NFSv3 use same structure */
 | 
						|
	memcpy(dirfh, rpc_pkt.u.reply.data + 1, NFS_FHSIZE);
 | 
						|
 | 
						|
	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;
 | 
						|
 | 
						|
	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  ||
 | 
						|
	    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 */
 | 
						|
			switch (ntohl(rpc_pkt.u.reply.data[0])) {
 | 
						|
			/* Minimal supported NFS version */
 | 
						|
			case 3:
 | 
						|
				debug("*** Waring: NFS version not supported: Requested: V%d, accepted: min V%d - max V%d\n",
 | 
						|
				      (supported_nfs_versions & NFSV2_FLAG) ?
 | 
						|
						2 : 3,
 | 
						|
				      ntohl(rpc_pkt.u.reply.data[0]),
 | 
						|
				      ntohl(rpc_pkt.u.reply.data[1]));
 | 
						|
				debug("Will retry with NFSv3\n");
 | 
						|
				/* Clear NFSV2_FLAG from supported versions */
 | 
						|
				supported_nfs_versions &= ~NFSV2_FLAG;
 | 
						|
				return -NFS_RPC_PROG_MISMATCH;
 | 
						|
			case 4:
 | 
						|
			default:
 | 
						|
				puts("*** ERROR: NFS version not supported");
 | 
						|
				debug(": Requested: V%d, accepted: min V%d - max V%d\n",
 | 
						|
				      (supported_nfs_versions & NFSV2_FLAG) ?
 | 
						|
						2 : 3,
 | 
						|
				      ntohl(rpc_pkt.u.reply.data[0]),
 | 
						|
				      ntohl(rpc_pkt.u.reply.data[1]));
 | 
						|
				puts("\n");
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		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;
 | 
						|
	}
 | 
						|
 | 
						|
	if (supported_nfs_versions & NFSV2_FLAG) {
 | 
						|
		memcpy(filefh, rpc_pkt.u.reply.data + 1, NFS_FHSIZE);
 | 
						|
	} else {  /* NFSV3_FLAG */
 | 
						|
		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 (ntohl(data[1]) != 0) {
 | 
						|
		/* '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 (!(supported_nfs_versions & NFSV2_FLAG)) { /* NFSV3_FLAG */
 | 
						|
		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 (*((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 (supported_nfs_versions & NFSV2_FLAG) {
 | 
						|
		rlen = ntohl(rpc_pkt.u.reply.data[18]);
 | 
						|
		data_ptr = (uchar *)&(rpc_pkt.u.reply.data[19]);
 | 
						|
	} else {  /* NFSV3_FLAG */
 | 
						|
		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 (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 (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 {
 | 
						|
			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 &&
 | 
						|
			   supported_nfs_versions != 0) {
 | 
						|
			/* 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);
 | 
						|
		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);
 | 
						|
		debug("*** ERROR: Fail allocate memory\n");
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	if (net_boot_file_name[0] == '\0') {
 | 
						|
		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);
 | 
						|
 | 
						|
		debug("*** Warning: no boot file name; using '%s'\n",
 | 
						|
		      nfs_path);
 | 
						|
	} else {
 | 
						|
		char *p = net_boot_file_name;
 | 
						|
 | 
						|
		p = strchr(p, ':');
 | 
						|
 | 
						|
		if (p != NULL) {
 | 
						|
			nfs_server_ip = string_to_ip(net_boot_file_name);
 | 
						|
			++p;
 | 
						|
			strcpy(nfs_path, p);
 | 
						|
		} else {
 | 
						|
			strcpy(nfs_path, net_boot_file_name);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	nfs_filename = basename(nfs_path);
 | 
						|
	nfs_path     = dirname(nfs_path);
 | 
						|
 | 
						|
	debug("Using %s device\n", eth_get_name());
 | 
						|
 | 
						|
	debug("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 = net_server_ip.s_addr & net_netmask.s_addr;
 | 
						|
		if (our_net.s_addr != server_net.s_addr)
 | 
						|
			debug("; sending through gateway %pI4",
 | 
						|
			      &net_gateway);
 | 
						|
	}
 | 
						|
	debug("\nFilename '%s/%s'.", nfs_path, nfs_filename);
 | 
						|
 | 
						|
	if (net_boot_file_expected_size_in_blocks) {
 | 
						|
		debug(" Size is 0x%x Bytes = ",
 | 
						|
		      net_boot_file_expected_size_in_blocks << 9);
 | 
						|
		print_size(net_boot_file_expected_size_in_blocks << 9, "");
 | 
						|
	}
 | 
						|
	debug("\nLoad address: 0x%lx\nLoading: *\b", 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();
 | 
						|
}
 |