1
/* Copyright (C) 2000-2004 DRIZZLE AB
3
This program is free software; you can redistribute it and/or modify
4
it under the terms of the GNU General Public License as published by
5
the Free Software Foundation.
7
There are special exceptions to the terms and conditions of the GPL as it
8
is applied to this software. View the full text of the exception in file
9
EXCEPTIONS-CLIENT in the directory of this software distribution.
11
This program is distributed in the hope that it will be useful,
12
but WITHOUT ANY WARRANTY; without even the implied warranty of
13
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
GNU General Public License for more details.
16
You should have received a copy of the GNU General Public License
17
along with this program; if not, write to the Free Software
18
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
20
#include <drizzled/global.h>
21
#include <mysys/my_sys.h>
32
#include <sys/socket.h>
33
#include <netinet/in.h>
34
#include <arpa/inet.h>
39
#ifdef HAVE_SYS_SELECT_H
40
#include <sys/select.h>
50
#define INADDR_NONE -1
53
#include <sql_common.h>
54
#include "client_settings.h"
55
#include <drizzled/version.h>
57
#undef net_buffer_length
58
#undef max_allowed_packet
60
uint32_t net_buffer_length= 8192;
61
uint32_t max_allowed_packet= 1024L*1024L*1024L;
64
#define SOCKET_ERROR -1
67
If allowed through some configuration, then this needs to
70
#define MAX_LONG_DATA_LENGTH 8192
71
#define unsigned_field(A) ((A)->flags & UNSIGNED_FLAG)
74
static DRIZZLE_PARAMETERS drizzle_internal_parameters=
75
{&max_allowed_packet, &net_buffer_length, 0};
77
const DRIZZLE_PARAMETERS * drizzle_get_parameters(void)
79
return &drizzle_internal_parameters;
82
bool drizzle_thread_init()
84
return my_thread_init();
87
void drizzle_thread_end()
94
Expand wildcard to a sql string
98
append_wild(char *to, char *end, const char *wild)
100
end-=5; /* Some extra */
103
to= strcpy(to," like '");
104
to+= 7; /* strlen(" like '"); */
106
while (*wild && to < end)
108
if (*wild == '\\' || *wild == '\'')
112
if (*wild) /* Too small buffer */
113
*to++='%'; /* Nicer this way */
119
/**************************************************************************
120
Ignore SIGPIPE handler
122
**************************************************************************/
125
my_pipe_sig_handler(int sig __attribute__((unused)))
127
#ifdef DONT_REMEMBER_SIGNAL
128
(void) signal(SIGPIPE, my_pipe_sig_handler);
133
/**************************************************************************
134
Change user and database
135
**************************************************************************/
137
int cli_read_change_user_result(DRIZZLE *drizzle)
141
pkt_length= cli_safe_read(drizzle);
143
if (pkt_length == packet_error)
149
bool drizzle_change_user(DRIZZLE *drizzle, const char *user,
150
const char *passwd, const char *db)
152
char buff[USERNAME_LENGTH+SCRAMBLED_PASSWORD_CHAR_LENGTH+NAME_LEN+2];
155
const CHARSET_INFO *saved_cs= drizzle->charset;
157
/* Get the connection-default character set. */
159
if (drizzle_init_character_set(drizzle))
161
drizzle->charset= saved_cs;
165
/* Use an empty string instead of NULL. */
172
/* Store user into the buffer */
173
end= strncpy(end, user, USERNAME_LENGTH) + USERNAME_LENGTH + 1;
175
/* write scrambled password according to server capabilities */
179
*end++= SCRAMBLE_LENGTH;
180
end+= SCRAMBLE_LENGTH;
184
*end++= '\0'; /* empty password */
185
/* Add database if needed */
186
end= strncpy(end, db ? db : "", NAME_LEN) + NAME_LEN + 1;
188
/* Add character set number. */
190
if (drizzle->server_capabilities & CLIENT_SECURE_CONNECTION)
192
int2store(end, (ushort) drizzle->charset->number);
196
/* Write authentication package */
197
(void)simple_command(drizzle,COM_CHANGE_USER, (uchar*) buff, (uint32_t) (end-buff), 1);
199
rc= (*drizzle->methods->read_change_user_result)(drizzle);
203
/* Free old connect information */
207
free(drizzle->passwd);
211
/* alloc new connect information */
212
drizzle->user= strdup(user);
213
drizzle->passwd= strdup(passwd);
214
drizzle->db= db ? strdup(db) : 0;
218
drizzle->charset= saved_cs;
224
#if defined(HAVE_GETPWUID) && defined(NO_GETPWUID_DECL)
225
struct passwd *getpwuid(uid_t);
226
char* getlogin(void);
229
/**************************************************************************
230
Do a query. If query returned rows, free old rows.
231
Read data by drizzle_store_result or by repeat call of drizzle_fetch_row
232
**************************************************************************/
235
drizzle_query(DRIZZLE *drizzle, const char *query)
237
return drizzle_real_query(drizzle,query, (uint) strlen(query));
241
/**************************************************************************
242
Return next field of the query results
243
**************************************************************************/
246
drizzle_fetch_field(DRIZZLE_RES *result)
248
if (result->current_field >= result->field_count)
250
return &result->fields[result->current_field++];
254
/**************************************************************************
255
Move to a specific row and column
256
**************************************************************************/
259
drizzle_data_seek(DRIZZLE_RES *result, uint64_t row)
263
for (tmp=result->data->data; row-- && tmp ; tmp = tmp->next) ;
264
result->current_row=0;
265
result->data_cursor = tmp;
269
/*************************************************************************
270
put the row or field cursor one a position one got from DRIZZLE_ROW_tell()
271
This doesn't restore any data. The next drizzle_fetch_row or
272
drizzle_fetch_field will return the next row or field after the last used
273
*************************************************************************/
276
drizzle_row_seek(DRIZZLE_RES *result, DRIZZLE_ROW_OFFSET row)
278
DRIZZLE_ROW_OFFSET return_value=result->data_cursor;
279
result->current_row= 0;
280
result->data_cursor= row;
286
drizzle_field_seek(DRIZZLE_RES *result, DRIZZLE_FIELD_OFFSET field_offset)
288
DRIZZLE_FIELD_OFFSET return_value=result->current_field;
289
result->current_field=field_offset;
294
/*****************************************************************************
295
List all tables in a database
296
If wild is given then only the tables matching wild is returned
297
*****************************************************************************/
300
drizzle_list_tables(DRIZZLE *drizzle, const char *wild)
303
char *ptr= strcpy(buff, "show tables");
304
ptr+= 11; /* strlen("show tables"); */
306
append_wild(ptr,buff+sizeof(buff),wild);
307
if (drizzle_query(drizzle,buff))
309
return (drizzle_store_result(drizzle));
313
DRIZZLE_FIELD *cli_list_fields(DRIZZLE *drizzle)
316
if (!(query= cli_read_rows(drizzle,(DRIZZLE_FIELD*) 0, 8)))
319
drizzle->field_count= (uint) query->rows;
320
return unpack_fields(query,&drizzle->field_alloc, drizzle->field_count, 1);
324
/**************************************************************************
325
List all fields in a table
326
If wild is given then only the fields matching wild is returned
327
Instead of this use query:
328
show fields in 'table' like "wild"
329
**************************************************************************/
332
drizzle_list_fields(DRIZZLE *drizzle, const char *table, const char *wild)
335
DRIZZLE_FIELD *fields;
336
char buff[257], *end;
338
end= strncpy(buff, table, 128) + 128;
339
end= strncpy(end+1, wild ? wild : "", 128) + 128;
341
free_old_query(drizzle);
342
if (simple_command(drizzle, COM_FIELD_LIST, (uchar*) buff,
343
(uint32_t) (end-buff), 1) ||
344
!(fields= (*drizzle->methods->list_fields)(drizzle)))
347
if (!(result = (DRIZZLE_RES *) malloc(sizeof(DRIZZLE_RES))))
350
memset(result, 0, sizeof(DRIZZLE_RES));
352
result->methods= drizzle->methods;
353
result->field_alloc=drizzle->field_alloc;
355
result->field_count = drizzle->field_count;
356
result->fields= fields;
361
/* List all running processes (threads) in server */
364
drizzle_list_processes(DRIZZLE *drizzle)
366
DRIZZLE_DATA *fields;
370
if (simple_command(drizzle,COM_PROCESS_INFO,0,0,0))
372
free_old_query(drizzle);
373
pos=(uchar*) drizzle->net.read_pos;
374
field_count=(uint) net_field_length(&pos);
375
if (!(fields = (*drizzle->methods->read_rows)(drizzle,(DRIZZLE_FIELD*) 0, 7)))
377
if (!(drizzle->fields=unpack_fields(fields,&drizzle->field_alloc,field_count,0)))
379
drizzle->status=DRIZZLE_STATUS_GET_RESULT;
380
drizzle->field_count=field_count;
381
return(drizzle_store_result(drizzle));
386
drizzle_shutdown(DRIZZLE *drizzle, enum drizzle_enum_shutdown_level shutdown_level)
389
level[0]= (uchar) shutdown_level;
390
return(simple_command(drizzle, COM_SHUTDOWN, level, 1, 0));
395
drizzle_refresh(DRIZZLE *drizzle,uint options)
398
bits[0]= (uchar) options;
399
return(simple_command(drizzle, COM_REFRESH, bits, 1, 0));
404
drizzle_kill(DRIZZLE *drizzle, uint32_t pid)
408
return(simple_command(drizzle,COM_PROCESS_KILL,buff,sizeof(buff),0));
413
drizzle_set_server_option(DRIZZLE *drizzle, enum enum_drizzle_set_option option)
416
int2store(buff, (uint) option);
417
return(simple_command(drizzle, COM_SET_OPTION, buff, sizeof(buff), 0));
421
const char *cli_read_statistics(DRIZZLE *drizzle)
423
drizzle->net.read_pos[drizzle->packet_length]=0; /* End of stat string */
424
if (!drizzle->net.read_pos[0])
426
set_drizzle_error(drizzle, CR_WRONG_HOST_INFO, unknown_sqlstate);
427
return drizzle->net.last_error;
429
return (char*) drizzle->net.read_pos;
434
drizzle_ping(DRIZZLE *drizzle)
437
res= simple_command(drizzle,COM_PING,0,0,0);
438
if (res == CR_SERVER_LOST && drizzle->reconnect)
439
res= simple_command(drizzle,COM_PING,0,0,0);
445
drizzle_get_server_info(const DRIZZLE *drizzle)
447
return((char*) drizzle->server_version);
452
drizzle_get_host_info(const DRIZZLE *drizzle)
454
return(drizzle->host_info);
459
drizzle_get_proto_info(const DRIZZLE *drizzle)
461
return (drizzle->protocol_version);
465
drizzle_get_client_info(void)
467
return (char*) DRIZZLE_SERVER_VERSION;
470
uint32_t drizzle_get_client_version(void)
472
return DRIZZLE_VERSION_ID;
475
bool drizzle_eof(const DRIZZLE_RES *res)
480
const DRIZZLE_FIELD * drizzle_fetch_field_direct(const DRIZZLE_RES *res, unsigned int fieldnr)
482
return &(res)->fields[fieldnr];
485
const DRIZZLE_FIELD * drizzle_fetch_fields(const DRIZZLE_RES *res)
490
DRIZZLE_ROW_OFFSET drizzle_row_tell(const DRIZZLE_RES *res)
492
return res->data_cursor;
495
DRIZZLE_FIELD_OFFSET drizzle_field_tell(const DRIZZLE_RES *res)
497
return res->current_field;
502
unsigned int drizzle_field_count(const DRIZZLE *drizzle)
504
return drizzle->field_count;
507
uint64_t drizzle_affected_rows(const DRIZZLE *drizzle)
509
return drizzle->affected_rows;
512
uint64_t drizzle_insert_id(const DRIZZLE *drizzle)
514
return drizzle->insert_id;
517
const char * drizzle_sqlstate(const DRIZZLE *drizzle)
519
return drizzle ? drizzle->net.sqlstate : cant_connect_sqlstate;
522
uint32_t drizzle_warning_count(const DRIZZLE *drizzle)
524
return drizzle->warning_count;
527
const char * drizzle_info(const DRIZZLE *drizzle)
529
return drizzle->info;
532
uint32_t drizzle_thread_id(const DRIZZLE *drizzle)
534
return drizzle->thread_id;
537
const char * drizzle_character_set_name(const DRIZZLE *drizzle)
539
return drizzle->charset->csname;
542
void drizzle_get_character_set_info(const DRIZZLE *drizzle, MY_CHARSET_INFO *csinfo)
544
csinfo->number = drizzle->charset->number;
545
csinfo->state = drizzle->charset->state;
546
csinfo->csname = drizzle->charset->csname;
547
csinfo->name = drizzle->charset->name;
548
csinfo->comment = drizzle->charset->comment;
549
csinfo->mbminlen = drizzle->charset->mbminlen;
550
csinfo->mbmaxlen = drizzle->charset->mbmaxlen;
552
if (drizzle->options.charset_dir)
553
csinfo->dir = drizzle->options.charset_dir;
555
csinfo->dir = charsets_dir;
558
uint drizzle_thread_safe(void)
564
bool drizzle_embedded(void)
569
/****************************************************************************
570
Some support functions
571
****************************************************************************/
574
Functions called my my_net_init() to set some application specific variables
577
void my_net_local_init(NET *net)
579
net->max_packet= (uint) net_buffer_length;
580
my_net_set_read_timeout(net, CLIENT_NET_READ_TIMEOUT);
581
my_net_set_write_timeout(net, CLIENT_NET_WRITE_TIMEOUT);
583
net->max_packet_size= max(net_buffer_length, max_allowed_packet);
587
This function is used to create HEX string that you
588
can use in a SQL statement in of the either ways:
589
INSERT INTO blob_column VALUES (0xAABBCC); (any DRIZZLE version)
590
INSERT INTO blob_column VALUES (X'AABBCC');
592
The string in "from" is encoded to a HEX string.
593
The result is placed in "to" and a terminating null byte is appended.
595
The string pointed to by "from" must be "length" bytes long.
596
You must allocate the "to" buffer to be at least length*2+1 bytes long.
597
Each character needs two bytes, and you need room for the terminating
598
null byte. When drizzle_hex_string() returns, the contents of "to" will
599
be a null-terminated string. The return value is the length of the
600
encoded string, not including the terminating null character.
602
The return value does not contain any leading 0x or a leading X' and
603
trailing '. The caller must supply whichever of those is desired.
607
drizzle_hex_string(char *to, const char *from, uint32_t length)
612
for (end= from + length; from < end; from++)
614
*to++= _dig_vec_upper[((unsigned char) *from) >> 4];
615
*to++= _dig_vec_upper[((unsigned char) *from) & 0x0F];
618
return (uint32_t) (to-to0);
622
Add escape characters to a string (blob?) to make it suitable for a insert
623
to should at least have place for length*2+1 chars
624
Returns the length of the to string
628
drizzle_escape_string(char *to,const char *from, uint32_t length)
630
return escape_string_for_drizzle(default_charset_info, to, 0, from, length);
634
drizzle_real_escape_string(DRIZZLE *drizzle, char *to,const char *from,
637
if (drizzle->server_status & SERVER_STATUS_NO_BACKSLASH_ESCAPES)
638
return escape_quotes_for_drizzle(drizzle->charset, to, 0, from, length);
639
return escape_string_for_drizzle(drizzle->charset, to, 0, from, length);
643
myodbc_remove_escape(const DRIZZLE *drizzle, char *name)
647
bool use_mb_flag= use_mb(drizzle->charset);
650
for (end=name; *end ; end++) ;
653
for (to=name ; *name ; name++)
657
if (use_mb_flag && (l = my_ismbchar( drizzle->charset, name , end ) ) )
665
if (*name == '\\' && name[1])
672
int cli_unbuffered_fetch(DRIZZLE *drizzle, char **row)
674
if (packet_error == cli_safe_read(drizzle))
677
*row= ((drizzle->net.read_pos[0] == DRIZZLE_PROTOCOL_NO_MORE_DATA) ? NULL :
678
(char*) (drizzle->net.read_pos+1));
682
/********************************************************************
684
*********************************************************************/
687
Commit the current transaction
690
bool drizzle_commit(DRIZZLE *drizzle)
692
return((bool) drizzle_real_query(drizzle, "commit", 6));
696
Rollback the current transaction
699
bool drizzle_rollback(DRIZZLE *drizzle)
701
return((bool) drizzle_real_query(drizzle, "rollback", 8));
706
Set autocommit to either true or false
709
bool drizzle_autocommit(DRIZZLE *drizzle, bool auto_mode)
711
return((bool) drizzle_real_query(drizzle, auto_mode ?
712
"set autocommit=1":"set autocommit=0",
717
/********************************************************************
718
Multi query execution + SPs APIs
719
*********************************************************************/
722
Returns true/false to indicate whether any more query results exist
723
to be read using drizzle_next_result()
726
bool drizzle_more_results(const DRIZZLE *drizzle)
728
return (drizzle->server_status & SERVER_MORE_RESULTS_EXISTS) ? true:false;
733
Reads and returns the next query results
735
int drizzle_next_result(DRIZZLE *drizzle)
737
if (drizzle->status != DRIZZLE_STATUS_READY)
739
set_drizzle_error(drizzle, CR_COMMANDS_OUT_OF_SYNC, unknown_sqlstate);
743
net_clear_error(&drizzle->net);
744
drizzle->affected_rows= ~(uint64_t) 0;
746
if (drizzle->server_status & SERVER_MORE_RESULTS_EXISTS)
747
return((*drizzle->methods->next_result)(drizzle));
749
return(-1); /* No more results */
753
DRIZZLE_RES * drizzle_use_result(DRIZZLE *drizzle)
755
return (*drizzle->methods->use_result)(drizzle);
758
bool drizzle_read_query_result(DRIZZLE *drizzle)
760
return (*drizzle->methods->read_query_result)(drizzle);