[Olsr-dev] [PATCH v1 21/23] gateway: setup/cleanup multi-gateway mode during startup/shutdown of olsrd
Ferry Huberts
(spam-protected)
Tue Dec 4 16:17:28 CET 2012
From: Ferry Huberts <(spam-protected)>
Signed-off-by: Ferry Huberts <(spam-protected)>
---
src/gateway.c | 221 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
src/gateway.h | 2 +
src/main.c | 10 +++
3 files changed, 233 insertions(+)
diff --git a/src/gateway.c b/src/gateway.c
index 6a43361..99c0961 100644
--- a/src/gateway.c
+++ b/src/gateway.c
@@ -26,6 +26,20 @@
#include <assert.h>
#include <net/if.h>
+/*
+ * Defines for the multi-gateway script
+ */
+
+#define SCRIPT_IPVERSION ((olsr_cnf->ip_version == AF_INET) ? "ipv4" : "ipv6")
+
+#define SCRIPT_MODE_GENERIC "generic"
+#define SCRIPT_MODE_OLSRIF "olsrif"
+#define SCRIPT_MODE_SGWSRVTUN "sgwsrvtun"
+#define SCRIPT_MODE_EGRESSIF "egressif"
+#define SCRIPT_MODE_SGWTUN "sgwtun"
+
+#define SCRIPT_ADDMODE(add) (add ? "add" : "del")
+
/** structure that holds an interface name, mark and a pointer to the gateway that uses it */
struct interfaceName {
char name[IFNAMSIZ]; /**< interface name */
@@ -218,6 +232,174 @@ static void set_unused_iptunnel_name(struct gateway_entry *gw) {
assert(ifn);
}
+/**
+ * Run the multi-gateway script/
+ *
+ * @param mode the mode (see SCRIPT_MODE_* defines)
+ * @param add true to add policy routing, false to remove it
+ * @param ifname the interface name (optional)
+ * @param ifmark the interface mark (optional
+ * @return true when successful
+ */
+static bool multiGwRunScript(const char * mode, bool add, const char * ifname, uint8_t * ifmark) {
+ struct autobuf buf;
+ int r;
+
+ abuf_init(&buf, 1024);
+
+ abuf_appendf(&buf, "\"%s\"", olsr_cnf->smart_gw_policyrouting_script);
+
+ abuf_appendf(&buf, " \"%s\"", SCRIPT_IPVERSION);
+
+ assert(!strcmp(mode, SCRIPT_MODE_GENERIC) || !strcmp(mode, SCRIPT_MODE_OLSRIF) ||
+ !strcmp(mode, SCRIPT_MODE_SGWSRVTUN) || !strcmp(mode, SCRIPT_MODE_EGRESSIF) ||
+ !strcmp(mode, SCRIPT_MODE_SGWTUN));
+ abuf_appendf(&buf, " \"%s\"", mode);
+
+ abuf_appendf(&buf, " \"%s\"", SCRIPT_ADDMODE(add));
+
+ if (ifname) {
+ assert(!strcmp(mode, SCRIPT_MODE_OLSRIF) || !strcmp(mode, SCRIPT_MODE_SGWSRVTUN) ||
+ !strcmp(mode, SCRIPT_MODE_EGRESSIF) || !strcmp(mode, SCRIPT_MODE_SGWTUN));
+ abuf_appendf(&buf, " \"%s\"", ifname);
+ } else {
+ assert(!strcmp(mode, SCRIPT_MODE_GENERIC));
+ }
+ if (ifmark) {
+ assert(!strcmp(mode, SCRIPT_MODE_EGRESSIF) || !strcmp(mode, SCRIPT_MODE_SGWTUN));
+ assert(ifname);
+ abuf_appendf(&buf, " \"%u\"", *ifmark);
+ } else {
+ assert(!strcmp(mode, SCRIPT_MODE_GENERIC) || !strcmp(mode, SCRIPT_MODE_OLSRIF) ||
+ !strcmp(mode, SCRIPT_MODE_SGWSRVTUN));
+ }
+
+ r = system(buf.buf);
+
+ abuf_free(&buf);
+
+ return (r == 0);
+}
+
+/**
+ * Setup generic multi-gateway iptables and ip rules
+ *
+ * - generic (on olsrd up/down)
+ * iptablesExecutable -t mangle -A OUTPUT -j CONNMARK --restore-mark
+ *
+ * @param add true to add policy routing, false to remove it
+ * @return true when successful
+ */
+static bool multiGwRulesGeneric(bool add) {
+ return multiGwRunScript(SCRIPT_MODE_GENERIC, add, NULL, NULL);
+}
+
+/**
+ * Setup multi-gateway iptables and ip rules for all OLSR interfaces.
+ *
+ * - olsr interfaces (on olsrd up/down)
+ * iptablesExecutable -t mangle -A PREROUTING -i ${olsrInterface} -j CONNMARK --restore-mark
+ *
+ * @param add true to add policy routing, false to remove it
+ * @return true when successful
+ */
+static bool multiGwRulesOlsrInterfaces(bool add) {
+ bool ok = true;
+ struct interface * ifn;
+
+ for (ifn = ifnet; ifn; ifn = ifn->int_next) {
+ if (!multiGwRunScript(SCRIPT_MODE_OLSRIF, add, ifn->int_name, NULL)) {
+ ok = false;
+ if (add) {
+ return ok;
+ }
+ }
+ }
+
+ return ok;
+}
+
+/**
+ * Setup multi-gateway iptables and ip rules for the smart gateway server tunnel.
+ *
+ * - sgw server tunnel interface (on olsrd up/down)
+ * iptablesExecutable -t mangle -A PREROUTING -i tunl0 -j CONNMARK --restore-mark
+ *
+ * @param add true to add policy routing, false to remove it
+ * @return true when successful
+ */
+static bool multiGwRulesSgwServerTunnel(bool add) {
+ return multiGwRunScript(SCRIPT_MODE_SGWSRVTUN, add, TUNNEL_NAME, NULL);
+}
+
+/**
+ * Setup multi-gateway iptables and ip rules for all egress interfaces.
+ *
+ * - egress interfaces (on interface up/down)
+ * iptablesExecutable -t mangle -A POSTROUTING -m conntrack --ctstate NEW -o ${egressInterface} -j CONNMARK --set-mark ${egressInterfaceMark}
+ * iptablesExecutable -t mangle -A INPUT -m conntrack --ctstate NEW -i ${egressInterface} -j CONNMARK --set-mark ${egressInterfaceMark}
+ * ip rule add fwmark ${egressInterfaceMark} table ${egressInterfaceMark} pref ${egressInterfaceMark}
+ *
+ * like table:
+ * ppp0 91
+ * eth0 92
+ *
+ * @param add true to add policy routing, false to remove it
+ * @return true when successful
+ */
+static bool multiGwRulesEgressInterfaces(bool add) {
+ bool ok = true;
+ unsigned int i = 0;
+
+ for (i = 0; i < olsr_cnf->smart_gw_use_count; i++) {
+ struct interfaceName * ifn = &sgwEgressInterfaceNames[i];
+ if (!multiGwRunScript(SCRIPT_MODE_EGRESSIF, add, ifn->name, &ifn->mark)) {
+ ok = false;
+ if (add) {
+ return ok;
+ }
+ }
+ }
+
+ return ok;
+}
+
+/**
+ * Setup multi-gateway iptables and ip rules for the smart gateway client tunnels.
+ *
+ * - sgw tunnels (on sgw tunnel up/down)
+ * iptablesExecutable -t mangle -A POSTROUTING -m conntrack --ctstate NEW -o ${sgwTunnelInterface} -j CONNMARK --set-mark ${sgwTunnelInterfaceMark}
+ * ip rule add fwmark ${sgwTunnelInterfaceMark} table ${sgwTunnelInterfaceMark} pref ${sgwTunnelInterfaceMark}
+ *
+ * like table:
+ * tnl_101 101
+ * tnl_102 102
+ * tnl_103 103
+ * tnl_104 104
+ * tnl_105 105
+ * tnl_106 106
+ * tnl_107 107
+ * tnl_108 108
+ */
+static bool multiGwRulesSgwTunnels(bool add) {
+ bool ok = true;
+ unsigned int i = 0;
+
+ while (i < olsr_cnf->smart_gw_use_count) {
+ struct interfaceName * ifn = (olsr_cnf->ip_version == AF_INET) ? &sgwTunnel4InterfaceNames[i] : &sgwTunnel6InterfaceNames[i];
+ if (!multiGwRunScript(SCRIPT_MODE_SGWTUN, add, ifn->name, &ifn->mark)) {
+ ok = false;
+ if (add) {
+ return ok;
+ }
+ }
+
+ i++;
+ }
+
+ return ok;
+}
+
/*
* Callback Functions
*/
@@ -325,6 +507,45 @@ int olsr_init_gateways(void) {
}
/**
+ * Startup gateway system
+ */
+int olsr_startup_gateways(void) {
+ bool ok = true;
+
+ if (olsr_cnf->smart_gw_use_count <= 1) {
+ return 0;
+ }
+
+ ok = ok && multiGwRulesGeneric(true);
+ ok = ok && multiGwRulesSgwServerTunnel(true);
+ ok = ok && multiGwRulesOlsrInterfaces(true);
+ ok = ok && multiGwRulesEgressInterfaces(true);
+ ok = ok && multiGwRulesSgwTunnels(true);
+ if (!ok) {
+ olsr_printf(0, "Could not setup multi-gateway iptables and ip rules\n");
+ olsr_shutdown_gateways();
+ return 1;
+ }
+
+ return 0;
+}
+
+/**
+ * Shutdown gateway tunnel system
+ */
+void olsr_shutdown_gateways(void) {
+ if (olsr_cnf->smart_gw_use_count <= 1) {
+ return;
+ }
+
+ (void)multiGwRulesSgwTunnels(false);
+ (void)multiGwRulesEgressInterfaces(false);
+ (void)multiGwRulesOlsrInterfaces(false);
+ (void)multiGwRulesSgwServerTunnel(false);
+ (void)multiGwRulesGeneric(false);
+}
+
+/**
* Cleanup gateway tunnel system
*/
void olsr_cleanup_gateways(void) {
diff --git a/src/gateway.h b/src/gateway.h
index 8ac6eab..7481218 100644
--- a/src/gateway.h
+++ b/src/gateway.h
@@ -138,6 +138,8 @@ struct olsr_gw_handler {
*/
int olsr_init_gateways(void);
+int olsr_startup_gateways(void);
+void olsr_shutdown_gateways(void);
void olsr_cleanup_gateways(void);
void olsr_trigger_inetgw_startup(void);
#ifndef NODEBUG
diff --git a/src/main.c b/src/main.c
index 2d9b3a5..20736d5 100644
--- a/src/main.c
+++ b/src/main.c
@@ -540,6 +540,15 @@ int main(int argc, char *argv[]) {
}
}
+#ifdef __linux__
+ /* startup gateway system */
+ if (olsr_cnf->smart_gw_active) {
+ if (olsr_startup_gateways()) {
+ olsr_exit("Cannot startup gateway tunnels", 1);
+ }
+ }
+#endif /* __linux__ */
+
olsr_do_startup_sleep();
/* Print heartbeat to stdout */
@@ -775,6 +784,7 @@ static void olsr_shutdown(int signo __attribute__ ((unused)))
#ifdef __linux__
/* trigger gateway selection */
if (olsr_cnf->smart_gw_active) {
+ olsr_shutdown_gateways();
olsr_cleanup_gateways();
}
--
1.7.11.7
More information about the Olsr-dev
mailing list