1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-23 17:34:34 +03:00

r24768: redo the pingpong torture using sync functions to make the code simpler

This commit is contained in:
Ronnie Sahlberg 2007-08-29 06:58:25 +00:00 committed by Gerald (Jerry) Carter
parent b58677fa74
commit b4a6a4c129

View File

@ -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;
/* 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);
}
}
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;
}
try_again:
ZERO_STRUCT(lock);
io.lockx.in.ulock_cnt = 0;
io.lockx.in.lock_cnt = 1;
lock.count = 1;
lock.pid = state->tree->session->pid;
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 = lock_timeout;
io.lockx.in.locks = &lock;
io.lockx.in.file.fnum = state->fnum;
io.lockx.in.file.fnum = fd;
state->req = smb_raw_lock_send(state->tree, &io);
if (state->req == NULL) {
DEBUG(0,("Failed to setup lock\n"));
lock_failed++;
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;
}
}
if (!NT_STATUS_IS_OK(status)) {
DEBUG(0,("Lock failed\n"));
exit(1);
}
state->req->async.private = state;
state->req->async.fn = lock_completion;
}
/*
called when a write completes
*/
static void write_completion(struct smbcli_request *req)
void unlock_byte(struct smbcli_state *cli, int fd, int offset)
{
struct pingpong_state *state = (struct pingpong_state *)req->async.private;
NTSTATUS status = smbcli_request_simple_recv(req);
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 = 100000;
io.lockx.in.locks = &lock;
io.lockx.in.file.fnum = fd;
status = smb_raw_lock(cli->tree, &io);
if (!NT_STATUS_IS_OK(status)) {
DEBUG(0,("Unlock failed\n"));
exit(1);
}
}
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 = fd;
io.writex.in.offset = offset;
io.writex.in.wmode = 0;
io.writex.in.remaining = 0;
io.writex.in.count = 1;
io.writex.in.data = &c;
status = smb_raw_write(cli->tree, &io);
if (!NT_STATUS_IS_OK(status)) {
printf("write failed\n");
exit(1);
}
lock_send(state, False);
}
static void write_send(struct pingpong_state *state)
void read_byte(struct smbcli_state *cli, int fd, uint8_t *c, int offset)
{
union smb_write io;
union smb_read 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.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.generic.level = RAW_READ_READX;
io.readx.in.file.fnum = fd;
io.readx.in.mincnt = 1;
io.readx.in.maxcnt = 1;
io.readx.in.offset = offset;
io.readx.in.remaining = 0;
io.readx.in.read_for_execute = False;
io.readx.out.data = c;
status = smb_raw_read(cli->tree, &io);
if (!NT_STATUS_IS_OK(status)) {
printf("read 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)
static struct timeval tp1, tp2;
static void start_timer(void)
{
union smb_read io;
io.generic.level = RAW_READ_READX;
io.readx.in.file.fnum = state->fnum;
io.readx.in.mincnt = 1;
io.readx.in.maxcnt = 1;
io.readx.in.offset = state->lock_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;
gettimeofday(&tp1, NULL);
}
/*
called when a lock completes
*/
static void lock_completion(struct smbcli_request *req)
static double end_timer(void)
{
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;
}
}
if (!NT_STATUS_IS_OK(status)) {
lock_failed++;
return;
}
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;
}
lock_send(state, False);
}
static void report_rate(struct event_context *ev, struct timed_event *te,
struct timeval t, void *private_data)
{
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;
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;
struct smbcli_state *cli;
int num_locks;
TALLOC_CTX *mem_ctx = talloc_new(torture);
static BOOL do_reads;
static BOOL do_writes;
int lock_timeout;
int fd;
struct smbcli_state *cli;
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);
if (lock_failed) {
DEBUG(0,("locking failed\n"));
goto failed;
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 (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;
}