OpenVPN_Management_GUI/Admin/Modules/Squid/squid_dynamic/squid_dyn.c

/* OpenVPN - Squid Load Balancer*
 *------------------------------*
 * Version 1.0                  *
 * Written by Christoph Haas    *
 * License: LGPL                *
 * 12.10.2010                   *
 */

#include <stdio.h>
#include <stdlib.h>

#include <stdarg.h>
#include <mysql/mysql.h>
#include <regex.h>
#include <string.h>

#define __MAX_LINE__   100
#define __MAX_FIELD__   255
#define false      0
#define true       1


MYSQL mysql;

int connect_db(char *config)
{
    FILE        *db_data                = NULL;
    regex_t     reg;
    char        user[__MAX_LINE__]      = "";
    char        passwd[__MAX_LINE__]    = "";
    char        database[__MAX_LINE__]  = "";
    char        host[__MAX_LINE__]      = "";
    int         port                    = 0;
    char        tmp[__MAX_LINE__];
    char        buf[__MAX_LINE__];
    int         a, b;
    int         length;

    if((db_data = fopen(config, "r")) == NULL)
    {
        fprintf(stderr, "Can't open %s for reading.\n", config);
        return false;
    }
    regcomp(&reg, "^([a-zA-Z0-9:]+)[ ][a-zA-Z0-9]+$", REG_EXTENDED | REG_NEWLINE);
    while(fgets(buf, __MAX_LINE__, db_data))
    {
        length = strlen(buf);
        if(length < 3)continue;
        if(length >= __MAX_LINE__)
        {
            fclose(db_data);
            regfree(&reg);
            fprintf(stderr, "To long line in config file.\n");
            return false;
        }
        if(buf[0] == '#')continue;
        if(regexec(&reg, buf, 0, 0, 0))
        {
            fclose(db_data);
            regfree(&reg);
            fprintf(stderr, "Syntax error in config file.\n");
            return false;
        }
        for(a = 0, b = 0; a < strlen(buf); a++)
        {
            if(buf[a] == ' ')b++;
        }
        if(b != 1)
        {
            fclose(db_data);
            regfree(&reg);
            fprintf(stderr, "Syntax error in config file.\n");
            return false;
        }
        if((strncmp(buf, "user: ", 6))      == 0)sscanf(buf, "%s %s", tmp, user);
        if((strncmp(buf, "passwd: ", 8))   == 0)sscanf(buf, "%s %s", tmp, passwd);
        if((strncmp(buf, "database: ", 10))   == 0)sscanf(buf, "%s %s", tmp, database);
        if((strncmp(buf, "host: ", 6))      == 0)sscanf(buf, "%s %s", tmp, host);
        if((strncmp(buf, "port: ", 6))      == 0)sscanf(buf, "%s %d", tmp, &port);
    }
    fclose(db_data);
    regfree(&reg);
    if((strlen(user) < 1) || (strlen(passwd) < 1) ||
      (strlen(database) < 1) || (strlen(host) < 1))
    {
        fprintf(stderr, "One value for MySQL connection isn't set. \
                        Please set user, passwd, database and host.\n");
        return false;
    }


    mysql_init(&mysql);
    if((mysql_real_connect(&mysql, host, user, passwd, database, port, NULL, 0)) == NULL)
    {
        fprintf(stderr, "%s\n", mysql_error(&mysql));
        return false;
    }
    return true;
}

int main(int argc, char **argv)
{
    MYSQL_RES *result;
    MYSQL_ROW row;
    char Query[200]= "SELECT * FROM userdata WHERE VPNId = '\0";

    if(argc == 2)
    {
        /*iID = atoi(argv[1]);
        printf("Int:%d\n", iID);
        sprintf(ID,"%-#10x",iID);
        printf("String:%s\n", ID);*/

        connect_db("/home/christoph/squid_dynamic/db.conf");      // Datenbank Verbindung aufbauen

        strcat(Query, argv[1]);     // Query builden
        strcat(Query,"'");          //

        mysql_query(&mysql, Query);             // Query abschicken
        result = mysql_store_result(&mysql);    // Result speichern

        while ((row = mysql_fetch_row(result))) // Alle Datensätze auslesen (in dem Fall eh nur einer)
        {
            printf("%s:%s\n", row[9], row[10]); // Feld 9 und 10 (login/pass) aus dem datensatz ausgeben
        }

        mysql_free_result(result);      // Result wieder löschen
        mysql_close(&mysql);            // Datenbank Verbindung schliesen
    }
    else
    {
        printf("USAGE: sqidy xx      | xx = VPNId\n");  // bei flascheingabe der Parameter
    }
}