Until the new ldb changes land, make ldb_wait set the error string.
[jelmer/samba4-debian.git] / source / auth / gensec / spnego.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    RFC2478 Compliant SPNEGO implementation
5    
6    Copyright (C) Jim McDonough <jmcd@us.ibm.com>      2003
7    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
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    
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "includes.h"
25 #include "auth/gensec/spnego.h"
26 #include "librpc/gen_ndr/ndr_dcerpc.h"
27 #include "auth/credentials/credentials.h"
28 #include "auth/gensec/gensec.h"
29
30 enum spnego_state_position {
31         SPNEGO_SERVER_START,
32         SPNEGO_CLIENT_START,
33         SPNEGO_SERVER_TARG,
34         SPNEGO_CLIENT_TARG,
35         SPNEGO_FALLBACK,
36         SPNEGO_DONE
37 };
38
39 struct spnego_state {
40         enum spnego_message_type expected_packet;
41         enum spnego_state_position state_position;
42         struct gensec_security *sub_sec_security;
43         bool no_response_expected;
44
45         const char *neg_oid;
46 };
47
48
49 static NTSTATUS gensec_spnego_client_start(struct gensec_security *gensec_security)
50 {
51         struct spnego_state *spnego_state;
52
53         spnego_state = talloc(gensec_security, struct spnego_state);
54         if (!spnego_state) {
55                 return NT_STATUS_NO_MEMORY;
56         }
57
58         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
59         spnego_state->state_position = SPNEGO_CLIENT_START;
60         spnego_state->sub_sec_security = NULL;
61         spnego_state->no_response_expected = false;
62
63         gensec_security->private_data = spnego_state;
64         return NT_STATUS_OK;
65 }
66
67 static NTSTATUS gensec_spnego_server_start(struct gensec_security *gensec_security)
68 {
69         struct spnego_state *spnego_state;
70
71         spnego_state = talloc(gensec_security, struct spnego_state);            
72         if (!spnego_state) {
73                 return NT_STATUS_NO_MEMORY;
74         }
75
76         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
77         spnego_state->state_position = SPNEGO_SERVER_START;
78         spnego_state->sub_sec_security = NULL;
79         spnego_state->no_response_expected = false;
80
81         gensec_security->private_data = spnego_state;
82         return NT_STATUS_OK;
83 }
84
85 /*
86   wrappers for the spnego_*() functions
87 */
88 static NTSTATUS gensec_spnego_unseal_packet(struct gensec_security *gensec_security, 
89                                             TALLOC_CTX *mem_ctx, 
90                                             uint8_t *data, size_t length, 
91                                             const uint8_t *whole_pdu, size_t pdu_length, 
92                                             const DATA_BLOB *sig)
93 {
94         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
95
96         if (spnego_state->state_position != SPNEGO_DONE 
97             && spnego_state->state_position != SPNEGO_FALLBACK) {
98                 return NT_STATUS_INVALID_PARAMETER;
99         }
100         
101         return gensec_unseal_packet(spnego_state->sub_sec_security, 
102                                     mem_ctx, 
103                                     data, length, 
104                                     whole_pdu, pdu_length,
105                                     sig); 
106 }
107
108 static NTSTATUS gensec_spnego_check_packet(struct gensec_security *gensec_security, 
109                                            TALLOC_CTX *mem_ctx, 
110                                            const uint8_t *data, size_t length, 
111                                            const uint8_t *whole_pdu, size_t pdu_length, 
112                                            const DATA_BLOB *sig)
113 {
114         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
115
116         if (spnego_state->state_position != SPNEGO_DONE 
117             && spnego_state->state_position != SPNEGO_FALLBACK) {
118                 return NT_STATUS_INVALID_PARAMETER;
119         }
120         
121         return gensec_check_packet(spnego_state->sub_sec_security, 
122                                    mem_ctx, 
123                                    data, length, 
124                                    whole_pdu, pdu_length,
125                                    sig);
126 }
127
128 static NTSTATUS gensec_spnego_seal_packet(struct gensec_security *gensec_security, 
129                                           TALLOC_CTX *mem_ctx, 
130                                           uint8_t *data, size_t length, 
131                                           const uint8_t *whole_pdu, size_t pdu_length, 
132                                           DATA_BLOB *sig)
133 {
134         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
135
136         if (spnego_state->state_position != SPNEGO_DONE 
137             && spnego_state->state_position != SPNEGO_FALLBACK) {
138                 return NT_STATUS_INVALID_PARAMETER;
139         }
140         
141         return gensec_seal_packet(spnego_state->sub_sec_security, 
142                                   mem_ctx, 
143                                   data, length, 
144                                   whole_pdu, pdu_length,
145                                   sig);
146 }
147
148 static NTSTATUS gensec_spnego_sign_packet(struct gensec_security *gensec_security, 
149                                           TALLOC_CTX *mem_ctx, 
150                                           const uint8_t *data, size_t length, 
151                                           const uint8_t *whole_pdu, size_t pdu_length, 
152                                           DATA_BLOB *sig)
153 {
154         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
155
156         if (spnego_state->state_position != SPNEGO_DONE 
157             && spnego_state->state_position != SPNEGO_FALLBACK) {
158                 return NT_STATUS_INVALID_PARAMETER;
159         }
160         
161         return gensec_sign_packet(spnego_state->sub_sec_security, 
162                                   mem_ctx, 
163                                   data, length, 
164                                   whole_pdu, pdu_length,
165                                   sig);
166 }
167
168 static NTSTATUS gensec_spnego_wrap(struct gensec_security *gensec_security, 
169                                    TALLOC_CTX *mem_ctx, 
170                                    const DATA_BLOB *in, 
171                                    DATA_BLOB *out)
172 {
173         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
174
175         if (spnego_state->state_position != SPNEGO_DONE 
176             && spnego_state->state_position != SPNEGO_FALLBACK) {
177                 DEBUG(1, ("gensec_spnego_wrap: wrong state for wrap\n"));
178                 return NT_STATUS_INVALID_PARAMETER;
179         }
180         
181         return gensec_wrap(spnego_state->sub_sec_security, 
182                            mem_ctx, in, out);
183 }
184
185 static NTSTATUS gensec_spnego_unwrap(struct gensec_security *gensec_security, 
186                                      TALLOC_CTX *mem_ctx, 
187                                      const DATA_BLOB *in, 
188                                      DATA_BLOB *out)
189 {
190         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
191
192         if (spnego_state->state_position != SPNEGO_DONE 
193             && spnego_state->state_position != SPNEGO_FALLBACK) {
194                 DEBUG(1, ("gensec_spnego_unwrap: wrong state for unwrap\n"));
195                 return NT_STATUS_INVALID_PARAMETER;
196         }
197         
198         return gensec_unwrap(spnego_state->sub_sec_security, 
199                              mem_ctx, in, out);
200 }
201
202 static NTSTATUS gensec_spnego_wrap_packets(struct gensec_security *gensec_security, 
203                                            TALLOC_CTX *mem_ctx, 
204                                            const DATA_BLOB *in, 
205                                            DATA_BLOB *out,
206                                            size_t *len_processed) 
207 {
208         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
209
210         if (spnego_state->state_position != SPNEGO_DONE 
211             && spnego_state->state_position != SPNEGO_FALLBACK) {
212                 DEBUG(1, ("gensec_spnego_wrap: wrong state for wrap\n"));
213                 return NT_STATUS_INVALID_PARAMETER;
214         }
215         
216         return gensec_wrap_packets(spnego_state->sub_sec_security, 
217                                    mem_ctx, in, out,
218                                    len_processed);
219 }
220
221 static NTSTATUS gensec_spnego_packet_full_request(struct gensec_security *gensec_security, 
222                                                 DATA_BLOB blob, size_t *size)
223 {
224         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
225
226         if (spnego_state->state_position != SPNEGO_DONE 
227             && spnego_state->state_position != SPNEGO_FALLBACK) {
228                 DEBUG(1, ("gensec_spnego_unwrap: wrong state for unwrap\n"));
229                 return NT_STATUS_INVALID_PARAMETER;
230         }
231         
232         return gensec_packet_full_request(spnego_state->sub_sec_security, 
233                                           blob, size);
234 }
235
236 static NTSTATUS gensec_spnego_unwrap_packets(struct gensec_security *gensec_security, 
237                                              TALLOC_CTX *mem_ctx, 
238                                              const DATA_BLOB *in, 
239                                              DATA_BLOB *out,
240                                              size_t *len_processed) 
241 {
242         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
243
244         if (spnego_state->state_position != SPNEGO_DONE 
245             && spnego_state->state_position != SPNEGO_FALLBACK) {
246                 DEBUG(1, ("gensec_spnego_unwrap: wrong state for unwrap\n"));
247                 return NT_STATUS_INVALID_PARAMETER;
248         }
249         
250         return gensec_unwrap_packets(spnego_state->sub_sec_security, 
251                                      mem_ctx, in, out,
252                                      len_processed);
253 }
254
255 static size_t gensec_spnego_sig_size(struct gensec_security *gensec_security, size_t data_size) 
256 {
257         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
258
259         if (spnego_state->state_position != SPNEGO_DONE 
260             && spnego_state->state_position != SPNEGO_FALLBACK) {
261                 return 0;
262         }
263         
264         return gensec_sig_size(spnego_state->sub_sec_security, data_size);
265 }
266
267 static size_t gensec_spnego_max_input_size(struct gensec_security *gensec_security) 
268 {
269         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
270
271         if (spnego_state->state_position != SPNEGO_DONE 
272             && spnego_state->state_position != SPNEGO_FALLBACK) {
273                 return 0;
274         }
275         
276         return gensec_max_input_size(spnego_state->sub_sec_security);
277 }
278
279 static size_t gensec_spnego_max_wrapped_size(struct gensec_security *gensec_security) 
280 {
281         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
282
283         if (spnego_state->state_position != SPNEGO_DONE 
284             && spnego_state->state_position != SPNEGO_FALLBACK) {
285                 return 0;
286         }
287         
288         return gensec_max_wrapped_size(spnego_state->sub_sec_security);
289 }
290
291 static NTSTATUS gensec_spnego_session_key(struct gensec_security *gensec_security, 
292                                           DATA_BLOB *session_key)
293 {
294         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
295         if (!spnego_state->sub_sec_security) {
296                 return NT_STATUS_INVALID_PARAMETER;
297         }
298         
299         return gensec_session_key(spnego_state->sub_sec_security, 
300                                   session_key);
301 }
302
303 static NTSTATUS gensec_spnego_session_info(struct gensec_security *gensec_security,
304                                                                       struct auth_session_info **session_info) 
305 {
306         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
307         if (!spnego_state->sub_sec_security) {
308                 return NT_STATUS_INVALID_PARAMETER;
309         }
310         
311         return gensec_session_info(spnego_state->sub_sec_security, 
312                                    session_info);
313 }
314
315 /** Fallback to another GENSEC mechanism, based on magic strings 
316  *
317  * This is the 'fallback' case, where we don't get SPNEGO, and have to
318  * try all the other options (and hope they all have a magic string
319  * they check)
320 */
321
322 static NTSTATUS gensec_spnego_server_try_fallback(struct gensec_security *gensec_security, 
323                                                   struct spnego_state *spnego_state,
324                                                   TALLOC_CTX *out_mem_ctx, 
325                                                   const DATA_BLOB in, DATA_BLOB *out) 
326 {
327         int i,j;
328         struct gensec_security_ops **all_ops
329                 = gensec_security_mechs(gensec_security, out_mem_ctx);
330         for (i=0; all_ops[i]; i++) {
331                 bool is_spnego;
332                 NTSTATUS nt_status;
333                 if (!all_ops[i]->oid) {
334                         continue;
335                 }
336
337                 is_spnego = false;
338                 for (j=0; all_ops[i]->oid[j]; j++) {
339                         if (strcasecmp(GENSEC_OID_SPNEGO,all_ops[i]->oid[j]) == 0) {
340                                 is_spnego = true;
341                         }
342                 }
343                 if (is_spnego) {
344                         continue;
345                 }
346
347                 if (!all_ops[i]->magic) {
348                         continue;
349                 }
350
351                 nt_status = all_ops[i]->magic(gensec_security, &in);
352                 if (!NT_STATUS_IS_OK(nt_status)) {
353                         continue;
354                 }
355
356                 spnego_state->state_position = SPNEGO_FALLBACK;
357
358                 nt_status = gensec_subcontext_start(spnego_state, 
359                                                     gensec_security, 
360                                                     &spnego_state->sub_sec_security);
361
362                 if (!NT_STATUS_IS_OK(nt_status)) {
363                         return nt_status;
364                 }
365                 /* select the sub context */
366                 nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
367                                                      all_ops[i]);
368                 if (!NT_STATUS_IS_OK(nt_status)) {
369                         return nt_status;
370                 }
371                 nt_status = gensec_update(spnego_state->sub_sec_security,
372                                           out_mem_ctx, in, out);
373                 return nt_status;
374         }
375         DEBUG(1, ("Failed to parse SPNEGO request\n"));
376         return NT_STATUS_INVALID_PARAMETER;
377         
378 }
379
380 /* 
381    Parse the netTokenInit, either from the client, to the server, or
382    from the server to the client.
383 */
384
385 static NTSTATUS gensec_spnego_parse_negTokenInit(struct gensec_security *gensec_security,
386                                                  struct spnego_state *spnego_state, 
387                                                  TALLOC_CTX *out_mem_ctx, 
388                                                  const char **mechType,
389                                                  const DATA_BLOB unwrapped_in, DATA_BLOB *unwrapped_out) 
390 {
391         int i;
392         NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
393         DATA_BLOB null_data_blob = data_blob(NULL,0);
394
395         const struct gensec_security_ops_wrapper *all_sec
396                 = gensec_security_by_oid_list(gensec_security, 
397                                               out_mem_ctx, 
398                                               mechType,
399                                               GENSEC_OID_SPNEGO);
400         if (spnego_state->state_position == SPNEGO_SERVER_START) {
401                 for (i=0; all_sec && all_sec[i].op; i++) {
402                         /* optomisitic token */
403                         if (strcmp(all_sec[i].oid, mechType[0]) == 0) {
404                                 nt_status = gensec_subcontext_start(spnego_state,
405                                                                     gensec_security,
406                                                                     &spnego_state->sub_sec_security);
407                                 if (!NT_STATUS_IS_OK(nt_status)) {
408                                         return nt_status;
409                                 }
410                                 /* select the sub context */
411                                 nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
412                                                                      all_sec[i].op);
413                                 if (!NT_STATUS_IS_OK(nt_status)) {
414                                         talloc_free(spnego_state->sub_sec_security);
415                                         spnego_state->sub_sec_security = NULL;
416                                         break;
417                                 }
418                                 
419                                 nt_status = gensec_update(spnego_state->sub_sec_security,
420                                                           out_mem_ctx, 
421                                                           unwrapped_in,
422                                                           unwrapped_out);
423                                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER) || 
424                                     NT_STATUS_EQUAL(nt_status, NT_STATUS_CANT_ACCESS_DOMAIN_INFO)) {
425                                         /* Pretend we never started it (lets the first run find some incompatible demand) */
426                                         
427                                         DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed to parse contents: %s\n", 
428                                                   spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
429                                         talloc_free(spnego_state->sub_sec_security);
430                                         spnego_state->sub_sec_security = NULL;
431                                         break;
432                                 }
433
434                                 spnego_state->neg_oid = all_sec[i].oid;
435                                 break;
436                         }
437                 }
438         }
439         
440         /* Having tried any optomisitc token from the client (if we
441          * were the server), if we didn't get anywhere, walk our list
442          * in our preference order */
443         
444         if (!spnego_state->sub_sec_security) {
445                 for (i=0; all_sec && all_sec[i].op; i++) {
446                         nt_status = gensec_subcontext_start(spnego_state,
447                                                             gensec_security,
448                                                             &spnego_state->sub_sec_security);
449                         if (!NT_STATUS_IS_OK(nt_status)) {
450                                 return nt_status;
451                         }
452                         /* select the sub context */
453                         nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
454                                                              all_sec[i].op);
455                         if (!NT_STATUS_IS_OK(nt_status)) {
456                                 talloc_free(spnego_state->sub_sec_security);
457                                 spnego_state->sub_sec_security = NULL;
458                                 continue;
459                         }
460                         
461                         spnego_state->neg_oid = all_sec[i].oid;
462
463                         /* only get the helping start blob for the first OID */
464                         nt_status = gensec_update(spnego_state->sub_sec_security,
465                                                   out_mem_ctx, 
466                                                   null_data_blob, 
467                                                   unwrapped_out);
468
469                         /* it is likely that a NULL input token will
470                          * not be liked by most server mechs, but if
471                          * we are in the client, we want the first
472                          * update packet to be able to abort the use
473                          * of this mech */
474                         if (spnego_state->state_position != SPNEGO_SERVER_START) {
475                                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER) || 
476                                     NT_STATUS_EQUAL(nt_status, NT_STATUS_CANT_ACCESS_DOMAIN_INFO)) {
477                                         /* Pretend we never started it (lets the first run find some incompatible demand) */
478                                         
479                                         DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed to parse: %s\n", 
480                                                   spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
481                                         talloc_free(spnego_state->sub_sec_security);
482                                         spnego_state->sub_sec_security = NULL;
483                                         continue;
484                                 }
485                         }
486
487                         break;
488                 }
489         }
490
491         if (spnego_state->sub_sec_security) {
492                 /* it is likely that a NULL input token will
493                  * not be liked by most server mechs, but this
494                  * does the right thing in the CIFS client.
495                  * just push us along the merry-go-round
496                  * again, and hope for better luck next
497                  * time */
498                 
499                 if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER)) {
500                         *unwrapped_out = data_blob(NULL, 0);
501                         nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
502                 }
503                 
504                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_PARAMETER) 
505                     && !NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) 
506                     && !NT_STATUS_IS_OK(nt_status)) {
507                         DEBUG(1, ("SPNEGO(%s) NEG_TOKEN_INIT failed: %s\n", 
508                                   spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
509                         talloc_free(spnego_state->sub_sec_security);
510                         spnego_state->sub_sec_security = NULL;
511                         
512                         /* We started the mech correctly, and the
513                          * input from the other side was valid.
514                          * Return the error (say bad password, invalid
515                          * ticket) */
516                         return nt_status;
517                 }
518         
519                 
520                 return nt_status; /* OK, INVALID_PARAMETER ore MORE PROCESSING */
521         }
522
523         DEBUG(1, ("SPNEGO: Could not find a suitable mechtype in NEG_TOKEN_INIT\n"));
524         /* we could re-negotiate here, but it would only work
525          * if the client or server lied about what it could
526          * support the first time.  Lets keep this code to
527          * reality */
528
529         return nt_status;
530 }
531
532 /** create a negTokenInit 
533  *
534  * This is the same packet, no matter if the client or server sends it first, but it is always the first packet
535 */
536 static NTSTATUS gensec_spnego_create_negTokenInit(struct gensec_security *gensec_security, 
537                                                   struct spnego_state *spnego_state,
538                                                   TALLOC_CTX *out_mem_ctx, 
539                                                   const DATA_BLOB in, DATA_BLOB *out) 
540 {
541         int i;
542         NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
543         DATA_BLOB null_data_blob = data_blob(NULL,0);
544         const char **mechTypes = NULL;
545         DATA_BLOB unwrapped_out = data_blob(NULL, 0);
546         const struct gensec_security_ops_wrapper *all_sec;
547         const char *principal = NULL;
548
549         mechTypes = gensec_security_oids(gensec_security, 
550                                          out_mem_ctx, GENSEC_OID_SPNEGO);
551
552         all_sec = gensec_security_by_oid_list(gensec_security, 
553                                               out_mem_ctx, 
554                                               mechTypes,
555                                               GENSEC_OID_SPNEGO);
556         for (i=0; all_sec && all_sec[i].op; i++) {
557                 struct spnego_data spnego_out;
558                 nt_status = gensec_subcontext_start(spnego_state,
559                                                     gensec_security,
560                                                     &spnego_state->sub_sec_security);
561                 if (!NT_STATUS_IS_OK(nt_status)) {
562                         return nt_status;
563                 }
564                 /* select the sub context */
565                 nt_status = gensec_start_mech_by_ops(spnego_state->sub_sec_security,
566                                                      all_sec[i].op);
567                 if (!NT_STATUS_IS_OK(nt_status)) {
568                         talloc_free(spnego_state->sub_sec_security);
569                         spnego_state->sub_sec_security = NULL;
570                         continue;
571                 }
572
573                 /* In the client, try and produce the first (optimistic) packet */
574                 if (spnego_state->state_position == SPNEGO_CLIENT_START) {
575                         nt_status = gensec_update(spnego_state->sub_sec_security,
576                                                   out_mem_ctx, 
577                                                   null_data_blob,
578                                                   &unwrapped_out);
579                         
580                         if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) 
581                             && !NT_STATUS_IS_OK(nt_status)) {
582                                 DEBUG(1, ("SPNEGO(%s) creating NEG_TOKEN_INIT failed: %s\n", 
583                                           spnego_state->sub_sec_security->ops->name, nt_errstr(nt_status)));
584                                 talloc_free(spnego_state->sub_sec_security);
585                                 spnego_state->sub_sec_security = NULL;
586                                 /* Pretend we never started it (lets the first run find some incompatible demand) */
587                                 
588                                 continue;
589                         }
590                 }
591
592                 spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
593                 
594                 /* List the remaining mechs as options */
595                 spnego_out.negTokenInit.mechTypes = gensec_security_oids_from_ops_wrapped(out_mem_ctx, 
596                                                                                           &all_sec[i]);
597                 spnego_out.negTokenInit.reqFlags = 0;
598                 
599                 if (spnego_state->state_position == SPNEGO_SERVER_START) {
600                         /* server credentials */
601                         struct cli_credentials *creds = gensec_get_credentials(gensec_security);
602                         if (creds) {
603                                 principal = cli_credentials_get_principal(creds, out_mem_ctx);
604                         }
605                 }
606                 if (principal) {
607                         spnego_out.negTokenInit.mechListMIC
608                                 = data_blob_string_const(principal);
609                 } else {
610                         spnego_out.negTokenInit.mechListMIC = null_data_blob;
611                 }
612
613                 spnego_out.negTokenInit.mechToken = unwrapped_out;
614                 
615                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
616                         DEBUG(1, ("Failed to write NEG_TOKEN_INIT\n"));
617                                 return NT_STATUS_INVALID_PARAMETER;
618                 }
619                 
620                 /* set next state */
621                 spnego_state->neg_oid = all_sec[i].oid;
622                 
623                 if (NT_STATUS_IS_OK(nt_status)) {
624                         spnego_state->no_response_expected = true;
625                 }
626
627                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
628         } 
629         talloc_free(spnego_state->sub_sec_security);
630         spnego_state->sub_sec_security = NULL;
631
632         DEBUG(1, ("Failed to setup SPNEGO negTokenInit request: %s\n", nt_errstr(nt_status)));
633         return NT_STATUS_INVALID_PARAMETER;
634 }
635
636
637 /** create a server negTokenTarg 
638  *
639  * This is the case, where the client is the first one who sends data
640 */
641
642 static NTSTATUS gensec_spnego_server_negTokenTarg(struct gensec_security *gensec_security, 
643                                                   struct spnego_state *spnego_state,
644                                                   TALLOC_CTX *out_mem_ctx, 
645                                                   NTSTATUS nt_status,
646                                                   const DATA_BLOB unwrapped_out, DATA_BLOB *out) 
647 {
648         struct spnego_data spnego_out;
649         DATA_BLOB null_data_blob = data_blob(NULL, 0);
650
651         /* compose reply */
652         spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
653         spnego_out.negTokenTarg.responseToken = unwrapped_out;
654         spnego_out.negTokenTarg.mechListMIC = null_data_blob;
655         spnego_out.negTokenTarg.supportedMech = NULL;
656
657         if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {   
658                 spnego_out.negTokenTarg.supportedMech = spnego_state->neg_oid;
659                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_INCOMPLETE;
660                 spnego_state->state_position = SPNEGO_SERVER_TARG;
661         } else if (NT_STATUS_IS_OK(nt_status)) {
662                 if (unwrapped_out.data) {
663                         spnego_out.negTokenTarg.supportedMech = spnego_state->neg_oid;
664                 }
665                 spnego_out.negTokenTarg.negResult = SPNEGO_ACCEPT_COMPLETED;
666                 spnego_state->state_position = SPNEGO_DONE;
667         } else {
668                 spnego_out.negTokenTarg.negResult = SPNEGO_REJECT;
669                 DEBUG(2, ("SPNEGO login failed: %s\n", nt_errstr(nt_status)));
670                 spnego_state->state_position = SPNEGO_DONE;
671         }
672
673         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
674                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
675                 return NT_STATUS_INVALID_PARAMETER;
676         }
677
678         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
679
680         return nt_status;
681 }
682
683
684 static NTSTATUS gensec_spnego_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx, 
685                                      const DATA_BLOB in, DATA_BLOB *out) 
686 {
687         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
688         DATA_BLOB null_data_blob = data_blob(NULL, 0);
689         DATA_BLOB unwrapped_out = data_blob(NULL, 0);
690         struct spnego_data spnego_out;
691         struct spnego_data spnego;
692
693         ssize_t len;
694
695         *out = data_blob(NULL, 0);
696
697         if (!out_mem_ctx) {
698                 out_mem_ctx = spnego_state;
699         }
700
701         /* and switch into the state machine */
702
703         switch (spnego_state->state_position) {
704         case SPNEGO_FALLBACK:
705                 return gensec_update(spnego_state->sub_sec_security,
706                                      out_mem_ctx, in, out);
707         case SPNEGO_SERVER_START:
708         {
709                 NTSTATUS nt_status;
710                 if (in.length) {
711
712                         len = spnego_read_data(gensec_security, in, &spnego);
713                         if (len == -1) {
714                                 return gensec_spnego_server_try_fallback(gensec_security, spnego_state, 
715                                                                          out_mem_ctx, in, out);
716                         }
717                         /* client sent NegTargetInit, we send NegTokenTarg */
718
719                         /* OK, so it's real SPNEGO, check the packet's the one we expect */
720                         if (spnego.type != spnego_state->expected_packet) {
721                                 DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type, 
722                                           spnego_state->expected_packet));
723                                 dump_data(1, in.data, in.length);
724                                 spnego_free_data(&spnego);
725                                 return NT_STATUS_INVALID_PARAMETER;
726                         }
727                         
728                         nt_status = gensec_spnego_parse_negTokenInit(gensec_security,
729                                                                      spnego_state,
730                                                                      out_mem_ctx, 
731                                                                      spnego.negTokenInit.mechTypes,
732                                                                      spnego.negTokenInit.mechToken, 
733                                                                      &unwrapped_out);
734                         
735                         nt_status = gensec_spnego_server_negTokenTarg(gensec_security,
736                                                                       spnego_state,
737                                                                       out_mem_ctx,
738                                                                       nt_status,
739                                                                       unwrapped_out, 
740                                                                       out);
741                         
742                         spnego_free_data(&spnego);
743                         
744                         return nt_status;
745                 } else {
746                         nt_status = gensec_spnego_create_negTokenInit(gensec_security, spnego_state, 
747                                                                       out_mem_ctx, in, out);
748                         spnego_state->state_position = SPNEGO_SERVER_START;
749                         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_INIT;
750                         return nt_status;
751                 }
752         }
753         
754         case SPNEGO_CLIENT_START:
755         {
756                 /* The server offers a list of mechanisms */
757                 
758                 const char *my_mechs[] = {NULL, NULL};
759                 NTSTATUS nt_status = NT_STATUS_INVALID_PARAMETER;
760
761                 if (!in.length) {
762                         /* client to produce negTokenInit */
763                         nt_status = gensec_spnego_create_negTokenInit(gensec_security, spnego_state, 
764                                                                  out_mem_ctx, in, out);
765                         spnego_state->state_position = SPNEGO_CLIENT_TARG;
766                         spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
767                         return nt_status;
768                 }
769                 
770                 len = spnego_read_data(gensec_security, in, &spnego);
771                 
772                 if (len == -1) {
773                         DEBUG(1, ("Invalid SPNEGO request:\n"));
774                         dump_data(1, in.data, in.length);
775                         return NT_STATUS_INVALID_PARAMETER;
776                 }
777                 
778                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
779                 if (spnego.type != spnego_state->expected_packet) {
780                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type, 
781                                   spnego_state->expected_packet));
782                         dump_data(1, in.data, in.length);
783                         spnego_free_data(&spnego);
784                         return NT_STATUS_INVALID_PARAMETER;
785                 }
786
787                 if (spnego.negTokenInit.targetPrincipal) {
788                         DEBUG(5, ("Server claims it's principal name is %s\n", spnego.negTokenInit.targetPrincipal));
789                         gensec_set_target_principal(gensec_security, spnego.negTokenInit.targetPrincipal);
790                 }
791
792                 nt_status = gensec_spnego_parse_negTokenInit(gensec_security,
793                                                              spnego_state,
794                                                              out_mem_ctx, 
795                                                              spnego.negTokenInit.mechTypes,
796                                                              spnego.negTokenInit.mechToken, 
797                                                              &unwrapped_out);
798
799                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(nt_status)) {
800                         spnego_free_data(&spnego);
801                         return nt_status;
802                 }
803
804                 my_mechs[0] = spnego_state->neg_oid;
805                 /* compose reply */
806                 spnego_out.type = SPNEGO_NEG_TOKEN_INIT;
807                 spnego_out.negTokenInit.mechTypes = my_mechs;
808                 spnego_out.negTokenInit.reqFlags = 0;
809                 spnego_out.negTokenInit.mechListMIC = null_data_blob;
810                 spnego_out.negTokenInit.mechToken = unwrapped_out;
811                 
812                 if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
813                         DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_INIT\n"));
814                                 return NT_STATUS_INVALID_PARAMETER;
815                 }
816                 
817                 /* set next state */
818                 spnego_state->expected_packet = SPNEGO_NEG_TOKEN_TARG;
819                 spnego_state->state_position = SPNEGO_CLIENT_TARG;
820
821                 if (NT_STATUS_IS_OK(nt_status)) {
822                         spnego_state->no_response_expected = true;
823                 }
824                 
825                 spnego_free_data(&spnego);
826                 return NT_STATUS_MORE_PROCESSING_REQUIRED;
827         }
828         case SPNEGO_SERVER_TARG:
829         {
830                 NTSTATUS nt_status;
831                 if (!in.length) {
832                         return NT_STATUS_INVALID_PARAMETER;
833                 }
834                 
835                 len = spnego_read_data(gensec_security, in, &spnego);
836                 
837                 if (len == -1) {
838                         DEBUG(1, ("Invalid SPNEGO request:\n"));
839                         dump_data(1, in.data, in.length);
840                         return NT_STATUS_INVALID_PARAMETER;
841                 }
842                 
843                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
844                 if (spnego.type != spnego_state->expected_packet) {
845                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type, 
846                                   spnego_state->expected_packet));
847                         dump_data(1, in.data, in.length);
848                         spnego_free_data(&spnego);
849                         return NT_STATUS_INVALID_PARAMETER;
850                 }
851
852                 if (!spnego_state->sub_sec_security) {
853                         DEBUG(1, ("SPNEGO: Did not setup a mech in NEG_TOKEN_INIT\n"));
854                         spnego_free_data(&spnego);
855                         return NT_STATUS_INVALID_PARAMETER;
856                 }
857
858                 nt_status = gensec_update(spnego_state->sub_sec_security,
859                                           out_mem_ctx, 
860                                           spnego.negTokenTarg.responseToken,
861                                           &unwrapped_out);
862
863                 nt_status = gensec_spnego_server_negTokenTarg(gensec_security,
864                                                               spnego_state,
865                                                               out_mem_ctx, 
866                                                               nt_status,
867                                                               unwrapped_out, 
868                                                               out);
869                 
870                 spnego_free_data(&spnego);
871                 
872                 return nt_status;
873         }
874         case SPNEGO_CLIENT_TARG:
875         {
876                 NTSTATUS nt_status;
877                 if (!in.length) {
878                         return NT_STATUS_INVALID_PARAMETER;
879                 }
880                 
881                 len = spnego_read_data(gensec_security, in, &spnego);
882                 
883                 if (len == -1) {
884                         DEBUG(1, ("Invalid SPNEGO request:\n"));
885                         dump_data(1, in.data, in.length);
886                         return NT_STATUS_INVALID_PARAMETER;
887                 }
888                 
889                 /* OK, so it's real SPNEGO, check the packet's the one we expect */
890                 if (spnego.type != spnego_state->expected_packet) {
891                         DEBUG(1, ("Invalid SPNEGO request: %d, expected %d\n", spnego.type, 
892                                   spnego_state->expected_packet));
893                         dump_data(1, in.data, in.length);
894                         spnego_free_data(&spnego);
895                         return NT_STATUS_INVALID_PARAMETER;
896                 }
897         
898                 if (spnego.negTokenTarg.negResult == SPNEGO_REJECT) {
899                         spnego_free_data(&spnego);
900                         return NT_STATUS_ACCESS_DENIED;
901                 }
902
903                 /* Server didn't like our choice of mech, and chose something else */
904                 if ((spnego.negTokenTarg.negResult == SPNEGO_ACCEPT_INCOMPLETE) &&
905                     spnego.negTokenTarg.supportedMech &&
906                     strcmp(spnego.negTokenTarg.supportedMech, spnego_state->neg_oid) != 0) {
907                         DEBUG(3,("GENSEC SPNEGO: client preferred mech (%s) not accepted, server wants: %s\n",
908                                  gensec_get_name_by_oid(spnego.negTokenTarg.supportedMech), 
909                                  gensec_get_name_by_oid(spnego_state->neg_oid)));
910                         
911                         talloc_free(spnego_state->sub_sec_security);
912                         nt_status = gensec_subcontext_start(spnego_state,
913                                                             gensec_security,
914                                                             &spnego_state->sub_sec_security);
915                         if (!NT_STATUS_IS_OK(nt_status)) {
916                                 spnego_free_data(&spnego);
917                                 return nt_status;
918                         }
919                         /* select the sub context */
920                         nt_status = gensec_start_mech_by_oid(spnego_state->sub_sec_security,
921                                                              spnego.negTokenTarg.supportedMech);
922                         if (!NT_STATUS_IS_OK(nt_status)) {
923                                 spnego_free_data(&spnego);
924                                 return nt_status;
925                         }
926
927                         nt_status = gensec_update(spnego_state->sub_sec_security,
928                                                   out_mem_ctx, 
929                                                   spnego.negTokenTarg.responseToken,
930                                                   &unwrapped_out);
931                         spnego_state->neg_oid = talloc_strdup(spnego_state, spnego.negTokenTarg.supportedMech);
932                 } else if (spnego_state->no_response_expected) {
933                         if (spnego.negTokenTarg.negResult != SPNEGO_ACCEPT_COMPLETED) {
934                                 DEBUG(3,("GENSEC SPNEGO: client GENSEC accepted, but server rejected (bad password?)\n"));
935                                 nt_status = NT_STATUS_INVALID_PARAMETER;
936                         } else if (spnego.negTokenTarg.responseToken.length) {
937                                 DEBUG(2,("GENSEC SPNEGO: client GENSEC accepted, but server continued negotiation!\n"));
938                                 nt_status = NT_STATUS_INVALID_PARAMETER;
939                         } else {
940                                 nt_status = NT_STATUS_OK;
941                         }
942                 } else {
943                         nt_status = gensec_update(spnego_state->sub_sec_security,
944                                                   out_mem_ctx, 
945                                                   spnego.negTokenTarg.responseToken, 
946                                                   &unwrapped_out);
947
948                         if (NT_STATUS_IS_OK(nt_status)) {
949                                 spnego_state->no_response_expected = true;
950                         }
951                 } 
952                 
953                 spnego_free_data(&spnego);
954
955                 if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)
956                         && !NT_STATUS_IS_OK(nt_status)) {
957                         DEBUG(1, ("SPNEGO(%s) login failed: %s\n", 
958                                   spnego_state->sub_sec_security->ops->name, 
959                                   nt_errstr(nt_status)));
960                         return nt_status;
961                 }
962
963                 if (unwrapped_out.length) {
964                         /* compose reply */
965                         spnego_out.type = SPNEGO_NEG_TOKEN_TARG;
966                         spnego_out.negTokenTarg.negResult = SPNEGO_NONE_RESULT;
967                         spnego_out.negTokenTarg.supportedMech = NULL;
968                         spnego_out.negTokenTarg.responseToken = unwrapped_out;
969                         spnego_out.negTokenTarg.mechListMIC = null_data_blob;
970                         
971                         if (spnego_write_data(out_mem_ctx, out, &spnego_out) == -1) {
972                                 DEBUG(1, ("Failed to write SPNEGO reply to NEG_TOKEN_TARG\n"));
973                                 return NT_STATUS_INVALID_PARAMETER;
974                         }
975                 
976                         spnego_state->state_position = SPNEGO_CLIENT_TARG;
977                         nt_status = NT_STATUS_MORE_PROCESSING_REQUIRED;
978                 } else {
979
980                         /* all done - server has accepted, and we agree */
981                         *out = null_data_blob;
982
983                         if (spnego.negTokenTarg.negResult != SPNEGO_ACCEPT_COMPLETED) {
984                                 /* unless of course it did not accept */
985                                 DEBUG(1,("gensec_update ok but not accepted\n"));
986                                 nt_status = NT_STATUS_INVALID_PARAMETER;
987                         }
988                 
989                         spnego_state->state_position = SPNEGO_DONE;
990                 }
991
992                 return nt_status;
993         }
994         case SPNEGO_DONE:
995                 /* We should not be called after we are 'done' */
996                 return NT_STATUS_INVALID_PARAMETER;
997         }
998         return NT_STATUS_INVALID_PARAMETER;
999 }
1000
1001 static bool gensec_spnego_have_feature(struct gensec_security *gensec_security,
1002                                        uint32_t feature) 
1003 {
1004         struct spnego_state *spnego_state = (struct spnego_state *)gensec_security->private_data;
1005         if (!spnego_state->sub_sec_security) {
1006                 return false;
1007         }
1008         
1009         return gensec_have_feature(spnego_state->sub_sec_security, 
1010                                    feature);
1011 }
1012
1013 static const char *gensec_spnego_oids[] = { 
1014         GENSEC_OID_SPNEGO,
1015         NULL 
1016 };
1017
1018 static const struct gensec_security_ops gensec_spnego_security_ops = {
1019         .name             = "spnego",
1020         .sasl_name        = "GSS-SPNEGO",
1021         .auth_type        = DCERPC_AUTH_TYPE_SPNEGO,
1022         .oid              = gensec_spnego_oids,
1023         .client_start     = gensec_spnego_client_start,
1024         .server_start     = gensec_spnego_server_start,
1025         .update           = gensec_spnego_update,
1026         .seal_packet      = gensec_spnego_seal_packet,
1027         .sign_packet      = gensec_spnego_sign_packet,
1028         .sig_size         = gensec_spnego_sig_size,
1029         .max_wrapped_size = gensec_spnego_max_wrapped_size,
1030         .max_input_size   = gensec_spnego_max_input_size,
1031         .check_packet     = gensec_spnego_check_packet,
1032         .unseal_packet    = gensec_spnego_unseal_packet,
1033         .packet_full_request = gensec_spnego_packet_full_request,
1034         .wrap             = gensec_spnego_wrap,
1035         .unwrap           = gensec_spnego_unwrap,
1036         .wrap_packets     = gensec_spnego_wrap_packets,
1037         .unwrap_packets   = gensec_spnego_unwrap_packets,
1038         .session_key      = gensec_spnego_session_key,
1039         .session_info     = gensec_spnego_session_info,
1040         .have_feature     = gensec_spnego_have_feature,
1041         .enabled          = true,
1042         .priority         = GENSEC_SPNEGO
1043 };
1044
1045 NTSTATUS gensec_spnego_init(void)
1046 {
1047         NTSTATUS ret;
1048         ret = gensec_register(&gensec_spnego_security_ops);
1049         if (!NT_STATUS_IS_OK(ret)) {
1050                 DEBUG(0,("Failed to register '%s' gensec backend!\n",
1051                         gensec_spnego_security_ops.name));
1052                 return ret;
1053         }
1054
1055         return ret;
1056 }