- Import new upstream release 4.1.8.

1 files changed, 183 insertions, 0 deletions

diff --git a/src/manager/database.c b/src/manager/database.c
new file mode 100644
index 000000000..a7776c81e
--- /dev/null
+++ b/src/manager/database.c
@@ -0,0 +1,183 @@
+/**
+ * @file database.c
+ *
+ * @brief Implementation of database_t.
+ *
+ */
+
+/*
+ * Copyright (C) 2007 Martin Willi
+ * Hochschule fuer Technik Rapperswil
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * for more details.
+ */
+
+#include "database.h"
+
+#include <sqlite3.h>
+#include <library.h>
+#include <crypto/hashers/hasher.h>
+
+
+typedef struct private_database_t private_database_t;
+
+/**
+ * private data of database
+ */
+struct private_database_t {
+
+	/**
+	 * public functions
+	 */
+	database_t public;
+	
+	/**
+	 * SQLite database handle
+	 */
+	sqlite3 *db;
+};
+
+/**
+ * database enumerator implements enumerator_t
+ */
+typedef struct  {
+	enumerator_t enumerator;
+	sqlite3_stmt *stmt;
+} db_enumerator_t;
+
+/**
+ * destroy a database enumerator
+ */
+static void db_enumerator_destroy(db_enumerator_t* this)
+{
+	sqlite3_finalize(this->stmt);
+	free(this);
+}
+
+/**
+ * create a database enumerator
+ */
+static enumerator_t *db_enumerator_create(bool(*enumerate)(db_enumerator_t*,void*,...),
+										 	 sqlite3_stmt *stmt)
+{
+	db_enumerator_t *this = malloc_thing(db_enumerator_t);
+	this->enumerator.enumerate = (void*)enumerate;
+	this->enumerator.destroy = (void*)db_enumerator_destroy;
+	this->stmt = stmt;
+	return &this->enumerator;
+}
+
+/**
+ * Implementation of database_t.login.
+ */
+static int login(private_database_t *this, char *username, char *password)
+{
+	sqlite3_stmt *stmt;
+	hasher_t *hasher;
+	chunk_t hash, data;
+	size_t username_len, password_len;
+	int uid = 0;
+	char *str;
+	
+	/* hash = SHA1( username | password ) */
+	hasher = hasher_create(HASH_SHA1);
+	hash = chunk_alloca(hasher->get_hash_size(hasher));
+	username_len = strlen(username);
+	password_len = strlen(password);
+	data = chunk_alloca(username_len + password_len);
+	memcpy(data.ptr, username, username_len);
+	memcpy(data.ptr + username_len, password, password_len);
+	hasher->get_hash(hasher, data, hash.ptr);
+	hasher->destroy(hasher);
+	str = chunk_to_hex(hash, FALSE);
+	
+	if (sqlite3_prepare_v2(this->db,
+			"SELECT oid FROM users WHERE username = ? AND password = ?;",
+			-1, &stmt, NULL) == SQLITE_OK)
+	{
+		if (sqlite3_bind_text(stmt, 1, username, -1, SQLITE_STATIC) == SQLITE_OK &&
+			sqlite3_bind_text(stmt, 2, str, -1, SQLITE_STATIC) == SQLITE_OK &&
+			sqlite3_step(stmt) == SQLITE_ROW)
+		{
+			uid = sqlite3_column_int(stmt, 0);
+		}
+		sqlite3_finalize(stmt);
+	}
+	free(str);
+	return uid;
+}
+
+/**
+ * enumerate function for gateway enumrator
+ */
+static bool gateway_enumerate(db_enumerator_t* e, int *id, const char **name,
+							  int *port, const char **address)
+{
+	if (sqlite3_step(e->stmt) == SQLITE_ROW)
+	{
+		*id = sqlite3_column_int(e->stmt, 0);
+		*name = sqlite3_column_text(e->stmt, 1);
+		*port = sqlite3_column_int(e->stmt, 2);
+		*address = sqlite3_column_text(e->stmt, 3);
+		return TRUE;
+	}
+	return FALSE;
+}
+
+/**
+ * Implementation of database_t.create_gateway_enumerator.
+ */
+static enumerator_t* create_gateway_enumerator(private_database_t *this, int user)
+{
+	sqlite3_stmt *stmt;
+	
+	if (sqlite3_prepare_v2(this->db,
+			"SELECT gateways.oid AS gid, name, port, address FROM "
+			"gateways, user_gateway AS ug ON gid = ug.gateway WHERE ug.user = ?;",
+			-1, &stmt, NULL) == SQLITE_OK)
+	{
+		if (sqlite3_bind_int(stmt, 1, user) == SQLITE_OK)
+		{
+			return db_enumerator_create((void*)gateway_enumerate, stmt);
+		}
+		sqlite3_finalize(stmt);
+	}
+	return enumerator_create_empty();
+}
+
+/**
+ * Implementation of database_t.destroy
+ */
+static void destroy(private_database_t *this)
+{
+	sqlite3_close(this->db);
+	free(this);
+}
+
+/*
+ * see header file
+ */
+database_t *database_create(char *dbfile)
+{
+	private_database_t *this = malloc_thing(private_database_t);
+	
+	this->public.login = (int(*)(database_t*, char *username, char *password))login;
+	this->public.create_gateway_enumerator = (enumerator_t*(*)(database_t*,int))create_gateway_enumerator;
+	this->public.destroy = (void(*)(database_t*))destroy;
+	
+	if (sqlite3_open(dbfile, &this->db) != SQLITE_OK)
+	{
+		destroy(this);
+		return NULL;
+	}
+	return &this->public;
+}
+