1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-24 21:34:56 +03:00
samba-mirror/source4/torture/util.c
Jelmer Vernooij 9d7856122e r19680: Allow specifying basedir for test data.
Create temporary directories in base directory.
(This used to be commit f111feb1d2)
2007-10-10 14:25:30 -05:00

163 lines
4.0 KiB
C

/*
Unix SMB/CIFS implementation.
SMB torture tester utility functions
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 "system/filesys.h"
#include "system/wait.h"
#include "torture/torture.h"
#include "libcli/raw/interfaces.h"
#include "libcli/raw/libcliraw.h"
/**
create a temporary directory.
*/
_PUBLIC_ NTSTATUS torture_temp_dir(TALLOC_CTX *mem_ctx, const char *prefix,
char **tempdir)
{
const char *basedir = lp_parm_string(-1, "torture", "basedir");
if (basedir == NULL) basedir = ".";
*tempdir = talloc_asprintf(mem_ctx, "%s/torture.tmp-%s.XXXXXX",
basedir, prefix);
if (mkdtemp(*tempdir) == NULL)
return NT_STATUS_UNSUCCESSFUL;
return NT_STATUS_OK;
}
/**
check if 2 NTTIMEs are equal.
*/
BOOL nt_time_equal(NTTIME *t1, NTTIME *t2)
{
return *t1 == *t2;
}
/**
* Provision a Samba installation using @param setupdir_script and start smbd.
*/
NTSTATUS torture_setup_server(TALLOC_CTX *mem_ctx,
const char *prefix,
const char *setupdir_script,
const char *smbd_path,
pid_t *smbd_pid)
{
char *tempdir;
NTSTATUS status;
pid_t pid;
int child_status;
char *configfile, *configparam;
pid_t closed_pid;
*smbd_pid = -1;
status = torture_temp_dir(mem_ctx, prefix, &tempdir);
if (NT_STATUS_IS_ERR(status)) {
return status;
}
if ((pid = fork()) == 0) {
execl(setupdir_script, setupdir_script, tempdir, NULL);
exit(1);
} else if (pid == -1) {
DEBUG(0, ("Unable to fork()\n"));
return NT_STATUS_UNSUCCESSFUL;
}
closed_pid = waitpid(pid, &child_status, 0);
if (closed_pid == -1) {
DEBUG(1, ("Error waiting for child"));
return NT_STATUS_UNSUCCESSFUL;
}
SMB_ASSERT(closed_pid == pid);
if (!WIFEXITED(child_status) || WEXITSTATUS(child_status) != 0) {
DEBUG(1, ("Invalid return code from setup script %s: %d\n",
setupdir_script,
WEXITSTATUS(child_status)));
return NT_STATUS_UNSUCCESSFUL;
}
configfile = talloc_asprintf(mem_ctx, "%s/etc/smb.conf",
tempdir);
if (!file_exist(configfile)) {
DEBUG(1, ("Setup script didn't create %s\n", configfile));
return NT_STATUS_UNSUCCESSFUL;
}
configparam = talloc_asprintf(mem_ctx, "--configfile=%s", configfile);
talloc_free(configfile);
if ((pid = fork()) == 0) {
execl(smbd_path, smbd_path, "-i", "--model=single",
configparam, NULL);
exit(1);
} else if (pid == -1) {
DEBUG(0, ("Unable to fork()\n"));
return NT_STATUS_UNSUCCESSFUL;
}
*smbd_pid = pid;
return NT_STATUS_OK;
}
NTSTATUS torture_second_tcon(TALLOC_CTX *mem_ctx,
struct smbcli_session *session,
const char *sharename,
struct smbcli_tree **res)
{
union smb_tcon tcon;
struct smbcli_tree *result;
TALLOC_CTX *tmp_ctx;
NTSTATUS status;
if ((tmp_ctx = talloc_new(mem_ctx)) == NULL) {
return NT_STATUS_NO_MEMORY;
}
result = smbcli_tree_init(session, tmp_ctx, False);
if (result == NULL) {
talloc_free(tmp_ctx);
return NT_STATUS_NO_MEMORY;
}
tcon.generic.level = RAW_TCON_TCONX;
tcon.tconx.in.flags = 0;
/* Ignore share mode security here */
tcon.tconx.in.password = data_blob(NULL, 0);
tcon.tconx.in.path = sharename;
tcon.tconx.in.device = "?????";
status = smb_raw_tcon(result, tmp_ctx, &tcon);
if (!NT_STATUS_IS_OK(status)) {
talloc_free(tmp_ctx);
return status;
}
result->tid = tcon.tconx.out.tid;
*res = talloc_steal(mem_ctx, result);
talloc_free(tmp_ctx);
return NT_STATUS_OK;
}