[net-next-2.6.git] / net / 9p / protocol.c

/*
 * net/9p/protocol.c
 *
 * 9P Protocol Support Code
 *
 *  Copyright (C) 2008 by Eric Van Hensbergen <ericvh@gmail.com>
 *
 *  Base on code from Anthony Liguori <aliguori@us.ibm.com>
 *  Copyright (C) 2008 by IBM, Corp.
 *
 *  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:
 *  Free Software Foundation
 *  51 Franklin Street, Fifth Floor
 *  Boston, MA  02111-1301  USA
 *
 */

#include <linux/module.h>
#include <linux/errno.h>
#include <linux/uaccess.h>
#include <linux/slab.h>
#include <linux/sched.h>
#include <linux/types.h>
#include <net/9p/9p.h>
#include <net/9p/client.h>
#include "protocol.h"

#ifndef MIN
#define MIN(a, b) (((a) < (b)) ? (a) : (b))
#endif

#ifndef MAX
#define MAX(a, b) (((a) > (b)) ? (a) : (b))
#endif

#ifndef offset_of
#define offset_of(type, memb) \
	((unsigned long)(&((type *)0)->memb))
#endif
#ifndef container_of
#define container_of(obj, type, memb) \
	((type *)(((char *)obj) - offset_of(type, memb)))
#endif

static int
p9pdu_writef(struct p9_fcall *pdu, int proto_version, const char *fmt, ...);

#ifdef CONFIG_NET_9P_DEBUG
void
p9pdu_dump(int way, struct p9_fcall *pdu)
{
	int i, n;
	u8 *data = pdu->sdata;
	int datalen = pdu->size;
	char buf[255];
	int buflen = 255;

	i = n = 0;
	if (datalen > (buflen-16))
		datalen = buflen-16;
	while (i < datalen) {
		n += scnprintf(buf + n, buflen - n, "%02x ", data[i]);
		if (i%4 == 3)
			n += scnprintf(buf + n, buflen - n, " ");
		if (i%32 == 31)
			n += scnprintf(buf + n, buflen - n, "\n");

		i++;
	}
	n += scnprintf(buf + n, buflen - n, "\n");

	if (way)
		P9_DPRINTK(P9_DEBUG_PKT, "[[[(%d) %s\n", datalen, buf);
	else
		P9_DPRINTK(P9_DEBUG_PKT, "]]](%d) %s\n", datalen, buf);
}
#else
void
p9pdu_dump(int way, struct p9_fcall *pdu)
{
}
#endif
EXPORT_SYMBOL(p9pdu_dump);

void p9stat_free(struct p9_wstat *stbuf)
{
	kfree(stbuf->name);
	kfree(stbuf->uid);
	kfree(stbuf->gid);
	kfree(stbuf->muid);
	kfree(stbuf->extension);
}
EXPORT_SYMBOL(p9stat_free);

static size_t pdu_read(struct p9_fcall *pdu, void *data, size_t size)
{
	size_t len = MIN(pdu->size - pdu->offset, size);
	memcpy(data, &pdu->sdata[pdu->offset], len);
	pdu->offset += len;
	return size - len;
}

static size_t pdu_write(struct p9_fcall *pdu, const void *data, size_t size)
{
	size_t len = MIN(pdu->capacity - pdu->size, size);
	memcpy(&pdu->sdata[pdu->size], data, len);
	pdu->size += len;
	return size - len;
}

static size_t
pdu_write_u(struct p9_fcall *pdu, const char __user *udata, size_t size)
{
	size_t len = MIN(pdu->capacity - pdu->size, size);
	int err = copy_from_user(&pdu->sdata[pdu->size], udata, len);
	if (err)
		printk(KERN_WARNING "pdu_write_u returning: %d\n", err);

	pdu->size += len;
	return size - len;
}

/*
	b - int8_t
	w - int16_t
	d - int32_t
	q - int64_t
	s - string
	S - stat
	Q - qid
	D - data blob (int32_t size followed by void *, results are not freed)
	T - array of strings (int16_t count, followed by strings)
	R - array of qids (int16_t count, followed by qids)
	? - if optional = 1, continue parsing
*/

static int
p9pdu_vreadf(struct p9_fcall *pdu, int proto_version, const char *fmt,
	va_list ap)
{
	const char *ptr;
	int errcode = 0;

	for (ptr = fmt; *ptr; ptr++) {
		switch (*ptr) {
		case 'b':{
				int8_t *val = va_arg(ap, int8_t *);
				if (pdu_read(pdu, val, sizeof(*val))) {
					errcode = -EFAULT;
					break;
				}
			}
			break;
		case 'w':{
				int16_t *val = va_arg(ap, int16_t *);
				__le16 le_val;
				if (pdu_read(pdu, &le_val, sizeof(le_val))) {
					errcode = -EFAULT;
					break;
				}
				*val = le16_to_cpu(le_val);
			}
			break;
		case 'd':{
				int32_t *val = va_arg(ap, int32_t *);
				__le32 le_val;
				if (pdu_read(pdu, &le_val, sizeof(le_val))) {
					errcode = -EFAULT;
					break;
				}
				*val = le32_to_cpu(le_val);
			}
			break;
		case 'q':{
				int64_t *val = va_arg(ap, int64_t *);
				__le64 le_val;
				if (pdu_read(pdu, &le_val, sizeof(le_val))) {
					errcode = -EFAULT;
					break;
				}
				*val = le64_to_cpu(le_val);
			}
			break;
		case 's':{
				char **sptr = va_arg(ap, char **);
				int16_t len;
				int size;

				errcode = p9pdu_readf(pdu, proto_version,
								"w", &len);
				if (errcode)
					break;

				size = MAX(len, 0);

				*sptr = kmalloc(size + 1, GFP_KERNEL);
				if (*sptr == NULL) {
					errcode = -EFAULT;
					break;
				}
				if (pdu_read(pdu, *sptr, size)) {
					errcode = -EFAULT;
					kfree(*sptr);
					*sptr = NULL;
				} else
					(*sptr)[size] = 0;
			}
			break;
		case 'Q':{
				struct p9_qid *qid =
				    va_arg(ap, struct p9_qid *);

				errcode = p9pdu_readf(pdu, proto_version, "bdq",
						      &qid->type, &qid->version,
						      &qid->path);
			}
			break;
		case 'S':{
				struct p9_wstat *stbuf =
				    va_arg(ap, struct p9_wstat *);

				memset(stbuf, 0, sizeof(struct p9_wstat));
				stbuf->n_uid = stbuf->n_gid = stbuf->n_muid =
									-1;
				errcode =
				    p9pdu_readf(pdu, proto_version,
						"wwdQdddqssss?sddd",
						&stbuf->size, &stbuf->type,
						&stbuf->dev, &stbuf->qid,
						&stbuf->mode, &stbuf->atime,
						&stbuf->mtime, &stbuf->length,
						&stbuf->name, &stbuf->uid,
						&stbuf->gid, &stbuf->muid,
						&stbuf->extension,
						&stbuf->n_uid, &stbuf->n_gid,
						&stbuf->n_muid);
				if (errcode)
					p9stat_free(stbuf);
			}
			break;
		case 'D':{
				int32_t *count = va_arg(ap, int32_t *);
				void **data = va_arg(ap, void **);

				errcode =
				    p9pdu_readf(pdu, proto_version, "d", count);
				if (!errcode) {
					*count =
					    MIN(*count,
						pdu->size - pdu->offset);
					*data = &pdu->sdata[pdu->offset];
				}
			}
			break;
		case 'T':{
				int16_t *nwname = va_arg(ap, int16_t *);
				char ***wnames = va_arg(ap, char ***);

				errcode = p9pdu_readf(pdu, proto_version,
								"w", nwname);
				if (!errcode) {
					*wnames =
					    kmalloc(sizeof(char *) * *nwname,
						    GFP_KERNEL);
					if (!*wnames)
						errcode = -ENOMEM;
				}

				if (!errcode) {
					int i;

					for (i = 0; i < *nwname; i++) {
						errcode =
						    p9pdu_readf(pdu,
								proto_version,
								"s",
								&(*wnames)[i]);
						if (errcode)
							break;
					}
				}

				if (errcode) {
					if (*wnames) {
						int i;

						for (i = 0; i < *nwname; i++)
							kfree((*wnames)[i]);
					}
					kfree(*wnames);
					*wnames = NULL;
				}
			}
			break;
		case 'R':{
				int16_t *nwqid = va_arg(ap, int16_t *);
				struct p9_qid **wqids =
				    va_arg(ap, struct p9_qid **);

				*wqids = NULL;

				errcode =
				    p9pdu_readf(pdu, proto_version, "w", nwqid);
				if (!errcode) {
					*wqids =
					    kmalloc(*nwqid *
						    sizeof(struct p9_qid),
						    GFP_KERNEL);
					if (*wqids == NULL)
						errcode = -ENOMEM;
				}

				if (!errcode) {
					int i;

					for (i = 0; i < *nwqid; i++) {
						errcode =
						    p9pdu_readf(pdu,
								proto_version,
								"Q",
								&(*wqids)[i]);
						if (errcode)
							break;
					}
				}

				if (errcode) {
					kfree(*wqids);
					*wqids = NULL;
				}
			}
			break;
		case '?':
			if ((proto_version != p9_proto_2000u) &&
				(proto_version != p9_proto_2000L))
				return 0;
			break;
		default:
			BUG();
			break;
		}

		if (errcode)
			break;
	}

	return errcode;
}

int
p9pdu_vwritef(struct p9_fcall *pdu, int proto_version, const char *fmt,
	va_list ap)
{
	const char *ptr;
	int errcode = 0;

	for (ptr = fmt; *ptr; ptr++) {
		switch (*ptr) {
		case 'b':{
				int8_t val = va_arg(ap, int);
				if (pdu_write(pdu, &val, sizeof(val)))
					errcode = -EFAULT;
			}
			break;
		case 'w':{
				__le16 val = cpu_to_le16(va_arg(ap, int));
				if (pdu_write(pdu, &val, sizeof(val)))
					errcode = -EFAULT;
			}
			break;
		case 'd':{
				__le32 val = cpu_to_le32(va_arg(ap, int32_t));
				if (pdu_write(pdu, &val, sizeof(val)))
					errcode = -EFAULT;
			}
			break;
		case 'q':{
				__le64 val = cpu_to_le64(va_arg(ap, int64_t));
				if (pdu_write(pdu, &val, sizeof(val)))
					errcode = -EFAULT;
			}
			break;
		case 's':{
				const char *sptr = va_arg(ap, const char *);
				int16_t len = 0;
				if (sptr)
					len = MIN(strlen(sptr), USHRT_MAX);

				errcode = p9pdu_writef(pdu, proto_version,
								"w", len);
				if (!errcode && pdu_write(pdu, sptr, len))
					errcode = -EFAULT;
			}
			break;
		case 'Q':{
				const struct p9_qid *qid =
				    va_arg(ap, const struct p9_qid *);
				errcode =
				    p9pdu_writef(pdu, proto_version, "bdq",
						 qid->type, qid->version,
						 qid->path);
			} break;
		case 'S':{
				const struct p9_wstat *stbuf =
				    va_arg(ap, const struct p9_wstat *);
				errcode =
				    p9pdu_writef(pdu, proto_version,
						 "wwdQdddqssss?sddd",
						 stbuf->size, stbuf->type,
						 stbuf->dev, &stbuf->qid,
						 stbuf->mode, stbuf->atime,
						 stbuf->mtime, stbuf->length,
						 stbuf->name, stbuf->uid,
						 stbuf->gid, stbuf->muid,
						 stbuf->extension, stbuf->n_uid,
						 stbuf->n_gid, stbuf->n_muid);
			} break;
		case 'D':{
				int32_t count = va_arg(ap, int32_t);
				const void *data = va_arg(ap, const void *);

				errcode = p9pdu_writef(pdu, proto_version, "d",
									count);
				if (!errcode && pdu_write(pdu, data, count))
					errcode = -EFAULT;
			}
			break;
		case 'U':{
				int32_t count = va_arg(ap, int32_t);
				const char __user *udata =
						va_arg(ap, const void __user *);
				errcode = p9pdu_writef(pdu, proto_version, "d",
									count);
				if (!errcode && pdu_write_u(pdu, udata, count))
					errcode = -EFAULT;
			}
			break;
		case 'T':{
				int16_t nwname = va_arg(ap, int);
				const char **wnames = va_arg(ap, const char **);

				errcode = p9pdu_writef(pdu, proto_version, "w",
									nwname);
				if (!errcode) {
					int i;

					for (i = 0; i < nwname; i++) {
						errcode =
						    p9pdu_writef(pdu,
								proto_version,
								 "s",
								 wnames[i]);
						if (errcode)
							break;
					}
				}
			}
			break;
		case 'R':{
				int16_t nwqid = va_arg(ap, int);
				struct p9_qid *wqids =
				    va_arg(ap, struct p9_qid *);

				errcode = p9pdu_writef(pdu, proto_version, "w",
									nwqid);
				if (!errcode) {
					int i;

					for (i = 0; i < nwqid; i++) {
						errcode =
						    p9pdu_writef(pdu,
								proto_version,
								 "Q",
								 &wqids[i]);
						if (errcode)
							break;
					}
				}
			}
			break;
		case '?':
			if ((proto_version != p9_proto_2000u) &&
				(proto_version != p9_proto_2000L))
				return 0;
			break;
		default:
			BUG();
			break;
		}

		if (errcode)
			break;
	}

	return errcode;
}

int p9pdu_readf(struct p9_fcall *pdu, int proto_version, const char *fmt, ...)
{
	va_list ap;
	int ret;

	va_start(ap, fmt);
	ret = p9pdu_vreadf(pdu, proto_version, fmt, ap);
	va_end(ap);

	return ret;
}

static int
p9pdu_writef(struct p9_fcall *pdu, int proto_version, const char *fmt, ...)
{
	va_list ap;
	int ret;

	va_start(ap, fmt);
	ret = p9pdu_vwritef(pdu, proto_version, fmt, ap);
	va_end(ap);

	return ret;
}

int p9stat_read(char *buf, int len, struct p9_wstat *st, int proto_version)
{
	struct p9_fcall fake_pdu;
	int ret;

	fake_pdu.size = len;
	fake_pdu.capacity = len;
	fake_pdu.sdata = buf;
	fake_pdu.offset = 0;

	ret = p9pdu_readf(&fake_pdu, proto_version, "S", st);
	if (ret) {
		P9_DPRINTK(P9_DEBUG_9P, "<<< p9stat_read failed: %d\n", ret);
		p9pdu_dump(1, &fake_pdu);
	}

	return ret;
}
EXPORT_SYMBOL(p9stat_read);

int p9pdu_prepare(struct p9_fcall *pdu, int16_t tag, int8_t type)
{
	return p9pdu_writef(pdu, 0, "dbw", 0, type, tag);
}

int p9pdu_finalize(struct p9_fcall *pdu)
{
	int size = pdu->size;
	int err;

	pdu->size = 0;
	err = p9pdu_writef(pdu, 0, "d", size);
	pdu->size = size;

#ifdef CONFIG_NET_9P_DEBUG
	if ((p9_debug_level & P9_DEBUG_PKT) == P9_DEBUG_PKT)
		p9pdu_dump(0, pdu);
#endif

	P9_DPRINTK(P9_DEBUG_9P, ">>> size=%d type: %d tag: %d\n", pdu->size,
							pdu->id, pdu->tag);

	return err;
}

void p9pdu_reset(struct p9_fcall *pdu)
{
	pdu->offset = 0;
	pdu->size = 0;
}
Commit	Line	Data
ace51c4d EVH	1	/*
	2	* net/9p/protocol.c
	3	*
	4	* 9P Protocol Support Code
	5	*
	6	* Copyright (C) 2008 by Eric Van Hensbergen <ericvh@gmail.com>
	7	*
	8	* Base on code from Anthony Liguori <aliguori@us.ibm.com>
	9	* Copyright (C) 2008 by IBM, Corp.
	10	*
	11	* This program is free software; you can redistribute it and/or modify
	12	* it under the terms of the GNU General Public License version 2
	13	* as published by the Free Software Foundation.
	14	*
	15	* This program is distributed in the hope that it will be useful,
	16	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	17	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	18	* GNU General Public License for more details.
	19	*
	20	* You should have received a copy of the GNU General Public License
	21	* along with this program; if not, write to:
	22	* Free Software Foundation
	23	* 51 Franklin Street, Fifth Floor
	24	* Boston, MA 02111-1301 USA
	25	*
	26	*/
	27
	28	#include <linux/module.h>
	29	#include <linux/errno.h>
51a87c55	30	#include <linux/uaccess.h>
5a0e3ad6	31	#include <linux/slab.h>
e7f4b8f1	32	#include <linux/sched.h>
beeebc92	33	#include <linux/types.h>
ace51c4d EVH	34	#include <net/9p/9p.h>
	35	#include <net/9p/client.h>
	36	#include "protocol.h"
	37
	38	#ifndef MIN
	39	#define MIN(a, b) (((a) < (b)) ? (a) : (b))
	40	#endif
	41
	42	#ifndef MAX
	43	#define MAX(a, b) (((a) > (b)) ? (a) : (b))
	44	#endif
	45
	46	#ifndef offset_of
	47	#define offset_of(type, memb) \
	48	((unsigned long)(&((type *)0)->memb))
	49	#endif
	50	#ifndef container_of
	51	#define container_of(obj, type, memb) \
	52	((type )(((char )obj) - offset_of(type, memb)))
	53	#endif
	54
	55	static int
342fee1d	56	p9pdu_writef(struct p9_fcall pdu, int proto_version, const char fmt, ...);
ace51c4d	57
0b15a3a5	58	#ifdef CONFIG_NET_9P_DEBUG
51a87c55 EVH	59	void
	60	p9pdu_dump(int way, struct p9_fcall *pdu)
	61	{
	62	int i, n;
	63	u8 *data = pdu->sdata;
	64	int datalen = pdu->size;
	65	char buf[255];
	66	int buflen = 255;
	67
	68	i = n = 0;
	69	if (datalen > (buflen-16))
	70	datalen = buflen-16;
	71	while (i < datalen) {
	72	n += scnprintf(buf + n, buflen - n, "%02x ", data[i]);
	73	if (i%4 == 3)
	74	n += scnprintf(buf + n, buflen - n, " ");
	75	if (i%32 == 31)
	76	n += scnprintf(buf + n, buflen - n, "\n");
	77
	78	i++;
	79	}
	80	n += scnprintf(buf + n, buflen - n, "\n");
	81
	82	if (way)
e7f4b8f1	83	P9_DPRINTK(P9_DEBUG_PKT, "[[[(%d) %s\n", datalen, buf);
51a87c55	84	else
e7f4b8f1	85	P9_DPRINTK(P9_DEBUG_PKT, "]]](%d) %s\n", datalen, buf);
51a87c55	86	}
0b15a3a5 EVH	87	#else
	88	void
	89	p9pdu_dump(int way, struct p9_fcall *pdu)
	90	{
	91	}
	92	#endif
51a87c55 EVH	93	EXPORT_SYMBOL(p9pdu_dump);
51a87c55 EVH	94
ace51c4d EVH	95	void p9stat_free(struct p9_wstat *stbuf)
	96	{
	97	kfree(stbuf->name);
	98	kfree(stbuf->uid);
	99	kfree(stbuf->gid);
	100	kfree(stbuf->muid);
	101	kfree(stbuf->extension);
	102	}
	103	EXPORT_SYMBOL(p9stat_free);
	104
	105	static size_t pdu_read(struct p9_fcall pdu, void data, size_t size)
	106	{
	107	size_t len = MIN(pdu->size - pdu->offset, size);
	108	memcpy(data, &pdu->sdata[pdu->offset], len);
	109	pdu->offset += len;
	110	return size - len;
	111	}
	112
	113	static size_t pdu_write(struct p9_fcall pdu, const void data, size_t size)
	114	{
	115	size_t len = MIN(pdu->capacity - pdu->size, size);
	116	memcpy(&pdu->sdata[pdu->size], data, len);
	117	pdu->size += len;
	118	return size - len;
	119	}
	120
51a87c55 EVH	121	static size_t
	122	pdu_write_u(struct p9_fcall pdu, const char __user udata, size_t size)
	123	{
	124	size_t len = MIN(pdu->capacity - pdu->size, size);
	125	int err = copy_from_user(&pdu->sdata[pdu->size], udata, len);
	126	if (err)
	127	printk(KERN_WARNING "pdu_write_u returning: %d\n", err);
	128
	129	pdu->size += len;
	130	return size - len;
	131	}
	132
ace51c4d EVH	133	/*
	134	b - int8_t
	135	w - int16_t
	136	d - int32_t
	137	q - int64_t
	138	s - string
	139	S - stat
	140	Q - qid
	141	D - data blob (int32_t size followed by void *, results are not freed)
	142	T - array of strings (int16_t count, followed by strings)
	143	R - array of qids (int16_t count, followed by qids)
	144	? - if optional = 1, continue parsing
	145	*/
	146
	147	static int
342fee1d SK	148	p9pdu_vreadf(struct p9_fcall pdu, int proto_version, const char fmt,
342fee1d SK	149	va_list ap)
ace51c4d EVH	150	{
	151	const char *ptr;
	152	int errcode = 0;
	153
	154	for (ptr = fmt; *ptr; ptr++) {
	155	switch (*ptr) {
	156	case 'b':{
	157	int8_t val = va_arg(ap, int8_t );
	158	if (pdu_read(pdu, val, sizeof(*val))) {
	159	errcode = -EFAULT;
	160	break;
	161	}
	162	}
	163	break;
	164	case 'w':{
	165	int16_t val = va_arg(ap, int16_t );
beeebc92 EVH	166	__le16 le_val;
beeebc92 EVH	167	if (pdu_read(pdu, &le_val, sizeof(le_val))) {
ace51c4d EVH	168	errcode = -EFAULT;
	169	break;
	170	}
beeebc92	171	*val = le16_to_cpu(le_val);
ace51c4d EVH	172	}
	173	break;
	174	case 'd':{
	175	int32_t val = va_arg(ap, int32_t );
beeebc92 EVH	176	__le32 le_val;
beeebc92 EVH	177	if (pdu_read(pdu, &le_val, sizeof(le_val))) {
ace51c4d EVH	178	errcode = -EFAULT;
	179	break;
	180	}
beeebc92	181	*val = le32_to_cpu(le_val);
ace51c4d EVH	182	}
	183	break;
	184	case 'q':{
	185	int64_t val = va_arg(ap, int64_t );
beeebc92 EVH	186	__le64 le_val;
beeebc92 EVH	187	if (pdu_read(pdu, &le_val, sizeof(le_val))) {
ace51c4d EVH	188	errcode = -EFAULT;
	189	break;
	190	}
beeebc92	191	*val = le64_to_cpu(le_val);
ace51c4d EVH	192	}
	193	break;
	194	case 's':{
e45c5405	195	char sptr = va_arg(ap, char );
ace51c4d EVH	196	int16_t len;
	197	int size;
	198
342fee1d SK	199	errcode = p9pdu_readf(pdu, proto_version,
342fee1d SK	200	"w", &len);
ace51c4d EVH	201	if (errcode)
	202	break;
	203
	204	size = MAX(len, 0);
	205
e45c5405 EVH	206	*sptr = kmalloc(size + 1, GFP_KERNEL);
e45c5405 EVH	207	if (*sptr == NULL) {
ace51c4d EVH	208	errcode = -EFAULT;
	209	break;
	210	}
e45c5405	211	if (pdu_read(pdu, *sptr, size)) {
ace51c4d	212	errcode = -EFAULT;
e45c5405 EVH	213	kfree(*sptr);
e45c5405 EVH	214	*sptr = NULL;
ace51c4d	215	} else
e45c5405	216	(*sptr)[size] = 0;
ace51c4d EVH	217	}
	218	break;
	219	case 'Q':{
	220	struct p9_qid *qid =
	221	va_arg(ap, struct p9_qid *);
	222
342fee1d	223	errcode = p9pdu_readf(pdu, proto_version, "bdq",
ace51c4d EVH	224	&qid->type, &qid->version,
	225	&qid->path);
	226	}
	227	break;
	228	case 'S':{
	229	struct p9_wstat *stbuf =
	230	va_arg(ap, struct p9_wstat *);
	231
f0a0ac2e	232	memset(stbuf, 0, sizeof(struct p9_wstat));
ace51c4d	233	stbuf->n_uid = stbuf->n_gid = stbuf->n_muid =
f0a0ac2e	234	-1;
ace51c4d	235	errcode =
342fee1d	236	p9pdu_readf(pdu, proto_version,
ace51c4d EVH	237	"wwdQdddqssss?sddd",
	238	&stbuf->size, &stbuf->type,
	239	&stbuf->dev, &stbuf->qid,
	240	&stbuf->mode, &stbuf->atime,
	241	&stbuf->mtime, &stbuf->length,
	242	&stbuf->name, &stbuf->uid,
	243	&stbuf->gid, &stbuf->muid,
	244	&stbuf->extension,
	245	&stbuf->n_uid, &stbuf->n_gid,
	246	&stbuf->n_muid);
	247	if (errcode)
	248	p9stat_free(stbuf);
	249	}
	250	break;
	251	case 'D':{
	252	int32_t count = va_arg(ap, int32_t );
	253	void data = va_arg(ap, void );
	254
	255	errcode =
342fee1d	256	p9pdu_readf(pdu, proto_version, "d", count);
ace51c4d EVH	257	if (!errcode) {
	258	*count =
	259	MIN(*count,
	260	pdu->size - pdu->offset);
	261	*data = &pdu->sdata[pdu->offset];
	262	}
	263	}
	264	break;
	265	case 'T':{
	266	int16_t nwname = va_arg(ap, int16_t );
	267	char *wnames = va_arg(ap, char *);
	268
342fee1d SK	269	errcode = p9pdu_readf(pdu, proto_version,
342fee1d SK	270	"w", nwname);
ace51c4d EVH	271	if (!errcode) {
	272	*wnames =
	273	kmalloc(sizeof(char ) *nwname,
	274	GFP_KERNEL);
	275	if (!*wnames)
	276	errcode = -ENOMEM;
	277	}
	278
	279	if (!errcode) {
	280	int i;
	281
	282	for (i = 0; i < *nwname; i++) {
	283	errcode =
342fee1d SK	284	p9pdu_readf(pdu,
342fee1d SK	285	proto_version,
ace51c4d EVH	286	"s",
	287	&(*wnames)[i]);
	288	if (errcode)
	289	break;
	290	}
	291	}
	292
	293	if (errcode) {
	294	if (*wnames) {
	295	int i;
	296
	297	for (i = 0; i < *nwname; i++)
	298	kfree((*wnames)[i]);
	299	}
	300	kfree(*wnames);
	301	*wnames = NULL;
	302	}
	303	}
	304	break;
	305	case 'R':{
	306	int16_t nwqid = va_arg(ap, int16_t );
	307	struct p9_qid **wqids =
	308	va_arg(ap, struct p9_qid **);
	309
	310	*wqids = NULL;
	311
	312	errcode =
342fee1d	313	p9pdu_readf(pdu, proto_version, "w", nwqid);
ace51c4d EVH	314	if (!errcode) {
	315	*wqids =
	316	kmalloc(nwqid
	317	sizeof(struct p9_qid),
	318	GFP_KERNEL);
	319	if (*wqids == NULL)
	320	errcode = -ENOMEM;
	321	}
	322
	323	if (!errcode) {
	324	int i;
	325
	326	for (i = 0; i < *nwqid; i++) {
	327	errcode =
342fee1d SK	328	p9pdu_readf(pdu,
342fee1d SK	329	proto_version,
ace51c4d EVH	330	"Q",
	331	&(*wqids)[i]);
	332	if (errcode)
	333	break;
	334	}
	335	}
	336
	337	if (errcode) {
	338	kfree(*wqids);
	339	*wqids = NULL;
	340	}
	341	}
	342	break;
	343	case '?':
c56e4acf SK	344	if ((proto_version != p9_proto_2000u) &&
c56e4acf SK	345	(proto_version != p9_proto_2000L))
ace51c4d EVH	346	return 0;
	347	break;
	348	default:
	349	BUG();
	350	break;
	351	}
	352
	353	if (errcode)
	354	break;
	355	}
	356
	357	return errcode;
	358	}
	359
	360	int
342fee1d SK	361	p9pdu_vwritef(struct p9_fcall pdu, int proto_version, const char fmt,
342fee1d SK	362	va_list ap)
ace51c4d EVH	363	{
	364	const char *ptr;
	365	int errcode = 0;
	366
	367	for (ptr = fmt; *ptr; ptr++) {
	368	switch (*ptr) {
	369	case 'b':{
	370	int8_t val = va_arg(ap, int);
	371	if (pdu_write(pdu, &val, sizeof(val)))
	372	errcode = -EFAULT;
	373	}
	374	break;
	375	case 'w':{
beeebc92	376	__le16 val = cpu_to_le16(va_arg(ap, int));
ace51c4d EVH	377	if (pdu_write(pdu, &val, sizeof(val)))
	378	errcode = -EFAULT;
	379	}
	380	break;
	381	case 'd':{
beeebc92	382	__le32 val = cpu_to_le32(va_arg(ap, int32_t));
ace51c4d EVH	383	if (pdu_write(pdu, &val, sizeof(val)))
	384	errcode = -EFAULT;
	385	}
	386	break;
	387	case 'q':{
beeebc92	388	__le64 val = cpu_to_le64(va_arg(ap, int64_t));
ace51c4d EVH	389	if (pdu_write(pdu, &val, sizeof(val)))
	390	errcode = -EFAULT;
	391	}
	392	break;
	393	case 's':{
e45c5405	394	const char sptr = va_arg(ap, const char );
ace51c4d	395	int16_t len = 0;
e45c5405	396	if (sptr)
4be929be	397	len = MIN(strlen(sptr), USHRT_MAX);
ace51c4d	398
342fee1d SK	399	errcode = p9pdu_writef(pdu, proto_version,
342fee1d SK	400	"w", len);
e45c5405	401	if (!errcode && pdu_write(pdu, sptr, len))
ace51c4d EVH	402	errcode = -EFAULT;
	403	}
	404	break;
	405	case 'Q':{
	406	const struct p9_qid *qid =
	407	va_arg(ap, const struct p9_qid *);
	408	errcode =
342fee1d	409	p9pdu_writef(pdu, proto_version, "bdq",
ace51c4d EVH	410	qid->type, qid->version,
	411	qid->path);
	412	} break;
	413	case 'S':{
	414	const struct p9_wstat *stbuf =
	415	va_arg(ap, const struct p9_wstat *);
	416	errcode =
342fee1d	417	p9pdu_writef(pdu, proto_version,
ace51c4d EVH	418	"wwdQdddqssss?sddd",
ace51c4d EVH	419	stbuf->size, stbuf->type,
51a87c55	420	stbuf->dev, &stbuf->qid,
ace51c4d EVH	421	stbuf->mode, stbuf->atime,
	422	stbuf->mtime, stbuf->length,
	423	stbuf->name, stbuf->uid,
	424	stbuf->gid, stbuf->muid,
	425	stbuf->extension, stbuf->n_uid,
	426	stbuf->n_gid, stbuf->n_muid);
	427	} break;
	428	case 'D':{
	429	int32_t count = va_arg(ap, int32_t);
	430	const void data = va_arg(ap, const void );
	431
342fee1d SK	432	errcode = p9pdu_writef(pdu, proto_version, "d",
342fee1d SK	433	count);
ace51c4d EVH	434	if (!errcode && pdu_write(pdu, data, count))
	435	errcode = -EFAULT;
	436	}
	437	break;
51a87c55 EVH	438	case 'U':{
	439	int32_t count = va_arg(ap, int32_t);
	440	const char __user *udata =
e45c5405	441	va_arg(ap, const void __user *);
342fee1d SK	442	errcode = p9pdu_writef(pdu, proto_version, "d",
342fee1d SK	443	count);
51a87c55 EVH	444	if (!errcode && pdu_write_u(pdu, udata, count))
	445	errcode = -EFAULT;
	446	}
	447	break;
ace51c4d EVH	448	case 'T':{
	449	int16_t nwname = va_arg(ap, int);
	450	const char wnames = va_arg(ap, const char );
	451
342fee1d SK	452	errcode = p9pdu_writef(pdu, proto_version, "w",
342fee1d SK	453	nwname);
ace51c4d EVH	454	if (!errcode) {
	455	int i;
	456
	457	for (i = 0; i < nwname; i++) {
	458	errcode =
342fee1d SK	459	p9pdu_writef(pdu,
342fee1d SK	460	proto_version,
ace51c4d EVH	461	"s",
	462	wnames[i]);
	463	if (errcode)
	464	break;
	465	}
	466	}
	467	}
	468	break;
	469	case 'R':{
	470	int16_t nwqid = va_arg(ap, int);
	471	struct p9_qid *wqids =
	472	va_arg(ap, struct p9_qid *);
	473
342fee1d SK	474	errcode = p9pdu_writef(pdu, proto_version, "w",
342fee1d SK	475	nwqid);
ace51c4d EVH	476	if (!errcode) {
	477	int i;
	478
	479	for (i = 0; i < nwqid; i++) {
	480	errcode =
342fee1d SK	481	p9pdu_writef(pdu,
342fee1d SK	482	proto_version,
ace51c4d EVH	483	"Q",
	484	&wqids[i]);
	485	if (errcode)
	486	break;
	487	}
	488	}
	489	}
	490	break;
	491	case '?':
c56e4acf SK	492	if ((proto_version != p9_proto_2000u) &&
c56e4acf SK	493	(proto_version != p9_proto_2000L))
ace51c4d EVH	494	return 0;
	495	break;
	496	default:
	497	BUG();
	498	break;
	499	}
	500
	501	if (errcode)
	502	break;
	503	}
	504
	505	return errcode;
	506	}
	507
342fee1d	508	int p9pdu_readf(struct p9_fcall pdu, int proto_version, const char fmt, ...)
ace51c4d EVH	509	{
	510	va_list ap;
	511	int ret;
	512
	513	va_start(ap, fmt);
342fee1d	514	ret = p9pdu_vreadf(pdu, proto_version, fmt, ap);
ace51c4d EVH	515	va_end(ap);
	516
	517	return ret;
	518	}
	519
	520	static int
342fee1d	521	p9pdu_writef(struct p9_fcall pdu, int proto_version, const char fmt, ...)
ace51c4d EVH	522	{
	523	va_list ap;
	524	int ret;
	525
	526	va_start(ap, fmt);
342fee1d	527	ret = p9pdu_vwritef(pdu, proto_version, fmt, ap);
ace51c4d EVH	528	va_end(ap);
	529
	530	return ret;
	531	}
51a87c55	532
342fee1d	533	int p9stat_read(char buf, int len, struct p9_wstat st, int proto_version)
02da398b EVH	534	{
02da398b EVH	535	struct p9_fcall fake_pdu;
e7f4b8f1	536	int ret;
02da398b EVH	537
	538	fake_pdu.size = len;
	539	fake_pdu.capacity = len;
	540	fake_pdu.sdata = buf;
	541	fake_pdu.offset = 0;
	542
342fee1d	543	ret = p9pdu_readf(&fake_pdu, proto_version, "S", st);
e7f4b8f1 EVH	544	if (ret) {
	545	P9_DPRINTK(P9_DEBUG_9P, "<<< p9stat_read failed: %d\n", ret);
	546	p9pdu_dump(1, &fake_pdu);
	547	}
	548
	549	return ret;
02da398b EVH	550	}
	551	EXPORT_SYMBOL(p9stat_read);
	552
51a87c55 EVH	553	int p9pdu_prepare(struct p9_fcall *pdu, int16_t tag, int8_t type)
	554	{
	555	return p9pdu_writef(pdu, 0, "dbw", 0, type, tag);
	556	}
	557
	558	int p9pdu_finalize(struct p9_fcall *pdu)
	559	{
	560	int size = pdu->size;
	561	int err;
	562
	563	pdu->size = 0;
	564	err = p9pdu_writef(pdu, 0, "d", size);
	565	pdu->size = size;
	566
0b15a3a5	567	#ifdef CONFIG_NET_9P_DEBUG
e7f4b8f1	568	if ((p9_debug_level & P9_DEBUG_PKT) == P9_DEBUG_PKT)
51a87c55	569	p9pdu_dump(0, pdu);
0b15a3a5	570	#endif
51a87c55	571
e7f4b8f1 EVH	572	P9_DPRINTK(P9_DEBUG_9P, ">>> size=%d type: %d tag: %d\n", pdu->size,
	573	pdu->id, pdu->tag);
	574
51a87c55 EVH	575	return err;
	576	}
	577
	578	void p9pdu_reset(struct p9_fcall *pdu)
	579	{
	580	pdu->offset = 0;
	581	pdu->size = 0;
	582	}