|
1 /* glpsql.c */ |
|
2 |
|
3 /*********************************************************************** |
|
4 * This code is part of GLPK (GNU Linear Programming Kit). |
|
5 * |
|
6 * Author: Heinrich Schuchardt <xypron.glpk@gmx.de>. |
|
7 * |
|
8 * Copyright (C) 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008, |
|
9 * 2009, 2010 Andrew Makhorin, Department for Applied Informatics, |
|
10 * Moscow Aviation Institute, Moscow, Russia. All rights reserved. |
|
11 * E-mail: <mao@gnu.org>. |
|
12 * |
|
13 * GLPK is free software: you can redistribute it and/or modify it |
|
14 * under the terms of the GNU General Public License as published by |
|
15 * the Free Software Foundation, either version 3 of the License, or |
|
16 * (at your option) any later version. |
|
17 * |
|
18 * GLPK is distributed in the hope that it will be useful, but WITHOUT |
|
19 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY |
|
20 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public |
|
21 * License for more details. |
|
22 * |
|
23 * You should have received a copy of the GNU General Public License |
|
24 * along with GLPK. If not, see <http://www.gnu.org/licenses/>. |
|
25 ***********************************************************************/ |
|
26 |
|
27 #ifdef HAVE_CONFIG_H |
|
28 #include <config.h> |
|
29 #endif |
|
30 |
|
31 #include "glpmpl.h" |
|
32 #include "glpsql.h" |
|
33 |
|
34 #ifdef ODBC_DLNAME |
|
35 #define HAVE_ODBC |
|
36 #define libodbc ODBC_DLNAME |
|
37 #define h_odbc (get_env_ptr()->h_odbc) |
|
38 #endif |
|
39 |
|
40 #ifdef MYSQL_DLNAME |
|
41 #define HAVE_MYSQL |
|
42 #define libmysql MYSQL_DLNAME |
|
43 #define h_mysql (get_env_ptr()->h_mysql) |
|
44 #endif |
|
45 |
|
46 static void *db_iodbc_open_int(TABDCA *dca, int mode, const char |
|
47 **sqllines); |
|
48 static void *db_mysql_open_int(TABDCA *dca, int mode, const char |
|
49 **sqllines); |
|
50 |
|
51 /**********************************************************************/ |
|
52 |
|
53 #if defined(HAVE_ODBC) || defined(HAVE_MYSQL) |
|
54 |
|
55 #define SQL_FIELD_MAX 100 |
|
56 /* maximal field count */ |
|
57 |
|
58 #define SQL_FDLEN_MAX 255 |
|
59 /* maximal field length */ |
|
60 |
|
61 /*********************************************************************** |
|
62 * NAME |
|
63 * |
|
64 * args_concat - concatenate arguments |
|
65 * |
|
66 * SYNOPSIS |
|
67 * |
|
68 * static char **args_concat(TABDCA *dca); |
|
69 * |
|
70 * DESCRIPTION |
|
71 * |
|
72 * The arguments passed in dca are SQL statements. A SQL statement may |
|
73 * be split over multiple arguments. The last argument of a SQL |
|
74 * statement will be terminated with a semilocon. Each SQL statement is |
|
75 * merged into a single zero terminated string. Boundaries between |
|
76 * arguments are replaced by space. |
|
77 * |
|
78 * RETURNS |
|
79 * |
|
80 * Buffer with SQL statements */ |
|
81 |
|
82 static char **args_concat(TABDCA *dca) |
|
83 { |
|
84 const char *arg; |
|
85 int i; |
|
86 int j; |
|
87 int j0; |
|
88 int j1; |
|
89 int len; |
|
90 int lentot; |
|
91 int narg; |
|
92 int nline = 0; |
|
93 void *ret; |
|
94 char **sqllines = NULL; |
|
95 |
|
96 narg = mpl_tab_num_args(dca); |
|
97 /* The SQL statements start with argument 3. */ |
|
98 if (narg < 3) |
|
99 return NULL; |
|
100 /* Count the SQL statements */ |
|
101 for (j = 3; j <= narg; j++) |
|
102 { |
|
103 arg = mpl_tab_get_arg(dca, j); |
|
104 len = strlen(arg); |
|
105 if (arg[len-1] == ';' || j == narg) |
|
106 nline ++; |
|
107 } |
|
108 /* Allocate string buffer. */ |
|
109 sqllines = (char **) xmalloc((nline+1) * sizeof(char **)); |
|
110 /* Join arguments */ |
|
111 sqllines[0] = NULL; |
|
112 j0 = 3; |
|
113 i = 0; |
|
114 lentot = 0; |
|
115 for (j = 3; j <= narg; j++) |
|
116 { |
|
117 arg = mpl_tab_get_arg(dca, j); |
|
118 len = strlen(arg); |
|
119 lentot += len; |
|
120 if (arg[len-1] == ';' || j == narg) |
|
121 { /* Join arguments for a single SQL statement */ |
|
122 sqllines[i] = xmalloc(lentot+1); |
|
123 sqllines[i+1] = NULL; |
|
124 sqllines[i][0] = 0x00; |
|
125 for (j1 = j0; j1 <= j; j1++) |
|
126 { if(j1>j0) |
|
127 strcat(sqllines[i], " "); |
|
128 strcat(sqllines[i], mpl_tab_get_arg(dca, j1)); |
|
129 } |
|
130 len = strlen(sqllines[i]); |
|
131 if (sqllines[i][len-1] == ';') |
|
132 sqllines[i][len-1] = 0x00; |
|
133 j0 = j+1; |
|
134 i++; |
|
135 lentot = 0; |
|
136 } |
|
137 } |
|
138 return sqllines; |
|
139 } |
|
140 |
|
141 /*********************************************************************** |
|
142 * NAME |
|
143 * |
|
144 * free_buffer - free multiline string buffer |
|
145 * |
|
146 * SYNOPSIS |
|
147 * |
|
148 * static void free_buffer(char **buf); |
|
149 * |
|
150 * DESCRIPTION |
|
151 * |
|
152 * buf is a list of strings terminated by NULL. |
|
153 * The memory for the strings and for the list is released. */ |
|
154 |
|
155 static void free_buffer(char **buf) |
|
156 { int i; |
|
157 |
|
158 for(i = 0; buf[i] != NULL; i++) |
|
159 xfree(buf[i]); |
|
160 xfree(buf); |
|
161 } |
|
162 |
|
163 static int db_escaped_string_length(const char* from) |
|
164 /* length of escaped string */ |
|
165 { |
|
166 int count; |
|
167 const char *pointer; |
|
168 |
|
169 for (pointer = from, count = 0; *pointer != (char) '\0'; pointer++, |
|
170 count++) |
|
171 { |
|
172 switch (*pointer) |
|
173 { |
|
174 case '\'': |
|
175 count++; |
|
176 break; |
|
177 } |
|
178 } |
|
179 |
|
180 return count; |
|
181 } |
|
182 |
|
183 static int db_escape_string (char *to, const char *from) |
|
184 /* escape string*/ |
|
185 { |
|
186 const char *source = from; |
|
187 char *target = to; |
|
188 unsigned int remaining; |
|
189 |
|
190 remaining = strlen(from); |
|
191 |
|
192 if (to == NULL) |
|
193 to = (char *) (from + remaining); |
|
194 |
|
195 while (remaining > 0) |
|
196 { |
|
197 switch (*source) |
|
198 { |
|
199 case '\'': |
|
200 *target = '\''; |
|
201 target++; |
|
202 *target = '\''; |
|
203 break; |
|
204 |
|
205 default: |
|
206 *target = *source; |
|
207 } |
|
208 source++; |
|
209 target++; |
|
210 remaining--; |
|
211 } |
|
212 |
|
213 /* Write the terminating NUL character. */ |
|
214 *target = '\0'; |
|
215 |
|
216 return target - to; |
|
217 } |
|
218 |
|
219 static char *db_generate_select_stmt(TABDCA *dca) |
|
220 /* generate select statement */ |
|
221 { |
|
222 char *arg; |
|
223 char const *field; |
|
224 char *query; |
|
225 int j; |
|
226 int narg; |
|
227 int nf; |
|
228 int total; |
|
229 |
|
230 total = 50; |
|
231 nf = mpl_tab_num_flds(dca); |
|
232 narg = mpl_tab_num_args(dca); |
|
233 for (j=1; j <= nf && j <= SQL_FIELD_MAX; j++) |
|
234 { |
|
235 field = mpl_tab_get_name(dca, j); |
|
236 total += strlen(field); |
|
237 total += 2; |
|
238 } |
|
239 arg = (char *) mpl_tab_get_arg(dca, narg); |
|
240 total += strlen(arg); |
|
241 query = xmalloc( total * sizeof(char)); |
|
242 strcpy (query, "SELECT "); |
|
243 for (j=1; j <= nf && j <= SQL_FIELD_MAX; j++) |
|
244 { |
|
245 field = mpl_tab_get_name(dca, j); |
|
246 strcat(query, field); |
|
247 if ( j < nf ) |
|
248 strcat(query, ", "); |
|
249 } |
|
250 strcat(query, " FROM "); |
|
251 strcat(query, arg); |
|
252 return query; |
|
253 } |
|
254 |
|
255 static char *db_generate_insert_stmt(TABDCA *dca) |
|
256 /* generate insert statement */ |
|
257 { |
|
258 char *arg; |
|
259 char const *field; |
|
260 char *query; |
|
261 int j; |
|
262 int narg; |
|
263 int nf; |
|
264 int total; |
|
265 |
|
266 total = 50; |
|
267 nf = mpl_tab_num_flds(dca); |
|
268 narg = mpl_tab_num_args(dca); |
|
269 for (j=1; j <= nf && j <= SQL_FIELD_MAX; j++) |
|
270 { |
|
271 field = mpl_tab_get_name(dca, j); |
|
272 total += strlen(field); |
|
273 total += 5; |
|
274 } |
|
275 arg = (char *) mpl_tab_get_arg(dca, narg); |
|
276 total += strlen(arg); |
|
277 query = xmalloc( (total+1) * sizeof(char)); |
|
278 strcpy (query, "INSERT INTO "); |
|
279 strcat(query, arg); |
|
280 strcat(query, " ( "); |
|
281 for (j=1; j <= nf && j <= SQL_FIELD_MAX; j++) |
|
282 { |
|
283 field = mpl_tab_get_name(dca, j); |
|
284 strcat(query, field); |
|
285 if ( j < nf ) |
|
286 strcat(query, ", "); |
|
287 } |
|
288 strcat(query, " ) VALUES ( "); |
|
289 for (j=1; j <= nf && j <= SQL_FIELD_MAX; j++) |
|
290 { |
|
291 strcat(query, "?"); |
|
292 if ( j < nf ) |
|
293 strcat(query, ", "); |
|
294 } |
|
295 strcat(query, " )"); |
|
296 return query; |
|
297 } |
|
298 |
|
299 #endif |
|
300 |
|
301 /**********************************************************************/ |
|
302 |
|
303 #ifndef HAVE_ODBC |
|
304 |
|
305 void *db_iodbc_open(TABDCA *dca, int mode) |
|
306 { xassert(dca == dca); |
|
307 xassert(mode == mode); |
|
308 xprintf("iODBC table driver not supported\n"); |
|
309 return NULL; |
|
310 } |
|
311 |
|
312 int db_iodbc_read(TABDCA *dca, void *link) |
|
313 { xassert(dca != dca); |
|
314 xassert(link != link); |
|
315 return 0; |
|
316 } |
|
317 |
|
318 int db_iodbc_write(TABDCA *dca, void *link) |
|
319 { xassert(dca != dca); |
|
320 xassert(link != link); |
|
321 return 0; |
|
322 } |
|
323 |
|
324 int db_iodbc_close(TABDCA *dca, void *link) |
|
325 { xassert(dca != dca); |
|
326 xassert(link != link); |
|
327 return 0; |
|
328 } |
|
329 |
|
330 #else |
|
331 |
|
332 #if defined(__CYGWIN__) || defined(__MINGW32__) || defined(__WOE__) |
|
333 #include <windows.h> |
|
334 #endif |
|
335 |
|
336 #include <sql.h> |
|
337 #include <sqlext.h> |
|
338 |
|
339 struct db_odbc |
|
340 { |
|
341 int mode; /*'R' = Read, 'W' = Write*/ |
|
342 SQLHDBC hdbc; /*connection handle*/ |
|
343 SQLHENV henv; /*environment handle*/ |
|
344 SQLHSTMT hstmt; /*statement handle*/ |
|
345 SQLSMALLINT nresultcols; /* columns in result*/ |
|
346 SQLULEN collen[SQL_FIELD_MAX+1]; |
|
347 SQLLEN outlen[SQL_FIELD_MAX+1]; |
|
348 SQLSMALLINT coltype[SQL_FIELD_MAX+1]; |
|
349 SQLCHAR data[SQL_FIELD_MAX+1][SQL_FDLEN_MAX+1]; |
|
350 SQLCHAR colname[SQL_FIELD_MAX+1][SQL_FDLEN_MAX+1]; |
|
351 int isnumeric[SQL_FIELD_MAX+1]; |
|
352 int nf; |
|
353 /* number of fields in the csv file */ |
|
354 int ref[1+SQL_FIELD_MAX]; |
|
355 /* ref[k] = k', if k-th field of the csv file corresponds to |
|
356 k'-th field in the table statement; if ref[k] = 0, k-th field |
|
357 of the csv file is ignored */ |
|
358 SQLCHAR *query; |
|
359 /* query generated by db_iodbc_open */ |
|
360 }; |
|
361 |
|
362 SQLRETURN SQL_API dl_SQLAllocHandle ( |
|
363 SQLSMALLINT HandleType, |
|
364 SQLHANDLE InputHandle, |
|
365 SQLHANDLE *OutputHandle) |
|
366 { |
|
367 typedef SQLRETURN SQL_API ep_SQLAllocHandle( |
|
368 SQLSMALLINT HandleType, |
|
369 SQLHANDLE InputHandle, |
|
370 SQLHANDLE *OutputHandle); |
|
371 |
|
372 ep_SQLAllocHandle *fn; |
|
373 fn = (ep_SQLAllocHandle *) xdlsym(h_odbc, "SQLAllocHandle"); |
|
374 xassert(fn != NULL); |
|
375 return (*fn)(HandleType, InputHandle, OutputHandle); |
|
376 } |
|
377 |
|
378 SQLRETURN SQL_API dl_SQLBindCol ( |
|
379 SQLHSTMT StatementHandle, |
|
380 SQLUSMALLINT ColumnNumber, |
|
381 SQLSMALLINT TargetType, |
|
382 SQLPOINTER TargetValue, |
|
383 SQLLEN BufferLength, |
|
384 SQLLEN *StrLen_or_Ind) |
|
385 { |
|
386 typedef SQLRETURN SQL_API ep_SQLBindCol( |
|
387 SQLHSTMT StatementHandle, |
|
388 SQLUSMALLINT ColumnNumber, |
|
389 SQLSMALLINT TargetType, |
|
390 SQLPOINTER TargetValue, |
|
391 SQLLEN BufferLength, |
|
392 SQLLEN *StrLen_or_Ind); |
|
393 ep_SQLBindCol *fn; |
|
394 fn = (ep_SQLBindCol *) xdlsym(h_odbc, "SQLBindCol"); |
|
395 xassert(fn != NULL); |
|
396 return (*fn)(StatementHandle, ColumnNumber, TargetType, |
|
397 TargetValue, BufferLength, StrLen_or_Ind); |
|
398 } |
|
399 |
|
400 SQLRETURN SQL_API dl_SQLCloseCursor ( |
|
401 SQLHSTMT StatementHandle) |
|
402 { |
|
403 typedef SQLRETURN SQL_API ep_SQLCloseCursor ( |
|
404 SQLHSTMT StatementHandle); |
|
405 |
|
406 ep_SQLCloseCursor *fn; |
|
407 fn = (ep_SQLCloseCursor *) xdlsym(h_odbc, "SQLCloseCursor"); |
|
408 xassert(fn != NULL); |
|
409 return (*fn)(StatementHandle); |
|
410 } |
|
411 |
|
412 |
|
413 SQLRETURN SQL_API dl_SQLDisconnect ( |
|
414 SQLHDBC ConnectionHandle) |
|
415 { |
|
416 typedef SQLRETURN SQL_API ep_SQLDisconnect( |
|
417 SQLHDBC ConnectionHandle); |
|
418 |
|
419 ep_SQLDisconnect *fn; |
|
420 fn = (ep_SQLDisconnect *) xdlsym(h_odbc, "SQLDisconnect"); |
|
421 xassert(fn != NULL); |
|
422 return (*fn)(ConnectionHandle); |
|
423 } |
|
424 |
|
425 SQLRETURN SQL_API dl_SQLDriverConnect ( |
|
426 SQLHDBC hdbc, |
|
427 SQLHWND hwnd, |
|
428 SQLCHAR *szConnStrIn, |
|
429 SQLSMALLINT cbConnStrIn, |
|
430 SQLCHAR *szConnStrOut, |
|
431 SQLSMALLINT cbConnStrOutMax, |
|
432 SQLSMALLINT *pcbConnStrOut, |
|
433 SQLUSMALLINT fDriverCompletion) |
|
434 { |
|
435 typedef SQLRETURN SQL_API ep_SQLDriverConnect( |
|
436 SQLHDBC hdbc, |
|
437 SQLHWND hwnd, |
|
438 SQLCHAR * szConnStrIn, |
|
439 SQLSMALLINT cbConnStrIn, |
|
440 SQLCHAR * szConnStrOut, |
|
441 SQLSMALLINT cbConnStrOutMax, |
|
442 SQLSMALLINT * pcbConnStrOut, |
|
443 SQLUSMALLINT fDriverCompletion); |
|
444 |
|
445 ep_SQLDriverConnect *fn; |
|
446 fn = (ep_SQLDriverConnect *) xdlsym(h_odbc, "SQLDriverConnect"); |
|
447 xassert(fn != NULL); |
|
448 return (*fn)(hdbc, hwnd, szConnStrIn, cbConnStrIn, szConnStrOut, |
|
449 cbConnStrOutMax, pcbConnStrOut, fDriverCompletion); |
|
450 } |
|
451 |
|
452 SQLRETURN SQL_API dl_SQLEndTran ( |
|
453 SQLSMALLINT HandleType, |
|
454 SQLHANDLE Handle, |
|
455 SQLSMALLINT CompletionType) |
|
456 { |
|
457 typedef SQLRETURN SQL_API ep_SQLEndTran ( |
|
458 SQLSMALLINT HandleType, |
|
459 SQLHANDLE Handle, |
|
460 SQLSMALLINT CompletionType); |
|
461 |
|
462 ep_SQLEndTran *fn; |
|
463 fn = (ep_SQLEndTran *) xdlsym(h_odbc, "SQLEndTran"); |
|
464 xassert(fn != NULL); |
|
465 return (*fn)(HandleType, Handle, CompletionType); |
|
466 } |
|
467 |
|
468 SQLRETURN SQL_API dl_SQLExecDirect ( |
|
469 SQLHSTMT StatementHandle, |
|
470 SQLCHAR * StatementText, |
|
471 SQLINTEGER TextLength) |
|
472 { |
|
473 typedef SQLRETURN SQL_API ep_SQLExecDirect ( |
|
474 SQLHSTMT StatementHandle, |
|
475 SQLCHAR * StatementText, |
|
476 SQLINTEGER TextLength); |
|
477 |
|
478 ep_SQLExecDirect *fn; |
|
479 fn = (ep_SQLExecDirect *) xdlsym(h_odbc, "SQLExecDirect"); |
|
480 xassert(fn != NULL); |
|
481 return (*fn)(StatementHandle, StatementText, TextLength); |
|
482 } |
|
483 |
|
484 SQLRETURN SQL_API dl_SQLFetch ( |
|
485 SQLHSTMT StatementHandle) |
|
486 { |
|
487 typedef SQLRETURN SQL_API ep_SQLFetch ( |
|
488 SQLHSTMT StatementHandle); |
|
489 |
|
490 ep_SQLFetch *fn; |
|
491 fn = (ep_SQLFetch*) xdlsym(h_odbc, "SQLFetch"); |
|
492 xassert(fn != NULL); |
|
493 return (*fn)(StatementHandle); |
|
494 } |
|
495 |
|
496 SQLRETURN SQL_API dl_SQLFreeHandle ( |
|
497 SQLSMALLINT HandleType, |
|
498 SQLHANDLE Handle) |
|
499 { |
|
500 typedef SQLRETURN SQL_API ep_SQLFreeHandle ( |
|
501 SQLSMALLINT HandleType, |
|
502 SQLHANDLE Handle); |
|
503 |
|
504 ep_SQLFreeHandle *fn; |
|
505 fn = (ep_SQLFreeHandle *) xdlsym(h_odbc, "SQLFreeHandle"); |
|
506 xassert(fn != NULL); |
|
507 return (*fn)(HandleType, Handle); |
|
508 } |
|
509 |
|
510 SQLRETURN SQL_API dl_SQLDescribeCol ( |
|
511 SQLHSTMT StatementHandle, |
|
512 SQLUSMALLINT ColumnNumber, |
|
513 SQLCHAR * ColumnName, |
|
514 SQLSMALLINT BufferLength, |
|
515 SQLSMALLINT * NameLength, |
|
516 SQLSMALLINT * DataType, |
|
517 SQLULEN * ColumnSize, |
|
518 SQLSMALLINT * DecimalDigits, |
|
519 SQLSMALLINT * Nullable) |
|
520 { |
|
521 typedef SQLRETURN SQL_API ep_SQLDescribeCol ( |
|
522 SQLHSTMT StatementHandle, |
|
523 SQLUSMALLINT ColumnNumber, |
|
524 SQLCHAR *ColumnName, |
|
525 SQLSMALLINT BufferLength, |
|
526 SQLSMALLINT *NameLength, |
|
527 SQLSMALLINT *DataType, |
|
528 SQLULEN *ColumnSize, |
|
529 SQLSMALLINT *DecimalDigits, |
|
530 SQLSMALLINT *Nullable); |
|
531 |
|
532 ep_SQLDescribeCol *fn; |
|
533 fn = (ep_SQLDescribeCol *) xdlsym(h_odbc, "SQLDescribeCol"); |
|
534 xassert(fn != NULL); |
|
535 return (*fn)(StatementHandle, ColumnNumber, ColumnName, |
|
536 BufferLength, NameLength, |
|
537 DataType, ColumnSize, DecimalDigits, Nullable); |
|
538 } |
|
539 |
|
540 SQLRETURN SQL_API dl_SQLGetDiagRec ( |
|
541 SQLSMALLINT HandleType, |
|
542 SQLHANDLE Handle, |
|
543 SQLSMALLINT RecNumber, |
|
544 SQLCHAR *Sqlstate, |
|
545 SQLINTEGER *NativeError, |
|
546 SQLCHAR *MessageText, |
|
547 SQLSMALLINT BufferLength, |
|
548 SQLSMALLINT *TextLength) |
|
549 { |
|
550 typedef SQLRETURN SQL_API ep_SQLGetDiagRec ( |
|
551 SQLSMALLINT HandleType, |
|
552 SQLHANDLE Handle, |
|
553 SQLSMALLINT RecNumber, |
|
554 SQLCHAR *Sqlstate, |
|
555 SQLINTEGER *NativeError, |
|
556 SQLCHAR *MessageText, |
|
557 SQLSMALLINT BufferLength, |
|
558 SQLSMALLINT *TextLength); |
|
559 |
|
560 ep_SQLGetDiagRec *fn; |
|
561 fn = (ep_SQLGetDiagRec *) xdlsym(h_odbc, "SQLGetDiagRec"); |
|
562 xassert(fn != NULL); |
|
563 return (*fn)(HandleType, Handle, RecNumber, Sqlstate, |
|
564 NativeError, MessageText, BufferLength, TextLength); |
|
565 } |
|
566 |
|
567 SQLRETURN SQL_API dl_SQLGetInfo ( |
|
568 SQLHDBC ConnectionHandle, |
|
569 SQLUSMALLINT InfoType, |
|
570 SQLPOINTER InfoValue, |
|
571 SQLSMALLINT BufferLength, |
|
572 SQLSMALLINT *StringLength) |
|
573 { |
|
574 typedef SQLRETURN SQL_API ep_SQLGetInfo ( |
|
575 SQLHDBC ConnectionHandle, |
|
576 SQLUSMALLINT InfoType, |
|
577 SQLPOINTER InfoValue, |
|
578 SQLSMALLINT BufferLength, |
|
579 SQLSMALLINT *StringLength); |
|
580 |
|
581 ep_SQLGetInfo *fn; |
|
582 fn = (ep_SQLGetInfo *) xdlsym(h_odbc, "SQLGetInfo"); |
|
583 xassert(fn != NULL); |
|
584 return (*fn)(ConnectionHandle, InfoType, InfoValue, BufferLength, |
|
585 StringLength); |
|
586 } |
|
587 |
|
588 SQLRETURN SQL_API dl_SQLNumResultCols ( |
|
589 SQLHSTMT StatementHandle, |
|
590 SQLSMALLINT *ColumnCount) |
|
591 { |
|
592 typedef SQLRETURN SQL_API ep_SQLNumResultCols ( |
|
593 SQLHSTMT StatementHandle, |
|
594 SQLSMALLINT *ColumnCount); |
|
595 |
|
596 ep_SQLNumResultCols *fn; |
|
597 fn = (ep_SQLNumResultCols *) xdlsym(h_odbc, "SQLNumResultCols"); |
|
598 xassert(fn != NULL); |
|
599 return (*fn)(StatementHandle, ColumnCount); |
|
600 } |
|
601 |
|
602 SQLRETURN SQL_API dl_SQLSetConnectAttr ( |
|
603 SQLHDBC ConnectionHandle, |
|
604 SQLINTEGER Attribute, |
|
605 SQLPOINTER Value, |
|
606 SQLINTEGER StringLength) |
|
607 { |
|
608 typedef SQLRETURN SQL_API ep_SQLSetConnectAttr ( |
|
609 SQLHDBC ConnectionHandle, |
|
610 SQLINTEGER Attribute, |
|
611 SQLPOINTER Value, |
|
612 SQLINTEGER StringLength); |
|
613 |
|
614 ep_SQLSetConnectAttr *fn; |
|
615 fn = (ep_SQLSetConnectAttr *) xdlsym(h_odbc, "SQLSetConnectAttr"); |
|
616 xassert(fn != NULL); |
|
617 return (*fn)(ConnectionHandle, Attribute, Value, StringLength); |
|
618 } |
|
619 |
|
620 SQLRETURN SQL_API dl_SQLSetEnvAttr ( |
|
621 SQLHENV EnvironmentHandle, |
|
622 SQLINTEGER Attribute, |
|
623 SQLPOINTER Value, |
|
624 SQLINTEGER StringLength) |
|
625 { |
|
626 typedef SQLRETURN SQL_API ep_SQLSetEnvAttr ( |
|
627 SQLHENV EnvironmentHandle, |
|
628 SQLINTEGER Attribute, |
|
629 SQLPOINTER Value, |
|
630 SQLINTEGER StringLength); |
|
631 |
|
632 ep_SQLSetEnvAttr *fn; |
|
633 fn = (ep_SQLSetEnvAttr *) xdlsym(h_odbc, "SQLSetEnvAttr"); |
|
634 xassert(fn != NULL); |
|
635 return (*fn)(EnvironmentHandle, Attribute, Value, StringLength); |
|
636 } |
|
637 |
|
638 static void extract_error( |
|
639 char *fn, |
|
640 SQLHANDLE handle, |
|
641 SQLSMALLINT type); |
|
642 |
|
643 static int is_numeric( |
|
644 SQLSMALLINT coltype); |
|
645 |
|
646 /*********************************************************************** |
|
647 * NAME |
|
648 * |
|
649 * db_iodbc_open - open connection to ODBC data base |
|
650 * |
|
651 * SYNOPSIS |
|
652 * |
|
653 * #include "glpsql.h" |
|
654 * void *db_iodbc_open(TABDCA *dca, int mode); |
|
655 * |
|
656 * DESCRIPTION |
|
657 * |
|
658 * The routine db_iodbc_open opens a connection to an ODBC data base. |
|
659 * It then executes the sql statements passed. |
|
660 * |
|
661 * In the case of table read the SELECT statement is executed. |
|
662 * |
|
663 * In the case of table write the INSERT statement is prepared. |
|
664 * RETURNS |
|
665 * |
|
666 * The routine returns a pointer to data storage area created. */ |
|
667 void *db_iodbc_open(TABDCA *dca, int mode) |
|
668 { void *ret; |
|
669 char **sqllines; |
|
670 |
|
671 sqllines = args_concat(dca); |
|
672 if (sqllines == NULL) |
|
673 { xprintf("Missing arguments in table statement.\n" |
|
674 "Please, supply table driver, dsn, and query.\n"); |
|
675 return NULL; |
|
676 } |
|
677 ret = db_iodbc_open_int(dca, mode, (const char **) sqllines); |
|
678 free_buffer(sqllines); |
|
679 return ret; |
|
680 } |
|
681 |
|
682 static void *db_iodbc_open_int(TABDCA *dca, int mode, const char |
|
683 **sqllines) |
|
684 { |
|
685 struct db_odbc *sql; |
|
686 SQLRETURN ret; |
|
687 SQLCHAR FAR *dsn; |
|
688 SQLCHAR info[256]; |
|
689 SQLSMALLINT colnamelen; |
|
690 SQLSMALLINT nullable; |
|
691 SQLSMALLINT scale; |
|
692 const char *arg; |
|
693 int narg; |
|
694 int i, j; |
|
695 int total; |
|
696 |
|
697 if (libodbc == NULL) |
|
698 { |
|
699 xprintf("No loader for shared ODBC library available\n"); |
|
700 return NULL; |
|
701 } |
|
702 |
|
703 if (h_odbc == NULL) |
|
704 { |
|
705 h_odbc = xdlopen(libodbc); |
|
706 if (h_odbc == NULL) |
|
707 { xprintf("unable to open library %s\n", libodbc); |
|
708 xprintf("%s\n", xerrmsg()); |
|
709 return NULL; |
|
710 } |
|
711 } |
|
712 |
|
713 sql = (struct db_odbc *) xmalloc(sizeof(struct db_odbc)); |
|
714 if (sql == NULL) |
|
715 return NULL; |
|
716 |
|
717 sql->mode = mode; |
|
718 sql->hdbc = NULL; |
|
719 sql->henv = NULL; |
|
720 sql->hstmt = NULL; |
|
721 sql->query = NULL; |
|
722 narg = mpl_tab_num_args(dca); |
|
723 |
|
724 dsn = (SQLCHAR FAR *) mpl_tab_get_arg(dca, 2); |
|
725 /* allocate an environment handle */ |
|
726 ret = dl_SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, |
|
727 &(sql->henv)); |
|
728 /* set attribute to enable application to run as ODBC 3.0 |
|
729 application */ |
|
730 ret = dl_SQLSetEnvAttr(sql->henv, SQL_ATTR_ODBC_VERSION, |
|
731 (void *) SQL_OV_ODBC3, 0); |
|
732 /* allocate a connection handle */ |
|
733 ret = dl_SQLAllocHandle(SQL_HANDLE_DBC, sql->henv, &(sql->hdbc)); |
|
734 /* connect */ |
|
735 ret = dl_SQLDriverConnect(sql->hdbc, NULL, dsn, SQL_NTS, NULL, 0, |
|
736 NULL, SQL_DRIVER_COMPLETE); |
|
737 if (SQL_SUCCEEDED(ret)) |
|
738 { /* output information about data base connection */ |
|
739 xprintf("Connected to "); |
|
740 dl_SQLGetInfo(sql->hdbc, SQL_DBMS_NAME, (SQLPOINTER)info, |
|
741 sizeof(info), NULL); |
|
742 xprintf("%s ", info); |
|
743 dl_SQLGetInfo(sql->hdbc, SQL_DBMS_VER, (SQLPOINTER)info, |
|
744 sizeof(info), NULL); |
|
745 xprintf("%s - ", info); |
|
746 dl_SQLGetInfo(sql->hdbc, SQL_DATABASE_NAME, (SQLPOINTER)info, |
|
747 sizeof(info), NULL); |
|
748 xprintf("%s\n", info); |
|
749 } |
|
750 else |
|
751 { /* describe error */ |
|
752 xprintf("Failed to connect\n"); |
|
753 extract_error("SQLDriverConnect", sql->hdbc, SQL_HANDLE_DBC); |
|
754 dl_SQLFreeHandle(SQL_HANDLE_DBC, sql->hdbc); |
|
755 dl_SQLFreeHandle(SQL_HANDLE_ENV, sql->henv); |
|
756 xfree(sql); |
|
757 return NULL; |
|
758 } |
|
759 /* set AUTOCOMMIT on*/ |
|
760 ret = dl_SQLSetConnectAttr(sql->hdbc, SQL_ATTR_AUTOCOMMIT, |
|
761 (SQLPOINTER)SQL_AUTOCOMMIT_ON, 0); |
|
762 /* allocate a statement handle */ |
|
763 ret = dl_SQLAllocHandle(SQL_HANDLE_STMT, sql->hdbc, &(sql->hstmt)); |
|
764 |
|
765 /* initialization queries */ |
|
766 for(j = 0; sqllines[j+1] != NULL; j++) |
|
767 { |
|
768 sql->query = (SQLCHAR *) sqllines[j]; |
|
769 xprintf("%s\n", sql->query); |
|
770 ret = dl_SQLExecDirect(sql->hstmt, sql->query, SQL_NTS); |
|
771 switch (ret) |
|
772 { |
|
773 case SQL_SUCCESS: |
|
774 case SQL_SUCCESS_WITH_INFO: |
|
775 case SQL_NO_DATA_FOUND: |
|
776 break; |
|
777 default: |
|
778 xprintf("db_iodbc_open: Query\n\"%s\"\nfailed.\n", |
|
779 sql->query); |
|
780 extract_error("SQLExecDirect", sql->hstmt, SQL_HANDLE_STMT); |
|
781 dl_SQLFreeHandle(SQL_HANDLE_STMT, sql->hstmt); |
|
782 dl_SQLDisconnect(sql->hdbc); |
|
783 dl_SQLFreeHandle(SQL_HANDLE_DBC, sql->hdbc); |
|
784 dl_SQLFreeHandle(SQL_HANDLE_ENV, sql->henv); |
|
785 xfree(sql); |
|
786 return NULL; |
|
787 } |
|
788 /* commit statement */ |
|
789 dl_SQLEndTran(SQL_HANDLE_ENV, sql->henv, SQL_COMMIT); |
|
790 } |
|
791 |
|
792 if ( sql->mode == 'R' ) |
|
793 { sql->nf = mpl_tab_num_flds(dca); |
|
794 for(j = 0; sqllines[j] != NULL; j++) |
|
795 arg = sqllines[j]; |
|
796 total = strlen(arg); |
|
797 if (total > 7 && 0 == strncmp(arg, "SELECT ", 7)) |
|
798 { |
|
799 total = strlen(arg); |
|
800 sql->query = xmalloc( (total+1) * sizeof(char)); |
|
801 strcpy (sql->query, arg); |
|
802 } |
|
803 else |
|
804 { |
|
805 sql->query = db_generate_select_stmt(dca); |
|
806 } |
|
807 xprintf("%s\n", sql->query); |
|
808 if (dl_SQLExecDirect(sql->hstmt, sql->query, SQL_NTS) != |
|
809 SQL_SUCCESS) |
|
810 { |
|
811 xprintf("db_iodbc_open: Query\n\"%s\"\nfailed.\n", sql->query); |
|
812 extract_error("SQLExecDirect", sql->hstmt, SQL_HANDLE_STMT); |
|
813 dl_SQLFreeHandle(SQL_HANDLE_STMT, sql->hstmt); |
|
814 dl_SQLDisconnect(sql->hdbc); |
|
815 dl_SQLFreeHandle(SQL_HANDLE_DBC, sql->hdbc); |
|
816 dl_SQLFreeHandle(SQL_HANDLE_ENV, sql->henv); |
|
817 xfree(sql->query); |
|
818 xfree(sql); |
|
819 return NULL; |
|
820 } |
|
821 xfree(sql->query); |
|
822 /* determine number of result columns */ |
|
823 ret = dl_SQLNumResultCols(sql->hstmt, &sql->nresultcols); |
|
824 total = sql->nresultcols; |
|
825 if (total > SQL_FIELD_MAX) |
|
826 { xprintf("db_iodbc_open: Too many fields (> %d) in query.\n" |
|
827 "\"%s\"\n", SQL_FIELD_MAX, sql->query); |
|
828 dl_SQLFreeHandle(SQL_HANDLE_STMT, sql->hstmt); |
|
829 dl_SQLDisconnect(sql->hdbc); |
|
830 dl_SQLFreeHandle(SQL_HANDLE_DBC, sql->hdbc); |
|
831 dl_SQLFreeHandle(SQL_HANDLE_ENV, sql->henv); |
|
832 xfree(sql->query); |
|
833 return NULL; |
|
834 } |
|
835 for (i = 1; i <= total; i++) |
|
836 { /* return a set of attributes for a column */ |
|
837 ret = dl_SQLDescribeCol(sql->hstmt, (SQLSMALLINT) i, |
|
838 sql->colname[i], SQL_FDLEN_MAX, |
|
839 &colnamelen, &(sql->coltype[i]), &(sql->collen[i]), &scale, |
|
840 &nullable); |
|
841 sql->isnumeric[i] = is_numeric(sql->coltype[i]); |
|
842 /* bind columns to program vars, converting all types to CHAR*/ |
|
843 dl_SQLBindCol(sql->hstmt, i, SQL_CHAR, sql->data[i], |
|
844 SQL_FDLEN_MAX, &(sql->outlen[i])); |
|
845 for (j = sql->nf; j >= 1; j--) |
|
846 { if (strcmp(mpl_tab_get_name(dca, j), sql->colname[i]) == 0) |
|
847 break; |
|
848 } |
|
849 sql->ref[i] = j; |
|
850 } |
|
851 } |
|
852 else if ( sql->mode == 'W' ) |
|
853 { for(j = 0; sqllines[j] != NULL; j++) |
|
854 arg = sqllines[j]; |
|
855 if ( NULL != strchr(arg, '?') ) |
|
856 { |
|
857 total = strlen(arg); |
|
858 sql->query = xmalloc( (total+1) * sizeof(char)); |
|
859 strcpy (sql->query, arg); |
|
860 } |
|
861 else |
|
862 { |
|
863 sql->query = db_generate_insert_stmt(dca); |
|
864 } |
|
865 xprintf("%s\n", sql->query); |
|
866 } |
|
867 return sql; |
|
868 } |
|
869 |
|
870 int db_iodbc_read(TABDCA *dca, void *link) |
|
871 { |
|
872 struct db_odbc *sql; |
|
873 SQLRETURN ret; |
|
874 char buf[SQL_FDLEN_MAX+1]; |
|
875 int i; |
|
876 int len; |
|
877 double num; |
|
878 |
|
879 sql = (struct db_odbc *) link; |
|
880 |
|
881 xassert(sql != NULL); |
|
882 xassert(sql->mode == 'R'); |
|
883 |
|
884 ret=dl_SQLFetch(sql->hstmt); |
|
885 if (ret== SQL_ERROR) |
|
886 return -1; |
|
887 if (ret== SQL_NO_DATA_FOUND) |
|
888 return -1; /*EOF*/ |
|
889 for (i=1; i <= sql->nresultcols; i++) |
|
890 { |
|
891 if (sql->ref[i] > 0) |
|
892 { |
|
893 len = sql->outlen[i]; |
|
894 if (len != SQL_NULL_DATA) |
|
895 { |
|
896 if (len > SQL_FDLEN_MAX) |
|
897 len = SQL_FDLEN_MAX; |
|
898 else if (len < 0) |
|
899 len = 0; |
|
900 strncpy(buf, (const char *) sql->data[i], len); |
|
901 buf[len] = 0x00; |
|
902 if (0 != (sql->isnumeric[i])) |
|
903 { strspx(buf); /* remove spaces*/ |
|
904 if (str2num(buf, &num) != 0) |
|
905 { xprintf("'%s' cannot be converted to a number.\n", |
|
906 buf); |
|
907 return 1; |
|
908 } |
|
909 mpl_tab_set_num(dca, sql->ref[i], num); |
|
910 } |
|
911 else |
|
912 { mpl_tab_set_str(dca, sql->ref[i], strtrim(buf)); |
|
913 } |
|
914 } |
|
915 } |
|
916 } |
|
917 return 0; |
|
918 } |
|
919 |
|
920 int db_iodbc_write(TABDCA *dca, void *link) |
|
921 { |
|
922 struct db_odbc *sql; |
|
923 char *part; |
|
924 char *query; |
|
925 char *template; |
|
926 char num[50]; |
|
927 int k; |
|
928 int len; |
|
929 int nf; |
|
930 |
|
931 sql = (struct db_odbc *) link; |
|
932 xassert(sql != NULL); |
|
933 xassert(sql->mode == 'W'); |
|
934 |
|
935 len = strlen(sql->query); |
|
936 template = (char *) xmalloc( (len + 1) * sizeof(char) ); |
|
937 strcpy(template, sql->query); |
|
938 |
|
939 nf = mpl_tab_num_flds(dca); |
|
940 for (k = 1; k <= nf; k++) |
|
941 { switch (mpl_tab_get_type(dca, k)) |
|
942 { case 'N': |
|
943 len += 20; |
|
944 break; |
|
945 case 'S': |
|
946 len += db_escaped_string_length(mpl_tab_get_str(dca, k)); |
|
947 len += 2; |
|
948 break; |
|
949 default: |
|
950 xassert(dca != dca); |
|
951 } |
|
952 } |
|
953 query = xmalloc( (len + 1 ) * sizeof(char) ); |
|
954 query[0] = 0x00; |
|
955 for (k = 1, part = strtok (template, "?"); (part != NULL); |
|
956 part = strtok (NULL, "?"), k++) |
|
957 { |
|
958 if (k > nf) break; |
|
959 strcat( query, part ); |
|
960 switch (mpl_tab_get_type(dca, k)) |
|
961 { case 'N': |
|
962 #if 0 /* 02/XI-2010 by xypron */ |
|
963 sprintf(num, "%-18g",mpl_tab_get_num(dca, k)); |
|
964 #else |
|
965 sprintf(num, "%.*g", DBL_DIG, mpl_tab_get_num(dca, k)); |
|
966 #endif |
|
967 strcat( query, num ); |
|
968 break; |
|
969 case 'S': |
|
970 strcat( query, "'"); |
|
971 db_escape_string( query + strlen(query), |
|
972 mpl_tab_get_str(dca, k) ); |
|
973 strcat( query, "'"); |
|
974 break; |
|
975 default: |
|
976 xassert(dca != dca); |
|
977 } |
|
978 } |
|
979 if (part != NULL) |
|
980 strcat(query, part); |
|
981 if (dl_SQLExecDirect(sql->hstmt, (SQLCHAR *) query, SQL_NTS) |
|
982 != SQL_SUCCESS) |
|
983 { |
|
984 xprintf("db_iodbc_write: Query\n\"%s\"\nfailed.\n", query); |
|
985 extract_error("SQLExecDirect", sql->hdbc, SQL_HANDLE_DBC); |
|
986 xfree(query); |
|
987 xfree(template); |
|
988 return 1; |
|
989 } |
|
990 |
|
991 xfree(query); |
|
992 xfree(template); |
|
993 return 0; |
|
994 } |
|
995 |
|
996 int db_iodbc_close(TABDCA *dca, void *link) |
|
997 { |
|
998 struct db_odbc *sql; |
|
999 |
|
1000 sql = (struct db_odbc *) link; |
|
1001 xassert(sql != NULL); |
|
1002 /* Commit */ |
|
1003 if ( sql->mode == 'W' ) |
|
1004 dl_SQLEndTran(SQL_HANDLE_ENV, sql->henv, SQL_COMMIT); |
|
1005 if ( sql->mode == 'R' ) |
|
1006 dl_SQLCloseCursor(sql->hstmt); |
|
1007 |
|
1008 dl_SQLFreeHandle(SQL_HANDLE_STMT, sql->hstmt); |
|
1009 dl_SQLDisconnect(sql->hdbc); |
|
1010 dl_SQLFreeHandle(SQL_HANDLE_DBC, sql->hdbc); |
|
1011 dl_SQLFreeHandle(SQL_HANDLE_ENV, sql->henv); |
|
1012 if ( sql->mode == 'W' ) |
|
1013 xfree(sql->query); |
|
1014 xfree(sql); |
|
1015 dca->link = NULL; |
|
1016 return 0; |
|
1017 } |
|
1018 |
|
1019 static void extract_error( |
|
1020 char *fn, |
|
1021 SQLHANDLE handle, |
|
1022 SQLSMALLINT type) |
|
1023 { |
|
1024 SQLINTEGER i = 0; |
|
1025 SQLINTEGER native; |
|
1026 SQLCHAR state[ 7 ]; |
|
1027 SQLCHAR text[256]; |
|
1028 SQLSMALLINT len; |
|
1029 SQLRETURN ret; |
|
1030 |
|
1031 xprintf("\nThe driver reported the following diagnostics whilst " |
|
1032 "running %s\n", fn); |
|
1033 |
|
1034 do |
|
1035 { |
|
1036 ret = dl_SQLGetDiagRec(type, handle, ++i, state, &native, text, |
|
1037 sizeof(text), &len ); |
|
1038 if (SQL_SUCCEEDED(ret)) |
|
1039 xprintf("%s:%ld:%ld:%s\n", state, i, native, text); |
|
1040 } |
|
1041 while( ret == SQL_SUCCESS ); |
|
1042 } |
|
1043 |
|
1044 static int is_numeric(SQLSMALLINT coltype) |
|
1045 { |
|
1046 int ret = 0; |
|
1047 switch (coltype) |
|
1048 { |
|
1049 case SQL_DECIMAL: |
|
1050 case SQL_NUMERIC: |
|
1051 case SQL_SMALLINT: |
|
1052 case SQL_INTEGER: |
|
1053 case SQL_REAL: |
|
1054 case SQL_FLOAT: |
|
1055 case SQL_DOUBLE: |
|
1056 case SQL_TINYINT: |
|
1057 case SQL_BIGINT: |
|
1058 ret = 1; |
|
1059 break; |
|
1060 } |
|
1061 return ret; |
|
1062 } |
|
1063 |
|
1064 #endif |
|
1065 |
|
1066 /**********************************************************************/ |
|
1067 |
|
1068 #ifndef HAVE_MYSQL |
|
1069 |
|
1070 void *db_mysql_open(TABDCA *dca, int mode) |
|
1071 { xassert(dca == dca); |
|
1072 xassert(mode == mode); |
|
1073 xprintf("MySQL table driver not supported\n"); |
|
1074 return NULL; |
|
1075 } |
|
1076 |
|
1077 int db_mysql_read(TABDCA *dca, void *link) |
|
1078 { xassert(dca != dca); |
|
1079 xassert(link != link); |
|
1080 return 0; |
|
1081 } |
|
1082 |
|
1083 int db_mysql_write(TABDCA *dca, void *link) |
|
1084 { xassert(dca != dca); |
|
1085 xassert(link != link); |
|
1086 return 0; |
|
1087 } |
|
1088 |
|
1089 int db_mysql_close(TABDCA *dca, void *link) |
|
1090 { xassert(dca != dca); |
|
1091 xassert(link != link); |
|
1092 return 0; |
|
1093 } |
|
1094 |
|
1095 #else |
|
1096 |
|
1097 #if defined(__CYGWIN__) || defined(__MINGW32__) || defined(__WOE__) |
|
1098 #include <windows.h> |
|
1099 #endif |
|
1100 |
|
1101 #ifdef __CYGWIN__ |
|
1102 #define byte_defined 1 |
|
1103 #endif |
|
1104 |
|
1105 #include <my_global.h> |
|
1106 #include <my_sys.h> |
|
1107 #include <mysql.h> |
|
1108 |
|
1109 struct db_mysql |
|
1110 { |
|
1111 int mode; /*'R' = Read, 'W' = Write*/ |
|
1112 MYSQL *con; /*connection*/ |
|
1113 MYSQL_RES *res; /*result*/ |
|
1114 int nf; |
|
1115 /* number of fields in the csv file */ |
|
1116 int ref[1+SQL_FIELD_MAX]; |
|
1117 /* ref[k] = k', if k-th field of the csv file corresponds to |
|
1118 k'-th field in the table statement; if ref[k] = 0, k-th field |
|
1119 of the csv file is ignored */ |
|
1120 char *query; |
|
1121 /* query generated by db_mysql_open */ |
|
1122 }; |
|
1123 |
|
1124 void STDCALL dl_mysql_close(MYSQL *sock) |
|
1125 { |
|
1126 typedef void STDCALL ep_mysql_close(MYSQL *sock); |
|
1127 |
|
1128 ep_mysql_close *fn; |
|
1129 fn = (ep_mysql_close *) xdlsym(h_mysql, "mysql_close"); |
|
1130 xassert(fn != NULL); |
|
1131 return (*fn)(sock); |
|
1132 } |
|
1133 |
|
1134 const char * STDCALL dl_mysql_error(MYSQL *mysql) |
|
1135 { |
|
1136 typedef const char * STDCALL ep_mysql_error(MYSQL *mysql); |
|
1137 |
|
1138 ep_mysql_error *fn; |
|
1139 fn = (ep_mysql_error *) xdlsym(h_mysql, "mysql_error"); |
|
1140 xassert(fn != NULL); |
|
1141 return (*fn)(mysql); |
|
1142 } |
|
1143 |
|
1144 MYSQL_FIELD * STDCALL dl_mysql_fetch_fields(MYSQL_RES *res) |
|
1145 { |
|
1146 typedef MYSQL_FIELD * STDCALL |
|
1147 ep_mysql_fetch_fields(MYSQL_RES *res); |
|
1148 |
|
1149 ep_mysql_fetch_fields *fn; |
|
1150 fn = (ep_mysql_fetch_fields *) xdlsym(h_mysql, "mysql_fetch_fields"); |
|
1151 xassert(fn != NULL); |
|
1152 return (*fn)(res); |
|
1153 } |
|
1154 |
|
1155 unsigned long * STDCALL dl_mysql_fetch_lengths(MYSQL_RES *result) |
|
1156 { |
|
1157 typedef unsigned long * STDCALL |
|
1158 ep_mysql_fetch_lengths(MYSQL_RES *result); |
|
1159 |
|
1160 ep_mysql_fetch_lengths *fn; |
|
1161 fn = (ep_mysql_fetch_lengths *) xdlsym(h_mysql, |
|
1162 "mysql_fetch_lengths"); |
|
1163 xassert(fn != NULL); |
|
1164 return (*fn)(result); |
|
1165 } |
|
1166 |
|
1167 MYSQL_ROW STDCALL dl_mysql_fetch_row(MYSQL_RES *result) |
|
1168 { |
|
1169 typedef MYSQL_ROW STDCALL ep_mysql_fetch_row(MYSQL_RES *result); |
|
1170 |
|
1171 ep_mysql_fetch_row *fn; |
|
1172 fn = (ep_mysql_fetch_row *) xdlsym(h_mysql, "mysql_fetch_row"); |
|
1173 xassert(fn != NULL); |
|
1174 return (*fn)(result); |
|
1175 } |
|
1176 |
|
1177 unsigned int STDCALL dl_mysql_field_count(MYSQL *mysql) |
|
1178 { |
|
1179 typedef unsigned int STDCALL ep_mysql_field_count(MYSQL *mysql); |
|
1180 |
|
1181 ep_mysql_field_count *fn; |
|
1182 fn = (ep_mysql_field_count *) xdlsym(h_mysql, "mysql_field_count"); |
|
1183 xassert(fn != NULL); |
|
1184 return (*fn)(mysql); |
|
1185 } |
|
1186 |
|
1187 MYSQL * STDCALL dl_mysql_init(MYSQL *mysql) |
|
1188 { |
|
1189 typedef MYSQL * STDCALL ep_mysql_init(MYSQL *mysql); |
|
1190 |
|
1191 ep_mysql_init *fn; |
|
1192 fn = (ep_mysql_init *) xdlsym(h_mysql, "mysql_init"); |
|
1193 xassert(fn != NULL); |
|
1194 return (*fn)(mysql); |
|
1195 } |
|
1196 |
|
1197 unsigned int STDCALL dl_mysql_num_fields(MYSQL_RES *res) |
|
1198 { |
|
1199 typedef unsigned int STDCALL ep_mysql_num_fields(MYSQL_RES *res); |
|
1200 |
|
1201 ep_mysql_num_fields *fn; |
|
1202 fn = (ep_mysql_num_fields *) xdlsym(h_mysql, "mysql_num_fields"); |
|
1203 xassert(fn != NULL); |
|
1204 return (*fn)(res); |
|
1205 } |
|
1206 |
|
1207 int STDCALL dl_mysql_query(MYSQL *mysql, const char *q) |
|
1208 { |
|
1209 typedef int STDCALL ep_mysql_query(MYSQL *mysql, const char *q); |
|
1210 |
|
1211 ep_mysql_query *fn; |
|
1212 fn = (ep_mysql_query *) xdlsym(h_mysql, "mysql_query"); |
|
1213 xassert(fn != NULL); |
|
1214 return (*fn)(mysql, q); |
|
1215 } |
|
1216 |
|
1217 MYSQL * STDCALL dl_mysql_real_connect(MYSQL *mysql, const char *host, |
|
1218 const char *user, |
|
1219 const char *passwd, |
|
1220 const char *db, |
|
1221 unsigned int port, |
|
1222 const char *unix_socket, |
|
1223 unsigned long clientflag) |
|
1224 { |
|
1225 typedef MYSQL * STDCALL ep_mysql_real_connect(MYSQL *mysql, |
|
1226 const char *host, |
|
1227 const char *user, |
|
1228 const char *passwd, |
|
1229 const char *db, |
|
1230 unsigned int port, |
|
1231 const char *unix_socket, |
|
1232 unsigned long clientflag); |
|
1233 |
|
1234 ep_mysql_real_connect *fn; |
|
1235 fn = (ep_mysql_real_connect *) xdlsym(h_mysql, |
|
1236 "mysql_real_connect"); |
|
1237 xassert(fn != NULL); |
|
1238 return (*fn)(mysql, host, user, passwd, db, port, unix_socket, |
|
1239 clientflag); |
|
1240 } |
|
1241 |
|
1242 MYSQL_RES * STDCALL dl_mysql_use_result(MYSQL *mysql) |
|
1243 { |
|
1244 typedef MYSQL_RES * STDCALL ep_mysql_use_result(MYSQL *mysql); |
|
1245 ep_mysql_use_result *fn; |
|
1246 fn = (ep_mysql_use_result *) xdlsym(h_mysql, "mysql_use_result"); |
|
1247 xassert(fn != NULL); |
|
1248 return (*fn)(mysql); |
|
1249 } |
|
1250 |
|
1251 /*********************************************************************** |
|
1252 * NAME |
|
1253 * |
|
1254 * db_mysql_open - open connection to ODBC data base |
|
1255 * |
|
1256 * SYNOPSIS |
|
1257 * |
|
1258 * #include "glpsql.h" |
|
1259 * void *db_mysql_open(TABDCA *dca, int mode); |
|
1260 * |
|
1261 * DESCRIPTION |
|
1262 * |
|
1263 * The routine db_mysql_open opens a connection to a MySQL data base. |
|
1264 * It then executes the sql statements passed. |
|
1265 * |
|
1266 * In the case of table read the SELECT statement is executed. |
|
1267 * |
|
1268 * In the case of table write the INSERT statement is prepared. |
|
1269 * RETURNS |
|
1270 * |
|
1271 * The routine returns a pointer to data storage area created. */ |
|
1272 |
|
1273 void *db_mysql_open(TABDCA *dca, int mode) |
|
1274 { void *ret; |
|
1275 char **sqllines; |
|
1276 |
|
1277 sqllines = args_concat(dca); |
|
1278 if (sqllines == NULL) |
|
1279 { xprintf("Missing arguments in table statement.\n" |
|
1280 "Please, supply table driver, dsn, and query.\n"); |
|
1281 return NULL; |
|
1282 } |
|
1283 ret = db_mysql_open_int(dca, mode, (const char **) sqllines); |
|
1284 free_buffer(sqllines); |
|
1285 return ret; |
|
1286 } |
|
1287 |
|
1288 static void *db_mysql_open_int(TABDCA *dca, int mode, const char |
|
1289 **sqllines) |
|
1290 { |
|
1291 struct db_mysql *sql = NULL; |
|
1292 char *arg = NULL; |
|
1293 const char *field; |
|
1294 MYSQL_FIELD *fields; |
|
1295 char *keyword; |
|
1296 char *value; |
|
1297 char *query; |
|
1298 char *dsn; |
|
1299 /* "Server=[server_name];Database=[database_name];UID=[username];*/ |
|
1300 /* PWD=[password];Port=[port]"*/ |
|
1301 char *server = NULL; /* Server */ |
|
1302 char *user = NULL; /* UID */ |
|
1303 char *password = NULL; /* PWD */ |
|
1304 char *database = NULL; /* Database */ |
|
1305 unsigned int port = 0; /* Port */ |
|
1306 int narg; |
|
1307 int i, j, total; |
|
1308 |
|
1309 if (libmysql == NULL) |
|
1310 { |
|
1311 xprintf("No loader for shared MySQL library available\n"); |
|
1312 return NULL; |
|
1313 } |
|
1314 |
|
1315 if (h_mysql == NULL) |
|
1316 { |
|
1317 h_mysql = xdlopen(libmysql); |
|
1318 if (h_mysql == NULL) |
|
1319 { xprintf("unable to open library %s\n", libmysql); |
|
1320 xprintf("%s\n", xerrmsg()); |
|
1321 return NULL; |
|
1322 } |
|
1323 } |
|
1324 |
|
1325 sql = (struct db_mysql *) xmalloc(sizeof(struct db_mysql)); |
|
1326 if (sql == NULL) |
|
1327 return NULL; |
|
1328 sql->mode = mode; |
|
1329 sql->res = NULL; |
|
1330 sql->query = NULL; |
|
1331 sql->nf = mpl_tab_num_flds(dca); |
|
1332 |
|
1333 narg = mpl_tab_num_args(dca); |
|
1334 if (narg < 3 ) |
|
1335 xprintf("MySQL driver: string list too short \n"); |
|
1336 |
|
1337 /* get connection string*/ |
|
1338 dsn = (char *) mpl_tab_get_arg(dca, 2); |
|
1339 /* copy connection string*/ |
|
1340 i = strlen(dsn); |
|
1341 i++; |
|
1342 arg = xmalloc(i * sizeof(char)); |
|
1343 strcpy(arg, dsn); |
|
1344 /*tokenize connection string*/ |
|
1345 for (i = 1, keyword = strtok (arg, "="); (keyword != NULL); |
|
1346 keyword = strtok (NULL, "="), i++) |
|
1347 { |
|
1348 value = strtok (NULL, ";"); |
|
1349 if (value==NULL) |
|
1350 { |
|
1351 xprintf("db_mysql_open: Missing value for keyword %s\n", |
|
1352 keyword); |
|
1353 xfree(arg); |
|
1354 xfree(sql); |
|
1355 return NULL; |
|
1356 } |
|
1357 if (0 == strcmp(keyword, "Server")) |
|
1358 server = value; |
|
1359 else if (0 == strcmp(keyword, "Database")) |
|
1360 database = value; |
|
1361 else if (0 == strcmp(keyword, "UID")) |
|
1362 user = value; |
|
1363 else if (0 == strcmp(keyword, "PWD")) |
|
1364 password = value; |
|
1365 else if (0 == strcmp(keyword, "Port")) |
|
1366 port = (unsigned int) atol(value); |
|
1367 } |
|
1368 /* Connect to database */ |
|
1369 sql->con = dl_mysql_init(NULL); |
|
1370 if (!dl_mysql_real_connect(sql->con, server, user, password, database, |
|
1371 port, NULL, 0)) |
|
1372 { |
|
1373 xprintf("db_mysql_open: Connect failed\n"); |
|
1374 xprintf("%s\n", dl_mysql_error(sql->con)); |
|
1375 xfree(arg); |
|
1376 xfree(sql); |
|
1377 return NULL; |
|
1378 } |
|
1379 xfree(arg); |
|
1380 |
|
1381 for(j = 0; sqllines[j+1] != NULL; j++) |
|
1382 { query = (char *) sqllines[j]; |
|
1383 xprintf("%s\n", query); |
|
1384 if (dl_mysql_query(sql->con, query)) |
|
1385 { |
|
1386 xprintf("db_mysql_open: Query\n\"%s\"\nfailed.\n", query); |
|
1387 xprintf("%s\n",dl_mysql_error(sql->con)); |
|
1388 dl_mysql_close(sql->con); |
|
1389 xfree(sql); |
|
1390 return NULL; |
|
1391 } |
|
1392 } |
|
1393 |
|
1394 if ( sql->mode == 'R' ) |
|
1395 { sql->nf = mpl_tab_num_flds(dca); |
|
1396 for(j = 0; sqllines[j] != NULL; j++) |
|
1397 arg = (char *) sqllines[j]; |
|
1398 total = strlen(arg); |
|
1399 if (total > 7 && 0 == strncmp(arg, "SELECT ", 7)) |
|
1400 { |
|
1401 total = strlen(arg); |
|
1402 query = xmalloc( (total+1) * sizeof(char)); |
|
1403 strcpy (query, arg); |
|
1404 } |
|
1405 else |
|
1406 { |
|
1407 query = db_generate_select_stmt(dca); |
|
1408 } |
|
1409 xprintf("%s\n", query); |
|
1410 if (dl_mysql_query(sql->con, query)) |
|
1411 { |
|
1412 xprintf("db_mysql_open: Query\n\"%s\"\nfailed.\n", query); |
|
1413 xprintf("%s\n",dl_mysql_error(sql->con)); |
|
1414 dl_mysql_close(sql->con); |
|
1415 xfree(query); |
|
1416 xfree(sql); |
|
1417 return NULL; |
|
1418 } |
|
1419 xfree(query); |
|
1420 sql->res = dl_mysql_use_result(sql->con); |
|
1421 if (sql->res) |
|
1422 { |
|
1423 /* create references between query results and table fields*/ |
|
1424 total = dl_mysql_num_fields(sql->res); |
|
1425 if (total > SQL_FIELD_MAX) |
|
1426 { xprintf("db_mysql_open: Too many fields (> %d) in query.\n" |
|
1427 "\"%s\"\n", SQL_FIELD_MAX, query); |
|
1428 xprintf("%s\n",dl_mysql_error(sql->con)); |
|
1429 dl_mysql_close(sql->con); |
|
1430 xfree(query); |
|
1431 xfree(sql); |
|
1432 return NULL; |
|
1433 } |
|
1434 fields = dl_mysql_fetch_fields(sql->res); |
|
1435 for (i = 1; i <= total; i++) |
|
1436 { |
|
1437 for (j = sql->nf; j >= 1; j--) |
|
1438 { |
|
1439 if (strcmp(mpl_tab_get_name(dca, j), fields[i-1].name) |
|
1440 == 0) |
|
1441 break; |
|
1442 } |
|
1443 sql->ref[i] = j; |
|
1444 } |
|
1445 } |
|
1446 else |
|
1447 { |
|
1448 if(dl_mysql_field_count(sql->con) == 0) |
|
1449 { |
|
1450 xprintf("db_mysql_open: Query was not a SELECT\n\"%s\"\n", |
|
1451 query); |
|
1452 xprintf("%s\n",dl_mysql_error(sql->con)); |
|
1453 xfree(query); |
|
1454 xfree(sql); |
|
1455 return NULL; |
|
1456 } |
|
1457 else |
|
1458 { |
|
1459 xprintf("db_mysql_open: Query\n\"%s\"\nfailed.\n", query); |
|
1460 xprintf("%s\n",dl_mysql_error(sql->con)); |
|
1461 xfree(query); |
|
1462 xfree(sql); |
|
1463 return NULL; |
|
1464 } |
|
1465 } |
|
1466 } |
|
1467 else if ( sql->mode == 'W' ) |
|
1468 { for(j = 0; sqllines[j] != NULL; j++) |
|
1469 arg = (char *) sqllines[j]; |
|
1470 if ( NULL != strchr(arg, '?') ) |
|
1471 { |
|
1472 total = strlen(arg); |
|
1473 query = xmalloc( (total+1) * sizeof(char)); |
|
1474 strcpy (query, arg); |
|
1475 } |
|
1476 else |
|
1477 query = db_generate_insert_stmt(dca); |
|
1478 sql->query = query; |
|
1479 xprintf("%s\n", query); |
|
1480 } |
|
1481 return sql; |
|
1482 } |
|
1483 |
|
1484 int db_mysql_read(TABDCA *dca, void *link) |
|
1485 { struct db_mysql *sql; |
|
1486 char buf[255+1]; |
|
1487 char **row; |
|
1488 unsigned long *lengths; |
|
1489 MYSQL_FIELD *fields; |
|
1490 double num; |
|
1491 int len; |
|
1492 unsigned long num_fields; |
|
1493 int i; |
|
1494 |
|
1495 sql = (struct db_mysql *) link; |
|
1496 |
|
1497 xassert(sql != NULL); |
|
1498 xassert(sql->mode == 'R'); |
|
1499 if (NULL == sql->res) |
|
1500 { |
|
1501 xprintf("db_mysql_read: no result set available"); |
|
1502 return 1; |
|
1503 } |
|
1504 if (NULL==(row = (char **)dl_mysql_fetch_row(sql->res))) { |
|
1505 return -1; /*EOF*/ |
|
1506 } |
|
1507 lengths = dl_mysql_fetch_lengths(sql->res); |
|
1508 fields = dl_mysql_fetch_fields(sql->res); |
|
1509 num_fields = dl_mysql_num_fields(sql->res); |
|
1510 for (i=1; i <= num_fields; i++) |
|
1511 { |
|
1512 if (row[i-1] != NULL) |
|
1513 { len = (size_t) lengths[i-1]; |
|
1514 if (len > 255) |
|
1515 len = 255; |
|
1516 strncpy(buf, (const char *) row[i-1], len); |
|
1517 buf[len] = 0x00; |
|
1518 if (0 != (fields[i-1].flags & NUM_FLAG)) |
|
1519 { strspx(buf); /* remove spaces*/ |
|
1520 if (str2num(buf, &num) != 0) |
|
1521 { xprintf("'%s' cannot be converted to a number.\n", buf); |
|
1522 return 1; |
|
1523 } |
|
1524 if (sql->ref[i] > 0) |
|
1525 mpl_tab_set_num(dca, sql->ref[i], num); |
|
1526 } |
|
1527 else |
|
1528 { if (sql->ref[i] > 0) |
|
1529 mpl_tab_set_str(dca, sql->ref[i], strtrim(buf)); |
|
1530 } |
|
1531 } |
|
1532 } |
|
1533 return 0; |
|
1534 } |
|
1535 |
|
1536 int db_mysql_write(TABDCA *dca, void *link) |
|
1537 { |
|
1538 struct db_mysql *sql; |
|
1539 char *part; |
|
1540 char *query; |
|
1541 char *template; |
|
1542 char num[50]; |
|
1543 int k; |
|
1544 int len; |
|
1545 int nf; |
|
1546 |
|
1547 sql = (struct db_mysql *) link; |
|
1548 xassert(sql != NULL); |
|
1549 xassert(sql->mode == 'W'); |
|
1550 |
|
1551 len = strlen(sql->query); |
|
1552 template = (char *) xmalloc( (len + 1) * sizeof(char) ); |
|
1553 strcpy(template, sql->query); |
|
1554 |
|
1555 nf = mpl_tab_num_flds(dca); |
|
1556 for (k = 1; k <= nf; k++) |
|
1557 { switch (mpl_tab_get_type(dca, k)) |
|
1558 { case 'N': |
|
1559 len += 20; |
|
1560 break; |
|
1561 case 'S': |
|
1562 len += db_escaped_string_length(mpl_tab_get_str(dca, k)); |
|
1563 len += 2; |
|
1564 break; |
|
1565 default: |
|
1566 xassert(dca != dca); |
|
1567 } |
|
1568 } |
|
1569 query = xmalloc( (len + 1 ) * sizeof(char) ); |
|
1570 query[0] = 0x00; |
|
1571 for (k = 1, part = strtok (template, "?"); (part != NULL); |
|
1572 part = strtok (NULL, "?"), k++) |
|
1573 { |
|
1574 if (k > nf) break; |
|
1575 strcat( query, part ); |
|
1576 switch (mpl_tab_get_type(dca, k)) |
|
1577 { case 'N': |
|
1578 #if 0 /* 02/XI-2010 by xypron */ |
|
1579 sprintf(num, "%-18g",mpl_tab_get_num(dca, k)); |
|
1580 #else |
|
1581 sprintf(num, "%.*g", DBL_DIG, mpl_tab_get_num(dca, k)); |
|
1582 #endif |
|
1583 strcat( query, num ); |
|
1584 break; |
|
1585 case 'S': |
|
1586 strcat( query, "'"); |
|
1587 db_escape_string( query + strlen(query), |
|
1588 mpl_tab_get_str(dca, k) ); |
|
1589 strcat( query, "'"); |
|
1590 break; |
|
1591 default: |
|
1592 xassert(dca != dca); |
|
1593 } |
|
1594 } |
|
1595 if (part != NULL) |
|
1596 strcat(query, part); |
|
1597 if (dl_mysql_query(sql->con, query)) |
|
1598 { |
|
1599 xprintf("db_mysql_write: Query\n\"%s\"\nfailed.\n", query); |
|
1600 xprintf("%s\n",dl_mysql_error(sql->con)); |
|
1601 xfree(query); |
|
1602 xfree(template); |
|
1603 return 1; |
|
1604 } |
|
1605 |
|
1606 xfree(query); |
|
1607 xfree(template); |
|
1608 return 0; |
|
1609 } |
|
1610 |
|
1611 int db_mysql_close(TABDCA *dca, void *link) |
|
1612 { |
|
1613 struct db_mysql *sql; |
|
1614 |
|
1615 sql = (struct db_mysql *) link; |
|
1616 xassert(sql != NULL); |
|
1617 dl_mysql_close(sql->con); |
|
1618 if ( sql->mode == 'W' ) |
|
1619 xfree(sql->query); |
|
1620 xfree(sql); |
|
1621 dca->link = NULL; |
|
1622 return 0; |
|
1623 } |
|
1624 |
|
1625 #endif |
|
1626 |
|
1627 /* eof */ |