2 Unix SMB/CIFS implementation.
3 SMB torture tester - mangling test
4 Copyright (C) Andrew Tridgell 2002
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 2 of the License, or
9 (at your option) any later version.
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.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 extern int torture_numops
;
25 static TDB_CONTEXT
*tdb
;
27 #define NAME_LENGTH 20
29 static unsigned total
, collisions
, failures
;
31 static BOOL
test_one(struct cli_state
*cli
, const char *name
)
41 fnum
= cli_open(cli
, name
, O_RDWR
|O_CREAT
|O_EXCL
, DENY_NONE
);
43 printf("open of %s failed (%s)\n", name
, cli_errstr(cli
));
47 if (!cli_close(cli
, fnum
)) {
48 printf("close of %s failed (%s)\n", name
, cli_errstr(cli
));
52 /* get the short name */
53 status
= cli_qpathinfo_alt_name(cli
, name
, shortname
);
54 if (!NT_STATUS_IS_OK(status
)) {
55 printf("query altname of %s failed (%s)\n", name
, cli_errstr(cli
));
59 fstr_sprintf(name2
, "\\mangle_test\\%s", shortname
);
60 if (!cli_unlink(cli
, name2
)) {
61 printf("unlink of %s (%s) failed (%s)\n",
62 name2
, name
, cli_errstr(cli
));
66 /* recreate by short name */
67 fnum
= cli_open(cli
, name2
, O_RDWR
|O_CREAT
|O_EXCL
, DENY_NONE
);
69 printf("open2 of %s failed (%s)\n", name2
, cli_errstr(cli
));
72 if (!cli_close(cli
, fnum
)) {
73 printf("close of %s failed (%s)\n", name
, cli_errstr(cli
));
77 /* and unlink by long name */
78 if (!cli_unlink(cli
, name
)) {
79 printf("unlink2 of %s (%s) failed (%s)\n",
80 name
, name2
, cli_errstr(cli
));
82 cli_unlink(cli
, name2
);
86 /* see if the short name is already in the tdb */
87 data
= tdb_fetch_bystring(tdb
, shortname
);
89 /* maybe its a duplicate long name? */
90 if (!strequal(name
, (const char *)data
.dptr
)) {
91 /* we have a collision */
93 printf("Collision between %s and %s -> %s "
94 " (coll/tot: %u/%u)\n",
95 name
, data
.dptr
, shortname
, collisions
, total
);
100 /* store it for later */
101 namedata
.dptr
= CONST_DISCARD(uint8
*, name
);
102 namedata
.dsize
= strlen(name
)+1;
103 tdb_store_bystring(tdb
, shortname
, namedata
, TDB_REPLACE
);
110 static void gen_name(char *name
)
112 const char *chars
= "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz._-$~... ";
113 unsigned max_idx
= strlen(chars
);
118 fstrcpy(name
, "\\mangle_test\\");
119 p
= name
+ strlen(name
);
121 len
= 1 + random() % NAME_LENGTH
;
123 for (i
=0;i
<len
;i
++) {
124 p
[i
] = chars
[random() % max_idx
];
129 if (strcmp(p
, ".") == 0 || strcmp(p
, "..") == 0) {
133 /* have a high probability of a common lead char */
134 if (random() % 2 == 0) {
138 /* and a medium probability of a common lead string */
139 if (random() % 10 == 0) {
140 if (strlen(p
) <= 5) {
143 /* try not to kill off the null termination */
144 memcpy(p
, "ABCDE", 5);
148 /* and a high probability of a good extension length */
149 if (random() % 2 == 0) {
150 char *s
= strrchr(p
, '.');
156 /* ..... and a 100% proability of a file not ending in "." */
157 if (p
[strlen(p
)-1] == '.')
158 p
[strlen(p
)-1] = '_';
162 BOOL
torture_mangle(int dummy
)
164 static struct cli_state
*cli
;
168 printf("starting mangle test\n");
170 if (!torture_open_connection(&cli
, 0)) {
174 /* we will use an internal tdb to store the names we have used */
175 tdb
= tdb_open(NULL
, 100000, TDB_INTERNAL
, 0, 0);
177 printf("ERROR: Failed to open tdb\n");
181 cli_unlink(cli
, "\\mangle_test\\*");
182 cli_rmdir(cli
, "\\mangle_test");
184 if (!cli_mkdir(cli
, "\\mangle_test")) {
185 printf("ERROR: Failed to make directory\n");
189 for (i
=0;i
<torture_numops
;i
++) {
195 if (!test_one(cli
, name
)) {
199 if (total
&& total
% 100 == 0) {
200 printf("collisions %u/%u - %.2f%% (%u failures)\r",
201 collisions
, total
, (100.0*collisions
) / total
, failures
);
205 cli_unlink(cli
, "\\mangle_test\\*");
206 if (!cli_rmdir(cli
, "\\mangle_test")) {
207 printf("ERROR: Failed to remove directory\n");
211 printf("\nTotal collisions %u/%u - %.2f%% (%u failures)\n",
212 collisions
, total
, (100.0*collisions
) / total
, failures
);
214 torture_close_connection(cli
);
216 printf("mangle test finished\n");
217 return (ret
&& (failures
== 0));