]> git.ipfire.org Git - ipfire-2.x.git/blobdiff - src/misc-progs/openvpnctrl.c
Merge branch 'kernel-update' of git.ipfire.org:/pub/git/ipfire-2.x into kernel-update
[ipfire-2.x.git] / src / misc-progs / openvpnctrl.c
index 68bbe2fff5e874c33180f2bb0a05f42aceb183ee..e7b128a3f57bdc8bd852282a51121f55db38ca07 100644 (file)
@@ -25,11 +25,13 @@ char enableorange[STRING_SIZE] = "off";
 char OVPNRED[STRING_SIZE] = "OVPN";
 char OVPNBLUE[STRING_SIZE] = "OVPN_BLUE_";
 char OVPNORANGE[STRING_SIZE] = "OVPN_ORANGE_";
-char WRAPPERVERSION[STRING_SIZE] = "ipfire-2.1.2";
+char WRAPPERVERSION[STRING_SIZE] = "ipfire-2.2.2";
 
 struct connection_struct {
        char name[STRING_SIZE];
+       char type[STRING_SIZE];
        char proto[STRING_SIZE];
+       char status[STRING_SIZE];
        int port;
        struct connection_struct *next;
 };
@@ -59,6 +61,12 @@ void usage(void)
        printf("      kills/stops OpenVPN\n");
        printf(" -r   --restart\n");
        printf("      restarts OpenVPN (implicitly creates chains and firewall rules)\n");
+       printf(" -sn2n --start-net-2-net\n");
+       printf("      starts all net2net connections\n");
+       printf("      you may pass a connection name to the switch to only start a specific one\n");
+       printf(" -kn2n --kill-net-2-net\n");
+       printf("      kills all net2net connections\n");
+       printf("      you may pass a connection name to the switch to only start a specific one\n");
        printf(" -d   --display\n");
        printf("      displays OpenVPN status to syslog\n");
        printf(" -fwr --firewall-rules\n");
@@ -81,7 +89,8 @@ connection *getConnections() {
        }
 
        char line[STRING_SIZE] = "";
-       char *result;
+       char result[STRING_SIZE] = "";
+       char *resultptr;
        int count;
        connection *conn_first = NULL;
        connection *conn_last = NULL;
@@ -102,17 +111,33 @@ connection *getConnections() {
                conn_last = conn_curr;
 
                count = 0;
-               result = strtok(line, ",");
-               while (result) {
-                       if (count == 2) {
+               char *lineptr = &line;
+               while (1) {
+                       if (*lineptr == NULL)
+                               break;
+
+                       resultptr = result;
+                       while (*lineptr != NULL) {
+                               if (*lineptr == ',') {
+                                       lineptr++;
+                                       break;
+                               }
+                               *resultptr++ = *lineptr++;
+                       }
+                       *resultptr = '\0';
+
+                       if (count == 1) {
+                               strcpy(conn_curr->status, result);
+                       } else if (count == 2) {
                                strcpy(conn_curr->name, result);
-                       } else if (count == 12) {
+                       } else if (count == 4) {
+                               strcpy(conn_curr->type, result);
+                       } else if (count == 29) {
                                strcpy(conn_curr->proto, result);
-                       } else if (count == 13) {
+                       } else if (count == 30) {
                                conn_curr->port = atoi(result);
                        }
 
-                       result = strtok(NULL, ",");
                        count++;
                }
        }
@@ -122,6 +147,20 @@ connection *getConnections() {
        return conn_first;
 }
 
+int readPidFile(const char *pidfile) {
+       FILE *fp = fopen(pidfile, "r");
+       if (fp == NULL) {
+               fprintf(stderr, "PID file not found: '%s'\n", pidfile);
+               exit(1);
+       }
+
+       int pid = 0;
+       fscanf(fp, "%d", &pid);
+       fclose(fp);
+
+       return pid;
+}
+
 void ovpnInit(void) {
        
        // Read OpenVPN configuration
@@ -268,39 +307,34 @@ void createChain(char *chain) {
 }
 
 void createAllChains(void) {
-       if (!((strcmp(enablered, "on")==0) || (strcmp(enableblue, "on")==0) || (strcmp(enableorange, "on")==0))){
-               fprintf(stderr, "OpenVPN is not enabled on any interface\n");
-               exit(1);
-       } else {
-               // create chain and chain references
-               if (!strcmp(enableorange, "on")) {
-                       if (strlen(orangeif)) {
-                               createChain(OVPNORANGE);
-                               createChainReference(OVPNORANGE);
-                       } else {
-                               fprintf(stderr, "OpenVPN enabled on orange but no orange interface found\n");
-                               //exit(1);
-                       }
+       // create chain and chain references
+       if (!strcmp(enableorange, "on")) {
+               if (strlen(orangeif)) {
+                       createChain(OVPNORANGE);
+                       createChainReference(OVPNORANGE);
+               } else {
+                       fprintf(stderr, "OpenVPN enabled on orange but no orange interface found\n");
+                       //exit(1);
                }
-       
-               if (!strcmp(enableblue, "on")) {
-                       if (strlen(blueif)) {
-                               createChain(OVPNBLUE);
-                               createChainReference(OVPNBLUE);
-                       } else {
-                               fprintf(stderr, "OpenVPN enabled on blue but no blue interface found\n");
-                               //exit(1);
-                       }
+       }
+
+       if (!strcmp(enableblue, "on")) {
+               if (strlen(blueif)) {
+                       createChain(OVPNBLUE);
+                       createChainReference(OVPNBLUE);
+               } else {
+                       fprintf(stderr, "OpenVPN enabled on blue but no blue interface found\n");
+                       //exit(1);
                }
-       
-               if (!strcmp(enablered, "on")) {
-                       if (strlen(redif)) {
-                               createChain(OVPNRED);
-                               createChainReference(OVPNRED);
-                       } else {
-                               fprintf(stderr, "OpenVPN enabled on red but no red interface found\n");
-                               //exit(1);
-                       }
+       }
+
+       if (!strcmp(enablered, "on")) {
+               if (strlen(redif)) {
+                       createChain(OVPNRED);
+                       createChainReference(OVPNRED);
+               } else {
+                       fprintf(stderr, "OpenVPN enabled on red but no red interface found\n");
+                       //exit(1);
                }
        }
 }
@@ -310,12 +344,6 @@ void setFirewallRules(void) {
        char dport[STRING_SIZE] = "";
        char dovpnip[STRING_SIZE] = "";
 
-       /* check if it makes sence to proceed further */
-       if (!((strcmp(enablered, "on")==0) || (strcmp(enableblue, "on")==0) || (strcmp(enableorange, "on")==0))){
-               fprintf(stderr, "Config error, at least one device must be enabled\n");
-               exit(1);
-       }
-
        kv = initkeyvalues();
        if (!readkeyvalues(kv, CONFIG_ROOT "/ovpn/settings"))
        {
@@ -340,9 +368,6 @@ void setFirewallRules(void) {
        }
        freekeyvalues(kv);
 
-       // read connection configuration
-       connection *conn = getConnections();
-
        // Flush all chains.
        flushChain(OVPNRED);
        flushChain(OVPNBLUE);
@@ -356,11 +381,18 @@ void setFirewallRules(void) {
        if (!strcmp(enableorange, "on") && strlen(orangeif))
                setChainRules(OVPNORANGE, orangeif, protocol, dport);
 
+       // read connection configuration
+       connection *conn = getConnections();
+
        // set firewall rules for n2n connections
-       char *port;
-       while (conn) {
-               sprintf(port, "%d", conn->port);
-               setChainRules(OVPNRED, redif, conn->proto, port);
+       char command[STRING_SIZE];
+       while (conn != NULL) {
+               if (strcmp(conn->type, "net") == 0) {
+                       sprintf(command, "/sbin/iptables -A %sINPUT -i %s -p %s --dport %d -j ACCEPT",
+                               OVPNRED, redif, conn->proto, conn->port);
+                       executeCommand(command);
+               }
+
                conn = conn->next;
        }
 }
@@ -369,7 +401,7 @@ void stopDaemon(void) {
        char command[STRING_SIZE];
 
        int pid = readPidFile("/var/run/openvpn.pid");
-       if (pid == NULL) {
+       if (!pid > 0) {
                exit(1);
        }
 
@@ -394,14 +426,14 @@ void startDaemon(void) {
        }
 }
 
-void startNet2Net(char *name) {
+int startNet2Net(char *name) {
        connection *conn = NULL;
        connection *conn_iter;
 
        conn_iter = getConnections();
 
        while (conn_iter) {
-               if (strcmp(conn_iter->name, name) == 0) {
+               if ((strcmp(conn_iter->type, "net") == 0) && (strcmp(conn_iter->name, name) == 0)) {
                        conn = conn_iter;
                        break;
                }
@@ -410,9 +442,16 @@ void startNet2Net(char *name) {
 
        if (conn == NULL) {
                fprintf(stderr, "Connection not found.\n");
-               exit(1);
+               return 1;
        }
 
+       if (strcmp(conn->status, "on") != 0) {
+               fprintf(stderr, "Connection '%s' is not enabled.\n", conn->name);
+               return 1;
+       }
+
+       fprintf(stderr, "Starting connection %s...\n", conn->name);
+
        char configfile[STRING_SIZE];
        snprintf(configfile, STRING_SIZE - 1, CONFIG_ROOT "/ovpn/n2nconf/%s/%s.conf",
                conn->name, conn->name);
@@ -421,7 +460,7 @@ void startNet2Net(char *name) {
        if (fp == NULL) {
                fprintf(stderr, "Could not find configuration file for connection '%s' at '%s'.\n",
                        conn->name, configfile);
-               exit(2);
+               return 2;
        }
        fclose(fp);
 
@@ -429,25 +468,15 @@ void startNet2Net(char *name) {
        setFirewallRules();
 
        char command[STRING_SIZE];
-       sprintf(command, "/usr/sbin/openvpn --config %s", configfile);
+       snprintf(command, STRING_SIZE-1, "/sbin/modprobe tun");
+       executeCommand(command);
+       snprintf(command, STRING_SIZE-1, "/usr/sbin/openvpn --config %s", configfile);
        executeCommand(command);
-}
-
-int readPidFile(const char *pidfile) {
-       FILE *fp = fopen(pidfile, "r");
-       if (fp == NULL) {
-               fprintf(stderr, "PID file not found: '%s'\n", pidfile);
-               exit(1);
-       }
-
-       int pid = NULL;
-       fscanf(fp, "%d", &pid);
-       fclose(fp);
 
-       return pid;
+       return 0;
 }
 
-void killNet2Net(char *name) {
+int killNet2Net(char *name) {
        connection *conn = NULL;
        connection *conn_iter;
 
@@ -463,21 +492,72 @@ void killNet2Net(char *name) {
 
        if (conn == NULL) {
                fprintf(stderr, "Connection not found.\n");
-               exit(1);
+               return 1;
        }
 
        char pidfile[STRING_SIZE];
-       snprintf(&pidfile, STRING_SIZE - 1, "/var/run/%sn2n.pid", conn->name);
+       snprintf(pidfile, STRING_SIZE - 1, "/var/run/%sn2n.pid", conn->name);
 
        int pid = readPidFile(pidfile);
-       if (pid == NULL) {
-               exit(1);
+       if (!pid > 0) {
+               fprintf(stderr, "Could not read pid file of connection %s.", conn->name);
+               return 1;
        }
 
-       fprintf(stderr, "Killing PID %d.\n", pid);
+       fprintf(stderr, "Killing connection %s (PID %d)...\n", conn->name, pid);
        kill(pid, SIGTERM);
 
-       exit(0);
+       char command[STRING_SIZE];
+       snprintf(command, STRING_SIZE - 1, "/bin/rm -f %s", pidfile);
+       executeCommand(command);
+
+       return 0;
+}
+
+void startAllNet2Net() {
+       int exitcode = 0, _exitcode = 0;
+
+       connection *conn = getConnections();
+
+       while(conn) {
+               /* Skip all connections that are not of type "net" or disabled. */
+               if ((strcmp(conn->type, "net") != 0) || (strcmp(conn->status, "on") != 0)) {
+                       conn = conn->next;
+                       continue;
+               }
+
+               _exitcode = startNet2Net(conn->name);
+               conn = conn->next;
+
+               if (_exitcode > exitcode) {
+                       exitcode = _exitcode;
+               }
+       }
+
+       exit(exitcode);
+}
+
+void killAllNet2Net() {
+       int exitcode = 0, _exitcode = 0;
+
+       connection *conn = getConnections();
+
+       while(conn) {
+               /* Skip all connections that are not of type "net". */
+               if (strcmp(conn->type, "net") != 0) {
+                       conn = conn->next;
+                       continue;
+               }
+
+               _exitcode = killNet2Net(conn->name);
+               conn = conn->next;
+
+               if (_exitcode > exitcode) {
+                       exitcode = _exitcode;
+               }
+       }
+
+       exit(exitcode);
 }
 
 void displayopenvpn(void) {
@@ -494,6 +574,8 @@ int main(int argc, char *argv[]) {
            usage();
 
        if(argc == 3) {
+               ovpnInit();
+
                if( (strcmp(argv[1], "-sn2n") == 0) || (strcmp(argv[1], "--start-net-2-net") == 0) ) {
                        startNet2Net(argv[2]);
                        return 0;
@@ -529,6 +611,14 @@ int main(int argc, char *argv[]) {
                                startDaemon();
                                return 0;
                        }
+                       else if( (strcmp(argv[1], "-sn2n") == 0) || (strcmp(argv[1], "--start-net-2-net") == 0) ) {
+                               startAllNet2Net();
+                               return 0;
+                       }
+                       else if( (strcmp(argv[1], "-kn2n") == 0) || (strcmp(argv[1], "--kill-net-2-net") == 0) ) {
+                               killAllNet2Net();
+                               return 0;
+                       }
                        else if( (strcmp(argv[1], "-sdo") == 0) || (strcmp(argv[1], "--start-daemon-only") == 0) ) {
                                startDaemon();
                                return 0;