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)29 uint32_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)51 uint32_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)74 static 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)84 bool 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)91 const struct GUID *fsp_client_guid(const files_struct *fsp)
92 {
93 	return &fsp->conn->sconn->client->connections->smb2.client.guid;
94 }
95