sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-03-08 04:58:40 +03:00
Code

r22836: started adding auto-reconnect logic to lockbench. This needs to be

Browse Source 
made async.
(This used to be commit 9e1ea3e09c80adb0b9809ee83e0c7b1ec9156e42)
This commit is contained in:
Andrew Tridgell 2007-05-14 01:05:09 +00:00 committed by Gerald (Jerry) Carter
parent 7df266a4f8
commit ba0424fddb
1 changed files with 37 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
source4/torture/raw/lockbench.c
View File

@ -44,7 +44,9 @@ static int nprocs;
static int lock_failed;
struct benchlock_state {
struct event_context *ev;
struct smbcli_state *cli;
int client_num;
int fnum;
int offset;
int count;
@ -76,6 +78,33 @@ static void lock_send(struct benchlock_state *state)
state->offset = (state->offset+1)%nprocs;
}
/*
reopen dead connections
*/
static void reopen_connection(struct benchlock_state *state)
{
do {
DEBUG(0,("reopening connection %u\n", state->client_num));
} while (!torture_open_connection_ev(&state->cli, state->client_num, state->ev));
state->fnum = smbcli_open(state->cli->tree, FNAME, O_RDWR|O_CREAT, DENY_NONE);
if (state->fnum == -1) {
printf("Failed to open %s on connection %d\n", FNAME, state->client_num);
exit(1);
}
state->lock[0].offset = state->offset;
state->io.lockx.in.ulock_cnt = 0;
state->req = smb_raw_lock_send(state->cli->tree, &state->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;
state->offset = (state->offset+1)%nprocs;
}
/*
called when a lock completes
*/
@ -87,6 +116,9 @@ static void lock_completion(struct smbcli_request *req)
if (!NT_STATUS_IS_OK(status)) {
lock_failed++;
DEBUG(0,("Lock failed - %s\n", nt_errstr(status)));
if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE)) {
reopen_connection(state);
}
} else {
state->count++;
lock_send(state);
@ -114,6 +146,8 @@ BOOL torture_bench_lock(struct torture_context *torture)
printf("Opening %d connections\n", nprocs);
for (i=0;i<nprocs;i++) {
state[i].client_num = i;
state[i].ev = ev;
if (!torture_open_connection_ev(&state[i].cli, i, ev)) {
return False;
}
@ -171,9 +205,10 @@ BOOL torture_bench_lock(struct torture_context *torture)
for (i=0;i<nprocs;i++) {
total += state[i].count;
}
if (torture_setting_bool(torture, "progress", true)) {
printf("%.2f ops/second\r", total/timeval_elapsed(&tv));
printf("%.2f ops/second (remaining=%u)\r",
total/timeval_elapsed(&tv),
(unsigned)(timelimit - timeval_elapsed(&tv)));
fflush(stdout);
}
}