[deliverable/linux.git] / fs / 9p / trans_sock.c

/*
 * linux/fs/9p/trans_socket.c
 *
 * Socket Transport Layer
 *
 *  Copyright (C) 2004-2005 by Latchesar Ionkov <lucho@ionkov.net>
 *  Copyright (C) 2004 by Eric Van Hensbergen <ericvh@gmail.com>
 *  Copyright (C) 1997-2002 by Ron Minnich <rminnich@sarnoff.com>
 *  Copyright (C) 1995, 1996 by Olaf Kirch <okir@monad.swb.de>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  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/config.h>
#include <linux/in.h>
#include <linux/module.h>
#include <linux/net.h>
#include <linux/ipv6.h>
#include <linux/errno.h>
#include <linux/kernel.h>
#include <linux/un.h>
#include <asm/uaccess.h>
#include <linux/inet.h>
#include <linux/idr.h>
#include <linux/file.h>

#include "debug.h"
#include "v9fs.h"
#include "transport.h"

#define V9FS_PORT 564

struct v9fs_trans_sock {
	struct socket *s;
	struct file *filp;
};

/**
 * v9fs_sock_recv - receive from a socket
 * @v9ses: session information
 * @v: buffer to receive data into
 * @len: size of receive buffer
 *
 */

static int v9fs_sock_recv(struct v9fs_transport *trans, void *v, int len)
{
	int ret;
	struct v9fs_trans_sock *ts;

	if (!trans || trans->status == Disconnected) {
		dprintk(DEBUG_ERROR, "disconnected ...\n");
		return -EREMOTEIO;
	}

	ts = trans->priv;

	if (!(ts->filp->f_flags & O_NONBLOCK))
		dprintk(DEBUG_ERROR, "blocking read ...\n");

	ret = kernel_read(ts->filp, ts->filp->f_pos, v, len);
	if (ret <= 0) {
		if (ret != -ERESTARTSYS && ret != -EAGAIN)
			trans->status = Disconnected;
	}

	return ret;
}

/**
 * v9fs_sock_send - send to a socket
 * @v9ses: session information
 * @v: buffer to send data from
 * @len: size of send buffer
 *
 */

static int v9fs_sock_send(struct v9fs_transport *trans, void *v, int len)
{
	int ret;
	mm_segment_t oldfs;
	struct v9fs_trans_sock *ts;

	if (!trans || trans->status == Disconnected) {
		dprintk(DEBUG_ERROR, "disconnected ...\n");
		return -EREMOTEIO;
	}

	ts = trans->priv;
	if (!ts) {
		dprintk(DEBUG_ERROR, "no transport ...\n");
		return -EREMOTEIO;
	}

	if (!(ts->filp->f_flags & O_NONBLOCK))
		dprintk(DEBUG_ERROR, "blocking write ...\n");

	oldfs = get_fs();
	set_fs(get_ds());
	ret = vfs_write(ts->filp, (void __user *)v, len, &ts->filp->f_pos);
	set_fs(oldfs);

	if (ret < 0) {
		if (ret != -ERESTARTSYS)
			trans->status = Disconnected;
	}

	return ret;
}

static unsigned int v9fs_sock_poll(struct v9fs_transport *trans,
	struct poll_table_struct *pt) {

	int ret;
	struct v9fs_trans_sock *ts;
	mm_segment_t oldfs;

	if (!trans) {
		dprintk(DEBUG_ERROR, "no transport\n");
		return -EIO;
	}

	ts = trans->priv;
	if (trans->status != Connected || !ts) {
		dprintk(DEBUG_ERROR, "transport disconnected: %d\n", trans->status);
		return -EIO;
	}

	oldfs = get_fs();
	set_fs(get_ds());

	if (!ts->filp->f_op || !ts->filp->f_op->poll) {
		dprintk(DEBUG_ERROR, "no poll operation\n");
		ret = -EIO;
		goto end;
	}

	ret = ts->filp->f_op->poll(ts->filp, pt);

end:
	set_fs(oldfs);
	return ret;
}


/**
 * v9fs_tcp_init - initialize TCP socket
 * @v9ses: session information
 * @addr: address of server to mount
 * @data: mount options
 *
 */

static int
v9fs_tcp_init(struct v9fs_session_info *v9ses, const char *addr, char *data)
{
	struct socket *csocket = NULL;
	struct sockaddr_in sin_server;
	int rc = 0;
	struct v9fs_trans_sock *ts = NULL;
	struct v9fs_transport *trans = v9ses->transport;
	int fd;

	trans->status = Disconnected;

	ts = kmalloc(sizeof(struct v9fs_trans_sock), GFP_KERNEL);

	if (!ts)
		return -ENOMEM;

	trans->priv = ts;
	ts->s = NULL;
	ts->filp = NULL;

	if (!addr)
		return -EINVAL;

	dprintk(DEBUG_TRANS, "Connecting to %s\n", addr);

	sin_server.sin_family = AF_INET;
	sin_server.sin_addr.s_addr = in_aton(addr);
	sin_server.sin_port = htons(v9ses->port);
	sock_create_kern(PF_INET, SOCK_STREAM, IPPROTO_TCP, &csocket);
	rc = csocket->ops->connect(csocket,
				   (struct sockaddr *)&sin_server,
				   sizeof(struct sockaddr_in), 0);
	if (rc < 0) {
		eprintk(KERN_ERR,
			"v9fs_trans_tcp: problem connecting socket to %s\n",
			addr);
		return rc;
	}
	csocket->sk->sk_allocation = GFP_NOIO;

	fd = sock_map_fd(csocket);
	if (fd < 0) {
		sock_release(csocket);
		kfree(ts);
		trans->priv = NULL;
		return fd;
	}

	ts->s = csocket;
	ts->filp = fget(fd);
	ts->filp->f_flags |= O_NONBLOCK;
	trans->status = Connected;

	return 0;
}

/**
 * v9fs_unix_init - initialize UNIX domain socket
 * @v9ses: session information
 * @dev_name: path to named pipe
 * @data: mount options
 *
 */

static int
v9fs_unix_init(struct v9fs_session_info *v9ses, const char *dev_name,
	       char *data)
{
	int rc, fd;
	struct socket *csocket;
	struct sockaddr_un sun_server;
	struct v9fs_transport *trans;
	struct v9fs_trans_sock *ts;

	rc = 0;
	csocket = NULL;
	trans = v9ses->transport;

	trans->status = Disconnected;

	if (strlen(dev_name) > UNIX_PATH_MAX) {
		eprintk(KERN_ERR, "v9fs_trans_unix: address too long: %s\n",
			dev_name);
		return -ENOMEM;
	}

	ts = kmalloc(sizeof(struct v9fs_trans_sock), GFP_KERNEL);
	if (!ts)
		return -ENOMEM;

	trans->priv = ts;
	ts->s = NULL;
	ts->filp = NULL;

	sun_server.sun_family = PF_UNIX;
	strcpy(sun_server.sun_path, dev_name);
	sock_create_kern(PF_UNIX, SOCK_STREAM, 0, &csocket);
	rc = csocket->ops->connect(csocket, (struct sockaddr *)&sun_server,
		sizeof(struct sockaddr_un) - 1, 0);	/* -1 *is* important */
	if (rc < 0) {
		eprintk(KERN_ERR,
			"v9fs_trans_unix: problem connecting socket: %s: %d\n",
			dev_name, rc);
		return rc;
	}
	csocket->sk->sk_allocation = GFP_NOIO;

	fd = sock_map_fd(csocket);
	if (fd < 0) {
		sock_release(csocket);
		kfree(ts);
		trans->priv = NULL;
		return fd;
	}

	ts->s = csocket;
	ts->filp = fget(fd);
	ts->filp->f_flags |= O_NONBLOCK;
	trans->status = Connected;

	return 0;
}

/**
 * v9fs_sock_close - shutdown socket
 * @trans: private socket structure
 *
 */

static void v9fs_sock_close(struct v9fs_transport *trans)
{
	struct v9fs_trans_sock *ts;

	if (!trans)
		return;

	ts = trans->priv;

	if ((ts) && (ts->filp)) {
		fput(ts->filp);
		ts->filp = NULL;
		ts->s = NULL;
		trans->status = Disconnected;
	}

	kfree(ts);

	trans->priv = NULL;
}

struct v9fs_transport v9fs_trans_tcp = {
	.init = v9fs_tcp_init,
	.write = v9fs_sock_send,
	.read = v9fs_sock_recv,
	.close = v9fs_sock_close,
	.poll = v9fs_sock_poll,
};

struct v9fs_transport v9fs_trans_unix = {
	.init = v9fs_unix_init,
	.write = v9fs_sock_send,
	.read = v9fs_sock_recv,
	.close = v9fs_sock_close,
	.poll = v9fs_sock_poll,
};
Commit	Line	Data
426cc91a EVH	1	/*
	2	* linux/fs/9p/trans_socket.c
	3	*
	4	* Socket Transport Layer
	5	*
3cf6429a	6	* Copyright (C) 2004-2005 by Latchesar Ionkov <lucho@ionkov.net>
426cc91a EVH	7	* Copyright (C) 2004 by Eric Van Hensbergen <ericvh@gmail.com>
	8	* Copyright (C) 1997-2002 by Ron Minnich <rminnich@sarnoff.com>
	9	* Copyright (C) 1995, 1996 by Olaf Kirch <okir@monad.swb.de>
	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 as published by
	13	* the Free Software Foundation; either version 2 of the License, or
	14	* (at your option) any later version.
	15	*
	16	* This program is distributed in the hope that it will be useful,
	17	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	18	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	19	* GNU General Public License for more details.
	20	*
	21	* You should have received a copy of the GNU General Public License
	22	* along with this program; if not, write to:
	23	* Free Software Foundation
	24	* 51 Franklin Street, Fifth Floor
	25	* Boston, MA 02111-1301 USA
	26	*
	27	*/
	28
	29	#include <linux/config.h>
14c85021	30	#include <linux/in.h>
426cc91a EVH	31	#include <linux/module.h>
	32	#include <linux/net.h>
	33	#include <linux/ipv6.h>
	34	#include <linux/errno.h>
	35	#include <linux/kernel.h>
	36	#include <linux/un.h>
	37	#include <asm/uaccess.h>
	38	#include <linux/inet.h>
	39	#include <linux/idr.h>
3cf6429a	40	#include <linux/file.h>
426cc91a EVH	41
	42	#include "debug.h"
	43	#include "v9fs.h"
	44	#include "transport.h"
	45
	46	#define V9FS_PORT 564
	47
	48	struct v9fs_trans_sock {
	49	struct socket *s;
3cf6429a	50	struct file *filp;
426cc91a EVH	51	};
	52
	53	/**
	54	* v9fs_sock_recv - receive from a socket
	55	* @v9ses: session information
	56	* @v: buffer to receive data into
	57	* @len: size of receive buffer
	58	*
	59	*/
	60
	61	static int v9fs_sock_recv(struct v9fs_transport trans, void v, int len)
	62	{
3cf6429a LI	63	int ret;
3cf6429a LI	64	struct v9fs_trans_sock *ts;
426cc91a	65
3cf6429a LI	66	if (!trans \|\| trans->status == Disconnected) {
3cf6429a LI	67	dprintk(DEBUG_ERROR, "disconnected ...\n");
426cc91a	68	return -EREMOTEIO;
3cf6429a	69	}
426cc91a	70
3cf6429a	71	ts = trans->priv;
426cc91a	72
3cf6429a LI	73	if (!(ts->filp->f_flags & O_NONBLOCK))
3cf6429a LI	74	dprintk(DEBUG_ERROR, "blocking read ...\n");
426cc91a	75
3cf6429a LI	76	ret = kernel_read(ts->filp, ts->filp->f_pos, v, len);
	77	if (ret <= 0) {
	78	if (ret != -ERESTARTSYS && ret != -EAGAIN)
426cc91a EVH	79	trans->status = Disconnected;
	80	}
	81
3cf6429a	82	return ret;
426cc91a EVH	83	}
	84
	85	/**
	86	* v9fs_sock_send - send to a socket
	87	* @v9ses: session information
	88	* @v: buffer to send data from
	89	* @len: size of send buffer
	90	*
	91	*/
	92
	93	static int v9fs_sock_send(struct v9fs_transport trans, void v, int len)
	94	{
3cf6429a	95	int ret;
426cc91a	96	mm_segment_t oldfs;
3cf6429a	97	struct v9fs_trans_sock *ts;
426cc91a	98
3cf6429a LI	99	if (!trans \|\| trans->status == Disconnected) {
	100	dprintk(DEBUG_ERROR, "disconnected ...\n");
	101	return -EREMOTEIO;
	102	}
	103
	104	ts = trans->priv;
	105	if (!ts) {
	106	dprintk(DEBUG_ERROR, "no transport ...\n");
	107	return -EREMOTEIO;
	108	}
426cc91a	109
3cf6429a LI	110	if (!(ts->filp->f_flags & O_NONBLOCK))
3cf6429a LI	111	dprintk(DEBUG_ERROR, "blocking write ...\n");
426cc91a EVH	112
	113	oldfs = get_fs();
	114	set_fs(get_ds());
3cf6429a	115	ret = vfs_write(ts->filp, (void __user *)v, len, &ts->filp->f_pos);
426cc91a EVH	116	set_fs(oldfs);
426cc91a EVH	117
3cf6429a LI	118	if (ret < 0) {
3cf6429a LI	119	if (ret != -ERESTARTSYS)
426cc91a EVH	120	trans->status = Disconnected;
	121	}
	122
3cf6429a LI	123	return ret;
	124	}
	125
	126	static unsigned int v9fs_sock_poll(struct v9fs_transport *trans,
	127	struct poll_table_struct *pt) {
	128
	129	int ret;
	130	struct v9fs_trans_sock *ts;
	131	mm_segment_t oldfs;
	132
	133	if (!trans) {
	134	dprintk(DEBUG_ERROR, "no transport\n");
	135	return -EIO;
	136	}
	137
	138	ts = trans->priv;
	139	if (trans->status != Connected \|\| !ts) {
	140	dprintk(DEBUG_ERROR, "transport disconnected: %d\n", trans->status);
	141	return -EIO;
	142	}
	143
	144	oldfs = get_fs();
	145	set_fs(get_ds());
	146
	147	if (!ts->filp->f_op \|\| !ts->filp->f_op->poll) {
	148	dprintk(DEBUG_ERROR, "no poll operation\n");
	149	ret = -EIO;
	150	goto end;
	151	}
	152
	153	ret = ts->filp->f_op->poll(ts->filp, pt);
	154
	155	end:
	156	set_fs(oldfs);
	157	return ret;
426cc91a EVH	158	}
426cc91a EVH	159
3cf6429a	160
426cc91a EVH	161	/**
	162	* v9fs_tcp_init - initialize TCP socket
	163	* @v9ses: session information
	164	* @addr: address of server to mount
	165	* @data: mount options
	166	*
	167	*/
	168
	169	static int
	170	v9fs_tcp_init(struct v9fs_session_info v9ses, const char addr, char *data)
	171	{
	172	struct socket *csocket = NULL;
	173	struct sockaddr_in sin_server;
	174	int rc = 0;
	175	struct v9fs_trans_sock *ts = NULL;
	176	struct v9fs_transport *trans = v9ses->transport;
3cf6429a	177	int fd;
426cc91a	178
3cf6429a	179	trans->status = Disconnected;
426cc91a EVH	180
	181	ts = kmalloc(sizeof(struct v9fs_trans_sock), GFP_KERNEL);
	182
	183	if (!ts)
	184	return -ENOMEM;
	185
	186	trans->priv = ts;
	187	ts->s = NULL;
3cf6429a	188	ts->filp = NULL;
426cc91a EVH	189
	190	if (!addr)
	191	return -EINVAL;
	192
	193	dprintk(DEBUG_TRANS, "Connecting to %s\n", addr);
	194
	195	sin_server.sin_family = AF_INET;
	196	sin_server.sin_addr.s_addr = in_aton(addr);
	197	sin_server.sin_port = htons(v9ses->port);
	198	sock_create_kern(PF_INET, SOCK_STREAM, IPPROTO_TCP, &csocket);
	199	rc = csocket->ops->connect(csocket,
	200	(struct sockaddr *)&sin_server,
	201	sizeof(struct sockaddr_in), 0);
	202	if (rc < 0) {
	203	eprintk(KERN_ERR,
	204	"v9fs_trans_tcp: problem connecting socket to %s\n",
	205	addr);
	206	return rc;
	207	}
	208	csocket->sk->sk_allocation = GFP_NOIO;
3cf6429a LI	209
	210	fd = sock_map_fd(csocket);
	211	if (fd < 0) {
	212	sock_release(csocket);
	213	kfree(ts);
	214	trans->priv = NULL;
	215	return fd;
	216	}
	217
426cc91a	218	ts->s = csocket;
3cf6429a LI	219	ts->filp = fget(fd);
3cf6429a LI	220	ts->filp->f_flags \|= O_NONBLOCK;
426cc91a EVH	221	trans->status = Connected;
	222
	223	return 0;
	224	}
	225
	226	/**
	227	* v9fs_unix_init - initialize UNIX domain socket
	228	* @v9ses: session information
	229	* @dev_name: path to named pipe
	230	* @data: mount options
	231	*
	232	*/
	233
	234	static int
	235	v9fs_unix_init(struct v9fs_session_info v9ses, const char dev_name,
	236	char *data)
	237	{
3cf6429a	238	int rc, fd;
426cc91a EVH	239	struct socket *csocket;
	240	struct sockaddr_un sun_server;
	241	struct v9fs_transport *trans;
	242	struct v9fs_trans_sock *ts;
	243
	244	rc = 0;
	245	csocket = NULL;
	246	trans = v9ses->transport;
	247
3cf6429a LI	248	trans->status = Disconnected;
3cf6429a LI	249
426cc91a EVH	250	if (strlen(dev_name) > UNIX_PATH_MAX) {
	251	eprintk(KERN_ERR, "v9fs_trans_unix: address too long: %s\n",
	252	dev_name);
	253	return -ENOMEM;
	254	}
	255
	256	ts = kmalloc(sizeof(struct v9fs_trans_sock), GFP_KERNEL);
	257	if (!ts)
	258	return -ENOMEM;
	259
	260	trans->priv = ts;
	261	ts->s = NULL;
3cf6429a	262	ts->filp = NULL;
426cc91a EVH	263
	264	sun_server.sun_family = PF_UNIX;
	265	strcpy(sun_server.sun_path, dev_name);
	266	sock_create_kern(PF_UNIX, SOCK_STREAM, 0, &csocket);
	267	rc = csocket->ops->connect(csocket, (struct sockaddr *)&sun_server,
	268	sizeof(struct sockaddr_un) - 1, 0); /* -1 is important */
	269	if (rc < 0) {
	270	eprintk(KERN_ERR,
	271	"v9fs_trans_unix: problem connecting socket: %s: %d\n",
	272	dev_name, rc);
	273	return rc;
	274	}
	275	csocket->sk->sk_allocation = GFP_NOIO;
3cf6429a LI	276
	277	fd = sock_map_fd(csocket);
	278	if (fd < 0) {
	279	sock_release(csocket);
	280	kfree(ts);
	281	trans->priv = NULL;
	282	return fd;
	283	}
	284
426cc91a	285	ts->s = csocket;
3cf6429a LI	286	ts->filp = fget(fd);
3cf6429a LI	287	ts->filp->f_flags \|= O_NONBLOCK;
426cc91a EVH	288	trans->status = Connected;
	289
	290	return 0;
	291	}
	292
	293	/**
	294	* v9fs_sock_close - shutdown socket
	295	* @trans: private socket structure
	296	*
	297	*/
	298
	299	static void v9fs_sock_close(struct v9fs_transport *trans)
	300	{
cb2e87a6 EVH	301	struct v9fs_trans_sock *ts;
	302
	303	if (!trans)
	304	return;
	305
	306	ts = trans->priv;
426cc91a	307
3cf6429a LI	308	if ((ts) && (ts->filp)) {
	309	fput(ts->filp);
	310	ts->filp = NULL;
426cc91a EVH	311	ts->s = NULL;
426cc91a EVH	312	trans->status = Disconnected;
426cc91a EVH	313	}
426cc91a EVH	314
f99d49ad	315	kfree(ts);
cb2e87a6 EVH	316
cb2e87a6 EVH	317	trans->priv = NULL;
426cc91a EVH	318	}
	319
	320	struct v9fs_transport v9fs_trans_tcp = {
	321	.init = v9fs_tcp_init,
	322	.write = v9fs_sock_send,
	323	.read = v9fs_sock_recv,
	324	.close = v9fs_sock_close,
3cf6429a	325	.poll = v9fs_sock_poll,
426cc91a EVH	326	};
	327
	328	struct v9fs_transport v9fs_trans_unix = {
	329	.init = v9fs_unix_init,
	330	.write = v9fs_sock_send,
	331	.read = v9fs_sock_recv,
	332	.close = v9fs_sock_close,
3cf6429a	333	.poll = v9fs_sock_poll,
426cc91a	334	};