net/cor/common.c

   1 /*
   2  *      Connection oriented routing
   3  *      Copyright (C) 2007-2008 Michael Blizek
   4  *
   5  *      This program is free software; you can redistribute it and/or
   6  *      modify it under the terms of the GNU General Public License
   7  *      as published by the Free Software Foundation; either version 2
   8  *      of the License, or (at your option) any later version.
   9  *
  10  *      This program is distributed in the hope that it will be useful,
  11  *      but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *      GNU General Public License for more details.
  14  *
  15  *      You should have received a copy of the GNU General Public License
  16  *      along with this program; if not, write to the Free Software
  17  *      Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  18  *      02110-1301, USA.
  19  */
  20
  21 #include <linux/mutex.h>
  22
  23 #include "cor.h"
  24
  25 DEFINE_MUTEX(cor_bindnodes);
  26 DEFINE_SPINLOCK(conn_free);
  27
  28 DEFINE_MUTEX(connid_gen);
  29
  30 LIST_HEAD(openports);
  31
  32
  33 struct cell_hdr{
  34         spinlock_t lock;
  35 };
  36
  37 struct kmem_cache *conn_slab;
  38
  39 struct htable connid_table;
  40
  41 struct kmem_cache *bindnode_slab;
  42 struct kmem_cache *connlistener_slab;
  43
  44 /* see cor.h/KP_ACK_CONN */
  45 static const __u32 window_sizes[] = {0,
  46                 64, 68, 73, 77, 82, 88, 93, 99, 106, 113, 120,
  47                 128, 136, 145, 155, 165, 175, 187, 199, 212, 226, 240,
  48                 256, 273, 290, 309, 329, 351, 374, 398, 424, 451, 481,
  49                 512, 545, 581, 619, 659, 702, 747, 796, 848, 903, 961,
  50                 1024, 1091, 1162, 1237, 1318, 1403, 1495, 1592, 1695, 1805,
  51                                 1923,
  52                 2048, 2181, 2323, 2474, 2635, 2806, 2989, 3183, 3390, 3611,
  53                                 3846,
  54                 4096, 4362, 4646, 4948, 5270, 5613, 5978, 6367, 6781, 7222,
  55                                 7692,
  56                 8192, 8725, 9292, 9897, 10540, 11226, 11956, 12734, 13562,
  57                                 14444, 15383,
  58                 16384, 17450, 18585, 19793, 21081, 22452, 23912, 25467, 27124,
  59                                 28888, 30767,
  60                 32768, 34899, 37169, 39587, 42161, 44904, 47824, 50935, 54248,
  61                                 57776, 61534,
  62                 65536, 69799, 74338, 79173, 84323, 89807, 95648, 101870, 108495,
  63                                 115552, 123068,
  64                 131072, 139597, 148677, 158347, 168646, 179615, 191297, 203739,
  65                                 216991, 231104, 246135,
  66                 262144, 279194, 297353, 316693, 337291, 359229, 382594, 407478,
  67                                 433981, 462208, 492270,
  68                 524288, 558388, 594706, 633387, 674583, 718459, 765188, 814957,
  69                                 867962, 924415, 984540,
  70                 1048576, 1116777, 1189413, 1266774, 1349166, 1436917, 1530376,
  71                                 1629913, 1735924, 1848831, 1969081,
  72                 2097152, 2233553, 2378826, 2533547, 2698332, 2873834, 3060752,
  73                                 3259826, 3471849, 3697662, 3938162,
  74                 4194304, 4467106, 4757652, 5067094, 5396664, 5747669, 6121503,
  75                                 6519652, 6943698, 7395323, 7876323,
  76                 8388608, 8934212, 9515303, 10134189, 10793327, 11495337,
  77                                 12243006, 13039305, 13887396, 14790647,
  78                                 15752647,
  79                 16777216, 17868424, 19030606, 20268378, 21586655, 22990674,
  80                                 24486013, 26078610, 27774791, 29581294,
  81                                 31505293,
  82                 33554432, 35736849, 38061212, 40536755, 43173310, 45981349,
  83                                 48972026, 52157220, 55549582, 59162588,
  84                                 63010587,
  85                 67108864, 71473698, 76122425, 81073510, 86346620, 91962698,
  86                                 97944052, 104314440, 111099165, 118325175,
  87                                 126021174,
  88                 134217728, 142947395, 152244850, 162147020, 172693239,
  89                                 183925396, 195888104, 208628880, 222198329,
  90                                 236650351, 252042347,
  91                 268435456, 285894791, 304489699, 324294041, 345386479,
  92                                 367850791, 391776208, 417257759, 444396658,
  93                                 473300701, 504084694,
  94                 536870912, 571789581};
  95
  96 __u8 enc_window(__u32 window_bytes)
  97 {
  98         int i;
  99         BUG_ON(window_sizes[255] != 571789581);
 100         for (i=1;i<256;i++) {
 101                 if (window_sizes[i] > window_bytes)
 102                         break;
 103         }
 104
 105         return (__u8)(i-1); /* round down */
 106 }
 107
 108 __u32 dec_window(__u8 window)
 109 {
 110         BUG_ON(window_sizes[255] != 571789581);
 111         return window_sizes[window];
 112 }
 113
 114 static inline int hdr_size(void)
 115 {
 116         return ((sizeof(struct cell_hdr) + sizeof(void *) - 1) / sizeof(void *)
 117                 ) * sizeof(void *);
 118 }
 119
 120 static inline int elements_per_cell(int cell_size)
 121 {
 122         return (cell_size - hdr_size())/sizeof(void *);
 123 }
 124
 125 static inline struct cell_hdr *cell_addr(struct htable *ht, __u32 id)
 126 {
 127         int idx = (id%ht->htable_size) / (elements_per_cell(ht->cell_size));
 128         return (struct cell_hdr *) (((char *)ht->htable) + ht->cell_size * idx);
 129 }
 130
 131 static inline char **element_addr(struct htable *ht, __u32 id)
 132 {
 133         int idx = (id%ht->htable_size) % (elements_per_cell(ht->cell_size));
 134         return (char **)
 135                         ( ((char *)cell_addr(ht, id)) +
 136                         hdr_size() + idx*sizeof(void *) );
 137 }
 138
 139
 140 static inline char **next_element(struct htable *ht, char *element)
 141 {
 142         return (char **)(element + ht->entry_offset);
 143 }
 144
 145 static inline struct kref *element_kref(struct htable *ht, char *element)
 146 {
 147         return (struct kref *)(element + ht->kref_offset);
 148 }
 149
 150
 151 static inline void unlock_element(struct htable *ht, __u32 key)
 152 {
 153         struct cell_hdr *hdr = cell_addr(ht, key);
 154         spin_unlock( &(hdr->lock) );
 155 }
 156
 157
 158 static char **get_element_nounlock(struct htable *ht, __u32 key,
 159                 void *searcheditem)
 160 {
 161         struct cell_hdr *hdr = cell_addr(ht, key);
 162         char **element = element_addr(ht, key);
 163
 164         BUG_ON(0 == element);
 165
 166         spin_lock( &(hdr->lock) );
 167
 168         while (1) {
 169                 if (*element == 0)
 170                         break;
 171                 if (searcheditem != 0 && ht->matches(*element, searcheditem))
 172                         break;
 173                 element = next_element(ht, *element);
 174         }
 175
 176         return element;
 177 }
 178
 179 char *htable_get(struct htable *ht, __u32 key, void *searcheditem)
 180 {
 181         unsigned long iflags;
 182         char *element;
 183
 184         if (unlikely(ht->htable == 0))
 185                 return 0;
 186
 187         local_irq_save(iflags);
 188         element = *(get_element_nounlock(ht, key, searcheditem));
 189         if (likely(element != 0))
 190                 kref_get(element_kref(ht, element));
 191         unlock_element(ht, key);
 192         local_irq_restore(iflags);
 193
 194         return element;
 195 }
 196
 197 int htable_delete(struct htable *ht, __u32 key,
 198                 void *searcheditem, void (*free) (struct kref *ref))
 199 {
 200         unsigned long iflags;
 201         char **element;
 202         char **next;
 203         int rc = 0;
 204
 205         if (unlikely(ht->htable == 0))
 206                 return 1;
 207
 208         local_irq_save(iflags);
 209
 210         element = get_element_nounlock(ht, key, searcheditem);
 211         BUG_ON(0 == element);
 212
 213         if (unlikely(*element == 0)) {
 214                 /* key not in table */
 215                 rc = 1;
 216                 goto out;
 217         }
 218
 219         next = next_element(ht, *element);
 220         kref_put(element_kref(ht, *element), free);
 221         *element = *next;
 222
 223 out:
 224         unlock_element(ht, key);
 225         local_irq_restore(iflags);
 226
 227         return rc;
 228 }
 229
 230 void htable_insert(struct htable *ht, char *newelement, __u32 key)
 231 {
 232         unsigned long iflags;
 233         char **element;
 234
 235         if (unlikely(ht->htable == 0))
 236                 return;
 237
 238         BUG_ON(*next_element(ht, newelement) != 0);
 239         local_irq_save(iflags);
 240
 241         element = get_element_nounlock(ht, key, 0);
 242
 243         BUG_ON(element == 0);
 244         BUG_ON(*element != 0);
 245
 246         *element = newelement;
 247         kref_get(element_kref(ht, newelement));
 248
 249         unlock_element(ht, key);
 250         local_irq_restore(iflags);
 251 }
 252
 253
 254 void htable_init(struct htable *ht, int (*matches)(void *htentry,
 255                 void *searcheditem), __u32 entry_offset, __u32 kref_offset)
 256 {
 257         int num_cells;
 258         int j;
 259
 260         BUG_ON(0 == ht);
 261
 262         ht->htable = kmalloc(PAGE_SIZE, GFP_KERNEL);
 263         if (unlikely(ht->htable == 0)) {
 264                 printk(KERN_CRIT "cor: error allocating htable (out of "
 265                                 "memory?)");
 266                 return;
 267         }
 268         memset(ht->htable, 0, PAGE_SIZE);
 269         ht->cell_size = 256;
 270
 271         num_cells = PAGE_SIZE/ht->cell_size;
 272
 273         for (j=0;j<num_cells;j++) {
 274                 struct cell_hdr *hdr = (struct cell_hdr *)
 275                                 ( ((char *) ht->htable) + j * ht->cell_size );
 276                 spin_lock_init(&(hdr->lock));
 277         }
 278
 279         ht->htable_size = num_cells * elements_per_cell(ht->cell_size);
 280         ht->num_elements = 0;
 281
 282         ht->matches = matches;
 283         ht->entry_offset = entry_offset;
 284         ht->kref_offset = kref_offset;
 285 }
 286
 287 struct conn *get_conn(__u32 conn_id)
 288 {
 289         return (struct conn *) htable_get(&connid_table, conn_id, &conn_id);
 290 }
 291
 292 static int connid_alloc(struct conn *sconn)
 293 {
 294         __u32 conn_id;
 295         int i;
 296
 297         BUG_ON(sconn->sourcetype != SOURCE_IN);
 298
 299         mutex_lock(&connid_gen);
 300         for(i=0;i<16;i++) {
 301                 struct conn *tmp;
 302
 303                 conn_id = 0;
 304                 get_random_bytes((char *) &conn_id, sizeof(conn_id));
 305
 306                 if (unlikely(conn_id == 0))
 307                         continue;
 308
 309                 tmp = get_conn(conn_id);
 310                 if (unlikely(tmp != 0)) {
 311                         kref_put(&(tmp->ref), free_conn);
 312                         continue;
 313                 }
 314
 315                 goto found;
 316         }
 317         mutex_unlock(&connid_gen);
 318
 319         return 1;
 320
 321 found:
 322         sconn->source.in.conn_id = conn_id;
 323         htable_insert(&connid_table, (char *) sconn, conn_id);
 324         mutex_unlock(&connid_gen);
 325         return 0;
 326 }
 327
 328 void free_conn(struct kref *ref)
 329 {
 330         unsigned long iflags;
 331         struct conn *conn = container_of(ref, struct conn, ref);
 332         struct conn *reversedir = 0;
 333
 334         BUG_ON(atomic_read(&(conn->isreset)) == 0);
 335
 336         spin_lock_irqsave(&conn_free, iflags);
 337
 338         if (conn->reversedir != 0)
 339                 atomic_set(&(conn->reversedir->isreset), 3);
 340
 341         if (atomic_read(&(conn->isreset)) != 3)
 342                 goto out;
 343
 344         if (conn->reversedir != 0) {
 345                 conn->reversedir->reversedir = 0;
 346                 reversedir = conn->reversedir;
 347                 conn->reversedir = 0;
 348         }
 349
 350         if (conn->sourcetype == SOURCE_IN) {
 351                 kref_put(&(conn->source.in.nb->ref), neighbor_free);
 352                 conn->source.in.nb = 0;
 353         }
 354
 355         if (conn->targettype == TARGET_OUT) {
 356                 kref_put(&(conn->target.out.nb->ref), neighbor_free);
 357                 conn->target.out.nb = 0;
 358         }
 359
 360         databuf_free(&(conn->buf));
 361
 362         kmem_cache_free(conn_slab, conn);
 363
 364 out:
 365         spin_unlock_irqrestore(&conn_free, iflags);
 366
 367         if (reversedir != 0)
 368                 free_conn(&(reversedir->ref));
 369 }
 370
 371 /*
 372  * rconn ==> the connection we received the commend from
 373  * ==> init rconn->target.out + rconn->reversedir->source.in
 374  * rc == 0 ==> ok
 375  * rc == 1 ==> connid allocation failed
 376  *
 377  * NOTE: call to this func *must* be protected by rcv_lock
 378  */
 379 int conn_init_out(struct conn *rconn, struct neighbor *nb)
 380 {
 381         struct conn *sconn = rconn->reversedir;
 382         __u32 stall_timeout_ms = rconn->target.unconnected.stall_timeout_ms;
 383
 384         BUG_ON(rconn->targettype != TARGET_UNCONNECTED);
 385         BUG_ON(sconn == 0);
 386         BUG_ON(sconn->sourcetype != SOURCE_NONE);
 387
 388         memset(&(rconn->target.out), 0, sizeof(rconn->target.out));
 389         memset(&(sconn->source.in), 0, sizeof(sconn->source.in));
 390
 391         rconn->targettype = TARGET_OUT;
 392         sconn->sourcetype = SOURCE_IN;
 393
 394         rconn->target.out.nb = nb;
 395         sconn->source.in.nb = nb;
 396
 397         /* neighbor pointer */
 398         kref_get(&(nb->ref));
 399         kref_get(&(nb->ref));
 400
 401         rconn->target.out.stall_timeout_ms = stall_timeout_ms;
 402         skb_queue_head_init(&(sconn->source.in.reorder_queue));
 403
 404         if (unlikely(connid_alloc(sconn)))
 405                 return 1;
 406
 407         INIT_LIST_HEAD(&(rconn->target.out.retrans_list));
 408
 409         mutex_lock(&(nb->conn_list_lock));
 410         list_add_tail(&(sconn->source.in.nb_list), &(nb->rcv_conn_list));
 411         list_add_tail(&(rconn->target.out.nb_list), &(nb->snd_conn_list));
 412         mutex_unlock(&(nb->conn_list_lock));
 413
 414         /* neighbor lists */
 415         kref_get(&(rconn->ref));
 416         kref_get(&(sconn->ref));
 417
 418         return 0;
 419 }
 420
 421 void conn_init_sock_source(struct conn *conn)
 422 {
 423         BUG_ON(conn == 0);
 424         conn->sourcetype = SOURCE_SOCK;
 425         memset(&(conn->source.sock), 0, sizeof(conn->source.sock));
 426         init_waitqueue_head(&(conn->source.sock.wait));
 427 }
 428
 429 void conn_init_sock_target(struct conn *conn)
 430 {
 431         BUG_ON(conn == 0);
 432         conn->targettype = TARGET_SOCK;
 433         memset(&(conn->target.sock), 0, sizeof(conn->target.sock));
 434         init_waitqueue_head(&(conn->target.sock.wait));
 435 }
 436
 437 struct conn* alloc_conn(gfp_t allocflags)
 438 {
 439         struct conn *rconn = 0;
 440         struct conn *sconn = 0;
 441
 442         rconn = kmem_cache_alloc(conn_slab, allocflags);
 443         if (unlikely(rconn == 0))
 444                 goto out_err0;
 445
 446         sconn = kmem_cache_alloc(conn_slab, allocflags);
 447         if (unlikely(sconn == 0))
 448                 goto out_err1;
 449
 450         memset(rconn, 0, sizeof(struct conn));
 451         memset(sconn, 0, sizeof(struct conn));
 452
 453         rconn->reversedir = sconn;
 454         sconn->reversedir = rconn;
 455
 456         kref_init(&(rconn->ref));
 457         kref_init(&(sconn->ref));
 458
 459         rconn->sockstate = SOCKSTATE_CONN;
 460         sconn->sockstate = SOCKSTATE_CONN;
 461
 462         rconn->sourcetype = SOURCE_NONE;
 463         sconn->sourcetype = SOURCE_NONE;
 464         rconn->targettype = TARGET_UNCONNECTED;
 465         sconn->targettype = TARGET_UNCONNECTED;
 466
 467         mutex_init(&(rconn->rcv_lock));
 468         mutex_init(&(sconn->rcv_lock));
 469
 470         rconn->target.unconnected.stall_timeout_ms =
 471                         CONN_STALL_DEFAULT_TIMEOUT_MS;
 472         sconn->target.unconnected.stall_timeout_ms =
 473                         CONN_STALL_DEFAULT_TIMEOUT_MS;
 474
 475         databuf_init(&(rconn->buf));
 476         databuf_init(&(sconn->buf));
 477
 478         return rconn;
 479
 480 out_err1:
 481         kmem_cache_free(conn_slab, rconn);
 482 out_err0:
 483         return 0;
 484 }
 485
 486 static struct connlistener *get_connlistener(__be64 port)
 487 {
 488         struct list_head *curr = openports.next;
 489
 490         while (curr != &openports) {
 491                 struct bindnode *currnode = ((struct bindnode *)
 492                         (((char *)curr) - offsetof(struct bindnode, lh)));
 493                 if (currnode->port == port) {
 494                         BUG_ON(currnode->owner == 0);
 495                         return currnode->owner;
 496                 }
 497
 498                 curr = curr->next;
 499         }
 500
 501         return 0;
 502 }
 503
 504 void close_port(struct connlistener *listener)
 505 {
 506         mutex_lock(&cor_bindnodes);
 507
 508         if (listener->bn != 0) {
 509                 list_del(&(listener->bn->lh));
 510                 kmem_cache_free(bindnode_slab, listener->bn);
 511                 listener->bn = 0;
 512         }
 513
 514         while (list_empty(&(listener->conn_queue)) == 0) {
 515                 struct conn *rconn = container_of(listener->conn_queue.next,
 516                                 struct conn, source.sock.cl_list);
 517                 list_del(&(rconn->source.sock.cl_list));
 518                 atomic_cmpxchg(&(rconn->reversedir->isreset), 0, 1);
 519                 reset_conn(rconn);
 520                 kref_put(&(rconn->ref), free_conn);
 521         }
 522
 523         kmem_cache_free(connlistener_slab, listener);
 524
 525         mutex_unlock(&cor_bindnodes);
 526 }
 527
 528 struct connlistener *open_port(__be64 port)
 529 {
 530
 531         struct bindnode *bn = 0;
 532         struct connlistener *listener = 0;
 533
 534         mutex_lock(&cor_bindnodes);
 535         if (get_connlistener(port) != 0)
 536                 goto out;
 537
 538
 539         bn = kmem_cache_alloc(bindnode_slab, GFP_KERNEL);
 540         listener = kmem_cache_alloc(connlistener_slab, GFP_KERNEL);
 541
 542         memset(bn, 0, sizeof(struct bindnode));
 543         memset(listener, 0, sizeof(struct connlistener));
 544
 545         bn->owner = listener;
 546         bn->port = port;
 547
 548         /* kref is not actually used */
 549         listener->sockstate = SOCKSTATE_LISTENER;
 550         listener->bn = bn;
 551         mutex_init(&(listener->lock));
 552         INIT_LIST_HEAD(&(listener->conn_queue));
 553         init_waitqueue_head(&(listener->wait));
 554
 555         list_add_tail((struct list_head *) &(bn->lh), &openports);
 556
 557 out:
 558         mutex_unlock(&cor_bindnodes);
 559
 560         return listener;
 561 }
 562
 563 /**
 564  * rc == 0 connected
 565  * rc == 2 port not open
 566  * rc == 3 listener queue full
 567  */
 568 int connect_port(struct conn *rconn, __be64 port)
 569 {
 570
 571         struct connlistener *listener;
 572         int rc = 0;
 573
 574         mutex_lock(&cor_bindnodes);
 575
 576         listener = get_connlistener(port);
 577         if (listener == 0) {
 578                 rc = 2;
 579                 goto out;
 580         }
 581
 582         mutex_lock(&(listener->lock));
 583
 584         if (unlikely(listener->queue_len >= listener->queue_maxlen)) {
 585                 if (listener->queue_maxlen <= 0)
 586                         rc = 2;
 587                 else
 588                         rc = 3;
 589
 590                 goto out2;
 591         }
 592
 593         kref_get(&(rconn->reversedir->ref));
 594
 595         conn_init_sock_target(rconn);
 596         conn_init_sock_source(rconn->reversedir);
 597
 598         list_add_tail(&(rconn->reversedir->source.sock.cl_list),
 599                         &(listener->conn_queue));
 600         listener->queue_len++;
 601         wake_up_interruptible(&(listener->wait));
 602
 603 out2:
 604         mutex_unlock(&(listener->lock));
 605
 606 out:
 607         mutex_unlock(&cor_bindnodes);
 608         return rc;
 609 }
 610
 611 /**
 612  * rc == 0 connected
 613  * rc == 2 addrtype not found
 614  * rc == 3 addr not found
 615  * rc == 4 ==> connid allocation failed
 616  * rc == 5 ==> control msg alloc failed
 617  */
 618 int connect_neigh(struct conn *rconn,
 619                 __u16 addrtypelen, __u8 *addrtype,
 620                 __u16 addrlen, __u8 *addr)
 621 {
 622         int rc = 0;
 623         struct control_msg_out *cm;
 624         struct neighbor *nb = find_neigh(addrtypelen, addrtype, addrlen, addr);
 625         if (nb == 0)
 626                 return 3;
 627         if (unlikely(conn_init_out(rconn, nb))) {
 628                 rc = 4;
 629                 goto neigh_kref;
 630         }
 631
 632         cm = alloc_control_msg();
 633         if (unlikely(cm == 0)) {
 634                 rc = 5;
 635                 goto neigh_kref;
 636         }
 637
 638         send_connect_nb(cm, nb, rconn->reversedir->source.in.conn_id);
 639
 640 neigh_kref:
 641         kref_put(&(nb->ref), neighbor_free);
 642
 643         return rc;
 644 }
 645
 646 static int _reset_conn(struct conn *conn)
 647 {
 648         /*
 649          * aktive conns have an additional ref to make sure that they are not
 650          * freed when only one direction is referenced by the connid hashtable
 651          */
 652         int krefput = 1;
 653         int isreset = atomic_cmpxchg(&(conn->isreset), 0, 2);
 654         if (isreset == 1)
 655                 isreset = atomic_cmpxchg(&(conn->isreset), 1, 2);
 656
 657         if (isreset == 2 || isreset == 3)
 658                 return 0;
 659
 660         /* lock sourcetype/targettype */
 661         mutex_lock(&(conn->rcv_lock));
 662
 663         if (conn->sourcetype == SOURCE_IN) {
 664                 mutex_lock(&(conn->source.in.nb->conn_list_lock));
 665                 list_del(&(conn->source.in.nb_list));
 666                 mutex_unlock(&(conn->source.in.nb->conn_list_lock));
 667
 668                 krefput++;
 669
 670                 if (conn->source.in.conn_id != 0) {
 671                         if (htable_delete(&connid_table,
 672                                         conn->source.in.conn_id,
 673                                         &(conn->source.in.conn_id), free_conn)){
 674                                 printk(KERN_ERR "error in _reset_conn: "
 675                                                 "htable_delete failed");
 676                         }
 677                 }
 678         } else if (conn->sourcetype == SOURCE_SOCK) {
 679                 wake_up_interruptible(&(conn->source.sock.wait));
 680         }
 681
 682         if (conn->targettype == TARGET_OUT) {
 683                 mutex_lock(&(conn->target.out.nb->conn_list_lock));
 684                 list_del(&(conn->target.out.nb_list));
 685                 mutex_unlock(&(conn->target.out.nb->conn_list_lock));
 686
 687                 krefput++;
 688
 689                 if (isreset == 0 && conn->target.out.conn_id != 0) {
 690                         struct control_msg_out *cm = alloc_control_msg();
 691                         if (likely(cm != 0))
 692                                 send_reset_conn(cm, conn->target.out.nb,
 693                                                 conn->target.out.conn_id);
 694                 }
 695
 696                 cancel_retrans(conn);
 697         } else if (conn->targettype == TARGET_SOCK) {
 698                 wake_up_interruptible(&(conn->target.sock.wait));
 699         }
 700
 701         databuf_free(&(conn->buf));
 702
 703         mutex_unlock(&(conn->rcv_lock));
 704
 705         return krefput;
 706 }
 707
 708 /* warning: do not hold the rcv_lock while calling this! */
 709 void reset_conn(struct conn *conn)
 710 {
 711         int put1 = _reset_conn(conn);
 712         int put2 = _reset_conn(conn->reversedir);
 713
 714         /*free_conn may not be called, before both _reset_conn have finished */
 715         while (put1 > 0) {
 716                 kref_put(&(conn->ref), free_conn);
 717                 put1--;
 718         }
 719
 720         while (put2 > 0) {
 721                 kref_put(&(conn->reversedir->ref), free_conn);
 722                 put2--;
 723         }
 724 }
 725
 726 static int matches_connid_in(void *htentry, void *searcheditem)
 727 {
 728         struct conn *conn = (struct conn *) htentry;
 729         __u32 conn_id = *((__u32 *) searcheditem);
 730         BUG_ON(conn->sourcetype != SOURCE_IN);
 731         return (conn->source.in.conn_id == conn_id);
 732 }
 733
 734 static int __init cor_common_init(void)
 735 {
 736         int rc;
 737
 738         struct conn c;
 739
 740         printk(KERN_ERR "sizeof conn: %d", sizeof(c));
 741         printk(KERN_ERR "  conn.source: %d", sizeof(c.source));
 742         printk(KERN_ERR "  conn.target: %d", sizeof(c.target));
 743         printk(KERN_ERR "  conn.target.out: %d", sizeof(c.target.out));
 744         printk(KERN_ERR "  conn.buf: %d", sizeof(c.buf));
 745
 746         printk(KERN_ERR "  mutex: %d", sizeof(struct mutex));
 747         printk(KERN_ERR "  spinlock: %d", sizeof(spinlock_t));
 748         printk(KERN_ERR "  kref: %d", sizeof(struct kref));
 749
 750
 751         conn_slab = kmem_cache_create("cor_conn", sizeof(struct conn), 8, 0, 0);
 752         htable_init(&connid_table, matches_connid_in,
 753                         offsetof(struct conn, source.in.htab_entry),
 754                         offsetof(struct conn, ref));
 755
 756         bindnode_slab = kmem_cache_create("cor_bindnode",
 757                         sizeof(struct bindnode), 8, 0, 0);
 758         connlistener_slab = kmem_cache_create("cor_connlistener",
 759                         sizeof(struct connlistener), 8, 0, 0);
 760
 761         forward_init();
 762
 763         cor_kgen_init();
 764
 765         rc = cor_snd_init();
 766         if (unlikely(rc != 0))
 767                 return rc;
 768
 769         rc = cor_neighbor_init();
 770         if (unlikely(rc != 0))
 771                 return rc;
 772
 773         rc = cor_rcv_init();
 774         if (unlikely(rc != 0))
 775                 return rc;
 776
 777         return 0;
 778 }
 779
 780 module_init(cor_common_init);
 781 MODULE_LICENSE("GPL");