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"
24 #include "../libcli/security/security.h"
25 #include "libsmb/clirap.h"
27 #define MAX_FILES 1000
29 static char buf
[70000];
30 extern int line_count
;
33 static struct timeval nb_start
;
40 static struct children
{
41 double bytes_in
, bytes_out
;
46 double nbio_total(void)
50 for (i
=0;i
<nprocs
;i
++) {
51 total
+= children
[i
].bytes_out
+ children
[i
].bytes_in
;
56 void nb_alarm(int ignore
)
59 int lines
=0, num_clients
=0;
60 if (nbio_id
!= -1) return;
62 for (i
=0;i
<nprocs
;i
++) {
63 lines
+= children
[i
].line
;
64 if (!children
[i
].done
) num_clients
++;
67 printf("%4d %8d %.2f MB/sec\r",
68 num_clients
, lines
/nprocs
,
69 1.0e-6 * nbio_total() / timeval_elapsed(&nb_start
));
71 signal(SIGALRM
, nb_alarm
);
75 void nbio_shmem(int n
)
78 children
= (struct children
*)shm_setup(sizeof(*children
) * nprocs
);
80 printf("Failed to setup shared memory!\n");
86 static int ne_find_handle(int handle
)
89 children
[nbio_id
].line
= line_count
;
90 for (i
=0;i
<MAX_FILES
;i
++) {
91 if (ftable
[i
].handle
== handle
) return i
;
97 static int find_handle(int handle
)
100 children
[nbio_id
].line
= line_count
;
101 for (i
=0;i
<MAX_FILES
;i
++) {
102 if (ftable
[i
].handle
== handle
) return i
;
104 printf("(%d) ERROR: handle %d was not found\n",
108 return -1; /* Not reached */
112 static struct cli_state
*c
;
114 static void sigsegv(int sig
)
117 printf("segv at line %d\n", line_count
);
118 slprintf(line
, sizeof(line
), "/usr/X11R6/bin/xterm -e gdb /proc/%d/exe %d",
119 (int)getpid(), (int)getpid());
120 if (system(line
) == -1) {
121 printf("system() failed\n");
126 void nb_setup(struct cli_state
*cli
)
128 signal(SIGSEGV
, sigsegv
);
130 nb_start
= timeval_current();
131 children
[nbio_id
].done
= 0;
135 void nb_unlink(const char *fname
)
137 if (!NT_STATUS_IS_OK(cli_unlink(c
, fname
, aSYSTEM
| aHIDDEN
))) {
139 printf("(%d) unlink %s failed (%s)\n",
140 line_count
, fname
, cli_errstr(c
));
146 void nb_createx(const char *fname
,
147 unsigned create_options
, unsigned create_disposition
, int handle
)
149 uint16_t fd
= (uint16_t)-1;
152 uint32 desired_access
;
154 if (create_options
& FILE_DIRECTORY_FILE
) {
155 desired_access
= FILE_READ_DATA
;
157 desired_access
= FILE_READ_DATA
| FILE_WRITE_DATA
;
160 status
= cli_ntcreate(c
, fname
, 0,
163 FILE_SHARE_READ
|FILE_SHARE_WRITE
,
165 create_options
, 0, &fd
);
166 if (!NT_STATUS_IS_OK(status
) && handle
!= -1) {
167 printf("ERROR: cli_ntcreate failed for %s - %s\n",
168 fname
, cli_errstr(c
));
171 if (NT_STATUS_IS_OK(status
) && handle
== -1) {
172 printf("ERROR: cli_ntcreate succeeded for %s\n", fname
);
175 if (fd
== (uint16_t)-1) return;
177 for (i
=0;i
<MAX_FILES
;i
++) {
178 if (ftable
[i
].handle
== 0) break;
180 if (i
== MAX_FILES
) {
181 printf("(%d) file table full for %s\n", line_count
,
185 ftable
[i
].handle
= handle
;
189 void nb_writex(int handle
, int offset
, int size
, int ret_size
)
193 if (buf
[0] == 0) memset(buf
, 1, sizeof(buf
));
195 i
= find_handle(handle
);
196 if (cli_write(c
, ftable
[i
].fd
, 0, buf
, offset
, size
) != ret_size
) {
197 printf("(%d) ERROR: write failed on handle %d, fd %d \
198 errno %d (%s)\n", line_count
, handle
, ftable
[i
].fd
, errno
, strerror(errno
));
202 children
[nbio_id
].bytes_out
+= ret_size
;
205 void nb_readx(int handle
, int offset
, int size
, int ret_size
)
209 i
= find_handle(handle
);
210 if ((ret
=cli_read(c
, ftable
[i
].fd
, buf
, offset
, size
)) != ret_size
) {
211 printf("(%d) ERROR: read failed on handle %d ofs=%d size=%d res=%d fd %d errno %d (%s)\n",
212 line_count
, handle
, offset
, size
, ret
, ftable
[i
].fd
, errno
, strerror(errno
));
215 children
[nbio_id
].bytes_in
+= ret_size
;
218 void nb_close(int handle
)
221 i
= find_handle(handle
);
222 if (!NT_STATUS_IS_OK(cli_close(c
, ftable
[i
].fd
))) {
223 printf("(%d) close failed on handle %d\n", line_count
, handle
);
226 ftable
[i
].handle
= 0;
229 void nb_rmdir(const char *fname
)
231 if (!NT_STATUS_IS_OK(cli_rmdir(c
, fname
))) {
232 printf("ERROR: rmdir %s failed (%s)\n",
233 fname
, cli_errstr(c
));
238 void nb_rename(const char *oldname
, const char *newname
)
240 if (!NT_STATUS_IS_OK(cli_rename(c
, oldname
, newname
))) {
241 printf("ERROR: rename %s %s failed (%s)\n",
242 oldname
, newname
, cli_errstr(c
));
248 void nb_qpathinfo(const char *fname
)
250 cli_qpathinfo1(c
, fname
, NULL
, NULL
, NULL
, NULL
, NULL
);
253 void nb_qfileinfo(int fnum
)
256 i
= find_handle(fnum
);
257 cli_qfileinfo_basic(c
, ftable
[i
].fd
, NULL
, NULL
, NULL
, NULL
, NULL
,
261 void nb_qfsinfo(int level
)
263 int bsize
, total
, avail
;
264 /* this is not the right call - we need cli_qfsinfo() */
265 cli_dskattr(c
, &bsize
, &total
, &avail
);
268 static NTSTATUS
find_fn(const char *mnt
, struct file_info
*finfo
, const char *name
,
275 void nb_findfirst(const char *mask
)
277 cli_list(c
, mask
, 0, find_fn
, NULL
);
280 void nb_flush(int fnum
)
283 i
= find_handle(fnum
);
284 /* hmmm, we don't have cli_flush() yet */
287 static int total_deleted
;
289 static NTSTATUS
delete_fn(const char *mnt
, struct file_info
*finfo
,
290 const char *name
, void *state
)
294 if (finfo
->name
[0] == '.') {
298 n
= SMB_STRDUP(name
);
300 if (asprintf(&s
, "%s%s", n
, finfo
->name
) == -1) {
301 printf("asprintf failed\n");
302 return NT_STATUS_NO_MEMORY
;
304 if (finfo
->mode
& aDIR
) {
306 if (asprintf(&s2
, "%s\\*", s
) == -1) {
307 printf("asprintf failed\n");
308 return NT_STATUS_NO_MEMORY
;
310 status
= cli_list(c
, s2
, aDIR
, delete_fn
, NULL
);
311 if (!NT_STATUS_IS_OK(status
)) {
326 void nb_deltree(const char *dname
)
329 if (asprintf(&mask
, "%s\\*", dname
) == -1) {
330 printf("asprintf failed\n");
335 cli_list(c
, mask
, aDIR
, delete_fn
, NULL
);
339 if (total_deleted
) printf("WARNING: Cleaned up %d files\n", total_deleted
);
343 void nb_cleanup(void)
345 cli_rmdir(c
, "clients");
346 children
[nbio_id
].done
= 1;