86 lines
2.6 KiB
C++
86 lines
2.6 KiB
C++
/*
|
|
* Copyright 2012 The WebRTC Project Authors. All rights reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include <iostream> // NOLINT
|
|
|
|
#include "webrtc/p2p/base/basicpacketsocketfactory.h"
|
|
#include "webrtc/p2p/base/turnserver.h"
|
|
#include "webrtc/base/asyncudpsocket.h"
|
|
#include "webrtc/base/optionsfile.h"
|
|
#include "webrtc/base/stringencode.h"
|
|
#include "webrtc/base/thread.h"
|
|
|
|
static const char kSoftware[] = "libjingle TurnServer";
|
|
|
|
class TurnFileAuth : public cricket::TurnAuthInterface {
|
|
public:
|
|
explicit TurnFileAuth(const std::string& path) : file_(path) {
|
|
file_.Load();
|
|
}
|
|
virtual bool GetKey(const std::string& username, const std::string& realm,
|
|
std::string* key) {
|
|
// File is stored as lines of <username>=<HA1>.
|
|
// Generate HA1 via "echo -n "<username>:<realm>:<password>" | md5sum"
|
|
std::string hex;
|
|
bool ret = file_.GetStringValue(username, &hex);
|
|
if (ret) {
|
|
char buf[32];
|
|
size_t len = rtc::hex_decode(buf, sizeof(buf), hex);
|
|
*key = std::string(buf, len);
|
|
}
|
|
return ret;
|
|
}
|
|
private:
|
|
rtc::OptionsFile file_;
|
|
};
|
|
|
|
int main(int argc, char **argv) {
|
|
if (argc != 5) {
|
|
std::cerr << "usage: turnserver int-addr ext-ip realm auth-file"
|
|
<< std::endl;
|
|
return 1;
|
|
}
|
|
|
|
rtc::SocketAddress int_addr;
|
|
if (!int_addr.FromString(argv[1])) {
|
|
std::cerr << "Unable to parse IP address: " << argv[1] << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
rtc::IPAddress ext_addr;
|
|
if (!IPFromString(argv[2], &ext_addr)) {
|
|
std::cerr << "Unable to parse IP address: " << argv[2] << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
rtc::Thread* main = rtc::Thread::Current();
|
|
rtc::AsyncUDPSocket* int_socket =
|
|
rtc::AsyncUDPSocket::Create(main->socketserver(), int_addr);
|
|
if (!int_socket) {
|
|
std::cerr << "Failed to create a UDP socket bound at"
|
|
<< int_addr.ToString() << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
cricket::TurnServer server(main);
|
|
TurnFileAuth auth(argv[4]);
|
|
server.set_realm(argv[3]);
|
|
server.set_software(kSoftware);
|
|
server.set_auth_hook(&auth);
|
|
server.AddInternalSocket(int_socket, cricket::PROTO_UDP);
|
|
server.SetExternalSocketFactory(new rtc::BasicPacketSocketFactory(),
|
|
rtc::SocketAddress(ext_addr, 0));
|
|
|
|
std::cout << "Listening internally at " << int_addr.ToString() << std::endl;
|
|
|
|
main->Run();
|
|
return 0;
|
|
}
|