r3428: switched to using minimal includes for the auto-generated RPC code.
[sfrench/samba-autobuild/.git] / source4 / torture / rpc / netlogon.c
index b08be80bb03322d234f8bf7f83fd8e9077547cc1..6971193f92ccf18120055eea5818eef7991eb584 100644 (file)
@@ -23,6 +23,7 @@
 */
 
 #include "includes.h"
+#include "librpc/gen_ndr/ndr_netlogon.h"
 
 
 static const char *machine_password;
@@ -272,6 +273,7 @@ enum ntlm_break {
 struct samlogon_state {
        TALLOC_CTX *mem_ctx;
        const char *account_name;
+       const char *account_domain;
        const char *password;
        struct dcerpc_pipe *p;
        struct netr_LogonSamLogon r;
@@ -295,131 +297,126 @@ static NTSTATUS check_samlogon(struct samlogon_state *samlogon_state,
 {
        NTSTATUS status;
        struct netr_LogonSamLogon *r = &samlogon_state->r;
-       int levels[] = { 2, 6 };
-       int i;
        struct netr_NetworkInfo ninfo;
 
-       for (i=0;i<ARRAY_SIZE(levels);i++) {
-               struct netr_SamBaseInfo *base;
-               
-               printf("testing netr_LogonSamLogon with logon level %d\n", levels[i]);
-
-               samlogon_state->r.in.logon_level = levels[i];
-               samlogon_state->r.in.logon.network = &ninfo;
+       struct netr_SamBaseInfo *base;
+       
+       printf("testing netr_LogonSamLogon\n");
        
-               ninfo.identity_info.domain_name.string = lp_workgroup();
-               ninfo.identity_info.parameter_control = 0;
-               ninfo.identity_info.logon_id_low = 0;
-               ninfo.identity_info.logon_id_high = 0;
-               ninfo.identity_info.account_name.string = samlogon_state->account_name;
-               ninfo.identity_info.workstation.string = TEST_MACHINE_NAME;
+       samlogon_state->r.in.logon.network = &ninfo;
+       
+       ninfo.identity_info.domain_name.string = samlogon_state->account_domain;
+       ninfo.identity_info.parameter_control = 0;
+       ninfo.identity_info.logon_id_low = 0;
+       ninfo.identity_info.logon_id_high = 0;
+       ninfo.identity_info.account_name.string = samlogon_state->account_name;
+       ninfo.identity_info.workstation.string = TEST_MACHINE_NAME;
                
-               memcpy(ninfo.challenge, chall->data, 8);
+       memcpy(ninfo.challenge, chall->data, 8);
                
-               switch (break_which) {
-               case BREAK_NONE:
-                       break;
-               case BREAK_LM:
-                       if (lm_response && lm_response->data) {
-                               lm_response->data[0]++;
-                       }
-                       break;
-               case BREAK_NT:
-                       if (nt_response && nt_response->data) {
-                               nt_response->data[0]++;
-                       }
-                       break;
-               case NO_LM:
-                       data_blob_free(lm_response);
-                       break;
-               case NO_NT:
-                       data_blob_free(nt_response);
-                       break;
+       switch (break_which) {
+       case BREAK_NONE:
+               break;
+       case BREAK_LM:
+               if (lm_response && lm_response->data) {
+                       lm_response->data[0]++;
                }
-               
-               if (nt_response) {
-                       ninfo.nt.data = nt_response->data;
-                       ninfo.nt.length = nt_response->length;
-               } else {
-                       ninfo.nt.data = NULL;
-                       ninfo.nt.length = 0;
+               break;
+       case BREAK_NT:
+               if (nt_response && nt_response->data) {
+                       nt_response->data[0]++;
                }
+               break;
+       case NO_LM:
+               data_blob_free(lm_response);
+               break;
+       case NO_NT:
+               data_blob_free(nt_response);
+               break;
+       }
                
-               if (lm_response) {
-                       ninfo.lm.data = lm_response->data;
-                       ninfo.lm.length = lm_response->length;
-               } else {
-                       ninfo.lm.data = NULL;
-                       ninfo.lm.length = 0;
-               }
+       if (nt_response) {
+               ninfo.nt.data = nt_response->data;
+               ninfo.nt.length = nt_response->length;
+       } else {
+               ninfo.nt.data = NULL;
+               ninfo.nt.length = 0;
+       }
                
-               ZERO_STRUCT(samlogon_state->auth2);
-               creds_client_authenticator(&samlogon_state->creds, &samlogon_state->auth);
+       if (lm_response) {
+               ninfo.lm.data = lm_response->data;
+               ninfo.lm.length = lm_response->length;
+       } else {
+               ninfo.lm.data = NULL;
+               ninfo.lm.length = 0;
+       }
                
-               r->out.return_authenticator = NULL;
-               status = dcerpc_netr_LogonSamLogon(samlogon_state->p, samlogon_state->mem_ctx, r);
-               if (!NT_STATUS_IS_OK(status)) {
-                       if (error_string) {
-                               *error_string = strdup(nt_errstr(status));
-                       }
-               }
+       ZERO_STRUCT(samlogon_state->auth2);
+       creds_client_authenticator(&samlogon_state->creds, &samlogon_state->auth);
                
-               if (!r->out.return_authenticator || 
-                   !creds_client_check(&samlogon_state->creds, &r->out.return_authenticator->cred)) {
-                       printf("Credential chaining failed\n");
-               }
-
-               if (!NT_STATUS_IS_OK(status)) {
-                       /* we cannot check the session key, if the logon failed... */
-                       return status;
+       r->out.return_authenticator = NULL;
+       status = dcerpc_netr_LogonSamLogon(samlogon_state->p, samlogon_state->mem_ctx, r);
+       if (!NT_STATUS_IS_OK(status)) {
+               if (error_string) {
+                       *error_string = strdup(nt_errstr(status));
                }
+       }
                
-               /* find and decyrpt the session keys, return in parameters above */
-               if (r->in.validation_level == 2) {
-                       base = &r->out.validation.sam2->base;
-               } else if (r->in.validation_level == 3) {
-                       base = &r->out.validation.sam3->base;
-               } else if (r->in.validation_level == 6) {
-                       base = &r->out.validation.sam6->base;
-               } else {
-                       base = NULL;
-               }
+       if (!r->out.return_authenticator || 
+           !creds_client_check(&samlogon_state->creds, &r->out.return_authenticator->cred)) {
+               printf("Credential chaining failed\n");
+       }
 
-               if (r->in.validation_level != 6) {
-                       static const char zeros[16];
+       if (!NT_STATUS_IS_OK(status)) {
+               /* we cannot check the session key, if the logon failed... */
+               return status;
+       }
+               
+       /* find and decyrpt the session keys, return in parameters above */
+       if (r->in.validation_level == 2) {
+               base = &r->out.validation.sam2->base;
+       } else if (r->in.validation_level == 3) {
+               base = &r->out.validation.sam3->base;
+       } else if (r->in.validation_level == 6) {
+               base = &r->out.validation.sam6->base;
+       } else {
+               base = NULL;
+       }
+
+       if (r->in.validation_level != 6) {
+               static const char zeros[16];
                        
-                       if (memcmp(base->key.key, zeros,  
-                                  sizeof(base->key.key)) != 0) {
-                               creds_arcfour_crypt(&samlogon_state->creds, 
-                                                   base->key.key, 
-                                                   sizeof(base->key.key));
-                       }
+               if (memcmp(base->key.key, zeros,  
+                          sizeof(base->key.key)) != 0) {
+                       creds_arcfour_crypt(&samlogon_state->creds, 
+                                           base->key.key, 
+                                           sizeof(base->key.key));
+               }
                        
-                       if (user_session_key) {
-                               memcpy(user_session_key, base->key.key, 16);
-                       }
+               if (user_session_key) {
+                       memcpy(user_session_key, base->key.key, 16);
+               }
                        
-                       if (memcmp(base->LMSessKey.key, zeros,  
-                                  sizeof(base->LMSessKey.key)) != 0) {
-                               creds_arcfour_crypt(&samlogon_state->creds, 
-                                                   base->LMSessKey.key, 
-                                                   sizeof(base->LMSessKey.key));
-                       }
+               if (memcmp(base->LMSessKey.key, zeros,  
+                          sizeof(base->LMSessKey.key)) != 0) {
+                       creds_arcfour_crypt(&samlogon_state->creds, 
+                                           base->LMSessKey.key, 
+                                           sizeof(base->LMSessKey.key));
+               }
                        
-                       if (lm_key) {
-                               memcpy(lm_key, base->LMSessKey.key, 8);
-                       }
-               } else {
-                       /* they aren't encrypted! */
-                       if (user_session_key) {
-                               memcpy(user_session_key, base->key.key, 16);
-                       }
-                       if (lm_key) {
-                               memcpy(lm_key, base->LMSessKey.key, 8);
-                       }
+               if (lm_key) {
+                       memcpy(lm_key, base->LMSessKey.key, 8);
+               }
+       } else {
+               /* they aren't encrypted! */
+               if (user_session_key) {
+                       memcpy(user_session_key, base->key.key, 16);
+               }
+               if (lm_key) {
+                       memcpy(lm_key, base->LMSessKey.key, 8);
                }
        }
-
+       
        return status;
 } 
 
@@ -431,6 +428,7 @@ static NTSTATUS check_samlogon(struct samlogon_state *samlogon_state,
 static BOOL test_lm_ntlm_broken(struct samlogon_state *samlogon_state, enum ntlm_break break_which, char **error_string) 
 {
        BOOL pass = True;
+       BOOL lm_good;
        NTSTATUS nt_status;
        DATA_BLOB lm_response = data_blob_talloc(samlogon_state->mem_ctx, NULL, 24);
        DATA_BLOB nt_response = data_blob_talloc(samlogon_state->mem_ctx, NULL, 24);
@@ -444,9 +442,13 @@ static BOOL test_lm_ntlm_broken(struct samlogon_state *samlogon_state, enum ntlm
        ZERO_STRUCT(lm_key);
        ZERO_STRUCT(user_session_key);
 
-       SMBencrypt(samlogon_state->password, samlogon_state->chall.data, lm_response.data);
-       E_deshash(samlogon_state->password, lm_hash); 
-
+       lm_good = SMBencrypt(samlogon_state->password, samlogon_state->chall.data, lm_response.data);
+       if (!lm_good) {
+               ZERO_STRUCT(lm_hash);
+       } else {
+               E_deshash(samlogon_state->password, lm_hash); 
+       }
+               
        SMBNTencrypt(samlogon_state->password, samlogon_state->chall.data, nt_response.data);
 
        E_md4hash(samlogon_state->password, nt_hash);
@@ -463,14 +465,27 @@ static BOOL test_lm_ntlm_broken(struct samlogon_state *samlogon_state, enum ntlm
        
        data_blob_free(&lm_response);
 
-       if (!NT_STATUS_IS_OK(nt_status)) {
+       if (NT_STATUS_EQUAL(NT_STATUS_WRONG_PASSWORD, nt_status)) {
+               /* for 'long' passwords, the LM password is invalid */
+               if (break_which == NO_NT && !lm_good) {
+                       return True;
+               }
                return break_which == BREAK_NT;
        }
 
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               return False;
+       }
+
+       if (break_which == NO_NT && !lm_good) {
+               printf("LM password is 'long' (> 14 chars and therefore invalid) but login did not fail!");
+               return False;
+       }
+
        if (memcmp(lm_hash, lm_key, 
                   sizeof(lm_key)) != 0) {
                printf("LM Key does not match expectations!\n");
-               printf("lm_key:\n");
+               printf("lm_key:\n");
                dump_data(1, (const char *)lm_key, 8);
                printf("expected:\n");
                dump_data(1, (const char *)lm_hash, 8);
@@ -589,6 +604,7 @@ static BOOL test_ntlm_in_lm(struct samlogon_state *samlogon_state, char **error_
 static BOOL test_ntlm_in_both(struct samlogon_state *samlogon_state, char **error_string) 
 {
        BOOL pass = True;
+       BOOL lm_good;
        NTSTATUS nt_status;
        DATA_BLOB nt_response = data_blob_talloc(samlogon_state->mem_ctx, NULL, 24);
        DATA_BLOB session_key = data_blob_talloc(samlogon_state->mem_ctx, NULL, 16);
@@ -607,7 +623,10 @@ static BOOL test_ntlm_in_both(struct samlogon_state *samlogon_state, char **erro
        SMBsesskeygen_ntv1((const uint8_t *)nt_hash, 
                           session_key.data);
 
-       E_deshash(samlogon_state->password, (uint8_t *)lm_hash); 
+       lm_good = E_deshash(samlogon_state->password, (uint8_t *)lm_hash); 
+       if (!lm_good) {
+               ZERO_STRUCT(lm_hash);
+       }
 
        nt_status = check_samlogon(samlogon_state,
                                   BREAK_NONE,
@@ -663,7 +682,7 @@ static BOOL test_lmv2_ntlmv2_broken(struct samlogon_state *samlogon_state, enum
        ZERO_STRUCT(user_session_key);
        
        /* TODO - test with various domain cases, and without domain */
-       if (!SMBNTLMv2encrypt(samlogon_state->account_name, lp_workgroup()
+       if (!SMBNTLMv2encrypt(samlogon_state->account_name, samlogon_state->account_domain
                              samlogon_state->password, &samlogon_state->chall,
                              &names_blob,
                              &lmv2_response, &ntlmv2_response, 
@@ -760,7 +779,7 @@ static BOOL test_plaintext(struct samlogon_state *samlogon_state, enum ntlm_brea
        DATA_BLOB lm_response = data_blob(NULL, 0);
        char *password;
        char *dospw;
-       smb_ucs2_t *unicodepw;
+       void *unicodepw;
 
        uint8_t user_session_key[16];
        uint8_t lm_key[16];
@@ -769,27 +788,24 @@ static BOOL test_plaintext(struct samlogon_state *samlogon_state, enum ntlm_brea
 
        ZERO_STRUCT(user_session_key);
        
-       if ((push_ucs2_talloc(samlogon_state->mem_ctx, (smb_ucs2_t **)&unicodepw, 
+       if ((push_ucs2_talloc(samlogon_state->mem_ctx, &unicodepw, 
                              samlogon_state->password)) == -1) {
                DEBUG(0, ("push_ucs2_allocate failed!\n"));
                exit(1);
        }
 
-       nt_response = data_blob_talloc(samlogon_state->mem_ctx, unicodepw, 
-                                      strlen_w(((void *)unicodepw))*sizeof(smb_ucs2_t));
+       nt_response = data_blob_talloc(samlogon_state->mem_ctx, unicodepw, utf16_len(unicodepw));
 
-       password = strdup_upper(samlogon_state->password);
+       password = strupper_talloc(samlogon_state->mem_ctx, samlogon_state->password);
 
        if ((convert_string_talloc(samlogon_state->mem_ctx, CH_UNIX, 
                                   CH_DOS, password,
                                   strlen(password)+1, 
-                                  (const void**)&dospw)) == -1) {
-               DEBUG(0, ("push_ascii_allocate failed!\n"));
+                                  (void**)&dospw)) == -1) {
+               DEBUG(0, ("convert_string_talloc failed!\n"));
                exit(1);
        }
 
-       SAFE_FREE(password);
-
        lm_response = data_blob_talloc(samlogon_state->mem_ctx, dospw, strlen(dospw));
 
        nt_status = check_samlogon(samlogon_state,
@@ -881,13 +897,15 @@ static const struct ntlm_tests {
 */
 static BOOL test_SamLogon(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
 {
-       int i, j;
+       int i, v, l;
        BOOL ret = True;
        int validation_levels[] = {2,3,6};
+       int logon_levels[] = { 2, 6 };
        struct samlogon_state samlogon_state;
        
        samlogon_state.mem_ctx = mem_ctx;
        samlogon_state.account_name = lp_parm_string(-1, "torture", "username");
+       samlogon_state.account_domain = lp_parm_string(-1, "torture", "userdomain");
        samlogon_state.password = lp_parm_string(-1, "torture", "password");
        samlogon_state.p = p;
 
@@ -912,21 +930,24 @@ static BOOL test_SamLogon(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
        samlogon_state.r.in.credential = &samlogon_state.auth;
        samlogon_state.r.in.return_authenticator = &samlogon_state.auth2;
 
-       for (i=0;i<ARRAY_SIZE(validation_levels);i++) {
-               samlogon_state.r.in.validation_level = validation_levels[i];
-               for (j=0; test_table[j].fn; j++) {
-                       char *error_string = NULL;
-                       printf("Testing SamLogon with '%s' at validation level %d\n", 
-                              test_table[j].name, validation_levels[i]);
+       for (i=0; test_table[i].fn; i++) {
+               for (v=0;v<ARRAY_SIZE(validation_levels);v++) {
+                       for (l=0;l<ARRAY_SIZE(logon_levels);l++) {
+                               char *error_string = NULL;
+                               samlogon_state.r.in.validation_level = validation_levels[v];
+                               samlogon_state.r.in.logon_level = logon_levels[l];
+                               printf("Testing SamLogon with '%s' at validation level %d, logon level %d\n", 
+                                      test_table[i].name, validation_levels[v], logon_levels[l]);
        
-                       if (!test_table[j].fn(&samlogon_state, &error_string)) {
-                               if (test_table[j].expect_fail) {
-                                       printf("Test %s failed (expected, test incomplete): %s\n", test_table[j].name, error_string);
-                               } else {
-                                       printf("Test %s failed: %s\n", test_table[j].name, error_string);
-                                       ret = False;
+                               if (!test_table[i].fn(&samlogon_state, &error_string)) {
+                                       if (test_table[i].expect_fail) {
+                                               printf("Test %s failed (expected, test incomplete): %s\n", test_table[i].name, error_string);
+                                       } else {
+                                               printf("Test %s failed: %s\n", test_table[i].name, error_string);
+                                               ret = False;
+                                       }
+                                       SAFE_FREE(error_string);
                                }
-                               SAFE_FREE(error_string);
                        }
                }
        }
@@ -979,7 +1000,7 @@ static BOOL test_SetPassword(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
           allows the password to be set to the same value twice in a
           row (match win2k3) */
        printf("Testing a second ServerPasswordSet on machine account\n");
-       printf("Changing machine account password to '%s' (same as pervsious run)\n", password);
+       printf("Changing machine account password to '%s' (same as previous run)\n", password);
 
        creds_client_authenticator(&creds, &r.in.credential);
 
@@ -1534,7 +1555,7 @@ static BOOL test_InteractiveLogin(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
        r.in.validation_level = 6;
        r.in.flags = 0;
 
-       pinfo.identity_info.domain_name.string = lp_workgroup();
+       pinfo.identity_info.domain_name.string = lp_parm_string(-1, "torture", "userdomain");
        pinfo.identity_info.parameter_control = 0;
        pinfo.identity_info.logon_id_low = 0;
        pinfo.identity_info.logon_id_high = 0;
@@ -1573,7 +1594,6 @@ static BOOL test_GetDomainInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
        struct netr_LogonGetDomainInfo r;
        struct netr_DomainQuery1 q1;
        struct netr_Authenticator a;
-       uint32_t i1;
        struct creds_CredentialState creds;
 
        if (!test_SetupCredentials3(p, mem_ctx, NETLOGON_NEG_AUTH2_ADS_FLAGS, &creds)) {
@@ -1591,8 +1611,8 @@ static BOOL test_GetDomainInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
        r.in.credential = &a;
        r.out.credential = &a;
 
-       i1 = 0;
-       r.in.i1 = &i1;
+       r.in.i1[0] = 0;
+       r.in.i1[1] = 0;
 
        r.in.query.query1 = &q1;
        ZERO_STRUCT(q1);
@@ -1624,7 +1644,92 @@ static BOOL test_GetDomainInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
 }
 
 
-BOOL torture_rpc_netlogon(int dummy)
+static void async_callback(struct rpc_request *req)
+{
+       int *counter = req->async.private;
+       if (NT_STATUS_IS_OK(req->status)) {
+               (*counter)++;
+       }
+}
+
+static BOOL test_GetDomainInfo_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
+{
+       NTSTATUS status;
+       struct netr_LogonGetDomainInfo r;
+       struct netr_DomainQuery1 q1;
+       struct netr_Authenticator a;
+#define ASYNC_COUNT 100
+       struct creds_CredentialState creds;
+       struct creds_CredentialState creds_async[ASYNC_COUNT];
+       struct rpc_request *req[ASYNC_COUNT];
+       int i;
+       int async_counter = 0;
+
+       if (!test_SetupCredentials3(p, mem_ctx, NETLOGON_NEG_AUTH2_ADS_FLAGS, &creds)) {
+               return False;
+       }
+
+       ZERO_STRUCT(r);
+       r.in.server_name = talloc_asprintf(mem_ctx, "\\\\%s", dcerpc_server_name(p));
+       r.in.computer_name = TEST_MACHINE_NAME;
+       r.in.unknown1 = 512;
+       r.in.level = 1;
+       r.in.credential = &a;
+       r.out.credential = &a;
+
+       r.in.i1[0] = 0;
+       r.in.i1[1] = 0;
+
+       r.in.query.query1 = &q1;
+       ZERO_STRUCT(q1);
+       
+       /* this should really be the fully qualified name */
+       q1.workstation_domain = TEST_MACHINE_NAME;
+       q1.workstation_site = "Default-First-Site-Name";
+       q1.blob2.length = 0;
+       q1.blob2.size = 0;
+       q1.blob2.data = NULL;
+       q1.product.string = "product string";
+
+       printf("Testing netr_LogonGetDomainInfo - async count %d\n", ASYNC_COUNT);
+
+       for (i=0;i<ASYNC_COUNT;i++) {
+               creds_client_authenticator(&creds, &a);
+
+               creds_async[i] = creds;
+               req[i] = dcerpc_netr_LogonGetDomainInfo_send(p, mem_ctx, &r);
+
+               req[i]->async.callback = async_callback;
+               req[i]->async.private = &async_counter;
+
+               /* even with this flush per request a w2k3 server seems to 
+                  clag with multiple outstanding requests. bleergh. */
+               if (event_loop_once(dcerpc_event_context(p)) != 0) {
+                       return False;
+               }
+       }
+
+       for (i=0;i<ASYNC_COUNT;i++) {
+               status = dcerpc_ndr_request_recv(req[i]);
+               if (!NT_STATUS_IS_OK(status) || !NT_STATUS_IS_OK(r.out.result)) {
+                       printf("netr_LogonGetDomainInfo_async(%d) - %s/%s\n", 
+                              i, nt_errstr(status), nt_errstr(r.out.result));
+                       break;
+               }
+
+               if (!creds_client_check(&creds_async[i], &a.cred)) {
+                       printf("Credential chaining failed at async %d\n", i);
+                       break;
+               }
+       }
+
+       printf("Testing netr_LogonGetDomainInfo - async count %d OK\n", async_counter);
+
+       return async_counter == ASYNC_COUNT;
+}
+
+
+BOOL torture_rpc_netlogon(void)
 {
         NTSTATUS status;
         struct dcerpc_pipe *p;
@@ -1713,6 +1818,10 @@ BOOL torture_rpc_netlogon(int dummy)
                ret = False;
        }
 
+       if (!test_GetDomainInfo_async(p, mem_ctx)) {
+               ret = False;
+       }
+
        talloc_destroy(mem_ctx);
 
        torture_rpc_close(p);