r4088: Get medieval on our ass about malloc.... :-). Take control of all our allocation
[tprouty/samba.git] / source / auth / auth_ntlmssp.c
1 /* 
2    Unix SMB/Netbios implementation.
3    Version 3.0
4    handle NLTMSSP, server side
5
6    Copyright (C) Andrew Tridgell      2001
7    Copyright (C) Andrew Bartlett 2001-2003
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 2 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, write to the Free Software
21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 */
23
24 #include "includes.h"
25
26 /**
27  * Return the challenge as determined by the authentication subsystem 
28  * @return an 8 byte random challenge
29  */
30
31 static const uint8 *auth_ntlmssp_get_challenge(const struct ntlmssp_state *ntlmssp_state)
32 {
33         AUTH_NTLMSSP_STATE *auth_ntlmssp_state = ntlmssp_state->auth_context;
34         return auth_ntlmssp_state->auth_context->get_ntlm_challenge(auth_ntlmssp_state->auth_context);
35 }
36
37 /**
38  * Some authentication methods 'fix' the challenge, so we may not be able to set it
39  *
40  * @return If the effective challenge used by the auth subsystem may be modified
41  */
42 static BOOL auth_ntlmssp_may_set_challenge(const struct ntlmssp_state *ntlmssp_state)
43 {
44         AUTH_NTLMSSP_STATE *auth_ntlmssp_state = ntlmssp_state->auth_context;
45         struct auth_context *auth_context = auth_ntlmssp_state->auth_context;
46
47         return auth_context->challenge_may_be_modified;
48 }
49
50 /**
51  * NTLM2 authentication modifies the effective challange, 
52  * @param challenge The new challenge value
53  */
54 static NTSTATUS auth_ntlmssp_set_challenge(struct ntlmssp_state *ntlmssp_state, DATA_BLOB *challenge)
55 {
56         AUTH_NTLMSSP_STATE *auth_ntlmssp_state = ntlmssp_state->auth_context;
57         struct auth_context *auth_context = auth_ntlmssp_state->auth_context;
58
59         SMB_ASSERT(challenge->length == 8);
60
61         auth_context->challenge = data_blob_talloc(auth_context->mem_ctx, 
62                                                    challenge->data, challenge->length);
63
64         auth_context->challenge_set_by = "NTLMSSP callback (NTLM2)";
65
66         DEBUG(5, ("auth_context challenge set by %s\n", auth_context->challenge_set_by));
67         DEBUG(5, ("challenge is: \n"));
68         dump_data(5, (const char *)auth_context->challenge.data, auth_context->challenge.length);
69         return NT_STATUS_OK;
70 }
71
72 /**
73  * Check the password on an NTLMSSP login.  
74  *
75  * Return the session keys used on the connection.
76  */
77
78 static NTSTATUS auth_ntlmssp_check_password(struct ntlmssp_state *ntlmssp_state, DATA_BLOB *user_session_key, DATA_BLOB *lm_session_key) 
79 {
80         AUTH_NTLMSSP_STATE *auth_ntlmssp_state = ntlmssp_state->auth_context;
81         auth_usersupplied_info *user_info = NULL;
82         NTSTATUS nt_status;
83
84         /* the client has given us its machine name (which we otherwise would not get on port 445).
85            we need to possibly reload smb.conf if smb.conf includes depend on the machine name */
86
87         set_remote_machine_name(auth_ntlmssp_state->ntlmssp_state->workstation, True);
88
89         /* setup the string used by %U */
90         /* sub_set_smb_name checks for weird internally */
91         sub_set_smb_name(auth_ntlmssp_state->ntlmssp_state->user);
92
93         reload_services(True);
94
95         nt_status = make_user_info_map(&user_info, 
96                                        auth_ntlmssp_state->ntlmssp_state->user, 
97                                        auth_ntlmssp_state->ntlmssp_state->domain, 
98                                        auth_ntlmssp_state->ntlmssp_state->workstation, 
99                                        auth_ntlmssp_state->ntlmssp_state->lm_resp.data ? &auth_ntlmssp_state->ntlmssp_state->lm_resp : NULL, 
100                                        auth_ntlmssp_state->ntlmssp_state->nt_resp.data ? &auth_ntlmssp_state->ntlmssp_state->nt_resp : NULL, 
101                                        NULL, NULL, NULL,
102                                        True);
103
104         if (!NT_STATUS_IS_OK(nt_status)) {
105                 return nt_status;
106         }
107
108         nt_status = auth_ntlmssp_state->auth_context->check_ntlm_password(auth_ntlmssp_state->auth_context, 
109                                                                           user_info, &auth_ntlmssp_state->server_info); 
110
111         free_user_info(&user_info);
112
113         if (!NT_STATUS_IS_OK(nt_status)) {
114                 return nt_status;
115         }
116         if (auth_ntlmssp_state->server_info->user_session_key.length) {
117                 DEBUG(10, ("Got NT session key of length %u\n", auth_ntlmssp_state->server_info->user_session_key.length));
118                 *user_session_key = data_blob_talloc(auth_ntlmssp_state->mem_ctx, 
119                                                    auth_ntlmssp_state->server_info->user_session_key.data,
120                                                    auth_ntlmssp_state->server_info->user_session_key.length);
121         }
122         if (auth_ntlmssp_state->server_info->lm_session_key.length) {
123                 DEBUG(10, ("Got LM session key of length %u\n", auth_ntlmssp_state->server_info->lm_session_key.length));
124                 *lm_session_key = data_blob_talloc(auth_ntlmssp_state->mem_ctx, 
125                                                    auth_ntlmssp_state->server_info->lm_session_key.data,
126                                                    auth_ntlmssp_state->server_info->lm_session_key.length);
127         }
128         return nt_status;
129 }
130
131 NTSTATUS auth_ntlmssp_start(AUTH_NTLMSSP_STATE **auth_ntlmssp_state)
132 {
133         NTSTATUS nt_status;
134         TALLOC_CTX *mem_ctx;
135
136         mem_ctx = talloc_init("AUTH NTLMSSP context");
137         
138         *auth_ntlmssp_state = TALLOC_ZERO_P(mem_ctx, AUTH_NTLMSSP_STATE);
139         if (!*auth_ntlmssp_state) {
140                 DEBUG(0,("auth_ntlmssp_start: talloc failed!\n"));
141                 talloc_destroy(mem_ctx);
142                 return NT_STATUS_NO_MEMORY;
143         }
144
145         ZERO_STRUCTP(*auth_ntlmssp_state);
146
147         (*auth_ntlmssp_state)->mem_ctx = mem_ctx;
148
149         if (!NT_STATUS_IS_OK(nt_status = ntlmssp_server_start(&(*auth_ntlmssp_state)->ntlmssp_state))) {
150                 return nt_status;
151         }
152
153         if (!NT_STATUS_IS_OK(nt_status = make_auth_context_subsystem(&(*auth_ntlmssp_state)->auth_context))) {
154                 return nt_status;
155         }
156
157         (*auth_ntlmssp_state)->ntlmssp_state->auth_context = (*auth_ntlmssp_state);
158         (*auth_ntlmssp_state)->ntlmssp_state->get_challenge = auth_ntlmssp_get_challenge;
159         (*auth_ntlmssp_state)->ntlmssp_state->may_set_challenge = auth_ntlmssp_may_set_challenge;
160         (*auth_ntlmssp_state)->ntlmssp_state->set_challenge = auth_ntlmssp_set_challenge;
161         (*auth_ntlmssp_state)->ntlmssp_state->check_password = auth_ntlmssp_check_password;
162         (*auth_ntlmssp_state)->ntlmssp_state->server_role = lp_server_role();
163
164         return NT_STATUS_OK;
165 }
166
167 void auth_ntlmssp_end(AUTH_NTLMSSP_STATE **auth_ntlmssp_state)
168 {
169         TALLOC_CTX *mem_ctx = (*auth_ntlmssp_state)->mem_ctx;
170
171         if ((*auth_ntlmssp_state)->ntlmssp_state) {
172                 ntlmssp_end(&(*auth_ntlmssp_state)->ntlmssp_state);
173         }
174         if ((*auth_ntlmssp_state)->auth_context) {
175                 ((*auth_ntlmssp_state)->auth_context->free)(&(*auth_ntlmssp_state)->auth_context);
176         }
177         if ((*auth_ntlmssp_state)->server_info) {
178                 free_server_info(&(*auth_ntlmssp_state)->server_info);
179         }
180         talloc_destroy(mem_ctx);
181         *auth_ntlmssp_state = NULL;
182 }
183
184 NTSTATUS auth_ntlmssp_update(AUTH_NTLMSSP_STATE *auth_ntlmssp_state, 
185                              const DATA_BLOB request, DATA_BLOB *reply) 
186 {
187         return ntlmssp_update(auth_ntlmssp_state->ntlmssp_state, request, reply);
188 }