2 Unix SMB/CIFS implementation.
3 SMB torture tester - NBENCH test
4 Copyright (C) Andrew Tridgell 1997-2004
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 #include "system/filesys.h"
25 int nbench_line_count = 0;
26 static int timelimit = 600;
28 static const char *loadfile;
30 #define ival(s) strtol(s, NULL, 0)
32 /* run a test that simulates an approximate netbench client load */
33 static BOOL run_netbench(struct smbcli_state *cli, int client)
35 extern int torture_nprocs;
44 if (torture_nprocs == 1) {
45 if (!torture_setup_dir(cli, "\\clients")) {
50 nb_setup(cli, client);
52 asprintf(&cname, "client%d", client+1);
54 f = fopen(loadfile, "r");
62 while (fgets(line, sizeof(line)-1, f)) {
67 line[strlen(line)-1] = 0;
69 all_string_sub(line,"client1", cname, sizeof(line));
73 i<19 && next_token(&p, params[i], " ", sizeof(fstring));
78 if (i < 2 || params[0][0] == '#') continue;
80 if (!strncmp(params[0],"SMB", 3)) {
81 printf("ERROR: You are using a dbench 1 load file\n");
85 if (strncmp(params[i-1], "NT_STATUS_", 10) != 0) {
86 printf("Badly formed status at line %d\n", nbench_line_count);
90 status = nt_status_string_to_code(params[i-1]);
92 DEBUG(9,("run_netbench(%d): %s %s\n", client, params[0], params[1]));
94 if (!strcmp(params[0],"NTCreateX")) {
95 nb_createx(params[1], ival(params[2]), ival(params[3]),
96 ival(params[4]), status);
97 } else if (!strcmp(params[0],"Close")) {
98 nb_close(ival(params[1]), status);
99 } else if (!strcmp(params[0],"Rename")) {
100 nb_rename(params[1], params[2], status);
101 } else if (!strcmp(params[0],"Unlink")) {
102 nb_unlink(params[1], ival(params[2]), status);
103 } else if (!strcmp(params[0],"Deltree")) {
104 nb_deltree(params[1]);
105 } else if (!strcmp(params[0],"Rmdir")) {
106 nb_rmdir(params[1], status);
107 } else if (!strcmp(params[0],"Mkdir")) {
108 nb_mkdir(params[1], status);
109 } else if (!strcmp(params[0],"QUERY_PATH_INFORMATION")) {
110 nb_qpathinfo(params[1], ival(params[2]), status);
111 } else if (!strcmp(params[0],"QUERY_FILE_INFORMATION")) {
112 nb_qfileinfo(ival(params[1]), ival(params[2]), status);
113 } else if (!strcmp(params[0],"QUERY_FS_INFORMATION")) {
114 nb_qfsinfo(ival(params[1]), status);
115 } else if (!strcmp(params[0],"SET_FILE_INFORMATION")) {
116 nb_sfileinfo(ival(params[1]), ival(params[2]), status);
117 } else if (!strcmp(params[0],"FIND_FIRST")) {
118 nb_findfirst(params[1], ival(params[2]),
119 ival(params[3]), ival(params[4]), status);
120 } else if (!strcmp(params[0],"WriteX")) {
121 nb_writex(ival(params[1]),
122 ival(params[2]), ival(params[3]), ival(params[4]),
124 } else if (!strcmp(params[0],"Write")) {
125 nb_write(ival(params[1]),
126 ival(params[2]), ival(params[3]), ival(params[4]),
128 } else if (!strcmp(params[0],"LockX")) {
129 nb_lockx(ival(params[1]),
130 ival(params[2]), ival(params[3]), status);
131 } else if (!strcmp(params[0],"UnlockX")) {
132 nb_unlockx(ival(params[1]),
133 ival(params[2]), ival(params[3]), status);
134 } else if (!strcmp(params[0],"ReadX")) {
135 nb_readx(ival(params[1]),
136 ival(params[2]), ival(params[3]), ival(params[4]),
138 } else if (!strcmp(params[0],"Flush")) {
139 nb_flush(ival(params[1]), status);
140 } else if (!strcmp(params[0],"Sleep")) {
141 nb_sleep(ival(params[1]), status);
143 printf("[%d] Unknown operation %s\n", nbench_line_count, params[0]);
146 if (nb_tick()) goto done;
155 if (!torture_close_connection(cli)) {
163 /* run a test that simulates an approximate netbench client load */
164 BOOL torture_nbench(void)
167 extern int torture_nprocs;
168 struct smbcli_state *cli;
171 p = lp_parm_string(-1, "torture", "timelimit");
176 warmup = timelimit / 20;
178 loadfile = lp_parm_string(-1, "torture", "loadfile");
179 if (!loadfile || !*loadfile) {
180 loadfile = "client.txt";
183 if (torture_nprocs > 1) {
184 if (!torture_open_connection(&cli)) {
188 if (!torture_setup_dir(cli, "\\clients")) {
193 nbio_shmem(torture_nprocs, timelimit, warmup);
195 printf("Running for %d seconds with load '%s' and warmup %d secs\n",
196 timelimit, loadfile, warmup);
198 signal(SIGALRM, nb_alarm);
200 torture_create_procs(run_netbench, &correct);
203 smbcli_deltree(cli->tree, "\\clients");
205 printf("\nThroughput %g MB/sec\n", nbio_result());