4 Unix SMB/CIFS implementation.
6 Copyright (C) Andrew Tridgell 1997-1998
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.
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.
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/>.
23 #include "torture/proto.h"
25 #define MAX_FILES 1000
27 static char buf[70000];
28 extern int line_count;
31 static struct timeval nb_start;
38 static struct children {
39 double bytes_in, bytes_out;
44 double nbio_total(void)
48 for (i=0;i<nprocs;i++) {
49 total += children[i].bytes_out + children[i].bytes_in;
54 void nb_alarm(int ignore)
57 int lines=0, num_clients=0;
58 if (nbio_id != -1) return;
60 for (i=0;i<nprocs;i++) {
61 lines += children[i].line;
62 if (!children[i].done) num_clients++;
65 printf("%4d %8d %.2f MB/sec\r",
66 num_clients, lines/nprocs,
67 1.0e-6 * nbio_total() / timeval_elapsed(&nb_start));
69 signal(SIGALRM, nb_alarm);
73 void nbio_shmem(int n)
76 children = (struct children *)shm_setup(sizeof(*children) * nprocs);
78 printf("Failed to setup shared memory!\n");
84 static int ne_find_handle(int handle)
87 children[nbio_id].line = line_count;
88 for (i=0;i<MAX_FILES;i++) {
89 if (ftable[i].handle == handle) return i;
95 static int find_handle(int handle)
98 children[nbio_id].line = line_count;
99 for (i=0;i<MAX_FILES;i++) {
100 if (ftable[i].handle == handle) return i;
102 printf("(%d) ERROR: handle %d was not found\n",
106 return -1; /* Not reached */
110 static struct cli_state *c;
112 static void sigsegv(int sig)
115 printf("segv at line %d\n", line_count);
116 slprintf(line, sizeof(line), "/usr/X11R6/bin/xterm -e gdb /proc/%d/exe %d",
117 (int)getpid(), (int)getpid());
118 if (system(line) == -1) {
119 printf("system() failed\n");
124 void nb_setup(struct cli_state *cli)
126 signal(SIGSEGV, sigsegv);
128 nb_start = timeval_current();
129 children[nbio_id].done = 0;
133 void nb_unlink(const char *fname)
135 if (!NT_STATUS_IS_OK(cli_unlink(c, fname, aSYSTEM | aHIDDEN))) {
137 printf("(%d) unlink %s failed (%s)\n",
138 line_count, fname, cli_errstr(c));
144 void nb_createx(const char *fname,
145 unsigned create_options, unsigned create_disposition, int handle)
147 uint16_t fd = (uint16_t)-1;
150 uint32 desired_access;
152 if (create_options & FILE_DIRECTORY_FILE) {
153 desired_access = FILE_READ_DATA;
155 desired_access = FILE_READ_DATA | FILE_WRITE_DATA;
158 status = cli_ntcreate(c, fname, 0,
161 FILE_SHARE_READ|FILE_SHARE_WRITE,
163 create_options, 0, &fd);
164 if (!NT_STATUS_IS_OK(status) && handle != -1) {
165 printf("ERROR: cli_ntcreate failed for %s - %s\n",
166 fname, cli_errstr(c));
169 if (NT_STATUS_IS_OK(status) && handle == -1) {
170 printf("ERROR: cli_ntcreate succeeded for %s\n", fname);
173 if (fd == (uint16_t)-1) return;
175 for (i=0;i<MAX_FILES;i++) {
176 if (ftable[i].handle == 0) break;
178 if (i == MAX_FILES) {
179 printf("(%d) file table full for %s\n", line_count,
183 ftable[i].handle = handle;
187 void nb_writex(int handle, int offset, int size, int ret_size)
191 if (buf[0] == 0) memset(buf, 1, sizeof(buf));
193 i = find_handle(handle);
194 if (cli_write(c, ftable[i].fd, 0, buf, offset, size) != ret_size) {
195 printf("(%d) ERROR: write failed on handle %d, fd %d \
196 errno %d (%s)\n", line_count, handle, ftable[i].fd, errno, strerror(errno));
200 children[nbio_id].bytes_out += ret_size;
203 void nb_readx(int handle, int offset, int size, int ret_size)
207 i = find_handle(handle);
208 if ((ret=cli_read(c, ftable[i].fd, buf, offset, size)) != ret_size) {
209 printf("(%d) ERROR: read failed on handle %d ofs=%d size=%d res=%d fd %d errno %d (%s)\n",
210 line_count, handle, offset, size, ret, ftable[i].fd, errno, strerror(errno));
213 children[nbio_id].bytes_in += ret_size;
216 void nb_close(int handle)
219 i = find_handle(handle);
220 if (!NT_STATUS_IS_OK(cli_close(c, ftable[i].fd))) {
221 printf("(%d) close failed on handle %d\n", line_count, handle);
224 ftable[i].handle = 0;
227 void nb_rmdir(const char *fname)
229 if (!NT_STATUS_IS_OK(cli_rmdir(c, fname))) {
230 printf("ERROR: rmdir %s failed (%s)\n",
231 fname, cli_errstr(c));
236 void nb_rename(const char *oldname, const char *newname)
238 if (!NT_STATUS_IS_OK(cli_rename(c, oldname, newname))) {
239 printf("ERROR: rename %s %s failed (%s)\n",
240 oldname, newname, cli_errstr(c));
246 void nb_qpathinfo(const char *fname)
248 cli_qpathinfo1(c, fname, NULL, NULL, NULL, NULL, NULL);
251 void nb_qfileinfo(int fnum)
254 i = find_handle(fnum);
255 cli_qfileinfo(c, ftable[i].fd, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
258 void nb_qfsinfo(int level)
260 int bsize, total, avail;
261 /* this is not the right call - we need cli_qfsinfo() */
262 cli_dskattr(c, &bsize, &total, &avail);
265 static void find_fn(const char *mnt, file_info *finfo, const char *name, void *state)
270 void nb_findfirst(const char *mask)
272 cli_list(c, mask, 0, find_fn, NULL);
275 void nb_flush(int fnum)
278 i = find_handle(fnum);
279 /* hmmm, we don't have cli_flush() yet */
282 static int total_deleted;
284 static void delete_fn(const char *mnt, file_info *finfo, const char *name, void *state)
287 if (finfo->name[0] == '.') return;
289 n = SMB_STRDUP(name);
291 if (asprintf(&s, "%s%s", n, finfo->name) == -1) {
292 printf("asprintf failed\n");
295 if (finfo->mode & aDIR) {
297 if (asprintf(&s2, "%s\\*", s) == -1) {
298 printf("asprintf failed\n");
301 cli_list(c, s2, aDIR, delete_fn, NULL);
311 void nb_deltree(const char *dname)
314 if (asprintf(&mask, "%s\\*", dname) == -1) {
315 printf("asprintf failed\n");
320 cli_list(c, mask, aDIR, delete_fn, NULL);
324 if (total_deleted) printf("WARNING: Cleaned up %d files\n", total_deleted);
328 void nb_cleanup(void)
330 cli_rmdir(c, "clients");
331 children[nbio_id].done = 1;