1 /*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21
22 /*
23 * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
24 * Copyright 2013 Nexenta Systems, Inc. All rights reserved.
25 */
26
27 /*
28 * Dispatch function for SMB2_QUERY_INFO
29 */
30
31 #include <smbsrv/smb2_kproto.h>
32 #include <smbsrv/smb_fsops.h>
33 #include <smbsrv/ntifs.h>
34
35 uint32_t
smb2_qinfo_quota(smb_request_t * sr,smb_queryinfo_t * qi)36 smb2_qinfo_quota(smb_request_t *sr, smb_queryinfo_t *qi)
37 {
38 _NOTE(ARGUNUSED(qi))
39 uint8_t single, restart;
40 uint32_t sidlistlen, startsidlen, startsidoff;
41 smb_node_t *tnode;
42 smb_ofile_t *ofile = sr->fid_ofile;
43 smb_quota_query_t request;
44 smb_quota_response_t reply;
45 uint32_t status = NT_STATUS_SUCCESS;
46 int rc;
47
48 bzero(&request, sizeof (smb_quota_query_t));
49 bzero(&reply, sizeof (smb_quota_response_t));
50
51 if (!smb_tree_has_feature(sr->tid_tree, SMB_TREE_QUOTA))
52 return (NT_STATUS_NOT_SUPPORTED);
53
54 if ((ofile->f_node == NULL) ||
55 (ofile->f_ftype != SMB_FTYPE_DISK))
56 return (NT_STATUS_NOT_SUPPORTED);
57
58 rc = smb_mbc_decodef(
59 &sr->smb_data, "bb..lll",
60 &single, /* b */
61 &restart, /* b */
62 /* reserved .. */
63 &sidlistlen, /* l */
64 &startsidlen, /* l */
65 &startsidoff); /* l */
66 if (rc)
67 return (NT_STATUS_INVALID_PARAMETER);
68
69 if ((sidlistlen != 0) && (startsidlen != 0))
70 return (NT_STATUS_INVALID_PARAMETER);
71
72
73 tnode = sr->tid_tree->t_snode;
74 request.qq_root_path = kmem_zalloc(MAXPATHLEN, KM_SLEEP);
75 if (smb_node_getmntpath(tnode, request.qq_root_path, MAXPATHLEN) != 0) {
76 kmem_free(request.qq_root_path, MAXPATHLEN);
77 return (NT_STATUS_INVALID_PARAMETER);
78 }
79
80 if (sidlistlen != 0)
81 request.qq_query_op = SMB_QUOTA_QUERY_SIDLIST;
82 else if (startsidlen != 0)
83 request.qq_query_op = SMB_QUOTA_QUERY_STARTSID;
84 else
85 request.qq_query_op = SMB_QUOTA_QUERY_ALL;
86
87 request.qq_single = single;
88 request.qq_restart = restart;
89 smb_quota_max_quota(&sr->raw_data, &request);
90
91 status = smb_quota_init_sids(&sr->smb_data, &request, ofile);
92
93 if (status == NT_STATUS_SUCCESS) {
94 if (smb_quota_query(sr->sr_server, &request, &reply) != 0) {
95 status = NT_STATUS_INTERNAL_ERROR;
96 } else {
97 status = reply.qr_status;
98 if (status == NT_STATUS_SUCCESS) {
99 status = smb_quota_encode_quotas(
100 &sr->raw_data,
101 &request, &reply, ofile);
102 }
103 xdr_free(smb_quota_response_xdr, (char *)&reply);
104 }
105 }
106
107 kmem_free(request.qq_root_path, MAXPATHLEN);
108 smb_quota_free_sids(&request);
109
110 if (status != NT_STATUS_SUCCESS) {
111 if (status == NT_STATUS_NO_MORE_ENTRIES) {
112 smb_ofile_set_quota_resume(ofile, NULL);
113 smbsr_warn(sr, status, 0, 0);
114 status = NT_STATUS_SUCCESS;
115 } else {
116 smbsr_error(sr, status, 0, 0);
117 }
118 }
119
120 return (status);
121 }
122