123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864 |
- /*
- +----------------------------------------------------------------------+
- | Copyright (c) The PHP Group |
- +----------------------------------------------------------------------+
- | This source file is subject to version 3.01 of the PHP license, |
- | that is bundled with this package in the file LICENSE, and is |
- | available through the world-wide-web at the following url: |
- | https://www.php.net/license/3_01.txt |
- | If you did not receive a copy of the PHP license and are unable to |
- | obtain it through the world-wide-web, please send a note to |
- | license@php.net so we can mail you a copy immediately. |
- +----------------------------------------------------------------------+
- | Author: Wez Furlong <wez@php.net> |
- +----------------------------------------------------------------------+
- */
- #ifdef HAVE_CONFIG_H
- #include "config.h"
- #endif
- #include "php.h"
- #include "php_ini.h"
- #include "ext/standard/info.h"
- #include "pdo/php_pdo.h"
- #include "pdo/php_pdo_driver.h"
- #include "php_pdo_odbc.h"
- #include "php_pdo_odbc_int.h"
- enum pdo_odbc_conv_result {
- PDO_ODBC_CONV_NOT_REQUIRED,
- PDO_ODBC_CONV_OK,
- PDO_ODBC_CONV_FAIL
- };
- static int pdo_odbc_sqltype_is_unicode(pdo_odbc_stmt *S, SWORD sqltype)
- {
- if (!S->assume_utf8) return 0;
- switch (sqltype) {
- #ifdef SQL_WCHAR
- case SQL_WCHAR:
- return 1;
- #endif
- #ifdef SQL_WLONGVARCHAR
- case SQL_WLONGVARCHAR:
- return 1;
- #endif
- #ifdef SQL_WVARCHAR
- case SQL_WVARCHAR:
- return 1;
- #endif
- default:
- return 0;
- }
- }
- static int pdo_odbc_utf82ucs2(pdo_stmt_t *stmt, int is_unicode, const char *buf,
- zend_ulong buflen, zend_ulong *outlen)
- {
- #ifdef PHP_WIN32
- if (is_unicode && buflen) {
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- DWORD ret;
- ret = MultiByteToWideChar(CP_UTF8, 0, buf, buflen, NULL, 0);
- if (ret == 0) {
- /*printf("%s:%d %d [%d] %.*s\n", __FILE__, __LINE__, GetLastError(), buflen, buflen, buf);*/
- return PDO_ODBC_CONV_FAIL;
- }
- ret *= sizeof(WCHAR);
- if (S->convbufsize <= ret) {
- S->convbufsize = ret + sizeof(WCHAR);
- S->convbuf = erealloc(S->convbuf, S->convbufsize);
- }
- ret = MultiByteToWideChar(CP_UTF8, 0, buf, buflen, (LPWSTR)S->convbuf, S->convbufsize / sizeof(WCHAR));
- if (ret == 0) {
- /*printf("%s:%d %d [%d] %.*s\n", __FILE__, __LINE__, GetLastError(), buflen, buflen, buf);*/
- return PDO_ODBC_CONV_FAIL;
- }
- ret *= sizeof(WCHAR);
- *outlen = ret;
- return PDO_ODBC_CONV_OK;
- }
- #endif
- return PDO_ODBC_CONV_NOT_REQUIRED;
- }
- static int pdo_odbc_ucs22utf8(pdo_stmt_t *stmt, int is_unicode, zval *result)
- {
- #ifdef PHP_WIN32
- ZEND_ASSERT(Z_TYPE_P(result) == IS_STRING);
- if (is_unicode && Z_STRLEN_P(result) != 0) {
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- DWORD ret;
- ret = WideCharToMultiByte(CP_UTF8, 0, (LPCWSTR) Z_STRVAL_P(result), Z_STRLEN_P(result)/sizeof(WCHAR), NULL, 0, NULL, NULL);
- if (ret == 0) {
- return PDO_ODBC_CONV_FAIL;
- }
- zend_string *str = zend_string_alloc(ret, 0);
- ret = WideCharToMultiByte(CP_UTF8, 0, (LPCWSTR) Z_STRVAL_P(result), Z_STRLEN_P(result)/sizeof(WCHAR), ZSTR_VAL(str), ZSTR_LEN(str), NULL, NULL);
- if (ret == 0) {
- return PDO_ODBC_CONV_FAIL;
- }
- ZSTR_VAL(str)[ret] = '\0';
- zval_ptr_dtor_str(result);
- ZVAL_STR(result, str);
- return PDO_ODBC_CONV_OK;
- }
- #endif
- return PDO_ODBC_CONV_NOT_REQUIRED;
- }
- static void free_cols(pdo_stmt_t *stmt, pdo_odbc_stmt *S)
- {
- if (S->cols) {
- int i;
- for (i = 0; i < S->col_count; i++) {
- if (S->cols[i].data) {
- efree(S->cols[i].data);
- }
- }
- efree(S->cols);
- S->cols = NULL;
- S->col_count = 0;
- }
- }
- static int odbc_stmt_dtor(pdo_stmt_t *stmt)
- {
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- if (S->stmt != SQL_NULL_HANDLE) {
- if (stmt->executed) {
- SQLCloseCursor(S->stmt);
- }
- SQLFreeHandle(SQL_HANDLE_STMT, S->stmt);
- S->stmt = SQL_NULL_HANDLE;
- }
- free_cols(stmt, S);
- if (S->convbuf) {
- efree(S->convbuf);
- }
- efree(S);
- return 1;
- }
- static int odbc_stmt_execute(pdo_stmt_t *stmt)
- {
- RETCODE rc;
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- char *buf = NULL;
- SQLLEN row_count = -1;
- if (stmt->executed) {
- SQLCloseCursor(S->stmt);
- }
- rc = SQLExecute(S->stmt);
- while (rc == SQL_NEED_DATA) {
- struct pdo_bound_param_data *param;
- rc = SQLParamData(S->stmt, (SQLPOINTER*)¶m);
- if (rc == SQL_NEED_DATA) {
- php_stream *stm;
- int len;
- pdo_odbc_param *P;
- zval *parameter;
- P = (pdo_odbc_param*)param->driver_data;
- if (Z_ISREF(param->parameter)) {
- parameter = Z_REFVAL(param->parameter);
- } else {
- parameter = ¶m->parameter;
- }
- if (Z_TYPE_P(parameter) != IS_RESOURCE) {
- /* they passed in a string */
- zend_ulong ulen;
- convert_to_string(parameter);
- switch (pdo_odbc_utf82ucs2(stmt, P->is_unicode,
- Z_STRVAL_P(parameter),
- Z_STRLEN_P(parameter),
- &ulen)) {
- case PDO_ODBC_CONV_NOT_REQUIRED:
- SQLPutData(S->stmt, Z_STRVAL_P(parameter),
- Z_STRLEN_P(parameter));
- break;
- case PDO_ODBC_CONV_OK:
- SQLPutData(S->stmt, S->convbuf, ulen);
- break;
- case PDO_ODBC_CONV_FAIL:
- pdo_odbc_stmt_error("error converting input string");
- SQLCloseCursor(S->stmt);
- if (buf) {
- efree(buf);
- }
- return 0;
- }
- continue;
- }
- /* we assume that LOBs are binary and don't need charset
- * conversion */
- php_stream_from_zval_no_verify(stm, parameter);
- if (!stm) {
- /* shouldn't happen either */
- pdo_odbc_stmt_error("input LOB is no longer a stream");
- SQLCloseCursor(S->stmt);
- if (buf) {
- efree(buf);
- }
- return 0;
- }
- /* now suck data from the stream and stick it into the database */
- if (buf == NULL) {
- buf = emalloc(8192);
- }
- do {
- len = php_stream_read(stm, buf, 8192);
- if (len == 0) {
- break;
- }
- SQLPutData(S->stmt, buf, len);
- } while (1);
- }
- }
- if (buf) {
- efree(buf);
- }
- switch (rc) {
- case SQL_SUCCESS:
- break;
- case SQL_NO_DATA_FOUND:
- case SQL_SUCCESS_WITH_INFO:
- pdo_odbc_stmt_error("SQLExecute");
- break;
- default:
- pdo_odbc_stmt_error("SQLExecute");
- return 0;
- }
- SQLRowCount(S->stmt, &row_count);
- stmt->row_count = row_count;
- if (S->cols == NULL) {
- /* do first-time-only definition of bind/mapping stuff */
- SQLSMALLINT colcount;
- /* how many columns do we have ? */
- SQLNumResultCols(S->stmt, &colcount);
- stmt->column_count = S->col_count = (int)colcount;
- S->cols = ecalloc(colcount, sizeof(pdo_odbc_column));
- S->going_long = 0;
- }
- return 1;
- }
- static int odbc_stmt_param_hook(pdo_stmt_t *stmt, struct pdo_bound_param_data *param,
- enum pdo_param_event event_type)
- {
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- RETCODE rc;
- SWORD sqltype = 0, ctype = 0, scale = 0, nullable = 0;
- SQLULEN precision = 0;
- pdo_odbc_param *P;
- zval *parameter;
- /* we're only interested in parameters for prepared SQL right now */
- if (param->is_param) {
- switch (event_type) {
- case PDO_PARAM_EVT_FETCH_PRE:
- case PDO_PARAM_EVT_FETCH_POST:
- case PDO_PARAM_EVT_NORMALIZE:
- /* Do nothing */
- break;
- case PDO_PARAM_EVT_FREE:
- P = param->driver_data;
- if (P) {
- efree(P);
- }
- break;
- case PDO_PARAM_EVT_ALLOC:
- {
- /* figure out what we're doing */
- switch (PDO_PARAM_TYPE(param->param_type)) {
- case PDO_PARAM_LOB:
- break;
- case PDO_PARAM_STMT:
- return 0;
- default:
- break;
- }
- rc = SQLDescribeParam(S->stmt, (SQLUSMALLINT) param->paramno+1, &sqltype, &precision, &scale, &nullable);
- if (rc != SQL_SUCCESS && rc != SQL_SUCCESS_WITH_INFO) {
- /* MS Access, for instance, doesn't support SQLDescribeParam,
- * so we need to guess */
- switch (PDO_PARAM_TYPE(param->param_type)) {
- case PDO_PARAM_INT:
- sqltype = SQL_INTEGER;
- break;
- case PDO_PARAM_LOB:
- sqltype = SQL_LONGVARBINARY;
- break;
- default:
- sqltype = SQL_LONGVARCHAR;
- }
- precision = 4000;
- scale = 5;
- nullable = 1;
- if (param->max_value_len > 0) {
- precision = param->max_value_len;
- }
- }
- if (sqltype == SQL_BINARY || sqltype == SQL_VARBINARY || sqltype == SQL_LONGVARBINARY) {
- ctype = SQL_C_BINARY;
- } else {
- ctype = SQL_C_CHAR;
- }
- P = emalloc(sizeof(*P));
- param->driver_data = P;
- P->len = 0; /* is re-populated each EXEC_PRE */
- P->outbuf = NULL;
- P->is_unicode = pdo_odbc_sqltype_is_unicode(S, sqltype);
- if (P->is_unicode) {
- /* avoid driver auto-translation: we'll do it ourselves */
- ctype = SQL_C_BINARY;
- }
- if ((param->param_type & PDO_PARAM_INPUT_OUTPUT) == PDO_PARAM_INPUT_OUTPUT) {
- P->paramtype = SQL_PARAM_INPUT_OUTPUT;
- } else if (param->max_value_len <= 0) {
- P->paramtype = SQL_PARAM_INPUT;
- } else {
- P->paramtype = SQL_PARAM_OUTPUT;
- }
- if (P->paramtype != SQL_PARAM_INPUT) {
- if (PDO_PARAM_TYPE(param->param_type) != PDO_PARAM_NULL) {
- /* need an explicit buffer to hold result */
- P->len = param->max_value_len > 0 ? param->max_value_len : precision;
- if (P->is_unicode) {
- P->len *= 2;
- }
- P->outbuf = emalloc(P->len + (P->is_unicode ? 2:1));
- }
- }
- if (PDO_PARAM_TYPE(param->param_type) == PDO_PARAM_LOB && P->paramtype != SQL_PARAM_INPUT) {
- pdo_odbc_stmt_error("Can't bind a lob for output");
- return 0;
- }
- rc = SQLBindParameter(S->stmt, (SQLUSMALLINT) param->paramno+1,
- P->paramtype, ctype, sqltype, precision, scale,
- P->paramtype == SQL_PARAM_INPUT ?
- (SQLPOINTER)param :
- P->outbuf,
- P->len,
- &P->len
- );
- if (rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO) {
- return 1;
- }
- pdo_odbc_stmt_error("SQLBindParameter");
- return 0;
- }
- case PDO_PARAM_EVT_EXEC_PRE:
- P = param->driver_data;
- if (!Z_ISREF(param->parameter)) {
- parameter = ¶m->parameter;
- } else {
- parameter = Z_REFVAL(param->parameter);
- }
- if (PDO_PARAM_TYPE(param->param_type) == PDO_PARAM_LOB) {
- if (Z_TYPE_P(parameter) == IS_RESOURCE) {
- php_stream *stm;
- php_stream_statbuf sb;
- php_stream_from_zval_no_verify(stm, parameter);
- if (!stm) {
- return 0;
- }
- if (0 == php_stream_stat(stm, &sb)) {
- if (P->outbuf) {
- int len, amount;
- char *ptr = P->outbuf;
- char *end = P->outbuf + P->len;
- P->len = 0;
- do {
- amount = end - ptr;
- if (amount == 0) {
- break;
- }
- if (amount > 8192)
- amount = 8192;
- len = php_stream_read(stm, ptr, amount);
- if (len == 0) {
- break;
- }
- ptr += len;
- P->len += len;
- } while (1);
- } else {
- P->len = SQL_LEN_DATA_AT_EXEC(sb.sb.st_size);
- }
- } else {
- if (P->outbuf) {
- P->len = 0;
- } else {
- P->len = SQL_LEN_DATA_AT_EXEC(0);
- }
- }
- } else {
- convert_to_string(parameter);
- if (P->outbuf) {
- P->len = Z_STRLEN_P(parameter);
- memcpy(P->outbuf, Z_STRVAL_P(parameter), P->len);
- } else {
- P->len = SQL_LEN_DATA_AT_EXEC(Z_STRLEN_P(parameter));
- }
- }
- } else if (Z_TYPE_P(parameter) == IS_NULL || PDO_PARAM_TYPE(param->param_type) == PDO_PARAM_NULL) {
- P->len = SQL_NULL_DATA;
- } else {
- convert_to_string(parameter);
- if (P->outbuf) {
- zend_ulong ulen;
- switch (pdo_odbc_utf82ucs2(stmt, P->is_unicode,
- Z_STRVAL_P(parameter),
- Z_STRLEN_P(parameter),
- &ulen)) {
- case PDO_ODBC_CONV_FAIL:
- case PDO_ODBC_CONV_NOT_REQUIRED:
- P->len = Z_STRLEN_P(parameter);
- memcpy(P->outbuf, Z_STRVAL_P(parameter), P->len);
- break;
- case PDO_ODBC_CONV_OK:
- P->len = ulen;
- memcpy(P->outbuf, S->convbuf, P->len);
- break;
- }
- } else {
- P->len = SQL_LEN_DATA_AT_EXEC(Z_STRLEN_P(parameter));
- }
- }
- return 1;
- case PDO_PARAM_EVT_EXEC_POST:
- P = param->driver_data;
- if (P->outbuf) {
- if (Z_ISREF(param->parameter)) {
- parameter = Z_REFVAL(param->parameter);
- } else {
- parameter = ¶m->parameter;
- }
- zval_ptr_dtor(parameter);
- if (P->len >= 0) {
- ZVAL_STRINGL(parameter, P->outbuf, P->len);
- switch (pdo_odbc_ucs22utf8(stmt, P->is_unicode, parameter)) {
- case PDO_ODBC_CONV_FAIL:
- /* something fishy, but allow it to come back as binary */
- case PDO_ODBC_CONV_NOT_REQUIRED:
- break;
- case PDO_ODBC_CONV_OK:
- break;
- }
- } else {
- ZVAL_NULL(parameter);
- }
- }
- return 1;
- }
- }
- return 1;
- }
- static int odbc_stmt_fetch(pdo_stmt_t *stmt,
- enum pdo_fetch_orientation ori, zend_long offset)
- {
- RETCODE rc;
- SQLSMALLINT odbcori;
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- switch (ori) {
- case PDO_FETCH_ORI_NEXT: odbcori = SQL_FETCH_NEXT; break;
- case PDO_FETCH_ORI_PRIOR: odbcori = SQL_FETCH_PRIOR; break;
- case PDO_FETCH_ORI_FIRST: odbcori = SQL_FETCH_FIRST; break;
- case PDO_FETCH_ORI_LAST: odbcori = SQL_FETCH_LAST; break;
- case PDO_FETCH_ORI_ABS: odbcori = SQL_FETCH_ABSOLUTE; break;
- case PDO_FETCH_ORI_REL: odbcori = SQL_FETCH_RELATIVE; break;
- default:
- strcpy(stmt->error_code, "HY106");
- return 0;
- }
- rc = SQLFetchScroll(S->stmt, odbcori, offset);
- if (rc == SQL_SUCCESS) {
- return 1;
- }
- if (rc == SQL_SUCCESS_WITH_INFO) {
- pdo_odbc_stmt_error("SQLFetchScroll");
- return 1;
- }
- if (rc == SQL_NO_DATA) {
- /* pdo_odbc_stmt_error("SQLFetchScroll"); */
- return 0;
- }
- pdo_odbc_stmt_error("SQLFetchScroll");
- return 0;
- }
- static int odbc_stmt_describe(pdo_stmt_t *stmt, int colno)
- {
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- struct pdo_column_data *col = &stmt->columns[colno];
- RETCODE rc;
- SWORD colnamelen;
- SQLULEN colsize;
- SQLLEN displaysize = 0;
- rc = SQLDescribeCol(S->stmt, colno+1, (SQLCHAR *) S->cols[colno].colname,
- sizeof(S->cols[colno].colname)-1, &colnamelen,
- &S->cols[colno].coltype, &colsize, NULL, NULL);
- /* This fixes a known issue with SQL Server and (max) lengths,
- may affect others as well. If we are SQL_VARCHAR,
- SQL_VARBINARY, or SQL_WVARCHAR (or any of the long variations)
- and zero is returned from colsize then consider it long */
- if (0 == colsize &&
- (S->cols[colno].coltype == SQL_VARCHAR ||
- S->cols[colno].coltype == SQL_LONGVARCHAR ||
- #ifdef SQL_WVARCHAR
- S->cols[colno].coltype == SQL_WVARCHAR ||
- #endif
- #ifdef SQL_WLONGVARCHAR
- S->cols[colno].coltype == SQL_WLONGVARCHAR ||
- #endif
- S->cols[colno].coltype == SQL_VARBINARY ||
- S->cols[colno].coltype == SQL_LONGVARBINARY)) {
- S->going_long = 1;
- }
- if (rc != SQL_SUCCESS) {
- pdo_odbc_stmt_error("SQLDescribeCol");
- if (rc != SQL_SUCCESS_WITH_INFO) {
- return 0;
- }
- }
- rc = SQLColAttribute(S->stmt, colno+1,
- SQL_DESC_DISPLAY_SIZE,
- NULL, 0, NULL, &displaysize);
- if (rc != SQL_SUCCESS) {
- pdo_odbc_stmt_error("SQLColAttribute");
- if (rc != SQL_SUCCESS_WITH_INFO) {
- return 0;
- }
- }
- colsize = displaysize;
- col->maxlen = S->cols[colno].datalen = colsize;
- col->name = zend_string_init(S->cols[colno].colname, colnamelen, 0);
- S->cols[colno].is_unicode = pdo_odbc_sqltype_is_unicode(S, S->cols[colno].coltype);
- /* tell ODBC to put it straight into our buffer, but only if it
- * isn't "long" data, and only if we haven't already bound a long
- * column. */
- if (colsize < 256 && !S->going_long) {
- S->cols[colno].data = emalloc(colsize+1);
- S->cols[colno].is_long = 0;
- rc = SQLBindCol(S->stmt, colno+1,
- S->cols[colno].is_unicode ? SQL_C_BINARY : SQL_C_CHAR,
- S->cols[colno].data,
- S->cols[colno].datalen+1, &S->cols[colno].fetched_len);
- if (rc != SQL_SUCCESS) {
- pdo_odbc_stmt_error("SQLBindCol");
- return 0;
- }
- } else {
- /* allocate a smaller buffer to keep around for smaller
- * "long" columns */
- S->cols[colno].data = emalloc(256);
- S->going_long = 1;
- S->cols[colno].is_long = 1;
- }
- return 1;
- }
- static int odbc_stmt_get_column_meta(pdo_stmt_t *stmt, zend_long colno, zval *return_value)
- {
- array_init(return_value);
- add_assoc_long(return_value, "pdo_type", PDO_PARAM_STR);
- return 1;
- }
- static int odbc_stmt_get_col(pdo_stmt_t *stmt, int colno, zval *result, enum pdo_param_type *type)
- {
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- pdo_odbc_column *C = &S->cols[colno];
- /* if it is a column containing "long" data, perform late binding now */
- if (C->is_long) {
- SQLLEN orig_fetched_len = SQL_NULL_DATA;
- RETCODE rc;
- /* fetch it into C->data, which is allocated with a length
- * of 256 bytes; if there is more to be had, we then allocate
- * bigger buffer for the caller to free */
- rc = SQLGetData(S->stmt, colno+1, C->is_unicode ? SQL_C_BINARY : SQL_C_CHAR, C->data,
- 256, &C->fetched_len);
- orig_fetched_len = C->fetched_len;
- if (rc == SQL_SUCCESS && C->fetched_len < 256) {
- /* all the data fit into our little buffer;
- * jump down to the generic bound data case */
- goto in_data;
- }
- if (rc == SQL_SUCCESS_WITH_INFO || rc == SQL_SUCCESS) {
- /* this is a 'long column'
- read the column in 255 byte blocks until the end of the column is reached, reassembling those blocks
- in order into the output buffer; 255 bytes are an optimistic assumption, since the driver may assert
- more or less NUL bytes at the end; we cater to that later, if actual length information is available
- this loop has to work whether or not SQLGetData() provides the total column length.
- calling SQLDescribeCol() or other, specifically to get the column length, then doing a single read
- for that size would be slower except maybe for extremely long columns.*/
- char *buf2 = emalloc(256);
- zend_string *str = zend_string_init(C->data, 256, 0);
- size_t used = 255; /* not 256; the driver NUL terminated the buffer */
- do {
- C->fetched_len = 0;
- /* read block. 256 bytes => 255 bytes are actually read, the last 1 is NULL */
- rc = SQLGetData(S->stmt, colno+1, C->is_unicode ? SQL_C_BINARY : SQL_C_CHAR, buf2, 256, &C->fetched_len);
- /* adjust `used` in case we have length info from the driver */
- if (orig_fetched_len >= 0 && C->fetched_len >= 0) {
- SQLLEN fixed_used = orig_fetched_len - C->fetched_len;
- ZEND_ASSERT(fixed_used <= used + 1);
- used = fixed_used;
- }
- /* resize output buffer and reassemble block */
- if (rc==SQL_SUCCESS_WITH_INFO || (rc==SQL_SUCCESS && C->fetched_len > 255)) {
- /* point 5, in section "Retrieving Data with SQLGetData" in http://msdn.microsoft.com/en-us/library/windows/desktop/ms715441(v=vs.85).aspx
- states that if SQL_SUCCESS_WITH_INFO, fetched_len will be > 255 (greater than buf2's size)
- (if a driver fails to follow that and wrote less than 255 bytes to buf2, this will AV or read garbage into buf) */
- str = zend_string_realloc(str, used + 256, 0);
- memcpy(ZSTR_VAL(str) + used, buf2, 256);
- used = used + 255;
- } else if (rc==SQL_SUCCESS) {
- str = zend_string_realloc(str, used + C->fetched_len, 0);
- memcpy(ZSTR_VAL(str) + used, buf2, C->fetched_len);
- used = used + C->fetched_len;
- } else {
- /* includes SQL_NO_DATA */
- break;
- }
- } while (1);
- efree(buf2);
- /* NULL terminate the buffer once, when finished, for use with the rest of PHP */
- ZSTR_VAL(str)[used] = '\0';
- ZVAL_STR(result, str);
- if (C->is_unicode) {
- goto unicode_conv;
- }
- return 1;
- }
- /* something went caca */
- return 1;
- }
- in_data:
- /* check the indicator to ensure that the data is intact */
- if (C->fetched_len == SQL_NULL_DATA) {
- /* A NULL value */
- ZVAL_NULL(result);
- return 1;
- } else if (C->fetched_len >= 0) {
- /* it was stored perfectly */
- ZVAL_STRINGL_FAST(result, C->data, C->fetched_len);
- if (C->is_unicode) {
- goto unicode_conv;
- }
- return 1;
- } else {
- /* no data? */
- ZVAL_NULL(result);
- return 1;
- }
- unicode_conv:
- switch (pdo_odbc_ucs22utf8(stmt, C->is_unicode, result)) {
- case PDO_ODBC_CONV_FAIL:
- /* oh well. They can have the binary version of it */
- case PDO_ODBC_CONV_NOT_REQUIRED:
- /* shouldn't happen... */
- return 1;
- case PDO_ODBC_CONV_OK:
- return 1;
- }
- return 1;
- }
- static int odbc_stmt_set_param(pdo_stmt_t *stmt, zend_long attr, zval *val)
- {
- SQLRETURN rc;
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- switch (attr) {
- case PDO_ATTR_CURSOR_NAME:
- convert_to_string(val);
- rc = SQLSetCursorName(S->stmt, (SQLCHAR *) Z_STRVAL_P(val), Z_STRLEN_P(val));
- if (rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO) {
- return 1;
- }
- pdo_odbc_stmt_error("SQLSetCursorName");
- return 0;
- case PDO_ODBC_ATTR_ASSUME_UTF8:
- S->assume_utf8 = zval_is_true(val);
- return 0;
- default:
- strcpy(S->einfo.last_err_msg, "Unknown Attribute");
- S->einfo.what = "setAttribute";
- strcpy(S->einfo.last_state, "IM001");
- return -1;
- }
- }
- static int odbc_stmt_get_attr(pdo_stmt_t *stmt, zend_long attr, zval *val)
- {
- SQLRETURN rc;
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- switch (attr) {
- case PDO_ATTR_CURSOR_NAME:
- {
- char buf[256];
- SQLSMALLINT len = 0;
- rc = SQLGetCursorName(S->stmt, (SQLCHAR *) buf, sizeof(buf), &len);
- if (rc == SQL_SUCCESS || rc == SQL_SUCCESS_WITH_INFO) {
- ZVAL_STRINGL(val, buf, len);
- return 1;
- }
- pdo_odbc_stmt_error("SQLGetCursorName");
- return 0;
- }
- case PDO_ODBC_ATTR_ASSUME_UTF8:
- ZVAL_BOOL(val, S->assume_utf8 ? 1 : 0);
- return 0;
- default:
- strcpy(S->einfo.last_err_msg, "Unknown Attribute");
- S->einfo.what = "getAttribute";
- strcpy(S->einfo.last_state, "IM001");
- return -1;
- }
- }
- static int odbc_stmt_next_rowset(pdo_stmt_t *stmt)
- {
- SQLRETURN rc;
- SQLSMALLINT colcount;
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- /* NOTE: can't guarantee that output or input/output parameters
- * are set until this fella returns SQL_NO_DATA, according to
- * MSDN ODBC docs */
- rc = SQLMoreResults(S->stmt);
- if (rc != SQL_SUCCESS && rc != SQL_SUCCESS_WITH_INFO) {
- return 0;
- }
- free_cols(stmt, S);
- /* how many columns do we have ? */
- SQLNumResultCols(S->stmt, &colcount);
- stmt->column_count = S->col_count = (int)colcount;
- S->cols = ecalloc(colcount, sizeof(pdo_odbc_column));
- S->going_long = 0;
- return 1;
- }
- static int odbc_stmt_close_cursor(pdo_stmt_t *stmt)
- {
- SQLRETURN rc;
- pdo_odbc_stmt *S = (pdo_odbc_stmt*)stmt->driver_data;
- rc = SQLCloseCursor(S->stmt);
- if (rc != SQL_SUCCESS && rc != SQL_SUCCESS_WITH_INFO) {
- return 0;
- }
- return 1;
- }
- const struct pdo_stmt_methods odbc_stmt_methods = {
- odbc_stmt_dtor,
- odbc_stmt_execute,
- odbc_stmt_fetch,
- odbc_stmt_describe,
- odbc_stmt_get_col,
- odbc_stmt_param_hook,
- odbc_stmt_set_param,
- odbc_stmt_get_attr,
- odbc_stmt_get_column_meta,
- odbc_stmt_next_rowset,
- odbc_stmt_close_cursor
- };
|