Line data Source code
1 : /*
2 : Unix SMB/CIFS implementation.
3 : test suite for lsa rpc operations
4 :
5 : Copyright (C) Andrew Tridgell 2003
6 : Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
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 "librpc/gen_ndr/ndr_lsa_c.h"
24 :
25 : #include "libcli/auth/libcli_auth.h"
26 : #include "torture/rpc/torture_rpc.h"
27 : #include "lib/cmdline/cmdline.h"
28 : #include "param/param.h"
29 :
30 2256 : static void init_lsa_String(struct lsa_String *name, const char *s)
31 : {
32 2256 : name->string = s;
33 2256 : }
34 :
35 2256 : static bool test_CreateSecret_basic(struct dcerpc_pipe *p,
36 : struct torture_context *tctx,
37 : struct policy_handle *handle,
38 : struct policy_handle *sec_handle)
39 : {
40 0 : NTSTATUS status;
41 0 : struct lsa_CreateSecret r;
42 0 : struct lsa_SetSecret r3;
43 0 : struct lsa_QuerySecret r4;
44 0 : struct lsa_DATA_BUF buf1;
45 0 : struct lsa_DATA_BUF_PTR bufp1;
46 0 : DATA_BLOB enc_key;
47 0 : DATA_BLOB session_key;
48 0 : NTTIME old_mtime, new_mtime;
49 0 : DATA_BLOB blob1;
50 2256 : const char *secret1 = "abcdef12345699qwerty";
51 0 : char *secret2;
52 0 : char *secname;
53 2256 : struct dcerpc_binding_handle *b = p->binding_handle;
54 :
55 2256 : secname = talloc_asprintf(tctx, "torturesecret-%08x", (unsigned int)random());
56 :
57 2256 : torture_comment(tctx, "Testing CreateSecret of %s\n", secname);
58 :
59 2256 : init_lsa_String(&r.in.name, secname);
60 :
61 2256 : r.in.handle = handle;
62 2256 : r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
63 2256 : r.out.sec_handle = sec_handle;
64 :
65 2256 : torture_assert_ntstatus_ok(tctx, dcerpc_lsa_CreateSecret_r(b, tctx, &r),
66 : "CreateSecret failed");
67 2256 : torture_assert_ntstatus_ok(tctx, r.out.result, "CreateSecret failed");
68 :
69 2256 : status = dcerpc_fetch_session_key(p, &session_key);
70 2256 : torture_assert_ntstatus_ok(tctx, status, "dcerpc_fetch_session_key failed");
71 :
72 1568 : enc_key = sess_encrypt_string(secret1, &session_key);
73 :
74 1568 : r3.in.sec_handle = sec_handle;
75 1568 : r3.in.new_val = &buf1;
76 1568 : r3.in.old_val = NULL;
77 1568 : r3.in.new_val->data = enc_key.data;
78 1568 : r3.in.new_val->length = enc_key.length;
79 1568 : r3.in.new_val->size = enc_key.length;
80 :
81 1568 : torture_comment(tctx, "Testing SetSecret\n");
82 :
83 1568 : torture_assert_ntstatus_ok(tctx, dcerpc_lsa_SetSecret_r(b, tctx, &r3),
84 : "SetSecret failed");
85 1568 : torture_assert_ntstatus_ok(tctx, r3.out.result, "SetSecret failed");
86 :
87 1568 : r3.in.sec_handle = sec_handle;
88 1568 : r3.in.new_val = &buf1;
89 1568 : r3.in.old_val = NULL;
90 1568 : r3.in.new_val->data = enc_key.data;
91 1568 : r3.in.new_val->length = enc_key.length;
92 1568 : r3.in.new_val->size = enc_key.length;
93 :
94 : /* break the encrypted data */
95 1568 : enc_key.data[0]++;
96 :
97 1568 : torture_comment(tctx, "Testing SetSecret with broken key\n");
98 :
99 1568 : torture_assert_ntstatus_ok(tctx, dcerpc_lsa_SetSecret_r(b, tctx, &r3),
100 : "SetSecret failed");
101 1568 : torture_assert_ntstatus_equal(tctx, r3.out.result, NT_STATUS_UNKNOWN_REVISION,
102 : "SetSecret should have failed UNKNOWN_REVISION");
103 :
104 1568 : data_blob_free(&enc_key);
105 :
106 1568 : ZERO_STRUCT(new_mtime);
107 1568 : ZERO_STRUCT(old_mtime);
108 :
109 : /* fetch the secret back again */
110 1568 : r4.in.sec_handle = sec_handle;
111 1568 : r4.in.new_val = &bufp1;
112 1568 : r4.in.new_mtime = &new_mtime;
113 1568 : r4.in.old_val = NULL;
114 1568 : r4.in.old_mtime = NULL;
115 :
116 1568 : bufp1.buf = NULL;
117 :
118 1568 : torture_comment(tctx, "Testing QuerySecret\n");
119 1568 : torture_assert_ntstatus_ok(tctx, dcerpc_lsa_QuerySecret_r(b, tctx, &r4),
120 : "QuerySecret failed");
121 1568 : torture_assert_ntstatus_ok(tctx, r4.out.result, "QuerySecret failed");
122 1568 : if (r4.out.new_val == NULL || r4.out.new_val->buf == NULL)
123 0 : torture_fail(tctx, "No secret buffer returned");
124 1568 : blob1.data = r4.out.new_val->buf->data;
125 1568 : blob1.length = r4.out.new_val->buf->size;
126 :
127 1568 : secret2 = sess_decrypt_string(tctx, &blob1, &session_key);
128 :
129 1568 : torture_assert_str_equal(tctx, secret1, secret2, "Returned secret invalid");
130 :
131 1568 : return true;
132 : }
133 :
134 : struct secret_settings {
135 : uint32_t bindoptions;
136 : bool keyexchange;
137 : bool ntlm2;
138 : bool lm_key;
139 : };
140 :
141 2256 : static bool test_secrets(struct torture_context *torture, const void *_data)
142 : {
143 0 : struct dcerpc_pipe *p;
144 0 : struct policy_handle *handle;
145 0 : struct dcerpc_binding *binding;
146 2256 : const struct secret_settings *settings =
147 : (const struct secret_settings *)_data;
148 0 : NTSTATUS status;
149 0 : struct dcerpc_binding_handle *b;
150 2256 : struct policy_handle sec_handle = {0};
151 0 : bool ok;
152 :
153 2256 : lpcfg_set_cmdline(torture->lp_ctx, "ntlmssp client:keyexchange", settings->keyexchange?"True":"False");
154 2256 : lpcfg_set_cmdline(torture->lp_ctx, "ntlmssp_client:ntlm2", settings->ntlm2?"True":"False");
155 2256 : lpcfg_set_cmdline(torture->lp_ctx, "ntlmssp_client:lm_key", settings->lm_key?"True":"False");
156 :
157 2256 : torture_assert_ntstatus_ok(torture, torture_rpc_binding(torture, &binding),
158 : "Getting bindoptions");
159 :
160 2256 : status = dcerpc_binding_set_flags(binding, settings->bindoptions, 0);
161 2256 : torture_assert_ntstatus_ok(torture, status, "dcerpc_binding_set_flags");
162 :
163 2256 : status = dcerpc_pipe_connect_b(torture, &p, binding,
164 : &ndr_table_lsarpc,
165 : samba_cmdline_get_creds(),
166 : torture->ev,
167 : torture->lp_ctx);
168 :
169 2256 : torture_assert_ntstatus_ok(torture, status, "connect");
170 2256 : b = p->binding_handle;
171 :
172 2256 : if (!test_lsa_OpenPolicy2(b, torture, &handle)) {
173 0 : talloc_free(p);
174 0 : return false;
175 : }
176 :
177 2256 : torture_assert(torture, handle, "OpenPolicy2 failed. This test cannot run against this server");
178 :
179 2256 : ok = test_CreateSecret_basic(p, torture, handle, &sec_handle);
180 :
181 2256 : if (is_valid_policy_hnd(&sec_handle)) {
182 0 : struct lsa_DeleteObject d;
183 :
184 2256 : d.in.handle = &sec_handle;
185 2256 : d.out.handle = &sec_handle;
186 :
187 2256 : status = dcerpc_lsa_DeleteObject_r(b, torture, &d);
188 2256 : if (!NT_STATUS_IS_OK(status) ||
189 2256 : !NT_STATUS_IS_OK(d.out.result)) {
190 0 : torture_warning(torture,
191 : "Failed to delete secrets object");
192 : }
193 : }
194 :
195 2256 : talloc_free(p);
196 2256 : return ok;
197 : }
198 :
199 56496 : static struct torture_tcase *add_test(struct torture_suite *suite, uint32_t bindoptions,
200 : bool keyexchange, bool ntlm2, bool lm_key)
201 : {
202 56496 : char *name = NULL;
203 3000 : struct secret_settings *settings;
204 :
205 56496 : settings = talloc_zero(suite, struct secret_settings);
206 56496 : settings->bindoptions = bindoptions;
207 :
208 56496 : if (bindoptions == DCERPC_PUSH_BIGENDIAN)
209 18832 : name = talloc_strdup(suite, "bigendian");
210 37664 : else if (bindoptions == DCERPC_SEAL)
211 18832 : name = talloc_strdup(suite, "seal");
212 18832 : else if (bindoptions == 0)
213 18832 : name = talloc_strdup(suite, "none");
214 : else
215 0 : name = talloc_strdup(suite, "unknown");
216 :
217 57996 : name = talloc_asprintf_append_buffer(name, " keyexchange:%s", keyexchange?"yes":"no");
218 56496 : settings->keyexchange = keyexchange;
219 :
220 57996 : name = talloc_asprintf_append_buffer(name, " ntlm2:%s", ntlm2?"yes":"no");
221 56496 : settings->ntlm2 = ntlm2;
222 :
223 57996 : name = talloc_asprintf_append_buffer(name, " lm_key:%s", lm_key?"yes":"no");
224 56496 : settings->lm_key = lm_key;
225 :
226 56496 : return torture_suite_add_simple_tcase_const(suite, name, test_secrets,
227 : settings);
228 : }
229 :
230 : static const bool bool_vals[] = { true, false };
231 :
232 : /* TEST session key correctness by pushing and pulling secrets */
233 2354 : struct torture_suite *torture_rpc_lsa_secrets(TALLOC_CTX *mem_ctx)
234 : {
235 2354 : struct torture_suite *suite = torture_suite_create(mem_ctx, "lsa.secrets");
236 125 : int keyexchange, ntlm2, lm_key;
237 :
238 7187 : for (keyexchange = 0; keyexchange < ARRAY_SIZE(bool_vals); keyexchange++) {
239 14124 : for (ntlm2 = 0; ntlm2 < ARRAY_SIZE(bool_vals); ntlm2++) {
240 28248 : for (lm_key = 0; lm_key < ARRAY_SIZE(bool_vals); lm_key++) {
241 18832 : add_test(suite, DCERPC_PUSH_BIGENDIAN, bool_vals[keyexchange], bool_vals[ntlm2],
242 18832 : bool_vals[lm_key]);
243 18832 : add_test(suite, DCERPC_SEAL, bool_vals[keyexchange], bool_vals[ntlm2], bool_vals[lm_key]);
244 18832 : add_test(suite, 0, bool_vals[keyexchange], bool_vals[ntlm2], bool_vals[lm_key]);
245 : }
246 : }
247 : }
248 :
249 2354 : return suite;
250 : }
|
