summaryrefslogtreecommitdiffstats
path: root/source4/torture/ndr/ndr.h
diff options
context:
space:
mode:
Diffstat (limited to 'source4/torture/ndr/ndr.h')
-rw-r--r--source4/torture/ndr/ndr.h249
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..82ccc69
--- /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,
+ ndr_flags_type ndr_flags,
+ libndr_flags 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,
+ libndr_flags 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,
+ ndr_flags_type ndr_flags,
+ libndr_flags 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__ */