include/libpgcli/pgconn.h

   1 /*Copyright (c) Brian B.
   2
   3   This library is free software; you can redistribute it and/or
   4   modify it under the terms of the GNU Lesser General Public
   5   License as published by the Free Software Foundation; either
   6   version 3 of the License, or (at your option) any later version.
   7   See the file LICENSE included with this distribution for more
   8   information.
   9 */
  10 #ifndef __PGCONN_H__
  11 #define __PGCONN_H__
  12
  13 #include <unistd.h>
  14 #include <endian.h>
  15 #include <sys/types.h>
  16 #include <sys/socket.h>
  17 #include <netdb.h>
  18 #include <netinet/in.h>
  19 #include <arpa/inet.h>
  20 #include <limits.h>
  21 #include <assert.h>
  22 #include <libex/list.h>
  23 #include <libex/net.h>
  24 #include "pgerrcodes.h"
  25 #include "pgprov3.h"
  26
  27 #define EPG_PORT 0x80000000
  28 #define EPG_ERRNO 0x40000000
  29 #define EPG_PROTO 0x20000000
  30 #define EPG_HOST 0x0fffffff
  31
  32 #define PG_NRECS(C) C->row_list->len
  33 #define PG_NFLDS(C) C->rowdesc->nflds
  34 #define PG_FLDLEN(C,R,F) C->rows[R]->fields[F].len
  35 #define PG_DATA(C,R,F) C->rows[R]->fields[F].data
  36 #define PG_ISNULL(C,R,F) C->rows[R]->fields[F].len >= 0 ? 0 : 1
  37
  38 #define CHECK_BOUNDS(C,R,F) (R >= 0 && R < PG_NRECS(C) && F >= 0 && F < PG_NFLDS(C))
  39
  40 typedef struct {
  41     char *code;
  42     char *text;
  43     char *msg;
  44     char *detail;
  45 } pgerror_t;
  46
  47 typedef struct pgpool pgpool_t;
  48
  49 typedef struct {
  50     int fd;
  51     list_t *msgs;
  52     list_t *row_list;
  53     int nflds;
  54     uint32_t intr_error;
  55     pgmsg_datarow_t **rows;
  56     pgmsg_error_t *error;
  57     pgmsg_ready_t *ready;
  58     pgmsg_rowdesc_t *rowdesc;
  59     pgmsg_cmd_complete_t *complete;
  60     pgmsg_t *suspended;
  61     pgmsg_t *emptyquery;
  62     char *date_style;
  63     char *client_encoding;
  64     char *server_encoding;
  65     char *session_auth;
  66     char *timezone;
  67     int is_int_datetimes;
  68     int is_superuser;
  69     int major_ver;
  70     int minor_ver;
  71     int authok;
  72     char *dbname;
  73     int cols;
  74     int fmt;
  75     cstr_t *nonce;
  76     time_t tm_released;
  77     list_item_t *li;
  78     pgpool_t *pool;
  79 } pgconn_t;
  80
  81 void pgconn_init (const char *pg_srv_addr, const char *pg_srv_service);
  82 pgconn_t *pg_connect (const char *url);
  83 const int pg_error (pgconn_t *conn, pgerror_t *e);
  84 static inline char pg_ready (pgconn_t *conn) {
  85     return conn->ready ? conn->ready->tr : 0;
  86 };
  87 int pg_execsql (pgconn_t *conn, const char *sql, size_t sql_len);
  88 int pg_copyin (pgconn_t *conn, const char *table_name);
  89 int pg_copyin_sendln (pgconn_t *conn, pgfld_t **flds);
  90 int pg_copyin_end (pgconn_t *conn);
  91 static inline int pg_start (pgconn_t* conn) {
  92     return pg_execsql(conn, CONST_STR_LEN("start transaction"));
  93 };
  94 static inline int pg_commit (pgconn_t *conn) {
  95     return pg_execsql(conn, CONST_STR_LEN("commit"));
  96 };
  97 static inline int pg_rollback (pgconn_t *conn) {
  98     return pg_execsql(conn, CONST_STR_LEN("rollback"));
  99 }
 100 int pg_prepareln (pgconn_t *conn, const char *name, size_t name_len,
 101                   const char *sql, size_t sql_len, int fld_len, pgfld_t **flds);
 102 static inline int pg_preparel (pgconn_t *conn, const char *sql, size_t sql_len, int fld_len, pgfld_t **flds) {
 103     return pg_prepareln(conn, CONST_STR_NULL, sql, sql_len, fld_len, flds);
 104 }
 105 int pg_preparen (pgconn_t *conn, const char *name, size_t name_len,
 106                  const char *sql, size_t sql_len, pgfld_t *fld, ...);
 107 int pg_prepare (pgconn_t *conn, const char *sql, size_t sql_len, pgfld_t *fld, ...);
 108 int pg_execln (pgconn_t *conn, const char *portal, size_t portal_len,
 109                const char *stmt, size_t stmt_len,
 110                int fld_len, pgfld_t **flds, int res_fmt_len, int *res_fmt, int max_data);
 111 static inline int pg_execl (pgconn_t *conn, int fld_len, pgfld_t **flds, int res_fmt_len, int *res_fmt, int max_data) {
 112     return pg_execln(conn, CONST_STR_NULL, CONST_STR_NULL, fld_len, flds, res_fmt_len, res_fmt, max_data);
 113 }
 114 int pg_execvn (pgconn_t *conn, const char *portal, size_t portal_len, const char *stmt, size_t stmt_len, int max_data, int fmt_out, pgfld_t *fld, va_list ap);
 115 static inline int pg_execv (pgconn_t *conn, int max_data, int out_fmt, pgfld_t *fld, va_list ap) {
 116     return pg_execvn(conn, CONST_STR_NULL, CONST_STR_NULL, max_data, out_fmt, fld, ap);
 117 }
 118 int pg_execn (pgconn_t *conn, const char *portal, size_t portal_len, const char *stmt, size_t stmt_len, int max_data, int out_fmt, pgfld_t *fld, ...);
 119 int pg_exec (pgconn_t *conn, int max_data, int out_fmt, pgfld_t *fld, ...);
 120 int pg_nextn (pgconn_t *conn, const char *portal, size_t portal_len, int max_data);
 121 static inline int pg_next (pgconn_t *conn, int max_data) {
 122     return pg_nextn(conn, CONST_STR_NULL, max_data);
 123 }
 124 static inline char pg_getready (pgconn_t *conn) {
 125     return conn->ready ? conn->ready->tr : 0;
 126 }
 127 int pg_release (pgconn_t *conn, const char *name, size_t name_len);
 128 void pg_close (pgconn_t *conn);
 129 void pg_disconnect(pgconn_t *conn);
 130
 131 static inline int pg_nrecs (pgconn_t *conn) { return PG_NRECS(conn); };
 132 static inline int pg_nflds (pgconn_t *conn) { return conn->rowdesc ? PG_NFLDS(conn) : 0; };
 133 static inline int pg_isnull (pgconn_t *conn, int row, int col) { return PG_ISNULL(conn, row, col); };
 134
 135 #define pg_getdate(C,R,F) pg_geti32(C,R,F)
 136 #define pg_gettm(C,R,F) pg_geti64(C,R,F)
 137 const strptr_t pg_get (pgconn_t *conn, int row, int col);
 138 static inline int pg_fldlen (pgconn_t *conn, int row, int col) {
 139     assert(CHECK_BOUNDS(conn, row, col));
 140     return PG_FLDLEN(conn, row, col);
 141 }
 142 cstr_t *pg_getstr (pgconn_t *conn, int row, int col);
 143 static inline int16_t pg_geti16 (pgconn_t *conn, int row, int col) {
 144     assert(CHECK_BOUNDS(conn, row, col));
 145     return be16toh(*(int16_t*)PG_DATA(conn, row, col));
 146 }
 147
 148 static inline int32_t pg_geti32 (pgconn_t *conn, int row, int col) {
 149     assert(CHECK_BOUNDS(conn, row, col));
 150     return be32toh(*(int32_t*)PG_DATA(conn, row, col));
 151 }
 152 static inline int64_t pg_geti64 (pgconn_t *conn, int row, int col) {
 153     assert(CHECK_BOUNDS(conn, row, col));
 154     return be64toh(*(int64_t*)PG_DATA(conn, row, col));
 155 }
 156 static inline double pg_getd (pgconn_t *conn, int row, int col) {
 157     assert(CHECK_BOUNDS(conn, row, col));
 158     return pg_conv_double(*(double*)PG_DATA(conn, row, col));
 159 }
 160 static inline float pg_getf (pgconn_t *conn, int row, int col) {
 161     assert(CHECK_BOUNDS(conn, row, col));
 162     return pg_conv_float(*(float*)PG_DATA(conn, row, col));
 163 }
 164 static inline int8_t pg_getbool (pgconn_t *conn, int row, int col) {
 165     assert(CHECK_BOUNDS(conn, row, col));
 166     return *PG_DATA(conn, row, col);
 167 }
 168 uint32_t pg_getx32 (pgconn_t *conn, int row, int col);
 169 pg_intv_t pg_getintv (pgconn_t *conn, int row, int col);
 170 int32_t pg_getinet4 (pgconn_t *conn, int row, int col);
 171 #ifdef HAVE_GMP
 172 static inline void pg_getnum (pgconn_t *conn, int row, int col, mpq_t res, uint16_t *dscale) {
 173     assert(CHECK_BOUNDS(conn, row, col));
 174     pg_get_numeric(PG_DATA(conn, row, col), res, dscale);
 175 }
 176 #endif
 177 static inline unsigned char *pg_getuuid (pgconn_t *conn, int row, int col) {
 178     assert(CHECK_BOUNDS(conn, row, col));
 179     return PG_DATA(conn, row, col);
 180 }
 181
 182 #endif