/*
 * 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.
 */

/*
 * Copyright 2020 RackTop Systems, Inc.
 */

/*
 * Dispatch function for SMB2_QUERY_INFO
 */

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

smb_sdrc_t
smb2_query_info(smb_request_t *sr)
{
        smb_queryinfo_t *qi;
        uint16_t StructSize;
        uint32_t oBufLength;
        uint16_t iBufOffset;
        uint32_t iBufLength;
        smb2fid_t smb2fid;
        uint16_t DataOff;
        uint32_t status;
        int rc = 0;

        qi = smb_srm_zalloc(sr, sizeof (*qi));

        /*
         * SMB2 Query Info request
         */
        rc = smb_mbc_decodef(
            &sr->smb_data, "wbblw..lllqq",
            &StructSize,                /* w */
            &qi->qi_InfoType,           /* b */
            &qi->qi_InfoClass,          /* b */
            &oBufLength,                /* l */
            &iBufOffset,                /* w */
            /* reserved                   .. */
            &iBufLength,                /* l */
            &qi->qi_AddlInfo,           /* l */
            &qi->qi_Flags,              /* l */
            &smb2fid.persistent,        /* q */
            &smb2fid.temporal);         /* q */
        if (rc || StructSize != 41)
                return (SDRC_ERROR);

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

        if (oBufLength > smb2_max_trans)
                oBufLength = smb2_max_trans;
        sr->raw_data.max_bytes = oBufLength;

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

        if (status)
                goto errout;

        switch (qi->qi_InfoType) {
        case SMB2_0_INFO_FILE:
                status = smb2_qinfo_file(sr, qi);
                break;
        case SMB2_0_INFO_FILESYSTEM:
                status = smb2_qinfo_fs(sr, qi);
                break;
        case SMB2_0_INFO_SECURITY:
                status = smb2_qinfo_sec(sr, qi);
                break;
        case SMB2_0_INFO_QUOTA:
                status = smb2_qinfo_quota(sr, qi);
                break;
        default:
                status = NT_STATUS_INVALID_PARAMETER;
                break;
        }

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

        switch (status) {

        case 0: /* success */
                break;

        case NT_STATUS_BUFFER_OVERFLOW:
                /* Not really an error, per se.  Advisory. */
                break;

        case NT_STATUS_BUFFER_TOO_SMALL:        /* only in smb2_qinfo_sec.c */
                /*
                 * [MS-SMB2] 3.3.5.20.3
                 * Handling SMB2_0_INFO_SECURITY
                 *  If dialect 3.1.1 must return 4-byte value
                 *  containing required buffer size.
                 *  ByteCount==12, ErrorContextCount==1,
                 *  ErrorData: ErrorDataLength==4,ErrorId==0
                 *  ErrorContextData==<buffer size>
                 *  Otherwise ByteCount==4
                 *
                 * When returning with data, 3.1.1 encapsulate.
                 */
                if (sr->session->dialect < SMB_VERS_3_11) {
                        smb2sr_put_error_data(sr, status, &sr->raw_data);
                } else {
                        smb2sr_put_error_ctx0(sr, status, &sr->raw_data);
                }
                return (SDRC_SUCCESS);

        case NT_STATUS_INFO_LENGTH_MISMATCH: /* there is no in smb2_qinfo_*.c */
                /*
                 * [MS-SMB2] 3.3.5.20.1
                 * SMB 3.1.1 Handling SMB2_0_INFO_FILE
                 * [MS-SMB2] 3.3.5.20.2
                 * SMB 3.1.1 Handling SMB2_0_INFO_FILESYSTEM
                 *
                 *  ByteCount==8, ErrorContextCount==1,
                 *  ErrorData: ErrorDataLength==0,ErrorId==0
                 *  Otherwise ByteCount==0
                 */
                if (sr->session->dialect < SMB_VERS_3_11) {
                        smb2sr_put_error_data(sr, status, NULL);
                } else {
                        smb2sr_put_error_ctx0(sr, status, NULL);
                }
                return (SDRC_SUCCESS);

        default:
                smb2sr_put_error(sr, status);
                return (SDRC_SUCCESS);
        }

        /*
         * SMB2 Query Info reply
         */
        DataOff = SMB2_HDR_SIZE + 8;
        oBufLength = MBC_LENGTH(&sr->raw_data);
        rc = smb_mbc_encodef(
            &sr->reply, "wwlC",
            9,  /* StructSize */        /* w */
            DataOff,                    /* w */
            oBufLength,                 /* l */
            &sr->raw_data);             /* C */
        if (rc)
                sr->smb2_status = NT_STATUS_INTERNAL_ERROR;

        return (SDRC_SUCCESS);
}