r12822: Given that talloc gives us this extra level of safety, use it.
[kai/samba.git] / source4 / scripting / ejs / smbcalls_auth.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    ejs auth functions
5
6    Copyright (C) Simo Sorce 2005
7    Copyright (C) Andrew Tridgell 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 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 #include "lib/appweb/ejs/ejs.h"
26 #include "auth/auth.h"
27 #include "scripting/ejs/smbcalls.h"
28
29 static int ejs_doauth(MprVarHandle eid,
30                       TALLOC_CTX *tmp_ctx, struct MprVar *auth, const char *username, 
31                       const char *password, const char *domain, const char *workstation,
32                       struct socket_address *remote_host, const char *authtype)
33 {
34         struct auth_usersupplied_info *user_info = NULL;
35         struct auth_serversupplied_info *server_info = NULL;
36         struct auth_session_info *session_info = NULL;
37         struct auth_context *auth_context;
38         const char *auth_types[] = { authtype, NULL };
39         NTSTATUS nt_status;
40
41         /*
42           darn, we need some way to get the right event_context here
43         */
44         nt_status = auth_context_create(tmp_ctx, auth_types, &auth_context, NULL);
45         if (!NT_STATUS_IS_OK(nt_status)) {
46                 mprSetPropertyValue(auth, "result", mprCreateBoolVar(False));
47                 mprSetPropertyValue(auth, "report", mprString("Auth System Failure"));
48                 goto done;
49         }
50
51         user_info = talloc(tmp_ctx, struct auth_usersupplied_info);
52         if (!user_info) {
53                 mprSetPropertyValue(auth, "result", mprCreateBoolVar(False));
54                 mprSetPropertyValue(auth, "report", mprString("talloc failed"));
55                 goto done;
56         }
57
58         user_info->mapped_state = True;
59         user_info->client.account_name = username;
60         user_info->mapped.account_name = username;
61         user_info->client.domain_name = domain;
62         user_info->mapped.domain_name = domain;
63
64         user_info->workstation_name = workstation;
65
66         user_info->remote_host = remote_host;
67
68         user_info->password_state = AUTH_PASSWORD_PLAIN;
69         user_info->password.plaintext = talloc_strdup(user_info, password);
70
71         user_info->flags = USER_INFO_CASE_INSENSITIVE_USERNAME |
72                 USER_INFO_DONT_CHECK_UNIX_ACCOUNT;
73
74         user_info->logon_parameters = 0;
75
76         nt_status = auth_check_password(auth_context, tmp_ctx, user_info, &server_info);
77         if (!NT_STATUS_IS_OK(nt_status)) {
78                 mprSetPropertyValue(auth, "report", 
79                                     mprString(talloc_asprintf(mprMemCtx(), "Login Failed: %s", 
80                                                               get_friendly_nt_error_msg(nt_status))));
81                 mprSetPropertyValue(auth, "result", mprCreateBoolVar(False));
82                 goto done;
83         }
84
85         nt_status = auth_generate_session_info(tmp_ctx, server_info, &session_info);
86         if (!NT_STATUS_IS_OK(nt_status)) {
87                 mprSetPropertyValue(auth, "report", mprString("Session Info generation failed"));
88                 mprSetPropertyValue(auth, "result", mprCreateBoolVar(False));
89                 goto done;
90         }
91
92         talloc_steal(mprMemCtx(), session_info);
93         mprSetThisPtr(eid, "session_info", session_info);
94
95         mprSetPropertyValue(auth, "result", mprCreateBoolVar(server_info->authenticated));
96         mprSetPropertyValue(auth, "username", mprString(server_info->account_name));
97         mprSetPropertyValue(auth, "domain", mprString(server_info->domain_name));
98
99 done:
100         return 0;
101 }
102
103 /*
104   perform user authentication, returning an array of results
105
106 */
107 static int ejs_userAuth(MprVarHandle eid, int argc, struct MprVar **argv)
108 {
109         TALLOC_CTX *tmp_ctx;
110         const char *username;
111         const char *password;
112         const char *domain;
113         const char *workstation;
114         struct MprVar auth;
115         struct cli_credentials *creds;
116         struct socket_address *remote_host;
117
118         if (argc != 2 || argv[0]->type != MPR_TYPE_OBJECT || argv[1]->type != MPR_TYPE_OBJECT) {
119                 ejsSetErrorMsg(eid, "userAuth invalid arguments, this function requires an object.");
120                 return -1;
121         }
122
123         /* get credential values from credentials object */
124         creds = mprGetPtr(argv[0], "creds");
125         if (creds == NULL) {
126                 ejsSetErrorMsg(eid, "userAuth requires a 'creds' first parameter");
127                 return -1;
128         }
129
130         remote_host = mprGetPtr(argv[1], "socket_address");
131         if (remote_host == NULL) {
132                 ejsSetErrorMsg(eid, "userAuth requires a socket address second parameter");
133                 return -1;
134         }
135
136         tmp_ctx = talloc_new(mprMemCtx());      
137         
138         username    = cli_credentials_get_username(creds);
139         password    = cli_credentials_get_password(creds);
140         domain      = cli_credentials_get_domain(creds);
141         workstation = cli_credentials_get_workstation(creds);
142
143         if (username == NULL || password == NULL || domain == NULL) {
144                 mpr_Return(eid, mprCreateUndefinedVar());
145                 talloc_free(tmp_ctx);
146                 return 0;
147         }
148
149         auth = mprObject("auth");
150
151         if (domain && (strcmp("SYSTEM USER", domain) == 0)) {
152                 ejs_doauth(eid, tmp_ctx, &auth, username, password, domain, workstation, remote_host, "unix");
153         } else {
154                 ejs_doauth(eid, tmp_ctx, &auth, username, password, domain, workstation, remote_host, "sam");
155         }
156
157         mpr_Return(eid, auth);
158         talloc_free(tmp_ctx);
159         return 0;
160 }
161
162 /*
163   initialise credentials ejs object
164 */
165 static int ejs_system_session(MprVarHandle eid, int argc, struct MprVar **argv)
166 {
167         struct MprVar *obj = mprInitObject(eid, "session_info", argc, argv);
168         struct auth_session_info *session_info = system_session(mprMemCtx());
169
170         if (session_info == NULL) {
171                 return -1;
172         }
173
174         mprSetPtrChild(obj, "session_info", session_info);
175         return 0;
176 }
177
178 /*
179   setup C functions that be called from ejs
180 */
181 void smb_setup_ejs_auth(void)
182 {
183         ejsDefineCFunction(-1, "userAuth", ejs_userAuth, NULL, MPR_VAR_SCRIPT_HANDLE);
184         ejsDefineCFunction(-1, "system_session", ejs_system_session, NULL, MPR_VAR_SCRIPT_HANDLE);
185 }