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_subscrp_report_overlap(s, publ->lower, publ->upper,
 334                                             TIPC_PUBLISHED, publ->ref,
 335                                             publ->node, created_subseq);
 336         }
 337         return publ;
 338 }
 339
 340 /**
 341  * tipc_nameseq_remove_publ
 342  *
 343  * NOTE: There may be cases where TIPC is asked to remove a publication
 344  * that is not in the name table.  For example, if another node issues a
 345  * publication for a name sequence that overlaps an existing name sequence
 346  * the publication will not be recorded, which means the publication won't
 347  * be found when the name sequence is later withdrawn by that node.
 348  * A failed withdraw request simply returns a failure indication and lets the
 349  * caller issue any error or warning messages associated with such a problem.
 350  */
 351 static struct publication *tipc_nameseq_remove_publ(struct net *net,
 352                                                     struct name_seq *nseq,
 353                                                     u32 inst, u32 node,
 354                                                     u32 ref, u32 key)
 355 {
 356         struct publication *publ;
 357         struct sub_seq *sseq = nameseq_find_subseq(nseq, inst);
 358         struct name_info *info;
 359         struct sub_seq *free;
 360         struct tipc_subscription *s, *st;
 361         int removed_subseq = 0;
 362
 363         if (!sseq)
 364                 return NULL;
 365
 366         info = sseq->info;
 367
 368         /* Locate publication, if it exists */
 369         list_for_each_entry(publ, &info->zone_list, zone_list) {
 370                 if ((publ->key == key) && (publ->ref == ref) &&
 371                     (!publ->node || (publ->node == node)))
 372                         goto found;
 373         }
 374         return NULL;
 375
 376 found:
 377         /* Remove publication from zone scope list */
 378         list_del(&publ->zone_list);
 379         info->zone_list_size--;
 380
 381         /* Remove publication from cluster scope list, if present */
 382         if (in_own_cluster(net, node)) {
 383                 list_del(&publ->cluster_list);
 384                 info->cluster_list_size--;
 385         }
 386
 387         /* Remove publication from node scope list, if present */
 388         if (in_own_node(net, node)) {
 389                 list_del(&publ->node_list);
 390                 info->node_list_size--;
 391         }
 392
 393         /* Contract subseq list if no more publications for that subseq */
 394         if (list_empty(&info->zone_list)) {
 395                 kfree(info);
 396                 free = &nseq->sseqs[nseq->first_free--];
 397                 memmove(sseq, sseq + 1, (free - (sseq + 1)) * sizeof(*sseq));
 398                 removed_subseq = 1;
 399         }
 400
 401         /* Notify any waiting subscriptions */
 402         list_for_each_entry_safe(s, st, &nseq->subscriptions, nameseq_list) {
 403                 tipc_subscrp_report_overlap(s, publ->lower, publ->upper,
 404                                             TIPC_WITHDRAWN, publ->ref,
 405                                             publ->node, removed_subseq);
 406         }
 407
 408         return publ;
 409 }
 410
 411 /**
 412  * tipc_nameseq_subscribe - attach a subscription, and issue
 413  * the prescribed number of events if there is any sub-
 414  * sequence overlapping with the requested sequence
 415  */
 416 static void tipc_nameseq_subscribe(struct name_seq *nseq,
 417                                    struct tipc_subscription *s)
 418 {
 419         struct sub_seq *sseq = nseq->sseqs;
 420
 421         list_add(&s->nameseq_list, &nseq->subscriptions);
 422
 423         if (!sseq)
 424                 return;
 425
 426         while (sseq != &nseq->sseqs[nseq->first_free]) {
 427                 if (tipc_subscrp_check_overlap(s, sseq->lower, sseq->upper)) {
 428                         struct publication *crs;
 429                         struct name_info *info = sseq->info;
 430                         int must_report = 1;
 431
 432                         list_for_each_entry(crs, &info->zone_list, zone_list) {
 433                                 tipc_subscrp_report_overlap(s, sseq->lower,
 434                                                             sseq->upper,
 435                                                             TIPC_PUBLISHED,
 436                                                             crs->ref, crs->node,
 437                                                             must_report);
 438                                 must_report = 0;
 439                         }
 440                 }
 441                 sseq++;
 442         }
 443 }
 444
 445 static struct name_seq *nametbl_find_seq(struct net *net, u32 type)
 446 {
 447         struct tipc_net *tn = net_generic(net, tipc_net_id);
 448         struct hlist_head *seq_head;
 449         struct name_seq *ns;
 450
 451         seq_head = &tn->nametbl->seq_hlist[hash(type)];
 452         hlist_for_each_entry_rcu(ns, seq_head, ns_list) {
 453                 if (ns->type == type)
 454                         return ns;
 455         }
 456
 457         return NULL;
 458 };
 459
 460 struct publication *tipc_nametbl_insert_publ(struct net *net, u32 type,
 461                                              u32 lower, u32 upper, u32 scope,
 462                                              u32 node, u32 port, u32 key)
 463 {
 464         struct tipc_net *tn = net_generic(net, tipc_net_id);
 465         struct publication *publ;
 466         struct name_seq *seq = nametbl_find_seq(net, type);
 467         int index = hash(type);
 468
 469         if ((scope < TIPC_ZONE_SCOPE) || (scope > TIPC_NODE_SCOPE) ||
 470             (lower > upper)) {
 471                 pr_debug("Failed to publish illegal {%u,%u,%u} with scope %u\n",
 472                          type, lower, upper, scope);
 473                 return NULL;
 474         }
 475
 476         if (!seq)
 477                 seq = tipc_nameseq_create(type, &tn->nametbl->seq_hlist[index]);
 478         if (!seq)
 479                 return NULL;
 480
 481         spin_lock_bh(&seq->lock);
 482         publ = tipc_nameseq_insert_publ(net, seq, type, lower, upper,
 483                                         scope, node, port, key);
 484         spin_unlock_bh(&seq->lock);
 485         return publ;
 486 }
 487
 488 struct publication *tipc_nametbl_remove_publ(struct net *net, u32 type,
 489                                              u32 lower, u32 node, u32 ref,
 490                                              u32 key)
 491 {
 492         struct publication *publ;
 493         struct name_seq *seq = nametbl_find_seq(net, type);
 494
 495         if (!seq)
 496                 return NULL;
 497
 498         spin_lock_bh(&seq->lock);
 499         publ = tipc_nameseq_remove_publ(net, seq, lower, node, ref, key);
 500         if (!seq->first_free && list_empty(&seq->subscriptions)) {
 501                 hlist_del_init_rcu(&seq->ns_list);
 502                 kfree(seq->sseqs);
 503                 spin_unlock_bh(&seq->lock);
 504                 kfree_rcu(seq, rcu);
 505                 return publ;
 506         }
 507         spin_unlock_bh(&seq->lock);
 508         return publ;
 509 }
 510
 511 /**
 512  * tipc_nametbl_translate - perform name translation
 513  *
 514  * On entry, 'destnode' is the search domain used during translation.
 515  *
 516  * On exit:
 517  * - if name translation is deferred to another node/cluster/zone,
 518  *   leaves 'destnode' unchanged (will be non-zero) and returns 0
 519  * - if name translation is attempted and succeeds, sets 'destnode'
 520  *   to publishing node and returns port reference (will be non-zero)
 521  * - if name translation is attempted and fails, sets 'destnode' to 0
 522  *   and returns 0
 523  */
 524 u32 tipc_nametbl_translate(struct net *net, u32 type, u32 instance,
 525                            u32 *destnode)
 526 {
 527         struct tipc_net *tn = net_generic(net, tipc_net_id);
 528         struct sub_seq *sseq;
 529         struct name_info *info;
 530         struct publication *publ;
 531         struct name_seq *seq;
 532         u32 ref = 0;
 533         u32 node = 0;
 534
 535         if (!tipc_in_scope(*destnode, tn->own_addr))
 536                 return 0;
 537
 538         rcu_read_lock();
 539         seq = nametbl_find_seq(net, type);
 540         if (unlikely(!seq))
 541                 goto not_found;
 542         spin_lock_bh(&seq->lock);
 543         sseq = nameseq_find_subseq(seq, instance);
 544         if (unlikely(!sseq))
 545                 goto no_match;
 546         info = sseq->info;
 547
 548         /* Closest-First Algorithm */
 549         if (likely(!*destnode)) {
 550                 if (!list_empty(&info->node_list)) {
 551                         publ = list_first_entry(&info->node_list,
 552                                                 struct publication,
 553                                                 node_list);
 554                         list_move_tail(&publ->node_list,
 555                                        &info->node_list);
 556                 } else if (!list_empty(&info->cluster_list)) {
 557                         publ = list_first_entry(&info->cluster_list,
 558                                                 struct publication,
 559                                                 cluster_list);
 560                         list_move_tail(&publ->cluster_list,
 561                                        &info->cluster_list);
 562                 } else {
 563                         publ = list_first_entry(&info->zone_list,
 564                                                 struct publication,
 565                                                 zone_list);
 566                         list_move_tail(&publ->zone_list,
 567                                        &info->zone_list);
 568                 }
 569         }
 570
 571         /* Round-Robin Algorithm */
 572         else if (*destnode == tn->own_addr) {
 573                 if (list_empty(&info->node_list))
 574                         goto no_match;
 575                 publ = list_first_entry(&info->node_list, struct publication,
 576                                         node_list);
 577                 list_move_tail(&publ->node_list, &info->node_list);
 578         } else if (in_own_cluster_exact(net, *destnode)) {
 579                 if (list_empty(&info->cluster_list))
 580                         goto no_match;
 581                 publ = list_first_entry(&info->cluster_list, struct publication,
 582                                         cluster_list);
 583                 list_move_tail(&publ->cluster_list, &info->cluster_list);
 584         } else {
 585                 publ = list_first_entry(&info->zone_list, struct publication,
 586                                         zone_list);
 587                 list_move_tail(&publ->zone_list, &info->zone_list);
 588         }
 589
 590         ref = publ->ref;
 591         node = publ->node;
 592 no_match:
 593         spin_unlock_bh(&seq->lock);
 594 not_found:
 595         rcu_read_unlock();
 596         *destnode = node;
 597         return ref;
 598 }
 599
 600 /**
 601  * tipc_nametbl_mc_translate - find multicast destinations
 602  *
 603  * Creates list of all local ports that overlap the given multicast address;
 604  * also determines if any off-node ports overlap.
 605  *
 606  * Note: Publications with a scope narrower than 'limit' are ignored.
 607  * (i.e. local node-scope publications mustn't receive messages arriving
 608  * from another node, even if the multcast link brought it here)
 609  *
 610  * Returns non-zero if any off-node ports overlap
 611  */
 612 int tipc_nametbl_mc_translate(struct net *net, u32 type, u32 lower, u32 upper,
 613                               u32 limit, struct tipc_plist *dports)
 614 {
 615         struct name_seq *seq;
 616         struct sub_seq *sseq;
 617         struct sub_seq *sseq_stop;
 618         struct name_info *info;
 619         int res = 0;
 620
 621         rcu_read_lock();
 622         seq = nametbl_find_seq(net, type);
 623         if (!seq)
 624                 goto exit;
 625
 626         spin_lock_bh(&seq->lock);
 627         sseq = seq->sseqs + nameseq_locate_subseq(seq, lower);
 628         sseq_stop = seq->sseqs + seq->first_free;
 629         for (; sseq != sseq_stop; sseq++) {
 630                 struct publication *publ;
 631
 632                 if (sseq->lower > upper)
 633                         break;
 634
 635                 info = sseq->info;
 636                 list_for_each_entry(publ, &info->node_list, node_list) {
 637                         if (publ->scope <= limit)
 638                                 tipc_plist_push(dports, publ->ref);
 639                 }
 640
 641                 if (info->cluster_list_size != info->node_list_size)
 642                         res = 1;
 643         }
 644         spin_unlock_bh(&seq->lock);
 645 exit:
 646         rcu_read_unlock();
 647         return res;
 648 }
 649
 650 /*
 651  * tipc_nametbl_publish - add name publication to network name tables
 652  */
 653 struct publication *tipc_nametbl_publish(struct net *net, u32 type, u32 lower,
 654                                          u32 upper, u32 scope, u32 port_ref,
 655                                          u32 key)
 656 {
 657         struct publication *publ;
 658         struct sk_buff *buf = NULL;
 659         struct tipc_net *tn = net_generic(net, tipc_net_id);
 660
 661         spin_lock_bh(&tn->nametbl_lock);
 662         if (tn->nametbl->local_publ_count >= TIPC_MAX_PUBLICATIONS) {
 663                 pr_warn("Publication failed, local publication limit reached (%u)\n",
 664                         TIPC_MAX_PUBLICATIONS);
 665                 spin_unlock_bh(&tn->nametbl_lock);
 666                 return NULL;
 667         }
 668
 669         publ = tipc_nametbl_insert_publ(net, type, lower, upper, scope,
 670                                         tn->own_addr, port_ref, key);
 671         if (likely(publ)) {
 672                 tn->nametbl->local_publ_count++;
 673                 buf = tipc_named_publish(net, publ);
 674                 /* Any pending external events? */
 675                 tipc_named_process_backlog(net);
 676         }
 677         spin_unlock_bh(&tn->nametbl_lock);
 678
 679         if (buf)
 680                 named_cluster_distribute(net, buf);
 681         return publ;
 682 }
 683
 684 /**
 685  * tipc_nametbl_withdraw - withdraw name publication from network name tables
 686  */
 687 int tipc_nametbl_withdraw(struct net *net, u32 type, u32 lower, u32 ref,
 688                           u32 key)
 689 {
 690         struct publication *publ;
 691         struct sk_buff *skb = NULL;
 692         struct tipc_net *tn = net_generic(net, tipc_net_id);
 693
 694         spin_lock_bh(&tn->nametbl_lock);
 695         publ = tipc_nametbl_remove_publ(net, type, lower, tn->own_addr,
 696                                         ref, key);
 697         if (likely(publ)) {
 698                 tn->nametbl->local_publ_count--;
 699                 skb = tipc_named_withdraw(net, publ);
 700                 /* Any pending external events? */
 701                 tipc_named_process_backlog(net);
 702                 list_del_init(&publ->pport_list);
 703                 kfree_rcu(publ, rcu);
 704         } else {
 705                 pr_err("Unable to remove local publication\n"
 706                        "(type=%u, lower=%u, ref=%u, key=%u)\n",
 707                        type, lower, ref, key);
 708         }
 709         spin_unlock_bh(&tn->nametbl_lock);
 710
 711         if (skb) {
 712                 named_cluster_distribute(net, skb);
 713                 return 1;
 714         }
 715         return 0;
 716 }
 717
 718 /**
 719  * tipc_nametbl_subscribe - add a subscription object to the name table
 720  */
 721 void tipc_nametbl_subscribe(struct tipc_subscription *s)
 722 {
 723         struct tipc_net *tn = net_generic(s->net, tipc_net_id);
 724         u32 type = s->seq.type;
 725         int index = hash(type);
 726         struct name_seq *seq;
 727
 728         spin_lock_bh(&tn->nametbl_lock);
 729         seq = nametbl_find_seq(s->net, type);
 730         if (!seq)
 731                 seq = tipc_nameseq_create(type, &tn->nametbl->seq_hlist[index]);
 732         if (seq) {
 733                 spin_lock_bh(&seq->lock);
 734                 tipc_nameseq_subscribe(seq, s);
 735                 spin_unlock_bh(&seq->lock);
 736         } else {
 737                 pr_warn("Failed to create subscription for {%u,%u,%u}\n",
 738                         s->seq.type, s->seq.lower, s->seq.upper);
 739         }
 740         spin_unlock_bh(&tn->nametbl_lock);
 741 }
 742
 743 /**
 744  * tipc_nametbl_unsubscribe - remove a subscription object from name table
 745  */
 746 void tipc_nametbl_unsubscribe(struct tipc_subscription *s)
 747 {
 748         struct tipc_net *tn = net_generic(s->net, tipc_net_id);
 749         struct name_seq *seq;
 750
 751         spin_lock_bh(&tn->nametbl_lock);
 752         seq = nametbl_find_seq(s->net, s->seq.type);
 753         if (seq != NULL) {
 754                 spin_lock_bh(&seq->lock);
 755                 list_del_init(&s->nameseq_list);
 756                 if (!seq->first_free && list_empty(&seq->subscriptions)) {
 757                         hlist_del_init_rcu(&seq->ns_list);
 758                         kfree(seq->sseqs);
 759                         spin_unlock_bh(&seq->lock);
 760                         kfree_rcu(seq, rcu);
 761                 } else {
 762                         spin_unlock_bh(&seq->lock);
 763                 }
 764         }
 765         spin_unlock_bh(&tn->nametbl_lock);
 766 }
 767
 768 int tipc_nametbl_init(struct net *net)
 769 {
 770         struct tipc_net *tn = net_generic(net, tipc_net_id);
 771         struct name_table *tipc_nametbl;
 772         int i;
 773
 774         tipc_nametbl = kzalloc(sizeof(*tipc_nametbl), GFP_ATOMIC);
 775         if (!tipc_nametbl)
 776                 return -ENOMEM;
 777
 778         for (i = 0; i < TIPC_NAMETBL_SIZE; i++)
 779                 INIT_HLIST_HEAD(&tipc_nametbl->seq_hlist[i]);
 780
 781         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_ZONE_SCOPE]);
 782         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_CLUSTER_SCOPE]);
 783         INIT_LIST_HEAD(&tipc_nametbl->publ_list[TIPC_NODE_SCOPE]);
 784         tn->nametbl = tipc_nametbl;
 785         spin_lock_init(&tn->nametbl_lock);
 786         return 0;
 787 }
 788
 789 /**
 790  * tipc_purge_publications - remove all publications for a given type
 791  *
 792  * tipc_nametbl_lock must be held when calling this function
 793  */
 794 static void tipc_purge_publications(struct net *net, struct name_seq *seq)
 795 {
 796         struct publication *publ, *safe;
 797         struct sub_seq *sseq;
 798         struct name_info *info;
 799
 800         spin_lock_bh(&seq->lock);
 801         sseq = seq->sseqs;
 802         info = sseq->info;
 803         list_for_each_entry_safe(publ, safe, &info->zone_list, zone_list) {
 804                 tipc_nameseq_remove_publ(net, seq, publ->lower, publ->node,
 805                                          publ->ref, publ->key);
 806                 kfree_rcu(publ, rcu);
 807         }
 808         hlist_del_init_rcu(&seq->ns_list);
 809         kfree(seq->sseqs);
 810         spin_unlock_bh(&seq->lock);
 811
 812         kfree_rcu(seq, rcu);
 813 }
 814
 815 void tipc_nametbl_stop(struct net *net)
 816 {
 817         u32 i;
 818         struct name_seq *seq;
 819         struct hlist_head *seq_head;
 820         struct tipc_net *tn = net_generic(net, tipc_net_id);
 821         struct name_table *tipc_nametbl = tn->nametbl;
 822
 823         /* Verify name table is empty and purge any lingering
 824          * publications, then release the name table
 825          */
 826         spin_lock_bh(&tn->nametbl_lock);
 827         for (i = 0; i < TIPC_NAMETBL_SIZE; i++) {
 828                 if (hlist_empty(&tipc_nametbl->seq_hlist[i]))
 829                         continue;
 830                 seq_head = &tipc_nametbl->seq_hlist[i];
 831                 hlist_for_each_entry_rcu(seq, seq_head, ns_list) {
 832                         tipc_purge_publications(net, seq);
 833                 }
 834         }
 835         spin_unlock_bh(&tn->nametbl_lock);
 836
 837         synchronize_net();
 838         kfree(tipc_nametbl);
 839
 840 }
 841
 842 static int __tipc_nl_add_nametable_publ(struct tipc_nl_msg *msg,
 843                                         struct name_seq *seq,
 844                                         struct sub_seq *sseq, u32 *last_publ)
 845 {
 846         void *hdr;
 847         struct nlattr *attrs;
 848         struct nlattr *publ;
 849         struct publication *p;
 850
 851         if (*last_publ) {
 852                 list_for_each_entry(p, &sseq->info->zone_list, zone_list)
 853                         if (p->key == *last_publ)
 854                                 break;
 855                 if (p->key != *last_publ)
 856                         return -EPIPE;
 857         } else {
 858                 p = list_first_entry(&sseq->info->zone_list, struct publication,
 859                                      zone_list);
 860         }
 861
 862         list_for_each_entry_from(p, &sseq->info->zone_list, zone_list) {
 863                 *last_publ = p->key;
 864
 865                 hdr = genlmsg_put(msg->skb, msg->portid, msg->seq,
 866                                   &tipc_genl_family, NLM_F_MULTI,
 867                                   TIPC_NL_NAME_TABLE_GET);
 868                 if (!hdr)
 869                         return -EMSGSIZE;
 870
 871                 attrs = nla_nest_start(msg->skb, TIPC_NLA_NAME_TABLE);
 872                 if (!attrs)
 873                         goto msg_full;
 874
 875                 publ = nla_nest_start(msg->skb, TIPC_NLA_NAME_TABLE_PUBL);
 876                 if (!publ)
 877                         goto attr_msg_full;
 878
 879                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_TYPE, seq->type))
 880                         goto publ_msg_full;
 881                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_LOWER, sseq->lower))
 882                         goto publ_msg_full;
 883                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_UPPER, sseq->upper))
 884                         goto publ_msg_full;
 885                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_SCOPE, p->scope))
 886                         goto publ_msg_full;
 887                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_NODE, p->node))
 888                         goto publ_msg_full;
 889                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_REF, p->ref))
 890                         goto publ_msg_full;
 891                 if (nla_put_u32(msg->skb, TIPC_NLA_PUBL_KEY, p->key))
 892                         goto publ_msg_full;
 893
 894                 nla_nest_end(msg->skb, publ);
 895                 nla_nest_end(msg->skb, attrs);
 896                 genlmsg_end(msg->skb, hdr);
 897         }
 898         *last_publ = 0;
 899
 900         return 0;
 901
 902 publ_msg_full:
 903         nla_nest_cancel(msg->skb, publ);
 904 attr_msg_full:
 905         nla_nest_cancel(msg->skb, attrs);
 906 msg_full:
 907         genlmsg_cancel(msg->skb, hdr);
 908
 909         return -EMSGSIZE;
 910 }
 911
 912 static int __tipc_nl_subseq_list(struct tipc_nl_msg *msg, struct name_seq *seq,
 913                                  u32 *last_lower, u32 *last_publ)
 914 {
 915         struct sub_seq *sseq;
 916         struct sub_seq *sseq_start;
 917         int err;
 918
 919         if (*last_lower) {
 920                 sseq_start = nameseq_find_subseq(seq, *last_lower);
 921                 if (!sseq_start)
 922                         return -EPIPE;
 923         } else {
 924                 sseq_start = seq->sseqs;
 925         }
 926
 927         for (sseq = sseq_start; sseq != &seq->sseqs[seq->first_free]; sseq++) {
 928                 err = __tipc_nl_add_nametable_publ(msg, seq, sseq, last_publ);
 929                 if (err) {
 930                         *last_lower = sseq->lower;
 931                         return err;
 932                 }
 933         }
 934         *last_lower = 0;
 935
 936         return 0;
 937 }
 938
 939 static int tipc_nl_seq_list(struct net *net, struct tipc_nl_msg *msg,
 940                             u32 *last_type, u32 *last_lower, u32 *last_publ)
 941 {
 942         struct tipc_net *tn = net_generic(net, tipc_net_id);
 943         struct hlist_head *seq_head;
 944         struct name_seq *seq = NULL;
 945         int err;
 946         int i;
 947
 948         if (*last_type)
 949                 i = hash(*last_type);
 950         else
 951                 i = 0;
 952
 953         for (; i < TIPC_NAMETBL_SIZE; i++) {
 954                 seq_head = &tn->nametbl->seq_hlist[i];
 955
 956                 if (*last_type) {
 957                         seq = nametbl_find_seq(net, *last_type);
 958                         if (!seq)
 959                                 return -EPIPE;
 960                 } else {
 961                         hlist_for_each_entry_rcu(seq, seq_head, ns_list)
 962                                 break;
 963                         if (!seq)
 964                                 continue;
 965                 }
 966
 967                 hlist_for_each_entry_from_rcu(seq, ns_list) {
 968                         spin_lock_bh(&seq->lock);
 969                         err = __tipc_nl_subseq_list(msg, seq, last_lower,
 970                                                     last_publ);
 971
 972                         if (err) {
 973                                 *last_type = seq->type;
 974                                 spin_unlock_bh(&seq->lock);
 975                                 return err;
 976                         }
 977                         spin_unlock_bh(&seq->lock);
 978                 }
 979                 *last_type = 0;
 980         }
 981         return 0;
 982 }
 983
 984 int tipc_nl_name_table_dump(struct sk_buff *skb, struct netlink_callback *cb)
 985 {
 986         int err;
 987         int done = cb->args[3];
 988         u32 last_type = cb->args[0];
 989         u32 last_lower = cb->args[1];
 990         u32 last_publ = cb->args[2];
 991         struct net *net = sock_net(skb->sk);
 992         struct tipc_nl_msg msg;
 993
 994         if (done)
 995                 return 0;
 996
 997         msg.skb = skb;
 998         msg.portid = NETLINK_CB(cb->skb).portid;
 999         msg.seq = cb->nlh->nlmsg_seq;
1000
1001         rcu_read_lock();
1002         err = tipc_nl_seq_list(net, &msg, &last_type, &last_lower, &last_publ);
1003         if (!err) {
1004                 done = 1;
1005         } else if (err != -EMSGSIZE) {
1006                 /* We never set seq or call nl_dump_check_consistent() this
1007                  * means that setting prev_seq here will cause the consistence
1008                  * check to fail in the netlink callback handler. Resulting in
1009                  * the NLMSG_DONE message having the NLM_F_DUMP_INTR flag set if
1010                  * we got an error.
1011                  */
1012                 cb->prev_seq = 1;
1013         }
1014         rcu_read_unlock();
1015
1016         cb->args[0] = last_type;
1017         cb->args[1] = last_lower;
1018         cb->args[2] = last_publ;
1019         cb->args[3] = done;
1020
1021         return skb->len;
1022 }
1023
1024 void tipc_plist_push(struct tipc_plist *pl, u32 port)
1025 {
1026         struct tipc_plist *nl;
1027
1028         if (likely(!pl->port)) {
1029                 pl->port = port;
1030                 return;
1031         }
1032         if (pl->port == port)
1033                 return;
1034         list_for_each_entry(nl, &pl->list, list) {
1035                 if (nl->port == port)
1036                         return;
1037         }
1038         nl = kmalloc(sizeof(*nl), GFP_ATOMIC);
1039         if (nl) {
1040                 nl->port = port;
1041                 list_add(&nl->list, &pl->list);
1042         }
1043 }
1044
1045 u32 tipc_plist_pop(struct tipc_plist *pl)
1046 {
1047         struct tipc_plist *nl;
1048         u32 port = 0;
1049
1050         if (likely(list_empty(&pl->list))) {
1051                 port = pl->port;
1052                 pl->port = 0;
1053                 return port;
1054         }
1055         nl = list_first_entry(&pl->list, typeof(*nl), list);
1056         port = nl->port;
1057         list_del(&nl->list);
1058         kfree(nl);
1059         return port;
1060 }