Migrated control_controller_t to INIT/METHOD macros.
[strongswan.git] / src / manager / storage.c
1 /*
2 * Copyright (C) 2007 Martin Willi
3 * Hochschule fuer Technik Rapperswil
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License as published by the
7 * Free Software Foundation; either version 2 of the License, or (at your
8 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
12 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
13 * for more details.
14 */
15
16 #include "storage.h"
17
18 #include <library.h>
19 #include <crypto/hashers/hasher.h>
20
21
22 typedef struct private_storage_t private_storage_t;
23
24 /**
25 * private data of storage
26 */
27 struct private_storage_t {
28
29 /**
30 * public functions
31 */
32 storage_t public;
33
34 /**
35 * database connection
36 */
37 database_t *db;
38 };
39
40 METHOD(storage_t, login, int,
41 private_storage_t *this, char *username, char *password)
42 {
43 hasher_t *hasher;
44 chunk_t hash, data, hex_str;
45 size_t username_len, password_len;
46 int uid = 0;
47 enumerator_t *enumerator;
48
49 /* hash = SHA1( username | password ) */
50 hasher = lib->crypto->create_hasher(lib->crypto, HASH_SHA1);
51 if (hasher == NULL)
52 {
53 return 0;
54 }
55 hash = chunk_alloca(hasher->get_hash_size(hasher));
56 username_len = strlen(username);
57 password_len = strlen(password);
58 data = chunk_alloca(username_len + password_len);
59 memcpy(data.ptr, username, username_len);
60 memcpy(data.ptr + username_len, password, password_len);
61 hasher->get_hash(hasher, data, hash.ptr);
62 hasher->destroy(hasher);
63 hex_str = chunk_to_hex(hash, NULL, FALSE);
64
65 enumerator = this->db->query(this->db,
66 "SELECT oid FROM users WHERE username = ? AND password = ?;",
67 DB_TEXT, username, DB_TEXT, hex_str.ptr,
68 DB_INT);
69 if (enumerator)
70 {
71 enumerator->enumerate(enumerator, &uid);
72 enumerator->destroy(enumerator);
73 }
74 free(hex_str.ptr);
75 return uid;
76 }
77
78 METHOD(storage_t, create_gateway_enumerator, enumerator_t*,
79 private_storage_t *this, int user)
80 {
81 enumerator_t *enumerator;
82
83 enumerator = this->db->query(this->db,
84 "SELECT gateways.oid AS gid, name, port, address FROM "
85 "gateways, user_gateway AS ug ON gid = ug.gateway WHERE ug.user = ?;",
86 DB_INT, user,
87 DB_INT, DB_TEXT, DB_INT, DB_TEXT);
88 if (!enumerator)
89 {
90 enumerator = enumerator_create_empty();
91 }
92 return enumerator;
93 }
94
95 METHOD(storage_t, destroy, void,
96 private_storage_t *this)
97 {
98 this->db->destroy(this->db);
99 free(this);
100 }
101
102 /*
103 * see header file
104 */
105 storage_t *storage_create(char *uri)
106 {
107 private_storage_t *this;
108
109 INIT(this,
110 .public = {
111 .login = _login,
112 .create_gateway_enumerator = _create_gateway_enumerator,
113 .destroy = _destroy,
114 },
115 .db = lib->db->create(lib->db, uri),
116 );
117 if (this->db == NULL)
118 {
119 free(this);
120 return NULL;
121 }
122 return &this->public;
123 }
124