nbio.c

samba最新软件

	if (retry) {
		/* Reopening after a disconnect. */
		io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN_IF;
	} else
	if (f != NULL &&
	    f->cp.create_disposition == NTCREATEX_DISP_CREATE &&
	    NT_STATUS_IS_OK(status))
	{
		/* Reopening after nb_createx() error. */
		io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN_IF;
	}

	ret = smb_raw_open(c->tree, mem_ctx, &io);

	talloc_free(mem_ctx);

	if (!check_status("NTCreateX", status, ret))
		return false;

	if (!NT_STATUS_IS_OK(ret))
		return true;

	if (f == NULL) {
		f = talloc (NULL, struct ftable);
		f->locks = NULL;
		nb_set_createx_params(f, fname, create_options, create_disposition, handle);
		DLIST_ADD_END(ftable, f, struct ftable *);
	}

	f->handle = handle;
	f->fd = io.ntcreatex.out.file.fnum;

	return true;
}

bool nb_createx(const char *fname, 
	       uint_t create_options, uint_t create_disposition, int handle,
	       NTSTATUS status)
{
	return nb_do_createx(NULL, fname, create_options, create_disposition, handle, status, false);
}

bool nb_writex(int handle, off_t offset, int size, int ret_size, NTSTATUS status)
{
	union smb_write io;
	int i;
	NTSTATUS ret;
	uint8_t *buf;

	i = find_handle(handle, NULL);

	if (bypass_io)
		return true;

	buf = malloc(size);
	memset(buf, 0xab, size);

	io.writex.level = RAW_WRITE_WRITEX;
	io.writex.in.file.fnum = i;
	io.writex.in.wmode = 0;
	io.writex.in.remaining = 0;
	io.writex.in.offset = offset;
	io.writex.in.count = size;
	io.writex.in.data = buf;

	ret = smb_raw_write(c->tree, &io);

	free(buf);

	if (!check_status("WriteX", status, ret))
		return false;

	if (NT_STATUS_IS_OK(ret) && io.writex.out.nwritten != ret_size) {
		printf("[%d] Warning: WriteX got count %d expected %d\n", 
		       nbench_line_count,
		       io.writex.out.nwritten, ret_size);
	}	

	children[nbio_id].bytes += ret_size;

	return true;
}

bool nb_write(int handle, off_t offset, int size, int ret_size, NTSTATUS status)
{
	union smb_write io;
	int i;
	NTSTATUS ret;
	uint8_t *buf;

	i = find_handle(handle, NULL);

	if (bypass_io)
		return true;

	buf = malloc(size);

	memset(buf, 0x12, size);

	io.write.level = RAW_WRITE_WRITE;
	io.write.in.file.fnum = i;
	io.write.in.remaining = 0;
	io.write.in.offset = offset;
	io.write.in.count = size;
	io.write.in.data = buf;

	ret = smb_raw_write(c->tree, &io);

	free(buf);

	if (!check_status("Write", status, ret))
		return false;

	if (NT_STATUS_IS_OK(ret) && io.write.out.nwritten != ret_size) {
		printf("[%d] Warning: Write got count %d expected %d\n", 
		       nbench_line_count,
		       io.write.out.nwritten, ret_size);
	}	

	children[nbio_id].bytes += ret_size;

	return true;
}

static bool nb_do_lockx(bool relock, int handle, off_t offset, int size, NTSTATUS status)
{
	union smb_lock io;
	int i;
	NTSTATUS ret;
	struct smb_lock_entry lck;
	struct ftable *f;

	i = find_handle(handle, &f);

	lck.pid = getpid();
	lck.offset = offset;
	lck.count = size;

	io.lockx.level = RAW_LOCK_LOCKX;
	io.lockx.in.file.fnum = i;
	io.lockx.in.mode = 0;
	io.lockx.in.timeout = 0;
	io.lockx.in.ulock_cnt = 0;
	io.lockx.in.lock_cnt = 1;
	io.lockx.in.locks = &lck;

	ret = smb_raw_lock(c->tree, &io);

	if (!check_status("LockX", status, ret))
		return false;

	if (f != NULL &&
	    !relock)
	{
		struct lock_info *linfo;
		linfo = talloc (f, struct lock_info);
		linfo->offset = offset;
		linfo->size = size;
		DLIST_ADD_END(f->locks, linfo, struct lock_info *);
	}

	return true;
}

bool nb_lockx(int handle, off_t offset, int size, NTSTATUS status)
{
	return nb_do_lockx(false, handle, offset, size, status);
}

bool nb_unlockx(int handle, uint_t offset, int size, NTSTATUS status)
{
	union smb_lock io;
	int i;
	NTSTATUS ret;
	struct smb_lock_entry lck;
	struct ftable *f;

	i = find_handle(handle, &f);

	lck.pid = getpid();
	lck.offset = offset;
	lck.count = size;

	io.lockx.level = RAW_LOCK_LOCKX;
	io.lockx.in.file.fnum = i;
	io.lockx.in.mode = 0;
	io.lockx.in.timeout = 0;
	io.lockx.in.ulock_cnt = 1;
	io.lockx.in.lock_cnt = 0;
	io.lockx.in.locks = &lck;

	ret = smb_raw_lock(c->tree, &io);

	if (!check_status("UnlockX", status, ret))
		return false;

	if (f != NULL) {
		struct lock_info *linfo;
		linfo = find_lock(f->locks, offset, size);
		if (linfo != NULL)
			DLIST_REMOVE(f->locks, linfo);
		else
			printf("nb_unlockx: unknown lock (%d)\n", handle);
	}

	return true;
}

bool nb_readx(int handle, off_t offset, int size, int ret_size, NTSTATUS status)
{
	union smb_read io;
	int i;
	NTSTATUS ret;
	uint8_t *buf;

	i = find_handle(handle, NULL);

	if (bypass_io)
		return true;

	buf = malloc(size);

	io.readx.level = RAW_READ_READX;
	io.readx.in.file.fnum = i;
	io.readx.in.offset    = offset;
	io.readx.in.mincnt    = size;
	io.readx.in.maxcnt    = size;
	io.readx.in.remaining = 0;
	io.readx.in.read_for_execute = false;
	io.readx.out.data     = buf;

	ret = smb_raw_read(c->tree, &io);

	free(buf);

	if (!check_status("ReadX", status, ret))
		return false;

	if (NT_STATUS_IS_OK(ret) && io.readx.out.nread != ret_size) {
		printf("[%d] ERROR: ReadX got count %d expected %d\n", 
		       nbench_line_count,
		       io.readx.out.nread, ret_size);
		nb_exit(1);
	}	

	children[nbio_id].bytes += ret_size;

	return true;
}

bool nb_close(int handle, NTSTATUS status)
{
	NTSTATUS ret;
	union smb_close io;
	int i;

	i = find_handle(handle, NULL);

	io.close.level = RAW_CLOSE_CLOSE;
	io.close.in.file.fnum = i;
	io.close.in.write_time = 0;

	ret = smb_raw_close(c->tree, &io);

	if (!check_status("Close", status, ret))
		return false;

	if (NT_STATUS_IS_OK(ret)) {
		struct ftable *f = find_ftable(handle);
		DLIST_REMOVE(ftable, f);
		talloc_free(f);
	}

	return true;
}

bool nb_rmdir(const char *dname, NTSTATUS status, bool retry)
{
	NTSTATUS ret;
	struct smb_rmdir io;

	io.in.path = dname;

	ret = smb_raw_rmdir(c->tree, &io);

	if (!retry)
		return check_status("Rmdir", status, ret);

	return true;
}

bool nb_mkdir(const char *dname, NTSTATUS status, bool retry)
{
	union smb_mkdir io;

	io.mkdir.level = RAW_MKDIR_MKDIR;
	io.mkdir.in.path = dname;

	/* NOTE! no error checking. Used for base fileset creation */
	smb_raw_mkdir(c->tree, &io);

	return true;
}

bool nb_rename(const char *old, const char *new, NTSTATUS status, bool retry)
{
	NTSTATUS ret;
	union smb_rename io;

	io.generic.level = RAW_RENAME_RENAME;
	io.rename.in.attrib = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY;
	io.rename.in.pattern1 = old;
	io.rename.in.pattern2 = new;

	ret = smb_raw_rename(c->tree, &io);

	if (!retry)
		return check_status("Rename", status, ret);

	return true;
}


bool nb_qpathinfo(const char *fname, int level, NTSTATUS status)
{
	union smb_fileinfo io;
	TALLOC_CTX *mem_ctx;
	NTSTATUS ret;

	mem_ctx = talloc_init("nb_qpathinfo");

	io.generic.level = level;
	io.generic.in.file.path = fname;

	ret = smb_raw_pathinfo(c->tree, mem_ctx, &io);

	talloc_free(mem_ctx);

	return check_status("Pathinfo", status, ret);
}


bool nb_qfileinfo(int fnum, int level, NTSTATUS status)
{
	union smb_fileinfo io;
	TALLOC_CTX *mem_ctx;
	NTSTATUS ret;
	int i;

	i = find_handle(fnum, NULL);

	mem_ctx = talloc_init("nb_qfileinfo");

	io.generic.level = level;
	io.generic.in.file.fnum = i;

	ret = smb_raw_fileinfo(c->tree, mem_ctx, &io);

	talloc_free(mem_ctx);

	return check_status("Fileinfo", status, ret);
}

bool nb_sfileinfo(int fnum, int level, NTSTATUS status)
{
	union smb_setfileinfo io;
	NTSTATUS ret;
	int i;

	if (level != RAW_SFILEINFO_BASIC_INFORMATION) {
		printf("[%d] Warning: setfileinfo level %d not handled\n", nbench_line_count, level);
		return true;
	}

	ZERO_STRUCT(io);

	i = find_handle(fnum, NULL);

	io.generic.level = level;
	io.generic.in.file.fnum = i;
	unix_to_nt_time(&io.basic_info.in.create_time, time(NULL));
	unix_to_nt_time(&io.basic_info.in.access_time, 0);
	unix_to_nt_time(&io.basic_info.in.write_time, 0);
	unix_to_nt_time(&io.basic_info.in.change_time, 0);
	io.basic_info.in.attrib = 0;

	ret = smb_raw_setfileinfo(c->tree, &io);

	return check_status("Setfileinfo", status, ret);
}

bool nb_qfsinfo(int level, NTSTATUS status)
{
	union smb_fsinfo io;
	TALLOC_CTX *mem_ctx;
	NTSTATUS ret;

	mem_ctx = talloc_init("smbcli_dskattr");

	io.generic.level = level;
	ret = smb_raw_fsinfo(c->tree, mem_ctx, &io);

	talloc_free(mem_ctx);
	
	return check_status("Fsinfo", status, ret);	
}

/* callback function used for trans2 search */
static bool findfirst_callback(void *private, const union smb_search_data *file)
{
	return true;
}

bool nb_findfirst(const char *mask, int level, int maxcnt, int count, NTSTATUS status)
{
	union smb_search_first io;
	TALLOC_CTX *mem_ctx;
	NTSTATUS ret;

	mem_ctx = talloc_init("smbcli_dskattr");

	io.t2ffirst.level = RAW_SEARCH_TRANS2;
	io.t2ffirst.data_level = level;
	io.t2ffirst.in.max_count = maxcnt;
	io.t2ffirst.in.search_attrib = FILE_ATTRIBUTE_DIRECTORY;
	io.t2ffirst.in.pattern = mask;
	io.t2ffirst.in.flags = FLAG_TRANS2_FIND_CLOSE;
	io.t2ffirst.in.storage_type = 0;
			
	ret = smb_raw_search_first(c->tree, mem_ctx, &io, NULL, findfirst_callback);

	talloc_free(mem_ctx);

	if (!check_status("Search", status, ret))
		return false;

	if (NT_STATUS_IS_OK(ret) && io.t2ffirst.out.count != count) {
		printf("[%d] Warning: got count %d expected %d\n", 
		       nbench_line_count,
		       io.t2ffirst.out.count, count);
	}

	return true;
}

bool nb_flush(int fnum, NTSTATUS status)
{
	union smb_flush io;
	NTSTATUS ret;
	int i;
	i = find_handle(fnum, NULL);

	io.flush.level		= RAW_FLUSH_FLUSH;
	io.flush.in.file.fnum	= i;

	ret = smb_raw_flush(c->tree, &io);

	return check_status("Flush", status, ret);
}

void nb_sleep(int usec, NTSTATUS status)
{
	usleep(usec);
}

bool nb_deltree(const char *dname, bool retry)
{
	int total_deleted;

	smb_raw_exit(c->session);

	while (ftable) {
		struct ftable *f = ftable;
		DLIST_REMOVE(ftable, f);
		talloc_free (f);
	}

	total_deleted = smbcli_deltree(c->tree, dname);

	if (total_deleted == -1) {
		printf("Failed to cleanup tree %s - exiting\n", dname);
		nb_exit(1);
	}

	smbcli_rmdir(c->tree, dname);

	return true;
}


void nb_exit(int status)
{
	children[nbio_id].connected = false;
	printf("[%d] client %d exiting with status %d\n",
	       nbench_line_count, nbio_id, status);
	exit(status);
}