/* * 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 2010 Sun Microsystems, Inc. All rights reserved. * Use is subject to license terms. */ #include #include #include /* * smb_fssize_t * volume_units and volume avail are the total allocated and * available units on the volume. * caller_units and caller_avail are the allocated and available * units on the volume for the user associated with the calling * thread. */ typedef struct smb_fssize { uint64_t fs_volume_units; uint64_t fs_volume_avail; uint64_t fs_caller_units; uint64_t fs_caller_avail; uint32_t fs_sectors_per_unit; uint32_t fs_bytes_per_sector; } smb_fssize_t; /* * File System Control Flags for smb_com_trans2_query|set_fs_information * level SMB_FILE_FS_CONTROL_INFORMATION */ #define FILE_VC_QUOTA_TRACK 0x00000001 #define FILE_VC_QUOTA_ENFORCE 0x00000002 #define FILE_VC_CONTENT_INDEX_DISABLED 0x00000008 #define FILE_VC_LOG_QUOTA_THRESHOLD 0x00000010 #define FILE_VC_LOG_QUOTA_LIMIT 0x00000020 #define FILE_VC_LOG_VOLUME_THRESHOLD 0x00000040 #define FILE_VC_LOG_VOLUME_LIMIT 0x00000080 #define FILE_VC_QUOTAS_INCOMPLETE 0x00000100 #define FILE_VC_QUOTAS_REBUILDING 0x00000200 static int smb_fssize(smb_request_t *, smb_fssize_t *); static int smb_trans2_set_fs_ctrl_info(smb_request_t *, smb_xa_t *); /* * smb_com_query_information_disk * * The SMB_COM_QUERY_INFORMATION_DISK command is used to determine the * capacity and remaining free space on the drive hosting the directory * structure indicated by Tid in the SMB header. * * The blocking/allocation units used in this response may be independent * of the actual physical or logical blocking/allocation algorithm(s) used * internally by the server. However, they must accurately reflect the * amount of space on the server. * * This SMB only returns 16 bits of information for each field, which may * not be large enough for some disk systems. In particular TotalUnits is * commonly > 64K. Fortunately, it turns out the all the client cares * about is the total disk size, in bytes, and the free space, in bytes. * So, it is reasonable for a server to adjust the relative values of * BlocksPerUnit and BlockSize to accommodate. If after all adjustment, * the numbers are still too high, the largest possible values for * TotalUnit or FreeUnits (i.e. 0xFFFF) should be returned. */ smb_sdrc_t smb_pre_query_information_disk(smb_request_t *sr) { DTRACE_SMB_1(op__QueryInformationDisk__start, smb_request_t *, sr); return (SDRC_SUCCESS); } void smb_post_query_information_disk(smb_request_t *sr) { DTRACE_SMB_1(op__QueryInformationDisk__done, smb_request_t *, sr); } smb_sdrc_t smb_com_query_information_disk(smb_request_t *sr) { int rc; fsblkcnt64_t total_blocks, free_blocks; unsigned long block_size, unit_size; unsigned short blocks_per_unit, bytes_per_block; unsigned short total_units, free_units; smb_fssize_t fssize; if (STYPE_ISIPC(sr->tid_tree->t_res_type)) { smbsr_error(sr, NT_STATUS_ACCESS_DENIED, ERRDOS, ERRnoaccess); return (SDRC_ERROR); } if (smb_fssize(sr, &fssize) != 0) return (SDRC_ERROR); unit_size = fssize.fs_sectors_per_unit; block_size = fssize.fs_bytes_per_sector; total_blocks = fssize.fs_caller_units; free_blocks = fssize.fs_caller_avail; /* * It seems that DOS clients cannot handle block sizes * bigger than 512 KB. So we have to set the block size at * most to 512 */ while (block_size > 512) { block_size >>= 1; unit_size <<= 1; } /* adjust blocks and sizes until they fit into a word */ while (total_blocks >= 0xFFFF) { total_blocks >>= 1; free_blocks >>= 1; if ((unit_size <<= 1) > 0xFFFF) { unit_size >>= 1; total_blocks = 0xFFFF; free_blocks <<= 1; break; } } total_units = (total_blocks >= 0xFFFF) ? 0xFFFF : (unsigned short)total_blocks; free_units = (free_blocks >= 0xFFFF) ? 0xFFFF : (unsigned short)free_blocks; bytes_per_block = (unsigned short)block_size; blocks_per_unit = (unsigned short)unit_size; rc = smbsr_encode_result(sr, 5, 0, "bwwww2.w", 5, total_units, /* total_units */ blocks_per_unit, /* blocks_per_unit */ bytes_per_block, /* blocksize */ free_units, /* free_units */ 0); /* bcc */ return ((rc == 0) ? SDRC_SUCCESS : SDRC_ERROR); } /* * smb_com_trans2_query_fs_information * * This transaction requests information about the filesystem. * The following information levels are supported: * * InformationLevel Value * ================================== ====== * SMB_INFO_ALLOCATION 1 * SMB_INFO_VOLUME 2 * SMB_QUERY_FS_VOLUME_INFO 0x102 * SMB_QUERY_FS_SIZE_INFO 0x103 * SMB_QUERY_FS_DEVICE_INFO 0x104 * SMB_QUERY_FS_ATTRIBUTE_INFO 0x105 * SMB_FILE_FS_VOLUME_INFORMATION 1001 * SMB_FILE_FS_SIZE_INFORMATION 1003 * SMB_FILE_FS_DEVICE_INFORMATION 1004 * SMB_FILE_FS_ATTRIBUTE_INFORMATION 1005 * SMB_FILE_FS_CONTROL_INFORMATION 1006 * SMB_FILE_FS_FULLSIZE_INFORMATION 1007 * * The fsid provides a system-wide unique file system ID. * fsid.val[0] is the 32-bit dev for the file system of the share root * smb_node. * fsid.val[1] is the file system type. */ smb_sdrc_t smb_com_trans2_query_fs_information(smb_request_t *sr, smb_xa_t *xa) { uint32_t flags; char *encode_str, *tmpbuf; uint64_t max_int; uint16_t infolev; int rc, length, buflen; smb_tree_t *tree; smb_node_t *snode; char *fsname = "NTFS"; fsid_t fsid; smb_fssize_t fssize; smb_msgbuf_t mb; tree = sr->tid_tree; if (!STYPE_ISDSK(tree->t_res_type)) { smbsr_error(sr, NT_STATUS_ACCESS_DENIED, ERRDOS, ERROR_ACCESS_DENIED); return (SDRC_ERROR); } if (smb_mbc_decodef(&xa->req_param_mb, "w", &infolev) != 0) return (SDRC_ERROR); snode = tree->t_snode; fsid = SMB_NODE_FSID(snode); switch (infolev) { case SMB_INFO_ALLOCATION: if (smb_fssize(sr, &fssize) != 0) return (SDRC_ERROR); max_int = (uint64_t)UINT_MAX; if (fssize.fs_caller_units > max_int) fssize.fs_caller_units = max_int; if (fssize.fs_caller_avail > max_int) fssize.fs_caller_avail = max_int; (void) smb_mbc_encodef(&xa->rep_data_mb, "llllw", 0, fssize.fs_sectors_per_unit, fssize.fs_caller_units, fssize.fs_caller_avail, fssize.fs_bytes_per_sector); break; case SMB_INFO_VOLUME: /* * In this response, the unicode volume label is NOT * expected to be aligned. Encode ('U') into a temporary * buffer, then encode buffer as a byte stream ('#c'). */ if ((sr->smb_flg2 & SMB_FLAGS2_UNICODE) || (sr->session->native_os == NATIVE_OS_WIN95)) { length = smb_wcequiv_strlen(tree->t_volume); buflen = length + sizeof (smb_wchar_t); tmpbuf = smb_srm_zalloc(sr, buflen); smb_msgbuf_init(&mb, (uint8_t *)tmpbuf, buflen, SMB_MSGBUF_UNICODE); rc = smb_msgbuf_encode(&mb, "U", tree->t_volume); if (rc >= 0) { rc = smb_mbc_encodef(&xa->rep_data_mb, "%lb#c", sr, fsid.val[0], length, length, tmpbuf); } smb_msgbuf_term(&mb); } else { length = strlen(tree->t_volume); rc = smb_mbc_encodef(&xa->rep_data_mb, "%lbs", sr, fsid.val[0], length, tree->t_volume); } if (rc < 0) return (SDRC_ERROR); break; case SMB_QUERY_FS_VOLUME_INFO: case SMB_FILE_FS_VOLUME_INFORMATION: if ((sr->smb_flg2 & SMB_FLAGS2_UNICODE) || (sr->session->native_os == NATIVE_OS_WIN95)) { length = smb_wcequiv_strlen(tree->t_volume); encode_str = "%qllb.U"; } else { length = strlen(tree->t_volume); encode_str = "%qllb.s"; } /* * NT has the "supports objects" flag set to 1. */ (void) smb_mbc_encodef(&xa->rep_data_mb, encode_str, sr, 0ll, /* Volume creation time */ fsid.val[0], /* Volume serial number */ length, /* label length */ 0, /* Supports objects */ tree->t_volume); break; case SMB_QUERY_FS_SIZE_INFO: case SMB_FILE_FS_SIZE_INFORMATION: if (smb_fssize(sr, &fssize) != 0) return (SDRC_ERROR); (void) smb_mbc_encodef(&xa->rep_data_mb, "qqll", fssize.fs_caller_units, fssize.fs_caller_avail, fssize.fs_sectors_per_unit, fssize.fs_bytes_per_sector); break; case SMB_QUERY_FS_DEVICE_INFO: case SMB_FILE_FS_DEVICE_INFORMATION: (void) smb_mbc_encodef(&xa->rep_data_mb, "ll", FILE_DEVICE_FILE_SYSTEM, FILE_DEVICE_IS_MOUNTED); break; case SMB_QUERY_FS_ATTRIBUTE_INFO: case SMB_FILE_FS_ATTRIBUTE_INFORMATION: if ((sr->smb_flg2 & SMB_FLAGS2_UNICODE) || (sr->session->native_os == NATIVE_OS_WINNT) || (sr->session->native_os == NATIVE_OS_WIN2000) || (sr->session->native_os == NATIVE_OS_WIN95) || (sr->session->native_os == NATIVE_OS_MACOS)) { length = smb_wcequiv_strlen(fsname); encode_str = "%lllU"; sr->smb_flg2 |= SMB_FLAGS2_UNICODE; } else { length = strlen(fsname); encode_str = "%llls"; } flags = FILE_CASE_PRESERVED_NAMES; if (tree->t_flags & SMB_TREE_UNICODE_ON_DISK) flags |= FILE_UNICODE_ON_DISK; if (tree->t_flags & SMB_TREE_SUPPORTS_ACLS) flags |= FILE_PERSISTENT_ACLS; if ((tree->t_flags & SMB_TREE_CASEINSENSITIVE) == 0) flags |= FILE_CASE_SENSITIVE_SEARCH; if (tree->t_flags & SMB_TREE_STREAMS) flags |= FILE_NAMED_STREAMS; if (tree->t_flags & SMB_TREE_QUOTA) flags |= FILE_VOLUME_QUOTAS; (void) smb_mbc_encodef(&xa->rep_data_mb, encode_str, sr, flags, MAXNAMELEN, /* max name */ length, /* label length */ fsname); break; case SMB_FILE_FS_CONTROL_INFORMATION: if (!smb_tree_has_feature(sr->tid_tree, SMB_TREE_QUOTA)) { smbsr_error(sr, NT_STATUS_NOT_SUPPORTED, ERRDOS, ERROR_NOT_SUPPORTED); return (SDRC_ERROR); } (void) smb_mbc_encodef(&xa->rep_data_mb, "qqqqqll", 0, /* free space start filtering - MUST be 0 */ 0, /* free space threshold - MUST be 0 */ 0, /* free space stop filtering - MUST be 0 */ SMB_QUOTA_UNLIMITED, /* default quota threshold */ SMB_QUOTA_UNLIMITED, /* default quota limit */ FILE_VC_QUOTA_ENFORCE, /* fs control flag */ 0); /* pad bytes */ break; case SMB_FILE_FS_FULLSIZE_INFORMATION: if (smb_fssize(sr, &fssize) != 0) return (SDRC_ERROR); (void) smb_mbc_encodef(&xa->rep_data_mb, "qqqll", fssize.fs_caller_units, fssize.fs_caller_avail, fssize.fs_volume_avail, fssize.fs_sectors_per_unit, fssize.fs_bytes_per_sector); break; case SMB_FILE_FS_LABEL_INFORMATION: case SMB_FILE_FS_OBJECTID_INFORMATION: case SMB_FILE_FS_DRIVERPATH_INFORMATION: smbsr_error(sr, NT_STATUS_NOT_SUPPORTED, ERRDOS, ERROR_NOT_SUPPORTED); return (SDRC_ERROR); default: smbsr_error(sr, NT_STATUS_INVALID_LEVEL, ERRDOS, ERROR_INVALID_LEVEL); return (SDRC_ERROR); } return (SDRC_SUCCESS); } /* * smb_fssize * * File system size information, for the volume and for the user * initiating the request. * * If there's no quota entry for the user initiating the request, * caller_units and caller_avail are the total and available units * for the volume (volume_units, volume_avail). * If there is a quota entry for the user initiating the request, * and it is not SMB_QUOTA_UNLIMITED, calculate caller_units and * caller_avail as follows: * caller_units = quota limit / bytes_per_unit * caller_avail = remaining quota / bytes_per_unit * * A quota limit of SMB_QUOTA_UNLIMITED means that the user's quota * is specfied as unlimited. A quota limit of 0 means there is no * quota specified for the user. * * Returns: 0 - success * -1 - error. Error status set in sr. */ static int smb_fssize(smb_request_t *sr, smb_fssize_t *fssize) { smb_node_t *node; struct statvfs64 df; uid_t uid; smb_quota_t quota; int rc, bytes_per_unit; bzero(fssize, sizeof (smb_fssize_t)); node = sr->tid_tree->t_snode; if ((rc = smb_fsop_statfs(sr->user_cr, node, &df)) != 0) { smbsr_errno(sr, rc); return (-1); } fssize->fs_bytes_per_sector = 512; fssize->fs_sectors_per_unit = df.f_frsize >> 9; if (df.f_bavail > df.f_blocks) df.f_bavail = 0; fssize->fs_volume_units = df.f_blocks; fssize->fs_volume_avail = df.f_bavail; fssize->fs_caller_units = df.f_blocks; fssize->fs_caller_avail = df.f_bavail; bytes_per_unit = fssize->fs_bytes_per_sector * fssize->fs_sectors_per_unit; if (!smb_tree_has_feature(sr->tid_tree, SMB_TREE_QUOTA)) return (0); uid = crgetuid(sr->uid_user->u_cred); if (smb_quota_query_user_quota(sr, uid, "a) != NT_STATUS_SUCCESS) return (0); if ((quota.q_limit != SMB_QUOTA_UNLIMITED) && (quota.q_limit != 0)) { fssize->fs_caller_units = quota.q_limit / bytes_per_unit; if (quota.q_limit <= quota.q_used) fssize->fs_caller_avail = 0; else fssize->fs_caller_avail = (quota.q_limit - quota.q_used) / bytes_per_unit; } return (0); } /* * smb_com_trans2_set_fs_information * * This transaction sets filesystem information. * The following information levels are supported: * * InformationLevel Value * ================================== ====== * SMB_FILE_FS_CONTROL_INFORMATION 1006 */ smb_sdrc_t smb_com_trans2_set_fs_information(smb_request_t *sr, smb_xa_t *xa) { smb_tree_t *tree; uint16_t infolev; tree = sr->tid_tree; if (!STYPE_ISDSK(tree->t_res_type)) { smbsr_error(sr, NT_STATUS_ACCESS_DENIED, ERRDOS, ERROR_ACCESS_DENIED); return (SDRC_ERROR); } if (smb_mbc_decodef(&xa->req_param_mb, "ww", &sr->smb_fid, &infolev) != 0) return (SDRC_ERROR); switch (infolev) { case SMB_FILE_FS_CONTROL_INFORMATION: if (smb_trans2_set_fs_ctrl_info(sr, xa) != 0) return (SDRC_ERROR); break; case SMB_FILE_FS_VOLUME_INFORMATION: case SMB_FILE_FS_LABEL_INFORMATION: case SMB_FILE_FS_SIZE_INFORMATION: case SMB_FILE_FS_DEVICE_INFORMATION: case SMB_FILE_FS_ATTRIBUTE_INFORMATION: case SMB_FILE_FS_FULLSIZE_INFORMATION: case SMB_FILE_FS_OBJECTID_INFORMATION: case SMB_FILE_FS_DRIVERPATH_INFORMATION: smbsr_error(sr, NT_STATUS_NOT_SUPPORTED, ERRDOS, ERROR_NOT_SUPPORTED); return (SDRC_ERROR); default: smbsr_error(sr, NT_STATUS_INVALID_LEVEL, ERRDOS, ERROR_INVALID_LEVEL); return (SDRC_ERROR); } return (SDRC_SUCCESS); } /* * smb_trans2_set_fs_ctrl_info * * Only users with Admin privileges (i.e. of the BUILTIN/Administrators * group) will be allowed to set quotas. * * Currently QUOTAS are always ENFORCED and the default values * are always SMB_QUOTA_UNLIMITED (none). Any attempt to set * values other than these will result in NT_STATUS_NOT_SUPPORTED. */ static int smb_trans2_set_fs_ctrl_info(smb_request_t *sr, smb_xa_t *xa) { int rc; uint64_t fstart, fthresh, fstop, qthresh, qlimit; uint32_t qctrl, qpad; if (!smb_tree_has_feature(sr->tid_tree, SMB_TREE_QUOTA)) { smbsr_error(sr, NT_STATUS_NOT_SUPPORTED, ERRDOS, ERROR_NOT_SUPPORTED); return (-1); } if (!smb_user_is_admin(sr->uid_user)) { smbsr_error(sr, NT_STATUS_ACCESS_DENIED, ERRDOS, ERROR_ACCESS_DENIED); return (-1); } rc = smb_mbc_decodef(&xa->req_data_mb, "qqqqqll", &fstart, &fthresh, &fstop, &qthresh, &qlimit, &qctrl, &qpad); if ((rc != 0) || (fstart != 0) || (fthresh != 0) || (fstop != 0)) { smbsr_error(sr, NT_STATUS_INVALID_PARAMETER, ERRDOS, ERROR_INVALID_PARAMETER); return (-1); } /* Only support ENFORCED quotas with UNLIMITED default */ if ((qctrl != FILE_VC_QUOTA_ENFORCE) || (qlimit != SMB_QUOTA_UNLIMITED) || (qthresh != SMB_QUOTA_UNLIMITED)) { smbsr_error(sr, NT_STATUS_NOT_SUPPORTED, ERRDOS, ERROR_NOT_SUPPORTED); return (-1); } return (0); }