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
  47         switch (version) {
  48         case 1:
  49                 ZERO_STRUCT(lease->parent_lease_key);
  50                 lease->lease_epoch = 0;
  51                 break;
  52         case 2:
  53                 memcpy(&lease->parent_lease_key, buf+32, 16);
  54                 lease->lease_epoch = SVAL(buf, 48);
  55                 break;
  56         }
  57
  58         return len;
  59 }
  60
  61 bool smb2_lease_push(const struct smb2_lease *lease, uint8_t *buf, size_t len)
  62 {
  63         int version;
  64
  65         switch (len) {
  66         case 32:
  67                 version = 1;
  68                 break;
  69         case 52:
  70                 version = 2;
  71                 break;
  72         default:
  73                 return false;
  74         }
  75
  76         memcpy(&buf[0], &lease->lease_key, 16);
  77         SIVAL(buf, 16, lease->lease_state);
  78         SIVAL(buf, 20, lease->lease_flags);
  79         SBVAL(buf, 24, lease->lease_duration);
  80
  81         if (version == 2) {
  82                 memcpy(&buf[32], &lease->parent_lease_key, 16);
  83                 SIVAL(buf, 48, lease->lease_epoch);
  84         }
  85
  86         return true;
  87 }