|
|
| version 1.36, 2011/03/11 07:57:42 | version 1.44, 2015/10/26 16:00:51 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser PgSQL driver. | Parser PgSQL 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) |
| 2007.10.25 using PgSQL 8.1.5 | 2007.10.25 using PgSQL 8.1.5 |
| */ | */ |
| static const char *RCSId="$Id$"; | |
| #include "config_includes.h" | #include "config_includes.h" |
| Line 16 static const char *RCSId="$Id$"; | Line 15 static const char *RCSId="$Id$"; |
| #include <libpq-fe.h> | #include <libpq-fe.h> |
| #include <libpq/libpq-fs.h> | #include <libpq/libpq-fs.h> |
| volatile const char * IDENT_PARSER3PGSQL_C="$Id$" IDENT_PA_SQL_DRIVER_H; | |
| // from catalog/pg_type.h | // from catalog/pg_type.h |
| #define BOOLOID 16 | #define BOOLOID 16 |
| #define INT8OID 20 | #define INT8OID 20 |
| Line 99 struct Connection { | Line 100 struct Connection { |
| const char* client_charset; | const char* client_charset; |
| bool autocommit; | bool autocommit; |
| bool without_default_transactions; | bool without_default_transactions; |
| bool standard_conforming_strings; | |
| }; | }; |
| /** | /** |
| Line 133 public: | Line 135 public: |
| charset=value& // transcode by server with 'SET CLIENT_ENCODING=value' | charset=value& // transcode by server with 'SET CLIENT_ENCODING=value' |
| datestyle=value& // 'SET DATESTYLE=value' available values are: ISO|SQL|Postgres|European|US|German [default=ISO] | datestyle=value& // 'SET DATESTYLE=value' available values are: ISO|SQL|Postgres|European|US|German [default=ISO] |
| autocommit=1& // each transaction is commited automatically (default) | autocommit=1& // each transaction is commited automatically (default) |
| WithoutDefaultTransaction=0 // 1 -- disable any BEGIN TRAN/COMMIT/ROLLBACK [can NOT be used with autocommit option] | standard_conforming_strings=1& // 0 -- escape \ char that could be needed for old servers |
| WithoutDefaultTransaction=0 // 1 -- disable any BEGIN TRAN/COMMIT/ROLLBACK [can NOT be used with autocommit option] | |
| */ | */ |
| void connect( | void connect( |
| char* url, | char* url, |
| Line 157 public: | Line 160 public: |
| connection.client_charset=0; | connection.client_charset=0; |
| connection.autocommit=true; | connection.autocommit=true; |
| connection.without_default_transactions=false; | connection.without_default_transactions=false; |
| connection.standard_conforming_strings=true; | |
| connection.conn=PQsetdbLogin( | connection.conn=PQsetdbLogin( |
| (host&&strcasecmp(host, "local")==0)?NULL/* local Unix domain socket */:host, port, | (host&&strcasecmp(host, "local")==0)?NULL/* local Unix domain socket */:host, port, |
| Line 191 public: | Line 195 public: |
| connection.without_default_transactions=true; | connection.without_default_transactions=true; |
| connection.autocommit=false; | connection.autocommit=false; |
| } | } |
| } else if(strcasecmp(key, "standard_conforming_strings")==0){ | |
| if(atoi(value)==0) | |
| connection.standard_conforming_strings=false; | |
| } else | } else |
| services._throw("unknown connect option" /*key*/); | services._throw("unknown connect option" /*key*/); |
| } else | } else |
| Line 200 public: | Line 207 public: |
| } | } |
| if(charset){ | if(charset){ |
| char statement[MAX_STRING]="SET CLIENT_ENCODING="; | char statement[MAX_STRING+1]="SET CLIENT_ENCODING="; |
| strncat(statement, charset, MAX_STRING); | strncat(statement, charset, MAX_STRING); |
| _execute_cmd(connection, statement); | _execute_cmd(connection, statement); |
| } | } |
| if(datestyle){ | if(datestyle){ |
| char statement[MAX_STRING]="SET DATESTYLE="; | char statement[MAX_STRING+1]="SET DATESTYLE="; |
| strncat(statement, datestyle, MAX_STRING); | strncat(statement, datestyle, MAX_STRING); |
| _execute_cmd(connection, statement); | _execute_cmd(connection, statement); |
| Line 243 public: | Line 250 public: |
| // thus we can't use the sql server quoting support | // thus we can't use the sql server quoting support |
| const char* quote(void *aconnection, const char *str, unsigned int length) | const char* quote(void *aconnection, const char *str, unsigned int length) |
| { | { |
| Connection& connection=*static_cast<Connection*>(aconnection); | |
| const char* from; | const char* from; |
| const char* from_end=str+length; | const char* from_end=str+length; |
| size_t quoted=0; | size_t quoted=0; |
| for(from=str; from<from_end; from++){ | if(connection.standard_conforming_strings){ |
| switch (*from) { | for(from=str; from<from_end; from++){ |
| case '\'': | if(*from=='\'') |
| case '\\': | quoted++; |
| quoted++; | } |
| } else { | |
| for(from=str; from<from_end; from++){ | |
| switch (*from) { | |
| case '\'': | |
| case '\\': | |
| quoted++; | |
| } | |
| } | } |
| } | } |
| if(!quoted) | if(!quoted) |
| return str; | return str; |
| Connection& connection=*static_cast<Connection*>(aconnection); | |
| char *result=(char*)connection.services->malloc_atomic(length + quoted + 1); | char *result=(char*)connection.services->malloc_atomic(length + quoted + 1); |
| char *to = result; | char *to = result; |
| for(from=str; from<from_end; from++){ | if(connection.standard_conforming_strings){ |
| switch (*from) { | for(from=str; from<from_end; from++){ |
| case '\'': // "'" -> "''" | if(*from=='\'') |
| *to++='\''; | *to++= '\''; // "'" -> "''" |
| break; | *to++=*from; |
| case '\\': // "\" -> "\\" | } |
| *to++='\\'; | } else { |
| break; | for(from=str; from<from_end; from++){ |
| switch (*from) { | |
| case '\'': // "'" -> "''" | |
| *to++= '\''; | |
| break; | |
| case '\\': // "\" -> "\\" | |
| *to++='\\'; | |
| break; | |
| } | |
| *to++=*from; | |
| } | } |
| *to++=*from; | |
| } | } |
| *to=0; | *to=0; |
| Line 545 private: | Line 568 private: |
| char *cur=result; | char *cur=result; |
| memcpy(cur, astatement, statement_size); cur+=statement_size; | memcpy(cur, astatement, statement_size); cur+=statement_size; |
| if(limit!=SQL_NO_LIMIT) | if(limit!=SQL_NO_LIMIT) |
| cur+=snprintf(cur, 7+MAX_NUMBER, " limit %u", limit); | cur+=snprintf(cur, 7+MAX_NUMBER, " limit %lu", limit); |
| if(offset) | if(offset) |
| cur+=snprintf(cur, 8+MAX_NUMBER, " offset %u", offset); | cur+=snprintf(cur, 8+MAX_NUMBER, " offset %lu", offset); |
| o=result; | o=result; |
| } else | } else |
| o=astatement; | o=astatement; |
| Line 688 private: // conn client library funcs | Line 711 private: // conn client library funcs |
| private: // conn client library funcs linking | private: // conn 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()) |
| return result; | |
| return "can not prepare to dynamic loading"; | |
| } | |
| lt_dlhandle handle=lt_dlopen(dlopen_file_spec); | lt_dlhandle handle=lt_dlopen(dlopen_file_spec); |
| if(!handle) | |
| 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 DSLINK(name, action) \ | #define DSLINK(name, action) \ |
| name=(t_##name)lt_dlsym(handle, #name); \ | name=(t_##name)lt_dlsym(handle, #name); \ |