qrtr.c 30 KB

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