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_qpathinfo(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;