[deliverable/linux.git] / net / unix / diag.c

#include <linux/types.h>
#include <linux/spinlock.h>
#include <linux/sock_diag.h>
#include <linux/unix_diag.h>
#include <linux/skbuff.h>
#include <net/netlink.h>
#include <net/af_unix.h>
#include <net/tcp_states.h>

#define UNIX_DIAG_PUT(skb, attrtype, attrlen) \
	RTA_DATA(__RTA_PUT(skb, attrtype, attrlen))

static int sk_diag_dump_name(struct sock *sk, struct sk_buff *nlskb)
{
	struct unix_address *addr = unix_sk(sk)->addr;
	char *s;

	if (addr) {
		s = UNIX_DIAG_PUT(nlskb, UNIX_DIAG_NAME, addr->len - sizeof(short));
		memcpy(s, addr->name->sun_path, addr->len - sizeof(short));
	}

	return 0;

rtattr_failure:
	return -EMSGSIZE;
}

static int sk_diag_dump_vfs(struct sock *sk, struct sk_buff *nlskb)
{
	struct dentry *dentry = unix_sk(sk)->dentry;
	struct unix_diag_vfs *uv;

	if (dentry) {
		uv = UNIX_DIAG_PUT(nlskb, UNIX_DIAG_VFS, sizeof(*uv));
		uv->udiag_vfs_ino = dentry->d_inode->i_ino;
		uv->udiag_vfs_dev = dentry->d_sb->s_dev;
	}

	return 0;

rtattr_failure:
	return -EMSGSIZE;
}

static int sk_diag_fill(struct sock *sk, struct sk_buff *skb, struct unix_diag_req *req,
		u32 pid, u32 seq, u32 flags, int sk_ino)
{
	unsigned char *b = skb_tail_pointer(skb);
	struct nlmsghdr *nlh;
	struct unix_diag_msg *rep;

	nlh = NLMSG_PUT(skb, pid, seq, SOCK_DIAG_BY_FAMILY, sizeof(*rep));
	nlh->nlmsg_flags = flags;

	rep = NLMSG_DATA(nlh);

	rep->udiag_family = AF_UNIX;
	rep->udiag_type = sk->sk_type;
	rep->udiag_state = sk->sk_state;
	rep->udiag_ino = sk_ino;
	sock_diag_save_cookie(sk, rep->udiag_cookie);

	if ((req->udiag_show & UDIAG_SHOW_NAME) &&
			sk_diag_dump_name(sk, skb))
		goto nlmsg_failure;

	if ((req->udiag_show & UDIAG_SHOW_VFS) &&
			sk_diag_dump_vfs(sk, skb))
		goto nlmsg_failure;

	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
	return skb->len;

nlmsg_failure:
	nlmsg_trim(skb, b);
	return -EMSGSIZE;
}

static int sk_diag_dump(struct sock *sk, struct sk_buff *skb, struct unix_diag_req *req,
		u32 pid, u32 seq, u32 flags)
{
	int sk_ino;

	unix_state_lock(sk);
	sk_ino = sock_i_ino(sk);
	unix_state_unlock(sk);

	if (!sk_ino)
		return 0;

	return sk_diag_fill(sk, skb, req, pid, seq, flags, sk_ino);
}

static int unix_diag_dump(struct sk_buff *skb, struct netlink_callback *cb)
{
	struct unix_diag_req *req;
	int num, s_num, slot, s_slot;

	req = NLMSG_DATA(cb->nlh);

	s_slot = cb->args[0];
	num = s_num = cb->args[1];

	spin_lock(&unix_table_lock);
	for (slot = s_slot; slot <= UNIX_HASH_SIZE; s_num = 0, slot++) {
		struct sock *sk;
		struct hlist_node *node;

		num = 0;
		sk_for_each(sk, node, &unix_socket_table[slot]) {
			if (num < s_num)
				goto next;
			if (!(req->udiag_states & (1 << sk->sk_state)))
				goto next;
			if (sk_diag_dump(sk, skb, req,
						NETLINK_CB(cb->skb).pid,
						cb->nlh->nlmsg_seq,
						NLM_F_MULTI) < 0)
				goto done;
next:
			num++;
		}
	}
done:
	spin_unlock(&unix_table_lock);
	cb->args[0] = slot;
	cb->args[1] = num;

	return skb->len;
}

static struct sock *unix_lookup_by_ino(int ino)
{
	int i;
	struct sock *sk;

	spin_lock(&unix_table_lock);
	for (i = 0; i <= UNIX_HASH_SIZE; i++) {
		struct hlist_node *node;

		sk_for_each(sk, node, &unix_socket_table[i])
			if (ino == sock_i_ino(sk)) {
				sock_hold(sk);
				spin_unlock(&unix_table_lock);

				return sk;
			}
	}

	spin_unlock(&unix_table_lock);
	return NULL;
}

static int unix_diag_get_exact(struct sk_buff *in_skb,
			       const struct nlmsghdr *nlh,
			       struct unix_diag_req *req)
{
	int err = -EINVAL;
	struct sock *sk;
	struct sk_buff *rep;
	unsigned int extra_len;

	if (req->udiag_ino == 0)
		goto out_nosk;

	sk = unix_lookup_by_ino(req->udiag_ino);
	err = -ENOENT;
	if (sk == NULL)
		goto out_nosk;

	err = sock_diag_check_cookie(sk, req->udiag_cookie);
	if (err)
		goto out;

	extra_len = 256;
again:
	err = -ENOMEM;
	rep = alloc_skb(NLMSG_SPACE((sizeof(struct unix_diag_msg) + extra_len)),
			GFP_KERNEL);
	if (!rep)
		goto out;

	err = sk_diag_fill(sk, rep, req, NETLINK_CB(in_skb).pid,
			   nlh->nlmsg_seq, 0, req->udiag_ino);
	if (err < 0) {
		kfree_skb(rep);
		extra_len += 256;
		if (extra_len >= PAGE_SIZE)
			goto out;

		goto again;
	}
	err = netlink_unicast(sock_diag_nlsk, rep, NETLINK_CB(in_skb).pid,
			      MSG_DONTWAIT);
	if (err > 0)
		err = 0;
out:
	if (sk)
		sock_put(sk);
out_nosk:
	return err;
}

static int unix_diag_handler_dump(struct sk_buff *skb, struct nlmsghdr *h)
{
	int hdrlen = sizeof(struct unix_diag_req);

	if (nlmsg_len(h) < hdrlen)
		return -EINVAL;

	if (h->nlmsg_flags & NLM_F_DUMP)
		return netlink_dump_start(sock_diag_nlsk, skb, h,
					  unix_diag_dump, NULL, 0);
	else
		return unix_diag_get_exact(skb, h, (struct unix_diag_req *)NLMSG_DATA(h));
}

static struct sock_diag_handler unix_diag_handler = {
	.family = AF_UNIX,
	.dump = unix_diag_handler_dump,
};

static int __init unix_diag_init(void)
{
	return sock_diag_register(&unix_diag_handler);
}

static void __exit unix_diag_exit(void)
{
	sock_diag_unregister(&unix_diag_handler);
}

module_init(unix_diag_init);
module_exit(unix_diag_exit);
MODULE_LICENSE("GPL");
MODULE_ALIAS_NET_PF_PROTO_TYPE(PF_NETLINK, NETLINK_SOCK_DIAG, 1 /* AF_LOCAL */);
Commit	Line	Data
22931d3b PE	1	#include <linux/types.h>
	2	#include <linux/spinlock.h>
	3	#include <linux/sock_diag.h>
	4	#include <linux/unix_diag.h>
	5	#include <linux/skbuff.h>
	6	#include <net/netlink.h>
	7	#include <net/af_unix.h>
	8	#include <net/tcp_states.h>
	9
	10	#define UNIX_DIAG_PUT(skb, attrtype, attrlen) \
	11	RTA_DATA(__RTA_PUT(skb, attrtype, attrlen))
	12
f5248b48 PE	13	static int sk_diag_dump_name(struct sock sk, struct sk_buff nlskb)
	14	{
	15	struct unix_address *addr = unix_sk(sk)->addr;
	16	char *s;
	17
	18	if (addr) {
	19	s = UNIX_DIAG_PUT(nlskb, UNIX_DIAG_NAME, addr->len - sizeof(short));
	20	memcpy(s, addr->name->sun_path, addr->len - sizeof(short));
	21	}
	22
	23	return 0;
	24
	25	rtattr_failure:
	26	return -EMSGSIZE;
	27	}
	28
5f7b0569 PE	29	static int sk_diag_dump_vfs(struct sock sk, struct sk_buff nlskb)
	30	{
	31	struct dentry *dentry = unix_sk(sk)->dentry;
	32	struct unix_diag_vfs *uv;
	33
	34	if (dentry) {
	35	uv = UNIX_DIAG_PUT(nlskb, UNIX_DIAG_VFS, sizeof(*uv));
	36	uv->udiag_vfs_ino = dentry->d_inode->i_ino;
	37	uv->udiag_vfs_dev = dentry->d_sb->s_dev;
	38	}
	39
	40	return 0;
	41
	42	rtattr_failure:
	43	return -EMSGSIZE;
	44	}
	45
45a96b9b PE	46	static int sk_diag_fill(struct sock sk, struct sk_buff skb, struct unix_diag_req *req,
	47	u32 pid, u32 seq, u32 flags, int sk_ino)
	48	{
	49	unsigned char *b = skb_tail_pointer(skb);
	50	struct nlmsghdr *nlh;
	51	struct unix_diag_msg *rep;
	52
	53	nlh = NLMSG_PUT(skb, pid, seq, SOCK_DIAG_BY_FAMILY, sizeof(*rep));
	54	nlh->nlmsg_flags = flags;
	55
	56	rep = NLMSG_DATA(nlh);
	57
	58	rep->udiag_family = AF_UNIX;
	59	rep->udiag_type = sk->sk_type;
	60	rep->udiag_state = sk->sk_state;
	61	rep->udiag_ino = sk_ino;
	62	sock_diag_save_cookie(sk, rep->udiag_cookie);
	63
f5248b48 PE	64	if ((req->udiag_show & UDIAG_SHOW_NAME) &&
	65	sk_diag_dump_name(sk, skb))
	66	goto nlmsg_failure;
	67
5f7b0569 PE	68	if ((req->udiag_show & UDIAG_SHOW_VFS) &&
	69	sk_diag_dump_vfs(sk, skb))
	70	goto nlmsg_failure;
	71
45a96b9b PE	72	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
	73	return skb->len;
	74
	75	nlmsg_failure:
	76	nlmsg_trim(skb, b);
	77	return -EMSGSIZE;
	78	}
	79
	80	static int sk_diag_dump(struct sock sk, struct sk_buff skb, struct unix_diag_req *req,
	81	u32 pid, u32 seq, u32 flags)
	82	{
	83	int sk_ino;
	84
	85	unix_state_lock(sk);
	86	sk_ino = sock_i_ino(sk);
	87	unix_state_unlock(sk);
	88
	89	if (!sk_ino)
	90	return 0;
	91
	92	return sk_diag_fill(sk, skb, req, pid, seq, flags, sk_ino);
	93	}
	94
22931d3b PE	95	static int unix_diag_dump(struct sk_buff skb, struct netlink_callback cb)
22931d3b PE	96	{
45a96b9b PE	97	struct unix_diag_req *req;
	98	int num, s_num, slot, s_slot;
	99
	100	req = NLMSG_DATA(cb->nlh);
	101
	102	s_slot = cb->args[0];
	103	num = s_num = cb->args[1];
	104
	105	spin_lock(&unix_table_lock);
	106	for (slot = s_slot; slot <= UNIX_HASH_SIZE; s_num = 0, slot++) {
	107	struct sock *sk;
	108	struct hlist_node *node;
	109
	110	num = 0;
	111	sk_for_each(sk, node, &unix_socket_table[slot]) {
	112	if (num < s_num)
	113	goto next;
	114	if (!(req->udiag_states & (1 << sk->sk_state)))
	115	goto next;
	116	if (sk_diag_dump(sk, skb, req,
	117	NETLINK_CB(cb->skb).pid,
	118	cb->nlh->nlmsg_seq,
	119	NLM_F_MULTI) < 0)
	120	goto done;
	121	next:
	122	num++;
	123	}
	124	}
	125	done:
	126	spin_unlock(&unix_table_lock);
	127	cb->args[0] = slot;
	128	cb->args[1] = num;
	129
	130	return skb->len;
22931d3b PE	131	}
22931d3b PE	132
5d3cae8b PE	133	static struct sock *unix_lookup_by_ino(int ino)
	134	{
	135	int i;
	136	struct sock *sk;
	137
	138	spin_lock(&unix_table_lock);
	139	for (i = 0; i <= UNIX_HASH_SIZE; i++) {
	140	struct hlist_node *node;
	141
	142	sk_for_each(sk, node, &unix_socket_table[i])
	143	if (ino == sock_i_ino(sk)) {
	144	sock_hold(sk);
	145	spin_unlock(&unix_table_lock);
	146
	147	return sk;
	148	}
	149	}
	150
	151	spin_unlock(&unix_table_lock);
	152	return NULL;
	153	}
	154
22931d3b PE	155	static int unix_diag_get_exact(struct sk_buff *in_skb,
	156	const struct nlmsghdr *nlh,
	157	struct unix_diag_req *req)
	158	{
5d3cae8b PE	159	int err = -EINVAL;
	160	struct sock *sk;
	161	struct sk_buff *rep;
	162	unsigned int extra_len;
	163
	164	if (req->udiag_ino == 0)
	165	goto out_nosk;
	166
	167	sk = unix_lookup_by_ino(req->udiag_ino);
	168	err = -ENOENT;
	169	if (sk == NULL)
	170	goto out_nosk;
	171
	172	err = sock_diag_check_cookie(sk, req->udiag_cookie);
	173	if (err)
	174	goto out;
	175
	176	extra_len = 256;
	177	again:
	178	err = -ENOMEM;
	179	rep = alloc_skb(NLMSG_SPACE((sizeof(struct unix_diag_msg) + extra_len)),
	180	GFP_KERNEL);
	181	if (!rep)
	182	goto out;
	183
	184	err = sk_diag_fill(sk, rep, req, NETLINK_CB(in_skb).pid,
	185	nlh->nlmsg_seq, 0, req->udiag_ino);
	186	if (err < 0) {
	187	kfree_skb(rep);
	188	extra_len += 256;
	189	if (extra_len >= PAGE_SIZE)
	190	goto out;
	191
	192	goto again;
	193	}
	194	err = netlink_unicast(sock_diag_nlsk, rep, NETLINK_CB(in_skb).pid,
	195	MSG_DONTWAIT);
	196	if (err > 0)
	197	err = 0;
	198	out:
	199	if (sk)
	200	sock_put(sk);
	201	out_nosk:
	202	return err;
22931d3b PE	203	}
	204
	205	static int unix_diag_handler_dump(struct sk_buff skb, struct nlmsghdr h)
	206	{
	207	int hdrlen = sizeof(struct unix_diag_req);
	208
	209	if (nlmsg_len(h) < hdrlen)
	210	return -EINVAL;
	211
	212	if (h->nlmsg_flags & NLM_F_DUMP)
	213	return netlink_dump_start(sock_diag_nlsk, skb, h,
	214	unix_diag_dump, NULL, 0);
	215	else
	216	return unix_diag_get_exact(skb, h, (struct unix_diag_req *)NLMSG_DATA(h));
	217	}
	218
	219	static struct sock_diag_handler unix_diag_handler = {
	220	.family = AF_UNIX,
	221	.dump = unix_diag_handler_dump,
	222	};
	223
	224	static int __init unix_diag_init(void)
	225	{
	226	return sock_diag_register(&unix_diag_handler);
	227	}
	228
	229	static void __exit unix_diag_exit(void)
	230	{
	231	sock_diag_unregister(&unix_diag_handler);
	232	}
	233
	234	module_init(unix_diag_init);
	235	module_exit(unix_diag_exit);
	236	MODULE_LICENSE("GPL");
	237	MODULE_ALIAS_NET_PF_PROTO_TYPE(PF_NETLINK, NETLINK_SOCK_DIAG, 1 /* AF_LOCAL */);