libcli/smb/smb2_lease.c

   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
  26 /**
  27  * Pull a lease off the wire into a struct smb2_lease.
  28  *
  29  * Note: the caller MUST zero initialize "lease".
  30  **/
  31 ssize_t smb2_lease_pull(const uint8_t *buf, size_t len,
  32                         struct smb2_lease *lease)
  33 {
  34         int version;
  35
  36         switch (len) {
  37         case 32:
  38                 version = 1;
  39                 break;
  40         case 52:
  41                 version = 2;
  42                 break;
  43         default:
  44                 return -1;
  45         }
  46
  47         memcpy(&lease->lease_key, buf, 16);
  48         lease->lease_state = IVAL(buf, 16);
  49         lease->lease_version = version;
  50
  51         switch (version) {
  52         case 1:
  53                 break;
  54         case 2:
  55                 lease->lease_flags = IVAL(buf, 20);
  56                 lease->lease_flags &= SMB2_LEASE_FLAG_PARENT_LEASE_KEY_SET;
  57                 if (lease->lease_flags & SMB2_LEASE_FLAG_PARENT_LEASE_KEY_SET) {
  58                         memcpy(&lease->parent_lease_key, buf+32, 16);
  59                 }
  60                 lease->lease_duration = BVAL(buf, 24);
  61                 lease->lease_epoch = SVAL(buf, 48);
  62                 break;
  63         }
  64
  65         return len;
  66 }
  67
  68 bool smb2_lease_push(const struct smb2_lease *lease, uint8_t *buf, size_t len)
  69 {
  70         int version;
  71
  72         switch (len) {
  73         case 32:
  74                 version = 1;
  75                 break;
  76         case 52:
  77                 version = 2;
  78                 break;
  79         default:
  80                 return false;
  81         }
  82
  83         memcpy(&buf[0], &lease->lease_key, 16);
  84         SIVAL(buf, 16, lease->lease_state);
  85         SIVAL(buf, 20, lease->lease_flags);
  86         SBVAL(buf, 24, lease->lease_duration);
  87
  88         if (version == 2) {
  89                 memcpy(&buf[32], &lease->parent_lease_key, 16);
  90                 SIVAL(buf, 48, lease->lease_epoch);
  91         }
  92
  93         return true;
  94 }
  95
  96 bool smb2_lease_key_equal(const struct smb2_lease_key *k1,
  97                           const struct smb2_lease_key *k2)
  98 {
  99         return ((k1->data[0] == k2->data[0]) && (k1->data[1] == k2->data[1]));
 100 }
 101
 102 bool smb2_lease_equal(const struct GUID *g1,
 103                       const struct smb2_lease_key *k1,
 104                       const struct GUID *g2,
 105                       const struct smb2_lease_key *k2)
 106 {
 107         return GUID_equal(g1, g2) && smb2_lease_key_equal(k1, k2);
 108 }