src/third_party/libjingle/source/talk/examples/turnserver/turnserver_main.cc

   1 /*
   2  * libjingle
   3  * Copyright 2012, Google Inc.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions are met:
   7  *
   8  *  1. Redistributions of source code must retain the above copyright notice,
   9  *     this list of conditions and the following disclaimer.
  10  *  2. Redistributions in binary form must reproduce the above copyright notice,
  11  *     this list of conditions and the following disclaimer in the documentation
  12  *     and/or other materials provided with the distribution.
  13  *  3. The name of the author may not be used to endorse or promote products
  14  *     derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
  17  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  18  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
  19  * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  20  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  22  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  23  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  24  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  25  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include <iostream>  // NOLINT
  29
  30 #include "webrtc/p2p/base/basicpacketsocketfactory.h"
  31 #include "webrtc/p2p/base/turnserver.h"
  32 #include "webrtc/base/asyncudpsocket.h"
  33 #include "webrtc/base/optionsfile.h"
  34 #include "webrtc/base/stringencode.h"
  35 #include "webrtc/base/thread.h"
  36
  37 static const char kSoftware[] = "libjingle TurnServer";
  38
  39 class TurnFileAuth : public cricket::TurnAuthInterface {
  40  public:
  41   explicit TurnFileAuth(const std::string& path) : file_(path) {
  42     file_.Load();
  43   }
  44   virtual bool GetKey(const std::string& username, const std::string& realm,
  45                       std::string* key) {
  46     // File is stored as lines of <username>=<HA1>.
  47     // Generate HA1 via "echo -n "<username>:<realm>:<password>" | md5sum"
  48     std::string hex;
  49     bool ret = file_.GetStringValue(username, &hex);
  50     if (ret) {
  51       char buf[32];
  52       size_t len = rtc::hex_decode(buf, sizeof(buf), hex);
  53       *key = std::string(buf, len);
  54     }
  55     return ret;
  56   }
  57  private:
  58   rtc::OptionsFile file_;
  59 };
  60
  61 int main(int argc, char **argv) {
  62   if (argc != 5) {
  63     std::cerr << "usage: turnserver int-addr ext-ip realm auth-file"
  64               << std::endl;
  65     return 1;
  66   }
  67
  68   rtc::SocketAddress int_addr;
  69   if (!int_addr.FromString(argv[1])) {
  70     std::cerr << "Unable to parse IP address: " << argv[1] << std::endl;
  71     return 1;
  72   }
  73
  74   rtc::IPAddress ext_addr;
  75   if (!IPFromString(argv[2], &ext_addr)) {
  76     std::cerr << "Unable to parse IP address: " << argv[2] << std::endl;
  77     return 1;
  78   }
  79
  80   rtc::Thread* main = rtc::Thread::Current();
  81   rtc::AsyncUDPSocket* int_socket =
  82       rtc::AsyncUDPSocket::Create(main->socketserver(), int_addr);
  83   if (!int_socket) {
  84     std::cerr << "Failed to create a UDP socket bound at"
  85               << int_addr.ToString() << std::endl;
  86     return 1;
  87   }
  88
  89   cricket::TurnServer server(main);
  90   TurnFileAuth auth(argv[4]);
  91   server.set_realm(argv[3]);
  92   server.set_software(kSoftware);
  93   server.set_auth_hook(&auth);
  94   server.AddInternalSocket(int_socket, cricket::PROTO_UDP);
  95   server.SetExternalSocketFactory(new rtc::BasicPacketSocketFactory(),
  96                                   rtc::SocketAddress(ext_addr, 0));
  97
  98   std::cout << "Listening internally at " << int_addr.ToString() << std::endl;
  99
 100   main->Run();
 101   return 0;
 102 }