]> git.stg.codes - stg.git/blob - projects/rlm_stg/conn.cpp
Added ability to optionally transfer resetable value to a real var.
[stg.git] / projects / rlm_stg / conn.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 "conn.h"
22
23 #include "radlog.h"
24 #include "stgpair.h"
25
26 #include "stg/json_parser.h"
27 #include "stg/json_generator.h"
28 #include "stg/locker.h"
29
30 #include <cerrno>
31 #include <cstring>
32
33 #include <sys/types.h>
34 #include <sys/socket.h>
35 #include <sys/un.h> // UNIX
36 #include <netinet/in.h> // IP
37 #include <netinet/tcp.h> // TCP
38 #include <netdb.h>
39
40 namespace RLM = STG::RLM;
41
42 using RLM::Conn;
43 using STG::JSON::Parser;
44 using STG::JSON::PairsParser;
45 using STG::JSON::EnumParser;
46 using STG::JSON::NodeParser;
47 using STG::JSON::Gen;
48 using STG::JSON::MapGen;
49 using STG::JSON::StringGen;
50
51 namespace
52 {
53
54 double CONN_TIMEOUT = 60;
55 double PING_TIMEOUT = 10;
56
57 struct ChannelConfig {
58     struct Error : std::runtime_error {
59         explicit Error(const std::string& message) : runtime_error(message) {}
60     };
61
62     explicit ChannelConfig(std::string address);
63
64     std::string transport;
65     std::string key;
66     std::string address;
67     std::string portStr;
68     uint16_t port;
69 };
70
71 std::string toStage(RLM::REQUEST_TYPE type)
72 {
73     switch (type)
74     {
75         case RLM::AUTHORIZE: return "authorize";
76         case RLM::AUTHENTICATE: return "authenticate";
77         case RLM::POST_AUTH: return "postauth";
78         case RLM::PRE_ACCT: return "preacct";
79         case RLM::ACCOUNT: return "accounting";
80     }
81     return "";
82 }
83
84 enum Packet
85 {
86     PING,
87     PONG,
88     DATA
89 };
90
91 std::map<std::string, Packet> packetCodes;
92 std::map<std::string, bool> resultCodes;
93 std::map<std::string, int> returnCodes;
94
95 class PacketParser : public EnumParser<Packet>
96 {
97     public:
98         PacketParser(NodeParser* next, Packet& packet, std::string& packetStr)
99             : EnumParser(next, packet, packetStr, packetCodes)
100         {
101             if (!packetCodes.empty())
102                 return;
103             packetCodes["ping"] = PING;
104             packetCodes["pong"] = PONG;
105             packetCodes["data"] = DATA;
106         }
107 };
108
109 class ResultParser : public EnumParser<bool>
110 {
111     public:
112         ResultParser(NodeParser* next, bool& result, std::string& resultStr)
113             : EnumParser(next, result, resultStr, resultCodes)
114         {
115             if (!resultCodes.empty())
116                 return;
117             resultCodes["no"] = false;
118             resultCodes["ok"] = true;
119         }
120 };
121
122 class ReturnCodeParser : public EnumParser<int>
123 {
124     public:
125         ReturnCodeParser(NodeParser* next, int& returnCode, std::string& returnCodeStr)
126             : EnumParser(next, returnCode, returnCodeStr, returnCodes)
127         {
128             if (!returnCodes.empty())
129                 return;
130             returnCodes["reject"]   = STG_REJECT;
131             returnCodes["fail"]     = STG_FAIL;
132             returnCodes["ok"]       = STG_OK;
133             returnCodes["handled"]  = STG_HANDLED;
134             returnCodes["invalid"]  = STG_INVALID;
135             returnCodes["userlock"] = STG_USERLOCK;
136             returnCodes["notfound"] = STG_NOTFOUND;
137             returnCodes["noop"]     = STG_NOOP;
138             returnCodes["updated"]  = STG_UPDATED;
139         }
140 };
141
142 class TopParser : public NodeParser
143 {
144     public:
145         typedef void (*Callback) (void* /*data*/);
146         TopParser(Callback callback, void* data)
147             : m_packet(PING),
148               m_result(false),
149               m_returnCode(STG_REJECT),
150               m_packetParser(this, m_packet, m_packetStr),
151               m_resultParser(this, m_result, m_resultStr),
152               m_returnCodeParser(this, m_returnCode, m_returnCodeStr),
153               m_replyParser(this, m_reply),
154               m_modifyParser(this, m_modify),
155               m_callback(callback), m_data(data)
156         {}
157
158         virtual NodeParser* parseStartMap() { return this; }
159         virtual NodeParser* parseMapKey(const std::string& value)
160         {
161             std::string key = ToLower(value);
162
163             if (key == "packet")
164                 return &m_packetParser;
165             else if (key == "result")
166                 return &m_resultParser;
167             else if (key == "reply")
168                 return &m_replyParser;
169             else if (key == "modify")
170                 return &m_modifyParser;
171             else if (key == "return_code")
172                 return &m_returnCodeParser;
173
174             return this;
175         }
176         virtual NodeParser* parseEndMap() { m_callback(m_data); return this; }
177
178         const std::string& packetStr() const { return m_packetStr; }
179         Packet packet() const { return m_packet; }
180         const std::string& resultStr() const { return m_resultStr; }
181         bool result() const { return m_result; }
182         const std::string& returnCodeStr() const { return m_returnCodeStr; }
183         int returnCode() const { return m_returnCode; }
184         const PairsParser::Pairs& reply() const { return m_reply; }
185         const PairsParser::Pairs& modify() const { return m_modify; }
186
187     private:
188         std::string m_packetStr;
189         Packet m_packet;
190         std::string m_resultStr;
191         bool m_result;
192         std::string m_returnCodeStr;
193         int m_returnCode;
194         PairsParser::Pairs m_reply;
195         PairsParser::Pairs m_modify;
196
197         PacketParser m_packetParser;
198         ResultParser m_resultParser;
199         ReturnCodeParser m_returnCodeParser;
200         PairsParser m_replyParser;
201         PairsParser m_modifyParser;
202
203         Callback m_callback;
204         void* m_data;
205 };
206
207 class ProtoParser : public Parser
208 {
209     public:
210         ProtoParser(TopParser::Callback callback, void* data)
211             : Parser( &m_topParser ),
212               m_topParser(callback, data)
213         {}
214
215         const std::string& packetStr() const { return m_topParser.packetStr(); }
216         Packet packet() const { return m_topParser.packet(); }
217         const std::string& resultStr() const { return m_topParser.resultStr(); }
218         bool result() const { return m_topParser.result(); }
219         const std::string& returnCodeStr() const { return m_topParser.returnCodeStr(); }
220         int returnCode() const { return m_topParser.returnCode(); }
221         const PairsParser::Pairs& reply() const { return m_topParser.reply(); }
222         const PairsParser::Pairs& modify() const { return m_topParser.modify(); }
223
224     private:
225         TopParser m_topParser;
226 };
227
228 class PacketGen : public Gen
229 {
230     public:
231         explicit PacketGen(const std::string& type)
232             : m_type(type)
233         {
234             m_gen.add("packet", m_type);
235         }
236         void run(yajl_gen_t* handle) const
237         {
238             m_gen.run(handle);
239         }
240         PacketGen& add(const std::string& key, const std::string& value)
241         {
242             m_gen.add(key, new StringGen(value));
243             return *this;
244         }
245         PacketGen& add(const std::string& key, MapGen& map)
246         {
247             m_gen.add(key, map);
248             return *this;
249         }
250     private:
251         MapGen m_gen;
252         StringGen m_type;
253 };
254
255 }
256
257 class Conn::Impl
258 {
259 public:
260     Impl(const std::string& address, Callback callback, void* data);
261     ~Impl();
262
263     bool stop();
264     bool connected() const { return m_connected; }
265
266     bool request(REQUEST_TYPE type, const std::string& userName, const std::string& password, const PAIRS& pairs);
267
268 private:
269     ChannelConfig m_config;
270
271     int m_sock;
272
273     bool m_running;
274     bool m_stopped;
275
276     time_t m_lastPing;
277     time_t m_lastActivity;
278
279     pthread_t m_thread;
280     pthread_mutex_t m_mutex;
281
282     Callback m_callback;
283     void* m_data;
284
285     ProtoParser m_parser;
286
287     bool m_connected;
288
289     void m_writeHeader(REQUEST_TYPE type, const std::string& userName, const std::string& password);
290     void m_writePairBlock(const PAIRS& source);
291     PAIRS m_readPairBlock();
292
293     static void* run(void* );
294
295     void runImpl();
296
297     bool start();
298
299     int connect();
300     int connectTCP();
301     int connectUNIX();
302
303     bool read();
304     bool tick();
305
306     static void process(void* data);
307     void processPing();
308     void processPong();
309     void processData();
310     bool sendPing();
311     bool sendPong();
312
313     static bool write(void* data, const char* buf, size_t size);
314 };
315
316 ChannelConfig::ChannelConfig(std::string addr)
317 {
318     // unix:pass@/var/run/stg.sock
319     // tcp:secret@192.168.0.1:12345
320     // udp:key@isp.com.ua:54321
321
322     size_t pos = addr.find_first_of(':');
323     if (pos == std::string::npos)
324         throw Error("Missing transport name.");
325     transport = ToLower(addr.substr(0, pos));
326     addr = addr.substr(pos + 1);
327     if (addr.empty())
328         throw Error("Missing address to connect to.");
329     pos = addr.find_first_of('@');
330     if (pos != std::string::npos) {
331         key = addr.substr(0, pos);
332         addr = addr.substr(pos + 1);
333         if (addr.empty())
334             throw Error("Missing address to connect to.");
335     }
336     if (transport == "unix")
337     {
338         address = addr;
339         return;
340     }
341     pos = addr.find_first_of(':');
342     if (pos == std::string::npos)
343         throw Error("Missing port.");
344     address = addr.substr(0, pos);
345     portStr = addr.substr(pos + 1);
346     if (str2x(portStr, port))
347         throw Error("Invalid port value.");
348 }
349
350 Conn::Conn(const std::string& address, Callback callback, void* data)
351     : m_impl(new Impl(address, callback, data))
352 {
353 }
354
355 Conn::~Conn()
356 {
357 }
358
359 bool Conn::stop()
360 {
361     return m_impl->stop();
362 }
363
364 bool Conn::connected() const
365 {
366     return m_impl->connected();
367 }
368
369 bool Conn::request(REQUEST_TYPE type, const std::string& userName, const std::string& password, const PAIRS& pairs)
370 {
371     return m_impl->request(type, userName, password, pairs);
372 }
373
374 Conn::Impl::Impl(const std::string& address, Callback callback, void* data)
375     : m_config(address),
376       m_sock(connect()),
377       m_running(false),
378       m_stopped(true),
379       m_lastPing(time(NULL)),
380       m_lastActivity(m_lastPing),
381       m_callback(callback),
382       m_data(data),
383       m_parser(&Conn::Impl::process, this),
384       m_connected(true)
385 {
386     pthread_mutex_init(&m_mutex, NULL);
387 }
388
389 Conn::Impl::~Impl()
390 {
391     stop();
392     shutdown(m_sock, SHUT_RDWR);
393     close(m_sock);
394     pthread_mutex_destroy(&m_mutex);
395 }
396
397 bool Conn::Impl::stop()
398 {
399     m_connected = false;
400
401     if (m_stopped)
402         return true;
403
404     m_running = false;
405
406     for (size_t i = 0; i < 25 && !m_stopped; i++) {
407         struct timespec ts = {0, 200000000};
408         nanosleep(&ts, NULL);
409     }
410
411     if (m_stopped) {
412         pthread_join(m_thread, NULL);
413         return true;
414     }
415
416     return false;
417 }
418
419 bool Conn::Impl::request(REQUEST_TYPE type, const std::string& userName, const std::string& password, const PAIRS& pairs)
420 {
421     if (!m_running)
422         if (!start())
423             return false;
424     MapGen map;
425     for (PAIRS::const_iterator it = pairs.begin(); it != pairs.end(); ++it)
426         map.add(it->first, new StringGen(it->second));
427     map.add("Radius-Username", new StringGen(userName));
428     map.add("Radius-Userpass", new StringGen(password));
429
430     PacketGen gen("data");
431     gen.add("stage", toStage(type))
432        .add("pairs", map);
433
434     STG_LOCKER lock(m_mutex);
435
436     m_lastPing = time(NULL);
437
438     return generate(gen, &Conn::Impl::write, this);
439 }
440
441 void Conn::Impl::runImpl()
442 {
443     m_running = true;
444
445     while (m_running) {
446         fd_set fds;
447
448         FD_ZERO(&fds);
449         FD_SET(m_sock, &fds);
450
451         struct timeval tv;
452         tv.tv_sec = 0;
453         tv.tv_usec = 500000;
454
455         int res = select(m_sock + 1, &fds, NULL, NULL, &tv);
456         if (res < 0)
457         {
458             if (errno == EINTR)
459                 continue;
460             RadLog("'select' is failed: %s", strerror(errno));
461             break;
462         }
463
464
465         if (!m_running)
466             break;
467
468         STG_LOCKER lock(m_mutex);
469
470         if (res > 0)
471         {
472             if (FD_ISSET(m_sock, &fds))
473                 m_running = read();
474         }
475         else
476             m_running = tick();
477     }
478
479     m_connected = false;
480     m_stopped = true;
481 }
482
483 bool Conn::Impl::start()
484 {
485     int res = pthread_create(&m_thread, NULL, &Conn::Impl::run, this);
486     if (res != 0)
487         return false;
488     return true;
489 }
490
491 int Conn::Impl::connect()
492 {
493     if (m_config.transport == "tcp")
494         return connectTCP();
495     else if (m_config.transport == "unix")
496         return connectUNIX();
497     throw Error("Invalid transport type: '" + m_config.transport + "'. Should be 'tcp' or 'unix'.");
498 }
499
500 int Conn::Impl::connectTCP()
501 {
502     addrinfo hints;
503     memset(&hints, 0, sizeof(addrinfo));
504
505     hints.ai_family = AF_INET;       /* Allow IPv4 */
506     hints.ai_socktype = SOCK_STREAM; /* Stream socket */
507     hints.ai_flags = 0;     /* For wildcard IP address */
508     hints.ai_protocol = 0;           /* Any protocol */
509     hints.ai_canonname = NULL;
510     hints.ai_addr = NULL;
511     hints.ai_next = NULL;
512
513     addrinfo* ais = NULL;
514     int res = getaddrinfo(m_config.address.c_str(), m_config.portStr.c_str(), &hints, &ais);
515     if (res != 0)
516         throw Error("Error resolvin address '" + m_config.address + "': " + gai_strerror(res));
517
518     for (addrinfo* ai = ais; ai != NULL; ai = ai->ai_next)
519     {
520         int fd = socket(AF_INET, SOCK_STREAM, 0);
521         if (fd == -1)
522         {
523             Error error(std::string("Error creating TCP socket: ") + strerror(errno));
524             freeaddrinfo(ais);
525             throw error;
526         }
527         if (::connect(fd, ai->ai_addr, ai->ai_addrlen) == -1)
528         {
529             shutdown(fd, SHUT_RDWR);
530             close(fd);
531             RadLog("'connect' is failed: %s", strerror(errno));
532             continue;
533         }
534         freeaddrinfo(ais);
535         return fd;
536     }
537
538     freeaddrinfo(ais);
539
540     throw Error("Failed to resolve '" + m_config.address);
541 };
542
543 int Conn::Impl::connectUNIX()
544 {
545     int fd = socket(AF_UNIX, SOCK_STREAM, 0);
546     if (fd == -1)
547         throw Error(std::string("Error creating UNIX socket: ") + strerror(errno));
548     struct sockaddr_un addr;
549     memset(&addr, 0, sizeof(addr));
550     addr.sun_family = AF_UNIX;
551     strncpy(addr.sun_path, m_config.address.c_str(), m_config.address.length());
552     if (::connect(fd, reinterpret_cast<struct sockaddr*>(&addr), sizeof(addr)) == -1)
553     {
554         Error error(std::string("Error connecting UNIX socket: ") + strerror(errno));
555         shutdown(fd, SHUT_RDWR);
556         close(fd);
557         throw error;
558     }
559     return fd;
560 }
561
562 bool Conn::Impl::read()
563 {
564     static std::vector<char> buffer(1024);
565     ssize_t res = ::read(m_sock, buffer.data(), buffer.size());
566     if (res < 0)
567     {
568         RadLog("Failed to read data: %s", strerror(errno));
569         return false;
570     }
571     m_lastActivity = time(NULL);
572     RadLog("Read %d bytes.\n%s\n", res, std::string(buffer.data(), res).c_str());
573     if (res == 0)
574     {
575         m_parser.last();
576         return false;
577     }
578     return m_parser.append(buffer.data(), res);
579 }
580
581 bool Conn::Impl::tick()
582 {
583     time_t now = time(NULL);
584     if (difftime(now, m_lastActivity) > CONN_TIMEOUT)
585     {
586         int delta = difftime(now, m_lastActivity);
587         RadLog("Connection timeout: %d sec.", delta);
588         //m_logger("Connection to " + m_remote + " timed out.");
589         return false;
590     }
591     if (difftime(now, m_lastPing) > PING_TIMEOUT)
592     {
593         int delta = difftime(now, m_lastPing);
594         RadLog("Ping timeout: %d sec. Sending ping...", delta);
595         sendPing();
596     }
597     return true;
598 }
599
600 void Conn::Impl::process(void* data)
601 {
602     Impl& impl = *static_cast<Impl*>(data);
603     switch (impl.m_parser.packet())
604     {
605         case PING:
606             impl.processPing();
607             return;
608         case PONG:
609             impl.processPong();
610             return;
611         case DATA:
612             impl.processData();
613             return;
614     }
615     RadLog("Received invalid packet type: '%s'.", impl.m_parser.packetStr().c_str());
616 }
617
618 void Conn::Impl::processPing()
619 {
620     sendPong();
621 }
622
623 void Conn::Impl::processPong()
624 {
625     m_lastActivity = time(NULL);
626 }
627
628 void Conn::Impl::processData()
629 {
630     RESULT data;
631     if (m_parser.result())
632     {
633         for (PairsParser::Pairs::const_iterator it = m_parser.reply().begin(); it != m_parser.reply().end(); ++it)
634             data.reply.push_back(std::make_pair(it->first, it->second));
635         for (PairsParser::Pairs::const_iterator it = m_parser.modify().begin(); it != m_parser.modify().end(); ++it)
636             data.modify.push_back(std::make_pair(it->first, it->second));
637         data.returnCode = STG_UPDATED;
638     }
639     else
640         data.returnCode = m_parser.returnCode();
641     m_callback(m_data, data);
642 }
643
644 bool Conn::Impl::sendPing()
645 {
646     PacketGen gen("ping");
647
648     m_lastPing = time(NULL);
649
650     return generate(gen, &Conn::Impl::write, this);
651 }
652
653 bool Conn::Impl::sendPong()
654 {
655     PacketGen gen("pong");
656
657     m_lastPing = time(NULL);
658
659     return generate(gen, &Conn::Impl::write, this);
660 }
661
662 bool Conn::Impl::write(void* data, const char* buf, size_t size)
663 {
664     std::string json(buf, size);
665     RadLog("Sending JSON: %s", json.c_str());
666     Conn::Impl& impl = *static_cast<Conn::Impl*>(data);
667     while (size > 0)
668     {
669         ssize_t res = ::send(impl.m_sock, buf, size, MSG_NOSIGNAL);
670         if (res < 0)
671         {
672             impl.m_connected = false;
673             RadLog("Failed to write data: %s.", strerror(errno));
674             return false;
675         }
676         size -= res;
677     }
678     return true;
679 }
680
681 void* Conn::Impl::run(void* data)
682 {
683     Impl& impl = *static_cast<Impl*>(data);
684     impl.runImpl();
685     return NULL;
686 }