[BACK]Return to parser3pgsql.C CVS log [TXT] [DIR] Up to [parser3project] / sql / pgsql

Diff for /sql/pgsql/parser3pgsql.C between versions 1.37 and 1.50

version 1.37, 2012/03/16 10:05:04 version 1.50, 2021/11/07 22:16:54
Line 1 Line 1
 /** @file  /** @file
         Parser PgSQL driver.          Parser PgSQL driver.
   
         Copyright (c) 2001-2012 Art. Lebedev Studio (http://www.artlebedev.com)          Copyright (c) 2001-2019 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)
   
Line 15 Line 15
 #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;  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
Line 78  static char* rsplit(char* string, char d Line 78  static char* rsplit(char* string, char d
                         return v+1;                          return v+1;
                 }                  }
         }          }
         return NULL;              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){
Line 99  struct Connection { Line 99  struct Connection {
         PGconn *conn;          PGconn *conn;
         const char* client_charset;          const char* client_charset;
         bool autocommit;          bool autocommit;
         bool without_default_transactions;          bool standard_conforming_strings;
 };  };
   
 /**  /**
Line 116  public: Line 116  public:
   
         /// 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";  
         }          }
   
         #define throwPQerror connection.services->_throw(PQerrorMessage(connection.conn))          #define throwPQerror connection.services->_throw(PQerrorMessage(connection.conn))
         #define PQclear_throw(msg) { \          #define PQclear_throw(msg) { \
                         PQclear(res); \                          PQclear(res); \
                         connection.services->_throw(msg); \                          connection.services->_throw(msg); \
                 }                                                                 }
         #define PQclear_throwPQerror PQclear_throw(PQerrorMessage(connection.conn))          #define PQclear_throwPQerror PQclear_throw(PQerrorMessage(connection.conn))
   
         /**     connect          /**     connect
Line 133  public: Line 132  public:
                         ClientCharset=charset&  // transcode by parser                          ClientCharset=charset&  // transcode by parser
                         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=0&                   // 1 -- each statement is commited automatically, only when with_default_transaction enabled
                         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
         */          */
         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* host=rsplit(user, '@');                  char* host=rsplit(user, '@');
                 char* db=lsplit(host, '/');                  char* db=lsplit(host, '/');
                 char* pwd=lsplit(user, ':');                  char* pwd=lsplit(user, ':');
                 char* port=lsplit(host, ':');  
   
                 char *options=lsplit(db, '?');                  char *options=lsplit(db, '?');
   
                 char* charset=0;                  char* charset=0;
                 char* datestyle=0;                  char* datestyle=0;
   
                   char* pq_options=0;
                   size_t  pq_options_len=options ? strlen(options) : 0;
   
                 Connection& connection=*(Connection *)services.malloc(sizeof(Connection));                  Connection& connection=*(Connection *)services.malloc(sizeof(Connection));
   
                 *connection_ref=&connection;                  *connection_ref=&connection;
                 connection.services=&services;                  connection.services=&services;
                 connection.client_charset=0;                      connection.client_charset=0;
                 connection.autocommit=true;                  connection.autocommit=true;
                 connection.without_default_transactions=false;                  connection.standard_conforming_strings=true;
   
                 connection.conn=PQsetdbLogin(  
                         (host&&strcasecmp(host, "local")==0)?NULL/* local Unix domain socket */:host, port,   
                         NULL, NULL, db, user, pwd);  
   
                 if(!connection.conn)  
                         services._throw("PQsetdbLogin failed");  
   
                 if(PQstatus(connection.conn)!=CONNECTION_OK)  
                         throwPQerror;  
   
                 while(options){                  while(options){
                         if(char *key=lsplit(&options, '&')){                          if(char *key=lsplit(&options, '&')){
Line 181  public: Line 169  public:
                                                 } else if(strcasecmp(key, "datestyle")==0){                                                  } else if(strcasecmp(key, "datestyle")==0){
                                                         datestyle=value;                                                          datestyle=value;
                                                 } else if(strcasecmp(key, "autocommit")==0){                                                  } else if(strcasecmp(key, "autocommit")==0){
                                                         if(connection.without_default_transactions)  
                                                                 services._throw("options WithoutDefaultTransaction and autocommit can't be used together");  
                                                         if(atoi(value)==0)                                                          if(atoi(value)==0)
                                                                 connection.autocommit=false;                                                                  connection.autocommit=false;
                                                 } else if(strcmp(key, "WithoutDefaultTransaction")==0){                                                  } else if(strcasecmp(key, "standard_conforming_strings")==0){
                                                         if(!connection.autocommit)                                                          if(atoi(value)==0)
                                                                 services._throw("options WithoutDefaultTransaction and autocommit can't be used together");                                                                  connection.standard_conforming_strings=false;
                                                         if(atoi(value)==1){                                                  } else {
                                                                 connection.without_default_transactions=true;                                                          if(!pq_options) {
                                                                 connection.autocommit=false;                                                                  pq_options=(char*)services.malloc_atomic(pq_options_len+2);
                                                                   strcpy(pq_options, "?");
                                                           } else {
                                                                   strcat(pq_options, "&");
                                                         }                                                          }
                                                 } else                                                          strcat(pq_options, key);
                                                         services._throw("unknown connect option" /*key*/);                                                          strcat(pq_options, "=");
                                                           strcat(pq_options, value);
                                                   }
                                         } else                                           } else 
                                                 services._throw("connect option without =value" /*key*/);                                                  services._throw("connect option without =value" /*key*/);
                                 }                                  }
                         }                          }
                 }                  }
   
                   if(host && (strchr(host, ',') || pq_options)){ // pq_options can exist only if host and db are not null
                           char pq_url[MAX_STRING+1];
                           snprintf(pq_url, MAX_STRING, "postgresql://%s/%s%s", host, db ? db : "", pq_options ? pq_options : "");
                           connection.conn=PQsetdbLogin(NULL, NULL, NULL, NULL, pq_url, user, pwd);
                   } else {
                           char* port=lsplit(host, ':');
                           connection.conn=PQsetdbLogin( (host && strcasecmp(host, "local") == 0) ? NULL /* local Unix domain socket */ : host, port, NULL, NULL, db, user, pwd);
                   }
   
                   if(!connection.conn)
                           services._throw("PQsetdbLogin failed");
   
                   if(PQstatus(connection.conn)!=CONNECTION_OK)
                           throwPQerror;
   
                 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);
                 }                  }
   
                 _transaction_begin(connection);                  if(!connection.autocommit)
                           _execute_cmd(connection, "BEGIN");
         }          }
   
         void disconnect(void *aconnection){          void disconnect(void *aconnection){
Line 225  public: Line 232  public:
   
         void commit(void *aconnection){          void commit(void *aconnection){
                 Connection& connection=*static_cast<Connection*>(aconnection);                  Connection& connection=*static_cast<Connection*>(aconnection);
                 _transaction_commit(connection);                  if(!connection.autocommit)
                 _transaction_begin(connection);                          _execute_cmd(connection, "COMMIT");
         }          }
   
         void rollback(void *aconnection){          void rollback(void *aconnection){
                 Connection& connection=*static_cast<Connection*>(aconnection);                  Connection& connection=*static_cast<Connection*>(aconnection);
                 _transaction_rollback(connection);                  if(!connection.autocommit)
                 _transaction_begin(connection);                          _execute_cmd(connection, "ROLLBACK");
         }          }
   
         bool ping(void *aconnection) {          bool ping(void *aconnection) {
Line 242  public: Line 249  public:
   
         // charset here is services.request_charset(), not connection.client_charset          // charset here is services.request_charset(), not connection.client_charset
         // 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;
                 return result;                  return result;
         }          }
   
         void query(void *aconnection,           void query(void *aconnection, const char *astatement, size_t placeholders_count, Placeholder* placeholders, unsigned long offset, unsigned long limit, SQL_Driver_query_event_handlers& handlers ){
                                 const char *astatement,   
                                 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;
                 PGconn *conn=connection.conn;                  PGconn *conn=connection.conn;
Line 305  public: Line 322  public:
                 if(transcode_needed){                  if(transcode_needed){
                         // transcode query from $request:charset to ?ClientCharset                          // transcode query from $request:charset to ?ClientCharset
                         statement_size=strlen(astatement);                          statement_size=strlen(astatement);
                         services.transcode(astatement, statement_size,                          services.transcode(astatement, statement_size, astatement, statement_size, request_charset, client_charset);
                                 astatement, statement_size,  
                                 request_charset,  
                                 client_charset);  
                 }                  }
   
                 const char *statement=_preprocess_statement(connection, astatement, statement_size, offset, limit);                  const char *statement=_preprocess_statement(connection, astatement, statement_size, offset, limit);
Line 327  public: Line 341  public:
                 SQL_Error sql_error;                  SQL_Error sql_error;
   
                 switch(PQresultStatus(res)) {                  switch(PQresultStatus(res)) {
                         case PGRES_EMPTY_QUERY:                           case PGRES_EMPTY_QUERY:
                                 PQclear_throw("no query");                                  PQclear_throw("no query");
                                 break;                                  break;
                         case PGRES_COMMAND_OK: // empty result: insert|delete|update|...                          case PGRES_COMMAND_OK: // empty result: insert|delete|update|...
                                 PQclear(res);                                  PQclear(res);
                                 if(connection.autocommit)  
                                         commit(aconnection);  
                                 return;                                  return;
                         case PGRES_TUPLES_OK:                           case PGRES_TUPLES_OK: 
                                 break;                                    break;
                         default:                          default:
                                 PQclear_throwPQerror;                                  PQclear_throwPQerror;
                                 break;                                  break;
Line 366  public: Line 378  public:
   
                         if(transcode_needed)                          if(transcode_needed)
                                 // transcode column name from ?ClientCharset to $request:charset                                  // transcode column name from ?ClientCharset to $request:charset
                                 services.transcode(str, length,                                  services.transcode(str, length, str, length, client_charset, request_charset);
                                         str, length,  
                                         client_charset,  
                                         request_charset);  
   
                         CHECK(handlers.add_column(sql_error, str, length));                          CHECK(handlers.add_column(sql_error, str, length));
                 }                  }
Line 460  cleanup: Line 469  cleanup:
                 PQclear(res);                  PQclear(res);
                 if(failed)                  if(failed)
                         services._throw(sql_error);                          services._throw(sql_error);
   
                 if(connection.autocommit)  
                         commit(aconnection);  
         }          }
   
 private:  private:
         void _bind_parameters(          void _bind_parameters( size_t placeholders_count, Placeholder* placeholders, const char** paramValues, Connection& connection, bool transcode_needed){
                                 size_t placeholders_count,   
                                 Placeholder* placeholders,   
                                 const char** paramValues,  
                                 Connection& connection,  
                                 bool transcode_needed  
         ){  
                 for(size_t i=0; i<placeholders_count; i++){                  for(size_t i=0; i<placeholders_count; i++){
                         Placeholder& ph=placeholders[i];                          Placeholder& ph=placeholders[i];
                         if(transcode_needed){                          if(transcode_needed){
                                 size_t name_length;                                  size_t name_length;
                                 connection.services->transcode(ph.name, strlen(ph.name),                                  connection.services->transcode(ph.name, strlen(ph.name), ph.name, name_length, connection.services->request_charset(), connection.client_charset);
                                         ph.name, name_length,  
                                         connection.services->request_charset(),  
                                         connection.client_charset);  
   
                                 if(ph.value) {                                  if(ph.value) {
                                         size_t value_length;                                          size_t value_length;
                                         connection.services->transcode(ph.value, strlen(ph.value),                                          connection.services->transcode(ph.value, strlen(ph.value), ph.value, value_length, connection.services->request_charset(), connection.client_charset);
                                                 ph.value, value_length,  
                                                 connection.services->request_charset(),  
                                                 connection.client_charset);  
                                 }                                  }
                         }                          }
                         int name_number=atoi(ph.name);                          int name_number=atoi(ph.name);
Line 498  private: Line 492  private:
                 }                  }
         }          }
                   
           
         void _transaction_begin(Connection& connection){  
                 _execute_transactions_cmd(connection, "BEGIN");  
         }  
   
         void _transaction_commit(Connection& connection){  
                 _execute_transactions_cmd(connection, "COMMIT");  
         }  
   
         void _transaction_rollback(Connection& connection){  
                 _execute_transactions_cmd(connection, "ROLLBACK");  
         }  
   
         void _execute_transactions_cmd(const Connection& connection, const char *query){  
                 if(!connection.without_default_transactions) // with option ?WithoutDefaultTransaction=1 user must execute BEGIN/COMMIT/ROLLBACK by himself  
                         _execute_cmd(connection, query);  
         }  
   
         // executes a query and throw away the result.          // executes a query and throw away the result.
         void _execute_cmd(const Connection& connection, const char *query){          void _execute_cmd(const Connection& connection, const char *query){
                 if(PGresult *res=PQexec(connection.conn, query))                  if(PGresult *res=PQexec(connection.conn, query))
Line 524  private: Line 500  private:
                         throwPQerror;                          throwPQerror;
         }          }
   
         const char *_preprocess_statement(          const char *_preprocess_statement(Connection& connection, const char *astatement, size_t statement_size, unsigned long offset, unsigned long limit){
                                         Connection& connection,  
                                         const char *astatement,  
                                         size_t statement_size,  
                                         unsigned long offset,  
                                         unsigned long limit  
         ){  
                 PGconn *conn=connection.conn;                  PGconn *conn=connection.conn;
   
                 if(!statement_size)                  if(!statement_size)
Line 546  private: Line 516  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 563  private: Line 533  private:
                                 const char* saved_o=o;                                  const char* saved_o=o;
                                 o+=3;                                  o+=3;
                                 while(*o)                                  while(*o)
                                         if(                                          if(o[0]=='*' &&         o[1]=='*' && o[2]=='/' && o[3]=='\'') { // name end
                                                 o[0]=='*' &&  
                                                 o[1]=='*' &&  
                                                 o[2]=='/' &&  
                                                 o[3]=='\'') { // name end  
                                                 saved_o=0; // found, marking that                                                  saved_o=0; // found, marking that
                                                 o+=4;                                                  o+=4;
                                                 Oid oid=lo_creat(conn, INV_READ|INV_WRITE);                                                  Oid oid=lo_creat(conn, INV_READ|INV_WRITE);
Line 689  private: // conn client library funcs Line 655  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); \
Line 717  private: // conn client library funcs li Line 691  private: // conn client library funcs li
                 DLINK(PQexecParams);                  DLINK(PQexecParams);
                 DLINK(PQftype);                  DLINK(PQftype);
                 DLINK(PQescapeStringConn);                  DLINK(PQescapeStringConn);
                 DLINK(lo_open);         DLINK(lo_close);                  DLINK(lo_open);   DLINK(lo_close);
                 DLINK(lo_read);         DLINK(lo_write);                  DLINK(lo_read);   DLINK(lo_write);
                 DLINK(lo_lseek);                DLINK(lo_creat);                  DLINK(lo_lseek);  DLINK(lo_creat);
                 DLINK(lo_tell);         DLINK(lo_unlink);                  DLINK(lo_tell);   DLINK(lo_unlink);
                 DLINK(lo_import);               DLINK(lo_export);                  DLINK(lo_import); DLINK(lo_export);
   
                 return 0;                  return 0;
         }          }
Removed from v.1.37  
changed lines
  Added in v.1.50

E-mail: