Adding upstream version 2:4.17.12+dfsg.upstream/2%4.17.12+dfsgupstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 249 insertions, 0 deletions

diff --git a/source4/torture/ndr/ndr.h b/source4/torture/ndr/ndr.h
new file mode 100644
index 0000000..8d553b8
--- /dev/null
+++ b/source4/torture/ndr/ndr.h
@@ -0,0 +1,249 @@
+/*
+   Unix SMB/CIFS implementation.
+   SMB torture tester
+   Copyright (C) Jelmer Vernooij 2007
+
+   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/>.
+*/
+
+#ifndef __TORTURE_NDR_H__
+#define __TORTURE_NDR_H__
+
+#include "torture/torture.h"
+#include "librpc/ndr/libndr.h"
+#include "libcli/security/security.h"
+
+_PUBLIC_ struct torture_test *_torture_suite_add_ndr_pullpush_test(
+					struct torture_suite *suite,
+					const char *name,
+					ndr_pull_flags_fn_t pull_fn,
+					ndr_push_flags_fn_t push_fn,
+					ndr_print_fn_t print_fn,
+					ndr_print_function_t print_function,
+					const char *db_name,
+					DATA_BLOB db,
+					size_t struct_size,
+					int ndr_flags,
+					int flags,
+					const char *check_fn_name,
+					bool (*check_fn) (struct torture_context *, void *data));
+
+_PUBLIC_ struct torture_test *_torture_suite_add_ndr_pull_inout_test(
+					struct torture_suite *suite,
+					const char *name,
+					ndr_pull_flags_fn_t pull_fn,
+					ndr_print_function_t print_fn,
+					const char *db_in_name,
+					DATA_BLOB db_in,
+					const char *db_out_name,
+					DATA_BLOB db_out,
+					size_t struct_size,
+					int flags,
+					const char *check_fn_name,
+					bool (*check_fn) (struct torture_context *ctx, void *data));
+
+_PUBLIC_ struct torture_test *_torture_suite_add_ndr_pull_invalid_data_test(
+	struct torture_suite *suite,
+	const char *name,
+	ndr_pull_flags_fn_t pull_fn,
+	const char *db_name,
+	DATA_BLOB db,
+	size_t struct_size,
+	int ndr_flags,
+	int flags,
+	enum ndr_err_code ndr_err);
+
+#define torture_suite_add_ndr_pull_test(suite,name,data,check_fn) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pullpush_test(suite, #name, \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 NULL, \
+			 (ndr_print_fn_t)ndr_print_ ## name, \
+			 NULL, \
+			 #data, \
+			 data_blob_const(data, sizeof(data)), \
+			 sizeof(struct name), \
+			 NDR_SCALARS|NDR_BUFFERS, 0, \
+			 #check_fn, \
+			 check_fn_anon); \
+	} while(0)
+
+#define torture_suite_add_ndr_pull_invalid_data_test(suite,name,data,ndr_err) \
+		_torture_suite_add_ndr_pull_invalid_data_test( \
+			suite, \
+			#name, \
+			(ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			#data, \
+			data_blob_const(data, sizeof(data)), \
+			sizeof(struct name), \
+			NDR_SCALARS|NDR_BUFFERS, 0, \
+			ndr_err);
+
+#define torture_suite_add_ndr_pull_fn_test(suite,name,data,flags,check_fn) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pullpush_test(suite, #name "_" #flags, \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 NULL, \
+			 NULL, \
+			 (ndr_print_function_t)ndr_print_ ## name, \
+			 #data, \
+			 data_blob_const(data, sizeof(data)), \
+			 sizeof(struct name), \
+			 flags, 0, \
+			 #check_fn, \
+			 check_fn_anon); \
+	} while(0)
+
+#define torture_suite_add_ndr_pull_fn_test_flags(suite,name,data,flags,flags2,check_fn) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pullpush_test(suite, #name "_" #flags "_" #flags2, \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 NULL, \
+			 NULL, \
+			 (ndr_print_function_t)ndr_print_ ## name, \
+			 #data, \
+			 data_blob_const(data, sizeof(data)), \
+			 sizeof(struct name), \
+			 flags, flags2, \
+			 #check_fn, \
+			 check_fn_anon); \
+	} while(0)
+
+#define torture_suite_add_ndr_pull_validate_test(suite,name,data,check_fn) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pullpush_test(suite, #name "_VALIDATE", \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 (ndr_push_flags_fn_t)ndr_push_ ## name, \
+			 (ndr_print_fn_t)ndr_print_ ## name, \
+			 NULL, \
+			 #data, \
+			 data_blob_const(data, sizeof(data)), \
+			 sizeof(struct name), \
+			 NDR_SCALARS|NDR_BUFFERS, 0, \
+			 #check_fn, \
+			 check_fn_anon); \
+	} while(0)
+
+#define torture_suite_add_ndr_pull_validate_test_blob(suite,name,data_blob,check_fn) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pullpush_test(suite, #name "_VALIDATE", \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 (ndr_push_flags_fn_t)ndr_push_ ## name, \
+			 (ndr_print_fn_t)ndr_print_ ## name, \
+			 NULL, \
+			 #data_blob, \
+			 data_blob, \
+			 sizeof(struct name), \
+			 NDR_SCALARS|NDR_BUFFERS, 0, \
+			 #check_fn, \
+			 check_fn_anon); \
+	} while(0)
+
+#define torture_suite_add_ndr_pull_validate_test_b64(suite,name,tname,b64,check_fn) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pullpush_test(suite, #name "_" tname, \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 (ndr_push_flags_fn_t)ndr_push_ ## name, \
+			 (ndr_print_fn_t)ndr_print_ ## name, \
+			 NULL, \
+			 #b64, \
+			 base64_decode_data_blob_talloc(suite, b64), \
+			 sizeof(struct name), \
+			 NDR_SCALARS|NDR_BUFFERS, 0, \
+			 #check_fn, \
+			 check_fn_anon); \
+	} while(0)
+
+#define torture_suite_add_ndr_pullpush_fn_test_flags(suite,name,data,flags,flags2,check_fn) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pullpush_test(suite, #name, \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 (ndr_push_flags_fn_t)ndr_push_ ## name, \
+			 NULL, \
+			 (ndr_print_function_t)ndr_print_ ## name, \
+			 #data, \
+			 data_blob_const(data, sizeof(data)), \
+			 sizeof(struct name), \
+			 flags, flags2, \
+			 #check_fn, \
+			 check_fn_anon); \
+	} while(0)
+
+#define torture_suite_add_ndr_pull_io_test(suite,name,data_in,data_out,check_fn_out) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn_out; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pull_inout_test(suite, #name "_INOUT", \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 (ndr_print_function_t)ndr_print_ ## name, \
+			 #data_in, \
+			 data_blob_const(data_in, sizeof(data_in)), \
+			 #data_out, \
+			 data_blob_const(data_out, sizeof(data_out)), \
+			 sizeof(struct name), \
+			 0, \
+			 #check_fn_out, \
+			 check_fn_anon); \
+	} while(0)
+
+#define torture_suite_add_ndr_pull_io_test_flags(suite,name,data_in,data_out,flags,check_fn_out) \
+	do { \
+		bool (*check_fn_typed) (struct torture_context *, struct name *) = \
+			check_fn_out; \
+		bool (*check_fn_anon) (struct torture_context *, void *) = \
+			(bool (*) (struct torture_context *, void *)) check_fn_typed; \
+		_torture_suite_add_ndr_pull_inout_test(suite, #name "_INOUT_" #flags, \
+			 (ndr_pull_flags_fn_t)ndr_pull_ ## name, \
+			 (ndr_print_function_t)ndr_print_ ## name, \
+			 #data_in, \
+			 data_blob_const(data_in, sizeof(data_in)), \
+			 #data_out, \
+			 data_blob_const(data_out, sizeof(data_out)), \
+			 sizeof(struct name), \
+			 flags, \
+			 #check_fn_out, \
+			 check_fn_anon); \
+	} while(0)
+
+#endif /* __TORTURE_NDR_H__ */