7e75195302853ee4b4abf75ff61e0237591b0e14
[vlendec/samba-autobuild/.git] / ctdb / tcp / tcp_init.c
1 /* 
2    ctdb over TCP
3
4    Copyright (C) Andrew Tridgell  2006
5
6    This program is free software; you can redistribute it and/or modify
7    it under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 2 of the License, or
9    (at your option) any later version.
10    
11    This program 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
14    GNU General Public License for more details.
15    
16    You should have received a copy of the GNU General Public License
17    along with this program; if not, write to the Free Software
18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19 */
20
21 #include "includes.h"
22 #include "lib/tdb/include/tdb.h"
23 #include "lib/events/events.h"
24 #include "system/network.h"
25 #include "system/filesys.h"
26 #include "../include/ctdb_private.h"
27 #include "ctdb_tcp.h"
28
29
30 /*
31   initialise tcp portion of a ctdb node 
32 */
33 static int ctdb_tcp_add_node(struct ctdb_node *node)
34 {
35         struct ctdb_tcp *ctcp = talloc_get_type(node->ctdb->private_data,
36                                                 struct ctdb_tcp);
37         struct ctdb_tcp_node *tnode;
38         tnode = talloc_zero(ctcp, struct ctdb_tcp_node);
39         CTDB_NO_MEMORY(node->ctdb, tnode);
40
41         tnode->fd = -1;
42         node->private_data = tnode;
43
44         tnode->queue = ctdb_queue_setup(node->ctdb, ctcp, tnode->fd, CTDB_TCP_ALIGNMENT,
45                                         ctdb_tcp_tnode_cb, node);
46         
47         return 0;
48 }
49
50 /*
51   initialise transport structures
52 */
53 static int ctdb_tcp_initialise(struct ctdb_context *ctdb)
54 {
55         int i;
56
57         /* listen on our own address */
58         if (ctdb_tcp_listen(ctdb) != 0) return -1;
59
60         for (i=0; i<ctdb->num_nodes; i++) {
61                 if (ctdb_tcp_add_node(ctdb->nodes[i]) != 0) {
62                         DEBUG(0, ("methods->add_node failed at %d\n", i));
63                         return -1;
64                 }
65         }
66         
67         return 0;
68 }
69
70 /*
71   start the protocol going
72 */
73 static int ctdb_tcp_start(struct ctdb_context *ctdb)
74 {
75         int i;
76
77         /* startup connections to the other servers - will happen on
78            next event loop */
79         for (i=0;i<ctdb->num_nodes;i++) {
80                 struct ctdb_node *node = *(ctdb->nodes + i);
81                 struct ctdb_tcp_node *tnode = talloc_get_type(
82                         node->private_data, struct ctdb_tcp_node);
83                 if (!(ctdb->flags & CTDB_FLAG_SELF_CONNECT) &&
84                     ctdb_same_address(&ctdb->address, &node->address)) continue;
85                 event_add_timed(ctdb->ev, tnode, timeval_zero(), 
86                                 ctdb_tcp_node_connect, node);
87         }
88
89         return 0;
90 }
91
92
93 /*
94   shutdown the transport
95 */
96 static void ctdb_tcp_shutdown(struct ctdb_context *ctdb)
97 {
98         struct ctdb_tcp *ctcp = talloc_get_type(ctdb->private_data,
99                                                 struct ctdb_tcp);
100         talloc_free(ctcp);
101         ctdb->private_data = NULL;
102 }
103
104
105 /*
106   transport packet allocator - allows transport to control memory for packets
107 */
108 static void *ctdb_tcp_allocate_pkt(TALLOC_CTX *mem_ctx, size_t size)
109 {
110         /* tcp transport needs to round to 8 byte alignment to ensure
111            that we can use a length header and 64 bit elements in
112            structures */
113         size = (size+(CTDB_TCP_ALIGNMENT-1)) & ~(CTDB_TCP_ALIGNMENT-1);
114         return talloc_size(mem_ctx, size);
115 }
116
117
118 static const struct ctdb_methods ctdb_tcp_methods = {
119         .initialise   = ctdb_tcp_initialise,
120         .start        = ctdb_tcp_start,
121         .queue_pkt    = ctdb_tcp_queue_pkt,
122         .add_node     = ctdb_tcp_add_node,
123         .allocate_pkt = ctdb_tcp_allocate_pkt,
124         .shutdown     = ctdb_tcp_shutdown,
125 };
126
127 /*
128   initialise tcp portion of ctdb 
129 */
130 int ctdb_tcp_init(struct ctdb_context *ctdb)
131 {
132         struct ctdb_tcp *ctcp;
133         ctcp = talloc_zero(ctdb, struct ctdb_tcp);
134         CTDB_NO_MEMORY(ctdb, ctcp);
135
136         ctcp->listen_fd = -1;
137         ctdb->private_data = ctcp;
138         ctdb->methods = &ctdb_tcp_methods;
139         return 0;
140 }
141