1 /*
2    Unix SMB/CIFS implementation.
3 
4    test suite for DCE/RPC verification trailer parsing
5 
6    Copyright (C) David Disseldorp 2014
7 
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 3 of the License, or
11    (at your option) any later version.
12 
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17 
18    You should have received a copy of the GNU General Public License
19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20 */
21 
22 #include "includes.h"
23 #include <unistd.h>
24 
25 #include "librpc/gen_ndr/security.h"
26 #include "lib/param/param.h"
27 #include "lib/util/dlinklist.h"
28 #include "libcli/resolve/resolve.h"
29 #include "librpc/gen_ndr/ndr_dcerpc.h"
30 #include "librpc/rpc/rpc_common.h"
31 #include "torture/torture.h"
32 #include "torture/local/proto.h"
33 
34 /* VT blob obtained from an FSRVP request */
35 uint8_t test_vt[] = {0x8a, 0xe3, 0x13, 0x71, 0x02, 0xf4, 0x36, 0x71,
36 		     0x02, 0x40, 0x28, 0x00, 0x3c, 0x65, 0xe0, 0xa8,
37 		     0x44, 0x27, 0x89, 0x43, 0xa6, 0x1d, 0x73, 0x73,
38 		     0xdf, 0x8b, 0x22, 0x92, 0x01, 0x00, 0x00, 0x00,
39 		     0x33, 0x05, 0x71, 0x71, 0xba, 0xbe, 0x37, 0x49,
40 		     0x83, 0x19, 0xb5, 0xdb, 0xef, 0x9c, 0xcc, 0x36,
41 		     0x01, 0x00, 0x00, 0x00};
42 
43 const char *vt_abstr_syntax = "a8e0653c-2744-4389-a61d-7373df8b2292/0x00000001";
44 const char *vt_trans_syntax = "71710533-beba-4937-8319-b5dbef9ccc36/0x00000001";
45 
test_verif_trailer_pctx(struct torture_context * tctx)46 static bool test_verif_trailer_pctx(struct torture_context *tctx)
47 {
48 	DATA_BLOB blob;
49 	bool ok;
50 	struct dcerpc_sec_vt_pcontext pctx;
51 	struct dcerpc_sec_verification_trailer *vt = NULL;
52 	struct ndr_pull *ndr;
53 	enum ndr_err_code ndr_err;
54 	struct ndr_print *ndr_print;
55 	TALLOC_CTX *mem_ctx = talloc_new(tctx);
56 	torture_assert(tctx, mem_ctx != NULL, "mem");
57 
58 	blob.data = test_vt;
59 	blob.length = ARRAY_SIZE(test_vt);
60 
61 	ndr = ndr_pull_init_blob(&blob, mem_ctx);
62 	torture_assert(tctx, ndr != NULL, "ndr");
63 
64 	ndr_err = ndr_pop_dcerpc_sec_verification_trailer(ndr, mem_ctx, &vt);
65 	torture_assert(tctx, NDR_ERR_CODE_IS_SUCCESS(ndr_err), "ndr");
66 
67 	ndr_print = talloc_zero(mem_ctx, struct ndr_print);
68 	torture_assert(tctx, ndr_print != NULL, "mem");
69 	ndr_print->print = ndr_print_printf_helper;
70 	ndr_print->depth = 1;
71 
72 	ndr_print_dcerpc_sec_verification_trailer(ndr_print,
73 						  "Verification Trailer", vt);
74 
75 	ZERO_STRUCT(pctx);
76 	ok = ndr_syntax_id_from_string(vt_abstr_syntax, &pctx.abstract_syntax);
77 	torture_assert(tctx, ok, "vt_abstr_syntax");
78 	ok = ndr_syntax_id_from_string(vt_trans_syntax, &pctx.transfer_syntax);
79 	torture_assert(tctx, ok, "vt_trans_syntax");
80 
81 	ok = dcerpc_sec_verification_trailer_check(vt, NULL, &pctx, NULL);
82 	torture_assert(tctx, ok, "VT check");
83 
84 	talloc_free(mem_ctx);
85 
86 	return true;
87 }
88 
torture_local_verif_trailer(TALLOC_CTX * mem_ctx)89 struct torture_suite *torture_local_verif_trailer(TALLOC_CTX *mem_ctx)
90 {
91 	struct torture_suite *suite = torture_suite_create(mem_ctx,
92 							   "verif_trailer");
93 
94 	torture_suite_add_simple_test(suite,
95 				      "pctx",
96 				      test_verif_trailer_pctx);
97 
98 	return suite;
99 }
100