2 Unix SMB/CIFS implementation.
6 Copyright (C) Andrew Tridgell 2005
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 2 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, write to the Free Software
20 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 #include "torture/torture.h"
25 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
26 #include "torture/rpc/rpc.h"
28 /**************************/
30 /**************************/
31 static BOOL
test_NetShareEnumAll(struct dcerpc_pipe
*p
, TALLOC_CTX
*mem_ctx
)
34 struct srvsvc_NetShareEnumAll r
;
35 struct srvsvc_NetShareCtr0 c0
;
36 uint32_t levels
[] = {0, 1, 2, 501, 502};
39 uint32_t resume_handle
;
43 r
.in
.server_unc
= talloc_asprintf(mem_ctx
,"\\\\%s",dcerpc_server_name(p
));
45 r
.in
.max_buffer
= (uint32_t)-1;
46 r
.in
.resume_handle
= &resume_handle
;
47 r
.out
.resume_handle
= &resume_handle
;
49 for (i
=0;i
<ARRAY_SIZE(levels
);i
++) {
52 r
.in
.level
= levels
[i
];
53 status
= dcerpc_srvsvc_NetShareEnumAll(p
, mem_ctx
, &r
);
54 if (!NT_STATUS_IS_OK(status
)) {
55 printf("NetShareEnumAll level %u failed - %s\n", r
.in
.level
, nt_errstr(status
));
59 if (!W_ERROR_IS_OK(r
.out
.result
)) {
60 printf("NetShareEnumAll level %u failed - %s\n", r
.in
.level
, win_errstr(r
.out
.result
));
69 benchmark srvsvc netshareenumall queries
71 static BOOL
bench_NetShareEnumAll(struct dcerpc_pipe
*p
, TALLOC_CTX
*mem_ctx
)
73 struct timeval tv
= timeval_current();
75 int timelimit
= lp_parm_int(-1, "torture", "timelimit", 10);
78 printf("Running for %d seconds\n", timelimit
);
79 while (timeval_elapsed(&tv
) < timelimit
) {
80 TALLOC_CTX
*tmp_ctx
= talloc_new(mem_ctx
);
81 if (!test_NetShareEnumAll(p
, tmp_ctx
)) break;
84 if (count
% 50 == 0) {
85 if (lp_parm_bool(-1, "torture", "progress", true)) {
86 printf("%.1f queries per second \r",
87 count
/ timeval_elapsed(&tv
));
92 printf("%.1f queries per second \n", count
/ timeval_elapsed(&tv
));
98 BOOL
torture_bench_rpc(struct torture_context
*torture
)
101 struct dcerpc_pipe
*p
;
105 mem_ctx
= talloc_init("torture_rpc_srvsvc");
107 status
= torture_rpc_connection(mem_ctx
,
109 &dcerpc_table_srvsvc
);
110 if (!NT_STATUS_IS_OK(status
)) {
111 talloc_free(mem_ctx
);
115 if (!bench_NetShareEnumAll(p
, mem_ctx
)) {
119 talloc_free(mem_ctx
);