--- parser3/src/sql/pgsql/Attic/parser3pgsql.C 2001/07/31 07:47:08 1.2 +++ parser3/src/sql/pgsql/Attic/parser3pgsql.C 2001/08/01 09:33:54 1.9 @@ -7,13 +7,21 @@ 2001.07.30 using PgSQL 7.1.2 */ -static const char *RCSId="$Id: parser3pgsql.C,v 1.2 2001/07/31 07:47:08 parser Exp $"; +static const char *RCSId="$Id: parser3pgsql.C,v 1.9 2001/08/01 09:33:54 parser Exp $"; #include "config_includes.h" #include "pa_sql_driver.h" #include +#include + +// OIDOID from #include +#define OIDOID 26 +// LO_BUFSIZE from interfaces\libpq\fe-lobj.c = 8192 (0x2000) +// actually writing chunks of that size failed, reduced it twice +#define LO_BUFSIZE 0x1000 + #include "ltdl.h" #define MAX_STRING 0x400 @@ -24,6 +32,11 @@ static const char *RCSId="$Id: parser3pg # define strcasecmp _stricmp #endif +#ifndef max +inline int max(int a,int b) { return a>b?a:b; } +inline int min(int a,int b){ return a '' unsigned int result=length; while(length--) { - if(*from=='\'') + switch(*from) { + case '\'': // "'" -> "''" *to++='\''; + break; + case '\\': // "\" -> "\\" + *to++='\''; + break; + } *to++=*from++; } return result; @@ -114,49 +141,40 @@ public: SQL_Driver_services& services, void *connection, const char *astatement, unsigned long offset, unsigned long limit, SQL_Driver_query_event_handlers& handlers) { +// _asm int 3; - PGconn *pgsql=(PGconn *)connection; + PGconn *conn=(PGconn *)connection; + #define PQclear_throw(msg) { \ + PQclear(res); \ + services._throw(msg); \ + } + #define PQclear_throwPQerror PQclear_throw(PQerrorMessage(conn)) - const char *statement; - if(offset || limit) { - size_t statement_size=strlen(astatement); - char *statement_limited=(char *)services.malloc( - statement_size+MAX_NUMBER*2+15/* limit # offset #*/+1); - char *cur=statement_limited; - memcpy(cur, astatement, statement_size); cur+=statement_size; - if(limit) - cur+=snprintf(cur, 7+MAX_NUMBER, " limit %u", limit); - if(offset) - cur+=snprintf(cur, 8+MAX_NUMBER, " offset %u", offset); - statement=statement_limited; - } else - statement=astatement; + const char *statement=preprocess_statement(services, conn, + astatement, offset, limit); - PGresult *res=PQexec(pgsql, statement); + PGresult *res=PQexec(conn, statement); if(!res) - services._throw(PQerrorMessage(pgsql)); + throwPQerror; switch(PQresultStatus(res)) { case PGRES_EMPTY_QUERY: - PQclear(res); - services._throw("no query"); + PQclear_throw("no query"); break; case PGRES_COMMAND_OK: // empty result: insert|delete|update|... + PQclear(res); return; case PGRES_TUPLES_OK: break; default: - PQclear(res); - services._throw("unknown PQexec error"); + PQclear_throwPQerror; break; } int column_count=PQnfields(res); - if(!column_count) { - PQclear(res); - services._throw("result contains no columns"); - } + if(!column_count) + PQclear_throw("result contains no columns"); for(int i=0; i=0) { + // seek to end + if(lo_lseek(conn, fd, 0, SEEK_END)<0) + PQclear_throwPQerror; + // get size + int size_tell=lo_tell(conn, fd); + if(size_tell<0) + PQclear_throwPQerror; + // seek to begin + if(lo_lseek(conn, fd, 0, SEEK_SET)<0) + PQclear_throwPQerror; + size=(size_t)size_tell; + if(size) { + // read + ptr=services.malloc(size); + if(!lo_read_ex(conn, fd, (const char *)ptr, size_tell)) + PQclear_throw("lo_read can not read all bytes of object"); + } else + ptr=0; + if(lo_close(conn, fd)<0) + PQclear_throwPQerror; + } else + PQclear_throwPQerror; + } else { + // normal column, read it normally + size=(size_t)PQgetlength(res, r, i); + if(size) { + ptr=services.malloc(size); + memcpy(ptr, cell, size); + } else + ptr=0; + } handlers.add_row_cell(ptr, size); } } @@ -186,7 +238,110 @@ public: PQclear(res); } -private: // pgsql client library funcs +private: // private funcs + + void begin_transaction(SQL_Driver_services& services, PGconn *conn) { + if(PGresult *res=PQexec(conn, "BEGIN")) + PQclear(res); + else + throwPQerror; + } + + const char *preprocess_statement(SQL_Driver_services& services, PGconn *conn, + const char *astatement, unsigned long offset, unsigned long limit) { + size_t statement_size=strlen(astatement); + + char *result=(char *)services.malloc(statement_size + +MAX_NUMBER*2+15 // limit # offset # + +MAX_STRING // in case of short 'strings' + +1); + // offset & limit -> suffixes + const char *o; + if(offset || limit) { + char *cur=result; + memcpy(cur, astatement, statement_size); cur+=statement_size; + if(limit) + cur+=snprintf(cur, 7+MAX_NUMBER, " limit %u", limit); + if(offset) + cur+=snprintf(cur, 8+MAX_NUMBER, " offset %u", offset); + o=result; + } else + o=astatement; + + // /**xxx**/'literal' -> oid + char *n=result; + while(*o) { + if( + o[0]=='/' && + o[1]=='*' && + o[2]=='*') { // name start + o+=3; + while(*o) + if( + o[0]=='*' && + o[1]=='*' && + o[2]=='/' && + o[3]=='\'') { // name end + o+=4; + Oid oid=lo_creat(conn, INV_READ|INV_WRITE); + if(oid==InvalidOid) + throwPQerror; + int fd=lo_open(conn, oid, INV_WRITE); + if(fd>=0) { + const char *start=o; + bool escaped=false; + while(*o && !(o[0]=='\'' && o[1]!='\'' && !escaped)) { + escaped=*o=='\\' || (o[0]=='\'' && o[1]=='\''); + if(escaped) { + // write pending, skip "\" or "'" + if(!lo_write_ex(conn, fd, start, o-start)) + services._throw("lo_write could not write all bytes of object (1)"); + start=++o; + } else + o++; + } + if(!lo_write_ex(conn, fd, start, o-start)) + services._throw("lo_write can not write all bytes of object (2)"); + if(lo_close(conn, fd)<0) + throwPQerror; + } else + throwPQerror; + if(*o) + o++; // skip "'" + + n+=snprintf(n, MAX_NUMBER, "%u", oid); + break; + } else + o++; // /**skip**/'xxx' + } else + *n++=*o++; + } + *n=0; + + return result; + } + +private: // lo_read/write exchancements + + bool lo_read_ex(PGconn *conn, int fd, const/*paf*/ char *buf, size_t len) { + int size_read; + while(len && (size_read=lo_read(conn, fd, buf, min(LO_BUFSIZE, len)))>0) { + buf+=size_read; + len-=size_read; + } + return len==0; + } + + bool lo_write_ex(PGconn *conn, int fd, const/*paf*/ char *buf, size_t len) { + int size_written; + while(len && (size_written=lo_write(conn, fd, buf, min(LO_BUFSIZE, len)))>0) { + buf+=size_written; + len-=size_written; + } + return len==0; + } + +private: // conn client library funcs typedef PGconn* (*t_PQsetdbLogin)( const char *pghost, @@ -214,8 +369,20 @@ private: // pgsql client library funcs typedef int (*t_PQnfields)(const PGresult *res); t_PQnfields PQnfields; typedef void (*t_PQclear)(PGresult *res); t_PQclear PQclear; + typedef Oid (*t_PQftype)(const PGresult *res, int field_num); t_PQftype PQftype; + + typedef int (*t_lo_open)(PGconn *conn, Oid lobjId, int mode); t_lo_open lo_open; + typedef int (*t_lo_close)(PGconn *conn, int fd); t_lo_close lo_close; + typedef int (*t_lo_read)(PGconn *conn, int fd, const/*paf*/ char *buf, size_t len); t_lo_read lo_read; + typedef int (*t_lo_write)(PGconn *conn, int fd, const/*paf*/ char *buf, size_t len); t_lo_write lo_write; + typedef int (*t_lo_lseek)(PGconn *conn, int fd, int offset, int whence); t_lo_lseek lo_lseek; + typedef Oid (*t_lo_creat)(PGconn *conn, int mode); t_lo_creat lo_creat; + typedef int (*t_lo_tell)(PGconn *conn, int fd); t_lo_tell lo_tell; + typedef int (*t_lo_unlink)(PGconn *conn, Oid lobjId); t_lo_unlink lo_unlink; + typedef Oid (*t_lo_import)(PGconn *conn, const char *filename); t_lo_import lo_import; + typedef int (*t_lo_export)(PGconn *conn, Oid lobjId, const char *filename); t_lo_export lo_export; -private: // pgsql client library funcs linking +private: // conn client library funcs linking const char *dlink(const char *dlopen_file_spec) { lt_dlhandle handle=lt_dlopen(dlopen_file_spec); @@ -241,6 +408,12 @@ private: // pgsql client library funcs l DLINK(PQclear); DLINK(PQresultStatus); DLINK(PQexec); + DLINK(PQftype); + DLINK(lo_open); DLINK(lo_close); + DLINK(lo_read); DLINK(lo_write); + DLINK(lo_lseek); DLINK(lo_creat); + DLINK(lo_tell); DLINK(lo_unlink); + DLINK(lo_import); DLINK(lo_export); return 0; }