/*

   Unix SMB/CIFS implementation.

   ACL get/set operations

   Copyright (C) Andrew Tridgell 2003-2004

   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/ndr_security.h"

/****************************************************************************

fetch file ACL (async send)

****************************************************************************/

struct smbcli_request *smb_raw_query_secdesc_send(struct smbcli_tree *tree,

              union smb_fileinfo *io)

{

  struct smb_nttrans nt;

  uint8_t params[8];

  nt.in.max_setup = 0;

  nt.in.max_param = 4;

  nt.in.max_data = 0xFFFF;

  nt.in.setup_count = 0;

  nt.in.function = NT_TRANSACT_QUERY_SECURITY_DESC;

  nt.in.setup = NULL;

  SSVAL(params, 0, io->query_secdesc.in.file.fnum);

  SSVAL(params, 2, 0); /* padding */

  SIVAL(params, 4, io->query_secdesc.in.secinfo_flags);

  nt.in.params.data = params;

  nt.in.params.length = 8;

  nt.in.data = data_blob(NULL, 0);

  return smb_raw_nttrans_send(tree, &nt);

}

/****************************************************************************

fetch file ACL (async recv)

****************************************************************************/

NTSTATUS smb_raw_query_secdesc_recv(struct smbcli_request *req,

            TALLOC_CTX *mem_ctx,

            union smb_fileinfo *io)

{

  NTSTATUS status;

  struct smb_nttrans nt;

  struct ndr_pull *ndr;

  enum ndr_err_code ndr_err;

  status = smb_raw_nttrans_recv(req, mem_ctx, &nt);

  if (!NT_STATUS_IS_OK(status)) {

    return status;

  }

  /* check that the basics are valid */

  if (nt.out.params.length != 4 ||

      IVAL(nt.out.params.data, 0) > nt.out.data.length) {

    return NT_STATUS_INVALID_PARAMETER;

  }

  nt.out.data.length = IVAL(nt.out.params.data, 0);

  ndr = ndr_pull_init_blob(&nt.out.data, mem_ctx);

  if (!ndr) {

    return NT_STATUS_INVALID_PARAMETER;

  }

  io->query_secdesc.out.sd = talloc(mem_ctx, struct security_descriptor);

  if (!io->query_secdesc.out.sd) {

    return NT_STATUS_NO_MEMORY;

  }

  ndr_err = ndr_pull_security_descriptor(ndr, NDR_SCALARS|NDR_BUFFERS,

                 io->query_secdesc.out.sd);

  if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {

    return ndr_map_error2ntstatus(ndr_err);

  }

  return NT_STATUS_OK;

}

/****************************************************************************

fetch file ACL (sync interface)

****************************************************************************/

NTSTATUS smb_raw_query_secdesc(struct smbcli_tree *tree,

             TALLOC_CTX *mem_ctx,

             union smb_fileinfo *io)

{

  struct smbcli_request *req = smb_raw_query_secdesc_send(tree, io);

  return smb_raw_query_secdesc_recv(req, mem_ctx, io);

}

/****************************************************************************

set file ACL (async send)

****************************************************************************/

struct smbcli_request *smb_raw_set_secdesc_send(struct smbcli_tree *tree,

            union smb_setfileinfo *io)

{

  struct smb_nttrans nt;

  uint8_t params[8];

  struct ndr_push *ndr;

  struct smbcli_request *req;

  enum ndr_err_code ndr_err;

  nt.in.max_setup = 0;

  nt.in.max_param = 0;

  nt.in.max_data = 0;

  nt.in.setup_count = 0;

  nt.in.function = NT_TRANSACT_SET_SECURITY_DESC;

  nt.in.setup = NULL;

  SSVAL(params, 0, io->set_secdesc.in.file.fnum);

  SSVAL(params, 2, 0); /* padding */

  SIVAL(params, 4, io->set_secdesc.in.secinfo_flags);

  nt.in.params.data = params;

  nt.in.params.length = 8;

  ndr = ndr_push_init_ctx(NULL);

  if (!ndr) return NULL;

  ndr_err = ndr_push_security_descriptor(ndr, NDR_SCALARS|NDR_BUFFERS, io->set_secdesc.in.sd);

  if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {

    talloc_free(ndr);

    return NULL;

  }

  nt.in.data = ndr_push_blob(ndr);

  req = smb_raw_nttrans_send(tree, &nt);

  talloc_free(ndr);

  return req;

}

/****************************************************************************

set file ACL (sync interface)

****************************************************************************/

NTSTATUS smb_raw_set_secdesc(struct smbcli_tree *tree,

           union smb_setfileinfo *io)

{

  struct smbcli_request *req = smb_raw_set_secdesc_send(tree, io);

  return smbcli_request_simple_recv(req);

}