Loading...
1// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2/*
3 * Copyright (c) 2015, Sony Mobile Communications Inc.
4 * Copyright (c) 2013, The Linux Foundation. All rights reserved.
5 * Copyright (c) 2020, Linaro Ltd.
6 */
7
8#include <linux/module.h>
9#include <linux/qrtr.h>
10#include <linux/workqueue.h>
11#include <net/sock.h>
12
13#include "qrtr.h"
14
15#define CREATE_TRACE_POINTS
16#include <trace/events/qrtr.h>
17
18static RADIX_TREE(nodes, GFP_KERNEL);
19
20static struct {
21 struct socket *sock;
22 struct sockaddr_qrtr bcast_sq;
23 struct list_head lookups;
24 struct workqueue_struct *workqueue;
25 struct work_struct work;
26 int local_node;
27} qrtr_ns;
28
29static const char * const qrtr_ctrl_pkt_strings[] = {
30 [QRTR_TYPE_HELLO] = "hello",
31 [QRTR_TYPE_BYE] = "bye",
32 [QRTR_TYPE_NEW_SERVER] = "new-server",
33 [QRTR_TYPE_DEL_SERVER] = "del-server",
34 [QRTR_TYPE_DEL_CLIENT] = "del-client",
35 [QRTR_TYPE_RESUME_TX] = "resume-tx",
36 [QRTR_TYPE_EXIT] = "exit",
37 [QRTR_TYPE_PING] = "ping",
38 [QRTR_TYPE_NEW_LOOKUP] = "new-lookup",
39 [QRTR_TYPE_DEL_LOOKUP] = "del-lookup",
40};
41
42struct qrtr_server_filter {
43 unsigned int service;
44 unsigned int instance;
45 unsigned int ifilter;
46};
47
48struct qrtr_lookup {
49 unsigned int service;
50 unsigned int instance;
51
52 struct sockaddr_qrtr sq;
53 struct list_head li;
54};
55
56struct qrtr_server {
57 unsigned int service;
58 unsigned int instance;
59
60 unsigned int node;
61 unsigned int port;
62
63 struct list_head qli;
64};
65
66struct qrtr_node {
67 unsigned int id;
68 struct radix_tree_root servers;
69};
70
71static struct qrtr_node *node_get(unsigned int node_id)
72{
73 struct qrtr_node *node;
74
75 node = radix_tree_lookup(&nodes, node_id);
76 if (node)
77 return node;
78
79 /* If node didn't exist, allocate and insert it to the tree */
80 node = kzalloc(sizeof(*node), GFP_KERNEL);
81 if (!node)
82 return NULL;
83
84 node->id = node_id;
85
86 if (radix_tree_insert(&nodes, node_id, node)) {
87 kfree(node);
88 return NULL;
89 }
90
91 return node;
92}
93
94static int server_match(const struct qrtr_server *srv,
95 const struct qrtr_server_filter *f)
96{
97 unsigned int ifilter = f->ifilter;
98
99 if (f->service != 0 && srv->service != f->service)
100 return 0;
101 if (!ifilter && f->instance)
102 ifilter = ~0;
103
104 return (srv->instance & ifilter) == f->instance;
105}
106
107static int service_announce_new(struct sockaddr_qrtr *dest,
108 struct qrtr_server *srv)
109{
110 struct qrtr_ctrl_pkt pkt;
111 struct msghdr msg = { };
112 struct kvec iv;
113
114 trace_qrtr_ns_service_announce_new(srv->service, srv->instance,
115 srv->node, srv->port);
116
117 iv.iov_base = &pkt;
118 iv.iov_len = sizeof(pkt);
119
120 memset(&pkt, 0, sizeof(pkt));
121 pkt.cmd = cpu_to_le32(QRTR_TYPE_NEW_SERVER);
122 pkt.server.service = cpu_to_le32(srv->service);
123 pkt.server.instance = cpu_to_le32(srv->instance);
124 pkt.server.node = cpu_to_le32(srv->node);
125 pkt.server.port = cpu_to_le32(srv->port);
126
127 msg.msg_name = (struct sockaddr *)dest;
128 msg.msg_namelen = sizeof(*dest);
129
130 return kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
131}
132
133static int service_announce_del(struct sockaddr_qrtr *dest,
134 struct qrtr_server *srv)
135{
136 struct qrtr_ctrl_pkt pkt;
137 struct msghdr msg = { };
138 struct kvec iv;
139 int ret;
140
141 trace_qrtr_ns_service_announce_del(srv->service, srv->instance,
142 srv->node, srv->port);
143
144 iv.iov_base = &pkt;
145 iv.iov_len = sizeof(pkt);
146
147 memset(&pkt, 0, sizeof(pkt));
148 pkt.cmd = cpu_to_le32(QRTR_TYPE_DEL_SERVER);
149 pkt.server.service = cpu_to_le32(srv->service);
150 pkt.server.instance = cpu_to_le32(srv->instance);
151 pkt.server.node = cpu_to_le32(srv->node);
152 pkt.server.port = cpu_to_le32(srv->port);
153
154 msg.msg_name = (struct sockaddr *)dest;
155 msg.msg_namelen = sizeof(*dest);
156
157 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
158 if (ret < 0)
159 pr_err("failed to announce del service\n");
160
161 return ret;
162}
163
164static void lookup_notify(struct sockaddr_qrtr *to, struct qrtr_server *srv,
165 bool new)
166{
167 struct qrtr_ctrl_pkt pkt;
168 struct msghdr msg = { };
169 struct kvec iv;
170 int ret;
171
172 iv.iov_base = &pkt;
173 iv.iov_len = sizeof(pkt);
174
175 memset(&pkt, 0, sizeof(pkt));
176 pkt.cmd = new ? cpu_to_le32(QRTR_TYPE_NEW_SERVER) :
177 cpu_to_le32(QRTR_TYPE_DEL_SERVER);
178 if (srv) {
179 pkt.server.service = cpu_to_le32(srv->service);
180 pkt.server.instance = cpu_to_le32(srv->instance);
181 pkt.server.node = cpu_to_le32(srv->node);
182 pkt.server.port = cpu_to_le32(srv->port);
183 }
184
185 msg.msg_name = (struct sockaddr *)to;
186 msg.msg_namelen = sizeof(*to);
187
188 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
189 if (ret < 0)
190 pr_err("failed to send lookup notification\n");
191}
192
193static int announce_servers(struct sockaddr_qrtr *sq)
194{
195 struct radix_tree_iter iter;
196 struct qrtr_server *srv;
197 struct qrtr_node *node;
198 void __rcu **slot;
199 int ret;
200
201 node = node_get(qrtr_ns.local_node);
202 if (!node)
203 return 0;
204
205 rcu_read_lock();
206 /* Announce the list of servers registered in this node */
207 radix_tree_for_each_slot(slot, &node->servers, &iter, 0) {
208 srv = radix_tree_deref_slot(slot);
209 if (!srv)
210 continue;
211 if (radix_tree_deref_retry(srv)) {
212 slot = radix_tree_iter_retry(&iter);
213 continue;
214 }
215 slot = radix_tree_iter_resume(slot, &iter);
216 rcu_read_unlock();
217
218 ret = service_announce_new(sq, srv);
219 if (ret < 0) {
220 pr_err("failed to announce new service\n");
221 return ret;
222 }
223
224 rcu_read_lock();
225 }
226
227 rcu_read_unlock();
228
229 return 0;
230}
231
232static struct qrtr_server *server_add(unsigned int service,
233 unsigned int instance,
234 unsigned int node_id,
235 unsigned int port)
236{
237 struct qrtr_server *srv;
238 struct qrtr_server *old;
239 struct qrtr_node *node;
240
241 if (!service || !port)
242 return NULL;
243
244 srv = kzalloc(sizeof(*srv), GFP_KERNEL);
245 if (!srv)
246 return NULL;
247
248 srv->service = service;
249 srv->instance = instance;
250 srv->node = node_id;
251 srv->port = port;
252
253 node = node_get(node_id);
254 if (!node)
255 goto err;
256
257 /* Delete the old server on the same port */
258 old = radix_tree_lookup(&node->servers, port);
259 if (old) {
260 radix_tree_delete(&node->servers, port);
261 kfree(old);
262 }
263
264 radix_tree_insert(&node->servers, port, srv);
265
266 trace_qrtr_ns_server_add(srv->service, srv->instance,
267 srv->node, srv->port);
268
269 return srv;
270
271err:
272 kfree(srv);
273 return NULL;
274}
275
276static int server_del(struct qrtr_node *node, unsigned int port)
277{
278 struct qrtr_lookup *lookup;
279 struct qrtr_server *srv;
280 struct list_head *li;
281
282 srv = radix_tree_lookup(&node->servers, port);
283 if (!srv)
284 return -ENOENT;
285
286 radix_tree_delete(&node->servers, port);
287
288 /* Broadcast the removal of local servers */
289 if (srv->node == qrtr_ns.local_node)
290 service_announce_del(&qrtr_ns.bcast_sq, srv);
291
292 /* Announce the service's disappearance to observers */
293 list_for_each(li, &qrtr_ns.lookups) {
294 lookup = container_of(li, struct qrtr_lookup, li);
295 if (lookup->service && lookup->service != srv->service)
296 continue;
297 if (lookup->instance && lookup->instance != srv->instance)
298 continue;
299
300 lookup_notify(&lookup->sq, srv, false);
301 }
302
303 kfree(srv);
304
305 return 0;
306}
307
308static int say_hello(struct sockaddr_qrtr *dest)
309{
310 struct qrtr_ctrl_pkt pkt;
311 struct msghdr msg = { };
312 struct kvec iv;
313 int ret;
314
315 iv.iov_base = &pkt;
316 iv.iov_len = sizeof(pkt);
317
318 memset(&pkt, 0, sizeof(pkt));
319 pkt.cmd = cpu_to_le32(QRTR_TYPE_HELLO);
320
321 msg.msg_name = (struct sockaddr *)dest;
322 msg.msg_namelen = sizeof(*dest);
323
324 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
325 if (ret < 0)
326 pr_err("failed to send hello msg\n");
327
328 return ret;
329}
330
331/* Announce the list of servers registered on the local node */
332static int ctrl_cmd_hello(struct sockaddr_qrtr *sq)
333{
334 int ret;
335
336 ret = say_hello(sq);
337 if (ret < 0)
338 return ret;
339
340 return announce_servers(sq);
341}
342
343static int ctrl_cmd_bye(struct sockaddr_qrtr *from)
344{
345 struct qrtr_node *local_node;
346 struct radix_tree_iter iter;
347 struct qrtr_ctrl_pkt pkt;
348 struct qrtr_server *srv;
349 struct sockaddr_qrtr sq;
350 struct msghdr msg = { };
351 struct qrtr_node *node;
352 void __rcu **slot;
353 struct kvec iv;
354 int ret;
355
356 iv.iov_base = &pkt;
357 iv.iov_len = sizeof(pkt);
358
359 node = node_get(from->sq_node);
360 if (!node)
361 return 0;
362
363 rcu_read_lock();
364 /* Advertise removal of this client to all servers of remote node */
365 radix_tree_for_each_slot(slot, &node->servers, &iter, 0) {
366 srv = radix_tree_deref_slot(slot);
367 if (!srv)
368 continue;
369 if (radix_tree_deref_retry(srv)) {
370 slot = radix_tree_iter_retry(&iter);
371 continue;
372 }
373 slot = radix_tree_iter_resume(slot, &iter);
374 rcu_read_unlock();
375 server_del(node, srv->port);
376 rcu_read_lock();
377 }
378 rcu_read_unlock();
379
380 /* Advertise the removal of this client to all local servers */
381 local_node = node_get(qrtr_ns.local_node);
382 if (!local_node)
383 return 0;
384
385 memset(&pkt, 0, sizeof(pkt));
386 pkt.cmd = cpu_to_le32(QRTR_TYPE_BYE);
387 pkt.client.node = cpu_to_le32(from->sq_node);
388
389 rcu_read_lock();
390 radix_tree_for_each_slot(slot, &local_node->servers, &iter, 0) {
391 srv = radix_tree_deref_slot(slot);
392 if (!srv)
393 continue;
394 if (radix_tree_deref_retry(srv)) {
395 slot = radix_tree_iter_retry(&iter);
396 continue;
397 }
398 slot = radix_tree_iter_resume(slot, &iter);
399 rcu_read_unlock();
400
401 sq.sq_family = AF_QIPCRTR;
402 sq.sq_node = srv->node;
403 sq.sq_port = srv->port;
404
405 msg.msg_name = (struct sockaddr *)&sq;
406 msg.msg_namelen = sizeof(sq);
407
408 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
409 if (ret < 0) {
410 pr_err("failed to send bye cmd\n");
411 return ret;
412 }
413 rcu_read_lock();
414 }
415
416 rcu_read_unlock();
417
418 return 0;
419}
420
421static int ctrl_cmd_del_client(struct sockaddr_qrtr *from,
422 unsigned int node_id, unsigned int port)
423{
424 struct qrtr_node *local_node;
425 struct radix_tree_iter iter;
426 struct qrtr_lookup *lookup;
427 struct qrtr_ctrl_pkt pkt;
428 struct msghdr msg = { };
429 struct qrtr_server *srv;
430 struct sockaddr_qrtr sq;
431 struct qrtr_node *node;
432 struct list_head *tmp;
433 struct list_head *li;
434 void __rcu **slot;
435 struct kvec iv;
436 int ret;
437
438 iv.iov_base = &pkt;
439 iv.iov_len = sizeof(pkt);
440
441 /* Don't accept spoofed messages */
442 if (from->sq_node != node_id)
443 return -EINVAL;
444
445 /* Local DEL_CLIENT messages comes from the port being closed */
446 if (from->sq_node == qrtr_ns.local_node && from->sq_port != port)
447 return -EINVAL;
448
449 /* Remove any lookups by this client */
450 list_for_each_safe(li, tmp, &qrtr_ns.lookups) {
451 lookup = container_of(li, struct qrtr_lookup, li);
452 if (lookup->sq.sq_node != node_id)
453 continue;
454 if (lookup->sq.sq_port != port)
455 continue;
456
457 list_del(&lookup->li);
458 kfree(lookup);
459 }
460
461 /* Remove the server belonging to this port */
462 node = node_get(node_id);
463 if (node)
464 server_del(node, port);
465
466 /* Advertise the removal of this client to all local servers */
467 local_node = node_get(qrtr_ns.local_node);
468 if (!local_node)
469 return 0;
470
471 memset(&pkt, 0, sizeof(pkt));
472 pkt.cmd = cpu_to_le32(QRTR_TYPE_DEL_CLIENT);
473 pkt.client.node = cpu_to_le32(node_id);
474 pkt.client.port = cpu_to_le32(port);
475
476 rcu_read_lock();
477 radix_tree_for_each_slot(slot, &local_node->servers, &iter, 0) {
478 srv = radix_tree_deref_slot(slot);
479 if (!srv)
480 continue;
481 if (radix_tree_deref_retry(srv)) {
482 slot = radix_tree_iter_retry(&iter);
483 continue;
484 }
485 slot = radix_tree_iter_resume(slot, &iter);
486 rcu_read_unlock();
487
488 sq.sq_family = AF_QIPCRTR;
489 sq.sq_node = srv->node;
490 sq.sq_port = srv->port;
491
492 msg.msg_name = (struct sockaddr *)&sq;
493 msg.msg_namelen = sizeof(sq);
494
495 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
496 if (ret < 0) {
497 pr_err("failed to send del client cmd\n");
498 return ret;
499 }
500 rcu_read_lock();
501 }
502
503 rcu_read_unlock();
504
505 return 0;
506}
507
508static int ctrl_cmd_new_server(struct sockaddr_qrtr *from,
509 unsigned int service, unsigned int instance,
510 unsigned int node_id, unsigned int port)
511{
512 struct qrtr_lookup *lookup;
513 struct qrtr_server *srv;
514 struct list_head *li;
515 int ret = 0;
516
517 /* Ignore specified node and port for local servers */
518 if (from->sq_node == qrtr_ns.local_node) {
519 node_id = from->sq_node;
520 port = from->sq_port;
521 }
522
523 srv = server_add(service, instance, node_id, port);
524 if (!srv)
525 return -EINVAL;
526
527 if (srv->node == qrtr_ns.local_node) {
528 ret = service_announce_new(&qrtr_ns.bcast_sq, srv);
529 if (ret < 0) {
530 pr_err("failed to announce new service\n");
531 return ret;
532 }
533 }
534
535 /* Notify any potential lookups about the new server */
536 list_for_each(li, &qrtr_ns.lookups) {
537 lookup = container_of(li, struct qrtr_lookup, li);
538 if (lookup->service && lookup->service != service)
539 continue;
540 if (lookup->instance && lookup->instance != instance)
541 continue;
542
543 lookup_notify(&lookup->sq, srv, true);
544 }
545
546 return ret;
547}
548
549static int ctrl_cmd_del_server(struct sockaddr_qrtr *from,
550 unsigned int service, unsigned int instance,
551 unsigned int node_id, unsigned int port)
552{
553 struct qrtr_node *node;
554
555 /* Ignore specified node and port for local servers*/
556 if (from->sq_node == qrtr_ns.local_node) {
557 node_id = from->sq_node;
558 port = from->sq_port;
559 }
560
561 /* Local servers may only unregister themselves */
562 if (from->sq_node == qrtr_ns.local_node && from->sq_port != port)
563 return -EINVAL;
564
565 node = node_get(node_id);
566 if (!node)
567 return -ENOENT;
568
569 return server_del(node, port);
570}
571
572static int ctrl_cmd_new_lookup(struct sockaddr_qrtr *from,
573 unsigned int service, unsigned int instance)
574{
575 struct radix_tree_iter node_iter;
576 struct qrtr_server_filter filter;
577 struct radix_tree_iter srv_iter;
578 struct qrtr_lookup *lookup;
579 struct qrtr_node *node;
580 void __rcu **node_slot;
581 void __rcu **srv_slot;
582
583 /* Accept only local observers */
584 if (from->sq_node != qrtr_ns.local_node)
585 return -EINVAL;
586
587 lookup = kzalloc(sizeof(*lookup), GFP_KERNEL);
588 if (!lookup)
589 return -ENOMEM;
590
591 lookup->sq = *from;
592 lookup->service = service;
593 lookup->instance = instance;
594 list_add_tail(&lookup->li, &qrtr_ns.lookups);
595
596 memset(&filter, 0, sizeof(filter));
597 filter.service = service;
598 filter.instance = instance;
599
600 rcu_read_lock();
601 radix_tree_for_each_slot(node_slot, &nodes, &node_iter, 0) {
602 node = radix_tree_deref_slot(node_slot);
603 if (!node)
604 continue;
605 if (radix_tree_deref_retry(node)) {
606 node_slot = radix_tree_iter_retry(&node_iter);
607 continue;
608 }
609 node_slot = radix_tree_iter_resume(node_slot, &node_iter);
610
611 radix_tree_for_each_slot(srv_slot, &node->servers,
612 &srv_iter, 0) {
613 struct qrtr_server *srv;
614
615 srv = radix_tree_deref_slot(srv_slot);
616 if (!srv)
617 continue;
618 if (radix_tree_deref_retry(srv)) {
619 srv_slot = radix_tree_iter_retry(&srv_iter);
620 continue;
621 }
622
623 if (!server_match(srv, &filter))
624 continue;
625
626 srv_slot = radix_tree_iter_resume(srv_slot, &srv_iter);
627
628 rcu_read_unlock();
629 lookup_notify(from, srv, true);
630 rcu_read_lock();
631 }
632 }
633 rcu_read_unlock();
634
635 /* Empty notification, to indicate end of listing */
636 lookup_notify(from, NULL, true);
637
638 return 0;
639}
640
641static void ctrl_cmd_del_lookup(struct sockaddr_qrtr *from,
642 unsigned int service, unsigned int instance)
643{
644 struct qrtr_lookup *lookup;
645 struct list_head *tmp;
646 struct list_head *li;
647
648 list_for_each_safe(li, tmp, &qrtr_ns.lookups) {
649 lookup = container_of(li, struct qrtr_lookup, li);
650 if (lookup->sq.sq_node != from->sq_node)
651 continue;
652 if (lookup->sq.sq_port != from->sq_port)
653 continue;
654 if (lookup->service != service)
655 continue;
656 if (lookup->instance && lookup->instance != instance)
657 continue;
658
659 list_del(&lookup->li);
660 kfree(lookup);
661 }
662}
663
664static void qrtr_ns_worker(struct work_struct *work)
665{
666 const struct qrtr_ctrl_pkt *pkt;
667 size_t recv_buf_size = 4096;
668 struct sockaddr_qrtr sq;
669 struct msghdr msg = { };
670 unsigned int cmd;
671 ssize_t msglen;
672 void *recv_buf;
673 struct kvec iv;
674 int ret;
675
676 msg.msg_name = (struct sockaddr *)&sq;
677 msg.msg_namelen = sizeof(sq);
678
679 recv_buf = kzalloc(recv_buf_size, GFP_KERNEL);
680 if (!recv_buf)
681 return;
682
683 for (;;) {
684 iv.iov_base = recv_buf;
685 iv.iov_len = recv_buf_size;
686
687 msglen = kernel_recvmsg(qrtr_ns.sock, &msg, &iv, 1,
688 iv.iov_len, MSG_DONTWAIT);
689
690 if (msglen == -EAGAIN)
691 break;
692
693 if (msglen < 0) {
694 pr_err("error receiving packet: %zd\n", msglen);
695 break;
696 }
697
698 pkt = recv_buf;
699 cmd = le32_to_cpu(pkt->cmd);
700 if (cmd < ARRAY_SIZE(qrtr_ctrl_pkt_strings) &&
701 qrtr_ctrl_pkt_strings[cmd])
702 trace_qrtr_ns_message(qrtr_ctrl_pkt_strings[cmd],
703 sq.sq_node, sq.sq_port);
704
705 ret = 0;
706 switch (cmd) {
707 case QRTR_TYPE_HELLO:
708 ret = ctrl_cmd_hello(&sq);
709 break;
710 case QRTR_TYPE_BYE:
711 ret = ctrl_cmd_bye(&sq);
712 break;
713 case QRTR_TYPE_DEL_CLIENT:
714 ret = ctrl_cmd_del_client(&sq,
715 le32_to_cpu(pkt->client.node),
716 le32_to_cpu(pkt->client.port));
717 break;
718 case QRTR_TYPE_NEW_SERVER:
719 ret = ctrl_cmd_new_server(&sq,
720 le32_to_cpu(pkt->server.service),
721 le32_to_cpu(pkt->server.instance),
722 le32_to_cpu(pkt->server.node),
723 le32_to_cpu(pkt->server.port));
724 break;
725 case QRTR_TYPE_DEL_SERVER:
726 ret = ctrl_cmd_del_server(&sq,
727 le32_to_cpu(pkt->server.service),
728 le32_to_cpu(pkt->server.instance),
729 le32_to_cpu(pkt->server.node),
730 le32_to_cpu(pkt->server.port));
731 break;
732 case QRTR_TYPE_EXIT:
733 case QRTR_TYPE_PING:
734 case QRTR_TYPE_RESUME_TX:
735 break;
736 case QRTR_TYPE_NEW_LOOKUP:
737 ret = ctrl_cmd_new_lookup(&sq,
738 le32_to_cpu(pkt->server.service),
739 le32_to_cpu(pkt->server.instance));
740 break;
741 case QRTR_TYPE_DEL_LOOKUP:
742 ctrl_cmd_del_lookup(&sq,
743 le32_to_cpu(pkt->server.service),
744 le32_to_cpu(pkt->server.instance));
745 break;
746 }
747
748 if (ret < 0)
749 pr_err("failed while handling packet from %d:%d",
750 sq.sq_node, sq.sq_port);
751 }
752
753 kfree(recv_buf);
754}
755
756static void qrtr_ns_data_ready(struct sock *sk)
757{
758 queue_work(qrtr_ns.workqueue, &qrtr_ns.work);
759}
760
761int qrtr_ns_init(void)
762{
763 struct sockaddr_qrtr sq;
764 int ret;
765
766 INIT_LIST_HEAD(&qrtr_ns.lookups);
767 INIT_WORK(&qrtr_ns.work, qrtr_ns_worker);
768
769 ret = sock_create_kern(&init_net, AF_QIPCRTR, SOCK_DGRAM,
770 PF_QIPCRTR, &qrtr_ns.sock);
771 if (ret < 0)
772 return ret;
773
774 ret = kernel_getsockname(qrtr_ns.sock, (struct sockaddr *)&sq);
775 if (ret < 0) {
776 pr_err("failed to get socket name\n");
777 goto err_sock;
778 }
779
780 qrtr_ns.workqueue = alloc_workqueue("qrtr_ns_handler", WQ_UNBOUND, 1);
781 if (!qrtr_ns.workqueue) {
782 ret = -ENOMEM;
783 goto err_sock;
784 }
785
786 qrtr_ns.sock->sk->sk_data_ready = qrtr_ns_data_ready;
787
788 sq.sq_port = QRTR_PORT_CTRL;
789 qrtr_ns.local_node = sq.sq_node;
790
791 ret = kernel_bind(qrtr_ns.sock, (struct sockaddr *)&sq, sizeof(sq));
792 if (ret < 0) {
793 pr_err("failed to bind to socket\n");
794 goto err_wq;
795 }
796
797 qrtr_ns.bcast_sq.sq_family = AF_QIPCRTR;
798 qrtr_ns.bcast_sq.sq_node = QRTR_NODE_BCAST;
799 qrtr_ns.bcast_sq.sq_port = QRTR_PORT_CTRL;
800
801 ret = say_hello(&qrtr_ns.bcast_sq);
802 if (ret < 0)
803 goto err_wq;
804
805 return 0;
806
807err_wq:
808 destroy_workqueue(qrtr_ns.workqueue);
809err_sock:
810 sock_release(qrtr_ns.sock);
811 return ret;
812}
813EXPORT_SYMBOL_GPL(qrtr_ns_init);
814
815void qrtr_ns_remove(void)
816{
817 cancel_work_sync(&qrtr_ns.work);
818 destroy_workqueue(qrtr_ns.workqueue);
819 sock_release(qrtr_ns.sock);
820}
821EXPORT_SYMBOL_GPL(qrtr_ns_remove);
822
823MODULE_AUTHOR("Manivannan Sadhasivam <manivannan.sadhasivam@linaro.org>");
824MODULE_DESCRIPTION("Qualcomm IPC Router Nameservice");
825MODULE_LICENSE("Dual BSD/GPL");
1// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2/*
3 * Copyright (c) 2015, Sony Mobile Communications Inc.
4 * Copyright (c) 2013, The Linux Foundation. All rights reserved.
5 * Copyright (c) 2020, Linaro Ltd.
6 */
7
8#include <linux/module.h>
9#include <linux/qrtr.h>
10#include <linux/workqueue.h>
11#include <net/sock.h>
12
13#include "qrtr.h"
14
15#include <trace/events/sock.h>
16#define CREATE_TRACE_POINTS
17#include <trace/events/qrtr.h>
18
19static DEFINE_XARRAY(nodes);
20
21static struct {
22 struct socket *sock;
23 struct sockaddr_qrtr bcast_sq;
24 struct list_head lookups;
25 struct workqueue_struct *workqueue;
26 struct work_struct work;
27 int local_node;
28} qrtr_ns;
29
30static const char * const qrtr_ctrl_pkt_strings[] = {
31 [QRTR_TYPE_HELLO] = "hello",
32 [QRTR_TYPE_BYE] = "bye",
33 [QRTR_TYPE_NEW_SERVER] = "new-server",
34 [QRTR_TYPE_DEL_SERVER] = "del-server",
35 [QRTR_TYPE_DEL_CLIENT] = "del-client",
36 [QRTR_TYPE_RESUME_TX] = "resume-tx",
37 [QRTR_TYPE_EXIT] = "exit",
38 [QRTR_TYPE_PING] = "ping",
39 [QRTR_TYPE_NEW_LOOKUP] = "new-lookup",
40 [QRTR_TYPE_DEL_LOOKUP] = "del-lookup",
41};
42
43struct qrtr_server_filter {
44 unsigned int service;
45 unsigned int instance;
46 unsigned int ifilter;
47};
48
49struct qrtr_lookup {
50 unsigned int service;
51 unsigned int instance;
52
53 struct sockaddr_qrtr sq;
54 struct list_head li;
55};
56
57struct qrtr_server {
58 unsigned int service;
59 unsigned int instance;
60
61 unsigned int node;
62 unsigned int port;
63
64 struct list_head qli;
65};
66
67struct qrtr_node {
68 unsigned int id;
69 struct xarray servers;
70};
71
72static struct qrtr_node *node_get(unsigned int node_id)
73{
74 struct qrtr_node *node;
75
76 node = xa_load(&nodes, node_id);
77 if (node)
78 return node;
79
80 /* If node didn't exist, allocate and insert it to the tree */
81 node = kzalloc(sizeof(*node), GFP_KERNEL);
82 if (!node)
83 return NULL;
84
85 node->id = node_id;
86 xa_init(&node->servers);
87
88 if (xa_store(&nodes, node_id, node, GFP_KERNEL)) {
89 kfree(node);
90 return NULL;
91 }
92
93 return node;
94}
95
96static int server_match(const struct qrtr_server *srv,
97 const struct qrtr_server_filter *f)
98{
99 unsigned int ifilter = f->ifilter;
100
101 if (f->service != 0 && srv->service != f->service)
102 return 0;
103 if (!ifilter && f->instance)
104 ifilter = ~0;
105
106 return (srv->instance & ifilter) == f->instance;
107}
108
109static int service_announce_new(struct sockaddr_qrtr *dest,
110 struct qrtr_server *srv)
111{
112 struct qrtr_ctrl_pkt pkt;
113 struct msghdr msg = { };
114 struct kvec iv;
115
116 trace_qrtr_ns_service_announce_new(srv->service, srv->instance,
117 srv->node, srv->port);
118
119 iv.iov_base = &pkt;
120 iv.iov_len = sizeof(pkt);
121
122 memset(&pkt, 0, sizeof(pkt));
123 pkt.cmd = cpu_to_le32(QRTR_TYPE_NEW_SERVER);
124 pkt.server.service = cpu_to_le32(srv->service);
125 pkt.server.instance = cpu_to_le32(srv->instance);
126 pkt.server.node = cpu_to_le32(srv->node);
127 pkt.server.port = cpu_to_le32(srv->port);
128
129 msg.msg_name = (struct sockaddr *)dest;
130 msg.msg_namelen = sizeof(*dest);
131
132 return kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
133}
134
135static void service_announce_del(struct sockaddr_qrtr *dest,
136 struct qrtr_server *srv)
137{
138 struct qrtr_ctrl_pkt pkt;
139 struct msghdr msg = { };
140 struct kvec iv;
141 int ret;
142
143 trace_qrtr_ns_service_announce_del(srv->service, srv->instance,
144 srv->node, srv->port);
145
146 iv.iov_base = &pkt;
147 iv.iov_len = sizeof(pkt);
148
149 memset(&pkt, 0, sizeof(pkt));
150 pkt.cmd = cpu_to_le32(QRTR_TYPE_DEL_SERVER);
151 pkt.server.service = cpu_to_le32(srv->service);
152 pkt.server.instance = cpu_to_le32(srv->instance);
153 pkt.server.node = cpu_to_le32(srv->node);
154 pkt.server.port = cpu_to_le32(srv->port);
155
156 msg.msg_name = (struct sockaddr *)dest;
157 msg.msg_namelen = sizeof(*dest);
158
159 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
160 if (ret < 0 && ret != -ENODEV)
161 pr_err("failed to announce del service\n");
162
163 return;
164}
165
166static void lookup_notify(struct sockaddr_qrtr *to, struct qrtr_server *srv,
167 bool new)
168{
169 struct qrtr_ctrl_pkt pkt;
170 struct msghdr msg = { };
171 struct kvec iv;
172 int ret;
173
174 iv.iov_base = &pkt;
175 iv.iov_len = sizeof(pkt);
176
177 memset(&pkt, 0, sizeof(pkt));
178 pkt.cmd = new ? cpu_to_le32(QRTR_TYPE_NEW_SERVER) :
179 cpu_to_le32(QRTR_TYPE_DEL_SERVER);
180 if (srv) {
181 pkt.server.service = cpu_to_le32(srv->service);
182 pkt.server.instance = cpu_to_le32(srv->instance);
183 pkt.server.node = cpu_to_le32(srv->node);
184 pkt.server.port = cpu_to_le32(srv->port);
185 }
186
187 msg.msg_name = (struct sockaddr *)to;
188 msg.msg_namelen = sizeof(*to);
189
190 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
191 if (ret < 0 && ret != -ENODEV)
192 pr_err("failed to send lookup notification\n");
193}
194
195static int announce_servers(struct sockaddr_qrtr *sq)
196{
197 struct qrtr_server *srv;
198 struct qrtr_node *node;
199 unsigned long index;
200 int ret;
201
202 node = node_get(qrtr_ns.local_node);
203 if (!node)
204 return 0;
205
206 /* Announce the list of servers registered in this node */
207 xa_for_each(&node->servers, index, srv) {
208 ret = service_announce_new(sq, srv);
209 if (ret < 0) {
210 if (ret == -ENODEV)
211 continue;
212
213 pr_err("failed to announce new service\n");
214 return ret;
215 }
216 }
217 return 0;
218}
219
220static struct qrtr_server *server_add(unsigned int service,
221 unsigned int instance,
222 unsigned int node_id,
223 unsigned int port)
224{
225 struct qrtr_server *srv;
226 struct qrtr_server *old;
227 struct qrtr_node *node;
228
229 if (!service || !port)
230 return NULL;
231
232 srv = kzalloc(sizeof(*srv), GFP_KERNEL);
233 if (!srv)
234 return NULL;
235
236 srv->service = service;
237 srv->instance = instance;
238 srv->node = node_id;
239 srv->port = port;
240
241 node = node_get(node_id);
242 if (!node)
243 goto err;
244
245 /* Delete the old server on the same port */
246 old = xa_store(&node->servers, port, srv, GFP_KERNEL);
247 if (old) {
248 if (xa_is_err(old)) {
249 pr_err("failed to add server [0x%x:0x%x] ret:%d\n",
250 srv->service, srv->instance, xa_err(old));
251 goto err;
252 } else {
253 kfree(old);
254 }
255 }
256
257 trace_qrtr_ns_server_add(srv->service, srv->instance,
258 srv->node, srv->port);
259
260 return srv;
261
262err:
263 kfree(srv);
264 return NULL;
265}
266
267static int server_del(struct qrtr_node *node, unsigned int port, bool bcast)
268{
269 struct qrtr_lookup *lookup;
270 struct qrtr_server *srv;
271 struct list_head *li;
272
273 srv = xa_load(&node->servers, port);
274 if (!srv)
275 return -ENOENT;
276
277 xa_erase(&node->servers, port);
278
279 /* Broadcast the removal of local servers */
280 if (srv->node == qrtr_ns.local_node && bcast)
281 service_announce_del(&qrtr_ns.bcast_sq, srv);
282
283 /* Announce the service's disappearance to observers */
284 list_for_each(li, &qrtr_ns.lookups) {
285 lookup = container_of(li, struct qrtr_lookup, li);
286 if (lookup->service && lookup->service != srv->service)
287 continue;
288 if (lookup->instance && lookup->instance != srv->instance)
289 continue;
290
291 lookup_notify(&lookup->sq, srv, false);
292 }
293
294 kfree(srv);
295
296 return 0;
297}
298
299static int say_hello(struct sockaddr_qrtr *dest)
300{
301 struct qrtr_ctrl_pkt pkt;
302 struct msghdr msg = { };
303 struct kvec iv;
304 int ret;
305
306 iv.iov_base = &pkt;
307 iv.iov_len = sizeof(pkt);
308
309 memset(&pkt, 0, sizeof(pkt));
310 pkt.cmd = cpu_to_le32(QRTR_TYPE_HELLO);
311
312 msg.msg_name = (struct sockaddr *)dest;
313 msg.msg_namelen = sizeof(*dest);
314
315 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
316 if (ret < 0)
317 pr_err("failed to send hello msg\n");
318
319 return ret;
320}
321
322/* Announce the list of servers registered on the local node */
323static int ctrl_cmd_hello(struct sockaddr_qrtr *sq)
324{
325 int ret;
326
327 ret = say_hello(sq);
328 if (ret < 0)
329 return ret;
330
331 return announce_servers(sq);
332}
333
334static int ctrl_cmd_bye(struct sockaddr_qrtr *from)
335{
336 struct qrtr_node *local_node;
337 struct qrtr_ctrl_pkt pkt;
338 struct qrtr_server *srv;
339 struct sockaddr_qrtr sq;
340 struct msghdr msg = { };
341 struct qrtr_node *node;
342 unsigned long index;
343 struct kvec iv;
344 int ret;
345
346 iv.iov_base = &pkt;
347 iv.iov_len = sizeof(pkt);
348
349 node = node_get(from->sq_node);
350 if (!node)
351 return 0;
352
353 /* Advertise removal of this client to all servers of remote node */
354 xa_for_each(&node->servers, index, srv)
355 server_del(node, srv->port, true);
356
357 /* Advertise the removal of this client to all local servers */
358 local_node = node_get(qrtr_ns.local_node);
359 if (!local_node)
360 return 0;
361
362 memset(&pkt, 0, sizeof(pkt));
363 pkt.cmd = cpu_to_le32(QRTR_TYPE_BYE);
364 pkt.client.node = cpu_to_le32(from->sq_node);
365
366 xa_for_each(&local_node->servers, index, srv) {
367 sq.sq_family = AF_QIPCRTR;
368 sq.sq_node = srv->node;
369 sq.sq_port = srv->port;
370
371 msg.msg_name = (struct sockaddr *)&sq;
372 msg.msg_namelen = sizeof(sq);
373
374 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
375 if (ret < 0 && ret != -ENODEV) {
376 pr_err("failed to send bye cmd\n");
377 return ret;
378 }
379 }
380 return 0;
381}
382
383static int ctrl_cmd_del_client(struct sockaddr_qrtr *from,
384 unsigned int node_id, unsigned int port)
385{
386 struct qrtr_node *local_node;
387 struct qrtr_lookup *lookup;
388 struct qrtr_ctrl_pkt pkt;
389 struct msghdr msg = { };
390 struct qrtr_server *srv;
391 struct sockaddr_qrtr sq;
392 struct qrtr_node *node;
393 struct list_head *tmp;
394 struct list_head *li;
395 unsigned long index;
396 struct kvec iv;
397 int ret;
398
399 iv.iov_base = &pkt;
400 iv.iov_len = sizeof(pkt);
401
402 /* Don't accept spoofed messages */
403 if (from->sq_node != node_id)
404 return -EINVAL;
405
406 /* Local DEL_CLIENT messages comes from the port being closed */
407 if (from->sq_node == qrtr_ns.local_node && from->sq_port != port)
408 return -EINVAL;
409
410 /* Remove any lookups by this client */
411 list_for_each_safe(li, tmp, &qrtr_ns.lookups) {
412 lookup = container_of(li, struct qrtr_lookup, li);
413 if (lookup->sq.sq_node != node_id)
414 continue;
415 if (lookup->sq.sq_port != port)
416 continue;
417
418 list_del(&lookup->li);
419 kfree(lookup);
420 }
421
422 /* Remove the server belonging to this port but don't broadcast
423 * DEL_SERVER. Neighbours would've already removed the server belonging
424 * to this port due to the DEL_CLIENT broadcast from qrtr_port_remove().
425 */
426 node = node_get(node_id);
427 if (node)
428 server_del(node, port, false);
429
430 /* Advertise the removal of this client to all local servers */
431 local_node = node_get(qrtr_ns.local_node);
432 if (!local_node)
433 return 0;
434
435 memset(&pkt, 0, sizeof(pkt));
436 pkt.cmd = cpu_to_le32(QRTR_TYPE_DEL_CLIENT);
437 pkt.client.node = cpu_to_le32(node_id);
438 pkt.client.port = cpu_to_le32(port);
439
440 xa_for_each(&local_node->servers, index, srv) {
441 sq.sq_family = AF_QIPCRTR;
442 sq.sq_node = srv->node;
443 sq.sq_port = srv->port;
444
445 msg.msg_name = (struct sockaddr *)&sq;
446 msg.msg_namelen = sizeof(sq);
447
448 ret = kernel_sendmsg(qrtr_ns.sock, &msg, &iv, 1, sizeof(pkt));
449 if (ret < 0 && ret != -ENODEV) {
450 pr_err("failed to send del client cmd\n");
451 return ret;
452 }
453 }
454 return 0;
455}
456
457static int ctrl_cmd_new_server(struct sockaddr_qrtr *from,
458 unsigned int service, unsigned int instance,
459 unsigned int node_id, unsigned int port)
460{
461 struct qrtr_lookup *lookup;
462 struct qrtr_server *srv;
463 struct list_head *li;
464 int ret = 0;
465
466 /* Ignore specified node and port for local servers */
467 if (from->sq_node == qrtr_ns.local_node) {
468 node_id = from->sq_node;
469 port = from->sq_port;
470 }
471
472 srv = server_add(service, instance, node_id, port);
473 if (!srv)
474 return -EINVAL;
475
476 if (srv->node == qrtr_ns.local_node) {
477 ret = service_announce_new(&qrtr_ns.bcast_sq, srv);
478 if (ret < 0) {
479 pr_err("failed to announce new service\n");
480 return ret;
481 }
482 }
483
484 /* Notify any potential lookups about the new server */
485 list_for_each(li, &qrtr_ns.lookups) {
486 lookup = container_of(li, struct qrtr_lookup, li);
487 if (lookup->service && lookup->service != service)
488 continue;
489 if (lookup->instance && lookup->instance != instance)
490 continue;
491
492 lookup_notify(&lookup->sq, srv, true);
493 }
494
495 return ret;
496}
497
498static int ctrl_cmd_del_server(struct sockaddr_qrtr *from,
499 unsigned int service, unsigned int instance,
500 unsigned int node_id, unsigned int port)
501{
502 struct qrtr_node *node;
503
504 /* Ignore specified node and port for local servers*/
505 if (from->sq_node == qrtr_ns.local_node) {
506 node_id = from->sq_node;
507 port = from->sq_port;
508 }
509
510 /* Local servers may only unregister themselves */
511 if (from->sq_node == qrtr_ns.local_node && from->sq_port != port)
512 return -EINVAL;
513
514 node = node_get(node_id);
515 if (!node)
516 return -ENOENT;
517
518 server_del(node, port, true);
519
520 return 0;
521}
522
523static int ctrl_cmd_new_lookup(struct sockaddr_qrtr *from,
524 unsigned int service, unsigned int instance)
525{
526 struct qrtr_server_filter filter;
527 struct qrtr_lookup *lookup;
528 struct qrtr_server *srv;
529 struct qrtr_node *node;
530 unsigned long node_idx;
531 unsigned long srv_idx;
532
533 /* Accept only local observers */
534 if (from->sq_node != qrtr_ns.local_node)
535 return -EINVAL;
536
537 lookup = kzalloc(sizeof(*lookup), GFP_KERNEL);
538 if (!lookup)
539 return -ENOMEM;
540
541 lookup->sq = *from;
542 lookup->service = service;
543 lookup->instance = instance;
544 list_add_tail(&lookup->li, &qrtr_ns.lookups);
545
546 memset(&filter, 0, sizeof(filter));
547 filter.service = service;
548 filter.instance = instance;
549
550 xa_for_each(&nodes, node_idx, node) {
551 xa_for_each(&node->servers, srv_idx, srv) {
552 if (!server_match(srv, &filter))
553 continue;
554
555 lookup_notify(from, srv, true);
556 }
557 }
558
559 /* Empty notification, to indicate end of listing */
560 lookup_notify(from, NULL, true);
561
562 return 0;
563}
564
565static void ctrl_cmd_del_lookup(struct sockaddr_qrtr *from,
566 unsigned int service, unsigned int instance)
567{
568 struct qrtr_lookup *lookup;
569 struct list_head *tmp;
570 struct list_head *li;
571
572 list_for_each_safe(li, tmp, &qrtr_ns.lookups) {
573 lookup = container_of(li, struct qrtr_lookup, li);
574 if (lookup->sq.sq_node != from->sq_node)
575 continue;
576 if (lookup->sq.sq_port != from->sq_port)
577 continue;
578 if (lookup->service != service)
579 continue;
580 if (lookup->instance && lookup->instance != instance)
581 continue;
582
583 list_del(&lookup->li);
584 kfree(lookup);
585 }
586}
587
588static void qrtr_ns_worker(struct work_struct *work)
589{
590 const struct qrtr_ctrl_pkt *pkt;
591 size_t recv_buf_size = 4096;
592 struct sockaddr_qrtr sq;
593 struct msghdr msg = { };
594 unsigned int cmd;
595 ssize_t msglen;
596 void *recv_buf;
597 struct kvec iv;
598 int ret;
599
600 msg.msg_name = (struct sockaddr *)&sq;
601 msg.msg_namelen = sizeof(sq);
602
603 recv_buf = kzalloc(recv_buf_size, GFP_KERNEL);
604 if (!recv_buf)
605 return;
606
607 for (;;) {
608 iv.iov_base = recv_buf;
609 iv.iov_len = recv_buf_size;
610
611 msglen = kernel_recvmsg(qrtr_ns.sock, &msg, &iv, 1,
612 iv.iov_len, MSG_DONTWAIT);
613
614 if (msglen == -EAGAIN)
615 break;
616
617 if (msglen < 0) {
618 pr_err("error receiving packet: %zd\n", msglen);
619 break;
620 }
621
622 pkt = recv_buf;
623 cmd = le32_to_cpu(pkt->cmd);
624 if (cmd < ARRAY_SIZE(qrtr_ctrl_pkt_strings) &&
625 qrtr_ctrl_pkt_strings[cmd])
626 trace_qrtr_ns_message(qrtr_ctrl_pkt_strings[cmd],
627 sq.sq_node, sq.sq_port);
628
629 ret = 0;
630 switch (cmd) {
631 case QRTR_TYPE_HELLO:
632 ret = ctrl_cmd_hello(&sq);
633 break;
634 case QRTR_TYPE_BYE:
635 ret = ctrl_cmd_bye(&sq);
636 break;
637 case QRTR_TYPE_DEL_CLIENT:
638 ret = ctrl_cmd_del_client(&sq,
639 le32_to_cpu(pkt->client.node),
640 le32_to_cpu(pkt->client.port));
641 break;
642 case QRTR_TYPE_NEW_SERVER:
643 ret = ctrl_cmd_new_server(&sq,
644 le32_to_cpu(pkt->server.service),
645 le32_to_cpu(pkt->server.instance),
646 le32_to_cpu(pkt->server.node),
647 le32_to_cpu(pkt->server.port));
648 break;
649 case QRTR_TYPE_DEL_SERVER:
650 ret = ctrl_cmd_del_server(&sq,
651 le32_to_cpu(pkt->server.service),
652 le32_to_cpu(pkt->server.instance),
653 le32_to_cpu(pkt->server.node),
654 le32_to_cpu(pkt->server.port));
655 break;
656 case QRTR_TYPE_EXIT:
657 case QRTR_TYPE_PING:
658 case QRTR_TYPE_RESUME_TX:
659 break;
660 case QRTR_TYPE_NEW_LOOKUP:
661 ret = ctrl_cmd_new_lookup(&sq,
662 le32_to_cpu(pkt->server.service),
663 le32_to_cpu(pkt->server.instance));
664 break;
665 case QRTR_TYPE_DEL_LOOKUP:
666 ctrl_cmd_del_lookup(&sq,
667 le32_to_cpu(pkt->server.service),
668 le32_to_cpu(pkt->server.instance));
669 break;
670 }
671
672 if (ret < 0)
673 pr_err("failed while handling packet from %d:%d",
674 sq.sq_node, sq.sq_port);
675 }
676
677 kfree(recv_buf);
678}
679
680static void qrtr_ns_data_ready(struct sock *sk)
681{
682 trace_sk_data_ready(sk);
683
684 queue_work(qrtr_ns.workqueue, &qrtr_ns.work);
685}
686
687int qrtr_ns_init(void)
688{
689 struct sockaddr_qrtr sq;
690 int ret;
691
692 INIT_LIST_HEAD(&qrtr_ns.lookups);
693 INIT_WORK(&qrtr_ns.work, qrtr_ns_worker);
694
695 ret = sock_create_kern(&init_net, AF_QIPCRTR, SOCK_DGRAM,
696 PF_QIPCRTR, &qrtr_ns.sock);
697 if (ret < 0)
698 return ret;
699
700 ret = kernel_getsockname(qrtr_ns.sock, (struct sockaddr *)&sq);
701 if (ret < 0) {
702 pr_err("failed to get socket name\n");
703 goto err_sock;
704 }
705
706 qrtr_ns.workqueue = alloc_ordered_workqueue("qrtr_ns_handler", 0);
707 if (!qrtr_ns.workqueue) {
708 ret = -ENOMEM;
709 goto err_sock;
710 }
711
712 qrtr_ns.sock->sk->sk_data_ready = qrtr_ns_data_ready;
713
714 sq.sq_port = QRTR_PORT_CTRL;
715 qrtr_ns.local_node = sq.sq_node;
716
717 ret = kernel_bind(qrtr_ns.sock, (struct sockaddr *)&sq, sizeof(sq));
718 if (ret < 0) {
719 pr_err("failed to bind to socket\n");
720 goto err_wq;
721 }
722
723 qrtr_ns.bcast_sq.sq_family = AF_QIPCRTR;
724 qrtr_ns.bcast_sq.sq_node = QRTR_NODE_BCAST;
725 qrtr_ns.bcast_sq.sq_port = QRTR_PORT_CTRL;
726
727 ret = say_hello(&qrtr_ns.bcast_sq);
728 if (ret < 0)
729 goto err_wq;
730
731 /* As the qrtr ns socket owner and creator is the same module, we have
732 * to decrease the qrtr module reference count to guarantee that it
733 * remains zero after the ns socket is created, otherwise, executing
734 * "rmmod" command is unable to make the qrtr module deleted after the
735 * qrtr module is inserted successfully.
736 *
737 * However, the reference count is increased twice in
738 * sock_create_kern(): one is to increase the reference count of owner
739 * of qrtr socket's proto_ops struct; another is to increment the
740 * reference count of owner of qrtr proto struct. Therefore, we must
741 * decrement the module reference count twice to ensure that it keeps
742 * zero after server's listening socket is created. Of course, we
743 * must bump the module reference count twice as well before the socket
744 * is closed.
745 */
746 module_put(qrtr_ns.sock->ops->owner);
747 module_put(qrtr_ns.sock->sk->sk_prot_creator->owner);
748
749 return 0;
750
751err_wq:
752 destroy_workqueue(qrtr_ns.workqueue);
753err_sock:
754 sock_release(qrtr_ns.sock);
755 return ret;
756}
757EXPORT_SYMBOL_GPL(qrtr_ns_init);
758
759void qrtr_ns_remove(void)
760{
761 cancel_work_sync(&qrtr_ns.work);
762 destroy_workqueue(qrtr_ns.workqueue);
763
764 /* sock_release() expects the two references that were put during
765 * qrtr_ns_init(). This function is only called during module remove,
766 * so try_stop_module() has already set the refcnt to 0. Use
767 * __module_get() instead of try_module_get() to successfully take two
768 * references.
769 */
770 __module_get(qrtr_ns.sock->ops->owner);
771 __module_get(qrtr_ns.sock->sk->sk_prot_creator->owner);
772 sock_release(qrtr_ns.sock);
773}
774EXPORT_SYMBOL_GPL(qrtr_ns_remove);
775
776MODULE_AUTHOR("Manivannan Sadhasivam <manivannan.sadhasivam@linaro.org>");
777MODULE_DESCRIPTION("Qualcomm IPC Router Nameservice");
778MODULE_LICENSE("Dual BSD/GPL");