anyrtpproxy/options.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 methodes 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 anytun.org <satp@wirdorange.org>
  15  *
  16  *  This program is free software; you can redistribute it and/or modify
  17  *  it under the terms of the GNU General Public License version 2
  18  *  as published by the Free Software Foundation.
  19  *
  20  *  This program is distributed in the hope that it will be useful,
  21  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  22  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23  *  GNU General Public License for more details.
  24  *
  25  *  You should have received a copy of the GNU General Public License
  26  *  along with this program (see the file COPYING included with this
  27  *  distribution); if not, write to the Free Software Foundation, Inc.,
  28  *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  29  */
  30
  31 #include <iostream>
  32 #include <queue>
  33 #include <string>
  34 #include <sstream>
  35
  36 #include "options.h"
  37
  38 Options* Options::inst = NULL;
  39 Mutex Options::instMutex;
  40 Options& gOpt = Options::instance();
  41
  42 Options& Options::instance()
  43 {
  44   Lock lock(instMutex);
  45   static instanceCleaner c;
  46   if(!inst)
  47     inst = new Options();
  48
  49   return *inst;
  50 }
  51
  52 Options::Options() : control_interface_("0.0.0.0", 22222)
  53
  54 {
  55   progname_ = "anyrtpproxy";
  56   chroot_ = false;
  57   username_ = "nobody";
  58   chroot_dir_ = "/var/run";
  59   daemonize_ = true;
  60         local_sync_port_ = 0;
  61         rtp_start_port_ = 34000;
  62         rtp_end_port_ = 35000;
  63         no_nat_once_ = false;
  64         nat_ = false;
  65 }
  66
  67 Options::~Options()
  68 {
  69 }
  70
  71 #define PARSE_BOOL_PARAM(SHORT, LONG, VALUE)             \
  72     else if(str == SHORT || str == LONG)                 \
  73       VALUE = true;
  74
  75 #define PARSE_INVERSE_BOOL_PARAM(SHORT, LONG, VALUE)     \
  76     else if(str == SHORT || str == LONG)                 \
  77       VALUE = false;
  78
  79 #define PARSE_SCALAR_PARAM(SHORT, LONG, VALUE)           \
  80     else if(str == SHORT || str == LONG)                 \
  81     {                                                    \
  82       if(argc < 1 || argv[i+1][0] == '-')                \
  83         return false;                                    \
  84       std::stringstream tmp;                             \
  85       tmp << argv[i+1];                                  \
  86       tmp >> VALUE;                                      \
  87       argc--;                                            \
  88       i++;                                               \
  89     }
  90
  91 #define PARSE_SCALAR_PARAM2(SHORT, LONG, VALUE1, VALUE2) \
  92     else if(str == SHORT || str == LONG)                 \
  93     {                                                    \
  94       if(argc < 2 ||                                     \
  95          argv[i+1][0] == '-' || argv[i+2][0] == '-')     \
  96         return false;                                    \
  97       std::stringstream tmp;                             \
  98       tmp << argv[i+1] << " " << argv[i+2];              \
  99       tmp >> VALUE1;                                     \
 100       tmp >> VALUE2;                                     \
 101       argc-=2;                                           \
 102       i+=2;                                              \
 103     }
 104
 105 #define PARSE_STRING_PARAM(SHORT, LONG, VALUE)           \
 106     else if(str == SHORT || str == LONG)                 \
 107     {                                                    \
 108       if(argc < 1 || argv[i+1][0] == '-')                \
 109         return false;                                    \
 110       VALUE = std::string(argv[i+1]);                    \
 111       argc--;                                            \
 112       i++;                                               \
 113     }
 114
 115 #define PARSE_HEXSTRING_PARAM_SEC(SHORT, LONG, VALUE)    \
 116     else if(str == SHORT || str == LONG)                 \
 117     {                                                    \
 118       if(argc < 1 || argv[i+1][0] == '-')                \
 119         return false;                                    \
 120       VALUE = Buffer(std::string(argv[i+1]));            \
 121       for(size_t j=0; j < strlen(argv[i+1]); ++j)        \
 122         argv[i+1][j] = '#';                              \
 123       argc--;                                            \
 124       i++;                                               \
 125     }
 126
 127 #define PARSE_CSLIST_PARAM(SHORT, LONG, LIST)            \
 128     else if(str == SHORT || str == LONG)                 \
 129     {                                                    \
 130       if(argc < 1 || argv[i+1][0] == '-')                \
 131         return false;                                    \
 132       std::stringstream tmp(argv[i+1]);                  \
 133       /* LIST.clear(); */                                \
 134                         while (tmp.good())                                 \
 135                         {                                                  \
 136                                 std::string tmp_line;                            \
 137                                 getline(tmp,tmp_line,',');                       \
 138                                 LIST.push(tmp_line);                        \
 139                         }                                                  \
 140       argc--;                                            \
 141       i++;                                               \
 142     }
 143
 144 bool Options::parse(int argc, char* argv[])
 145 {
 146   Lock lock(mutex);
 147
 148   progname_ = argv[0];
 149   std::queue<std::string> host_port_queue;
 150   argc--;
 151   for(int i=1; argc > 0; ++i)
 152   {
 153     std::string str(argv[i]);
 154     argc--;
 155
 156     if(str == "-h" || str == "--help")
 157       return false;
 158     PARSE_BOOL_PARAM("-t","--chroot", chroot_)
 159     PARSE_BOOL_PARAM("-n","--nat", nat_)
 160     PARSE_BOOL_PARAM("-o","--no-nat-once", no_nat_once_)
 161     PARSE_SCALAR_PARAM("-u","--user", username_)
 162     PARSE_SCALAR_PARAM("-c","--chroot-dir", chroot_dir_)
 163     PARSE_INVERSE_BOOL_PARAM("-d","--nodaemonize", daemonize_)
 164     PARSE_STRING_PARAM("-s","--control", control_interface_)
 165     PARSE_SCALAR_PARAM2("-p","--port-range", rtp_start_port_, rtp_end_port_)
 166                 PARSE_CSLIST_PARAM("-M","--sync-hosts", host_port_queue)
 167     PARSE_SCALAR_PARAM("-S","--sync-port", local_sync_port_)
 168 //    PARSE_SCALAR_PARAM("-I","--sync-interface", local_sync_addr_)
 169     else
 170       return false;
 171   }
 172   while(!host_port_queue.empty())
 173   {
 174     std::stringstream tmp_stream(host_port_queue.front());
 175     OptionConnectTo oct;
 176     getline(tmp_stream,oct.host,':');
 177     if(!tmp_stream.good())
 178       return false;
 179     tmp_stream >> oct.port;
 180     host_port_queue.pop();
 181     connect_to_.push_back(oct);
 182   }
 183
 184   return sanityCheck();
 185 }
 186
 187 bool Options::sanityCheck()
 188 {
 189   if(!control_interface_.port_) control_interface_.port_ = 22220;
 190   return true;
 191 }
 192
 193 void Options::printUsage()
 194 {
 195   std::cout << "USAGE: anyrtpproxy" << std::endl;
 196   std::cout << "  [-h|--help]                      prints this..." << std::endl;
 197   std::cout << "  [-t|--chroot]                    chroot and drop priviledges" << std::endl;
 198   std::cout << "  [-u|--username] <username>       in case of chroot run as this user" << std::endl;
 199   std::cout << "  [-c|--chroot-dir] <directory>    directory to make a chroot to" << std::endl;
 200   std::cout << "  [-d|--nodaemonize]               don't run in background" << std::endl;
 201   std::cout << "  [-s|--control] <addr[:port]>     the address/port to listen on for control commands" << std::endl;
 202   std::cout << "  [-p|--port-range] <start> <end>  port range used to relay rtp connections" << std::endl;
 203   std::cout << "  [-n|--nat]                       enable permantent automatic nat detection(use only with anytun)" << std::endl;
 204   std::cout << "  [-o|--no-nat-once]               disable automatic nat detection for new connections" << std::endl;
 205 //  std::cout << "       [-I|--sync-interface] <ip-address>  local unicast(sync) ip address to bind to" << std::endl;
 206   std::cout << "  [-S|--sync-port] <port>          local unicast(sync) port to bind to" << std::endl;
 207   std::cout << "  [-M|--sync-hosts] <hostname|ip>:<port>[,<hostname|ip>:<port>[...]]"<< std::endl;
 208   std::cout << "                                   List of Remote Sync Hosts/Ports"<< std::endl;
 209 }
 210
 211 void Options::printOptions()
 212 {
 213   Lock lock(mutex);
 214   std::cout << "Options:" << std::endl;
 215   std::cout << "chroot='" << chroot_ << "'" << std::endl;
 216   std::cout << "username='" << username_ << "'" << std::endl;
 217   std::cout << "chroot-dir='" << chroot_dir_ << "'" << std::endl;
 218   std::cout << "daemonize='" << daemonize_ << "'" << std::endl;
 219   std::cout << "control-interface='" << control_interface_.toString() << "'" << std::endl;
 220 }
 221
 222 std::string Options::getProgname()
 223 {
 224   Lock lock(mutex);
 225   return progname_;
 226 }
 227
 228 bool Options::getChroot()
 229 {
 230   Lock lock(mutex);
 231   return chroot_;
 232 }
 233
 234 bool Options::getNat()
 235 {
 236   Lock lock(mutex);
 237   return nat_;
 238 }
 239
 240 bool Options::getNoNatOnce()
 241 {
 242   Lock lock(mutex);
 243   return no_nat_once_;
 244 }
 245
 246 std::string Options::getUsername()
 247 {
 248   Lock lock(mutex);
 249   return username_;
 250 }
 251
 252 std::string Options::getChrootDir()
 253 {
 254   Lock lock(mutex);
 255   return chroot_dir_;
 256 }
 257
 258 bool Options::getDaemonize()
 259 {
 260   Lock lock(mutex);
 261   return daemonize_;
 262 }
 263
 264 Host Options::getControlInterface()
 265 {
 266   Lock lock(mutex);
 267   return control_interface_;
 268 }
 269
 270 u_int16_t Options::getLocalSyncPort()
 271 {
 272   return local_sync_port_;
 273 }
 274
 275 Options& Options::setLocalSyncPort(u_int16_t l)
 276 {
 277   local_sync_port_ = l;
 278   return *this;
 279 }
 280
 281 u_int16_t Options::getRtpStartPort()
 282 {
 283   return rtp_start_port_;
 284 }
 285
 286 Options& Options::setRtpStartPort(u_int16_t l)
 287 {
 288   rtp_start_port_ = l;
 289   return *this;
 290 }
 291
 292 u_int16_t Options::getRtpEndPort()
 293 {
 294   return rtp_end_port_;
 295 }
 296
 297 Options& Options::setRtpEndPort(u_int16_t l)
 298 {
 299   rtp_end_port_ = l;
 300   return *this;
 301 }
 302
 303 ConnectToList Options::getConnectTo()
 304 {
 305   Lock lock(mutex);
 306   return connect_to_;
 307 }
 308