Annotation of embedaddon/php/ext/mysqli/tests/local_infile_tools.inc, revision 1.1.1.1
1.1 misho 1: <?php
2: /* Utility function for mysqli_set_local_infile*.phpt tests */
3: function shutdown_clean($file) {
4: if ($file) {
5: unlink($file);
6: }
7: }
8:
9: function check_local_infile_support($link, $engine, $table_name = 'test') {
10:
11: if (!$res = mysqli_query($link, 'SHOW VARIABLES LIKE "local_infile"'))
12: return "Cannot check if Server variable 'local_infile' is set to 'ON'";
13:
14: $row = mysqli_fetch_assoc($res);
15: mysqli_free_result($res);
16: if ('ON' != $row['Value'])
17: return sprintf("Server variable 'local_infile' seems not set to 'ON', found '%s'", $row['Value']);
18:
19: if (!mysqli_query($link, sprintf('DROP TABLE IF EXISTS %s', $table_name))) {
20: return "Failed to drop old test table";
21: }
22:
23: if (!mysqli_query($link, $sql = sprintf('CREATE TABLE %s(id INT, label CHAR(1), PRIMARY KEY(id)) ENGINE=%s',
24: $table_name, $engine)))
25: return "Failed to create test table: $sql";
26:
27: $file = create_standard_csv(1, false);
28: if (!$file) {
29: mysqli_query($link, sprintf('DROP TABLE IF EXISTS %s', $table_name));
30: return "Cannot create CSV file";
31: }
32:
33: if (!@mysqli_query($link, sprintf("LOAD DATA LOCAL INFILE '%s'
34: INTO TABLE %s
35: FIELDS TERMINATED BY ';' OPTIONALLY ENCLOSED BY '\''
36: LINES TERMINATED BY '\n'",
37: mysqli_real_escape_string($link, $file),
38: $table_name))) {
39: if (1148 == mysqli_errno($link)) {
40: mysqli_query($link, sprintf('DROP TABLE IF EXISTS %s', $table_name));
41: return "Cannot test LOAD DATA LOCAL INFILE, [1148] The used command is not allowed with this MySQL version";
42: } else if ($link->errno) {
43: return $link->error;
44: }
45: }
46: mysqli_query($link, sprintf('DROP TABLE IF EXISTS %s', $table_name));
47: return "";
48: }
49:
50: function create_standard_csv($offset, $verbose = true) {
51: // create a CVS file
52: $file = tempnam(sys_get_temp_dir(), 'mysqli_test');
53: if (!$fp = fopen($file, 'w')) {
54: if ($verbose)
55: printf("[%03d + 1] Cannot create CVS file '%s'\n", $offset, $file);
56: return NULL;
57: } else {
58: /* Looks ugly? No, handy if you have crashes... */
59: register_shutdown_function("shutdown_clean", $file);
60: }
61:
62: if ((version_compare(PHP_VERSION, '5.9.9', '>') == 1)) {
63: if (!fwrite($fp, (binary)"'97';'x';\n") ||
64: !fwrite($fp, (binary)"'98';'y';\n") ||
65: !fwrite($fp, (binary)"99;'z';\n")) {
66: if ($verbose)
67: printf("[%03d + 2] Cannot write CVS file '%s'\n", $offset, $file);
68: return NULL;
69: }
70: } else {
71: if (!fwrite($fp, "97;'x';\n") ||
72: !fwrite($fp, "98;'y';\n") ||
73: !fwrite($fp, "99;'z';\n")) {
74: if ($verbose)
75: printf("[%03d + 3] Cannot write CVS file '%s'\n", $offset, $file);
76: return NULL;
77: }
78: }
79:
80: fclose($fp);
81:
82: if (!chmod($file, 0644)) {
83: if ($verbose)
84: printf("[%03d + 4] Cannot change the file perms of '%s' from 0600 to 0644, MySQL might not be able to read it\n",
85: $offset, $file);
86: return NULL;
87: }
88: return $file;
89: }
90:
91: function try_handler($offset, $link, $file, $handler, $expected = null) {
92:
93: if ('default' == $handler) {
94: mysqli_set_local_infile_default($link);
95: } else if (!mysqli_set_local_infile_handler($link, $handler)) {
96: printf("[%03d] Cannot set infile handler to '%s'\n", $offset, $handler);
97: return false;
98: }
99: printf("Callback set to '%s'\n", $handler);
100:
101: if (!mysqli_query($link, sprintf("DELETE FROM test"))) {
102: printf("[%03d] Cannot remove records, [%d] %s\n", $offset + 1, mysqli_errno($link), mysqli_error($link));
103: return false;
104: }
105:
106: if (!@mysqli_query($link, sprintf("LOAD DATA LOCAL INFILE '%s'
107: INTO TABLE test
108: FIELDS TERMINATED BY ';' OPTIONALLY ENCLOSED BY '\''
109: LINES TERMINATED BY '\n'",
110: mysqli_real_escape_string($link, $file)))) {
111: printf("[%03d] LOAD DATA failed, [%d] %s\n",
112: $offset + 2,
113: mysqli_errno($link), mysqli_error($link));
114: }
115:
116: if (!$res = mysqli_query($link, "SELECT id, label FROM test ORDER BY id")) {
117: printf("[%03d] [%d] %s\n", $offset + 3, mysqli_errno($link), mysqli_error($link));
118: return false;
119: }
120:
121: if (!is_array($expected))
122: return true;
123:
124: foreach ($expected as $k => $values) {
125: if (!$tmp = mysqli_fetch_assoc($res)) {
126: printf("[%03d/%d] [%d] '%s'\n", $offset + 4, $k, mysqli_errno($link), mysqli_error($link));
127: return false;
128: }
129: if ($values['id'] != $tmp['id']) {
130: printf("[%03d/%d] Expecting %s got %s\n",
131: $offset + 5, $k,
132: $values['id'], $tmp['id']);
133: return false;
134: }
135: if ($values['label'] != $tmp['label']) {
136: printf("[%03d/%d] Expecting %s got %s\n",
137: $offset + 6, $k,
138: $values['label'], $tmp['label']);
139: return false;
140: }
141: }
142:
143: if ($res && $tmp = mysqli_fetch_assoc($res)) {
144: printf("[%03d] More results than expected!\n", $offset + 7);
145: do {
146: var_dump($tmp);
147: } while ($tmp = mysqli_fetch_assoc($res));
148: return false;
149: }
150:
151: if ($res)
152: mysqli_free_result($res);
153:
154: return true;
155: }
156: ?>
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>