/*
* 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) 2010, Oracle and/or its affiliates. All rights reserved.
* Copyright 2013 Nexenta Systems, Inc. All rights reserved.
*/
#include <smbsrv/smb_kproto.h>
#include <smbsrv/smb_fsops.h>
/*
* smb_nt_transact_query_quota
*
* This method allows the client to retrieve quota information from
* the server. The result of the call is returned to the client in the
* Data part of the transaction response.
*
* On entry, the 'TotalParameterCount' field must be equal to 16, and the
* client parameter block must be encoded with the following parameters:
*
* Request Description
* ========================== ==================================
* WORD fid SMB file identifier of the target directory
* BYTE ReturnSingleEntry A boolean indicating whether to return
* a single entry (TRUE) or multiple entries (FALSE).
* BYTE RestartScan A boolean indicating whether to continue from
* the previous request (FALSE) or restart a new
* sequence (TRUE).
* DWORD SidListLength The length, in bytes, of the SidList in the
* data block or 0 if there is no SidList.
* DWORD StartSidLength If SidListLength is 0 (i.e. there is no SidList
* in the data block), then this is either:
* 1) the (non-zero) length in bytes of the
* StartSid in the parameter buffer, or
* 2) if 0, there is no StartSid in the
* parameter buffer, in which case, all SIDs
* are to be enumerated as if they were
* passed in the SidList.
* Otherwise, StartSidLength is ignored.
* DWORD StartSidOffset The offset, in bytes, to the StartSid in the
* parameter block (if one exists).
*
* One of SidListLength and StartSidLength must be 0.
*
* An SMB_COM_NT_TRANSACTION response is sent in reply when the request
* is successful. The 'TotalParameterCount' is set to 4, and the parameter
* block in the server response contains a 32-bit unsigned integer
* indicating the length, in bytes, of the returned quota information.
* The 'TotalDataCount' is set to indicate the length of the data buffer,
* and the data buffer contains the following quota information:
*
* Data Block Encoding Description
* ================================== =================================
* ULONG NextEntryOffset; Offset to start of next entry from
* start of this entry, or 0 for the
* final entry
* ULONG SidLength; Length (bytes) of SID
* SMB_TIME ChangeTime; Time that the quota was last changed
* LARGE_INTEGER QuotaUsed; Amount of quota (bytes) used by user
* LARGE_INTEGER QuotaThreshold; Quota warning limit (bytes) for user
* LARGE_INTEGER QuotaLimit; The quota limit (bytes) for this user
* USHORT Sid; Search handle
*/
smb_sdrc_t
smb_nt_transact_query_quota(smb_request_t *sr, smb_xa_t *xa)
{
uint8_t single, restart;
uint32_t sidlistlen, startsidlen, startsidoff;
smb_node_t *tnode;
smb_ofile_t *ofile;
smb_quota_query_t request;
smb_quota_response_t reply;
uint32_t status = NT_STATUS_SUCCESS;
bzero(&request, sizeof (smb_quota_query_t));
bzero(&reply, sizeof (smb_quota_response_t));
if (!smb_tree_has_feature(sr->tid_tree, SMB_TREE_QUOTA)) {
smbsr_error(sr, NT_STATUS_NOT_SUPPORTED, 0, 0);
return (SDRC_ERROR);
}
if (xa->smb_tpscnt != 16) {
smbsr_error(sr, NT_STATUS_INVALID_PARAMETER, 0, 0);
return (SDRC_ERROR);
}
if (smb_mbc_decodef(&xa->req_param_mb, "%wbblll", sr, &sr->smb_fid,
&single, &restart, &sidlistlen, &startsidlen, &startsidoff)) {
smbsr_error(sr, NT_STATUS_INVALID_PARAMETER, 0, 0);
return (SDRC_ERROR);
}
if ((sidlistlen != 0) && (startsidlen != 0)) {
smbsr_error(sr, NT_STATUS_INVALID_PARAMETER, 0, 0);
return (SDRC_ERROR);
}
smbsr_lookup_file(sr);
ofile = sr->fid_ofile;
if (ofile == NULL) {
smbsr_error(sr, NT_STATUS_INVALID_HANDLE, ERRDOS, ERRbadfid);
return (SDRC_ERROR);
}
if ((ofile->f_node == NULL) || (ofile->f_ftype != SMB_FTYPE_DISK)) {
smbsr_error(sr, NT_STATUS_ACCESS_DENIED, ERRDOS,
ERROR_ACCESS_DENIED);
smbsr_release_file(sr);
return (SDRC_ERROR);
}
tnode = sr->tid_tree->t_snode;
request.qq_root_path = kmem_zalloc(MAXPATHLEN, KM_SLEEP);
if (smb_node_getmntpath(tnode, request.qq_root_path, MAXPATHLEN) != 0) {
smbsr_error(sr, NT_STATUS_INVALID_PARAMETER, ERRDOS,
ERROR_INVALID_PARAMETER);
smbsr_release_file(sr);
kmem_free(request.qq_root_path, MAXPATHLEN);
return (SDRC_ERROR);
}
if (sidlistlen != 0)
request.qq_query_op = SMB_QUOTA_QUERY_SIDLIST;
else if (startsidlen != 0)
request.qq_query_op = SMB_QUOTA_QUERY_STARTSID;
else
request.qq_query_op = SMB_QUOTA_QUERY_ALL;
request.qq_single = single;
request.qq_restart = restart;
smb_quota_max_quota(&xa->rep_data_mb, &request);
status = smb_quota_init_sids(&xa->req_data_mb, &request, ofile);
if (status == NT_STATUS_SUCCESS) {
if (smb_quota_query(sr->sr_server, &request, &reply) != 0) {
status = NT_STATUS_INTERNAL_ERROR;
} else {
status = reply.qr_status;
if (status == NT_STATUS_SUCCESS) {
status = smb_quota_encode_quotas(
&xa->rep_data_mb,
&request, &reply, ofile);
}
(void) smb_mbc_encodef(&xa->rep_param_mb, "l",
xa->rep_data_mb.chain_offset);
xdr_free(smb_quota_response_xdr, (char *)&reply);
}
}
kmem_free(request.qq_root_path, MAXPATHLEN);
smb_quota_free_sids(&request);
if (status != NT_STATUS_SUCCESS) {
if (status == NT_STATUS_NO_MORE_ENTRIES) {
smb_ofile_set_quota_resume(ofile, NULL);
smbsr_warn(sr, status, 0, 0);
status = NT_STATUS_SUCCESS;
} else {
smbsr_error(sr, status, 0, 0);
}
(void) smb_mbc_encodef(&xa->rep_param_mb, "l", 0);
}
smbsr_release_file(sr);
return ((status == NT_STATUS_SUCCESS) ? SDRC_SUCCESS : SDRC_ERROR);
}
/*
* smb_nt_transact_set_quota
*
* This method allows the client to set quota information on the server.
* The result status of the call is returned to the client in the
* 'status' field of the SMB response header.
*
* On entry, the 'TotalParameterCount' field must be equal to 2, and the
* client parameter block must be encoded with the following parameters:
*
* Data Block Encoding Description
* ================================== =================================
* ULONG NextEntryOffset; Offset to start of next entry from
* start of this entry, or 0 for the
* final entry
* ULONG SidLength; Length (bytes) of SID
* SMB_TIME ChangeTime; Time that the quota was last changed
* LARGE_INTEGER QuotaUsed; Amount of quota (bytes) used by user
* LARGE_INTEGER QuotaThreshold; Quota warning limit (bytes) for user
* LARGE_INTEGER QuotaLimit; The quota limit (bytes) for this user
* VARIABLE Sid; Security identifier of the user
*
* An SMB_COM_NT_TRANSACTION response is sent in reply when the request
* is successful. The 'TotalParameterCount' and the 'TotalDataCount' are set
* to 0, and the parameter block 'Status' field in the server SMB response
* header contains a 32-bit unsigned integer indicating the result status
* (NT_STATUS_SUCCESS if successful).
*
* Only users with Admin privileges (i.e. of the BUILTIN/Administrators
* group) will be allowed to set quotas.
*/
smb_sdrc_t
smb_nt_transact_set_quota(smb_request_t *sr, smb_xa_t *xa)
{
char *root_path;
uint32_t status = NT_STATUS_SUCCESS;
smb_node_t *tnode;
smb_ofile_t *ofile;
smb_quota_set_t request;
uint32_t reply;
list_t *quota_list;
bzero(&request, sizeof (smb_quota_set_t));
if (!smb_tree_has_feature(sr->tid_tree, SMB_TREE_QUOTA)) {
smbsr_error(sr, NT_STATUS_NOT_SUPPORTED, 0, 0);
return (SDRC_ERROR);
}
if (!smb_user_is_admin(sr->uid_user)) {
smbsr_error(sr, NT_STATUS_ACCESS_DENIED, ERRSRV, ERRaccess);
return (-1);
}
if (xa->smb_tpscnt != 2) {
smbsr_error(sr, NT_STATUS_INVALID_PARAMETER, 0, 0);
return (SDRC_ERROR);
}
if (smb_mbc_decodef(&xa->req_param_mb, "%w", sr,
&sr->smb_fid)) {
smbsr_error(sr, NT_STATUS_INVALID_PARAMETER, 0, 0);
return (SDRC_ERROR);
}
smbsr_lookup_file(sr);
ofile = sr->fid_ofile;
if (ofile == NULL) {
smbsr_error(sr, NT_STATUS_INVALID_HANDLE, ERRDOS, ERRbadfid);
return (SDRC_ERROR);
}
if ((ofile->f_node == NULL) || (ofile->f_ftype != SMB_FTYPE_DISK)) {
smbsr_error(sr, NT_STATUS_ACCESS_DENIED, ERRDOS,
ERROR_ACCESS_DENIED);
smbsr_release_file(sr);
return (SDRC_ERROR);
}
tnode = sr->tid_tree->t_snode;
root_path = kmem_zalloc(MAXPATHLEN, KM_SLEEP);
if (smb_node_getmntpath(tnode, root_path, MAXPATHLEN) != 0) {
smbsr_error(sr, NT_STATUS_INVALID_PARAMETER, ERRDOS,
ERROR_INVALID_PARAMETER);
smbsr_release_file(sr);
kmem_free(root_path, MAXPATHLEN);
return (SDRC_ERROR);
}
quota_list = &request.qs_quota_list;
list_create(quota_list, sizeof (smb_quota_t),
offsetof(smb_quota_t, q_list_node));
status = smb_quota_decode_quotas(&xa->req_data_mb, quota_list);
if (status == NT_STATUS_SUCCESS) {
request.qs_root_path = root_path;
if (smb_quota_set(sr->sr_server, &request, &reply) != 0) {
status = NT_STATUS_INTERNAL_ERROR;
} else {
status = reply;
xdr_free(xdr_uint32_t, (char *)&reply);
}
}
kmem_free(root_path, MAXPATHLEN);
smb_quota_free_quotas(&request.qs_quota_list);
smbsr_release_file(sr);
if (status != NT_STATUS_SUCCESS) {
smbsr_error(sr, status, 0, 0);
(void) smb_mbc_encodef(&xa->rep_param_mb, "l", 0);
return (SDRC_ERROR);
}
return (SDRC_SUCCESS);
}