/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
 * Copyright 2017 Nexenta Systems, Inc.  All rights reserved.
 */

#include <smbsrv/smb_kproto.h>

#define SMB_CREATE_NAMEBUF_SZ   16

/*
 * Create a new file, or truncate an existing file to zero length,
 * open the file and return a fid.  The file is specified using a
 * fully qualified name relative to the tree.
 */
smb_sdrc_t
smb_pre_create(smb_request_t *sr)
{
        struct open_param *op = &sr->arg.open;
        int rc;

        bzero(op, sizeof (sr->arg.open));

        rc = smbsr_decode_vwv(sr, "wl", &op->dattr, &op->mtime.tv_sec);
        if (rc == 0)
                rc = smbsr_decode_data(sr, "%S", sr, &op->fqi.fq_path.pn_path);

        op->create_disposition = FILE_OVERWRITE_IF;
        op->create_options = FILE_NON_DIRECTORY_FILE;

        DTRACE_SMB_START(op__Create, smb_request_t *, sr); /* arg.open */

        return ((rc == 0) ? SDRC_SUCCESS : SDRC_ERROR);
}

void
smb_post_create(smb_request_t *sr)
{
        DTRACE_SMB_DONE(op__Create, smb_request_t *, sr);
}

smb_sdrc_t
smb_com_create(smb_request_t *sr)
{
        if (smb_common_create(sr) != NT_STATUS_SUCCESS)
                return (SDRC_ERROR);

        if (smbsr_encode_result(sr, 1, 0, "bww", 1, sr->smb_fid, 0))
                return (SDRC_ERROR);

        return (SDRC_SUCCESS);
}

/*
 * Create a new file and return a fid.  The file is specified using
 * a fully qualified name relative to the tree.
 */
smb_sdrc_t
smb_pre_create_new(smb_request_t *sr)
{
        struct open_param *op = &sr->arg.open;
        int rc;

        bzero(op, sizeof (sr->arg.open));

        rc = smbsr_decode_vwv(sr, "wl", &op->dattr, &op->mtime.tv_sec);
        if (rc == 0)
                rc = smbsr_decode_data(sr, "%S", sr, &op->fqi.fq_path.pn_path);

        op->create_disposition = FILE_CREATE;

        DTRACE_SMB_START(op__CreateNew, smb_request_t *, sr); /* arg.open */

        return ((rc == 0) ? SDRC_SUCCESS : SDRC_ERROR);
}

void
smb_post_create_new(smb_request_t *sr)
{
        DTRACE_SMB_DONE(op__CreateNew, smb_request_t *, sr);
}

smb_sdrc_t
smb_com_create_new(smb_request_t *sr)
{
        if (smb_common_create(sr) != NT_STATUS_SUCCESS)
                return (SDRC_ERROR);

        if (smbsr_encode_result(sr, 1, 0, "bww", 1, sr->smb_fid, 0))
                return (SDRC_ERROR);

        return (SDRC_SUCCESS);
}

/*
 * Create a unique file in the specified directory relative to the
 * current tree.  No attributes are specified.
 */
smb_sdrc_t
smb_pre_create_temporary(smb_request_t *sr)
{
        struct open_param *op = &sr->arg.open;
        uint16_t reserved;
        int rc;

        bzero(op, sizeof (sr->arg.open));

        rc = smbsr_decode_vwv(sr, "wl", &reserved, &op->mtime.tv_sec);
        if (rc == 0)
                rc = smbsr_decode_data(sr, "%S", sr, &op->fqi.fq_path.pn_path);

        op->create_disposition = FILE_CREATE;

        DTRACE_SMB_START(op__CreateTemporary, smb_request_t *, sr);

        return ((rc == 0) ? SDRC_SUCCESS : SDRC_ERROR);
}

void
smb_post_create_temporary(smb_request_t *sr)
{
        DTRACE_SMB_DONE(op__CreateTemporary, smb_request_t *, sr);
}

smb_sdrc_t
smb_com_create_temporary(smb_request_t *sr)
{
        static uint16_t tmp_id = 10000;
        struct open_param *op = &sr->arg.open;
        char name[SMB_CREATE_NAMEBUF_SZ];
        char *buf;
        uint16_t bcc;

        ++tmp_id;
        bcc = 1; /* null terminator */
        bcc += snprintf(name, SMB_CREATE_NAMEBUF_SZ, "tt%05d.tmp", tmp_id);

        buf = smb_srm_zalloc(sr, MAXPATHLEN);
        (void) snprintf(buf, MAXPATHLEN, "%s\\%s",
            op->fqi.fq_path.pn_path, name);
        op->fqi.fq_path.pn_path = buf;

        if (smb_common_create(sr) != NT_STATUS_SUCCESS)
                return (SDRC_ERROR);

        if (smbsr_encode_result(sr, 1, VAR_BCC, "bww%s", 1, sr->smb_fid,
            VAR_BCC, sr, name))
                return (SDRC_ERROR);

        return (SDRC_SUCCESS);
}

/*
 * Common create file function.  The file is opened in compatibility
 * mode with read/write access.
 */
uint32_t
smb_common_create(smb_request_t *sr)
{
        struct open_param *op = &sr->arg.open;
        uint32_t status;

        if ((op->mtime.tv_sec != 0) && (op->mtime.tv_sec != UINT_MAX))
                op->mtime.tv_sec = smb_time_local_to_gmt(sr, op->mtime.tv_sec);
        op->mtime.tv_nsec = 0;
        op->dsize = 0;
        op->omode = SMB_DA_ACCESS_READ_WRITE | SMB_DA_SHARE_COMPATIBILITY;
        op->desired_access = smb_omode_to_amask(op->omode);
        op->share_access = smb_denymode_to_sharemode(op->omode,
            op->fqi.fq_path.pn_path);

        if (sr->smb_flg & SMB_FLAGS_OPLOCK) {
                if (sr->smb_flg & SMB_FLAGS_OPLOCK_NOTIFY_ANY)
                        op->op_oplock_level = SMB_OPLOCK_BATCH;
                else
                        op->op_oplock_level = SMB_OPLOCK_EXCLUSIVE;
        } else {
                op->op_oplock_level = SMB_OPLOCK_NONE;
        }

        status = smb_common_open(sr);
        if (status == 0 && op->op_oplock_level != SMB_OPLOCK_NONE) {
                /* Oplock req. in op->op_oplock_level etc. */
                smb1_oplock_acquire(sr, B_FALSE);
        }

        if (op->op_oplock_level == SMB_OPLOCK_NONE) {
                sr->smb_flg &=
                    ~(SMB_FLAGS_OPLOCK | SMB_FLAGS_OPLOCK_NOTIFY_ANY);
        }

        if (status)
                smbsr_status(sr, status, 0, 0);

        return (status);
}