Update year specification of the Copyright-field to 2020
[anytun.git] / src / anyrtpproxy / commandHandler.cpp
1 /*
2  *  anytun
3  *
4  *  The secure anycast tunneling protocol (satp) defines a protocol used
5  *  for communication between any combination of unicast and anycast
6  *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
7  *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
8  *  ethernet, ip, arp ...). satp directly includes cryptography and
9  *  message authentication based on the methods used by SRTP.  It is
10  *  intended to deliver a generic, scaleable and secure solution for
11  *  tunneling and relaying of packets of any protocol.
12  *
13  *
14  *  Copyright (C) 2007-2014 Markus Grüneis, Othmar Gsenger, Erwin Nindl,
15  *                          Christian Pointner <satp@wirdorange.org>
16  *
17  *  This file is part of Anytun.
18  *
19  *  Anytun is free software: you can redistribute it and/or modify
20  *  it under the terms of the GNU General Public License as published by
21  *  the Free Software Foundation, either version 3 of the License, or
22  *  any later version.
23  *
24  *  Anytun is distributed in the hope that it will be useful,
25  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
26  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
27  *  GNU General Public License for more details.
28  *
29  *  You should have received a copy of the GNU General Public License
30  *  along with Anytun.  If not, see <http://www.gnu.org/licenses/>.
31  *
32  *  In addition, as a special exception, the copyright holders give
33  *  permission to link the code of portions of this program with the
34  *  OpenSSL library under certain conditions as described in each
35  *  individual source file, and distribute linked combinations
36  *  including the two.
37  *  You must obey the GNU General Public License in all respects
38  *  for all of the code used other than OpenSSL.  If you modify
39  *  file(s) with this exception, you may extend this exception to your
40  *  version of the file(s), but you are not obligated to do so.  If you
41  *  do not wish to do so, delete this exception statement from your
42  *  version.  If you delete this exception statement from all source
43  *  files in the program, then also delete it here.
44  */
45
46 #include <sstream>
47 #include <vector>
48
49 #include <iomanip>
50 #include <iostream>
51 #include <sstream>
52
53 #include <boost/bind.hpp>
54
55 #include "commandHandler.h"
56 #include "../buffer.h"
57 #include "../log.h"
58 #include "../syncQueue.h"
59 #include "syncRtpCommand.h"
60 #include "rtpSessionTable.h"
61 #include "callIdQueue.h"
62 #include "options.h"
63
64 #define MAX_COMMAND_LENGTH 1000
65
66 CommandHandler::CommandHandler(SyncQueue& q, std::string lp,PortWindow& pw) : thread_(boost::bind(run,this)),
67   queue_(q), running_(true), control_sock_(io_service_),
68   local_address_(""), local_port_(lp),port_window_(pw)
69 {
70   proto::resolver resolver(io_service_);
71   proto::resolver::query query(local_port_);
72   proto::endpoint e = *resolver.resolve(query);
73   control_sock_.open(e.protocol());
74   control_sock_.bind(e);
75 }
76
77 CommandHandler::CommandHandler(SyncQueue& q, string la, std::string lp, PortWindow& pw) : thread_(boost::bind(run,this)),
78   queue_(q), running_(true), control_sock_(io_service_),
79   local_address_(la), local_port_(lp),port_window_(pw)
80 {
81   proto::resolver resolver(io_service_);
82   proto::resolver::query query(local_address_, local_port_);
83   proto::endpoint e = *resolver.resolve(query);
84   control_sock_.open(e.protocol());
85   control_sock_.bind(e);
86 }
87
88 void CommandHandler::run(void* s)
89 {
90   CommandHandler* self = reinterpret_cast<CommandHandler*>(s);
91
92   Buffer buf(uint32_t(MAX_COMMAND_LENGTH));
93   try {
94     proto::endpoint remote_end;
95
96     int len;
97     for(;;) {
98       buf.setLength(MAX_COMMAND_LENGTH);
99
100       len = self->control_sock_.receive_from(boost::asio::buffer(buf.getBuf(), buf.getLength()), remote_end);
101       buf.setLength(len);
102
103       std::string ret = self->handle(std::string(reinterpret_cast<char*>(buf.getBuf()), buf.getLength())); // TODO: reinterpret is ugly
104
105       cLog.msg(Log::PRIO_DEBUG) << "CommandHandler received Command from " << remote_end << ", ret='" << ret << "'";
106
107       self->control_sock_.send_to(boost::asio::buffer(ret.c_str(), ret.length()), remote_end);
108     }
109   } catch(std::exception& e) {
110     self->running_ = false;
111   }
112   self->running_ = false;
113 }
114
115 bool CommandHandler::isRunning()
116 {
117   return running_;
118 }
119
120
121
122 std::string CommandHandler::handle(std::string command)
123 {
124   istringstream iss(command);
125   ostringstream oss;
126   std::string cookie;
127   std::string cmd;
128
129   iss >> cookie;
130   oss << cookie << " ";
131
132   if(iss.bad() || iss.eof()) {
133     oss << RET_ERR_SYNTAX;
134     return oss.str();
135   }
136   iss >> cmd;
137
138   std::vector<std::string> params;
139   while(!iss.bad() && !iss.eof()) {
140     std::string tmp;
141     iss >> tmp;
142     params.push_back(tmp);
143   }
144
145   switch(std::toupper(cmd[0])) {
146   case CMD_REQUEST:
147     if(params.size() < 4) { oss << RET_ERR_SYNTAX; break; }
148     oss << handleRequest(cmd.erase(0,1), params[0], params[1], params[2], params[3], (params.size() < 5) ? "" : params[4]);
149     break;
150   case CMD_RESPONSE:
151     if(params.size() < 4) { oss << RET_ERR_SYNTAX; break; }
152     oss << handleResponse(cmd.erase(0,1), params[0], params[1], params[2], params[3], (params.size() < 5) ? "" : params[4]);
153     break;
154   case CMD_DELETE:
155     if(params.size() < 2) { oss << RET_ERR_SYNTAX; break; }
156     oss << handleDelete(params[0], params[1], (params.size() < 3) ? "" : params[2]);
157     break;
158   case CMD_VERSION:
159     if(cmd.length() > 1 && cmd[1] == 'F') {
160       if(params.size() < 1) { oss << RET_ERR_SYNTAX; break; }
161       oss << handleVersionF(params[0]);
162       break;
163     }
164     oss << handleVersion();
165     break;
166   case CMD_INFO:
167     oss << handleInfo();
168     break;
169   default:
170     oss << RET_ERR_SYNTAX;
171     break;
172   }
173
174   return oss.str();
175 }
176
177 string CommandHandler::handleRequest(string modifiers, string call_id, string addr, string port, string from_tag, string to_tag)
178 {
179   std::cout << "received request[" << modifiers << "] command ('" << call_id << "','" << addr  << "','" << port
180             << "','" << from_tag << "','" << to_tag << "')" << std::endl;
181
182   try {
183     RtpSession::proto::resolver resolver(io_service_);
184     bool is_new;
185     RtpSession& session = gRtpSessionTable.getOrNewSession(call_id, is_new);
186     if(is_new) {
187       uint16_t port1 = port_window_.newPort(); // TODO: get next available port
188       uint16_t port2 = port_window_.newPort(); // TODO: get next available port
189       if(!port1 || !port2) {
190         if(port1) { port_window_.freePort(port1); }
191         if(port2) { port_window_.freePort(port2); }
192         throw std::runtime_error("no free port found");
193       }
194       std::stringstream ps1, ps2;
195       ps1 << port1;
196       ps2 << port2;
197
198       RtpSession::proto::endpoint e1, e2;
199       if(gOpt.getLocalAddr() == "") {
200         RtpSession::proto::resolver::query query1(ps1.str());
201         e1 = *resolver.resolve(query1);
202         RtpSession::proto::resolver::query query2(ps2.str());
203         e2 = *resolver.resolve(query2);
204       } else {
205         RtpSession::proto::resolver::query query1(gOpt.getLocalAddr(),ps1.str());
206         e1 = *resolver.resolve(query1);
207         RtpSession::proto::resolver::query query2(gOpt.getLocalAddr(),ps2.str());
208         e2 = *resolver.resolve(query2);
209       }
210
211       session.setLocalEnd1(e1);
212       session.setLocalEnd2(e2);
213     }
214     RtpSession::proto::resolver::query query(addr,port);
215     session.setRemoteEnd1(*resolver.resolve(query));
216
217     ostringstream oss;
218     oss << session.getLocalEnd2().port();
219     return oss.str();
220   } catch(std::exception& e) {
221     return RET_ERR_UNKNOWN; // TODO: change to corret error value
222   }
223 }
224
225 string CommandHandler::handleResponse(string modifiers, string call_id, string addr, string port, string from_tag, string to_tag)
226 {
227   std::cout << "received response[" << modifiers << "] command ('" << call_id << "','" << addr  << "','" << port
228             << "','" << from_tag << "','" << to_tag << "')" << std::endl;
229
230   try {
231     RtpSession& session = gRtpSessionTable.getSession(call_id);
232     RtpSession::proto::resolver resolver(io_service_);
233     RtpSession::proto::resolver::query query(addr,port);
234     session.setRemoteEnd2(*resolver.resolve(query));
235     session.isComplete(true);
236     SyncRtpCommand sc(call_id);
237     queue_.push(sc);
238
239     ostringstream oss;
240     oss << session.getLocalEnd1().port();
241     return oss.str();
242   } catch(std::exception& e) {
243     return RET_ERR_UNKNOWN; // TODO: change to corret error value
244   }
245 }
246
247 string CommandHandler::handleDelete(string call_id, string from_tag, string to_tag)
248 {
249   std::cout << "received delete command ('" << call_id << "','" << from_tag << "','" << to_tag << "')" << std::endl;
250
251   try {
252     RtpSession& session = gRtpSessionTable.getSession(call_id);
253     session.isDead(true);
254     SyncRtpCommand sc(call_id);
255     queue_.push(sc);
256
257     return RET_OK;
258   } catch(std::exception& e) {
259     return RET_ERR_UNKNOWN; // TODO: change to corret error value
260   }
261 }
262
263 string CommandHandler::handleVersion()
264 {
265   std::cout << "received version command" << std::endl;
266   return BASE_VERSION;
267 }
268
269 string CommandHandler::handleVersionF(string date_code)
270 {
271   std::cout << "received version[F] command ('" << date_code << "')" << std::endl;
272   if(!date_code.compare(SUP_VERSION)) {
273     return "1";
274   }
275
276   return "0";
277 }
278
279 string CommandHandler::handleInfo()
280 {
281   std::cout << "received info command, ignoring" << std::endl;
282   return RET_OK;
283 }
284