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 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 "datatypes.h"
  37 #include "anyctrOptions.h"
  38
  39 Options* Options::inst = NULL;
  40 Mutex Options::instMutex;
  41 Options& gOpt = Options::instance();
  42
  43 Options& Options::instance()
  44 {
  45   Lock lock(instMutex);
  46   static instanceCleaner c;
  47   if(!inst)
  48     inst = new Options();
  49
  50   return *inst;
  51 }
  52
  53 Options::Options() : key_(u_int32_t(0)), salt_(u_int32_t(0))
  54 {
  55   progname_ = "anyctr";
  56   remote_addr_ = "";
  57   remote_port_ = 4444;
  58   ifconfig_param_remote_netmask_ = "255.255.255.0";
  59   seq_window_size_ = 100;
  60   kd_prf_ = "aes-ctr";
  61   mux_ = 0;
  62   network_prefix_length_ = 32;
  63 }
  64
  65 Options::~Options()
  66 {
  67 }
  68
  69 #define PARSE_BOOL_PARAM(SHORT, LONG, VALUE)             \
  70     else if(str == SHORT || str == LONG)                 \
  71       VALUE = true;
  72
  73 #define PARSE_INVERSE_BOOL_PARAM(SHORT, LONG, VALUE)     \
  74     else if(str == SHORT || str == LONG)                 \
  75       VALUE = false;
  76
  77 #define PARSE_SCALAR_PARAM(SHORT, LONG, VALUE)           \
  78     else if(str == SHORT || str == LONG)                 \
  79     {                                                    \
  80       if(argc < 1 || argv[i+1][0] == '-')                \
  81         return false;                                    \
  82       std::stringstream tmp;                             \
  83       tmp << argv[i+1];                                  \
  84       tmp >> VALUE;                                      \
  85       argc--;                                            \
  86       i++;                                               \
  87     }
  88
  89 #define PARSE_SCALAR_PARAM2(SHORT, LONG, VALUE1, VALUE2) \
  90     else if(str == SHORT || str == LONG)                 \
  91     {                                                    \
  92       if(argc < 2 ||                                     \
  93          argv[i+1][0] == '-' || argv[i+2][0] == '-')     \
  94         return false;                                    \
  95       std::stringstream tmp;                             \
  96       tmp << argv[i+1] << " " << argv[i+2];              \
  97       tmp >> VALUE1;                                     \
  98       tmp >> VALUE2;                                     \
  99       argc-=2;                                           \
 100       i+=2;                                              \
 101     }
 102
 103 #define PARSE_HEXSTRING_PARAM(SHORT, LONG, VALUE)        \
 104     else if(str == SHORT || str == LONG)                 \
 105     {                                                    \
 106       if(argc < 1 || argv[i+1][0] == '-')                \
 107         return false;                                    \
 108       VALUE = Buffer(std::string(argv[i+1]));            \
 109       argc--;                                            \
 110       i++;                                               \
 111     }
 112
 113 #define PARSE_CSLIST_PARAM(SHORT, LONG, LIST)            \
 114     else if(str == SHORT || str == LONG)                 \
 115     {                                                    \
 116       if(argc < 1 || argv[i+1][0] == '-')                \
 117         return false;                                    \
 118       std::stringstream tmp(argv[i+1]);                  \
 119                         while (tmp.good())                                 \
 120                         {                                                  \
 121                                 std::string tmp_line;                            \
 122                                 getline(tmp,tmp_line,',');                       \
 123                                 LIST.push(tmp_line);                             \
 124                         }                                                  \
 125       argc--;                                            \
 126       i++;                                               \
 127     }
 128
 129 bool Options::parse(int argc, char* argv[])
 130 {
 131   Lock lock(mutex);
 132
 133   progname_ = argv[0];
 134   argc--;
 135   std::queue<std::string> host_port_queue;
 136   for(int i=1; argc > 0; ++i)
 137   {
 138     std::string str(argv[i]);
 139     argc--;
 140
 141     if(str == "-h" || str == "--help")
 142       return false;
 143     PARSE_SCALAR_PARAM("-r","--remote-host", remote_addr_)
 144     PARSE_SCALAR_PARAM("-o","--remote-port", remote_port_)
 145     PARSE_SCALAR_PARAM("-n","--prefix", ifconfig_param_remote_netmask_)
 146     PARSE_SCALAR_PARAM("-w","--window-size", seq_window_size_)
 147     PARSE_SCALAR_PARAM("-m","--mux", mux_)
 148     PARSE_SCALAR_PARAM("-l","--prefix-len", network_prefix_length_)
 149     PARSE_HEXSTRING_PARAM("-K","--key", key_)
 150     PARSE_HEXSTRING_PARAM("-A","--salt", salt_)
 151     PARSE_SCALAR_PARAM("-k","--kd-prf", kd_prf_)
 152     else
 153       return false;
 154   }
 155
 156         while(!host_port_queue.empty())
 157         {
 158                 std::stringstream tmp_stream(host_port_queue.front());
 159                 OptionConnectTo oct;
 160                 getline(tmp_stream,oct.host,':');
 161                 if(!tmp_stream.good())
 162                         return false;
 163                 tmp_stream >> oct.port;
 164                 host_port_queue.pop();
 165                 connect_to_.push_back(oct);
 166         }
 167   return true;
 168 }
 169
 170 void Options::printUsage()
 171 {
 172   std::cout << "USAGE:" << std::endl;
 173   std::cout << "anyctr [-h|--help]                         prints this..." << std::endl;
 174   std::cout << "       [-r|--remote-host] <hostname|ip>    remote host" << std::endl;
 175   std::cout << "       [-o|--remote-port] <port>           remote port" << std::endl;
 176   std::cout << "       [-n|--prefix] <remote net>          remote subnet for route" << std::endl;
 177   std::cout << "       [-w|--window-size] <window size>    seqence number window size" << std::endl;
 178   std::cout << "       [-m|--mux] <mux-id>                 the multiplex id to use" << std::endl;
 179   std::cout << "       [-l|--prefix-len] <prefix length>   network prefix length" << std::endl;
 180   std::cout << "       [-K|--key] <master key>             master key to use for encryption" << std::endl;
 181   std::cout << "       [-A|--salt] <master salt>           master salt to use for encryption" << std::endl;
 182   std::cout << "       [-k|--kd-prf] <kd-prf type>         key derivation pseudo random function" << std::endl;
 183 }
 184
 185 void Options::printOptions()
 186 {
 187   Lock lock(mutex);
 188   std::cout << "Options:" << std::endl;
 189   std::cout << "remote_addr='" << remote_addr_ << "'" << std::endl;
 190   std::cout << "remote_port='" << remote_port_ << "'" << std::endl;
 191   std::cout << "ifconfig_param_local='" << ifconfig_param_local_ << "'" << std::endl;
 192   std::cout << "ifconfig_param_remote_netmask='" << ifconfig_param_remote_netmask_ << "'" << std::endl;
 193   std::cout << "seq_window_size='" << seq_window_size_ << "'" << std::endl;
 194   std::cout << "mux_id='" << mux_ << "'" << std::endl;
 195   std::cout << "network_prefix_length='" << network_prefix_length_ << "'" << std::endl;
 196   std::cout << "key=" << key_.getHexDumpOneLine() << std::endl;
 197   std::cout << "salt=" << salt_.getHexDumpOneLine() << std::endl;
 198   std::cout << "kd_prf='" << kd_prf_ << "'" << std::endl;
 199 }
 200
 201 std::string Options::getProgname()
 202 {
 203   Lock lock(mutex);
 204   return progname_;
 205 }
 206
 207
 208 Options& Options::setProgname(std::string p)
 209 {
 210   Lock lock(mutex);
 211   progname_ = p;
 212   return *this;
 213 }
 214
 215
 216 std::string Options::getRemoteAddr()
 217 {
 218   Lock lock(mutex);
 219   return remote_addr_;
 220 }
 221
 222 Options& Options::setRemoteAddr(std::string r)
 223 {
 224   Lock lock(mutex);
 225   remote_addr_ = r;
 226   return *this;
 227 }
 228
 229 u_int16_t Options::getRemotePort()
 230 {
 231   return remote_port_;
 232 }
 233
 234 Options& Options::setRemotePort(u_int16_t r)
 235 {
 236   remote_port_ = r;
 237   return *this;
 238 }
 239
 240 Options& Options::setRemoteAddrPort(std::string addr, u_int16_t port)
 241 {
 242   Lock lock(mutex);
 243   remote_addr_ = addr;
 244   remote_port_ = port;
 245   return *this;
 246 }
 247
 248 std::string Options::getIfconfigParamRemoteNetmask()
 249 {
 250   Lock lock(mutex);
 251   return ifconfig_param_remote_netmask_;
 252 }
 253
 254 Options& Options::setIfconfigParamRemoteNetmask(std::string i)
 255 {
 256   Lock lock(mutex);
 257   ifconfig_param_remote_netmask_ = i;
 258   return *this;
 259 }
 260
 261 window_size_t Options::getSeqWindowSize()
 262 {
 263   return seq_window_size_;
 264 }
 265
 266 Options& Options::setSeqWindowSize(window_size_t s)
 267 {
 268   seq_window_size_ = s;
 269   return *this;
 270 }
 271
 272
 273 std::string Options::getKdPrf()
 274 {
 275   Lock lock(mutex);
 276   return kd_prf_;
 277 }
 278
 279 Options& Options::setKdPrf(std::string k)
 280 {
 281   Lock lock(mutex);
 282   kd_prf_ = k;
 283   return *this;
 284 }
 285
 286 u_int16_t Options::getMux()
 287 {
 288   Lock lock(mutex);
 289   return mux_;
 290 }
 291
 292 Options& Options::setMux(u_int16_t m)
 293 {
 294   Lock lock(mutex);
 295   mux_ = m;
 296   return *this;
 297 }
 298
 299 u_int16_t Options::getNetworkPrefixLength()
 300 {
 301   Lock lock(mutex);
 302   return network_prefix_length_;
 303 }
 304
 305 Options& Options::setNetworkPrefixLength(u_int16_t l)
 306 {
 307   Lock lock(mutex);
 308   network_prefix_length_ = l;
 309   return *this;
 310 }
 311
 312 Buffer Options::getKey()
 313 {
 314   Lock lock(mutex);
 315   return key_;
 316 }
 317
 318 Options& Options::setKey(std::string k)
 319 {
 320   Lock lock(mutex);
 321   key_ = k;
 322   return *this;
 323 }
 324
 325 Buffer Options::getSalt()
 326 {
 327   Lock lock(mutex);
 328   return salt_;
 329 }
 330
 331 Options& Options::setSalt(std::string s)
 332 {
 333   Lock lock(mutex);
 334   salt_ = s;
 335   return *this;
 336 }