« back to all changes in this revision
Viewing changes to libdrizzleclient/local_infile.c
- browse files at revision 919.1.3
- compare with another revision
- download diff
- download tarball
- view history from revision 919.1.3
- Committer: Brian Aker
- Date: 2009-03-07 17:09:37 UTC
- mto: This revision was merged to the branch mainline in revision 920.
- Revision ID: brian@tangent.org-20090307170937-tnd4eq9tlkvfpddh
Remove client code around LOAD DATE LOCAL
added
removed
libdrizzleclient/local_infile.c
1
/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
*
4
* Copyright (C) 2008 Sun Microsystems, Inc.
5
*
6
* This program is free software; you can redistribute it and/or modify
7
* it under the terms of the GNU General Public License as published by
8
* the Free Software Foundation; version 2 of the License.
9
*
10
* This program is distributed in the hope that it will be useful,
11
* but WITHOUT ANY WARRANTY; without even the implied warranty of
12
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
* GNU General Public License for more details.
14
*
15
* You should have received a copy of the GNU General Public License
16
* along with this program; if not, write to the Free Software
17
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
18
*/
19
20
#include <drizzled/global.h>
21
22
#include "libdrizzle.h"
23
#include "libdrizzle_priv.h"
24
#include "errmsg.h"
25
#include <sys/stat.h>
26
#include <signal.h>
27
#include <time.h>
28
#include <fcntl.h>
29
#include <stdio.h>
30
31
#ifdef HAVE_PWD_H
32
#include <pwd.h>
33
#endif
34
35
#include <sys/socket.h>
36
#include <netinet/in.h>
37
#include <arpa/inet.h>
38
#include <netdb.h>
39
#ifdef HAVE_SELECT_H
40
#include <select.h>
41
#endif
42
#ifdef HAVE_SYS_SELECT_H
43
#include <sys/select.h>
44
#endif
45
46
#ifdef HAVE_POLL
47
#include <sys/poll.h>
48
#endif
49
#ifdef HAVE_SYS_UN_H
50
#include <sys/un.h>
51
#endif
52
#ifndef INADDR_NONE
53
#define INADDR_NONE -1
54
#endif
55
56
#include "local_infile.h"
57
#include <drizzled/gettext.h>
58
59
#define MY_ALIGN(A,L) (((A) + (L) - 1) & ~((L) - 1))
60
61
bool drizzleclient_handle_local_infile(DRIZZLE *drizzle, const char *net_filename)
62
{
63
bool result= true;
64
uint32_t packet_length=MY_ALIGN(drizzle->net.max_packet-16,IO_SIZE);
65
NET *net= &drizzle->net;
66
int readcount;
67
void *li_ptr; /* pass state to local_infile functions */
68
char *buf; /* buffer to be filled by local_infile_read */
69
struct st_drizzleclient_options *options= &drizzle->options;
70
71
/* check that we've got valid callback functions */
72
if (!(options->local_infile_init &&
73
options->local_infile_read &&
74
options->local_infile_end &&
75
options->local_infile_error))
76
{
77
/* if any of the functions is invalid, set the default */
78
drizzleclient_set_local_infile_default(drizzle);
79
}
80
81
/* copy filename into local memory and allocate read buffer */
82
if (!(buf=malloc(packet_length)))
83
{
84
drizzleclient_set_error(drizzle, CR_OUT_OF_MEMORY, drizzleclient_sqlstate_get_unknown());
85
return(1);
86
}
87
88
/* initialize local infile (open file, usually) */
89
if ((*options->local_infile_init)(&li_ptr, net_filename,
90
options->local_infile_userdata))
91
{
92
(void)drizzleclient_net_write(net,(const unsigned char*) "",0); /* Server needs one packet */
93
drizzleclient_net_flush(net);
94
strcpy(net->sqlstate, drizzleclient_sqlstate_get_unknown());
95
net->last_errno=
96
(*options->local_infile_error)(li_ptr,
97
net->last_error,
98
sizeof(net->last_error)-1);
99
goto err;
100
}
101
102
/* read blocks of data from local infile callback */
103
while ((readcount =
104
(*options->local_infile_read)(li_ptr, buf,
105
packet_length)) > 0)
106
{
107
if (drizzleclient_net_write(net, (unsigned char*) buf, readcount))
108
{
109
goto err;
110
}
111
}
112
113
/* Send empty packet to mark end of file */
114
if (drizzleclient_net_write(net, (const unsigned char*) "", 0) || drizzleclient_net_flush(net))
115
{
116
drizzleclient_set_error(drizzle, CR_SERVER_LOST, drizzleclient_sqlstate_get_unknown());
117
goto err;
118
}
119
120
if (readcount < 0)
121
{
122
net->last_errno=
123
(*options->local_infile_error)(li_ptr,
124
net->last_error,
125
sizeof(net->last_error)-1);
126
goto err;
127
}
128
129
result=false; /* Ok */
130
131
err:
132
/* free up memory allocated with _init, usually */
133
(*options->local_infile_end)(li_ptr);
134
if(buf)
135
free(buf);
136
return(result);
137
}
138
139
140
/****************************************************************************
141
Default handlers for LOAD LOCAL INFILE
142
****************************************************************************/
143
144
typedef struct st_default_local_infile
145
{
146
int fd;
147
int error_num;
148
const char *filename;
149
char error_msg[LOCAL_INFILE_ERROR_LEN];
150
} default_local_infile_data;
151
152
153
/*
154
Open file for LOAD LOCAL INFILE
155
156
SYNOPSIS
157
default_local_infile_init()
158
ptr Store pointer to internal data here
159
filename File name to open. This may be in unix format !
160
161
162
NOTES
163
Even if this function returns an error, the load data interface
164
guarantees that default_local_infile_end() is called.
165
166
RETURN
167
0 ok
168
1 error
169
*/
170
171
static int default_local_infile_init(void **ptr, const char *filename,
172
void *userdata)
173
{
174
(void)userdata;
175
default_local_infile_data *data;
176
char tmp_name[FN_REFLEN];
177
178
if (!(*ptr= data= ((default_local_infile_data *)
179
malloc(sizeof(default_local_infile_data)))))
180
return 1; /* out of memory */
181
182
data->error_msg[0]= 0;
183
data->error_num= 0;
184
data->filename= filename;
185
186
if ((data->fd = open(tmp_name, O_RDONLY)) < 0)
187
{
188
data->error_num= errno;
189
snprintf(data->error_msg, sizeof(data->error_msg)-1,
190
_("File '%s' not found (Errcode: %d)"), tmp_name, data->error_num);
191
return 1;
192
}
193
return 0; /* ok */
194
}
195
196
197
/*
198
Read data for LOAD LOCAL INFILE
199
200
SYNOPSIS
201
default_local_infile_read()
202
ptr Points to handle allocated by _init
203
buf Read data here
204
buf_len Ammount of data to read
205
206
RETURN
207
> 0 number of bytes read
208
== 0 End of data
209
< 0 Error
210
*/
211
212
static int default_local_infile_read(void *ptr, char *buf, uint32_t buf_len)
213
{
214
int count;
215
default_local_infile_data*data = (default_local_infile_data *) ptr;
216
217
if ((count= (int) read(data->fd, (unsigned char *) buf, buf_len)) < 0)
218
{
219
data->error_num= 2; /* the errmsg for not entire file read */
220
snprintf(data->error_msg, sizeof(data->error_msg)-1,
221
_("Error reading file '%s' (Errcode: %d)"),
222
data->filename, errno);
223
}
224
return count;
225
}
226
227
228
/*
229
Read data for LOAD LOCAL INFILE
230
231
SYNOPSIS
232
default_local_infile_end()
233
ptr Points to handle allocated by _init
234
May be NULL if _init failed!
235
236
RETURN
237
*/
238
239
static void default_local_infile_end(void *ptr)
240
{
241
default_local_infile_data *data= (default_local_infile_data *) ptr;
242
if (data) /* If not error on open */
243
{
244
if (data->fd >= 0)
245
close(data->fd);
246
free(ptr);
247
}
248
}
249
250
251
/*
252
Return error from LOAD LOCAL INFILE
253
254
SYNOPSIS
255
default_local_infile_end()
256
ptr Points to handle allocated by _init
257
May be NULL if _init failed!
258
error_msg Store error text here
259
error_msg_len Max lenght of error_msg
260
261
RETURN
262
error message number
263
*/
264
265
static int
266
default_local_infile_error(void *ptr, char *error_msg, uint32_t error_msg_len)
267
{
268
default_local_infile_data *data = (default_local_infile_data *) ptr;
269
if (data) /* If not error on open */
270
{
271
strncpy(error_msg, data->error_msg, error_msg_len);
272
return data->error_num;
273
}
274
/* This can only happen if we got error on malloc of handle */
275
strcpy(error_msg, ER(CR_OUT_OF_MEMORY));
276
return CR_OUT_OF_MEMORY;
277
}
278
279
280
void
281
drizzleclient_set_local_infile_handler(DRIZZLE *drizzle,
282
int (*local_infile_init)(void **, const char *,
283
void *),
284
int (*local_infile_read)(void *, char *, uint32_t),
285
void (*local_infile_end)(void *),
286
int (*local_infile_error)(void *, char *, uint32_t),
287
void *userdata)
288
{
289
drizzle->options.local_infile_init= local_infile_init;
290
drizzle->options.local_infile_read= local_infile_read;
291
drizzle->options.local_infile_end= local_infile_end;
292
drizzle->options.local_infile_error= local_infile_error;
293
drizzle->options.local_infile_userdata = userdata;
294
}
295
296
297
void drizzleclient_set_local_infile_default(DRIZZLE *drizzle)
298
{
299
drizzle->options.local_infile_init= default_local_infile_init;
300
drizzle->options.local_infile_read= default_local_infile_read;
301
drizzle->options.local_infile_end= default_local_infile_end;
302
drizzle->options.local_infile_error= default_local_infile_error;
303
}
Loggerhead is a web-based interface for Breezy
Version: 2.0.1