server_detect.c

/* 
    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:https://www.gnu.org/licenses/>.

    Authors:        Andrew Hutchings, SkySQL (andrew at skysql dot com)
*/

#include <pcre.h>
#include <glib.h>
#include <string.h>
#include "server_detect.h"

int detect_server(MYSQL *conn) {
	pcre *re= NULL;
	const char *error;
	int erroroffset;
	int ovector[9]= {0};
	int rc;
	const char* db_version= mysql_get_server_info(conn);

	re= pcre_compile(DETECT_MYSQL_REGEX, 0, &error, &erroroffset, NULL);
	if (!re) {
		g_critical("Regular expression fail: %s", error);
		exit(EXIT_FAILURE);
	}

	rc = pcre_exec(re, NULL, db_version, strlen(db_version), 0, 0, ovector, 9);
	pcre_free(re);

	if (rc > 0) {
		return SERVER_TYPE_MYSQL;
	}

	re= pcre_compile(DETECT_DRIZZLE_REGEX, 0, &error, &erroroffset, NULL);
	if (!re) {
		g_critical("Regular expression fail: %s", error);
		exit(EXIT_FAILURE);
	}

	rc = pcre_exec(re, NULL, db_version, strlen(db_version), 0, 0, ovector, 9);     
 	pcre_free(re);

	if (rc > 0) {
		return SERVER_TYPE_DRIZZLE;
	}

	re= pcre_compile(DETECT_MARIADB_REGEX, 0, &error, &erroroffset, NULL);
        if (!re) {
                g_critical("Regular expression fail: %s", error);
                exit(EXIT_FAILURE);
        }

        rc = pcre_exec(re, NULL, db_version, strlen(db_version), 0, 0, ovector, 9);
        pcre_free(re);

        if (rc > 0) {
                return SERVER_TYPE_MYSQL;
        }

	return SERVER_TYPE_UNKNOWN;
}