src/anyCtrOptions.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-2008 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 version 3 as
  21  *  published by the Free Software Foundation.
  22  *
  23  *  Anytun is distributed in the hope that it will be useful,
  24  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  25  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  26  *  GNU General Public License for more details.
  27  *
  28  *  You should have received a copy of the GNU General Public License
  29  *  along with anytun.  If not, see <http://www.gnu.org/licenses/>.
  30  */
  31
  32 #include <iostream>
  33 #include <queue>
  34 #include <string>
  35 #include <sstream>
  36
  37 #include "datatypes.h"
  38 #include "anyCtrOptions.h"
  39
  40 Options* Options::inst = NULL;
  41 Mutex Options::instMutex;
  42 Options& gOpt = Options::instance();
  43
  44 Options& Options::instance()
  45 {
  46   Lock lock(instMutex);
  47   static instanceCleaner c;
  48   if(!inst)
  49     inst = new Options();
  50
  51   return *inst;
  52 }
  53
  54 Options::Options()
  55 {
  56   progname_ = "anytun-controld";
  57   file_name_ = "";
  58   daemonize_ = true;
  59   chroot_ = false;
  60   username_ = "nobody";
  61   chroot_dir_ = "/var/run/anytun-controld";
  62   pid_file_ = "";
  63   bind_to_addr_ = "127.0.0.1";
  64   bind_to_port_ = "2323";
  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_HEXSTRING_PARAM_SEC(SHORT, LONG, VALUE)    \
 106     else if(str == SHORT || str == LONG)                 \
 107     {                                                    \
 108       if(argc < 1 || argv[i+1][0] == '-')                \
 109         return false;                                    \
 110       VALUE = Buffer(std::string(argv[i+1]));            \
 111       for(size_t j=0; j < strlen(argv[i+1]); ++j)        \
 112         argv[i+1][j] = '#';                              \
 113       argc--;                                            \
 114       i++;                                               \
 115     }
 116
 117 #define PARSE_CSLIST_PARAM(SHORT, LONG, LIST)            \
 118     else if(str == SHORT || str == LONG)                 \
 119     {                                                    \
 120       if(argc < 1 || argv[i+1][0] == '-')                \
 121         return false;                                    \
 122       std::stringstream tmp(argv[i+1]);                  \
 123                         while (tmp.good())                                 \
 124                         {                                                  \
 125                                 std::string tmp_line;                            \
 126                                 getline(tmp,tmp_line,',');                       \
 127                                 LIST.push(tmp_line);                             \
 128                         }                                                  \
 129       argc--;                                            \
 130       i++;                                               \
 131     }
 132
 133 bool Options::parse(int argc, char* argv[])
 134 {
 135   Lock lock(mutex);
 136
 137   progname_ = argv[0];
 138   argc--;
 139
 140   std::string control_host("");
 141   for(int i=1; argc > 0; ++i)
 142   {
 143     std::string str(argv[i]);
 144     argc--;
 145
 146     if(str == "-h" || str == "--help")
 147       return false;
 148     PARSE_SCALAR_PARAM("-f","--file", file_name_)
 149     PARSE_INVERSE_BOOL_PARAM("-D","--nodaemonize", daemonize_)
 150     PARSE_BOOL_PARAM("-C","--chroot", chroot_)
 151     PARSE_SCALAR_PARAM("-u","--username", username_)
 152     PARSE_SCALAR_PARAM("-H","--chroot-dir", chroot_dir_)
 153     PARSE_SCALAR_PARAM("-P","--write-pid", pid_file_)
 154     PARSE_SCALAR_PARAM("-X","--control-host", control_host)
 155     else
 156       return false;
 157   }
 158
 159   if(control_host != "")
 160     return splitAndSetHostPort(control_host);
 161
 162   return true;
 163 }
 164
 165 bool Options::splitAndSetHostPort(std::string hostPort)
 166 {
 167   if(hostPort.length() >= 2 && hostPort[0] == ':' && hostPort[1] != ':') {
 168     bind_to_addr_ = "";
 169     hostPort.erase(0,1);
 170     std::stringstream tmp_stream(hostPort);
 171     tmp_stream >> bind_to_port_;
 172     return true;
 173   }
 174
 175   size_t pos = hostPort.find_first_of("[");
 176
 177   if(pos != std::string::npos && pos != 0)
 178     return false; // an [ was found but not at the beginning
 179
 180   bool hasPort = false;
 181   if(pos != std::string::npos) {
 182     hostPort.erase(pos, 1);
 183     pos = hostPort.find_first_of("]");
 184
 185     if(pos == std::string::npos)
 186       return false; // no trailing ] although an leading [ was found
 187
 188     if(pos < hostPort.length()-2) {
 189
 190       if(hostPort[pos+1] != ':')
 191         return false; // wrong port delimieter
 192
 193       hostPort[pos+1] = '/';
 194       hasPort = true;
 195     }
 196     else if(pos != hostPort.length()-1)
 197       return false; // to few characters left
 198
 199     hostPort.erase(pos, 1);
 200   }
 201   else {
 202     pos = hostPort.find_first_of(":");
 203     if(pos != std::string::npos && pos == hostPort.find_last_of(":")) {
 204       // an ':' has been found and it is the only one -> assuming port present
 205       hasPort = true;
 206       hostPort[pos] = '/';
 207     }
 208   }
 209
 210   if(hasPort) {
 211     std::stringstream tmp_stream(hostPort);
 212
 213     getline(tmp_stream, bind_to_addr_, '/');
 214     if(!tmp_stream.good())
 215       return false;
 216
 217     tmp_stream >> bind_to_port_;
 218   }
 219   else {
 220     bind_to_addr_ = hostPort;
 221     bind_to_port_ = "2323"; // default sync port
 222   }
 223
 224   return true;
 225 }
 226
 227 void Options::printUsage()
 228 {
 229   std::cout << "USAGE:" << std::endl;
 230   std::cout << "anytun-controld [-h|--help]                  prints this..." << std::endl;
 231   std::cout << "                [-D|--nodaemonize]           don't run in background" << std::endl;
 232   std::cout << "                [-C|--chroot]                chroot and drop privileges" << std::endl;
 233   std::cout << "                [-u|--username] <username>   if chroot change to this user" << std::endl;
 234   std::cout << "                [-H|--chroot-dir] <path>     chroot to this directory" << std::endl;
 235   std::cout << "                [-P|--write-pid] <path>      write pid to this file" << std::endl;
 236   std::cout << "                [-f|--file] <path>           path to file" << std::endl;
 237   std::cout << "                [-X|--control-host] < <hostname|ip>[:<port>] | :<port> >" << std::endl;
 238   std::cout << "                                             local tcp port and or ip address to bind to" << std::endl;
 239 }
 240
 241 void Options::printOptions()
 242 {
 243   Lock lock(mutex);
 244   std::cout << "Options:" << std::endl;
 245   std::cout << "daemonize=" << daemonize_ << std::endl;
 246   std::cout << "chroot=" << chroot_ << std::endl;
 247   std::cout << "username='" << username_ << "'" << std::endl;
 248   std::cout << "chroot_dir='" << chroot_dir_ << "'" << std::endl;
 249   std::cout << "pid_file='" << pid_file_ << "'" << std::endl;
 250   std::cout << "bind_to_addr_='" << bind_to_addr_ << "'" << std::endl;
 251   std::cout << "bind_to_port_='" << bind_to_port_ << "'" << std::endl;
 252 }
 253
 254 std::string Options::getProgname()
 255 {
 256   Lock lock(mutex);
 257   return progname_;
 258 }
 259
 260
 261 Options& Options::setProgname(std::string p)
 262 {
 263   Lock lock(mutex);
 264   progname_ = p;
 265   return *this;
 266 }
 267
 268 bool Options::getDaemonize()
 269 {
 270   return daemonize_;
 271 }
 272
 273 Options& Options::setDaemonize(bool d)
 274 {
 275   daemonize_ = d;
 276   return *this;
 277 }
 278
 279 bool Options::getChroot()
 280 {
 281   return chroot_;
 282 }
 283
 284 Options& Options::setChroot(bool c)
 285 {
 286   chroot_ = c;
 287   return *this;
 288 }
 289
 290 std::string Options::getUsername()
 291 {
 292   Lock lock(mutex);
 293   return username_;
 294 }
 295
 296 Options& Options::setUsername(std::string u)
 297 {
 298   Lock lock(mutex);
 299   username_ = u;
 300   return *this;
 301 }
 302
 303 std::string Options::getChrootDir()
 304 {
 305   Lock lock(mutex);
 306   return chroot_dir_;
 307 }
 308
 309 Options& Options::setChrootDir(std::string c)
 310 {
 311   Lock lock(mutex);
 312   chroot_dir_ = c;
 313   return *this;
 314 }
 315
 316 std::string Options::getPidFile()
 317 {
 318   Lock lock(mutex);
 319   return pid_file_;
 320 }
 321
 322 Options& Options::setPidFile(std::string p)
 323 {
 324   Lock lock(mutex);
 325   pid_file_ = p;
 326   return *this;
 327 }
 328
 329 std::string Options::getFileName()
 330 {
 331   Lock lock(mutex);
 332   return file_name_;
 333 }
 334
 335 Options& Options::setFileName(std::string f)
 336 {
 337   Lock lock(mutex);
 338   file_name_ = f;
 339   return *this;
 340 }
 341
 342 std::string Options::getBindToAddr()
 343 {
 344   Lock lock(mutex);
 345   return bind_to_addr_;
 346 }
 347
 348 Options& Options::setBindToAddr(std::string b)
 349 {
 350   Lock lock(mutex);
 351   bind_to_addr_ = b;
 352   return *this;
 353 }
 354
 355 std::string Options::getBindToPort()
 356 {
 357   return bind_to_port_;
 358 }
 359
 360 Options& Options::setBindToPort(std::string b)
 361 {
 362   bind_to_port_ = b;
 363   return *this;
 364 }