]> git.stg.codes - stg.git/blob - rlm_stg/stg_client.cpp
217272255e5c448afc92c0b4583fdd2e259878a3
[stg.git] / rlm_stg / stg_client.cpp
1 /*
2  *    This program is free software; you can redistribute it and/or modify
3  *    it under the terms of the GNU General Public License as published by
4  *    the Free Software Foundation; either version 2 of the License, or
5  *    (at your option) any later version.
6  *
7  *    This program is distributed in the hope that it will be useful,
8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  *    GNU General Public License for more details.
11  *
12  *    You should have received a copy of the GNU General Public License
13  *    along with this program; if not, write to the Free Software
14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
15  */
16
17 /*
18  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
19  */
20
21 #include "stg_client.h"
22
23 #include "conn.h"
24 #include "radlog.h"
25
26 #include "stg/locker.h"
27 #include "stg/common.h"
28
29 #include <map>
30 #include <utility>
31
32 using STG::RLM::Client;
33 using STG::RLM::Conn;
34 using STG::RLM::RESULT;
35
36 namespace {
37
38 Client* stgClient = NULL;
39
40 }
41
42 class Client::Impl
43 {
44     public:
45         explicit Impl(const std::string& address);
46         ~Impl();
47
48         bool stop() { return m_conn ? m_conn->stop() : true; }
49
50         RESULT request(REQUEST_TYPE type, const std::string& userName, const std::string& password, const PAIRS& pairs);
51
52     private:
53         std::string m_address;
54         std::unique_ptr<Conn> m_conn;
55
56         pthread_mutex_t m_mutex;
57         pthread_cond_t m_cond;
58         bool m_done;
59         RESULT m_result;
60
61         static bool callback(void* data, const RESULT& result)
62         {
63             Impl& impl = *static_cast<Impl*>(data);
64             STG_LOCKER lock(impl.m_mutex);
65             impl.m_result = result;
66             impl.m_done = true;
67             pthread_cond_signal(&impl.m_cond);
68             return true;
69         }
70 };
71
72 Client::Impl::Impl(const std::string& address)
73     : m_address(address)
74 {
75     try
76     {
77         m_conn.reset(new Conn(m_address, &Impl::callback, this));
78     }
79     catch (const std::runtime_error& ex)
80     {
81         RadLog("Connection error: %s.", ex.what());
82     }
83     pthread_mutex_init(&m_mutex, NULL);
84     pthread_cond_init(&m_cond, NULL);
85     m_done = false;
86 }
87
88 Client::Impl::~Impl()
89 {
90     pthread_cond_destroy(&m_cond);
91     pthread_mutex_destroy(&m_mutex);
92 }
93
94 RESULT Client::Impl::request(REQUEST_TYPE type, const std::string& userName, const std::string& password, const PAIRS& pairs)
95 {
96     STG_LOCKER lock(m_mutex);
97     if (!m_conn || !m_conn->connected())
98         m_conn.reset(new Conn(m_address, &Impl::callback, this));
99     if (!m_conn->connected())
100         throw Conn::Error("Failed to create connection to '" + m_address + "'.");
101
102     m_done = false;
103     m_conn->request(type, userName, password, pairs);
104     timespec ts;
105     clock_gettime(CLOCK_REALTIME, &ts);
106     ts.tv_sec += 5;
107     int res = 0;
108     while (!m_done && res == 0)
109         res = pthread_cond_timedwait(&m_cond, &m_mutex, &ts);
110     if (res != 0)
111         throw Conn::Error("Request failed.");
112     return m_result;
113 }
114
115 Client::Client(const std::string& address)
116     : m_impl(new Impl(address))
117 {
118 }
119
120 Client::~Client()
121 {
122 }
123
124 bool Client::stop()
125 {
126     return m_impl->stop();
127 }
128
129 RESULT Client::request(REQUEST_TYPE type, const std::string& userName, const std::string& password, const PAIRS& pairs)
130 {
131     return m_impl->request(type, userName, password, pairs);
132 }
133
134 Client* Client::get()
135 {
136     return stgClient;
137 }
138
139 bool Client::configure(const std::string& address)
140 {
141     if ( stgClient != NULL )
142         return stgClient->configure(address);
143     try {
144         stgClient = new Client(address);
145         return true;
146     } catch (const std::exception& ex) {
147         RadLog("Client configuration error: %s.", ex.what());
148     }
149     return false;
150 }