Merge tag 'for_linus' of git://git.kernel.org/pub/scm/linux/kernel/git/mst/vhost
[linux-2.6-microblaze.git] / net / tipc / name_table.c
1 /*
2  * net/tipc/name_table.c: TIPC name table code
3  *
4  * Copyright (c) 2000-2006, 2014-2015, Ericsson AB
5  * Copyright (c) 2004-2008, 2010-2014, Wind River Systems
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions are met:
10  *
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. Neither the names of the copyright holders nor the names of its
17  *    contributors may be used to endorse or promote products derived from
18  *    this software without specific prior written permission.
19  *
20  * Alternatively, this software may be distributed under the terms of the
21  * GNU General Public License ("GPL") version 2 as published by the Free
22  * Software Foundation.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
25  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
28  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
29  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
30  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
31  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
32  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
33  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
34  * POSSIBILITY OF SUCH DAMAGE.
35  */
36
37 #include <net/sock.h>
38 #include "core.h"
39 #include "netlink.h"
40 #include "name_table.h"
41 #include "name_distr.h"
42 #include "subscr.h"
43 #include "bcast.h"
44 #include "addr.h"
45 #include "node.h"
46 #include "group.h"
47 #include <net/genetlink.h>
48
49 #define TIPC_NAMETBL_SIZE 1024          /* must be a power of 2 */
50
51 /**
52  * struct name_info - name sequence publication info
53  * @node_list: circular list of publications made by own node
54  * @cluster_list: circular list of publications made by own cluster
55  * @zone_list: circular list of publications made by own zone
56  * @node_list_size: number of entries in "node_list"
57  * @cluster_list_size: number of entries in "cluster_list"
58  * @zone_list_size: number of entries in "zone_list"
59  *
60  * Note: The zone list always contains at least one entry, since all
61  *       publications of the associated name sequence belong to it.
62  *       (The cluster and node lists may be empty.)
63  */
64 struct name_info {
65         struct list_head node_list;
66         struct list_head cluster_list;
67         struct list_head zone_list;
68         u32 node_list_size;
69         u32 cluster_list_size;
70         u32 zone_list_size;
71 };
72
73 /**
74  * struct sub_seq - container for all published instances of a name sequence
75  * @lower: name sequence lower bound
76  * @upper: name sequence upper bound
77  * @info: pointer to name sequence publication info
78  */
79 struct sub_seq {
80         u32 lower;
81         u32 upper;
82         struct name_info *info;
83 };
84
85 /**
86  * struct name_seq - container for all published instances of a name type
87  * @type: 32 bit 'type' value for name sequence
88  * @sseq: pointer to dynamically-sized array of sub-sequences of this 'type';
89  *        sub-sequences are sorted in ascending order
90  * @alloc: number of sub-sequences currently in array
91  * @first_free: array index of first unused sub-sequence entry
92  * @ns_list: links to adjacent name sequences in hash chain
93  * @subscriptions: list of subscriptions for this 'type'
94  * @lock: spinlock controlling access to publication lists of all sub-sequences
95  * @rcu: RCU callback head used for deferred freeing
96  */
97 struct name_seq {
98         u32 type;
99         struct sub_seq *sseqs;
100         u32 alloc;
101         u32 first_free;
102         struct hlist_node ns_list;
103         struct list_head subscriptions;
104         spinlock_t lock;
105         struct rcu_head rcu;
106 };
107
108 static int hash(int x)
109 {
110         return x & (TIPC_NAMETBL_SIZE - 1);
111 }
112
113 /**
114  * publ_create - create a publication structure
115  */
116 static struct publication *publ_create(u32 type, u32 lower, u32 upper,
117                                        u32 scope, u32 node, u32 port_ref,
118                                        u32 key)
119 {
120         struct publication *publ = kzalloc(sizeof(*publ), GFP_ATOMIC);
121         if (publ == NULL) {
122                 pr_warn("Publication creation failure, no memory\n");
123                 return NULL;
124         }
125
126         publ->type = type;
127         publ->lower = lower;
128         publ->upper = upper;
129         publ->scope = scope;
130         publ->node = node;
131         publ->ref = port_ref;
132         publ->key = key;
133         INIT_LIST_HEAD(&publ->pport_list);
134         return publ;
135 }
136
137 /**
138  * tipc_subseq_alloc - allocate a specified number of sub-sequence structures
139  */
140 static struct sub_seq *tipc_subseq_alloc(u32 cnt)
141 {
142         return kcalloc(cnt, sizeof(struct sub_seq), GFP_ATOMIC);
143 }
144
145 /**
146  * tipc_nameseq_create - create a name sequence structure for the specified 'type'
147  *
148  * Allocates a single sub-sequence structure and sets it to all 0's.
149  */
150 static struct name_seq *tipc_nameseq_create(u32 type, struct hlist_head *seq_head)
151 {
152         struct name_seq *nseq = kzalloc(sizeof(*nseq), GFP_ATOMIC);
153         struct sub_seq *sseq = tipc_subseq_alloc(1);
154
155         if (!nseq || !sseq) {
156                 pr_warn("Name sequence creation failed, no memory\n");
157                 kfree(nseq);
158                 kfree(sseq);
159                 return NULL;
160         }
161
162         spin_lock_init(&nseq->lock);
163         nseq->type = type;
164         nseq->sseqs = sseq;
165         nseq->alloc = 1;
166         INIT_HLIST_NODE(&nseq->ns_list);
167         INIT_LIST_HEAD(&nseq->subscriptions);
168         hlist_add_head_rcu(&nseq->ns_list, seq_head);
169         return nseq;
170 }
171
172 /**
173  * nameseq_find_subseq - find sub-sequence (if any) matching a name instance
174  *
175  * Very time-critical, so binary searches through sub-sequence array.
176  */
177 static struct sub_seq *nameseq_find_subseq(struct name_seq *nseq,
178                                            u32 instance)
179 {
180         struct sub_seq *sseqs = nseq->sseqs;
181         int low = 0;
182         int high = nseq->first_free - 1;
183         int mid;
184
185         while (low <= high) {
186                 mid = (low + high) / 2;
187                 if (instance < sseqs[mid].lower)
188                         high = mid - 1;
189                 else if (instance > sseqs[mid].upper)
190                         low = mid + 1;
191                 else
192                         return &sseqs[mid];
193         }
194         return NULL;
195 }
196
197 /**
198  * nameseq_locate_subseq - determine position of name instance in sub-sequence
199  *
200  * Returns index in sub-sequence array of the entry that contains the specified
201  * instance value; if no entry contains that value, returns the position
202  * where a new entry for it would be inserted in the array.
203  *
204  * Note: Similar to binary search code for locating a sub-sequence.
205  */
206 static u32 nameseq_locate_subseq(struct name_seq *nseq, u32 instance)
207 {
208         struct sub_seq *sseqs = nseq->sseqs;
209         int low = 0;
210         int high = nseq->first_free - 1;
211         int mid;
212
213         while (low <= high) {
214                 mid = (low + high) / 2;
215                 if (instance < sseqs[mid].lower)
216                         high = mid - 1;
217                 else if (instance > sseqs[mid].upper)
218                         low = mid + 1;
219                 else
220                         return mid;
221         }
222         return low;
223 }
224
225 /**
226  * tipc_nameseq_insert_publ
227  */
228 static struct publication *tipc_nameseq_insert_publ(struct net *net,
229                                                     struct name_seq *nseq,
230                                                     u32 type, u32 lower,
231                                                     u32 upper, u32 scope,
232                                                     u32 node, u32 port, u32 key)
233 {
234         struct tipc_subscription *s;
235         struct tipc_subscription *st;
236         struct publication *publ;
237         struct sub_seq *sseq;
238         struct name_info *info;
239         int created_subseq = 0;
240
241         sseq = nameseq_find_subseq(nseq, lower);
242         if (sseq) {
243
244                 /* Lower end overlaps existing entry => need an exact match */
245                 if ((sseq->lower != lower) || (sseq->upper != upper)) {
246                         return NULL;
247                 }
248
249                 info = sseq->info;
250
251                 /* Check if an identical publication already exists */
252                 list_for_each_entry(publ, &info->zone_list, zone_list) {
253                         if ((publ->ref == port) && (publ->key == key) &&
254                             (!publ->node || (publ->node == node)))
255                                 return NULL;
256                 }
257         } else {
258                 u32 inspos;
259                 struct sub_seq *freesseq;
260
261                 /* Find where lower end should be inserted */
262                 inspos = nameseq_locate_subseq(nseq, lower);
263
264                 /* Fail if upper end overlaps into an existing entry */
265                 if ((inspos < nseq->first_free) &&
266                     (upper >= nseq->sseqs[inspos].lower)) {
267                         return NULL;
268                 }
269
270                 /* Ensure there is space for new sub-sequence */
271                 if (nseq->first_free == nseq->alloc) {
272                         struct sub_seq *sseqs = tipc_subseq_alloc(nseq->alloc * 2);
273
274                         if (!sseqs) {
275                                 pr_warn("Cannot publish {%u,%u,%u}, no memory\n",
276                                         type, lower, upper);
277                                 return NULL;
278                         }
279                         memcpy(sseqs, nseq->sseqs,
280                                nseq->alloc * sizeof(struct sub_seq));
281                         kfree(nseq->sseqs);
282                         nseq->sseqs = sseqs;
283                         nseq->alloc *= 2;
284                 }
285
286                 info = kzalloc(sizeof(*info), GFP_ATOMIC);
287                 if (!info) {
288                         pr_warn("Cannot publish {%u,%u,%u}, no memory\n",
289                                 type, lower, upper);
290                         return NULL;
291                 }
292
293                 INIT_LIST_HEAD(&info->node_list);
294                 INIT_LIST_HEAD(&info->cluster_list);
295                 INIT_LIST_HEAD(&info->zone_list);
296
297                 /* Insert new sub-sequence */
298                 sseq = &nseq->sseqs[inspos];
299                 freesseq = &nseq->sseqs[nseq->first_free];
300                 memmove(sseq + 1, sseq, (freesseq - sseq) * sizeof(*sseq));
301                 memset(sseq, 0, sizeof(*sseq));
302                 nseq->first_free++;
303                 sseq->lower = lower;
304                 sseq->upper = upper;
305                 sseq->info = info;
306                 created_subseq = 1;
307         }
308
309         /* Insert a publication */
310         publ = publ_create(type, lower, upper, scope, node, port, key);
311         if (!publ)
312                 return NULL;
313
314         list_add(&publ->zone_list, &info->zone_list);
315         info->zone_list_size++;
316
317         if (in_own_cluster(net, node)) {
318                 list_add(&publ->cluster_list, &info->cluster_list);
319                 info->cluster_list_size++;
320         }
321
322         if (in_own_node(net, node)) {
323                 list_add(&publ->node_list, &info->node_list);
324                 info->node_list_size++;
325         }
326
327         /* Any subscriptions waiting for notification?  */
328         list_for_each_entry_safe(s, st, &nseq->subscriptions, nameseq_list) {
329                 tipc_subscrp_report_overlap(s, publ->lower, publ->upper,
330                                             TIPC_PUBLISHED, publ->ref,
331                                             publ->node, publ->scope,
332                                             created_subseq);
333         }
334         return publ;
335 }
336
337 /**
338  * tipc_nameseq_remove_publ
339  *
340  * NOTE: There may be cases where TIPC is asked to remove a publication
341  * that is not in the name table.  For example, if another node issues a
342  * publication for a name sequence that overlaps an existing name sequence
343  * the publication will not be recorded, which means the publication won't
344  * be found when the name sequence is later withdrawn by that node.
345  * A failed withdraw request simply returns a failure indication and lets the
346  * caller issue any error or warning messages associated with such a problem.
347  */
348 static struct publication *tipc_nameseq_remove_publ(struct net *net,
349                                                     struct name_seq *nseq,
350                                                     u32 inst, u32 node,
351                                                     u32 ref, u32 key)
352 {
353         struct publication *publ;
354         struct sub_seq *sseq = nameseq_find_subseq(nseq, inst);
355         struct name_info *info;
356         struct sub_seq *free;
357         struct tipc_subscription *s, *st;
358         int removed_subseq = 0;
359
360         if (!sseq)
361                 return NULL;
362
363         info = sseq->info;
364
365         /* Locate publication, if it exists */
366         list_for_each_entry(publ, &info->zone_list, zone_list) {
367                 if ((publ->key == key) && (publ->ref == ref) &&
368                     (!publ->node || (publ->node == node)))
369                         goto found;
370         }
371         return NULL;
372
373 found:
374         /* Remove publication from zone scope list */
375         list_del(&publ->zone_list);
376         info->zone_list_size--;
377
378         /* Remove publication from cluster scope list, if present */
379         if (in_own_cluster(net, node)) {
380                 list_del(&publ->cluster_list);
381                 info->cluster_list_size--;
382         }
383
384         /* Remove publication from node scope list, if present */
385         if (in_own_node(net, node)) {
386                 list_del(&publ->node_list);
387                 info->node_list_size--;
388         }
389
390         /* Contract subseq list if no more publications for that subseq */
391         if (list_empty(&info->zone_list)) {
392                 kfree(info);
393                 free = &nseq->sseqs[nseq->first_free--];
394                 memmove(sseq, sseq + 1, (free - (sseq + 1)) * sizeof(*sseq));
395                 removed_subseq = 1;
396         }
397
398         /* Notify any waiting subscriptions */
399         list_for_each_entry_safe(s, st, &nseq->subscriptions, nameseq_list) {
400                 tipc_subscrp_report_overlap(s, publ->lower, publ->upper,
401                                             TIPC_WITHDRAWN, publ->ref,
402                                             publ->node, publ->scope,
403                                             removed_subseq);
404         }
405
406         return publ;
407 }
408
409 /**
410  * tipc_nameseq_subscribe - attach a subscription, and optionally
411  * issue the prescribed number of events if there is any sub-
412  * sequence overlapping with the requested sequence
413  */
414 static void tipc_nameseq_subscribe(struct name_seq *nseq,
415                                    struct tipc_subscription *s,
416                                    bool status)
417 {
418         struct sub_seq *sseq = nseq->sseqs;
419         struct tipc_name_seq ns;
420
421         tipc_subscrp_convert_seq(&s->evt.s.seq, s->swap, &ns);
422
423         tipc_subscrp_get(s);
424         list_add(&s->nameseq_list, &nseq->subscriptions);
425
426         if (!status || !sseq)
427                 return;
428
429         while (sseq != &nseq->sseqs[nseq->first_free]) {
430                 if (tipc_subscrp_check_overlap(&ns, sseq->lower, sseq->upper)) {
431                         struct publication *crs;
432                         struct name_info *info = sseq->info;
433                         int must_report = 1;
434
435                         list_for_each_entry(crs, &info->zone_list, zone_list) {
436                                 tipc_subscrp_report_overlap(s, sseq->lower,
437                                                             sseq->upper,
438                                                             TIPC_PUBLISHED,
439                                                             crs->ref, crs->node,
440                                                             crs->scope,
441                                                             must_report);
442                                 must_report = 0;
443                         }
444                 }
445                 sseq++;
446         }
447 }
448
449 static struct name_seq *nametbl_find_seq(struct net *net, u32 type)
450 {
451         struct tipc_net *tn = net_generic(net, tipc_net_id);
452         struct hlist_head *seq_head;
453         struct name_seq *ns;
454
455         seq_head = &tn->nametbl->seq_hlist[hash(type)];
456         hlist_for_each_entry_rcu(ns, seq_head, ns_list) {
457                 if (ns->type == type)
458                         return ns;
459         }
460
461         return NULL;
462 };
463
464 struct publication *tipc_nametbl_insert_publ(struct net *net, u32 type,
465                                              u32 lower, u32 upper, u32 scope,
466                                              u32 node, u32 port, u32 key)
467 {
468         struct tipc_net *tn = net_generic(net, tipc_net_id);
469         struct publication *publ;
470         struct name_seq *seq = nametbl_find_seq(net, type);
471         int index = hash(type);
472
473         if ((scope < TIPC_ZONE_SCOPE) || (scope > TIPC_NODE_SCOPE) ||
474             (lower > upper)) {
475                 pr_debug("Failed to publish illegal {%u,%u,%u} with scope %u\n",
476                          type, lower, upper, scope);
477                 return NULL;
478         }
479
480         if (!seq)
481                 seq = tipc_nameseq_create(type, &tn->nametbl->seq_hlist[index]);
482         if (!seq)
483                 return NULL;
484
485         spin_lock_bh(&seq->lock);
486         publ = tipc_nameseq_insert_publ(net, seq, type, lower, upper,
487                                         scope, node, port, key);
488         spin_unlock_bh(&seq->lock);
489         return publ;
490 }
491
492 struct publication *tipc_nametbl_remove_publ(struct net *net, u32 type,
493                                              u32 lower, u32 node, u32 ref,
494                                              u32 key)
495 {
496         struct publication *publ;
497         struct name_seq *seq = nametbl_find_seq(net, type);
498
499         if (!seq)
500                 return NULL;
501
502         spin_lock_bh(&seq->lock);
503         publ = tipc_nameseq_remove_publ(net, seq, lower, node, ref, key);
504         if (!seq->first_free && list_empty(&seq->subscriptions)) {
505                 hlist_del_init_rcu(&seq->ns_list);
506                 kfree(seq->sseqs);
507                 spin_unlock_bh(&seq->lock);
508                 kfree_rcu(seq, rcu);
509                 return publ;
510         }
511         spin_unlock_bh(&seq->lock);
512         return publ;
513 }
514
515 /**
516  * tipc_nametbl_translate - perform name translation
517  *
518  * On entry, 'destnode' is the search domain used during translation.
519  *
520  * On exit:
521  * - if name translation is deferred to another node/cluster/zone,
522  *   leaves 'destnode' unchanged (will be non-zero) and returns 0
523  * - if name translation is attempted and succeeds, sets 'destnode'
524  *   to publishing node and returns port reference (will be non-zero)
525  * - if name translation is attempted and fails, sets 'destnode' to 0
526  *   and returns 0
527  */
528 u32 tipc_nametbl_translate(struct net *net, u32 type, u32 instance,
529                            u32 *destnode)
530 {
531         struct tipc_net *tn = net_generic(net, tipc_net_id);
532         struct sub_seq *sseq;
533         struct name_info *info;
534         struct publication *publ;
535         struct name_seq *seq;
536         u32 ref = 0;
537         u32 node = 0;
538
539         if (!tipc_in_scope(*destnode, tn->own_addr))
540                 return 0;
541
542         rcu_read_lock();
543         seq = nametbl_find_seq(net, type);
544         if (unlikely(!seq))
545                 goto not_found;
546         spin_lock_bh(&seq->lock);
547         sseq = nameseq_find_subseq(seq, instance);
548         if (unlikely(!sseq))
549                 goto no_match;
550         info = sseq->info;
551
552         /* Closest-First Algorithm */
553         if (likely(!*destnode)) {
554                 if (!list_empty(&info->node_list)) {
555                         publ = list_first_entry(&info->node_list,
556                                                 struct publication,
557                                                 node_list);
558                         list_move_tail(&publ->node_list,
559                                        &info->node_list);
560                 } else if (!list_empty(&info->cluster_list)) {
561                         publ = list_first_entry(&info->cluster_list,
562                                                 struct publication,
563                                                 cluster_list);
564                         list_move_tail(&publ->cluster_list,
565                                        &info->cluster_list);
566                 } else {
567                         publ = list_first_entry(&info->zone_list,
568                                                 struct publication,
569                                                 zone_list);
570                         list_move_tail(&publ->zone_list,
571                                        &info->zone_list);
572                 }
573         }
574
575         /* Round-Robin Algorithm */
576         else if (*destnode == tn->own_addr) {
577                 if (list_empty(&info->node_list))
578                         goto no_match;
579                 publ = list_first_entry(&info->node_list, struct publication,
580                                         node_list);
581                 list_move_tail(&publ->node_list, &info->node_list);
582         } else if (in_own_cluster_exact(net, *destnode)) {
583                 if (list_empty(&info->cluster_list))
584                         goto no_match;
585                 publ = list_first_entry(&info->cluster_list, struct publication,
586                                         cluster_list);
587                 list_move_tail(&publ->cluster_list, &info->cluster_list);
588         } else {
589                 publ = list_first_entry(&info->zone_list, struct publication,
590                                         zone_list);
591                 list_move_tail(&publ->zone_list, &info->zone_list);
592         }
593
594         ref = publ->ref;
595         node = publ->node;
596 no_match:
597         spin_unlock_bh(&seq->lock);
598 not_found:
599         rcu_read_unlock();
600         *destnode = node;
601         return ref;
602 }
603
604 bool tipc_nametbl_lookup(struct net *net, u32 type, u32 instance, u32 scope,
605                          struct list_head *dsts, int *dstcnt, u32 exclude,
606                          bool all)
607 {
608         u32 self = tipc_own_addr(net);
609         struct publication *publ;
610         struct name_info *info;
611         struct name_seq *seq;
612         struct sub_seq *sseq;
613
614         *dstcnt = 0;
615         rcu_read_lock();
616         seq = nametbl_find_seq(net, type);
617         if (unlikely(!seq))
618                 goto exit;
619         spin_lock_bh(&seq->lock);
620         sseq = nameseq_find_subseq(seq, instance);
621         if (likely(sseq)) {
622                 info = sseq->info;
623                 list_for_each_entry(publ, &info->zone_list, zone_list) {
624                         if (publ->scope != scope)
625                                 continue;
626                         if (publ->ref == exclude && publ->node == self)
627                                 continue;
628                         tipc_dest_push(dsts, publ->node, publ->ref);
629                         (*dstcnt)++;
630                         if (all)
631                                 continue;
632                         list_move_tail(&publ->zone_list, &info->zone_list);
633                         break;
634                 }
635         }
636         spin_unlock_bh(&seq->lock);
637 exit:
638         rcu_read_unlock();
639         return !list_empty(dsts);
640 }
641
642 int tipc_nametbl_mc_lookup(struct net *net, u32 type, u32 lower, u32 upper,
643                            u32 scope, bool exact, struct list_head *dports)
644 {
645         struct sub_seq *sseq_stop;
646         struct name_info *info;
647         struct publication *p;
648         struct name_seq *seq;
649         struct sub_seq *sseq;
650         int res = 0;
651
652         rcu_read_lock();
653         seq = nametbl_find_seq(net, type);
654         if (!seq)
655                 goto exit;
656
657         spin_lock_bh(&seq->lock);
658         sseq = seq->sseqs + nameseq_locate_subseq(seq, lower);
659         sseq_stop = seq->sseqs + seq->first_free;
660         for (; sseq != sseq_stop; sseq++) {
661                 if (sseq->lower > upper)
662                         break;
663                 info = sseq->info;
664                 list_for_each_entry(p, &info->node_list, node_list) {
665                         if (p->scope == scope || (!exact && p->scope < scope))
666                                 tipc_dest_push(dports, 0, p->ref);
667                 }
668
669                 if (info->cluster_list_size != info->node_list_size)
670                         res = 1;
671         }
672         spin_unlock_bh(&seq->lock);
673 exit:
674         rcu_read_unlock();
675         return res;
676 }
677
678 /* tipc_nametbl_lookup_dst_nodes - find broadcast destination nodes
679  * - Creates list of nodes that overlap the given multicast address
680  * - Determines if any node local ports overlap
681  */
682 void tipc_nametbl_lookup_dst_nodes(struct net *net, u32 type, u32 lower,
683                                    u32 upper, struct tipc_nlist *nodes)
684 {
685         struct sub_seq *sseq, *stop;
686         struct publication *publ;
687         struct name_info *info;
688         struct name_seq *seq;
689
690         rcu_read_lock();
691         seq = nametbl_find_seq(net, type);
692         if (!seq)
693                 goto exit;
694
695         spin_lock_bh(&seq->lock);
696         sseq = seq->sseqs + nameseq_locate_subseq(seq, lower);
697         stop = seq->sseqs + seq->first_free;
698         for (; sseq != stop && sseq->lower <= upper; sseq++) {
699                 info = sseq->info;
700                 list_for_each_entry(publ, &info->zone_list, zone_list) {
701                         tipc_nlist_add(nodes, publ->node);
702                 }
703         }
704         spin_unlock_bh(&seq->lock);
705 exit:
706         rcu_read_unlock();
707 }
708
709 /* tipc_nametbl_build_group - build list of communication group members
710  */
711 void tipc_nametbl_build_group(struct net *net, struct tipc_group *grp,
712                               u32 type, u32 scope)
713 {
714         struct sub_seq *sseq, *stop;
715         struct name_info *info;
716         struct publication *p;
717         struct name_seq *seq;
718
719         rcu_read_lock();
720         seq = nametbl_find_seq(net, type);
721         if (!seq)
722                 goto exit;
723
724         spin_lock_bh(&seq->lock);
725         sseq = seq->sseqs;
726         stop = seq->sseqs + seq->first_free;
727         for (; sseq != stop; sseq++) {
728                 info = sseq->info;
729                 list_for_each_entry(p, &info->zone_list, zone_list) {
730                         if (p->scope != scope)
731                                 continue;
732                         tipc_group_add_member(grp, p->node, p->ref, p->lower);
733                 }
734         }
735         spin_unlock_bh(&seq->lock);
736 exit:
737         rcu_read_unlock();
738 }
739
740 /*
741  * tipc_nametbl_publish - add name publication to network name tables
742  */
743 struct publication *tipc_nametbl_publish(struct net *net, u32 type, u32 lower,
744                                          u32 upper, u32 scope, u32 port_ref,
745                                          u32 key)
746 {
747         struct publication *publ;
748         struct sk_buff *buf = NULL;
749         struct tipc_net *tn = net_generic(net, tipc_net_id);
750
751         spin_lock_bh(&tn->nametbl_lock);
752         if (tn->nametbl->local_publ_count >= TIPC_MAX_PUBLICATIONS) {
753                 pr_warn("Publication failed, local publication limit reached (%u)\n",
754                         TIPC_MAX_PUBLICATIONS);
755                 spin_unlock_bh(&tn->nametbl_lock);
756                 return NULL;
757         }
758
759         publ = tipc_nametbl_insert_publ(net, type, lower, upper, scope,
760                                         tn->own_addr, port_ref, key);
761         if (likely(publ)) {
762                 tn->nametbl->local_publ_count++;
763                 buf = tipc_named_publish(net, publ);
764                 /* Any pending external events? */
765                 tipc_named_process_backlog(net);
766         }
767         spin_unlock_bh(&tn->nametbl_lock);
768
769         if (buf)
770                 tipc_node_broadcast(net, buf);
771         return publ;
772 }
773
774 /**
775  * tipc_nametbl_withdraw - withdraw name publication from network name tables
776  */
777 int tipc_nametbl_withdraw(struct net *net, u32 type, u32 lower, u32 ref,
778                           u32 key)
779 {
780         struct publication *publ;
781         struct sk_buff *skb = NULL;
782         struct tipc_net *tn = net_generic(net, tipc_net_id);
783
784         spin_lock_bh(&tn->nametbl_lock);
785         publ = tipc_nametbl_remove_publ(net, type, lower, tn->own_addr,
786                                         ref, key);
787         if (likely(publ)) {
788                 tn->nametbl->local_publ_count--;
789                 skb = tipc_named_withdraw(net, publ);
790                 /* Any pending external events? */
791                 tipc_named_process_backlog(net);
792                 list_del_init(&publ->pport_list);
793                 kfree_rcu(publ, rcu);
794         } else {
795                 pr_err("Unable to remove local publication\n"
796                        "(type=%u, lower=%u, ref=%u, key=%u)\n",
797                        type, lower, ref, key);
798         }
799         spin_unlock_bh(&tn->nametbl_lock);
800
801         if (skb) {
802                 tipc_node_broadcast(net, skb);
803                 return 1;
804         }
805         return 0;
806 }
807
808 /**
809  * tipc_nametbl_subscribe - add a subscription object to the name table
810  */
811 void tipc_nametbl_subscribe(struct tipc_subscription *s, bool status)
812 {
813         struct tipc_net *tn = net_generic(s->net, tipc_net_id);
814         u32 type = tipc_subscrp_convert_seq_type(s->evt.s.seq.type, s->swap);
815         int index = hash(type);
816         struct name_seq *seq;
817         struct tipc_name_seq ns;
818
819         spin_lock_bh(&tn->nametbl_lock);
820         seq = nametbl_find_seq(s->net, type);
821         if (!seq)
822                 seq = tipc_nameseq_create(type, &tn->nametbl->seq_hlist[index]);
823         if (seq) {
824                 spin_lock_bh(&seq->lock);
825                 tipc_nameseq_subscribe(seq, s, status);
826                 spin_unlock_bh(&seq->lock);
827         } else {
828                 tipc_subscrp_convert_seq(&s->evt.s.seq, s->swap, &ns);
829                 pr_warn("Failed to create subscription for {%u,%u,%u}\n",
830                         ns.type, ns.lower, ns.upper);
831         }
832         spin_unlock_bh(&tn->nametbl_lock);
833 }
834
835 /**
836  * tipc_nametbl_unsubscribe - remove a subscription object from name table
837  */
838 void tipc_nametbl_unsubscribe(struct tipc_subscription *s)
839 {
840         struct tipc_net *tn = net_generic(s->net, tipc_net_id);
841         struct name_seq *seq;
842         u32 type = tipc_subscrp_convert_seq_type(s->evt.s.seq.type, s->swap);
843
844         spin_lock_bh(&tn->nametbl_lock);
845         seq = nametbl_find_seq(s->net, type);
846         if (seq != NULL) {
847                 spin_lock_bh(&seq->lock);
848                 list_del_init(&s->nameseq_list);
849                 tipc_subscrp_put(s);
850                 if (!seq->first_free && list_empty(&seq->subscriptions)) {
851                         hlist_del_init_rcu(&seq->ns_list);
852                         kfree(seq->sseqs);
853                         spin_unlock_bh(&seq->lock);
854                         kfree_rcu(seq, rcu);
855                 } else {
856                         spin_unlock_bh(&seq->lock);
857                 }
858         }
859         spin_unlock_bh(&tn->nametbl_lock);
860 }
861
862 int tipc_nametbl_init(struct net *net)
863 {
864         struct tipc_net *tn = net_generic(net, tipc_net_id);
865         struct name_table *tipc_nametbl;
866         int i;
867
868         tipc_nametbl = kzalloc(sizeof(*tipc_nametbl), GFP_ATOMIC);
869         if (!tipc_nametbl)
870                 return -ENOMEM;
871
872         for (i = 0; i < TIPC_NAMETBL_SIZE; i++)
873                 INIT_HLIST_HEAD(&tipc_nametbl->seq_hlist[i]);
874
875         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_ZONE_SCOPE]);
876         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_CLUSTER_SCOPE]);
877         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_NODE_SCOPE]);
878         tn->nametbl = tipc_nametbl;
879         spin_lock_init(&tn->nametbl_lock);
880         return 0;
881 }
882
883 /**
884  * tipc_purge_publications - remove all publications for a given type
885  *
886  * tipc_nametbl_lock must be held when calling this function
887  */
888 static void tipc_purge_publications(struct net *net, struct name_seq *seq)
889 {
890         struct publication *publ, *safe;
891         struct sub_seq *sseq;
892         struct name_info *info;
893
894         spin_lock_bh(&seq->lock);
895         sseq = seq->sseqs;
896         info = sseq->info;
897         list_for_each_entry_safe(publ, safe, &info->zone_list, zone_list) {
898                 tipc_nameseq_remove_publ(net, seq, publ->lower, publ->node,
899                                          publ->ref, publ->key);
900                 kfree_rcu(publ, rcu);
901         }
902         hlist_del_init_rcu(&seq->ns_list);
903         kfree(seq->sseqs);
904         spin_unlock_bh(&seq->lock);
905
906         kfree_rcu(seq, rcu);
907 }
908
909 void tipc_nametbl_stop(struct net *net)
910 {
911         u32 i;
912         struct name_seq *seq;
913         struct hlist_head *seq_head;
914         struct tipc_net *tn = net_generic(net, tipc_net_id);
915         struct name_table *tipc_nametbl = tn->nametbl;
916
917         /* Verify name table is empty and purge any lingering
918          * publications, then release the name table
919          */
920         spin_lock_bh(&tn->nametbl_lock);
921         for (i = 0; i < TIPC_NAMETBL_SIZE; i++) {
922                 if (hlist_empty(&tipc_nametbl->seq_hlist[i]))
923                         continue;
924                 seq_head = &tipc_nametbl->seq_hlist[i];
925                 hlist_for_each_entry_rcu(seq, seq_head, ns_list) {
926                         tipc_purge_publications(net, seq);
927                 }
928         }
929         spin_unlock_bh(&tn->nametbl_lock);
930
931         synchronize_net();
932         kfree(tipc_nametbl);
933
934 }
935
936 static int __tipc_nl_add_nametable_publ(struct tipc_nl_msg *msg,
937                                         struct name_seq *seq,
938                                         struct sub_seq *sseq, u32 *last_publ)
939 {
940         void *hdr;
941         struct nlattr *attrs;
942         struct nlattr *publ;
943         struct publication *p;
944
945         if (*last_publ) {
946                 list_for_each_entry(p, &sseq->info->zone_list, zone_list)
947                         if (p->key == *last_publ)
948                                 break;
949                 if (p->key != *last_publ)
950                         return -EPIPE;
951         } else {
952                 p = list_first_entry(&sseq->info->zone_list, struct publication,
953                                      zone_list);
954         }
955
956         list_for_each_entry_from(p, &sseq->info->zone_list, zone_list) {
957                 *last_publ = p->key;
958
959                 hdr = genlmsg_put(msg->skb, msg->portid, msg->seq,
960                                   &tipc_genl_family, NLM_F_MULTI,
961                                   TIPC_NL_NAME_TABLE_GET);
962                 if (!hdr)
963                         return -EMSGSIZE;
964
965                 attrs = nla_nest_start(msg->skb, TIPC_NLA_NAME_TABLE);
966                 if (!attrs)
967                         goto msg_full;
968
969                 publ = nla_nest_start(msg->skb, TIPC_NLA_NAME_TABLE_PUBL);
970                 if (!publ)
971                         goto attr_msg_full;
972
973                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_TYPE, seq->type))
974                         goto publ_msg_full;
975                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_LOWER, sseq->lower))
976                         goto publ_msg_full;
977                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_UPPER, sseq->upper))
978                         goto publ_msg_full;
979                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_SCOPE, p->scope))
980                         goto publ_msg_full;
981                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_NODE, p->node))
982                         goto publ_msg_full;
983                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_REF, p->ref))
984                         goto publ_msg_full;
985                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_KEY, p->key))
986                         goto publ_msg_full;
987
988                 nla_nest_end(msg->skb, publ);
989                 nla_nest_end(msg->skb, attrs);
990                 genlmsg_end(msg->skb, hdr);
991         }
992         *last_publ = 0;
993
994         return 0;
995
996 publ_msg_full:
997         nla_nest_cancel(msg->skb, publ);
998 attr_msg_full:
999         nla_nest_cancel(msg->skb, attrs);
1000 msg_full:
1001         genlmsg_cancel(msg->skb, hdr);
1002
1003         return -EMSGSIZE;
1004 }
1005
1006 static int __tipc_nl_subseq_list(struct tipc_nl_msg *msg, struct name_seq *seq,
1007                                  u32 *last_lower, u32 *last_publ)
1008 {
1009         struct sub_seq *sseq;
1010         struct sub_seq *sseq_start;
1011         int err;
1012
1013         if (*last_lower) {
1014                 sseq_start = nameseq_find_subseq(seq, *last_lower);
1015                 if (!sseq_start)
1016                         return -EPIPE;
1017         } else {
1018                 sseq_start = seq->sseqs;
1019         }
1020
1021         for (sseq = sseq_start; sseq != &seq->sseqs[seq->first_free]; sseq++) {
1022                 err = __tipc_nl_add_nametable_publ(msg, seq, sseq, last_publ);
1023                 if (err) {
1024                         *last_lower = sseq->lower;
1025                         return err;
1026                 }
1027         }
1028         *last_lower = 0;
1029
1030         return 0;
1031 }
1032
1033 static int tipc_nl_seq_list(struct net *net, struct tipc_nl_msg *msg,
1034                             u32 *last_type, u32 *last_lower, u32 *last_publ)
1035 {
1036         struct tipc_net *tn = net_generic(net, tipc_net_id);
1037         struct hlist_head *seq_head;
1038         struct name_seq *seq = NULL;
1039         int err;
1040         int i;
1041
1042         if (*last_type)
1043                 i = hash(*last_type);
1044         else
1045                 i = 0;
1046
1047         for (; i < TIPC_NAMETBL_SIZE; i++) {
1048                 seq_head = &tn->nametbl->seq_hlist[i];
1049
1050                 if (*last_type) {
1051                         seq = nametbl_find_seq(net, *last_type);
1052                         if (!seq)
1053                                 return -EPIPE;
1054                 } else {
1055                         hlist_for_each_entry_rcu(seq, seq_head, ns_list)
1056                                 break;
1057                         if (!seq)
1058                                 continue;
1059                 }
1060
1061                 hlist_for_each_entry_from_rcu(seq, ns_list) {
1062                         spin_lock_bh(&seq->lock);
1063                         err = __tipc_nl_subseq_list(msg, seq, last_lower,
1064                                                     last_publ);
1065
1066                         if (err) {
1067                                 *last_type = seq->type;
1068                                 spin_unlock_bh(&seq->lock);
1069                                 return err;
1070                         }
1071                         spin_unlock_bh(&seq->lock);
1072                 }
1073                 *last_type = 0;
1074         }
1075         return 0;
1076 }
1077
1078 int tipc_nl_name_table_dump(struct sk_buff *skb, struct netlink_callback *cb)
1079 {
1080         int err;
1081         int done = cb->args[3];
1082         u32 last_type = cb->args[0];
1083         u32 last_lower = cb->args[1];
1084         u32 last_publ = cb->args[2];
1085         struct net *net = sock_net(skb->sk);
1086         struct tipc_nl_msg msg;
1087
1088         if (done)
1089                 return 0;
1090
1091         msg.skb = skb;
1092         msg.portid = NETLINK_CB(cb->skb).portid;
1093         msg.seq = cb->nlh->nlmsg_seq;
1094
1095         rcu_read_lock();
1096         err = tipc_nl_seq_list(net, &msg, &last_type, &last_lower, &last_publ);
1097         if (!err) {
1098                 done = 1;
1099         } else if (err != -EMSGSIZE) {
1100                 /* We never set seq or call nl_dump_check_consistent() this
1101                  * means that setting prev_seq here will cause the consistence
1102                  * check to fail in the netlink callback handler. Resulting in
1103                  * the NLMSG_DONE message having the NLM_F_DUMP_INTR flag set if
1104                  * we got an error.
1105                  */
1106                 cb->prev_seq = 1;
1107         }
1108         rcu_read_unlock();
1109
1110         cb->args[0] = last_type;
1111         cb->args[1] = last_lower;
1112         cb->args[2] = last_publ;
1113         cb->args[3] = done;
1114
1115         return skb->len;
1116 }
1117
1118 struct tipc_dest *tipc_dest_find(struct list_head *l, u32 node, u32 port)
1119 {
1120         u64 value = (u64)node << 32 | port;
1121         struct tipc_dest *dst;
1122
1123         list_for_each_entry(dst, l, list) {
1124                 if (dst->value != value)
1125                         continue;
1126                 return dst;
1127         }
1128         return NULL;
1129 }
1130
1131 bool tipc_dest_push(struct list_head *l, u32 node, u32 port)
1132 {
1133         u64 value = (u64)node << 32 | port;
1134         struct tipc_dest *dst;
1135
1136         if (tipc_dest_find(l, node, port))
1137                 return false;
1138
1139         dst = kmalloc(sizeof(*dst), GFP_ATOMIC);
1140         if (unlikely(!dst))
1141                 return false;
1142         dst->value = value;
1143         list_add(&dst->list, l);
1144         return true;
1145 }
1146
1147 bool tipc_dest_pop(struct list_head *l, u32 *node, u32 *port)
1148 {
1149         struct tipc_dest *dst;
1150
1151         if (list_empty(l))
1152                 return false;
1153         dst = list_first_entry(l, typeof(*dst), list);
1154         if (port)
1155                 *port = dst->port;
1156         if (node)
1157                 *node = dst->node;
1158         list_del(&dst->list);
1159         kfree(dst);
1160         return true;
1161 }
1162
1163 bool tipc_dest_del(struct list_head *l, u32 node, u32 port)
1164 {
1165         struct tipc_dest *dst;
1166
1167         dst = tipc_dest_find(l, node, port);
1168         if (!dst)
1169                 return false;
1170         list_del(&dst->list);
1171         kfree(dst);
1172         return true;
1173 }
1174
1175 void tipc_dest_list_purge(struct list_head *l)
1176 {
1177         struct tipc_dest *dst, *tmp;
1178
1179         list_for_each_entry_safe(dst, tmp, l, list) {
1180                 list_del(&dst->list);
1181                 kfree(dst);
1182         }
1183 }
1184
1185 int tipc_dest_list_len(struct list_head *l)
1186 {
1187         struct tipc_dest *dst;
1188         int i = 0;
1189
1190         list_for_each_entry(dst, l, list) {
1191                 i++;
1192         }
1193         return i;
1194 }