1 /* 2 Unix SMB/CIFS implementation. 3 Lease utility functions 4 5 Copyright (C) Jeremy Allison 2017. 6 Copyright (C) Stefan (metze) Metzmacher 2017. 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 #define DBGC_CLASS DBGC_LOCKING 23 #include "includes.h" 24 #include "../librpc/gen_ndr/open_files.h" 25 #include "locking/proto.h" 26 #include "smbd/globals.h" 27 #include "locking/leases_db.h" 28 map_oplock_to_lease_type(uint16_t op_type)29uint32_t map_oplock_to_lease_type(uint16_t op_type) 30 { 31 uint32_t ret; 32 33 switch(op_type) { 34 case BATCH_OPLOCK: 35 case BATCH_OPLOCK|EXCLUSIVE_OPLOCK: 36 ret = SMB2_LEASE_READ|SMB2_LEASE_WRITE|SMB2_LEASE_HANDLE; 37 break; 38 case EXCLUSIVE_OPLOCK: 39 ret = SMB2_LEASE_READ|SMB2_LEASE_WRITE; 40 break; 41 case LEVEL_II_OPLOCK: 42 ret = SMB2_LEASE_READ; 43 break; 44 default: 45 ret = SMB2_LEASE_NONE; 46 break; 47 } 48 return ret; 49 } 50 fsp_lease_type(struct files_struct * fsp)51uint32_t fsp_lease_type(struct files_struct *fsp) 52 { 53 NTSTATUS status; 54 55 if (fsp->oplock_type != LEASE_OPLOCK) { 56 uint32_t type = map_oplock_to_lease_type(fsp->oplock_type); 57 return type; 58 } 59 60 status = leases_db_get_current_state( 61 fsp_client_guid(fsp), 62 &fsp->lease->lease.lease_key, 63 &fsp->leases_db_seqnum, 64 &fsp->lease_type); 65 if (!NT_STATUS_IS_OK(status)) { 66 DBG_DEBUG("leases_db_get_current_state failed: %s\n", 67 nt_errstr(status)); 68 fsp->lease_type = 0; /* no lease */ 69 } 70 71 return fsp->lease_type; 72 } 73 lease_type_is_exclusive(uint32_t lease_type)74static uint32_t lease_type_is_exclusive(uint32_t lease_type) 75 { 76 if ((lease_type & (SMB2_LEASE_READ | SMB2_LEASE_WRITE)) == 77 (SMB2_LEASE_READ | SMB2_LEASE_WRITE)) { 78 return true; 79 } 80 81 return false; 82 } 83 fsp_lease_type_is_exclusive(struct files_struct * fsp)84bool fsp_lease_type_is_exclusive(struct files_struct *fsp) 85 { 86 uint32_t lease_type = fsp_lease_type(fsp); 87 88 return lease_type_is_exclusive(lease_type); 89 } 90 fsp_client_guid(const files_struct * fsp)91const struct GUID *fsp_client_guid(const files_struct *fsp) 92 { 93 return &fsp->conn->sconn->client->connections->smb2.client.guid; 94 } 95