From 38fa35571ec9c86dfcd22ed603f9249c071771d8 Mon Sep 17 00:00:00 2001
From: Andrew Tridgell <tridge@samba.org>
Date: Tue, 5 Feb 2002 01:31:47 +0000
Subject: rewrote smbtorture to use the new dbench 2 format and methods (This
 used to be commit 36f816a5e12a80f0184f43cbd44ef5fac53fcb81)

---
 source3/torture/nbio.c | 271 +++++++++++++++++++++++++++++--------------------
 1 file changed, 159 insertions(+), 112 deletions(-)

(limited to 'source3/torture/nbio.c')

diff --git a/source3/torture/nbio.c b/source3/torture/nbio.c
index 678f3ab556..0785c4e196 100644
--- a/source3/torture/nbio.c
+++ b/source3/torture/nbio.c
@@ -28,12 +28,68 @@
 
 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 {
+	double bytes_in, bytes_out;
+	int line;
+} *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(void)
+{
+	int i;
+	int lines=0;
+	if (nbio_id != -1) return;
+
+	for (i=0;i<nprocs;i++) {
+		lines += children[i].line;
+	}
+
+	printf("%8d  %.2f MB/sec\r", lines/nprocs, 1.0e-6 * nbio_total() / end_timer());
+
+	signal(SIGALRM, nb_alarm);
+	alarm(1);	
+}
+
+void nbio_shmem(int n)
+{
+	nprocs = n;
+	children = shm_setup(sizeof(*children) * nprocs);
+	if (!children) {
+		printf("Failed to setup shared memory!\n");
+		exit(1);
+	}
+}
+
+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("ERROR: handle %d was not found\n", 
+	       line_count, handle);
+	exit(1);
+}
+
+
 static struct cli_state *c;
 
 static void sigsegv(int sig)
@@ -50,13 +106,12 @@ void nb_setup(struct cli_state *cli)
 {
 	signal(SIGSEGV, sigsegv);
 	c = cli;
+	start_timer();
 }
 
 
 void nb_unlink(char *fname)
 {
-	strupper(fname);
-
 	if (!cli_unlink(c, fname)) {
 #if NBDEBUG
 		printf("(%d) unlink %s failed (%s)\n", 
@@ -65,175 +120,167 @@ void nb_unlink(char *fname)
 	}
 }
 
-void nb_open(char *fname, int handle, int size)
+
+void nb_createx(char *fname, 
+		unsigned create_options, unsigned create_disposition, int handle)
 {
 	int fd, i;
-	int flags = O_RDWR|O_CREAT;
-	size_t st_size;
 	static int count;
 
-	strupper(fname);
-
-	if (size == 0) flags |= O_TRUNC;
-
-	fd = cli_open(c, fname, flags, DENY_NONE);
-	if (fd == -1) {
-#if NBDEBUG
-		printf("(%d) open %s failed for handle %d (%s)\n", 
-		       line_count, fname, handle, cli_errstr(c));
-#endif
-		return;
+	fd = cli_nt_create_full(c, fname, 
+				FILE_READ_DATA | FILE_WRITE_DATA,
+				0,
+				FILE_SHARE_READ|FILE_SHARE_WRITE, 
+				create_disposition, 
+				create_options);
+	if (fd == -1 && handle != -1) {
+		printf("ERROR: cli_nt_create_full failed for %s - %s\n",
+		       fname, cli_errstr(c));
+		exit(1);
 	}
-	cli_getattrE(c, fd, NULL, &st_size, NULL, NULL, NULL);
-	if (size > st_size) {
-#if NBDEBUG
-		printf("(%d) needs expanding %s to %d from %d\n", 
-		       line_count, fname, size, (int)st_size);
-#endif
-	} else if (size < st_size) {
-#if NBDEBUG
-		printf("(%d) needs truncating %s to %d from %d\n", 
-		       line_count, fname, size, (int)st_size);
-#endif
+	if (fd != -1 && handle == -1) {
+		printf("ERROR: cli_nt_create_full succeeded for %s\n", fname);
+		exit(1);
 	}
+	if (fd == -1) return;
+
 	for (i=0;i<MAX_FILES;i++) {
 		if (ftable[i].handle == 0) break;
 	}
 	if (i == MAX_FILES) {
-		printf("file table full for %s\n", fname);
+		printf("(%d) file table full for %s\n", line_count, 
+		       fname);
 		exit(1);
 	}
 	ftable[i].handle = handle;
 	ftable[i].fd = fd;
-	if (count++ % 100 == 0) {
-		printf(".");
-	}
 }
 
-void nb_write(int handle, int size, int offset)
+void nb_writex(int handle, int offset, int size, int ret_size)
 {
 	int i;
 
 	if (buf[0] == 0) memset(buf, 1, sizeof(buf));
 
-	for (i=0;i<MAX_FILES;i++) {
-		if (ftable[i].handle == handle) break;
-	}
-	if (i == MAX_FILES) {
-#if NBDEBUG
-		printf("(%d) nb_write: handle %d was not open size=%d ofs=%d\n", 
-		       line_count, handle, size, offset);
-#endif
-		return;
-	}
-	if (cli_smbwrite(c, ftable[i].fd, buf, offset, size) != size) {
-		printf("(%d) write failed on handle %d, fd %d \
+	i = find_handle(handle);
+	if (cli_write(c, ftable[i].fd, 0, buf, offset, size) != ret_size) {
+		printf("(%d) ERROR: write failed on handle %d, fd %d \
 errno %d (%s)\n", line_count, handle, ftable[i].fd, errno, strerror(errno));
-		if (errno == ENOSPC) {
-			printf("Halting.\n");
-			fflush(stdout);
-			fflush(stderr);
-			exit(3);
-		}
+		exit(1);
 	}
+
+	children[nbio_id].bytes_out += ret_size;
 }
 
-void nb_read(int handle, int size, int offset)
+void nb_readx(int handle, int offset, int size, int ret_size)
 {
 	int i, ret;
 
-	for (i=0;i<MAX_FILES;i++) {
-		if (ftable[i].handle == handle) break;
-	}
-	if (i == MAX_FILES) {
-		printf("(%d) nb_read: handle %d was not open size=%d ofs=%d\n", 
-		       line_count, handle, size, offset);
-		return;
-	}
-	if ((ret=cli_read(c, ftable[i].fd, buf, offset, size)) != size) {
-#if NBDEBUG
-		printf("(%d) read failed on handle %d ofs=%d size=%d res=%d fd %d errno %d (%s)\n",
+	i = find_handle(handle);
+	if ((ret=cli_read(c, ftable[i].fd, buf, offset, size)) != ret_size) {
+		printf("(%d) ERROR: read failed on handle %d ofs=%d size=%d res=%d fd %d errno %d (%s)\n",
 			line_count, handle, offset, size, ret, ftable[i].fd, errno, strerror(errno));
-#endif
+		exit(1);
 	}
+	children[nbio_id].bytes_in += ret_size;
 }
 
 void nb_close(int handle)
 {
 	int i;
-	for (i=0;i<MAX_FILES;i++) {
-		if (ftable[i].handle == handle) break;
-	}
-	if (i == MAX_FILES) {
-		printf("(%d) nb_close: handle %d was not open\n", 
-		       line_count, handle);
-		return;
+	i = find_handle(handle);
+	if (!cli_close(c, ftable[i].fd)) {
+		printf("(%d) close failed on handle %d\n", line_count, handle);
+		exit(1);
 	}
-	cli_close(c, ftable[i].fd);
 	ftable[i].handle = 0;
 }
 
-void nb_mkdir(char *fname)
-{
-	strupper(fname);
-
-	if (!cli_mkdir(c, fname)) {
-#if NBDEBUG
-		printf("mkdir %s failed (%s)\n", 
-		       fname, cli_errstr(c));
-#endif
-	}
-}
-
 void nb_rmdir(char *fname)
 {
-	strupper(fname);
-
 	if (!cli_rmdir(c, fname)) {
-#if NBDEBUG
-		printf("rmdir %s failed (%s)\n", 
+		printf("ERROR: rmdir %s failed (%s)\n", 
 		       fname, cli_errstr(c));
-#endif
+		exit(1);
 	}
 }
 
 void nb_rename(char *old, char *new)
 {
-	strupper(old);
-	strupper(new);
-
 	if (!cli_rename(c, old, new)) {
-#if NBDEBUG
-		printf("rename %s %s failed (%s)\n", 
+		printf("ERROR: rename %s %s failed (%s)\n", 
 		       old, new, cli_errstr(c));
-#endif
+		exit(1);
 	}
 }
 
 
-void nb_stat(char *fname, int size)
+void nb_qpathinfo(char *fname)
+{
+	cli_qpathinfo(c, fname, NULL, NULL, NULL, NULL, NULL);
+}
+
+void nb_qfileinfo(int fnum)
+{
+	int i;
+	i = find_handle(fnum);
+	cli_qfileinfo(c, ftable[i].fd, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
+}
+
+void nb_qfsinfo(int level)
 {
-	size_t st_size;
+	int bsize, total, avail;
+	/* this is not the right call - we need cli_qfsinfo() */
+	cli_dskattr(c, &bsize, &total, &avail);
+}
 
-	strupper(fname);
+static void find_fn(file_info *finfo, const char *name, void *state)
+{
+	/* noop */
+}
 
-	if (!cli_getatr(c, fname, NULL, &st_size, NULL)) {
-#if NBDEBUG
-		printf("(%d) nb_stat: %s size=%d %s\n", 
-		       line_count, fname, size, cli_errstr(c));
-#endif
-		return;
-	}
-	if (st_size != size) {
-#if NBDEBUG
-		printf("(%d) nb_stat: %s wrong size %d %d\n", 
-		       line_count, fname, (int)st_size, size);
-#endif
+void nb_findfirst(char *mask)
+{
+	cli_list(c, mask, 0, find_fn, NULL);
+}
+
+void nb_flush(int fnum)
+{
+	int i;
+	i = find_handle(fnum);
+	/* hmmm, we don't have cli_flush() yet */
+}
+
+static int total_deleted;
+
+static void delete_fn(file_info *finfo, const char *name, void *state)
+{
+	char *s, *n;
+	if (finfo->name[0] == '.') return;
+
+	n = strdup(name);
+	n[strlen(n)-1] = 0;
+	asprintf(&s, "%s%s", n, finfo->name);
+	if (finfo->mode & aDIR) {
+		nb_deltree(s);
+	} else {
+		total_deleted++;
+		nb_unlink(s);
 	}
+	free(s);
+	free(n);
 }
 
-void nb_create(char *fname, int size)
+void nb_deltree(char *dname)
 {
-	nb_open(fname, 5000, size);
-	nb_close(5000);
+	char *mask;
+	asprintf(&mask, "%s\\*", dname);
+
+	total_deleted = 0;
+
+	cli_list(c, mask, aDIR, delete_fn, NULL);
+	free(mask);
+	nb_rmdir(dname);
+
+	if (total_deleted) printf("WARNING: Cleaned up %d files\n", total_deleted);
 }
-- 
cgit