mirror of
https://github.com/samba-team/samba.git
synced 2024-12-25 23:21:54 +03:00
0f62c11199
(This used to be commit c7d1d86853
)
781 lines
19 KiB
C
781 lines
19 KiB
C
/*
|
|
Unix SMB/CIFS implementation.
|
|
SMB torture tester
|
|
Copyright (C) Andrew Tridgell 1997-2003
|
|
Copyright (C) Jelmer Vernooij 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 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"
|
|
#include "libcli/raw/libcliraw.h"
|
|
#include "system/time.h"
|
|
#include "system/wait.h"
|
|
#include "system/filesys.h"
|
|
#include "libcli/raw/ioctl.h"
|
|
#include "libcli/libcli.h"
|
|
#include "lib/ldb/include/ldb.h"
|
|
#include "lib/events/events.h"
|
|
#include "libcli/resolve/resolve.h"
|
|
#include "auth/credentials/credentials.h"
|
|
#include "libcli/ldap/ldap_client.h"
|
|
#include "librpc/gen_ndr/ndr_nbt.h"
|
|
#include "torture/torture.h"
|
|
|
|
#include "torture/raw/proto.h"
|
|
#include "libcli/smb2/smb2.h"
|
|
#include "torture/smb2/proto.h"
|
|
#include "torture/rap/proto.h"
|
|
#include "torture/auth/proto.h"
|
|
#include "torture/local/proto.h"
|
|
#include "torture/nbench/proto.h"
|
|
#include "torture/ldap/proto.h"
|
|
#include "torture/nbt/proto.h"
|
|
#include "torture/libnet/proto.h"
|
|
#include "torture/util.h"
|
|
#include "build.h"
|
|
#include "dlinklist.h"
|
|
|
|
_PUBLIC_ int torture_nprocs=4;
|
|
_PUBLIC_ int torture_numops=10;
|
|
_PUBLIC_ int torture_entries=1000;
|
|
_PUBLIC_ int torture_failures=1;
|
|
_PUBLIC_ int torture_seed=0;
|
|
_PUBLIC_ int torture_numasync=100;
|
|
_PUBLIC_ BOOL use_oplocks;
|
|
static int procnum; /* records process count number when forking */
|
|
static struct smbcli_state *current_cli;
|
|
_PUBLIC_ BOOL use_level_II_oplocks;
|
|
_PUBLIC_ BOOL torture_showall = False;
|
|
|
|
|
|
|
|
|
|
static BOOL wait_lock(struct smbcli_state *c, int fnum, uint32_t offset, uint32_t len)
|
|
{
|
|
while (NT_STATUS_IS_ERR(smbcli_lock(c->tree, fnum, offset, len, -1, WRITE_LOCK))) {
|
|
if (!check_error(__location__, c, ERRDOS, ERRlock, NT_STATUS_LOCK_NOT_GRANTED)) return False;
|
|
}
|
|
return True;
|
|
}
|
|
|
|
|
|
static BOOL rw_torture(struct smbcli_state *c)
|
|
{
|
|
const char *lockfname = "\\torture.lck";
|
|
char *fname;
|
|
int fnum;
|
|
int fnum2;
|
|
pid_t pid2, pid = getpid();
|
|
int i, j;
|
|
uint8_t buf[1024];
|
|
BOOL correct = True;
|
|
|
|
fnum2 = smbcli_open(c->tree, lockfname, O_RDWR | O_CREAT | O_EXCL,
|
|
DENY_NONE);
|
|
if (fnum2 == -1)
|
|
fnum2 = smbcli_open(c->tree, lockfname, O_RDWR, DENY_NONE);
|
|
if (fnum2 == -1) {
|
|
printf("open of %s failed (%s)\n", lockfname, smbcli_errstr(c->tree));
|
|
return False;
|
|
}
|
|
|
|
|
|
for (i=0;i<torture_numops;i++) {
|
|
uint_t n = (uint_t)random()%10;
|
|
if (i % 10 == 0) {
|
|
printf("%d\r", i); fflush(stdout);
|
|
}
|
|
asprintf(&fname, "\\torture.%u", n);
|
|
|
|
if (!wait_lock(c, fnum2, n*sizeof(int), sizeof(int))) {
|
|
return False;
|
|
}
|
|
|
|
fnum = smbcli_open(c->tree, fname, O_RDWR | O_CREAT | O_TRUNC, DENY_ALL);
|
|
if (fnum == -1) {
|
|
printf("open failed (%s)\n", smbcli_errstr(c->tree));
|
|
correct = False;
|
|
break;
|
|
}
|
|
|
|
if (smbcli_write(c->tree, fnum, 0, &pid, 0, sizeof(pid)) != sizeof(pid)) {
|
|
printf("write failed (%s)\n", smbcli_errstr(c->tree));
|
|
correct = False;
|
|
}
|
|
|
|
for (j=0;j<50;j++) {
|
|
if (smbcli_write(c->tree, fnum, 0, buf,
|
|
sizeof(pid)+(j*sizeof(buf)),
|
|
sizeof(buf)) != sizeof(buf)) {
|
|
printf("write failed (%s)\n", smbcli_errstr(c->tree));
|
|
correct = False;
|
|
}
|
|
}
|
|
|
|
pid2 = 0;
|
|
|
|
if (smbcli_read(c->tree, fnum, &pid2, 0, sizeof(pid)) != sizeof(pid)) {
|
|
printf("read failed (%s)\n", smbcli_errstr(c->tree));
|
|
correct = False;
|
|
}
|
|
|
|
if (pid2 != pid) {
|
|
printf("data corruption!\n");
|
|
correct = False;
|
|
}
|
|
|
|
if (NT_STATUS_IS_ERR(smbcli_close(c->tree, fnum))) {
|
|
printf("close failed (%s)\n", smbcli_errstr(c->tree));
|
|
correct = False;
|
|
}
|
|
|
|
if (NT_STATUS_IS_ERR(smbcli_unlink(c->tree, fname))) {
|
|
printf("unlink failed (%s)\n", smbcli_errstr(c->tree));
|
|
correct = False;
|
|
}
|
|
|
|
if (NT_STATUS_IS_ERR(smbcli_unlock(c->tree, fnum2, n*sizeof(int), sizeof(int)))) {
|
|
printf("unlock failed (%s)\n", smbcli_errstr(c->tree));
|
|
correct = False;
|
|
}
|
|
free(fname);
|
|
}
|
|
|
|
smbcli_close(c->tree, fnum2);
|
|
smbcli_unlink(c->tree, lockfname);
|
|
|
|
printf("%d\n", i);
|
|
|
|
return correct;
|
|
}
|
|
|
|
static BOOL run_torture(struct smbcli_state *cli, int dummy)
|
|
{
|
|
BOOL ret;
|
|
|
|
ret = rw_torture(cli);
|
|
|
|
if (!torture_close_connection(cli)) {
|
|
ret = False;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
/*
|
|
see how many RPC pipes we can open at once
|
|
*/
|
|
static BOOL run_pipe_number(struct torture_context *torture)
|
|
{
|
|
struct smbcli_state *cli1;
|
|
const char *pipe_name = "\\WKSSVC";
|
|
int fnum;
|
|
int num_pipes = 0;
|
|
|
|
printf("starting pipenumber test\n");
|
|
if (!torture_open_connection(&cli1)) {
|
|
return False;
|
|
}
|
|
|
|
while(1) {
|
|
fnum = smbcli_nt_create_full(cli1->tree, pipe_name, 0, SEC_FILE_READ_DATA, FILE_ATTRIBUTE_NORMAL,
|
|
NTCREATEX_SHARE_ACCESS_READ|NTCREATEX_SHARE_ACCESS_WRITE, NTCREATEX_DISP_OPEN_IF, 0, 0);
|
|
|
|
if (fnum == -1) {
|
|
printf("Open of pipe %s failed with error (%s)\n", pipe_name, smbcli_errstr(cli1->tree));
|
|
break;
|
|
}
|
|
num_pipes++;
|
|
printf("%d\r", num_pipes);
|
|
fflush(stdout);
|
|
}
|
|
|
|
printf("pipe_number test - we can open %d %s pipes.\n", num_pipes, pipe_name );
|
|
torture_close_connection(cli1);
|
|
return True;
|
|
}
|
|
|
|
|
|
|
|
|
|
/*
|
|
open N connections to the server and just hold them open
|
|
used for testing performance when there are N idle users
|
|
already connected
|
|
*/
|
|
static BOOL torture_holdcon(struct torture_context *torture)
|
|
{
|
|
int i;
|
|
struct smbcli_state **cli;
|
|
int num_dead = 0;
|
|
|
|
printf("Opening %d connections\n", torture_numops);
|
|
|
|
cli = malloc_array_p(struct smbcli_state *, torture_numops);
|
|
|
|
for (i=0;i<torture_numops;i++) {
|
|
if (!torture_open_connection(&cli[i])) {
|
|
return False;
|
|
}
|
|
printf("opened %d connections\r", i);
|
|
fflush(stdout);
|
|
}
|
|
|
|
printf("\nStarting pings\n");
|
|
|
|
while (1) {
|
|
for (i=0;i<torture_numops;i++) {
|
|
NTSTATUS status;
|
|
if (cli[i]) {
|
|
status = smbcli_chkpath(cli[i]->tree, "\\");
|
|
if (!NT_STATUS_IS_OK(status)) {
|
|
printf("Connection %d is dead\n", i);
|
|
cli[i] = NULL;
|
|
num_dead++;
|
|
}
|
|
usleep(100);
|
|
}
|
|
}
|
|
|
|
if (num_dead == torture_numops) {
|
|
printf("All connections dead - finishing\n");
|
|
break;
|
|
}
|
|
|
|
printf(".");
|
|
fflush(stdout);
|
|
}
|
|
|
|
return True;
|
|
}
|
|
|
|
/*
|
|
test how many open files this server supports on the one socket
|
|
*/
|
|
static BOOL run_maxfidtest(struct smbcli_state *cli, int dummy)
|
|
{
|
|
#define MAXFID_TEMPLATE "\\maxfid\\fid%d\\maxfid.%d.%d"
|
|
char *fname;
|
|
int fnums[0x11000], i;
|
|
int retries=4, maxfid;
|
|
BOOL correct = True;
|
|
|
|
if (retries <= 0) {
|
|
printf("failed to connect\n");
|
|
return False;
|
|
}
|
|
|
|
if (smbcli_deltree(cli->tree, "\\maxfid") == -1) {
|
|
printf("Failed to deltree \\maxfid - %s\n",
|
|
smbcli_errstr(cli->tree));
|
|
return False;
|
|
}
|
|
if (NT_STATUS_IS_ERR(smbcli_mkdir(cli->tree, "\\maxfid"))) {
|
|
printf("Failed to mkdir \\maxfid, error=%s\n",
|
|
smbcli_errstr(cli->tree));
|
|
return False;
|
|
}
|
|
|
|
printf("Testing maximum number of open files\n");
|
|
|
|
for (i=0; i<0x11000; i++) {
|
|
if (i % 1000 == 0) {
|
|
asprintf(&fname, "\\maxfid\\fid%d", i/1000);
|
|
if (NT_STATUS_IS_ERR(smbcli_mkdir(cli->tree, fname))) {
|
|
printf("Failed to mkdir %s, error=%s\n",
|
|
fname, smbcli_errstr(cli->tree));
|
|
return False;
|
|
}
|
|
free(fname);
|
|
}
|
|
asprintf(&fname, MAXFID_TEMPLATE, i/1000, i,(int)getpid());
|
|
if ((fnums[i] = smbcli_open(cli->tree, fname,
|
|
O_RDWR|O_CREAT|O_TRUNC, DENY_NONE)) ==
|
|
-1) {
|
|
printf("open of %s failed (%s)\n",
|
|
fname, smbcli_errstr(cli->tree));
|
|
printf("maximum fnum is %d\n", i);
|
|
break;
|
|
}
|
|
free(fname);
|
|
printf("%6d\r", i);
|
|
}
|
|
printf("%6d\n", i);
|
|
i--;
|
|
|
|
maxfid = i;
|
|
|
|
printf("cleaning up\n");
|
|
for (i=0;i<maxfid/2;i++) {
|
|
asprintf(&fname, MAXFID_TEMPLATE, i/1000, i,(int)getpid());
|
|
if (NT_STATUS_IS_ERR(smbcli_close(cli->tree, fnums[i]))) {
|
|
printf("Close of fnum %d failed - %s\n", fnums[i], smbcli_errstr(cli->tree));
|
|
}
|
|
if (NT_STATUS_IS_ERR(smbcli_unlink(cli->tree, fname))) {
|
|
printf("unlink of %s failed (%s)\n",
|
|
fname, smbcli_errstr(cli->tree));
|
|
correct = False;
|
|
}
|
|
free(fname);
|
|
|
|
asprintf(&fname, MAXFID_TEMPLATE, (maxfid-i)/1000, maxfid-i,(int)getpid());
|
|
if (NT_STATUS_IS_ERR(smbcli_close(cli->tree, fnums[maxfid-i]))) {
|
|
printf("Close of fnum %d failed - %s\n", fnums[maxfid-i], smbcli_errstr(cli->tree));
|
|
}
|
|
if (NT_STATUS_IS_ERR(smbcli_unlink(cli->tree, fname))) {
|
|
printf("unlink of %s failed (%s)\n",
|
|
fname, smbcli_errstr(cli->tree));
|
|
correct = False;
|
|
}
|
|
free(fname);
|
|
|
|
printf("%6d %6d\r", i, maxfid-i);
|
|
}
|
|
printf("%6d\n", 0);
|
|
|
|
if (smbcli_deltree(cli->tree, "\\maxfid") == -1) {
|
|
printf("Failed to deltree \\maxfid - %s\n",
|
|
smbcli_errstr(cli->tree));
|
|
return False;
|
|
}
|
|
|
|
printf("maxfid test finished\n");
|
|
if (!torture_close_connection(cli)) {
|
|
correct = False;
|
|
}
|
|
return correct;
|
|
#undef MAXFID_TEMPLATE
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
sees what IOCTLs are supported
|
|
*/
|
|
static BOOL torture_ioctl_test(struct torture_context *torture)
|
|
{
|
|
struct smbcli_state *cli;
|
|
uint16_t device, function;
|
|
int fnum;
|
|
const char *fname = "\\ioctl.dat";
|
|
NTSTATUS status;
|
|
union smb_ioctl parms;
|
|
TALLOC_CTX *mem_ctx;
|
|
|
|
if (!torture_open_connection(&cli)) {
|
|
return False;
|
|
}
|
|
|
|
mem_ctx = talloc_init("ioctl_test");
|
|
|
|
printf("starting ioctl test\n");
|
|
|
|
smbcli_unlink(cli->tree, fname);
|
|
|
|
fnum = smbcli_open(cli->tree, fname, O_RDWR|O_CREAT|O_EXCL, DENY_NONE);
|
|
if (fnum == -1) {
|
|
printf("open of %s failed (%s)\n", fname, smbcli_errstr(cli->tree));
|
|
return False;
|
|
}
|
|
|
|
parms.ioctl.level = RAW_IOCTL_IOCTL;
|
|
parms.ioctl.in.file.fnum = fnum;
|
|
parms.ioctl.in.request = IOCTL_QUERY_JOB_INFO;
|
|
status = smb_raw_ioctl(cli->tree, mem_ctx, &parms);
|
|
printf("ioctl job info: %s\n", smbcli_errstr(cli->tree));
|
|
|
|
for (device=0;device<0x100;device++) {
|
|
printf("testing device=0x%x\n", device);
|
|
for (function=0;function<0x100;function++) {
|
|
parms.ioctl.in.request = (device << 16) | function;
|
|
status = smb_raw_ioctl(cli->tree, mem_ctx, &parms);
|
|
|
|
if (NT_STATUS_IS_OK(status)) {
|
|
printf("ioctl device=0x%x function=0x%x OK : %d bytes\n",
|
|
device, function, (int)parms.ioctl.out.blob.length);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!torture_close_connection(cli)) {
|
|
return False;
|
|
}
|
|
|
|
return True;
|
|
}
|
|
|
|
|
|
static void sigcont(int sig)
|
|
{
|
|
}
|
|
|
|
double torture_create_procs(BOOL (*fn)(struct smbcli_state *, int), BOOL *result)
|
|
{
|
|
int i, status;
|
|
volatile pid_t *child_status;
|
|
volatile BOOL *child_status_out;
|
|
int synccount;
|
|
int tries = 8;
|
|
double start_time_limit = 10 + (torture_nprocs * 1.5);
|
|
char **unc_list = NULL;
|
|
const char *p;
|
|
int num_unc_names = 0;
|
|
struct timeval tv;
|
|
|
|
*result = True;
|
|
|
|
synccount = 0;
|
|
|
|
signal(SIGCONT, sigcont);
|
|
|
|
child_status = (volatile pid_t *)shm_setup(sizeof(pid_t)*torture_nprocs);
|
|
if (!child_status) {
|
|
printf("Failed to setup shared memory\n");
|
|
return -1;
|
|
}
|
|
|
|
child_status_out = (volatile BOOL *)shm_setup(sizeof(BOOL)*torture_nprocs);
|
|
if (!child_status_out) {
|
|
printf("Failed to setup result status shared memory\n");
|
|
return -1;
|
|
}
|
|
|
|
p = lp_parm_string(-1, "torture", "unclist");
|
|
if (p) {
|
|
unc_list = file_lines_load(p, &num_unc_names, NULL);
|
|
if (!unc_list || num_unc_names <= 0) {
|
|
printf("Failed to load unc names list from '%s'\n", p);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
for (i = 0; i < torture_nprocs; i++) {
|
|
child_status[i] = 0;
|
|
child_status_out[i] = True;
|
|
}
|
|
|
|
tv = timeval_current();
|
|
|
|
for (i=0;i<torture_nprocs;i++) {
|
|
procnum = i;
|
|
if (fork() == 0) {
|
|
char *myname;
|
|
char *hostname=NULL, *sharename;
|
|
|
|
pid_t mypid = getpid();
|
|
srandom(((int)mypid) ^ ((int)time(NULL)));
|
|
|
|
asprintf(&myname, "CLIENT%d", i);
|
|
lp_set_cmdline("netbios name", myname);
|
|
free(myname);
|
|
|
|
|
|
if (unc_list) {
|
|
if (!smbcli_parse_unc(unc_list[i % num_unc_names],
|
|
NULL, &hostname, &sharename)) {
|
|
printf("Failed to parse UNC name %s\n",
|
|
unc_list[i % num_unc_names]);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
while (1) {
|
|
if (hostname) {
|
|
if (torture_open_connection_share(NULL,
|
|
¤t_cli,
|
|
hostname,
|
|
sharename,
|
|
NULL)) {
|
|
break;
|
|
}
|
|
} else if (torture_open_connection(¤t_cli)) {
|
|
break;
|
|
}
|
|
if (tries-- == 0) {
|
|
printf("pid %d failed to start\n", (int)getpid());
|
|
_exit(1);
|
|
}
|
|
msleep(100);
|
|
}
|
|
|
|
child_status[i] = getpid();
|
|
|
|
pause();
|
|
|
|
if (child_status[i]) {
|
|
printf("Child %d failed to start!\n", i);
|
|
child_status_out[i] = 1;
|
|
_exit(1);
|
|
}
|
|
|
|
child_status_out[i] = fn(current_cli, i);
|
|
_exit(0);
|
|
}
|
|
}
|
|
|
|
do {
|
|
synccount = 0;
|
|
for (i=0;i<torture_nprocs;i++) {
|
|
if (child_status[i]) synccount++;
|
|
}
|
|
if (synccount == torture_nprocs) break;
|
|
msleep(100);
|
|
} while (timeval_elapsed(&tv) < start_time_limit);
|
|
|
|
if (synccount != torture_nprocs) {
|
|
printf("FAILED TO START %d CLIENTS (started %d)\n", torture_nprocs, synccount);
|
|
*result = False;
|
|
return timeval_elapsed(&tv);
|
|
}
|
|
|
|
printf("Starting %d clients\n", torture_nprocs);
|
|
|
|
/* start the client load */
|
|
tv = timeval_current();
|
|
for (i=0;i<torture_nprocs;i++) {
|
|
child_status[i] = 0;
|
|
}
|
|
|
|
printf("%d clients started\n", torture_nprocs);
|
|
|
|
kill(0, SIGCONT);
|
|
|
|
for (i=0;i<torture_nprocs;i++) {
|
|
int ret;
|
|
while ((ret=waitpid(0, &status, 0)) == -1 && errno == EINTR) /* noop */ ;
|
|
if (ret == -1 || WEXITSTATUS(status) != 0) {
|
|
*result = False;
|
|
}
|
|
}
|
|
|
|
printf("\n");
|
|
|
|
for (i=0;i<torture_nprocs;i++) {
|
|
if (!child_status_out[i]) {
|
|
*result = False;
|
|
}
|
|
}
|
|
return timeval_elapsed(&tv);
|
|
}
|
|
|
|
#define FLAG_MULTIPROC 1
|
|
|
|
static struct {
|
|
const char *name;
|
|
BOOL (*fn)(struct torture_context *);
|
|
BOOL (*multi_fn)(struct smbcli_state *, int );
|
|
} builtin_torture_ops[] = {
|
|
/* benchmarking tests */
|
|
{"BENCH-HOLDCON", torture_holdcon, 0},
|
|
{"BENCH-NBENCH", torture_nbench, 0},
|
|
{"BENCH-TORTURE", NULL, run_torture},
|
|
{"BENCH-NBT", torture_bench_nbt, 0},
|
|
{"BENCH-WINS", torture_bench_wins, 0},
|
|
{"BENCH-CLDAP", torture_bench_cldap, 0},
|
|
{"BENCH-OPLOCK", torture_bench_oplock, 0},
|
|
|
|
/* RAW smb tests */
|
|
{"RAW-QFSINFO", torture_raw_qfsinfo, 0},
|
|
{"RAW-QFILEINFO", torture_raw_qfileinfo, 0},
|
|
{"RAW-SFILEINFO", torture_raw_sfileinfo, 0},
|
|
{"RAW-SFILEINFO-BUG", torture_raw_sfileinfo_bug, 0},
|
|
{"RAW-SEARCH", torture_raw_search, 0},
|
|
{"RAW-CLOSE", torture_raw_close, 0},
|
|
{"RAW-OPEN", torture_raw_open, 0},
|
|
{"RAW-MKDIR", torture_raw_mkdir, 0},
|
|
{"RAW-OPLOCK", torture_raw_oplock, 0},
|
|
{"RAW-NOTIFY", torture_raw_notify, 0},
|
|
{"RAW-MUX", torture_raw_mux, 0},
|
|
{"RAW-IOCTL", torture_raw_ioctl, 0},
|
|
{"RAW-CHKPATH", torture_raw_chkpath, 0},
|
|
{"RAW-UNLINK", torture_raw_unlink, 0},
|
|
{"RAW-READ", torture_raw_read, 0},
|
|
{"RAW-WRITE", torture_raw_write, 0},
|
|
{"RAW-LOCK", torture_raw_lock, 0},
|
|
{"RAW-CONTEXT", torture_raw_context, 0},
|
|
{"RAW-RENAME", torture_raw_rename, 0},
|
|
{"RAW-SEEK", torture_raw_seek, 0},
|
|
{"RAW-EAS", torture_raw_eas, 0},
|
|
{"RAW-STREAMS", torture_raw_streams, 0},
|
|
{"RAW-ACLS", torture_raw_acls, 0},
|
|
{"RAW-COMPOSITE", torture_raw_composite, 0},
|
|
{"RAW-SAMBA3HIDE", torture_samba3_hide, 0},
|
|
|
|
/* SMB2 tests */
|
|
{"SMB2-CONNECT", torture_smb2_connect, 0},
|
|
{"SMB2-SCAN", torture_smb2_scan, 0},
|
|
{"SMB2-SCANGETINFO", torture_smb2_getinfo_scan, 0},
|
|
{"SMB2-SCANSETINFO", torture_smb2_setinfo_scan, 0},
|
|
{"SMB2-SCANFIND", torture_smb2_find_scan, 0},
|
|
{"SMB2-GETINFO", torture_smb2_getinfo, 0},
|
|
{"SMB2-SETINFO", torture_smb2_setinfo, 0},
|
|
{"SMB2-FIND", torture_smb2_find, 0},
|
|
{"SMB2-LOCK", torture_smb2_lock, 0},
|
|
|
|
/* RAP tests */
|
|
{"RAP-BASIC", torture_rap_basic, 0},
|
|
|
|
/* protocol scanners */
|
|
{"SCAN-MAXFID", NULL, run_maxfidtest},
|
|
{"SCAN-PIPE_NUMBER", run_pipe_number, 0},
|
|
{"SCAN-IOCTL", torture_ioctl_test, 0},
|
|
{"SCAN-RAP", torture_rap_scan, 0},
|
|
{"SCAN-EAMAX", torture_max_eas, 0},
|
|
|
|
/* local (no server) testers */
|
|
{"LOCAL-TALLOC", torture_local_talloc, 0},
|
|
|
|
/* ldap testers */
|
|
{"LDAP-BASIC", torture_ldap_basic, 0},
|
|
{"LDAP-SCHEMA", torture_ldap_schema, 0},
|
|
{"LDAP-CLDAP", torture_cldap, 0},
|
|
|
|
/* nbt tests */
|
|
{"NBT-REGISTER", torture_nbt_register, 0},
|
|
{"NBT-WINS", torture_nbt_wins, 0},
|
|
{"NBT-DGRAM", torture_nbt_dgram, 0},
|
|
{"NBT-BROWSE", torture_nbt_browse, 0},
|
|
{"NBT-WINSREPLICATION-SIMPLE", torture_nbt_winsreplication_simple, 0},
|
|
{"NBT-WINSREPLICATION-REPLICA", torture_nbt_winsreplication_replica, 0},
|
|
{"NBT-WINSREPLICATION-OWNED", torture_nbt_winsreplication_owned, 0},
|
|
|
|
{NULL, NULL, 0}
|
|
};
|
|
|
|
/* ignore me */ static struct torture_suite *
|
|
(*suite_generators[]) (TALLOC_CTX *mem_ctx) =
|
|
{
|
|
torture_local_binding_string,
|
|
torture_ntlmssp,
|
|
torture_local_messaging,
|
|
torture_local_irpc,
|
|
torture_local_util_strlist,
|
|
torture_local_util_file,
|
|
torture_local_idtree,
|
|
torture_local_iconv,
|
|
torture_local_socket,
|
|
torture_pac,
|
|
torture_registry,
|
|
torture_local_resolve,
|
|
torture_local_sddl,
|
|
torture_local_ndr,
|
|
torture_local_event,
|
|
NULL
|
|
};
|
|
|
|
static void register_builtin_ops(void)
|
|
{
|
|
int i;
|
|
TALLOC_CTX *mem_ctx = talloc_autofree_context();
|
|
|
|
for (i = 0; builtin_torture_ops[i].name; i++) {
|
|
register_torture_op(builtin_torture_ops[i].name,
|
|
builtin_torture_ops[i].fn,
|
|
builtin_torture_ops[i].multi_fn);
|
|
}
|
|
|
|
for (i = 0; suite_generators[i]; i++)
|
|
torture_register_suite(suite_generators[i](mem_ctx));
|
|
}
|
|
|
|
struct torture_suite_list *torture_suites = NULL;
|
|
|
|
NTSTATUS torture_register_suite(struct torture_suite *suite)
|
|
{
|
|
struct torture_suite_list *p, *n;
|
|
|
|
n = talloc(talloc_autofree_context(), struct torture_suite_list);
|
|
n->suite = suite;
|
|
|
|
for (p = torture_suites; p; p = p->next) {
|
|
if (strcmp(p->suite->name, suite->name) == 0) {
|
|
/* Check for duplicates */
|
|
DEBUG(0,("There already is a suite registered with the name %s!\n", suite->name));
|
|
return NT_STATUS_OBJECT_NAME_COLLISION;
|
|
}
|
|
|
|
if (strcmp(p->suite->name, suite->name) < 0 &&
|
|
(!p->next || strcmp(p->next->suite->name, suite->name) > 0)) {
|
|
DLIST_ADD_AFTER(torture_suites, n, p);
|
|
return NT_STATUS_OK;
|
|
}
|
|
}
|
|
|
|
DLIST_ADD(torture_suites, n);
|
|
|
|
return NT_STATUS_OK;
|
|
}
|
|
|
|
static BOOL wrap_old_torture_fn(struct torture_context *torture,
|
|
const void *_fn)
|
|
{
|
|
BOOL (*fn)(struct torture_context *) = _fn;
|
|
return fn(torture);
|
|
}
|
|
|
|
static BOOL wrap_old_torture_multifn(struct torture_context *torture,
|
|
const void *_fn)
|
|
{
|
|
BOOL (*fn)(struct smbcli_state *, int ) = _fn;
|
|
BOOL result;
|
|
|
|
torture_create_procs(fn, &result);
|
|
|
|
return result;
|
|
}
|
|
|
|
_PUBLIC_ NTSTATUS register_torture_op(const char *name, BOOL (*fn)(struct torture_context *), BOOL (*multi_fn)(struct smbcli_state *, int ))
|
|
{
|
|
struct torture_suite *suite;
|
|
|
|
if (fn != NULL) {
|
|
suite = torture_suite_create(talloc_autofree_context(), name);
|
|
|
|
torture_suite_add_simple_tcase(suite, name,
|
|
wrap_old_torture_fn,
|
|
fn);
|
|
torture_register_suite(suite);
|
|
}
|
|
|
|
if (multi_fn != NULL) {
|
|
suite = torture_suite_create(talloc_autofree_context(), name);
|
|
|
|
torture_suite_add_simple_tcase(suite, name,
|
|
wrap_old_torture_multifn,
|
|
multi_fn);
|
|
torture_register_suite(suite);
|
|
}
|
|
|
|
|
|
return NT_STATUS_OK;
|
|
}
|
|
|
|
int torture_init(void)
|
|
{
|
|
init_module_fn static_init[] = STATIC_torture_MODULES;
|
|
init_module_fn *shared_init = load_samba_modules(NULL, "torture");
|
|
|
|
register_builtin_ops();
|
|
|
|
run_init_functions(static_init);
|
|
run_init_functions(shared_init);
|
|
|
|
talloc_free(shared_init);
|
|
|
|
return 0;
|
|
}
|