|
|
| version 1.23, 2004/05/25 07:07:07 | version 1.53, 2017/08/01 17:09:02 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser MySQL driver. | Parser MySQL driver. |
| Copyright(c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001-2015 Art. Lebedev Studio (http://www.artlebedev.com) |
| Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) | Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) |
| 2001.07.30 using MySQL 3.23.22b | 2001-07-30 using MySQL 3.23.22b |
| 2001.11.06 numrows on "HP-UX istok1 B.11.00 A 9000/869 448594332 two-user license" | 2001-11-06 numrows on "HP-UX istok1 B.11.00 A 9000/869 448594332 two-user license" |
| 3.23.42 & 4.0.0.alfa never worked, both subst & .sl version returned 0 | 3.23.42 & 4.0.0.alfa never worked, both subst & .sl version returned 0 |
| */ | */ |
| static const char *RCSId="$Id$"; | |
| #include "config_includes.h" | #include "config_includes.h" |
| #include "pa_sql_driver.h" | #include "pa_sql_driver.h" |
| volatile const char * IDENT_PARSER3MYSQL_C="$Id$" IDENT_PA_SQL_DRIVER_H; | |
| #define NO_CLIENT_LONG_LONG | #define NO_CLIENT_LONG_LONG |
| #include "mysql.h" | #include "mysql.h" |
| #include "ltdl.h" | #include "ltdl.h" |
| Line 28 static const char *RCSId="$Id$"; | Line 29 static const char *RCSId="$Id$"; |
| # define strcasecmp _stricmp | # define strcasecmp _stricmp |
| #endif | #endif |
| static char *lsplit(char *string, char delim) { | static char *lsplit(char *string, char delim){ |
| if(string) { | if(string) { |
| char *v=strchr(string, delim); | if(char *v=strchr(string, delim)){ |
| if(v) { | |
| *v=0; | *v=0; |
| return v+1; | return v+1; |
| } | } |
| } | } |
| return 0; | return 0; |
| } | } |
| static char *lsplit(char **string_ref, char delim) { | static char *lsplit(char **string_ref, char delim){ |
| char *result=*string_ref; | char *result=*string_ref; |
| char *next=lsplit(*string_ref, delim); | char *next=lsplit(*string_ref, delim); |
| *string_ref=next; | *string_ref=next; |
| return result; | return result; |
| } | |
| static char* rsplit(char* string, char delim){ | |
| if(string){ | |
| if(char* v=strrchr(string, delim)){ | |
| *v=0; | |
| return v+1; | |
| } | |
| } | |
| return NULL; | |
| } | } |
| static void toupper_str(char *out, const char *in, size_t size) { | static void toupper_str(char *out, const char *in, size_t size){ |
| while(size--) | while(size--) |
| *out++=(char)toupper(*in++); | *out++=(char)toupper(*in++); |
| } | } |
| inline static bool is_column_transcode_required(enum_field_types type) { | |
| switch(type) { | |
| case MYSQL_TYPE_NULL: | |
| #ifdef FIELD_TYPE_NEWDECIMAL | |
| case MYSQL_TYPE_NEWDECIMAL: | |
| #endif | |
| case MYSQL_TYPE_DECIMAL: | |
| case MYSQL_TYPE_FLOAT: | |
| case MYSQL_TYPE_DOUBLE: | |
| case MYSQL_TYPE_TINY: | |
| case MYSQL_TYPE_SHORT: | |
| case MYSQL_TYPE_LONG: | |
| case MYSQL_TYPE_LONGLONG: | |
| case MYSQL_TYPE_INT24: | |
| #ifdef FIELD_TYPE_BIT | |
| case MYSQL_TYPE_BIT: | |
| #endif | |
| case MYSQL_TYPE_DATE: | |
| case MYSQL_TYPE_NEWDATE: | |
| case MYSQL_TYPE_TIME: | |
| case MYSQL_TYPE_DATETIME: | |
| case MYSQL_TYPE_YEAR: | |
| case MYSQL_TYPE_TIMESTAMP: | |
| case MYSQL_TYPE_BLOB: | |
| case MYSQL_TYPE_TINY_BLOB: | |
| case MYSQL_TYPE_MEDIUM_BLOB: | |
| case MYSQL_TYPE_LONG_BLOB: | |
| return false; | |
| break; | |
| default: | |
| return true; | |
| } | |
| } | |
| inline static char* strdup(SQL_Driver_services& services, char* str, size_t length) { | |
| char *strm=(char*)services.malloc_atomic(length+1); | |
| memcpy(strm, str, length); | |
| strm[length]=0; | |
| return strm; | |
| } | |
| struct Connection { | struct Connection { |
| SQL_Driver_services* services; | SQL_Driver_services* services; |
| MYSQL* handle; | MYSQL* handle; |
| const char* cstrClientCharset; | const char* client_charset; |
| bool autocommit; | bool autocommit; |
| }; | }; |
| /** | /** |
| MySQL server driver | MySQL server driver |
| */ | */ |
| class MySQL_Driver : public SQL_Driver { | class MySQL_Driver : public SQL_Driver { |
| public: | public: |
| MySQL_Driver() : SQL_Driver() { | MySQL_Driver() : SQL_Driver() {} |
| } | |
| /// get api version | /// get api version |
| int api_version() { return SQL_DRIVER_API_VERSION; } | int api_version() { return SQL_DRIVER_API_VERSION; } |
| /// initialize driver by loading sql dynamic link library | /// initialize driver by loading sql dynamic link library |
| const char *initialize(char *dlopen_file_spec) { | const char *initialize(char *dlopen_file_spec) { |
| return dlopen_file_spec? | return dlopen_file_spec? |
| dlink(dlopen_file_spec):"client library column is empty"; | dlink(dlopen_file_spec):"client library column is empty"; |
| } | } |
| /** connect | /** connect |
| @param url | @param url |
| format: @b user:pass@host[:port]|[/unix/socket]/database? | format: @b user:pass@host[:port]|[/unix/socket]/database? |
| charset=cp1251_koi8& | charset=value& // transcode by server with command 'SET NAMES value' |
| ClientCharset=charset& // transcode by parser | |
| timeout=3& | timeout=3& |
| compress=1& | compress=0& |
| named_pipe=1 | named_pipe=1& |
| 3.23.22b | autocommit=1& |
| Currently the only option for @b character_set_name is cp1251_koi8. | multi_statements=0 // allows more then one statement in one query |
| WARNING: must be used only to connect, for buffer doesn't live long | 4.1+ accept not 'cp1251_koi8' but 'cp1251', 'utf8' and much more |
| it is usable for transcoding using sql server | |
| */ | */ |
| void connect( | void connect(char *url, SQL_Driver_services& services, void **connection_ref /*< output: Connection* */){ |
| char *url, | |
| SQL_Driver_services& services, | |
| void **connection_ref ///< output: Connection* | |
| ) { | |
| char *user=url; | char *user=url; |
| char *s=lsplit(user, '@'); | char *s=rsplit(user, '@'); |
| char *host=0; | char *host=0; |
| char *unix_socket=0; | char *unix_socket=0; |
| if(s && s[0]=='[') { // unix socket | if(s && s[0]=='[') { // unix socket |
| Line 104 public: | Line 159 public: |
| char *db=lsplit(s, '/'); | char *db=lsplit(s, '/'); |
| char *pwd=lsplit(user, ':'); | char *pwd=lsplit(user, ':'); |
| char *error_pos=0; | char *error_pos=0; |
| char *port_cstr=lsplit(host, ':'); | |
| int port=port_cstr?strtol(port_cstr, &error_pos, 0):0; | |
| char *options=lsplit(db, '?'); | char *options=lsplit(db, '?'); |
| char *charset=0; | |
| int client_flag=CLIENT_MULTI_RESULTS; | |
| char *cstrBackwardCompAskServerToTranscode=0; | Connection& connection=*(Connection *)services.malloc(sizeof(Connection)); |
| *connection_ref=&connection; | |
| Connection& connection=*(Connection *)services.malloc(sizeof(Connection)); | |
| connection.services=&services; | connection.services=&services; |
| connection.handle=mysql_init(NULL); | connection.handle=mysql_init(NULL); |
| connection.cstrClientCharset=0; | connection.client_charset=0; |
| connection.autocommit=true; | connection.autocommit=true; |
| *connection_ref=&connection; | |
| while(options) { | while(1){ |
| if(char *key=lsplit(&options, '&')) { | char *next_host=lsplit(host, ','); |
| if(*key) { | char *host_options=next_host && options ? strdup(services, options, strlen(options)) : options; |
| if(char *value=lsplit(key, '=')) { | |
| if(strcmp(key, "ClientCharset" ) == 0) { | char *port_cstr=lsplit(host, ':'); |
| int port=port_cstr?strtol(port_cstr, &error_pos, 0):0; | |
| while(host_options){ | |
| char *key=lsplit(&host_options, '&'); | |
| if(key && *key){ | |
| if(char *value=lsplit(key, '=')){ | |
| if(strcmp(key, "ClientCharset")==0){ // transcoding with parser | |
| toupper_str(value, value, strlen(value)); | toupper_str(value, value, strlen(value)); |
| connection.cstrClientCharset=value; | connection.client_charset=value; |
| } else if(strcasecmp(key, "charset")==0) { // left for backward compatibility, consider using ClientCharset | } else if(strcasecmp(key, "charset")==0){ // transcoding with server |
| cstrBackwardCompAskServerToTranscode=value; | charset=value; |
| } else if(strcasecmp(key, "timeout")==0) { | } else if(strcasecmp(key, "timeout")==0){ |
| unsigned int timeout=(unsigned int)atoi(value); | unsigned int timeout=(unsigned int)atoi(value); |
| if(mysql_options(connection.handle, MYSQL_OPT_CONNECT_TIMEOUT, (const char *)&timeout)!=0) | if(mysql_options(connection.handle, MYSQL_OPT_CONNECT_TIMEOUT, (const char *)&timeout)!=0) |
| services._throw(mysql_error(connection.handle)); | services._throw(mysql_error(connection.handle)); |
| } else if(strcasecmp(key, "compress")==0) { | } else if(strcasecmp(key, "compress")==0){ |
| if(atoi(value)) | if(atoi(value)) |
| if(mysql_options(connection.handle, MYSQL_OPT_COMPRESS, 0)!=0) | if(mysql_options(connection.handle, MYSQL_OPT_COMPRESS, 0)!=0) |
| services._throw(mysql_error(connection.handle)); | services._throw(mysql_error(connection.handle)); |
| } else if(strcasecmp(key, "named_pipe")==0) { | } else if(strcasecmp(key, "named_pipe")==0){ |
| if(atoi(value)) | if(atoi(value)) |
| if(mysql_options(connection.handle, MYSQL_OPT_NAMED_PIPE , 0)!=0) | if(mysql_options(connection.handle, MYSQL_OPT_NAMED_PIPE, 0)!=0) |
| services._throw(mysql_error(connection.handle)); | services._throw(mysql_error(connection.handle)); |
| } else if(strcasecmp(key, "autocommit")==0) { | } else if(strcasecmp(key, "local_infile")==0){ |
| if(atoi(value)==0) { | if(atoi(value)) |
| if(mysql_options(connection.handle, MYSQL_OPT_LOCAL_INFILE, 0)!=0) | |
| services._throw(mysql_error(connection.handle)); | |
| } else if(strcasecmp(key, "autocommit")==0){ | |
| if(atoi(value)==0) | |
| connection.autocommit=false; | connection.autocommit=false; |
| } | } else if(strcasecmp(key, "multi_statements")==0){ |
| if(atoi(value)!=0) | |
| client_flag=CLIENT_MULTI_STATEMENTS; | |
| } else | } else |
| services._throw("unknown connect option" /*key*/); | services._throw("unknown connect option" /*key*/); |
| } else | } else |
| services._throw("connect option without =value" /*key*/); | services._throw("connect option without =value" /*key*/); |
| } | } |
| } | } |
| if(mysql_real_connect(connection.handle, host, user, pwd, db, port, unix_socket, client_flag)) | |
| break; | |
| if(!next_host) | |
| services._throw(mysql_error(connection.handle)); | |
| host=next_host; | |
| } | } |
| if(connection.cstrClientCharset && cstrBackwardCompAskServerToTranscode) | if(charset){ |
| services._throw("use 'ClientCharset' option only, " | char statement[MAX_STRING+1]="SET NAMES "; |
| "'charset' option is obsolete and should not be used with new 'ClientCharset' option"); | strncat(statement, charset, MAX_STRING); |
| _exec(connection, statement); | |
| if(!mysql_real_connect(connection.handle, | |
| host, user, pwd, db, port?port:MYSQL_PORT, unix_socket, 0)) | |
| services._throw(mysql_error(connection.handle)); | |
| if(cstrBackwardCompAskServerToTranscode) { | |
| // set charset | |
| char statement[MAX_STRING]="set CHARACTER SET "; // cp1251_koi8 | |
| strncat(statement, cstrBackwardCompAskServerToTranscode, MAX_STRING); | |
| exec(connection, statement); | |
| } | } |
| if(!connection.autocommit) | if(!connection.autocommit) |
| exec(connection, "set autocommit=0"); | _exec(connection, "SET AUTOCOMMIT=0"); |
| } | |
| void exec(Connection& connection, const char* statement) { | |
| if(mysql_query(connection.handle, statement)) | |
| connection.services->_throw(mysql_error(connection.handle)); | |
| (*mysql_store_result)(connection.handle); // throw out the result [don't need but must call] | |
| } | } |
| void disconnect(void *aconnection) { | void disconnect(void *aconnection) { |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| mysql_close(connection.handle); | mysql_close(connection.handle); |
| connection.handle=0; | connection.handle=0; |
| } | } |
| void commit(void *aconnection) { | void commit(void *aconnection) { |
| //_asm int 3; | |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| if(!connection.autocommit) | if(!connection.autocommit) |
| exec(connection, "commit"); | _exec(connection, "COMMIT"); |
| } | } |
| void rollback(void *aconnection) { | void rollback(void *aconnection) { |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| if(!connection.autocommit) | if(!connection.autocommit) |
| exec(connection, "rollback"); | _exec(connection, "ROLLBACK"); |
| } | } |
| bool ping(void *aconnection) { | bool ping(void *aconnection) { |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| return mysql_ping(connection.handle)==0; | return mysql_ping(connection.handle)==0; |
| } | } |
| const char* quote(void *aconnection, const char *from, unsigned int length) { | // charset here is services.request_charset(), not connection.client_charset |
| // thus we can't use the sql server quoting support | |
| const char* quote(void *aconnection, const char *str, unsigned int length) { | |
| const char* from; | |
| const char* from_end=str+length; | |
| size_t quoted=0; | |
| for(from=str; from<from_end; from++){ | |
| switch (*from) { | |
| case 0: | |
| case '\n': | |
| case '\r': | |
| case '\032': | |
| case '\\': | |
| case '\'': | |
| case '"': | |
| quoted++; | |
| } | |
| } | |
| if(!quoted) | |
| return str; | |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| /* | char *result=(char*)connection.services->malloc_atomic(length + quoted + 1); |
| 3.23.22b | char *to = result; |
| You must allocate the to buffer to be at least length*2+1 bytes long. | |
| (In the worse case, each character may need to be encoded as using two bytes, | for(from=str; from<from_end; from++){ |
| and you need room for the terminating null byte.) | char escape; |
| */ | switch (*from) { |
| char *result=(char*)connection.services->malloc_atomic(length*2+1); | case 0: |
| mysql_escape_string(result, from, length); | escape= '0'; |
| break; | |
| case '\n': | |
| escape= 'n'; | |
| break; | |
| case '\r': | |
| escape= 'r'; | |
| break; | |
| case '\032': | |
| escape= 'Z'; | |
| break; | |
| case '\\': | |
| case '\'': | |
| case '"': | |
| escape= *from; | |
| break; | |
| default: | |
| *to++=*from; | |
| continue; | |
| } | |
| *to++= '\\'; | |
| *to++= escape; | |
| } | |
| *to=0; | |
| return result; | return result; |
| } | } |
| void query( | |
| void *aconnection, | void query(void *aconnection, |
| const char *astatement, unsigned long offset, unsigned long limit, | const char *astatement, |
| SQL_Driver_query_event_handlers& handlers) { | size_t placeholders_count, Placeholder* placeholders, |
| unsigned long offset, unsigned long limit, | |
| SQL_Driver_query_event_handlers& handlers | |
| ){ | |
| Connection& connection=*static_cast<Connection*>(aconnection); | Connection& connection=*static_cast<Connection*>(aconnection); |
| SQL_Driver_services& services=*connection.services; | SQL_Driver_services& services=*connection.services; |
| const char* cstrClientCharset=connection.cstrClientCharset; | |
| MYSQL_RES *res=NULL; | MYSQL_RES *res=NULL; |
| // transcode from $request:charset to connect-string?client_charset | if(placeholders_count>0) |
| if(cstrClientCharset) { | services._throw("bind variables not supported (yet)"); |
| size_t transcoded_statement_size; | |
| services.transcode(astatement, strlen(astatement), | bool transcode_needed=_transcode_required(connection); |
| astatement, transcoded_statement_size, | |
| size_t statement_size=0; | |
| if(transcode_needed) { | |
| statement_size=strlen(astatement); | |
| // transcode query from $request:charset to ?ClientCharset | |
| services.transcode(astatement, statement_size, | |
| astatement, statement_size, | |
| services.request_charset(), | services.request_charset(), |
| cstrClientCharset); | connection.client_charset); |
| } | } |
| const char *statement; | const char *statement; |
| if(offset || limit) { | if(offset || limit!=SQL_NO_LIMIT) { |
| size_t statement_size=strlen(astatement); | if(!statement_size) |
| statement_size=strlen(astatement); | |
| char *statement_limited=(char *)services.malloc_atomic( | char *statement_limited=(char *)services.malloc_atomic( |
| statement_size+MAX_NUMBER*2+8/* limit #,#*/+1); | statement_size+MAX_NUMBER*2+8/* LIMIT #,#*/+1); |
| char *cur=statement_limited; | char *cur=statement_limited; |
| memcpy(cur, astatement, statement_size); cur+=statement_size; | memcpy(cur, astatement, statement_size); cur+=statement_size; |
| cur+=sprintf(cur, " limit "); | cur+=sprintf(cur, " LIMIT "); |
| if(offset) | if(offset) |
| cur+=snprintf(cur, MAX_NUMBER+1, "%u,", offset); | cur+=snprintf(cur, MAX_NUMBER, "%lu,", offset); |
| if(limit) | if(limit!=SQL_NO_LIMIT) |
| cur+=snprintf(cur, MAX_NUMBER, "%u", limit); | cur+=snprintf(cur, MAX_NUMBER, "%lu", limit); |
| statement=statement_limited; | statement=statement_limited; |
| } else | } else |
| statement=astatement; | statement=astatement; |
| if(mysql_query(connection.handle, statement)) | if(mysql_query(connection.handle, statement)) |
| services._throw(mysql_error(connection.handle)); | _throw(connection, mysql_error(connection.handle)); |
| if(!(res=mysql_store_result(connection.handle)) && mysql_field_count(connection.handle)) | if(!(res=mysql_store_result(connection.handle)) && mysql_field_count(connection.handle)) |
| services._throw(mysql_error(connection.handle)); | _throw(connection, mysql_error(connection.handle)); |
| if(!res) // empty result: insert|delete|update|... | if(!res) // empty result: insert|delete|update|... |
| return; | return; |
| int column_count=mysql_num_fields(res); | size_t column_count=mysql_num_fields(res); |
| if(!column_count) // old client | if(!column_count) // old client |
| column_count=mysql_field_count(connection.handle); | column_count=mysql_field_count(connection.handle); |
| if(!column_count) { | if(!column_count){ |
| mysql_free_result(res); | mysql_free_result(res); |
| services._throw("result contains no columns"); | services._throw("result contains no columns"); |
| } | } |
| bool failed=false; | bool failed=false; |
| SQL_Error sql_error; | SQL_Error sql_error; |
| #define CHECK(afailed) \ | |
| if(afailed) { \ | #define CHECK(afailed) \ |
| failed=true; \ | if(afailed) { \ |
| failed=true; \ | |
| goto cleanup; \ | goto cleanup; \ |
| } | } |
| for(int i=0; i<column_count; i++){ | #define DO_FETCH_FIELDS(transcode_column_name) { \ |
| if(MYSQL_FIELD *field=mysql_fetch_field(res)) { | for(size_t i=0; i<column_count; i++) { \ |
| size_t length=strlen(field->name); | if(MYSQL_FIELD *field = mysql_fetch_field(res)){ \ |
| char* strm=(char*)services.malloc_atomic(length+1); | size_t length=field->name_length; \ |
| memcpy(strm, field->name, length+1); | const char* str=strdup(services, field->name, length); \ |
| const char* str=strm; | transcode_column_name \ |
| CHECK(handlers.add_column(sql_error, str, length)); \ | |
| } else { \ | |
| /* seen broken client, that reported "44" column count for "select 2+2" */ \ | |
| column_count=i; \ | |
| break; \ | |
| } \ | |
| } \ | |
| } | |
| #define DO_FETCH_ROWS(transcode_cell_value) { \ | |
| while(MYSQL_ROW mysql_row=mysql_fetch_row(res)) { \ | |
| CHECK(handlers.add_row(sql_error)); \ | |
| unsigned long *lengths=mysql_fetch_lengths(res); \ | |
| for(size_t i=0; i<column_count; i++) { \ | |
| const char* str=0; \ | |
| size_t length=lengths[i]; \ | |
| if(length) { \ | |
| str=strdup(services, mysql_row[i], length); \ | |
| transcode_cell_value \ | |
| } \ | |
| CHECK(handlers.add_row_cell(sql_error, str, length)); \ | |
| } \ | |
| } \ | |
| } | |
| // transcode to $request:charset from connect-string?client_charset | bool* transcode_column=0; |
| if(cstrClientCharset) { | if(transcode_needed) { |
| transcode_column = new bool[column_count]; | |
| DO_FETCH_FIELDS( | |
| transcode_column[i] = is_column_transcode_required(field->type); | |
| // transcode column's name from ?ClientCharset to $request:charset | |
| services.transcode(str, length, | |
| str, length, | |
| connection.client_charset, | |
| services.request_charset()); | |
| ) | |
| CHECK(handlers.before_rows(sql_error)); | |
| DO_FETCH_ROWS( | |
| if(transcode_column[i]) | |
| // transcode cell's value from ?ClientCharset to $request:charset | |
| services.transcode(str, length, | services.transcode(str, length, |
| str, length, | str, length, |
| cstrClientCharset, | connection.client_charset, |
| services.request_charset()); | services.request_charset()); |
| } | ) |
| } else { | |
| CHECK(handlers.add_column(sql_error, str, length)); | // without transcoding |
| } else { | DO_FETCH_FIELDS() |
| // seen some broken client, | CHECK(handlers.before_rows(sql_error)); |
| // which reported "44" for column count of response to "select 2+2" | DO_FETCH_ROWS() |
| column_count=i; | |
| break; | |
| } | |
| } | } |
| CHECK(handlers.before_rows(sql_error)); | |
| while(MYSQL_ROW mysql_row=mysql_fetch_row(res)) { | |
| CHECK(handlers.add_row(sql_error)); | |
| unsigned long *lengths=mysql_fetch_lengths(res); | |
| for(int i=0; i<column_count; i++){ | |
| size_t length=(size_t)lengths[i]; | |
| const char* str; | |
| if(length) { | |
| char *strm=(char*)services.malloc_atomic(length+1); | |
| memcpy(strm, mysql_row[i], length); | |
| strm[length]=0; | |
| str=strm; | |
| // transcode to $request:charset from connect-string?client_charset | |
| if(cstrClientCharset) | |
| services.transcode(str, length, | |
| str, length, | |
| cstrClientCharset, | |
| services.request_charset()); | |
| } else | |
| str=0; | |
| CHECK(handlers.add_row_cell(sql_error, str, length)); | |
| } | |
| } | |
| cleanup: | cleanup: |
| if(transcode_column) | |
| delete transcode_column; | |
| mysql_free_result(res); | mysql_free_result(res); |
| if(failed) | if(failed) |
| services._throw(sql_error); | services._throw(sql_error); |
| } | } |
| private: | |
| void _exec(Connection& connection, const char* statement) { | |
| if(mysql_query(connection.handle, statement)) | |
| _throw(connection, mysql_error(connection.handle)); | |
| (*mysql_store_result)(connection.handle); // throw out the result [don't need but must call] | |
| } | |
| void _throw(Connection& connection, const char* aerr_msg) { | |
| size_t length=strlen(aerr_msg); | |
| if(length && _transcode_required(connection)) { | |
| connection.services->transcode(aerr_msg, length, | |
| aerr_msg, length, | |
| connection.client_charset, | |
| connection.services->request_charset()); | |
| } | |
| connection.services->_throw(aerr_msg); | |
| } | |
| bool _transcode_required(Connection& connection){ | |
| return (connection.client_charset && strcmp(connection.client_charset, connection.services->request_charset())!=0); | |
| } | |
| private: // mysql client library funcs | private: // mysql client library funcs |
| typedef MYSQL* (STDCALL *t_mysql_init)(MYSQL *); t_mysql_init mysql_init; | typedef MYSQL* (STDCALL *t_mysql_init)(MYSQL *); t_mysql_init mysql_init; |
| typedef void (STDCALL *t_mysql_server_end)(); t_mysql_server_end mysql_server_end; | |
| typedef int (STDCALL *t_mysql_options)(MYSQL *mysql, enum mysql_option option, const char *arg); t_mysql_options mysql_options; | typedef int (STDCALL *t_mysql_options)(MYSQL *mysql, enum mysql_option option, const char *arg); t_mysql_options mysql_options; |
| typedef MYSQL_RES* (STDCALL *t_mysql_store_result)(MYSQL *); t_mysql_store_result mysql_store_result; | typedef MYSQL_RES* (STDCALL *t_mysql_store_result)(MYSQL *); t_mysql_store_result mysql_store_result; |
| typedef int (STDCALL *t_mysql_query)(MYSQL *, const char *q); t_mysql_query mysql_query; | typedef int (STDCALL *t_mysql_query)(MYSQL *, const char *q); t_mysql_query mysql_query; |
| typedef char * (STDCALL *t_mysql_error)(MYSQL *); t_mysql_error mysql_error; | typedef char* (STDCALL *t_mysql_error)(MYSQL *); t_mysql_error mysql_error; |
| static char* STDCALL subst_mysql_error(MYSQL *mysql) { return (mysql)->net.last_error; } | static char* STDCALL subst_mysql_error(MYSQL *mysql) { return (mysql)->net.last_error; } |
| typedef MYSQL* (STDCALL *t_mysql_real_connect)(MYSQL *, const char *host, | typedef MYSQL* (STDCALL *t_mysql_real_connect)(MYSQL *, const char *host, const char *user, const char *passwd, const char *db, unsigned int port, const char *unix_socket, unsigned int clientflag); t_mysql_real_connect mysql_real_connect; |
| const char *user, | |
| const char *passwd, | typedef void (STDCALL *t_mysql_close)(MYSQL *); t_mysql_close mysql_close; |
| const char *db, | |
| unsigned int port, | typedef int (STDCALL *t_mysql_ping)(MYSQL *); t_mysql_ping mysql_ping; |
| const char *unix_socket, | |
| unsigned int clientflag); t_mysql_real_connect mysql_real_connect; | typedef unsigned long (STDCALL *t_mysql_escape_string)(char *to,const char *from, unsigned long from_length); t_mysql_escape_string mysql_escape_string; |
| typedef void (STDCALL *t_mysql_close)(MYSQL *); t_mysql_close mysql_close; | typedef void (STDCALL *t_mysql_free_result)(MYSQL_RES *result); t_mysql_free_result mysql_free_result; |
| typedef int (STDCALL *t_mysql_ping)(MYSQL *); t_mysql_ping mysql_ping; | |
| typedef unsigned long (STDCALL *t_mysql_escape_string)(char *to,const char *from, | |
| unsigned long from_length); t_mysql_escape_string mysql_escape_string; | |
| typedef void (STDCALL *t_mysql_free_result)(MYSQL_RES *result); t_mysql_free_result mysql_free_result; | |
| typedef unsigned long* (STDCALL *t_mysql_fetch_lengths)(MYSQL_RES *result); t_mysql_fetch_lengths mysql_fetch_lengths; | typedef unsigned long* (STDCALL *t_mysql_fetch_lengths)(MYSQL_RES *result); t_mysql_fetch_lengths mysql_fetch_lengths; |
| typedef MYSQL_ROW (STDCALL *t_mysql_fetch_row)(MYSQL_RES *result); t_mysql_fetch_row mysql_fetch_row; | |
| typedef MYSQL_FIELD* (STDCALL *t_mysql_fetch_field)(MYSQL_RES *result); t_mysql_fetch_field mysql_fetch_field; | |
| typedef unsigned int (STDCALL *t_mysql_num_fields)(MYSQL_RES *); t_mysql_num_fields mysql_num_fields; | |
| static unsigned int STDCALL subst_mysql_num_fields(MYSQL_RES *res) { return res->field_count; } | |
| typedef MYSQL_ROW (STDCALL *t_mysql_fetch_row)(MYSQL_RES *result); t_mysql_fetch_row mysql_fetch_row; | |
| typedef MYSQL_FIELD* (STDCALL *t_mysql_fetch_field)(MYSQL_RES *result); t_mysql_fetch_field mysql_fetch_field; | |
| typedef unsigned int (STDCALL *t_mysql_num_fields)(MYSQL_RES *); t_mysql_num_fields mysql_num_fields; | |
| typedef unsigned int (STDCALL *t_mysql_field_count)(MYSQL *); t_mysql_field_count mysql_field_count; | typedef unsigned int (STDCALL *t_mysql_field_count)(MYSQL *); t_mysql_field_count mysql_field_count; |
| static unsigned int STDCALL subst_mysql_num_fields(MYSQL_RES *res) { return res->field_count; } | |
| static unsigned int STDCALL subst_mysql_field_count(MYSQL *mysql) { return mysql->field_count; } | static unsigned int STDCALL subst_mysql_field_count(MYSQL *mysql) { return mysql->field_count; } |
| private: // mysql client library funcs linking | private: // mysql client library funcs linking |
| const char *dlink(const char *dlopen_file_spec) { | const char *dlink(const char *dlopen_file_spec) { |
| if(lt_dlinit()) | if(lt_dlinit()){ |
| return lt_dlerror(); | if(const char* result=lt_dlerror()) |
| lt_dlhandle handle=lt_dlopen(dlopen_file_spec); | return result; |
| if (!handle) | return "can not prepare to dynamic loading"; |
| } | |
| lt_dlhandle handle=lt_dlopen(dlopen_file_spec); | |
| if(!handle){ | |
| if(const char* result=lt_dlerror()) | |
| return result; | |
| return "can not open the dynamic link module"; | return "can not open the dynamic link module"; |
| } | |
| #define GLINK(name) \ | |
| name=(t_##name)lt_dlsym(handle, #name); | |
| #define DSLINK(name, action) \ | #define DSLINK(name, action) \ |
| name=(t_##name)lt_dlsym(handle, #name); \ | GLINK(name) \ |
| if(!name) \ | if(!name) \ |
| action; | action; |
| Line 387 private: // mysql client library funcs l | Line 537 private: // mysql client library funcs l |
| #define SLINK(name) DSLINK(name, name=subst_##name) | #define SLINK(name) DSLINK(name, name=subst_##name) |
| DLINK(mysql_init); | DLINK(mysql_init); |
| GLINK(mysql_server_end); | |
| DLINK(mysql_options); | DLINK(mysql_options); |
| DLINK(mysql_store_result); | DLINK(mysql_store_result); |
| DLINK(mysql_query); | DLINK(mysql_query); |
| Line 407 private: // mysql client library funcs l | Line 558 private: // mysql client library funcs l |
| }; | }; |
| extern "C" SQL_Driver *SQL_DRIVER_CREATE() { | extern "C" SQL_Driver *SQL_DRIVER_CREATE() { |
| return new MySQL_Driver(); | static MySQL_Driver Driver; |
| return &Driver; | |
| } | } |