stun_server.h - mozsearch

Enable keyboard shortcuts

/*

 *  Copyright 2004 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.

*/

#ifndef P2P_TEST_STUN_SERVER_H_

#define P2P_TEST_STUN_SERVER_H_

#include <stddef.h>

#include <memory>

#include "absl/strings/string_view.h"

#include "api/sequence_checker.h"

#include "api/transport/stun.h"

#include "rtc_base/async_packet_socket.h"

#include "rtc_base/async_udp_socket.h"

#include "rtc_base/network/received_packet.h"

#include "rtc_base/socket_address.h"

namespace webrtc {

const int STUN_SERVER_PORT = 3478;

class StunServer {

 public:

  // Creates a STUN server, which will listen on the given socket.

  explicit StunServer(AsyncUDPSocket* socket);

  // Removes the STUN server from the socket and deletes the socket.

  virtual ~StunServer();

 protected:

  // Callback for packets from socket.

  void OnPacket(AsyncPacketSocket* socket, const rtc::ReceivedPacket& packet);

  // Handlers for the different types of STUN/TURN requests:

  virtual void OnBindingRequest(cricket::StunMessage* msg,

                                const SocketAddress& addr);

  void OnAllocateRequest(cricket::StunMessage* msg, const SocketAddress& addr);

  void OnSharedSecretRequest(cricket::StunMessage* msg,

                             const SocketAddress& addr);

  void OnSendRequest(cricket::StunMessage* msg, const SocketAddress& addr);

  // Sends an error response to the given message back to the user.

  void SendErrorResponse(const cricket::StunMessage& msg,

                         const SocketAddress& addr,

                         int error_code,

                         absl::string_view error_desc);

  // Sends the given message to the appropriate destination.

  void SendResponse(const cricket::StunMessage& msg, const SocketAddress& addr);

  // A helper method to compose a STUN binding response.

  void GetStunBindResponse(cricket::StunMessage* message,

                           const SocketAddress& remote_addr,

                           cricket::StunMessage* response) const;

 private:

  SequenceChecker sequence_checker_;

  std::unique_ptr<AsyncUDPSocket> socket_;

};

}  //  namespace webrtc

// Re-export symbols from the webrtc namespace for backwards compatibility.

// TODO(bugs.webrtc.org/4222596): Remove once all references are updated.

namespace cricket {

using ::webrtc::STUN_SERVER_PORT;

using ::webrtc::StunServer;

}  // namespace cricket

#endif  // P2P_TEST_STUN_SERVER_H_