Line data Source code
1 : /*
2 : Unix SMB/CIFS implementation.
3 : SMB torture tester - unicode table dumper
4 : Copyright (C) Andrew Tridgell 2001
5 :
6 : This program is free software; you can redistribute it and/or modify
7 : it under the terms of the GNU General Public License as published by
8 : the Free Software Foundation; either version 3 of the License, or
9 : (at your option) any later version.
10 :
11 : This program is distributed in the hope that it will be useful,
12 : but WITHOUT ANY WARRANTY; without even the implied warranty of
13 : MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 : GNU General Public License for more details.
15 :
16 : You should have received a copy of the GNU General Public License
17 : along with this program. If not, see <http://www.gnu.org/licenses/>.
18 : */
19 :
20 : #include "includes.h"
21 : #include "system/filesys.h"
22 : #include "torture/proto.h"
23 : #include "../libcli/security/security.h"
24 : #include "libsmb/libsmb.h"
25 : #include "libsmb/clirap.h"
26 : #include "lib/util/string_wrappers.h"
27 :
28 0 : bool torture_utable(int dummy)
29 : {
30 : struct cli_state *cli;
31 : fstring fname, alt_name;
32 : uint16_t fnum;
33 : smb_ucs2_t c2;
34 : int c, len, fd;
35 0 : int chars_allowed=0, alt_allowed=0;
36 : uint8_t valid[0x10000];
37 :
38 0 : printf("starting utable\n");
39 :
40 0 : if (!torture_open_connection(&cli, 0)) {
41 0 : return False;
42 : }
43 :
44 0 : memset(valid, 0, sizeof(valid));
45 :
46 0 : cli_mkdir(cli, "\\utable");
47 0 : cli_unlink(cli, "\\utable\\*", FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN);
48 :
49 0 : for (c=1; c < 0x10000; c++) {
50 0 : size_t size = 0;
51 : char *p;
52 :
53 0 : SSVAL(&c2, 0, c);
54 0 : fstrcpy(fname, "\\utable\\x");
55 0 : p = fname+strlen(fname);
56 0 : if (!convert_string(CH_UTF16LE, CH_UNIX,
57 : &c2, 2,
58 0 : p, sizeof(fname)-strlen(fname),&size)) {
59 0 : d_printf("convert_string %s failed !\n", fname);
60 0 : continue;
61 : }
62 0 : len = size;
63 0 : p[len] = 0;
64 0 : fstrcat(fname,"_a_long_extension");
65 :
66 0 : if (!NT_STATUS_IS_OK(cli_openx(cli, fname, O_RDWR | O_CREAT | O_TRUNC,
67 : DENY_NONE, &fnum))) {
68 0 : continue;
69 : }
70 :
71 0 : chars_allowed++;
72 :
73 0 : cli_qpathinfo_alt_name(cli, fname, alt_name);
74 :
75 0 : if (strncmp(alt_name, "X_A_L", 5) != 0) {
76 0 : alt_allowed++;
77 0 : valid[c] = 1;
78 0 : d_printf("fname=[%s] alt_name=[%s]\n", fname, alt_name);
79 : }
80 :
81 0 : cli_close(cli, fnum);
82 0 : cli_unlink(cli, fname, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN);
83 :
84 0 : if (c % 100 == 0) {
85 0 : printf("%d (%d/%d)\r", c, chars_allowed, alt_allowed);
86 : }
87 : }
88 0 : printf("%d (%d/%d)\n", c, chars_allowed, alt_allowed);
89 :
90 0 : cli_rmdir(cli, "\\utable");
91 :
92 0 : d_printf("%d chars allowed %d alt chars allowed\n", chars_allowed, alt_allowed);
93 :
94 0 : fd = open("valid.dat", O_WRONLY|O_CREAT|O_TRUNC, 0644);
95 0 : if (fd == -1) {
96 0 : d_printf("Failed to create valid.dat - %s", strerror(errno));
97 0 : return False;
98 : }
99 0 : if (write(fd, valid, 0x10000) != 0x10000) {
100 0 : d_printf("Failed to create valid.dat - %s", strerror(errno));
101 0 : close(fd);
102 0 : return false;
103 : }
104 0 : close(fd);
105 0 : d_printf("wrote valid.dat\n");
106 :
107 0 : return True;
108 : }
109 :
110 :
111 0 : static char *form_name(int c)
112 : {
113 : static fstring fname;
114 : smb_ucs2_t c2;
115 : char *p;
116 0 : size_t len = 0;
117 :
118 0 : fstrcpy(fname, "\\utable\\");
119 0 : p = fname+strlen(fname);
120 0 : SSVAL(&c2, 0, c);
121 :
122 0 : if (!convert_string(CH_UTF16LE, CH_UNIX,
123 : &c2, 2,
124 0 : p, sizeof(fname)-strlen(fname), &len)) {
125 0 : d_printf("form_name: convert string %s failed\n",
126 : fname);
127 0 : return NULL;
128 : }
129 0 : p[len] = 0;
130 0 : return fname;
131 : }
132 :
133 0 : bool torture_casetable(int dummy)
134 : {
135 : static struct cli_state *cli;
136 : char *fname;
137 : uint16_t fnum;
138 : int c, i;
139 : #define MAX_EQUIVALENCE 8
140 : smb_ucs2_t equiv[0x10000][MAX_EQUIVALENCE];
141 0 : printf("starting casetable\n");
142 :
143 0 : if (!torture_open_connection(&cli, 0)) {
144 0 : return False;
145 : }
146 :
147 0 : memset(equiv, 0, sizeof(equiv));
148 :
149 0 : cli_unlink(cli, "\\utable\\*", FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN);
150 0 : cli_rmdir(cli, "\\utable");
151 0 : if (!NT_STATUS_IS_OK(cli_mkdir(cli, "\\utable"))) {
152 0 : printf("Failed to create utable directory!\n");
153 0 : return False;
154 : }
155 :
156 0 : for (c=1; c < 0x10000; c++) {
157 : off_t size;
158 :
159 0 : if (c == '.' || c == '\\') continue;
160 :
161 0 : printf("%04x (%c)\n", c, isprint(c)?c:'.');
162 :
163 0 : fname = form_name(c);
164 0 : if (!NT_STATUS_IS_OK(cli_ntcreate(cli, fname, 0,
165 : GENERIC_ALL_ACCESS,
166 : FILE_ATTRIBUTE_NORMAL,
167 : FILE_SHARE_NONE,
168 : FILE_OPEN_IF, 0, 0, &fnum, NULL))) {
169 0 : printf("Failed to create file with char %04x\n", c);
170 0 : continue;
171 : }
172 :
173 0 : size = 0;
174 :
175 0 : if (!NT_STATUS_IS_OK(cli_qfileinfo_basic(
176 : cli, fnum, NULL, &size,
177 : NULL, NULL, NULL, NULL, NULL))) {
178 0 : continue;
179 : }
180 :
181 0 : if (size > 0) {
182 : /* found a character equivalence! */
183 : int c2[MAX_EQUIVALENCE];
184 :
185 0 : if (size/sizeof(int) >= MAX_EQUIVALENCE) {
186 0 : printf("too many chars match?? size=%ld c=0x%04x\n",
187 : (unsigned long)size, c);
188 0 : cli_close(cli, fnum);
189 0 : return False;
190 : }
191 :
192 0 : cli_read(cli, fnum, (char *)c2, 0, size, NULL);
193 0 : printf("%04x: ", c);
194 0 : equiv[c][0] = c;
195 0 : for (i=0; i<size/sizeof(int); i++) {
196 0 : printf("%04x ", c2[i]);
197 0 : equiv[c][i+1] = c2[i];
198 : }
199 0 : printf("\n");
200 0 : fflush(stdout);
201 : }
202 :
203 0 : cli_writeall(cli, fnum, 0, (uint8_t *)&c, size, sizeof(c),
204 : NULL);
205 0 : cli_close(cli, fnum);
206 : }
207 :
208 0 : cli_unlink(cli, "\\utable\\*", FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN);
209 0 : cli_rmdir(cli, "\\utable");
210 :
211 0 : return True;
212 : }
|
