source4/libcli/raw/rawfsinfo.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    RAW_QFS_* operations
   5
   6    Copyright (C) Andrew Tridgell 2003
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 3 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23 #include "libcli/raw/libcliraw.h"
  24 #include "libcli/raw/raw_proto.h"
  25 #include "librpc/gen_ndr/ndr_misc.h"
  26
  27 /****************************************************************************
  28  Query FS Info - SMBdskattr call (async send)
  29 ****************************************************************************/
  30 static struct smbcli_request *smb_raw_dskattr_send(struct smbcli_tree *tree,
  31                                                 union smb_fsinfo *fsinfo)
  32 {
  33         struct smbcli_request *req;
  34
  35         req = smbcli_request_setup(tree, SMBdskattr, 0, 0);
  36         if (req == NULL) {
  37                 return NULL;
  38         }
  39
  40         if (!smbcli_request_send(req)) {
  41                 smbcli_request_destroy(req);
  42                 return NULL;
  43         }
  44
  45         return req;
  46 }
  47
  48 /****************************************************************************
  49  Query FS Info - SMBdskattr call (async recv)
  50 ****************************************************************************/
  51 static NTSTATUS smb_raw_dskattr_recv(struct smbcli_request *req,
  52                                      union smb_fsinfo *fsinfo)
  53 {
  54         if (!smbcli_request_receive(req) ||
  55             smbcli_request_is_error(req)) {
  56                 goto failed;
  57         }
  58
  59         SMBCLI_CHECK_WCT(req, 5);
  60         fsinfo->dskattr.out.units_total =     SVAL(req->in.vwv, VWV(0));
  61         fsinfo->dskattr.out.blocks_per_unit = SVAL(req->in.vwv, VWV(1));
  62         fsinfo->dskattr.out.block_size =      SVAL(req->in.vwv, VWV(2));
  63         fsinfo->dskattr.out.units_free =      SVAL(req->in.vwv, VWV(3));
  64
  65 failed:
  66         return smbcli_request_destroy(req);
  67 }
  68
  69
  70 /****************************************************************************
  71  RAW_QFS_ trans2 interface via blobs (async send)
  72 ****************************************************************************/
  73 static struct smbcli_request *smb_raw_qfsinfo_send(struct smbcli_tree *tree,
  74                                                 TALLOC_CTX *mem_ctx,
  75                                                 uint16_t info_level)
  76 {
  77         struct smb_trans2 tp;
  78         uint16_t setup = TRANSACT2_QFSINFO;
  79
  80         tp.in.max_setup = 0;
  81         tp.in.flags = 0;
  82         tp.in.timeout = 0;
  83         tp.in.setup_count = 1;
  84         tp.in.max_param = 0;
  85         tp.in.max_data = 0xFFFF;
  86         tp.in.setup = &setup;
  87         tp.in.data = data_blob(NULL, 0);
  88         tp.in.timeout = 0;
  89
  90         tp.in.params = data_blob_talloc(mem_ctx, NULL, 2);
  91         if (!tp.in.params.data) {
  92                 return NULL;
  93         }
  94         SSVAL(tp.in.params.data, 0, info_level);
  95
  96         return smb_raw_trans2_send(tree, &tp);
  97 }
  98
  99 /****************************************************************************
 100  RAW_QFS_ trans2 interface via blobs (async recv)
 101 ****************************************************************************/
 102 static NTSTATUS smb_raw_qfsinfo_blob_recv(struct smbcli_request *req,
 103                                           TALLOC_CTX *mem_ctx,
 104                                           DATA_BLOB *blob)
 105 {
 106         struct smb_trans2 tp;
 107         NTSTATUS status;
 108
 109         status = smb_raw_trans2_recv(req, mem_ctx, &tp);
 110
 111         if (NT_STATUS_IS_OK(status)) {
 112                 (*blob) = tp.out.data;
 113         }
 114
 115         return status;
 116 }
 117
 118
 119 /* local macros to make the code more readable */
 120 #define QFS_CHECK_MIN_SIZE(size) if (blob.length < (size)) { \
 121       DEBUG(1,("Unexpected QFS reply size %d for level %u - expected min of %d\n", \
 122                (int)blob.length, fsinfo->generic.level, (size))); \
 123       status = NT_STATUS_INFO_LENGTH_MISMATCH; \
 124       goto failed; \
 125 }
 126 #define QFS_CHECK_SIZE(size) if (blob.length != (size)) { \
 127       DEBUG(1,("Unexpected QFS reply size %d for level %u - expected %d\n", \
 128                (int)blob.length, fsinfo->generic.level, (size))); \
 129       status = NT_STATUS_INFO_LENGTH_MISMATCH; \
 130       goto failed; \
 131 }
 132
 133
 134 /****************************************************************************
 135  Query FSInfo raw interface (async send)
 136 ****************************************************************************/
 137 struct smbcli_request *smb_raw_fsinfo_send(struct smbcli_tree *tree,
 138                                         TALLOC_CTX *mem_ctx,
 139                                         union smb_fsinfo *fsinfo)
 140 {
 141         uint16_t info_level;
 142
 143         /* handle the only non-trans2 call separately */
 144         if (fsinfo->generic.level == RAW_QFS_DSKATTR) {
 145                 return smb_raw_dskattr_send(tree, fsinfo);
 146         }
 147         if (fsinfo->generic.level >= RAW_QFS_GENERIC) {
 148                 return NULL;
 149         }
 150
 151         /* the headers map the trans2 levels direct to info levels */
 152         info_level = (uint16_t)fsinfo->generic.level;
 153
 154         return smb_raw_qfsinfo_send(tree, mem_ctx, info_level);
 155 }
 156
 157 /*
 158   parse the fsinfo 'passthru' level replies
 159 */
 160 NTSTATUS smb_raw_fsinfo_passthru_parse(DATA_BLOB blob, TALLOC_CTX *mem_ctx,
 161                                        enum smb_fsinfo_level level,
 162                                        union smb_fsinfo *fsinfo)
 163 {
 164         NTSTATUS status = NT_STATUS_OK;
 165         int i;
 166
 167         /* parse the results */
 168         switch (level) {
 169         case RAW_QFS_VOLUME_INFORMATION:
 170                 QFS_CHECK_MIN_SIZE(18);
 171                 fsinfo->volume_info.out.create_time   = smbcli_pull_nttime(blob.data, 0);
 172                 fsinfo->volume_info.out.serial_number = IVAL(blob.data, 8);
 173                 smbcli_blob_pull_string(NULL, mem_ctx, &blob,
 174                                         &fsinfo->volume_info.out.volume_name,
 175                                         12, 18, STR_UNICODE);
 176                 break;
 177
 178         case RAW_QFS_SIZE_INFORMATION:
 179                 QFS_CHECK_SIZE(24);
 180                 fsinfo->size_info.out.total_alloc_units = BVAL(blob.data,  0);
 181                 fsinfo->size_info.out.avail_alloc_units = BVAL(blob.data,  8);
 182                 fsinfo->size_info.out.sectors_per_unit =  IVAL(blob.data, 16);
 183                 fsinfo->size_info.out.bytes_per_sector =  IVAL(blob.data, 20);
 184                 break;
 185
 186         case RAW_QFS_DEVICE_INFORMATION:
 187                 QFS_CHECK_SIZE(8);
 188                 fsinfo->device_info.out.device_type     = IVAL(blob.data,  0);
 189                 fsinfo->device_info.out.characteristics = IVAL(blob.data,  4);
 190                 break;
 191
 192         case RAW_QFS_ATTRIBUTE_INFORMATION:
 193                 QFS_CHECK_MIN_SIZE(12);
 194                 fsinfo->attribute_info.out.fs_attr   =                 IVAL(blob.data, 0);
 195                 fsinfo->attribute_info.out.max_file_component_length = IVAL(blob.data, 4);
 196                 smbcli_blob_pull_string(NULL, mem_ctx, &blob,
 197                                         &fsinfo->attribute_info.out.fs_type,
 198                                         8, 12, STR_UNICODE);
 199                 break;
 200
 201         case RAW_QFS_QUOTA_INFORMATION:
 202                 QFS_CHECK_SIZE(48);
 203                 fsinfo->quota_information.out.unknown[0] =  BVAL(blob.data,  0);
 204                 fsinfo->quota_information.out.unknown[1] =  BVAL(blob.data,  8);
 205                 fsinfo->quota_information.out.unknown[2] =  BVAL(blob.data, 16);
 206                 fsinfo->quota_information.out.quota_soft =  BVAL(blob.data, 24);
 207                 fsinfo->quota_information.out.quota_hard =  BVAL(blob.data, 32);
 208                 fsinfo->quota_information.out.quota_flags = BVAL(blob.data, 40);
 209                 break;
 210
 211         case RAW_QFS_FULL_SIZE_INFORMATION:
 212                 QFS_CHECK_SIZE(32);
 213                 fsinfo->full_size_information.out.total_alloc_units =        BVAL(blob.data,  0);
 214                 fsinfo->full_size_information.out.call_avail_alloc_units =   BVAL(blob.data,  8);
 215                 fsinfo->full_size_information.out.actual_avail_alloc_units = BVAL(blob.data, 16);
 216                 fsinfo->full_size_information.out.sectors_per_unit =         IVAL(blob.data, 24);
 217                 fsinfo->full_size_information.out.bytes_per_sector =         IVAL(blob.data, 28);
 218                 break;
 219
 220         case RAW_QFS_OBJECTID_INFORMATION: {
 221                 DATA_BLOB b2 = data_blob_const(blob.data, MIN(16, blob.length));
 222                 QFS_CHECK_SIZE(64);
 223                 status = GUID_from_ndr_blob(&b2, &fsinfo->objectid_information.out.guid);
 224                 NT_STATUS_NOT_OK_RETURN(status);
 225                 for (i=0;i<6;i++) {
 226                         fsinfo->objectid_information.out.unknown[i] = BVAL(blob.data, 16 + i*8);
 227                 }
 228                 break;
 229         }
 230
 231         default:
 232                 status = NT_STATUS_INVALID_INFO_CLASS;
 233         }
 234
 235 failed:
 236         return status;
 237 }
 238
 239
 240 /****************************************************************************
 241  Query FSInfo raw interface (async recv)
 242 ****************************************************************************/
 243 NTSTATUS smb_raw_fsinfo_recv(struct smbcli_request *req,
 244                              TALLOC_CTX *mem_ctx,
 245                              union smb_fsinfo *fsinfo)
 246 {
 247         DATA_BLOB blob;
 248         NTSTATUS status;
 249         struct smbcli_session *session = req?req->session:NULL;
 250
 251         if (fsinfo->generic.level == RAW_QFS_DSKATTR) {
 252                 return smb_raw_dskattr_recv(req, fsinfo);
 253         }
 254
 255         status = smb_raw_qfsinfo_blob_recv(req, mem_ctx, &blob);
 256         if (!NT_STATUS_IS_OK(status)) {
 257                 return status;
 258         }
 259
 260         /* parse the results */
 261         switch (fsinfo->generic.level) {
 262         case RAW_QFS_GENERIC:
 263         case RAW_QFS_DSKATTR:
 264                 /* handled above */
 265                 break;
 266
 267         case RAW_QFS_ALLOCATION:
 268                 QFS_CHECK_SIZE(18);
 269                 fsinfo->allocation.out.fs_id =             IVAL(blob.data,  0);
 270                 fsinfo->allocation.out.sectors_per_unit =  IVAL(blob.data,  4);
 271                 fsinfo->allocation.out.total_alloc_units = IVAL(blob.data,  8);
 272                 fsinfo->allocation.out.avail_alloc_units = IVAL(blob.data, 12);
 273                 fsinfo->allocation.out.bytes_per_sector =  SVAL(blob.data, 16);
 274                 break;
 275
 276         case RAW_QFS_VOLUME:
 277                 QFS_CHECK_MIN_SIZE(5);
 278                 fsinfo->volume.out.serial_number = IVAL(blob.data, 0);
 279                 smbcli_blob_pull_string(session, mem_ctx, &blob,
 280                                      &fsinfo->volume.out.volume_name,
 281                                      4, 5, STR_LEN8BIT | STR_NOALIGN);
 282                 break;
 283
 284         case RAW_QFS_VOLUME_INFO:
 285         case RAW_QFS_VOLUME_INFORMATION:
 286                 return smb_raw_fsinfo_passthru_parse(blob, mem_ctx,
 287                                                      RAW_QFS_VOLUME_INFORMATION, fsinfo);
 288
 289         case RAW_QFS_SIZE_INFO:
 290         case RAW_QFS_SIZE_INFORMATION:
 291                 return smb_raw_fsinfo_passthru_parse(blob, mem_ctx,
 292                                                      RAW_QFS_SIZE_INFORMATION, fsinfo);
 293
 294         case RAW_QFS_DEVICE_INFO:
 295         case RAW_QFS_DEVICE_INFORMATION:
 296                 return smb_raw_fsinfo_passthru_parse(blob, mem_ctx,
 297                                                      RAW_QFS_DEVICE_INFORMATION, fsinfo);
 298
 299         case RAW_QFS_ATTRIBUTE_INFO:
 300         case RAW_QFS_ATTRIBUTE_INFORMATION:
 301                 return smb_raw_fsinfo_passthru_parse(blob, mem_ctx,
 302                                                      RAW_QFS_ATTRIBUTE_INFORMATION, fsinfo);
 303
 304         case RAW_QFS_UNIX_INFO:
 305                 QFS_CHECK_SIZE(12);
 306                 fsinfo->unix_info.out.major_version = SVAL(blob.data, 0);
 307                 fsinfo->unix_info.out.minor_version = SVAL(blob.data, 2);
 308                 fsinfo->unix_info.out.capability    = SVAL(blob.data, 4);
 309                 break;
 310
 311         case RAW_QFS_QUOTA_INFORMATION:
 312                 return smb_raw_fsinfo_passthru_parse(blob, mem_ctx,
 313                                                      RAW_QFS_QUOTA_INFORMATION, fsinfo);
 314
 315         case RAW_QFS_FULL_SIZE_INFORMATION:
 316                 return smb_raw_fsinfo_passthru_parse(blob, mem_ctx,
 317                                                      RAW_QFS_FULL_SIZE_INFORMATION, fsinfo);
 318
 319         case RAW_QFS_OBJECTID_INFORMATION:
 320                 return smb_raw_fsinfo_passthru_parse(blob, mem_ctx,
 321                                                      RAW_QFS_OBJECTID_INFORMATION, fsinfo);
 322         }
 323
 324 failed:
 325         return status;
 326 }
 327
 328 /****************************************************************************
 329  Query FSInfo raw interface (sync interface)
 330 ****************************************************************************/
 331 _PUBLIC_ NTSTATUS smb_raw_fsinfo(struct smbcli_tree *tree,
 332                         TALLOC_CTX *mem_ctx,
 333                         union smb_fsinfo *fsinfo)
 334 {
 335         struct smbcli_request *req = smb_raw_fsinfo_send(tree, mem_ctx, fsinfo);
 336         return smb_raw_fsinfo_recv(req, mem_ctx, fsinfo);
 337 }
 338
 339 /****************************************************************************
 340  Set FSInfo raw interface (async recv)
 341 ****************************************************************************/
 342 static NTSTATUS smb_raw_setfsinfo_recv(struct smbcli_request *req,
 343                              TALLOC_CTX *mem_ctx,
 344                              union smb_setfsinfo *set_fsinfo)
 345 {
 346         DATA_BLOB blob = data_blob_null;
 347         NTSTATUS status;
 348
 349         if (set_fsinfo->generic.level != RAW_SETFS_UNIX_INFO) {
 350                 return NT_STATUS_INVALID_PARAMETER;
 351         }
 352
 353         status = smb_raw_qfsinfo_blob_recv(req, mem_ctx, &blob);
 354         data_blob_free(&blob);
 355         return status;
 356 }
 357
 358 /****************************************************************************
 359  Set FSInfo raw interface (async send)
 360 ****************************************************************************/
 361 static struct smbcli_request *smb_raw_setfsinfo_send(struct smbcli_tree *tree,
 362                                                 TALLOC_CTX *mem_ctx,
 363                                                 union smb_setfsinfo *set_fsinfo)
 364 {
 365         struct smb_trans2 tp;
 366         uint16_t info_level;
 367         uint16_t setup = TRANSACT2_SETFSINFO;
 368
 369         if (set_fsinfo->generic.level != RAW_SETFS_UNIX_INFO) {
 370                 return NULL;
 371         }
 372         tp.in.max_setup = 0;
 373         tp.in.flags = 0;
 374         tp.in.timeout = 0;
 375         tp.in.setup_count = 1;
 376         tp.in.max_param = 0;
 377         tp.in.max_data = 0xFFFF;
 378         tp.in.setup = &setup;
 379         tp.in.timeout = 0;
 380
 381         tp.in.params = data_blob_talloc(mem_ctx, NULL, 4);
 382         if (!tp.in.params.data) {
 383                 return NULL;
 384         }
 385         info_level = (uint16_t)set_fsinfo->generic.level;
 386         SSVAL(tp.in.params.data, 0, 0);
 387         SSVAL(tp.in.params.data, 2, info_level);
 388
 389         tp.in.data = data_blob_talloc(mem_ctx, NULL, 12);
 390         if (!tp.in.data.data) {
 391                 return NULL;
 392         }
 393
 394         SSVAL(tp.in.data.data, 0, set_fsinfo->unix_info.in.major_version);
 395         SSVAL(tp.in.data.data, 2, set_fsinfo->unix_info.in.minor_version);
 396         SBVAL(tp.in.data.data, 4, set_fsinfo->unix_info.in.capability);
 397
 398         return smb_raw_trans2_send(tree, &tp);
 399 }
 400
 401 /****************************************************************************
 402  Set FSInfo raw interface (sync interface)
 403 ****************************************************************************/
 404 _PUBLIC_ NTSTATUS smb_raw_setfsinfo(struct smbcli_tree *tree,
 405                         TALLOC_CTX *mem_ctx,
 406                         union smb_setfsinfo *set_fsinfo)
 407 {
 408         struct smbcli_request *req = smb_raw_setfsinfo_send(tree, mem_ctx, set_fsinfo);
 409         return smb_raw_setfsinfo_recv(req, mem_ctx, set_fsinfo);
 410 }