4 Copyright (C) Andrew Tridgell 2006
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 3 of the License, or
9 (at your option) any later version.
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.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "system/network.h"
23 #include "system/filesys.h"
24 #include "../include/ctdb_private.h"
27 static int tnode_destructor(struct ctdb_tcp_node *tnode)
29 // struct ctdb_node *node = talloc_find_parent_bytype(tnode, struct ctdb_node);
31 if (tnode->fd != -1) {
40 initialise tcp portion of a ctdb node
42 static int ctdb_tcp_add_node(struct ctdb_node *node)
44 struct ctdb_tcp_node *tnode;
45 tnode = talloc_zero(node, struct ctdb_tcp_node);
46 CTDB_NO_MEMORY(node->ctdb, tnode);
49 node->private_data = tnode;
50 talloc_set_destructor(tnode, tnode_destructor);
52 tnode->out_queue = ctdb_queue_setup(node->ctdb, node, tnode->fd, CTDB_TCP_ALIGNMENT,
53 ctdb_tcp_tnode_cb, node, "to-node-%s", node->name);
59 initialise transport structures
61 static int ctdb_tcp_initialise(struct ctdb_context *ctdb)
65 /* listen on our own address */
66 if (ctdb_tcp_listen(ctdb) != 0) {
67 DEBUG(DEBUG_CRIT, (__location__ " Failed to start listening on the CTDB socket\n"));
71 for (i=0; i < ctdb->num_nodes; i++) {
72 if (ctdb->nodes[i]->flags & NODE_FLAGS_DELETED) {
75 if (ctdb_tcp_add_node(ctdb->nodes[i]) != 0) {
76 DEBUG(DEBUG_CRIT, ("methods->add_node failed at %d\n", i));
85 start the protocol going
87 static int ctdb_tcp_connect_node(struct ctdb_node *node)
89 struct ctdb_context *ctdb = node->ctdb;
90 struct ctdb_tcp_node *tnode = talloc_get_type(
91 node->private_data, struct ctdb_tcp_node);
93 /* startup connection to the other server - will happen on
95 if (!ctdb_same_address(&ctdb->address, &node->address)) {
96 tnode->connect_te = event_add_timed(ctdb->ev, tnode,
98 ctdb_tcp_node_connect, node);
105 shutdown and try to restart a connection to a node after it has been
108 static void ctdb_tcp_restart(struct ctdb_node *node)
110 struct ctdb_tcp_node *tnode = talloc_get_type(
111 node->private_data, struct ctdb_tcp_node);
113 DEBUG(DEBUG_NOTICE,("Tearing down connection to dead node :%d\n", node->pnn));
115 ctdb_tcp_stop_connection(node);
117 tnode->connect_te = event_add_timed(node->ctdb->ev, tnode, timeval_zero(),
118 ctdb_tcp_node_connect, node);
123 shutdown the transport
125 static void ctdb_tcp_shutdown(struct ctdb_context *ctdb)
127 struct ctdb_tcp *ctcp = talloc_get_type(ctdb->private_data,
130 ctdb->private_data = NULL;
136 static int ctdb_tcp_start(struct ctdb_context *ctdb)
140 for (i=0; i < ctdb->num_nodes; i++) {
141 if (ctdb->nodes[i]->flags & NODE_FLAGS_DELETED) {
144 ctdb_tcp_connect_node(ctdb->nodes[i]);
152 transport packet allocator - allows transport to control memory for packets
154 static void *ctdb_tcp_allocate_pkt(TALLOC_CTX *mem_ctx, size_t size)
156 /* tcp transport needs to round to 8 byte alignment to ensure
157 that we can use a length header and 64 bit elements in
159 size = (size+(CTDB_TCP_ALIGNMENT-1)) & ~(CTDB_TCP_ALIGNMENT-1);
160 return talloc_size(mem_ctx, size);
164 static const struct ctdb_methods ctdb_tcp_methods = {
165 .initialise = ctdb_tcp_initialise,
166 .start = ctdb_tcp_start,
167 .queue_pkt = ctdb_tcp_queue_pkt,
168 .add_node = ctdb_tcp_add_node,
169 .connect_node = ctdb_tcp_connect_node,
170 .allocate_pkt = ctdb_tcp_allocate_pkt,
171 .shutdown = ctdb_tcp_shutdown,
172 .restart = ctdb_tcp_restart,
175 static int tcp_ctcp_destructor(struct ctdb_tcp *ctcp)
177 ctcp->ctdb->private_data = NULL;
178 ctcp->ctdb->methods = NULL;
185 initialise tcp portion of ctdb
187 int ctdb_tcp_init(struct ctdb_context *ctdb)
189 struct ctdb_tcp *ctcp;
190 ctcp = talloc_zero(ctdb, struct ctdb_tcp);
191 CTDB_NO_MEMORY(ctdb, ctcp);
193 ctcp->listen_fd = -1;
195 ctdb->private_data = ctcp;
196 ctdb->methods = &ctdb_tcp_methods;
198 talloc_set_destructor(ctcp, tcp_ctcp_destructor);