//gcc bug37162.c -Wall -g -o bug37162 -L/home/sbester/mysql/5.1/mysql-5.1.26-rc-linux-i686/lib -I/home/sbester/mysql/5.1/mysql-5.1.26-rc-linux-i686/include   -lmysqlclient_r -lz -lpthread

#include <unistd.h>
#include <pthread.h>

#include <string.h>
#include <stdlib.h>
#include <time.h>
#include <mysql.h>
#include <stdio.h>
#define TESTTIME (6000)
#define NUMTHREADS (10)
#define DATASIZE (16*1024*1024)
char host[]="192.168.250.3";
int port=3307;
char username[]="root";
char password[]="";
char database[]="test";
pthread_t pthreads[NUMTHREADS];
unsigned long client_version=0;
unsigned long server_version=0;
unsigned long num_queries=0;
int threaddone=0;


char *databucket=NULL;

int db_query(MYSQL *dbc,char *sql,int showresults);
char* alocmem(size_t num);
int write_string(char *buf,const size_t minlen,const size_t maxlen);
/* prepared statement function declarations: 3 */
int ps_func_0000(MYSQL *db,MYSQL_STMT *stmt);
int ps_func_0001(MYSQL *db,MYSQL_STMT *stmt);
int ps_func_0002(MYSQL *db,MYSQL_STMT *stmt);

void *worker_thread(void *arg)
{
	MYSQL *dbc=NULL;
	my_bool auto_reconnect=1;
	int cancelstate=0;
	pthread_setcancelstate(PTHREAD_CANCEL_DISABLE,&cancelstate);

	dbc = mysql_init(NULL);
	if(NULL == dbc)
	{
		printf("mysql_init failed\n");
		dbc=NULL;
		goto threadexit;
	}
	else
	{
		mysql_options(dbc,MYSQL_OPT_RECONNECT,(char*)&auto_reconnect);
		if (!mysql_real_connect(dbc,host,username,password,database,port, NULL, CLIENT_FOUND_ROWS|CLIENT_MULTI_STATEMENTS|CLIENT_MULTI_RESULTS))
		{
			printf("mysql_real_connect failed: %s (%d)", mysql_error(dbc),mysql_errno(dbc));
			dbc=NULL;
		}
	}

	
	char shortquery[1024];
	memset(shortquery,0,1024);
	char *longquery;
	longquery=NULL;
	char *c;
	c=NULL;
	
	
	MYSQL_STMT *stmt=NULL;
        stmt= mysql_stmt_init(dbc);
	if (NULL == stmt)
	{
	      printf("mysql_stmt_init failed: %s (%d)\n",mysql_error(dbc),mysql_errno(dbc));
	      //continue;
	}
	while(0==threaddone && NULL!=dbc)
	{

                
                
                if(lrand48()%3==0)
		{
		        //select a from t1
		        ps_func_0000(dbc,stmt);
		}

                if(lrand48()%3==0)
		{        
		        //select h,a,b from t2
		        ps_func_0001(dbc,stmt);
                }
                
                if(lrand48()%3==0)
		{ 
		        //select d,e from t3
		        ps_func_0002(dbc,stmt);
		}

                mysql_stmt_free_result(stmt);
                
                
                

	}
	//mysql_stmt_close(stmt);
threadexit:
	mysql_close(dbc);
	mysql_thread_end();
	pthread_exit(0);
}


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

	srand48((unsigned long)time(NULL));
	time_t timestart=0,timenow=0;

	unsigned int counter=0;
	counter=0;
	char shortquery[1024]={0};
	char *longquery=NULL;
	longquery=NULL;
	char *c=NULL;
	my_init();
	if (!(dbc = mysql_init(NULL)))
	{
		printf("mysql_init\n");
		dbc=NULL;
		goto threadexit;
	}
	else
	{
		if (!mysql_real_connect(dbc,host,username,password,database,port, NULL, CLIENT_FOUND_ROWS|CLIENT_MULTI_STATEMENTS|CLIENT_MULTI_RESULTS))
		{
			printf("mysql_real_connect failed: %s (%d)", mysql_error(dbc),mysql_errno(dbc));
			dbc=NULL;
			goto threadexit;
		}
	}

	printf("running initializations..\n");
	client_version=mysql_get_client_version();
	server_version=mysql_get_server_version(dbc);
	printf("client version=%lu\n",client_version);
	printf("server version=%lu\n",server_version);
	if((client_version/10000) < (server_version/10000))
	{
		printf("incompatible client and server version!  please upgrade client library!\n");
		goto threadexit;
	}

	if (!mysql_thread_safe())
	{
		printf("non-threadsafe client detected!  please rebuild and link with libmysql_r!\n");
	}

	printf("pre-generating %d bytes of random data\n",DATASIZE);
   databucket=(char*)calloc(DATASIZE,sizeof(char*));
   if(NULL == databucket)
	{
		printf("error: cannot calloc data buffer.\n");
		exit(1);
	}
	for(i=0;i<DATASIZE-1;i++)
	{
		databucket[i]=0x61+(lrand48()%26);
	}
				c=shortquery;
				c+=sprintf(c,"%s","drop table if exists t1,t2,t3");
				db_query(dbc,shortquery,1);

				c=shortquery;
				c+=sprintf(c,"%s","create table t1(a varchar(125),b varchar(5),c int, d int, e bigint, f datetime, g char(2), h varchar(10))engine=myisam");
				db_query(dbc,shortquery,1);

			longquery = alocmem(2164);
			for(counter=0;counter<5000;counter++)
			{
				c=longquery;
				c+=sprintf(c,"%s","insert into t1 set a='");
				c+=write_string(c,0,255);
				c+=sprintf(c,"%s","',b='");
				c+=write_string(c,0,255);
				c+=sprintf(c,"%s","',c=");
				c+=sprintf(c,"%ld",-2147483647 + lrand48()%4294967294lu);
				c+=sprintf(c,"%s",",d=");
				c+=sprintf(c,"%ld",-2147483647 + lrand48()%4294967294lu);
				c+=sprintf(c,"%s",",e=");
				c+=sprintf(c,"%ld",-2147483647 + lrand48()%4294967294lu);
				c+=sprintf(c,"%s",",f='");
				c+=sprintf(c,"%04ld-%02ld-%02ld %02ld:%02ld:%02ld",1000 + (lrand48()%(9999-1000)),01 + (lrand48()%(12-1)),01 + (lrand48()%(31-1)),(lrand48()%(24-0)),(lrand48()%(60-0)),(lrand48()%(60)));
				c+=sprintf(c,"%s","',g='");
				c+=write_string(c,0,255);
				c+=sprintf(c,"%s","',h='");
				c+=write_string(c,0,255);
				c+=sprintf(c,"%s","'");
				db_query(dbc,longquery,1);
			}
			free(longquery);

			c=shortquery;
			c+=sprintf(c,"%s","create table t2 like t1");
			db_query(dbc,shortquery,1);
				
			c=shortquery;
			c+=sprintf(c,"%s","create table t3 like t1");
			db_query(dbc,shortquery,1);
			
			c=shortquery;
			c+=sprintf(c,"%s","insert into t2 select * from t1 order by rand() limit 1000");
			db_query(dbc,shortquery,1);
			
			c=shortquery;
			c+=sprintf(c,"%s","insert into t3 select * from t2 order by rand() limit 100");
			db_query(dbc,shortquery,1);

	mysql_close(dbc);

	printf("about to spawn %d threads\n",NUMTHREADS);
	for (i=0;i<NUMTHREADS;i++)
	{
		err=pthread_create(&pthreads[i], NULL, worker_thread, (void *)i);
		if(err!=0)
		{
			printf("error spawning thread %lu, pthread_create returned %lu\n",(unsigned long)i,(unsigned long)err);
		}
		printf(".");
	}
	printf("\n");
	printf("completed spawning new database worker threads\n");

	printf("testcase is now running, so watch for error output\n");

	timestart=time(NULL);
	timenow=time(NULL);
	for(i=0;(timenow-timestart) < TESTTIME;timenow=time(NULL))
	{
		usleep(1000*1000);
		printf("queries: %09lu\n",num_queries);
	}
	threaddone=1;

	printf("waiting for worker threads to finish...\n");

	for (i=0;i<NUMTHREADS;i++)
	{
		pthread_join(pthreads[i], NULL);
	}

	if(NULL!=databucket)free(databucket);
	exit(0);
threadexit:
	exit(-1);
}


int db_query(MYSQL *dbc,char *sql,int showresults)
{
	int res=0;
	MYSQL_RES *r=NULL;
	MYSQL_ROW w;
	MYSQL_FIELD *field=NULL;
	int moreresult=0;
	unsigned int i=0;
	if(NULL == dbc) return 0;
	res = mysql_query(dbc,sql);
	if(res != 0 && showresults > 0)
	{
		printf("query failed '%s' : %d (%s)\n",sql,mysql_errno(dbc),mysql_error(dbc));
		return 0;
	}

	num_queries++;
	do
	{
		r = mysql_use_result(dbc);
		if(r)
		{
			unsigned int numfields = mysql_num_fields(r);
			//unsigned int numrows=mysql_num_rows(r);
			while(0!=(field = mysql_fetch_field(r)))
			{
					//print metadata information about each field
					if(showresults > 1)
					{
						printf("%s	",field->name);
					}
			}
			if(showresults > 1)
			{
				printf("\n------------------------------------\n");
			}

			while (0!=(w = mysql_fetch_row(r)))
			{
				for(i = 0; i < numfields; i++)
				{
					//print each field here
					if(showresults > 1)
					{
						printf("%s\t",w[i]);
					}
				}

				if(showresults > 1)
				{
						printf("\n");
				}
			}
			if(showresults > 1)
			{
				printf("\n");
			}
			mysql_free_result(r);
		}
		else //no rows returned. was it a select?
		{
			if(mysql_field_count(dbc) > 0 && showresults > 0)
			{
				printf("No results for '%s'.  (%d) - %s\n",sql,mysql_errno(dbc),mysql_error(dbc));
				return 0;
			}
			else //it could have been some insert/update/delete
			{
				//this is successful query
			}
		}
		moreresult=mysql_next_result(dbc);
		if(moreresult > 0 && showresults > 0)
		{
			printf("mysql_next_result returned %d, mysql error %s, (%d)\n",moreresult,mysql_error(dbc),mysql_errno(dbc));
			return 0;
		}
	} while (0==moreresult);
	return 1;
}

char* alocmem(size_t num)
{
	char *r=(char*)calloc(num,1);
	if(NULL == r)
	{
		printf("cannot calloc %I64u bytes of memory\n",num);
		exit(1);
	}
	return r;
}

int write_string(char *buf,const size_t minlen,const size_t maxlen)
{
	if(!buf)return 0;
	if(maxlen==0 ){buf[0]=0;return 0;}
	size_t lengt=minlen + (maxlen<=minlen?0:lrand48() % (maxlen-minlen));
	int start_lengt = lrand48()%(DATASIZE - lengt - 1);
	if(lengt<0) lengt=0;
	if(start_lengt<0) start_lengt=0;
	memcpy(buf,databucket+start_lengt,lengt);
	buf[lengt]=0;
	return lengt;
}

int ps_func_0000(MYSQL *db,MYSQL_STMT *stmt)
{
	char query[1024]="select a from t1";
	//MYSQL_STMT *stmt=NULL;
	unsigned int val=0;
	MYSQL_BIND bind_result[1];
	unsigned long length_result[1];
	my_bool is_null_result[1];
	my_bool error_result[1];
	memset(bind_result, 0, sizeof(bind_result));
	//stmt= mysql_stmt_init(db);
	//if (NULL == stmt)
	//{
	//	printf("mysql_stmt_init failed: %s (%d)\n",mysql_error(db),mysql_errno(db));
	//	return 0;
	//}

	if(mysql_stmt_prepare(stmt,query,(unsigned long)strlen(query)))
	{
		printf("mysql_stmt_prepare failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=1;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_UPDATE_MAX_LENGTH, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=CURSOR_TYPE_READ_ONLY;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_CURSOR_TYPE, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=5000;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_PREFETCH_ROWS, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

	bind_result[0].buffer_type=MYSQL_TYPE_VAR_STRING;
	bind_result[0].buffer=alocmem(255);
	bind_result[0].buffer_length=255;
	bind_result[0].length=&length_result[0];
	bind_result[0].is_null=&is_null_result[0];
	bind_result[0].error=&error_result[0];
        int i=0;
	for(i=0;i<5;i++)
	{
		if(mysql_stmt_execute(stmt))
		{
			printf("mysql_stmt_execute failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}
	        num_queries++;
		if(mysql_stmt_bind_result(stmt,bind_result))
		{
			printf("mysql_stmt_bind_result failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}

		if(mysql_stmt_store_result(stmt))
		{
			printf("mysql_stmt_store_result failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}

		while (!mysql_stmt_fetch(stmt))
		{
		}
	}


        

	//if(mysql_stmt_close(stmt))
	//{
	//	printf("mysql_stmt_close failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
	//	return 0;
	//}

	if(NULL!=bind_result[0].buffer) free(bind_result[0].buffer);


	return 1;
}
int ps_func_0001(MYSQL *db,MYSQL_STMT *stmt)
{
	char *query="select h,a,b from t2";
	//MYSQL_STMT *stmt=NULL;
	unsigned int val=0;
	MYSQL_BIND bind_result[3];
	unsigned long length_result[3];
	my_bool is_null_result[3];
	my_bool error_result[3];
	memset(bind_result, 0, sizeof(bind_result));
	//stmt= mysql_stmt_init(db);
	//if (NULL == stmt)
	//{
	//	printf("mysql_stmt_init failed: %s (%d)\n",mysql_error(db),mysql_errno(db));
	//	return 0;
	//}

	if(mysql_stmt_prepare(stmt,query,(unsigned long)strlen(query)))
	{
		printf("mysql_stmt_prepare failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=1;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_UPDATE_MAX_LENGTH, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=CURSOR_TYPE_READ_ONLY;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_CURSOR_TYPE, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=5000;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_PREFETCH_ROWS, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

	bind_result[0].buffer_type=MYSQL_TYPE_VAR_STRING;
	bind_result[0].buffer=alocmem(10240);
	bind_result[0].buffer_length=10240;
	bind_result[0].length=&length_result[0];
	bind_result[0].is_null=&is_null_result[0];
	bind_result[0].error=&error_result[0];

	bind_result[1].buffer_type=MYSQL_TYPE_VAR_STRING;
	bind_result[1].buffer=alocmem(255);
	bind_result[1].buffer_length=255;
	bind_result[1].length=&length_result[1];
	bind_result[1].is_null=&is_null_result[1];
	bind_result[1].error=&error_result[1];

	bind_result[2].buffer_type=MYSQL_TYPE_VAR_STRING;
	bind_result[2].buffer=alocmem(255);
	bind_result[2].buffer_length=255;
	bind_result[2].length=&length_result[2];
	bind_result[2].is_null=&is_null_result[2];
	bind_result[2].error=&error_result[2];
        int i=0;
	for(i=0;i<5;i++)
	{
		if(mysql_stmt_execute(stmt))
		{
			printf("mysql_stmt_execute failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}
	        num_queries++;
		if(mysql_stmt_bind_result(stmt,bind_result))
		{
			printf("mysql_stmt_bind_result failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}

		if(mysql_stmt_store_result(stmt))
		{
			printf("mysql_stmt_store_result failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}

		while (!mysql_stmt_fetch(stmt))
		{
		}
	}

	//if(mysql_stmt_close(stmt))
	//{
	//	printf("mysql_stmt_close failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
	//	return 0;
	//}

	if(NULL!=bind_result[0].buffer) free(bind_result[0].buffer);
	if(NULL!=bind_result[1].buffer) free(bind_result[1].buffer);
	if(NULL!=bind_result[2].buffer) free(bind_result[2].buffer);


	return 1;
}
int ps_func_0002(MYSQL *db,MYSQL_STMT *stmt)
{
	char *query="select d,e from t3";
	//MYSQL_STMT *stmt=NULL;
	unsigned int val=0;
	MYSQL_BIND bind_result[2];
	unsigned long length_result[2];
	my_bool is_null_result[2];
	my_bool error_result[2];
	memset(bind_result, 0, sizeof(bind_result));
	//stmt= mysql_stmt_init(db);
	//if (NULL == stmt)
	//{
	//	printf("mysql_stmt_init failed: %s (%d)\n",mysql_error(db),mysql_errno(db));
	//	return 0;
	//}

	if(mysql_stmt_prepare(stmt,query,(unsigned long)strlen(query)))
	{
		printf("mysql_stmt_prepare failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=1;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_UPDATE_MAX_LENGTH, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=CURSOR_TYPE_READ_ONLY;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_CURSOR_TYPE, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

        val=5000;
        if(mysql_stmt_attr_set(stmt, STMT_ATTR_PREFETCH_ROWS, (void *)&val))
	{
		printf("mysql_stmt_attr_set: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
		return 0;
	}

	bind_result[0].buffer_type=MYSQL_TYPE_LONG;
	bind_result[0].buffer=alocmem(4);
	bind_result[0].buffer_length=4;
	bind_result[0].length=&length_result[0];
	bind_result[0].is_null=&is_null_result[0];
	bind_result[0].error=&error_result[0];

	bind_result[1].buffer_type=MYSQL_TYPE_LONGLONG;
	bind_result[1].buffer=alocmem(8);
	bind_result[1].buffer_length=8;
	bind_result[1].length=&length_result[1];
	bind_result[1].is_null=&is_null_result[1];
	bind_result[1].error=&error_result[1];
        int i=0;
	for(i=0;i<5;i++)
	{
		if(mysql_stmt_execute(stmt))
		{
			printf("mysql_stmt_execute failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}
	        num_queries++;
		if(mysql_stmt_bind_result(stmt,bind_result))
		{
			printf("mysql_stmt_bind_result failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}

		if(mysql_stmt_store_result(stmt))
		{
			printf("mysql_stmt_store_result failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
			return 0;
		}

		while (!mysql_stmt_fetch(stmt))
		{
		}
	}

	//if(mysql_stmt_close(stmt))
	//{
	//	printf("mysql_stmt_close failed: %s (%d)\n",mysql_stmt_error(stmt),mysql_stmt_errno(stmt));
	//	return 0;
	//}

	if(NULL!=bind_result[0].buffer) free(bind_result[0].buffer);
	if(NULL!=bind_result[1].buffer) free(bind_result[1].buffer);


	return 1;
}