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