1 files changed, 137 insertions, 0 deletions
diff --git a/source4/torture/smb2/maxwrite.c b/source4/torture/smb2/maxwrite.c
new file mode 100644
index 0000000..bc52ebc
--- /dev/null
+++ b/source4/torture/smb2/maxwrite.c
@@ -0,0 +1,137 @@
+/* 
+   Unix SMB/CIFS implementation.
+
+   test suite for SMB2 write operations
+
+   Copyright (C) Andrew Tridgell 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 3 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, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "includes.h"
+#include "librpc/gen_ndr/security.h"
+#include "libcli/smb2/smb2.h"
+#include "libcli/smb2/smb2_calls.h"
+#include "torture/torture.h"
+#include "torture/smb2/proto.h"
+
+#define FNAME "testmaxwrite.dat"
+
+/*
+  test writing
+*/
+static NTSTATUS torture_smb2_write(struct torture_context *tctx,
+				   struct smb2_tree *tree, 
+				   struct smb2_handle handle)
+{
+	struct smb2_write w;
+	struct smb2_read r;
+	NTSTATUS status;
+	int i, len;
+	int max = 80000000;
+	int min = 1;
+
+	while (max > min) {
+		TALLOC_CTX *tmp_ctx = talloc_new(tctx);
+
+
+		len = 1+(min+max)/2;
+
+		ZERO_STRUCT(w);
+		w.in.file.handle = handle;
+		w.in.offset      = 0;
+		w.in.data        = data_blob_talloc(tmp_ctx, NULL, len);
+
+		for (i=0;i<len;i++) {
+			w.in.data.data[i] = i % 256;
+		}
+
+		torture_comment(tctx, "trying to write %d bytes (min=%d max=%d)\n",
+		       len, min, max);
+
+		status = smb2_write(tree, &w);
+		if (!NT_STATUS_IS_OK(status)) {
+			torture_comment(tctx, "write failed - %s\n", nt_errstr(status));
+			max = len-1;
+			status = smb2_util_close(tree, handle);
+			if (!NT_STATUS_IS_OK(status)) {
+				/* vista bug */
+				torture_comment(tctx, "coping with server disconnect\n");
+				talloc_free(tree);
+				if (!torture_smb2_connection(tctx, &tree)) {
+					torture_comment(tctx, "failed to reconnect\n");
+					return NT_STATUS_NET_WRITE_FAULT;
+				}
+			}
+			status = torture_smb2_createfile(tctx, tree, FNAME, &handle);
+			if (!NT_STATUS_IS_OK(status)) {
+				torture_comment(tctx, "failed to create file handle\n");
+				talloc_free(tmp_ctx);
+				return status;
+			}
+			continue;
+		} else {
+			min = len;
+		}
+
+		
+		ZERO_STRUCT(r);
+		r.in.file.handle = handle;
+		r.in.length      = len;
+		r.in.offset      = 0;
+		
+		torture_comment(tctx, "reading %d bytes\n", len);
+
+		status = smb2_read(tree, tmp_ctx, &r);
+		if (!NT_STATUS_IS_OK(status)) {
+			torture_comment(tctx, "read failed - %s\n", nt_errstr(status));
+		} else if (w.in.data.length != r.out.data.length ||
+		    memcmp(w.in.data.data, r.out.data.data, len) != 0) {
+			torture_comment(tctx, "read data mismatch\n");
+		}
+
+		talloc_free(tmp_ctx);
+	}
+
+	torture_comment(tctx, "converged: len=%d\n", max);
+	smb2_util_close(tree, handle);
+	smb2_util_unlink(tree, FNAME);
+
+	return NT_STATUS_OK;
+}
+
+
+
+/* 
+   basic testing of SMB2 connection calls
+*/
+bool torture_smb2_maxwrite(struct torture_context *tctx)
+{
+	struct smb2_tree *tree;
+	struct smb2_handle h1;
+
+	if (!torture_smb2_connection(tctx, &tree)) {
+		return false;
+	}
+
+	torture_assert_ntstatus_ok(tctx,
+		torture_smb2_createfile(tctx, tree, FNAME, &h1),
+		"failed to create file handle");
+
+	torture_assert_ntstatus_ok(tctx,
+		torture_smb2_write(tctx, tree, h1),
+		"Write failed");
+
+	return true;
+}