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