/* * 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 #include /* * 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); }