r19598: Ahead of a merge to current lorikeet-heimdal:
[metze/samba/wip.git] / source4 / smb_server / smb2 / sesssetup.c
1 /* 
2    Unix SMB2 implementation.
3    
4    Copyright (C) Andrew Bartlett        2001-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 "auth/credentials/credentials.h"
24 #include "auth/gensec/gensec.h"
25 #include "auth/auth.h"
26 #include "libcli/smb2/smb2.h"
27 #include "libcli/smb2/smb2_calls.h"
28 #include "smb_server/smb_server.h"
29 #include "smb_server/service_smb_proto.h"
30 #include "smb_server/smb2/smb2_server.h"
31 #include "smbd/service_stream.h"
32
33 static void smb2srv_sesssetup_send(struct smb2srv_request *req, union smb_sesssetup *io)
34 {
35         uint16_t unknown1;
36
37         if (NT_STATUS_IS_OK(req->status)) {
38                 unknown1 = 0x0003;
39         } else if (NT_STATUS_EQUAL(req->status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
40                 unknown1 = 0x0002;
41         } else {
42                 smb2srv_send_error(req, req->status);
43                 return;
44         }
45
46         SMB2SRV_CHECK(smb2srv_setup_reply(req, 0x08, True, io->smb2.out.secblob.length));
47
48         SSVAL(req->out.hdr, SMB2_HDR_UNKNOWN1,  unknown1);
49         SBVAL(req->out.hdr, SMB2_HDR_UID,       io->smb2.out.uid);
50
51         SSVAL(req->out.body, 0x02, io->smb2.out._pad);
52         SMB2SRV_CHECK(smb2_push_o16s16_blob(&req->out, 0x04, io->smb2.out.secblob));
53
54         smb2srv_send_reply(req);
55 }
56
57 struct smb2srv_sesssetup_callback_ctx {
58         struct smb2srv_request *req;
59         union smb_sesssetup *io;
60         struct smbsrv_session *smb_sess;
61 };
62
63 static void smb2srv_sesssetup_callback(struct gensec_update_request *greq, void *private_data)
64 {
65         struct smb2srv_sesssetup_callback_ctx *ctx = talloc_get_type(private_data,
66                                                      struct smb2srv_sesssetup_callback_ctx);
67         struct smb2srv_request *req = ctx->req;
68         union smb_sesssetup *io = ctx->io;
69         struct smbsrv_session *smb_sess = ctx->smb_sess;
70         struct auth_session_info *session_info = NULL;
71         NTSTATUS status;
72
73         status = gensec_update_recv(greq, req, &io->smb2.out.secblob);
74         if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
75                 goto done;
76         } else if (!NT_STATUS_IS_OK(status)) {
77                 goto failed;
78         }
79
80         status = gensec_session_info(smb_sess->gensec_ctx, &session_info);
81         if (!NT_STATUS_IS_OK(status)) {
82                 goto failed;
83         }
84
85         /* Ensure this is marked as a 'real' vuid, not one
86          * simply valid for the session setup leg */
87         status = smbsrv_session_sesssetup_finished(smb_sess, session_info);
88         if (!NT_STATUS_IS_OK(status)) {
89                 goto failed;
90         }
91         req->session = smb_sess;
92
93 done:
94         io->smb2.out.uid = smb_sess->vuid;
95 failed:
96         req->status = auth_nt_status_squash(status);
97         smb2srv_sesssetup_send(req, io);
98 }
99
100 static void smb2srv_sesssetup_backend(struct smb2srv_request *req, union smb_sesssetup *io)
101 {
102         NTSTATUS status;
103         struct smb2srv_sesssetup_callback_ctx *callback_ctx;
104         struct smbsrv_session *smb_sess = NULL;
105         uint64_t vuid;
106
107         io->smb2.out._pad       = 0;
108         io->smb2.out.uid        = 0;
109         io->smb2.out.secblob = data_blob(NULL, 0);
110
111         vuid = BVAL(req->in.hdr, SMB2_HDR_UID);
112
113         /*
114          * only when we got '0' we should allocate a new session
115          */
116         if (vuid == 0) {
117                 struct gensec_security *gensec_ctx;
118
119                 status = gensec_server_start(req,
120                                              req->smb_conn->connection->event.ctx,
121                                              req->smb_conn->connection->msg_ctx,
122                                              &gensec_ctx);
123                 if (!NT_STATUS_IS_OK(status)) {
124                         DEBUG(1, ("Failed to start GENSEC server code: %s\n", nt_errstr(status)));
125                         goto failed;
126                 }
127
128                 gensec_set_credentials(gensec_ctx, req->smb_conn->negotiate.server_credentials);
129
130                 gensec_set_target_service(gensec_ctx, "cifs");
131
132                 gensec_want_feature(gensec_ctx, GENSEC_FEATURE_SESSION_KEY);
133
134                 status = gensec_start_mech_by_oid(gensec_ctx, GENSEC_OID_SPNEGO);
135                 if (!NT_STATUS_IS_OK(status)) {
136                         DEBUG(1, ("Failed to start GENSEC SPNEGO server code: %s\n", nt_errstr(status)));
137                         goto failed;
138                 }
139
140                 /* allocate a new session */
141                 smb_sess = smbsrv_session_new(req->smb_conn, gensec_ctx);
142                 if (!smb_sess) {
143                         status = NT_STATUS_INSUFFICIENT_RESOURCES;
144                         goto failed;
145                 }
146                 status = smbsrv_smb2_init_tcons(smb_sess);
147                 if (!NT_STATUS_IS_OK(status)) {
148                         goto failed;
149                 }
150         } else {
151                 /* lookup an existing session */
152                 smb_sess = smbsrv_session_find_sesssetup(req->smb_conn, vuid);
153         }
154
155         if (!smb_sess) {
156                 status = NT_STATUS_USER_SESSION_DELETED;
157                 goto failed;
158         }
159
160         if (!smb_sess->gensec_ctx) {
161                 status = NT_STATUS_INTERNAL_ERROR;
162                 DEBUG(1, ("Internal ERROR: no gensec_ctx on session: %s\n", nt_errstr(status)));
163                 goto failed;
164         }
165
166         callback_ctx = talloc(req, struct smb2srv_sesssetup_callback_ctx);
167         if (!callback_ctx) goto nomem;
168         callback_ctx->req       = req;
169         callback_ctx->io        = io;
170         callback_ctx->smb_sess  = smb_sess;
171
172         gensec_update_send(smb_sess->gensec_ctx, io->smb2.in.secblob,
173                            smb2srv_sesssetup_callback, callback_ctx);
174         return;
175 nomem:
176         status = NT_STATUS_NO_MEMORY;
177 failed:
178         talloc_free(smb_sess);
179         req->status = auth_nt_status_squash(status);
180         smb2srv_sesssetup_send(req, io);
181 }
182
183 void smb2srv_sesssetup_recv(struct smb2srv_request *req)
184 {
185         union smb_sesssetup *io;
186
187         SMB2SRV_CHECK_BODY_SIZE(req, 0x18, True);
188         SMB2SRV_TALLOC_IO_PTR(io, union smb_sesssetup);
189
190         io->smb2.level          = RAW_SESSSETUP_SMB2;
191         io->smb2.in._pad        = SVAL(req->in.body, 0x02);
192         io->smb2.in.unknown2    = IVAL(req->in.body, 0x04);
193         io->smb2.in.unknown3    = IVAL(req->in.body, 0x08);
194         SMB2SRV_CHECK(smb2_pull_o16s16_blob(&req->in, io, req->in.body+0x0C, &io->smb2.in.secblob));
195         io->smb2.in.unknown4    = BVAL(req->in.body, 0x10);
196
197         smb2srv_sesssetup_backend(req, io);
198 }
199
200 static NTSTATUS smb2srv_logoff_backend(struct smb2srv_request *req)
201 {
202         /* TODO: call ntvfs backends to close file of this session */
203         talloc_free(req->session);
204         req->session = NULL;
205         return NT_STATUS_OK;
206 }
207
208 static void smb2srv_logoff_send(struct smb2srv_request *req)
209 {
210         if (NT_STATUS_IS_ERR(req->status)) {
211                 smb2srv_send_error(req, req->status);
212                 return;
213         }
214
215         SMB2SRV_CHECK(smb2srv_setup_reply(req, 0x04, False, 0));
216
217         SSVAL(req->out.body, 0x02, 0);
218
219         smb2srv_send_reply(req);
220 }
221
222 void smb2srv_logoff_recv(struct smb2srv_request *req)
223 {
224         uint16_t _pad;
225
226         SMB2SRV_CHECK_BODY_SIZE(req, 0x04, False);
227
228         _pad    = SVAL(req->in.body, 0x02);
229
230         req->status = smb2srv_logoff_backend(req);
231
232         if (req->control_flags & SMB2SRV_REQ_CTRL_FLAG_NOT_REPLY) {
233                 talloc_free(req);
234                 return;
235         }
236         smb2srv_logoff_send(req);
237 }