mail archive of the barebox mailing list
 help / color / mirror / Atom feed
* [PATCH] fs: Add NFS support
@ 2012-05-13 10:48 Sascha Hauer
  0 siblings, 0 replies; only message in thread
From: Sascha Hauer @ 2012-05-13 10:48 UTC (permalink / raw)
  To: barebox

This patch adds readonly NFS support. Currently no links are supported.
This is based on the previous U-Boot/NetBSD based code and some Kernel
bits.

Signed-off-by: Sascha Hauer <s.hauer@pengutronix.de>
---
 fs/Kconfig  |    4 +
 fs/Makefile |    1 +
 fs/nfs.c    | 1054 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 1059 insertions(+)
 create mode 100644 fs/nfs.c

diff --git a/fs/Kconfig b/fs/Kconfig
index e5f307f..769a99e 100644
--- a/fs/Kconfig
+++ b/fs/Kconfig
@@ -23,6 +23,10 @@ config FS_TFTP
 	bool
 	prompt "tftp support"
 
+config FS_NFS
+	bool
+	prompt "nfs support"
+
 source fs/fat/Kconfig
 
 config PARTITION_NEED_MTD
diff --git a/fs/Makefile b/fs/Makefile
index d204093..1b52bee 100644
--- a/fs/Makefile
+++ b/fs/Makefile
@@ -5,3 +5,4 @@ obj-$(CONFIG_FS_DEVFS)	+= devfs.o
 obj-$(CONFIG_FS_FAT)	+= fat/
 obj-y	+= fs.o
 obj-$(CONFIG_FS_TFTP)	+= tftp.o
+obj-$(CONFIG_FS_NFS)	+= nfs.o
diff --git a/fs/nfs.c b/fs/nfs.c
new file mode 100644
index 0000000..75a8f25
--- /dev/null
+++ b/fs/nfs.c
@@ -0,0 +1,1054 @@
+/*
+ * nfs.c - barebox NFS driver
+ *
+ * Copyright (c) 2012 Sascha Hauer <s.hauer@pengutronix.de>, Pengutronix
+ * Copyright (c) Masami Komiya <mkomiya@sonare.it> 2004
+ *
+ * Based on U-Boot NFS code which is based on NetBSD code
+ *
+ * See file CREDITS for list of people who contributed to this
+ * project.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#include <common.h>
+#include <net.h>
+#include <driver.h>
+#include <fs.h>
+#include <errno.h>
+#include <libgen.h>
+#include <fcntl.h>
+#include <fs.h>
+#include <init.h>
+#include <linux/stat.h>
+#include <linux/err.h>
+#include <kfifo.h>
+#include <sizes.h>
+
+#define SUNRPC_PORT     111
+
+#define PROG_PORTMAP    100000
+#define PROG_NFS        100003
+#define PROG_MOUNT      100005
+
+#define MSG_CALL        0
+#define MSG_REPLY       1
+
+#define PORTMAP_GETPORT 3
+
+#define MOUNT_ADDENTRY	1
+#define MOUNT_UMOUNT	3
+
+#define NFS_GETATTR     1
+#define NFS_LOOKUP      4
+#define NFS_READLINK    5
+#define NFS_READ        6
+#define NFS_READDIR	16
+
+#define NFS_FHSIZE      32
+
+enum nfs_stat {
+	NFS_OK		= 0,
+	NFSERR_PERM	= 1,
+	NFSERR_NOENT	= 2,
+	NFSERR_IO	= 5,
+	NFSERR_NXIO	= 6,
+	NFSERR_ACCES	= 13,
+	NFSERR_EXIST	= 17,
+	NFSERR_NODEV	= 19,
+	NFSERR_NOTDIR	= 20,
+	NFSERR_ISDIR	= 21,
+	NFSERR_FBIG	= 27,
+	NFSERR_NOSPC	= 28,
+	NFSERR_ROFS	= 30,
+	NFSERR_NAMETOOLONG=63,
+	NFSERR_NOTEMPTY	= 66,
+	NFSERR_DQUOT	= 69,
+	NFSERR_STALE	= 70,
+	NFSERR_WFLUSH	= 99,
+};
+
+static void *nfs_packet;
+static int nfs_len;
+
+struct rpc_call {
+	uint32_t id;
+	uint32_t type;
+	uint32_t rpcvers;
+	uint32_t prog;
+	uint32_t vers;
+	uint32_t proc;
+	uint32_t data[0];
+};
+
+struct rpc_reply {
+	uint32_t id;
+	uint32_t type;
+	uint32_t rstatus;
+	uint32_t verifier;
+	uint32_t v2;
+	uint32_t astatus;
+	uint32_t data[0];
+};
+
+#define NFS_TIMEOUT	(2 * SECOND)
+#define NFS_MAX_RESEND	5
+
+struct nfs_priv {
+	struct net_connection *con;
+	IPaddr_t server;
+	char *path;
+	int mount_port;
+	int nfs_port;
+	unsigned long rpc_id;
+	char rootfh[NFS_FHSIZE];
+};
+
+struct file_priv {
+	struct kfifo *fifo;
+	void *buf;
+	char filefh[NFS_FHSIZE];
+	struct nfs_priv *npriv;
+};
+
+static uint64_t nfs_timer_start;
+
+static int	nfs_state;
+#define STATE_DONE			1
+#define STATE_START			2
+
+enum ftype {
+	NFNON = 0,
+	NFREG = 1,
+	NFDIR = 2,
+	NFBLK = 3,
+	NFCHR = 4,
+	NFLNK = 5
+};
+
+struct fattr {
+	uint32_t type;
+	uint32_t mode;
+	uint32_t nlink;
+	uint32_t uid;
+	uint32_t gid;
+	uint32_t size;
+	uint32_t blocksize;
+	uint32_t rdev;
+	uint32_t blocks;
+};
+
+struct readdirargs {
+	char filefh[NFS_FHSIZE];
+	uint32_t cookie;
+	uint32_t count;
+};
+
+struct xdr_stream {
+	__be32 *p;
+	void *buf;
+	__be32 *end;
+};
+
+#define xdr_zero 0
+#define XDR_QUADLEN(l)		(((l) + 3) >> 2)
+
+static void xdr_init(struct xdr_stream *stream, void *buf, int len)
+{
+	stream->p = stream->buf = buf;
+	stream->end = stream->buf + len;
+}
+
+static __be32 *__xdr_inline_decode(struct xdr_stream *xdr, size_t nbytes)
+{
+	__be32 *p = xdr->p;
+	__be32 *q = p + XDR_QUADLEN(nbytes);
+
+        if (q > xdr->end || q < p)
+		return NULL;
+	xdr->p = q;
+	return p;
+}
+
+static __be32 *xdr_inline_decode(struct xdr_stream *xdr, size_t nbytes)
+{
+	__be32 *p;
+
+	if (nbytes == 0)
+		return xdr->p;
+	if (xdr->p == xdr->end)
+		return NULL;
+	p = __xdr_inline_decode(xdr, nbytes);
+
+	return p;
+}
+
+static int decode_filename(struct xdr_stream *xdr,
+		char *name, u32 *length)
+{
+	__be32 *p;
+	u32 count;
+
+	p = xdr_inline_decode(xdr, 4);
+	if (!p)
+		goto out_overflow;
+	count = be32_to_cpup(p);
+	if (count > 255)
+		goto out_nametoolong;
+	p = xdr_inline_decode(xdr, count);
+	if (!p)
+		goto out_overflow;
+	memcpy(name, p, count);
+	name[count] = 0;
+	*length = count;
+	return 0;
+out_nametoolong:
+	printk("NFS: returned filename too long: %u\n", count);
+	return -ENAMETOOLONG;
+out_overflow:
+	printf("%s overflow\n",__func__);
+	return -EIO;
+}
+
+/*
+ * rpc_add_credentials - Add RPC authentication/verifier entries
+ */
+static uint32_t *rpc_add_credentials(uint32_t *p)
+{
+	int hl;
+	int hostnamelen = 0;
+
+	/*
+	 * 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).
+	 */
+
+	hl = (hostnamelen + 3) & ~3;
+
+	/* Provide an AUTH_UNIX credential.  */
+	*p++ = htonl(1);		/* AUTH_UNIX */
+	*p++ = htonl(hl + 20);		/* auth length */
+	*p++ = htonl(0);		/* stamp */
+	*p++ = htonl(hostnamelen);	/* hostname string */
+
+	if (hostnamelen & 3)
+		*(p + hostnamelen / 4) = 0; /* add zero padding */
+
+	/* memcpy(p, hostname, hostnamelen); */ /* empty hostname */
+
+	p += hl / 4;
+	*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;
+}
+
+static int rpc_check_reply(unsigned char *pkt, int rpc_prog, unsigned long rpc_id, int *nfserr)
+{
+	uint32_t *data;
+	struct rpc_reply rpc;
+
+	*nfserr = 0;
+
+	if (!pkt)
+		return -EAGAIN;
+
+	memcpy(&rpc, pkt, sizeof(rpc));
+
+	if (ntohl(rpc.id) != rpc_id)
+		return -EINVAL;
+
+	if (rpc.rstatus  ||
+	    rpc.verifier ||
+	    rpc.astatus ) {
+		return -EINVAL;
+	}
+
+	if (rpc_prog == PROG_PORTMAP)
+		return 0;
+
+	data = (uint32_t *)(pkt + sizeof(struct rpc_reply));
+	*nfserr = ntohl(net_read_uint32(data));
+	*nfserr = -*nfserr;
+
+	debug("%s: state: %d, err %d\n", __func__, nfs_state, *nfserr);
+
+	return 0;
+}
+
+/*
+ * rpc_req - synchronous RPC request
+ */
+static int rpc_req(struct nfs_priv *npriv, int rpc_prog, int rpc_proc,
+		uint32_t *data, int datalen)
+{
+	struct rpc_call pkt;
+	unsigned long id;
+	int dport;
+	int ret;
+	unsigned char *payload = net_udp_get_payload(npriv->con);
+	int nfserr;
+	int tries = 0;
+
+	npriv->rpc_id++;
+	id = npriv->rpc_id;
+
+	pkt.id = htonl(id);
+	pkt.type = htonl(MSG_CALL);
+	pkt.rpcvers = htonl(2);	/* use RPC version 2 */
+	pkt.prog = htonl(rpc_prog);
+	pkt.vers = htonl(2);	/* portmapper is version 2 */
+	pkt.proc = htonl(rpc_proc);
+
+	memcpy(payload, &pkt, sizeof(pkt));
+	memcpy(payload + sizeof(pkt), data, datalen * sizeof(uint32_t));
+
+	if (rpc_prog == PROG_PORTMAP)
+		dport = SUNRPC_PORT;
+	else if (rpc_prog == PROG_MOUNT)
+		dport = npriv->mount_port;
+	else
+		dport = npriv->nfs_port;
+
+	npriv->con->udp->uh_dport = htons(dport);
+
+again:
+	ret = net_udp_send(npriv->con, sizeof(pkt) + datalen * sizeof(uint32_t));
+
+	nfs_timer_start = get_time_ns();
+
+	nfs_state = STATE_START;
+	nfs_packet = NULL;
+
+	while (nfs_state != STATE_DONE) {
+		if (ctrlc()) {
+			ret = -EINTR;
+			break;
+		}
+		net_poll();
+
+		if (is_timeout(nfs_timer_start, NFS_TIMEOUT)) {
+			tries++;
+			if (tries == NFS_MAX_RESEND)
+				return -ETIMEDOUT;
+			goto again;
+		}
+
+		ret = rpc_check_reply(nfs_packet, rpc_prog,
+				npriv->rpc_id, &nfserr);
+		if (!ret) {
+			ret = nfserr;
+			break;
+		}
+	}
+
+	return ret;
+}
+
+/*
+ * rpc_lookup_req - Lookup RPC Port numbers
+ */
+static int rpc_lookup_req(struct nfs_priv *npriv, int prog, int ver)
+{
+	uint32_t data[16];
+	int ret;
+	uint32_t port;
+
+	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;
+
+	ret = rpc_req(npriv, PROG_PORTMAP, PORTMAP_GETPORT, data, 8);
+	if (ret)
+		return ret;
+
+	port = net_read_uint32((uint32_t *)(nfs_packet + sizeof(struct rpc_reply)));
+
+	switch (prog) {
+	case PROG_MOUNT:
+		npriv->mount_port = ntohl(port);
+		debug("mount port: %d\n", npriv->mount_port);
+		break;
+	case PROG_NFS:
+		npriv->nfs_port = ntohl(port);
+		debug("nfs port: %d\n", npriv->nfs_port);
+		break;
+	default:
+		return -EINVAL;
+	}
+
+	return 0;
+}
+
+/*
+ * nfs_mount_req - Mount an NFS Filesystem
+ */
+static int nfs_mount_req(struct nfs_priv *npriv)
+{
+	uint32_t data[1024];
+	uint32_t *p;
+	int len;
+	int pathlen;
+	int ret;
+
+	pathlen = strlen(npriv->path);
+
+	debug("%s: %s\n", __func__, npriv->path);
+
+	p = &(data[0]);
+	p = rpc_add_credentials(p);
+
+	*p++ = htonl(pathlen);
+	if (pathlen & 3)
+		*(p + pathlen / 4) = 0;
+
+	memcpy (p, npriv->path, pathlen);
+	p += (pathlen + 3) / 4;
+
+	len = p - &(data[0]);
+
+	ret = rpc_req(npriv, PROG_MOUNT, MOUNT_ADDENTRY, data, len);
+	if (ret)
+		return ret;
+
+	memcpy(npriv->rootfh, nfs_packet + sizeof(struct rpc_reply) + 4, NFS_FHSIZE);
+
+	return 0;
+}
+
+/*
+ * nfs_umountall_req - Unmount all our NFS Filesystems on the Server
+ */
+static void nfs_umount_req(struct nfs_priv *npriv)
+{
+	uint32_t data[1024];
+	uint32_t *p;
+	int len;
+	int pathlen;
+
+	pathlen = strlen(npriv->path);
+
+	p = &(data[0]);
+	p = rpc_add_credentials(p);
+
+	*p++ = htonl(pathlen);
+	if (pathlen & 3)
+		*(p + pathlen / 4) = 0;
+
+	memcpy (p, npriv->path, pathlen);
+	p += (pathlen + 3) / 4;
+
+	len = p - &(data[0]);
+
+	rpc_req(npriv, PROG_MOUNT, MOUNT_UMOUNT, data, len);
+}
+
+/*
+ * nfs_lookup_req - Lookup Pathname
+ */
+static int nfs_lookup_req(struct file_priv *priv, const char *filename,
+		int fnamelen)
+{
+	uint32_t data[1024];
+	uint32_t *p;
+	int len;
+	int ret;
+
+	p = &(data[0]);
+	p = rpc_add_credentials(p);
+
+	memcpy(p, priv->filefh, NFS_FHSIZE);
+
+	p += (NFS_FHSIZE / 4);
+	*p++ = htonl(fnamelen);
+
+	if (fnamelen & 3)
+		*(p + fnamelen / 4) = 0;
+
+	memcpy(p, filename, fnamelen);
+	p += (fnamelen + 3) / 4;
+
+	len = p - &(data[0]);
+
+	ret = rpc_req(priv->npriv, PROG_NFS, NFS_LOOKUP, data, len);
+	if (ret)
+		return ret;
+
+	memcpy(priv->filefh, nfs_packet + sizeof(struct rpc_reply) + 4, NFS_FHSIZE);
+
+	return 0;
+}
+
+static int nfs_attr_req(struct file_priv *priv, struct stat *s)
+{
+	uint32_t data[1024];
+	uint32_t *p;
+	int len;
+	int ret;
+	struct fattr *fattr;
+	uint32_t type;
+
+	p = &(data[0]);
+	p = rpc_add_credentials(p);
+
+	memcpy(p, priv->filefh, NFS_FHSIZE);
+	p += (NFS_FHSIZE / 4);
+	*p++ = 0;
+
+	len = p - &(data[0]);
+
+	ret = rpc_req(priv->npriv, PROG_NFS, NFS_GETATTR, data, len);
+	if (ret)
+		return ret;
+
+	fattr = nfs_packet + sizeof(struct rpc_reply) + 4;
+
+	type = ntohl(net_read_uint32(&fattr->type));
+
+	s->st_size = ntohl(net_read_uint32(&fattr->size));
+	s->st_mode = ntohl(net_read_uint32(&fattr->mode));
+
+	return 0;
+}
+
+static void *nfs_readdirattr_req(struct file_priv *priv, int *plen, uint32_t cookie)
+{
+	uint32_t data[1024];
+	uint32_t *p;
+	int len;
+	int ret;
+	void *buf;
+
+	p = &(data[0]);
+	p = rpc_add_credentials(p);
+
+	memcpy(p, priv->filefh, NFS_FHSIZE);
+	p += (NFS_FHSIZE / 4);
+	*p++ = htonl(cookie); /* cookie */
+	*p++ = htonl(1024); /* count */
+	*p++ = 0;
+
+	len = p - &(data[0]);
+
+	ret = rpc_req(priv->npriv, PROG_NFS, NFS_READDIR, data, len);
+	if (ret)
+		return NULL;
+
+	*plen = nfs_len - sizeof(struct rpc_reply) + 4;
+
+	buf = xzalloc(*plen);
+
+	memcpy(buf, nfs_packet + sizeof(struct rpc_reply) + 4, *plen);
+
+	return buf;
+}
+
+/*
+ * nfs_read_req - Read File on NFS Server
+ */
+static int nfs_read_req(struct file_priv *priv, int offset, int readlen)
+{
+	uint32_t data[1024];
+	uint32_t *p;
+	uint32_t *filedata;
+	int len;
+	int ret;
+	int rlen;
+
+	p = &(data[0]);
+	p = rpc_add_credentials(p);
+
+	memcpy (p, priv->filefh, NFS_FHSIZE);
+	p += (NFS_FHSIZE / 4);
+	*p++ = htonl(offset);
+	*p++ = htonl(readlen);
+	*p++ = 0;
+
+	len = p - &(data[0]);
+
+	ret = rpc_req(priv->npriv, PROG_NFS, NFS_READ, data, len);
+	if (ret)
+		return ret;
+
+	filedata = (uint32_t *)(nfs_packet + sizeof(struct rpc_reply));
+
+	rlen = ntohl(net_read_uint32(filedata + 18));
+
+	kfifo_put(priv->fifo, (char *)(filedata + 19), rlen);
+
+	return 0;
+}
+
+#if 0
+static int nfs_readlink_reply(unsigned char *pkt, unsigned len)
+{
+	uint32_t *data;
+	char *path;
+	int rlen;
+//	int ret;
+
+	data = (uint32_t *)(pkt + sizeof(struct rpc_reply));
+
+	data++;
+
+	rlen = ntohl(net_read_uint32(data)); /* new path length */
+
+	data++;
+	path = (char *)data;
+
+	if (*path != '/') {
+		strcat(nfs_path, "/");
+		strncat(nfs_path, path, rlen);
+	} else {
+		memcpy(nfs_path, path, rlen);
+		nfs_path[rlen] = 0;
+	}
+	return 0;
+}
+#endif
+
+static void nfs_handler(void *ctx, char *packet, unsigned len)
+{
+	char *pkt = net_eth_to_udp_payload(packet);
+
+	nfs_state = STATE_DONE;
+	nfs_packet = pkt;
+	nfs_len = len;
+}
+
+static int nfs_create(struct device_d *dev, const char *pathname, mode_t mode)
+{
+	return -ENOSYS;
+}
+
+static int nfs_unlink(struct device_d *dev, const char *pathname)
+{
+	return -ENOSYS;
+}
+
+static int nfs_mkdir(struct device_d *dev, const char *pathname)
+{
+	return -ENOSYS;
+}
+
+static int nfs_rmdir(struct device_d *dev, const char *pathname)
+{
+	return -ENOSYS;
+}
+
+static int nfs_truncate(struct device_d *dev, FILE *f, ulong size)
+{
+	return -ENOSYS;
+}
+
+static struct file_priv *nfs_do_open(struct device_d *dev, const char *filename)
+{
+	struct file_priv *priv;
+	struct nfs_priv *npriv = dev->priv;
+	int ret;
+	const char *fname, *tok;
+
+	priv = xzalloc(sizeof(*priv));
+
+	priv->npriv = npriv;
+
+	if (!*filename) {
+		memcpy(priv->filefh, npriv->rootfh, NFS_FHSIZE);
+		return priv;
+	}
+
+	filename++;
+
+	fname = filename;
+
+	memcpy(priv->filefh, npriv->rootfh, NFS_FHSIZE);
+
+	while (*fname) {
+		int flen;
+
+		tok = strchr(fname, '/');
+		if (tok)
+			flen = tok - fname;
+		else
+			flen = strlen(fname);
+
+		ret = nfs_lookup_req(priv, fname, flen);
+		if (ret)
+			goto out;
+
+		if (tok)
+			fname += flen + 1;
+		else
+			break;
+	}
+
+	return priv;
+
+out:
+	free(priv);
+
+	return ERR_PTR(ret);
+}
+
+static void nfs_do_close(struct file_priv *priv)
+{
+	if (priv->fifo)
+		kfifo_free(priv->fifo);
+
+	free(priv);
+}
+
+static struct file_priv *nfs_do_stat(struct device_d *dev, const char *filename, struct stat *s)
+{
+	struct file_priv *priv;
+	int ret;
+
+	priv = nfs_do_open(dev, filename);
+	if (IS_ERR(priv))
+		return priv;
+
+	ret = nfs_attr_req(priv, s);
+	if (ret) {
+		nfs_do_close(priv);
+		return ERR_PTR(ret);
+	}
+
+	return priv;
+}
+
+static int nfs_open(struct device_d *dev, FILE *file, const char *filename)
+{
+	struct file_priv *priv;
+	struct stat s;
+
+	priv = nfs_do_stat(dev, filename, &s);
+	if (IS_ERR(priv))
+		return PTR_ERR(priv);
+
+	file->inode = priv;
+	file->size = s.st_size;
+
+	priv->fifo = kfifo_alloc(1024);
+	if (!priv->fifo) {
+		free(priv);
+		return -ENOMEM;
+	}
+
+	return 0;
+}
+
+static int nfs_close(struct device_d *dev, FILE *file)
+{
+	struct file_priv *priv = file->inode;
+
+	nfs_do_close(priv);
+
+	return 0;
+}
+
+static int nfs_write(struct device_d *_dev, FILE *file, const void *inbuf,
+		size_t insize)
+{
+	return -ENOSYS;
+}
+
+static int nfs_read(struct device_d *dev, FILE *file, void *buf, size_t insize)
+{
+	struct file_priv *priv = file->inode;
+	int now, outsize = 0, ret, pos = file->pos;
+
+	while (insize) {
+		now = kfifo_get(priv->fifo, buf, insize);
+		outsize += now;
+		buf += now;
+		insize -= now;
+
+		if (insize) {
+			now = 1024;
+
+			if (pos + now > file->size)
+				now = file->size - pos;
+
+			ret = nfs_read_req(priv, pos, now);
+			if (ret)
+				return ret;
+			pos += now;
+		}
+	}
+
+	return outsize;
+}
+
+static off_t nfs_lseek(struct device_d *dev, FILE *file, off_t pos)
+{
+	struct file_priv *priv = file->inode;
+
+	file->pos = pos;
+	kfifo_reset(priv->fifo);
+
+	return file->pos;
+}
+
+struct nfs_dir {
+	DIR dir;
+	struct xdr_stream stream;
+	struct dirent ent;
+	struct file_priv *priv;
+	uint32_t cookie;
+};
+
+static DIR *nfs_opendir(struct device_d *dev, const char *pathname)
+{
+	struct file_priv *priv;
+	struct stat s;
+	int ret;
+	void *buf;
+	struct nfs_dir *dir;
+	int len;
+
+	priv = nfs_do_open(dev, pathname);
+	if (IS_ERR(priv))
+		return NULL;
+
+	ret = nfs_attr_req(priv, &s);
+	if (ret)
+		return NULL;
+
+	if (!S_ISDIR(s.st_mode))
+		return NULL;
+
+	dir = xzalloc(sizeof(*dir));
+	dir->priv = priv;
+
+	buf = nfs_readdirattr_req(priv, &len, 0);
+	if (!buf)
+		return NULL;
+
+	xdr_init(&dir->stream, buf, len);
+
+	return &dir->dir;
+}
+
+static struct dirent *nfs_readdir(struct device_d *dev, DIR *dir)
+{
+	struct nfs_dir *ndir = (void *)dir;
+	__be32 *p;
+	int ret;
+	int len;
+	struct xdr_stream *xdr = &ndir->stream;
+
+again:
+	p = xdr_inline_decode(xdr, 4);
+	if (!p)
+		goto out_overflow;
+
+	if (*p++ == xdr_zero) {
+		p = xdr_inline_decode(xdr, 4);
+		if (!p)
+			goto out_overflow;
+		if (*p++ == xdr_zero) {
+			void *buf;
+			int len;
+
+			/*
+			 * End of current entries, read next chunk.
+			 */
+
+			free(ndir->stream.buf);
+
+			buf = nfs_readdirattr_req(ndir->priv, &len, ndir->cookie);
+			if (!buf)
+				return NULL;
+
+			xdr_init(&ndir->stream, buf, len);
+
+			goto again;
+		}
+		return NULL; /* -EINVAL */
+	}
+
+	p = xdr_inline_decode(xdr, 4);
+	if (!p)
+		goto out_overflow;
+
+	ret = decode_filename(xdr, ndir->ent.d_name, &len);
+	if (ret)
+		return NULL;
+
+	/*
+	 * The type (size and byte order) of nfscookie isn't defined in
+	 * RFC 1094.  This implementation assumes that it's an XDR uint32.
+	 */
+	p = xdr_inline_decode(xdr, 4);
+	if (!p)
+		goto out_overflow;
+
+	ndir->cookie = be32_to_cpup(p);
+
+	return &ndir->ent;
+
+out_overflow:
+
+	printf("nfs: overflow error\n");
+
+	return NULL;
+
+}
+
+static int nfs_closedir(struct device_d *dev, DIR *dir)
+{
+	struct nfs_dir *ndir = (void *)dir;
+
+	nfs_do_close(ndir->priv);
+	free(ndir->stream.buf);
+	free(ndir);
+
+	return 0;
+}
+
+static int nfs_stat(struct device_d *dev, const char *filename, struct stat *s)
+{
+	struct file_priv *priv;
+
+	priv = nfs_do_stat(dev, filename, s);
+	if (IS_ERR(priv)) {
+		return PTR_ERR(priv);
+	} else {
+		nfs_do_close(priv);
+		return 0;
+	}
+}
+
+static int nfs_probe(struct device_d *dev)
+{
+	struct fs_device_d *fsdev = dev_to_fs_device(dev);
+	struct nfs_priv *npriv = xzalloc(sizeof(struct nfs_priv));
+	char *tmp = xstrdup(fsdev->backingstore);
+	char *path;
+	int ret;
+
+	dev->priv = npriv;
+
+	debug("nfs: mount: %s\n", fsdev->backingstore);
+
+	path = strchr(tmp, ':');
+	if (!path) {
+		ret = -EINVAL;
+		goto err;
+	}
+
+	*path = 0;
+
+	npriv->path = xstrdup(path + 1);
+
+	npriv->server = resolv(tmp);
+
+	debug("nfs: server: %s path: %s\n", tmp, npriv->path);
+
+	npriv->con = net_udp_new(npriv->server, 0, nfs_handler, npriv);
+	if (IS_ERR(npriv->con)) {
+		ret = PTR_ERR(npriv->con);
+		goto err1;
+	}
+
+	/* Need a priviliged source port */
+	net_udp_bind(npriv->con, 1000);
+
+	ret = rpc_lookup_req(npriv, PROG_MOUNT, 1);
+	if (ret) {
+		printf("lookup mount port failed with %d\n", ret);
+		goto err2;
+	}
+
+	ret = rpc_lookup_req(npriv, PROG_NFS, 2);
+	if (ret) {
+		printf("lookup nfs port failed with %d\n", ret);
+		goto err2;
+	}
+
+	ret = nfs_mount_req(npriv);
+	if (ret) {
+		printf("mounting failed with %d\n", ret);
+		goto err2;
+	}
+
+	free(tmp);
+
+	return 0;
+
+err2:
+	net_unregister(npriv->con);
+err1:
+	free(npriv->path);
+err:
+	free(tmp);
+	free(npriv);
+
+	return ret;
+}
+
+static void nfs_remove(struct device_d *dev)
+{
+	struct nfs_priv *npriv = dev->priv;
+
+	nfs_umount_req(npriv);
+
+	net_unregister(npriv->con);
+	free(npriv->path);
+	free(npriv);
+}
+
+static struct fs_driver_d nfs_driver = {
+	.open      = nfs_open,
+	.close     = nfs_close,
+	.read      = nfs_read,
+	.lseek     = nfs_lseek,
+	.opendir   = nfs_opendir,
+	.readdir   = nfs_readdir,
+	.closedir  = nfs_closedir,
+	.stat      = nfs_stat,
+	.create    = nfs_create,
+	.unlink    = nfs_unlink,
+	.mkdir     = nfs_mkdir,
+	.rmdir     = nfs_rmdir,
+	.write     = nfs_write,
+	.truncate  = nfs_truncate,
+	.flags     = 0,
+	.drv = {
+		.probe  = nfs_probe,
+		.remove = nfs_remove,
+		.name = "nfs",
+	}
+};
+
+static int nfs_init(void)
+{
+	return register_fs_driver(&nfs_driver);
+}
+coredevice_initcall(nfs_init);
-- 
1.7.10


_______________________________________________
barebox mailing list
barebox@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/barebox

^ permalink raw reply	[flat|nested] only message in thread

only message in thread, other threads:[~2012-05-13 10:49 UTC | newest]

Thread overview: (only message) (download: mbox.gz / follow: Atom feed)
-- links below jump to the message on this page --
2012-05-13 10:48 [PATCH] fs: Add NFS support Sascha Hauer

This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox