LCOV - code coverage report
Current view: top level - source4/torture/rpc - session_key.c (source / functions) Hit Total Coverage
Test: coverage report for abartlet/fix-coverage dd10fb34 Lines: 104 109 95.4 %
Date: 2021-09-23 10:06:22 Functions: 5 5 100.0 %

          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        2304 : static void init_lsa_String(struct lsa_String *name, const char *s)
      31             : {
      32        2304 :         name->string = s;
      33        2304 : }
      34             : 
      35        2304 : 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             :         NTSTATUS status;
      41             :         struct lsa_CreateSecret r;
      42             :         struct lsa_SetSecret r3;
      43             :         struct lsa_QuerySecret r4;
      44             :         struct lsa_DATA_BUF buf1;
      45             :         struct lsa_DATA_BUF_PTR bufp1;
      46             :         DATA_BLOB enc_key;
      47             :         DATA_BLOB session_key;
      48             :         NTTIME old_mtime, new_mtime;
      49             :         DATA_BLOB blob1;
      50        2304 :         const char *secret1 = "abcdef12345699qwerty";
      51             :         char *secret2;
      52             :         char *secname;
      53        2304 :         struct dcerpc_binding_handle *b = p->binding_handle;
      54             : 
      55        2304 :         secname = talloc_asprintf(tctx, "torturesecret-%08x", (unsigned int)random());
      56             : 
      57        2304 :         torture_comment(tctx, "Testing CreateSecret of %s\n", secname);
      58             : 
      59        2304 :         init_lsa_String(&r.in.name, secname);
      60             : 
      61        2304 :         r.in.handle = handle;
      62        2304 :         r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      63        2304 :         r.out.sec_handle = sec_handle;
      64             : 
      65        2304 :         torture_assert_ntstatus_ok(tctx, dcerpc_lsa_CreateSecret_r(b, tctx, &r),
      66             :                 "CreateSecret failed");
      67        2304 :         torture_assert_ntstatus_ok(tctx, r.out.result, "CreateSecret failed");
      68             : 
      69        2304 :         status = dcerpc_fetch_session_key(p, &session_key);
      70        2304 :         torture_assert_ntstatus_ok(tctx, status, "dcerpc_fetch_session_key failed");
      71             : 
      72        1600 :         enc_key = sess_encrypt_string(secret1, &session_key);
      73             : 
      74        1600 :         r3.in.sec_handle = sec_handle;
      75        1600 :         r3.in.new_val = &buf1;
      76        1600 :         r3.in.old_val = NULL;
      77        1600 :         r3.in.new_val->data = enc_key.data;
      78        1600 :         r3.in.new_val->length = enc_key.length;
      79        1600 :         r3.in.new_val->size = enc_key.length;
      80             : 
      81        1600 :         torture_comment(tctx, "Testing SetSecret\n");
      82             : 
      83        1600 :         torture_assert_ntstatus_ok(tctx, dcerpc_lsa_SetSecret_r(b, tctx, &r3),
      84             :                 "SetSecret failed");
      85        1600 :         torture_assert_ntstatus_ok(tctx, r3.out.result, "SetSecret failed");
      86             : 
      87        1600 :         r3.in.sec_handle = sec_handle;
      88        1600 :         r3.in.new_val = &buf1;
      89        1600 :         r3.in.old_val = NULL;
      90        1600 :         r3.in.new_val->data = enc_key.data;
      91        1600 :         r3.in.new_val->length = enc_key.length;
      92        1600 :         r3.in.new_val->size = enc_key.length;
      93             : 
      94             :         /* break the encrypted data */
      95        1600 :         enc_key.data[0]++;
      96             : 
      97        1600 :         torture_comment(tctx, "Testing SetSecret with broken key\n");
      98             : 
      99        1600 :         torture_assert_ntstatus_ok(tctx, dcerpc_lsa_SetSecret_r(b, tctx, &r3),
     100             :                 "SetSecret failed");
     101        1600 :         torture_assert_ntstatus_equal(tctx, r3.out.result, NT_STATUS_UNKNOWN_REVISION,
     102             :                 "SetSecret should have failed UNKNOWN_REVISION");
     103             : 
     104        1600 :         data_blob_free(&enc_key);
     105             : 
     106        1600 :         ZERO_STRUCT(new_mtime);
     107        1600 :         ZERO_STRUCT(old_mtime);
     108             : 
     109             :         /* fetch the secret back again */
     110        1600 :         r4.in.sec_handle = sec_handle;
     111        1600 :         r4.in.new_val = &bufp1;
     112        1600 :         r4.in.new_mtime = &new_mtime;
     113        1600 :         r4.in.old_val = NULL;
     114        1600 :         r4.in.old_mtime = NULL;
     115             : 
     116        1600 :         bufp1.buf = NULL;
     117             : 
     118        1600 :         torture_comment(tctx, "Testing QuerySecret\n");
     119        1600 :         torture_assert_ntstatus_ok(tctx, dcerpc_lsa_QuerySecret_r(b, tctx, &r4),
     120             :                 "QuerySecret failed");
     121        1600 :         torture_assert_ntstatus_ok(tctx, r4.out.result, "QuerySecret failed");
     122        1600 :         if (r4.out.new_val == NULL || r4.out.new_val->buf == NULL)
     123           0 :                 torture_fail(tctx, "No secret buffer returned");
     124        1600 :         blob1.data = r4.out.new_val->buf->data;
     125        1600 :         blob1.length = r4.out.new_val->buf->size;
     126             : 
     127        1600 :         secret2 = sess_decrypt_string(tctx, &blob1, &session_key);
     128             : 
     129        1600 :         torture_assert_str_equal(tctx, secret1, secret2, "Returned secret invalid");
     130             : 
     131        1600 :         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        2304 : static bool test_secrets(struct torture_context *torture, const void *_data)
     142             : {
     143             :         struct dcerpc_pipe *p;
     144             :         struct policy_handle *handle;
     145             :         struct dcerpc_binding *binding;
     146        2304 :         const struct secret_settings *settings =
     147             :                 (const struct secret_settings *)_data;
     148             :         NTSTATUS status;
     149             :         struct dcerpc_binding_handle *b;
     150        2304 :         struct policy_handle sec_handle = {0};
     151             :         bool ok;
     152             : 
     153        2304 :         lpcfg_set_cmdline(torture->lp_ctx, "ntlmssp client:keyexchange", settings->keyexchange?"True":"False");
     154        2304 :         lpcfg_set_cmdline(torture->lp_ctx, "ntlmssp_client:ntlm2", settings->ntlm2?"True":"False");
     155        2304 :         lpcfg_set_cmdline(torture->lp_ctx, "ntlmssp_client:lm_key", settings->lm_key?"True":"False");
     156             : 
     157        2304 :         torture_assert_ntstatus_ok(torture, torture_rpc_binding(torture, &binding),
     158             :                                    "Getting bindoptions");
     159             : 
     160        2304 :         status = dcerpc_binding_set_flags(binding, settings->bindoptions, 0);
     161        2304 :         torture_assert_ntstatus_ok(torture, status, "dcerpc_binding_set_flags");
     162             : 
     163        2304 :         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        2304 :         torture_assert_ntstatus_ok(torture, status, "connect");
     170        2304 :         b = p->binding_handle;
     171             : 
     172        2304 :         if (!test_lsa_OpenPolicy2(b, torture, &handle)) {
     173           0 :                 talloc_free(p);
     174           0 :                 return false;
     175             :         }
     176             : 
     177        2304 :         torture_assert(torture, handle, "OpenPolicy2 failed.  This test cannot run against this server");
     178             : 
     179        2304 :         ok = test_CreateSecret_basic(p, torture, handle, &sec_handle);
     180             : 
     181        2304 :         if (is_valid_policy_hnd(&sec_handle)) {
     182             :                 struct lsa_DeleteObject d;
     183             : 
     184        2304 :                 d.in.handle = &sec_handle;
     185        2304 :                 d.out.handle = &sec_handle;
     186             : 
     187        2304 :                 status = dcerpc_lsa_DeleteObject_r(b, torture, &d);
     188        3912 :                 if (!NT_STATUS_IS_OK(status) ||
     189        2304 :                     !NT_STATUS_IS_OK(d.out.result)) {
     190           0 :                         torture_warning(torture,
     191             :                                         "Failed to delete secrets object");
     192             :                 }
     193             :         }
     194             : 
     195        2304 :         talloc_free(p);
     196        2304 :         return ok;
     197             : }
     198             : 
     199       56520 : static struct torture_tcase *add_test(struct torture_suite *suite, uint32_t bindoptions,
     200             :                                      bool keyexchange, bool ntlm2, bool lm_key)
     201             : {
     202       56520 :         char *name = NULL;
     203             :         struct secret_settings *settings;
     204             : 
     205       56520 :         settings = talloc_zero(suite, struct secret_settings);
     206       56520 :         settings->bindoptions = bindoptions;
     207             : 
     208       56520 :         if (bindoptions == DCERPC_PUSH_BIGENDIAN)
     209       18840 :                 name = talloc_strdup(suite, "bigendian");
     210       37680 :         else if (bindoptions == DCERPC_SEAL)
     211       18840 :                 name = talloc_strdup(suite, "seal");
     212       18840 :         else if (bindoptions == 0)
     213       18840 :                 name = talloc_strdup(suite, "none");
     214             :         else
     215           0 :                 name = talloc_strdup(suite, "unknown");
     216             : 
     217       56520 :         name = talloc_asprintf_append_buffer(name, " keyexchange:%s", keyexchange?"yes":"no");
     218       56520 :         settings->keyexchange = keyexchange;
     219             : 
     220       56520 :         name = talloc_asprintf_append_buffer(name, " ntlm2:%s", ntlm2?"yes":"no");
     221       56520 :         settings->ntlm2 = ntlm2;
     222             : 
     223       56520 :         name = talloc_asprintf_append_buffer(name, " lm_key:%s", lm_key?"yes":"no");
     224       56520 :         settings->lm_key = lm_key;
     225             : 
     226       56520 :         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        2355 : struct torture_suite *torture_rpc_lsa_secrets(TALLOC_CTX *mem_ctx)
     234             : {
     235        2355 :         struct torture_suite *suite = torture_suite_create(mem_ctx, "lsa.secrets");
     236             :         int keyexchange, ntlm2, lm_key;
     237             : 
     238        7065 :         for (keyexchange = 0; keyexchange < ARRAY_SIZE(bool_vals); keyexchange++) {
     239       13884 :                 for (ntlm2 = 0; ntlm2 < ARRAY_SIZE(bool_vals); ntlm2++) {
     240       27768 :                         for (lm_key = 0; lm_key < ARRAY_SIZE(bool_vals); lm_key++) {
     241       18840 :                                 add_test(suite, DCERPC_PUSH_BIGENDIAN, bool_vals[keyexchange], bool_vals[ntlm2],
     242       18840 :                                          bool_vals[lm_key]);
     243       18840 :                                 add_test(suite, DCERPC_SEAL, bool_vals[keyexchange], bool_vals[ntlm2], bool_vals[lm_key]);
     244       18840 :                                 add_test(suite, 0, bool_vals[keyexchange], bool_vals[ntlm2], bool_vals[lm_key]);
     245             :                         }
     246             :                 }
     247             :         }
     248             : 
     249        2355 :         return suite;
     250             : }

Generated by: LCOV version 1.13