Add the rt linux 4.1.3-rt3 as base
[kvmfornfv.git] / kernel / 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 <net/genetlink.h>
46
47 #define TIPC_NAMETBL_SIZE 1024          /* must be a power of 2 */
48
49 static const struct nla_policy
50 tipc_nl_name_table_policy[TIPC_NLA_NAME_TABLE_MAX + 1] = {
51         [TIPC_NLA_NAME_TABLE_UNSPEC]    = { .type = NLA_UNSPEC },
52         [TIPC_NLA_NAME_TABLE_PUBL]      = { .type = NLA_NESTED }
53 };
54
55 /**
56  * struct name_info - name sequence publication info
57  * @node_list: circular list of publications made by own node
58  * @cluster_list: circular list of publications made by own cluster
59  * @zone_list: circular list of publications made by own zone
60  * @node_list_size: number of entries in "node_list"
61  * @cluster_list_size: number of entries in "cluster_list"
62  * @zone_list_size: number of entries in "zone_list"
63  *
64  * Note: The zone list always contains at least one entry, since all
65  *       publications of the associated name sequence belong to it.
66  *       (The cluster and node lists may be empty.)
67  */
68 struct name_info {
69         struct list_head node_list;
70         struct list_head cluster_list;
71         struct list_head zone_list;
72         u32 node_list_size;
73         u32 cluster_list_size;
74         u32 zone_list_size;
75 };
76
77 /**
78  * struct sub_seq - container for all published instances of a name sequence
79  * @lower: name sequence lower bound
80  * @upper: name sequence upper bound
81  * @info: pointer to name sequence publication info
82  */
83 struct sub_seq {
84         u32 lower;
85         u32 upper;
86         struct name_info *info;
87 };
88
89 /**
90  * struct name_seq - container for all published instances of a name type
91  * @type: 32 bit 'type' value for name sequence
92  * @sseq: pointer to dynamically-sized array of sub-sequences of this 'type';
93  *        sub-sequences are sorted in ascending order
94  * @alloc: number of sub-sequences currently in array
95  * @first_free: array index of first unused sub-sequence entry
96  * @ns_list: links to adjacent name sequences in hash chain
97  * @subscriptions: list of subscriptions for this 'type'
98  * @lock: spinlock controlling access to publication lists of all sub-sequences
99  * @rcu: RCU callback head used for deferred freeing
100  */
101 struct name_seq {
102         u32 type;
103         struct sub_seq *sseqs;
104         u32 alloc;
105         u32 first_free;
106         struct hlist_node ns_list;
107         struct list_head subscriptions;
108         spinlock_t lock;
109         struct rcu_head rcu;
110 };
111
112 static int hash(int x)
113 {
114         return x & (TIPC_NAMETBL_SIZE - 1);
115 }
116
117 /**
118  * publ_create - create a publication structure
119  */
120 static struct publication *publ_create(u32 type, u32 lower, u32 upper,
121                                        u32 scope, u32 node, u32 port_ref,
122                                        u32 key)
123 {
124         struct publication *publ = kzalloc(sizeof(*publ), GFP_ATOMIC);
125         if (publ == NULL) {
126                 pr_warn("Publication creation failure, no memory\n");
127                 return NULL;
128         }
129
130         publ->type = type;
131         publ->lower = lower;
132         publ->upper = upper;
133         publ->scope = scope;
134         publ->node = node;
135         publ->ref = port_ref;
136         publ->key = key;
137         INIT_LIST_HEAD(&publ->pport_list);
138         return publ;
139 }
140
141 /**
142  * tipc_subseq_alloc - allocate a specified number of sub-sequence structures
143  */
144 static struct sub_seq *tipc_subseq_alloc(u32 cnt)
145 {
146         return kcalloc(cnt, sizeof(struct sub_seq), GFP_ATOMIC);
147 }
148
149 /**
150  * tipc_nameseq_create - create a name sequence structure for the specified 'type'
151  *
152  * Allocates a single sub-sequence structure and sets it to all 0's.
153  */
154 static struct name_seq *tipc_nameseq_create(u32 type, struct hlist_head *seq_head)
155 {
156         struct name_seq *nseq = kzalloc(sizeof(*nseq), GFP_ATOMIC);
157         struct sub_seq *sseq = tipc_subseq_alloc(1);
158
159         if (!nseq || !sseq) {
160                 pr_warn("Name sequence creation failed, no memory\n");
161                 kfree(nseq);
162                 kfree(sseq);
163                 return NULL;
164         }
165
166         spin_lock_init(&nseq->lock);
167         nseq->type = type;
168         nseq->sseqs = sseq;
169         nseq->alloc = 1;
170         INIT_HLIST_NODE(&nseq->ns_list);
171         INIT_LIST_HEAD(&nseq->subscriptions);
172         hlist_add_head_rcu(&nseq->ns_list, seq_head);
173         return nseq;
174 }
175
176 /**
177  * nameseq_find_subseq - find sub-sequence (if any) matching a name instance
178  *
179  * Very time-critical, so binary searches through sub-sequence array.
180  */
181 static struct sub_seq *nameseq_find_subseq(struct name_seq *nseq,
182                                            u32 instance)
183 {
184         struct sub_seq *sseqs = nseq->sseqs;
185         int low = 0;
186         int high = nseq->first_free - 1;
187         int mid;
188
189         while (low <= high) {
190                 mid = (low + high) / 2;
191                 if (instance < sseqs[mid].lower)
192                         high = mid - 1;
193                 else if (instance > sseqs[mid].upper)
194                         low = mid + 1;
195                 else
196                         return &sseqs[mid];
197         }
198         return NULL;
199 }
200
201 /**
202  * nameseq_locate_subseq - determine position of name instance in sub-sequence
203  *
204  * Returns index in sub-sequence array of the entry that contains the specified
205  * instance value; if no entry contains that value, returns the position
206  * where a new entry for it would be inserted in the array.
207  *
208  * Note: Similar to binary search code for locating a sub-sequence.
209  */
210 static u32 nameseq_locate_subseq(struct name_seq *nseq, u32 instance)
211 {
212         struct sub_seq *sseqs = nseq->sseqs;
213         int low = 0;
214         int high = nseq->first_free - 1;
215         int mid;
216
217         while (low <= high) {
218                 mid = (low + high) / 2;
219                 if (instance < sseqs[mid].lower)
220                         high = mid - 1;
221                 else if (instance > sseqs[mid].upper)
222                         low = mid + 1;
223                 else
224                         return mid;
225         }
226         return low;
227 }
228
229 /**
230  * tipc_nameseq_insert_publ
231  */
232 static struct publication *tipc_nameseq_insert_publ(struct net *net,
233                                                     struct name_seq *nseq,
234                                                     u32 type, u32 lower,
235                                                     u32 upper, u32 scope,
236                                                     u32 node, u32 port, u32 key)
237 {
238         struct tipc_subscription *s;
239         struct tipc_subscription *st;
240         struct publication *publ;
241         struct sub_seq *sseq;
242         struct name_info *info;
243         int created_subseq = 0;
244
245         sseq = nameseq_find_subseq(nseq, lower);
246         if (sseq) {
247
248                 /* Lower end overlaps existing entry => need an exact match */
249                 if ((sseq->lower != lower) || (sseq->upper != upper)) {
250                         return NULL;
251                 }
252
253                 info = sseq->info;
254
255                 /* Check if an identical publication already exists */
256                 list_for_each_entry(publ, &info->zone_list, zone_list) {
257                         if ((publ->ref == port) && (publ->key == key) &&
258                             (!publ->node || (publ->node == node)))
259                                 return NULL;
260                 }
261         } else {
262                 u32 inspos;
263                 struct sub_seq *freesseq;
264
265                 /* Find where lower end should be inserted */
266                 inspos = nameseq_locate_subseq(nseq, lower);
267
268                 /* Fail if upper end overlaps into an existing entry */
269                 if ((inspos < nseq->first_free) &&
270                     (upper >= nseq->sseqs[inspos].lower)) {
271                         return NULL;
272                 }
273
274                 /* Ensure there is space for new sub-sequence */
275                 if (nseq->first_free == nseq->alloc) {
276                         struct sub_seq *sseqs = tipc_subseq_alloc(nseq->alloc * 2);
277
278                         if (!sseqs) {
279                                 pr_warn("Cannot publish {%u,%u,%u}, no memory\n",
280                                         type, lower, upper);
281                                 return NULL;
282                         }
283                         memcpy(sseqs, nseq->sseqs,
284                                nseq->alloc * sizeof(struct sub_seq));
285                         kfree(nseq->sseqs);
286                         nseq->sseqs = sseqs;
287                         nseq->alloc *= 2;
288                 }
289
290                 info = kzalloc(sizeof(*info), GFP_ATOMIC);
291                 if (!info) {
292                         pr_warn("Cannot publish {%u,%u,%u}, no memory\n",
293                                 type, lower, upper);
294                         return NULL;
295                 }
296
297                 INIT_LIST_HEAD(&info->node_list);
298                 INIT_LIST_HEAD(&info->cluster_list);
299                 INIT_LIST_HEAD(&info->zone_list);
300
301                 /* Insert new sub-sequence */
302                 sseq = &nseq->sseqs[inspos];
303                 freesseq = &nseq->sseqs[nseq->first_free];
304                 memmove(sseq + 1, sseq, (freesseq - sseq) * sizeof(*sseq));
305                 memset(sseq, 0, sizeof(*sseq));
306                 nseq->first_free++;
307                 sseq->lower = lower;
308                 sseq->upper = upper;
309                 sseq->info = info;
310                 created_subseq = 1;
311         }
312
313         /* Insert a publication */
314         publ = publ_create(type, lower, upper, scope, node, port, key);
315         if (!publ)
316                 return NULL;
317
318         list_add(&publ->zone_list, &info->zone_list);
319         info->zone_list_size++;
320
321         if (in_own_cluster(net, node)) {
322                 list_add(&publ->cluster_list, &info->cluster_list);
323                 info->cluster_list_size++;
324         }
325
326         if (in_own_node(net, node)) {
327                 list_add(&publ->node_list, &info->node_list);
328                 info->node_list_size++;
329         }
330
331         /* Any subscriptions waiting for notification?  */
332         list_for_each_entry_safe(s, st, &nseq->subscriptions, nameseq_list) {
333                 tipc_subscr_report_overlap(s,
334                                            publ->lower,
335                                            publ->upper,
336                                            TIPC_PUBLISHED,
337                                            publ->ref,
338                                            publ->node,
339                                            created_subseq);
340         }
341         return publ;
342 }
343
344 /**
345  * tipc_nameseq_remove_publ
346  *
347  * NOTE: There may be cases where TIPC is asked to remove a publication
348  * that is not in the name table.  For example, if another node issues a
349  * publication for a name sequence that overlaps an existing name sequence
350  * the publication will not be recorded, which means the publication won't
351  * be found when the name sequence is later withdrawn by that node.
352  * A failed withdraw request simply returns a failure indication and lets the
353  * caller issue any error or warning messages associated with such a problem.
354  */
355 static struct publication *tipc_nameseq_remove_publ(struct net *net,
356                                                     struct name_seq *nseq,
357                                                     u32 inst, u32 node,
358                                                     u32 ref, u32 key)
359 {
360         struct publication *publ;
361         struct sub_seq *sseq = nameseq_find_subseq(nseq, inst);
362         struct name_info *info;
363         struct sub_seq *free;
364         struct tipc_subscription *s, *st;
365         int removed_subseq = 0;
366
367         if (!sseq)
368                 return NULL;
369
370         info = sseq->info;
371
372         /* Locate publication, if it exists */
373         list_for_each_entry(publ, &info->zone_list, zone_list) {
374                 if ((publ->key == key) && (publ->ref == ref) &&
375                     (!publ->node || (publ->node == node)))
376                         goto found;
377         }
378         return NULL;
379
380 found:
381         /* Remove publication from zone scope list */
382         list_del(&publ->zone_list);
383         info->zone_list_size--;
384
385         /* Remove publication from cluster scope list, if present */
386         if (in_own_cluster(net, node)) {
387                 list_del(&publ->cluster_list);
388                 info->cluster_list_size--;
389         }
390
391         /* Remove publication from node scope list, if present */
392         if (in_own_node(net, node)) {
393                 list_del(&publ->node_list);
394                 info->node_list_size--;
395         }
396
397         /* Contract subseq list if no more publications for that subseq */
398         if (list_empty(&info->zone_list)) {
399                 kfree(info);
400                 free = &nseq->sseqs[nseq->first_free--];
401                 memmove(sseq, sseq + 1, (free - (sseq + 1)) * sizeof(*sseq));
402                 removed_subseq = 1;
403         }
404
405         /* Notify any waiting subscriptions */
406         list_for_each_entry_safe(s, st, &nseq->subscriptions, nameseq_list) {
407                 tipc_subscr_report_overlap(s,
408                                            publ->lower,
409                                            publ->upper,
410                                            TIPC_WITHDRAWN,
411                                            publ->ref,
412                                            publ->node,
413                                            removed_subseq);
414         }
415
416         return publ;
417 }
418
419 /**
420  * tipc_nameseq_subscribe - attach a subscription, and issue
421  * the prescribed number of events if there is any sub-
422  * sequence overlapping with the requested sequence
423  */
424 static void tipc_nameseq_subscribe(struct name_seq *nseq,
425                                    struct tipc_subscription *s)
426 {
427         struct sub_seq *sseq = nseq->sseqs;
428
429         list_add(&s->nameseq_list, &nseq->subscriptions);
430
431         if (!sseq)
432                 return;
433
434         while (sseq != &nseq->sseqs[nseq->first_free]) {
435                 if (tipc_subscr_overlap(s, sseq->lower, sseq->upper)) {
436                         struct publication *crs;
437                         struct name_info *info = sseq->info;
438                         int must_report = 1;
439
440                         list_for_each_entry(crs, &info->zone_list, zone_list) {
441                                 tipc_subscr_report_overlap(s,
442                                                            sseq->lower,
443                                                            sseq->upper,
444                                                            TIPC_PUBLISHED,
445                                                            crs->ref,
446                                                            crs->node,
447                                                            must_report);
448                                 must_report = 0;
449                         }
450                 }
451                 sseq++;
452         }
453 }
454
455 static struct name_seq *nametbl_find_seq(struct net *net, u32 type)
456 {
457         struct tipc_net *tn = net_generic(net, tipc_net_id);
458         struct hlist_head *seq_head;
459         struct name_seq *ns;
460
461         seq_head = &tn->nametbl->seq_hlist[hash(type)];
462         hlist_for_each_entry_rcu(ns, seq_head, ns_list) {
463                 if (ns->type == type)
464                         return ns;
465         }
466
467         return NULL;
468 };
469
470 struct publication *tipc_nametbl_insert_publ(struct net *net, u32 type,
471                                              u32 lower, u32 upper, u32 scope,
472                                              u32 node, u32 port, u32 key)
473 {
474         struct tipc_net *tn = net_generic(net, tipc_net_id);
475         struct publication *publ;
476         struct name_seq *seq = nametbl_find_seq(net, type);
477         int index = hash(type);
478
479         if ((scope < TIPC_ZONE_SCOPE) || (scope > TIPC_NODE_SCOPE) ||
480             (lower > upper)) {
481                 pr_debug("Failed to publish illegal {%u,%u,%u} with scope %u\n",
482                          type, lower, upper, scope);
483                 return NULL;
484         }
485
486         if (!seq)
487                 seq = tipc_nameseq_create(type, &tn->nametbl->seq_hlist[index]);
488         if (!seq)
489                 return NULL;
490
491         spin_lock_bh(&seq->lock);
492         publ = tipc_nameseq_insert_publ(net, seq, type, lower, upper,
493                                         scope, node, port, key);
494         spin_unlock_bh(&seq->lock);
495         return publ;
496 }
497
498 struct publication *tipc_nametbl_remove_publ(struct net *net, u32 type,
499                                              u32 lower, u32 node, u32 ref,
500                                              u32 key)
501 {
502         struct publication *publ;
503         struct name_seq *seq = nametbl_find_seq(net, type);
504
505         if (!seq)
506                 return NULL;
507
508         spin_lock_bh(&seq->lock);
509         publ = tipc_nameseq_remove_publ(net, seq, lower, node, ref, key);
510         if (!seq->first_free && list_empty(&seq->subscriptions)) {
511                 hlist_del_init_rcu(&seq->ns_list);
512                 kfree(seq->sseqs);
513                 spin_unlock_bh(&seq->lock);
514                 kfree_rcu(seq, rcu);
515                 return publ;
516         }
517         spin_unlock_bh(&seq->lock);
518         return publ;
519 }
520
521 /**
522  * tipc_nametbl_translate - perform name translation
523  *
524  * On entry, 'destnode' is the search domain used during translation.
525  *
526  * On exit:
527  * - if name translation is deferred to another node/cluster/zone,
528  *   leaves 'destnode' unchanged (will be non-zero) and returns 0
529  * - if name translation is attempted and succeeds, sets 'destnode'
530  *   to publishing node and returns port reference (will be non-zero)
531  * - if name translation is attempted and fails, sets 'destnode' to 0
532  *   and returns 0
533  */
534 u32 tipc_nametbl_translate(struct net *net, u32 type, u32 instance,
535                            u32 *destnode)
536 {
537         struct tipc_net *tn = net_generic(net, tipc_net_id);
538         struct sub_seq *sseq;
539         struct name_info *info;
540         struct publication *publ;
541         struct name_seq *seq;
542         u32 ref = 0;
543         u32 node = 0;
544
545         if (!tipc_in_scope(*destnode, tn->own_addr))
546                 return 0;
547
548         rcu_read_lock();
549         seq = nametbl_find_seq(net, type);
550         if (unlikely(!seq))
551                 goto not_found;
552         spin_lock_bh(&seq->lock);
553         sseq = nameseq_find_subseq(seq, instance);
554         if (unlikely(!sseq))
555                 goto no_match;
556         info = sseq->info;
557
558         /* Closest-First Algorithm */
559         if (likely(!*destnode)) {
560                 if (!list_empty(&info->node_list)) {
561                         publ = list_first_entry(&info->node_list,
562                                                 struct publication,
563                                                 node_list);
564                         list_move_tail(&publ->node_list,
565                                        &info->node_list);
566                 } else if (!list_empty(&info->cluster_list)) {
567                         publ = list_first_entry(&info->cluster_list,
568                                                 struct publication,
569                                                 cluster_list);
570                         list_move_tail(&publ->cluster_list,
571                                        &info->cluster_list);
572                 } else {
573                         publ = list_first_entry(&info->zone_list,
574                                                 struct publication,
575                                                 zone_list);
576                         list_move_tail(&publ->zone_list,
577                                        &info->zone_list);
578                 }
579         }
580
581         /* Round-Robin Algorithm */
582         else if (*destnode == tn->own_addr) {
583                 if (list_empty(&info->node_list))
584                         goto no_match;
585                 publ = list_first_entry(&info->node_list, struct publication,
586                                         node_list);
587                 list_move_tail(&publ->node_list, &info->node_list);
588         } else if (in_own_cluster_exact(net, *destnode)) {
589                 if (list_empty(&info->cluster_list))
590                         goto no_match;
591                 publ = list_first_entry(&info->cluster_list, struct publication,
592                                         cluster_list);
593                 list_move_tail(&publ->cluster_list, &info->cluster_list);
594         } else {
595                 publ = list_first_entry(&info->zone_list, struct publication,
596                                         zone_list);
597                 list_move_tail(&publ->zone_list, &info->zone_list);
598         }
599
600         ref = publ->ref;
601         node = publ->node;
602 no_match:
603         spin_unlock_bh(&seq->lock);
604 not_found:
605         rcu_read_unlock();
606         *destnode = node;
607         return ref;
608 }
609
610 /**
611  * tipc_nametbl_mc_translate - find multicast destinations
612  *
613  * Creates list of all local ports that overlap the given multicast address;
614  * also determines if any off-node ports overlap.
615  *
616  * Note: Publications with a scope narrower than 'limit' are ignored.
617  * (i.e. local node-scope publications mustn't receive messages arriving
618  * from another node, even if the multcast link brought it here)
619  *
620  * Returns non-zero if any off-node ports overlap
621  */
622 int tipc_nametbl_mc_translate(struct net *net, u32 type, u32 lower, u32 upper,
623                               u32 limit, struct tipc_plist *dports)
624 {
625         struct name_seq *seq;
626         struct sub_seq *sseq;
627         struct sub_seq *sseq_stop;
628         struct name_info *info;
629         int res = 0;
630
631         rcu_read_lock();
632         seq = nametbl_find_seq(net, type);
633         if (!seq)
634                 goto exit;
635
636         spin_lock_bh(&seq->lock);
637         sseq = seq->sseqs + nameseq_locate_subseq(seq, lower);
638         sseq_stop = seq->sseqs + seq->first_free;
639         for (; sseq != sseq_stop; sseq++) {
640                 struct publication *publ;
641
642                 if (sseq->lower > upper)
643                         break;
644
645                 info = sseq->info;
646                 list_for_each_entry(publ, &info->node_list, node_list) {
647                         if (publ->scope <= limit)
648                                 tipc_plist_push(dports, publ->ref);
649                 }
650
651                 if (info->cluster_list_size != info->node_list_size)
652                         res = 1;
653         }
654         spin_unlock_bh(&seq->lock);
655 exit:
656         rcu_read_unlock();
657         return res;
658 }
659
660 /*
661  * tipc_nametbl_publish - add name publication to network name tables
662  */
663 struct publication *tipc_nametbl_publish(struct net *net, u32 type, u32 lower,
664                                          u32 upper, u32 scope, u32 port_ref,
665                                          u32 key)
666 {
667         struct publication *publ;
668         struct sk_buff *buf = NULL;
669         struct tipc_net *tn = net_generic(net, tipc_net_id);
670
671         spin_lock_bh(&tn->nametbl_lock);
672         if (tn->nametbl->local_publ_count >= TIPC_MAX_PUBLICATIONS) {
673                 pr_warn("Publication failed, local publication limit reached (%u)\n",
674                         TIPC_MAX_PUBLICATIONS);
675                 spin_unlock_bh(&tn->nametbl_lock);
676                 return NULL;
677         }
678
679         publ = tipc_nametbl_insert_publ(net, type, lower, upper, scope,
680                                         tn->own_addr, port_ref, key);
681         if (likely(publ)) {
682                 tn->nametbl->local_publ_count++;
683                 buf = tipc_named_publish(net, publ);
684                 /* Any pending external events? */
685                 tipc_named_process_backlog(net);
686         }
687         spin_unlock_bh(&tn->nametbl_lock);
688
689         if (buf)
690                 named_cluster_distribute(net, buf);
691         return publ;
692 }
693
694 /**
695  * tipc_nametbl_withdraw - withdraw name publication from network name tables
696  */
697 int tipc_nametbl_withdraw(struct net *net, u32 type, u32 lower, u32 ref,
698                           u32 key)
699 {
700         struct publication *publ;
701         struct sk_buff *skb = NULL;
702         struct tipc_net *tn = net_generic(net, tipc_net_id);
703
704         spin_lock_bh(&tn->nametbl_lock);
705         publ = tipc_nametbl_remove_publ(net, type, lower, tn->own_addr,
706                                         ref, key);
707         if (likely(publ)) {
708                 tn->nametbl->local_publ_count--;
709                 skb = tipc_named_withdraw(net, publ);
710                 /* Any pending external events? */
711                 tipc_named_process_backlog(net);
712                 list_del_init(&publ->pport_list);
713                 kfree_rcu(publ, rcu);
714         } else {
715                 pr_err("Unable to remove local publication\n"
716                        "(type=%u, lower=%u, ref=%u, key=%u)\n",
717                        type, lower, ref, key);
718         }
719         spin_unlock_bh(&tn->nametbl_lock);
720
721         if (skb) {
722                 named_cluster_distribute(net, skb);
723                 return 1;
724         }
725         return 0;
726 }
727
728 /**
729  * tipc_nametbl_subscribe - add a subscription object to the name table
730  */
731 void tipc_nametbl_subscribe(struct tipc_subscription *s)
732 {
733         struct tipc_net *tn = net_generic(s->net, tipc_net_id);
734         u32 type = s->seq.type;
735         int index = hash(type);
736         struct name_seq *seq;
737
738         spin_lock_bh(&tn->nametbl_lock);
739         seq = nametbl_find_seq(s->net, type);
740         if (!seq)
741                 seq = tipc_nameseq_create(type, &tn->nametbl->seq_hlist[index]);
742         if (seq) {
743                 spin_lock_bh(&seq->lock);
744                 tipc_nameseq_subscribe(seq, s);
745                 spin_unlock_bh(&seq->lock);
746         } else {
747                 pr_warn("Failed to create subscription for {%u,%u,%u}\n",
748                         s->seq.type, s->seq.lower, s->seq.upper);
749         }
750         spin_unlock_bh(&tn->nametbl_lock);
751 }
752
753 /**
754  * tipc_nametbl_unsubscribe - remove a subscription object from name table
755  */
756 void tipc_nametbl_unsubscribe(struct tipc_subscription *s)
757 {
758         struct tipc_net *tn = net_generic(s->net, tipc_net_id);
759         struct name_seq *seq;
760
761         spin_lock_bh(&tn->nametbl_lock);
762         seq = nametbl_find_seq(s->net, s->seq.type);
763         if (seq != NULL) {
764                 spin_lock_bh(&seq->lock);
765                 list_del_init(&s->nameseq_list);
766                 if (!seq->first_free && list_empty(&seq->subscriptions)) {
767                         hlist_del_init_rcu(&seq->ns_list);
768                         kfree(seq->sseqs);
769                         spin_unlock_bh(&seq->lock);
770                         kfree_rcu(seq, rcu);
771                 } else {
772                         spin_unlock_bh(&seq->lock);
773                 }
774         }
775         spin_unlock_bh(&tn->nametbl_lock);
776 }
777
778 int tipc_nametbl_init(struct net *net)
779 {
780         struct tipc_net *tn = net_generic(net, tipc_net_id);
781         struct name_table *tipc_nametbl;
782         int i;
783
784         tipc_nametbl = kzalloc(sizeof(*tipc_nametbl), GFP_ATOMIC);
785         if (!tipc_nametbl)
786                 return -ENOMEM;
787
788         for (i = 0; i < TIPC_NAMETBL_SIZE; i++)
789                 INIT_HLIST_HEAD(&tipc_nametbl->seq_hlist[i]);
790
791         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_ZONE_SCOPE]);
792         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_CLUSTER_SCOPE]);
793         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_NODE_SCOPE]);
794         tn->nametbl = tipc_nametbl;
795         spin_lock_init(&tn->nametbl_lock);
796         return 0;
797 }
798
799 /**
800  * tipc_purge_publications - remove all publications for a given type
801  *
802  * tipc_nametbl_lock must be held when calling this function
803  */
804 static void tipc_purge_publications(struct net *net, struct name_seq *seq)
805 {
806         struct publication *publ, *safe;
807         struct sub_seq *sseq;
808         struct name_info *info;
809
810         spin_lock_bh(&seq->lock);
811         sseq = seq->sseqs;
812         info = sseq->info;
813         list_for_each_entry_safe(publ, safe, &info->zone_list, zone_list) {
814                 tipc_nameseq_remove_publ(net, seq, publ->lower, publ->node,
815                                          publ->ref, publ->key);
816                 kfree_rcu(publ, rcu);
817         }
818         hlist_del_init_rcu(&seq->ns_list);
819         kfree(seq->sseqs);
820         spin_unlock_bh(&seq->lock);
821
822         kfree_rcu(seq, rcu);
823 }
824
825 void tipc_nametbl_stop(struct net *net)
826 {
827         u32 i;
828         struct name_seq *seq;
829         struct hlist_head *seq_head;
830         struct tipc_net *tn = net_generic(net, tipc_net_id);
831         struct name_table *tipc_nametbl = tn->nametbl;
832
833         /* Verify name table is empty and purge any lingering
834          * publications, then release the name table
835          */
836         spin_lock_bh(&tn->nametbl_lock);
837         for (i = 0; i < TIPC_NAMETBL_SIZE; i++) {
838                 if (hlist_empty(&tipc_nametbl->seq_hlist[i]))
839                         continue;
840                 seq_head = &tipc_nametbl->seq_hlist[i];
841                 hlist_for_each_entry_rcu(seq, seq_head, ns_list) {
842                         tipc_purge_publications(net, seq);
843                 }
844         }
845         spin_unlock_bh(&tn->nametbl_lock);
846
847         synchronize_net();
848         kfree(tipc_nametbl);
849
850 }
851
852 static int __tipc_nl_add_nametable_publ(struct tipc_nl_msg *msg,
853                                         struct name_seq *seq,
854                                         struct sub_seq *sseq, u32 *last_publ)
855 {
856         void *hdr;
857         struct nlattr *attrs;
858         struct nlattr *publ;
859         struct publication *p;
860
861         if (*last_publ) {
862                 list_for_each_entry(p, &sseq->info->zone_list, zone_list)
863                         if (p->key == *last_publ)
864                                 break;
865                 if (p->key != *last_publ)
866                         return -EPIPE;
867         } else {
868                 p = list_first_entry(&sseq->info->zone_list, struct publication,
869                                      zone_list);
870         }
871
872         list_for_each_entry_from(p, &sseq->info->zone_list, zone_list) {
873                 *last_publ = p->key;
874
875                 hdr = genlmsg_put(msg->skb, msg->portid, msg->seq,
876                                   &tipc_genl_family, NLM_F_MULTI,
877                                   TIPC_NL_NAME_TABLE_GET);
878                 if (!hdr)
879                         return -EMSGSIZE;
880
881                 attrs = nla_nest_start(msg->skb, TIPC_NLA_NAME_TABLE);
882                 if (!attrs)
883                         goto msg_full;
884
885                 publ = nla_nest_start(msg->skb, TIPC_NLA_NAME_TABLE_PUBL);
886                 if (!publ)
887                         goto attr_msg_full;
888
889                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_TYPE, seq->type))
890                         goto publ_msg_full;
891                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_LOWER, sseq->lower))
892                         goto publ_msg_full;
893                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_UPPER, sseq->upper))
894                         goto publ_msg_full;
895                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_SCOPE, p->scope))
896                         goto publ_msg_full;
897                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_NODE, p->node))
898                         goto publ_msg_full;
899                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_REF, p->ref))
900                         goto publ_msg_full;
901                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_KEY, p->key))
902                         goto publ_msg_full;
903
904                 nla_nest_end(msg->skb, publ);
905                 nla_nest_end(msg->skb, attrs);
906                 genlmsg_end(msg->skb, hdr);
907         }
908         *last_publ = 0;
909
910         return 0;
911
912 publ_msg_full:
913         nla_nest_cancel(msg->skb, publ);
914 attr_msg_full:
915         nla_nest_cancel(msg->skb, attrs);
916 msg_full:
917         genlmsg_cancel(msg->skb, hdr);
918
919         return -EMSGSIZE;
920 }
921
922 static int __tipc_nl_subseq_list(struct tipc_nl_msg *msg, struct name_seq *seq,
923                                  u32 *last_lower, u32 *last_publ)
924 {
925         struct sub_seq *sseq;
926         struct sub_seq *sseq_start;
927         int err;
928
929         if (*last_lower) {
930                 sseq_start = nameseq_find_subseq(seq, *last_lower);
931                 if (!sseq_start)
932                         return -EPIPE;
933         } else {
934                 sseq_start = seq->sseqs;
935         }
936
937         for (sseq = sseq_start; sseq != &seq->sseqs[seq->first_free]; sseq++) {
938                 err = __tipc_nl_add_nametable_publ(msg, seq, sseq, last_publ);
939                 if (err) {
940                         *last_lower = sseq->lower;
941                         return err;
942                 }
943         }
944         *last_lower = 0;
945
946         return 0;
947 }
948
949 static int tipc_nl_seq_list(struct net *net, struct tipc_nl_msg *msg,
950                             u32 *last_type, u32 *last_lower, u32 *last_publ)
951 {
952         struct tipc_net *tn = net_generic(net, tipc_net_id);
953         struct hlist_head *seq_head;
954         struct name_seq *seq = NULL;
955         int err;
956         int i;
957
958         if (*last_type)
959                 i = hash(*last_type);
960         else
961                 i = 0;
962
963         for (; i < TIPC_NAMETBL_SIZE; i++) {
964                 seq_head = &tn->nametbl->seq_hlist[i];
965
966                 if (*last_type) {
967                         seq = nametbl_find_seq(net, *last_type);
968                         if (!seq)
969                                 return -EPIPE;
970                 } else {
971                         hlist_for_each_entry_rcu(seq, seq_head, ns_list)
972                                 break;
973                         if (!seq)
974                                 continue;
975                 }
976
977                 hlist_for_each_entry_from_rcu(seq, ns_list) {
978                         spin_lock_bh(&seq->lock);
979                         err = __tipc_nl_subseq_list(msg, seq, last_lower,
980                                                     last_publ);
981
982                         if (err) {
983                                 *last_type = seq->type;
984                                 spin_unlock_bh(&seq->lock);
985                                 return err;
986                         }
987                         spin_unlock_bh(&seq->lock);
988                 }
989                 *last_type = 0;
990         }
991         return 0;
992 }
993
994 int tipc_nl_name_table_dump(struct sk_buff *skb, struct netlink_callback *cb)
995 {
996         int err;
997         int done = cb->args[3];
998         u32 last_type = cb->args[0];
999         u32 last_lower = cb->args[1];
1000         u32 last_publ = cb->args[2];
1001         struct net *net = sock_net(skb->sk);
1002         struct tipc_nl_msg msg;
1003
1004         if (done)
1005                 return 0;
1006
1007         msg.skb = skb;
1008         msg.portid = NETLINK_CB(cb->skb).portid;
1009         msg.seq = cb->nlh->nlmsg_seq;
1010
1011         rcu_read_lock();
1012         err = tipc_nl_seq_list(net, &msg, &last_type, &last_lower, &last_publ);
1013         if (!err) {
1014                 done = 1;
1015         } else if (err != -EMSGSIZE) {
1016                 /* We never set seq or call nl_dump_check_consistent() this
1017                  * means that setting prev_seq here will cause the consistence
1018                  * check to fail in the netlink callback handler. Resulting in
1019                  * the NLMSG_DONE message having the NLM_F_DUMP_INTR flag set if
1020                  * we got an error.
1021                  */
1022                 cb->prev_seq = 1;
1023         }
1024         rcu_read_unlock();
1025
1026         cb->args[0] = last_type;
1027         cb->args[1] = last_lower;
1028         cb->args[2] = last_publ;
1029         cb->args[3] = done;
1030
1031         return skb->len;
1032 }
1033
1034 void tipc_plist_push(struct tipc_plist *pl, u32 port)
1035 {
1036         struct tipc_plist *nl;
1037
1038         if (likely(!pl->port)) {
1039                 pl->port = port;
1040                 return;
1041         }
1042         if (pl->port == port)
1043                 return;
1044         list_for_each_entry(nl, &pl->list, list) {
1045                 if (nl->port == port)
1046                         return;
1047         }
1048         nl = kmalloc(sizeof(*nl), GFP_ATOMIC);
1049         if (nl) {
1050                 nl->port = port;
1051                 list_add(&nl->list, &pl->list);
1052         }
1053 }
1054
1055 u32 tipc_plist_pop(struct tipc_plist *pl)
1056 {
1057         struct tipc_plist *nl;
1058         u32 port = 0;
1059
1060         if (likely(list_empty(&pl->list))) {
1061                 port = pl->port;
1062                 pl->port = 0;
1063                 return port;
1064         }
1065         nl = list_first_entry(&pl->list, typeof(*nl), list);
1066         port = nl->port;
1067         list_del(&nl->list);
1068         kfree(nl);
1069         return port;
1070 }