summaryrefslogtreecommitdiffstats
path: root/src/lib/dhcpsrv/cfg_db_access.cc
blob: e93631dc7b6cd1c9803903703f3e53c6d3409eb5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
// Copyright (C) 2016-2017 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include <config.h>
#include <dhcpsrv/cfg_db_access.h>
#include <dhcpsrv/host_data_source_factory.h>
#include <dhcpsrv/host_mgr.h>
#include <dhcpsrv/lease_mgr_factory.h>
#include <boost/algorithm/string.hpp>
#include <boost/foreach.hpp>
#include <boost/lexical_cast.hpp>
#include <sstream>
#include <vector>

using namespace isc::data;

namespace isc {
namespace dhcp {

CfgDbAccess::CfgDbAccess()
    : appended_parameters_(), lease_db_access_("type=memfile"),
      host_db_access_() {
}

std::string
CfgDbAccess::getLeaseDbAccessString() const {
    return (getAccessString(lease_db_access_));
}


std::string
CfgDbAccess::getHostDbAccessString() const {
    return (getAccessString(host_db_access_));
}


void
CfgDbAccess::createManagers() const {
    // Recreate lease manager.
    LeaseMgrFactory::destroy();
    LeaseMgrFactory::create(getLeaseDbAccessString());

    // Recreate host data source.
    HostDataSourceFactory::destroy();
    if (!host_db_access_.empty()) {
        HostMgr::create(getHostDbAccessString());
    }
}

std::string 
CfgDbAccess::getAccessString(const std::string& access_string) const {
    std::ostringstream s;
    s << access_string;
    // Only append additional parameters if any parameters are specified
    // in a configuration. For host database, no parameters mean that
    // database access is disabled and thus we don't want to append any
    // parameters.
    if ((s.tellp() != std::streampos(0)) && (!appended_parameters_.empty())) {
        s << " " << appended_parameters_;
    }

    return (s.str());
}

ElementPtr
CfgDbAccess::toElementDbAccessString(const std::string& dbaccess) {
    ElementPtr result = Element::createMap();
    // Code from DatabaseConnection::parse
    if (dbaccess.empty()) {
        return (result);
    }
    std::vector<std::string> tokens;
    boost::split(tokens, dbaccess, boost::is_any_of(std::string("\t ")));
    BOOST_FOREACH(std::string token, tokens) {
        size_t pos = token.find("=");
        if (pos != std::string::npos) {
            std::string keyword = token.substr(0, pos);
            std::string value = token.substr(pos + 1);
            if ((keyword == "lfc-interval") ||
                (keyword == "connect-timeout") ||
                (keyword == "port")) {
                // integer parameters
                int64_t int_value;
                try {
                    int_value = boost::lexical_cast<int64_t>(value);
                    result->set(keyword, Element::create(int_value));
                } catch (...) {
                    isc_throw(ToElementError, "invalid DB access "
                              << "integer parameter: "
                              << keyword << "=" << value);
                }
            } else if ((keyword == "persist") ||
                       (keyword == "readonly")) {
                if (value == "true") {
                    result->set(keyword, Element::create(true));
                } else if (value == "false") {
                    result->set(keyword, Element::create(false));
                } else {
                    isc_throw(ToElementError, "invalid DB access "
                              << "boolean parameter: "
                              << keyword << "=" << value);
                }
            } else if ((keyword == "type") ||
                       (keyword == "user") ||
                       (keyword == "password") ||
                       (keyword == "host") ||
                       (keyword == "name") ||
                       (keyword == "contact_points") ||
                       (keyword == "keyspace")) {
                result->set(keyword, Element::create(value));
            } else {
                isc_throw(ToElementError, "unknown DB access parameter: "
                          << keyword << "=" << value);
            }
        } else {
            isc_throw(ToElementError, "Cannot unparse " << token
                      << ", expected format is name=value");
        }
    }
    return (result);
}

} // end of isc::dhcp namespace
} // end of isc namespace