#include <winsock2.h>
#include <ws2tcpip.h>
#include <stdio.h>
#pragma comment(lib, "Ws2_32.lib")
#include "mysql.h"
#include <my_getopt.h>

static my_bool opt_help=0;
static char *opt_user="root";
char *opt_password="";
char *opt_socket="MYSQL";
static unsigned long opt_num_connections=1000;

static struct my_option my_long_options[]=
{
	{"help", '?', "Display this help and exit",
   	(uchar **) &opt_help, (uchar **) &opt_help, 0,
  	 GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
	
	{"user", 'u', "Username for login",
	&opt_user, &opt_user, 0,
	GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},

	{"password", 'p',"Password to use when connecting to server",
	&opt_password, &opt_password, 0,
	GET_STR, REQUIRED_ARG,0, 0, 0, 0, 0, 0},

	{"shared-memory-base-name", 'S',"Shared memory base name",
	&opt_socket, &opt_socket, 0,
	GET_STR, REQUIRED_ARG,0, 0, 0, 0, 0, 0},

	{"num-connections", 'n', "Number of consecutive connections to open/close",
	&opt_num_connections,&opt_num_connections, 0,
	GET_UINT, REQUIRED_ARG, 1000, 1,4294967295, 0, 0,  0},

	{0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
};


int main(int argc,char* argv[])
{
	MYSQL *dbc=NULL;
	MYSQL_RES *r;
	unsigned int err=0,i=0;

	err= handle_options(&argc, &argv, my_long_options, NULL);
	if (err)
	{
		printf("handle_options returned error\n");
    	return err;
	}

	if (opt_help)
	{
		my_print_help(my_long_options);
		return 1;
	}

	if(mysql_library_init(0,NULL,NULL))
	{
		printf("mysql_library_init() failed\n");
		return 1;
	}
        
	

	
	
	printf("about to make %lu connections using shared memory, watch for errors!\n\n",opt_num_connections);
	for(i=0;i<opt_num_connections;i++)
	{
		dbc = mysql_init(NULL);
		if (!dbc)
		{
			printf("mysql_init() failed\n");
			Sleep(1000);
			continue;
		}
		if(mysql_options(dbc,MYSQL_SHARED_MEMORY_BASE_NAME,opt_socket))
		{
			printf("cannot set MYSQL_SHARED_MEMORY_BASE_NAME to %s\n",opt_socket);
		}
		if(!mysql_real_connect(dbc,"localhost",opt_user,opt_password,NULL,3306,"MYSQL",0))
		{
			printf("mysql_real_connect failed: %s (%d)\n", mysql_error(dbc),mysql_errno(dbc));
		}

		if(0 == i && dbc != NULL)
			printf("connected: %s [%s] \n",mysql_get_host_info(dbc),mysql_get_server_info(dbc));
		
		if(0 == i%10000)
			printf("tried %10lu connections..\n",i);
		
		if(dbc)
			mysql_close(dbc);
	}
				
	return 0;
}


static my_bool get_one_option(int optid, const struct my_option *opt __attribute__((unused)), char *argument)
{
  	return 0;
}