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 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
  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 }