ctdb/tcp/tcp_init.c

   1 /*
   2    ctdb over TCP
   3
   4    Copyright (C) Andrew Tridgell  2006
   5
   6    This library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2 of the License, or (at your option) any later version.
  10
  11    This library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with this library; if not, write to the Free Software
  18    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 #include "includes.h"
  22 #include "lib/events/events.h"
  23 #include "system/network.h"
  24 #include "system/filesys.h"
  25 #include "ctdb_private.h"
  26 #include "ctdb_tcp.h"
  27
  28 /*
  29   start the protocol going
  30 */
  31 int ctdb_tcp_start(struct ctdb_context *ctdb)
  32 {
  33         int i;
  34
  35         /* listen on our own address */
  36         if (ctdb_tcp_listen(ctdb) != 0) return -1;
  37
  38         /* startup connections to the other servers - will happen on
  39            next event loop */
  40         for (i=0;i<ctdb->num_nodes;i++) {
  41                 struct ctdb_node *node = *(ctdb->nodes + i);
  42                 if (!(ctdb->flags & CTDB_FLAG_SELF_CONNECT) &&
  43                     ctdb_same_address(&ctdb->address, &node->address)) continue;
  44                 event_add_timed(ctdb->ev, node, timeval_zero(),
  45                                 ctdb_tcp_node_connect, node);
  46         }
  47
  48         return 0;
  49 }
  50
  51
  52 /*
  53   initialise tcp portion of a ctdb node
  54 */
  55 int ctdb_tcp_add_node(struct ctdb_node *node)
  56 {
  57         struct ctdb_tcp_node *tnode;
  58         tnode = talloc_zero(node, struct ctdb_tcp_node);
  59         CTDB_NO_MEMORY(node->ctdb, tnode);
  60
  61         tnode->fd = -1;
  62         node->private = tnode;
  63         return 0;
  64 }
  65
  66
  67 /*
  68   transport packet allocator - allows transport to control memory for packets
  69 */
  70 void *ctdb_tcp_allocate_pkt(struct ctdb_node *node, size_t size)
  71 {
  72         /* tcp transport needs to round to 8 byte alignment to ensure
  73            that we can use a length header and 64 bit elements in
  74            structures */
  75         size = (size+(CTDB_TCP_ALIGNMENT-1)) & ~(CTDB_TCP_ALIGNMENT-1);
  76         return talloc_size(node, size);
  77 }
  78
  79 void ctdb_tcp_dealloc_pkt(struct ctdb_node *node, void *buf)
  80 {
  81         talloc_free(buf);
  82 }
  83
  84 int ctdb_tcp_stop(struct ctdb_context *ctdb)
  85 {
  86         return 0;
  87 }
  88
  89 static const struct ctdb_methods ctdb_tcp_methods = {
  90         .start     = ctdb_tcp_start,
  91         .add_node  = ctdb_tcp_add_node,
  92         .queue_pkt = ctdb_tcp_queue_pkt,
  93         .allocate_pkt = ctdb_tcp_allocate_pkt,
  94         .dealloc_pkt = ctdb_tcp_dealloc_pkt
  95 };
  96
  97 /*
  98   initialise tcp portion of ctdb
  99 */
 100 int ctdb_tcp_init(struct ctdb_context *ctdb)
 101 {
 102         struct ctdb_tcp *ctcp;
 103         ctcp = talloc_zero(ctdb, struct ctdb_tcp);
 104         CTDB_NO_MEMORY(ctdb, ctcp);
 105
 106         ctcp->listen_fd = -1;
 107         ctdb->private = ctcp;
 108         ctdb->methods = &ctdb_tcp_methods;
 109         return 0;
 110 }
 111