summaryrefslogtreecommitdiff
path: root/source4
diff options
context:
space:
mode:
Diffstat (limited to 'source4')
-rwxr-xr-xsource4/torture/raw/pingpong.c400
1 files changed, 146 insertions, 254 deletions
diff --git a/source4/torture/raw/pingpong.c b/source4/torture/raw/pingpong.c
index a3ee5b5809..027034275a 100755
--- a/source4/torture/raw/pingpong.c
+++ b/source4/torture/raw/pingpong.c
@@ -2,6 +2,27 @@
Unix SMB/CIFS implementation.
ping pong test
+
+ Copyright (C) Ronnie Sahlberg 2007
+
+ Significantly based on and borrowed from lockbench.c by
+ Copyright (C) Andrew Tridgell 2006
+
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 3 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+/*
filename is specified by
--option=torture:filename=...
@@ -21,26 +42,7 @@
writing to the file can be enabled with
--option=torture:write=True
-
- Copyright (C) Ronnie Sahlberg
-
- Significantly based on and borrowed from lockbench.c by
- Copyright (C) Andrew Tridgell 2006
-
- This program is free software; you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation; either version 3 of the License, or
- (at your option) any later version.
-
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
-
- You should have received a copy of the GNU General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
-
#include "includes.h"
#include "torture/torture.h"
#include "libcli/raw/libcliraw.h"
@@ -48,264 +50,129 @@
#include "system/filesys.h"
#include "libcli/libcli.h"
#include "torture/util.h"
-#include "lib/events/events.h"
-#include "lib/cmdline/popt_common.h"
-#include "libcli/composite/composite.h"
-#include "libcli/smb_composite/smb_composite.h"
-
-static BOOL do_reads;
-static BOOL do_writes;
-static int lock_failed;
-enum lock_stage {LOCK_INITIAL, LOCK_LOCK, LOCK_UNLOCK};
-
-struct pingpong_state {
- struct event_context *ev;
- struct smbcli_tree *tree;
- TALLOC_CTX *mem_ctx;
- int fnum;
- enum lock_stage stage;
- int num_locks;
- int lock_offset;
- int unlock_offset;
- struct smbcli_request *req;
- int count;
- int lastcount;
- int lock_timeout;
- uint8_t c;
- uint8_t incr;
- uint8_t last_incr;
- uint8_t *val;
-};
-
-
-static void lock_completion(struct smbcli_request *);
-
-/*
- send the next lock request
-*/
-static void lock_send(struct pingpong_state *state, BOOL retry_lock)
+void lock_byte(struct smbcli_state *cli, int fd, int offset, int lock_timeout)
{
union smb_lock io;
struct smb_lock_entry lock;
+ NTSTATUS status;
+try_again:
+ ZERO_STRUCT(lock);
+ io.lockx.in.ulock_cnt = 0;
+ io.lockx.in.lock_cnt = 1;
- /* we have completed one lock/unlock pair */
- if (state->stage == LOCK_UNLOCK) {
- if ( (state->count > state->num_locks)
- &&(state->incr != state->last_incr) ) {
- state->last_incr = state->incr;
- printf("data increment = %u\n", state->incr);
+ lock.count = 1;
+ lock.offset = offset;
+ lock.pid = cli->tree->session->pid;
+ io.lockx.level = RAW_LOCK_LOCKX;
+ io.lockx.in.mode = LOCKING_ANDX_LARGE_FILES;
+ io.lockx.in.timeout = lock_timeout;
+ io.lockx.in.locks = &lock;
+ io.lockx.in.file.fnum = fd;
+
+ status = smb_raw_lock(cli->tree, &io);
+
+ /* If we dont use timeouts and we got file lock conflict
+ just try the lock again.
+ */
+ if (lock_timeout==0) {
+ if ( (NT_STATUS_EQUAL(NT_STATUS_FILE_LOCK_CONFLICT, status))
+ ||(NT_STATUS_EQUAL(NT_STATUS_LOCK_NOT_GRANTED, status)) ) {
+ goto try_again;
}
}
- switch (state->stage) {
- case LOCK_INITIAL:
- io.lockx.in.ulock_cnt = 0;
- io.lockx.in.lock_cnt = 1;
- state->lock_offset = -1;
- state->unlock_offset = 0;
- lock.offset = (state->lock_offset+1)%state->num_locks;
- break;
- case LOCK_LOCK:
- io.lockx.in.ulock_cnt = 0;
- io.lockx.in.lock_cnt = 1;
- if (!retry_lock) {
- state->lock_offset = (state->lock_offset+1)%state->num_locks;
- }
- lock.offset = (state->lock_offset+1)%state->num_locks;
- break;
- case LOCK_UNLOCK:
- state->count++;
- io.lockx.in.ulock_cnt = 1;
- io.lockx.in.lock_cnt = 0;
- lock.offset = state->lock_offset;
- break;
+ if (!NT_STATUS_IS_OK(status)) {
+ DEBUG(0,("Lock failed\n"));
+ exit(1);
}
+}
- lock.count = 1;
- lock.pid = state->tree->session->pid;
+void unlock_byte(struct smbcli_state *cli, int fd, int offset)
+{
+ union smb_lock io;
+ struct smb_lock_entry lock;
+ NTSTATUS status;
+ ZERO_STRUCT(lock);
+ io.lockx.in.ulock_cnt = 1;
+ io.lockx.in.lock_cnt = 0;
+
+ lock.count = 1;
+ lock.offset = offset;
+ lock.pid = cli->tree->session->pid;
io.lockx.level = RAW_LOCK_LOCKX;
io.lockx.in.mode = LOCKING_ANDX_LARGE_FILES;
- io.lockx.in.timeout = state->lock_timeout;
+ io.lockx.in.timeout = 100000;
io.lockx.in.locks = &lock;
- io.lockx.in.file.fnum = state->fnum;
-
- state->req = smb_raw_lock_send(state->tree, &io);
- if (state->req == NULL) {
- DEBUG(0,("Failed to setup lock\n"));
- lock_failed++;
- }
- state->req->async.private = state;
- state->req->async.fn = lock_completion;
-}
+ io.lockx.in.file.fnum = fd;
-
-/*
- called when a write completes
-*/
-static void write_completion(struct smbcli_request *req)
-{
- struct pingpong_state *state = (struct pingpong_state *)req->async.private;
- NTSTATUS status = smbcli_request_simple_recv(req);
+ status = smb_raw_lock(cli->tree, &io);
if (!NT_STATUS_IS_OK(status)) {
- printf("write failed\n");
+ DEBUG(0,("Unlock failed\n"));
exit(1);
}
-
- lock_send(state, False);
}
-
-static void write_send(struct pingpong_state *state)
+void write_byte(struct smbcli_state *cli, int fd, uint8_t c, int offset)
{
union smb_write io;
+ NTSTATUS status;
io.generic.level = RAW_WRITE_WRITEX;
- io.writex.in.file.fnum = state->fnum;
- io.writex.in.offset = state->lock_offset;
+ io.writex.in.file.fnum = fd;
+ io.writex.in.offset = offset;
io.writex.in.wmode = 0;
io.writex.in.remaining = 0;
io.writex.in.count = 1;
- state->c = state->val[state->lock_offset]+1;
- io.writex.in.data = &state->c;
-
- state->req = smb_raw_write_send(state->tree, &io);
- if (state->req == NULL) {
- DEBUG(0,("Failed to setup write\n"));
- exit(1);
- }
- state->req->async.private = state;
- state->req->async.fn = write_completion;
-}
-
-
-/*
- called when a read completes
-*/
-static void read_completion(struct smbcli_request *req)
-{
- struct pingpong_state *state = (struct pingpong_state *)req->async.private;
- NTSTATUS status = smbcli_request_simple_recv(req);
+ io.writex.in.data = &c;
+ status = smb_raw_write(cli->tree, &io);
if (!NT_STATUS_IS_OK(status)) {
- printf("read failed\n");
+ printf("write failed\n");
exit(1);
}
+}
- state->req = NULL;
-
- state->incr = state->c - state->val[state->lock_offset];
- state->val[state->lock_offset] = state->c;
-
- /* a read just completed, now spawn off to the write handler, if
- write is enabled. othervise spawn off to the lock handler
- to proceed to unlock the previous lock
- */
- if (do_writes) {
- write_send(state);
- return;
- }
- lock_send(state, False);
-}
-
-
-static void read_send(struct pingpong_state *state)
+void read_byte(struct smbcli_state *cli, int fd, uint8_t *c, int offset)
{
union smb_read io;
+ NTSTATUS status;
io.generic.level = RAW_READ_READX;
- io.readx.in.file.fnum = state->fnum;
+ io.readx.in.file.fnum = fd;
io.readx.in.mincnt = 1;
io.readx.in.maxcnt = 1;
- io.readx.in.offset = state->lock_offset;
+ io.readx.in.offset = offset;
io.readx.in.remaining = 0;
io.readx.in.read_for_execute = False;
- io.readx.out.data = &state->c;
-
- state->req = smb_raw_read_send(state->tree, &io);
- if (state->req == NULL) {
- DEBUG(0,("Failed to setup read\n"));
- exit(1);
- }
- state->req->async.private = state;
- state->req->async.fn = read_completion;
-}
-
-
-/*
- called when a lock completes
-*/
-static void lock_completion(struct smbcli_request *req)
-{
- struct pingpong_state *state = (struct pingpong_state *)req->async.private;
- NTSTATUS status = smbcli_request_simple_recv(req);
- state->req = NULL;
- /* If we dont use timeouts and we got file lock conflict
- just try the lock again.
- */
- if (state->lock_timeout==0) {
- if ( (NT_STATUS_EQUAL(NT_STATUS_FILE_LOCK_CONFLICT, status))
- ||(NT_STATUS_EQUAL(NT_STATUS_LOCK_NOT_GRANTED, status)) ) {
- lock_send(state, True);
- return;
- }
- }
+ io.readx.out.data = c;
+ status = smb_raw_read(cli->tree, &io);
if (!NT_STATUS_IS_OK(status)) {
- lock_failed++;
- return;
+ printf("read failed\n");
+ exit(1);
}
+}
- switch (state->stage) {
- case LOCK_INITIAL:
- state->stage = LOCK_LOCK;
- break;
- case LOCK_LOCK:
- state->stage = LOCK_UNLOCK;
- break;
- case LOCK_UNLOCK:
- state->stage = LOCK_LOCK;
- break;
- }
- /* if we just completed a lock and we have read enabled
- then spawn off to the read handler instead of sending an unlock
- */
- if ( (state->stage == LOCK_UNLOCK)
- &&(do_reads) ){
- read_send(state);
- return;
- }
- /* if we just completed a lock and we we didnt have reads enabled
- but we do have write enabled, thenspawn off to the write handler
- instead of sending an unlock
- */
- if ( (state->stage == LOCK_UNLOCK)
- &&(do_writes) ){
- write_send(state);
- return;
- }
+static struct timeval tp1, tp2;
- lock_send(state, False);
+static void start_timer(void)
+{
+ gettimeofday(&tp1, NULL);
}
-static void report_rate(struct event_context *ev, struct timed_event *te,
- struct timeval t, void *private_data)
+static double end_timer(void)
{
- struct pingpong_state *state = talloc_get_type(private_data,
- struct pingpong_state);
- printf("%5u ", 2*(unsigned)(state->count - state->lastcount));
- state->lastcount = state->count;
-
- printf("\r");
- fflush(stdout);
- event_add_timed(ev, state, timeval_current_ofs(1, 0), report_rate, state);
+ gettimeofday(&tp2, NULL);
+ return (tp2.tv_sec + (tp2.tv_usec*1.0e-6)) -
+ (tp1.tv_sec + (tp1.tv_usec*1.0e-6));
}
/*
@@ -314,13 +181,17 @@ static void report_rate(struct event_context *ev, struct timed_event *te,
BOOL torture_ping_pong(struct torture_context *torture)
{
const char *fn;
+ int num_locks;
TALLOC_CTX *mem_ctx = talloc_new(torture);
- struct event_context *ev = event_context_find(mem_ctx);
- int timelimit = torture_setting_int(torture, "timelimit", 10);
- struct timeval tv;
- struct pingpong_state *state;
+ static BOOL do_reads;
+ static BOOL do_writes;
+ int lock_timeout;
+ int fd;
struct smbcli_state *cli;
- int num_locks;
+ int i;
+ uint8_t incr=0, last_incr=0;
+ uint8_t *val;
+ int count, loops;
fn = lp_parm_string(-1, "torture", "filename");
if (fn == NULL) {
@@ -334,44 +205,65 @@ BOOL torture_ping_pong(struct torture_context *torture)
return false;
}
- do_reads = lp_parm_bool(-1, "torture", "read", False);
- do_writes = lp_parm_bool(-1, "torture", "write", False);
+ do_reads = lp_parm_bool(-1, "torture", "read", False);
+ do_writes = lp_parm_bool(-1, "torture", "write", False);
+ lock_timeout = lp_parm_int(-1, "torture", "lock_timeout", 100000);
- if (!torture_open_connection_ev(&cli, 0, ev)) {
- DEBUG(0,("Could not open connection to share\n"));
- return False;
+ if (!torture_open_connection(&cli, 0)) {
+ DEBUG(0,("Could not open connection\n"));
+ return false;
}
-
- state = talloc_zero(mem_ctx, struct pingpong_state);
- state->ev = ev;
- state->tree = cli->tree;
- state->num_locks = num_locks;
- state->lock_timeout = lp_parm_int(-1, "torture", "lock_timeout", 100000);
- state->fnum = smbcli_open(state->tree, fn, O_RDWR|O_CREAT, DENY_NONE);
- if (state->fnum == -1) {
+ fd = smbcli_open(cli->tree, fn, O_RDWR|O_CREAT, DENY_NONE);
+ if (fd == -1) {
printf("Failed to open %s\n", fn);
exit(1);
}
- state->val = talloc_zero_array(state, uint8_t, state->num_locks+1);
- state->stage = LOCK_INITIAL;
- lock_send(state, False);
- tv = timeval_current();
+ write_byte(cli, fd, 0, num_locks);
+ lock_byte(cli, fd, 0, lock_timeout);
+
- printf("Running for %d seconds\n", timelimit);
- event_add_timed(ev, state, timeval_current_ofs(1, 0), report_rate, state);
- while (timeval_elapsed(&tv) < timelimit) {
- event_loop_once(ev);
+ start_timer();
+ val = talloc_zero_array(mem_ctx, uint8_t, num_locks);
+ i = 0;
+ count = 0;
+ loops = 0;
+ while (1) {
+ lock_byte(cli, fd, (i+1)%num_locks, lock_timeout);
- if (lock_failed) {
- DEBUG(0,("locking failed\n"));
- goto failed;
+ if (do_reads) {
+ uint8_t c;
+ read_byte(cli, fd, &c, i);
+ incr = c-val[i];
+ val[i] = c;
}
- }
+ if (do_writes) {
+ uint8_t c = val[i] + 1;
+ write_byte(cli, fd, c, i);
+ }
+
+ unlock_byte(cli, fd, i);
+
+ i = (i+1)%num_locks;
+ count++;
+ if (loops>num_locks && incr!=last_incr) {
+ last_incr = incr;
+ printf("data increment = %u\n", incr);
+ fflush(stdout);
+ }
+ if (end_timer() > 1.0) {
+ printf("%8u locks/sec\r",
+ (unsigned)(2*count/end_timer()));
+ fflush(stdout);
+ start_timer();
+ count=0;
+ }
+ loops++;
+ }
-failed:
talloc_free(mem_ctx);
return true;
}
+