/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2017 Nexenta Systems, Inc.  All rights reserved.
 */

/*
 * Dispatch function for SMB2_SET_INFO
 */

#include <smbsrv/smb2_kproto.h>
#include <smbsrv/smb_fsops.h>
#include <smbsrv/ntifs.h>

smb_sdrc_t
smb2_set_info(smb_request_t *sr)
{
        smb_setinfo_t sinfo;
        uint16_t StructSize;
        uint16_t iBufOffset;
        uint32_t iBufLength;
        uint32_t AddlInfo;
        smb2fid_t smb2fid;
        uint32_t status;
        uint8_t InfoType, InfoClass;
        int rc = 0;

        bzero(&sinfo, sizeof (sinfo));

        /*
         * Decode SMB2 Set Info request
         */
        rc = smb_mbc_decodef(
            &sr->smb_data, "wbblw..lqq",
            &StructSize,                /* w */
            &InfoType,                  /* b */
            &InfoClass,                 /* b */
            &iBufLength,                /* l */
            &iBufOffset,                /* w */
            /* reserved                   .. */
            &AddlInfo,                  /* l */
            &smb2fid.persistent,        /* q */
            &smb2fid.temporal);         /* q */
        if (rc || StructSize != 33)
                return (SDRC_ERROR);

        /*
         * If there's an input buffer, setup a shadow.
         */
        if (iBufLength) {
                rc = MBC_SHADOW_CHAIN(&sinfo.si_data, &sr->smb_data,
                    sr->smb2_cmd_hdr + iBufOffset, iBufLength);
                if (rc) {
                        return (SDRC_ERROR);
                }
        }

        /* No output data. */
        sr->raw_data.max_bytes = 0;

        status = smb2sr_lookup_fid(sr, &smb2fid);
        DTRACE_SMB2_START(op__SetInfo, smb_request_t *, sr);

        if (status)
                goto errout;

        if (iBufLength > smb2_max_trans) {
                status = NT_STATUS_INVALID_PARAMETER;
                goto errout;
        }

        sinfo.si_node = sr->fid_ofile->f_node;
        sr->user_cr = sr->fid_ofile->f_cr;

        switch (InfoType) {
        case SMB2_0_INFO_FILE:
                status = smb2_setinfo_file(sr, &sinfo, InfoClass);
                break;
        case SMB2_0_INFO_FILESYSTEM:
                status = smb2_setinfo_fs(sr, &sinfo, InfoClass);
                break;
        case SMB2_0_INFO_SECURITY:
                status = smb2_setinfo_sec(sr, &sinfo, AddlInfo);
                break;
        case SMB2_0_INFO_QUOTA:
                status = smb2_setinfo_quota(sr, &sinfo);
                break;
        default:
                status = NT_STATUS_INVALID_PARAMETER;
                break;
        }

errout:
        sr->smb2_status = status;
        DTRACE_SMB2_DONE(op__SetInfo, smb_request_t *, sr);

        if (status) {
                smb2sr_put_error(sr, status);
                return (SDRC_SUCCESS);
        }

        /*
         * SMB2 Query Info reply
         */
        (void) smb_mbc_encodef(
            &sr->reply, "w..",
            2); /* StructSize */        /* w */

        return (SDRC_SUCCESS);
}