1/*
2  +----------------------------------------------------------------------+
3  | PHP Version 7                                                        |
4  +----------------------------------------------------------------------+
5  | Copyright (c) 1997-2015 The PHP Group                                |
6  +----------------------------------------------------------------------+
7  | This source file is subject to version 3.01 of the PHP license,      |
8  | that is bundled with this package in the file LICENSE, and is        |
9  | available through the world-wide-web at the following url:           |
10  | http://www.php.net/license/3_01.txt                                  |
11  | If you did not receive a copy of the PHP license and are unable to   |
12  | obtain it through the world-wide-web, please send a note to          |
13  | license@php.net so we can mail you a copy immediately.               |
14  +----------------------------------------------------------------------+
15  | Author: Wez Furlong <wez@php.net>                                    |
16  |         Marcus Boerger <helly@php.net>                               |
17  |         Sterling Hughes <sterling@php.net>                           |
18  +----------------------------------------------------------------------+
19*/
20
21/* $Id$ */
22
23/* The PDO Database Handle Class */
24
25#ifdef HAVE_CONFIG_H
26#include "config.h"
27#endif
28
29#include "php.h"
30#include "php_ini.h"
31#include "ext/standard/info.h"
32#include "php_pdo.h"
33#include "php_pdo_driver.h"
34#include "php_pdo_int.h"
35#include "zend_exceptions.h"
36#include "zend_object_handlers.h"
37#include "zend_hash.h"
38
39static int pdo_dbh_attribute_set(pdo_dbh_t *dbh, zend_long attr, zval *value);
40
41void pdo_raise_impl_error(pdo_dbh_t *dbh, pdo_stmt_t *stmt, const char *sqlstate, const char *supp) /* {{{ */
42{
43    pdo_error_type *pdo_err = &dbh->error_code;
44    char *message = NULL;
45    const char *msg;
46
47    if (dbh && dbh->error_mode == PDO_ERRMODE_SILENT) {
48#if 0
49        /* BUG: if user is running in silent mode and hits an error at the driver level
50         * when they use the PDO methods to call up the error information, they may
51         * get bogus information */
52        return;
53#endif
54    }
55
56    if (stmt) {
57        pdo_err = &stmt->error_code;
58    }
59
60    strncpy(*pdo_err, sqlstate, 6);
61
62    /* hash sqlstate to error messages */
63    msg = pdo_sqlstate_state_to_description(*pdo_err);
64    if (!msg) {
65        msg = "<<Unknown error>>";
66    }
67
68    if (supp) {
69        spprintf(&message, 0, "SQLSTATE[%s]: %s: %s", *pdo_err, msg, supp);
70    } else {
71        spprintf(&message, 0, "SQLSTATE[%s]: %s", *pdo_err, msg);
72    }
73
74    if (dbh && dbh->error_mode != PDO_ERRMODE_EXCEPTION) {
75        php_error_docref(NULL, E_WARNING, "%s", message);
76    } else {
77        zval ex, info;
78        zend_class_entry *def_ex = php_pdo_get_exception_base(1), *pdo_ex = php_pdo_get_exception();
79
80        object_init_ex(&ex, pdo_ex);
81
82        zend_update_property_string(def_ex, &ex, "message", sizeof("message")-1, message);
83        zend_update_property_string(def_ex, &ex, "code", sizeof("code")-1, *pdo_err);
84
85        array_init(&info);
86
87        add_next_index_string(&info, *pdo_err);
88        add_next_index_long(&info, 0);
89        zend_update_property(pdo_ex, &ex, "errorInfo", sizeof("errorInfo")-1, &info);
90        zval_ptr_dtor(&info);
91
92        zend_throw_exception_object(&ex);
93    }
94
95    if (message) {
96        efree(message);
97    }
98}
99/* }}} */
100
101PDO_API void pdo_handle_error(pdo_dbh_t *dbh, pdo_stmt_t *stmt) /* {{{ */
102{
103    pdo_error_type *pdo_err = &dbh->error_code;
104    const char *msg = "<<Unknown>>";
105    char *supp = NULL;
106    zend_long native_code = 0;
107    zend_string *message = NULL;
108    zval info;
109
110    if (dbh == NULL || dbh->error_mode == PDO_ERRMODE_SILENT) {
111        return;
112    }
113
114    if (stmt) {
115        pdo_err = &stmt->error_code;
116    }
117
118    /* hash sqlstate to error messages */
119    msg = pdo_sqlstate_state_to_description(*pdo_err);
120    if (!msg) {
121        msg = "<<Unknown error>>";
122    }
123
124    ZVAL_UNDEF(&info);
125    if (dbh->methods->fetch_err) {
126        array_init(&info);
127
128        add_next_index_string(&info, *pdo_err);
129
130        if (dbh->methods->fetch_err(dbh, stmt, &info)) {
131            zval *item;
132
133            if ((item = zend_hash_index_find(Z_ARRVAL(info), 1)) != NULL) {
134                native_code = Z_LVAL_P(item);
135            }
136
137            if ((item = zend_hash_index_find(Z_ARRVAL(info), 2)) != NULL) {
138                supp = estrndup(Z_STRVAL_P(item), Z_STRLEN_P(item));
139            }
140        }
141    }
142
143    if (supp) {
144        message = strpprintf(0, "SQLSTATE[%s]: %s: %ld %s", *pdo_err, msg, native_code, supp);
145    } else {
146        message = strpprintf(0, "SQLSTATE[%s]: %s", *pdo_err, msg);
147    }
148
149    if (dbh->error_mode == PDO_ERRMODE_WARNING) {
150        php_error_docref(NULL, E_WARNING, "%s", message->val);
151    } else if (EG(exception) == NULL) {
152        zval ex;
153        zend_class_entry *def_ex = php_pdo_get_exception_base(1), *pdo_ex = php_pdo_get_exception();
154
155        object_init_ex(&ex, pdo_ex);
156
157        zend_update_property_str(def_ex, &ex, "message", sizeof("message") - 1, message);
158        zend_update_property_string(def_ex, &ex, "code", sizeof("code") - 1, *pdo_err);
159
160        if (!Z_ISUNDEF(info)) {
161            zend_update_property(pdo_ex, &ex, "errorInfo", sizeof("errorInfo") - 1, &info);
162        }
163
164        zend_throw_exception_object(&ex);
165    }
166
167    if (!Z_ISUNDEF(info)) {
168        zval_ptr_dtor(&info);
169    }
170
171    if (message) {
172        zend_string_release(message);
173    }
174
175    if (supp) {
176        efree(supp);
177    }
178}
179/* }}} */
180
181static char *dsn_from_uri(char *uri, char *buf, size_t buflen) /* {{{ */
182{
183    php_stream *stream;
184    char *dsn = NULL;
185
186    stream = php_stream_open_wrapper(uri, "rb", REPORT_ERRORS, NULL);
187    if (stream) {
188        dsn = php_stream_get_line(stream, buf, buflen, NULL);
189        php_stream_close(stream);
190    }
191    return dsn;
192}
193/* }}} */
194
195/* {{{ proto void PDO::__construct(string dsn[, string username[, string passwd [, array options]]])
196   */
197static PHP_METHOD(PDO, dbh_constructor)
198{
199    zval *object = getThis();
200    pdo_dbh_t *dbh = NULL;
201    zend_bool is_persistent = 0;
202    char *data_source;
203    size_t data_source_len;
204    char *colon;
205    char *username=NULL, *password=NULL;
206    size_t usernamelen, passwordlen;
207    pdo_driver_t *driver = NULL;
208    zval *options = NULL;
209    char alt_dsn[512];
210    int call_factory = 1;
211    zend_error_handling zeh;
212
213    if (FAILURE == zend_parse_parameters_throw(ZEND_NUM_ARGS(),
214                "s|s!s!a!", &data_source, &data_source_len,
215                &username, &usernamelen, &password, &passwordlen, &options)) {
216        return;
217    }
218
219    /* parse the data source name */
220    colon = strchr(data_source, ':');
221
222    if (!colon) {
223        /* let's see if this string has a matching dsn in the php.ini */
224        char *ini_dsn = NULL;
225
226        snprintf(alt_dsn, sizeof(alt_dsn), "pdo.dsn.%s", data_source);
227        if (FAILURE == cfg_get_string(alt_dsn, &ini_dsn)) {
228            zend_throw_exception_ex(php_pdo_get_exception(), 0, "invalid data source name");
229            return;
230        }
231
232        data_source = ini_dsn;
233        colon = strchr(data_source, ':');
234
235        if (!colon) {
236            zend_throw_exception_ex(php_pdo_get_exception(), 0, "invalid data source name (via INI: %s)", alt_dsn);
237            return;
238        }
239    }
240
241    if (!strncmp(data_source, "uri:", sizeof("uri:")-1)) {
242        /* the specified URI holds connection details */
243        data_source = dsn_from_uri(data_source + sizeof("uri:")-1, alt_dsn, sizeof(alt_dsn));
244        if (!data_source) {
245            zend_throw_exception_ex(php_pdo_get_exception(), 0, "invalid data source URI");
246            return;
247        }
248        colon = strchr(data_source, ':');
249        if (!colon) {
250            zend_throw_exception_ex(php_pdo_get_exception(), 0, "invalid data source name (via URI)");
251            return;
252        }
253    }
254
255    driver = pdo_find_driver(data_source, colon - data_source);
256
257    if (!driver) {
258        /* NB: don't want to include the data_source in the error message as
259         * it might contain a password */
260        zend_throw_exception_ex(php_pdo_get_exception(), 0, "could not find driver");
261        return;
262    }
263
264    dbh = Z_PDO_DBH_P(object);
265
266    /* is this supposed to be a persistent connection ? */
267    if (options) {
268        int plen = 0;
269        char *hashkey = NULL;
270        zend_resource *le;
271        pdo_dbh_t *pdbh = NULL;
272        zval *v;
273
274        if ((v = zend_hash_index_find(Z_ARRVAL_P(options), PDO_ATTR_PERSISTENT)) != NULL) {
275            if (Z_TYPE_P(v) == IS_STRING &&
276                !is_numeric_string(Z_STRVAL_P(v), Z_STRLEN_P(v), NULL, NULL, 0) && Z_STRLEN_P(v) > 0) {
277                /* user specified key */
278                plen = spprintf(&hashkey, 0, "PDO:DBH:DSN=%s:%s:%s:%s", data_source,
279                        username ? username : "",
280                        password ? password : "",
281                        Z_STRVAL_P(v));
282                is_persistent = 1;
283            } else {
284                convert_to_long_ex(v);
285                is_persistent = Z_LVAL_P(v) ? 1 : 0;
286                plen = spprintf(&hashkey, 0, "PDO:DBH:DSN=%s:%s:%s", data_source,
287                        username ? username : "",
288                        password ? password : "");
289            }
290        }
291
292        if (is_persistent) {
293            /* let's see if we have one cached.... */
294            if ((le = zend_hash_str_find_ptr(&EG(persistent_list), hashkey, plen)) != NULL) {
295                if (le->type == php_pdo_list_entry()) {
296                    pdbh = (pdo_dbh_t*)le->ptr;
297
298                    /* is the connection still alive ? */
299                    if (pdbh->methods->check_liveness && FAILURE == (pdbh->methods->check_liveness)(pdbh)) {
300                        /* nope... need to kill it */
301                        /*??? memory leak */
302                        zend_list_close(le);
303                        pdbh = NULL;
304                    }
305                }
306            }
307
308            if (pdbh) {
309                call_factory = 0;
310            } else {
311                /* need a brand new pdbh */
312                pdbh = pecalloc(1, sizeof(*pdbh), 1);
313
314                pdbh->is_persistent = 1;
315                pdbh->persistent_id = pemalloc(plen + 1, 1);
316                memcpy((char *)pdbh->persistent_id, hashkey, plen+1);
317                pdbh->persistent_id_len = plen;
318                pdbh->def_stmt_ce = dbh->def_stmt_ce;
319            }
320        }
321
322        if (pdbh) {
323            efree(dbh);
324            /* switch over to the persistent one */
325            Z_PDO_OBJECT_P(object)->inner = pdbh;
326            dbh = pdbh;
327        }
328
329        if (hashkey) {
330            efree(hashkey);
331        }
332    }
333
334    if (call_factory) {
335        dbh->data_source_len = strlen(colon + 1);
336        dbh->data_source = (const char*)pestrdup(colon + 1, is_persistent);
337        dbh->username = username ? pestrdup(username, is_persistent) : NULL;
338        dbh->password = password ? pestrdup(password, is_persistent) : NULL;
339        dbh->default_fetch_type = PDO_FETCH_BOTH;
340    }
341
342    dbh->auto_commit = pdo_attr_lval(options, PDO_ATTR_AUTOCOMMIT, 1);
343
344    if (!dbh->data_source || (username && !dbh->username) || (password && !dbh->password)) {
345        php_error_docref(NULL, E_ERROR, "out of memory");
346    }
347
348    zend_replace_error_handling(EH_THROW, pdo_exception_ce, &zeh);
349
350    if (!call_factory) {
351        /* we got a persistent guy from our cache */
352        goto options;
353    }
354
355    if (driver->db_handle_factory(dbh, options)) {
356        /* all set */
357
358        if (is_persistent) {
359            zend_resource le;
360
361            /* register in the persistent list etc. */
362            /* we should also need to replace the object store entry,
363               since it was created with emalloc */
364
365            le.type = php_pdo_list_entry();
366            le.ptr = dbh;
367            GC_REFCOUNT(&le) = 1;
368
369            if ((zend_hash_str_update_mem(&EG(persistent_list),
370                        (char*)dbh->persistent_id, dbh->persistent_id_len, &le, sizeof(le))) == NULL) {
371                php_error_docref(NULL, E_ERROR, "Failed to register persistent entry");
372            }
373        }
374
375        dbh->driver = driver;
376options:
377        if (options) {
378            zval *attr_value;
379            zend_ulong long_key;
380            zend_string *str_key = NULL;
381
382            ZEND_HASH_FOREACH_KEY_VAL(Z_ARRVAL_P(options), long_key, str_key, attr_value) {
383                if (str_key) {
384                    continue;
385                }
386                pdo_dbh_attribute_set(dbh, long_key, attr_value);
387            } ZEND_HASH_FOREACH_END();
388        }
389
390        zend_restore_error_handling(&zeh);
391        return;
392    }
393
394    /* the connection failed; things will tidy up in free_storage */
395    /* XXX raise exception */
396    zend_restore_error_handling(&zeh);
397    if (!EG(exception)) {
398        zend_throw_exception(pdo_exception_ce, "Constructor failed", 0);
399    }
400}
401/* }}} */
402
403static zval *pdo_stmt_instantiate(pdo_dbh_t *dbh, zval *object, zend_class_entry *dbstmt_ce, zval *ctor_args) /* {{{ */
404{
405    if (!Z_ISUNDEF_P(ctor_args)) {
406        if (Z_TYPE_P(ctor_args) != IS_ARRAY) {
407            pdo_raise_impl_error(dbh, NULL, "HY000", "constructor arguments must be passed as an array");
408            return NULL;
409        }
410        if (!dbstmt_ce->constructor) {
411            pdo_raise_impl_error(dbh, NULL, "HY000", "user-supplied statement does not accept constructor arguments");
412            return NULL;
413        }
414    }
415
416    if (UNEXPECTED(object_init_ex(object, dbstmt_ce) != SUCCESS)) {
417        return NULL;
418    }
419    // ??? Z_SET_REFCOUNT_P(object, 1);
420    //Z_SET_ISREF_P(object);
421
422    return object;
423} /* }}} */
424
425static void pdo_stmt_construct(zend_execute_data *execute_data, pdo_stmt_t *stmt, zval *object, zend_class_entry *dbstmt_ce, zval *ctor_args) /* {{{ */
426{
427    zval query_string;
428    zval z_key;
429
430    ZVAL_STRINGL(&query_string, stmt->query_string, stmt->query_stringlen);
431    ZVAL_STRINGL(&z_key, "queryString", sizeof("queryString") - 1);
432    std_object_handlers.write_property(object, &z_key, &query_string, NULL);
433    zval_ptr_dtor(&query_string);
434    zval_ptr_dtor(&z_key);
435
436    if (dbstmt_ce->constructor) {
437        zend_fcall_info fci;
438        zend_fcall_info_cache fcc;
439        zval retval;
440
441        fci.size = sizeof(zend_fcall_info);
442        fci.function_table = &dbstmt_ce->function_table;
443        ZVAL_UNDEF(&fci.function_name);
444        fci.object = Z_OBJ_P(object);
445        fci.symbol_table = NULL;
446        fci.retval = &retval;
447        fci.param_count = 0;
448        fci.params = NULL;
449        fci.no_separation = 1;
450
451        zend_fcall_info_args(&fci, ctor_args);
452
453        fcc.initialized = 1;
454        fcc.function_handler = dbstmt_ce->constructor;
455        fcc.calling_scope = EG(scope);
456        fcc.called_scope = Z_OBJCE_P(object);
457        fcc.object = Z_OBJ_P(object);
458
459        if (zend_call_function(&fci, &fcc) != FAILURE) {
460            zval_ptr_dtor(&retval);
461        }
462
463        zend_fcall_info_args_clear(&fci, 1);
464    }
465}
466/* }}} */
467
468/* {{{ proto object PDO::prepare(string statement [, array options])
469   Prepares a statement for execution and returns a statement object */
470static PHP_METHOD(PDO, prepare)
471{
472    pdo_stmt_t *stmt;
473    char *statement;
474    size_t statement_len;
475    zval *options = NULL, *opt, *item, ctor_args;
476    zend_class_entry *dbstmt_ce, *pce;
477    pdo_dbh_object_t *dbh_obj = Z_PDO_OBJECT_P(getThis());
478    pdo_dbh_t *dbh = dbh_obj->inner;
479
480    if (FAILURE == zend_parse_parameters(ZEND_NUM_ARGS(), "s|a", &statement,
481            &statement_len, &options)) {
482        RETURN_FALSE;
483    }
484
485    PDO_DBH_CLEAR_ERR();
486    PDO_CONSTRUCT_CHECK;
487
488    if (ZEND_NUM_ARGS() > 1 && (opt = zend_hash_index_find(Z_ARRVAL_P(options), PDO_ATTR_STATEMENT_CLASS)) != NULL) {
489        if (Z_TYPE_P(opt) != IS_ARRAY || (item = zend_hash_index_find(Z_ARRVAL_P(opt), 0)) == NULL
490            || Z_TYPE_P(item) != IS_STRING
491            || (pce = zend_lookup_class(Z_STR_P(item))) == NULL
492        ) {
493            pdo_raise_impl_error(dbh, NULL, "HY000",
494                "PDO::ATTR_STATEMENT_CLASS requires format array(classname, array(ctor_args)); "
495                "the classname must be a string specifying an existing class"
496                );
497            PDO_HANDLE_DBH_ERR();
498            RETURN_FALSE;
499        }
500        dbstmt_ce = pce;
501        if (!instanceof_function(dbstmt_ce, pdo_dbstmt_ce)) {
502            pdo_raise_impl_error(dbh, NULL, "HY000",
503                "user-supplied statement class must be derived from PDOStatement");
504            PDO_HANDLE_DBH_ERR();
505            RETURN_FALSE;
506        }
507        if (dbstmt_ce->constructor && !(dbstmt_ce->constructor->common.fn_flags & (ZEND_ACC_PRIVATE|ZEND_ACC_PROTECTED))) {
508            pdo_raise_impl_error(dbh, NULL, "HY000",
509                "user-supplied statement class cannot have a public constructor");
510            PDO_HANDLE_DBH_ERR();
511            RETURN_FALSE;
512        }
513        if ((item = zend_hash_index_find(Z_ARRVAL_P(opt), 1)) != NULL) {
514            if (Z_TYPE_P(item) != IS_ARRAY) {
515                pdo_raise_impl_error(dbh, NULL, "HY000",
516                    "PDO::ATTR_STATEMENT_CLASS requires format array(classname, ctor_args); "
517                    "ctor_args must be an array"
518                );
519                PDO_HANDLE_DBH_ERR();
520                RETURN_FALSE;
521            }
522            ZVAL_COPY_VALUE(&ctor_args, item);
523        } else {
524            ZVAL_UNDEF(&ctor_args);
525        }
526    } else {
527        dbstmt_ce = dbh->def_stmt_ce;
528        ZVAL_COPY_VALUE(&ctor_args, &dbh->def_stmt_ctor_args);
529    }
530
531    if (!pdo_stmt_instantiate(dbh, return_value, dbstmt_ce, &ctor_args)) {
532        if (EXPECTED(!EG(exception))) {
533            pdo_raise_impl_error(dbh, NULL, "HY000",
534                "failed to instantiate user-supplied statement class"
535                );
536        }
537        PDO_HANDLE_DBH_ERR();
538        RETURN_FALSE;
539    }
540    stmt = Z_PDO_STMT_P(return_value);
541
542    /* unconditionally keep this for later reference */
543    stmt->query_string = estrndup(statement, statement_len);
544    stmt->query_stringlen = statement_len;
545    stmt->default_fetch_type = dbh->default_fetch_type;
546    stmt->dbh = dbh;
547    /* give it a reference to me */
548    ZVAL_OBJ(&stmt->database_object_handle, &dbh_obj->std);
549    Z_ADDREF(stmt->database_object_handle);
550    /* we haven't created a lazy object yet */
551    ZVAL_UNDEF(&stmt->lazy_object_ref);
552
553    if (dbh->methods->preparer(dbh, statement, statement_len, stmt, options)) {
554        pdo_stmt_construct(execute_data, stmt, return_value, dbstmt_ce, &ctor_args);
555        return;
556    }
557
558    PDO_HANDLE_DBH_ERR();
559
560    /* kill the object handle for the stmt here */
561    zval_dtor(return_value);
562
563    RETURN_FALSE;
564}
565/* }}} */
566
567/* {{{ proto bool PDO::beginTransaction()
568   Initiates a transaction */
569static PHP_METHOD(PDO, beginTransaction)
570{
571    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
572
573    if (zend_parse_parameters_none() == FAILURE) {
574        return;
575    }
576    PDO_CONSTRUCT_CHECK;
577
578    if (dbh->in_txn) {
579        zend_throw_exception_ex(php_pdo_get_exception(), 0, "There is already an active transaction");
580        RETURN_FALSE;
581    }
582
583    if (!dbh->methods->begin) {
584        /* TODO: this should be an exception; see the auto-commit mode
585         * comments below */
586        zend_throw_exception_ex(php_pdo_get_exception(), 0, "This driver doesn't support transactions");
587        RETURN_FALSE;
588    }
589
590    if (dbh->methods->begin(dbh)) {
591        dbh->in_txn = 1;
592        RETURN_TRUE;
593    }
594
595    PDO_HANDLE_DBH_ERR();
596    RETURN_FALSE;
597}
598/* }}} */
599
600/* {{{ proto bool PDO::commit()
601   Commit a transaction */
602static PHP_METHOD(PDO, commit)
603{
604    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
605
606    if (zend_parse_parameters_none() == FAILURE) {
607        return;
608    }
609    PDO_CONSTRUCT_CHECK;
610
611    if (!dbh->in_txn) {
612        zend_throw_exception_ex(php_pdo_get_exception(), 0, "There is no active transaction");
613        RETURN_FALSE;
614    }
615
616    if (dbh->methods->commit(dbh)) {
617        dbh->in_txn = 0;
618        RETURN_TRUE;
619    }
620
621    PDO_HANDLE_DBH_ERR();
622    RETURN_FALSE;
623}
624/* }}} */
625
626/* {{{ proto bool PDO::rollBack()
627   roll back a transaction */
628static PHP_METHOD(PDO, rollBack)
629{
630    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
631
632    if (zend_parse_parameters_none() == FAILURE) {
633        return;
634    }
635    PDO_CONSTRUCT_CHECK;
636
637    if (!dbh->in_txn) {
638        zend_throw_exception_ex(php_pdo_get_exception(), 0, "There is no active transaction");
639        RETURN_FALSE;
640    }
641
642    if (dbh->methods->rollback(dbh)) {
643        dbh->in_txn = 0;
644        RETURN_TRUE;
645    }
646
647    PDO_HANDLE_DBH_ERR();
648    RETURN_FALSE;
649}
650/* }}} */
651
652/* {{{ proto bool PDO::inTransaction()
653   determine if inside a transaction */
654static PHP_METHOD(PDO, inTransaction)
655{
656    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
657
658    if (zend_parse_parameters_none() == FAILURE) {
659        return;
660    }
661    PDO_CONSTRUCT_CHECK;
662
663    if (!dbh->methods->in_transaction) {
664        RETURN_BOOL(dbh->in_txn);
665    }
666
667    RETURN_BOOL(dbh->methods->in_transaction(dbh));
668}
669/* }}} */
670
671static int pdo_dbh_attribute_set(pdo_dbh_t *dbh, zend_long attr, zval *value) /* {{{ */
672{
673
674#define PDO_LONG_PARAM_CHECK \
675    if (Z_TYPE_P(value) != IS_LONG && Z_TYPE_P(value) != IS_STRING && Z_TYPE_P(value) != IS_FALSE && Z_TYPE_P(value) != IS_TRUE) { \
676        pdo_raise_impl_error(dbh, NULL, "HY000", "attribute value must be an integer"); \
677        PDO_HANDLE_DBH_ERR(); \
678        return FAILURE; \
679    } \
680
681    switch (attr) {
682        case PDO_ATTR_ERRMODE:
683            PDO_LONG_PARAM_CHECK;
684            convert_to_long(value);
685            switch (Z_LVAL_P(value)) {
686                case PDO_ERRMODE_SILENT:
687                case PDO_ERRMODE_WARNING:
688                case PDO_ERRMODE_EXCEPTION:
689                    dbh->error_mode = Z_LVAL_P(value);
690                    return SUCCESS;
691                default:
692                    pdo_raise_impl_error(dbh, NULL, "HY000", "invalid error mode");
693                    PDO_HANDLE_DBH_ERR();
694                    return FAILURE;
695            }
696            return FAILURE;
697
698        case PDO_ATTR_CASE:
699            PDO_LONG_PARAM_CHECK;
700            convert_to_long(value);
701            switch (Z_LVAL_P(value)) {
702                case PDO_CASE_NATURAL:
703                case PDO_CASE_UPPER:
704                case PDO_CASE_LOWER:
705                    dbh->desired_case = Z_LVAL_P(value);
706                    return SUCCESS;
707                default:
708                    pdo_raise_impl_error(dbh, NULL, "HY000", "invalid case folding mode");
709                    PDO_HANDLE_DBH_ERR();
710                    return FAILURE;
711            }
712            return FAILURE;
713
714        case PDO_ATTR_ORACLE_NULLS:
715            PDO_LONG_PARAM_CHECK;
716            convert_to_long(value);
717            dbh->oracle_nulls = Z_LVAL_P(value);
718            return SUCCESS;
719
720        case PDO_ATTR_DEFAULT_FETCH_MODE:
721            if (Z_TYPE_P(value) == IS_ARRAY) {
722                zval *tmp;
723                if ((tmp = zend_hash_index_find(Z_ARRVAL_P(value), 0)) != NULL && Z_TYPE_P(tmp) == IS_LONG) {
724                    if (Z_LVAL_P(tmp) == PDO_FETCH_INTO || Z_LVAL_P(tmp) == PDO_FETCH_CLASS) {
725                        pdo_raise_impl_error(dbh, NULL, "HY000", "FETCH_INTO and FETCH_CLASS are not yet supported as default fetch modes");
726                        return FAILURE;
727                    }
728                }
729            } else {
730                PDO_LONG_PARAM_CHECK;
731            }
732            convert_to_long(value);
733            if (Z_LVAL_P(value) == PDO_FETCH_USE_DEFAULT) {
734                pdo_raise_impl_error(dbh, NULL, "HY000", "invalid fetch mode type");
735                return FAILURE;
736            }
737            dbh->default_fetch_type = Z_LVAL_P(value);
738            return SUCCESS;
739
740        case PDO_ATTR_STRINGIFY_FETCHES:
741            PDO_LONG_PARAM_CHECK;
742            convert_to_long(value);
743            dbh->stringify = Z_LVAL_P(value) ? 1 : 0;
744            return SUCCESS;
745
746        case PDO_ATTR_STATEMENT_CLASS: {
747            /* array(string classname, array(mixed ctor_args)) */
748            zend_class_entry *pce;
749            zval *item;
750
751            if (dbh->is_persistent) {
752                pdo_raise_impl_error(dbh, NULL, "HY000",
753                    "PDO::ATTR_STATEMENT_CLASS cannot be used with persistent PDO instances"
754                    );
755                PDO_HANDLE_DBH_ERR();
756                return FAILURE;
757            }
758            if (Z_TYPE_P(value) != IS_ARRAY
759                || (item = zend_hash_index_find(Z_ARRVAL_P(value), 0)) == NULL
760                || Z_TYPE_P(item) != IS_STRING
761                || (pce = zend_lookup_class(Z_STR_P(item))) == NULL
762            ) {
763                pdo_raise_impl_error(dbh, NULL, "HY000",
764                    "PDO::ATTR_STATEMENT_CLASS requires format array(classname, array(ctor_args)); "
765                    "the classname must be a string specifying an existing class"
766                    );
767                PDO_HANDLE_DBH_ERR();
768                return FAILURE;
769            }
770            if (!instanceof_function(pce, pdo_dbstmt_ce)) {
771                pdo_raise_impl_error(dbh, NULL, "HY000",
772                    "user-supplied statement class must be derived from PDOStatement");
773                PDO_HANDLE_DBH_ERR();
774                return FAILURE;
775            }
776            if (pce->constructor && !(pce->constructor->common.fn_flags & (ZEND_ACC_PRIVATE|ZEND_ACC_PROTECTED))) {
777                pdo_raise_impl_error(dbh, NULL, "HY000",
778                    "user-supplied statement class cannot have a public constructor");
779                PDO_HANDLE_DBH_ERR();
780                return FAILURE;
781            }
782            dbh->def_stmt_ce = pce;
783            if (!Z_ISUNDEF(dbh->def_stmt_ctor_args)) {
784                zval_ptr_dtor(&dbh->def_stmt_ctor_args);
785                ZVAL_UNDEF(&dbh->def_stmt_ctor_args);
786            }
787            if ((item = zend_hash_index_find(Z_ARRVAL_P(value), 1)) != NULL) {
788                if (Z_TYPE_P(item) != IS_ARRAY) {
789                    pdo_raise_impl_error(dbh, NULL, "HY000",
790                        "PDO::ATTR_STATEMENT_CLASS requires format array(classname, array(ctor_args)); "
791                        "ctor_args must be an array"
792                    );
793                    PDO_HANDLE_DBH_ERR();
794                    return FAILURE;
795                }
796                ZVAL_COPY(&dbh->def_stmt_ctor_args, item);
797            }
798            return SUCCESS;
799        }
800
801        default:
802            ;
803    }
804
805    if (!dbh->methods->set_attribute) {
806        goto fail;
807    }
808
809    PDO_DBH_CLEAR_ERR();
810    if (dbh->methods->set_attribute(dbh, attr, value)) {
811        return SUCCESS;
812    }
813
814fail:
815    if (attr == PDO_ATTR_AUTOCOMMIT) {
816        zend_throw_exception_ex(php_pdo_get_exception(), 0, "The auto-commit mode cannot be changed for this driver");
817    } else if (!dbh->methods->set_attribute) {
818        pdo_raise_impl_error(dbh, NULL, "IM001", "driver does not support setting attributes");
819    } else {
820        PDO_HANDLE_DBH_ERR();
821    }
822    return FAILURE;
823}
824/* }}} */
825
826/* {{{ proto bool PDO::setAttribute(long attribute, mixed value)
827   Set an attribute */
828static PHP_METHOD(PDO, setAttribute)
829{
830    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
831    zend_long attr;
832    zval *value;
833
834    if (FAILURE == zend_parse_parameters(ZEND_NUM_ARGS(), "lz", &attr, &value)) {
835        RETURN_FALSE;
836    }
837
838    PDO_DBH_CLEAR_ERR();
839    PDO_CONSTRUCT_CHECK;
840
841    if (pdo_dbh_attribute_set(dbh, attr, value) != FAILURE) {
842        RETURN_TRUE;
843    }
844    RETURN_FALSE;
845}
846/* }}} */
847
848/* {{{ proto mixed PDO::getAttribute(long attribute)
849   Get an attribute */
850static PHP_METHOD(PDO, getAttribute)
851{
852    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
853    zend_long attr;
854
855    if (FAILURE == zend_parse_parameters(ZEND_NUM_ARGS(), "l", &attr)) {
856        RETURN_FALSE;
857    }
858
859    PDO_DBH_CLEAR_ERR();
860    PDO_CONSTRUCT_CHECK;
861
862    /* handle generic PDO-level attributes */
863    switch (attr) {
864        case PDO_ATTR_PERSISTENT:
865            RETURN_BOOL(dbh->is_persistent);
866
867        case PDO_ATTR_CASE:
868            RETURN_LONG(dbh->desired_case);
869
870        case PDO_ATTR_ORACLE_NULLS:
871            RETURN_LONG(dbh->oracle_nulls);
872
873        case PDO_ATTR_ERRMODE:
874            RETURN_LONG(dbh->error_mode);
875
876        case PDO_ATTR_DRIVER_NAME:
877            RETURN_STRINGL((char*)dbh->driver->driver_name, dbh->driver->driver_name_len);
878
879        case PDO_ATTR_STATEMENT_CLASS:
880            array_init(return_value);
881            add_next_index_str(return_value, zend_string_copy(dbh->def_stmt_ce->name));
882            if (!Z_ISUNDEF(dbh->def_stmt_ctor_args)) {
883                if (Z_REFCOUNTED(dbh->def_stmt_ctor_args)) Z_ADDREF(dbh->def_stmt_ctor_args);
884                add_next_index_zval(return_value, &dbh->def_stmt_ctor_args);
885            }
886            return;
887        case PDO_ATTR_DEFAULT_FETCH_MODE:
888            RETURN_LONG(dbh->default_fetch_type);
889        default:
890            break;
891    }
892
893    if (!dbh->methods->get_attribute) {
894        pdo_raise_impl_error(dbh, NULL, "IM001", "driver does not support getting attributes");
895        RETURN_FALSE;
896    }
897
898    switch (dbh->methods->get_attribute(dbh, attr, return_value)) {
899        case -1:
900            PDO_HANDLE_DBH_ERR();
901            RETURN_FALSE;
902
903        case 0:
904            pdo_raise_impl_error(dbh, NULL, "IM001", "driver does not support that attribute");
905            RETURN_FALSE;
906
907        default:
908            return;
909    }
910}
911/* }}} */
912
913/* {{{ proto long PDO::exec(string query)
914   Execute a query that does not return a row set, returning the number of affected rows */
915static PHP_METHOD(PDO, exec)
916{
917    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
918    char *statement;
919    size_t statement_len;
920    zend_long ret;
921
922    if (FAILURE == zend_parse_parameters(ZEND_NUM_ARGS(), "s", &statement, &statement_len)) {
923        RETURN_FALSE;
924    }
925
926    if (!statement_len) {
927        pdo_raise_impl_error(dbh, NULL, "HY000",  "trying to execute an empty query");
928        RETURN_FALSE;
929    }
930    PDO_DBH_CLEAR_ERR();
931    PDO_CONSTRUCT_CHECK;
932    ret = dbh->methods->doer(dbh, statement, statement_len);
933    if(ret == -1) {
934        PDO_HANDLE_DBH_ERR();
935        RETURN_FALSE;
936    } else {
937        RETURN_LONG(ret);
938    }
939}
940/* }}} */
941
942/* {{{ proto string PDO::lastInsertId([string seqname])
943   Returns the id of the last row that we affected on this connection.  Some databases require a sequence or table name to be passed in.  Not always meaningful. */
944static PHP_METHOD(PDO, lastInsertId)
945{
946    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
947    char *name = NULL;
948    size_t namelen;
949
950    if (FAILURE == zend_parse_parameters(ZEND_NUM_ARGS(), "|s!", &name, &namelen)) {
951        RETURN_FALSE;
952    }
953
954    PDO_DBH_CLEAR_ERR();
955    PDO_CONSTRUCT_CHECK;
956    if (!dbh->methods->last_id) {
957        pdo_raise_impl_error(dbh, NULL, "IM001", "driver does not support lastInsertId()");
958        RETURN_FALSE;
959    } else {
960        size_t id_len;
961        char *id;
962        id = dbh->methods->last_id(dbh, name, &id_len);
963        if (!id) {
964            PDO_HANDLE_DBH_ERR();
965            RETURN_FALSE;
966        } else {
967            //??? use zend_string ?
968            RETVAL_STRINGL(id, id_len);
969            efree(id);
970        }
971    }
972}
973/* }}} */
974
975/* {{{ proto string PDO::errorCode()
976   Fetch the error code associated with the last operation on the database handle */
977static PHP_METHOD(PDO, errorCode)
978{
979    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
980
981    if (zend_parse_parameters_none() == FAILURE) {
982        return;
983    }
984    PDO_CONSTRUCT_CHECK;
985
986    if (dbh->query_stmt) {
987        RETURN_STRING(dbh->query_stmt->error_code);
988    }
989
990    if (dbh->error_code[0] == '\0') {
991        RETURN_NULL();
992    }
993
994    /**
995     * Making sure that we fallback to the default implementation
996     * if the dbh->error_code is not null.
997     */
998    RETURN_STRING(dbh->error_code);
999}
1000/* }}} */
1001
1002/* {{{ proto int PDO::errorInfo()
1003   Fetch extended error information associated with the last operation on the database handle */
1004static PHP_METHOD(PDO, errorInfo)
1005{
1006    int error_count;
1007    int error_count_diff     = 0;
1008    int error_expected_count = 3;
1009
1010    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
1011
1012    if (zend_parse_parameters_none() == FAILURE) {
1013        return;
1014    }
1015
1016    PDO_CONSTRUCT_CHECK;
1017
1018    array_init(return_value);
1019
1020    if (dbh->query_stmt) {
1021        add_next_index_string(return_value, dbh->query_stmt->error_code);
1022    } else {
1023        add_next_index_string(return_value, dbh->error_code);
1024    }
1025
1026    if (dbh->methods->fetch_err) {
1027        dbh->methods->fetch_err(dbh, dbh->query_stmt, return_value);
1028    }
1029
1030    /**
1031     * In order to be consistent, we have to make sure we add the good amount
1032     * of nulls depending on the current number of elements. We make a simple
1033     * difference and add the needed elements
1034     */
1035    error_count = zend_hash_num_elements(Z_ARRVAL_P(return_value));
1036
1037    if (error_expected_count > error_count) {
1038        int current_index;
1039
1040        error_count_diff = error_expected_count - error_count;
1041        for (current_index = 0; current_index < error_count_diff; current_index++) {
1042            add_next_index_null(return_value);
1043        }
1044    }
1045}
1046/* }}} */
1047
1048/* {{{ proto object PDO::query(string sql [, PDOStatement::setFetchMode() args])
1049   Prepare and execute $sql; returns the statement object for iteration */
1050static PHP_METHOD(PDO, query)
1051{
1052    pdo_stmt_t *stmt;
1053    char *statement;
1054    size_t statement_len;
1055    pdo_dbh_object_t *dbh_obj = Z_PDO_OBJECT_P(getThis());
1056    pdo_dbh_t *dbh = dbh_obj->inner;
1057
1058    /* Return a meaningful error when no parameters were passed */
1059    if (!ZEND_NUM_ARGS()) {
1060        zend_parse_parameters(0, "z|z", NULL, NULL);
1061        RETURN_FALSE;
1062    }
1063
1064    if (FAILURE == zend_parse_parameters(1, "s", &statement,
1065            &statement_len)) {
1066        RETURN_FALSE;
1067    }
1068
1069    PDO_DBH_CLEAR_ERR();
1070    PDO_CONSTRUCT_CHECK;
1071
1072    if (!pdo_stmt_instantiate(dbh, return_value, dbh->def_stmt_ce, &dbh->def_stmt_ctor_args)) {
1073        if (EXPECTED(!EG(exception))) {
1074            pdo_raise_impl_error(dbh, NULL, "HY000", "failed to instantiate user supplied statement class");
1075        }
1076        return;
1077    }
1078    stmt = Z_PDO_STMT_P(return_value);
1079
1080    /* unconditionally keep this for later reference */
1081    stmt->query_string = estrndup(statement, statement_len);
1082    stmt->query_stringlen = statement_len;
1083
1084    stmt->default_fetch_type = dbh->default_fetch_type;
1085    stmt->active_query_string = stmt->query_string;
1086    stmt->active_query_stringlen = statement_len;
1087    stmt->dbh = dbh;
1088    /* give it a reference to me */
1089    ZVAL_OBJ(&stmt->database_object_handle, &dbh_obj->std);
1090    Z_ADDREF(stmt->database_object_handle);
1091    /* we haven't created a lazy object yet */
1092    ZVAL_UNDEF(&stmt->lazy_object_ref);
1093
1094    if (dbh->methods->preparer(dbh, statement, statement_len, stmt, NULL)) {
1095        PDO_STMT_CLEAR_ERR();
1096        if (ZEND_NUM_ARGS() == 1 || SUCCESS == pdo_stmt_setup_fetch_mode(INTERNAL_FUNCTION_PARAM_PASSTHRU, stmt, 1)) {
1097
1098            /* now execute the statement */
1099            PDO_STMT_CLEAR_ERR();
1100            if (stmt->methods->executer(stmt)) {
1101                int ret = 1;
1102                if (!stmt->executed) {
1103                    if (stmt->dbh->alloc_own_columns) {
1104                        ret = pdo_stmt_describe_columns(stmt);
1105                    }
1106                    stmt->executed = 1;
1107                }
1108                if (ret) {
1109                    pdo_stmt_construct(execute_data, stmt, return_value, dbh->def_stmt_ce, &dbh->def_stmt_ctor_args);
1110                    return;
1111                }
1112            }
1113        }
1114        /* something broke */
1115        dbh->query_stmt = stmt;
1116        ZVAL_COPY_VALUE(&dbh->query_stmt_zval, return_value);
1117        Z_DELREF(stmt->database_object_handle);
1118        ZVAL_UNDEF(&stmt->database_object_handle);
1119        PDO_HANDLE_STMT_ERR();
1120    } else {
1121        PDO_HANDLE_DBH_ERR();
1122        zval_ptr_dtor(return_value);
1123    }
1124
1125    RETURN_FALSE;
1126}
1127/* }}} */
1128
1129/* {{{ proto string PDO::quote(string string [, int paramtype])
1130   quotes string for use in a query.  The optional paramtype acts as a hint for drivers that have alternate quoting styles.  The default value is PDO_PARAM_STR */
1131static PHP_METHOD(PDO, quote)
1132{
1133    pdo_dbh_t *dbh = Z_PDO_DBH_P(getThis());
1134    char *str;
1135    size_t str_len;
1136    zend_long paramtype = PDO_PARAM_STR;
1137    char *qstr;
1138    size_t qlen;
1139
1140    if (FAILURE == zend_parse_parameters(ZEND_NUM_ARGS(), "s|l", &str, &str_len, &paramtype)) {
1141        RETURN_FALSE;
1142    }
1143
1144    PDO_DBH_CLEAR_ERR();
1145    PDO_CONSTRUCT_CHECK;
1146    if (!dbh->methods->quoter) {
1147        pdo_raise_impl_error(dbh, NULL, "IM001", "driver does not support quoting");
1148        RETURN_FALSE;
1149    }
1150
1151    if (dbh->methods->quoter(dbh, str, str_len, &qstr, &qlen, paramtype)) {
1152        RETVAL_STRINGL(qstr, qlen);
1153        efree(qstr);
1154        return;
1155    }
1156    PDO_HANDLE_DBH_ERR();
1157    RETURN_FALSE;
1158}
1159/* }}} */
1160
1161/* {{{ proto int PDO::__wakeup()
1162   Prevents use of a PDO instance that has been unserialized */
1163static PHP_METHOD(PDO, __wakeup)
1164{
1165    zend_throw_exception_ex(php_pdo_get_exception(), 0, "You cannot serialize or unserialize PDO instances");
1166}
1167/* }}} */
1168
1169/* {{{ proto int PDO::__sleep()
1170   Prevents serialization of a PDO instance */
1171static PHP_METHOD(PDO, __sleep)
1172{
1173    zend_throw_exception_ex(php_pdo_get_exception(), 0, "You cannot serialize or unserialize PDO instances");
1174}
1175/* }}} */
1176
1177/* {{{ proto array PDO::getAvailableDrivers()
1178   Return array of available PDO drivers */
1179static PHP_METHOD(PDO, getAvailableDrivers)
1180{
1181    pdo_driver_t *pdriver;
1182
1183    if (zend_parse_parameters_none() == FAILURE) {
1184        return;
1185    }
1186
1187    array_init(return_value);
1188
1189    ZEND_HASH_FOREACH_PTR(&pdo_driver_hash, pdriver) {
1190        add_next_index_stringl(return_value, (char*)pdriver->driver_name, pdriver->driver_name_len);
1191    } ZEND_HASH_FOREACH_END();
1192}
1193/* }}} */
1194
1195/* {{{ arginfo */
1196ZEND_BEGIN_ARG_INFO_EX(arginfo_pdo___construct, 0, 0, 1)
1197    ZEND_ARG_INFO(0, dsn)
1198    ZEND_ARG_INFO(0, username)
1199    ZEND_ARG_INFO(0, passwd)
1200    ZEND_ARG_INFO(0, options) /* array */
1201ZEND_END_ARG_INFO()
1202
1203ZEND_BEGIN_ARG_INFO_EX(arginfo_pdo_prepare, 0, 0, 1)
1204    ZEND_ARG_INFO(0, statement)
1205    ZEND_ARG_INFO(0, options) /* array */
1206ZEND_END_ARG_INFO()
1207
1208ZEND_BEGIN_ARG_INFO(arginfo_pdo_setattribute, 0)
1209    ZEND_ARG_INFO(0, attribute)
1210    ZEND_ARG_INFO(0, value)
1211ZEND_END_ARG_INFO()
1212
1213ZEND_BEGIN_ARG_INFO(arginfo_pdo_getattribute, 0)
1214    ZEND_ARG_INFO(0, attribute)
1215ZEND_END_ARG_INFO()
1216
1217ZEND_BEGIN_ARG_INFO(arginfo_pdo_exec, 0)
1218    ZEND_ARG_INFO(0, query)
1219ZEND_END_ARG_INFO()
1220
1221ZEND_BEGIN_ARG_INFO_EX(arginfo_pdo_lastinsertid, 0, 0, 0)
1222    ZEND_ARG_INFO(0, seqname)
1223ZEND_END_ARG_INFO()
1224
1225ZEND_BEGIN_ARG_INFO_EX(arginfo_pdo_quote, 0, 0, 1)
1226    ZEND_ARG_INFO(0, string)
1227    ZEND_ARG_INFO(0, paramtype)
1228ZEND_END_ARG_INFO()
1229
1230ZEND_BEGIN_ARG_INFO(arginfo_pdo__void, 0)
1231ZEND_END_ARG_INFO()
1232/* }}} */
1233
1234const zend_function_entry pdo_dbh_functions[] = /* {{{ */ {
1235    ZEND_MALIAS(PDO, __construct, dbh_constructor,  arginfo_pdo___construct,    ZEND_ACC_PUBLIC)
1236    PHP_ME(PDO, prepare,                arginfo_pdo_prepare,        ZEND_ACC_PUBLIC)
1237    PHP_ME(PDO, beginTransaction,       arginfo_pdo__void,         ZEND_ACC_PUBLIC)
1238    PHP_ME(PDO, commit,                 arginfo_pdo__void,         ZEND_ACC_PUBLIC)
1239    PHP_ME(PDO, rollBack,               arginfo_pdo__void,         ZEND_ACC_PUBLIC)
1240    PHP_ME(PDO, inTransaction,          arginfo_pdo__void,         ZEND_ACC_PUBLIC)
1241    PHP_ME(PDO, setAttribute,   arginfo_pdo_setattribute,   ZEND_ACC_PUBLIC)
1242    PHP_ME(PDO, exec,           arginfo_pdo_exec,       ZEND_ACC_PUBLIC)
1243    PHP_ME(PDO, query,          NULL,                   ZEND_ACC_PUBLIC)
1244    PHP_ME(PDO, lastInsertId,   arginfo_pdo_lastinsertid,   ZEND_ACC_PUBLIC)
1245    PHP_ME(PDO, errorCode,              arginfo_pdo__void,         ZEND_ACC_PUBLIC)
1246    PHP_ME(PDO, errorInfo,              arginfo_pdo__void,         ZEND_ACC_PUBLIC)
1247    PHP_ME(PDO, getAttribute,   arginfo_pdo_getattribute,   ZEND_ACC_PUBLIC)
1248    PHP_ME(PDO, quote,          arginfo_pdo_quote,      ZEND_ACC_PUBLIC)
1249    PHP_ME(PDO, __wakeup,               arginfo_pdo__void,         ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
1250    PHP_ME(PDO, __sleep,                arginfo_pdo__void,         ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
1251    PHP_ME(PDO, getAvailableDrivers,    arginfo_pdo__void,         ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
1252    {NULL, NULL, NULL}
1253};
1254/* }}} */
1255
1256static void cls_method_dtor(zval *el) /* {{{ */ {
1257    zend_function *func = (zend_function*)Z_PTR_P(el);
1258    if (func->common.function_name) {
1259        zend_string_release(func->common.function_name);
1260    }
1261    efree(func);
1262}
1263/* }}} */
1264
1265/* {{{ overloaded object handlers for PDO class */
1266int pdo_hash_methods(pdo_dbh_object_t *dbh_obj, int kind)
1267{
1268    const zend_function_entry *funcs;
1269    zend_function func;
1270    zend_internal_function *ifunc = (zend_internal_function*)&func;
1271    size_t namelen;
1272    char *lc_name;
1273    pdo_dbh_t *dbh = dbh_obj->inner;
1274
1275    if (!dbh || !dbh->methods || !dbh->methods->get_driver_methods) {
1276        return 0;
1277    }
1278    funcs = dbh->methods->get_driver_methods(dbh, kind);
1279    if (!funcs) {
1280        return 0;
1281    }
1282
1283    if (!(dbh->cls_methods[kind] = pemalloc(sizeof(HashTable), dbh->is_persistent))) {
1284        php_error_docref(NULL, E_ERROR, "out of memory while allocating PDO methods.");
1285    }
1286    zend_hash_init_ex(dbh->cls_methods[kind], 8, NULL, cls_method_dtor, dbh->is_persistent, 0);
1287
1288    while (funcs->fname) {
1289        ifunc->type = ZEND_INTERNAL_FUNCTION;
1290        ifunc->handler = funcs->handler;
1291        ifunc->function_name = zend_string_init(funcs->fname, strlen(funcs->fname), 0);
1292        ifunc->scope = dbh_obj->std.ce;
1293        ifunc->prototype = NULL;
1294        if (funcs->flags) {
1295            ifunc->fn_flags = funcs->flags | ZEND_ACC_NEVER_CACHE;
1296        } else {
1297            ifunc->fn_flags = ZEND_ACC_PUBLIC | ZEND_ACC_NEVER_CACHE;
1298        }
1299        if (funcs->arg_info) {
1300            zend_internal_function_info *info = (zend_internal_function_info*)funcs->arg_info;
1301
1302            ifunc->arg_info = (zend_internal_arg_info*)funcs->arg_info + 1;
1303            ifunc->num_args = funcs->num_args;
1304            if (info->required_num_args == -1) {
1305                ifunc->required_num_args = funcs->num_args;
1306            } else {
1307                ifunc->required_num_args = info->required_num_args;
1308            }
1309            if (info->return_reference) {
1310                ifunc->fn_flags |= ZEND_ACC_RETURN_REFERENCE;
1311            }
1312            if (funcs->arg_info[funcs->num_args].is_variadic) {
1313                ifunc->fn_flags |= ZEND_ACC_VARIADIC;
1314                /* Don't count the variadic argument */
1315                ifunc->num_args--;
1316            }
1317        } else {
1318            ifunc->arg_info = NULL;
1319            ifunc->num_args = 0;
1320            ifunc->required_num_args = 0;
1321        }
1322        namelen = strlen(funcs->fname);
1323        lc_name = emalloc(namelen+1);
1324        zend_str_tolower_copy(lc_name, funcs->fname, namelen);
1325        zend_hash_str_add_mem(dbh->cls_methods[kind], lc_name, namelen, &func, sizeof(func));
1326        efree(lc_name);
1327        funcs++;
1328    }
1329
1330    return 1;
1331}
1332
1333static union _zend_function *dbh_method_get(zend_object **object, zend_string *method_name, const zval *key)
1334{
1335    zend_function *fbc = NULL;
1336    pdo_dbh_object_t *dbh_obj = php_pdo_dbh_fetch_object(*object);
1337    zend_string *lc_method_name;
1338
1339    lc_method_name = zend_string_init(method_name->val, method_name->len, 0);
1340    zend_str_tolower_copy(lc_method_name->val, method_name->val, method_name->len);
1341
1342    if ((fbc = std_object_handlers.get_method(object, method_name, key)) == NULL) {
1343        /* not a pre-defined method, nor a user-defined method; check
1344         * the driver specific methods */
1345        if (!dbh_obj->inner->cls_methods[PDO_DBH_DRIVER_METHOD_KIND_DBH]) {
1346            if (!pdo_hash_methods(dbh_obj,
1347                PDO_DBH_DRIVER_METHOD_KIND_DBH)
1348                || !dbh_obj->inner->cls_methods[PDO_DBH_DRIVER_METHOD_KIND_DBH]) {
1349                goto out;
1350            }
1351        }
1352
1353        fbc = zend_hash_find_ptr(dbh_obj->inner->cls_methods[PDO_DBH_DRIVER_METHOD_KIND_DBH], lc_method_name);
1354    }
1355
1356out:
1357    zend_string_release(lc_method_name);
1358    return fbc;
1359}
1360
1361static int dbh_compare(zval *object1, zval *object2)
1362{
1363    return -1;
1364}
1365
1366static zend_object_handlers pdo_dbh_object_handlers;
1367static void pdo_dbh_free_storage(zend_object *std);
1368
1369void pdo_dbh_init(void)
1370{
1371    zend_class_entry ce;
1372
1373    INIT_CLASS_ENTRY(ce, "PDO", pdo_dbh_functions);
1374    pdo_dbh_ce = zend_register_internal_class(&ce);
1375    pdo_dbh_ce->create_object = pdo_dbh_new;
1376
1377    memcpy(&pdo_dbh_object_handlers, &std_object_handlers, sizeof(zend_object_handlers));
1378    pdo_dbh_object_handlers.offset = XtOffsetOf(pdo_dbh_object_t, std);
1379    pdo_dbh_object_handlers.dtor_obj = zend_objects_destroy_object;
1380    pdo_dbh_object_handlers.free_obj = pdo_dbh_free_storage;
1381    pdo_dbh_object_handlers.get_method = dbh_method_get;
1382    pdo_dbh_object_handlers.compare_objects = dbh_compare;
1383
1384    REGISTER_PDO_CLASS_CONST_LONG("PARAM_BOOL", (zend_long)PDO_PARAM_BOOL);
1385    REGISTER_PDO_CLASS_CONST_LONG("PARAM_NULL", (zend_long)PDO_PARAM_NULL);
1386    REGISTER_PDO_CLASS_CONST_LONG("PARAM_INT",  (zend_long)PDO_PARAM_INT);
1387    REGISTER_PDO_CLASS_CONST_LONG("PARAM_STR",  (zend_long)PDO_PARAM_STR);
1388    REGISTER_PDO_CLASS_CONST_LONG("PARAM_LOB",  (zend_long)PDO_PARAM_LOB);
1389    REGISTER_PDO_CLASS_CONST_LONG("PARAM_STMT", (zend_long)PDO_PARAM_STMT);
1390    REGISTER_PDO_CLASS_CONST_LONG("PARAM_INPUT_OUTPUT", (zend_long)PDO_PARAM_INPUT_OUTPUT);
1391
1392    REGISTER_PDO_CLASS_CONST_LONG("PARAM_EVT_ALLOC",        (zend_long)PDO_PARAM_EVT_ALLOC);
1393    REGISTER_PDO_CLASS_CONST_LONG("PARAM_EVT_FREE",         (zend_long)PDO_PARAM_EVT_FREE);
1394    REGISTER_PDO_CLASS_CONST_LONG("PARAM_EVT_EXEC_PRE",     (zend_long)PDO_PARAM_EVT_EXEC_PRE);
1395    REGISTER_PDO_CLASS_CONST_LONG("PARAM_EVT_EXEC_POST",    (zend_long)PDO_PARAM_EVT_EXEC_POST);
1396    REGISTER_PDO_CLASS_CONST_LONG("PARAM_EVT_FETCH_PRE",    (zend_long)PDO_PARAM_EVT_FETCH_PRE);
1397    REGISTER_PDO_CLASS_CONST_LONG("PARAM_EVT_FETCH_POST",   (zend_long)PDO_PARAM_EVT_FETCH_POST);
1398    REGISTER_PDO_CLASS_CONST_LONG("PARAM_EVT_NORMALIZE",    (zend_long)PDO_PARAM_EVT_NORMALIZE);
1399
1400    REGISTER_PDO_CLASS_CONST_LONG("FETCH_LAZY", (zend_long)PDO_FETCH_LAZY);
1401    REGISTER_PDO_CLASS_CONST_LONG("FETCH_ASSOC", (zend_long)PDO_FETCH_ASSOC);
1402    REGISTER_PDO_CLASS_CONST_LONG("FETCH_NUM",  (zend_long)PDO_FETCH_NUM);
1403    REGISTER_PDO_CLASS_CONST_LONG("FETCH_BOTH", (zend_long)PDO_FETCH_BOTH);
1404    REGISTER_PDO_CLASS_CONST_LONG("FETCH_OBJ",  (zend_long)PDO_FETCH_OBJ);
1405    REGISTER_PDO_CLASS_CONST_LONG("FETCH_BOUND", (zend_long)PDO_FETCH_BOUND);
1406    REGISTER_PDO_CLASS_CONST_LONG("FETCH_COLUMN", (zend_long)PDO_FETCH_COLUMN);
1407    REGISTER_PDO_CLASS_CONST_LONG("FETCH_CLASS", (zend_long)PDO_FETCH_CLASS);
1408    REGISTER_PDO_CLASS_CONST_LONG("FETCH_INTO", (zend_long)PDO_FETCH_INTO);
1409    REGISTER_PDO_CLASS_CONST_LONG("FETCH_FUNC", (zend_long)PDO_FETCH_FUNC);
1410    REGISTER_PDO_CLASS_CONST_LONG("FETCH_GROUP", (zend_long)PDO_FETCH_GROUP);
1411    REGISTER_PDO_CLASS_CONST_LONG("FETCH_UNIQUE", (zend_long)PDO_FETCH_UNIQUE);
1412    REGISTER_PDO_CLASS_CONST_LONG("FETCH_KEY_PAIR", (zend_long)PDO_FETCH_KEY_PAIR);
1413    REGISTER_PDO_CLASS_CONST_LONG("FETCH_CLASSTYPE", (zend_long)PDO_FETCH_CLASSTYPE);
1414
1415#if PHP_VERSION_ID >= 50100
1416    REGISTER_PDO_CLASS_CONST_LONG("FETCH_SERIALIZE",(zend_long)PDO_FETCH_SERIALIZE);
1417#endif
1418    REGISTER_PDO_CLASS_CONST_LONG("FETCH_PROPS_LATE", (zend_long)PDO_FETCH_PROPS_LATE);
1419    REGISTER_PDO_CLASS_CONST_LONG("FETCH_NAMED", (zend_long)PDO_FETCH_NAMED);
1420
1421    REGISTER_PDO_CLASS_CONST_LONG("ATTR_AUTOCOMMIT",    (zend_long)PDO_ATTR_AUTOCOMMIT);
1422    REGISTER_PDO_CLASS_CONST_LONG("ATTR_PREFETCH",      (zend_long)PDO_ATTR_PREFETCH);
1423    REGISTER_PDO_CLASS_CONST_LONG("ATTR_TIMEOUT",       (zend_long)PDO_ATTR_TIMEOUT);
1424    REGISTER_PDO_CLASS_CONST_LONG("ATTR_ERRMODE",       (zend_long)PDO_ATTR_ERRMODE);
1425    REGISTER_PDO_CLASS_CONST_LONG("ATTR_SERVER_VERSION",    (zend_long)PDO_ATTR_SERVER_VERSION);
1426    REGISTER_PDO_CLASS_CONST_LONG("ATTR_CLIENT_VERSION",    (zend_long)PDO_ATTR_CLIENT_VERSION);
1427    REGISTER_PDO_CLASS_CONST_LONG("ATTR_SERVER_INFO",       (zend_long)PDO_ATTR_SERVER_INFO);
1428    REGISTER_PDO_CLASS_CONST_LONG("ATTR_CONNECTION_STATUS",     (zend_long)PDO_ATTR_CONNECTION_STATUS);
1429    REGISTER_PDO_CLASS_CONST_LONG("ATTR_CASE",          (zend_long)PDO_ATTR_CASE);
1430    REGISTER_PDO_CLASS_CONST_LONG("ATTR_CURSOR_NAME",   (zend_long)PDO_ATTR_CURSOR_NAME);
1431    REGISTER_PDO_CLASS_CONST_LONG("ATTR_CURSOR",        (zend_long)PDO_ATTR_CURSOR);
1432    REGISTER_PDO_CLASS_CONST_LONG("ATTR_ORACLE_NULLS",  (zend_long)PDO_ATTR_ORACLE_NULLS);
1433    REGISTER_PDO_CLASS_CONST_LONG("ATTR_PERSISTENT",    (zend_long)PDO_ATTR_PERSISTENT);
1434    REGISTER_PDO_CLASS_CONST_LONG("ATTR_STATEMENT_CLASS",       (zend_long)PDO_ATTR_STATEMENT_CLASS);
1435    REGISTER_PDO_CLASS_CONST_LONG("ATTR_FETCH_TABLE_NAMES",     (zend_long)PDO_ATTR_FETCH_TABLE_NAMES);
1436    REGISTER_PDO_CLASS_CONST_LONG("ATTR_FETCH_CATALOG_NAMES",       (zend_long)PDO_ATTR_FETCH_CATALOG_NAMES);
1437    REGISTER_PDO_CLASS_CONST_LONG("ATTR_DRIVER_NAME",       (zend_long)PDO_ATTR_DRIVER_NAME);
1438    REGISTER_PDO_CLASS_CONST_LONG("ATTR_STRINGIFY_FETCHES", (zend_long)PDO_ATTR_STRINGIFY_FETCHES);
1439    REGISTER_PDO_CLASS_CONST_LONG("ATTR_MAX_COLUMN_LEN", (zend_long)PDO_ATTR_MAX_COLUMN_LEN);
1440    REGISTER_PDO_CLASS_CONST_LONG("ATTR_EMULATE_PREPARES", (zend_long)PDO_ATTR_EMULATE_PREPARES);
1441    REGISTER_PDO_CLASS_CONST_LONG("ATTR_DEFAULT_FETCH_MODE", (zend_long)PDO_ATTR_DEFAULT_FETCH_MODE);
1442
1443    REGISTER_PDO_CLASS_CONST_LONG("ERRMODE_SILENT", (zend_long)PDO_ERRMODE_SILENT);
1444    REGISTER_PDO_CLASS_CONST_LONG("ERRMODE_WARNING",    (zend_long)PDO_ERRMODE_WARNING);
1445    REGISTER_PDO_CLASS_CONST_LONG("ERRMODE_EXCEPTION",  (zend_long)PDO_ERRMODE_EXCEPTION);
1446
1447    REGISTER_PDO_CLASS_CONST_LONG("CASE_NATURAL",   (zend_long)PDO_CASE_NATURAL);
1448    REGISTER_PDO_CLASS_CONST_LONG("CASE_LOWER", (zend_long)PDO_CASE_LOWER);
1449    REGISTER_PDO_CLASS_CONST_LONG("CASE_UPPER", (zend_long)PDO_CASE_UPPER);
1450
1451    REGISTER_PDO_CLASS_CONST_LONG("NULL_NATURAL",   (zend_long)PDO_NULL_NATURAL);
1452    REGISTER_PDO_CLASS_CONST_LONG("NULL_EMPTY_STRING",  (zend_long)PDO_NULL_EMPTY_STRING);
1453    REGISTER_PDO_CLASS_CONST_LONG("NULL_TO_STRING", (zend_long)PDO_NULL_TO_STRING);
1454
1455    REGISTER_PDO_CLASS_CONST_STRING("ERR_NONE", PDO_ERR_NONE);
1456
1457    REGISTER_PDO_CLASS_CONST_LONG("FETCH_ORI_NEXT", (zend_long)PDO_FETCH_ORI_NEXT);
1458    REGISTER_PDO_CLASS_CONST_LONG("FETCH_ORI_PRIOR", (zend_long)PDO_FETCH_ORI_PRIOR);
1459    REGISTER_PDO_CLASS_CONST_LONG("FETCH_ORI_FIRST", (zend_long)PDO_FETCH_ORI_FIRST);
1460    REGISTER_PDO_CLASS_CONST_LONG("FETCH_ORI_LAST", (zend_long)PDO_FETCH_ORI_LAST);
1461    REGISTER_PDO_CLASS_CONST_LONG("FETCH_ORI_ABS", (zend_long)PDO_FETCH_ORI_ABS);
1462    REGISTER_PDO_CLASS_CONST_LONG("FETCH_ORI_REL", (zend_long)PDO_FETCH_ORI_REL);
1463
1464    REGISTER_PDO_CLASS_CONST_LONG("CURSOR_FWDONLY", (zend_long)PDO_CURSOR_FWDONLY);
1465    REGISTER_PDO_CLASS_CONST_LONG("CURSOR_SCROLL", (zend_long)PDO_CURSOR_SCROLL);
1466
1467#if 0
1468    REGISTER_PDO_CLASS_CONST_LONG("ERR_CANT_MAP",       (zend_long)PDO_ERR_CANT_MAP);
1469    REGISTER_PDO_CLASS_CONST_LONG("ERR_SYNTAX",         (zend_long)PDO_ERR_SYNTAX);
1470    REGISTER_PDO_CLASS_CONST_LONG("ERR_CONSTRAINT",     (zend_long)PDO_ERR_CONSTRAINT);
1471    REGISTER_PDO_CLASS_CONST_LONG("ERR_NOT_FOUND",      (zend_long)PDO_ERR_NOT_FOUND);
1472    REGISTER_PDO_CLASS_CONST_LONG("ERR_ALREADY_EXISTS",     (zend_long)PDO_ERR_ALREADY_EXISTS);
1473    REGISTER_PDO_CLASS_CONST_LONG("ERR_NOT_IMPLEMENTED",    (zend_long)PDO_ERR_NOT_IMPLEMENTED);
1474    REGISTER_PDO_CLASS_CONST_LONG("ERR_MISMATCH",       (zend_long)PDO_ERR_MISMATCH);
1475    REGISTER_PDO_CLASS_CONST_LONG("ERR_TRUNCATED",      (zend_long)PDO_ERR_TRUNCATED);
1476    REGISTER_PDO_CLASS_CONST_LONG("ERR_DISCONNECTED",   (zend_long)PDO_ERR_DISCONNECTED);
1477    REGISTER_PDO_CLASS_CONST_LONG("ERR_NO_PERM",        (zend_long)PDO_ERR_NO_PERM);
1478#endif
1479
1480}
1481
1482static void dbh_free(pdo_dbh_t *dbh, zend_bool free_persistent)
1483{
1484    int i;
1485
1486    if (dbh->is_persistent && !free_persistent) {
1487        return;
1488    }
1489
1490    if (dbh->query_stmt) {
1491        zval_ptr_dtor(&dbh->query_stmt_zval);
1492        dbh->query_stmt = NULL;
1493    }
1494
1495    if (dbh->methods) {
1496        dbh->methods->closer(dbh);
1497    }
1498
1499    if (dbh->data_source) {
1500        pefree((char *)dbh->data_source, dbh->is_persistent);
1501    }
1502    if (dbh->username) {
1503        pefree(dbh->username, dbh->is_persistent);
1504    }
1505    if (dbh->password) {
1506        pefree(dbh->password, dbh->is_persistent);
1507    }
1508
1509    if (dbh->persistent_id) {
1510        pefree((char *)dbh->persistent_id, dbh->is_persistent);
1511    }
1512
1513    if (!Z_ISUNDEF(dbh->def_stmt_ctor_args)) {
1514        zval_ptr_dtor(&dbh->def_stmt_ctor_args);
1515    }
1516
1517    for (i = 0; i < PDO_DBH_DRIVER_METHOD_KIND__MAX; i++) {
1518        if (dbh->cls_methods[i]) {
1519            zend_hash_destroy(dbh->cls_methods[i]);
1520            pefree(dbh->cls_methods[i], dbh->is_persistent);
1521        }
1522    }
1523
1524    pefree(dbh, dbh->is_persistent);
1525}
1526
1527static void pdo_dbh_free_storage(zend_object *std)
1528{
1529    pdo_dbh_t *dbh = php_pdo_dbh_fetch_inner(std);
1530    if (dbh->in_txn && dbh->methods && dbh->methods->rollback) {
1531        dbh->methods->rollback(dbh);
1532        dbh->in_txn = 0;
1533    }
1534
1535    if (dbh->is_persistent && dbh->methods && dbh->methods->persistent_shutdown) {
1536        dbh->methods->persistent_shutdown(dbh);
1537    }
1538    zend_object_std_dtor(std);
1539    dbh_free(dbh, 0);
1540}
1541
1542zend_object *pdo_dbh_new(zend_class_entry *ce)
1543{
1544    pdo_dbh_object_t *dbh;
1545
1546    dbh = ecalloc(1, sizeof(pdo_dbh_object_t) + zend_object_properties_size(ce));
1547    zend_object_std_init(&dbh->std, ce);
1548    object_properties_init(&dbh->std, ce);
1549    rebuild_object_properties(&dbh->std);
1550    dbh->inner = ecalloc(1, sizeof(pdo_dbh_t));
1551    dbh->inner->def_stmt_ce = pdo_dbstmt_ce;
1552
1553    dbh->std.handlers = &pdo_dbh_object_handlers;
1554
1555    return &dbh->std;
1556}
1557
1558/* }}} */
1559
1560ZEND_RSRC_DTOR_FUNC(php_pdo_pdbh_dtor) /* {{{ */
1561{
1562    if (res->ptr) {
1563        pdo_dbh_t *dbh = (pdo_dbh_t*)res->ptr;
1564        dbh_free(dbh, 1);
1565        res->ptr = NULL;
1566    }
1567}
1568/* }}} */
1569
1570/*
1571 * Local variables:
1572 * tab-width: 4
1573 * c-basic-offset: 4
1574 * End:
1575 * vim600: noet sw=4 ts=4 fdm=marker
1576 * vim<600: noet sw=4 ts=4
1577 */
1578