/*------------------------------------------------------------------------- * * oracle_utils.c * routines that use OCI (Oracle's C API) * *------------------------------------------------------------------------- */ #include #include #include #include #include #if defined _WIN32 || defined _WIN64 /* for getpid */ #include #endif /* Oracle header */ #include #include "oracle_fdw.h" /* number of rows and memory limit for prefetch */ #define PREFETCH_ROWS 200 #define PREFETCH_MEMORY 24576 /* number of bytes to read per LOB chunk */ #define LOB_CHUNK_SIZE 8132 /* emit no error messages when set, used for shutdown */ static int silent = 0; /* contains Oracle error messages, set by checkerr() */ #define ERRBUFSIZE 1000 static char oraMessage[ERRBUFSIZE]; static sb4 err_code; /* set to "1" as soon as OCIEnvCreate is called */ static int oci_initialized = 0; /* * Linked list for temporary Oracle handles and descriptors. * Stores statement and describe handles as well as timetamp and LOB descriptors. * Other handles are stored in the handle cache below. */ struct handleEntry { dvoid *handlep; ub4 type; int isDescriptor; struct handleEntry *next; }; /* * Linked list for connection, server and environment handles. * "envlist" is the starting point. * Oracle sessions can be multiplexed over one server connection. */ struct connEntry { char *user; OCISvcCtx *svchp; OCISession *userhp; struct handleEntry *handlelist; int xact_level; /* 0 = none, 1 = main, else subtransaction */ struct connEntry *next; }; struct srvEntry { char *connectstring; OCIServer *srvhp; struct srvEntry *next; struct connEntry *connlist; }; static struct envEntry { char *nls_lang; OCIEnv *envhp; OCIError *errhp; struct envEntry *next; struct srvEntry *srvlist; } *envlist = NULL; /* * Represents one Oracle connection, points to cached entries. * This is necessary to be able to pass them back to * oracle_fdw.c without having to #include oci.h there. */ struct oracleSession { struct envEntry *envp; struct srvEntry *srvp; struct connEntry *connp; OCIStmt *stmthp; }; /* * Helper functions */ static void oracleSetSavepoint(oracleSession *session, int nest_level); static void setOracleEnvironment(char *nls_lang); static void oracleQueryPlan(oracleSession *session, const char *query, const char *desc_query, int nres, dvoid **res, sb4 *res_size, ub2 *res_type, ub2 *res_len, sb2 *res_ind); static sword checkerr(sword status, dvoid *handle, ub4 handleType); static char *copyOraText(const char *string, int size, int quote); static void closeSession(OCIEnv *envhp, OCIServer *srvhp, OCISession *userhp, int disconnect); static void disconnectServer(OCIEnv *envhp, OCIServer *srvhp); static void removeEnvironment(OCIEnv *envhp); static void allocHandle(dvoid **handlep, ub4 type, int isDescriptor, OCIEnv *envhp, struct connEntry *connp, oraError error, const char *errmsg); static void freeHandle(dvoid *handlep, struct connEntry *connp); static ub2 getOraType(oraType arg); static sb4 bind_out_callback(void *octxp, OCIBind *bindp, ub4 iter, ub4 index, void **bufpp, ub4 **alenp, ub1 *piecep, void **indp, ub2 **rcodep); static sb4 bind_in_callback(void *ictxp, OCIBind *bindp, ub4 iter, ub4 index, void **bufpp, ub4 *alenp, ub1 *piecep, void **indpp); static void oracleWriteLob(oracleSession *session, OCILobLocator *locptr, char *value); /* * oracleGetSession * Look up an Oracle connection in the cache, create a new one if there is none. * The result is a palloc'ed data structure containing the connection. * "curlevel" is the current PostgreSQL transaction level. */ oracleSession *oracleGetSession(const char *connectstring, char *user, char *password, const char *nls_lang, const char *tablename, int curlevel) { OCIEnv *envhp = NULL; OCIError *errhp = NULL; OCISvcCtx *svchp = NULL; OCIServer *srvhp = NULL; OCISession *userhp = NULL; OCITrans *txnhp = NULL; oracleSession *session; struct envEntry *envp; struct srvEntry *srvp; struct connEntry *connp; char pid[30], *nlscopy = NULL; ub4 is_connected; /* it's easier to deal with empty strings */ if (!connectstring) connectstring = ""; if (!user) user = ""; if (!password) password = ""; if (!nls_lang) nls_lang = ""; /* search environment and server handle in cache */ for (envp = envlist; envp != NULL; envp = envp->next) { if (strcmp(envp->nls_lang, nls_lang) == 0) { envhp = envp->envhp; errhp = envp->errhp; break; } } if (envhp == NULL) { /* * Create environment and error handle. */ /* create persistent copy of "nls_lang" */ if ((nlscopy = strdup(nls_lang)) == NULL) oracleError_i(FDW_OUT_OF_MEMORY, "error connecting to Oracle: failed to allocate %d bytes of memory", strlen(nls_lang) + 1); /* set Oracle environment */ setOracleEnvironment(nlscopy); /* create environment handle */ if (checkerr( OCIEnvCreate((OCIEnv **) &envhp, (ub4) OCI_DEFAULT, (dvoid *) 0, (dvoid * (*)(dvoid *,size_t)) 0, (dvoid * (*)(dvoid *, dvoid *, size_t)) 0, (void (*)(dvoid *, dvoid *)) 0, (size_t) 0, (dvoid **) 0), (dvoid *)envhp, OCI_HTYPE_ENV) != 0) { free(nlscopy); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIEnvCreate failed to create environment handle", oraMessage); } /* we can call OCITerminate now */ oci_initialized = 1; /* allocate error handle */ if (checkerr( OCIHandleAlloc((dvoid *) envhp, (dvoid **) &errhp, (ub4) OCI_HTYPE_ERROR, (size_t) 0, (dvoid **) 0), (dvoid *)envhp, OCI_HTYPE_ENV) != OCI_SUCCESS) { free(nlscopy); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIHandleAlloc failed to allocate error handle", oraMessage); } /* add handles to cache */ if ((envp = malloc(sizeof(struct envEntry))) == NULL) { oracleError_i(FDW_OUT_OF_MEMORY, "error connecting to Oracle: failed to allocate %d bytes of memory", sizeof(struct envEntry)); } envp->nls_lang = nlscopy; envp->envhp = envhp; envp->errhp = errhp; envp->srvlist = NULL; envp->next = envlist; envlist = envp; } /* search connect string in cache */ for (srvp = envp->srvlist; srvp != NULL; srvp = srvp->next) { if (strcmp(srvp->connectstring, connectstring) == 0) { srvhp = srvp->srvhp; break; } } if (srvp != NULL) { /* * Test if we are still connected. * If not, clean up the mess. */ if (checkerr( OCIAttrGet((dvoid *)srvhp, (ub4)OCI_HTYPE_SERVER, (dvoid *)&is_connected, (ub4 *)0, (ub4)OCI_ATTR_SERVER_STATUS, errhp), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error connecting to Oracle: OCIAttrGet failed to get connection status", oraMessage); } if (is_connected == OCI_SERVER_NOT_CONNECTED) { /* clean up */ silent = 1; while (srvp->connlist != NULL) { closeSession(envhp, srvhp, srvp->connlist->userhp, 0); } disconnectServer(envhp, srvhp); silent = 0; srvp = NULL; } } if (srvp == NULL) { /* * No cache entry was found, we have to create a new server connection. */ /* create new server handle */ if (checkerr( OCIHandleAlloc((dvoid *) envhp, (dvoid **) &srvhp, (ub4) OCI_HTYPE_SERVER, (size_t) 0, (dvoid **) 0), (dvoid *)envhp, OCI_HTYPE_ENV) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIHandleAlloc failed to allocate server handle", oraMessage); } /* connect to the Oracle server */ if (checkerr( OCIServerAttach(srvhp, errhp, (text *)connectstring, strlen(connectstring), 0), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_sd(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "connection for foreign table \"%s\" cannot be established", tablename, oraMessage); } /* add server handle to cache */ if ((srvp = malloc(sizeof(struct srvEntry))) == NULL) { oracleError_i(FDW_OUT_OF_MEMORY, "error connecting to Oracle: failed to allocate %d bytes of memory", sizeof(struct srvEntry)); } if ((srvp->connectstring = strdup(connectstring)) == NULL) { oracleError_i(FDW_OUT_OF_MEMORY, "error connecting to Oracle: failed to allocate %d bytes of memory", strlen(connectstring) + 1); } srvp->srvhp = srvhp; srvp->next = envp->srvlist; srvp->connlist = NULL; envp->srvlist = srvp; } /* search user session for this server in cache */ for (connp = srvp->connlist; connp != NULL; connp = connp->next) { if (strcmp(connp->user, user) == 0) { svchp = connp->svchp; userhp = connp->userhp; break; } } if (userhp == NULL) { /* * If no cached user session was found, authenticate. */ /* allocate service handle */ if (checkerr( OCIHandleAlloc((dvoid *) envhp, (dvoid **) &svchp, (ub4) OCI_HTYPE_SVCCTX, (size_t) 0, (dvoid **) 0), (dvoid *)envhp, OCI_HTYPE_ENV) != OCI_SUCCESS) { free(nlscopy); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIHandleAlloc failed to allocate service handle", oraMessage); } /* set server handle in service handle */ if (checkerr( OCIAttrSet(svchp, OCI_HTYPE_SVCCTX, srvhp, 0, OCI_ATTR_SERVER, errhp), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIAttrSet failed to set server handle in service handle", oraMessage); } /* create transaction handle */ if (checkerr( OCIHandleAlloc((dvoid *) envhp, (dvoid **) &txnhp, (ub4) OCI_HTYPE_TRANS, (size_t) 0, (dvoid **) 0), (dvoid *)envhp, OCI_HTYPE_ENV) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIHandleAlloc failed to allocate transaction handle", oraMessage); } /* set transaction handle in service handle */ if (checkerr( OCIAttrSet(svchp, OCI_HTYPE_SVCCTX, txnhp, 0, OCI_ATTR_TRANS, errhp), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIAttrSet failed to set transaction handle in service handle", oraMessage); } /* create session handle */ if (checkerr( OCIHandleAlloc((dvoid *) envhp, (dvoid **) &userhp, (ub4) OCI_HTYPE_SESSION, (size_t) 0, (dvoid **) 0), (dvoid *)envhp, OCI_HTYPE_ENV) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIHandleAlloc failed to allocate session handle", oraMessage); } /* set MODULE for the Oracle session */ sprintf(pid, "%lu", (unsigned long)getpid()); pid[29] = '\0'; if (checkerr( OCIAttrSet(userhp, OCI_HTYPE_SESSION, "postgres", (ub4)8, OCI_ATTR_MODULE, errhp), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIAttrSet failed to set module in session handle", oraMessage); } /* set ACTION for the Oracle session */ if (checkerr( OCIAttrSet(userhp, OCI_HTYPE_SESSION, pid, (ub4)strlen(pid), OCI_ATTR_ACTION, errhp), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIAttrSet failed to set action in session handle", oraMessage); } /* set user name */ if (checkerr( OCIAttrSet(userhp, OCI_HTYPE_SESSION, user, strlen(user), OCI_ATTR_USERNAME, errhp), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIAttrSet failed to set user name in session handle", oraMessage); } /* set password */ if (checkerr( OCIAttrSet(userhp, OCI_HTYPE_SESSION, password, strlen(password), OCI_ATTR_PASSWORD, errhp), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIAttrSet failed to set password in session handle", oraMessage); } /* authenticate; use external authentication if no username has been supplied */ if (checkerr( OCISessionBegin(svchp, errhp, userhp, (user[0] == '\0' ? OCI_CRED_EXT : OCI_CRED_RDBMS), OCI_DEFAULT), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_sd(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "session for foreign table \"%s\" cannot be authenticated", tablename, oraMessage); } /* set session handle in service handle */ if (checkerr( OCIAttrSet(svchp, OCI_HTYPE_SVCCTX, userhp, 0, OCI_ATTR_SESSION, errhp), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCIAttrSet failed to set session handle in service handle", oraMessage); } /* add session handle to cache */ if ((connp = malloc(sizeof(struct connEntry))) == NULL) { oracleError_i(FDW_OUT_OF_MEMORY, "error connecting to Oracle: failed to allocate %d bytes of memory", sizeof(struct connEntry)); } if ((connp->user = strdup(user)) == NULL) { oracleError_i(FDW_OUT_OF_MEMORY, "error connecting to Oracle: failed to allocate %d bytes of memory", sizeof(strlen(user) + 1)); } connp->svchp = svchp; connp->userhp = userhp; connp->handlelist = NULL; connp->xact_level = 0; connp->next = srvp->connlist; srvp->connlist = connp; /* register callback for PostgreSQL transaction events */ oracleRegisterCallback(connp); } if (connp->xact_level <= 0) { oracleDebug2("oracle_fdw: begin serializable remote transaction"); /* start a read-only or "serializable" (= repeatable read) transaction */ if (checkerr( OCITransStart(svchp, errhp, (uword)0, OCI_TRANS_SERIALIZABLE), (dvoid *)errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle: OCITransStart failed to start a transaction", oraMessage); } connp->xact_level = 1; } /* palloc a data structure pointing to the cached entries */ session = oracleAlloc(sizeof(struct oracleSession)); session->envp = envp; session->srvp = srvp; session->connp = connp; session->stmthp = NULL; /* set savepoints up to the current level */ oracleSetSavepoint(session, curlevel); return session; } /* * oracleCloseStatement * Close any open statement associated with the session. */ void oracleCloseStatement(oracleSession *session) { /* free statement handle, if it exists */ if (session->stmthp != NULL) { /* free the statement handle */ freeHandle(session->stmthp, session->connp); session->stmthp = NULL; } } /* * oracleCloseConnections * Close everything in the cache. */ void oracleCloseConnections(void) { while (envlist != NULL) { while (envlist->srvlist != NULL) { while (envlist->srvlist->connlist != NULL) { closeSession(envlist->envhp, envlist->srvlist->srvhp, envlist->srvlist->connlist->userhp, 0); } disconnectServer(envlist->envhp, envlist->srvlist->srvhp); } removeEnvironment(envlist->envhp); } } /* * oracleShutdown * Close all open connections, free handles, terminate Oracle. * This will be called at the end of the PostgreSQL session. */ void oracleShutdown(void) { /* don't report error messages */ silent = 1; oracleCloseConnections(); /* done with Oracle */ if (oci_initialized) (void)OCITerminate(OCI_DEFAULT); } /* * oracleEndTransaction * Commit or rollback the transaction. * The first argument must be a connEntry. * If "noerror" is true, don't throw errors. */ void oracleEndTransaction(void *arg, int is_commit, int noerror) { struct connEntry *connp = NULL; struct srvEntry *srvp = NULL; struct envEntry *envp = NULL; int found = 0; /* do nothing if there is no transaction */ if (((struct connEntry *)arg)->xact_level == 0) return; /* find the cached handles for the argument */ envp = envlist; while (envp) { srvp = envp->srvlist; while (srvp) { connp = srvp->connlist; while (connp) { if (connp == (struct connEntry *)arg) { found = 1; break; } connp = connp->next; } if (found) break; srvp = srvp->next; } if (found) break; envp = envp->next; } if (! found) oracleError(FDW_ERROR, "oracleEndTransaction internal error: handle not found in cache"); /* free handles */ while (connp->handlelist != NULL) freeHandle(connp->handlelist->handlep, connp); /* commit or rollback */ if (is_commit) { oracleDebug2("oracle_fdw: commit remote transaction"); if (checkerr( OCITransCommit(connp->svchp, envp->errhp, OCI_DEFAULT), (dvoid *)envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS && !noerror) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error committing transaction: OCITransCommit failed", oraMessage); } } else { oracleDebug2("oracle_fdw: roll back remote transaction"); if (checkerr( OCITransRollback(connp->svchp, envp->errhp, OCI_DEFAULT), (dvoid *)envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS && !noerror) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error rolling back transaction: OCITransRollback failed", oraMessage); } } connp->xact_level = 0; } /* * oracleEndSubtransaction * Commit or rollback all subtransaction up to savepoint "nest_nevel". * The first argument must be a connEntry. * If "is_commit" is not true, rollback. */ void oracleEndSubtransaction(void *arg, int nest_level, int is_commit) { char query[50], message[60]; struct connEntry *connp = NULL; struct srvEntry *srvp = NULL; struct envEntry *envp = NULL; OCIStmt *stmthp = NULL; int found = 0; /* do nothing if the transaction level is lower than nest_level */ if (((struct connEntry *)arg)->xact_level < nest_level) return; ((struct connEntry *)arg)->xact_level = nest_level - 1; if (is_commit) { /* * There is nothing to do as savepoints don't get released in Oracle: * Setting the same savepoint again just overwrites the previous one. */ return; } /* find the cached handles for the argument */ envp = envlist; while (envp) { srvp = envp->srvlist; while (srvp) { connp = srvp->connlist; while (connp) { if (connp == (struct connEntry *)arg) { found = 1; break; } connp = connp->next; } if (found) break; srvp = srvp->next; } if (found) break; envp = envp->next; } if (! found) oracleError(FDW_ERROR, "oracleRollbackSavepoint internal error: handle not found in cache"); snprintf(message, 59, "oracle_fdw: rollback to savepoint s%d", nest_level); oracleDebug2(message); snprintf(query, 49, "ROLLBACK TO SAVEPOINT s%d", nest_level); /* create statement handle */ allocHandle((void **)&stmthp, OCI_HTYPE_STMT, 0, envp->envhp, connp, FDW_OUT_OF_MEMORY, "error rolling back to savepoint: OCIHandleAlloc failed to allocate statement handle"); /* prepare the query */ if (checkerr( OCIStmtPrepare(stmthp, envp->errhp, (text *)query, (ub4) strlen(query), (ub4) OCI_NTV_SYNTAX, (ub4) OCI_DEFAULT), (dvoid *)envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error rolling back to savepoint: OCIStmtPrepare failed to prepare rollback statement", oraMessage); } /* rollback to savepoint */ if (checkerr( OCIStmtExecute(connp->svchp, stmthp, envp->errhp, (ub4)1, (ub4)0, (CONST OCISnapshot *)NULL, (OCISnapshot *)NULL, OCI_DEFAULT), (dvoid *)envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error rolling back to savepoint: OCIStmtExecute failed to set savepoint", oraMessage); } /* free statement handle */ freeHandle(stmthp, connp); } /* * oracleIsStatementOpen * Return 1 if there is a statement handle, else 0. */ int oracleIsStatementOpen(oracleSession *session) { return (session->stmthp != NULL); } /* * oracleDescribe * Find the remote Oracle table and describe it. * Returns a palloc'ed data structure with the results. */ struct oraTable *oracleDescribe(oracleSession *session, char *schema, char *table, char *pgname, long max_long) { struct oraTable *reply; OCIStmt *stmthp; OCIParam *colp; ub2 oraType, charsize, bin_size; ub1 csfrm; sb2 precision; sb1 scale; char *qtable, *qschema = NULL, *tablename, *query; OraText *ident; ub4 ncols, ident_size; int i, length; /* get a complete quoted table name */ qtable = copyOraText(table, strlen(table), 1); length = strlen(qtable); if (schema != NULL) { qschema = copyOraText(schema, strlen(schema), 1); length += strlen(qschema) + 1; } tablename = oracleAlloc(length + 1); tablename[0] = '\0'; /* empty */ if (schema != NULL) { strcat(tablename, qschema); strcat(tablename, "."); } strcat(tablename, qtable); oracleFree(qtable); if (schema != NULL) oracleFree(qschema); /* construct a "SELECT * FROM ..." query to describe columns */ length += 14; query = oracleAlloc(length + 1); strcpy(query, "SELECT * FROM "); strcat(query, tablename); /* create statement handle */ allocHandle((void **)&stmthp, OCI_HTYPE_STMT, 0, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIHandleAlloc failed to allocate statement handle"); /* prepare the query */ if (checkerr( OCIStmtPrepare(stmthp, session->envp->errhp, (text *)query, (ub4) strlen(query), (ub4) OCI_NTV_SYNTAX, (ub4) OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIStmtPrepare failed to prepare query", oraMessage); } if (checkerr( OCIStmtExecute(session->connp->svchp, stmthp, session->envp->errhp, (ub4)0, (ub4)0, (CONST OCISnapshot *)NULL, (OCISnapshot *)NULL, OCI_DESCRIBE_ONLY), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { if (err_code == 942) oracleError_ssdh(FDW_TABLE_NOT_FOUND, "Oracle table %s for foreign table \"%s\" does not exist or does not allow read access", tablename, pgname, oraMessage, "Oracle table names are case sensitive (normally all uppercase)."); else oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIStmtExecute failed to describe table", oraMessage); } /* allocate an oraTable struct for the results */ reply = oracleAlloc(sizeof(struct oraTable)); reply->name = tablename; reply->pgname = pgname; reply->npgcols = 0; /* get the number of columns */ if (checkerr( OCIAttrGet((dvoid *)stmthp, (ub4)OCI_HTYPE_STMT, (dvoid *)&ncols, (ub4 *)0, (ub4)OCI_ATTR_PARAM_COUNT, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIAttrGet failed to get number of columns", oraMessage); } reply->ncols = ncols; reply->cols = (struct oraColumn **)oracleAlloc(sizeof(struct oraColumn *) * reply->ncols); /* loop through the column list */ for (i=1; i<=reply->ncols; ++i) { /* allocate an oraColumn struct for the column */ reply->cols[i-1] = (struct oraColumn *)oracleAlloc(sizeof(struct oraColumn)); reply->cols[i-1]->pgname = NULL; reply->cols[i-1]->pgattnum = 0; reply->cols[i-1]->pgtype = 0; reply->cols[i-1]->pgtypmod = 0; reply->cols[i-1]->used = 0; reply->cols[i-1]->pkey = 0; reply->cols[i-1]->val = NULL; reply->cols[i-1]->val_len = 0; reply->cols[i-1]->val_null = 1; /* get the parameter descriptor for the column */ if (checkerr( OCIParamGet((void *)stmthp, OCI_HTYPE_STMT, session->envp->errhp, (dvoid **)&colp, i), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIParamGet failed to get column data", oraMessage); } /* get the column name */ if (checkerr( OCIAttrGet((dvoid*)colp, OCI_DTYPE_PARAM, (dvoid*)&ident, &ident_size, (ub4)OCI_ATTR_NAME, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIAttrGet failed to get column name", oraMessage); } reply->cols[i-1]->name = copyOraText((char *)ident, (int)ident_size, 1); /* get the data type */ if (checkerr( OCIAttrGet((dvoid*)colp, OCI_DTYPE_PARAM, (dvoid*)&oraType, (ub4 *)0, (ub4)OCI_ATTR_DATA_TYPE, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIAttrGet failed to get column type", oraMessage); } /* get the character set form */ if (checkerr( OCIAttrGet((dvoid*)colp, OCI_DTYPE_PARAM, (dvoid*)&csfrm, (ub4 *)0, (ub4)OCI_ATTR_CHARSET_FORM, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIAttrGet failed to get character set form", oraMessage); } /* get the number of characters for string fields */ if (checkerr( OCIAttrGet((dvoid*)colp, OCI_DTYPE_PARAM, (dvoid*)&charsize, (ub4 *)0, (ub4)OCI_ATTR_CHAR_SIZE, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIAttrGet failed to get column length", oraMessage); } /* get the binary length for RAW fields */ if (checkerr( OCIAttrGet((dvoid*)colp, OCI_DTYPE_PARAM, (dvoid*)&bin_size, (ub4 *)0, (ub4)OCI_ATTR_DATA_SIZE, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIAttrGet failed to get column size", oraMessage); } /* get the precision */ if (checkerr( OCIAttrGet((dvoid*)colp, OCI_DTYPE_PARAM, (dvoid*)&precision, (ub4 *)0, (ub4)OCI_ATTR_PRECISION, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIAttrGet failed to get column precision", oraMessage); } /* get the scale */ if (checkerr( OCIAttrGet((dvoid*)colp, OCI_DTYPE_PARAM, (dvoid*)&scale, (ub4 *)0, (ub4)OCI_ATTR_SCALE, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error describing remote table: OCIAttrGet failed to get column scale", oraMessage); } reply->cols[i-1]->scale = scale; /* determine oraType and length to allocate */ switch (oraType) { case SQLT_AFC: /* CHAR(n) */ reply->cols[i-1]->oratype = ORA_TYPE_CHAR; reply->cols[i-1]->val_size = charsize * 4 + 1; break; case SQLT_CHR: case SQLT_VCS: /* VARCHAR(n) and VARCHAR2(n) */ reply->cols[i-1]->oratype = ORA_TYPE_VARCHAR2; reply->cols[i-1]->val_size = charsize * 4 + 1; break; case SQLT_BLOB: /* BLOB */ reply->cols[i-1]->oratype = ORA_TYPE_BLOB; /* for LOB columns, "val" will contain a pointer to the locator */ reply->cols[i-1]->val_size = sizeof(OCILobLocator *); break; case SQLT_BFILE: /* BFILE */ reply->cols[i-1]->oratype = ORA_TYPE_BFILE; /* for LOB columns, "val" will contain a pointer to the locator */ reply->cols[i-1]->val_size = sizeof(OCILobLocator *); break; case SQLT_CLOB: case SQLT_CFILE: /* CLOB and CFILE */ if (csfrm == SQLCS_NCHAR) { /* * We don't support NCLOB because Oracle cannot * transform it to the client character set automatically. */ reply->cols[i-1]->oratype = ORA_TYPE_OTHER; reply->cols[i-1]->val_size = 0; } else { reply->cols[i-1]->oratype = ORA_TYPE_CLOB; /* for LOB columns, "val" will contain a pointer to the locator */ reply->cols[i-1]->val_size = sizeof(OCILobLocator *); } break; case SQLT_NUM: /* NUMBER */ reply->cols[i-1]->oratype = ORA_TYPE_NUMBER; if (precision == 0) /* this should be big enough for unrestricted NUMBERs */ reply->cols[i-1]->val_size = 140; else reply->cols[i-1]->val_size = ((-scale) > precision ? (-scale) : precision) + 5; break; case SQLT_FLT: /* FLOAT */ reply->cols[i-1]->oratype = ORA_TYPE_NUMBER; reply->cols[i-1]->val_size = 140; break; case SQLT_IBFLOAT: /* BINARY_FLOAT */ reply->cols[i-1]->oratype = ORA_TYPE_BINARYFLOAT; reply->cols[i-1]->val_size = 42; break; case SQLT_IBDOUBLE: /* BINARY_DOUBLE */ reply->cols[i-1]->oratype = ORA_TYPE_BINARYDOUBLE; reply->cols[i-1]->val_size = 310; break; case SQLT_DAT: /* DATE */ reply->cols[i-1]->oratype = ORA_TYPE_DATE; reply->cols[i-1]->val_size = 20; break; case SQLT_TIMESTAMP: /* TIMESTAMP */ reply->cols[i-1]->oratype = ORA_TYPE_TIMESTAMP; reply->cols[i-1]->val_size = 30; break; case SQLT_TIMESTAMP_TZ: case SQLT_TIMESTAMP_LTZ: /* TIMESTAMP WITH TIMEZONE and TIMESTAMP WITH LOCAL TIMEZONE */ reply->cols[i-1]->oratype = ORA_TYPE_TIMESTAMPTZ; reply->cols[i-1]->val_size = 36; break; case SQLT_INTERVAL_YM: /* INTERVAL YEAR TO MONTH */ reply->cols[i-1]->oratype = ORA_TYPE_INTERVALY2M; reply->cols[i-1]->val_size = precision + 5; break; case SQLT_INTERVAL_DS: /* INTERVAL DAY TO SECOND */ reply->cols[i-1]->oratype = ORA_TYPE_INTERVALD2S; reply->cols[i-1]->val_size = precision + scale + 12; break; case SQLT_LBI: /* LONG RAW */ reply->cols[i-1]->oratype = ORA_TYPE_LONGRAW; reply->cols[i-1]->val_size = max_long + 4; break; case SQLT_LNG: /* LONG */ reply->cols[i-1]->oratype = ORA_TYPE_LONG; reply->cols[i-1]->val_size = max_long + 4; break; case SQLT_BIN: /* RAW(n) */ reply->cols[i-1]->oratype = ORA_TYPE_RAW; /* use binary size for RAWs */ reply->cols[i-1]->val_size = 2 * bin_size + 1; break; default: reply->cols[i-1]->oratype = ORA_TYPE_OTHER; reply->cols[i-1]->val_size = 0; } } /* free statement handle, this takes care of the parameter handles */ freeHandle(stmthp, session->connp); return reply; } /* * oracleEstimate * Sets the estimates for startup_cost, total_cost, rows and width * as far as Oracle can tell. * startup_cost and total_cost will not be very good estimates. */ void oracleEstimate(oracleSession *session, const char *query, double seq_page_cost, int block_size, double *startup_cost, double *total_cost, double *rows, int *width) { double time, bytes; dvoid *res[3]; sb4 res_size[3]; ub2 res_type[3], res_len[3]; sb2 res_ind[3]; const char * const desc_query = "SELECT time, cardinality, bytes FROM v$sql_plan WHERE sql_id = :sql_id AND child_number = :child_number AND id = 1"; res[0] = (dvoid *)&time; res_size[0] = sizeof(double); res_type[0] = SQLT_BDOUBLE; res[1] = (dvoid *)rows; res_size[1] = sizeof(double); res_type[1] = SQLT_BDOUBLE; res[2] = (dvoid *)&bytes; res_size[2] = sizeof(double); res_type[2] = SQLT_BDOUBLE; oracleQueryPlan(session, query, desc_query, 3, res, res_size, res_type, res_len, res_ind); /* close the statement */ oracleCloseStatement(session); /* width now contains the total bytes estimated - divide by number of rows */ *width = (int)(bytes / *rows); /* * Guess startup_cost and total_cost from Oracle's "time". * This is really shady as Oracle's time has a granularity of 1. * Assume that we can sequentially read 81920000 bytes per second. */ *total_cost = time * 81920000.0 / block_size * seq_page_cost; *startup_cost = *total_cost; } #define EXPLAIN_LINE_SIZE 1000 /* * oracleExplain * Returns the EXPLAIN PLAN for the query. * "plan" will contain "nrows" palloc'ed strings. */ void oracleExplain(oracleSession *session, const char *query, int *nrows, char ***plan) { char res[EXPLAIN_LINE_SIZE], *r=res; sb4 res_size = EXPLAIN_LINE_SIZE; ub2 res_len, res_type = SQLT_STR; sb2 res_ind; sword result; const char * const desc_query = "SELECT rtrim(lpad(' ',2*level-2)||operation||' '||options||' '||object_name||' '" "||CASE WHEN access_predicates IS NULL THEN NULL ELSE '(condition '||access_predicates||')' END" "||' '||CASE WHEN filter_predicates IS NULL THEN NULL ELSE '(filter '||filter_predicates||')' END)" " FROM v$sql_plan" " CONNECT BY prior id = parent_id AND prior sql_id = sql_id AND prior child_number = child_number" " START WITH id=0 AND sql_id=:sql_id and child_number=:child_number" " ORDER BY id"; /* execute the query and get the first result row */ oracleQueryPlan(session, query, desc_query, 1, (dvoid **)&r, &res_size, &res_type, &res_len, &res_ind); *nrows = 0; do { /* increase result "array" */ if (++(*nrows) == 1) *plan = (char **)oracleAlloc(sizeof(char *)); else *plan = (char **)oracleRealloc(*plan, sizeof(char *) * (*nrows)); /* add entry */ (*plan)[(*nrows)-1] = oracleAlloc(strlen(res) + 1); strcpy((*plan)[(*nrows)-1], res); /* fetch next row */ result = checkerr( OCIStmtFetch2(session->stmthp, session->envp->errhp, 1, OCI_FETCH_NEXT, 0, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR); if (result != OCI_SUCCESS && result != OCI_NO_DATA) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error fetching result: OCIStmtFetch2 failed to fetch next result row", oraMessage); } } while (result != OCI_NO_DATA); /* close the statement */ oracleCloseStatement(session); } /* * oracleSetSavepoint * Set savepoints up to level "nest_level". */ void oracleSetSavepoint(oracleSession *session, int nest_level) { while (session->connp->xact_level < nest_level) { char query[40], message[50]; snprintf(message, 49, "oracle_fdw: set savepoint s%d", session->connp->xact_level + 1); oracleDebug2(message); snprintf(query, 39, "SAVEPOINT s%d", session->connp->xact_level + 1); /* create statement handle */ allocHandle((void **)&(session->stmthp), OCI_HTYPE_STMT, 0, session->envp->envhp, session->connp, FDW_OUT_OF_MEMORY, "error setting savepoint: OCIHandleAlloc failed to allocate statement handle"); /* prepare the query */ if (checkerr( OCIStmtPrepare(session->stmthp, session->envp->errhp, (text *)query, (ub4) strlen(query), (ub4) OCI_NTV_SYNTAX, (ub4) OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error setting savepoint: OCIStmtPrepare failed to prepare savepoint statement", oraMessage); } /* set savepoint */ if (checkerr( OCIStmtExecute(session->connp->svchp, session->stmthp, session->envp->errhp, (ub4)1, (ub4)0, (CONST OCISnapshot *)NULL, (OCISnapshot *)NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error setting savepoint: OCIStmtExecute failed to set savepoint", oraMessage); } /* free statement handle */ freeHandle(session->stmthp, session->connp); session->stmthp = NULL; ++session->connp->xact_level; } } /* * setOracleEnvironment * Set environment variables do that Oracle works as we want. * * NLS_LANG sets the language and client encoding * NLS_NCHAR is unset so that N* data types are converted to the * character set specified in NLS_LANG. * * The following variables are set to values that make Oracle convert * numeric and date/time values to strings PostgreSQL can parse: * NLS_DATE_FORMAT * NLS_TIMESTAMP_FORMAT * NLS_TIMESTAMP_TZ_FORMAT * NLS_NUMERIC_CHARACTERS * NLS_CALENDAR */ void setOracleEnvironment(char *nls_lang) { if (putenv(nls_lang) != 0) { free(nls_lang); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle", "Environment variable NLS_LANG cannot be set."); } /* other environment variables that control Oracle formats */ if (putenv("NLS_DATE_FORMAT=YYYY-MM-DD HH24:MI:SS") != 0) { free(nls_lang); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle", "Environment variable NLS_DATE_FORMAT cannot be set."); } if (putenv("NLS_TIMESTAMP_FORMAT=YYYY-MM-DD HH24:MI:SS.FF") != 0) { free(nls_lang); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle", "Environment variable NLS_TIMESTAMP_FORMAT cannot be set."); } if (putenv("NLS_TIMESTAMP_TZ_FORMAT=YYYY-MM-DD HH24:MI:SS.FFTZH:TZM") != 0) { free(nls_lang); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle", "Environment variable NLS_TIMESTAMP_TZ_FORMAT cannot be set."); } if (putenv("NLS_NUMERIC_CHARACTERS=.,") != 0) { free(nls_lang); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle", "Environment variable NLS_NUMERIC_CHARACTERS cannot be set."); } if (putenv("NLS_CALENDAR=") != 0) { free(nls_lang); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle", "Environment variable NLS_CALENDAR cannot be set."); } if (putenv("NLS_NCHAR=") != 0) { free(nls_lang); oracleError_d(FDW_UNABLE_TO_ESTABLISH_CONNECTION, "error connecting to Oracle", "Environment variable NLS_NCHAR cannot be set."); } } /* * oracleQueryPlan * This is a utility function called by oracleEstimate and oracleExplain. * First the "query" is described explicitly. * This will parse the query and add it to the library cache. * Then a "desc_query" is executed against V$SQL_PLAN to extract * planner info. The Oracle statement is stored in "session" for further fetches. * * "nres" is the number of elements in the select list of "query", * "res_size" and "res_type" are arrays of length "nres" containing * result buffer size and Oracle type of the select list entries, * while the arrays "res", "res_len" and "res_ind" contain output parameters * for the result, the actual lenth of the result and NULL indicators. */ void oracleQueryPlan(oracleSession *session, const char *query, const char *desc_query, int nres, dvoid **res, sb4 *res_size, ub2 *res_type, ub2 *res_len, sb2 *res_ind) { int child_nr, i; const char * const sql_id_query = "SELECT sql_id, child_number FROM (SELECT sql_id, child_number FROM v$sql WHERE sql_text LIKE :sql ORDER BY last_active_time DESC) WHERE rownum=1"; char sql_id[20], query_head[50], *p; OCIDefine *defnhp; OCIBind *bndhp; sb2 ind1, ind2, ind3; ub2 len1, len2; ub4 prefetch_rows = PREFETCH_ROWS, prefetch_memory = PREFETCH_MEMORY; /* make sure there is no statement handle stored in session */ if (session->stmthp != NULL) { oracleError(FDW_ERROR, "oracleQueryPlan internal error: statement handle is not NULL"); } /* create statement handle */ allocHandle((void **)&(session->stmthp), OCI_HTYPE_STMT, 0, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIHandleAlloc failed to allocate statement handle"); /* set prefetch options */ if (checkerr( OCIAttrSet((dvoid *)session->stmthp, OCI_HTYPE_STMT, (dvoid *)&prefetch_rows, 0, OCI_ATTR_PREFETCH_ROWS, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIAttrSet failed to set number of prefetched rows in statement handle", oraMessage); } if (checkerr( OCIAttrSet((dvoid *)session->stmthp, OCI_HTYPE_STMT, (dvoid *)&prefetch_memory, 0, OCI_ATTR_PREFETCH_MEMORY, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIAttrSet failed to set prefetch memory in statement handle", oraMessage); } /* prepare the query */ if (checkerr( OCIStmtPrepare(session->stmthp, session->envp->errhp, (text *)query, (ub4) strlen(query), (ub4) OCI_NTV_SYNTAX, (ub4) OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIStmtPrepare failed to prepare remote query", oraMessage); } /* parse and describe the query, store it in the library cache */ if (checkerr( OCIStmtExecute(session->connp->svchp, session->stmthp, session->envp->errhp, (ub4)0, (ub4)0, (CONST OCISnapshot *)NULL, (OCISnapshot *)NULL, OCI_DESCRIBE_ONLY), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query query: OCIStmtExecute failed to describe remote query", oraMessage); } freeHandle(session->stmthp, session->connp); session->stmthp = NULL; /* * Get the SQL_ID and CHILD_NUMBER from V$SQL. */ /* get the first part of the SQL query with '%' appended */ if ((p = strchr(query + 7, ' ')) == NULL) { oracleError(FDW_ERROR, "oracleQueryPlan internal error: no space found in query"); } strncpy(query_head, query, p-query); query_head[p-query] = '%'; query_head[p-query+1] = '\0'; /* create statement handle */ allocHandle((void **)&(session->stmthp), OCI_HTYPE_STMT, 0, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIHandleAlloc failed to allocate statement handle"); /* prepare */ if (checkerr( OCIStmtPrepare(session->stmthp, session->envp->errhp, (text *)sql_id_query, (ub4) strlen(sql_id_query), (ub4) OCI_NTV_SYNTAX, (ub4) OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIStmtPrepare failed to prepare remote query for sql_id", oraMessage); } /* bind */ bndhp = NULL; ind3 = 0; if (checkerr( OCIBindByName(session->stmthp, &bndhp, session->envp->errhp, (text *)":sql", (sb4)4, (dvoid *)query_head, (sb4)(strlen(query_head) + 1), SQLT_STR, (dvoid *)&ind3, NULL, NULL, (ub4)0, NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIBindByName failed to bind parameter", oraMessage); } /* define result values */ sql_id[19] = '\0'; defnhp = NULL; if (checkerr( OCIDefineByPos(session->stmthp, &defnhp, session->envp->errhp, (ub4)1, (dvoid *)sql_id, (sb4)19, SQLT_STR, (dvoid *)&ind1, (ub2 *)&len1, NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIDefineByPos failed to define result value", oraMessage); } defnhp = NULL; if (checkerr( OCIDefineByPos(session->stmthp, &defnhp, session->envp->errhp, (ub4)2, (dvoid *)&child_nr, (sb4)sizeof(int), SQLT_INT, (dvoid *)&ind2, (ub2 *)&len2, NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIDefineByPos failed to define result value", oraMessage); } /* execute */ if (checkerr( OCIStmtExecute(session->connp->svchp, session->stmthp, session->envp->errhp, (ub4)1, (ub4)0, (CONST OCISnapshot *)NULL, (OCISnapshot *)NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { /* create a better message if we lack permissions on V$SQL */ if (err_code == 942) oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "no SELECT privilege on V$SQL in the remote database", oraMessage); else oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIStmtExecute failed to execute remote query for sql_id", oraMessage); } freeHandle(session->stmthp, session->connp); session->stmthp = NULL; /* * Run the final desc_query. */ /* create statement handle */ allocHandle((void **)&(session->stmthp), OCI_HTYPE_STMT, 0, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIHandleAlloc failed to allocate statement handle"); /* prepare */ if (checkerr( OCIStmtPrepare(session->stmthp, session->envp->errhp, (text *)desc_query, (ub4) strlen(desc_query), (ub4) OCI_NTV_SYNTAX, (ub4) OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIStmtPrepare failed to prepare remote plan query", oraMessage); } /* bind */ bndhp = NULL; ind1 = 0; if (checkerr( OCIBindByName(session->stmthp, &bndhp, session->envp->errhp, (text *)":sql_id", (sb4)7, (dvoid *)sql_id, (sb4)strlen(sql_id) + 1, SQLT_STR, (dvoid *)&ind1, NULL, NULL, (ub4)0, NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIBindByName failed to bind parameter", oraMessage); } bndhp = NULL; ind2 = 0; if (checkerr( OCIBindByName(session->stmthp, &bndhp, session->envp->errhp, (text *)":child_number", (sb4)13, (dvoid *)&child_nr, (sb4)sizeof(int), SQLT_INT, (dvoid *)&ind2, NULL, NULL, (ub4)0, NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIBindByName failed to bind parameter", oraMessage); } /* define result values */ for (i=0; istmthp, &defnhp, session->envp->errhp, (ub4)(i + 1), (dvoid *)res[i], res_size[i], res_type[i], (dvoid *)&res_ind[i], &res_len[i], NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIDefineByPos failed to define result value", oraMessage); } } /* execute */ if (checkerr( OCIStmtExecute(session->connp->svchp, session->stmthp, session->envp->errhp, (ub4)1, (ub4)0, (CONST OCISnapshot *)NULL, (OCISnapshot *)NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { /* create a better message if we lack permissions on V$SQL_PLAN */ if (err_code == 942) oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "no SELECT privilege on V$SQL_PLAN in the remote database", oraMessage); else oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error describing query: OCIStmtExecute failed to execute remote plan query", oraMessage); } } /* * oraclePrepareQuery * Prepares an SQL statement for execution. * This function should handle everything that has to be done only once * even if the statement is executed multiple times, that is: * - For SELECT statements, defines the result values to be stored in oraTable. * - For DML statements, allocates LOB locators for the RETURNING clause in oraTable. * - Set the prefetch options. */ void oraclePrepareQuery(oracleSession *session, const char *query, const struct oraTable *oraTable) { int i, col_pos, is_select; OCIDefine *defnhp; static char dummy[4]; static sb4 dummy_size = 4; static sb2 dummy_null; ub4 prefetch_rows = PREFETCH_ROWS, prefetch_memory = PREFETCH_MEMORY; /* figure out if the query is FOR UPDATE */ is_select = (strncmp(query, "SELECT", 6) == 0); /* make sure there is no statement handle stored in "session" */ if (session->stmthp != NULL) { oracleError(FDW_ERROR, "oraclePrepareQuery internal error: statement handle is not NULL"); } /* create statement handle */ allocHandle((void **)&(session->stmthp), OCI_HTYPE_STMT, 0, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIHandleAlloc failed to allocate statement handle"); /* prepare the statement */ if (checkerr( OCIStmtPrepare(session->stmthp, session->envp->errhp, (text *)query, (ub4) strlen(query), (ub4) OCI_NTV_SYNTAX, (ub4) OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIStmtPrepare failed to prepare remote query", oraMessage); } /* loop through table columns */ col_pos = 0; for (i=0; incols; ++i) { if (oraTable->cols[i]->used) { /* * Unfortunately Oracle handles DML statements with a RETURNING clause * quite different from SELECT statements. In the latter, the result * columns are "defined", i.e. bound to some storage space. * This definition is only necessary once, even if the query is executed * multiple times, so we do this here. * RETURNING clause are handled in oracleExecuteQuery, here we only * allocate locators for LOB columns in RETURNING clauses. */ if (is_select) { ub2 type; /* figure out in which format we want the results */ type = getOraType(oraTable->cols[i]->oratype); if (oraTable->cols[i]->pgtype == UUIDOID) type = SQLT_STR; /* check if it is a LOB column */ if (type == SQLT_BLOB || type == SQLT_BFILE || type == SQLT_CLOB) { /* allocate a LOB locator, store a pointer to it in "val" */ allocHandle((void **)oraTable->cols[i]->val, OCI_DTYPE_LOB, 1, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIDescriptorAlloc failed to allocate LOB descriptor"); } /* define result value */ defnhp = NULL; if (checkerr( OCIDefineByPos(session->stmthp, &defnhp, session->envp->errhp, (ub4)++col_pos, (dvoid *)oraTable->cols[i]->val, (sb4)oraTable->cols[i]->val_size, type, (dvoid *)&oraTable->cols[i]->val_null, (ub2 *)&oraTable->cols[i]->val_len, NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIDefineByPos failed to define result value", oraMessage); } } else { ub2 type; /* for other statements, allocate LOB locators for RETURNING parameters */ type = getOraType(oraTable->cols[i]->oratype); if (type == ORA_TYPE_BLOB || type == SQLT_BFILE || type == ORA_TYPE_CLOB) { /* allocate a LOB locator, store a pointer to it in "val" */ allocHandle((void **)oraTable->cols[i]->val, OCI_DTYPE_LOB, 1, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIDescriptorAlloc failed to allocate LOB descriptor"); } } } } if (is_select && col_pos == 0) { /* * No columns selected (i.e., SELECT '1' FROM). * Define dummy result columnn. */ defnhp = NULL; if (checkerr( OCIDefineByPos(session->stmthp, &defnhp, session->envp->errhp, (ub4)1, (dvoid *)dummy, dummy_size, SQLT_STR, (dvoid *)&dummy_null, NULL, NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIDefineByPos failed to define result value", oraMessage); } } /* set prefetch options */ if (checkerr( OCIAttrSet((dvoid *)session->stmthp, OCI_HTYPE_STMT, (dvoid *)&prefetch_rows, 0, OCI_ATTR_PREFETCH_ROWS, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIAttrSet failed to set number of prefetched rows in statement handle", oraMessage); } if (checkerr( OCIAttrSet((dvoid *)session->stmthp, OCI_HTYPE_STMT, (dvoid *)&prefetch_memory, 0, OCI_ATTR_PREFETCH_MEMORY, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIAttrSet failed to set prefetch memory in statement handle", oraMessage); } } /* * oracleExecuteQuery * Execute a prepared statement and fetches the first result row. * The parameters ("bind variables") are filled from paramList. * Returns the count of processed rows. * This can be called several times for a prepared SQL statement. */ int oracleExecuteQuery(oracleSession *session, const struct oraTable *oraTable, struct paramDesc *paramList) { sb2 *indicators; struct paramDesc *param; sword result; ub4 rowcount, lob_empty = 0; OCILobLocator **lob_locators; /* for input parameters that are LOBs */ int i, param_count = 0; for (param=paramList; param; param=param->next) ++param_count; /* allocate a temporary array to hold the LOB locators */ lob_locators = oracleAlloc(param_count * sizeof(OCILobLocator *)); memset(lob_locators, 0, param_count * sizeof(OCILobLocator *)); /* allocate a temporary array of indicators */ indicators = oracleAlloc(param_count * sizeof(sb2 *)); /* bind the parameters */ param_count = -1; for (param=paramList; param; param=param->next) { dvoid *value = NULL; /* will contain the value as bound */ sb4 value_len = 0; /* length of "value" */ ub2 value_type = SQLT_STR; /* SQL_STR works for NULLs of all types */ ub4 oci_mode = OCI_DEFAULT; /* changed only for output parameters */ OCIDateTime **timest; OCINumber *number; char *num_format, *pos; ++param_count; indicators[param_count] = (sb2)((param->value == NULL) ? -1 : 0); if (param->value != NULL) { /* * For non-NULL LOB input parameters we insert empty LOBs in a first step. * Such columns have also been added to the RETURNING clause so that the * actual LOB contents can be written immediately after the DML statement. */ if (param->bindType == BIND_BLOB || param->bindType == BIND_CLOB) { /* allocate LOB locator for LOB columns */ allocHandle((void **)&lob_locators[param_count], OCI_DTYPE_LOB, 1, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIDescriptorAlloc failed to allocate LOB descriptor"); /* empty the LOB */ if (checkerr( OCIAttrSet(lob_locators[param_count], OCI_DTYPE_LOB, &lob_empty, 0, OCI_ATTR_LOBEMPTY, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIAttrSet failed to set LOB empty", oraMessage); } } switch (param->bindType) { case BIND_NUMBER: /* allocate a new NUMBER */ number = oracleAlloc(sizeof(OCINumber)); /* * Construct number format. */ value_len = strlen(param->value); num_format = oracleAlloc(value_len + 1); /* fill everything with '9' */ memset(num_format, '9', value_len); num_format[value_len] = '\0'; /* write 'D' in the decimal point position */ if ((pos = strchr(param->value, '.')) != NULL) num_format[pos - param->value] = 'D'; /* replace the scientific notation part with 'E' */ if ((pos = strchr(param->value, 'e')) != NULL) memset(num_format + (pos - param->value), 'E', value_len - (pos - param->value)); /* convert parameter string to NUMBER */ if (checkerr( OCINumberFromText(session->envp->errhp, (const OraText *)param->value, (ub4)value_len, (const OraText *)num_format, (ub4)value_len, (const OraText *)NULL, (ub4)0, number), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCINumberFromText failed to convert parameter", oraMessage); } oracleFree(num_format); value = (dvoid *)number; value_len = sizeof(OCINumber); value_type = SQLT_VNU; break; case BIND_TIMESTAMP: /* timest must exist until the OCIStmtExecute call */ timest = (OCIDateTime **)oracleAlloc(sizeof(OCIDateTime *)); /* allocate timestamp descriptor */ allocHandle((void **)timest, OCI_DTYPE_TIMESTAMP_TZ, 1, session->envp->envhp, session->connp, FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIDescriptorAlloc failed to allocate Timestamp descriptor"); /* convert parameter string to TIMESTAMP WITH TIME ZONE */ if (checkerr( OCIDateTimeFromText((dvoid *)session->connp->userhp, session->envp->errhp, (const OraText *)param->value, strlen(param->value), (const OraText *)NULL, (ub1)0, (const OraText *)NULL, (size_t)0, *timest), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIDateTimeFromText failed to convert parameter", oraMessage); } /* store in param->value to be able to free it later */ param->value = (char *)*timest; value = (dvoid *)timest; value_len = sizeof(OCIDateTime **); value_type = SQLT_TIMESTAMP_TZ; break; case BIND_STRING: value = param->value; value_len = strlen(param->value)+1; value_type = SQLT_STR; break; case BIND_LONGRAW: value = param->value; value_len = *((sb4 *)param->value) + 4; value_type = SQLT_LVB; break; case BIND_LONG: value = param->value; value_len = *((sb4 *)param->value) + 4; value_type = SQLT_LVC; break; case BIND_BLOB: value = &lob_locators[param_count]; value_len = sizeof(OCILobLocator *); value_type = SQLT_BLOB; break; case BIND_CLOB: value = &lob_locators[param_count]; value_len = sizeof(OCILobLocator *); value_type = SQLT_CLOB; break; case BIND_OUTPUT: value = NULL; value_len = oraTable->cols[param->colnum]->val_size; value_type = getOraType(oraTable->cols[param->colnum]->oratype); if (oraTable->cols[param->colnum]->pgtype == UUIDOID) { /* the type input function will interpret the string value correctly */ value_type = SQLT_STR; } oci_mode = OCI_DATA_AT_EXEC; break; } } /* bind the value to the parameter */ if (checkerr( OCIBindByName(session->stmthp, (OCIBind **)¶m->bindh, session->envp->errhp, (text *)param->name, (sb4)strlen(param->name), value, value_len, value_type, (dvoid *)&indicators[param_count], NULL, NULL, (ub4)0, NULL, oci_mode), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIBindByName failed to bind parameter", oraMessage); } /* for output parameters, define callbacks that provide storage space */ if (param->bindType == BIND_OUTPUT) { if (checkerr( OCIBindDynamic((OCIBind *)param->bindh, session->envp->errhp, oraTable->cols[param->colnum], &bind_in_callback, oraTable->cols[param->colnum], &bind_out_callback), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIBindDynamic failed to bind callback for parameter", oraMessage); } } } /* execute the query and get the first result row */ result = checkerr( OCIStmtExecute(session->connp->svchp, session->stmthp, session->envp->errhp, (ub4)1, (ub4)0, (CONST OCISnapshot *)NULL, (OCISnapshot *)NULL, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR); if (result != OCI_SUCCESS && result != OCI_NO_DATA) { /* use the correct SQLSTATE for serialization failures */ oracleError_d(err_code == 8177 ? FDW_SERIALIZATION_FAILURE : FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIStmtExecute failed to execute remote query", oraMessage); } /* free all LOB locators */ for (i=0; i<=param_count; ++i) if (lob_locators[i] != NULL) freeHandle((dvoid *)lob_locators[i], session->connp); oracleFree(lob_locators); /* free indicators */ oracleFree(indicators); if (result == OCI_NO_DATA) return 0; /* get the number of processed rows (important for DML) */ if (checkerr( OCIAttrGet((dvoid *)session->stmthp, (ub4)OCI_HTYPE_STMT, (dvoid *)&rowcount, (ub4 *)0, (ub4)OCI_ATTR_ROW_COUNT, session->envp->errhp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error executing query: OCIAttrGet failed to get number of affected rows", oraMessage); } /* post processing of output parameters */ for (param=paramList; param; param=param->next) if (param->bindType == BIND_OUTPUT) { /* * Store the received length in the table column. * Should not lose any data in all possible cases * since LONG and LONG RAW don't work with RETURNING anyway. */ oraTable->cols[param->colnum]->val_len = (unsigned short)oraTable->cols[param->colnum]->val_len4; } if (rowcount > 0) { /* * For LOB columns in DML statements we have added an empty LOB above. * Now we have to write the actual contents. */ for (param=paramList; param; param=param->next) if ((param->bindType == BIND_BLOB || param->bindType == BIND_CLOB) && oraTable->cols[param->colnum]->val_null != -1) oracleWriteLob(session, *((OCILobLocator **)oraTable->cols[param->colnum]->val), param->value); } return rowcount; } /* * oracleFetchNext * Fetch the next result row, return 1 if there is one, else 0. */ int oracleFetchNext(oracleSession *session) { sword result; /* make sure there is a statement handle stored in "session" */ if (session->stmthp == NULL) { oracleError(FDW_ERROR, "oracleFetchNext internal error: statement handle is NULL"); } /* fetch the next result row */ result = checkerr( OCIStmtFetch2(session->stmthp, session->envp->errhp, 1, OCI_FETCH_NEXT, 0, OCI_DEFAULT), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR); if (result != OCI_SUCCESS && result != OCI_NO_DATA) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error fetching result: OCIStmtFetch2 failed to fetch next result row", oraMessage); } return (result == OCI_SUCCESS); } /* * oracleGetLob * Get the LOB contents and store them in *value and *value_len. * If "trunc" is nonzero, it contains the number of bytes or characters to get. */ void oracleGetLob(oracleSession *session, void *locptr, oraType type, char **value, long *value_len, unsigned long trunc) { OCILobLocator *locp = *(OCILobLocator **)locptr; oraub8 amount_byte, amount_char; sword result = OCI_SUCCESS; /* initialize result buffer length */ *value_len = 0; /* open the LOB */ if (type == ORA_TYPE_BFILE) { /* function to open BFILES - required */ if (checkerr( OCILobFileOpen(session->connp->svchp, session->envp->errhp, locp, OCI_FILE_READONLY), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error fetching result: OCILobFileOpen failed to open BFILE", oraMessage); } } else { /* function to open internal LOBs - for better performance */ if (checkerr( OCILobOpen(session->connp->svchp, session->envp->errhp, locp, OCI_FILE_READONLY), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error fetching result: OCILobOpen failed to open LOB", oraMessage); } } /* read the LOB in chunks */ do { /* extend result buffer */ if (*value_len == 0) *value = oracleAlloc(LOB_CHUNK_SIZE + 1); else *value = oracleRealloc(*value, *value_len + LOB_CHUNK_SIZE + 1); /* * The first time round, "amount_* = 0" tells OCILobRead to read the whole LOB. * On subsequent reads, the amount_* parameters are ignored. * After the call, "amount_byte" contains the number of bytes read. */ amount_byte = (oraub8)trunc; /* ignored for CLOBs */ amount_char = amount_byte; /* ignored for binary LOBs */ result = checkerr( OCILobRead2(session->connp->svchp, session->envp->errhp, locp, &amount_byte, &amount_char, (oraub8)1, (dvoid *)(*value + *value_len), (oraub8)LOB_CHUNK_SIZE, (result == OCI_NEED_DATA) ? OCI_NEXT_PIECE : OCI_FIRST_PIECE, NULL, NULL, (ub2)0, (ub1)0), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR); if (result == OCI_ERROR) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error fetching result: OCILobRead failed to read LOB chunk", oraMessage); } /* update LOB length */ *value_len += (long)amount_byte; } while (result == OCI_NEED_DATA); /* string end for CLOBs */ (*value)[*value_len] = '\0'; /* close the LOB */ if (type == ORA_TYPE_BFILE) { if (checkerr( OCILobFileClose(session->connp->svchp, session->envp->errhp, locp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error fetching result: OCILobFileClose failed to close BFILE", oraMessage); } } else { if (checkerr( OCILobClose(session->connp->svchp, session->envp->errhp, locp), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error fetching result: OCILobClose failed to close LOB", oraMessage); } } } /* * checkerr * Call OCIErrorGet to get error message and error code. */ sword checkerr(sword status, dvoid *handle, ub4 handleType) { int length; oraMessage[0] = '\0'; if (status == OCI_SUCCESS_WITH_INFO || status == OCI_ERROR) { OCIErrorGet(handle, (ub4)1, NULL, &err_code, (text *)oraMessage, (ub4)ERRBUFSIZE, handleType); length = strlen(oraMessage); if (length > 0 && oraMessage[length-1] == '\n') oraMessage[length-1] = '\0'; } if (status == OCI_SUCCESS_WITH_INFO) status = OCI_SUCCESS; if (status == OCI_NO_DATA) { strcpy(oraMessage, "ORA-00100: no data found"); err_code = (sb4)100; } return status; } /* * copyOraText * Returns a palloc'ed string containing a (possibly quoted) copy of "string". */ char *copyOraText(const char *string, int size, int quote) { int resultsize = (quote ? size + 2 : size); register int i, j=-1; char *result; if (quote) { for (i=0; inext) { if (envp->envhp == envhp) break; } if (envp == NULL) { if (silent) return; else oracleError(FDW_ERROR, "closeSession internal error: environment handle not found in cache"); } /* search server handle in cache */ for (srvp = envp->srvlist; srvp != NULL; srvp = srvp->next) { if (srvp->srvhp == srvhp) break; } if (srvp == NULL) { if (silent) return; else oracleError(FDW_ERROR, "closeSession internal error: server handle not found in cache"); } /* search connection in cache */ for (connp = srvp->connlist; connp != NULL; connp = connp->next) { if (connp->userhp == userhp) break; prevconnp = connp; } if (connp == NULL) { if (silent) return; else oracleError(FDW_ERROR, "closeSession internal error: user handle not found in cache"); } /* terminate the session */ if (checkerr( OCISessionEnd(connp->svchp, envp->errhp, connp->userhp, OCI_DEFAULT), (dvoid *)envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS && !silent) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error closing session: OCISessionEnd failed to terminate session", oraMessage); } /* free the session handle */ (void)OCIHandleFree((dvoid *)connp->userhp, OCI_HTYPE_SESSION); /* get the transaction handle */ if (checkerr( OCIAttrGet((dvoid *)connp->svchp, (ub4)OCI_HTYPE_SVCCTX, (dvoid *)&txnhp, (ub4 *)0, (ub4)OCI_ATTR_TRANS, envp->errhp), (dvoid *)envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS && !silent) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error closing session: OCIAttrGet failed to get transaction handle", oraMessage); } /* free the service handle */ (void)OCIHandleFree((dvoid *)connp->svchp, OCI_HTYPE_SVCCTX); /* free the transaction handle */ (void)OCIHandleFree((dvoid *)txnhp, OCI_HTYPE_TRANS); /* remove the session handle from the cache */ if (prevconnp == NULL) srvp->connlist = connp->next; else prevconnp->next = connp->next; /* close the server session if desired and this is the last session */ if (disconnect && srvp->connlist == NULL) disconnectServer(envhp, srvhp); /* unregister callback for rolled back transactions */ oracleUnregisterCallback(connp); /* free the memory */ free(connp->user); free(connp); } /* * disconnectServer * Disconnect from the server and remove it from the cache. */ void disconnectServer(OCIEnv *envhp, OCIServer *srvhp) { struct envEntry *envp; struct srvEntry *srvp, *prevsrvp = NULL; /* search environment handle in cache */ for (envp = envlist; envp != NULL; envp = envp->next) { if (envp->envhp == envhp) break; } if (envp == NULL) { if (silent) return; else oracleError(FDW_ERROR, "disconnectServer internal error: environment handle not found in cache"); } /* search server handle in cache */ for (srvp = envp->srvlist; srvp != NULL; srvp = srvp->next) { if (srvp->srvhp == srvhp) break; prevsrvp = srvp; } if (srvp == NULL) { if (silent) return; else oracleError(FDW_ERROR, "disconnectServer internal error: server handle not found in cache"); } /* disconnect server */ if (!silent && checkerr( OCIServerDetach(srvp->srvhp, envp->errhp, OCI_DEFAULT), (dvoid *)envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_REPLY, "error closing session: OCIServerDetach failed to detach from server", oraMessage); } /* free the server handle */ (void)OCIHandleFree((dvoid *)srvp->srvhp, OCI_HTYPE_SERVER); /* remove server entry from the linked list */ if (prevsrvp == NULL) envp->srvlist = srvp->next; else prevsrvp->next = srvp->next; /* free the memory */ free(srvp->connectstring); free(srvp); } /* * removeEnvironment * Deallocate environment, error and service handle and remove cache entry. */ void removeEnvironment(OCIEnv *envhp) { struct envEntry *envp, *prevenvp = NULL; /* search environment handle in cache */ for (envp = envlist; envp != NULL; envp = envp->next) { if (envp->envhp == envhp) break; prevenvp = envp; } if (envp == NULL) { if (silent) return; else oracleError(FDW_ERROR, "removeEnvironment internal error: environment handle not found in cache"); } /* free the error handle */ (void)OCIHandleFree((dvoid *)envp->errhp, OCI_HTYPE_ERROR); /* free the environment handle */ (void)OCIHandleFree((dvoid *)envp->envhp, OCI_HTYPE_ENV); /* remove environment entry from the linked list */ if (prevenvp == NULL) envlist = envp->next; else prevenvp->next = envp->next; /* free the memory */ free(envp->nls_lang); free(envp); } /* * allocHandle * Allocate an Oracle handle or descriptor, keep it in the cached list. */ void allocHandle(dvoid **handlepp, ub4 type, int isDescriptor, OCIEnv *envhp, struct connEntry *connp, oraError error, const char *errmsg) { struct handleEntry *entry; sword rc; /* create entry for linked list */ if ((entry = malloc(sizeof(struct handleEntry))) == NULL) { oracleError_i(FDW_OUT_OF_MEMORY, "error allocating handle: failed to allocate %d bytes of memory", sizeof(struct handleEntry)); } if (isDescriptor) rc = OCIDescriptorAlloc((const dvoid *)envhp, handlepp, type, (size_t)0, NULL); else rc = OCIHandleAlloc((const dvoid *)envhp, handlepp, type, (size_t)0, NULL); if (rc != OCI_SUCCESS) { free(entry); oracleError(error, errmsg); } /* add handle to linked list */ entry->handlep = *handlepp; entry->type = type; entry->isDescriptor = isDescriptor; entry->next = connp->handlelist; connp->handlelist = entry; } /* * freeHandle * Free an Oracle handle or descriptor, remove it from the cached list. */ void freeHandle(dvoid *handlep, struct connEntry *connp) { struct handleEntry *entry, *preventry = NULL; /* find it in the linked list */ for (entry = connp->handlelist; entry != NULL; entry = entry->next) { if (entry->handlep == handlep) break; preventry = entry; } if (entry == NULL) oracleError(FDW_ERROR, "internal error freeing handle: not found in cache"); /* free the handle */ if (entry->isDescriptor) (void)OCIDescriptorFree(handlep, entry->type); else (void)OCIHandleFree(handlep, entry->type); /* remove it */ if (preventry == NULL) connp->handlelist = entry->next; else preventry->next = entry->next; free(entry); } /* * getOraType * Find oracle's name for a given oraType. */ ub2 getOraType(oraType arg) { switch (arg) { case ORA_TYPE_BLOB: return SQLT_BLOB; case ORA_TYPE_BFILE: return SQLT_BFILE; case ORA_TYPE_CLOB: return SQLT_CLOB; case ORA_TYPE_RAW: return SQLT_BIN; case ORA_TYPE_LONG: return SQLT_LVC; case ORA_TYPE_LONGRAW: return SQLT_LVB; default: /* all other columns are converted to strings */ return SQLT_STR; } } /* * bind_out_callback * Point Oracle to where it should write the value for the output parameter. */ sb4 bind_out_callback(void *octxp, OCIBind *bindp, ub4 iter, ub4 index, void **bufpp, ub4 **alenp, ub1 *piecep, void **indp, ub2 **rcodep) { struct oraColumn *column = (struct oraColumn *)octxp; if (column->oratype == ORA_TYPE_BLOB || column->oratype == ORA_TYPE_CLOB || column->oratype == ORA_TYPE_BFILE) { /* for LOBs, data should be written to the LOB locator */ *bufpp = *((OCILobLocator **)column->val); } else { /* for other types, data should be written directly to the buffer */ *bufpp = column->val; } column->val_len4 = (unsigned int)column->val_size; *alenp = &(column->val_len4); *indp = &(column->val_null); *rcodep = NULL; if (*piecep == OCI_ONE_PIECE) return OCI_CONTINUE; else return OCI_ERROR; } /* * bind_in_callback * Provide a NULL value. * This is necessary for output parameters to keep Oracle from crashing. */ sb4 bind_in_callback(void *ictxp, OCIBind *bindp, ub4 iter, ub4 index, void **bufpp, ub4 *alenp, ub1 *piecep, void **indpp) { struct oraColumn *column = (struct oraColumn *)ictxp; *piecep = OCI_ONE_PIECE; column->val_null = -1; *indpp = &(column->val_null); return OCI_CONTINUE; } /* * oracleWriteLob * Write the value in "value" to the LOB Locator in "locptr". * "value" is supposed to be in LVB/LVC format (first 4 bytes are the length). */ void oracleWriteLob(oracleSession *session, OCILobLocator *locptr, char *value) { oraub8 byte_count = *((sb4 *)value), char_count = 0; /* empty LOB */ if (byte_count == 0) return; /* open the LOB */ if (checkerr( OCILobOpen(session->connp->svchp, session->envp->errhp, locptr, OCI_LOB_READWRITE), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error writing data: OCILobOpen failed to open LOB", oraMessage); } /* write the whole LOB in one piece */ if (checkerr( OCILobWrite2(session->connp->svchp, session->envp->errhp, locptr, &byte_count, &char_count, (oraub8)1, value + 4, byte_count, OCI_ONE_PIECE, NULL, NULL, (ub2)0, (ub1)0), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error writing data: OCILobWrite2 failed to write LOB", oraMessage); } /* close the LOB */ if (checkerr( OCILobClose(session->connp->svchp, session->envp->errhp, locptr), (dvoid *)session->envp->errhp, OCI_HTYPE_ERROR) != OCI_SUCCESS) { oracleError_d(FDW_UNABLE_TO_CREATE_EXECUTION, "error writing data: OCILobClose failed to close LOB", oraMessage); } }