X-Git-Url: http://git.bytex64.net/?a=blobdiff_plain;f=common%2Fauth.c;h=0dbe6a4a12094bd1cf2e5d1eee8da0d1fd5f9cb2;hb=30465019b067284f33ffe514e970edcf3ca6f1e5;hp=cdfbf85f396fed3cbb2d640db412baa8c5815b73;hpb=96ec261b36bdbb701e05f5ee1aab70dec44085f9;p=blerg.git

diff --git a/common/auth.c b/common/auth.c
index cdfbf85..0dbe6a4 100644
--- a/common/auth.c
+++ b/common/auth.c
@@ -1,3 +1,6 @@
+/* Blerg is (C) 2011 The Dominion of Awesome, and is distributed under a
+ * BSD-style license.  Please see the COPYING file for details.
+ */
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <fcntl.h>
@@ -9,17 +12,22 @@
 #include "util.h"
 
 #define TOKEN_SIZE 16
+#define MAX_PASSWORD_LENGTH 64
 
 int auth_set_password(const char *username, const char *password) {
 	char filename[512];
-	int fd;
+	int fd, n;
 
 	if (!valid_name(username) || !blerg_exists(username))
 		return 0;
 
+	n = strlen(password);
+	if (n > MAX_PASSWORD_LENGTH)
+		return 0;
+
 	snprintf(filename, 512, "%s/%s/password", DATA_PATH, username);
 	fd = open(filename, O_WRONLY | O_CREAT, 0600);
-	write(fd, password, strlen(password));
+	write(fd, password, n);
 	close(fd);
 	
 	return 1;
@@ -37,7 +45,7 @@ int auth_get_password(const char *username, char *password) {
 	fd = open(filename, O_RDONLY);
 	if (fd == -1)
 		return 0;
-	len = read(fd, password, 32);
+	len = read(fd, password, MAX_PASSWORD_LENGTH);
 	close(fd);
 
 	password[len] = 0;
@@ -51,7 +59,7 @@ int auth_check_password(const char *username, const char *password) {
 	if (auth_get_password(username, epw) == 0)
 		return 0;
 
-	if (strncmp(password, epw, 32) == 0)
+	if (strncmp(password, epw, MAX_PASSWORD_LENGTH) == 0)
 		return 1;
 	else
 		return 0;
@@ -87,70 +95,56 @@ char *create_random_token() {
 	return token;
 }
 
-int auth_login(const char *username, const char *password) {
+char * auth_login(const char *username, const char *password) {
 	char filename[512];
 	int token_fd;
 
 	if (!auth_check_password(username, password))
-		return 0;
+		return NULL;
 
-	sprintf(filename, "%s/%s/token", DATA_PATH, username);
+	char *token = create_random_token();
+
+	sprintf(filename, "%s/%s/tokens", DATA_PATH, username);
+	if (access(filename, F_OK) != 0) {
+		if (mkdir(filename, 0700) == -1) {
+			perror("Could not create auth token dir");
+			return NULL;
+		}
+	}
+
+	sprintf(filename, "%s/%s/tokens/%s", DATA_PATH, username, token);
 	token_fd = open(filename, O_WRONLY | O_CREAT, 0600);
 	if (token_fd == -1) {
 		perror("Could not open token");
-		return 0;
+		return NULL;
 	}
-
-	char *token = create_random_token();
-	write(token_fd, token, TOKEN_SIZE * 2);
 	close(token_fd);
-	free(token);
 
-	return 1;
+	return token;
 }
 
-int auth_logout(const char *username) {
+int auth_logout(const char *username, const char *token) {
 	char filename[512];
 
 	if (!valid_name(username))
 		return 0;
 
-	sprintf(filename, "%s/%s/token", DATA_PATH, username);
+	sprintf(filename, "%s/%s/tokens", DATA_PATH, username);
+	if (access(filename, F_OK) != 0) {
+		return 0;
+	}
+
+	sprintf(filename, "%s/%s/tokens/%s", DATA_PATH, username, token);
 	if (unlink(filename) == -1)
 		return 0;
 
 	return 1;
 }
 
-char *auth_get_token(const char *username) {
+int auth_check_token(const char *username, const char *given_token) {
 	char filename[512];
-	char *token;
-	int token_fd;
-
-	if (!valid_name(username))
-		return 0;
-
-	sprintf(filename, "%s/%s/token", DATA_PATH, username);
-	token_fd = open(filename, O_RDONLY, 0600);
-	if (token_fd == -1) {
-		perror("Could not open token");
-		return NULL;
-	}
 
-	token = malloc(TOKEN_SIZE * 2 + 1);
-	read(token_fd, token, TOKEN_SIZE * 2);
-	close(token_fd);
+	sprintf(filename, "%s/%s/tokens/%s", DATA_PATH, username, given_token);
 
-	return token;
-}
-
-int auth_check_token(const char *username, const char *given_token) {
-	char *token = auth_get_token(username);
-	if (token != NULL && given_token != NULL) {
-		int ret = (strncmp(token, given_token, TOKEN_SIZE * 2) == 0);
-		free(token);
-		return ret;
-	} else {
-		return 0;
-	}
+	return (access(filename, F_OK) == 0);
 }