2 * Copyright (c) 2015, Sony Mobile Communications Inc.
3 * Copyright (c) 2013, The Linux Foundation. All rights reserved.
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 and
7 * only version 2 as published by the Free Software Foundation.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 #include <linux/module.h>
15 #include <linux/netlink.h>
16 #include <linux/qrtr.h>
17 #include <linux/termios.h> /* For TIOCINQ/OUTQ */
23 #define QRTR_PROTO_VER_1 1
24 #define QRTR_PROTO_VER_2 3
27 #define QRTR_MIN_EPH_SOCKET 0x4000
28 #define QRTR_MAX_EPH_SOCKET 0x7fff
31 * struct qrtr_hdr_v1 - (I|R)PCrouter packet header version 1
32 * @version: protocol version
33 * @type: packet type; one of QRTR_TYPE_*
34 * @src_node_id: source node
35 * @src_port_id: source port
36 * @confirm_rx: boolean; whether a resume-tx packet should be send in reply
37 * @size: length of packet, excluding this header
38 * @dst_node_id: destination node
39 * @dst_port_id: destination port
53 * struct qrtr_hdr_v2 - (I|R)PCrouter packet header later versions
54 * @version: protocol version
55 * @type: packet type; one of QRTR_TYPE_*
56 * @flags: bitmask of QRTR_FLAGS_*
57 * @optlen: length of optional header data
58 * @size: length of packet, excluding this header and optlen
59 * @src_node_id: source node
60 * @src_port_id: source port
61 * @dst_node_id: destination node
62 * @dst_port_id: destination port
76 #define QRTR_FLAGS_CONFIRM_RX BIT(0)
88 #define QRTR_HDR_MAX_SIZE max_t(size_t, sizeof(struct qrtr_hdr_v1), \
89 sizeof(struct qrtr_hdr_v2))
92 /* WARNING: sk must be the first member */
94 struct sockaddr_qrtr us;
95 struct sockaddr_qrtr peer;
98 static inline struct qrtr_sock *qrtr_sk(struct sock *sk)
100 BUILD_BUG_ON(offsetof(struct qrtr_sock, sk) != 0);
101 return container_of(sk, struct qrtr_sock, sk);
104 static unsigned int qrtr_local_nid = -1;
107 static RADIX_TREE(qrtr_nodes, GFP_KERNEL);
109 static LIST_HEAD(qrtr_all_nodes);
110 /* lock for qrtr_nodes, qrtr_all_nodes and node reference */
111 static DEFINE_MUTEX(qrtr_node_lock);
113 /* local port allocation management */
114 static DEFINE_IDR(qrtr_ports);
115 static DEFINE_MUTEX(qrtr_port_lock);
118 * struct qrtr_node - endpoint node
119 * @ep_lock: lock for endpoint management and callbacks
121 * @ref: reference count for node
123 * @rx_queue: receive queue
124 * @work: scheduled work struct for recv work
125 * @item: list item for broadcast list
128 struct mutex ep_lock;
129 struct qrtr_endpoint *ep;
133 struct sk_buff_head rx_queue;
134 struct work_struct work;
135 struct list_head item;
138 static int qrtr_local_enqueue(struct qrtr_node *node, struct sk_buff *skb,
139 int type, struct sockaddr_qrtr *from,
140 struct sockaddr_qrtr *to);
141 static int qrtr_bcast_enqueue(struct qrtr_node *node, struct sk_buff *skb,
142 int type, struct sockaddr_qrtr *from,
143 struct sockaddr_qrtr *to);
145 /* Release node resources and free the node.
147 * Do not call directly, use qrtr_node_release. To be used with
148 * kref_put_mutex. As such, the node mutex is expected to be locked on call.
150 static void __qrtr_node_release(struct kref *kref)
152 struct qrtr_node *node = container_of(kref, struct qrtr_node, ref);
154 if (node->nid != QRTR_EP_NID_AUTO)
155 radix_tree_delete(&qrtr_nodes, node->nid);
157 list_del(&node->item);
158 mutex_unlock(&qrtr_node_lock);
160 cancel_work_sync(&node->work);
161 skb_queue_purge(&node->rx_queue);
165 /* Increment reference to node. */
166 static struct qrtr_node *qrtr_node_acquire(struct qrtr_node *node)
169 kref_get(&node->ref);
173 /* Decrement reference to node and release as necessary. */
174 static void qrtr_node_release(struct qrtr_node *node)
178 kref_put_mutex(&node->ref, __qrtr_node_release, &qrtr_node_lock);
181 /* Pass an outgoing packet socket buffer to the endpoint driver. */
182 static int qrtr_node_enqueue(struct qrtr_node *node, struct sk_buff *skb,
183 int type, struct sockaddr_qrtr *from,
184 struct sockaddr_qrtr *to)
186 struct qrtr_hdr_v1 *hdr;
187 size_t len = skb->len;
190 hdr = skb_push(skb, sizeof(*hdr));
191 hdr->version = cpu_to_le32(QRTR_PROTO_VER_1);
192 hdr->type = cpu_to_le32(type);
193 hdr->src_node_id = cpu_to_le32(from->sq_node);
194 hdr->src_port_id = cpu_to_le32(from->sq_port);
195 if (to->sq_port == QRTR_PORT_CTRL) {
196 hdr->dst_node_id = cpu_to_le32(node->nid);
197 hdr->dst_port_id = cpu_to_le32(QRTR_PORT_CTRL);
199 hdr->dst_node_id = cpu_to_le32(to->sq_node);
200 hdr->dst_port_id = cpu_to_le32(to->sq_port);
203 hdr->size = cpu_to_le32(len);
206 rc = skb_put_padto(skb, ALIGN(len, 4) + sizeof(*hdr));
209 mutex_lock(&node->ep_lock);
212 rc = node->ep->xmit(node->ep, skb);
215 mutex_unlock(&node->ep_lock);
220 /* Lookup node by id.
222 * callers must release with qrtr_node_release()
224 static struct qrtr_node *qrtr_node_lookup(unsigned int nid)
226 struct qrtr_node *node;
228 mutex_lock(&qrtr_node_lock);
229 node = radix_tree_lookup(&qrtr_nodes, nid);
230 node = qrtr_node_acquire(node);
231 mutex_unlock(&qrtr_node_lock);
236 /* Assign node id to node.
238 * This is mostly useful for automatic node id assignment, based on
239 * the source id in the incoming packet.
241 static void qrtr_node_assign(struct qrtr_node *node, unsigned int nid)
243 if (node->nid != QRTR_EP_NID_AUTO || nid == QRTR_EP_NID_AUTO)
246 mutex_lock(&qrtr_node_lock);
247 radix_tree_insert(&qrtr_nodes, nid, node);
249 mutex_unlock(&qrtr_node_lock);
253 * qrtr_endpoint_post() - post incoming data
254 * @ep: endpoint handle
255 * @data: data pointer
256 * @len: size of data in bytes
258 * Return: 0 on success; negative error code on failure
260 int qrtr_endpoint_post(struct qrtr_endpoint *ep, const void *data, size_t len)
262 struct qrtr_node *node = ep->node;
263 const struct qrtr_hdr_v1 *v1;
264 const struct qrtr_hdr_v2 *v2;
271 if (len == 0 || len & 3)
274 skb = __netdev_alloc_skb(NULL, len, GFP_ATOMIC | __GFP_NOWARN);
278 cb = (struct qrtr_cb *)skb->cb;
280 /* Version field in v1 is little endian, so this works for both cases */
284 case QRTR_PROTO_VER_1:
285 if (len < sizeof(*v1))
288 hdrlen = sizeof(*v1);
290 cb->type = le32_to_cpu(v1->type);
291 cb->src_node = le32_to_cpu(v1->src_node_id);
292 cb->src_port = le32_to_cpu(v1->src_port_id);
293 cb->confirm_rx = !!v1->confirm_rx;
294 cb->dst_node = le32_to_cpu(v1->dst_node_id);
295 cb->dst_port = le32_to_cpu(v1->dst_port_id);
297 size = le32_to_cpu(v1->size);
299 case QRTR_PROTO_VER_2:
300 if (len < sizeof(*v2))
303 hdrlen = sizeof(*v2) + v2->optlen;
306 cb->confirm_rx = !!(v2->flags & QRTR_FLAGS_CONFIRM_RX);
307 cb->src_node = le16_to_cpu(v2->src_node_id);
308 cb->src_port = le16_to_cpu(v2->src_port_id);
309 cb->dst_node = le16_to_cpu(v2->dst_node_id);
310 cb->dst_port = le16_to_cpu(v2->dst_port_id);
312 if (cb->src_port == (u16)QRTR_PORT_CTRL)
313 cb->src_port = QRTR_PORT_CTRL;
314 if (cb->dst_port == (u16)QRTR_PORT_CTRL)
315 cb->dst_port = QRTR_PORT_CTRL;
317 size = le32_to_cpu(v2->size);
320 pr_err("qrtr: Invalid version %d\n", ver);
324 if (!size || len != ALIGN(size, 4) + hdrlen)
327 if (cb->dst_port != QRTR_PORT_CTRL && cb->type != QRTR_TYPE_DATA)
330 skb_put_data(skb, data + hdrlen, size);
332 skb_queue_tail(&node->rx_queue, skb);
333 schedule_work(&node->work);
342 EXPORT_SYMBOL_GPL(qrtr_endpoint_post);
345 * qrtr_alloc_ctrl_packet() - allocate control packet skb
346 * @pkt: reference to qrtr_ctrl_pkt pointer
348 * Returns newly allocated sk_buff, or NULL on failure
350 * This function allocates a sk_buff large enough to carry a qrtr_ctrl_pkt and
351 * on success returns a reference to the control packet in @pkt.
353 static struct sk_buff *qrtr_alloc_ctrl_packet(struct qrtr_ctrl_pkt **pkt)
355 const int pkt_len = sizeof(struct qrtr_ctrl_pkt);
358 skb = alloc_skb(QRTR_HDR_MAX_SIZE + pkt_len, GFP_KERNEL);
362 skb_reserve(skb, QRTR_HDR_MAX_SIZE);
363 *pkt = skb_put_zero(skb, pkt_len);
368 static struct qrtr_sock *qrtr_port_lookup(int port);
369 static void qrtr_port_put(struct qrtr_sock *ipc);
371 /* Handle and route a received packet.
373 * This will auto-reply with resume-tx packet as necessary.
375 static void qrtr_node_rx_work(struct work_struct *work)
377 struct qrtr_node *node = container_of(work, struct qrtr_node, work);
378 struct qrtr_ctrl_pkt *pkt;
379 struct sockaddr_qrtr dst;
380 struct sockaddr_qrtr src;
383 while ((skb = skb_dequeue(&node->rx_queue)) != NULL) {
384 struct qrtr_sock *ipc;
388 cb = (struct qrtr_cb *)skb->cb;
389 src.sq_node = cb->src_node;
390 src.sq_port = cb->src_port;
391 dst.sq_node = cb->dst_node;
392 dst.sq_port = cb->dst_port;
393 confirm = !!cb->confirm_rx;
395 qrtr_node_assign(node, cb->src_node);
397 ipc = qrtr_port_lookup(cb->dst_port);
401 if (sock_queue_rcv_skb(&ipc->sk, skb))
408 skb = qrtr_alloc_ctrl_packet(&pkt);
412 pkt->cmd = cpu_to_le32(QRTR_TYPE_RESUME_TX);
413 pkt->client.node = cpu_to_le32(dst.sq_node);
414 pkt->client.port = cpu_to_le32(dst.sq_port);
416 if (qrtr_node_enqueue(node, skb, QRTR_TYPE_RESUME_TX,
424 * qrtr_endpoint_register() - register a new endpoint
425 * @ep: endpoint to register
426 * @nid: desired node id; may be QRTR_EP_NID_AUTO for auto-assignment
427 * Return: 0 on success; negative error code on failure
429 * The specified endpoint must have the xmit function pointer set on call.
431 int qrtr_endpoint_register(struct qrtr_endpoint *ep, unsigned int nid)
433 struct qrtr_node *node;
435 if (!ep || !ep->xmit)
438 node = kzalloc(sizeof(*node), GFP_KERNEL);
442 INIT_WORK(&node->work, qrtr_node_rx_work);
443 kref_init(&node->ref);
444 mutex_init(&node->ep_lock);
445 skb_queue_head_init(&node->rx_queue);
446 node->nid = QRTR_EP_NID_AUTO;
449 qrtr_node_assign(node, nid);
451 mutex_lock(&qrtr_node_lock);
452 list_add(&node->item, &qrtr_all_nodes);
453 mutex_unlock(&qrtr_node_lock);
458 EXPORT_SYMBOL_GPL(qrtr_endpoint_register);
461 * qrtr_endpoint_unregister - unregister endpoint
462 * @ep: endpoint to unregister
464 void qrtr_endpoint_unregister(struct qrtr_endpoint *ep)
466 struct qrtr_node *node = ep->node;
467 struct sockaddr_qrtr src = {AF_QIPCRTR, node->nid, QRTR_PORT_CTRL};
468 struct sockaddr_qrtr dst = {AF_QIPCRTR, qrtr_local_nid, QRTR_PORT_CTRL};
469 struct qrtr_ctrl_pkt *pkt;
472 mutex_lock(&node->ep_lock);
474 mutex_unlock(&node->ep_lock);
476 /* Notify the local controller about the event */
477 skb = qrtr_alloc_ctrl_packet(&pkt);
479 pkt->cmd = cpu_to_le32(QRTR_TYPE_BYE);
480 qrtr_local_enqueue(NULL, skb, QRTR_TYPE_BYE, &src, &dst);
483 qrtr_node_release(node);
486 EXPORT_SYMBOL_GPL(qrtr_endpoint_unregister);
488 /* Lookup socket by port.
490 * Callers must release with qrtr_port_put()
492 static struct qrtr_sock *qrtr_port_lookup(int port)
494 struct qrtr_sock *ipc;
496 if (port == QRTR_PORT_CTRL)
499 mutex_lock(&qrtr_port_lock);
500 ipc = idr_find(&qrtr_ports, port);
503 mutex_unlock(&qrtr_port_lock);
508 /* Release acquired socket. */
509 static void qrtr_port_put(struct qrtr_sock *ipc)
514 /* Remove port assignment. */
515 static void qrtr_port_remove(struct qrtr_sock *ipc)
517 struct qrtr_ctrl_pkt *pkt;
519 int port = ipc->us.sq_port;
520 struct sockaddr_qrtr to;
522 to.sq_family = AF_QIPCRTR;
523 to.sq_node = QRTR_NODE_BCAST;
524 to.sq_port = QRTR_PORT_CTRL;
526 skb = qrtr_alloc_ctrl_packet(&pkt);
528 pkt->cmd = cpu_to_le32(QRTR_TYPE_DEL_CLIENT);
529 pkt->client.node = cpu_to_le32(ipc->us.sq_node);
530 pkt->client.port = cpu_to_le32(ipc->us.sq_port);
532 skb_set_owner_w(skb, &ipc->sk);
533 qrtr_bcast_enqueue(NULL, skb, QRTR_TYPE_DEL_CLIENT, &ipc->us,
537 if (port == QRTR_PORT_CTRL)
540 __sock_put(&ipc->sk);
542 mutex_lock(&qrtr_port_lock);
543 idr_remove(&qrtr_ports, port);
544 mutex_unlock(&qrtr_port_lock);
547 /* Assign port number to socket.
549 * Specify port in the integer pointed to by port, and it will be adjusted
550 * on return as necesssary.
553 * 0: Assign ephemeral port in [QRTR_MIN_EPH_SOCKET, QRTR_MAX_EPH_SOCKET]
554 * <QRTR_MIN_EPH_SOCKET: Specified; requires CAP_NET_ADMIN
555 * >QRTR_MIN_EPH_SOCKET: Specified; available to all
557 static int qrtr_port_assign(struct qrtr_sock *ipc, int *port)
562 mutex_lock(&qrtr_port_lock);
564 min_port = QRTR_MIN_EPH_SOCKET;
565 rc = idr_alloc_u32(&qrtr_ports, ipc, &min_port, QRTR_MAX_EPH_SOCKET, GFP_ATOMIC);
568 } else if (*port < QRTR_MIN_EPH_SOCKET && !capable(CAP_NET_ADMIN)) {
570 } else if (*port == QRTR_PORT_CTRL) {
572 rc = idr_alloc_u32(&qrtr_ports, ipc, &min_port, 0, GFP_ATOMIC);
575 rc = idr_alloc_u32(&qrtr_ports, ipc, &min_port, *port, GFP_ATOMIC);
579 mutex_unlock(&qrtr_port_lock);
591 /* Reset all non-control ports */
592 static void qrtr_reset_ports(void)
594 struct qrtr_sock *ipc;
597 mutex_lock(&qrtr_port_lock);
598 idr_for_each_entry(&qrtr_ports, ipc, id) {
599 /* Don't reset control port */
604 ipc->sk.sk_err = ENETRESET;
605 ipc->sk.sk_error_report(&ipc->sk);
608 mutex_unlock(&qrtr_port_lock);
611 /* Bind socket to address.
613 * Socket should be locked upon call.
615 static int __qrtr_bind(struct socket *sock,
616 const struct sockaddr_qrtr *addr, int zapped)
618 struct qrtr_sock *ipc = qrtr_sk(sock->sk);
619 struct sock *sk = sock->sk;
624 if (!zapped && addr->sq_port == ipc->us.sq_port)
627 port = addr->sq_port;
628 rc = qrtr_port_assign(ipc, &port);
632 /* unbind previous, if any */
634 qrtr_port_remove(ipc);
635 ipc->us.sq_port = port;
637 sock_reset_flag(sk, SOCK_ZAPPED);
639 /* Notify all open ports about the new controller */
640 if (port == QRTR_PORT_CTRL)
646 /* Auto bind to an ephemeral port. */
647 static int qrtr_autobind(struct socket *sock)
649 struct sock *sk = sock->sk;
650 struct sockaddr_qrtr addr;
652 if (!sock_flag(sk, SOCK_ZAPPED))
655 addr.sq_family = AF_QIPCRTR;
656 addr.sq_node = qrtr_local_nid;
659 return __qrtr_bind(sock, &addr, 1);
662 /* Bind socket to specified sockaddr. */
663 static int qrtr_bind(struct socket *sock, struct sockaddr *saddr, int len)
665 DECLARE_SOCKADDR(struct sockaddr_qrtr *, addr, saddr);
666 struct qrtr_sock *ipc = qrtr_sk(sock->sk);
667 struct sock *sk = sock->sk;
670 if (len < sizeof(*addr) || addr->sq_family != AF_QIPCRTR)
673 if (addr->sq_node != ipc->us.sq_node)
677 rc = __qrtr_bind(sock, addr, sock_flag(sk, SOCK_ZAPPED));
683 /* Queue packet to local peer socket. */
684 static int qrtr_local_enqueue(struct qrtr_node *node, struct sk_buff *skb,
685 int type, struct sockaddr_qrtr *from,
686 struct sockaddr_qrtr *to)
688 struct qrtr_sock *ipc;
691 ipc = qrtr_port_lookup(to->sq_port);
692 if (!ipc || &ipc->sk == skb->sk) { /* do not send to self */
697 cb = (struct qrtr_cb *)skb->cb;
698 cb->src_node = from->sq_node;
699 cb->src_port = from->sq_port;
701 if (sock_queue_rcv_skb(&ipc->sk, skb)) {
712 /* Queue packet for broadcast. */
713 static int qrtr_bcast_enqueue(struct qrtr_node *node, struct sk_buff *skb,
714 int type, struct sockaddr_qrtr *from,
715 struct sockaddr_qrtr *to)
717 struct sk_buff *skbn;
719 mutex_lock(&qrtr_node_lock);
720 list_for_each_entry(node, &qrtr_all_nodes, item) {
721 skbn = skb_clone(skb, GFP_KERNEL);
724 skb_set_owner_w(skbn, skb->sk);
725 qrtr_node_enqueue(node, skbn, type, from, to);
727 mutex_unlock(&qrtr_node_lock);
729 qrtr_local_enqueue(NULL, skb, type, from, to);
734 static int qrtr_sendmsg(struct socket *sock, struct msghdr *msg, size_t len)
736 DECLARE_SOCKADDR(struct sockaddr_qrtr *, addr, msg->msg_name);
737 int (*enqueue_fn)(struct qrtr_node *, struct sk_buff *, int,
738 struct sockaddr_qrtr *, struct sockaddr_qrtr *);
739 struct qrtr_sock *ipc = qrtr_sk(sock->sk);
740 struct sock *sk = sock->sk;
741 struct qrtr_node *node;
744 u32 type = QRTR_TYPE_DATA;
747 if (msg->msg_flags & ~(MSG_DONTWAIT))
756 if (msg->msg_namelen < sizeof(*addr)) {
761 if (addr->sq_family != AF_QIPCRTR) {
766 rc = qrtr_autobind(sock);
771 } else if (sk->sk_state == TCP_ESTABLISHED) {
779 if (addr->sq_node == QRTR_NODE_BCAST) {
780 if (addr->sq_port != QRTR_PORT_CTRL &&
781 qrtr_local_nid != QRTR_NODE_BCAST) {
785 enqueue_fn = qrtr_bcast_enqueue;
786 } else if (addr->sq_node == ipc->us.sq_node) {
787 enqueue_fn = qrtr_local_enqueue;
789 node = qrtr_node_lookup(addr->sq_node);
794 enqueue_fn = qrtr_node_enqueue;
797 plen = (len + 3) & ~3;
798 skb = sock_alloc_send_skb(sk, plen + QRTR_HDR_MAX_SIZE,
799 msg->msg_flags & MSG_DONTWAIT, &rc);
805 skb_reserve(skb, QRTR_HDR_MAX_SIZE);
807 rc = memcpy_from_msg(skb_put(skb, len), msg, len);
813 if (ipc->us.sq_port == QRTR_PORT_CTRL) {
820 /* control messages already require the type as 'command' */
821 skb_copy_bits(skb, 0, &type, 4);
822 type = le32_to_cpu(type);
825 rc = enqueue_fn(node, skb, type, &ipc->us, addr);
830 qrtr_node_release(node);
836 static int qrtr_recvmsg(struct socket *sock, struct msghdr *msg,
837 size_t size, int flags)
839 DECLARE_SOCKADDR(struct sockaddr_qrtr *, addr, msg->msg_name);
840 struct sock *sk = sock->sk;
847 if (sock_flag(sk, SOCK_ZAPPED)) {
849 return -EADDRNOTAVAIL;
852 skb = skb_recv_datagram(sk, flags & ~MSG_DONTWAIT,
853 flags & MSG_DONTWAIT, &rc);
862 msg->msg_flags |= MSG_TRUNC;
865 rc = skb_copy_datagram_msg(skb, 0, msg, copied);
871 /* There is an anonymous 2-byte hole after sq_family,
872 * make sure to clear it.
874 memset(addr, 0, sizeof(*addr));
876 cb = (struct qrtr_cb *)skb->cb;
877 addr->sq_family = AF_QIPCRTR;
878 addr->sq_node = cb->src_node;
879 addr->sq_port = cb->src_port;
880 msg->msg_namelen = sizeof(*addr);
884 skb_free_datagram(sk, skb);
890 static int qrtr_connect(struct socket *sock, struct sockaddr *saddr,
893 DECLARE_SOCKADDR(struct sockaddr_qrtr *, addr, saddr);
894 struct qrtr_sock *ipc = qrtr_sk(sock->sk);
895 struct sock *sk = sock->sk;
898 if (len < sizeof(*addr) || addr->sq_family != AF_QIPCRTR)
903 sk->sk_state = TCP_CLOSE;
904 sock->state = SS_UNCONNECTED;
906 rc = qrtr_autobind(sock);
913 sock->state = SS_CONNECTED;
914 sk->sk_state = TCP_ESTABLISHED;
921 static int qrtr_getname(struct socket *sock, struct sockaddr *saddr,
924 struct qrtr_sock *ipc = qrtr_sk(sock->sk);
925 struct sockaddr_qrtr qaddr;
926 struct sock *sk = sock->sk;
930 if (sk->sk_state != TCP_ESTABLISHED) {
941 qaddr.sq_family = AF_QIPCRTR;
943 memcpy(saddr, &qaddr, sizeof(qaddr));
945 return sizeof(qaddr);
948 static int qrtr_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
950 void __user *argp = (void __user *)arg;
951 struct qrtr_sock *ipc = qrtr_sk(sock->sk);
952 struct sock *sk = sock->sk;
953 struct sockaddr_qrtr *sq;
963 len = sk->sk_sndbuf - sk_wmem_alloc_get(sk);
966 rc = put_user(len, (int __user *)argp);
969 skb = skb_peek(&sk->sk_receive_queue);
972 rc = put_user(len, (int __user *)argp);
975 if (copy_from_user(&ifr, argp, sizeof(ifr))) {
980 sq = (struct sockaddr_qrtr *)&ifr.ifr_addr;
982 if (copy_to_user(argp, &ifr, sizeof(ifr))) {
988 rc = sock_get_timestamp(sk, argp);
1011 static int qrtr_release(struct socket *sock)
1013 struct sock *sk = sock->sk;
1014 struct qrtr_sock *ipc;
1022 sk->sk_shutdown = SHUTDOWN_MASK;
1023 if (!sock_flag(sk, SOCK_DEAD))
1024 sk->sk_state_change(sk);
1026 sock_set_flag(sk, SOCK_DEAD);
1030 if (!sock_flag(sk, SOCK_ZAPPED))
1031 qrtr_port_remove(ipc);
1033 skb_queue_purge(&sk->sk_receive_queue);
1041 static const struct proto_ops qrtr_proto_ops = {
1042 .owner = THIS_MODULE,
1043 .family = AF_QIPCRTR,
1045 .connect = qrtr_connect,
1046 .socketpair = sock_no_socketpair,
1047 .accept = sock_no_accept,
1048 .listen = sock_no_listen,
1049 .sendmsg = qrtr_sendmsg,
1050 .recvmsg = qrtr_recvmsg,
1051 .getname = qrtr_getname,
1052 .ioctl = qrtr_ioctl,
1053 .poll = datagram_poll,
1054 .shutdown = sock_no_shutdown,
1055 .setsockopt = sock_no_setsockopt,
1056 .getsockopt = sock_no_getsockopt,
1057 .release = qrtr_release,
1058 .mmap = sock_no_mmap,
1059 .sendpage = sock_no_sendpage,
1062 static struct proto qrtr_proto = {
1064 .owner = THIS_MODULE,
1065 .obj_size = sizeof(struct qrtr_sock),
1068 static int qrtr_create(struct net *net, struct socket *sock,
1069 int protocol, int kern)
1071 struct qrtr_sock *ipc;
1074 if (sock->type != SOCK_DGRAM)
1077 sk = sk_alloc(net, AF_QIPCRTR, GFP_KERNEL, &qrtr_proto, kern);
1081 sock_set_flag(sk, SOCK_ZAPPED);
1083 sock_init_data(sock, sk);
1084 sock->ops = &qrtr_proto_ops;
1087 ipc->us.sq_family = AF_QIPCRTR;
1088 ipc->us.sq_node = qrtr_local_nid;
1089 ipc->us.sq_port = 0;
1094 static const struct nla_policy qrtr_policy[IFA_MAX + 1] = {
1095 [IFA_LOCAL] = { .type = NLA_U32 },
1098 static int qrtr_addr_doit(struct sk_buff *skb, struct nlmsghdr *nlh,
1099 struct netlink_ext_ack *extack)
1101 struct nlattr *tb[IFA_MAX + 1];
1102 struct ifaddrmsg *ifm;
1105 if (!netlink_capable(skb, CAP_NET_ADMIN))
1108 if (!netlink_capable(skb, CAP_SYS_ADMIN))
1113 rc = nlmsg_parse(nlh, sizeof(*ifm), tb, IFA_MAX, qrtr_policy, extack);
1117 ifm = nlmsg_data(nlh);
1121 qrtr_local_nid = nla_get_u32(tb[IFA_LOCAL]);
1125 static const struct net_proto_family qrtr_family = {
1126 .owner = THIS_MODULE,
1127 .family = AF_QIPCRTR,
1128 .create = qrtr_create,
1131 static int __init qrtr_proto_init(void)
1135 rc = proto_register(&qrtr_proto, 1);
1139 rc = sock_register(&qrtr_family);
1141 proto_unregister(&qrtr_proto);
1145 rc = rtnl_register_module(THIS_MODULE, PF_QIPCRTR, RTM_NEWADDR, qrtr_addr_doit, NULL, 0);
1147 sock_unregister(qrtr_family.family);
1148 proto_unregister(&qrtr_proto);
1153 postcore_initcall(qrtr_proto_init);
1155 static void __exit qrtr_proto_fini(void)
1157 rtnl_unregister(PF_QIPCRTR, RTM_NEWADDR);
1158 sock_unregister(qrtr_family.family);
1159 proto_unregister(&qrtr_proto);
1161 module_exit(qrtr_proto_fini);
1163 MODULE_DESCRIPTION("Qualcomm IPC-router driver");
1164 MODULE_LICENSE("GPL v2");
1165 MODULE_ALIAS_NETPROTO(PF_QIPCRTR);