source: trunk/server/source3/torture/nbio.c

Last change on this file was 745, checked in by Silvan Scherrer, 13 years ago

Samba Server: updated trunk to 3.6.0
File size: 7.8 KB
Line 
1#define NBDEBUG 0
2
3/*
4 Unix SMB/CIFS implementation.
5 SMB torture tester
6 Copyright (C) Andrew Tridgell 1997-1998
7
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
12
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
17
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
20*/
21
22#include "includes.h"
23#include "torture/proto.h"
24#include "../libcli/security/security.h"
25#include "libsmb/libsmb.h"
26#include "libsmb/clirap.h"
27
28#define MAX_FILES 1000
29
30static char buf[70000];
31extern int line_count;
32extern int nbio_id;
33static int nprocs;
34static struct timeval nb_start;
35
36static struct {
37 int fd;
38 int handle;
39} ftable[MAX_FILES];
40
41static struct children {
42 double bytes_in, bytes_out;
43 int line;
44 int done;
45} *children;
46
47double nbio_total(void)
48{
49 int i;
50 double total = 0;
51 for (i=0;i<nprocs;i++) {
52 total += children[i].bytes_out + children[i].bytes_in;
53 }
54 return total;
55}
56
57void nb_alarm(int ignore)
58{
59 int i;
60 int lines=0, num_clients=0;
61 if (nbio_id != -1) return;
62
63 for (i=0;i<nprocs;i++) {
64 lines += children[i].line;
65 if (!children[i].done) num_clients++;
66 }
67
68 printf("%4d %8d %.2f MB/sec\r",
69 num_clients, lines/nprocs,
70 1.0e-6 * nbio_total() / timeval_elapsed(&nb_start));
71
72 signal(SIGALRM, nb_alarm);
73 alarm(1);
74}
75
76void nbio_shmem(int n)
77{
78 nprocs = n;
79 children = (struct children *)shm_setup(sizeof(*children) * nprocs);
80 if (!children) {
81 printf("Failed to setup shared memory!\n");
82 exit(1);
83 }
84}
85
86#if 0
87static int ne_find_handle(int handle)
88{
89 int i;
90 children[nbio_id].line = line_count;
91 for (i=0;i<MAX_FILES;i++) {
92 if (ftable[i].handle == handle) return i;
93 }
94 return -1;
95}
96#endif
97
98static int find_handle(int handle)
99{
100 int i;
101 children[nbio_id].line = line_count;
102 for (i=0;i<MAX_FILES;i++) {
103 if (ftable[i].handle == handle) return i;
104 }
105 printf("(%d) ERROR: handle %d was not found\n",
106 line_count, handle);
107 exit(1);
108
109 return -1; /* Not reached */
110}
111
112
113static struct cli_state *c;
114
115static void sigsegv(int sig)
116{
117 char line[200];
118 printf("segv at line %d\n", line_count);
119 slprintf(line, sizeof(line), "/usr/X11R6/bin/xterm -e gdb /proc/%d/exe %d",
120 (int)getpid(), (int)getpid());
121 if (system(line) == -1) {
122 printf("system() failed\n");
123 }
124 exit(1);
125}
126
127void nb_setup(struct cli_state *cli)
128{
129 signal(SIGSEGV, sigsegv);
130 c = cli;
131 nb_start = timeval_current();
132 children[nbio_id].done = 0;
133}
134
135
136void nb_unlink(const char *fname)
137{
138 if (!NT_STATUS_IS_OK(cli_unlink(c, fname, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN))) {
139#if NBDEBUG
140 printf("(%d) unlink %s failed (%s)\n",
141 line_count, fname, cli_errstr(c));
142#endif
143 }
144}
145
146
147void nb_createx(const char *fname,
148 unsigned create_options, unsigned create_disposition, int handle)
149{
150 uint16_t fd = (uint16_t)-1;
151 int i;
152 NTSTATUS status;
153 uint32 desired_access;
154
155 if (create_options & FILE_DIRECTORY_FILE) {
156 desired_access = FILE_READ_DATA;
157 } else {
158 desired_access = FILE_READ_DATA | FILE_WRITE_DATA;
159 }
160
161 status = cli_ntcreate(c, fname, 0,
162 desired_access,
163 0x0,
164 FILE_SHARE_READ|FILE_SHARE_WRITE,
165 create_disposition,
166 create_options, 0, &fd);
167 if (!NT_STATUS_IS_OK(status) && handle != -1) {
168 printf("ERROR: cli_ntcreate failed for %s - %s\n",
169 fname, cli_errstr(c));
170 exit(1);
171 }
172 if (NT_STATUS_IS_OK(status) && handle == -1) {
173 printf("ERROR: cli_ntcreate succeeded for %s\n", fname);
174 exit(1);
175 }
176 if (fd == (uint16_t)-1) return;
177
178 for (i=0;i<MAX_FILES;i++) {
179 if (ftable[i].handle == 0) break;
180 }
181 if (i == MAX_FILES) {
182 printf("(%d) file table full for %s\n", line_count,
183 fname);
184 exit(1);
185 }
186 ftable[i].handle = handle;
187 ftable[i].fd = fd;
188}
189
190void nb_writex(int handle, int offset, int size, int ret_size)
191{
192 int i;
193 NTSTATUS status;
194
195 if (buf[0] == 0) memset(buf, 1, sizeof(buf));
196
197 i = find_handle(handle);
198 status = cli_writeall(c, ftable[i].fd, 0, (uint8_t *)buf, offset, size,
199 NULL);
200 if (!NT_STATUS_IS_OK(status)) {
201 printf("(%d) ERROR: write failed on handle %d, fd %d "
202 "error %s\n", line_count, handle, ftable[i].fd,
203 nt_errstr(status));
204 exit(1);
205 }
206
207 children[nbio_id].bytes_out += ret_size;
208}
209
210void nb_readx(int handle, int offset, int size, int ret_size)
211{
212 int i, ret;
213
214 i = find_handle(handle);
215 if ((ret=cli_read(c, ftable[i].fd, buf, offset, size)) != ret_size) {
216 printf("(%d) ERROR: read failed on handle %d ofs=%d size=%d res=%d fd %d errno %d (%s)\n",
217 line_count, handle, offset, size, ret, ftable[i].fd, errno, strerror(errno));
218 exit(1);
219 }
220 children[nbio_id].bytes_in += ret_size;
221}
222
223void nb_close(int handle)
224{
225 int i;
226 i = find_handle(handle);
227 if (!NT_STATUS_IS_OK(cli_close(c, ftable[i].fd))) {
228 printf("(%d) close failed on handle %d\n", line_count, handle);
229 exit(1);
230 }
231 ftable[i].handle = 0;
232}
233
234void nb_rmdir(const char *fname)
235{
236 if (!NT_STATUS_IS_OK(cli_rmdir(c, fname))) {
237 printf("ERROR: rmdir %s failed (%s)\n",
238 fname, cli_errstr(c));
239 exit(1);
240 }
241}
242
243void nb_rename(const char *oldname, const char *newname)
244{
245 if (!NT_STATUS_IS_OK(cli_rename(c, oldname, newname))) {
246 printf("ERROR: rename %s %s failed (%s)\n",
247 oldname, newname, cli_errstr(c));
248 exit(1);
249 }
250}
251
252
253void nb_qpathinfo(const char *fname)
254{
255 cli_qpathinfo1(c, fname, NULL, NULL, NULL, NULL, NULL);
256}
257
258void nb_qfileinfo(int fnum)
259{
260 int i;
261 i = find_handle(fnum);
262 cli_qfileinfo_basic(c, ftable[i].fd, NULL, NULL, NULL, NULL, NULL,
263 NULL, NULL);
264}
265
266void nb_qfsinfo(int level)
267{
268 int bsize, total, avail;
269 /* this is not the right call - we need cli_qfsinfo() */
270 cli_dskattr(c, &bsize, &total, &avail);
271}
272
273static NTSTATUS find_fn(const char *mnt, struct file_info *finfo, const char *name,
274 void *state)
275{
276 /* noop */
277 return NT_STATUS_OK;
278}
279
280void nb_findfirst(const char *mask)
281{
282 cli_list(c, mask, 0, find_fn, NULL);
283}
284
285void nb_flush(int fnum)
286{
287 int i;
288 i = find_handle(fnum);
289 /* hmmm, we don't have cli_flush() yet */
290}
291
292static int total_deleted;
293
294static NTSTATUS delete_fn(const char *mnt, struct file_info *finfo,
295 const char *name, void *state)
296{
297 NTSTATUS status;
298 char *s, *n;
299 if (finfo->name[0] == '.') {
300 return NT_STATUS_OK;
301 }
302
303 n = SMB_STRDUP(name);
304 n[strlen(n)-1] = 0;
305 if (asprintf(&s, "%s%s", n, finfo->name) == -1) {
306 printf("asprintf failed\n");
307 return NT_STATUS_NO_MEMORY;
308 }
309 if (finfo->mode & FILE_ATTRIBUTE_DIRECTORY) {
310 char *s2;
311 if (asprintf(&s2, "%s\\*", s) == -1) {
312 printf("asprintf failed\n");
313 return NT_STATUS_NO_MEMORY;
314 }
315 status = cli_list(c, s2, FILE_ATTRIBUTE_DIRECTORY, delete_fn, NULL);
316 if (!NT_STATUS_IS_OK(status)) {
317 free(n);
318 free(s2);
319 return status;
320 }
321 nb_rmdir(s);
322 } else {
323 total_deleted++;
324 nb_unlink(s);
325 }
326 free(s);
327 free(n);
328 return NT_STATUS_OK;
329}
330
331void nb_deltree(const char *dname)
332{
333 char *mask;
334 if (asprintf(&mask, "%s\\*", dname) == -1) {
335 printf("asprintf failed\n");
336 return;
337 }
338
339 total_deleted = 0;
340 cli_list(c, mask, FILE_ATTRIBUTE_DIRECTORY, delete_fn, NULL);
341 free(mask);
342 cli_rmdir(c, dname);
343
344 if (total_deleted) printf("WARNING: Cleaned up %d files\n", total_deleted);
345}
346
347
348void nb_cleanup(void)
349{
350 cli_rmdir(c, "clients");
351 children[nbio_id].done = 1;
352}
Note: See TracBrowser for help on using the repository browser.