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 #ifndef TALK_P2P_BASE_TESTTURNSERVER_H_ 29 #define TALK_P2P_BASE_TESTTURNSERVER_H_ 30 31 #include <string> 32 33 #include "talk/base/asyncudpsocket.h" 34 #include "talk/base/thread.h" 35 #include "talk/p2p/base/basicpacketsocketfactory.h" 36 #include "talk/p2p/base/stun.h" 37 #include "talk/p2p/base/turnserver.h" 38 39 namespace cricket { 40 41 static const char kTestRealm[] = "example.org"; 42 static const char kTestSoftware[] = "TestTurnServer"; 43 44 class TestTurnServer : public TurnAuthInterface { 45 public: TestTurnServer(talk_base::Thread * thread,const talk_base::SocketAddress & udp_int_addr,const talk_base::SocketAddress & udp_ext_addr)46 TestTurnServer(talk_base::Thread* thread, 47 const talk_base::SocketAddress& udp_int_addr, 48 const talk_base::SocketAddress& udp_ext_addr) 49 : server_(thread) { 50 AddInternalSocket(udp_int_addr, cricket::PROTO_UDP); 51 server_.SetExternalSocketFactory(new talk_base::BasicPacketSocketFactory(), 52 udp_ext_addr); 53 server_.set_realm(kTestRealm); 54 server_.set_software(kTestSoftware); 55 server_.set_auth_hook(this); 56 } 57 set_enable_otu_nonce(bool enable)58 void set_enable_otu_nonce(bool enable) { 59 server_.set_enable_otu_nonce(enable); 60 } 61 server()62 TurnServer* server() { return &server_; } 63 AddInternalSocket(const talk_base::SocketAddress & int_addr,ProtocolType proto)64 void AddInternalSocket(const talk_base::SocketAddress& int_addr, 65 ProtocolType proto) { 66 talk_base::Thread* thread = talk_base::Thread::Current(); 67 if (proto == cricket::PROTO_UDP) { 68 server_.AddInternalSocket(talk_base::AsyncUDPSocket::Create( 69 thread->socketserver(), int_addr), proto); 70 } else if (proto == cricket::PROTO_TCP) { 71 // For TCP we need to create a server socket which can listen for incoming 72 // new connections. 73 talk_base::AsyncSocket* socket = 74 thread->socketserver()->CreateAsyncSocket(SOCK_STREAM); 75 socket->Bind(int_addr); 76 socket->Listen(5); 77 server_.AddInternalServerSocket(socket, proto); 78 } 79 } 80 81 private: 82 // For this test server, succeed if the password is the same as the username. 83 // Obviously, do not use this in a production environment. GetKey(const std::string & username,const std::string & realm,std::string * key)84 virtual bool GetKey(const std::string& username, const std::string& realm, 85 std::string* key) { 86 return ComputeStunCredentialHash(username, realm, username, key); 87 } 88 89 TurnServer server_; 90 }; 91 92 } // namespace cricket 93 94 #endif // TALK_P2P_BASE_TESTTURNSERVER_H_ 95