merge from peter
[vlendec/samba-autobuild/.git] / ctdb / tcp / tcp_init.c
index b3378677035fd4e10b9ccf6433aeeb6cd3869ea4..35b43bd5e2969a77772e67a0e7dac650972cb583 100644 (file)
 */
 
 #include "includes.h"
+#include "lib/tdb/include/tdb.h"
 #include "lib/events/events.h"
 #include "system/network.h"
 #include "system/filesys.h"
-#include "ctdb_private.h"
+#include "../include/ctdb_private.h"
 #include "ctdb_tcp.h"
 
+
+/*
+  initialise tcp portion of a ctdb node 
+*/
+static int ctdb_tcp_add_node(struct ctdb_node *node)
+{
+       struct ctdb_tcp_node *tnode;
+       tnode = talloc_zero(node, struct ctdb_tcp_node);
+       CTDB_NO_MEMORY(node->ctdb, tnode);
+
+       tnode->fd = -1;
+       node->private_data = tnode;
+
+       tnode->queue = ctdb_queue_setup(node->ctdb, node, tnode->fd, CTDB_TCP_ALIGNMENT,
+                                       ctdb_tcp_tnode_cb, node);
+       
+       return 0;
+}
+
 /*
   start the protocol going
 */
-int ctdb_tcp_start(struct ctdb_context *ctdb)
+static int ctdb_tcp_start(struct ctdb_context *ctdb)
 {
        int i;
 
+       for (i=0; i<ctdb->num_nodes; i++) {
+               if (ctdb_tcp_add_node(ctdb->nodes[i]) != 0) {
+                       DEBUG(0, ("methods->add_node failed at %d\n", i));
+                       return -1;
+               }
+       }
+       
        /* listen on our own address */
        if (ctdb_tcp_listen(ctdb) != 0) return -1;
 
@@ -50,24 +77,23 @@ int ctdb_tcp_start(struct ctdb_context *ctdb)
 
 
 /*
-  initialise tcp portion of a ctdb node 
+  transport packet allocator - allows transport to control memory for packets
 */
-int ctdb_tcp_add_node(struct ctdb_node *node)
+static void *ctdb_tcp_allocate_pkt(TALLOC_CTX *mem_ctx, size_t size)
 {
-       struct ctdb_tcp_node *tnode;
-       tnode = talloc_zero(node, struct ctdb_tcp_node);
-       CTDB_NO_MEMORY(node->ctdb, tnode);
-
-       tnode->fd = -1;
-       node->private = tnode;
-       return 0;
+       /* tcp transport needs to round to 8 byte alignment to ensure
+          that we can use a length header and 64 bit elements in
+          structures */
+       size = (size+(CTDB_TCP_ALIGNMENT-1)) & ~(CTDB_TCP_ALIGNMENT-1);
+       return talloc_size(mem_ctx, size);
 }
 
 
 static const struct ctdb_methods ctdb_tcp_methods = {
-       .start     = ctdb_tcp_start,
-       .add_node  = ctdb_tcp_add_node,
-       .queue_pkt = ctdb_tcp_queue_pkt
+       .start        = ctdb_tcp_start,
+       .queue_pkt    = ctdb_tcp_queue_pkt,
+       .add_node     = ctdb_tcp_add_node,
+       .allocate_pkt = ctdb_tcp_allocate_pkt
 };
 
 /*
@@ -80,7 +106,7 @@ int ctdb_tcp_init(struct ctdb_context *ctdb)
        CTDB_NO_MEMORY(ctdb, ctcp);
 
        ctcp->listen_fd = -1;
-       ctdb->private = ctcp;
+       ctdb->private_data = ctcp;
        ctdb->methods = &ctdb_tcp_methods;
        return 0;
 }