mirror of
https://github.com/samba-team/samba.git
synced 2024-12-22 13:34:15 +03:00
5e5701a7b8
- shutdown sockets before calling shutdown script (This used to be ctdb commit c5e099feef94a014a77742b6cc1d0afe78ef9da9)
142 lines
3.5 KiB
C
142 lines
3.5 KiB
C
/*
|
|
ctdb over TCP
|
|
|
|
Copyright (C) Andrew Tridgell 2006
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*/
|
|
|
|
#include "includes.h"
|
|
#include "lib/tdb/include/tdb.h"
|
|
#include "lib/events/events.h"
|
|
#include "system/network.h"
|
|
#include "system/filesys.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 *ctcp = talloc_get_type(node->ctdb->private_data,
|
|
struct ctdb_tcp);
|
|
struct ctdb_tcp_node *tnode;
|
|
tnode = talloc_zero(ctcp, struct ctdb_tcp_node);
|
|
CTDB_NO_MEMORY(node->ctdb, tnode);
|
|
|
|
tnode->fd = -1;
|
|
node->private_data = tnode;
|
|
|
|
tnode->queue = ctdb_queue_setup(node->ctdb, ctcp, tnode->fd, CTDB_TCP_ALIGNMENT,
|
|
ctdb_tcp_tnode_cb, node);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
initialise transport structures
|
|
*/
|
|
static int ctdb_tcp_initialise(struct ctdb_context *ctdb)
|
|
{
|
|
int i;
|
|
|
|
/* listen on our own address */
|
|
if (ctdb_tcp_listen(ctdb) != 0) return -1;
|
|
|
|
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;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
start the protocol going
|
|
*/
|
|
static int ctdb_tcp_start(struct ctdb_context *ctdb)
|
|
{
|
|
int i;
|
|
|
|
/* startup connections to the other servers - will happen on
|
|
next event loop */
|
|
for (i=0;i<ctdb->num_nodes;i++) {
|
|
struct ctdb_node *node = *(ctdb->nodes + i);
|
|
struct ctdb_tcp_node *tnode = talloc_get_type(
|
|
node->private_data, struct ctdb_tcp_node);
|
|
if (!(ctdb->flags & CTDB_FLAG_SELF_CONNECT) &&
|
|
ctdb_same_address(&ctdb->address, &node->address)) continue;
|
|
event_add_timed(ctdb->ev, tnode, timeval_zero(),
|
|
ctdb_tcp_node_connect, node);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
/*
|
|
shutdown the transport
|
|
*/
|
|
static void ctdb_tcp_shutdown(struct ctdb_context *ctdb)
|
|
{
|
|
struct ctdb_tcp *ctcp = talloc_get_type(ctdb->private_data,
|
|
struct ctdb_tcp);
|
|
talloc_free(ctcp);
|
|
ctdb->private_data = NULL;
|
|
}
|
|
|
|
|
|
/*
|
|
transport packet allocator - allows transport to control memory for packets
|
|
*/
|
|
static void *ctdb_tcp_allocate_pkt(TALLOC_CTX *mem_ctx, size_t size)
|
|
{
|
|
/* 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 = {
|
|
.initialise = ctdb_tcp_initialise,
|
|
.start = ctdb_tcp_start,
|
|
.queue_pkt = ctdb_tcp_queue_pkt,
|
|
.add_node = ctdb_tcp_add_node,
|
|
.allocate_pkt = ctdb_tcp_allocate_pkt,
|
|
.shutdown = ctdb_tcp_shutdown,
|
|
};
|
|
|
|
/*
|
|
initialise tcp portion of ctdb
|
|
*/
|
|
int ctdb_tcp_init(struct ctdb_context *ctdb)
|
|
{
|
|
struct ctdb_tcp *ctcp;
|
|
ctcp = talloc_zero(ctdb, struct ctdb_tcp);
|
|
CTDB_NO_MEMORY(ctdb, ctcp);
|
|
|
|
ctcp->listen_fd = -1;
|
|
ctdb->private_data = ctcp;
|
|
ctdb->methods = &ctdb_tcp_methods;
|
|
return 0;
|
|
}
|
|
|