a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross/*
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * CDDL HEADER START
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * The contents of this file are subject to the terms of the
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * Common Development and Distribution License (the "License").
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * You may not use this file except in compliance with the License.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * or http://www.opensolaris.org/os/licensing.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * See the License for the specific language governing permissions
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * and limitations under the License.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * When distributing Covered Code, include this CDDL HEADER in each
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * If applicable, add the following below this CDDL HEADER, with the
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * fields enclosed by brackets "[]" replaced with your own identifying
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * information: Portions Copyright [yyyy] [name of copyright owner]
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * CDDL HEADER END
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross/*
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * Copyright 2013 Nexenta Systems, Inc. All rights reserved.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross/*
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * Dispatch function for SMB2_QUERY_INFO
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <smbsrv/smb2_kproto.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <smbsrv/smb_fsops.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <smbsrv/ntifs.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossuint32_t
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rosssmb2_qinfo_quota(smb_request_t *sr, smb_queryinfo_t *qi)
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross{
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross _NOTE(ARGUNUSED(qi))
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross uint8_t single, restart;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross uint32_t sidlistlen, startsidlen, startsidoff;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smb_node_t *tnode;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smb_ofile_t *ofile = sr->fid_ofile;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smb_quota_query_t request;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smb_quota_response_t reply;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross uint32_t status = NT_STATUS_SUCCESS;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross int rc;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross bzero(&request, sizeof (smb_quota_query_t));
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross bzero(&reply, sizeof (smb_quota_response_t));
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (!smb_tree_has_feature(sr->tid_tree, SMB_TREE_QUOTA))
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross return (NT_STATUS_NOT_SUPPORTED);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if ((ofile->f_node == NULL) ||
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross (ofile->f_ftype != SMB_FTYPE_DISK))
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross return (NT_STATUS_NOT_SUPPORTED);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross rc = smb_mbc_decodef(
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross &sr->smb_data, "bb..lll",
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross &single, /* b */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross &restart, /* b */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross /* reserved .. */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross &sidlistlen, /* l */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross &startsidlen, /* l */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross &startsidoff); /* l */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (rc)
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross return (NT_STATUS_INVALID_PARAMETER);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if ((sidlistlen != 0) && (startsidlen != 0))
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross return (NT_STATUS_INVALID_PARAMETER);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross tnode = sr->tid_tree->t_snode;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross request.qq_root_path = kmem_zalloc(MAXPATHLEN, KM_SLEEP);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (smb_node_getmntpath(tnode, request.qq_root_path, MAXPATHLEN) != 0) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross kmem_free(request.qq_root_path, MAXPATHLEN);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross return (NT_STATUS_INVALID_PARAMETER);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (sidlistlen != 0)
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross request.qq_query_op = SMB_QUOTA_QUERY_SIDLIST;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross else if (startsidlen != 0)
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross request.qq_query_op = SMB_QUOTA_QUERY_STARTSID;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross else
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross request.qq_query_op = SMB_QUOTA_QUERY_ALL;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross request.qq_single = single;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross request.qq_restart = restart;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smb_quota_max_quota(&sr->raw_data, &request);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross status = smb_quota_init_sids(&sr->smb_data, &request, ofile);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (status == NT_STATUS_SUCCESS) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (smb_quota_query(sr->sr_server, &request, &reply) != 0) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross status = NT_STATUS_INTERNAL_ERROR;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross } else {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross status = reply.qr_status;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (status == NT_STATUS_SUCCESS) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross status = smb_quota_encode_quotas(
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross &sr->raw_data,
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross &request, &reply, ofile);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross xdr_free(smb_quota_response_xdr, (char *)&reply);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross kmem_free(request.qq_root_path, MAXPATHLEN);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smb_quota_free_sids(&request);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (status != NT_STATUS_SUCCESS) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross if (status == NT_STATUS_NO_MORE_ENTRIES) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smb_ofile_set_quota_resume(ofile, NULL);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smbsr_warn(sr, status, 0, 0);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross status = NT_STATUS_SUCCESS;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross } else {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross smbsr_error(sr, status, 0, 0);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross return (status);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross}