Logo Search packages:      
Sourcecode: samba4 version File versions  Download package

samba3rpc.c

/*
   Unix SMB/CIFS implementation.

   dcerpc torture tests, designed to walk Samba3 code paths

   Copyright (C) Volker Lendecke 2006

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 3 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "includes.h"
#include "libcli/raw/libcliraw.h"
#include "libcli/raw/raw_proto.h"
#include "../librpc/gen_ndr/rap.h"
#include "torture/util.h"
#include "torture/rap/proto.h"
#include "librpc/gen_ndr/ndr_lsa_c.h"
#include "librpc/gen_ndr/ndr_samr_c.h"
#include "librpc/gen_ndr/ndr_netlogon_c.h"
#include "librpc/gen_ndr/ndr_srvsvc_c.h"
#include "librpc/gen_ndr/ndr_spoolss_c.h"
#include "librpc/gen_ndr/ndr_winreg_c.h"
#include "librpc/gen_ndr/ndr_wkssvc_c.h"
#include "lib/cmdline/popt_common.h"
#include "torture/rpc/torture_rpc.h"
#include "libcli/libcli.h"
#include "libcli/smb_composite/smb_composite.h"
#include "libcli/auth/libcli_auth.h"
#include "../lib/crypto/crypto.h"
#include "libcli/security/security.h"
#include "param/param.h"
#include "lib/registry/registry.h"
#include "libcli/resolve/resolve.h"

/*
 * This tests a RPC call using an invalid vuid
 */

bool torture_bind_authcontext(struct torture_context *torture)
{
      TALLOC_CTX *mem_ctx;
      NTSTATUS status;
      bool ret = false;
      struct lsa_ObjectAttribute objectattr;
      struct lsa_OpenPolicy2 openpolicy;
      struct policy_handle handle;
      struct lsa_Close close_handle;
      struct smbcli_session *tmp;
      struct smbcli_session *session2;
      struct smbcli_state *cli;
      struct dcerpc_pipe *lsa_pipe;
      struct dcerpc_binding_handle *lsa_handle;
      struct cli_credentials *anon_creds;
      struct smb_composite_sesssetup setup;
      struct smbcli_options options;
      struct smbcli_session_options session_options;

      mem_ctx = talloc_init("torture_bind_authcontext");

      if (mem_ctx == NULL) {
            torture_comment(torture, "talloc_init failed\n");
            return false;
      }

      lpcfg_smbcli_options(torture->lp_ctx, &options);
      lpcfg_smbcli_session_options(torture->lp_ctx, &session_options);

      status = smbcli_full_connection(mem_ctx, &cli,
                              torture_setting_string(torture, "host", NULL),
                              lpcfg_smb_ports(torture->lp_ctx),
                              "IPC$", NULL,
                              lpcfg_socket_options(torture->lp_ctx),
                              cmdline_credentials,
                              lpcfg_resolve_context(torture->lp_ctx),
                              torture->ev, &options, &session_options,
                              lpcfg_gensec_settings(torture, torture->lp_ctx));
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "smbcli_full_connection failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      lsa_pipe = dcerpc_pipe_init(mem_ctx, cli->transport->socket->event.ctx);
      if (lsa_pipe == NULL) {
            torture_comment(torture, "dcerpc_pipe_init failed\n");
            goto done;
      }
      lsa_handle = lsa_pipe->binding_handle;

      status = dcerpc_pipe_open_smb(lsa_pipe, cli->tree, "\\lsarpc");
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "dcerpc_pipe_open_smb failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      status = dcerpc_bind_auth_none(lsa_pipe, &ndr_table_lsarpc);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "dcerpc_bind_auth_none failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      openpolicy.in.system_name =talloc_asprintf(
            mem_ctx, "\\\\%s", dcerpc_server_name(lsa_pipe));
      ZERO_STRUCT(objectattr);
      openpolicy.in.attr = &objectattr;
      openpolicy.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      openpolicy.out.handle = &handle;

      status = dcerpc_lsa_OpenPolicy2_r(lsa_handle, mem_ctx, &openpolicy);

      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "dcerpc_lsa_OpenPolicy2 failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      if (!NT_STATUS_IS_OK(openpolicy.out.result)) {
            torture_comment(torture, "dcerpc_lsa_OpenPolicy2 failed: %s\n",
                   nt_errstr(openpolicy.out.result));
            goto done;
      }

      close_handle.in.handle = &handle;
      close_handle.out.handle = &handle;

      status = dcerpc_lsa_Close_r(lsa_handle, mem_ctx, &close_handle);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "dcerpc_lsa_Close failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      if (!NT_STATUS_IS_OK(close_handle.out.result)) {
            torture_comment(torture, "dcerpc_lsa_Close failed: %s\n",
                   nt_errstr(close_handle.out.result));
            goto done;
      }

      session2 = smbcli_session_init(cli->transport, mem_ctx, false, session_options);
      if (session2 == NULL) {
            torture_comment(torture, "smbcli_session_init failed\n");
            goto done;
      }

      if (!(anon_creds = cli_credentials_init_anon(mem_ctx))) {
            torture_comment(torture, "create_anon_creds failed\n");
            goto done;
      }

      setup.in.sesskey = cli->transport->negotiate.sesskey;
      setup.in.capabilities = cli->transport->negotiate.capabilities;
      setup.in.workgroup = "";
      setup.in.credentials = anon_creds;
      setup.in.gensec_settings = lpcfg_gensec_settings(torture, torture->lp_ctx);

      status = smb_composite_sesssetup(session2, &setup);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "anon session setup failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      session2->vuid = setup.out.vuid;

      tmp = cli->tree->session;
      cli->tree->session = session2;

      status = dcerpc_lsa_OpenPolicy2_r(lsa_handle, mem_ctx, &openpolicy);

      cli->tree->session = tmp;
      talloc_free(lsa_pipe);
      lsa_pipe = NULL;

      if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_HANDLE)) {
            torture_comment(torture, "dcerpc_lsa_OpenPolicy2 with wrong vuid gave %s, "
                   "expected NT_STATUS_INVALID_HANDLE\n",
                   nt_errstr(status));
            goto done;
      }

      ret = true;
 done:
      talloc_free(mem_ctx);
      return ret;
}

/*
 * Bind to lsa using a specific auth method
 */

static bool bindtest(struct torture_context *tctx,
                 struct smbcli_state *cli,
                 struct cli_credentials *credentials,
                 uint8_t auth_type, uint8_t auth_level)
{
      TALLOC_CTX *mem_ctx;
      bool ret = false;
      NTSTATUS status;

      struct dcerpc_pipe *lsa_pipe;
      struct dcerpc_binding_handle *lsa_handle;
      struct lsa_ObjectAttribute objectattr;
      struct lsa_OpenPolicy2 openpolicy;
      struct lsa_QueryInfoPolicy query;
      union lsa_PolicyInformation *info = NULL;
      struct policy_handle handle;
      struct lsa_Close close_handle;

      if ((mem_ctx = talloc_init("bindtest")) == NULL) {
            torture_comment(tctx, "talloc_init failed\n");
            return false;
      }

      lsa_pipe = dcerpc_pipe_init(mem_ctx,
                            cli->transport->socket->event.ctx); 
      if (lsa_pipe == NULL) {
            torture_comment(tctx, "dcerpc_pipe_init failed\n");
            goto done;
      }
      lsa_handle = lsa_pipe->binding_handle;

      status = dcerpc_pipe_open_smb(lsa_pipe, cli->tree, "\\lsarpc");
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_pipe_open_smb failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      status = dcerpc_bind_auth(lsa_pipe, &ndr_table_lsarpc,
                          credentials, lpcfg_gensec_settings(tctx->lp_ctx, tctx->lp_ctx), auth_type, auth_level,
                          NULL);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_bind_auth failed: %s\n", nt_errstr(status));
            goto done;
      }

      openpolicy.in.system_name =talloc_asprintf(
            mem_ctx, "\\\\%s", dcerpc_server_name(lsa_pipe));
      ZERO_STRUCT(objectattr);
      openpolicy.in.attr = &objectattr;
      openpolicy.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      openpolicy.out.handle = &handle;

      status = dcerpc_lsa_OpenPolicy2_r(lsa_handle, mem_ctx, &openpolicy);

      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_lsa_OpenPolicy2 failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      if (!NT_STATUS_IS_OK(openpolicy.out.result)) {
            torture_comment(tctx, "dcerpc_lsa_OpenPolicy2 failed: %s\n",
                   nt_errstr(openpolicy.out.result));
            goto done;
      }

      query.in.handle = &handle;
      query.in.level = LSA_POLICY_INFO_DOMAIN;
      query.out.info = &info;

      status = dcerpc_lsa_QueryInfoPolicy_r(lsa_handle, mem_ctx, &query);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_lsa_QueryInfoPolicy failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      if (!NT_STATUS_IS_OK(query.out.result)) {
            torture_comment(tctx, "dcerpc_lsa_QueryInfoPolicy failed: %s\n",
                   nt_errstr(query.out.result));
            goto done;
      }

      close_handle.in.handle = &handle;
      close_handle.out.handle = &handle;

      status = dcerpc_lsa_Close_r(lsa_handle, mem_ctx, &close_handle);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_lsa_Close failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      if (!NT_STATUS_IS_OK(close_handle.out.result)) {
            torture_comment(tctx, "dcerpc_lsa_Close failed: %s\n",
                   nt_errstr(close_handle.out.result));
            goto done;
      }


      ret = true;
 done:
      talloc_free(mem_ctx);
      return ret;
}

/*
 * test authenticated RPC binds with the variants Samba3 does support
 */

static bool torture_bind_samba3(struct torture_context *torture)
{
      TALLOC_CTX *mem_ctx;
      NTSTATUS status;
      bool ret = false;
      struct smbcli_state *cli;
      struct smbcli_options options;
      struct smbcli_session_options session_options;

      mem_ctx = talloc_init("torture_bind_authcontext");

      if (mem_ctx == NULL) {
            torture_comment(torture, "talloc_init failed\n");
            return false;
      }

      lpcfg_smbcli_options(torture->lp_ctx, &options);
      lpcfg_smbcli_session_options(torture->lp_ctx, &session_options);

      status = smbcli_full_connection(mem_ctx, &cli,
                              torture_setting_string(torture, "host", NULL),
                              lpcfg_smb_ports(torture->lp_ctx),
                              "IPC$", NULL,
                              lpcfg_socket_options(torture->lp_ctx),
                              cmdline_credentials,
                              lpcfg_resolve_context(torture->lp_ctx),
                              torture->ev, &options, &session_options,
                              lpcfg_gensec_settings(torture, torture->lp_ctx));
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "smbcli_full_connection failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      ret = true;

      ret &= bindtest(torture, cli, cmdline_credentials, DCERPC_AUTH_TYPE_NTLMSSP,
                  DCERPC_AUTH_LEVEL_INTEGRITY);
      ret &= bindtest(torture, cli, cmdline_credentials, DCERPC_AUTH_TYPE_NTLMSSP,
                  DCERPC_AUTH_LEVEL_PRIVACY);
      ret &= bindtest(torture, cli, cmdline_credentials, DCERPC_AUTH_TYPE_SPNEGO,
                  DCERPC_AUTH_LEVEL_INTEGRITY);
      ret &= bindtest(torture, cli, cmdline_credentials, DCERPC_AUTH_TYPE_SPNEGO,
                  DCERPC_AUTH_LEVEL_PRIVACY);

 done:
      talloc_free(mem_ctx);
      return ret;
}

/*
 * Lookup or create a user and return all necessary info
 */

static bool get_usr_handle(struct torture_context *tctx,
                     struct smbcli_state *cli,
                     TALLOC_CTX *mem_ctx,
                     struct cli_credentials *admin_creds,
                     uint8_t auth_type,
                     uint8_t auth_level,
                     const char *username,
                     char **domain,
                     struct dcerpc_pipe **result_pipe,
                     struct policy_handle **result_handle,
                     struct dom_sid **sid_p)
{
      struct dcerpc_pipe *samr_pipe;
      struct dcerpc_binding_handle *samr_handle;
      NTSTATUS status;
      struct policy_handle conn_handle;
      struct policy_handle domain_handle;
      struct policy_handle *user_handle;
      struct samr_Connect2 conn;
      struct samr_EnumDomains enumdom;
      uint32_t resume_handle = 0;
      uint32_t num_entries = 0;
      struct samr_SamArray *sam = NULL;
      struct samr_LookupDomain l;
      struct dom_sid2 *sid = NULL;
      int dom_idx;
      struct lsa_String domain_name;
      struct lsa_String user_name;
      struct samr_OpenDomain o;
      struct samr_CreateUser2 c;
      uint32_t user_rid,access_granted;

      samr_pipe = dcerpc_pipe_init(mem_ctx,
                             cli->transport->socket->event.ctx);
      torture_assert(tctx, samr_pipe, "dcerpc_pipe_init failed");

      samr_handle = samr_pipe->binding_handle;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_pipe_open_smb(samr_pipe, cli->tree, "\\samr"),
            "dcerpc_pipe_open_smb failed");

      if (admin_creds != NULL) {
            torture_assert_ntstatus_ok(tctx,
                  dcerpc_bind_auth(samr_pipe, &ndr_table_samr,
                                admin_creds, lpcfg_gensec_settings(tctx->lp_ctx, tctx->lp_ctx), auth_type, auth_level,
                                NULL),
                  "dcerpc_bind_auth failed");
      } else {
            /* We must have an authenticated SMB connection */
            torture_assert_ntstatus_ok(tctx,
                  dcerpc_bind_auth_none(samr_pipe, &ndr_table_samr),
                  "dcerpc_bind_auth_none failed");
      }

      conn.in.system_name = talloc_asprintf(
            mem_ctx, "\\\\%s", dcerpc_server_name(samr_pipe));
      conn.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      conn.out.connect_handle = &conn_handle;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_samr_Connect2_r(samr_handle, mem_ctx, &conn),
            "samr_Connect2 failed");
      torture_assert_ntstatus_ok(tctx, conn.out.result,
            "samr_Connect2 failed");

      enumdom.in.connect_handle = &conn_handle;
      enumdom.in.resume_handle = &resume_handle;
      enumdom.in.buf_size = (uint32_t)-1;
      enumdom.out.resume_handle = &resume_handle;
      enumdom.out.num_entries = &num_entries;
      enumdom.out.sam = &sam;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_samr_EnumDomains_r(samr_handle, mem_ctx, &enumdom),
            "samr_EnumDomains failed");
      torture_assert_ntstatus_ok(tctx, enumdom.out.result,
            "samr_EnumDomains failed");

      torture_assert_int_equal(tctx, *enumdom.out.num_entries, 2,
            "samr_EnumDomains returned unexpected num_entries");

      dom_idx = strequal(sam->entries[0].name.string,
                     "builtin") ? 1:0;

      l.in.connect_handle = &conn_handle;
      domain_name.string = sam->entries[dom_idx].name.string;
      *domain = talloc_strdup(mem_ctx, domain_name.string);
      l.in.domain_name = &domain_name;
      l.out.sid = &sid;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_samr_LookupDomain_r(samr_handle, mem_ctx, &l),
            "samr_LookupDomain failed");
      torture_assert_ntstatus_ok(tctx, l.out.result,
            "samr_LookupDomain failed");

      o.in.connect_handle = &conn_handle;
      o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      o.in.sid = *l.out.sid;
      o.out.domain_handle = &domain_handle;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_samr_OpenDomain_r(samr_handle, mem_ctx, &o),
            "samr_OpenDomain failed");
      torture_assert_ntstatus_ok(tctx, o.out.result,
            "samr_OpenDomain failed");

      c.in.domain_handle = &domain_handle;
      user_name.string = username;
      c.in.account_name = &user_name;
      c.in.acct_flags = ACB_NORMAL;
      c.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      user_handle = talloc(mem_ctx, struct policy_handle);
      c.out.user_handle = user_handle;
      c.out.access_granted = &access_granted;
      c.out.rid = &user_rid;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_samr_CreateUser2_r(samr_handle, mem_ctx, &c),
            "samr_CreateUser2 failed");

      if (NT_STATUS_EQUAL(c.out.result, NT_STATUS_USER_EXISTS)) {
            struct samr_LookupNames ln;
            struct samr_OpenUser ou;
            struct samr_Ids rids, types;

            ln.in.domain_handle = &domain_handle;
            ln.in.num_names = 1;
            ln.in.names = &user_name;
            ln.out.rids = &rids;
            ln.out.types = &types;

            torture_assert_ntstatus_ok(tctx,
                  dcerpc_samr_LookupNames_r(samr_handle, mem_ctx, &ln),
                  "samr_LookupNames failed");
            torture_assert_ntstatus_ok(tctx, ln.out.result,
                  "samr_LookupNames failed");

            ou.in.domain_handle = &domain_handle;
            ou.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
            user_rid = ou.in.rid = ln.out.rids->ids[0];
            ou.out.user_handle = user_handle;

            torture_assert_ntstatus_ok(tctx,
                  dcerpc_samr_OpenUser_r(samr_handle, mem_ctx, &ou),
                  "samr_OpenUser failed");
            status = ou.out.result;
      } else {
            status = c.out.result;
      }

      torture_assert_ntstatus_ok(tctx, status,
            "samr_CreateUser failed");

      *result_pipe = samr_pipe;
      *result_handle = user_handle;
      if (sid_p != NULL) {
            *sid_p = dom_sid_add_rid(mem_ctx, *l.out.sid, user_rid);
      }
      return true;

}

/*
 * Create a test user
 */

static bool create_user(struct torture_context *tctx,
                  TALLOC_CTX *mem_ctx, struct smbcli_state *cli,
                  struct cli_credentials *admin_creds,
                  const char *username, const char *password,
                  char **domain_name,
                  struct dom_sid **user_sid)
{
      TALLOC_CTX *tmp_ctx;
      NTSTATUS status;
      struct dcerpc_pipe *samr_pipe;
      struct dcerpc_binding_handle *samr_handle;
      struct policy_handle *wks_handle;
      bool ret = false;

      if (!(tmp_ctx = talloc_new(mem_ctx))) {
            torture_comment(tctx, "talloc_init failed\n");
            return false;
      }

      ret = get_usr_handle(tctx, cli, tmp_ctx, admin_creds,
                       DCERPC_AUTH_TYPE_NTLMSSP,
                       DCERPC_AUTH_LEVEL_INTEGRITY,
                       username, domain_name, &samr_pipe, &wks_handle,
                       user_sid);
      if (ret == false) {
            torture_comment(tctx, "get_usr_handle failed\n");
            goto done;
      }
      samr_handle = samr_pipe->binding_handle;

      {
            struct samr_SetUserInfo2 sui2;
            struct samr_SetUserInfo sui;
            struct samr_QueryUserInfo qui;
            union samr_UserInfo u_info;
            union samr_UserInfo *info;
            DATA_BLOB session_key;


            ZERO_STRUCT(u_info);
            encode_pw_buffer(u_info.info23.password.data, password,
                         STR_UNICODE);

            status = dcerpc_fetch_session_key(samr_pipe, &session_key);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "dcerpc_fetch_session_key failed\n");
                  goto done;
            }
            arcfour_crypt_blob(u_info.info23.password.data, 516,
                           &session_key);
            u_info.info23.info.password_expired = 0;
            u_info.info23.info.fields_present = SAMR_FIELD_NT_PASSWORD_PRESENT |
                                        SAMR_FIELD_LM_PASSWORD_PRESENT |
                                        SAMR_FIELD_EXPIRED_FLAG;
            sui2.in.user_handle = wks_handle;
            sui2.in.info = &u_info;
            sui2.in.level = 23;

            status = dcerpc_samr_SetUserInfo2_r(samr_handle, tmp_ctx, &sui2);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "samr_SetUserInfo(23) failed: %s\n",
                         nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(sui2.out.result)) {
                  torture_comment(tctx, "samr_SetUserInfo(23) failed: %s\n",
                         nt_errstr(sui2.out.result));
                  goto done;
            }

            u_info.info16.acct_flags = ACB_NORMAL;
            sui.in.user_handle = wks_handle;
            sui.in.info = &u_info;
            sui.in.level = 16;

            status = dcerpc_samr_SetUserInfo_r(samr_handle, tmp_ctx, &sui);
            if (!NT_STATUS_IS_OK(status) || !NT_STATUS_IS_OK(sui.out.result)) {
                  torture_comment(tctx, "samr_SetUserInfo(16) failed\n");
                  goto done;
            }

            qui.in.user_handle = wks_handle;
            qui.in.level = 21;
            qui.out.info = &info;

            status = dcerpc_samr_QueryUserInfo_r(samr_handle, tmp_ctx, &qui);
            if (!NT_STATUS_IS_OK(status) || !NT_STATUS_IS_OK(qui.out.result)) {
                  torture_comment(tctx, "samr_QueryUserInfo(21) failed\n");
                  goto done;
            }

            info->info21.allow_password_change = 0;
            info->info21.force_password_change = 0;
            info->info21.account_name.string = NULL;
            info->info21.rid = 0;
            info->info21.acct_expiry = 0;
            info->info21.fields_present = 0x81827fa; /* copy usrmgr.exe */

            u_info.info21 = info->info21;
            sui.in.user_handle = wks_handle;
            sui.in.info = &u_info;
            sui.in.level = 21;

            status = dcerpc_samr_SetUserInfo_r(samr_handle, tmp_ctx, &sui);
            if (!NT_STATUS_IS_OK(status) || !NT_STATUS_IS_OK(sui.out.result)) {
                  torture_comment(tctx, "samr_SetUserInfo(21) failed\n");
                  goto done;
            }
      }

      *domain_name= talloc_steal(mem_ctx, *domain_name);
      *user_sid = talloc_steal(mem_ctx, *user_sid);
      ret = true;
 done:
      talloc_free(tmp_ctx);
      return ret;
}

/*
 * Delete a test user
 */

static bool delete_user(struct torture_context *tctx,
                  struct smbcli_state *cli,
                  struct cli_credentials *admin_creds,
                  const char *username)
{
      TALLOC_CTX *mem_ctx;
      NTSTATUS status;
      char *dom_name;
      struct dcerpc_pipe *samr_pipe;
      struct dcerpc_binding_handle *samr_handle;
      struct policy_handle *user_handle;
      bool ret = false;

      if ((mem_ctx = talloc_init("leave")) == NULL) {
            torture_comment(tctx, "talloc_init failed\n");
            return false;
      }

      ret = get_usr_handle(tctx, cli, mem_ctx, admin_creds,
                       DCERPC_AUTH_TYPE_NTLMSSP,
                       DCERPC_AUTH_LEVEL_INTEGRITY,
                       username, &dom_name, &samr_pipe,
                       &user_handle, NULL);
      if (ret == false) {
            torture_comment(tctx, "get_wks_handle failed\n");
            goto done;
      }
      samr_handle = samr_pipe->binding_handle;

      {
            struct samr_DeleteUser d;

            d.in.user_handle = user_handle;
            d.out.user_handle = user_handle;

            status = dcerpc_samr_DeleteUser_r(samr_handle, mem_ctx, &d);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "samr_DeleteUser failed %s\n", nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(d.out.result)) {
                  torture_comment(tctx, "samr_DeleteUser failed %s\n", nt_errstr(d.out.result));
                  goto done;
            }

      }

      ret = true;

 done:
      talloc_free(mem_ctx);
      return ret;
}

/*
 * Do a Samba3-style join
 */

static bool join3(struct torture_context *tctx,
              struct smbcli_state *cli,
              bool use_level25,
              struct cli_credentials *admin_creds,
              struct cli_credentials *wks_creds)
{
      TALLOC_CTX *mem_ctx;
      NTSTATUS status;
      char *dom_name;
      struct dcerpc_pipe *samr_pipe;
      struct dcerpc_binding_handle *samr_handle;
      struct policy_handle *wks_handle;
      bool ret = false;
      NTTIME last_password_change;

      if ((mem_ctx = talloc_init("join3")) == NULL) {
            torture_comment(tctx, "talloc_init failed\n");
            return false;
      }

      ret = get_usr_handle(
            tctx, cli, mem_ctx, admin_creds,
            DCERPC_AUTH_TYPE_NTLMSSP,
            DCERPC_AUTH_LEVEL_PRIVACY,
            talloc_asprintf(mem_ctx, "%s$",
                        cli_credentials_get_workstation(wks_creds)),
            &dom_name, &samr_pipe, &wks_handle, NULL);
      if (ret == false) {
            torture_comment(tctx, "get_wks_handle failed\n");
            goto done;
      }
      samr_handle = samr_pipe->binding_handle;

      {
            struct samr_QueryUserInfo q;
            union samr_UserInfo *info;

            q.in.user_handle = wks_handle;
            q.in.level = 21;
            q.out.info = &info;

            status = dcerpc_samr_QueryUserInfo_r(samr_handle, mem_ctx, &q);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_warning(tctx, "QueryUserInfo failed: %s\n",
                          nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(q.out.result)) {
                  torture_warning(tctx, "QueryUserInfo failed: %s\n",
                          nt_errstr(q.out.result));
                  goto done;
            }


            last_password_change = info->info21.last_password_change;
      }

      cli_credentials_set_domain(wks_creds, dom_name, CRED_SPECIFIED);

      if (use_level25) {
            struct samr_SetUserInfo2 sui2;
            union samr_UserInfo u_info;
            struct samr_UserInfo21 *i21 = &u_info.info25.info;
            DATA_BLOB session_key;
            DATA_BLOB confounded_session_key = data_blob_talloc(
                  mem_ctx, NULL, 16);
            struct MD5Context ctx;
            uint8_t confounder[16];

            ZERO_STRUCT(u_info);

            i21->full_name.string = talloc_asprintf(
                  mem_ctx, "%s$",
                  cli_credentials_get_workstation(wks_creds));
            i21->acct_flags = ACB_WSTRUST;
            i21->fields_present = SAMR_FIELD_FULL_NAME |
                  SAMR_FIELD_ACCT_FLAGS | SAMR_FIELD_NT_PASSWORD_PRESENT;
            /* this would break the test result expectations
            i21->fields_present |= SAMR_FIELD_EXPIRED_FLAG;
            i21->password_expired = 1;
            */

            encode_pw_buffer(u_info.info25.password.data,
                         cli_credentials_get_password(wks_creds),
                         STR_UNICODE);
            status = dcerpc_fetch_session_key(samr_pipe, &session_key);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "dcerpc_fetch_session_key failed: %s\n",
                         nt_errstr(status));
                  goto done;
            }
            generate_random_buffer((uint8_t *)confounder, 16);

            MD5Init(&ctx);
            MD5Update(&ctx, confounder, 16);
            MD5Update(&ctx, session_key.data, session_key.length);
            MD5Final(confounded_session_key.data, &ctx);

            arcfour_crypt_blob(u_info.info25.password.data, 516,
                           &confounded_session_key);
            memcpy(&u_info.info25.password.data[516], confounder, 16);

            sui2.in.user_handle = wks_handle;
            sui2.in.level = 25;
            sui2.in.info = &u_info;

            status = dcerpc_samr_SetUserInfo2_r(samr_handle, mem_ctx, &sui2);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "samr_SetUserInfo2(25) failed: %s\n",
                         nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(sui2.out.result)) {
                  torture_comment(tctx, "samr_SetUserInfo2(25) failed: %s\n",
                         nt_errstr(sui2.out.result));
                  goto done;
            }
      } else {
            struct samr_SetUserInfo2 sui2;
            struct samr_SetUserInfo sui;
            union samr_UserInfo u_info;
            DATA_BLOB session_key;

            encode_pw_buffer(u_info.info24.password.data,
                         cli_credentials_get_password(wks_creds),
                         STR_UNICODE);
            /* just to make this test pass */
            u_info.info24.password_expired = 1;

            status = dcerpc_fetch_session_key(samr_pipe, &session_key);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "dcerpc_fetch_session_key failed\n");
                  goto done;
            }
            arcfour_crypt_blob(u_info.info24.password.data, 516,
                           &session_key);
            sui2.in.user_handle = wks_handle;
            sui2.in.info = &u_info;
            sui2.in.level = 24;

            status = dcerpc_samr_SetUserInfo2_r(samr_handle, mem_ctx, &sui2);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "samr_SetUserInfo(24) failed: %s\n",
                         nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(sui2.out.result)) {
                  torture_comment(tctx, "samr_SetUserInfo(24) failed: %s\n",
                         nt_errstr(sui2.out.result));
                  goto done;
            }

            u_info.info16.acct_flags = ACB_WSTRUST;
            sui.in.user_handle = wks_handle;
            sui.in.info = &u_info;
            sui.in.level = 16;

            status = dcerpc_samr_SetUserInfo_r(samr_handle, mem_ctx, &sui);
            if (!NT_STATUS_IS_OK(status) || !NT_STATUS_IS_OK(sui.out.result)) {
                  torture_comment(tctx, "samr_SetUserInfo(16) failed\n");
                  goto done;
            }
      }

      {
            struct samr_QueryUserInfo q;
            union samr_UserInfo *info;

            q.in.user_handle = wks_handle;
            q.in.level = 21;
            q.out.info = &info;

            status = dcerpc_samr_QueryUserInfo_r(samr_handle, mem_ctx, &q);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_warning(tctx, "QueryUserInfo failed: %s\n",
                          nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(q.out.result)) {
                  torture_warning(tctx, "QueryUserInfo failed: %s\n",
                          nt_errstr(q.out.result));
                  goto done;
            }

            if (use_level25) {
                  if (last_password_change
                      == info->info21.last_password_change) {
                        torture_warning(tctx, "last_password_change unchanged "
                               "during join, level25 must change "
                               "it\n");
                        goto done;
                  }
            }
            else {
                  if (last_password_change
                      != info->info21.last_password_change) {
                        torture_warning(tctx, "last_password_change changed "
                               "during join, level24 doesn't "
                               "change it\n");
                        goto done;
                  }
            }
      }

      ret = true;

 done:
      talloc_free(mem_ctx);
      return ret;
}

/*
 * Do a ReqChallenge/Auth2 and get the wks creds
 */

static bool auth2(struct torture_context *tctx,
              struct smbcli_state *cli,
              struct cli_credentials *wks_cred)
{
      TALLOC_CTX *mem_ctx;
      struct dcerpc_pipe *net_pipe;
      struct dcerpc_binding_handle *net_handle;
      bool result = false;
      NTSTATUS status;
      struct netr_ServerReqChallenge r;
      struct netr_Credential netr_cli_creds;
      struct netr_Credential netr_srv_creds;
      uint32_t negotiate_flags;
      struct netr_ServerAuthenticate2 a;
      struct netlogon_creds_CredentialState *creds_state;
      struct netr_Credential netr_cred;
      struct samr_Password mach_pw;

      mem_ctx = talloc_new(NULL);
      if (mem_ctx == NULL) {
            torture_comment(tctx, "talloc_new failed\n");
            return false;
      }

      net_pipe = dcerpc_pipe_init(mem_ctx,
                            cli->transport->socket->event.ctx);
      if (net_pipe == NULL) {
            torture_comment(tctx, "dcerpc_pipe_init failed\n");
            goto done;
      }
      net_handle = net_pipe->binding_handle;

      status = dcerpc_pipe_open_smb(net_pipe, cli->tree, "\\netlogon");
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_pipe_open_smb failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      status = dcerpc_bind_auth_none(net_pipe, &ndr_table_netlogon);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_bind_auth_none failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      r.in.computer_name = cli_credentials_get_workstation(wks_cred);
      r.in.server_name = talloc_asprintf(
            mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
      if (r.in.server_name == NULL) {
            torture_comment(tctx, "talloc_asprintf failed\n");
            goto done;
      }
      generate_random_buffer(netr_cli_creds.data,
                         sizeof(netr_cli_creds.data));
      r.in.credentials = &netr_cli_creds;
      r.out.return_credentials = &netr_srv_creds;

      status = dcerpc_netr_ServerReqChallenge_r(net_handle, mem_ctx, &r);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "netr_ServerReqChallenge failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      if (!NT_STATUS_IS_OK(r.out.result)) {
            torture_comment(tctx, "netr_ServerReqChallenge failed: %s\n",
                   nt_errstr(r.out.result));
            goto done;
      }

      negotiate_flags = NETLOGON_NEG_AUTH2_FLAGS;
      E_md4hash(cli_credentials_get_password(wks_cred), mach_pw.hash);

      a.in.server_name = talloc_asprintf(
            mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
      a.in.account_name = talloc_asprintf(
            mem_ctx, "%s$", cli_credentials_get_workstation(wks_cred));
      a.in.computer_name = cli_credentials_get_workstation(wks_cred);
      a.in.secure_channel_type = SEC_CHAN_WKSTA;
      a.in.negotiate_flags = &negotiate_flags;
      a.out.negotiate_flags = &negotiate_flags;
      a.in.credentials = &netr_cred;
      a.out.return_credentials = &netr_cred;

      creds_state = netlogon_creds_client_init(mem_ctx,
                                     a.in.account_name,
                                     a.in.computer_name,
                                     r.in.credentials,
                                     r.out.return_credentials, &mach_pw,
                                     &netr_cred, negotiate_flags);

      status = dcerpc_netr_ServerAuthenticate2_r(net_handle, mem_ctx, &a);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "netr_ServerServerAuthenticate2 failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      if (!NT_STATUS_IS_OK(a.out.result)) {
            torture_comment(tctx, "netr_ServerServerAuthenticate2 failed: %s\n",
                   nt_errstr(a.out.result));
            goto done;
      }

      if (!netlogon_creds_client_check(creds_state, a.out.return_credentials)) {
            torture_comment(tctx, "creds_client_check failed\n");
            goto done;
      }

      cli_credentials_set_netlogon_creds(wks_cred, creds_state);

      result = true;

 done:
      talloc_free(mem_ctx);
      return result;
}

/*
 * Do a couple of schannel protected Netlogon ops: Interactive and Network
 * login, and change the wks password
 */

static bool schan(struct torture_context *tctx,
              struct smbcli_state *cli,
              struct cli_credentials *wks_creds,
              struct cli_credentials *user_creds)
{
      TALLOC_CTX *mem_ctx;
      NTSTATUS status;
      bool ret = false;
      struct dcerpc_pipe *net_pipe;
      struct dcerpc_binding_handle *net_handle;
      int i;

      mem_ctx = talloc_new(NULL);
      if (mem_ctx == NULL) {
            torture_comment(tctx, "talloc_new failed\n");
            return false;
      }

      net_pipe = dcerpc_pipe_init(mem_ctx,
                            cli->transport->socket->event.ctx);
      if (net_pipe == NULL) {
            torture_comment(tctx, "dcerpc_pipe_init failed\n");
            goto done;
      }
      net_handle = net_pipe->binding_handle;

      status = dcerpc_pipe_open_smb(net_pipe, cli->tree, "\\netlogon");
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_pipe_open_smb failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

#if 0
      net_pipe->conn->flags |= DCERPC_DEBUG_PRINT_IN |
            DCERPC_DEBUG_PRINT_OUT;
#endif
#if 1
      net_pipe->conn->flags |= (DCERPC_SIGN | DCERPC_SEAL);
      status = dcerpc_bind_auth(net_pipe, &ndr_table_netlogon,
                          wks_creds, lpcfg_gensec_settings(tctx->lp_ctx, tctx->lp_ctx), DCERPC_AUTH_TYPE_SCHANNEL,
                          DCERPC_AUTH_LEVEL_PRIVACY,
                          NULL);
#else
      status = dcerpc_bind_auth_none(net_pipe, &ndr_table_netlogon);
#endif
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "schannel bind failed: %s\n", nt_errstr(status));
            goto done;
      }


      for (i=2; i<4; i++) {
            int flags;
            DATA_BLOB chal, nt_resp, lm_resp, names_blob, session_key;
            struct netlogon_creds_CredentialState *creds_state;
            struct netr_Authenticator netr_auth, netr_auth2;
            struct netr_NetworkInfo ninfo;
            struct netr_PasswordInfo pinfo;
            struct netr_LogonSamLogon r;
            union netr_LogonLevel logon;
            union netr_Validation validation;
            uint8_t authoritative;
            struct netr_Authenticator return_authenticator;

            flags = CLI_CRED_LANMAN_AUTH | CLI_CRED_NTLM_AUTH |
                  CLI_CRED_NTLMv2_AUTH;

            chal = data_blob_talloc(mem_ctx, NULL, 8);
            if (chal.data == NULL) {
                  torture_comment(tctx, "data_blob_talloc failed\n");
                  goto done;
            }

            generate_random_buffer(chal.data, chal.length);
            names_blob = NTLMv2_generate_names_blob(
                  mem_ctx,
                  cli_credentials_get_workstation(user_creds),
                  cli_credentials_get_domain(user_creds));
            status = cli_credentials_get_ntlm_response(
                  user_creds, mem_ctx, &flags, chal, names_blob,
                  &lm_resp, &nt_resp, NULL, NULL);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "cli_credentials_get_ntlm_response failed:"
                         " %s\n", nt_errstr(status));
                  goto done;
            }

            creds_state = cli_credentials_get_netlogon_creds(wks_creds);
            netlogon_creds_client_authenticator(creds_state, &netr_auth);

            ninfo.identity_info.account_name.string =
                  cli_credentials_get_username(user_creds);
            ninfo.identity_info.domain_name.string =
                  cli_credentials_get_domain(user_creds);
            ninfo.identity_info.parameter_control = 0;
            ninfo.identity_info.logon_id_low = 0;
            ninfo.identity_info.logon_id_high = 0;
            ninfo.identity_info.workstation.string =
                  cli_credentials_get_workstation(user_creds);
            memcpy(ninfo.challenge, chal.data, sizeof(ninfo.challenge));
            ninfo.nt.length = nt_resp.length;
            ninfo.nt.data = nt_resp.data;
            ninfo.lm.length = lm_resp.length;
            ninfo.lm.data = lm_resp.data;

            logon.network = &ninfo;

            r.in.server_name = talloc_asprintf(
                  mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
            ZERO_STRUCT(netr_auth2);
            r.in.computer_name =
                  cli_credentials_get_workstation(wks_creds);
            r.in.credential = &netr_auth;
            r.in.return_authenticator = &netr_auth2;
            r.in.logon_level = 2;
            r.in.validation_level = i;
            r.in.logon = &logon;
            r.out.validation = &validation;
            r.out.authoritative = &authoritative;
            r.out.return_authenticator = &return_authenticator;

            status = dcerpc_netr_LogonSamLogon_r(net_handle, mem_ctx, &r);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "netr_LogonSamLogon failed: %s\n",
                         nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(r.out.result)) {
                  torture_comment(tctx, "netr_LogonSamLogon failed: %s\n",
                         nt_errstr(r.out.result));
                  goto done;
            }

            if ((r.out.return_authenticator == NULL) ||
                (!netlogon_creds_client_check(creds_state,
                               &r.out.return_authenticator->cred))) {
                  torture_comment(tctx, "Credentials check failed!\n");
                  goto done;
            }

            netlogon_creds_client_authenticator(creds_state, &netr_auth);

            pinfo.identity_info = ninfo.identity_info;
            ZERO_STRUCT(pinfo.lmpassword.hash);
            E_md4hash(cli_credentials_get_password(user_creds),
                    pinfo.ntpassword.hash);
            session_key = data_blob_talloc(mem_ctx,
                                     creds_state->session_key, 16);
            arcfour_crypt_blob(pinfo.ntpassword.hash,
                           sizeof(pinfo.ntpassword.hash),
                           &session_key);

            logon.password = &pinfo;

            r.in.logon_level = 1;
            r.in.logon = &logon;
            r.out.return_authenticator = &return_authenticator;

            status = dcerpc_netr_LogonSamLogon_r(net_handle, mem_ctx, &r);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "netr_LogonSamLogon failed: %s\n",
                         nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(r.out.result)) {
                  torture_comment(tctx, "netr_LogonSamLogon failed: %s\n",
                         nt_errstr(r.out.result));
                  goto done;
            }

            if ((r.out.return_authenticator == NULL) ||
                (!netlogon_creds_client_check(creds_state,
                               &r.out.return_authenticator->cred))) {
                  torture_comment(tctx, "Credentials check failed!\n");
                  goto done;
            }
      }

      {
            struct netr_ServerPasswordSet s;
            char *password = generate_random_password(wks_creds, 8, 255);
            struct netlogon_creds_CredentialState *creds_state;
            struct netr_Authenticator credential, return_authenticator;
            struct samr_Password new_password;

            s.in.server_name = talloc_asprintf(
                  mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
            s.in.computer_name = cli_credentials_get_workstation(wks_creds);
            s.in.account_name = talloc_asprintf(
                  mem_ctx, "%s$", s.in.computer_name);
            s.in.secure_channel_type = SEC_CHAN_WKSTA;
            s.in.credential = &credential;
            s.in.new_password = &new_password;
            s.out.return_authenticator = &return_authenticator;

            E_md4hash(password, new_password.hash);

            creds_state = cli_credentials_get_netlogon_creds(wks_creds);
            netlogon_creds_des_encrypt(creds_state, &new_password);
            netlogon_creds_client_authenticator(creds_state, &credential);

            status = dcerpc_netr_ServerPasswordSet_r(net_handle, mem_ctx, &s);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_comment(tctx, "ServerPasswordSet - %s\n", nt_errstr(status));
                  goto done;
            }
            if (!NT_STATUS_IS_OK(s.out.result)) {
                  torture_comment(tctx, "ServerPasswordSet - %s\n", nt_errstr(s.out.result));
                  goto done;
            }

            if (!netlogon_creds_client_check(creds_state,
                                     &s.out.return_authenticator->cred)) {
                  torture_comment(tctx, "Credential chaining failed\n");
            }

            cli_credentials_set_password(wks_creds, password,
                                   CRED_SPECIFIED);
      }

      ret = true;
 done:
      talloc_free(mem_ctx);
      return ret;
}

/*
 * Delete the wks account again
 */

static bool leave(struct torture_context *tctx,
              struct smbcli_state *cli,
              struct cli_credentials *admin_creds,
              struct cli_credentials *wks_creds)
{
      char *wks_name = talloc_asprintf(
            NULL, "%s$", cli_credentials_get_workstation(wks_creds));
      bool ret;

      ret = delete_user(tctx, cli, admin_creds, wks_name);
      talloc_free(wks_name);
      return ret;
}

/*
 * Test the Samba3 DC code a bit. Join, do some schan netlogon ops, leave
 */

static bool torture_netlogon_samba3(struct torture_context *torture)
{
      TALLOC_CTX *mem_ctx;
      NTSTATUS status;
      bool ret = false;
      struct smbcli_state *cli;
      struct cli_credentials *anon_creds;
      struct cli_credentials *wks_creds;
      const char *wks_name;
      int i;
      struct smbcli_options options;
      struct smbcli_session_options session_options;

      wks_name = torture_setting_string(torture, "wksname", NULL);
      if (wks_name == NULL) {
            wks_name = get_myname(torture);
      }

      mem_ctx = talloc_init("torture_netlogon_samba3");

      if (mem_ctx == NULL) {
            torture_comment(torture, "talloc_init failed\n");
            return false;
      }

      if (!(anon_creds = cli_credentials_init_anon(mem_ctx))) {
            torture_comment(torture, "create_anon_creds failed\n");
            goto done;
      }

      lpcfg_smbcli_options(torture->lp_ctx, &options);
      lpcfg_smbcli_session_options(torture->lp_ctx, &session_options);

      status = smbcli_full_connection(mem_ctx, &cli,
                              torture_setting_string(torture, "host", NULL),
                              lpcfg_smb_ports(torture->lp_ctx),
                              "IPC$", NULL,
                              lpcfg_socket_options(torture->lp_ctx),
                              anon_creds,
                              lpcfg_resolve_context(torture->lp_ctx),
                              torture->ev, &options, &session_options,
                              lpcfg_gensec_settings(torture, torture->lp_ctx));
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "smbcli_full_connection failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      wks_creds = cli_credentials_init(mem_ctx);
      if (wks_creds == NULL) {
            torture_comment(torture, "cli_credentials_init failed\n");
            goto done;
      }

      cli_credentials_set_conf(wks_creds, torture->lp_ctx);
      cli_credentials_set_secure_channel_type(wks_creds, SEC_CHAN_WKSTA);
      cli_credentials_set_username(wks_creds, wks_name, CRED_SPECIFIED);
      cli_credentials_set_workstation(wks_creds, wks_name, CRED_SPECIFIED);
      cli_credentials_set_password(wks_creds,
                             generate_random_password(wks_creds, 8, 255),
                             CRED_SPECIFIED);

      if (!join3(torture, cli, false, cmdline_credentials, wks_creds)) {
            torture_comment(torture, "join failed\n");
            goto done;
      }

      cli_credentials_set_domain(
            cmdline_credentials, cli_credentials_get_domain(wks_creds),
            CRED_SPECIFIED);

      for (i=0; i<2; i++) {

            /* Do this more than once, the routine "schan" changes
             * the workstation password using the netlogon
             * password change routine */

            int j;

            if (!auth2(torture, cli, wks_creds)) {
                  torture_comment(torture, "auth2 failed\n");
                  goto done;
            }

            for (j=0; j<2; j++) {
                  if (!schan(torture, cli, wks_creds, cmdline_credentials)) {
                        torture_comment(torture, "schan failed\n");
                        goto done;
                  }
            }
      }

      if (!leave(torture, cli, cmdline_credentials, wks_creds)) {
            torture_comment(torture, "leave failed\n");
            goto done;
      }

      ret = true;

 done:
      talloc_free(mem_ctx);
      return ret;
}

/*
 * Do a simple join, testjoin and leave using specified smb and samr
 * credentials
 */

static bool test_join3(struct torture_context *tctx,
                   bool use_level25,
                   struct cli_credentials *smb_creds,
                   struct cli_credentials *samr_creds,
                   const char *wks_name)
{
      NTSTATUS status;
      struct smbcli_state *cli;
      struct cli_credentials *wks_creds;
      struct smbcli_options options;
      struct smbcli_session_options session_options;

      lpcfg_smbcli_options(tctx->lp_ctx, &options);
      lpcfg_smbcli_session_options(tctx->lp_ctx, &session_options);

      status = smbcli_full_connection(tctx, &cli,
                              torture_setting_string(tctx, "host", NULL),
                              lpcfg_smb_ports(tctx->lp_ctx),
                              "IPC$", NULL, lpcfg_socket_options(tctx->lp_ctx),
                              smb_creds, lpcfg_resolve_context(tctx->lp_ctx),
                              tctx->ev, &options, &session_options,
                              lpcfg_gensec_settings(tctx, tctx->lp_ctx));
      torture_assert_ntstatus_ok(tctx, status,
            "smbcli_full_connection failed");

      wks_creds = cli_credentials_init(cli);
      torture_assert(tctx, wks_creds, "cli_credentials_init failed");

      cli_credentials_set_conf(wks_creds, tctx->lp_ctx);
      cli_credentials_set_secure_channel_type(wks_creds, SEC_CHAN_WKSTA);
      cli_credentials_set_username(wks_creds, wks_name, CRED_SPECIFIED);
      cli_credentials_set_workstation(wks_creds, wks_name, CRED_SPECIFIED);
      cli_credentials_set_password(wks_creds,
                             generate_random_password(wks_creds, 8, 255),
                             CRED_SPECIFIED);

      torture_assert(tctx,
            join3(tctx, cli, use_level25, samr_creds, wks_creds),
            "join failed");

      cli_credentials_set_domain(
            cmdline_credentials, cli_credentials_get_domain(wks_creds),
            CRED_SPECIFIED);

      torture_assert(tctx,
            auth2(tctx, cli, wks_creds),
            "auth2 failed");

      torture_assert(tctx,
            leave(tctx, cli, samr_creds, wks_creds),
            "leave failed");

      talloc_free(cli);

      return true;
}

/*
 * Test the different session key variants. Do it by joining, this uses the
 * session key in the setpassword routine. Test the join by doing the auth2.
 */

static bool torture_samba3_sessionkey(struct torture_context *torture)
{
      struct cli_credentials *anon_creds;
      const char *wks_name;

      wks_name = torture_setting_string(torture, "wksname", get_myname(torture));

      if (!(anon_creds = cli_credentials_init_anon(torture))) {
            torture_fail(torture, "create_anon_creds failed\n");
      }

      cli_credentials_set_workstation(anon_creds, wks_name, CRED_SPECIFIED);


      if (!torture_setting_bool(torture, "samba3", false)) {

            /* Samba3 in the build farm right now does this happily. Need
             * to fix :-) */

            if (test_join3(torture, false, anon_creds, NULL, wks_name)) {
                  torture_fail(torture, "join using anonymous bind on an anonymous smb "
                         "connection succeeded -- HUH??\n");
            }
      }

      torture_assert(torture,
            test_join3(torture, false, anon_creds, cmdline_credentials, wks_name),
            "join using ntlmssp bind on an anonymous smb connection failed");

      torture_assert(torture,
            test_join3(torture, false, cmdline_credentials, NULL, wks_name),
            "join using anonymous bind on an authenticated smb connection failed");

      torture_assert(torture,
            test_join3(torture, false, cmdline_credentials, cmdline_credentials, wks_name),
            "join using ntlmssp bind on an authenticated smb connection failed");

      /*
       * The following two are tests for setuserinfolevel 25
       */

      torture_assert(torture,
            test_join3(torture, true, anon_creds, cmdline_credentials, wks_name),
            "join using ntlmssp bind on an anonymous smb connection failed");

      torture_assert(torture,
            test_join3(torture, true, cmdline_credentials, NULL, wks_name),
            "join using anonymous bind on an authenticated smb connection failed");

      return true;
}

/*
 * open pipe and bind, given an IPC$ context
 */

static NTSTATUS pipe_bind_smb(struct torture_context *tctx,
                        TALLOC_CTX *mem_ctx,
                        struct smbcli_tree *tree,
                        const char *pipe_name,
                        const struct ndr_interface_table *iface,
                        struct dcerpc_pipe **p)
{
      struct dcerpc_pipe *result;
      NTSTATUS status;

      if (!(result = dcerpc_pipe_init(
                  mem_ctx, tree->session->transport->socket->event.ctx))) {
            return NT_STATUS_NO_MEMORY;
      }

      status = dcerpc_pipe_open_smb(result, tree, pipe_name);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_pipe_open_smb failed: %s\n",
                   nt_errstr(status));
            talloc_free(result);
            return status;
      }

      status = dcerpc_bind_auth_none(result, iface);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "schannel bind failed: %s\n", nt_errstr(status));
            talloc_free(result);
            return status;
      }

      *p = result;
      return NT_STATUS_OK;
}

/*
 * Sane wrapper around lsa_LookupNames
 */

static struct dom_sid *name2sid(struct torture_context *tctx,
                        TALLOC_CTX *mem_ctx,
                        struct dcerpc_pipe *p,
                        const char *name,
                        const char *domain)
{
      struct lsa_ObjectAttribute attr;
      struct lsa_QosInfo qos;
      struct lsa_OpenPolicy2 r;
      struct lsa_Close c;
      NTSTATUS status;
      struct policy_handle handle;
      struct lsa_LookupNames l;
      struct lsa_TransSidArray sids;
      struct lsa_RefDomainList *domains = NULL;
      struct lsa_String lsa_name;
      uint32_t count = 0;
      struct dom_sid *result;
      TALLOC_CTX *tmp_ctx;
      struct dcerpc_binding_handle *b = p->binding_handle;

      if (!(tmp_ctx = talloc_new(mem_ctx))) {
            return NULL;
      }

      qos.len = 0;
      qos.impersonation_level = 2;
      qos.context_mode = 1;
      qos.effective_only = 0;

      attr.len = 0;
      attr.root_dir = NULL;
      attr.object_name = NULL;
      attr.attributes = 0;
      attr.sec_desc = NULL;
      attr.sec_qos = &qos;

      r.in.system_name = "\\";
      r.in.attr = &attr;
      r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      r.out.handle = &handle;

      status = dcerpc_lsa_OpenPolicy2_r(b, tmp_ctx, &r);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "OpenPolicy2 failed - %s\n", nt_errstr(status));
            talloc_free(tmp_ctx);
            return NULL;
      }
      if (!NT_STATUS_IS_OK(r.out.result)) {
            torture_comment(tctx, "OpenPolicy2 failed - %s\n", nt_errstr(r.out.result));
            talloc_free(tmp_ctx);
            return NULL;
      }

      sids.count = 0;
      sids.sids = NULL;

      lsa_name.string = talloc_asprintf(tmp_ctx, "%s\\%s", domain, name);

      l.in.handle = &handle;
      l.in.num_names = 1;
      l.in.names = &lsa_name;
      l.in.sids = &sids;
      l.in.level = 1;
      l.in.count = &count;
      l.out.count = &count;
      l.out.sids = &sids;
      l.out.domains = &domains;

      status = dcerpc_lsa_LookupNames_r(b, tmp_ctx, &l);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "LookupNames of %s failed - %s\n", lsa_name.string,
                   nt_errstr(status));
            talloc_free(tmp_ctx);
            return NULL;
      }
      if (!NT_STATUS_IS_OK(l.out.result)) {
            torture_comment(tctx, "LookupNames of %s failed - %s\n", lsa_name.string,
                   nt_errstr(l.out.result));
            talloc_free(tmp_ctx);
            return NULL;
      }

      result = dom_sid_add_rid(mem_ctx, domains->domains[0].sid,
                         l.out.sids->sids[0].rid);

      c.in.handle = &handle;
      c.out.handle = &handle;

      status = dcerpc_lsa_Close_r(b, tmp_ctx, &c);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "dcerpc_lsa_Close failed - %s\n", nt_errstr(status));
            talloc_free(tmp_ctx);
            return NULL;
      }
      if (!NT_STATUS_IS_OK(c.out.result)) {
            torture_comment(tctx, "dcerpc_lsa_Close failed - %s\n", nt_errstr(c.out.result));
            talloc_free(tmp_ctx);
            return NULL;
      }

      talloc_free(tmp_ctx);
      return result;
}

/*
 * Find out the user SID on this connection
 */

static struct dom_sid *whoami(struct torture_context *tctx,
                        TALLOC_CTX *mem_ctx,
                        struct smbcli_tree *tree)
{
      struct dcerpc_pipe *lsa;
      struct dcerpc_binding_handle *lsa_handle;
      struct lsa_GetUserName r;
      NTSTATUS status;
      struct lsa_String *authority_name_p = NULL;
      struct lsa_String *account_name_p = NULL;
      struct dom_sid *result;

      status = pipe_bind_smb(tctx, mem_ctx, tree, "\\pipe\\lsarpc",
                         &ndr_table_lsarpc, &lsa);
      if (!NT_STATUS_IS_OK(status)) {
            torture_warning(tctx, "Could not bind to LSA: %s\n",
                   nt_errstr(status));
            return NULL;
      }
      lsa_handle = lsa->binding_handle;

      r.in.system_name = "\\";
      r.in.account_name = &account_name_p;
      r.in.authority_name = &authority_name_p;
      r.out.account_name = &account_name_p;

      status = dcerpc_lsa_GetUserName_r(lsa_handle, mem_ctx, &r);

      authority_name_p = *r.out.authority_name;

      if (!NT_STATUS_IS_OK(status)) {
            torture_warning(tctx, "GetUserName failed - %s\n",
                   nt_errstr(status));
            talloc_free(lsa);
            return NULL;
      }
      if (!NT_STATUS_IS_OK(r.out.result)) {
            torture_warning(tctx, "GetUserName failed - %s\n",
                   nt_errstr(r.out.result));
            talloc_free(lsa);
            return NULL;
      }

      result = name2sid(tctx, mem_ctx, lsa, account_name_p->string,
                    authority_name_p->string);

      talloc_free(lsa);
      return result;
}

static int destroy_tree(struct smbcli_tree *tree)
{
      smb_tree_disconnect(tree);
      return 0;
}

/*
 * Do a tcon, given a session
 */

static NTSTATUS secondary_tcon(struct torture_context *tctx,
                         TALLOC_CTX *mem_ctx,
                         struct smbcli_session *session,
                         const char *sharename,
                         struct smbcli_tree **res)
{
      struct smbcli_tree *result;
      TALLOC_CTX *tmp_ctx;
      union smb_tcon tcon;
      NTSTATUS status;

      if (!(tmp_ctx = talloc_new(mem_ctx))) {
            return NT_STATUS_NO_MEMORY;
      }

      if (!(result = smbcli_tree_init(session, mem_ctx, false))) {
            talloc_free(tmp_ctx);
            return NT_STATUS_NO_MEMORY;
      }

      tcon.generic.level = RAW_TCON_TCONX;
      tcon.tconx.in.flags = 0;
      tcon.tconx.in.password = data_blob(NULL, 0);
      tcon.tconx.in.path = sharename;
      tcon.tconx.in.device = "?????";

      status = smb_raw_tcon(result, tmp_ctx, &tcon);
      if (!NT_STATUS_IS_OK(status)) {
            torture_warning(tctx, "smb_raw_tcon failed: %s\n",
                   nt_errstr(status));
            talloc_free(tmp_ctx);
            return status;
      }

      result->tid = tcon.tconx.out.tid;
      result = talloc_steal(mem_ctx, result);
      talloc_set_destructor(result, destroy_tree);
      talloc_free(tmp_ctx);
      *res = result;
      return NT_STATUS_OK;
}

/*
 * Test the getusername behaviour
 */

static bool torture_samba3_rpc_getusername(struct torture_context *torture)
{
      NTSTATUS status;
      struct smbcli_state *cli;
      TALLOC_CTX *mem_ctx;
      bool ret = true;
      struct dom_sid *user_sid;
      struct dom_sid *created_sid;
      struct cli_credentials *anon_creds;
      struct cli_credentials *user_creds;
      char *domain_name;
      struct smbcli_options options;
      struct smbcli_session_options session_options;

      if (!(mem_ctx = talloc_new(torture))) {
            return false;
      }

      lpcfg_smbcli_options(torture->lp_ctx, &options);
      lpcfg_smbcli_session_options(torture->lp_ctx, &session_options);

      status = smbcli_full_connection(
            mem_ctx, &cli, torture_setting_string(torture, "host", NULL),
            lpcfg_smb_ports(torture->lp_ctx),
            "IPC$", NULL, lpcfg_socket_options(torture->lp_ctx), cmdline_credentials,
            lpcfg_resolve_context(torture->lp_ctx), torture->ev, &options,
            &session_options, lpcfg_gensec_settings(torture, torture->lp_ctx));
      if (!NT_STATUS_IS_OK(status)) {
            torture_warning(torture, "smbcli_full_connection failed: %s\n",
                   nt_errstr(status));
            ret = false;
            goto done;
      }

      if (!(user_sid = whoami(torture, mem_ctx, cli->tree))) {
            torture_warning(torture, "whoami on auth'ed connection failed\n");
            ret = false;
      }

      talloc_free(cli);

      if (!(anon_creds = cli_credentials_init_anon(mem_ctx))) {
            torture_warning(torture, "create_anon_creds failed\n");
            ret = false;
            goto done;
      }

      status = smbcli_full_connection(
            mem_ctx, &cli, torture_setting_string(torture, "host", NULL),
            lpcfg_smb_ports(torture->lp_ctx), "IPC$", NULL,
            lpcfg_socket_options(torture->lp_ctx), anon_creds,
            lpcfg_resolve_context(torture->lp_ctx),
            torture->ev, &options, &session_options,
            lpcfg_gensec_settings(torture, torture->lp_ctx));
      if (!NT_STATUS_IS_OK(status)) {
            torture_warning(torture, "anon smbcli_full_connection failed: %s\n",
                   nt_errstr(status));
            ret = false;
            goto done;
      }

      if (!(user_sid = whoami(torture, mem_ctx, cli->tree))) {
            torture_warning(torture, "whoami on anon connection failed\n");
            ret = false;
            goto done;
      }

      if (!dom_sid_equal(user_sid,
                     dom_sid_parse_talloc(mem_ctx, "s-1-5-7"))) {
            torture_warning(torture, "Anon lsa_GetUserName returned %s, expected "
                   "S-1-5-7",
                   dom_sid_string(mem_ctx, user_sid));
            ret = false;
      }

      if (!(user_creds = cli_credentials_init(mem_ctx))) {
            torture_warning(torture, "cli_credentials_init failed\n");
            ret = false;
            goto done;
      }

      cli_credentials_set_conf(user_creds, torture->lp_ctx);
      cli_credentials_set_username(user_creds, "torture_username",
                             CRED_SPECIFIED);
      cli_credentials_set_password(user_creds,
                             generate_random_password(user_creds, 8, 255),
                             CRED_SPECIFIED);

      if (!create_user(torture, mem_ctx, cli, cmdline_credentials,
                   cli_credentials_get_username(user_creds),
                   cli_credentials_get_password(user_creds),
                   &domain_name, &created_sid)) {
            torture_warning(torture, "create_user failed\n");
            ret = false;
            goto done;
      }

      cli_credentials_set_domain(user_creds, domain_name,
                           CRED_SPECIFIED);

      {
            struct smbcli_session *session2;
            struct smb_composite_sesssetup setup;
            struct smbcli_tree *tree;

            session2 = smbcli_session_init(cli->transport, mem_ctx, false, session_options);
            if (session2 == NULL) {
                  torture_warning(torture, "smbcli_session_init failed\n");
                  goto done;
            }

            setup.in.sesskey = cli->transport->negotiate.sesskey;
            setup.in.capabilities = cli->transport->negotiate.capabilities;
            setup.in.workgroup = "";
            setup.in.credentials = user_creds;
            setup.in.gensec_settings = lpcfg_gensec_settings(torture, torture->lp_ctx);

            status = smb_composite_sesssetup(session2, &setup);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_warning(torture, "session setup with new user failed: "
                         "%s\n", nt_errstr(status));
                  ret = false;
                  goto done;
            }
            session2->vuid = setup.out.vuid;

            if (!NT_STATUS_IS_OK(secondary_tcon(torture, mem_ctx, session2,
                                        "IPC$", &tree))) {
                  torture_warning(torture, "secondary_tcon failed\n");
                  ret = false;
                  goto done;
            }

            if (!(user_sid = whoami(torture, mem_ctx, tree))) {
                  torture_warning(torture, "whoami on user connection failed\n");
                  ret = false;
                  goto del;
            }

            talloc_free(tree);
      }

      torture_comment(torture, "Created %s, found %s\n",
             dom_sid_string(mem_ctx, created_sid),
             dom_sid_string(mem_ctx, user_sid));

      if (!dom_sid_equal(created_sid, user_sid)) {
            ret = false;
      }

 del:
      if (!delete_user(torture, cli,
                   cmdline_credentials,
                   cli_credentials_get_username(user_creds))) {
            torture_warning(torture, "delete_user failed\n");
            ret = false;
      }

 done:
      talloc_free(mem_ctx);
      return ret;
}

static bool test_NetShareGetInfo(struct torture_context *tctx,
                         struct dcerpc_pipe *p,
                         const char *sharename)
{
      NTSTATUS status;
      struct srvsvc_NetShareGetInfo r;
      union srvsvc_NetShareInfo info;
      uint32_t levels[] = { 0, 1, 2, 501, 502, 1004, 1005, 1006, 1007, 1501 };
      int i;
      bool ret = true;
      struct dcerpc_binding_handle *b = p->binding_handle;

      r.in.server_unc = talloc_asprintf(tctx, "\\\\%s",
                                dcerpc_server_name(p));
      r.in.share_name = sharename;
      r.out.info = &info;

      for (i=0;i<ARRAY_SIZE(levels);i++) {
            r.in.level = levels[i];

            torture_comment(tctx, "Testing NetShareGetInfo level %u on share '%s'\n",
                   r.in.level, r.in.share_name);

            status = dcerpc_srvsvc_NetShareGetInfo_r(b, tctx, &r);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_warning(tctx, "NetShareGetInfo level %u on share '%s' failed"
                         " - %s\n", r.in.level, r.in.share_name,
                         nt_errstr(status));
                  ret = false;
                  continue;
            }
            if (!W_ERROR_IS_OK(r.out.result)) {
                  torture_warning(tctx, "NetShareGetInfo level %u on share '%s' failed "
                         "- %s\n", r.in.level, r.in.share_name,
                         win_errstr(r.out.result));
                  ret = false;
                  continue;
            }
      }

      return ret;
}

static bool test_NetShareEnum(struct torture_context *tctx,
                        struct dcerpc_pipe *p,
                        const char **one_sharename)
{
      NTSTATUS status;
      struct srvsvc_NetShareEnum r;
      struct srvsvc_NetShareInfoCtr info_ctr;
      struct srvsvc_NetShareCtr0 c0;
      struct srvsvc_NetShareCtr1 c1;
      struct srvsvc_NetShareCtr2 c2;
      struct srvsvc_NetShareCtr501 c501;
      struct srvsvc_NetShareCtr502 c502;
      struct srvsvc_NetShareCtr1004 c1004;
      struct srvsvc_NetShareCtr1005 c1005;
      struct srvsvc_NetShareCtr1006 c1006;
      struct srvsvc_NetShareCtr1007 c1007;
      uint32_t totalentries = 0;
      uint32_t levels[] = { 0, 1, 2, 501, 502, 1004, 1005, 1006, 1007 };
      int i;
      bool ret = true;
      struct dcerpc_binding_handle *b = p->binding_handle;

      ZERO_STRUCT(info_ctr);

      r.in.server_unc = talloc_asprintf(tctx,"\\\\%s",dcerpc_server_name(p));
      r.in.info_ctr = &info_ctr;
      r.in.max_buffer = (uint32_t)-1;
      r.in.resume_handle = NULL;
      r.out.totalentries = &totalentries;
      r.out.info_ctr = &info_ctr;

      for (i=0;i<ARRAY_SIZE(levels);i++) {
            info_ctr.level = levels[i];

            switch (info_ctr.level) {
            case 0:
                  ZERO_STRUCT(c0);
                  info_ctr.ctr.ctr0 = &c0;
                  break;
            case 1:
                  ZERO_STRUCT(c1);
                  info_ctr.ctr.ctr1 = &c1;
                  break;
            case 2:
                  ZERO_STRUCT(c2);
                  info_ctr.ctr.ctr2 = &c2;
                  break;
            case 501:
                  ZERO_STRUCT(c501);
                  info_ctr.ctr.ctr501 = &c501;
                  break;
            case 502:
                  ZERO_STRUCT(c502);
                  info_ctr.ctr.ctr502 = &c502;
                  break;
            case 1004:
                  ZERO_STRUCT(c1004);
                  info_ctr.ctr.ctr1004 = &c1004;
                  break;
            case 1005:
                  ZERO_STRUCT(c1005);
                  info_ctr.ctr.ctr1005 = &c1005;
                  break;
            case 1006:
                  ZERO_STRUCT(c1006);
                  info_ctr.ctr.ctr1006 = &c1006;
                  break;
            case 1007:
                  ZERO_STRUCT(c1007);
                  info_ctr.ctr.ctr1007 = &c1007;
                  break;
            }

            torture_comment(tctx, "Testing NetShareEnum level %u\n", info_ctr.level);

            status = dcerpc_srvsvc_NetShareEnum_r(b, tctx, &r);
            if (!NT_STATUS_IS_OK(status)) {
                  torture_warning(tctx, "NetShareEnum level %u failed - %s\n",
                         info_ctr.level, nt_errstr(status));
                  ret = false;
                  continue;
            }
            if (!W_ERROR_IS_OK(r.out.result)) {
                  torture_warning(tctx, "NetShareEnum level %u failed - %s\n",
                         info_ctr.level, win_errstr(r.out.result));
                  continue;
            }
            if (info_ctr.level == 0) {
                  struct srvsvc_NetShareCtr0 *ctr = r.out.info_ctr->ctr.ctr0;
                  if (ctr->count > 0) {
                        *one_sharename = ctr->array[0].name;
                  }
            }
      }

      return ret;
}

static bool torture_samba3_rpc_srvsvc(struct torture_context *torture)
{
      struct dcerpc_pipe *p;
      const char *sharename = NULL;
      bool ret = true;

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &p, &ndr_table_srvsvc),
            "failed to open srvsvc");

      ret &= test_NetShareEnum(torture, p, &sharename);
      if (sharename == NULL) {
            torture_comment(torture, "did not get sharename\n");
      } else {
            ret &= test_NetShareGetInfo(torture, p, sharename);
      }

      return ret;
}

/*
 * Do a ReqChallenge/Auth2 with a random wks name, make sure it returns
 * NT_STATUS_NO_SAM_ACCOUNT
 */

static bool torture_samba3_rpc_randomauth2(struct torture_context *torture)
{
      TALLOC_CTX *mem_ctx;
      struct dcerpc_pipe *net_pipe;
      struct dcerpc_binding_handle *net_handle;
      char *wksname;
      bool result = false;
      NTSTATUS status;
      struct netr_ServerReqChallenge r;
      struct netr_Credential netr_cli_creds;
      struct netr_Credential netr_srv_creds;
      uint32_t negotiate_flags;
      struct netr_ServerAuthenticate2 a;
      struct netlogon_creds_CredentialState *creds_state;
      struct netr_Credential netr_cred;
      struct samr_Password mach_pw;
      struct smbcli_state *cli;

      if (!(mem_ctx = talloc_new(torture))) {
            torture_comment(torture, "talloc_new failed\n");
            return false;
      }

      if (!(wksname = generate_random_str_list(
                  mem_ctx, 14, "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"))) {
            torture_comment(torture, "generate_random_str_list failed\n");
            goto done;
      }

      if (!(torture_open_connection_share(
                  mem_ctx, &cli,
                  torture, torture_setting_string(torture, "host", NULL),
                  "IPC$", torture->ev))) {
            torture_comment(torture, "IPC$ connection failed\n");
            goto done;
      }

      if (!(net_pipe = dcerpc_pipe_init(
                  mem_ctx, cli->transport->socket->event.ctx))) {
            torture_comment(torture, "dcerpc_pipe_init failed\n");
            goto done;
      }
      net_handle = net_pipe->binding_handle;

      status = dcerpc_pipe_open_smb(net_pipe, cli->tree, "\\netlogon");
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "dcerpc_pipe_open_smb failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      status = dcerpc_bind_auth_none(net_pipe, &ndr_table_netlogon);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "dcerpc_bind_auth_none failed: %s\n",
                   nt_errstr(status));
            goto done;
      }

      r.in.computer_name = wksname;
      r.in.server_name = talloc_asprintf(
            mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
      if (r.in.server_name == NULL) {
            torture_comment(torture, "talloc_asprintf failed\n");
            goto done;
      }
      generate_random_buffer(netr_cli_creds.data,
                         sizeof(netr_cli_creds.data));
      r.in.credentials = &netr_cli_creds;
      r.out.return_credentials = &netr_srv_creds;

      status = dcerpc_netr_ServerReqChallenge_r(net_handle, mem_ctx, &r);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(torture, "netr_ServerReqChallenge failed: %s\n",
                   nt_errstr(status));
            goto done;
      }
      if (!NT_STATUS_IS_OK(r.out.result)) {
            torture_comment(torture, "netr_ServerReqChallenge failed: %s\n",
                   nt_errstr(r.out.result));
            goto done;
      }

      negotiate_flags = NETLOGON_NEG_AUTH2_FLAGS;
      E_md4hash("foobar", mach_pw.hash);

      a.in.server_name = talloc_asprintf(
            mem_ctx, "\\\\%s", dcerpc_server_name(net_pipe));
      a.in.account_name = talloc_asprintf(
            mem_ctx, "%s$", wksname);
      a.in.computer_name = wksname;
      a.in.secure_channel_type = SEC_CHAN_WKSTA;
      a.in.negotiate_flags = &negotiate_flags;
      a.out.negotiate_flags = &negotiate_flags;
      a.in.credentials = &netr_cred;
      a.out.return_credentials = &netr_cred;

      creds_state = netlogon_creds_client_init(mem_ctx,
                                     a.in.account_name,
                                     a.in.computer_name,
                                     r.in.credentials,
                                     r.out.return_credentials, &mach_pw,
                                     &netr_cred, negotiate_flags);


      status = dcerpc_netr_ServerAuthenticate2_r(net_handle, mem_ctx, &a);
      if (!NT_STATUS_IS_OK(status)) {
            goto done;
      }
      if (!NT_STATUS_EQUAL(a.out.result, NT_STATUS_NO_TRUST_SAM_ACCOUNT)) {
            torture_comment(torture, "dcerpc_netr_ServerAuthenticate2 returned %s, "
                   "expected NT_STATUS_NO_TRUST_SAM_ACCOUNT\n",
                   nt_errstr(a.out.result));
            goto done;
      }

      result = true;
 done:
      talloc_free(mem_ctx);
      return result;
}

static struct security_descriptor *get_sharesec(struct torture_context *tctx,
                                    TALLOC_CTX *mem_ctx,
                                    struct smbcli_session *sess,
                                    const char *sharename)
{
      struct smbcli_tree *tree;
      TALLOC_CTX *tmp_ctx;
      struct dcerpc_pipe *p;
      struct dcerpc_binding_handle *b;
      NTSTATUS status;
      struct srvsvc_NetShareGetInfo r;
      union srvsvc_NetShareInfo info;
      struct security_descriptor *result;

      if (!(tmp_ctx = talloc_new(mem_ctx))) {
            torture_comment(tctx, "talloc_new failed\n");
            return NULL;
      }

      if (!NT_STATUS_IS_OK(secondary_tcon(tctx, tmp_ctx, sess, "IPC$", &tree))) {
            torture_comment(tctx, "secondary_tcon failed\n");
            talloc_free(tmp_ctx);
            return NULL;
      }

      status = pipe_bind_smb(tctx, mem_ctx, tree, "\\pipe\\srvsvc",
                         &ndr_table_srvsvc, &p);
      if (!NT_STATUS_IS_OK(status)) {
            torture_warning(tctx, "could not bind to srvsvc pipe: %s\n",
                   nt_errstr(status));
            talloc_free(tmp_ctx);
            return NULL;
      }
      b = p->binding_handle;

#if 0
      p->conn->flags |= DCERPC_DEBUG_PRINT_IN | DCERPC_DEBUG_PRINT_OUT;
#endif

      r.in.server_unc = talloc_asprintf(tmp_ctx, "\\\\%s",
                                dcerpc_server_name(p));
      r.in.share_name = sharename;
      r.in.level = 502;
      r.out.info = &info;

      status = dcerpc_srvsvc_NetShareGetInfo_r(b, tmp_ctx, &r);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "srvsvc_NetShareGetInfo failed: %s\n",
                   nt_errstr(status));
            talloc_free(tmp_ctx);
            return NULL;
      }
      if (!W_ERROR_IS_OK(r.out.result)) {
            torture_comment(tctx, "srvsvc_NetShareGetInfo failed: %s\n",
                   win_errstr(r.out.result));
            talloc_free(tmp_ctx);
            return NULL;
      }

      result = talloc_steal(mem_ctx, info.info502->sd_buf.sd);
      talloc_free(tmp_ctx);
      return result;
}

static NTSTATUS set_sharesec(struct torture_context *tctx,
                       TALLOC_CTX *mem_ctx,
                       struct smbcli_session *sess,
                       const char *sharename,
                       struct security_descriptor *sd)
{
      struct smbcli_tree *tree;
      TALLOC_CTX *tmp_ctx;
      struct dcerpc_pipe *p;
      struct dcerpc_binding_handle *b;
      NTSTATUS status;
      struct sec_desc_buf i;
      struct srvsvc_NetShareSetInfo r;
      union srvsvc_NetShareInfo info;
      uint32_t error = 0;

      if (!(tmp_ctx = talloc_new(mem_ctx))) {
            torture_comment(tctx, "talloc_new failed\n");
            return NT_STATUS_NO_MEMORY;
      }

      if (!NT_STATUS_IS_OK(secondary_tcon(tctx, tmp_ctx, sess, "IPC$", &tree))) {
            torture_comment(tctx, "secondary_tcon failed\n");
            talloc_free(tmp_ctx);
            return NT_STATUS_UNSUCCESSFUL;
      }

      status = pipe_bind_smb(tctx, mem_ctx, tree, "\\pipe\\srvsvc",
                         &ndr_table_srvsvc, &p);
      if (!NT_STATUS_IS_OK(status)) {
            torture_warning(tctx, "could not bind to srvsvc pipe: %s\n",
                   nt_errstr(status));
            talloc_free(tmp_ctx);
            return NT_STATUS_UNSUCCESSFUL;
      }
      b = p->binding_handle;

#if 0
      p->conn->flags |= DCERPC_DEBUG_PRINT_IN | DCERPC_DEBUG_PRINT_OUT;
#endif

      r.in.server_unc = talloc_asprintf(tmp_ctx, "\\\\%s",
                                dcerpc_server_name(p));
      r.in.share_name = sharename;
      r.in.level = 1501;
      i.sd = sd;
      info.info1501 = &i;
      r.in.info = &info;
      r.in.parm_error = &error;

      status = dcerpc_srvsvc_NetShareSetInfo_r(b, tmp_ctx, &r);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "srvsvc_NetShareSetInfo failed: %s\n",
                   nt_errstr(status));
      }
      if (!W_ERROR_IS_OK(r.out.result)) {
            torture_comment(tctx, "srvsvc_NetShareSetInfo failed: %s\n",
                  win_errstr(r.out.result));
            status = werror_to_ntstatus(r.out.result);
      }
      talloc_free(tmp_ctx);
      return status;
}

bool try_tcon(struct torture_context *tctx,
            TALLOC_CTX *mem_ctx,
            struct security_descriptor *orig_sd,
            struct smbcli_session *session,
            const char *sharename, const struct dom_sid *user_sid,
            unsigned int access_mask, NTSTATUS expected_tcon,
            NTSTATUS expected_mkdir)
{
      TALLOC_CTX *tmp_ctx;
      struct smbcli_tree *rmdir_tree, *tree;
      struct dom_sid *domain_sid;
      uint32_t rid;
      struct security_descriptor *sd;
      NTSTATUS status;
      bool ret = true;

      if (!(tmp_ctx = talloc_new(mem_ctx))) {
            torture_comment(tctx, "talloc_new failed\n");
            return false;
      }

      status = secondary_tcon(tctx, tmp_ctx, session, sharename, &rmdir_tree);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "first tcon to delete dir failed\n");
            talloc_free(tmp_ctx);
            return false;
      }

      smbcli_rmdir(rmdir_tree, "sharesec_testdir");

      if (!NT_STATUS_IS_OK(dom_sid_split_rid(tmp_ctx, user_sid,
                                     &domain_sid, &rid))) {
            torture_comment(tctx, "dom_sid_split_rid failed\n");
            talloc_free(tmp_ctx);
            return false;
      }

      sd = security_descriptor_dacl_create(
            tmp_ctx, 0, "S-1-5-32-544",
            dom_sid_string(mem_ctx, dom_sid_add_rid(mem_ctx, domain_sid,
                                          DOMAIN_RID_USERS)),
            dom_sid_string(mem_ctx, user_sid),
            SEC_ACE_TYPE_ACCESS_ALLOWED, access_mask, 0, NULL);
      if (sd == NULL) {
            torture_comment(tctx, "security_descriptor_dacl_create failed\n");
            talloc_free(tmp_ctx);
                return false;
        }

      status = set_sharesec(tctx, mem_ctx, session, sharename, sd);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "custom set_sharesec failed: %s\n",
                   nt_errstr(status));
            talloc_free(tmp_ctx);
                return false;
      }

      status = secondary_tcon(tctx, tmp_ctx, session, sharename, &tree);
      if (!NT_STATUS_EQUAL(status, expected_tcon)) {
            torture_comment(tctx, "Expected %s, got %s\n", nt_errstr(expected_tcon),
                   nt_errstr(status));
            ret = false;
            goto done;
      }

      if (!NT_STATUS_IS_OK(status)) {
            /* An expected non-access, no point in trying to write */
            goto done;
      }

      status = smbcli_mkdir(tree, "sharesec_testdir");
      if (!NT_STATUS_EQUAL(status, expected_mkdir)) {
            torture_warning(tctx, "Expected %s, got %s\n",
                   nt_errstr(expected_mkdir), nt_errstr(status));
            ret = false;
      }

 done:
      smbcli_rmdir(rmdir_tree, "sharesec_testdir");

      status = set_sharesec(tctx, mem_ctx, session, sharename, orig_sd);
      if (!NT_STATUS_IS_OK(status)) {
            torture_comment(tctx, "custom set_sharesec failed: %s\n",
                   nt_errstr(status));
            talloc_free(tmp_ctx);
                return false;
      }

      talloc_free(tmp_ctx);
      return ret;
}

static bool torture_samba3_rpc_sharesec(struct torture_context *torture)
{
      TALLOC_CTX *mem_ctx;
      bool ret = true;
      struct smbcli_state *cli;
      struct security_descriptor *sd;
      struct dom_sid *user_sid;

      if (!(mem_ctx = talloc_new(torture))) {
            return false;
      }

      if (!(torture_open_connection_share(
                  mem_ctx, &cli, torture, torture_setting_string(torture, "host", NULL),
                  "IPC$", torture->ev))) {
            torture_comment(torture, "IPC$ connection failed\n");
            talloc_free(mem_ctx);
            return false;
      }

      if (!(user_sid = whoami(torture, mem_ctx, cli->tree))) {
            torture_comment(torture, "whoami failed\n");
            talloc_free(mem_ctx);
            return false;
      }

      sd = get_sharesec(torture, mem_ctx, cli->session,
                    torture_setting_string(torture, "share", NULL));

      ret &= try_tcon(torture, mem_ctx, sd, cli->session,
                  torture_setting_string(torture, "share", NULL),
                  user_sid, 0, NT_STATUS_ACCESS_DENIED, NT_STATUS_OK);

      ret &= try_tcon(torture, mem_ctx, sd, cli->session,
                  torture_setting_string(torture, "share", NULL),
                  user_sid, SEC_FILE_READ_DATA, NT_STATUS_OK,
                  NT_STATUS_MEDIA_WRITE_PROTECTED);

      ret &= try_tcon(torture, mem_ctx, sd, cli->session,
                  torture_setting_string(torture, "share", NULL),
                  user_sid, SEC_FILE_ALL, NT_STATUS_OK, NT_STATUS_OK);

      talloc_free(mem_ctx);
      return ret;
}

static bool torture_samba3_rpc_lsa(struct torture_context *torture)
{
      struct dcerpc_pipe *p;
      struct dcerpc_binding_handle *b;
      struct policy_handle lsa_handle;

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &p, &ndr_table_lsarpc),
            "failed to setup lsarpc");

      b = p->binding_handle;

      {
            struct lsa_ObjectAttribute attr;
            struct lsa_OpenPolicy2 o;
            o.in.system_name = talloc_asprintf(
                  torture, "\\\\%s", dcerpc_server_name(p));
            ZERO_STRUCT(attr);
            o.in.attr = &attr;
            o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
            o.out.handle = &lsa_handle;

            torture_assert_ntstatus_ok(torture,
                  dcerpc_lsa_OpenPolicy2_r(b, torture, &o),
                  "dcerpc_lsa_OpenPolicy2 failed");
            torture_assert_ntstatus_ok(torture, o.out.result,
                  "dcerpc_lsa_OpenPolicy2 failed");
      }

      {
            int i;
            int levels[] = { 2,3,5,6 };

            for (i=0; i<ARRAY_SIZE(levels); i++) {
                  struct lsa_QueryInfoPolicy r;
                  union lsa_PolicyInformation *info = NULL;
                  r.in.handle = &lsa_handle;
                  r.in.level = levels[i];
                  r.out.info = &info;

                  torture_assert_ntstatus_ok(torture,
                        dcerpc_lsa_QueryInfoPolicy_r(b, torture, &r),
                        talloc_asprintf(torture, "dcerpc_lsa_QueryInfoPolicy level %d failed", levels[i]));
                  torture_assert_ntstatus_ok(torture, r.out.result,
                        talloc_asprintf(torture, "dcerpc_lsa_QueryInfoPolicy level %d failed", levels[i]));
            }
      }

      return true;
}

static NTSTATUS get_servername(TALLOC_CTX *mem_ctx, struct smbcli_tree *tree,
                         char **name)
{
      struct rap_WserverGetInfo r;
      NTSTATUS status;
      char servername[17];

      r.in.level = 0;
      r.in.bufsize = 0xffff;

      status = smbcli_rap_netservergetinfo(tree, mem_ctx, &r);
      if (!NT_STATUS_IS_OK(status)) {
            return status;
      }

      memcpy(servername, r.out.info.info0.name, 16);
      servername[16] = '\0';

      if (!pull_ascii_talloc(mem_ctx, name, servername, NULL)) {
            return NT_STATUS_NO_MEMORY;
      }

      return NT_STATUS_OK;
}

static bool rap_get_servername(struct torture_context *tctx,
                         char **servername)
{
      struct smbcli_state *cli;

      torture_assert(tctx,
            torture_open_connection_share(tctx, &cli, tctx, torture_setting_string(tctx, "host", NULL),
                                    "IPC$", tctx->ev),
            "IPC$ connection failed");

      torture_assert_ntstatus_ok(tctx,
            get_servername(tctx, cli->tree, servername),
            "get_servername failed");

      talloc_free(cli);

      return true;
}

static bool find_printers(struct torture_context *tctx,
                    struct dcerpc_pipe *p,
                    const char ***printers,
                    int *num_printers)
{
      struct srvsvc_NetShareEnum r;
      struct srvsvc_NetShareInfoCtr info_ctr;
      struct srvsvc_NetShareCtr1 c1_in;
      struct srvsvc_NetShareCtr1 *c1;
      uint32_t totalentries = 0;
      int i;
      struct dcerpc_binding_handle *b = p->binding_handle;

      ZERO_STRUCT(c1_in);
      info_ctr.level = 1;
      info_ctr.ctr.ctr1 = &c1_in;

      r.in.server_unc = talloc_asprintf(
            tctx, "\\\\%s", dcerpc_server_name(p));
      r.in.info_ctr = &info_ctr;
      r.in.max_buffer = (uint32_t)-1;
      r.in.resume_handle = NULL;
      r.out.totalentries = &totalentries;
      r.out.info_ctr = &info_ctr;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_srvsvc_NetShareEnum_r(b, tctx, &r),
            "NetShareEnum level 1 failed");
      torture_assert_werr_ok(tctx, r.out.result,
            "NetShareEnum level 1 failed");

      *printers = NULL;
      *num_printers = 0;
      c1 = r.out.info_ctr->ctr.ctr1;
      for (i=0; i<c1->count; i++) {
            if (c1->array[i].type != STYPE_PRINTQ) {
                  continue;
            }
            if (!add_string_to_array(tctx, c1->array[i].name,
                               printers, num_printers)) {
                  return false;
            }
      }

      return true;
}

static bool enumprinters(struct torture_context *tctx,
                   struct dcerpc_binding_handle *b,
                   const char *servername, int level, int *num_printers)
{
      struct spoolss_EnumPrinters r;
      DATA_BLOB blob;
      uint32_t needed;
      uint32_t count;
      union spoolss_PrinterInfo *info;

      r.in.flags = PRINTER_ENUM_LOCAL;
      r.in.server = talloc_asprintf(tctx, "\\\\%s", servername);
      r.in.level = level;
      r.in.buffer = NULL;
      r.in.offered = 0;
      r.out.needed = &needed;
      r.out.count = &count;
      r.out.info = &info;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_spoolss_EnumPrinters_r(b, tctx, &r),
            "dcerpc_spoolss_EnumPrinters failed");
      torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
            "EnumPrinters unexpected return code should be WERR_INSUFFICIENT_BUFFER");

      blob = data_blob_talloc_zero(tctx, needed);
      if (blob.data == NULL) {
            return false;
      }

      r.in.buffer = &blob;
      r.in.offered = needed;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_spoolss_EnumPrinters_r(b, tctx, &r),
            "dcerpc_spoolss_EnumPrinters failed");
      torture_assert_werr_ok(tctx, r.out.result,
            "dcerpc_spoolss_EnumPrinters failed");

      *num_printers = count;

      return true;
}

static bool getprinterinfo(struct torture_context *tctx,
                     struct dcerpc_binding_handle *b,
                     struct policy_handle *handle, int level,
                     union spoolss_PrinterInfo **res)
{
      struct spoolss_GetPrinter r;
      DATA_BLOB blob;
      uint32_t needed;

      r.in.handle = handle;
      r.in.level = level;
      r.in.buffer = NULL;
      r.in.offered = 0;
      r.out.needed = &needed;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_spoolss_GetPrinter_r(b, tctx, &r),
            "dcerpc_spoolss_GetPrinter failed");
      torture_assert_werr_equal(tctx, r.out.result, WERR_INSUFFICIENT_BUFFER,
            "GetPrinter unexpected return code should be WERR_INSUFFICIENT_BUFFER");

      r.in.handle = handle;
      r.in.level = level;
      blob = data_blob_talloc_zero(tctx, needed);
      if (blob.data == NULL) {
            return false;
      }
      r.in.buffer = &blob;
      r.in.offered = needed;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_spoolss_GetPrinter_r(b, tctx, &r),
            "dcerpc_spoolss_GetPrinter failed");
      torture_assert_werr_ok(tctx, r.out.result,
            "dcerpc_spoolss_GetPrinter failed");

      if (res != NULL) {
            *res = talloc_steal(tctx, r.out.info);
      }

      return true;
}

static bool torture_samba3_rpc_spoolss(struct torture_context *torture)
{
      struct dcerpc_pipe *p, *p2;
      struct dcerpc_binding_handle *b;
      struct policy_handle server_handle, printer_handle;
      const char **printers;
      int num_printers;
      struct spoolss_UserLevel1 userlevel1;
      char *servername;

      torture_assert(torture,
            rap_get_servername(torture, &servername),
            "failed to rap servername");

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &p2, &ndr_table_srvsvc),
            "failed to setup srvsvc");

      torture_assert(torture,
            find_printers(torture, p2, &printers, &num_printers),
            "failed to find printers via srvsvc");

      talloc_free(p2);

      if (num_printers == 0) {
            torture_skip(torture, "Did not find printers\n");
            return true;
      }

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &p, &ndr_table_spoolss),
            "failed to setup spoolss");

      b = p->binding_handle;

      ZERO_STRUCT(userlevel1);
      userlevel1.client = talloc_asprintf(
            torture, "\\\\%s", lpcfg_netbios_name(torture->lp_ctx));
      userlevel1.user = cli_credentials_get_username(cmdline_credentials);
      userlevel1.build = 2600;
      userlevel1.major = 3;
      userlevel1.minor = 0;
      userlevel1.processor = 0;

      {
            struct spoolss_OpenPrinterEx r;

            ZERO_STRUCT(r);
            r.in.printername = talloc_asprintf(torture, "\\\\%s",
                                       servername);
            r.in.datatype = NULL;
            r.in.access_mask = 0;
            r.in.level = 1;
            r.in.userlevel.level1 = &userlevel1;
            r.out.handle = &server_handle;

            torture_assert_ntstatus_ok(torture,
                  dcerpc_spoolss_OpenPrinterEx_r(b, torture, &r),
                  "dcerpc_spoolss_OpenPrinterEx failed");
            torture_assert_werr_ok(torture, r.out.result,
                  "dcerpc_spoolss_OpenPrinterEx failed");
      }

      {
            struct spoolss_ClosePrinter r;

            r.in.handle = &server_handle;
            r.out.handle = &server_handle;

            torture_assert_ntstatus_ok(torture,
                  dcerpc_spoolss_ClosePrinter_r(b, torture, &r),
                  "dcerpc_spoolss_ClosePrinter failed");
            torture_assert_werr_ok(torture, r.out.result,
                  "dcerpc_spoolss_ClosePrinter failed");
      }

      {
            struct spoolss_OpenPrinterEx r;

            ZERO_STRUCT(r);
            r.in.printername = talloc_asprintf(
                  torture, "\\\\%s\\%s", servername, printers[0]);
            r.in.datatype = NULL;
            r.in.access_mask = 0;
            r.in.level = 1;
            r.in.userlevel.level1 = &userlevel1;
            r.out.handle = &printer_handle;

            torture_assert_ntstatus_ok(torture,
                  dcerpc_spoolss_OpenPrinterEx_r(b, torture, &r),
                  "dcerpc_spoolss_OpenPrinterEx failed");
            torture_assert_werr_ok(torture, r.out.result,
                  "dcerpc_spoolss_OpenPrinterEx failed");
      }

      {
            int i;

            for (i=0; i<8; i++) {
                  torture_assert(torture,
                        getprinterinfo(torture, b, &printer_handle, i, NULL),
                        talloc_asprintf(torture, "getprinterinfo %d failed", i));
            }
      }

      {
            struct spoolss_ClosePrinter r;

            r.in.handle = &printer_handle;
            r.out.handle = &printer_handle;

            torture_assert_ntstatus_ok(torture,
                  dcerpc_spoolss_ClosePrinter_r(b, torture, &r),
                  "dcerpc_spoolss_ClosePrinter failed");
            torture_assert_werr_ok(torture, r.out.result,
                  "dcerpc_spoolss_ClosePrinter failed");
      }

      {
            int num_enumerated;

            torture_assert(torture,
                  enumprinters(torture, b, servername, 1, &num_enumerated),
                  "enumprinters failed");

            torture_assert_int_equal(torture, num_printers, num_enumerated,
                  "netshareenum / enumprinters lvl 1 numprinter mismatch");
      }

      {
            int num_enumerated;

            torture_assert(torture,
                  enumprinters(torture, b, servername, 2, &num_enumerated),
                  "enumprinters failed");

            torture_assert_int_equal(torture, num_printers, num_enumerated,
                  "netshareenum / enumprinters lvl 2 numprinter mismatch");
      }

      return true;
}

static bool torture_samba3_rpc_wkssvc(struct torture_context *torture)
{
      struct dcerpc_pipe *p;
      struct dcerpc_binding_handle *b;
      char *servername;

      torture_assert(torture,
            rap_get_servername(torture, &servername),
            "failed to rap servername");

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &p, &ndr_table_wkssvc),
            "failed to setup wkssvc");

      b = p->binding_handle;

      {
            struct wkssvc_NetWkstaInfo100 wks100;
            union wkssvc_NetWkstaInfo info;
            struct wkssvc_NetWkstaGetInfo r;

            r.in.server_name = "\\foo";
            r.in.level = 100;
            info.info100 = &wks100;
            r.out.info = &info;

            torture_assert_ntstatus_ok(torture,
                  dcerpc_wkssvc_NetWkstaGetInfo_r(b, torture, &r),
                  "dcerpc_wkssvc_NetWksGetInfo failed");
            torture_assert_werr_ok(torture, r.out.result,
                  "dcerpc_wkssvc_NetWksGetInfo failed");

            torture_assert_str_equal(torture, servername, r.out.info->info100->server_name,
                  "servername RAP / DCERPC inconsistency");
      }

      return true;
}

static bool winreg_close(struct torture_context *tctx,
                   struct dcerpc_binding_handle *b,
                   struct policy_handle *handle)
{
      struct winreg_CloseKey c;

      c.in.handle = c.out.handle = handle;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_winreg_CloseKey_r(b, tctx, &c),
            "winreg_CloseKey failed");
      torture_assert_werr_ok(tctx, c.out.result,
            "winreg_CloseKey failed");

      return true;
}

static bool enumvalues(struct torture_context *tctx,
                   struct dcerpc_binding_handle *b,
                   struct policy_handle *handle)
{
      uint32_t enum_index = 0;

      while (1) {
            struct winreg_EnumValue r;
            struct winreg_ValNameBuf name;
            enum winreg_Type type = 0;
            uint8_t buf8[1024];
            NTSTATUS status;
            uint32_t size, length;

            r.in.handle = handle;
            r.in.enum_index = enum_index;
            name.name = "";
            name.size = 1024;
            r.in.name = r.out.name = &name;
            size = 1024;
            length = 5;
            r.in.type = &type;
            r.in.value = buf8;
            r.in.size = &size;
            r.in.length = &length;

            status = dcerpc_winreg_EnumValue_r(b, tctx, &r);
            if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
                  return true;
            }
            enum_index += 1;
      }
}

static bool enumkeys(struct torture_context *tctx,
                 struct dcerpc_binding_handle *b,
                 struct policy_handle *handle,
                 int depth)
{
      struct winreg_EnumKey r;
      struct winreg_StringBuf kclass, name;
      NTSTATUS status;
      NTTIME t = 0;

      if (depth <= 0) {
            return true;
      }

      kclass.name   = "";
      kclass.size   = 1024;

      r.in.handle = handle;
      r.in.enum_index = 0;
      r.in.name = &name;
      r.in.keyclass = &kclass;
      r.out.name = &name;
      r.in.last_changed_time = &t;

      do {
            struct winreg_OpenKey o;
            struct policy_handle key_handle;
            int i;

            name.name = NULL;
            name.size = 1024;

            status = dcerpc_winreg_EnumKey_r(b, tctx, &r);
            if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
                  /* We're done enumerating */
                  return true;
            }

            for (i=0; i<10-depth; i++) {
                  torture_comment(tctx, " ");
            }
            torture_comment(tctx, "%s\n", r.out.name->name);

            o.in.parent_handle = handle;
            o.in.keyname.name = r.out.name->name;
            o.in.options = 0;
            o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
            o.out.handle = &key_handle;

            status = dcerpc_winreg_OpenKey_r(b, tctx, &o);
            if (NT_STATUS_IS_OK(status) && W_ERROR_IS_OK(o.out.result)) {
                  enumkeys(tctx, b, &key_handle, depth-1);
                  enumvalues(tctx, b, &key_handle);
                  torture_assert(tctx, winreg_close(tctx, b, &key_handle), "");
            }

            r.in.enum_index += 1;
      } while(true);

      return true;
}

typedef NTSTATUS (*winreg_open_fn)(struct dcerpc_binding_handle *, TALLOC_CTX *, void *);

static bool test_Open3(struct torture_context *tctx,
                   struct dcerpc_binding_handle *b,
                   const char *name, winreg_open_fn open_fn)
{
      struct policy_handle handle;
      struct winreg_OpenHKLM r;

      r.in.system_name = 0;
      r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      r.out.handle = &handle;

      torture_assert_ntstatus_ok(tctx,
            open_fn(b, tctx, &r),
            talloc_asprintf(tctx, "%s failed", name));
      torture_assert_werr_ok(tctx, r.out.result,
            talloc_asprintf(tctx, "%s failed", name));

      enumkeys(tctx, b, &handle, 4);

      torture_assert(tctx,
            winreg_close(tctx, b, &handle),
            "dcerpc_CloseKey failed");

      return true;
}

static bool torture_samba3_rpc_winreg(struct torture_context *torture)
{
      struct dcerpc_pipe *p;
      struct dcerpc_binding_handle *b;
      bool ret = true;
      struct {
            const char *name;
            winreg_open_fn fn;
      } open_fns[] = {
            {"OpenHKLM", (winreg_open_fn)dcerpc_winreg_OpenHKLM_r },
            {"OpenHKU",  (winreg_open_fn)dcerpc_winreg_OpenHKU_r },
            {"OpenHKPD", (winreg_open_fn)dcerpc_winreg_OpenHKPD_r },
            {"OpenHKPT", (winreg_open_fn)dcerpc_winreg_OpenHKPT_r },
            {"OpenHKCR", (winreg_open_fn)dcerpc_winreg_OpenHKCR_r }};
#if 0
      int i;
#endif

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &p, &ndr_table_winreg),
            "failed to setup winreg");

      b = p->binding_handle;

#if 1
      ret = test_Open3(torture, b, open_fns[0].name, open_fns[0].fn);
#else
      for (i = 0; i < ARRAY_SIZE(open_fns); i++) {
            if (!test_Open3(torture, b, open_fns[i].name, open_fns[i].fn))
                  ret = false;
      }
#endif
      return ret;
}

static bool get_shareinfo(struct torture_context *tctx,
                    struct dcerpc_binding_handle *b,
                    const char *servername,
                    const char *share,
                    struct srvsvc_NetShareInfo502 **info502)
{
      struct srvsvc_NetShareGetInfo r;
      union srvsvc_NetShareInfo info;

      r.in.server_unc = talloc_asprintf(tctx, "\\\\%s", servername);
      r.in.share_name = share;
      r.in.level = 502;
      r.out.info = &info;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_srvsvc_NetShareGetInfo_r(b, tctx, &r),
            "srvsvc_NetShareGetInfo failed");
      torture_assert_werr_ok(tctx, r.out.result,
            "srvsvc_NetShareGetInfo failed");

      *info502 = talloc_move(tctx, &info.info502);

      return true;
}

/*
 * Get us a handle on HKLM\
 */

static bool get_hklm_handle(struct torture_context *tctx,
                      struct dcerpc_binding_handle *b,
                      struct policy_handle *handle)
{
      struct winreg_OpenHKLM r;
      struct policy_handle result;

      r.in.system_name = 0;
        r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
        r.out.handle = &result;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_winreg_OpenHKLM_r(b, tctx, &r),
            "OpenHKLM failed");
      torture_assert_werr_ok(tctx, r.out.result,
            "OpenHKLM failed");

      *handle = result;

      return true;
}

static bool torture_samba3_createshare(struct torture_context *tctx,
                               struct dcerpc_binding_handle *b,
                               const char *sharename)
{
      struct policy_handle hklm;
      struct policy_handle new_handle;
      struct winreg_CreateKey c;
      struct winreg_CloseKey cl;
      enum winreg_CreateAction action_taken;

      c.in.handle = &hklm;
      c.in.name.name = talloc_asprintf(
            tctx, "software\\samba\\smbconf\\%s", sharename);
      torture_assert(tctx, c.in.name.name, "talloc_asprintf failed");

      c.in.keyclass.name = "";
      c.in.options = 0;
      c.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      c.in.secdesc = NULL;
      c.in.action_taken = &action_taken;
      c.out.new_handle = &new_handle;
      c.out.action_taken = &action_taken;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_winreg_CreateKey_r(b, tctx, &c),
            "OpenKey failed");
      torture_assert_werr_ok(tctx, c.out.result,
            "OpenKey failed");

      cl.in.handle = &new_handle;
      cl.out.handle = &new_handle;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_winreg_CloseKey_r(b, tctx, &cl),
            "CloseKey failed");
      torture_assert_werr_ok(tctx, cl.out.result,
            "CloseKey failed");

      return true;
}

static bool torture_samba3_deleteshare(struct torture_context *tctx,
                               struct dcerpc_binding_handle *b,
                               const char *sharename)
{
      struct policy_handle hklm;
      struct winreg_DeleteKey d;

      torture_assert(tctx,
            get_hklm_handle(tctx, b, &hklm),
            "get_hklm_handle failed");

      d.in.handle = &hklm;
      d.in.key.name = talloc_asprintf(
            tctx, "software\\samba\\smbconf\\%s", sharename);
      torture_assert(tctx, d.in.key.name, "talloc_asprintf failed");

      torture_assert_ntstatus_ok(tctx,
            dcerpc_winreg_DeleteKey_r(b, tctx, &d),
            "DeleteKey failed");
      torture_assert_werr_ok(tctx, d.out.result,
            "DeleteKey failed");

      return true;
}

static bool torture_samba3_setconfig(struct torture_context *tctx,
                             struct dcerpc_binding_handle *b,
                             const char *sharename,
                             const char *parameter,
                             const char *value)
{
      struct policy_handle hklm, key_handle;
      struct winreg_OpenKey o;
      struct winreg_SetValue s;
      uint32_t type;
      DATA_BLOB val;

      torture_assert(tctx,
            get_hklm_handle(tctx, b, &hklm),
            "get_hklm_handle failed");

      o.in.parent_handle = &hklm;
      o.in.keyname.name = talloc_asprintf(
            tctx, "software\\samba\\smbconf\\%s", sharename);
      torture_assert(tctx, o.in.keyname.name, "talloc_asprintf failed");

      o.in.options = 0;
      o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
      o.out.handle = &key_handle;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_winreg_OpenKey_r(b, tctx, &o),
            "OpenKey failed");
      torture_assert_werr_ok(tctx, o.out.result,
            "OpenKey failed");

      torture_assert(tctx,
            reg_string_to_val(tctx, "REG_SZ", value, &type, &val),
            "reg_string_to_val failed");

      s.in.handle = &key_handle;
      s.in.name.name = parameter;
      s.in.type = type;
      s.in.data = val.data;
      s.in.size = val.length;

      torture_assert_ntstatus_ok(tctx,
            dcerpc_winreg_SetValue_r(b, tctx, &s),
            "SetValue failed");
      torture_assert_werr_ok(tctx, s.out.result,
            "SetValue failed");

      return true;
}

static bool torture_samba3_regconfig(struct torture_context *torture)
{
      struct srvsvc_NetShareInfo502 *i = NULL;
      const char *comment = "Dummer Kommentar";
      struct dcerpc_pipe *srvsvc_pipe, *winreg_pipe;

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &srvsvc_pipe, &ndr_table_srvsvc),
            "failed to setup srvsvc");

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &winreg_pipe, &ndr_table_winreg),
            "failed to setup winreg");

      torture_assert(torture,
            torture_samba3_createshare(torture, winreg_pipe->binding_handle, "blubber"),
            "torture_samba3_createshare failed");

      torture_assert(torture,
            torture_samba3_setconfig(torture, winreg_pipe->binding_handle, "blubber", "comment", comment),
            "torture_samba3_setconfig failed");

      torture_assert(torture,
            get_shareinfo(torture, srvsvc_pipe->binding_handle, dcerpc_server_name(srvsvc_pipe), "blubber", &i),
            "get_shareinfo failed");

      torture_assert_str_equal(torture, comment, i->comment,
            "got unexpected comment");

      torture_assert(torture,
            torture_samba3_deleteshare(torture, winreg_pipe->binding_handle, "blubber"),
            "torture_samba3_deleteshare failed");

      return true;
}

/*
 * Test that even with a result of 0 rids the array is returned as a
 * non-NULL pointer. Yes, XP does notice.
 */

bool torture_samba3_getaliasmembership_0(struct torture_context *torture)
{
      struct dcerpc_pipe *p;
      struct dcerpc_binding_handle *b;
      struct samr_Connect2 c;
      struct samr_OpenDomain o;
      struct dom_sid sid;
      struct lsa_SidPtr ptr;
      struct lsa_SidArray sids;
      struct samr_GetAliasMembership g;
      struct samr_Ids rids;
      struct policy_handle samr, domain;

      torture_assert_ntstatus_ok(torture,
            torture_rpc_connection(torture, &p, &ndr_table_samr),
            "failed to setup samr");

      b = p->binding_handle;

      c.in.system_name = NULL;
      c.in.access_mask = SAMR_ACCESS_LOOKUP_DOMAIN;
      c.out.connect_handle = &samr;
      torture_assert_ntstatus_ok(torture,
            dcerpc_samr_Connect2_r(b, torture, &c),
            "");
      torture_assert_ntstatus_ok(torture, c.out.result,
            "");
      dom_sid_parse("S-1-5-32", &sid);
      o.in.connect_handle = &samr;
      o.in.access_mask = SAMR_DOMAIN_ACCESS_LOOKUP_ALIAS;
      o.in.sid = &sid;
      o.out.domain_handle = &domain;
      torture_assert_ntstatus_ok(torture,
            dcerpc_samr_OpenDomain_r(b, torture, &o),
            "");
      torture_assert_ntstatus_ok(torture, o.out.result,
            "");
      dom_sid_parse("S-1-2-3-4-5", &sid);
      ptr.sid = &sid;
      sids.num_sids = 1;
      sids.sids = &ptr;
      g.in.domain_handle = &domain;
      g.in.sids = &sids;
      g.out.rids = &rids;
      torture_assert_ntstatus_ok(torture,
            dcerpc_samr_GetAliasMembership_r(b, torture, &g),
            "");
      torture_assert_ntstatus_ok(torture, g.out.result,
            "");
      if (rids.ids == NULL) {
            /* This is the piece to test here */
            torture_fail(torture,
                  "torture_samba3_getaliasmembership_0: "
                  "Server returns NULL rids array\n");
      }

      return true;
}

struct torture_suite *torture_rpc_samba3(TALLOC_CTX *mem_ctx)
{
      struct torture_suite *suite = torture_suite_create(mem_ctx, "SAMBA3");

      torture_suite_add_simple_test(suite, "BIND", torture_bind_samba3);
      torture_suite_add_simple_test(suite, "NETLOGON", torture_netlogon_samba3);
      torture_suite_add_simple_test(suite, "SESSIONKEY", torture_samba3_sessionkey);
      torture_suite_add_simple_test(suite, "SRVSVC", torture_samba3_rpc_srvsvc);
      torture_suite_add_simple_test(suite, "SHARESEC", torture_samba3_rpc_sharesec);
      torture_suite_add_simple_test(suite, "GETUSERNAME", torture_samba3_rpc_getusername);
      torture_suite_add_simple_test(suite, "RANDOMAUTH2", torture_samba3_rpc_randomauth2);
      torture_suite_add_simple_test(suite, "LSA", torture_samba3_rpc_lsa);
      torture_suite_add_simple_test(suite, "SPOOLSS", torture_samba3_rpc_spoolss);
      torture_suite_add_simple_test(suite, "WKSSVC", torture_samba3_rpc_wkssvc);
      torture_suite_add_simple_test(suite, "WINREG", torture_samba3_rpc_winreg);
      torture_suite_add_simple_test(suite, "GETALIASMEMBERSHIP-0", torture_samba3_getaliasmembership_0);
      torture_suite_add_simple_test(suite, "REGCONFIG", torture_samba3_regconfig);

      suite->description = talloc_strdup(suite, "samba3 DCERPC interface tests");

      return suite;
}

Generated by  Doxygen 1.6.0   Back to index