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_flags = IVAL(buf, 20);
  50         lease->lease_duration = BVAL(buf, 24);
  51         lease->lease_version = version;
  52
  53         switch (version) {
  54         case 1:
  55                 break;
  56         case 2:
  57                 memcpy(&lease->parent_lease_key, buf+32, 16);
  58                 lease->lease_epoch = SVAL(buf, 48);
  59                 break;
  60         }
  61
  62         return len;
  63 }
  64
  65 bool smb2_lease_push(const struct smb2_lease *lease, uint8_t *buf, size_t len)
  66 {
  67         int version;
  68
  69         switch (len) {
  70         case 32:
  71                 version = 1;
  72                 break;
  73         case 52:
  74                 version = 2;
  75                 break;
  76         default:
  77                 return false;
  78         }
  79
  80         memcpy(&buf[0], &lease->lease_key, 16);
  81         SIVAL(buf, 16, lease->lease_state);
  82         SIVAL(buf, 20, lease->lease_flags);
  83         SBVAL(buf, 24, lease->lease_duration);
  84
  85         if (version == 2) {
  86                 memcpy(&buf[32], &lease->parent_lease_key, 16);
  87                 SIVAL(buf, 48, lease->lease_epoch);
  88         }
  89
  90         return true;
  91 }
  92
  93 bool smb2_lease_key_equal(const struct smb2_lease_key *k1,
  94                           const struct smb2_lease_key *k2)
  95 {
  96         return ((k1->data[0] == k2->data[0]) && (k1->data[1] == k2->data[1]));
  97 }
  98
  99 bool smb2_lease_equal(const struct GUID *g1,
 100                       const struct smb2_lease_key *k1,
 101                       const struct GUID *g2,
 102                       const struct smb2_lease_key *k2)
 103 {
 104         return GUID_equal(g1, g2) && smb2_lease_key_equal(k1, k2);
 105 }