source/torture/rpc/samba3rpc.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    dcerpc torture tests, designed to walk Samba3 code paths
   5
   6    Copyright (C) Volker Lendecke 2006
   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 2 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, write to the Free Software
  20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21 */
  22
  23 #include "includes.h"
  24 #include "libcli/raw/libcliraw.h"
  25 #include "libcli/rap/rap.h"
  26 #include "torture/torture.h"
  27 #include "torture/util.h"
  28 #include "torture/rap/proto.h"
  29 #include "librpc/gen_ndr/ndr_lsa.h"
  30 #include "librpc/gen_ndr/ndr_lsa_c.h"
  31 #include "librpc/gen_ndr/ndr_samr.h"
  32 #include "librpc/gen_ndr/ndr_samr_c.h"
  33 #include "librpc/gen_ndr/ndr_netlogon.h"
  34 #include "librpc/gen_ndr/ndr_netlogon_c.h"
  35 #include "librpc/gen_ndr/ndr_srvsvc.h"
  36 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
  37 #include "librpc/gen_ndr/ndr_spoolss.h"
  38 #include "librpc/gen_ndr/ndr_spoolss_c.h"
  39 #include "librpc/gen_ndr/ndr_winreg.h"
  40 #include "librpc/gen_ndr/ndr_winreg_c.h"
  41 #include "librpc/gen_ndr/ndr_wkssvc.h"
  42 #include "librpc/gen_ndr/ndr_wkssvc_c.h"
  43 #include "lib/cmdline/popt_common.h"
  44 #include "librpc/rpc/dcerpc.h"
  45 #include "torture/rpc/rpc.h"
  46 #include "libcli/libcli.h"
  47 #include "libcli/composite/composite.h"
  48 #include "libcli/smb_composite/smb_composite.h"
  49 #include "libcli/auth/libcli_auth.h"
  50 #include "libcli/auth/credentials.h"
  51 #include "lib/crypto/crypto.h"
  52 #include "libcli/security/proto.h"
  53
  54 static struct cli_credentials *create_anon_creds(TALLOC_CTX *mem_ctx)
  55 {
  56         struct cli_credentials *result;
  57
  58         if (!(result = cli_credentials_init(mem_ctx))) {
  59                 return NULL;
  60         }
  61
  62         cli_credentials_set_conf(result);
  63         cli_credentials_set_anonymous(result);
  64
  65         return result;
  66 }
  67
  68 /*
  69  * This tests a RPC call using an invalid vuid
  70  */
  71
  72 BOOL torture_bind_authcontext(struct torture_context *torture)
  73 {
  74         TALLOC_CTX *mem_ctx;
  75         NTSTATUS status;
  76         BOOL ret = False;
  77         struct lsa_ObjectAttribute objectattr;
  78         struct lsa_OpenPolicy2 openpolicy;
  79         struct policy_handle handle;
  80         struct lsa_Close close_handle;
  81         struct smbcli_session *tmp;
  82         struct smbcli_session *session2;
  83         struct smbcli_state *cli;
  84         struct dcerpc_pipe *lsa_pipe;
  85         struct cli_credentials *anon_creds;
  86         struct smb_composite_sesssetup setup;
  87
  88         mem_ctx = talloc_init("torture_bind_authcontext");
  89
  90         if (mem_ctx == NULL) {
  91                 d_printf("talloc_init failed\n");
  92                 return False;
  93         }
  94
  95         status = smbcli_full_connection(mem_ctx, &cli,
  96                                         lp_parm_string(-1, "torture", "host"),
  97                                         "IPC$", NULL, cmdline_credentials,
  98                                         NULL);
  99         if (!NT_STATUS_IS_OK(status)) {
 100                 d_printf("smbcli_full_connection failed: %s\n",
 101                          nt_errstr(status));
 102                 goto done;
 103         }
 104
 105         lsa_pipe = dcerpc_pipe_init(mem_ctx, cli->transport->socket->event.ctx);
 106         if (lsa_pipe == NULL) {
 107                 d_printf("dcerpc_pipe_init failed\n");
 108                 goto done;
 109         }
 110
 111         status = dcerpc_pipe_open_smb(lsa_pipe->conn, cli->tree, "\\lsarpc");
 112         if (!NT_STATUS_IS_OK(status)) {
 113                 d_printf("dcerpc_pipe_open_smb failed: %s\n",
 114                          nt_errstr(status));
 115                 goto done;
 116         }
 117
 118         status = dcerpc_bind_auth_none(lsa_pipe, &dcerpc_table_lsarpc);
 119         if (!NT_STATUS_IS_OK(status)) {
 120                 d_printf("dcerpc_bind_auth_none failed: %s\n",
 121                          nt_errstr(status));
 122                 goto done;
 123         }
 124
 125         openpolicy.in.system_name =talloc_asprintf(
 126                 mem_ctx, "\\\\%s", dcerpc_server_name(lsa_pipe));
 127         ZERO_STRUCT(objectattr);
 128         openpolicy.in.attr = &objectattr;
 129         openpolicy.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 130         openpolicy.out.handle = &handle;
 131
 132         status = dcerpc_lsa_OpenPolicy2(lsa_pipe, mem_ctx, &openpolicy);
 133
 134         if (!NT_STATUS_IS_OK(status)) {
 135                 d_printf("dcerpc_lsa_OpenPolicy2 failed: %s\n",
 136                          nt_errstr(status));
 137                 goto done;
 138         }
 139
 140         close_handle.in.handle = &handle;
 141         close_handle.out.handle = &handle;
 142
 143         status = dcerpc_lsa_Close(lsa_pipe, mem_ctx, &close_handle);
 144         if (!NT_STATUS_IS_OK(status)) {
 145                 d_printf("dcerpc_lsa_Close failed: %s\n",
 146                          nt_errstr(status));
 147                 goto done;
 148         }
 149
 150         session2 = smbcli_session_init(cli->transport, mem_ctx, False);
 151         if (session2 == NULL) {
 152                 d_printf("smbcli_session_init failed\n");
 153                 goto done;
 154         }
 155
 156         if (!(anon_creds = create_anon_creds(mem_ctx))) {
 157                 d_printf("create_anon_creds failed\n");
 158                 goto done;
 159         }
 160
 161         setup.in.sesskey = cli->transport->negotiate.sesskey;
 162         setup.in.capabilities = cli->transport->negotiate.capabilities;
 163         setup.in.workgroup = "";
 164         setup.in.credentials = anon_creds;
 165
 166         status = smb_composite_sesssetup(session2, &setup);
 167         if (!NT_STATUS_IS_OK(status)) {
 168                 d_printf("anon session setup failed: %s\n",
 169                          nt_errstr(status));
 170                 goto done;
 171         }
 172
 173         tmp = cli->tree->session;
 174         cli->tree->session = session2;
 175
 176         status = dcerpc_lsa_OpenPolicy2(lsa_pipe, mem_ctx, &openpolicy);
 177
 178         cli->tree->session = tmp;
 179         talloc_free(lsa_pipe);
 180         lsa_pipe = NULL;
 181
 182         if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_HANDLE)) {
 183                 d_printf("dcerpc_lsa_OpenPolicy2 with wrong vuid gave %s, "
 184                          "expected NT_STATUS_INVALID_HANDLE\n",
 185                          nt_errstr(status));
 186                 goto done;
 187         }
 188
 189         ret = True;
 190  done:
 191         talloc_free(mem_ctx);
 192         return ret;
 193 }
 194
 195 /*
 196  * Bind to lsa using a specific auth method
 197  */
 198
 199 static BOOL bindtest(struct smbcli_state *cli,
 200                      struct cli_credentials *credentials,
 201                      uint8_t auth_type, uint8_t auth_level)
 202 {
 203         TALLOC_CTX *mem_ctx;
 204         BOOL ret = False;
 205         NTSTATUS status;
 206
 207         struct dcerpc_pipe *lsa_pipe;
 208         struct lsa_ObjectAttribute objectattr;
 209         struct lsa_OpenPolicy2 openpolicy;
 210         struct lsa_QueryInfoPolicy query;
 211         struct policy_handle handle;
 212         struct lsa_Close close_handle;
 213
 214         if ((mem_ctx = talloc_init("bindtest")) == NULL) {
 215                 d_printf("talloc_init failed\n");
 216                 return False;
 217         }
 218
 219         lsa_pipe = dcerpc_pipe_init(mem_ctx,
 220                                     cli->transport->socket->event.ctx);
 221         if (lsa_pipe == NULL) {
 222                 d_printf("dcerpc_pipe_init failed\n");
 223                 goto done;
 224         }
 225
 226         status = dcerpc_pipe_open_smb(lsa_pipe->conn, cli->tree, "\\lsarpc");
 227         if (!NT_STATUS_IS_OK(status)) {
 228                 d_printf("dcerpc_pipe_open_smb failed: %s\n",
 229                          nt_errstr(status));
 230                 goto done;
 231         }
 232
 233         status = dcerpc_bind_auth(lsa_pipe, &dcerpc_table_lsarpc,
 234                                   credentials, auth_type, auth_level,
 235                                   NULL);
 236         if (!NT_STATUS_IS_OK(status)) {
 237                 d_printf("dcerpc_bind_auth failed: %s\n", nt_errstr(status));
 238                 goto done;
 239         }
 240
 241         openpolicy.in.system_name =talloc_asprintf(
 242                 mem_ctx, "\\\\%s", dcerpc_server_name(lsa_pipe));
 243         ZERO_STRUCT(objectattr);
 244         openpolicy.in.attr = &objectattr;
 245         openpolicy.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 246         openpolicy.out.handle = &handle;
 247
 248         status = dcerpc_lsa_OpenPolicy2(lsa_pipe, mem_ctx, &openpolicy);
 249
 250         if (!NT_STATUS_IS_OK(status)) {
 251                 d_printf("dcerpc_lsa_OpenPolicy2 failed: %s\n",
 252                          nt_errstr(status));
 253                 goto done;
 254         }
 255
 256         query.in.handle = &handle;
 257         query.in.level = LSA_POLICY_INFO_DOMAIN;
 258
 259         status = dcerpc_lsa_QueryInfoPolicy(lsa_pipe, mem_ctx, &query);
 260         if (!NT_STATUS_IS_OK(status)) {
 261                 d_printf("dcerpc_lsa_QueryInfoPolicy failed: %s\n",
 262                          nt_errstr(status));
 263                 goto done;
 264         }
 265
 266         close_handle.in.handle = &handle;
 267         close_handle.out.handle = &handle;
 268
 269         status = dcerpc_lsa_Close(lsa_pipe, mem_ctx, &close_handle);
 270         if (!NT_STATUS_IS_OK(status)) {
 271                 d_printf("dcerpc_lsa_Close failed: %s\n",
 272                          nt_errstr(status));
 273                 goto done;
 274         }
 275
 276         ret = True;
 277  done:
 278         talloc_free(mem_ctx);
 279         return ret;
 280 }
 281
 282 /*
 283  * test authenticated RPC binds with the variants Samba3 does support
 284  */
 285
 286 BOOL torture_bind_samba3(struct torture_context *torture)
 287 {
 288         TALLOC_CTX *mem_ctx;
 289         NTSTATUS status;
 290         BOOL ret = False;
 291         struct smbcli_state *cli;
 292
 293         mem_ctx = talloc_init("torture_bind_authcontext");
 294
 295         if (mem_ctx == NULL) {
 296                 d_printf("talloc_init failed\n");
 297                 return False;
 298         }
 299
 300         status = smbcli_full_connection(mem_ctx, &cli,
 301                                         lp_parm_string(-1, "torture", "host"),
 302                                         "IPC$", NULL, cmdline_credentials,
 303                                         NULL);
 304         if (!NT_STATUS_IS_OK(status)) {
 305                 d_printf("smbcli_full_connection failed: %s\n",
 306                          nt_errstr(status));
 307                 goto done;
 308         }
 309
 310         ret = True;
 311
 312         ret &= bindtest(cli, cmdline_credentials, DCERPC_AUTH_TYPE_NTLMSSP,
 313                         DCERPC_AUTH_LEVEL_INTEGRITY);
 314         ret &= bindtest(cli, cmdline_credentials, DCERPC_AUTH_TYPE_NTLMSSP,
 315                         DCERPC_AUTH_LEVEL_PRIVACY);
 316         ret &= bindtest(cli, cmdline_credentials, DCERPC_AUTH_TYPE_SPNEGO,
 317                         DCERPC_AUTH_LEVEL_INTEGRITY);
 318         ret &= bindtest(cli, cmdline_credentials, DCERPC_AUTH_TYPE_SPNEGO,
 319                         DCERPC_AUTH_LEVEL_PRIVACY);
 320
 321  done:
 322         talloc_free(mem_ctx);
 323         return ret;
 324 }
 325
 326 /*
 327  * Lookup or create a user and return all necessary info
 328  */
 329
 330 static NTSTATUS get_usr_handle(struct smbcli_state *cli,
 331                                TALLOC_CTX *mem_ctx,
 332                                struct cli_credentials *admin_creds,
 333                                uint8_t auth_type,
 334                                uint8_t auth_level,
 335                                const char *username,
 336                                char **domain,
 337                                struct dcerpc_pipe **result_pipe,
 338                                struct policy_handle **result_handle,
 339                                struct dom_sid **sid)
 340 {
 341         struct dcerpc_pipe *samr_pipe;
 342         NTSTATUS status;
 343         struct policy_handle conn_handle;
 344         struct policy_handle domain_handle;
 345         struct policy_handle *user_handle;
 346         struct samr_Connect2 conn;
 347         struct samr_EnumDomains enumdom;
 348         uint32_t resume_handle = 0;
 349         struct samr_LookupDomain l;
 350         int dom_idx;
 351         struct lsa_String domain_name;
 352         struct lsa_String user_name;
 353         struct samr_OpenDomain o;
 354         struct samr_CreateUser2 c;
 355         uint32_t user_rid,access_granted;
 356
 357         samr_pipe = dcerpc_pipe_init(mem_ctx,
 358                                      cli->transport->socket->event.ctx);
 359         if (samr_pipe == NULL) {
 360                 d_printf("dcerpc_pipe_init failed\n");
 361                 status = NT_STATUS_NO_MEMORY;
 362                 goto fail;
 363         }
 364
 365         status = dcerpc_pipe_open_smb(samr_pipe->conn, cli->tree, "\\samr");
 366         if (!NT_STATUS_IS_OK(status)) {
 367                 d_printf("dcerpc_pipe_open_smb failed: %s\n",
 368                          nt_errstr(status));
 369                 goto fail;
 370         }
 371
 372         if (admin_creds != NULL) {
 373                 status = dcerpc_bind_auth(samr_pipe, &dcerpc_table_samr,
 374                                           admin_creds, auth_type, auth_level,
 375                                           NULL);
 376                 if (!NT_STATUS_IS_OK(status)) {
 377                         d_printf("dcerpc_bind_auth failed: %s\n",
 378                                  nt_errstr(status));
 379                         goto fail;
 380                 }
 381         } else {
 382                 /* We must have an authenticated SMB connection */
 383                 status = dcerpc_bind_auth_none(samr_pipe, &dcerpc_table_samr);
 384                 if (!NT_STATUS_IS_OK(status)) {
 385                         d_printf("dcerpc_bind_auth_none failed: %s\n",
 386                                  nt_errstr(status));
 387                         goto fail;
 388                 }
 389         }
 390
 391         conn.in.system_name = talloc_asprintf(
 392                 mem_ctx, "\\\\%s", dcerpc_server_name(samr_pipe));
 393         conn.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 394         conn.out.connect_handle = &conn_handle;
 395
 396         status = dcerpc_samr_Connect2(samr_pipe, mem_ctx, &conn);
 397         if (!NT_STATUS_IS_OK(status)) {
 398                 d_printf("samr_Connect2 failed: %s\n", nt_errstr(status));
 399                 goto fail;
 400         }
 401
 402         enumdom.in.connect_handle = &conn_handle;
 403         enumdom.in.resume_handle = &resume_handle;
 404         enumdom.in.buf_size = (uint32_t)-1;
 405         enumdom.out.resume_handle = &resume_handle;
 406
 407         status = dcerpc_samr_EnumDomains(samr_pipe, mem_ctx, &enumdom);
 408         if (!NT_STATUS_IS_OK(status)) {
 409                 d_printf("samr_EnumDomains failed: %s\n", nt_errstr(status));
 410                 goto fail;
 411         }
 412
 413         if (enumdom.out.num_entries != 2) {
 414                 d_printf("samr_EnumDomains returned %d entries, expected 2\n",
 415                          enumdom.out.num_entries);
 416                 status = NT_STATUS_UNSUCCESSFUL;
 417                 goto fail;
 418         }
 419
 420         dom_idx = strequal(enumdom.out.sam->entries[0].name.string,
 421                            "builtin") ? 1:0;
 422
 423         l.in.connect_handle = &conn_handle;
 424         domain_name.string = enumdom.out.sam->entries[0].name.string;
 425         *domain = talloc_strdup(mem_ctx, domain_name.string);
 426         l.in.domain_name = &domain_name;
 427
 428         status = dcerpc_samr_LookupDomain(samr_pipe, mem_ctx, &l);
 429         if (!NT_STATUS_IS_OK(status)) {
 430                 d_printf("samr_LookupDomain failed: %s\n", nt_errstr(status));
 431                 goto fail;
 432         }
 433
 434         o.in.connect_handle = &conn_handle;
 435         o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 436         o.in.sid = l.out.sid;
 437         o.out.domain_handle = &domain_handle;
 438
 439         status = dcerpc_samr_OpenDomain(samr_pipe, mem_ctx, &o);
 440         if (!NT_STATUS_IS_OK(status)) {
 441                 d_printf("samr_OpenDomain failed: %s\n", nt_errstr(status));
 442                 goto fail;
 443         }
 444
 445         c.in.domain_handle = &domain_handle;
 446         user_name.string = username;
 447         c.in.account_name = &user_name;
 448         c.in.acct_flags = ACB_NORMAL;
 449         c.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 450         user_handle = talloc(mem_ctx, struct policy_handle);
 451         c.out.user_handle = user_handle;
 452         c.out.access_granted = &access_granted;
 453         c.out.rid = &user_rid;
 454
 455         status = dcerpc_samr_CreateUser2(samr_pipe, mem_ctx, &c);
 456
 457         if (NT_STATUS_EQUAL(status, NT_STATUS_USER_EXISTS)) {
 458                 struct samr_LookupNames ln;
 459                 struct samr_OpenUser ou;
 460
 461                 ln.in.domain_handle = &domain_handle;
 462                 ln.in.num_names = 1;
 463                 ln.in.names = &user_name;
 464
 465                 status = dcerpc_samr_LookupNames(samr_pipe, mem_ctx, &ln);
 466                 if (!NT_STATUS_IS_OK(status)) {
 467                         d_printf("samr_LookupNames failed: %s\n",
 468                                  nt_errstr(status));
 469                         goto fail;
 470                 }
 471
 472                 ou.in.domain_handle = &domain_handle;
 473                 ou.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 474                 user_rid = ou.in.rid = ln.out.rids.ids[0];
 475                 ou.out.user_handle = user_handle;
 476
 477                 status = dcerpc_samr_OpenUser(samr_pipe, mem_ctx, &ou);
 478                 if (!NT_STATUS_IS_OK(status)) {
 479                         d_printf("samr_OpenUser failed: %s\n",
 480                                  nt_errstr(status));
 481                         goto fail;
 482                 }
 483         }
 484
 485         if (!NT_STATUS_IS_OK(status)) {
 486                 d_printf("samr_CreateUser failed: %s\n", nt_errstr(status));
 487                 goto fail;
 488         }
 489
 490         *result_pipe = samr_pipe;
 491         *result_handle = user_handle;
 492         if (sid != NULL) {
 493                 *sid = dom_sid_add_rid(mem_ctx, l.out.sid, user_rid);
 494         }
 495         return NT_STATUS_OK;
 496
 497  fail:
 498         return status;
 499 }
 500
 501 /*
 502  * Create a test user
 503  */
 504
 505 static BOOL create_user(TALLOC_CTX *mem_ctx, struct smbcli_state *cli,
 506                         struct cli_credentials *admin_creds,
 507                         const char *username, const char *password,
 508                         char **domain_name,
 509                         struct dom_sid **user_sid)
 510 {
 511         TALLOC_CTX *tmp_ctx;
 512         NTSTATUS status;
 513         struct dcerpc_pipe *samr_pipe;
 514         struct policy_handle *wks_handle;
 515         BOOL ret = False;
 516
 517         if (!(tmp_ctx = talloc_new(mem_ctx))) {
 518                 d_printf("talloc_init failed\n");
 519                 return False;
 520         }
 521
 522         status = get_usr_handle(cli, tmp_ctx, admin_creds,
 523                                 DCERPC_AUTH_TYPE_NTLMSSP,
 524                                 DCERPC_AUTH_LEVEL_INTEGRITY,
 525                                 username, domain_name, &samr_pipe, &wks_handle,
 526                                 user_sid);
 527         if (!NT_STATUS_IS_OK(status)) {
 528                 d_printf("get_wks_handle failed: %s\n", nt_errstr(status));
 529                 goto done;
 530         }
 531
 532         {
 533                 struct samr_SetUserInfo2 sui2;
 534                 struct samr_SetUserInfo sui;
 535                 struct samr_QueryUserInfo qui;
 536                 union samr_UserInfo u_info;
 537                 DATA_BLOB session_key;
 538
 539                 encode_pw_buffer(u_info.info24.password.data, password,
 540                                  STR_UNICODE);
 541                 u_info.info24.pw_len =  strlen_m(password)*2;
 542
 543                 status = dcerpc_fetch_session_key(samr_pipe, &session_key);
 544                 if (!NT_STATUS_IS_OK(status)) {
 545                         d_printf("dcerpc_fetch_session_key failed\n");
 546                         goto done;
 547                 }
 548                 arcfour_crypt_blob(u_info.info24.password.data, 516,
 549                                    &session_key);
 550                 sui2.in.user_handle = wks_handle;
 551                 sui2.in.info = &u_info;
 552                 sui2.in.level = 24;
 553
 554                 status = dcerpc_samr_SetUserInfo2(samr_pipe, tmp_ctx, &sui2);
 555                 if (!NT_STATUS_IS_OK(status)) {
 556                         d_printf("samr_SetUserInfo(24) failed: %s\n",
 557                                  nt_errstr(status));
 558                         goto done;
 559                 }
 560
 561                 u_info.info16.acct_flags = ACB_NORMAL;
 562                 sui.in.user_handle = wks_handle;
 563                 sui.in.info = &u_info;
 564                 sui.in.level = 16;
 565
 566                 status = dcerpc_samr_SetUserInfo(samr_pipe, tmp_ctx, &sui);
 567                 if (!NT_STATUS_IS_OK(status)) {
 568                         d_printf("samr_SetUserInfo(16) failed\n");
 569                         goto done;
 570                 }
 571
 572                 qui.in.user_handle = wks_handle;
 573                 qui.in.level = 21;
 574
 575                 status = dcerpc_samr_QueryUserInfo(samr_pipe, tmp_ctx, &qui);
 576                 if (!NT_STATUS_IS_OK(status)) {
 577                         d_printf("samr_QueryUserInfo(21) failed\n");
 578                         goto done;
 579                 }
 580
 581                 qui.out.info->info21.allow_password_change = 0;
 582                 qui.out.info->info21.force_password_change = 0;
 583                 qui.out.info->info21.account_name.string = NULL;
 584                 qui.out.info->info21.rid = 0;
 585                 qui.out.info->info21.fields_present = 0x81827fa; /* copy usrmgr.exe */
 586
 587                 u_info.info21 = qui.out.info->info21;
 588                 sui.in.user_handle = wks_handle;
 589                 sui.in.info = &u_info;
 590                 sui.in.level = 21;
 591
 592                 status = dcerpc_samr_SetUserInfo(samr_pipe, tmp_ctx, &sui);
 593                 if (!NT_STATUS_IS_OK(status)) {
 594                         d_printf("samr_SetUserInfo(21) failed\n");
 595                         goto done;
 596                 }
 597         }
 598
 599         *domain_name= talloc_steal(mem_ctx, *domain_name);
 600         *user_sid = talloc_steal(mem_ctx, *user_sid);
 601         ret = True;
 602  done:
 603         talloc_free(tmp_ctx);
 604         return ret;
 605 }
 606
 607 /*
 608  * Delete a test user
 609  */
 610
 611 static BOOL delete_user(struct smbcli_state *cli,
 612                         struct cli_credentials *admin_creds,
 613                         const char *username)
 614 {
 615         TALLOC_CTX *mem_ctx;
 616         NTSTATUS status;
 617         char *dom_name;
 618         struct dcerpc_pipe *samr_pipe;
 619         struct policy_handle *user_handle;
 620         BOOL ret = False;
 621
 622         if ((mem_ctx = talloc_init("leave")) == NULL) {
 623                 d_printf("talloc_init failed\n");
 624                 return False;
 625         }
 626
 627         status = get_usr_handle(cli, mem_ctx, admin_creds,
 628                                 DCERPC_AUTH_TYPE_NTLMSSP,
 629                                 DCERPC_AUTH_LEVEL_INTEGRITY,
 630                                 username, &dom_name, &samr_pipe,
 631                                 &user_handle, NULL);
 632
 633         if (!NT_STATUS_IS_OK(status)) {
 634                 d_printf("get_wks_handle failed: %s\n", nt_errstr(status));
 635                 goto done;
 636         }
 637
 638         {
 639                 struct samr_DeleteUser d;
 640
 641                 d.in.user_handle = user_handle;
 642                 d.out.user_handle = user_handle;
 643
 644                 status = dcerpc_samr_DeleteUser(samr_pipe, mem_ctx, &d);
 645                 if (!NT_STATUS_IS_OK(status)) {
 646                         d_printf("samr_DeleteUser failed\n");
 647                         goto done;
 648                 }
 649         }
 650
 651         ret = True;
 652
 653  done:
 654         talloc_free(mem_ctx);
 655         return ret;
 656 }
 657
 658 /*
 659  * Do a Samba3-style join
 660  */
 661
 662 static BOOL join3(struct smbcli_state *cli,
 663                   BOOL use_level25,
 664                   struct cli_credentials *admin_creds,
 665                   struct cli_credentials *wks_creds)
 666 {
 667         TALLOC_CTX *mem_ctx;
 668         NTSTATUS status;
 669         char *dom_name;
 670         struct dcerpc_pipe *samr_pipe;
 671         struct policy_handle *wks_handle;
 672         BOOL ret = False;
 673
 674         if ((mem_ctx = talloc_init("join3")) == NULL) {
 675                 d_printf("talloc_init failed\n");
 676                 return False;
 677         }
 678
 679         status = get_usr_handle(
 680                 cli, mem_ctx, admin_creds,
 681                 DCERPC_AUTH_TYPE_NTLMSSP,
 682                 DCERPC_AUTH_LEVEL_PRIVACY,
 683                 talloc_asprintf(mem_ctx, "%s$",
 684                                 cli_credentials_get_workstation(wks_creds)),
 685                 &dom_name, &samr_pipe, &wks_handle, NULL);
 686
 687         if (!NT_STATUS_IS_OK(status)) {
 688                 d_printf("get_wks_handle failed: %s\n", nt_errstr(status));
 689                 goto done;
 690         }
 691
 692         cli_credentials_set_domain(wks_creds, dom_name, CRED_SPECIFIED);
 693
 694         if (use_level25) {
 695                 struct samr_SetUserInfo2 sui2;
 696                 union samr_UserInfo u_info;
 697                 struct samr_UserInfo21 *i21 = &u_info.info25.info;
 698                 DATA_BLOB session_key;
 699                 DATA_BLOB confounded_session_key = data_blob_talloc(
 700                         mem_ctx, NULL, 16);
 701                 struct MD5Context ctx;
 702                 uint8_t confounder[16];
 703
 704                 ZERO_STRUCT(u_info);
 705
 706                 i21->full_name.string = talloc_asprintf(
 707                         mem_ctx, "%s$",
 708                         cli_credentials_get_workstation(wks_creds));
 709                 i21->acct_flags = ACB_WSTRUST;
 710                 i21->fields_present = SAMR_FIELD_FULL_NAME |
 711                         SAMR_FIELD_ACCT_FLAGS | SAMR_FIELD_PASSWORD;
 712
 713                 encode_pw_buffer(u_info.info25.password.data,
 714                                  cli_credentials_get_password(wks_creds),
 715                                  STR_UNICODE);
 716                 status = dcerpc_fetch_session_key(samr_pipe, &session_key);
 717                 if (!NT_STATUS_IS_OK(status)) {
 718                         d_printf("dcerpc_fetch_session_key failed: %s\n",
 719                                  nt_errstr(status));
 720                         goto done;
 721                 }
 722                 generate_random_buffer((uint8_t *)confounder, 16);
 723
 724                 MD5Init(&ctx);
 725                 MD5Update(&ctx, confounder, 16);
 726                 MD5Update(&ctx, session_key.data, session_key.length);
 727                 MD5Final(confounded_session_key.data, &ctx);
 728
 729                 arcfour_crypt_blob(u_info.info25.password.data, 516,
 730                                    &confounded_session_key);
 731                 memcpy(&u_info.info25.password.data[516], confounder, 16);
 732
 733                 sui2.in.user_handle = wks_handle;
 734                 sui2.in.level = 25;
 735                 sui2.in.info = &u_info;
 736
 737                 status = dcerpc_samr_SetUserInfo2(samr_pipe, mem_ctx, &sui2);
 738                 if (!NT_STATUS_IS_OK(status)) {
 739                         d_printf("samr_SetUserInfo2(25) failed: %s\n",
 740                                  nt_errstr(status));
 741                         goto done;
 742                 }
 743         } else {
 744                 struct samr_SetUserInfo2 sui2;
 745                 struct samr_SetUserInfo sui;
 746                 union samr_UserInfo u_info;
 747                 DATA_BLOB session_key;
 748
 749                 encode_pw_buffer(u_info.info24.password.data,
 750                                  cli_credentials_get_password(wks_creds),
 751                                  STR_UNICODE);
 752                 u_info.info24.pw_len =
 753                         strlen_m(cli_credentials_get_password(wks_creds))*2;
 754
 755                 status = dcerpc_fetch_session_key(samr_pipe, &session_key);
 756                 if (!NT_STATUS_IS_OK(status)) {
 757                         d_printf("dcerpc_fetch_session_key failed\n");
 758                         goto done;
 759                 }
 760                 arcfour_crypt_blob(u_info.info24.password.data, 516,
 761                                    &session_key);
 762                 sui2.in.user_handle = wks_handle;
 763                 sui2.in.info = &u_info;
 764                 sui2.in.level = 24;
 765
 766                 status = dcerpc_samr_SetUserInfo2(samr_pipe, mem_ctx, &sui2);
 767                 if (!NT_STATUS_IS_OK(status)) {
 768                         d_printf("samr_SetUserInfo(24) failed: %s\n",
 769                                  nt_errstr(status));
 770                         goto done;
 771                 }
 772
 773                 u_info.info16.acct_flags = ACB_WSTRUST;
 774                 sui.in.user_handle = wks_handle;
 775                 sui.in.info = &u_info;
 776                 sui.in.level = 16;
 777
 778                 status = dcerpc_samr_SetUserInfo(samr_pipe, mem_ctx, &sui);
 779                 if (!NT_STATUS_IS_OK(status)) {
 780                         d_printf("samr_SetUserInfo(16) failed\n");
 781                         goto done;
 782                 }
 783         }
 784
 785         ret = True;
 786
 787  done:
 788         talloc_free(mem_ctx);
 789         return ret;
 790 }
 791
 792 /*
 793  * Do a ReqChallenge/Auth2 and get the wks creds
 794  */
 795
 796 static BOOL auth2(struct smbcli_state *cli,
 797                   struct cli_credentials *wks_cred)
 798 {
 799         TALLOC_CTX *mem_ctx;
 800         struct dcerpc_pipe *net_pipe;
 801         BOOL result = False;
 802         NTSTATUS status;
 803         struct netr_ServerReqChallenge r;
 804         struct netr_Credential netr_cli_creds;
 805         struct netr_Credential netr_srv_creds;
 806         uint32_t negotiate_flags;
 807         struct netr_ServerAuthenticate2 a;
 808         struct creds_CredentialState *creds_state;
 809         struct netr_Credential netr_cred;
 810         struct samr_Password mach_pw;
 811
 812         mem_ctx = talloc_new(NULL);
 813         if (mem_ctx == NULL) {
 814                 d_printf("talloc_new failed\n");
 815                 return False;
 816         }
 817
 818         net_pipe = dcerpc_pipe_init(mem_ctx,
 819                                     cli->transport->socket->event.ctx);
 820         if (net_pipe == NULL) {
 821                 d_printf("dcerpc_pipe_init failed\n");
 822                 goto done;
 823         }
 824
 825         status = dcerpc_pipe_open_smb(net_pipe->conn, cli->tree, "\\netlogon");
 826         if (!NT_STATUS_IS_OK(status)) {
 827                 d_printf("dcerpc_pipe_open_smb failed: %s\n",
 828                          nt_errstr(status));
 829                 goto done;
 830         }
 831
 832         status = dcerpc_bind_auth_none(net_pipe, &dcerpc_table_netlogon);
 833         if (!NT_STATUS_IS_OK(status)) {
 834                 d_printf("dcerpc_bind_auth_none failed: %s\n",
 835                          nt_errstr(status));
 836                 goto done;
 837         }
 838
 839         r.in.computer_name = cli_credentials_get_workstation(wks_cred);
 840         r.in.server_name = talloc_asprintf(
 841                 mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
 842         if (r.in.server_name == NULL) {
 843                 d_printf("talloc_asprintf failed\n");
 844                 goto done;
 845         }
 846         generate_random_buffer(netr_cli_creds.data,
 847                                sizeof(netr_cli_creds.data));
 848         r.in.credentials = &netr_cli_creds;
 849         r.out.credentials = &netr_srv_creds;
 850
 851         status = dcerpc_netr_ServerReqChallenge(net_pipe, mem_ctx, &r);
 852         if (!NT_STATUS_IS_OK(status)) {
 853                 d_printf("netr_ServerReqChallenge failed: %s\n",
 854                          nt_errstr(status));
 855                 goto done;
 856         }
 857
 858         negotiate_flags = NETLOGON_NEG_AUTH2_FLAGS;
 859         E_md4hash(cli_credentials_get_password(wks_cred), mach_pw.hash);
 860
 861         creds_state = talloc(mem_ctx, struct creds_CredentialState);
 862         creds_client_init(creds_state, r.in.credentials,
 863                           r.out.credentials, &mach_pw,
 864                           &netr_cred, negotiate_flags);
 865
 866         a.in.server_name = talloc_asprintf(
 867                 mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
 868         a.in.account_name = talloc_asprintf(
 869                 mem_ctx, "%s$", cli_credentials_get_workstation(wks_cred));
 870         a.in.computer_name = cli_credentials_get_workstation(wks_cred);
 871         a.in.secure_channel_type = SEC_CHAN_WKSTA;
 872         a.in.negotiate_flags = &negotiate_flags;
 873         a.out.negotiate_flags = &negotiate_flags;
 874         a.in.credentials = &netr_cred;
 875         a.out.credentials = &netr_cred;
 876
 877         status = dcerpc_netr_ServerAuthenticate2(net_pipe, mem_ctx, &a);
 878         if (!NT_STATUS_IS_OK(status)) {
 879                 d_printf("netr_ServerServerAuthenticate2 failed: %s\n",
 880                          nt_errstr(status));
 881                 goto done;
 882         }
 883
 884         if (!creds_client_check(creds_state, a.out.credentials)) {
 885                 d_printf("creds_client_check failed\n");
 886                 goto done;
 887         }
 888
 889         cli_credentials_set_netlogon_creds(wks_cred, creds_state);
 890
 891         result = True;
 892
 893  done:
 894         talloc_free(mem_ctx);
 895         return result;
 896 }
 897
 898 /*
 899  * Do a couple of schannel protected Netlogon ops: Interactive and Network
 900  * login, and change the wks password
 901  */
 902
 903 static BOOL schan(struct smbcli_state *cli,
 904                   struct cli_credentials *wks_creds,
 905                   struct cli_credentials *user_creds)
 906 {
 907         TALLOC_CTX *mem_ctx;
 908         NTSTATUS status;
 909         BOOL ret = False;
 910         struct dcerpc_pipe *net_pipe;
 911         int i;
 912
 913         mem_ctx = talloc_new(NULL);
 914         if (mem_ctx == NULL) {
 915                 d_printf("talloc_new failed\n");
 916                 return False;
 917         }
 918
 919         net_pipe = dcerpc_pipe_init(mem_ctx,
 920                                     cli->transport->socket->event.ctx);
 921         if (net_pipe == NULL) {
 922                 d_printf("dcerpc_pipe_init failed\n");
 923                 goto done;
 924         }
 925
 926         status = dcerpc_pipe_open_smb(net_pipe->conn, cli->tree, "\\netlogon");
 927         if (!NT_STATUS_IS_OK(status)) {
 928                 d_printf("dcerpc_pipe_open_smb failed: %s\n",
 929                          nt_errstr(status));
 930                 goto done;
 931         }
 932
 933 #if 0
 934         net_pipe->conn->flags |= DCERPC_DEBUG_PRINT_IN |
 935                 DCERPC_DEBUG_PRINT_OUT;
 936 #endif
 937 #if 1
 938         net_pipe->conn->flags |= (DCERPC_SIGN | DCERPC_SEAL);
 939         status = dcerpc_bind_auth(net_pipe, &dcerpc_table_netlogon,
 940                                   wks_creds, DCERPC_AUTH_TYPE_SCHANNEL,
 941                                   DCERPC_AUTH_LEVEL_PRIVACY,
 942                                   NULL);
 943 #else
 944         status = dcerpc_bind_auth_none(net_pipe, &dcerpc_table_netlogon);
 945 #endif
 946         if (!NT_STATUS_IS_OK(status)) {
 947                 d_printf("schannel bind failed: %s\n", nt_errstr(status));
 948                 goto done;
 949         }
 950
 951
 952         for (i=2; i<4; i++) {
 953                 int flags;
 954                 DATA_BLOB chal, nt_resp, lm_resp, names_blob, session_key;
 955                 struct creds_CredentialState *creds_state;
 956                 struct netr_Authenticator netr_auth, netr_auth2;
 957                 struct netr_NetworkInfo ninfo;
 958                 struct netr_PasswordInfo pinfo;
 959                 struct netr_LogonSamLogon r;
 960
 961                 flags = CLI_CRED_LANMAN_AUTH | CLI_CRED_NTLM_AUTH |
 962                         CLI_CRED_NTLMv2_AUTH;
 963
 964                 chal = data_blob_talloc(mem_ctx, NULL, 8);
 965                 if (chal.data == NULL) {
 966                         d_printf("data_blob_talloc failed\n");
 967                         goto done;
 968                 }
 969
 970                 generate_random_buffer(chal.data, chal.length);
 971                 names_blob = NTLMv2_generate_names_blob(
 972                         mem_ctx, cli_credentials_get_workstation(user_creds),
 973                         cli_credentials_get_domain(user_creds));
 974                 status = cli_credentials_get_ntlm_response(
 975                         user_creds, mem_ctx, &flags, chal, names_blob,
 976                         &lm_resp, &nt_resp, NULL, NULL);
 977                 if (!NT_STATUS_IS_OK(status)) {
 978                         d_printf("cli_credentials_get_ntlm_response failed:"
 979                                  " %s\n", nt_errstr(status));
 980                         goto done;
 981                 }
 982
 983                 creds_state = cli_credentials_get_netlogon_creds(wks_creds);
 984                 creds_client_authenticator(creds_state, &netr_auth);
 985
 986                 ninfo.identity_info.account_name.string =
 987                         cli_credentials_get_username(user_creds);
 988                 ninfo.identity_info.domain_name.string =
 989                         cli_credentials_get_domain(user_creds);
 990                 ninfo.identity_info.parameter_control = 0;
 991                 ninfo.identity_info.logon_id_low = 0;
 992                 ninfo.identity_info.logon_id_high = 0;
 993                 ninfo.identity_info.workstation.string =
 994                         cli_credentials_get_workstation(user_creds);
 995                 memcpy(ninfo.challenge, chal.data, sizeof(ninfo.challenge));
 996                 ninfo.nt.length = nt_resp.length;
 997                 ninfo.nt.data = nt_resp.data;
 998                 ninfo.lm.length = lm_resp.length;
 999                 ninfo.lm.data = lm_resp.data;
1000
1001                 r.in.server_name = talloc_asprintf(
1002                         mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
1003                 ZERO_STRUCT(netr_auth2);
1004                 r.in.computer_name =
1005                         cli_credentials_get_workstation(wks_creds);
1006                 r.in.credential = &netr_auth;
1007                 r.in.return_authenticator = &netr_auth2;
1008                 r.in.logon_level = 2;
1009                 r.in.validation_level = i;
1010                 r.in.logon.network = &ninfo;
1011                 r.out.return_authenticator = NULL;
1012
1013                 status = dcerpc_netr_LogonSamLogon(net_pipe, mem_ctx, &r);
1014                 if (!NT_STATUS_IS_OK(status)) {
1015                         d_printf("netr_LogonSamLogon failed: %s\n",
1016                                  nt_errstr(status));
1017                         goto done;
1018                 }
1019
1020                 if ((r.out.return_authenticator == NULL) ||
1021                     (!creds_client_check(creds_state,
1022                                          &r.out.return_authenticator->cred))) {
1023                         d_printf("Credentials check failed!\n");
1024                         goto done;
1025                 }
1026
1027                 creds_client_authenticator(creds_state, &netr_auth);
1028
1029                 pinfo.identity_info = ninfo.identity_info;
1030                 ZERO_STRUCT(pinfo.lmpassword.hash);
1031                 E_md4hash(cli_credentials_get_password(user_creds),
1032                           pinfo.ntpassword.hash);
1033                 session_key = data_blob_talloc(mem_ctx,
1034                                                creds_state->session_key, 16);
1035                 arcfour_crypt_blob(pinfo.ntpassword.hash,
1036                                    sizeof(pinfo.ntpassword.hash),
1037                                    &session_key);
1038
1039                 r.in.logon_level = 1;
1040                 r.in.logon.password = &pinfo;
1041                 r.out.return_authenticator = NULL;
1042
1043                 status = dcerpc_netr_LogonSamLogon(net_pipe, mem_ctx, &r);
1044                 if (!NT_STATUS_IS_OK(status)) {
1045                         d_printf("netr_LogonSamLogon failed: %s\n",
1046                                  nt_errstr(status));
1047                         goto done;
1048                 }
1049
1050                 if ((r.out.return_authenticator == NULL) ||
1051                     (!creds_client_check(creds_state,
1052                                          &r.out.return_authenticator->cred))) {
1053                         d_printf("Credentials check failed!\n");
1054                         goto done;
1055                 }
1056         }
1057
1058         {
1059                 struct netr_ServerPasswordSet s;
1060                 char *password = generate_random_str(wks_creds, 8);
1061                 struct creds_CredentialState *creds_state;
1062
1063                 s.in.server_name = talloc_asprintf(
1064                         mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
1065                 s.in.computer_name = cli_credentials_get_workstation(wks_creds);
1066                 s.in.account_name = talloc_asprintf(
1067                         mem_ctx, "%s$", s.in.computer_name);
1068                 s.in.secure_channel_type = SEC_CHAN_WKSTA;
1069                 E_md4hash(password, s.in.new_password.hash);
1070
1071                 creds_state = cli_credentials_get_netlogon_creds(wks_creds);
1072                 creds_des_encrypt(creds_state, &s.in.new_password);
1073                 creds_client_authenticator(creds_state, &s.in.credential);
1074
1075                 status = dcerpc_netr_ServerPasswordSet(net_pipe, mem_ctx, &s);
1076                 if (!NT_STATUS_IS_OK(status)) {
1077                         printf("ServerPasswordSet - %s\n", nt_errstr(status));
1078                         goto done;
1079                 }
1080
1081                 if (!creds_client_check(creds_state,
1082                                         &s.out.return_authenticator.cred)) {
1083                         printf("Credential chaining failed\n");
1084                 }
1085
1086                 cli_credentials_set_password(wks_creds, password,
1087                                              CRED_SPECIFIED);
1088         }
1089
1090         ret = True;
1091  done:
1092         talloc_free(mem_ctx);
1093         return ret;
1094 }
1095
1096 /*
1097  * Delete the wks account again
1098  */
1099
1100 static BOOL leave(struct smbcli_state *cli,
1101                   struct cli_credentials *admin_creds,
1102                   struct cli_credentials *wks_creds)
1103 {
1104         char *wks_name = talloc_asprintf(
1105                 NULL, "%s$", cli_credentials_get_workstation(wks_creds));
1106         BOOL ret;
1107
1108         ret = delete_user(cli, admin_creds, wks_name);
1109         talloc_free(wks_name);
1110         return ret;
1111 }
1112
1113 /*
1114  * Test the Samba3 DC code a bit. Join, do some schan netlogon ops, leave
1115  */
1116
1117 BOOL torture_netlogon_samba3(struct torture_context *torture)
1118 {
1119         TALLOC_CTX *mem_ctx;
1120         NTSTATUS status;
1121         BOOL ret = False;
1122         struct smbcli_state *cli;
1123         struct cli_credentials *anon_creds;
1124         struct cli_credentials *wks_creds;
1125         const char *wks_name;
1126         int i;
1127
1128         wks_name = lp_parm_string(-1, "torture", "wksname");
1129         if (wks_name == NULL) {
1130                 wks_name = get_myname();
1131         }
1132
1133         mem_ctx = talloc_init("torture_bind_authcontext");
1134
1135         if (mem_ctx == NULL) {
1136                 d_printf("talloc_init failed\n");
1137                 return False;
1138         }
1139
1140         if (!(anon_creds = create_anon_creds(mem_ctx))) {
1141                 d_printf("create_anon_creds failed\n");
1142                 goto done;
1143         }
1144
1145         status = smbcli_full_connection(mem_ctx, &cli,
1146                                         lp_parm_string(-1, "torture", "host"),
1147                                         "IPC$", NULL, anon_creds, NULL);
1148         if (!NT_STATUS_IS_OK(status)) {
1149                 d_printf("smbcli_full_connection failed: %s\n",
1150                          nt_errstr(status));
1151                 goto done;
1152         }
1153
1154         wks_creds = cli_credentials_init(mem_ctx);
1155         if (wks_creds == NULL) {
1156                 d_printf("cli_credentials_init failed\n");
1157                 goto done;
1158         }
1159
1160         cli_credentials_set_conf(wks_creds);
1161         cli_credentials_set_secure_channel_type(wks_creds, SEC_CHAN_WKSTA);
1162         cli_credentials_set_username(wks_creds, wks_name, CRED_SPECIFIED);
1163         cli_credentials_set_workstation(wks_creds, wks_name, CRED_SPECIFIED);
1164         cli_credentials_set_password(wks_creds,
1165                                      generate_random_str(wks_creds, 8),
1166                                      CRED_SPECIFIED);
1167
1168         if (!join3(cli, False, cmdline_credentials, wks_creds)) {
1169                 d_printf("join failed\n");
1170                 goto done;
1171         }
1172
1173         cli_credentials_set_domain(
1174                 cmdline_credentials, cli_credentials_get_domain(wks_creds),
1175                 CRED_SPECIFIED);
1176
1177         for (i=0; i<2; i++) {
1178
1179                 /* Do this more than once, the routine "schan" changes
1180                  * the workstation password using the netlogon
1181                  * password change routine */
1182
1183                 int j;
1184
1185                 if (!auth2(cli, wks_creds)) {
1186                         d_printf("auth2 failed\n");
1187                         goto done;
1188                 }
1189
1190                 for (j=0; j<2; j++) {
1191                         if (!schan(cli, wks_creds, cmdline_credentials)) {
1192                                 d_printf("schan failed\n");
1193                                 goto done;
1194                         }
1195                 }
1196         }
1197
1198         if (!leave(cli, cmdline_credentials, wks_creds)) {
1199                 d_printf("leave failed\n");
1200                 goto done;
1201         }
1202
1203         ret = True;
1204
1205  done:
1206         talloc_free(mem_ctx);
1207         return ret;
1208 }
1209
1210 /*
1211  * Do a simple join, testjoin and leave using specified smb and samr
1212  * credentials
1213  */
1214
1215 static BOOL test_join3(TALLOC_CTX *mem_ctx,
1216                        BOOL use_level25,
1217                        struct cli_credentials *smb_creds,
1218                        struct cli_credentials *samr_creds,
1219                        const char *wks_name)
1220 {
1221         NTSTATUS status;
1222         BOOL ret = False;
1223         struct smbcli_state *cli;
1224         struct cli_credentials *wks_creds;
1225
1226         status = smbcli_full_connection(mem_ctx, &cli,
1227                                         lp_parm_string(-1, "torture", "host"),
1228                                         "IPC$", NULL, smb_creds, NULL);
1229         if (!NT_STATUS_IS_OK(status)) {
1230                 d_printf("smbcli_full_connection failed: %s\n",
1231                          nt_errstr(status));
1232                 goto done;
1233         }
1234
1235         wks_creds = cli_credentials_init(cli);
1236         if (wks_creds == NULL) {
1237                 d_printf("cli_credentials_init failed\n");
1238                 goto done;
1239         }
1240
1241         cli_credentials_set_conf(wks_creds);
1242         cli_credentials_set_secure_channel_type(wks_creds, SEC_CHAN_WKSTA);
1243         cli_credentials_set_username(wks_creds, wks_name, CRED_SPECIFIED);
1244         cli_credentials_set_workstation(wks_creds, wks_name, CRED_SPECIFIED);
1245         cli_credentials_set_password(wks_creds,
1246                                      generate_random_str(wks_creds, 8),
1247                                      CRED_SPECIFIED);
1248
1249         if (!join3(cli, use_level25, samr_creds, wks_creds)) {
1250                 d_printf("join failed\n");
1251                 goto done;
1252         }
1253
1254         cli_credentials_set_domain(
1255                 cmdline_credentials, cli_credentials_get_domain(wks_creds),
1256                 CRED_SPECIFIED);
1257
1258         if (!auth2(cli, wks_creds)) {
1259                 d_printf("auth2 failed\n");
1260                 goto done;
1261         }
1262
1263         if (!leave(cli, samr_creds, wks_creds)) {
1264                 d_printf("leave failed\n");
1265                 goto done;
1266         }
1267
1268         talloc_free(cli);
1269
1270         ret = True;
1271
1272  done:
1273         return ret;
1274 }
1275
1276 /*
1277  * Test the different session key variants. Do it by joining, this uses the
1278  * session key in the setpassword routine. Test the join by doing the auth2.
1279  */
1280
1281 BOOL torture_samba3_sessionkey(struct torture_context *torture)
1282 {
1283         TALLOC_CTX *mem_ctx;
1284         BOOL ret = False;
1285         struct cli_credentials *anon_creds;
1286         const char *wks_name;
1287
1288         wks_name = lp_parm_string(-1, "torture", "wksname");
1289         if (wks_name == NULL) {
1290                 wks_name = get_myname();
1291         }
1292
1293         mem_ctx = talloc_init("torture_samba3_sessionkey");
1294
1295         if (mem_ctx == NULL) {
1296                 d_printf("talloc_init failed\n");
1297                 return False;
1298         }
1299
1300         if (!(anon_creds = create_anon_creds(mem_ctx))) {
1301                 d_printf("create_anon_creds failed\n");
1302                 goto done;
1303         }
1304
1305         ret = True;
1306
1307         if (!lp_parm_bool(-1, "target", "samba3", False)) {
1308
1309                 /* Samba3 in the build farm right now does this happily. Need
1310                  * to fix :-) */
1311
1312                 if (test_join3(mem_ctx, False, anon_creds, NULL, wks_name)) {
1313                         d_printf("join using anonymous bind on an anonymous smb "
1314                                  "connection succeeded -- HUH??\n");
1315                         ret = False;
1316                 }
1317         }
1318
1319         if (!test_join3(mem_ctx, False, anon_creds, cmdline_credentials,
1320                         wks_name)) {
1321                 d_printf("join using ntlmssp bind on an anonymous smb "
1322                          "connection failed\n");
1323                 ret = False;
1324         }
1325
1326         if (!test_join3(mem_ctx, False, cmdline_credentials, NULL, wks_name)) {
1327                 d_printf("join using anonymous bind on an authenticated smb "
1328                          "connection failed\n");
1329                 ret = False;
1330         }
1331
1332         if (!test_join3(mem_ctx, False, cmdline_credentials,
1333                         cmdline_credentials,
1334                         wks_name)) {
1335                 d_printf("join using ntlmssp bind on an authenticated smb "
1336                          "connection failed\n");
1337                 ret = False;
1338         }
1339
1340         /*
1341          * The following two are tests for setuserinfolevel 25
1342          */
1343
1344         if (!test_join3(mem_ctx, True, anon_creds, cmdline_credentials,
1345                         wks_name)) {
1346                 d_printf("join using ntlmssp bind on an anonymous smb "
1347                          "connection failed\n");
1348                 ret = False;
1349         }
1350
1351         if (!test_join3(mem_ctx, True, cmdline_credentials, NULL, wks_name)) {
1352                 d_printf("join using anonymous bind on an authenticated smb "
1353                          "connection failed\n");
1354                 ret = False;
1355         }
1356
1357  done:
1358
1359         return ret;
1360 }
1361
1362 /*
1363  * open pipe and bind, given an IPC$ context
1364  */
1365
1366 static NTSTATUS pipe_bind_smb(TALLOC_CTX *mem_ctx,
1367                               struct smbcli_tree *tree,
1368                               const char *pipe_name,
1369                               const struct dcerpc_interface_table *iface,
1370                               struct dcerpc_pipe **p)
1371 {
1372         struct dcerpc_pipe *result;
1373         NTSTATUS status;
1374
1375         if (!(result = dcerpc_pipe_init(
1376                       mem_ctx, tree->session->transport->socket->event.ctx))) {
1377                 return NT_STATUS_NO_MEMORY;
1378         }
1379
1380         status = dcerpc_pipe_open_smb(result->conn, tree, pipe_name);
1381         if (!NT_STATUS_IS_OK(status)) {
1382                 d_printf("dcerpc_pipe_open_smb failed: %s\n",
1383                          nt_errstr(status));
1384                 talloc_free(result);
1385                 return status;
1386         }
1387
1388         status = dcerpc_bind_auth_none(result, iface);
1389         if (!NT_STATUS_IS_OK(status)) {
1390                 d_printf("schannel bind failed: %s\n", nt_errstr(status));
1391                 talloc_free(result);
1392                 return status;
1393         }
1394
1395         *p = result;
1396         return NT_STATUS_OK;
1397 }
1398
1399 /*
1400  * Sane wrapper around lsa_LookupNames
1401  */
1402
1403 static struct dom_sid *name2sid(TALLOC_CTX *mem_ctx,
1404                                 struct dcerpc_pipe *p,
1405                                 const char *name,
1406                                 const char *domain)
1407 {
1408         struct lsa_ObjectAttribute attr;
1409         struct lsa_QosInfo qos;
1410         struct lsa_OpenPolicy2 r;
1411         struct lsa_Close c;
1412         NTSTATUS status;
1413         struct policy_handle handle;
1414         struct lsa_LookupNames l;
1415         struct lsa_TransSidArray sids;
1416         struct lsa_String lsa_name;
1417         uint32_t count = 0;
1418         struct dom_sid *result;
1419         TALLOC_CTX *tmp_ctx;
1420
1421         if (!(tmp_ctx = talloc_new(mem_ctx))) {
1422                 return NULL;
1423         }
1424
1425         qos.len = 0;
1426         qos.impersonation_level = 2;
1427         qos.context_mode = 1;
1428         qos.effective_only = 0;
1429
1430         attr.len = 0;
1431         attr.root_dir = NULL;
1432         attr.object_name = NULL;
1433         attr.attributes = 0;
1434         attr.sec_desc = NULL;
1435         attr.sec_qos = &qos;
1436
1437         r.in.system_name = "\\";
1438         r.in.attr = &attr;
1439         r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1440         r.out.handle = &handle;
1441
1442         status = dcerpc_lsa_OpenPolicy2(p, tmp_ctx, &r);
1443         if (!NT_STATUS_IS_OK(status)) {
1444                 printf("OpenPolicy2 failed - %s\n", nt_errstr(status));
1445                 talloc_free(tmp_ctx);
1446                 return NULL;
1447         }
1448
1449         sids.count = 0;
1450         sids.sids = NULL;
1451
1452         lsa_name.string = talloc_asprintf(tmp_ctx, "%s\\%s", domain, name);
1453
1454         l.in.handle = &handle;
1455         l.in.num_names = 1;
1456         l.in.names = &lsa_name;
1457         l.in.sids = &sids;
1458         l.in.level = 1;
1459         l.in.count = &count;
1460         l.out.count = &count;
1461         l.out.sids = &sids;
1462
1463         status = dcerpc_lsa_LookupNames(p, tmp_ctx, &l);
1464         if (!NT_STATUS_IS_OK(status)) {
1465                 printf("LookupNames failed - %s\n", nt_errstr(status));
1466                 talloc_free(tmp_ctx);
1467                 return NULL;
1468         }
1469
1470         result = dom_sid_add_rid(mem_ctx, l.out.domains->domains[0].sid,
1471                                  l.out.sids->sids[0].rid);
1472
1473         c.in.handle = &handle;
1474         c.out.handle = &handle;
1475
1476         status = dcerpc_lsa_Close(p, tmp_ctx, &c);
1477         if (!NT_STATUS_IS_OK(status)) {
1478                 printf("dcerpc_lsa_Close failed - %s\n", nt_errstr(status));
1479                 talloc_free(tmp_ctx);
1480                 return NULL;
1481         }
1482
1483         talloc_free(tmp_ctx);
1484         return result;
1485 }
1486
1487 /*
1488  * Find out the user SID on this connection
1489  */
1490
1491 static struct dom_sid *whoami(TALLOC_CTX *mem_ctx, struct smbcli_tree *tree)
1492 {
1493         struct dcerpc_pipe *lsa;
1494         struct lsa_GetUserName r;
1495         NTSTATUS status;
1496         struct lsa_StringPointer authority_name_p;
1497         struct dom_sid *result;
1498
1499         status = pipe_bind_smb(mem_ctx, tree, "\\pipe\\lsarpc",
1500                                &dcerpc_table_lsarpc, &lsa);
1501         if (!NT_STATUS_IS_OK(status)) {
1502                 d_printf("(%s) Could not bind to LSA: %s\n",
1503                          __location__, nt_errstr(status));
1504                 return NULL;
1505         }
1506
1507         r.in.system_name = "\\";
1508         r.in.account_name = NULL;
1509         authority_name_p.string = NULL;
1510         r.in.authority_name = &authority_name_p;
1511
1512         status = dcerpc_lsa_GetUserName(lsa, mem_ctx, &r);
1513
1514         if (!NT_STATUS_IS_OK(status)) {
1515                 printf("(%s) GetUserName failed - %s\n",
1516                        __location__, nt_errstr(status));
1517                 talloc_free(lsa);
1518                 return NULL;
1519         }
1520
1521         result = name2sid(mem_ctx, lsa, r.out.account_name->string,
1522                           r.out.authority_name->string->string);
1523
1524         talloc_free(lsa);
1525         return result;
1526 }
1527
1528 /*
1529  * Do a tcon, given a session
1530  */
1531
1532 NTSTATUS secondary_tcon(TALLOC_CTX *mem_ctx,
1533                         struct smbcli_session *session,
1534                         const char *sharename,
1535                         struct smbcli_tree **res)
1536 {
1537         struct smbcli_tree *result;
1538         TALLOC_CTX *tmp_ctx;
1539         union smb_tcon tcon;
1540         NTSTATUS status;
1541
1542         if (!(tmp_ctx = talloc_new(mem_ctx))) {
1543                 return NT_STATUS_NO_MEMORY;
1544         }
1545
1546         if (!(result = smbcli_tree_init(session, mem_ctx, False))) {
1547                 talloc_free(tmp_ctx);
1548                 return NT_STATUS_NO_MEMORY;
1549         }
1550
1551         tcon.generic.level = RAW_TCON_TCONX;
1552         tcon.tconx.in.flags = 0;
1553         tcon.tconx.in.password = data_blob(NULL, 0);
1554         tcon.tconx.in.path = sharename;
1555         tcon.tconx.in.device = "?????";
1556
1557         status = smb_raw_tcon(result, tmp_ctx, &tcon);
1558         if (!NT_STATUS_IS_OK(status)) {
1559                 d_printf("(%s) smb_raw_tcon failed: %s\n", __location__,
1560                          nt_errstr(status));
1561                 talloc_free(tmp_ctx);
1562                 return status;
1563         }
1564
1565         result->tid = tcon.tconx.out.tid;
1566         result = talloc_steal(mem_ctx, result);
1567         talloc_free(tmp_ctx);
1568         *res = result;
1569         return NT_STATUS_OK;
1570 }
1571
1572 /*
1573  * Test the getusername behaviour
1574  */
1575
1576 BOOL torture_samba3_rpc_getusername(struct torture_context *torture)
1577 {
1578         NTSTATUS status;
1579         struct smbcli_state *cli;
1580         TALLOC_CTX *mem_ctx;
1581         BOOL ret = True;
1582         struct dom_sid *user_sid;
1583         struct dom_sid *created_sid;
1584         struct cli_credentials *anon_creds;
1585         struct cli_credentials *user_creds;
1586         char *domain_name;
1587
1588         if (!(mem_ctx = talloc_new(torture))) {
1589                 return False;
1590         }
1591
1592         status = smbcli_full_connection(
1593                 mem_ctx, &cli, lp_parm_string(-1, "torture", "host"),
1594                 "IPC$", NULL, cmdline_credentials, NULL);
1595         if (!NT_STATUS_IS_OK(status)) {
1596                 d_printf("(%s) smbcli_full_connection failed: %s\n",
1597                          __location__, nt_errstr(status));
1598                 ret = False;
1599                 goto done;
1600         }
1601
1602         if (!(user_sid = whoami(mem_ctx, cli->tree))) {
1603                 d_printf("(%s) whoami on auth'ed connection failed\n",
1604                          __location__);
1605                 ret = False;
1606         }
1607
1608         talloc_free(cli);
1609
1610         if (!(anon_creds = create_anon_creds(mem_ctx))) {
1611                 d_printf("(%s) create_anon_creds failed\n", __location__);
1612                 ret = False;
1613                 goto done;
1614         }
1615
1616         status = smbcli_full_connection(
1617                 mem_ctx, &cli, lp_parm_string(-1, "torture", "host"),
1618                 "IPC$", NULL, anon_creds, NULL);
1619         if (!NT_STATUS_IS_OK(status)) {
1620                 d_printf("(%s) anon smbcli_full_connection failed: %s\n",
1621                          __location__, nt_errstr(status));
1622                 ret = False;
1623                 goto done;
1624         }
1625
1626         if (!(user_sid = whoami(mem_ctx, cli->tree))) {
1627                 d_printf("(%s) whoami on anon connection failed\n",
1628                          __location__);
1629                 ret = False;
1630                 goto done;
1631         }
1632
1633         if (!dom_sid_equal(user_sid,
1634                            dom_sid_parse_talloc(mem_ctx, "s-1-5-7"))) {
1635                 d_printf("(%s) Anon lsa_GetUserName returned %s, expected "
1636                          "S-1-5-7", __location__,
1637                          dom_sid_string(mem_ctx, user_sid));
1638                 ret = False;
1639         }
1640
1641         if (!(user_creds = cli_credentials_init(mem_ctx))) {
1642                 d_printf("(%s) cli_credentials_init failed\n", __location__);
1643                 ret = False;
1644                 goto done;
1645         }
1646
1647         cli_credentials_set_conf(user_creds);
1648         cli_credentials_set_username(user_creds, "torture_username",
1649                                      CRED_SPECIFIED);
1650         cli_credentials_set_password(user_creds,
1651                                      generate_random_str(user_creds, 8),
1652                                      CRED_SPECIFIED);
1653
1654         if (!create_user(mem_ctx, cli, cmdline_credentials,
1655                          cli_credentials_get_username(user_creds),
1656                          cli_credentials_get_password(user_creds),
1657                          &domain_name, &created_sid)) {
1658                 d_printf("(%s) create_user failed\n", __location__);
1659                 ret = False;
1660                 goto done;
1661         }
1662
1663         cli_credentials_set_domain(user_creds, domain_name,
1664                                    CRED_SPECIFIED);
1665
1666         {
1667                 struct smbcli_session *session2;
1668                 struct smb_composite_sesssetup setup;
1669                 struct smbcli_tree *tree;
1670
1671                 session2 = smbcli_session_init(cli->transport, mem_ctx, False);
1672                 if (session2 == NULL) {
1673                         d_printf("(%s) smbcli_session_init failed\n",
1674                                  __location__);
1675                         goto done;
1676                 }
1677
1678                 setup.in.sesskey = cli->transport->negotiate.sesskey;
1679                 setup.in.capabilities = cli->transport->negotiate.capabilities;
1680                 setup.in.workgroup = "";
1681                 setup.in.credentials = user_creds;
1682
1683                 status = smb_composite_sesssetup(session2, &setup);
1684                 if (!NT_STATUS_IS_OK(status)) {
1685                         d_printf("(%s) session setup with new user failed: "
1686                                  "%s\n", __location__, nt_errstr(status));
1687                         ret = False;
1688                         goto done;
1689                 }
1690
1691                 if (!NT_STATUS_IS_OK(secondary_tcon(mem_ctx, session2,
1692                                                     "IPC$", &tree))) {
1693                         d_printf("(%s) secondary_tcon failed\n",
1694                                  __location__);
1695                         ret = False;
1696                         goto done;
1697                 }
1698
1699                 if (!(user_sid = whoami(mem_ctx, tree))) {
1700                         d_printf("(%s) whoami on user connection failed\n",
1701                                  __location__);
1702                         ret = False;
1703                         goto delete;
1704                 }
1705
1706                 talloc_free(tree);
1707         }
1708
1709         d_printf("Created %s, found %s\n",
1710                  dom_sid_string(mem_ctx, created_sid),
1711                  dom_sid_string(mem_ctx, user_sid));
1712
1713         if (!dom_sid_equal(created_sid, user_sid)) {
1714                 ret = False;
1715         }
1716
1717  delete:
1718         if (!delete_user(cli, cmdline_credentials,
1719                          cli_credentials_get_username(user_creds))) {
1720                 d_printf("(%s) delete_user failed\n", __location__);
1721                 ret = False;
1722         }
1723
1724  done:
1725         talloc_free(mem_ctx);
1726         return ret;
1727 }
1728
1729 static BOOL test_NetShareGetInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
1730                                  const char *sharename)
1731 {
1732         NTSTATUS status;
1733         struct srvsvc_NetShareGetInfo r;
1734         uint32_t levels[] = { 0, 1, 2, 501, 502, 1004, 1005, 1006, 1007, 1501 };
1735         int i;
1736         BOOL ret = True;
1737
1738         r.in.server_unc = talloc_asprintf(mem_ctx, "\\\\%s",
1739                                           dcerpc_server_name(p));
1740         r.in.share_name = sharename;
1741
1742         for (i=0;i<ARRAY_SIZE(levels);i++) {
1743                 r.in.level = levels[i];
1744
1745                 ZERO_STRUCT(r.out);
1746
1747                 printf("testing NetShareGetInfo level %u on share '%s'\n",
1748                        r.in.level, r.in.share_name);
1749
1750                 status = dcerpc_srvsvc_NetShareGetInfo(p, mem_ctx, &r);
1751                 if (!NT_STATUS_IS_OK(status)) {
1752                         printf("NetShareGetInfo level %u on share '%s' failed"
1753                                " - %s\n", r.in.level, r.in.share_name,
1754                                nt_errstr(status));
1755                         ret = False;
1756                         continue;
1757                 }
1758                 if (!W_ERROR_IS_OK(r.out.result)) {
1759                         printf("NetShareGetInfo level %u on share '%s' failed "
1760                                "- %s\n", r.in.level, r.in.share_name,
1761                                win_errstr(r.out.result));
1762                         ret = False;
1763                         continue;
1764                 }
1765         }
1766
1767         return ret;
1768 }
1769
1770 static BOOL test_NetShareEnum(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
1771                               const char **one_sharename)
1772 {
1773         NTSTATUS status;
1774         struct srvsvc_NetShareEnum r;
1775         struct srvsvc_NetShareCtr0 c0;
1776         uint32_t levels[] = { 0, 1, 2, 501, 502, 1004, 1005, 1006, 1007 };
1777         int i;
1778         BOOL ret = True;
1779
1780         r.in.server_unc = talloc_asprintf(mem_ctx,"\\\\%s",dcerpc_server_name(p));
1781         r.in.ctr.ctr0 = &c0;
1782         r.in.ctr.ctr0->count = 0;
1783         r.in.ctr.ctr0->array = NULL;
1784         r.in.max_buffer = (uint32_t)-1;
1785         r.in.resume_handle = NULL;
1786
1787         for (i=0;i<ARRAY_SIZE(levels);i++) {
1788                 r.in.level = levels[i];
1789
1790                 ZERO_STRUCT(r.out);
1791
1792                 printf("testing NetShareEnum level %u\n", r.in.level);
1793                 status = dcerpc_srvsvc_NetShareEnum(p, mem_ctx, &r);
1794                 if (!NT_STATUS_IS_OK(status)) {
1795                         printf("NetShareEnum level %u failed - %s\n",
1796                                r.in.level, nt_errstr(status));
1797                         ret = False;
1798                         continue;
1799                 }
1800                 if (!W_ERROR_IS_OK(r.out.result)) {
1801                         printf("NetShareEnum level %u failed - %s\n",
1802                                r.in.level, win_errstr(r.out.result));
1803                         continue;
1804                 }
1805                 if (r.in.level == 0) {
1806                         struct srvsvc_NetShareCtr0 *ctr = r.out.ctr.ctr0;
1807                         if (ctr->count > 0) {
1808                                 *one_sharename = ctr->array[0].name;
1809                         }
1810                 }
1811         }
1812
1813         return ret;
1814 }
1815
1816 BOOL torture_samba3_rpc_srvsvc(struct torture_context *torture)
1817 {
1818         struct dcerpc_pipe *p;
1819         TALLOC_CTX *mem_ctx;
1820         BOOL ret = True;
1821         const char *sharename = NULL;
1822         struct smbcli_state *cli;
1823         NTSTATUS status;
1824
1825         if (!(mem_ctx = talloc_new(torture))) {
1826                 return False;
1827         }
1828
1829         if (!(torture_open_connection_share(
1830                       mem_ctx, &cli, lp_parm_string(-1, "torture", "host"),
1831                       "IPC$", NULL))) {
1832                 talloc_free(mem_ctx);
1833                 return False;
1834         }
1835
1836         status = pipe_bind_smb(mem_ctx, cli->tree, "\\pipe\\srvsvc",
1837                                &dcerpc_table_srvsvc, &p);
1838         if (!NT_STATUS_IS_OK(status)) {
1839                 d_printf("(%s) could not bind to srvsvc pipe: %s\n",
1840                          __location__, nt_errstr(status));
1841                 ret = False;
1842                 goto done;
1843         }
1844
1845         ret &= test_NetShareEnum(p, mem_ctx, &sharename);
1846         if (sharename == NULL) {
1847                 printf("did not get sharename\n");
1848         } else {
1849                 ret &= test_NetShareGetInfo(p, mem_ctx, sharename);
1850         }
1851
1852  done:
1853         talloc_free(mem_ctx);
1854         return ret;
1855 }
1856
1857 static struct security_descriptor *get_sharesec(TALLOC_CTX *mem_ctx,
1858                                                 struct smbcli_session *sess,
1859                                                 const char *sharename)
1860 {
1861         struct smbcli_tree *tree;
1862         TALLOC_CTX *tmp_ctx;
1863         struct dcerpc_pipe *p;
1864         NTSTATUS status;
1865         struct srvsvc_NetShareGetInfo r;
1866         struct security_descriptor *result;
1867
1868         if (!(tmp_ctx = talloc_new(mem_ctx))) {
1869                 d_printf("talloc_new failed\n");
1870                 return NULL;
1871         }
1872
1873         if (!NT_STATUS_IS_OK(secondary_tcon(tmp_ctx, sess, "IPC$", &tree))) {
1874                 d_printf("secondary_tcon failed\n");
1875                 talloc_free(tmp_ctx);
1876                 return NULL;
1877         }
1878
1879         status = pipe_bind_smb(mem_ctx, tree, "\\pipe\\srvsvc",
1880                                &dcerpc_table_srvsvc, &p);
1881         if (!NT_STATUS_IS_OK(status)) {
1882                 d_printf("(%s) could not bind to srvsvc pipe: %s\n",
1883                          __location__, nt_errstr(status));
1884                 talloc_free(tmp_ctx);
1885                 return NULL;
1886         }
1887
1888 #if 0
1889         p->conn->flags |= DCERPC_DEBUG_PRINT_IN | DCERPC_DEBUG_PRINT_OUT;
1890 #endif
1891
1892         r.in.server_unc = talloc_asprintf(tmp_ctx, "\\\\%s",
1893                                           dcerpc_server_name(p));
1894         r.in.share_name = sharename;
1895         r.in.level = 502;
1896
1897         status = dcerpc_srvsvc_NetShareGetInfo(p, tmp_ctx, &r);
1898         if (!NT_STATUS_IS_OK(status)) {
1899                 d_printf("srvsvc_NetShareGetInfo failed: %s\n",
1900                          nt_errstr(status));
1901                 talloc_free(tmp_ctx);
1902                 return NULL;
1903         }
1904
1905         result = talloc_steal(mem_ctx, r.out.info.info502->sd);
1906         talloc_free(tmp_ctx);
1907         return result;
1908 }
1909
1910 static NTSTATUS set_sharesec(TALLOC_CTX *mem_ctx,
1911                              struct smbcli_session *sess,
1912                              const char *sharename,
1913                              struct security_descriptor *sd)
1914 {
1915         struct smbcli_tree *tree;
1916         TALLOC_CTX *tmp_ctx;
1917         struct dcerpc_pipe *p;
1918         NTSTATUS status;
1919         struct sec_desc_buf i;
1920         struct srvsvc_NetShareSetInfo r;
1921         uint32_t error = 0;
1922
1923         if (!(tmp_ctx = talloc_new(mem_ctx))) {
1924                 d_printf("talloc_new failed\n");
1925                 return NT_STATUS_NO_MEMORY;
1926         }
1927
1928         if (!NT_STATUS_IS_OK(secondary_tcon(tmp_ctx, sess, "IPC$", &tree))) {
1929                 d_printf("secondary_tcon failed\n");
1930                 talloc_free(tmp_ctx);
1931                 return NT_STATUS_UNSUCCESSFUL;
1932         }
1933
1934         status = pipe_bind_smb(mem_ctx, tree, "\\pipe\\srvsvc",
1935                                &dcerpc_table_srvsvc, &p);
1936         if (!NT_STATUS_IS_OK(status)) {
1937                 d_printf("(%s) could not bind to srvsvc pipe: %s\n",
1938                          __location__, nt_errstr(status));
1939                 talloc_free(tmp_ctx);
1940                 return NT_STATUS_UNSUCCESSFUL;
1941         }
1942
1943 #if 0
1944         p->conn->flags |= DCERPC_DEBUG_PRINT_IN | DCERPC_DEBUG_PRINT_OUT;
1945 #endif
1946
1947         r.in.server_unc = talloc_asprintf(tmp_ctx, "\\\\%s",
1948                                           dcerpc_server_name(p));
1949         r.in.share_name = sharename;
1950         r.in.level = 1501;
1951         i.sd = sd;
1952         r.in.info.info1501 = &i;
1953         r.in.parm_error = &error;
1954
1955         status = dcerpc_srvsvc_NetShareSetInfo(p, tmp_ctx, &r);
1956         if (!NT_STATUS_IS_OK(status)) {
1957                 d_printf("srvsvc_NetShareGetInfo failed: %s\n",
1958                          nt_errstr(status));
1959         }
1960
1961         talloc_free(tmp_ctx);
1962         return status;
1963 }
1964
1965 BOOL try_tcon(TALLOC_CTX *mem_ctx,
1966               struct security_descriptor *orig_sd,
1967               struct smbcli_session *session,
1968               const char *sharename, const struct dom_sid *user_sid,
1969               unsigned int access_mask, NTSTATUS expected_tcon,
1970               NTSTATUS expected_mkdir)
1971 {
1972         TALLOC_CTX *tmp_ctx;
1973         struct smbcli_tree *rmdir_tree, *tree;
1974         struct dom_sid *domain_sid;
1975         uint32_t rid;
1976         struct security_descriptor *sd;
1977         NTSTATUS status;
1978         BOOL ret = True;
1979
1980         if (!(tmp_ctx = talloc_new(mem_ctx))) {
1981                 d_printf("talloc_new failed\n");
1982                 return False;
1983         }
1984
1985         status = secondary_tcon(tmp_ctx, session, sharename, &rmdir_tree);
1986         if (!NT_STATUS_IS_OK(status)) {
1987                 d_printf("first tcon to delete dir failed\n");
1988                 talloc_free(tmp_ctx);
1989                 return False;
1990         }
1991
1992         smbcli_rmdir(rmdir_tree, "sharesec_testdir");
1993
1994         if (!NT_STATUS_IS_OK(dom_sid_split_rid(tmp_ctx, user_sid,
1995                                                &domain_sid, &rid))) {
1996                 d_printf("dom_sid_split_rid failed\n");
1997                 talloc_free(tmp_ctx);
1998                 return False;
1999         }
2000
2001         sd = security_descriptor_create(
2002                 tmp_ctx, "S-1-5-32-544",
2003                 dom_sid_string(mem_ctx, dom_sid_add_rid(mem_ctx, domain_sid,
2004                                                         DOMAIN_RID_USERS)),
2005                 dom_sid_string(mem_ctx, user_sid),
2006                 SEC_ACE_TYPE_ACCESS_ALLOWED, access_mask, 0, NULL);
2007         if (sd == NULL) {
2008                 d_printf("security_descriptor_create failed\n");
2009                 talloc_free(tmp_ctx);
2010                 return False;
2011         }
2012
2013         status = set_sharesec(mem_ctx, session, sharename, sd);
2014         if (!NT_STATUS_IS_OK(status)) {
2015                 d_printf("custom set_sharesec failed: %s\n",
2016                          nt_errstr(status));
2017                 talloc_free(tmp_ctx);
2018                 return False;
2019         }
2020
2021         status = secondary_tcon(tmp_ctx, session, sharename, &tree);
2022         if (!NT_STATUS_EQUAL(status, expected_tcon)) {
2023                 d_printf("Expected %s, got %s\n", nt_errstr(expected_tcon),
2024                          nt_errstr(status));
2025                 ret = False;
2026                 goto done;
2027         }
2028
2029         if (!NT_STATUS_IS_OK(status)) {
2030                 /* An expected non-access, no point in trying to write */
2031                 goto done;
2032         }
2033
2034         status = smbcli_mkdir(tree, "sharesec_testdir");
2035         if (!NT_STATUS_EQUAL(status, expected_mkdir)) {
2036                 d_printf("Expected %s, got %s\n", nt_errstr(expected_mkdir),
2037                          nt_errstr(status));
2038                 ret = False;
2039         }
2040
2041  done:
2042         smbcli_rmdir(rmdir_tree, "sharesec_testdir");
2043
2044         status = set_sharesec(mem_ctx, session, sharename, orig_sd);
2045         if (!NT_STATUS_IS_OK(status)) {
2046                 d_printf("custom set_sharesec failed: %s\n",
2047                          nt_errstr(status));
2048                 talloc_free(tmp_ctx);
2049                 return False;
2050         }
2051
2052         talloc_free(tmp_ctx);
2053         return ret;
2054 }
2055
2056 BOOL torture_samba3_rpc_sharesec(struct torture_context *torture)
2057 {
2058         TALLOC_CTX *mem_ctx;
2059         BOOL ret = True;
2060         struct smbcli_state *cli;
2061         struct security_descriptor *sd;
2062         struct dom_sid *user_sid;
2063
2064         if (!(mem_ctx = talloc_new(torture))) {
2065                 return False;
2066         }
2067
2068         if (!(torture_open_connection_share(
2069                       mem_ctx, &cli, lp_parm_string(-1, "torture", "host"),
2070                       "IPC$", NULL))) {
2071                 d_printf("IPC$ connection failed\n");
2072                 talloc_free(mem_ctx);
2073                 return False;
2074         }
2075
2076         if (!(user_sid = whoami(mem_ctx, cli->tree))) {
2077                 d_printf("whoami failed\n");
2078                 talloc_free(mem_ctx);
2079                 return False;
2080         }
2081
2082         sd = get_sharesec(mem_ctx, cli->session, lp_parm_string(-1, "torture",
2083                                                                 "share"));
2084
2085         ret &= try_tcon(mem_ctx, sd, cli->session,
2086                         lp_parm_string(-1, "torture", "share"),
2087                         user_sid, 0, NT_STATUS_ACCESS_DENIED, NT_STATUS_OK);
2088
2089         ret &= try_tcon(mem_ctx, sd, cli->session,
2090                         lp_parm_string(-1, "torture", "share"),
2091                         user_sid, SEC_FILE_READ_DATA, NT_STATUS_OK,
2092                         NT_STATUS_NETWORK_ACCESS_DENIED);
2093
2094         ret &= try_tcon(mem_ctx, sd, cli->session,
2095                         lp_parm_string(-1, "torture", "share"),
2096                         user_sid, SEC_FILE_ALL, NT_STATUS_OK, NT_STATUS_OK);
2097
2098         talloc_free(mem_ctx);
2099         return ret;
2100 }
2101
2102 BOOL torture_samba3_rpc_lsa(struct torture_context *torture)
2103 {
2104         TALLOC_CTX *mem_ctx;
2105         BOOL ret = True;
2106         struct smbcli_state *cli;
2107         struct dcerpc_pipe *p;
2108         struct policy_handle lsa_handle;
2109         NTSTATUS status;
2110         struct dom_sid *domain_sid;
2111
2112         if (!(mem_ctx = talloc_new(torture))) {
2113                 return False;
2114         }
2115
2116         if (!(torture_open_connection_share(
2117                       mem_ctx, &cli, lp_parm_string(-1, "torture", "host"),
2118                       "IPC$", NULL))) {
2119                 d_printf("IPC$ connection failed\n");
2120                 talloc_free(mem_ctx);
2121                 return False;
2122         }
2123
2124         status = pipe_bind_smb(mem_ctx, cli->tree, "\\lsarpc",
2125                                &dcerpc_table_lsarpc, &p);
2126         if (!NT_STATUS_IS_OK(status)) {
2127                 d_printf("(%s) pipe_bind_smb failed: %s\n", __location__,
2128                          nt_errstr(status));
2129                 talloc_free(mem_ctx);
2130                 return False;
2131         }
2132
2133         {
2134                 struct lsa_ObjectAttribute attr;
2135                 struct lsa_OpenPolicy2 o;
2136                 o.in.system_name = talloc_asprintf(
2137                         mem_ctx, "\\\\%s", dcerpc_server_name(p));
2138                 ZERO_STRUCT(attr);
2139                 o.in.attr = &attr;
2140                 o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2141                 o.out.handle = &lsa_handle;
2142                 status = dcerpc_lsa_OpenPolicy2(p, mem_ctx, &o);
2143                 if (!NT_STATUS_IS_OK(status)) {
2144                         d_printf("(%s) dcerpc_lsa_OpenPolicy2 failed: %s\n",
2145                                  __location__, nt_errstr(status));
2146                         talloc_free(mem_ctx);
2147                         return False;
2148                 }
2149         }
2150
2151 #if 0
2152         p->conn->flags |= DCERPC_DEBUG_PRINT_IN | DCERPC_DEBUG_PRINT_OUT;
2153 #endif
2154
2155         {
2156                 int i;
2157                 int levels[] = { 2,3,5,6 };
2158
2159                 for (i=0; i<ARRAY_SIZE(levels); i++) {
2160                         struct lsa_QueryInfoPolicy r;
2161                         r.in.handle = &lsa_handle;
2162                         r.in.level = levels[i];
2163                         status = dcerpc_lsa_QueryInfoPolicy(p, mem_ctx, &r);
2164                         if (!NT_STATUS_IS_OK(status)) {
2165                                 d_printf("(%s) dcerpc_lsa_QueryInfoPolicy %d "
2166                                          "failed: %s\n", __location__,
2167                                          levels[i], nt_errstr(status));
2168                                 talloc_free(mem_ctx);
2169                                 return False;
2170                         }
2171                         if (levels[i] == 5) {
2172                                 domain_sid = r.out.info->account_domain.sid;
2173                         }
2174                 }
2175         }
2176
2177         return ret;
2178 }
2179
2180 static NTSTATUS get_servername(TALLOC_CTX *mem_ctx, struct smbcli_tree *tree,
2181                                char **name)
2182 {
2183         struct rap_WserverGetInfo r;
2184         NTSTATUS status;
2185         char servername[17];
2186
2187         r.in.level = 0;
2188         r.in.bufsize = 0xffff;
2189
2190         status = smbcli_rap_netservergetinfo(tree, mem_ctx, &r);
2191         if (!NT_STATUS_IS_OK(status)) {
2192                 return status;
2193         }
2194
2195         memcpy(servername, r.out.info.info0.name, 16);
2196         servername[16] = '\0';
2197
2198         if (pull_ascii_talloc(mem_ctx, name, servername) < 0) {
2199                 return NT_STATUS_NO_MEMORY;
2200         }
2201
2202         return NT_STATUS_OK;
2203 }
2204
2205
2206 static NTSTATUS find_printers(TALLOC_CTX *ctx, struct smbcli_tree *tree,
2207                               const char ***printers, int *num_printers)
2208 {
2209         TALLOC_CTX *mem_ctx;
2210         NTSTATUS status;
2211         struct dcerpc_pipe *p;
2212         struct srvsvc_NetShareEnum r;
2213         struct srvsvc_NetShareCtr1 c1_in;
2214         struct srvsvc_NetShareCtr1 *c1;
2215         int i;
2216
2217         mem_ctx = talloc_new(ctx);
2218         if (mem_ctx == NULL) {
2219                 return NT_STATUS_NO_MEMORY;
2220         }
2221
2222         status = pipe_bind_smb(mem_ctx, tree, "\\srvsvc", &dcerpc_table_srvsvc,
2223                                &p);
2224         if (!NT_STATUS_IS_OK(status)) {
2225                 d_printf("could not bind to srvsvc pipe\n");
2226                 talloc_free(mem_ctx);
2227                 return status;
2228         }
2229
2230         r.in.server_unc = talloc_asprintf(
2231                 mem_ctx, "\\\\%s", dcerpc_server_name(p));
2232         r.in.level = 1;
2233         ZERO_STRUCT(c1_in);
2234         r.in.ctr.ctr1 = &c1_in;
2235         r.in.max_buffer = (uint32_t)-1;
2236         r.in.resume_handle = NULL;
2237
2238         status = dcerpc_srvsvc_NetShareEnum(p, mem_ctx, &r);
2239         if (!NT_STATUS_IS_OK(status)) {
2240                 d_printf("NetShareEnum level %u failed - %s\n",
2241                          r.in.level, nt_errstr(status));
2242                 talloc_free(mem_ctx);
2243                 return status;
2244         }
2245
2246         *printers = NULL;
2247         *num_printers = 0;
2248         c1 = r.out.ctr.ctr1;
2249         for (i=0; i<c1->count; i++) {
2250                 if (c1->array[i].type != STYPE_PRINTQ) {
2251                         continue;
2252                 }
2253                 if (!add_string_to_array(ctx, c1->array[i].name,
2254                                          printers, num_printers)) {
2255                         talloc_free(ctx);
2256                         return NT_STATUS_NO_MEMORY;
2257                 }
2258         }
2259
2260         talloc_free(mem_ctx);
2261         return NT_STATUS_OK;
2262 }
2263
2264 static BOOL enumprinters(TALLOC_CTX *mem_ctx, struct dcerpc_pipe *pipe,
2265                          const char *servername, int level, int *num_printers)
2266 {
2267         struct spoolss_EnumPrinters r;
2268         NTSTATUS status;
2269         DATA_BLOB blob;
2270
2271         r.in.flags = PRINTER_ENUM_LOCAL;
2272         r.in.server = talloc_asprintf(mem_ctx, "\\\\%s", servername);
2273         r.in.level = level;
2274         r.in.buffer = NULL;
2275         r.in.offered = 0;
2276
2277         status = dcerpc_spoolss_EnumPrinters(pipe, mem_ctx, &r);
2278         if (!NT_STATUS_IS_OK(status)) {
2279                 d_printf("(%s) dcerpc_spoolss_EnumPrinters failed: %s\n",
2280                          __location__, nt_errstr(status));
2281                 return False;
2282         }
2283
2284         if (!W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
2285                 d_printf("(%s) EnumPrinters unexpected return code %s, should "
2286                          "be WERR_INSUFFICIENT_BUFFER\n", __location__,
2287                          win_errstr(r.out.result));
2288                 return False;
2289         }
2290
2291         blob = data_blob_talloc(mem_ctx, NULL, r.out.needed);
2292         if (blob.data == NULL) {
2293                 d_printf("(%s) data_blob_talloc failed\n", __location__);
2294                 return False;
2295         }
2296
2297         r.in.buffer = &blob;
2298         r.in.offered = r.out.needed;
2299
2300         status = dcerpc_spoolss_EnumPrinters(pipe, mem_ctx, &r);
2301         if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2302                 d_printf("(%s) dcerpc_spoolss_EnumPrinters failed: %s, "
2303                          "%s\n", __location__, nt_errstr(status),
2304                          win_errstr(r.out.result));
2305                 return False;
2306         }
2307
2308         *num_printers = r.out.count;
2309
2310         return True;
2311 }
2312
2313 static NTSTATUS getprinterinfo(TALLOC_CTX *ctx, struct dcerpc_pipe *pipe,
2314                                struct policy_handle *handle, int level,
2315                                union spoolss_PrinterInfo **res)
2316 {
2317         TALLOC_CTX *mem_ctx;
2318         struct spoolss_GetPrinter r;
2319         DATA_BLOB blob;
2320         NTSTATUS status;
2321
2322         mem_ctx = talloc_new(ctx);
2323         if (mem_ctx == NULL) {
2324                 return NT_STATUS_NO_MEMORY;
2325         }
2326
2327         r.in.handle = handle;
2328         r.in.level = level;
2329         r.in.buffer = NULL;
2330         r.in.offered = 0;
2331
2332         status = dcerpc_spoolss_GetPrinter(pipe, mem_ctx, &r);
2333         if (!NT_STATUS_IS_OK(status)) {
2334                 d_printf("(%s) dcerpc_spoolss_GetPrinter failed: %s\n",
2335                          __location__, nt_errstr(status));
2336                 talloc_free(mem_ctx);
2337                 return status;
2338         }
2339
2340         if (!W_ERROR_EQUAL(r.out.result, WERR_INSUFFICIENT_BUFFER)) {
2341                 printf("GetPrinter unexpected return code %s, should "
2342                        "be WERR_INSUFFICIENT_BUFFER\n",
2343                        win_errstr(r.out.result));
2344                 talloc_free(mem_ctx);
2345                 return NT_STATUS_UNSUCCESSFUL;
2346         }
2347
2348         r.in.handle = handle;
2349         r.in.level = level;
2350         blob = data_blob_talloc(mem_ctx, NULL, r.out.needed);
2351         if (blob.data == NULL) {
2352                 talloc_free(mem_ctx);
2353                 return NT_STATUS_NO_MEMORY;
2354         }
2355         memset(blob.data, 0, blob.length);
2356         r.in.buffer = &blob;
2357         r.in.offered = r.out.needed;
2358
2359         status = dcerpc_spoolss_GetPrinter(pipe, mem_ctx, &r);
2360         if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2361                 d_printf("(%s) dcerpc_spoolss_GetPrinter failed: %s, "
2362                          "%s\n", __location__, nt_errstr(status),
2363                          win_errstr(r.out.result));
2364                 talloc_free(mem_ctx);
2365                 return NT_STATUS_IS_OK(status) ?
2366                         NT_STATUS_UNSUCCESSFUL : status;
2367         }
2368
2369         if (res != NULL) {
2370                 *res = talloc_steal(ctx, r.out.info);
2371         }
2372
2373         talloc_free(mem_ctx);
2374         return NT_STATUS_OK;
2375 }
2376
2377 BOOL torture_samba3_rpc_spoolss(struct torture_context *torture)
2378 {
2379         TALLOC_CTX *mem_ctx;
2380         BOOL ret = True;
2381         struct smbcli_state *cli;
2382         struct dcerpc_pipe *p;
2383         NTSTATUS status;
2384         struct policy_handle server_handle, printer_handle;
2385         const char **printers;
2386         int num_printers;
2387         struct spoolss_UserLevel1 userlevel1;
2388         char *servername;
2389
2390         if (!(mem_ctx = talloc_new(torture))) {
2391                 return False;
2392         }
2393
2394         if (!(torture_open_connection_share(
2395                       mem_ctx, &cli, lp_parm_string(-1, "torture", "host"),
2396                       "IPC$", NULL))) {
2397                 d_printf("IPC$ connection failed\n");
2398                 talloc_free(mem_ctx);
2399                 return False;
2400         }
2401
2402         status = get_servername(mem_ctx, cli->tree, &servername);
2403         if (!NT_STATUS_IS_OK(status)) {
2404                 d_fprintf(stderr, "(%s) get_servername returned %s\n",
2405                           __location__, nt_errstr(status));
2406                 talloc_free(mem_ctx);
2407                 return False;
2408         }
2409
2410         if (!NT_STATUS_IS_OK(find_printers(mem_ctx, cli->tree,
2411                                            &printers, &num_printers))) {
2412                 talloc_free(mem_ctx);
2413                 return False;
2414         }
2415
2416         if (num_printers == 0) {
2417                 d_printf("Did not find printers\n");
2418                 talloc_free(mem_ctx);
2419                 return True;
2420         }
2421
2422         status = pipe_bind_smb(mem_ctx, cli->tree, "\\spoolss",
2423                                &dcerpc_table_spoolss, &p);
2424         if (!NT_STATUS_IS_OK(status)) {
2425                 d_printf("(%s) pipe_bind_smb failed: %s\n", __location__,
2426                          nt_errstr(status));
2427                 talloc_free(mem_ctx);
2428                 return False;
2429         }
2430
2431         ZERO_STRUCT(userlevel1);
2432         userlevel1.client = talloc_asprintf(
2433                 mem_ctx, "\\\\%s", lp_netbios_name());
2434         userlevel1.user = cli_credentials_get_username(cmdline_credentials);
2435         userlevel1.build = 2600;
2436         userlevel1.major = 3;
2437         userlevel1.minor = 0;
2438         userlevel1.processor = 0;
2439
2440         {
2441                 struct spoolss_OpenPrinterEx r;
2442
2443                 ZERO_STRUCT(r);
2444                 r.in.printername = talloc_asprintf(mem_ctx, "\\\\%s",
2445                                                    servername);
2446                 r.in.datatype = NULL;
2447                 r.in.access_mask = 0;
2448                 r.in.level = 1;
2449                 r.in.userlevel.level1 = &userlevel1;
2450                 r.out.handle = &server_handle;
2451
2452                 status = dcerpc_spoolss_OpenPrinterEx(p, mem_ctx, &r);
2453                 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2454                         d_printf("(%s) dcerpc_spoolss_OpenPrinterEx failed: "
2455                                  "%s, %s\n", __location__, nt_errstr(status),
2456                                  win_errstr(r.out.result));
2457                         talloc_free(mem_ctx);
2458                         return False;
2459                 }
2460         }
2461
2462         {
2463                 struct spoolss_ClosePrinter r;
2464
2465                 r.in.handle = &server_handle;
2466                 r.out.handle = &server_handle;
2467
2468                 status = dcerpc_spoolss_ClosePrinter(p, mem_ctx, &r);
2469                 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2470                         d_printf("(%s) dcerpc_spoolss_ClosePrinter failed: "
2471                                  "%s, %s\n", __location__, nt_errstr(status),
2472                                  win_errstr(r.out.result));
2473                         talloc_free(mem_ctx);
2474                         return False;
2475                 }
2476         }
2477
2478         {
2479                 struct spoolss_OpenPrinterEx r;
2480
2481                 ZERO_STRUCT(r);
2482                 r.in.printername = talloc_asprintf(
2483                         mem_ctx, "\\\\%s\\%s", servername, printers[0]);
2484                 r.in.datatype = NULL;
2485                 r.in.access_mask = 0;
2486                 r.in.level = 1;
2487                 r.in.userlevel.level1 = &userlevel1;
2488                 r.out.handle = &printer_handle;
2489
2490                 status = dcerpc_spoolss_OpenPrinterEx(p, mem_ctx, &r);
2491                 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2492                         d_printf("(%s) dcerpc_spoolss_OpenPrinterEx failed: "
2493                                  "%s, %s\n", __location__, nt_errstr(status),
2494                                  win_errstr(r.out.result));
2495                         talloc_free(mem_ctx);
2496                         return False;
2497                 }
2498         }
2499
2500         {
2501                 int i;
2502
2503                 for (i=0; i<8; i++) {
2504                         status = getprinterinfo(mem_ctx, p, &printer_handle,
2505                                                 i, NULL);
2506                         if (!NT_STATUS_IS_OK(status)) {
2507                                 d_printf("(%s) getprinterinfo %d failed: %s\n",
2508                                          __location__, i, nt_errstr(status));
2509                                 ret = False;
2510                         }
2511                 }
2512         }
2513
2514         {
2515                 struct spoolss_ClosePrinter r;
2516
2517                 r.in.handle = &printer_handle;
2518                 r.out.handle = &printer_handle;
2519
2520                 status = dcerpc_spoolss_ClosePrinter(p, mem_ctx, &r);
2521                 if (!NT_STATUS_IS_OK(status)) {
2522                         d_printf("(%s) dcerpc_spoolss_ClosePrinter failed: "
2523                                  "%s\n", __location__, nt_errstr(status));
2524                         talloc_free(mem_ctx);
2525                         return False;
2526                 }
2527         }
2528
2529         {
2530                 int num_enumerated;
2531                 if (!enumprinters(mem_ctx, p, servername, 1,
2532                                   &num_enumerated)) {
2533                         d_printf("(%s) enumprinters failed\n", __location__);
2534                         talloc_free(mem_ctx);
2535                         return False;
2536                 }
2537                 if (num_printers != num_enumerated) {
2538                         d_printf("(%s) netshareenum gave %d printers, "
2539                                  "enumprinters lvl 1 gave %d\n", __location__,
2540                                  num_printers, num_enumerated);
2541                         talloc_free(mem_ctx);
2542                         return False;
2543                 }
2544         }
2545
2546         {
2547                 int num_enumerated;
2548                 if (!enumprinters(mem_ctx, p, servername, 2,
2549                                   &num_enumerated)) {
2550                         d_printf("(%s) enumprinters failed\n", __location__);
2551                         talloc_free(mem_ctx);
2552                         return False;
2553                 }
2554                 if (num_printers != num_enumerated) {
2555                         d_printf("(%s) netshareenum gave %d printers, "
2556                                  "enumprinters lvl 2 gave %d\n", __location__,
2557                                  num_printers, num_enumerated);
2558                         talloc_free(mem_ctx);
2559                         return False;
2560                 }
2561         }
2562
2563         talloc_free(mem_ctx);
2564
2565         return ret;
2566 }
2567
2568 BOOL torture_samba3_rpc_wkssvc(struct torture_context *torture)
2569 {
2570         TALLOC_CTX *mem_ctx;
2571         struct smbcli_state *cli;
2572         struct dcerpc_pipe *p;
2573         NTSTATUS status;
2574         char *servername;
2575
2576         if (!(mem_ctx = talloc_new(torture))) {
2577                 return False;
2578         }
2579
2580         if (!(torture_open_connection_share(
2581                       mem_ctx, &cli, lp_parm_string(-1, "torture", "host"),
2582                       "IPC$", NULL))) {
2583                 d_printf("IPC$ connection failed\n");
2584                 talloc_free(mem_ctx);
2585                 return False;
2586         }
2587
2588         status = get_servername(mem_ctx, cli->tree, &servername);
2589         if (!NT_STATUS_IS_OK(status)) {
2590                 d_fprintf(stderr, "(%s) get_servername returned %s\n",
2591                           __location__, nt_errstr(status));
2592                 talloc_free(mem_ctx);
2593                 return False;
2594         }
2595
2596         status = pipe_bind_smb(mem_ctx, cli->tree, "\\wkssvc",
2597                                &dcerpc_table_wkssvc, &p);
2598         if (!NT_STATUS_IS_OK(status)) {
2599                 d_printf("(%s) pipe_bind_smb failed: %s\n", __location__,
2600                          nt_errstr(status));
2601                 talloc_free(mem_ctx);
2602                 return False;
2603         }
2604
2605         {
2606                 struct wkssvc_NetWkstaInfo100 wks100;
2607                 union wkssvc_NetWkstaInfo info;
2608                 struct wkssvc_NetWkstaGetInfo r;
2609
2610                 r.in.server_name = "\\foo";
2611                 r.in.level = 100;
2612                 info.info100 = &wks100;
2613                 r.out.info = &info;
2614
2615                 status = dcerpc_wkssvc_NetWkstaGetInfo(p, mem_ctx, &r);
2616                 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2617                         d_printf("(%s) dcerpc_wkssvc_NetWksGetInfo failed: "
2618                                  "%s, %s\n", __location__, nt_errstr(status),
2619                                  win_errstr(r.out.result));
2620                         talloc_free(mem_ctx);
2621                         return False;
2622                 }
2623
2624                 if (strcmp(servername,
2625                            r.out.info->info100->server_name) != 0) {
2626                         d_printf("(%s) servername inconsistency: RAP=%s, "
2627                                  "dcerpc_wkssvc_NetWksGetInfo=%s",
2628                                  __location__, servername,
2629                                  r.out.info->info100->server_name);
2630                         talloc_free(mem_ctx);
2631                         return False;
2632                 }
2633         }
2634
2635         talloc_free(mem_ctx);
2636         return True;
2637 }
2638
2639 static NTSTATUS winreg_close(struct dcerpc_pipe *p,
2640                              struct policy_handle *handle)
2641 {
2642         struct winreg_CloseKey c;
2643         NTSTATUS status;
2644         TALLOC_CTX *mem_ctx;
2645
2646         c.in.handle = c.out.handle = handle;
2647
2648         if (!(mem_ctx = talloc_new(p))) {
2649                 return NT_STATUS_NO_MEMORY;
2650         }
2651
2652         status = dcerpc_winreg_CloseKey(p, mem_ctx, &c);
2653         talloc_free(mem_ctx);
2654
2655         if (!NT_STATUS_IS_OK(status)) {
2656                 return status;
2657         }
2658
2659         if (!W_ERROR_IS_OK(c.out.result)) {
2660                 return werror_to_ntstatus(c.out.result);
2661         }
2662
2663         return NT_STATUS_OK;
2664 }
2665
2666 static NTSTATUS enumvalues(struct dcerpc_pipe *p, struct policy_handle *handle,
2667                            TALLOC_CTX *mem_ctx)
2668 {
2669         uint32_t enum_index = 0;
2670
2671         while (1) {
2672                 struct winreg_EnumValue r;
2673                 struct winreg_StringBuf name;
2674                 enum winreg_Type type = 0;
2675                 uint8_t buf8[1024];
2676                 NTSTATUS status;
2677                 uint32_t size, length;
2678
2679                 r.in.handle = handle;
2680                 r.in.enum_index = enum_index;
2681                 name.name = "";
2682                 name.size = 1024;
2683                 r.in.name = r.out.name = &name;
2684                 size = 1024;
2685                 length = 5;
2686                 r.in.type = &type;
2687                 r.in.value = buf8;
2688                 r.in.size = &size;
2689                 r.in.length = &length;
2690
2691                 status = dcerpc_winreg_EnumValue(p, mem_ctx, &r);
2692                 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2693                         return NT_STATUS_OK;
2694                 }
2695                 enum_index += 1;
2696         }
2697 }
2698
2699 static NTSTATUS enumkeys(struct dcerpc_pipe *p, struct policy_handle *handle,
2700                          TALLOC_CTX *mem_ctx, int depth)
2701 {
2702         struct winreg_EnumKey r;
2703         struct winreg_StringBuf class, name;
2704         NTSTATUS status;
2705         NTTIME t = 0;
2706
2707         if (depth <= 0) {
2708                 return NT_STATUS_OK;
2709         }
2710
2711         class.name   = "";
2712         class.size   = 1024;
2713
2714         r.in.handle = handle;
2715         r.in.enum_index = 0;
2716         r.in.name = &name;
2717         r.in.keyclass = &class;
2718         r.out.name = &name;
2719         r.in.last_changed_time = &t;
2720
2721         do {
2722                 TALLOC_CTX *tmp_ctx;
2723                 struct winreg_OpenKey o;
2724                 struct policy_handle key_handle;
2725                 int i;
2726
2727                 if (!(tmp_ctx = talloc_new(mem_ctx))) {
2728                         return NT_STATUS_NO_MEMORY;
2729                 }
2730
2731                 name.name = NULL;
2732                 name.size = 1024;
2733
2734                 status = dcerpc_winreg_EnumKey(p, tmp_ctx, &r);
2735                 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2736                         /* We're done enumerating */
2737                         talloc_free(tmp_ctx);
2738                         return NT_STATUS_OK;
2739                 }
2740
2741                 for (i=0; i<10-depth; i++)
2742                         printf(" ");
2743                 printf("%s\n", r.out.name->name);
2744
2745
2746                 o.in.parent_handle = handle;
2747                 o.in.keyname.name = r.out.name->name;
2748                 o.in.unknown = 0;
2749                 o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2750                 o.out.handle = &key_handle;
2751
2752                 status = dcerpc_winreg_OpenKey(p, tmp_ctx, &o);
2753                 if (NT_STATUS_IS_OK(status) && W_ERROR_IS_OK(o.out.result)) {
2754                         enumkeys(p, &key_handle, tmp_ctx, depth-1);
2755                         enumvalues(p, &key_handle, tmp_ctx);
2756                         status = winreg_close(p, &key_handle);
2757                         if (!NT_STATUS_IS_OK(status)) {
2758                                 return status;
2759                         }
2760                 }
2761
2762                 talloc_free(tmp_ctx);
2763
2764                 r.in.enum_index += 1;
2765         } while(True);
2766
2767         return NT_STATUS_OK;
2768 }
2769
2770 typedef NTSTATUS (*winreg_open_fn)(struct dcerpc_pipe *, TALLOC_CTX *, void *);
2771
2772 static BOOL test_Open3(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
2773                        const char *name, winreg_open_fn open_fn)
2774 {
2775         struct policy_handle handle;
2776         struct winreg_OpenHKLM r;
2777         NTSTATUS status;
2778
2779         r.in.system_name = 0;
2780         r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2781         r.out.handle = &handle;
2782
2783         status = open_fn(p, mem_ctx, &r);
2784         if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2785                 d_printf("(%s) %s failed: %s, %s\n", __location__, name,
2786                          nt_errstr(status), win_errstr(r.out.result));
2787                 return False;
2788         }
2789
2790         enumkeys(p, &handle, mem_ctx, 4);
2791
2792         status = winreg_close(p, &handle);
2793         if (!NT_STATUS_IS_OK(status)) {
2794                 d_printf("(%s) dcerpc_CloseKey failed: %s\n",
2795                          __location__, nt_errstr(status));
2796                 return False;
2797         }
2798
2799         return True;
2800 }
2801
2802 BOOL torture_samba3_rpc_winreg(struct torture_context *torture)
2803 {
2804         NTSTATUS status;
2805         struct dcerpc_pipe *p;
2806         TALLOC_CTX *mem_ctx;
2807         BOOL ret = True;
2808         struct {
2809                 const char *name;
2810                 winreg_open_fn fn;
2811         } open_fns[] = {
2812                 {"OpenHKLM", (winreg_open_fn)dcerpc_winreg_OpenHKLM },
2813                 {"OpenHKU",  (winreg_open_fn)dcerpc_winreg_OpenHKU },
2814                 {"OpenHKPD", (winreg_open_fn)dcerpc_winreg_OpenHKPD },
2815                 {"OpenHKPT", (winreg_open_fn)dcerpc_winreg_OpenHKPT },
2816                 {"OpenHKCR", (winreg_open_fn)dcerpc_winreg_OpenHKCR }};
2817         int i;
2818
2819         mem_ctx = talloc_init("torture_rpc_winreg");
2820
2821         status = torture_rpc_connection(mem_ctx, &p, &dcerpc_table_winreg);
2822
2823         if (!NT_STATUS_IS_OK(status)) {
2824                 talloc_free(mem_ctx);
2825                 return False;
2826         }
2827
2828 #if 1
2829         ret = test_Open3(p, mem_ctx, open_fns[0].name, open_fns[0].fn);
2830 #else
2831         for (i = 0; i < ARRAY_SIZE(open_fns); i++) {
2832                 if (!test_Open3(p, mem_ctx, open_fns[i].name, open_fns[i].fn))
2833                         ret = False;
2834         }
2835 #endif
2836
2837         talloc_free(mem_ctx);
2838
2839         return ret;
2840 }