1 /*
2    Unix SMB/CIFS implementation.
3 
4    SMB2 Lease context handling
5 
6    Copyright (C) Stefan Metzmacher 2012
7    Copyright (C) Volker Lendecke 2013
8 
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 3 of the License, or
12    (at your option) any later version.
13 
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18 
19    You should have received a copy of the GNU General Public License
20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
21 */
22 
23 #include "includes.h"
24 #include "../libcli/smb/smb_common.h"
25 
smb2_lease_pull(const uint8_t * buf,size_t len,struct smb2_lease * lease)26 ssize_t smb2_lease_pull(const uint8_t *buf, size_t len,
27 			struct smb2_lease *lease)
28 {
29 	int version;
30 
31 	switch (len) {
32 	case 32:
33 		version = 1;
34 		break;
35 	case 52:
36 		version = 2;
37 		break;
38 	default:
39 		return -1;
40 	}
41 
42 	memcpy(&lease->lease_key, buf, 16);
43 	lease->lease_state = IVAL(buf, 16);
44 	lease->lease_flags = IVAL(buf, 20);
45 	lease->lease_duration = BVAL(buf, 24);
46 	lease->lease_version = version;
47 
48 	switch (version) {
49 	case 1:
50 		ZERO_STRUCT(lease->parent_lease_key);
51 		lease->lease_epoch = 0;
52 		break;
53 	case 2:
54 		memcpy(&lease->parent_lease_key, buf+32, 16);
55 		lease->lease_epoch = SVAL(buf, 48);
56 		break;
57 	}
58 
59 	return len;
60 }
61 
smb2_lease_push(const struct smb2_lease * lease,uint8_t * buf,size_t len)62 bool smb2_lease_push(const struct smb2_lease *lease, uint8_t *buf, size_t len)
63 {
64 	int version;
65 
66 	switch (len) {
67 	case 32:
68 		version = 1;
69 		break;
70 	case 52:
71 		version = 2;
72 		break;
73 	default:
74 		return false;
75 	}
76 
77 	memcpy(&buf[0], &lease->lease_key, 16);
78 	SIVAL(buf, 16, lease->lease_state);
79 	SIVAL(buf, 20, lease->lease_flags);
80 	SBVAL(buf, 24, lease->lease_duration);
81 
82 	if (version == 2) {
83 		memcpy(&buf[32], &lease->parent_lease_key, 16);
84 		SIVAL(buf, 48, lease->lease_epoch);
85 	}
86 
87 	return true;
88 }
89 
smb2_lease_key_equal(const struct smb2_lease_key * k1,const struct smb2_lease_key * k2)90 bool smb2_lease_key_equal(const struct smb2_lease_key *k1,
91 			  const struct smb2_lease_key *k2)
92 {
93 	return ((k1->data[0] == k2->data[0]) && (k1->data[1] == k2->data[1]));
94 }
95 
smb2_lease_equal(const struct GUID * g1,const struct smb2_lease_key * k1,const struct GUID * g2,const struct smb2_lease_key * k2)96 bool smb2_lease_equal(const struct GUID *g1,
97 		      const struct smb2_lease_key *k1,
98 		      const struct GUID *g2,
99 		      const struct smb2_lease_key *k2)
100 {
101 	return GUID_equal(g1, g2) && smb2_lease_key_equal(k1, k2);
102 }
103