source4/smb_server/smb2/receive.c

   1 /*
   2    Unix SMB2 implementation.
   3
   4    Copyright (C) Andrew Tridgell        2005
   5    Copyright (C) Stefan Metzmacher      2005
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program; if not, write to the Free Software
  19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "includes.h"
  23 #include "system/time.h"
  24 #include "smbd/service_stream.h"
  25 #include "libcli/smb2/smb2.h"
  26 #include "smb_server/smb_server.h"
  27 #include "smb_server/smb2/smb2_server.h"
  28 #include "lib/stream/packet.h"
  29
  30
  31 static struct smb2srv_request *smb2srv_init_request(struct smbsrv_connection *smb_conn)
  32 {
  33         struct smb2srv_request *req;
  34
  35         req = talloc_zero(smb_conn, struct smb2srv_request);
  36         if (!req) return NULL;
  37
  38         req->smb_conn = smb_conn;
  39
  40         return req;
  41 }
  42
  43 NTSTATUS smb2srv_setup_reply(struct smb2srv_request *req, uint_t body_fixed_size, uint_t body_dynamic_size)
  44 {
  45         req->out.size           = SMB2_HDR_BODY+NBT_HDR_SIZE+body_fixed_size;
  46
  47         req->out.allocated      = req->out.size + body_dynamic_size;
  48         req->out.buffer         = talloc_size(req, req->out.allocated);
  49         NT_STATUS_HAVE_NO_MEMORY(req->out.buffer);
  50
  51         req->out.hdr            = req->out.buffer       + NBT_HDR_SIZE;
  52         req->out.body           = req->out.hdr          + SMB2_HDR_BODY;
  53         req->out.body_size      = body_fixed_size;
  54         req->out.dynamic        = (body_dynamic_size ? req->out.body + body_fixed_size : NULL);
  55
  56         SIVAL(req->out.hdr, 0,                SMB2_MAGIC);
  57         SSVAL(req->out.hdr, SMB2_HDR_LENGTH,  SMB2_HDR_BODY);
  58         SSVAL(req->out.hdr, SMB2_HDR_PAD1,    0);
  59         SIVAL(req->out.hdr, SMB2_HDR_STATUS,  NT_STATUS_V(req->status));
  60         SSVAL(req->out.hdr, SMB2_HDR_OPCODE,  SVAL(req->in.hdr, SMB2_HDR_OPCODE));
  61         SSVAL(req->out.hdr, SMB2_HDR_PAD2,    0);
  62         SIVAL(req->out.hdr, SMB2_HDR_FLAGS,   0x00000001);
  63         SIVAL(req->out.hdr, SMB2_HDR_UNKNOWN, 0);
  64         SBVAL(req->out.hdr, SMB2_HDR_SEQNUM,  req->seqnum);
  65         SIVAL(req->out.hdr, SMB2_HDR_PID,     IVAL(req->in.hdr, SMB2_HDR_PID));
  66         SIVAL(req->out.hdr, SMB2_HDR_TID,     IVAL(req->in.hdr, SMB2_HDR_TID));
  67         SBVAL(req->out.hdr, SMB2_HDR_UID,     BVAL(req->in.hdr, SMB2_HDR_UID));
  68         memset(req->out.hdr+SMB2_HDR_SIG, 0, 16);
  69
  70         /* set the length of the fixed body part and +1 if there's a dynamic part also */
  71         SSVAL(req->out.body, 0, body_fixed_size + (body_dynamic_size?1:0));
  72
  73         /*
  74          * if we have a dynamic part, make sure the first byte
  75          * which is always be part of the packet is initialized
  76          */
  77         if (body_dynamic_size) {
  78                 SCVAL(req->out.dynamic, 0, 0);
  79         }
  80
  81         return NT_STATUS_OK;
  82 }
  83
  84 void smb2srv_send_reply(struct smb2srv_request *req)
  85 {
  86         DATA_BLOB blob;
  87         NTSTATUS status;
  88
  89         if (req->out.size > NBT_HDR_SIZE) {
  90                 _smb_setlen(req->out.buffer, req->out.size - NBT_HDR_SIZE);
  91         }
  92
  93         blob = data_blob_const(req->out.buffer, req->out.size);
  94         status = packet_send(req->smb_conn->packet, blob);
  95         if (!NT_STATUS_IS_OK(status)) {
  96                 smbsrv_terminate_connection(req->smb_conn, nt_errstr(status));
  97         }
  98         talloc_free(req);
  99 }
 100
 101 void smb2srv_send_error(struct smb2srv_request *req, NTSTATUS error)
 102 {
 103         NTSTATUS status;
 104
 105         status = smb2srv_setup_reply(req, 8, 1);
 106         if (!NT_STATUS_IS_OK(status)) {
 107                 smbsrv_terminate_connection(req->smb_conn, nt_errstr(status));
 108                 talloc_free(req);
 109                 return;
 110         }
 111
 112         SIVAL(req->out.hdr, SMB2_HDR_STATUS, NT_STATUS_V(error));
 113
 114         SSVAL(req->out.body, 0x02, 0);
 115         SIVAL(req->out.body, 0x04, 0);
 116
 117         smb2srv_send_reply(req);
 118 }
 119
 120 static NTSTATUS smb2srv_reply(struct smb2srv_request *req)
 121 {
 122         uint16_t opcode;
 123
 124         opcode          = SVAL(req->in.hdr, SMB2_HDR_OPCODE);
 125         req->seqnum     = BVAL(req->in.hdr, SMB2_HDR_SEQNUM);
 126
 127         errno = 0;
 128
 129         switch (opcode) {
 130         case SMB2_OP_NEGPROT:
 131                 smb2srv_negprot_recv(req);
 132                 return NT_STATUS_OK;
 133         case SMB2_OP_SESSSETUP:
 134                 smb2srv_sesssetup_recv(req);
 135                 return NT_STATUS_OK;
 136         case SMB2_OP_LOGOFF:
 137                 smb2srv_logoff_recv(req);
 138                 return NT_STATUS_OK;
 139         case SMB2_OP_TCON:
 140                 smb2srv_tcon_recv(req);
 141                 return NT_STATUS_OK;
 142         case SMB2_OP_TDIS:
 143                 smb2srv_tdis_recv(req);
 144                 return NT_STATUS_OK;
 145         case SMB2_OP_CREATE:
 146                 smb2srv_create_recv(req);
 147                 return NT_STATUS_OK;
 148         case SMB2_OP_CLOSE:
 149                 smb2srv_close_recv(req);
 150                 return NT_STATUS_OK;
 151         case SMB2_OP_FLUSH:
 152                 smb2srv_flush_recv(req);
 153                 return NT_STATUS_OK;
 154         case SMB2_OP_READ:
 155                 smb2srv_read_recv(req);
 156                 return NT_STATUS_OK;
 157         case SMB2_OP_WRITE:
 158                 smb2srv_write_recv(req);
 159                 return NT_STATUS_OK;
 160         case SMB2_OP_LOCK:
 161                 smb2srv_lock_recv(req);
 162                 return NT_STATUS_OK;
 163         case SMB2_OP_IOCTL:
 164                 smb2srv_ioctl_recv(req);
 165                 return NT_STATUS_OK;
 166         case SMB2_OP_CANCEL:
 167                 smb2srv_cancel_recv(req);
 168                 return NT_STATUS_OK;
 169         case SMB2_OP_KEEPALIVE:
 170                 smb2srv_keepalive_recv(req);
 171                 return NT_STATUS_OK;
 172         case SMB2_OP_FIND:
 173                 smb2srv_find_recv(req);
 174                 return NT_STATUS_OK;
 175         case SMB2_OP_NOTIFY:
 176                 smb2srv_notify_recv(req);
 177                 return NT_STATUS_OK;
 178         case SMB2_OP_GETINFO:
 179                 smb2srv_getinfo_recv(req);
 180                 return NT_STATUS_OK;
 181         case SMB2_OP_SETINFO:
 182                 smb2srv_setinfo_recv(req);
 183                 return NT_STATUS_OK;
 184         case SMB2_OP_BREAK:
 185                 smb2srv_break_recv(req);
 186                 return NT_STATUS_OK;
 187         }
 188
 189         DEBUG(1,("Invalid SMB2 opcode: 0x%04X\n", opcode));
 190         smbsrv_terminate_connection(req->smb_conn, "Invalid SMB2 opcode");
 191         return NT_STATUS_OK;
 192 }
 193
 194 NTSTATUS smbsrv_recv_smb2_request(void *private, DATA_BLOB blob)
 195 {
 196         struct smbsrv_connection *smb_conn = talloc_get_type(private, struct smbsrv_connection);
 197         struct smb2srv_request *req;
 198         uint32_t protocol_version;
 199         uint16_t buffer_code;
 200         uint32_t dynamic_size;
 201
 202         /* see if its a special NBT packet */
 203         if (CVAL(blob.data,0) != 0) {
 204                 DEBUG(2,("Special NBT packet on SMB2 connection"));
 205                 smbsrv_terminate_connection(smb_conn, "Special NBT packet on SMB2 connection");
 206                 return NT_STATUS_OK;
 207         }
 208
 209         if (blob.length < (NBT_HDR_SIZE + SMB2_MIN_SIZE)) {
 210                 DEBUG(2,("Invalid SMB2 packet length count %ld\n", (long)blob.length));
 211                 smbsrv_terminate_connection(smb_conn, "Invalid SMB2 packet");
 212                 return NT_STATUS_OK;
 213         }
 214
 215         protocol_version = IVAL(blob.data, NBT_HDR_SIZE);
 216
 217         if (protocol_version != SMB2_MAGIC) {
 218                 DEBUG(2,("Invalid SMB packet: protocl prefix: 0x%08X\n", protocol_version));
 219                 smbsrv_terminate_connection(smb_conn, "NON-SMB2 packet");
 220                 return NT_STATUS_OK;
 221         }
 222
 223         req = smb2srv_init_request(smb_conn);
 224         NT_STATUS_HAVE_NO_MEMORY(req);
 225
 226         req->in.buffer          = talloc_steal(req, blob.data);
 227         req->in.size            = blob.length;
 228         req->request_time       = timeval_current();
 229         req->in.allocated       = req->in.size;
 230
 231         req->in.hdr             = req->in.buffer+ NBT_HDR_SIZE;
 232         req->in.body            = req->in.hdr   + SMB2_HDR_BODY;
 233         req->in.body_size       = req->in.size  - (SMB2_HDR_BODY+NBT_HDR_SIZE);
 234         req->in.dynamic         = NULL;
 235
 236         buffer_code             = SVAL(req->in.body, 0);
 237         dynamic_size            = req->in.body_size - (buffer_code & ~1);
 238
 239         if (dynamic_size != 0 && (buffer_code & 1)) {
 240                 req->in.dynamic = req->in.body + (buffer_code & ~1);
 241                 if (smb2_oob(&req->in, req->in.dynamic, dynamic_size)) {
 242                         DEBUG(1,("SMB2 request invalid dynamic size 0x%x\n",
 243                                  dynamic_size));
 244                         smb2srv_send_error(req, NT_STATUS_INVALID_PARAMETER);
 245                         return NT_STATUS_OK;
 246                 }
 247         }
 248
 249         /*
 250          * TODO: - make sure the length field is 64
 251          *       - make sure it's a request
 252          */
 253
 254         return smb2srv_reply(req);
 255 }
 256
 257 /*
 258  * init the SMB2 protocol related stuff
 259  */
 260 NTSTATUS smbsrv_init_smb2_connection(struct smbsrv_connection *smb_conn)
 261 {
 262         NTSTATUS status;
 263
 264         /* now initialise a few default values associated with this smb socket */
 265         smb_conn->negotiate.max_send = 0xFFFF;
 266
 267         /* this is the size that w2k uses, and it appears to be important for
 268            good performance */
 269         smb_conn->negotiate.max_recv = lp_max_xmit();
 270
 271         smb_conn->negotiate.zone_offset = get_time_zone(time(NULL));
 272
 273         smb_conn->config.security = SEC_USER;
 274         smb_conn->config.nt_status_support = True;
 275
 276         status = smbsrv_init_sessions(smb_conn, UINT64_MAX);
 277         NT_STATUS_NOT_OK_RETURN(status);
 278
 279         status = smbsrv_init_tcons(smb_conn, UINT32_MAX);
 280         NT_STATUS_NOT_OK_RETURN(status);
 281
 282         return NT_STATUS_OK;
 283
 284 }