mirror of https://github.com/sysown/proxysql
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
264 lines
8.0 KiB
264 lines
8.0 KiB
#include <cstdlib>
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <unistd.h>
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
#include <sstream>
|
|
#include "mysql.h"
|
|
|
|
#include "tap.h"
|
|
#include "command_line.h"
|
|
#include "utils.h"
|
|
|
|
|
|
/*
|
|
this test uses a lot of code from test_ssl_fast_forward-t.cpp
|
|
*/
|
|
|
|
char * username = (char *)"user1459";
|
|
char * password = (char *)"pass1459";
|
|
|
|
std::vector<std::string> queries_set1 = {
|
|
"SET mysql-have_ssl='false'",
|
|
"LOAD MYSQL VARIABLES TO RUNTIME",
|
|
"DELETE FROM mysql_servers WHERE hostgroup_id = 1459",
|
|
"INSERT INTO mysql_servers (hostgroup_id, hostname, port, use_ssl) VALUES (1459, '127.0.0.1', 6030, 0)",
|
|
"LOAD MYSQL SERVERS TO RUNTIME",
|
|
"DELETE FROM mysql_users WHERE username = 'user1459'",
|
|
"INSERT INTO mysql_users (username,password,fast_forward,default_hostgroup) VALUES ('" + std::string(username) + "','" + std::string(password) + "',1,1459)",
|
|
"LOAD MYSQL USERS TO RUNTIME",
|
|
};
|
|
|
|
std::vector<std::string> queries_set2 = {
|
|
"SET mysql-have_ssl='true'",
|
|
"LOAD MYSQL VARIABLES TO RUNTIME",
|
|
};
|
|
|
|
std::vector<std::string> queries_set3 = {
|
|
"SET mysql-have_ssl='false'",
|
|
"LOAD MYSQL VARIABLES TO RUNTIME",
|
|
"UPDATE mysql_servers SET use_ssl=1 WHERE hostgroup_id = 1459",
|
|
"LOAD MYSQL SERVERS TO RUNTIME",
|
|
};
|
|
|
|
std::vector<std::string> queries_set4 = {
|
|
"SET mysql-have_ssl='true'",
|
|
"LOAD MYSQL VARIABLES TO RUNTIME",
|
|
"UPDATE mysql_servers SET use_ssl=1 WHERE hostgroup_id = 1459",
|
|
"LOAD MYSQL SERVERS TO RUNTIME",
|
|
};
|
|
|
|
std::vector<std::string> queries_SQL3 = {
|
|
"DROP TABLE IF EXISTS tbl1459v",
|
|
"CREATE TABLE tbl1459v (id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL , t1 VARCHAR)",
|
|
};
|
|
|
|
std::vector<std::string> queries_SQL4 = {
|
|
"DROP TABLE IF EXISTS tbl1459v",
|
|
"VACUUM",
|
|
};
|
|
|
|
|
|
int run_queries_sets(std::vector<std::string>& queries, MYSQL *my, const std::string& message_prefix) {
|
|
for (std::vector<std::string>::iterator it = queries.begin(); it != queries.end(); it++) {
|
|
std::string q = *it;
|
|
diag("%s: %s", message_prefix.c_str(), q.c_str());
|
|
MYSQL_QUERY(my, q.c_str());
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#define ITER1 1
|
|
|
|
#define LL 16000 // lower limit
|
|
#define UL 96000 // upper limit
|
|
|
|
int main(int argc, char** argv) {
|
|
CommandLine cl;
|
|
|
|
if(cl.getEnv())
|
|
return exit_status();
|
|
|
|
unsigned int p = 0;
|
|
p += 5*ITER1;
|
|
//p += (5-3)*ITER2*queries_limits.size(); // only on encrypted backend connections
|
|
p += ((UL-LL)/1000*2);
|
|
plan(p);
|
|
diag("Testing SSL and fast_forward");
|
|
|
|
MYSQL* mysqladmin = mysql_init(NULL);
|
|
if (!mysqladmin)
|
|
return exit_status();
|
|
|
|
if (!mysql_real_connect(mysqladmin, cl.host, cl.admin_username, cl.admin_password, NULL, cl.admin_port, NULL, 0)) {
|
|
fprintf(stderr, "File %s, line %d, Error: %s\n",
|
|
__FILE__, __LINE__, mysql_error(mysqladmin));
|
|
return exit_status();
|
|
}
|
|
|
|
MYSQL * mysqls[5];
|
|
for (int i = 0 ; i<5 ; i++) {
|
|
mysqls[i] = NULL;
|
|
}
|
|
// We will loop ITER1 times.
|
|
// On each iteration we create 5 connections with different configuration and run a simple SELECT 1
|
|
|
|
for (int it = 0 ; it<ITER1 ; it++) {
|
|
|
|
diag("We will reconfigure ProxySQL to use SQLite3 Server on hostgroup 1459, IP 127.0.0.1 and port 6030");
|
|
if (run_queries_sets(queries_set1, mysqladmin, "Running on Admin"))
|
|
return exit_status();
|
|
|
|
|
|
diag("We now create a connection not using SSL for either client or backend");
|
|
mysqls[0] = mysql_init(NULL);
|
|
if (!mysqls[0])
|
|
return exit_status();
|
|
|
|
if (!mysql_real_connect(mysqls[0], cl.host, username, password, NULL, cl.port, NULL, 0)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysqls[0]));
|
|
return exit_status();
|
|
}
|
|
MYSQL_QUERY(mysqls[0], "select 1");
|
|
MYSQL_RES* result = mysql_store_result(mysqls[0]);
|
|
ok(mysql_num_rows(result) == 1, "Line %d: Select statement should be executed on connection 1" , __LINE__);
|
|
mysql_free_result(result);
|
|
|
|
|
|
|
|
|
|
diag("We now create a connection using SSL for client connection only and no SSL for backend");
|
|
if (run_queries_sets(queries_set2, mysqladmin, "Running on Admin"))
|
|
return exit_status();
|
|
|
|
mysqls[1] = mysql_init(NULL);
|
|
if (!mysqls[1])
|
|
return exit_status();
|
|
|
|
mysql_ssl_set(mysqls[1], NULL, NULL, NULL, NULL, NULL);
|
|
if (!mysql_real_connect(mysqls[1], cl.host, username, password, NULL, cl.port, NULL, CLIENT_SSL)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysqls[1]));
|
|
return exit_status();
|
|
}
|
|
MYSQL_QUERY(mysqls[1], "select 1");
|
|
result = mysql_store_result(mysqls[1]);
|
|
ok(mysql_num_rows(result) == 1, "Line %d: Select statement should be executed on connection 2" , __LINE__);
|
|
mysql_free_result(result);
|
|
|
|
|
|
|
|
|
|
diag("We now create a connection trying to use SSL for backend connection (but SSL is disabled globally) and not SSL for frontend");
|
|
if (run_queries_sets(queries_set3, mysqladmin, "Running on Admin"))
|
|
return exit_status();
|
|
mysqls[2] = mysql_init(NULL);
|
|
if (!mysqls[2])
|
|
return exit_status();
|
|
|
|
if (!mysql_real_connect(mysqls[2], cl.host, username, password, NULL, cl.port, NULL, 0)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysqls[2]));
|
|
return exit_status();
|
|
}
|
|
MYSQL_QUERY(mysqls[2], "select 1");
|
|
result = mysql_store_result(mysqls[2]);
|
|
ok(mysql_num_rows(result) == 1, "Line %d: Select statement should be executed on connection 3" , __LINE__);
|
|
mysql_free_result(result);
|
|
|
|
|
|
|
|
diag("We now create a connection trying to use SSL for backend connection and not SSL for frontend");
|
|
if (run_queries_sets(queries_set4, mysqladmin, "Running on Admin"))
|
|
return exit_status();
|
|
mysqls[3] = mysql_init(NULL);
|
|
if (!mysqls[3])
|
|
return exit_status();
|
|
|
|
if (!mysql_real_connect(mysqls[3], cl.host, username, password, NULL, cl.port, NULL, 0)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysqls[3]));
|
|
return exit_status();
|
|
}
|
|
MYSQL_QUERY(mysqls[3], "select 1");
|
|
result = mysql_store_result(mysqls[3]);
|
|
ok(mysql_num_rows(result) == 1, "Line %d: Select statement should be executed on connection 4" , __LINE__);
|
|
mysql_free_result(result);
|
|
|
|
|
|
|
|
diag("We now create a connection using SSL for both client or backend");
|
|
if (run_queries_sets(queries_set4, mysqladmin, "Running on Admin")) // note: we use queries_set4 again
|
|
return exit_status();
|
|
mysqls[4] = mysql_init(NULL);
|
|
if (!mysqls[4])
|
|
return exit_status();
|
|
|
|
mysql_ssl_set(mysqls[4], NULL, NULL, NULL, NULL, NULL);
|
|
if (!mysql_real_connect(mysqls[4], cl.host, username, password, NULL, cl.port, NULL, CLIENT_SSL)) {
|
|
fprintf(stderr, "Failed to connect to database: Error: %s\n",
|
|
mysql_error(mysqls[4]));
|
|
return exit_status();
|
|
}
|
|
MYSQL_QUERY(mysqls[4], "select 1");
|
|
result = mysql_store_result(mysqls[4]);
|
|
ok(mysql_num_rows(result) == 1, "Line %d: Select statement should be executed on connection 5" , __LINE__);
|
|
mysql_free_result(result);
|
|
|
|
|
|
if (it != ITER1 - 1) {
|
|
for (int i = 0 ; i<5 ; i++) {
|
|
mysql_close(mysqls[i]);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
for (int i=0; i<5; i++) {
|
|
diag("Connection %d has thread_id: %lu", i, mysqls[i]->thread_id);
|
|
}
|
|
|
|
// We now sends long INSERTs. This code is similar of test_ssl_large_query-t.cpp
|
|
|
|
// We now populate a table named tbl1459v
|
|
if (run_queries_sets(queries_SQL3, mysqls[0], "Running on SQLite3"))
|
|
return exit_status();
|
|
|
|
|
|
std::string s0 = "0";
|
|
//for (int i=16001; i<=48000; i++) {
|
|
for (int i=LL; i<UL; i+=10) {
|
|
std::string s1= "";
|
|
for (int j=0; j<i; j++) {
|
|
s1 += s0;
|
|
}
|
|
s1 += "')";
|
|
// we intentionally have the connections in the inner loop so to use all the connection through the test
|
|
// we only loop on the backend connections with SSL
|
|
for (int c=3; c<5; c++) {
|
|
unsigned int id = c*UL+i;
|
|
std::string s = "INSERT INTO tbl1459v VALUES (" + std::to_string(id) + ",'";
|
|
s += s1;
|
|
std::string del = "DELETE FROM tbl1459v WHERE id = " + std::to_string(id);
|
|
MYSQL_QUERY(mysqls[c], del.c_str());
|
|
MYSQL_QUERY(mysqls[c], s.c_str());
|
|
if (i%1000 == 0) {
|
|
ok(i, "Executed INSERT with id=%d (%lu bytes) on connection %d", id, s.length(), c); // this can be a simple diag, but we use ok() to track progress
|
|
}
|
|
}
|
|
}
|
|
// clean up
|
|
if (run_queries_sets(queries_SQL4, mysqls[0], "Running on SQLite3"))
|
|
return exit_status();
|
|
|
|
|
|
mysql_close(mysqladmin);
|
|
|
|
return exit_status();
|
|
}
|
|
|