3 * @brief Implementation of Login related functionality
13 #define LDAP_DEPRECATED 1 // Required to use ldap_simple_bind_s
17 #undef _GNU_SOURCE // HACK to silence compiler warning on redefinition in my_global.h
18 #include <my_global.h>
26 * Attempt to login by searching a MySQL database
27 * @param user - Username
28 * @param pass - Password
29 * @param db_host - Host running the DataBase
30 * @param db_user - User to search the database as
31 * @param db_pass - Password for the database user
32 * @param db_name - Name of the database to use
33 * @param db_table - Table to search in
34 * @returns Privelage level of the user or USER_UNAUTH for failure to authenticate
36 UserType Login_MySQL(const char * user, const char * pass,
37 const char * db_host, const char * db_user, const char * db_pass, const char * db_name, const char * db_table)
39 MYSQL * con = mysql_init(NULL);
42 Log(LOGERR, "mysql_init failed - %s", mysql_error(con));
46 if (mysql_real_connect(con, db_host, db_user, db_pass, NULL, 0, NULL, 0) == NULL)
48 Log(LOGERR, "mysql_real_connect failed - %s", mysql_error(con));
55 // Select the database
56 sprintf(buffer, "USE %s;", db_name);
57 if (mysql_query(con, buffer))
59 Log(LOGERR, "mysql_query failed - %s", mysql_error(con));
64 // Search for the user
65 sprintf(buffer, "SELECT password FROM %s WHERE user_name = \"%s\";", db_table, user);
66 if (mysql_query(con, buffer))
68 Log(LOGERR, "mysql_query failed - %s", mysql_error(con));
74 MYSQL_RES * result = mysql_store_result(con);
77 Log(LOGERR, "mysql_store_result failed - %s", mysql_error(con));
82 int num_fields = mysql_num_fields(result);
85 Log(LOGERR, "The database may be corrupt; %d fields found, expected %d", num_fields, 1);
90 UserType user_type = USER_UNAUTH;
94 if ((row = mysql_fetch_row(result)))
96 if (strcmp(crypt(pass, row[0]), row[0]) == 0)
98 user_type = USER_NORMAL;
101 // There should only be one row. Through a hissy fit if we see any more.
102 if ((row = mysql_fetch_row(result)))
104 Log(LOGERR, "Too many rows found.");
105 user_type = USER_UNAUTH;
110 Log(LOGERR, "No user matching %s", user);
114 mysql_free_result(result);
120 * Attempt to login using a file formatted like /etc/shadow
121 * This is here... because all better options have been exhausted
122 * @param user - The username
123 * @param pass - The password
124 * @param shadow - The file to use
125 * @returns Privelage level of the user or USER_UNAUTH for failure to authenticate
127 UserType Login_Shadow(const char * user, const char * pass, const char * shadow)
129 if (strlen(user) + strlen(pass) >= BUFSIZ-1)
131 Log(LOGERR, "User/Password too long!\n");
135 FILE * f = fopen(shadow, "r");
138 Log(LOGERR,"Can't open %s - %s\n", shadow, strerror(errno));
143 int passwd_index = -1;
145 while (fgets(buffer, BUFSIZ, f) != NULL) // NOTE: Restrict username+password strings to BUFSIZ... what could possibly go wrong?
148 //Log(LOGDEBUG,"Scanning %d: %s", strlen(buffer), buffer);
150 for (int i = 0; i < BUFSIZ-1 && buffer[i] != '\0'; ++i)
152 if (buffer[i] == ':')
160 if (strcmp(user,buffer) == 0)
162 //Log(LOGDEBUG,"User matches! %s\n", buffer);
168 if (passwd_index <= 0)
170 //Log(LOGDEBUG,"No user found matching %s\n", user);
175 for (int i = passwd_index; i < BUFSIZ-1 && buffer[i] != '\0'; ++i)
177 if (buffer[i] == ':')
182 if (buffer[i] == '\n')
185 char * end = buffer+gid_index;
186 UserType user_type = USER_NORMAL;
187 if (gid_index > passwd_index && gid_index < BUFSIZ-1)
189 int gid = strtol(buffer+gid_index, &end,10);
190 Log(LOGDEBUG, "Usertype %d %s", gid, buffer+gid_index);
191 if (*end == '\0' && gid == 0)
193 Log(LOGDEBUG, "Admin");
194 user_type = USER_ADMIN;
198 // Determine the salt
200 int s = 0; int count = 0;
201 for (int i = passwd_index; i < BUFSIZ-1; ++i)
203 salt[s++] = buffer[i];
204 if (salt[s] == '$' && ++count >= 3)
208 // Log(LOGDEBUG,"Salted Entry: %s\n", buffer+passwd_index);
209 // Log(LOGDEBUG,"Salted Attempt: %s\n", crypt(pass, salt));
211 if (strcmp(crypt(pass, salt), buffer+passwd_index) == 0)
219 * Attempt to bind to a LDAP uri
220 * @param uri - The uri
222 * @param pass - The password
223 * @returns An error code according to libldap; LDAP_SUCCESS if everything worked
225 int Login_LDAP_Bind(const char * uri, const char * dn, const char * pass)
227 Log(LOGDEBUG, "Bind to %s with dn %s and pass %s", uri, dn, pass);
229 // Initialise LDAP; prepares to connect to the server
231 int err = ldap_initialize(&ld, uri);
232 if (err != LDAP_SUCCESS || ld == NULL)
234 Log(LOGERR,"ldap_initialize failed - %s (ld = %p)", ldap_err2string(err), ld);
238 // Set the LDAP version...
239 int version = LDAP_VERSION3;
240 err = ldap_set_option(ld, LDAP_OPT_PROTOCOL_VERSION, &version); // specify the version
241 if (err != LDAP_SUCCESS)
243 Log(LOGERR,"ldap_set_option failed - %s", ldap_err2string(err));
247 // Attempt to bind using the supplied credentials.
248 // NOTE: ldap_simple_bind_s is "deprecated" in <ldap.h>, but not listed as such in the man pages :S
249 err = ldap_simple_bind_s(ld, dn, pass);
250 if (err != LDAP_SUCCESS)
252 Log(LOGERR, "ldap_simple_bind_s failed - %s", ldap_err2string(err));
256 Log(LOGDEBUG, "Successfully bound to %s with dn %s", uri, dn);
259 int err2 = ldap_unbind_s(ld);
260 if (err2 != LDAP_SUCCESS)
262 Log(LOGERR, "ldap_unbind_s failed - %s", ldap_err2string(err2));
270 * @param context - The context. The key will be cleared.
271 * @param params - Parameter string, UNUSED
273 void Logout_Handler(FCGIContext * context, char * params)
275 FCGI_ReleaseControl(context);
276 FCGI_AcceptJSON(context, "Logged out", "0");
281 * Handle a Login Request
282 * @param context - The context
283 * @param params - Parameter string, should contain username and password
285 void Login_Handler(FCGIContext * context, char * params)
287 char * user; // The username supplied through CGI
288 char * pass; // The password supplied through CGI
290 FCGIValue values[] = {
291 {"user", &user, FCGI_REQUIRED(FCGI_STRING_T)},
292 {"pass", &pass, FCGI_REQUIRED(FCGI_STRING_T)},
295 //enum to avoid the use of magic numbers
302 // Fill values appropriately
303 if (!FCGI_ParseRequest(context, params, values, sizeof(values)/sizeof(FCGIValue)))
305 // Error occured; FCGI_RejectJSON already called
309 // Trim leading whitespace
311 for (i = 0; isspace(user[0]) && user[0] != '\0'; ++i, ++user);
313 // Truncate string at first non alphanumeric character
314 for (i = 0; isalnum(user[i]) && user[i] != '\0'; ++i);
318 UserType user_type = USER_UNAUTH;
320 switch (g_options.auth_method)
325 if (strlen(pass) <= 0)
327 FCGI_RejectJSON(context, "No password supplied.");
331 //TODO: Generate the DN in some sane way
336 // On a simple LDAP server:
337 //int len = sprintf(dn, "uid=%s,%s", user, g_options.auth_options);
340 char * user_group = "Students";
342 user_group = "Staff";
343 int len = sprintf(dn, "cn=%s,ou=%s,%s", user, user_group, g_options.auth_options);
348 FCGI_RejectJSON(context, "DN too long! Recompile with increased BUFSIZ");
352 if (Login_LDAP_Bind(g_options.auth_uri, dn, pass) == LDAP_SUCCESS)
355 user_type = USER_ADMIN;
357 user_type = USER_NORMAL;
363 user_type = Login_Shadow(user, pass, g_options.auth_uri);
368 //WARNING: C string manipulation code approaching!
369 // Non reentrent; uses strsep and modifies g_options.auth_options
370 // If problems happen, try strdup ...
371 static char * db_opts[] = {"root", "", "users", "uc_users"};
372 static bool db_init_opts = false;
376 db_opts[0] = (char*)g_options.auth_options;
377 for (int i = 1; i < sizeof(db_opts)/sizeof(char*); ++i)
379 char * def = db_opts[i];
380 db_opts[i] = db_opts[i-1];
382 strsep(db_opts+i, ",");
383 if (db_opts[i] == NULL)
389 Log(LOGDEBUG, "MySQL: user %s pass %s name %s table %s", db_opts[0], db_opts[1], db_opts[2], db_opts[3]);
392 user_type = Login_MySQL(user, pass, g_options.auth_uri, db_opts[0],db_opts[1], db_opts[2], db_opts[3]);
397 Log(LOGWARN, "No authentication!");
398 user_type = USER_ADMIN;
405 if (user_type == USER_UNAUTH)
407 Log(LOGDEBUG, "Authentication failure for %s", user);
408 FCGI_RejectJSONEx(context, STATUS_UNAUTHORIZED, "Authentication failure.");
412 // Try and gain control over the system
413 if (FCGI_LockControl(context, user, user_type))
415 FCGI_EscapeText(context->user_name); //Don't break javascript pls
416 // Give the user a cookie
417 FCGI_AcceptJSON(context, "Logged in", context->control_key);
418 Log(LOGDEBUG, "Successful authentication for %s", user);
422 Log(LOGDEBUG, "%s successfully authenticated but system was in use by %s", user, context->user_name);
423 FCGI_RejectJSON(context, "Someone else is already logged in (and you are not an admin)");