net/qrtr/qrtr.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Copyright (c) 2015, Sony Mobile Communications Inc.
   4  * Copyright (c) 2013, The Linux Foundation. All rights reserved.
   5  */
   6 #include <linux/module.h>
   7 #include <linux/netlink.h>
   8 #include <linux/qrtr.h>
   9 #include <linux/termios.h>      /* For TIOCINQ/OUTQ */
  10 #include <linux/numa.h>
  11
  12 #include <net/sock.h>
  13
  14 #include "qrtr.h"
  15
  16 #define QRTR_PROTO_VER_1 1
  17 #define QRTR_PROTO_VER_2 3
  18
  19 /* auto-bind range */
  20 #define QRTR_MIN_EPH_SOCKET 0x4000
  21 #define QRTR_MAX_EPH_SOCKET 0x7fff
  22
  23 /**
  24  * struct qrtr_hdr_v1 - (I|R)PCrouter packet header version 1
  25  * @version: protocol version
  26  * @type: packet type; one of QRTR_TYPE_*
  27  * @src_node_id: source node
  28  * @src_port_id: source port
  29  * @confirm_rx: boolean; whether a resume-tx packet should be send in reply
  30  * @size: length of packet, excluding this header
  31  * @dst_node_id: destination node
  32  * @dst_port_id: destination port
  33  */
  34 struct qrtr_hdr_v1 {
  35         __le32 version;
  36         __le32 type;
  37         __le32 src_node_id;
  38         __le32 src_port_id;
  39         __le32 confirm_rx;
  40         __le32 size;
  41         __le32 dst_node_id;
  42         __le32 dst_port_id;
  43 } __packed;
  44
  45 /**
  46  * struct qrtr_hdr_v2 - (I|R)PCrouter packet header later versions
  47  * @version: protocol version
  48  * @type: packet type; one of QRTR_TYPE_*
  49  * @flags: bitmask of QRTR_FLAGS_*
  50  * @optlen: length of optional header data
  51  * @size: length of packet, excluding this header and optlen
  52  * @src_node_id: source node
  53  * @src_port_id: source port
  54  * @dst_node_id: destination node
  55  * @dst_port_id: destination port
  56  */
  57 struct qrtr_hdr_v2 {
  58         u8 version;
  59         u8 type;
  60         u8 flags;
  61         u8 optlen;
  62         __le32 size;
  63         __le16 src_node_id;
  64         __le16 src_port_id;
  65         __le16 dst_node_id;
  66         __le16 dst_port_id;
  67 };
  68
  69 #define QRTR_FLAGS_CONFIRM_RX   BIT(0)
  70
  71 struct qrtr_cb {
  72         u32 src_node;
  73         u32 src_port;
  74         u32 dst_node;
  75         u32 dst_port;
  76
  77         u8 type;
  78         u8 confirm_rx;
  79 };
  80
  81 #define QRTR_HDR_MAX_SIZE max_t(size_t, sizeof(struct qrtr_hdr_v1), \
  82                                         sizeof(struct qrtr_hdr_v2))
  83
  84 struct qrtr_sock {
  85         /* WARNING: sk must be the first member */
  86         struct sock sk;
  87         struct sockaddr_qrtr us;
  88         struct sockaddr_qrtr peer;
  89 };
  90
  91 static inline struct qrtr_sock *qrtr_sk(struct sock *sk)
  92 {
  93         BUILD_BUG_ON(offsetof(struct qrtr_sock, sk) != 0);
  94         return container_of(sk, struct qrtr_sock, sk);
  95 }
  96
  97 static unsigned int qrtr_local_nid = NUMA_NO_NODE;
  98
  99 /* for node ids */
 100 static RADIX_TREE(qrtr_nodes, GFP_KERNEL);
 101 /* broadcast list */
 102 static LIST_HEAD(qrtr_all_nodes);
 103 /* lock for qrtr_nodes, qrtr_all_nodes and node reference */
 104 static DEFINE_MUTEX(qrtr_node_lock);
 105
 106 /* local port allocation management */
 107 static DEFINE_IDR(qrtr_ports);
 108 static DEFINE_MUTEX(qrtr_port_lock);
 109
 110 /**
 111  * struct qrtr_node - endpoint node
 112  * @ep_lock: lock for endpoint management and callbacks
 113  * @ep: endpoint
 114  * @ref: reference count for node
 115  * @nid: node id
 116  * @rx_queue: receive queue
 117  * @work: scheduled work struct for recv work
 118  * @item: list item for broadcast list
 119  */
 120 struct qrtr_node {
 121         struct mutex ep_lock;
 122         struct qrtr_endpoint *ep;
 123         struct kref ref;
 124         unsigned int nid;
 125
 126         struct sk_buff_head rx_queue;
 127         struct work_struct work;
 128         struct list_head item;
 129 };
 130
 131 static int qrtr_local_enqueue(struct qrtr_node *node, struct sk_buff *skb,
 132                               int type, struct sockaddr_qrtr *from,
 133                               struct sockaddr_qrtr *to);
 134 static int qrtr_bcast_enqueue(struct qrtr_node *node, struct sk_buff *skb,
 135                               int type, struct sockaddr_qrtr *from,
 136                               struct sockaddr_qrtr *to);
 137
 138 /* Release node resources and free the node.
 139  *
 140  * Do not call directly, use qrtr_node_release.  To be used with
 141  * kref_put_mutex.  As such, the node mutex is expected to be locked on call.
 142  */
 143 static void __qrtr_node_release(struct kref *kref)
 144 {
 145         struct qrtr_node *node = container_of(kref, struct qrtr_node, ref);
 146
 147         if (node->nid != QRTR_EP_NID_AUTO)
 148                 radix_tree_delete(&qrtr_nodes, node->nid);
 149
 150         list_del(&node->item);
 151         mutex_unlock(&qrtr_node_lock);
 152
 153         cancel_work_sync(&node->work);
 154         skb_queue_purge(&node->rx_queue);
 155         kfree(node);
 156 }
 157
 158 /* Increment reference to node. */
 159 static struct qrtr_node *qrtr_node_acquire(struct qrtr_node *node)
 160 {
 161         if (node)
 162                 kref_get(&node->ref);
 163         return node;
 164 }
 165
 166 /* Decrement reference to node and release as necessary. */
 167 static void qrtr_node_release(struct qrtr_node *node)
 168 {
 169         if (!node)
 170                 return;
 171         kref_put_mutex(&node->ref, __qrtr_node_release, &qrtr_node_lock);
 172 }
 173
 174 /* Pass an outgoing packet socket buffer to the endpoint driver. */
 175 static int qrtr_node_enqueue(struct qrtr_node *node, struct sk_buff *skb,
 176                              int type, struct sockaddr_qrtr *from,
 177                              struct sockaddr_qrtr *to)
 178 {
 179         struct qrtr_hdr_v1 *hdr;
 180         size_t len = skb->len;
 181         int rc = -ENODEV;
 182
 183         hdr = skb_push(skb, sizeof(*hdr));
 184         hdr->version = cpu_to_le32(QRTR_PROTO_VER_1);
 185         hdr->type = cpu_to_le32(type);
 186         hdr->src_node_id = cpu_to_le32(from->sq_node);
 187         hdr->src_port_id = cpu_to_le32(from->sq_port);
 188         if (to->sq_port == QRTR_PORT_CTRL) {
 189                 hdr->dst_node_id = cpu_to_le32(node->nid);
 190                 hdr->dst_port_id = cpu_to_le32(QRTR_NODE_BCAST);
 191         } else {
 192                 hdr->dst_node_id = cpu_to_le32(to->sq_node);
 193                 hdr->dst_port_id = cpu_to_le32(to->sq_port);
 194         }
 195
 196         hdr->size = cpu_to_le32(len);
 197         hdr->confirm_rx = 0;
 198
 199         skb_put_padto(skb, ALIGN(len, 4) + sizeof(*hdr));
 200
 201         mutex_lock(&node->ep_lock);
 202         if (node->ep)
 203                 rc = node->ep->xmit(node->ep, skb);
 204         else
 205                 kfree_skb(skb);
 206         mutex_unlock(&node->ep_lock);
 207
 208         return rc;
 209 }
 210
 211 /* Lookup node by id.
 212  *
 213  * callers must release with qrtr_node_release()
 214  */
 215 static struct qrtr_node *qrtr_node_lookup(unsigned int nid)
 216 {
 217         struct qrtr_node *node;
 218
 219         mutex_lock(&qrtr_node_lock);
 220         node = radix_tree_lookup(&qrtr_nodes, nid);
 221         node = qrtr_node_acquire(node);
 222         mutex_unlock(&qrtr_node_lock);
 223
 224         return node;
 225 }
 226
 227 /* Assign node id to node.
 228  *
 229  * This is mostly useful for automatic node id assignment, based on
 230  * the source id in the incoming packet.
 231  */
 232 static void qrtr_node_assign(struct qrtr_node *node, unsigned int nid)
 233 {
 234         if (node->nid != QRTR_EP_NID_AUTO || nid == QRTR_EP_NID_AUTO)
 235                 return;
 236
 237         mutex_lock(&qrtr_node_lock);
 238         radix_tree_insert(&qrtr_nodes, nid, node);
 239         node->nid = nid;
 240         mutex_unlock(&qrtr_node_lock);
 241 }
 242
 243 /**
 244  * qrtr_endpoint_post() - post incoming data
 245  * @ep: endpoint handle
 246  * @data: data pointer
 247  * @len: size of data in bytes
 248  *
 249  * Return: 0 on success; negative error code on failure
 250  */
 251 int qrtr_endpoint_post(struct qrtr_endpoint *ep, const void *data, size_t len)
 252 {
 253         struct qrtr_node *node = ep->node;
 254         const struct qrtr_hdr_v1 *v1;
 255         const struct qrtr_hdr_v2 *v2;
 256         struct sk_buff *skb;
 257         struct qrtr_cb *cb;
 258         unsigned int size;
 259         unsigned int ver;
 260         size_t hdrlen;
 261
 262         if (len & 3)
 263                 return -EINVAL;
 264
 265         skb = netdev_alloc_skb(NULL, len);
 266         if (!skb)
 267                 return -ENOMEM;
 268
 269         cb = (struct qrtr_cb *)skb->cb;
 270
 271         /* Version field in v1 is little endian, so this works for both cases */
 272         ver = *(u8*)data;
 273
 274         switch (ver) {
 275         case QRTR_PROTO_VER_1:
 276                 v1 = data;
 277                 hdrlen = sizeof(*v1);
 278
 279                 cb->type = le32_to_cpu(v1->type);
 280                 cb->src_node = le32_to_cpu(v1->src_node_id);
 281                 cb->src_port = le32_to_cpu(v1->src_port_id);
 282                 cb->confirm_rx = !!v1->confirm_rx;
 283                 cb->dst_node = le32_to_cpu(v1->dst_node_id);
 284                 cb->dst_port = le32_to_cpu(v1->dst_port_id);
 285
 286                 size = le32_to_cpu(v1->size);
 287                 break;
 288         case QRTR_PROTO_VER_2:
 289                 v2 = data;
 290                 hdrlen = sizeof(*v2) + v2->optlen;
 291
 292                 cb->type = v2->type;
 293                 cb->confirm_rx = !!(v2->flags & QRTR_FLAGS_CONFIRM_RX);
 294                 cb->src_node = le16_to_cpu(v2->src_node_id);
 295                 cb->src_port = le16_to_cpu(v2->src_port_id);
 296                 cb->dst_node = le16_to_cpu(v2->dst_node_id);
 297                 cb->dst_port = le16_to_cpu(v2->dst_port_id);
 298
 299                 if (cb->src_port == (u16)QRTR_PORT_CTRL)
 300                         cb->src_port = QRTR_PORT_CTRL;
 301                 if (cb->dst_port == (u16)QRTR_PORT_CTRL)
 302                         cb->dst_port = QRTR_PORT_CTRL;
 303
 304                 size = le32_to_cpu(v2->size);
 305                 break;
 306         default:
 307                 pr_err("qrtr: Invalid version %d\n", ver);
 308                 goto err;
 309         }
 310
 311         if (len != ALIGN(size, 4) + hdrlen)
 312                 goto err;
 313
 314         if (cb->dst_port != QRTR_PORT_CTRL && cb->type != QRTR_TYPE_DATA)
 315                 goto err;
 316
 317         skb_put_data(skb, data + hdrlen, size);
 318
 319         skb_queue_tail(&node->rx_queue, skb);
 320         schedule_work(&node->work);
 321
 322         return 0;
 323
 324 err:
 325         kfree_skb(skb);
 326         return -EINVAL;
 327
 328 }
 329 EXPORT_SYMBOL_GPL(qrtr_endpoint_post);
 330
 331 /**
 332  * qrtr_alloc_ctrl_packet() - allocate control packet skb
 333  * @pkt: reference to qrtr_ctrl_pkt pointer
 334  *
 335  * Returns newly allocated sk_buff, or NULL on failure
 336  *
 337  * This function allocates a sk_buff large enough to carry a qrtr_ctrl_pkt and
 338  * on success returns a reference to the control packet in @pkt.
 339  */
 340 static struct sk_buff *qrtr_alloc_ctrl_packet(struct qrtr_ctrl_pkt **pkt)
 341 {
 342         const int pkt_len = sizeof(struct qrtr_ctrl_pkt);
 343         struct sk_buff *skb;
 344
 345         skb = alloc_skb(QRTR_HDR_MAX_SIZE + pkt_len, GFP_KERNEL);
 346         if (!skb)
 347                 return NULL;
 348
 349         skb_reserve(skb, QRTR_HDR_MAX_SIZE);
 350         *pkt = skb_put_zero(skb, pkt_len);
 351
 352         return skb;
 353 }
 354
 355 static struct qrtr_sock *qrtr_port_lookup(int port);
 356 static void qrtr_port_put(struct qrtr_sock *ipc);
 357
 358 /* Handle and route a received packet.
 359  *
 360  * This will auto-reply with resume-tx packet as necessary.
 361  */
 362 static void qrtr_node_rx_work(struct work_struct *work)
 363 {
 364         struct qrtr_node *node = container_of(work, struct qrtr_node, work);
 365         struct qrtr_ctrl_pkt *pkt;
 366         struct sockaddr_qrtr dst;
 367         struct sockaddr_qrtr src;
 368         struct sk_buff *skb;
 369
 370         while ((skb = skb_dequeue(&node->rx_queue)) != NULL) {
 371                 struct qrtr_sock *ipc;
 372                 struct qrtr_cb *cb;
 373                 int confirm;
 374
 375                 cb = (struct qrtr_cb *)skb->cb;
 376                 src.sq_node = cb->src_node;
 377                 src.sq_port = cb->src_port;
 378                 dst.sq_node = cb->dst_node;
 379                 dst.sq_port = cb->dst_port;
 380                 confirm = !!cb->confirm_rx;
 381
 382                 qrtr_node_assign(node, cb->src_node);
 383
 384                 ipc = qrtr_port_lookup(cb->dst_port);
 385                 if (!ipc) {
 386                         kfree_skb(skb);
 387                 } else {
 388                         if (sock_queue_rcv_skb(&ipc->sk, skb))
 389                                 kfree_skb(skb);
 390
 391                         qrtr_port_put(ipc);
 392                 }
 393
 394                 if (confirm) {
 395                         skb = qrtr_alloc_ctrl_packet(&pkt);
 396                         if (!skb)
 397                                 break;
 398
 399                         pkt->cmd = cpu_to_le32(QRTR_TYPE_RESUME_TX);
 400                         pkt->client.node = cpu_to_le32(dst.sq_node);
 401                         pkt->client.port = cpu_to_le32(dst.sq_port);
 402
 403                         if (qrtr_node_enqueue(node, skb, QRTR_TYPE_RESUME_TX,
 404                                               &dst, &src))
 405                                 break;
 406                 }
 407         }
 408 }
 409
 410 /**
 411  * qrtr_endpoint_register() - register a new endpoint
 412  * @ep: endpoint to register
 413  * @nid: desired node id; may be QRTR_EP_NID_AUTO for auto-assignment
 414  * Return: 0 on success; negative error code on failure
 415  *
 416  * The specified endpoint must have the xmit function pointer set on call.
 417  */
 418 int qrtr_endpoint_register(struct qrtr_endpoint *ep, unsigned int nid)
 419 {
 420         struct qrtr_node *node;
 421
 422         if (!ep || !ep->xmit)
 423                 return -EINVAL;
 424
 425         node = kzalloc(sizeof(*node), GFP_KERNEL);
 426         if (!node)
 427                 return -ENOMEM;
 428
 429         INIT_WORK(&node->work, qrtr_node_rx_work);
 430         kref_init(&node->ref);
 431         mutex_init(&node->ep_lock);
 432         skb_queue_head_init(&node->rx_queue);
 433         node->nid = QRTR_EP_NID_AUTO;
 434         node->ep = ep;
 435
 436         qrtr_node_assign(node, nid);
 437
 438         mutex_lock(&qrtr_node_lock);
 439         list_add(&node->item, &qrtr_all_nodes);
 440         mutex_unlock(&qrtr_node_lock);
 441         ep->node = node;
 442
 443         return 0;
 444 }
 445 EXPORT_SYMBOL_GPL(qrtr_endpoint_register);
 446
 447 /**
 448  * qrtr_endpoint_unregister - unregister endpoint
 449  * @ep: endpoint to unregister
 450  */
 451 void qrtr_endpoint_unregister(struct qrtr_endpoint *ep)
 452 {
 453         struct qrtr_node *node = ep->node;
 454         struct sockaddr_qrtr src = {AF_QIPCRTR, node->nid, QRTR_PORT_CTRL};
 455         struct sockaddr_qrtr dst = {AF_QIPCRTR, qrtr_local_nid, QRTR_PORT_CTRL};
 456         struct qrtr_ctrl_pkt *pkt;
 457         struct sk_buff *skb;
 458
 459         mutex_lock(&node->ep_lock);
 460         node->ep = NULL;
 461         mutex_unlock(&node->ep_lock);
 462
 463         /* Notify the local controller about the event */
 464         skb = qrtr_alloc_ctrl_packet(&pkt);
 465         if (skb) {
 466                 pkt->cmd = cpu_to_le32(QRTR_TYPE_BYE);
 467                 qrtr_local_enqueue(NULL, skb, QRTR_TYPE_BYE, &src, &dst);
 468         }
 469
 470         qrtr_node_release(node);
 471         ep->node = NULL;
 472 }
 473 EXPORT_SYMBOL_GPL(qrtr_endpoint_unregister);
 474
 475 /* Lookup socket by port.
 476  *
 477  * Callers must release with qrtr_port_put()
 478  */
 479 static struct qrtr_sock *qrtr_port_lookup(int port)
 480 {
 481         struct qrtr_sock *ipc;
 482
 483         if (port == QRTR_PORT_CTRL)
 484                 port = 0;
 485
 486         mutex_lock(&qrtr_port_lock);
 487         ipc = idr_find(&qrtr_ports, port);
 488         if (ipc)
 489                 sock_hold(&ipc->sk);
 490         mutex_unlock(&qrtr_port_lock);
 491
 492         return ipc;
 493 }
 494
 495 /* Release acquired socket. */
 496 static void qrtr_port_put(struct qrtr_sock *ipc)
 497 {
 498         sock_put(&ipc->sk);
 499 }
 500
 501 /* Remove port assignment. */
 502 static void qrtr_port_remove(struct qrtr_sock *ipc)
 503 {
 504         struct qrtr_ctrl_pkt *pkt;
 505         struct sk_buff *skb;
 506         int port = ipc->us.sq_port;
 507         struct sockaddr_qrtr to;
 508
 509         to.sq_family = AF_QIPCRTR;
 510         to.sq_node = QRTR_NODE_BCAST;
 511         to.sq_port = QRTR_PORT_CTRL;
 512
 513         skb = qrtr_alloc_ctrl_packet(&pkt);
 514         if (skb) {
 515                 pkt->cmd = cpu_to_le32(QRTR_TYPE_DEL_CLIENT);
 516                 pkt->client.node = cpu_to_le32(ipc->us.sq_node);
 517                 pkt->client.port = cpu_to_le32(ipc->us.sq_port);
 518
 519                 skb_set_owner_w(skb, &ipc->sk);
 520                 qrtr_bcast_enqueue(NULL, skb, QRTR_TYPE_DEL_CLIENT, &ipc->us,
 521                                    &to);
 522         }
 523
 524         if (port == QRTR_PORT_CTRL)
 525                 port = 0;
 526
 527         __sock_put(&ipc->sk);
 528
 529         mutex_lock(&qrtr_port_lock);
 530         idr_remove(&qrtr_ports, port);
 531         mutex_unlock(&qrtr_port_lock);
 532 }
 533
 534 /* Assign port number to socket.
 535  *
 536  * Specify port in the integer pointed to by port, and it will be adjusted
 537  * on return as necesssary.
 538  *
 539  * Port may be:
 540  *   0: Assign ephemeral port in [QRTR_MIN_EPH_SOCKET, QRTR_MAX_EPH_SOCKET]
 541  *   <QRTR_MIN_EPH_SOCKET: Specified; requires CAP_NET_ADMIN
 542  *   >QRTR_MIN_EPH_SOCKET: Specified; available to all
 543  */
 544 static int qrtr_port_assign(struct qrtr_sock *ipc, int *port)
 545 {
 546         int rc;
 547
 548         mutex_lock(&qrtr_port_lock);
 549         if (!*port) {
 550                 rc = idr_alloc(&qrtr_ports, ipc,
 551                                QRTR_MIN_EPH_SOCKET, QRTR_MAX_EPH_SOCKET + 1,
 552                                GFP_ATOMIC);
 553                 if (rc >= 0)
 554                         *port = rc;
 555         } else if (*port < QRTR_MIN_EPH_SOCKET && !capable(CAP_NET_ADMIN)) {
 556                 rc = -EACCES;
 557         } else if (*port == QRTR_PORT_CTRL) {
 558                 rc = idr_alloc(&qrtr_ports, ipc, 0, 1, GFP_ATOMIC);
 559         } else {
 560                 rc = idr_alloc(&qrtr_ports, ipc, *port, *port + 1, GFP_ATOMIC);
 561                 if (rc >= 0)
 562                         *port = rc;
 563         }
 564         mutex_unlock(&qrtr_port_lock);
 565
 566         if (rc == -ENOSPC)
 567                 return -EADDRINUSE;
 568         else if (rc < 0)
 569                 return rc;
 570
 571         sock_hold(&ipc->sk);
 572
 573         return 0;
 574 }
 575
 576 /* Reset all non-control ports */
 577 static void qrtr_reset_ports(void)
 578 {
 579         struct qrtr_sock *ipc;
 580         int id;
 581
 582         mutex_lock(&qrtr_port_lock);
 583         idr_for_each_entry(&qrtr_ports, ipc, id) {
 584                 /* Don't reset control port */
 585                 if (id == 0)
 586                         continue;
 587
 588                 sock_hold(&ipc->sk);
 589                 ipc->sk.sk_err = ENETRESET;
 590                 ipc->sk.sk_error_report(&ipc->sk);
 591                 sock_put(&ipc->sk);
 592         }
 593         mutex_unlock(&qrtr_port_lock);
 594 }
 595
 596 /* Bind socket to address.
 597  *
 598  * Socket should be locked upon call.
 599  */
 600 static int __qrtr_bind(struct socket *sock,
 601                        const struct sockaddr_qrtr *addr, int zapped)
 602 {
 603         struct qrtr_sock *ipc = qrtr_sk(sock->sk);
 604         struct sock *sk = sock->sk;
 605         int port;
 606         int rc;
 607
 608         /* rebinding ok */
 609         if (!zapped && addr->sq_port == ipc->us.sq_port)
 610                 return 0;
 611
 612         port = addr->sq_port;
 613         rc = qrtr_port_assign(ipc, &port);
 614         if (rc)
 615                 return rc;
 616
 617         /* unbind previous, if any */
 618         if (!zapped)
 619                 qrtr_port_remove(ipc);
 620         ipc->us.sq_port = port;
 621
 622         sock_reset_flag(sk, SOCK_ZAPPED);
 623
 624         /* Notify all open ports about the new controller */
 625         if (port == QRTR_PORT_CTRL)
 626                 qrtr_reset_ports();
 627
 628         return 0;
 629 }
 630
 631 /* Auto bind to an ephemeral port. */
 632 static int qrtr_autobind(struct socket *sock)
 633 {
 634         struct sock *sk = sock->sk;
 635         struct sockaddr_qrtr addr;
 636
 637         if (!sock_flag(sk, SOCK_ZAPPED))
 638                 return 0;
 639
 640         addr.sq_family = AF_QIPCRTR;
 641         addr.sq_node = qrtr_local_nid;
 642         addr.sq_port = 0;
 643
 644         return __qrtr_bind(sock, &addr, 1);
 645 }
 646
 647 /* Bind socket to specified sockaddr. */
 648 static int qrtr_bind(struct socket *sock, struct sockaddr *saddr, int len)
 649 {
 650         DECLARE_SOCKADDR(struct sockaddr_qrtr *, addr, saddr);
 651         struct qrtr_sock *ipc = qrtr_sk(sock->sk);
 652         struct sock *sk = sock->sk;
 653         int rc;
 654
 655         if (len < sizeof(*addr) || addr->sq_family != AF_QIPCRTR)
 656                 return -EINVAL;
 657
 658         if (addr->sq_node != ipc->us.sq_node)
 659                 return -EINVAL;
 660
 661         lock_sock(sk);
 662         rc = __qrtr_bind(sock, addr, sock_flag(sk, SOCK_ZAPPED));
 663         release_sock(sk);
 664
 665         return rc;
 666 }
 667
 668 /* Queue packet to local peer socket. */
 669 static int qrtr_local_enqueue(struct qrtr_node *node, struct sk_buff *skb,
 670                               int type, struct sockaddr_qrtr *from,
 671                               struct sockaddr_qrtr *to)
 672 {
 673         struct qrtr_sock *ipc;
 674         struct qrtr_cb *cb;
 675
 676         ipc = qrtr_port_lookup(to->sq_port);
 677         if (!ipc || &ipc->sk == skb->sk) { /* do not send to self */
 678                 kfree_skb(skb);
 679                 return -ENODEV;
 680         }
 681
 682         cb = (struct qrtr_cb *)skb->cb;
 683         cb->src_node = from->sq_node;
 684         cb->src_port = from->sq_port;
 685
 686         if (sock_queue_rcv_skb(&ipc->sk, skb)) {
 687                 qrtr_port_put(ipc);
 688                 kfree_skb(skb);
 689                 return -ENOSPC;
 690         }
 691
 692         qrtr_port_put(ipc);
 693
 694         return 0;
 695 }
 696
 697 /* Queue packet for broadcast. */
 698 static int qrtr_bcast_enqueue(struct qrtr_node *node, struct sk_buff *skb,
 699                               int type, struct sockaddr_qrtr *from,
 700                               struct sockaddr_qrtr *to)
 701 {
 702         struct sk_buff *skbn;
 703
 704         mutex_lock(&qrtr_node_lock);
 705         list_for_each_entry(node, &qrtr_all_nodes, item) {
 706                 skbn = skb_clone(skb, GFP_KERNEL);
 707                 if (!skbn)
 708                         break;
 709                 skb_set_owner_w(skbn, skb->sk);
 710                 qrtr_node_enqueue(node, skbn, type, from, to);
 711         }
 712         mutex_unlock(&qrtr_node_lock);
 713
 714         qrtr_local_enqueue(NULL, skb, type, from, to);
 715
 716         return 0;
 717 }
 718
 719 static int qrtr_sendmsg(struct socket *sock, struct msghdr *msg, size_t len)
 720 {
 721         DECLARE_SOCKADDR(struct sockaddr_qrtr *, addr, msg->msg_name);
 722         int (*enqueue_fn)(struct qrtr_node *, struct sk_buff *, int,
 723                           struct sockaddr_qrtr *, struct sockaddr_qrtr *);
 724         __le32 qrtr_type = cpu_to_le32(QRTR_TYPE_DATA);
 725         struct qrtr_sock *ipc = qrtr_sk(sock->sk);
 726         struct sock *sk = sock->sk;
 727         struct qrtr_node *node;
 728         struct sk_buff *skb;
 729         size_t plen;
 730         u32 type;
 731         int rc;
 732
 733         if (msg->msg_flags & ~(MSG_DONTWAIT))
 734                 return -EINVAL;
 735
 736         if (len > 65535)
 737                 return -EMSGSIZE;
 738
 739         lock_sock(sk);
 740
 741         if (addr) {
 742                 if (msg->msg_namelen < sizeof(*addr)) {
 743                         release_sock(sk);
 744                         return -EINVAL;
 745                 }
 746
 747                 if (addr->sq_family != AF_QIPCRTR) {
 748                         release_sock(sk);
 749                         return -EINVAL;
 750                 }
 751
 752                 rc = qrtr_autobind(sock);
 753                 if (rc) {
 754                         release_sock(sk);
 755                         return rc;
 756                 }
 757         } else if (sk->sk_state == TCP_ESTABLISHED) {
 758                 addr = &ipc->peer;
 759         } else {
 760                 release_sock(sk);
 761                 return -ENOTCONN;
 762         }
 763
 764         node = NULL;
 765         if (addr->sq_node == QRTR_NODE_BCAST) {
 766                 if (addr->sq_port != QRTR_PORT_CTRL &&
 767                     qrtr_local_nid != QRTR_NODE_BCAST) {
 768                         release_sock(sk);
 769                         return -ENOTCONN;
 770                 }
 771                 enqueue_fn = qrtr_bcast_enqueue;
 772         } else if (addr->sq_node == ipc->us.sq_node) {
 773                 enqueue_fn = qrtr_local_enqueue;
 774         } else {
 775                 node = qrtr_node_lookup(addr->sq_node);
 776                 if (!node) {
 777                         release_sock(sk);
 778                         return -ECONNRESET;
 779                 }
 780                 enqueue_fn = qrtr_node_enqueue;
 781         }
 782
 783         plen = (len + 3) & ~3;
 784         skb = sock_alloc_send_skb(sk, plen + QRTR_HDR_MAX_SIZE,
 785                                   msg->msg_flags & MSG_DONTWAIT, &rc);
 786         if (!skb)
 787                 goto out_node;
 788
 789         skb_reserve(skb, QRTR_HDR_MAX_SIZE);
 790
 791         rc = memcpy_from_msg(skb_put(skb, len), msg, len);
 792         if (rc) {
 793                 kfree_skb(skb);
 794                 goto out_node;
 795         }
 796
 797         if (ipc->us.sq_port == QRTR_PORT_CTRL) {
 798                 if (len < 4) {
 799                         rc = -EINVAL;
 800                         kfree_skb(skb);
 801                         goto out_node;
 802                 }
 803
 804                 /* control messages already require the type as 'command' */
 805                 skb_copy_bits(skb, 0, &qrtr_type, 4);
 806         }
 807
 808         type = le32_to_cpu(qrtr_type);
 809         rc = enqueue_fn(node, skb, type, &ipc->us, addr);
 810         if (rc >= 0)
 811                 rc = len;
 812
 813 out_node:
 814         qrtr_node_release(node);
 815         release_sock(sk);
 816
 817         return rc;
 818 }
 819
 820 static int qrtr_recvmsg(struct socket *sock, struct msghdr *msg,
 821                         size_t size, int flags)
 822 {
 823         DECLARE_SOCKADDR(struct sockaddr_qrtr *, addr, msg->msg_name);
 824         struct sock *sk = sock->sk;
 825         struct sk_buff *skb;
 826         struct qrtr_cb *cb;
 827         int copied, rc;
 828
 829         lock_sock(sk);
 830
 831         if (sock_flag(sk, SOCK_ZAPPED)) {
 832                 release_sock(sk);
 833                 return -EADDRNOTAVAIL;
 834         }
 835
 836         skb = skb_recv_datagram(sk, flags & ~MSG_DONTWAIT,
 837                                 flags & MSG_DONTWAIT, &rc);
 838         if (!skb) {
 839                 release_sock(sk);
 840                 return rc;
 841         }
 842
 843         copied = skb->len;
 844         if (copied > size) {
 845                 copied = size;
 846                 msg->msg_flags |= MSG_TRUNC;
 847         }
 848
 849         rc = skb_copy_datagram_msg(skb, 0, msg, copied);
 850         if (rc < 0)
 851                 goto out;
 852         rc = copied;
 853
 854         if (addr) {
 855                 cb = (struct qrtr_cb *)skb->cb;
 856                 addr->sq_family = AF_QIPCRTR;
 857                 addr->sq_node = cb->src_node;
 858                 addr->sq_port = cb->src_port;
 859                 msg->msg_namelen = sizeof(*addr);
 860         }
 861
 862 out:
 863         skb_free_datagram(sk, skb);
 864         release_sock(sk);
 865
 866         return rc;
 867 }
 868
 869 static int qrtr_connect(struct socket *sock, struct sockaddr *saddr,
 870                         int len, int flags)
 871 {
 872         DECLARE_SOCKADDR(struct sockaddr_qrtr *, addr, saddr);
 873         struct qrtr_sock *ipc = qrtr_sk(sock->sk);
 874         struct sock *sk = sock->sk;
 875         int rc;
 876
 877         if (len < sizeof(*addr) || addr->sq_family != AF_QIPCRTR)
 878                 return -EINVAL;
 879
 880         lock_sock(sk);
 881
 882         sk->sk_state = TCP_CLOSE;
 883         sock->state = SS_UNCONNECTED;
 884
 885         rc = qrtr_autobind(sock);
 886         if (rc) {
 887                 release_sock(sk);
 888                 return rc;
 889         }
 890
 891         ipc->peer = *addr;
 892         sock->state = SS_CONNECTED;
 893         sk->sk_state = TCP_ESTABLISHED;
 894
 895         release_sock(sk);
 896
 897         return 0;
 898 }
 899
 900 static int qrtr_getname(struct socket *sock, struct sockaddr *saddr,
 901                         int peer)
 902 {
 903         struct qrtr_sock *ipc = qrtr_sk(sock->sk);
 904         struct sockaddr_qrtr qaddr;
 905         struct sock *sk = sock->sk;
 906
 907         lock_sock(sk);
 908         if (peer) {
 909                 if (sk->sk_state != TCP_ESTABLISHED) {
 910                         release_sock(sk);
 911                         return -ENOTCONN;
 912                 }
 913
 914                 qaddr = ipc->peer;
 915         } else {
 916                 qaddr = ipc->us;
 917         }
 918         release_sock(sk);
 919
 920         qaddr.sq_family = AF_QIPCRTR;
 921
 922         memcpy(saddr, &qaddr, sizeof(qaddr));
 923
 924         return sizeof(qaddr);
 925 }
 926
 927 static int qrtr_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
 928 {
 929         void __user *argp = (void __user *)arg;
 930         struct qrtr_sock *ipc = qrtr_sk(sock->sk);
 931         struct sock *sk = sock->sk;
 932         struct sockaddr_qrtr *sq;
 933         struct sk_buff *skb;
 934         struct ifreq ifr;
 935         long len = 0;
 936         int rc = 0;
 937
 938         lock_sock(sk);
 939
 940         switch (cmd) {
 941         case TIOCOUTQ:
 942                 len = sk->sk_sndbuf - sk_wmem_alloc_get(sk);
 943                 if (len < 0)
 944                         len = 0;
 945                 rc = put_user(len, (int __user *)argp);
 946                 break;
 947         case TIOCINQ:
 948                 skb = skb_peek(&sk->sk_receive_queue);
 949                 if (skb)
 950                         len = skb->len;
 951                 rc = put_user(len, (int __user *)argp);
 952                 break;
 953         case SIOCGIFADDR:
 954                 if (copy_from_user(&ifr, argp, sizeof(ifr))) {
 955                         rc = -EFAULT;
 956                         break;
 957                 }
 958
 959                 sq = (struct sockaddr_qrtr *)&ifr.ifr_addr;
 960                 *sq = ipc->us;
 961                 if (copy_to_user(argp, &ifr, sizeof(ifr))) {
 962                         rc = -EFAULT;
 963                         break;
 964                 }
 965                 break;
 966         case SIOCADDRT:
 967         case SIOCDELRT:
 968         case SIOCSIFADDR:
 969         case SIOCGIFDSTADDR:
 970         case SIOCSIFDSTADDR:
 971         case SIOCGIFBRDADDR:
 972         case SIOCSIFBRDADDR:
 973         case SIOCGIFNETMASK:
 974         case SIOCSIFNETMASK:
 975                 rc = -EINVAL;
 976                 break;
 977         default:
 978                 rc = -ENOIOCTLCMD;
 979                 break;
 980         }
 981
 982         release_sock(sk);
 983
 984         return rc;
 985 }
 986
 987 static int qrtr_release(struct socket *sock)
 988 {
 989         struct sock *sk = sock->sk;
 990         struct qrtr_sock *ipc;
 991
 992         if (!sk)
 993                 return 0;
 994
 995         lock_sock(sk);
 996
 997         ipc = qrtr_sk(sk);
 998         sk->sk_shutdown = SHUTDOWN_MASK;
 999         if (!sock_flag(sk, SOCK_DEAD))
1000                 sk->sk_state_change(sk);
1001
1002         sock_set_flag(sk, SOCK_DEAD);
1003         sock->sk = NULL;
1004
1005         if (!sock_flag(sk, SOCK_ZAPPED))
1006                 qrtr_port_remove(ipc);
1007
1008         skb_queue_purge(&sk->sk_receive_queue);
1009
1010         release_sock(sk);
1011         sock_put(sk);
1012
1013         return 0;
1014 }
1015
1016 static const struct proto_ops qrtr_proto_ops = {
1017         .owner          = THIS_MODULE,
1018         .family         = AF_QIPCRTR,
1019         .bind           = qrtr_bind,
1020         .connect        = qrtr_connect,
1021         .socketpair     = sock_no_socketpair,
1022         .accept         = sock_no_accept,
1023         .listen         = sock_no_listen,
1024         .sendmsg        = qrtr_sendmsg,
1025         .recvmsg        = qrtr_recvmsg,
1026         .getname        = qrtr_getname,
1027         .ioctl          = qrtr_ioctl,
1028         .gettstamp      = sock_gettstamp,
1029         .poll           = datagram_poll,
1030         .shutdown       = sock_no_shutdown,
1031         .setsockopt     = sock_no_setsockopt,
1032         .getsockopt     = sock_no_getsockopt,
1033         .release        = qrtr_release,
1034         .mmap           = sock_no_mmap,
1035         .sendpage       = sock_no_sendpage,
1036 };
1037
1038 static struct proto qrtr_proto = {
1039         .name           = "QIPCRTR",
1040         .owner          = THIS_MODULE,
1041         .obj_size       = sizeof(struct qrtr_sock),
1042 };
1043
1044 static int qrtr_create(struct net *net, struct socket *sock,
1045                        int protocol, int kern)
1046 {
1047         struct qrtr_sock *ipc;
1048         struct sock *sk;
1049
1050         if (sock->type != SOCK_DGRAM)
1051                 return -EPROTOTYPE;
1052
1053         sk = sk_alloc(net, AF_QIPCRTR, GFP_KERNEL, &qrtr_proto, kern);
1054         if (!sk)
1055                 return -ENOMEM;
1056
1057         sock_set_flag(sk, SOCK_ZAPPED);
1058
1059         sock_init_data(sock, sk);
1060         sock->ops = &qrtr_proto_ops;
1061
1062         ipc = qrtr_sk(sk);
1063         ipc->us.sq_family = AF_QIPCRTR;
1064         ipc->us.sq_node = qrtr_local_nid;
1065         ipc->us.sq_port = 0;
1066
1067         return 0;
1068 }
1069
1070 static const struct nla_policy qrtr_policy[IFA_MAX + 1] = {
1071         [IFA_LOCAL] = { .type = NLA_U32 },
1072 };
1073
1074 static int qrtr_addr_doit(struct sk_buff *skb, struct nlmsghdr *nlh,
1075                           struct netlink_ext_ack *extack)
1076 {
1077         struct nlattr *tb[IFA_MAX + 1];
1078         struct ifaddrmsg *ifm;
1079         int rc;
1080
1081         if (!netlink_capable(skb, CAP_NET_ADMIN))
1082                 return -EPERM;
1083
1084         if (!netlink_capable(skb, CAP_SYS_ADMIN))
1085                 return -EPERM;
1086
1087         ASSERT_RTNL();
1088
1089         rc = nlmsg_parse_deprecated(nlh, sizeof(*ifm), tb, IFA_MAX,
1090                                     qrtr_policy, extack);
1091         if (rc < 0)
1092                 return rc;
1093
1094         ifm = nlmsg_data(nlh);
1095         if (!tb[IFA_LOCAL])
1096                 return -EINVAL;
1097
1098         qrtr_local_nid = nla_get_u32(tb[IFA_LOCAL]);
1099         return 0;
1100 }
1101
1102 static const struct net_proto_family qrtr_family = {
1103         .owner  = THIS_MODULE,
1104         .family = AF_QIPCRTR,
1105         .create = qrtr_create,
1106 };
1107
1108 static int __init qrtr_proto_init(void)
1109 {
1110         int rc;
1111
1112         rc = proto_register(&qrtr_proto, 1);
1113         if (rc)
1114                 return rc;
1115
1116         rc = sock_register(&qrtr_family);
1117         if (rc) {
1118                 proto_unregister(&qrtr_proto);
1119                 return rc;
1120         }
1121
1122         rc = rtnl_register_module(THIS_MODULE, PF_QIPCRTR, RTM_NEWADDR, qrtr_addr_doit, NULL, 0);
1123         if (rc) {
1124                 sock_unregister(qrtr_family.family);
1125                 proto_unregister(&qrtr_proto);
1126         }
1127
1128         return rc;
1129 }
1130 postcore_initcall(qrtr_proto_init);
1131
1132 static void __exit qrtr_proto_fini(void)
1133 {
1134         rtnl_unregister(PF_QIPCRTR, RTM_NEWADDR);
1135         sock_unregister(qrtr_family.family);
1136         proto_unregister(&qrtr_proto);
1137 }
1138 module_exit(qrtr_proto_fini);
1139
1140 MODULE_DESCRIPTION("Qualcomm IPC-router driver");
1141 MODULE_LICENSE("GPL v2");
1142 MODULE_ALIAS_NETPROTO(PF_QIPCRTR);