diff options
Diffstat (limited to 'ctdb/utils/ipmux/ipmux.c')
| -rw-r--r-- | ctdb/utils/ipmux/ipmux.c | 268 |
1 files changed, 0 insertions, 268 deletions
diff --git a/ctdb/utils/ipmux/ipmux.c b/ctdb/utils/ipmux/ipmux.c deleted file mode 100644 index 598e9dcb6f2..00000000000 --- a/ctdb/utils/ipmux/ipmux.c +++ /dev/null @@ -1,268 +0,0 @@ -/* - simple ip multiplexer - - Copyright (C) Ronnie Sahlberg 2007 - - 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 3 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, see <http://www.gnu.org/licenses/>. -*/ - -#include "includes.h" -#include "lib/events/events.h" -#include "system/filesys.h" -#include "system/network.h" -#include "popt.h" -#include "cmdline.h" -#include "ctdb.h" -#include "ctdb_private.h" - -#if defined(HAVE_LINUX_NETFILTER_H) && defined(HAVE_LIBIPQ_H) -#include <linux/netfilter.h> -#include <libipq.h> - -#define CONTROL_TIMEOUT() timeval_current_ofs(5, 0) - -struct ipmux_node { - uint32_t pnn; - ctdb_sock_addr addr; -}; -struct ipmux_node *ipmux_nodes; - -/* - This function is used to open a raw socket to send tickles from - */ -int ctdb_sys_open_sending_socket(void) -{ - int s, ret; - uint32_t one = 1; - - s = socket(AF_INET, SOCK_RAW, htons(IPPROTO_RAW)); - if (s == -1) { - DEBUG(DEBUG_CRIT,(__location__ " failed to open raw socket (%s)\n", - strerror(errno))); - return -1; - } - - ret = setsockopt(s, SOL_IP, IP_HDRINCL, &one, sizeof(one)); - if (ret != 0) { - DEBUG(DEBUG_CRIT,(__location__ " failed to setup IP headers (%s)\n", - strerror(errno))); - close(s); - return -1; - } - - set_nonblocking(s); - set_close_on_exec(s); - - return s; -} - - -/* - main program -*/ -int main(int argc, const char *argv[]) -{ - struct ctdb_context *ctdb; - struct poptOption popt_options[] = { - POPT_AUTOHELP - POPT_CTDB_CMDLINE - POPT_TABLEEND - }; - int opt; - const char **extra_argv; - int extra_argc = 0; - int ret; - poptContext pc; - struct event_context *ev; - uint32_t mypnn, recmaster; - TALLOC_CTX *mem_ctx=NULL; - struct ctdb_node_map *nodemap; - int i, num_nodes; - int s; - struct ipq_handle *ipqh; -#define PKTSIZE 65535 - unsigned char pktbuf[PKTSIZE]; - ipq_packet_msg_t *ipqp; - struct iphdr *ip; - int hash; - - pc = poptGetContext(argv[0], argc, argv, popt_options, POPT_CONTEXT_KEEP_FIRST); - - while ((opt = poptGetNextOpt(pc)) != -1) { - switch (opt) { - default: - fprintf(stderr, "Invalid option %s: %s\n", - poptBadOption(pc, 0), poptStrerror(opt)); - exit(1); - } - } - - /* talloc_enable_leak_report_full(); */ - - /* setup the remaining options for the main program to use */ - extra_argv = poptGetArgs(pc); - if (extra_argv) { - extra_argv++; - while (extra_argv[extra_argc]) extra_argc++; - } - - ev = event_context_init(NULL); - - ctdb = ctdb_cmdline_client(ev); - - - mem_ctx = talloc_new(ctdb); - - /* get our pnn */ - mypnn = ctdb_ctrl_getpnn(ctdb, CONTROL_TIMEOUT(), CTDB_CURRENT_NODE); - if (mypnn == (uint32_t)-1) { - DEBUG(0,("IPMUX: Failed to get local pnn - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - - /* get the recmaster */ - ret = ctdb_ctrl_getrecmaster(ctdb, mem_ctx, CONTROL_TIMEOUT(), CTDB_CURRENT_NODE, &recmaster); - if (ret != 0) { - DEBUG(0,("IPMUX: Failed to get recmaster - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - - /* verify we are the recmaster */ - if (recmaster != mypnn) { - DEBUG(0,("IPMUX: we are not the recmaster - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - - /* get the list of nodes */ - ret = ctdb_ctrl_getnodemap(ctdb, CONTROL_TIMEOUT(), CTDB_CURRENT_NODE, mem_ctx, &nodemap); - if (ret != 0) { - DEBUG(0,("IPMUX: failed to get the nodemap - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - - /* count how many connected nodes we have */ - num_nodes = 0; - for (i=0; i<nodemap->num; i++) { - if (nodemap->nodes[i].flags & NODE_FLAGS_DISCONNECTED) { - continue; - } - - num_nodes++; - } - if (num_nodes == 0) { - DEBUG(0,("IPMUX: no connected nodes - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - ipmux_nodes = talloc_array(mem_ctx, struct ipmux_node, num_nodes); - if (ipmux_nodes == NULL) { - DEBUG(0,("IPMUX: failed to allocate ipmux node array - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - - /* populate the ipmux node array */ - num_nodes = 0; - for (i=0; i<nodemap->num; i++) { - if (nodemap->nodes[i].flags & NODE_FLAGS_DISCONNECTED) { - continue; - } - ipmux_nodes[num_nodes].pnn = i; - ipmux_nodes[num_nodes].addr = nodemap->nodes[i].addr; - num_nodes++; - } - - - /* open a raw socket to send the packets out through */ - s = ctdb_sys_open_sending_socket(); - if (s == -1) { - DEBUG(0,("IPMUX: failed to open raw socket - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - - /* open the ipq handle */ - ipqh = ipq_create_handle(0, PF_INET); - if (ipqh == NULL) { - DEBUG(0,("IPMUX: failed to create ipq handle - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - ret = ipq_set_mode(ipqh, IPQ_COPY_PACKET, PKTSIZE); - if (ret < 0) { - DEBUG(0,("IPMUX: failed to set ipq mode. make sure the ip_queue module is loaded - exiting\n")); - talloc_free(mem_ctx); - exit(10); - } - - while (1) { - /* wait for the next packet */ - ret = ipq_read(ipqh, pktbuf, PKTSIZE, 0); - if (ret <= 0) { - continue; - } - - /* read the packet */ - ipqp = ipq_get_packet(pktbuf); - if (ipqp == NULL) { - continue; - } - - /* calculate a hash based on the clients ip address */ - ip = (struct iphdr *)&ipqp->payload[0]; - /* ntohl here since the client ip addresses are much more - likely to differ in the lower bits than the hight bits */ - hash = ntohl(ip->saddr) % num_nodes; - - - /* if the packet is hashed to the current host, then - just accept it and let the kernel pass it onto - the local stack - */ - if (ipmux_nodes[hash].pnn == mypnn) { - ipq_set_verdict(ipqh, ipqp->packet_id, NF_ACCEPT, 0, pktbuf); - continue; - } - - /* we have hashed it to one of the other nodes, so - send the packet off and tell the kernel to not worry - about this packet any more - */ - ret = sendto(s, &ipqp->payload[0], ipqp->data_len, 0, (struct sockaddr_in *)&ipmux_nodes[hash].addr, sizeof(ctdb_sock_addr)); - ipq_set_verdict(ipqh, ipqp->packet_id, NF_DROP, 0, pktbuf); - - } - - return 0; -} - -#else -int main(void) -{ - printf("ipmux tool disabled - lacking netfilter and libipq development libs\n"); - return 1; -} -#endif |