Return to storage.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / strongswan / src / manager

Annotation of embedaddon/strongswan/src/manager/storage.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Copyright (C) 2007 Martin Willi
        !             3:  * HSR 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:        if (!hasher->get_hash(hasher, data, hash.ptr))
        !            62:        {
        !            63:                hasher->destroy(hasher);
        !            64:                return 0;
        !            65:        }
        !            66:        hasher->destroy(hasher);
        !            67:        hex_str = chunk_to_hex(hash, NULL, FALSE);
        !            68: 
        !            69:        enumerator = this->db->query(this->db,
        !            70:                        "SELECT oid FROM users WHERE username = ? AND password = ?;",
        !            71:                        DB_TEXT, username, DB_TEXT, hex_str.ptr,
        !            72:                        DB_INT);
        !            73:        if (enumerator)
        !            74:        {
        !            75:                enumerator->enumerate(enumerator, &uid);
        !            76:                enumerator->destroy(enumerator);
        !            77:        }
        !            78:        free(hex_str.ptr);
        !            79:        return uid;
        !            80: }
        !            81: 
        !            82: METHOD(storage_t, create_gateway_enumerator, enumerator_t*,
        !            83:        private_storage_t *this, int user)
        !            84: {
        !            85:        enumerator_t *enumerator;
        !            86: 
        !            87:        enumerator = this->db->query(this->db,
        !            88:                        "SELECT gateways.oid AS gid, name, port, address FROM "
        !            89:                        "gateways, user_gateway AS ug ON gid = ug.gateway WHERE ug.user = ?;",
        !            90:                        DB_INT, user,
        !            91:                        DB_INT, DB_TEXT, DB_INT, DB_TEXT);
        !            92:        if (!enumerator)
        !            93:        {
        !            94:                enumerator = enumerator_create_empty();
        !            95:        }
        !            96:        return enumerator;
        !            97: }
        !            98: 
        !            99: METHOD(storage_t, destroy, void,
        !           100:        private_storage_t *this)
        !           101: {
        !           102:        this->db->destroy(this->db);
        !           103:        free(this);
        !           104: }
        !           105: 
        !           106: /*
        !           107:  * see header file
        !           108:  */
        !           109: storage_t *storage_create(char *uri)
        !           110: {
        !           111:        private_storage_t *this;
        !           112: 
        !           113:        INIT(this,
        !           114:                .public = {
        !           115:                        .login = _login,
        !           116:                        .create_gateway_enumerator = _create_gateway_enumerator,
        !           117:                        .destroy = _destroy,
        !           118:                },
        !           119:                .db = lib->db->create(lib->db, uri),
        !           120:        );
        !           121:        if (this->db == NULL)
        !           122:        {
        !           123:                free(this);
        !           124:                return NULL;
        !           125:        }
        !           126:        return &this->public;
        !           127: }
        !           128: