[linux-block.git] / net / phonet / datagram.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * File: datagram.c
 *
 * Datagram (ISI) Phonet sockets
 *
 * Copyright (C) 2008 Nokia Corporation.
 *
 * Authors: Sakari Ailus <sakari.ailus@nokia.com>
 *          Rémi Denis-Courmont
 */

#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/socket.h>
#include <asm/ioctls.h>
#include <net/sock.h>

#include <linux/phonet.h>
#include <linux/export.h>
#include <net/phonet/phonet.h>

static int pn_backlog_rcv(struct sock *sk, struct sk_buff *skb);

/* associated socket ceases to exist */
static void pn_sock_close(struct sock *sk, long timeout)
{
	sk_common_release(sk);
}

static int pn_ioctl(struct sock *sk, int cmd, unsigned long arg)
{
	struct sk_buff *skb;
	int answ;

	switch (cmd) {
	case SIOCINQ:
		lock_sock(sk);
		skb = skb_peek(&sk->sk_receive_queue);
		answ = skb ? skb->len : 0;
		release_sock(sk);
		return put_user(answ, (int __user *)arg);

	case SIOCPNADDRESOURCE:
	case SIOCPNDELRESOURCE: {
			u32 res;
			if (get_user(res, (u32 __user *)arg))
				return -EFAULT;
			if (res >= 256)
				return -EINVAL;
			if (cmd == SIOCPNADDRESOURCE)
				return pn_sock_bind_res(sk, res);
			else
				return pn_sock_unbind_res(sk, res);
		}
	}

	return -ENOIOCTLCMD;
}

/* Destroy socket. All references are gone. */
static void pn_destruct(struct sock *sk)
{
	skb_queue_purge(&sk->sk_receive_queue);
}

static int pn_init(struct sock *sk)
{
	sk->sk_destruct = pn_destruct;
	return 0;
}

static int pn_sendmsg(struct sock *sk, struct msghdr *msg, size_t len)
{
	DECLARE_SOCKADDR(struct sockaddr_pn *, target, msg->msg_name);
	struct sk_buff *skb;
	int err;

	if (msg->msg_flags & ~(MSG_DONTWAIT|MSG_EOR|MSG_NOSIGNAL|
				MSG_CMSG_COMPAT))
		return -EOPNOTSUPP;

	if (target == NULL)
		return -EDESTADDRREQ;

	if (msg->msg_namelen < sizeof(struct sockaddr_pn))
		return -EINVAL;

	if (target->spn_family != AF_PHONET)
		return -EAFNOSUPPORT;

	skb = sock_alloc_send_skb(sk, MAX_PHONET_HEADER + len,
					msg->msg_flags & MSG_DONTWAIT, &err);
	if (skb == NULL)
		return err;
	skb_reserve(skb, MAX_PHONET_HEADER);

	err = memcpy_from_msg((void *)skb_put(skb, len), msg, len);
	if (err < 0) {
		kfree_skb(skb);
		return err;
	}

	/*
	 * Fill in the Phonet header and
	 * finally pass the packet forwards.
	 */
	err = pn_skb_send(sk, skb, target);

	/* If ok, return len. */
	return (err >= 0) ? len : err;
}

static int pn_recvmsg(struct sock *sk, struct msghdr *msg, size_t len,
		      int flags, int *addr_len)
{
	struct sk_buff *skb = NULL;
	struct sockaddr_pn sa;
	int rval = -EOPNOTSUPP;
	int copylen;

	if (flags & ~(MSG_PEEK|MSG_TRUNC|MSG_DONTWAIT|MSG_NOSIGNAL|
			MSG_CMSG_COMPAT))
		goto out_nofree;

	skb = skb_recv_datagram(sk, flags, &rval);
	if (skb == NULL)
		goto out_nofree;

	pn_skb_get_src_sockaddr(skb, &sa);

	copylen = skb->len;
	if (len < copylen) {
		msg->msg_flags |= MSG_TRUNC;
		copylen = len;
	}

	rval = skb_copy_datagram_msg(skb, 0, msg, copylen);
	if (rval) {
		rval = -EFAULT;
		goto out;
	}

	rval = (flags & MSG_TRUNC) ? skb->len : copylen;

	if (msg->msg_name != NULL) {
		__sockaddr_check_size(sizeof(sa));
		memcpy(msg->msg_name, &sa, sizeof(sa));
		*addr_len = sizeof(sa);
	}

out:
	skb_free_datagram(sk, skb);

out_nofree:
	return rval;
}

/* Queue an skb for a sock. */
static int pn_backlog_rcv(struct sock *sk, struct sk_buff *skb)
{
	int err = sock_queue_rcv_skb(sk, skb);

	if (err < 0)
		kfree_skb(skb);
	return err ? NET_RX_DROP : NET_RX_SUCCESS;
}

/* Module registration */
static struct proto pn_proto = {
	.close		= pn_sock_close,
	.ioctl		= pn_ioctl,
	.init		= pn_init,
	.sendmsg	= pn_sendmsg,
	.recvmsg	= pn_recvmsg,
	.backlog_rcv	= pn_backlog_rcv,
	.hash		= pn_sock_hash,
	.unhash		= pn_sock_unhash,
	.get_port	= pn_sock_get_port,
	.obj_size	= sizeof(struct pn_sock),
	.owner		= THIS_MODULE,
	.name		= "PHONET",
};

static const struct phonet_protocol pn_dgram_proto = {
	.ops		= &phonet_dgram_ops,
	.prot		= &pn_proto,
	.sock_type	= SOCK_DGRAM,
};

int __init isi_register(void)
{
	return phonet_proto_register(PN_PROTO_PHONET, &pn_dgram_proto);
}

void __exit isi_unregister(void)
{
	phonet_proto_unregister(PN_PROTO_PHONET, &pn_dgram_proto);
}
Commit	Line	Data
2b27bdcc	1	// SPDX-License-Identifier: GPL-2.0-only
107d0d9b RDC	2	/*
	3	* File: datagram.c
	4	*
	5	* Datagram (ISI) Phonet sockets
	6	*
	7	* Copyright (C) 2008 Nokia Corporation.
	8	*
31fdc555 RDC	9	* Authors: Sakari Ailus <sakari.ailus@nokia.com>
31fdc555 RDC	10	* Rémi Denis-Courmont
107d0d9b RDC	11	*/
	12
	13	#include <linux/kernel.h>
5a0e3ad6	14	#include <linux/slab.h>
107d0d9b RDC	15	#include <linux/socket.h>
	16	#include <asm/ioctls.h>
	17	#include <net/sock.h>
	18
	19	#include <linux/phonet.h>
bc3b2d7f	20	#include <linux/export.h>
107d0d9b RDC	21	#include <net/phonet/phonet.h>
	22
	23	static int pn_backlog_rcv(struct sock sk, struct sk_buff skb);
	24
	25	/* associated socket ceases to exist */
	26	static void pn_sock_close(struct sock *sk, long timeout)
	27	{
	28	sk_common_release(sk);
	29	}
	30
	31	static int pn_ioctl(struct sock *sk, int cmd, unsigned long arg)
	32	{
	33	struct sk_buff *skb;
	34	int answ;
	35
	36	switch (cmd) {
	37	case SIOCINQ:
	38	lock_sock(sk);
	39	skb = skb_peek(&sk->sk_receive_queue);
	40	answ = skb ? skb->len : 0;
	41	release_sock(sk);
	42	return put_user(answ, (int __user *)arg);
7417fa83 RDC	43
	44	case SIOCPNADDRESOURCE:
	45	case SIOCPNDELRESOURCE: {
	46	u32 res;
	47	if (get_user(res, (u32 __user *)arg))
	48	return -EFAULT;
	49	if (res >= 256)
	50	return -EINVAL;
	51	if (cmd == SIOCPNADDRESOURCE)
	52	return pn_sock_bind_res(sk, res);
	53	else
	54	return pn_sock_unbind_res(sk, res);
	55	}
107d0d9b RDC	56	}
	57
	58	return -ENOIOCTLCMD;
	59	}
	60
	61	/* Destroy socket. All references are gone. */
	62	static void pn_destruct(struct sock *sk)
	63	{
	64	skb_queue_purge(&sk->sk_receive_queue);
	65	}
	66
	67	static int pn_init(struct sock *sk)
	68	{
	69	sk->sk_destruct = pn_destruct;
	70	return 0;
	71	}
	72
1b784140	73	static int pn_sendmsg(struct sock sk, struct msghdr msg, size_t len)
107d0d9b	74	{
342dfc30	75	DECLARE_SOCKADDR(struct sockaddr_pn *, target, msg->msg_name);
107d0d9b RDC	76	struct sk_buff *skb;
	77	int err;
	78
82ecbcb9 RDC	79	if (msg->msg_flags & ~(MSG_DONTWAIT\|MSG_EOR\|MSG_NOSIGNAL\|
82ecbcb9 RDC	80	MSG_CMSG_COMPAT))
107d0d9b RDC	81	return -EOPNOTSUPP;
107d0d9b RDC	82
342dfc30	83	if (target == NULL)
107d0d9b RDC	84	return -EDESTADDRREQ;
	85
	86	if (msg->msg_namelen < sizeof(struct sockaddr_pn))
	87	return -EINVAL;
	88
107d0d9b RDC	89	if (target->spn_family != AF_PHONET)
	90	return -EAFNOSUPPORT;
	91
	92	skb = sock_alloc_send_skb(sk, MAX_PHONET_HEADER + len,
	93	msg->msg_flags & MSG_DONTWAIT, &err);
	94	if (skb == NULL)
	95	return err;
	96	skb_reserve(skb, MAX_PHONET_HEADER);
	97
6ce8e9ce	98	err = memcpy_from_msg((void *)skb_put(skb, len), msg, len);
107d0d9b RDC	99	if (err < 0) {
	100	kfree_skb(skb);
	101	return err;
	102	}
	103
	104	/*
	105	* Fill in the Phonet header and
	106	* finally pass the packet forwards.
	107	*/
	108	err = pn_skb_send(sk, skb, target);
	109
	110	/* If ok, return len. */
	111	return (err >= 0) ? len : err;
	112	}
	113
1b784140	114	static int pn_recvmsg(struct sock sk, struct msghdr msg, size_t len,
ec095263	115	int flags, int *addr_len)
107d0d9b RDC	116	{
	117	struct sk_buff *skb = NULL;
	118	struct sockaddr_pn sa;
	119	int rval = -EOPNOTSUPP;
	120	int copylen;
	121
82ecbcb9 RDC	122	if (flags & ~(MSG_PEEK\|MSG_TRUNC\|MSG_DONTWAIT\|MSG_NOSIGNAL\|
82ecbcb9 RDC	123	MSG_CMSG_COMPAT))
107d0d9b RDC	124	goto out_nofree;
107d0d9b RDC	125
f4b41f06	126	skb = skb_recv_datagram(sk, flags, &rval);
107d0d9b RDC	127	if (skb == NULL)
	128	goto out_nofree;
	129
	130	pn_skb_get_src_sockaddr(skb, &sa);
	131
	132	copylen = skb->len;
	133	if (len < copylen) {
	134	msg->msg_flags \|= MSG_TRUNC;
	135	copylen = len;
	136	}
	137
51f3d02b	138	rval = skb_copy_datagram_msg(skb, 0, msg, copylen);
107d0d9b RDC	139	if (rval) {
	140	rval = -EFAULT;
	141	goto out;
	142	}
	143
	144	rval = (flags & MSG_TRUNC) ? skb->len : copylen;
	145
bceaa902	146	if (msg->msg_name != NULL) {
342dfc30	147	__sockaddr_check_size(sizeof(sa));
bceaa902 HFS	148	memcpy(msg->msg_name, &sa, sizeof(sa));
	149	*addr_len = sizeof(sa);
	150	}
107d0d9b RDC	151
	152	out:
	153	skb_free_datagram(sk, skb);
	154
	155	out_nofree:
	156	return rval;
	157	}
	158
	159	/* Queue an skb for a sock. */
	160	static int pn_backlog_rcv(struct sock sk, struct sk_buff skb)
	161	{
	162	int err = sock_queue_rcv_skb(sk, skb);
766e9037 ED	163
766e9037 ED	164	if (err < 0)
107d0d9b RDC	165	kfree_skb(skb);
	166	return err ? NET_RX_DROP : NET_RX_SUCCESS;
	167	}
	168
	169	/* Module registration */
	170	static struct proto pn_proto = {
	171	.close = pn_sock_close,
	172	.ioctl = pn_ioctl,
	173	.init = pn_init,
	174	.sendmsg = pn_sendmsg,
	175	.recvmsg = pn_recvmsg,
	176	.backlog_rcv = pn_backlog_rcv,
	177	.hash = pn_sock_hash,
	178	.unhash = pn_sock_unhash,
	179	.get_port = pn_sock_get_port,
	180	.obj_size = sizeof(struct pn_sock),
	181	.owner = THIS_MODULE,
	182	.name = "PHONET",
	183	};
	184
548ec114	185	static const struct phonet_protocol pn_dgram_proto = {
107d0d9b RDC	186	.ops = &phonet_dgram_ops,
	187	.prot = &pn_proto,
	188	.sock_type = SOCK_DGRAM,
	189	};
	190
	191	int __init isi_register(void)
	192	{
	193	return phonet_proto_register(PN_PROTO_PHONET, &pn_dgram_proto);
	194	}
	195
	196	void __exit isi_unregister(void)
	197	{
	198	phonet_proto_unregister(PN_PROTO_PHONET, &pn_dgram_proto);
	199	}