1--TEST--
2MySQL PDO->exec(), affected rows
3--EXTENSIONS--
4pdo_mysql
5--SKIPIF--
6<?php
7require_once __DIR__ . '/inc/mysql_pdo_test.inc';
8MySQLPDOTest::skip();
9
10// Run test only locally - not against remote hosts
11$db = MySQLPDOTest::factory();
12$stmt = $db->query('SELECT USER() as _user');
13$row = $stmt->fetch(PDO::FETCH_ASSOC);
14$tmp = explode('@', $row['_user']);
15if (count($tmp) < 2)
16    die("skip Cannot detect if test is run against local or remote database server");
17if (($tmp[1] !== 'localhost') && ($tmp[1] !== '127.0.0.1'))
18    die("skip Test cannot be run against remote database server");
19
20$stmt = $db->query("SHOW VARIABLES LIKE 'secure_file_priv'");
21if (($row = $stmt->fetch(PDO::FETCH_ASSOC)) && ($row['value'] != '')) {
22    if (!is_writable($row['value']))
23        die("skip secure_file_priv directory not writable: {$row['value']}");
24
25    $filename = $row['value'] . DIRECTORY_SEPARATOR  . "pdo_mysql_exec_load_data.csv";
26
27    if (file_exists($filename) && !is_writable($filename))
28        die("skip {$filename} not writable");
29}
30?>
31--FILE--
32<?php
33function exec_and_count($offset, &$db, $sql, $exp) {
34    try {
35        $ret = $db->exec($sql);
36        if ($ret !== $exp) {
37            printf("[%03d] Expecting '%s'/%s got '%s'/%s when running '%s', [%s] %s\n",
38                $offset, $exp, gettype($exp), $ret, gettype($ret), $sql,
39                $db->errorCode(), implode(' ', $db->errorInfo()));
40            return false;
41        }
42    } catch (PDOException $e) {
43        if (42000 == $db->errorCode()) {
44            // Error: 1148 SQLSTATE: 42000  (ER_NOT_ALLOWED_COMMAND)
45            // Load data infile not allowed
46            return false;
47        }
48
49        printf("[%03d] '%s' has failed, [%s] %s\n",
50            $offset, $sql, $db->errorCode(), implode(' ', $db->errorInfo()));
51        return false;
52    }
53
54    return true;
55}
56
57require_once __DIR__ . '/inc/mysql_pdo_test.inc';
58$db = MySQLPDOTest::factoryWithAttr([PDO::MYSQL_ATTR_LOCAL_INFILE=>true]);
59
60/* affected rows related */
61
62exec_and_count(3, $db, sprintf('CREATE TABLE test_mysql_exec_load_data(id INT NOT NULL PRIMARY KEY, col1 CHAR(10)) ENGINE=%s', PDO_MYSQL_TEST_ENGINE), 0);
63
64$stmt = $db->query("SHOW VARIABLES LIKE 'secure_file_priv'");
65if (($row = $stmt->fetch(PDO::FETCH_ASSOC)) && ($row['value'] != '')) {
66    $filename = $row['value'] . DIRECTORY_SEPARATOR  . "pdo_mysql_exec_load_data.csv";
67} else {
68    $filename =  MySQLPDOTest::getTempDir() . DIRECTORY_SEPARATOR  . "pdo_mysql_exec_load_data.csv";
69}
70
71$fp = fopen($filename, "w");
72fwrite($fp, "1;foo\n");
73fwrite($fp, "2;bar");
74fclose($fp);
75
76$sql = sprintf("LOAD DATA LOCAL INFILE %s INTO TABLE test_mysql_exec_load_data FIELDS TERMINATED BY ';' LINES TERMINATED  BY '\n'", $db->quote($filename));
77
78if (exec_and_count(4, $db, $sql, 2)) {
79    $stmt = $db->query('SELECT id, col1 FROM test_mysql_exec_load_data ORDER BY id ASC');
80    $expected = array(array("id" => 1, "col1" => "foo"), array("id" => 2, "col1" => "bar"));
81    $ret = $stmt->fetchAll(PDO::FETCH_ASSOC);
82    foreach ($expected as $offset => $exp) {
83        foreach ($exp as $key => $value) {
84            if ($ret[$offset][$key] != $value) {
85                printf("Results seem wrong, check manually\n");
86                var_dump($ret);
87                var_dump($expected);
88                break 2;
89            }
90        }
91    }
92}
93
94unlink($filename);
95
96print "done!";
97?>
98--CLEAN--
99<?php
100require __DIR__ . '/inc/mysql_pdo_test.inc';
101$db = MySQLPDOTest::factory();
102$db->query('DROP TABLE IF EXISTS test_mysql_exec_load_data');
103?>
104--EXPECT--
105done!
106