source: branches/samba-3.0/source/torture/nbio.c@ 314

#define NBDEBUG 0

/* 
   Unix SMB/CIFS implementation.
   SMB torture tester
   Copyright (C) Andrew Tridgell 1997-1998
   
   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 2 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, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#include "includes.h"

#define MAX_FILES 1000

static char buf[70000];
extern int line_count;
extern int nbio_id;
static int nprocs;

static struct {
        int fd;
        int handle;
} ftable[MAX_FILES];

static struct children {
        double bytes_in, bytes_out;
        int line;
        int done;
} *children;

double nbio_total(void)
{
        int i;
        double total = 0;
        for (i=0;i<nprocs;i++) {
                total += children[i].bytes_out + children[i].bytes_in;
        }
        return total;
}

void nb_alarm(int ignore)
{
        int i;
        int lines=0, num_clients=0;
        if (nbio_id != -1) return;

        for (i=0;i<nprocs;i++) {
                lines += children[i].line;
                if (!children[i].done) num_clients++;
        }

        printf("%4d  %8d  %.2f MB/sec\r", num_clients, lines/nprocs, 1.0e-6 * nbio_total() / end_timer());

        signal(SIGALRM, nb_alarm);
        alarm(1);       
}

void nbio_shmem(int n)
{
        nprocs = n;
        children = (struct children *)shm_setup(sizeof(*children) * nprocs);
        if (!children) {
                printf("Failed to setup shared memory!\n");
                exit(1);
        }
}

#if 0
static int ne_find_handle(int handle)
{
        int i;
        children[nbio_id].line = line_count;
        for (i=0;i<MAX_FILES;i++) {
                if (ftable[i].handle == handle) return i;
        }
        return -1;
}
#endif

static int find_handle(int handle)
{
        int i;
        children[nbio_id].line = line_count;
        for (i=0;i<MAX_FILES;i++) {
                if (ftable[i].handle == handle) return i;
        }
        printf("(%d) ERROR: handle %d was not found\n", 
               line_count, handle);
        exit(1);

        return -1;              /* Not reached */
}


static struct cli_state *c;

static void sigsegv(int sig)
{
        char line[200];
        printf("segv at line %d\n", line_count);
        slprintf(line, sizeof(line), "/usr/X11R6/bin/xterm -e gdb /proc/%d/exe %d", 
                (int)getpid(), (int)getpid());
        system(line);
        exit(1);
}

void nb_setup(struct cli_state *cli)
{
        signal(SIGSEGV, sigsegv);
        c = cli;
        start_timer();
        children[nbio_id].done = 0;
}


void nb_unlink(const char *fname)
{
        if (!cli_unlink(c, fname)) {
#if NBDEBUG
                printf("(%d) unlink %s failed (%s)\n", 
                       line_count, fname, cli_errstr(c));
#endif
        }
}


void nb_createx(const char *fname, 
                unsigned create_options, unsigned create_disposition, int handle)
{
        int fd, i;
        uint32 desired_access;

        if (create_options & FILE_DIRECTORY_FILE) {
                desired_access = FILE_READ_DATA;
        } else {
                desired_access = FILE_READ_DATA | FILE_WRITE_DATA;
        }

        fd = cli_nt_create_full(c, fname, 0, 
                                desired_access,
                                0x0,
                                FILE_SHARE_READ|FILE_SHARE_WRITE, 
                                create_disposition, 
                                create_options, 0);
        if (fd == -1 && handle != -1) {