sql.h

网页抓取程序

#ifndef __SQLFNCT
#define __SQLFNCT


int sqlConnect(char* hostname, char* username, char* password, char* table ,int port, MYSQL* rMysql)
{

	mysql_init(rMysql);

	mysql_options(rMysql,MYSQL_READ_DEFAULT_GROUP,"Sers");
	mysql_options(rMysql,MYSQL_OPT_COMPRESS,0);

	if (!mysql_real_connect(rMysql,hostname,username,password,table,port,NULL,0))
		return 0;

return 1;
}

int my_mysql_query_and_store_results(MYSQL*mysql, char* sqlQuery,MYSQL_RES** tRes,MYSQL_RES* srRes)
{
int ret;

	ret=mysql_query(mysql, sqlQuery);

	if(!ret)
	{
		*tRes=mysql_store_result(mysql);
		if (*tRes)
		{
			memcpy(srRes,*tRes,sizeof(MYSQL_RES));
		}
	}
	else
		*tRes=NULL;

return ret;
}

/* this is licensed under the GPL. Ben Johnson is the author who borrowed heavily from man pages. */

/* because some limit is required, string
* must be less than 9999 characters long. */
#define MAX_STRMODIFIER_LEN 4
static int
snprintf_mysql_escaped_sql_statement(MYSQL * mysql, char * buf, size_t size, const char * fmt, ...)
{
	int added_length = 0;
	int offset_from_buf = 0;
	int oversized = 0;
	va_list ap;
	int d;
	double f;
	char c, *s;
	char current_char;
	int expecting = 0;
	char strmodifier[MAX_STRMODIFIER_LEN+1];
	int strmodifier_offset = 0;
	int keep_expecting = 0;
	int isoverflowed = 0;
	
	strmodifier[0] = '\0';
	
	va_start(ap, fmt);
	while ((current_char = *fmt++))
	{
		added_length = 0; /* this might not change */
		
		if(! expecting)
		{
			switch(current_char)
			{
			case '%': /* start expecting */
				expecting = 1;
				break;
			default:
				if( offset_from_buf < size-2 )
				{
					buf[offset_from_buf] = current_char;
					buf[offset_from_buf+1] = '\0'; /* null terminate every change */
				}
				else
					isoverflowed = 1;
				added_length = 1;
				break;
			}
		}
		else
		{
			keep_expecting = 0;
			
			switch(current_char)
			{
			case '%': /* literal '%' character */
				if( offset_from_buf < size-2 )
				{
					buf[offset_from_buf] = current_char;
					buf[offset_from_buf+1] = '\0'; /* null terminate every change */
				}
				else
					isoverflowed = 1;
				added_length = 1;
				break;
				/* string length modifiers */
			case '0':
			case '1':
			case '2':
			case '3':
			case '4':
			case '5':
			case '6':
			case '7':
			case '8':
			case '9':
				if( strmodifier_offset < MAX_STRMODIFIER_LEN )
				{
					strmodifier[strmodifier_offset] = current_char;
					strmodifier[strmodifier_offset+1] = '\0'; /* null term every change */
					strmodifier_offset++;
				}
				keep_expecting = 1;
				break;
			case 's': /* string */
				{
					int len = 0;
					char * p = NULL; /* for quick malloc */
					s = va_arg(ap, char *);
					
					/* need string length, one way or another */
					if( *strmodifier )
						len = atoi(strmodifier);
					else
						len = strlen(s);
					
						/* if remaining space is sufficient, use buf
					* as mysql_real_escape_string()'s "to" arg */
					if( size - offset_from_buf > len*2+1 )
						added_length = mysql_real_escape_string(mysql, &buf[offset_from_buf], s, len);
					else
					{
						p = malloc(len*2+1);
						if( p )
						{
							added_length = mysql_real_escape_string(mysql, p, s, len);
							if( offset_from_buf+added_length < size )
								memcpy(&buf[offset_from_buf], p, added_length+1); /* +1 is null */
							else
								isoverflowed = 1;
							free(p);
						}
						else
						{
							oversized = -1;
							goto error_out;
						}
					}
				}
				break;
			case 'd': /* int */
				d = va_arg(ap, int);
				added_length = snprintf(&buf[offset_from_buf], (size-offset_from_buf)+1, "%d", d);
				if( added_length > (size-offset_from_buf)+1 )
					isoverflowed = 1;
				break;
			case 'c': /* char */
					  /* need a cast here since va_arg only
				takes fully promoted types */
				c = (char) va_arg(ap, int);
				added_length = snprintf(&buf[offset_from_buf], (size-offset_from_buf)+1, "%c", c);
				if( added_length > (size-offset_from_buf)+1 )
					isoverflowed = 1;
				break;
			case 'f': /* float/double */
				f = va_arg(ap, double); /* promote floats */
				added_length = snprintf(&buf[offset_from_buf], (size-offset_from_buf)+1, "%f", f);
				if( added_length > (size-offset_from_buf)+1 )
					isoverflowed = 1;
				break;
			default:
			/* invalid format. maybe do an error here.
			printf("%s[%d]: default? not adding: %c\n", __FUNCTION__, offset_from_buf, current_char);
				*/
				break;
			}
			
			if( ! keep_expecting )
			{
				strmodifier[0] = '\0';
				strmodifier_offset = 0;
				expecting = 0;
			}
}

oversized += added_length;
if( added_length <= size - offset_from_buf )
offset_from_buf += added_length;
}

#if 0
if( isoverflowed )
memset(buf, 0, size);
#endif

error_out:
va_end(ap);

return oversized;
}



#endif

/*EOF*/