1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
<?php
/* Utility function for mysqli_set_local_infile*.phpt tests */
function shutdown_clean($file) {
if ($file) {
unlink($file);
}
}
function create_standard_csv($offset) {
// create a CVS file
$file = tempnam(sys_get_temp_dir(), 'mysqli_test');
if (!$fp = fopen($file, 'w')) {
printf("[%03d + 1] Cannot create CVS file '%s'\n", $offset, $file);
return NULL;
} else {
/* Looks ugly? No, handy if you have crashes... */
register_shutdown_function("shutdown_clean", $file);
}
if ((version_compare(PHP_VERSION, '5.9.9', '>') == 1)) {
if (!fwrite($fp, (binary)"'97';'x';\n") ||
!fwrite($fp, (binary)"'98';'y';\n") ||
!fwrite($fp, (binary)"99;'z';\n")) {
printf("[%03d + 2] Cannot write CVS file '%s'\n", $offset, $file);
return NULL;
}
} else {
if (!fwrite($fp, "97;'x';\n") ||
!fwrite($fp, "98;'y';\n") ||
!fwrite($fp, "99;'z';\n")) {
printf("[%03d + 3] Cannot write CVS file '%s'\n", $offset, $file);
return NULL;
}
}
fclose($fp);
if (!chmod($file, 0644)) {
printf("[%03d + 4] Cannot change the file perms of '%s' from 0600 to 0644, MySQL might not be able to read it\n",
$offset, $file);
return NULL;
}
return $file;
}
function try_handler($offset, $link, $file, $handler, $expected = null) {
if ('default' == $handler) {
mysqli_set_local_infile_default($link);
} else if (!mysqli_set_local_infile_handler($link, $handler)) {
printf("[%03d] Cannot set infile handler to '%s'\n", $offset, $handler);
return false;
}
printf("Callback set to '%s'\n", $handler);
if (!mysqli_query($link, sprintf("DELETE FROM test"))) {
printf("[%03d] Cannot remove records, [%d] %s\n", $offset + 1, mysqli_errno($link), mysqli_error($link));
return false;
}
if (!@mysqli_query($link, sprintf("LOAD DATA LOCAL INFILE '%s'
INTO TABLE test
FIELDS TERMINATED BY ';' OPTIONALLY ENCLOSED BY '\''
LINES TERMINATED BY '\n'",
mysqli_real_escape_string($link, $file)))) {
printf("[%03d] LOAD DATA failed, [%d] %s\n",
$offset + 2,
mysqli_errno($link), mysqli_error($link));
}
if (!$res = mysqli_query($link, "SELECT id, label FROM test ORDER BY id")) {
printf("[%03d] [%d] %s\n", $offset + 3, mysqli_errno($link), mysqli_error($link));
return false;
}
if (!is_array($expected))
return true;
foreach ($expected as $k => $values) {
if (!$tmp = mysqli_fetch_assoc($res)) {
printf("[%03d/%d] [%d] '%s'\n", $offset + 4, $k, mysqli_errno($link), mysqli_error($link));
return false;
}
if ($values['id'] != $tmp['id']) {
printf("[%03d/%d] Expecting %s got %s\n",
$offset + 5, $k,
$values['id'], $tmp['id']);
return false;
}
if ($values['label'] != $tmp['label']) {
printf("[%03d/%d] Expecting %s got %s\n",
$offset + 6, $k,
$values['label'], $tmp['label']);
return false;
}
}
if ($res && $tmp = mysqli_fetch_assoc($res)) {
printf("[%03d] More results than expected!\n", $offset + 7);
do {
var_dump($tmp);
} while ($tmp = mysqli_fetch_assoc($res));
return false;
}
if ($res)
mysqli_free_result($res);
return true;
}
?>
|
