static char *
CC_lookup_cs_new(ConnectionClass *self)
{
char *encstr = NULL;
QResultClass *res;
res = CC_send_query(self, "select pg_client_encoding()", NULL, IGNORE_ABORT_ON_CONN | ROLLBACK_ON_ERROR, NULL);
if (QR_command_maybe_successful(res))
{
const char *enc = QR_get_value_backend_text(res, 0, 0);
if (enc)
encstr = strdup(enc);
}
QR_Destructor(res);
return encstr;
}
RETCODE
DiscardStatementSvp(StatementClass *stmt, RETCODE ret, BOOL errorOnly)
{
CSTR func = "DiscardStatementSvp";
char esavepoint[32], cmd[64];
ConnectionClass *conn = SC_get_conn(stmt);
QResultClass *res;
BOOL cmd_success, start_stmt = FALSE;
inolog("%s:%p->accessed=%d is_in=%d is_rb=%d is_tc=%d\n", func, stmt, SC_accessed_db(stmt),
CC_is_in_trans(conn), SC_is_rb_stmt(stmt), SC_is_tc_stmt(stmt));
switch (ret)
{
case SQL_NEED_DATA:
break;
case SQL_ERROR:
start_stmt = TRUE;
break;
default:
if (!errorOnly)
start_stmt = TRUE;
break;
}
if (!SC_accessed_db(stmt) || !CC_is_in_trans(conn))
goto cleanup;
if (!SC_is_rb_stmt(stmt) && !SC_is_tc_stmt(stmt))
goto cleanup;
sprintf(esavepoint, "_EXEC_SVP_%p", stmt);
if (SQL_ERROR == ret)
{
if (SC_started_rbpoint(stmt))
{
snprintf(cmd, sizeof(cmd), "ROLLBACK to %s", esavepoint);
res = CC_send_query(conn, cmd, NULL, IGNORE_ABORT_ON_CONN, NULL);
cmd_success = QR_command_maybe_successful(res);
QR_Destructor(res);
if (!cmd_success)
{
SC_set_error(stmt, STMT_INTERNAL_ERROR, "internal ROLLBACK failed", func);
CC_abort(conn);
goto cleanup;
}
}
else
{
CC_abort(conn);
goto cleanup;
}
}
else if (errorOnly)
return ret;
inolog("ret=%d\n", ret);
if (SQL_NEED_DATA != ret && SC_started_rbpoint(stmt))
{
snprintf(cmd, sizeof(cmd), "RELEASE %s", esavepoint);
res = CC_send_query(conn, cmd, NULL, IGNORE_ABORT_ON_CONN, NULL);
cmd_success = QR_command_maybe_successful(res);
QR_Destructor(res);
if (!cmd_success)
{
SC_set_error(stmt, STMT_INTERNAL_ERROR, "internal RELEASE failed", func);
CC_abort(conn);
ret = SQL_ERROR;
}
}
cleanup:
if (SQL_NEED_DATA != ret)
SC_forget_unnamed(stmt); /* unnamed plan is no longer reliable */
if (!SC_is_prepare_statement(stmt) && ONCE_DESCRIBED == stmt->prepared)
SC_set_prepared(stmt, NOT_YET_PREPARED);
if (start_stmt || SQL_ERROR == ret)
{
if (stmt->lock_CC_for_rb > 0)
{
LEAVE_CONN_CS(conn);
stmt->lock_CC_for_rb--;
}
SC_start_stmt(stmt);
}
return ret;
}
/*
* Must be in a transaction or the subsequent execution
* invokes a transaction.
*/
RETCODE
SetStatementSvp(StatementClass *stmt)
{
CSTR func = "SetStatementSvp";
char esavepoint[32], cmd[64];
ConnectionClass *conn = SC_get_conn(stmt);
QResultClass *res;
RETCODE ret = SQL_SUCCESS_WITH_INFO;
if (CC_is_in_error_trans(conn))
return ret;
if (0 == stmt->lock_CC_for_rb)
{
ENTER_CONN_CS(conn);
stmt->lock_CC_for_rb++;
}
switch (stmt->statement_type)
{
case STMT_TYPE_SPECIAL:
case STMT_TYPE_TRANSACTION:
return ret;
}
if (!SC_accessed_db(stmt))
{
BOOL need_savep = FALSE;
if (stmt->internal)
{
if (PG_VERSION_GE(conn, 8.0))
SC_start_rb_stmt(stmt);
else
SC_start_tc_stmt(stmt);
}
if (SC_is_rb_stmt(stmt))
{
if (CC_is_in_trans(conn))
{
need_savep = TRUE;
}
}
if (need_savep)
{
sprintf(esavepoint, "_EXEC_SVP_%p", stmt);
snprintf(cmd, sizeof(cmd), "SAVEPOINT %s", esavepoint);
res = CC_send_query(conn, cmd, NULL, 0, NULL);
if (QR_command_maybe_successful(res))
{
SC_set_accessed_db(stmt);
SC_start_rbpoint(stmt);
ret = SQL_SUCCESS;
}
else
{
SC_set_error(stmt, STMT_INTERNAL_ERROR, "internal SAVEPOINT failed", func);
ret = SQL_ERROR;
}
QR_Destructor(res);
}
else
SC_set_accessed_db(stmt);
}
inolog("%s:%p->accessed=%d\n", func, stmt, SC_accessed_db(stmt));
return ret;
}
/*
* The execution after all parameters were resolved.
*/
static
RETCODE Exec_with_parameters_resolved(StatementClass *stmt, BOOL *exec_end)
{
CSTR func = "Exec_with_parameters_resolved";
RETCODE retval;
SQLLEN end_row;
SQLINTEGER cursor_type, scroll_concurrency;
ConnectionClass *conn;
QResultClass *res;
APDFields *apdopts;
IPDFields *ipdopts;
BOOL prepare_before_exec = FALSE;
*exec_end = FALSE;
conn = SC_get_conn(stmt);
mylog("%s: copying statement params: trans_status=%d, len=%d, stmt='%s'\n", func, conn->transact_status, strlen(stmt->statement), stmt->statement);
/* save the cursor's info before the execution */
cursor_type = stmt->options.cursor_type;
scroll_concurrency = stmt->options.scroll_concurrency;
/* Prepare the statement if possible at backend side */
if (!stmt->inaccurate_result)
{
if (HowToPrepareBeforeExec(stmt, FALSE) >= allowParse)
prepare_before_exec = TRUE;
}
inolog("prepare_before_exec=%d srv=%d\n", prepare_before_exec, conn->connInfo.use_server_side_prepare);
/* Create the statement with parameters substituted. */
retval = copy_statement_with_parameters(stmt, prepare_before_exec);
stmt->current_exec_param = -1;
if (retval != SQL_SUCCESS)
{
stmt->exec_current_row = -1;
*exec_end = TRUE;
return retval; /* error msg is passed from the above */
}
mylog(" stmt_with_params = '%s'\n", stmt->stmt_with_params);
/*
* Dummy exection to get the column info.
*/
if (stmt->inaccurate_result && SC_is_parse_tricky(stmt))
{
BOOL in_trans = CC_is_in_trans(conn);
BOOL issued_begin = FALSE;
QResultClass *curres;
stmt->exec_current_row = -1;
*exec_end = TRUE;
if (!SC_is_pre_executable(stmt))
return SQL_SUCCESS;
if (strnicmp(stmt->stmt_with_params, "BEGIN;", 6) == 0)
{
/* do nothing */
}
else if (!in_trans)
{
if (issued_begin = CC_begin(conn), !issued_begin)
{
SC_set_error(stmt, STMT_EXEC_ERROR, "Handle prepare error", func);
return SQL_ERROR;
}
}
/* we are now in a transaction */
res = CC_send_query(conn, stmt->stmt_with_params, NULL, 0, SC_get_ancestor(stmt));
if (!QR_command_maybe_successful(res))
{
#ifndef _LEGACY_MODE_
if (PG_VERSION_LT(conn, 8.0))
CC_abort(conn);
#endif /* LEGACY_MODE_ */
SC_set_error(stmt, STMT_EXEC_ERROR, "Handle prepare error", func);
QR_Destructor(res);
return SQL_ERROR;
}
SC_set_Result(stmt, res);
for (curres = res; !curres->num_fields; curres = curres->next)
;
SC_set_Curres(stmt, curres);
if (CC_does_autocommit(conn))
{
if (issued_begin)
CC_commit(conn);
}
stmt->status = STMT_FINISHED;
return SQL_SUCCESS;
}
/*
* The real execution.
*/
mylog("about to begin SC_execute\n");
retval = SC_execute(stmt);
if (retval == SQL_ERROR)
{
stmt->exec_current_row = -1;
*exec_end = TRUE;
return retval;
}
res = SC_get_Result(stmt);
/* special handling of result for keyset driven cursors */
if (SQL_CURSOR_KEYSET_DRIVEN == stmt->options.cursor_type &&
SQL_CONCUR_READ_ONLY != stmt->options.scroll_concurrency)
{
QResultClass *kres;
if (kres = res->next, kres)
{
if (kres->fields)
CI_Destructor(kres->fields);
kres->fields = res->fields;
res->fields = NULL;
kres->num_fields = res->num_fields;
res->next = NULL;
SC_set_Result(stmt, kres);
res = kres;
}
}
#ifdef NOT_USED
else if (SC_is_concat_prepare_exec(stmt))
{
if (res && QR_command_maybe_successful(res))
{
QResultClass *kres;
kres = res->next;
inolog("res->next=%p\n", kres);
res->next = NULL;
SC_set_Result(stmt, kres);
res = kres;
SC_set_prepared(stmt, PREPARED_PERMANENTLY);
}
else
{
retval = SQL_ERROR;
if (stmt->execute_statement)
free(stmt->execute_statement);
stmt->execute_statement = NULL;
}
}
#endif /* NOT_USED */
#if (ODBCVER >= 0x0300)
ipdopts = SC_get_IPDF(stmt);
if (ipdopts->param_status_ptr)
{
switch (retval)
{
case SQL_SUCCESS:
ipdopts->param_status_ptr[stmt->exec_current_row] = SQL_PARAM_SUCCESS;
break;
case SQL_SUCCESS_WITH_INFO:
ipdopts->param_status_ptr[stmt->exec_current_row] = SQL_PARAM_SUCCESS_WITH_INFO;
break;
default:
ipdopts->param_status_ptr[stmt->exec_current_row] = SQL_PARAM_ERROR;
break;
}
}
#endif /* ODBCVER */
if (end_row = stmt->exec_end_row, end_row < 0)
{
apdopts = SC_get_APDF(stmt);
end_row = (SQLINTEGER) apdopts->paramset_size - 1;
}
if (stmt->inaccurate_result ||
stmt->exec_current_row >= end_row)
{
*exec_end = TRUE;
stmt->exec_current_row = -1;
}
else
stmt->exec_current_row++;
if (res)
{
#if (ODBCVER >= 0x0300)
EnvironmentClass *env = (EnvironmentClass *) CC_get_env(conn);
const char *cmd = QR_get_command(res);
SQLLEN start_row;
if (start_row = stmt->exec_start_row, start_row < 0)
start_row = 0;
if (retval == SQL_SUCCESS &&
NULL != cmd &&
start_row >= end_row &&
NULL != env &&
EN_is_odbc3(env))
{
int count;
if (sscanf(cmd , "UPDATE %d", &count) == 1)
;
else if (sscanf(cmd , "DELETE %d", &count) == 1)
;
else
count = -1;
if (0 == count)
retval = SQL_NO_DATA;
}
#endif /* ODBCVER */
stmt->diag_row_count = res->recent_processed_row_count;
}
/*
* The cursor's info was changed ?
*/
if (retval == SQL_SUCCESS &&
(stmt->options.cursor_type != cursor_type ||
stmt->options.scroll_concurrency != scroll_concurrency))
{
SC_set_error(stmt, STMT_OPTION_VALUE_CHANGED, "cursor updatability changed", func);
retval = SQL_SUCCESS_WITH_INFO;
}
return retval;
}
void
CC_lookup_characterset(ConnectionClass *self)
{
char *encspec = NULL, *currenc = NULL, *tencstr;
CSTR func = "CC_lookup_characterset";
mylog("%s: entering...\n", func);
if (self->original_client_encoding)
encspec = strdup(self->original_client_encoding);
if (self->current_client_encoding)
currenc = strdup(self->current_client_encoding);
else if (PG_VERSION_LT(self, 7.2))
currenc = CC_lookup_cs_old(self);
else
currenc = CC_lookup_cs_new(self);
tencstr = encspec ? encspec : currenc;
if (self->original_client_encoding)
{
if (stricmp(self->original_client_encoding, tencstr))
{
char msg[256];
snprintf(msg, sizeof(msg), "The client_encoding '%s' was changed to '%s'", self->original_client_encoding, tencstr);
CC_set_error(self, CONN_OPTION_VALUE_CHANGED, msg, func);
}
free(self->original_client_encoding);
}
#ifndef UNICODE_SUPPORT
else
{
const char *wenc = get_environment_encoding(self, encspec, currenc, FALSE);
if (wenc && (!tencstr || stricmp(tencstr, wenc)))
{
QResultClass *res;
char query[64];
int errnum = CC_get_errornumber(self);
BOOL cmd_success;
sprintf(query, "set client_encoding to '%s'", wenc);
res = CC_send_query(self, query, NULL, IGNORE_ABORT_ON_CONN | ROLLBACK_ON_ERROR, NULL);
cmd_success = QR_command_maybe_successful(res);
QR_Destructor(res);
CC_set_errornumber(self, errnum);
if (cmd_success)
{
self->original_client_encoding = strdup(wenc);
self->ccsc = pg_CS_code(self->original_client_encoding);
if (encspec)
free(encspec);
if (currenc)
free(currenc);
return;
}
}
}
#endif /* UNICODE_SUPPORT */
if (tencstr)
{
self->original_client_encoding = tencstr;
if (encspec && currenc)
free(currenc);
self->ccsc = pg_CS_code(tencstr);
qlog(" [ Client encoding = '%s' (code = %d) ]\n", self->original_client_encoding, self->ccsc);
if (self->ccsc < 0)
{
char msg[256];
snprintf(msg, sizeof(msg), "would handle the encoding '%s' like ASCII", tencstr);
CC_set_error(self, CONN_OPTION_VALUE_CHANGED, msg, func);
}
}
else
{
self->ccsc = SQL_ASCII;
self->original_client_encoding = NULL;
}
self->mb_maxbyte_per_char = pg_mb_maxlen(self->ccsc);
}
