[GLUE] Rsync SAMBA_3_2_0 SVN r25598 in order to create the v3-2-test branch.
[samba.git] / source / rpc_client / cli_lsarpc.c
index aa1cb95fda14ea0067ad399d8dc89f1936f396d7..be85e49476a8c8a4318d0181ddb84e989638a615 100644 (file)
@@ -1,24 +1,24 @@
-/* 
+/*
    Unix SMB/CIFS implementation.
    RPC pipe client
    Copyright (C) Tim Potter                        2000-2001,
    Copyright (C) Andrew Tridgell              1992-1997,2000,
    Copyright (C) Rafal Szczesniak                       2002
    Copyright (C) Jeremy Allison                                2005.
-   
+   Copyright (C) Michael Adam                          2007.
+
    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 2 of the License, or
+   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, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
@@ -78,9 +78,6 @@ NTSTATUS rpccli_lsa_open_policy(struct rpc_pipe_client *cli,
 
        if (NT_STATUS_IS_OK(result)) {
                *pol = r.pol;
-#ifdef __INSURE__
-               pol->marker = MALLOC(1);
-#endif
        }
 
        return result;
@@ -88,7 +85,7 @@ NTSTATUS rpccli_lsa_open_policy(struct rpc_pipe_client *cli,
 
 /** Open a LSA policy handle
   *
-  * @param cli Handle on an initialised SMB connection 
+  * @param cli Handle on an initialised SMB connection
   */
 
 NTSTATUS rpccli_lsa_open_policy2(struct rpc_pipe_client *cli,
@@ -125,121 +122,85 @@ NTSTATUS rpccli_lsa_open_policy2(struct rpc_pipe_client *cli,
 
        if (NT_STATUS_IS_OK(result)) {
                *pol = r.pol;
-#ifdef __INSURE__
-               pol->marker = (char *)malloc(1);
-#endif
        }
 
        return result;
 }
 
-/** Close a LSA policy handle */
-
-NTSTATUS rpccli_lsa_close(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx, 
-                         POLICY_HND *pol)
-{
-       prs_struct qbuf, rbuf;
-       LSA_Q_CLOSE q;
-       LSA_R_CLOSE r;
-       NTSTATUS result;
-
-       ZERO_STRUCT(q);
-       ZERO_STRUCT(r);
-
-       init_lsa_q_close(&q, pol);
-
-       CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_CLOSE,
-                       q, r,
-                       qbuf, rbuf,
-                       lsa_io_q_close,
-                       lsa_io_r_close,
-                       NT_STATUS_UNSUCCESSFUL );
-
-       /* Return output parameters */
-
-       result = r.status;
-
-       if (NT_STATUS_IS_OK(result)) {
-#ifdef __INSURE__
-               SAFE_FREE(pol->marker);
-#endif
-               *pol = r.pol;
-       }
-
-       return result;
-}
-
-/** Lookup a list of sids */
-
-NTSTATUS rpccli_lsa_lookup_sids(struct rpc_pipe_client *cli,
-                               TALLOC_CTX *mem_ctx,
-                               POLICY_HND *pol, int num_sids,
-                               const DOM_SID *sids, 
-                               char ***domains, char ***names, uint32 **types)
+/* Lookup a list of sids
+ *
+ * internal version withOUT memory allocation of the target arrays.
+ * this assumes suffciently sized arrays to store domains, names and types. */
+
+static NTSTATUS rpccli_lsa_lookup_sids_noalloc(struct rpc_pipe_client *cli,
+                                              TALLOC_CTX *mem_ctx,
+                                              POLICY_HND *pol,
+                                              int num_sids,
+                                              const DOM_SID *sids,
+                                              char **domains,
+                                              char **names,
+                                              uint32 *types)
 {
        prs_struct qbuf, rbuf;
        LSA_Q_LOOKUP_SIDS q;
        LSA_R_LOOKUP_SIDS r;
        DOM_R_REF ref;
-       LSA_TRANS_NAME_ENUM t_names;
        NTSTATUS result = NT_STATUS_OK;
+       TALLOC_CTX *tmp_ctx = NULL;
        int i;
 
+       tmp_ctx = talloc_new(mem_ctx);
+       if (!tmp_ctx) {
+               DEBUG(0, ("rpccli_lsa_lookup_sids_noalloc: out of memory!\n"));
+               result = NT_STATUS_UNSUCCESSFUL;
+               goto done;
+       }
+
        ZERO_STRUCT(q);
        ZERO_STRUCT(r);
 
-       init_q_lookup_sids(mem_ctx, &q, pol, num_sids, sids, 1);
+       init_q_lookup_sids(tmp_ctx, &q, pol, num_sids, sids, 1);
 
        ZERO_STRUCT(ref);
-       ZERO_STRUCT(t_names);
 
        r.dom_ref = &ref;
-       r.names = &t_names;
 
-       CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_LOOKUPSIDS,
+       CLI_DO_RPC( cli, tmp_ctx, PI_LSARPC, LSA_LOOKUPSIDS,
                        q, r,
                        qbuf, rbuf,
                        lsa_io_q_lookup_sids,
                        lsa_io_r_lookup_sids,
                        NT_STATUS_UNSUCCESSFUL );
 
+       DEBUG(10, ("LSA_LOOKUPSIDS returned '%s', mapped count = %d'\n",
+                  nt_errstr(r.status), r.mapped_count));
+
        if (!NT_STATUS_IS_OK(r.status) &&
-           NT_STATUS_V(r.status) != NT_STATUS_V(STATUS_SOME_UNMAPPED)) {
-         
+           !NT_STATUS_EQUAL(r.status, NT_STATUS_NONE_MAPPED) &&
+           !NT_STATUS_EQUAL(r.status, STATUS_SOME_UNMAPPED))
+       {
                /* An actual error occured */
                result = r.status;
-
                goto done;
        }
 
        /* Return output parameters */
 
-       if (r.mapped_count == 0) {
+       if (NT_STATUS_EQUAL(r.status, NT_STATUS_NONE_MAPPED) ||
+           (r.mapped_count == 0))
+       {
+               for (i = 0; i < num_sids; i++) {
+                       (names)[i] = NULL;
+                       (domains)[i] = NULL;
+                       (types)[i] = SID_NAME_UNKNOWN;
+               }
                result = NT_STATUS_NONE_MAPPED;
                goto done;
        }
 
-       if (!((*domains) = TALLOC_ARRAY(mem_ctx, char *, num_sids))) {
-               DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
-               result = NT_STATUS_UNSUCCESSFUL;
-               goto done;
-       }
-
-       if (!((*names) = TALLOC_ARRAY(mem_ctx, char *, num_sids))) {
-               DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
-               result = NT_STATUS_UNSUCCESSFUL;
-               goto done;
-       }
-
-       if (!((*types) = TALLOC_ARRAY(mem_ctx, uint32, num_sids))) {
-               DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
-               result = NT_STATUS_UNSUCCESSFUL;
-               goto done;
-       }
-               
        for (i = 0; i < num_sids; i++) {
                fstring name, dom_name;
-               uint32 dom_idx = t_names.name[i].domain_idx;
+               uint32 dom_idx = r.names.name[i].domain_idx;
 
                /* Translate optimised name through domain index array */
 
@@ -248,36 +209,156 @@ NTSTATUS rpccli_lsa_lookup_sids(struct rpc_pipe_client *cli,
                        rpcstr_pull_unistr2_fstring(
                                 dom_name, &ref.ref_dom[dom_idx].uni_dom_name);
                        rpcstr_pull_unistr2_fstring(
-                                name, &t_names.uni_name[i]);
+                                name, &r.names.uni_name[i]);
 
-                       (*names)[i] = talloc_strdup(mem_ctx, name);
-                       (*domains)[i] = talloc_strdup(mem_ctx, dom_name);
-                       (*types)[i] = t_names.name[i].sid_name_use;
-                       
-                       if (((*names)[i] == NULL) || ((*domains)[i] == NULL)) {
-                               DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
+                       (names)[i] = talloc_strdup(mem_ctx, name);
+                       (domains)[i] = talloc_strdup(mem_ctx, dom_name);
+                       (types)[i] = r.names.name[i].sid_name_use;
+
+                       if (((names)[i] == NULL) || ((domains)[i] == NULL)) {
+                               DEBUG(0, ("cli_lsa_lookup_sids_noalloc(): out of memory\n"));
                                result = NT_STATUS_UNSUCCESSFUL;
                                goto done;
                        }
 
                } else {
-                       (*names)[i] = NULL;
-                       (*domains)[i] = NULL;
-                       (*types)[i] = SID_NAME_UNKNOWN;
+                       (names)[i] = NULL;
+                       (domains)[i] = NULL;
+                       (types)[i] = SID_NAME_UNKNOWN;
                }
        }
 
- done:
+done:
+       TALLOC_FREE(tmp_ctx);
+       return result;
+}
+
+/* Lookup a list of sids
+ *
+ * do it the right way: there is a limit (of 20480 for w2k3) entries
+ * returned by this call. when the sids list contains more entries,
+ * empty lists are returned. This version of lsa_lookup_sids passes
+ * the list of sids in hunks of LOOKUP_SIDS_HUNK_SIZE to the lsa call. */
+
+/* This constant defines the limit of how many sids to look up
+ * in one call (maximum). the limit from the server side is
+ * at 20480 for win2k3, but we keep it at a save 1000 for now. */
+#define LOOKUP_SIDS_HUNK_SIZE 1000
+
+NTSTATUS rpccli_lsa_lookup_sids(struct rpc_pipe_client *cli,
+                               TALLOC_CTX *mem_ctx,
+                               POLICY_HND *pol,
+                               int num_sids,
+                               const DOM_SID *sids,
+                               char ***domains,
+                               char ***names,
+                               uint32 **types)
+{
+       NTSTATUS result = NT_STATUS_OK;
+       int sids_left = 0;
+       int sids_processed = 0;
+       const DOM_SID *hunk_sids = sids;
+       char **hunk_domains = NULL;
+       char **hunk_names = NULL;
+       uint32 *hunk_types = NULL;
+
+       if (num_sids) {
+               if (!((*domains) = TALLOC_ARRAY(mem_ctx, char *, num_sids))) {
+                       DEBUG(0, ("rpccli_lsa_lookup_sids(): out of memory\n"));
+                       result = NT_STATUS_NO_MEMORY;
+                       goto fail;
+               }
+
+               if (!((*names) = TALLOC_ARRAY(mem_ctx, char *, num_sids))) {
+                       DEBUG(0, ("rpccli_lsa_lookup_sids(): out of memory\n"));
+                       result = NT_STATUS_NO_MEMORY;
+                       goto fail;
+               }
+
+               if (!((*types) = TALLOC_ARRAY(mem_ctx, enum lsa_SidType, num_sids))) {
+                       DEBUG(0, ("rpccli_lsa_lookup_sids(): out of memory\n"));
+                       result = NT_STATUS_NO_MEMORY;
+                       goto fail;
+               }
+       } else {
+               (*domains) = NULL;
+               (*names) = NULL;
+               (*types) = NULL;
+       }
+
+       sids_left = num_sids;
+       hunk_domains = *domains;
+       hunk_names = *names;
+       hunk_types = *types;
+
+       while (sids_left > 0) {
+               int hunk_num_sids;
+               NTSTATUS hunk_result = NT_STATUS_OK;
+
+               hunk_num_sids = ((sids_left > LOOKUP_SIDS_HUNK_SIZE)
+                               ? LOOKUP_SIDS_HUNK_SIZE
+                               : sids_left);
+
+               DEBUG(10, ("rpccli_lsa_lookup_sids: processing items "
+                          "%d -- %d of %d.\n",
+                          sids_processed,
+                          sids_processed + hunk_num_sids - 1,
+                          num_sids));
+
+               hunk_result = rpccli_lsa_lookup_sids_noalloc(cli,
+                                                            mem_ctx,
+                                                            pol,
+                                                            hunk_num_sids,
+                                                            hunk_sids,
+                                                            hunk_domains,
+                                                            hunk_names,
+                                                            hunk_types);
+
+               if (!NT_STATUS_IS_OK(hunk_result) &&
+                   !NT_STATUS_EQUAL(hunk_result, STATUS_SOME_UNMAPPED) &&
+                   !NT_STATUS_EQUAL(hunk_result, NT_STATUS_NONE_MAPPED))
+               {
+                       /* An actual error occured */
+                       result = hunk_result;
+                       goto fail;
+               }
+
+               /* adapt overall result */
+               if (( NT_STATUS_IS_OK(result) &&
+                    !NT_STATUS_IS_OK(hunk_result))
+                   ||
+                   ( NT_STATUS_EQUAL(result, NT_STATUS_NONE_MAPPED) &&
+                    !NT_STATUS_EQUAL(hunk_result, NT_STATUS_NONE_MAPPED)))
+               {
+                       result = STATUS_SOME_UNMAPPED;
+               }
+
+               sids_left -= hunk_num_sids;
+               sids_processed += hunk_num_sids; /* only used in DEBUG */
+               hunk_sids += hunk_num_sids;
+               hunk_domains += hunk_num_sids;
+               hunk_names += hunk_num_sids;
+               hunk_types += hunk_num_sids;
+       }
 
        return result;
+
+fail:
+       TALLOC_FREE(*domains);
+       TALLOC_FREE(*names);
+       TALLOC_FREE(*types);
+       return result;
 }
 
 /** Lookup a list of names */
 
 NTSTATUS rpccli_lsa_lookup_names(struct rpc_pipe_client *cli,
                                 TALLOC_CTX *mem_ctx,
-                                POLICY_HND *pol, int num_names, 
-                                const char **names, DOM_SID **sids, 
+                                POLICY_HND *pol, int num_names,
+                                const char **names,
+                                const char ***dom_names,
+                                int level,
+                                DOM_SID **sids,
                                 uint32 **types)
 {
        prs_struct qbuf, rbuf;
@@ -286,14 +367,14 @@ NTSTATUS rpccli_lsa_lookup_names(struct rpc_pipe_client *cli,
        DOM_R_REF ref;
        NTSTATUS result;
        int i;
-       
+
        ZERO_STRUCT(q);
        ZERO_STRUCT(r);
 
        ZERO_STRUCT(ref);
        r.dom_ref = &ref;
 
-       init_q_lookup_names(mem_ctx, &q, pol, num_names, names);
+       init_q_lookup_names(mem_ctx, &q, pol, num_names, names, level);
 
        CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_LOOKUPNAMES,
                        q, r,
@@ -319,39 +400,64 @@ NTSTATUS rpccli_lsa_lookup_names(struct rpc_pipe_client *cli,
                goto done;
        }
 
-       if (!((*sids = TALLOC_ARRAY(mem_ctx, DOM_SID, num_names)))) {
-               DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
-               result = NT_STATUS_UNSUCCESSFUL;
-               goto done;
-       }
+       if (num_names) {
+               if (!((*sids = TALLOC_ARRAY(mem_ctx, DOM_SID, num_names)))) {
+                       DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
+                       result = NT_STATUS_NO_MEMORY;
+                       goto done;
+               }
 
-       if (!((*types = TALLOC_ARRAY(mem_ctx, uint32, num_names)))) {
-               DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
-               result = NT_STATUS_UNSUCCESSFUL;
-               goto done;
+               if (!((*types = TALLOC_ARRAY(mem_ctx, uint32, num_names)))) {
+                       DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
+                       result = NT_STATUS_NO_MEMORY;
+                       goto done;
+               }
+
+               if (dom_names != NULL) {
+                       *dom_names = TALLOC_ARRAY(mem_ctx, const char *, num_names);
+                       if (*dom_names == NULL) {
+                               DEBUG(0, ("cli_lsa_lookup_sids(): out of memory\n"));
+                               result = NT_STATUS_NO_MEMORY;
+                               goto done;
+                       }
+               }
+       } else {
+               *sids = NULL;
+               *types = NULL;
+               if (dom_names != NULL) {
+                       *dom_names = NULL;
+               }
        }
 
        for (i = 0; i < num_names; i++) {
-               DOM_RID2 *t_rids = r.dom_rid;
+               DOM_RID *t_rids = r.dom_rid;
                uint32 dom_idx = t_rids[i].rid_idx;
                uint32 dom_rid = t_rids[i].rid;
                DOM_SID *sid = &(*sids)[i];
 
                /* Translate optimised sid through domain index array */
 
-               if (dom_idx != 0xffffffff) {
+               if (dom_idx == 0xffffffff) {
+                       /* Nothing to do, this is unknown */
+                       ZERO_STRUCTP(sid);
+                       (*types)[i] = SID_NAME_UNKNOWN;
+                       continue;
+               }
 
-                       sid_copy(sid, &ref.ref_dom[dom_idx].ref_dom.sid);
+               sid_copy(sid, &ref.ref_dom[dom_idx].ref_dom.sid);
 
-                       if (dom_rid != 0xffffffff) {
-                               sid_append_rid(sid, dom_rid);
-                       }
+               if (dom_rid != 0xffffffff) {
+                       sid_append_rid(sid, dom_rid);
+               }
 
-                       (*types)[i] = t_rids[i].type;
-               } else {
-                       ZERO_STRUCTP(sid);
-                       (*types)[i] = SID_NAME_UNKNOWN;
+               (*types)[i] = t_rids[i].type;
+
+               if (dom_names == NULL) {
+                       continue;
                }
+
+               (*dom_names)[i] = rpcstr_pull_unistr2_talloc(
+                       *dom_names, &ref.ref_dom[dom_idx].uni_dom_name);
        }
 
  done:
@@ -359,13 +465,83 @@ NTSTATUS rpccli_lsa_lookup_names(struct rpc_pipe_client *cli,
        return result;
 }
 
+NTSTATUS rpccli_lsa_query_info_policy_new(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
+                                         POLICY_HND *pol, uint16 info_class,
+                                         LSA_INFO_CTR *ctr)
+{
+       prs_struct qbuf, rbuf;
+       LSA_Q_QUERY_INFO q;
+       LSA_R_QUERY_INFO r;
+       NTSTATUS result;
+
+       ZERO_STRUCT(q);
+       ZERO_STRUCT(r);
+
+       init_q_query(&q, pol, info_class);
+
+       CLI_DO_RPC(cli, mem_ctx, PI_LSARPC, LSA_QUERYINFOPOLICY,
+               q, r,
+               qbuf, rbuf,
+               lsa_io_q_query,
+               lsa_io_r_query,
+               NT_STATUS_UNSUCCESSFUL);
+
+       result = r.status;
+
+       if (!NT_STATUS_IS_OK(result)) {
+               goto done;
+       }
+
+ done:
+
+       *ctr = r.ctr;
+
+       return result;
+}
+
+NTSTATUS rpccli_lsa_query_info_policy2_new(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
+                                         POLICY_HND *pol, uint16 info_class,
+                                         LSA_INFO_CTR2 *ctr)
+{
+       prs_struct qbuf, rbuf;
+       LSA_Q_QUERY_INFO2 q;
+       LSA_R_QUERY_INFO2 r;
+       NTSTATUS result;
+
+       ZERO_STRUCT(q);
+       ZERO_STRUCT(r);
+
+       init_q_query2(&q, pol, info_class);
+
+       CLI_DO_RPC(cli, mem_ctx, PI_LSARPC, LSA_QUERYINFO2,
+               q, r,
+               qbuf, rbuf,
+               lsa_io_q_query_info2,
+               lsa_io_r_query_info2,
+               NT_STATUS_UNSUCCESSFUL);
+
+       result = r.status;
+
+       if (!NT_STATUS_IS_OK(result)) {
+               goto done;
+       }
+
+ done:
+
+       *ctr = r.ctr;
+
+       return result;
+}
+
+
+
 /** Query info policy
  *
  *  @param domain_sid - returned remote server's domain sid */
 
 NTSTATUS rpccli_lsa_query_info_policy(struct rpc_pipe_client *cli,
                                      TALLOC_CTX *mem_ctx,
-                                     POLICY_HND *pol, uint16 info_class, 
+                                     POLICY_HND *pol, uint16 info_class,
                                      char **domain_name, DOM_SID **domain_sid)
 {
        prs_struct qbuf, rbuf;
@@ -396,42 +572,50 @@ NTSTATUS rpccli_lsa_query_info_policy(struct rpc_pipe_client *cli,
        switch (info_class) {
 
        case 3:
-               if (domain_name && (r.dom.id3.buffer_dom_name != 0)) {
-                       *domain_name = unistr2_tdup(mem_ctx, 
-                                                  &r.dom.id3.
+               if (domain_name && (r.ctr.info.id3.buffer_dom_name != 0)) {
+                       *domain_name = unistr2_tdup(mem_ctx,
+                                                  &r.ctr.info.id3.
                                                   uni_domain_name);
+                       if (!*domain_name) {
+                               return NT_STATUS_NO_MEMORY;
+                       }
                }
 
-               if (domain_sid && (r.dom.id3.buffer_dom_sid != 0)) {
+               if (domain_sid && (r.ctr.info.id3.buffer_dom_sid != 0)) {
                        *domain_sid = TALLOC_P(mem_ctx, DOM_SID);
-                       if (*domain_sid) {
-                               sid_copy(*domain_sid, &r.dom.id3.dom_sid.sid);
+                       if (!*domain_sid) {
+                               return NT_STATUS_NO_MEMORY;
                        }
+                       sid_copy(*domain_sid, &r.ctr.info.id3.dom_sid.sid);
                }
 
                break;
 
        case 5:
-               
-               if (domain_name && (r.dom.id5.buffer_dom_name != 0)) {
-                       *domain_name = unistr2_tdup(mem_ctx, 
-                                                  &r.dom.id5.
+
+               if (domain_name && (r.ctr.info.id5.buffer_dom_name != 0)) {
+                       *domain_name = unistr2_tdup(mem_ctx,
+                                                  &r.ctr.info.id5.
                                                   uni_domain_name);
+                       if (!*domain_name) {
+                               return NT_STATUS_NO_MEMORY;
+                       }
                }
-                       
-               if (domain_sid && (r.dom.id5.buffer_dom_sid != 0)) {
+
+               if (domain_sid && (r.ctr.info.id5.buffer_dom_sid != 0)) {
                        *domain_sid = TALLOC_P(mem_ctx, DOM_SID);
-                       if (*domain_sid) {
-                               sid_copy(*domain_sid, &r.dom.id5.dom_sid.sid);
+                       if (!*domain_sid) {
+                               return NT_STATUS_NO_MEMORY;
                        }
+                       sid_copy(*domain_sid, &r.ctr.info.id5.dom_sid.sid);
                }
                break;
-                       
+
        default:
                DEBUG(3, ("unknown info class %d\n", info_class));
-               break;                
+               break;
        }
-       
+
  done:
 
        return result;
@@ -447,10 +631,10 @@ NTSTATUS rpccli_lsa_query_info_policy(struct rpc_pipe_client *cli,
 
 NTSTATUS rpccli_lsa_query_info_policy2(struct rpc_pipe_client *cli,
                                       TALLOC_CTX *mem_ctx,
-                                      POLICY_HND *pol, uint16 info_class, 
+                                      POLICY_HND *pol, uint16 info_class,
                                       char **domain_name, char **dns_name,
                                       char **forest_name,
-                                      struct uuid **domain_guid,
+                                      struct GUID **domain_guid,
                                       DOM_SID **domain_sid)
 {
        prs_struct qbuf, rbuf;
@@ -483,42 +667,90 @@ NTSTATUS rpccli_lsa_query_info_policy2(struct rpc_pipe_client *cli,
 
        ZERO_STRUCTP(domain_guid);
 
-       if (domain_name && r.info.dns_dom_info.hdr_nb_dom_name.buffer) {
-               *domain_name = unistr2_tdup(mem_ctx, 
-                                           &r.info.dns_dom_info
+       if (domain_name && r.ctr.info.id12.hdr_nb_dom_name.buffer) {
+               *domain_name = unistr2_tdup(mem_ctx,
+                                           &r.ctr.info.id12
                                            .uni_nb_dom_name);
+               if (!*domain_name) {
+                       return NT_STATUS_NO_MEMORY;
+               }
        }
-       if (dns_name && r.info.dns_dom_info.hdr_dns_dom_name.buffer) {
-               *dns_name = unistr2_tdup(mem_ctx, 
-                                        &r.info.dns_dom_info
+       if (dns_name && r.ctr.info.id12.hdr_dns_dom_name.buffer) {
+               *dns_name = unistr2_tdup(mem_ctx,
+                                        &r.ctr.info.id12
                                         .uni_dns_dom_name);
+               if (!*dns_name) {
+                       return NT_STATUS_NO_MEMORY;
+               }
        }
-       if (forest_name && r.info.dns_dom_info.hdr_forest_name.buffer) {
-               *forest_name = unistr2_tdup(mem_ctx, 
-                                           &r.info.dns_dom_info
+       if (forest_name && r.ctr.info.id12.hdr_forest_name.buffer) {
+               *forest_name = unistr2_tdup(mem_ctx,
+                                           &r.ctr.info.id12
                                            .uni_forest_name);
+               if (!*forest_name) {
+                       return NT_STATUS_NO_MEMORY;
+               }
        }
-       
+
        if (domain_guid) {
-               *domain_guid = TALLOC_P(mem_ctx, struct uuid);
-               memcpy(*domain_guid, 
-                      &r.info.dns_dom_info.dom_guid, 
-                      sizeof(struct uuid));
+               *domain_guid = TALLOC_P(mem_ctx, struct GUID);
+               if (!*domain_guid) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+               memcpy(*domain_guid,
+                      &r.ctr.info.id12.dom_guid,
+                      sizeof(struct GUID));
        }
 
-       if (domain_sid && r.info.dns_dom_info.ptr_dom_sid != 0) {
+       if (domain_sid && r.ctr.info.id12.ptr_dom_sid != 0) {
                *domain_sid = TALLOC_P(mem_ctx, DOM_SID);
-               if (*domain_sid) {
-                       sid_copy(*domain_sid, 
-                                &r.info.dns_dom_info.dom_sid.sid);
+               if (!*domain_sid) {
+                       return NT_STATUS_NO_MEMORY;
                }
+               sid_copy(*domain_sid,
+                        &r.ctr.info.id12.dom_sid.sid);
        }
-       
+
  done:
 
        return result;
 }
 
+NTSTATUS rpccli_lsa_set_info_policy(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
+                                   POLICY_HND *pol, uint16 info_class,
+                                   LSA_INFO_CTR ctr)
+{
+       prs_struct qbuf, rbuf;
+       LSA_Q_SET_INFO q;
+       LSA_R_SET_INFO r;
+       NTSTATUS result;
+
+       ZERO_STRUCT(q);
+       ZERO_STRUCT(r);
+
+       init_q_set(&q, pol, info_class, ctr);
+
+       CLI_DO_RPC(cli, mem_ctx, PI_LSARPC, LSA_SETINFOPOLICY,
+               q, r,
+               qbuf, rbuf,
+               lsa_io_q_set,
+               lsa_io_r_set,
+               NT_STATUS_UNSUCCESSFUL);
+
+       result = r.status;
+
+       if (!NT_STATUS_IS_OK(result)) {
+               goto done;
+       }
+
+       /* Return output parameters */
+
+ done:
+
+       return result;
+}
+
+
 /**
  * Enumerate list of trusted domains
  *
@@ -536,7 +768,7 @@ NTSTATUS rpccli_lsa_query_info_policy2(struct rpc_pipe_client *cli,
 
 NTSTATUS rpccli_lsa_enum_trust_dom(struct rpc_pipe_client *cli,
                                   TALLOC_CTX *mem_ctx,
-                                  POLICY_HND *pol, uint32 *enum_ctx, 
+                                  POLICY_HND *pol, uint32 *enum_ctx,
                                   uint32 *num_domains,
                                   char ***domain_names, DOM_SID **domain_sids)
 {
@@ -550,31 +782,31 @@ NTSTATUS rpccli_lsa_enum_trust_dom(struct rpc_pipe_client *cli,
        ZERO_STRUCT(out);
 
        /* 64k is enough for about 2000 trusted domains */
-       
+
         init_q_enum_trust_dom(&in, pol, *enum_ctx, 0x10000);
 
-       CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_ENUMTRUSTDOM, 
-                   in, out, 
+       CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_ENUMTRUSTDOM,
+                   in, out,
                    qbuf, rbuf,
                    lsa_io_q_enum_trust_dom,
-                   lsa_io_r_enum_trust_dom, 
+                   lsa_io_r_enum_trust_dom,
                    NT_STATUS_UNSUCCESSFUL );
 
 
        /* check for an actual error */
 
-       if ( !NT_STATUS_IS_OK(out.status) 
-               && !NT_STATUS_EQUAL(out.status, NT_STATUS_NO_MORE_ENTRIES) 
+       if ( !NT_STATUS_IS_OK(out.status)
+               && !NT_STATUS_EQUAL(out.status, NT_STATUS_NO_MORE_ENTRIES)
                && !NT_STATUS_EQUAL(out.status, STATUS_MORE_ENTRIES) )
        {
                return out.status;
        }
-               
+
        /* Return output parameters */
 
        *num_domains  = out.count;
        *enum_ctx     = out.enum_context;
-       
+
        if ( out.count ) {
 
                /* Allocate memory for trusted domain names and sids */
@@ -593,7 +825,7 @@ NTSTATUS rpccli_lsa_enum_trust_dom(struct rpc_pipe_client *cli,
 
                for (i = 0; i < out.count; i++) {
 
-                       rpcstr_pull( tmp, out.domlist->domains[i].name.string->buffer, 
+                       rpcstr_pull( tmp, out.domlist->domains[i].name.string->buffer,
                                sizeof(tmp), out.domlist->domains[i].name.length, 0);
                        (*domain_names)[i] = talloc_strdup(mem_ctx, tmp);
 
@@ -639,22 +871,28 @@ NTSTATUS rpccli_lsa_enum_privilege(struct rpc_pipe_client *cli, TALLOC_CTX *mem_
        *enum_context = r.enum_context;
        *count = r.count;
 
-       if (!((*privs_name = TALLOC_ARRAY(mem_ctx, char *, r.count)))) {
-               DEBUG(0, ("(cli_lsa_enum_privilege): out of memory\n"));
-               result = NT_STATUS_UNSUCCESSFUL;
-               goto done;
-       }
+       if (r.count) {
+               if (!((*privs_name = TALLOC_ARRAY(mem_ctx, char *, r.count)))) {
+                       DEBUG(0, ("(cli_lsa_enum_privilege): out of memory\n"));
+                       result = NT_STATUS_UNSUCCESSFUL;
+                       goto done;
+               }
 
-       if (!((*privs_high = TALLOC_ARRAY(mem_ctx, uint32, r.count)))) {
-               DEBUG(0, ("(cli_lsa_enum_privilege): out of memory\n"));
-               result = NT_STATUS_UNSUCCESSFUL;
-               goto done;
-       }
+               if (!((*privs_high = TALLOC_ARRAY(mem_ctx, uint32, r.count)))) {
+                       DEBUG(0, ("(cli_lsa_enum_privilege): out of memory\n"));
+                       result = NT_STATUS_UNSUCCESSFUL;
+                       goto done;
+               }
 
-       if (!((*privs_low = TALLOC_ARRAY(mem_ctx, uint32, r.count)))) {
-               DEBUG(0, ("(cli_lsa_enum_privilege): out of memory\n"));
-               result = NT_STATUS_UNSUCCESSFUL;
-               goto done;
+               if (!((*privs_low = TALLOC_ARRAY(mem_ctx, uint32, r.count)))) {
+                       DEBUG(0, ("(cli_lsa_enum_privilege): out of memory\n"));
+                       result = NT_STATUS_UNSUCCESSFUL;
+                       goto done;
+               }
+       } else {
+               *privs_name = NULL;
+               *privs_high = NULL;
+               *privs_low = NULL;
        }
 
        for (i = 0; i < r.count; i++) {
@@ -676,7 +914,7 @@ NTSTATUS rpccli_lsa_enum_privilege(struct rpc_pipe_client *cli, TALLOC_CTX *mem_
 /** Get privilege name */
 
 NTSTATUS rpccli_lsa_get_dispname(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
-                             POLICY_HND *pol, const char *name, 
+                             POLICY_HND *pol, const char *name,
                              uint16 lang_id, uint16 lang_id_sys,
                              fstring description, uint16 *lang_id_desc)
 {
@@ -704,7 +942,7 @@ NTSTATUS rpccli_lsa_get_dispname(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ct
        }
 
        /* Return output parameters */
-       
+
        rpcstr_pull_unistr2_fstring(description , &r.desc);
        *lang_id_desc = r.lang_id;
 
@@ -716,7 +954,7 @@ NTSTATUS rpccli_lsa_get_dispname(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ct
 /** Enumerate list of SIDs  */
 
 NTSTATUS rpccli_lsa_enum_sids(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
-                                POLICY_HND *pol, uint32 *enum_ctx, uint32 pref_max_length, 
+                                POLICY_HND *pol, uint32 *enum_ctx, uint32 pref_max_length,
                                 uint32 *num_sids, DOM_SID **sids)
 {
        prs_struct qbuf, rbuf;
@@ -779,7 +1017,7 @@ NTSTATUS rpccli_lsa_enum_sids(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
  * */
 
 NTSTATUS rpccli_lsa_create_account(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
-                             POLICY_HND *dom_pol, DOM_SID *sid, uint32 desired_access, 
+                             POLICY_HND *dom_pol, DOM_SID *sid, uint32 desired_access,
                             POLICY_HND *user_pol)
 {
        prs_struct qbuf, rbuf;
@@ -817,7 +1055,7 @@ NTSTATUS rpccli_lsa_create_account(struct rpc_pipe_client *cli, TALLOC_CTX *mem_
  * @param cli Handle on an initialised SMB connection */
 
 NTSTATUS rpccli_lsa_open_account(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
-                             POLICY_HND *dom_pol, DOM_SID *sid, uint32 des_access, 
+                             POLICY_HND *dom_pol, DOM_SID *sid, uint32 des_access,
                             POLICY_HND *user_pol)
 {
        prs_struct qbuf, rbuf;
@@ -949,7 +1187,7 @@ NTSTATUS rpccli_lsa_lookup_priv_value(struct rpc_pipe_client *cli, TALLOC_CTX *m
 /** Query LSA security object */
 
 NTSTATUS rpccli_lsa_query_secobj(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
-                             POLICY_HND *pol, uint32 sec_info, 
+                             POLICY_HND *pol, uint32 sec_info,
                              SEC_DESC_BUF **psdb)
 {
        prs_struct qbuf, rbuf;
@@ -1028,10 +1266,16 @@ NTSTATUS rpccli_lsa_enum_account_rights(struct rpc_pipe_client *cli, TALLOC_CTX
                goto done;
        }
 
-       
+
        privileges = TALLOC_ARRAY( mem_ctx, fstring, *count );
        names      = TALLOC_ARRAY( mem_ctx, char *, *count );
 
+       if ((privileges == NULL) || (names == NULL)) {
+               TALLOC_FREE(privileges);
+               TALLOC_FREE(names);
+               return NT_STATUS_NO_MEMORY;
+       }
+
        for ( i=0; i<*count; i++ ) {
                UNISTR4 *uni_string = &r.rights->strings[i];
 
@@ -1039,11 +1283,11 @@ NTSTATUS rpccli_lsa_enum_account_rights(struct rpc_pipe_client *cli, TALLOC_CTX
                        continue;
 
                rpcstr_pull( privileges[i], uni_string->string->buffer, sizeof(privileges[i]), -1, STR_TERMINATE );
-                       
+
                /* now copy to the return array */
                names[i] = talloc_strdup( mem_ctx, privileges[i] );
        }
-       
+
        *priv_names = names;
 
 done:
@@ -1135,18 +1379,18 @@ BOOL fetch_domain_sid( char *domain, char *remote_machine, DOM_SID *psid)
        NTSTATUS result;
        POLICY_HND lsa_pol;
        BOOL ret = False;
+
        ZERO_STRUCT(cli);
        if(cli_initialise(&cli) == False) {
                DEBUG(0,("fetch_domain_sid: unable to initialize client connection.\n"));
                return False;
        }
+
        if(!resolve_name( remote_machine, &cli.dest_ip, 0x20)) {
                DEBUG(0,("fetch_domain_sid: Can't resolve address for %s\n", remote_machine));
                goto done;
        }
+
        if (!cli_connect(&cli, remote_machine, &cli.dest_ip)) {
                DEBUG(0,("fetch_domain_sid: unable to connect to SMB server on \
 machine %s. Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
@@ -1154,35 +1398,35 @@ machine %s. Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
        }
 
        if (!attempt_netbios_session_request(&cli, global_myname, remote_machine, &cli.dest_ip)) {
-               DEBUG(0,("fetch_domain_sid: machine %s rejected the NetBIOS session request.\n", 
+               DEBUG(0,("fetch_domain_sid: machine %s rejected the NetBIOS session request.\n",
                        remote_machine));
                goto done;
        }
+
        cli.protocol = PROTOCOL_NT1;
+
        if (!cli_negprot(&cli)) {
                DEBUG(0,("fetch_domain_sid: machine %s rejected the negotiate protocol. \
 Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
                goto done;
        }
+
        if (cli.protocol != PROTOCOL_NT1) {
                DEBUG(0,("fetch_domain_sid: machine %s didn't negotiate NT protocol.\n",
                        remote_machine));
                goto done;
        }
+
        /*
         * Do an anonymous session setup.
         */
+
        if (!cli_session_setup(&cli, "", "", 0, "", 0, "")) {
                DEBUG(0,("fetch_domain_sid: machine %s rejected the session setup. \
 Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
                goto done;
        }
+
        if (!(cli.sec_mode & NEGOTIATE_SECURITY_USER_LEVEL)) {
                DEBUG(0,("fetch_domain_sid: machine %s isn't in user level security mode\n",
                        remote_machine));
@@ -1196,26 +1440,26 @@ Error was : %s.\n", remote_machine, cli_errstr(&cli) ));
        }
 
        /* Fetch domain sid */
+
        if (!cli_nt_session_open(&cli, PI_LSARPC)) {
                DEBUG(0, ("fetch_domain_sid: Error connecting to SAM pipe\n"));
                goto done;
        }
+
        result = cli_lsa_open_policy(&cli, cli.mem_ctx, True, SEC_RIGHTS_QUERY_VALUE, &lsa_pol);
        if (!NT_STATUS_IS_OK(result)) {
                DEBUG(0, ("fetch_domain_sid: Error opening lsa policy handle. %s\n",
                        nt_errstr(result) ));
                goto done;
        }
+
        result = cli_lsa_query_info_policy(&cli, cli.mem_ctx, &lsa_pol, 5, domain, psid);
        if (!NT_STATUS_IS_OK(result)) {
                DEBUG(0, ("fetch_domain_sid: Error querying lsa policy handle. %s\n",
                        nt_errstr(result) ));
                goto done;
        }
+
        ret = True;
 
   done:
@@ -1252,7 +1496,7 @@ NTSTATUS rpccli_lsa_open_trusted_domain(struct rpc_pipe_client *cli, TALLOC_CTX
                NT_STATUS_UNSUCCESSFUL);
 
        /* Return output parameters */
-       
+
        result = r.status;
 
        if (NT_STATUS_IS_OK(result)) {
@@ -1263,8 +1507,8 @@ NTSTATUS rpccli_lsa_open_trusted_domain(struct rpc_pipe_client *cli, TALLOC_CTX
 }
 
 NTSTATUS rpccli_lsa_query_trusted_domain_info(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
-                                          POLICY_HND *pol, 
-                                          uint16 info_class,  
+                                          POLICY_HND *pol,
+                                          uint16 info_class,
                                           LSA_TRUSTED_DOMAIN_INFO **info)
 {
        prs_struct qbuf, rbuf;
@@ -1277,7 +1521,7 @@ NTSTATUS rpccli_lsa_query_trusted_domain_info(struct rpc_pipe_client *cli, TALLO
 
        /* Marshall data and send request */
 
-       init_q_query_trusted_domain_info(&q, pol, info_class); 
+       init_q_query_trusted_domain_info(&q, pol, info_class);
 
        CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_QUERYTRUSTDOMINFO,
                q, r,
@@ -1293,15 +1537,51 @@ NTSTATUS rpccli_lsa_query_trusted_domain_info(struct rpc_pipe_client *cli, TALLO
        }
 
        *info = r.info;
-               
+
 done:
        return result;
 }
 
+NTSTATUS rpccli_lsa_open_trusted_domain_by_name(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
+                                               POLICY_HND *pol, const char *name, uint32 access_mask,
+                                               POLICY_HND *trustdom_pol)
+{
+       prs_struct qbuf, rbuf;
+       LSA_Q_OPEN_TRUSTED_DOMAIN_BY_NAME q;
+       LSA_R_OPEN_TRUSTED_DOMAIN_BY_NAME r;
+       NTSTATUS result;
+
+       ZERO_STRUCT(q);
+       ZERO_STRUCT(r);
+
+       /* Initialise input parameters */
+
+       init_lsa_q_open_trusted_domain_by_name(&q, pol, name, access_mask);
+
+       /* Marshall data and send request */
+
+       CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_OPENTRUSTDOMBYNAME,
+               q, r,
+               qbuf, rbuf,
+               lsa_io_q_open_trusted_domain_by_name,
+               lsa_io_r_open_trusted_domain_by_name,
+               NT_STATUS_UNSUCCESSFUL);
+
+       /* Return output parameters */
+
+       result = r.status;
+
+       if (NT_STATUS_IS_OK(result)) {
+               *trustdom_pol = r.handle;
+       }
+
+       return result;
+}
+
 
 NTSTATUS rpccli_lsa_query_trusted_domain_info_by_sid(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
-                                                 POLICY_HND *pol, 
-                                                 uint16 info_class, DOM_SID *dom_sid, 
+                                                 POLICY_HND *pol,
+                                                 uint16 info_class, DOM_SID *dom_sid,
                                                  LSA_TRUSTED_DOMAIN_INFO **info)
 {
        prs_struct qbuf, rbuf;
@@ -1314,7 +1594,7 @@ NTSTATUS rpccli_lsa_query_trusted_domain_info_by_sid(struct rpc_pipe_client *cli
 
        /* Marshall data and send request */
 
-       init_q_query_trusted_domain_info_by_sid(&q, pol, info_class, dom_sid); 
+       init_q_query_trusted_domain_info_by_sid(&q, pol, info_class, dom_sid);
 
        CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_QUERYTRUSTDOMINFOBYSID,
                q, r,
@@ -1337,8 +1617,8 @@ done:
 }
 
 NTSTATUS rpccli_lsa_query_trusted_domain_info_by_name(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
-                                                  POLICY_HND *pol, 
-                                                  uint16 info_class, const char *domain_name, 
+                                                  POLICY_HND *pol,
+                                                  uint16 info_class, const char *domain_name,
                                                   LSA_TRUSTED_DOMAIN_INFO **info)
 {
        prs_struct qbuf, rbuf;
@@ -1351,7 +1631,7 @@ NTSTATUS rpccli_lsa_query_trusted_domain_info_by_name(struct rpc_pipe_client *cl
 
        /* Marshall data and send request */
 
-       init_q_query_trusted_domain_info_by_name(&q, pol, info_class, domain_name); 
+       init_q_query_trusted_domain_info_by_name(&q, pol, info_class, domain_name);
 
        CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_QUERYTRUSTDOMINFOBYNAME,
                q, r,
@@ -1369,6 +1649,42 @@ NTSTATUS rpccli_lsa_query_trusted_domain_info_by_name(struct rpc_pipe_client *cl
        *info = r.info;
 
 done:
-       
+
+       return result;
+}
+
+NTSTATUS cli_lsa_query_domain_info_policy(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
+                                         POLICY_HND *pol,
+                                         uint16 info_class, LSA_DOM_INFO_UNION **info)
+{
+       prs_struct qbuf, rbuf;
+       LSA_Q_QUERY_DOM_INFO_POLICY q;
+       LSA_R_QUERY_DOM_INFO_POLICY r;
+       NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
+
+       ZERO_STRUCT(q);
+       ZERO_STRUCT(r);
+
+       /* Marshall data and send request */
+
+       init_q_query_dom_info(&q, pol, info_class);
+
+       CLI_DO_RPC( cli, mem_ctx, PI_LSARPC, LSA_QUERYDOMINFOPOL,
+               q, r,
+               qbuf, rbuf,
+               lsa_io_q_query_dom_info,
+               lsa_io_r_query_dom_info,
+               NT_STATUS_UNSUCCESSFUL);
+
+       result = r.status;
+
+       if (!NT_STATUS_IS_OK(result)) {
+               goto done;
+       }
+
+       *info = r.info;
+
+done:
        return result;
 }
+