mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-18 16:17:50 +01:00

Bug: webrtc:13579 Change-Id: Ia33afa2a9ad12d1a586087d49f581a93fddb565d Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/262766 Reviewed-by: Harald Alvestrand <hta@webrtc.org> Commit-Queue: Ali Tofigh <alito@webrtc.org> Cr-Commit-Position: refs/heads/main@{#37381}
72 lines
2.5 KiB
C++
72 lines
2.5 KiB
C++
/*
|
|
* 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_BASE_STUN_SERVER_H_
|
|
#define P2P_BASE_STUN_SERVER_H_
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
#include <memory>
|
|
|
|
#include "absl/strings/string_view.h"
|
|
#include "api/transport/stun.h"
|
|
#include "rtc_base/async_packet_socket.h"
|
|
#include "rtc_base/async_udp_socket.h"
|
|
#include "rtc_base/socket_address.h"
|
|
#include "rtc_base/third_party/sigslot/sigslot.h"
|
|
|
|
namespace cricket {
|
|
|
|
const int STUN_SERVER_PORT = 3478;
|
|
|
|
class StunServer : public sigslot::has_slots<> {
|
|
public:
|
|
// Creates a STUN server, which will listen on the given socket.
|
|
explicit StunServer(rtc::AsyncUDPSocket* socket);
|
|
// Removes the STUN server from the socket and deletes the socket.
|
|
~StunServer() override;
|
|
|
|
protected:
|
|
// Slot for Socket.PacketRead:
|
|
void OnPacket(rtc::AsyncPacketSocket* socket,
|
|
const char* buf,
|
|
size_t size,
|
|
const rtc::SocketAddress& remote_addr,
|
|
const int64_t& packet_time_us);
|
|
|
|
// Handlers for the different types of STUN/TURN requests:
|
|
virtual void OnBindingRequest(StunMessage* msg,
|
|
const rtc::SocketAddress& addr);
|
|
void OnAllocateRequest(StunMessage* msg, const rtc::SocketAddress& addr);
|
|
void OnSharedSecretRequest(StunMessage* msg, const rtc::SocketAddress& addr);
|
|
void OnSendRequest(StunMessage* msg, const rtc::SocketAddress& addr);
|
|
|
|
// Sends an error response to the given message back to the user.
|
|
void SendErrorResponse(const StunMessage& msg,
|
|
const rtc::SocketAddress& addr,
|
|
int error_code,
|
|
absl::string_view error_desc);
|
|
|
|
// Sends the given message to the appropriate destination.
|
|
void SendResponse(const StunMessage& msg, const rtc::SocketAddress& addr);
|
|
|
|
// A helper method to compose a STUN binding response.
|
|
void GetStunBindResponse(StunMessage* message,
|
|
const rtc::SocketAddress& remote_addr,
|
|
StunMessage* response) const;
|
|
|
|
private:
|
|
std::unique_ptr<rtc::AsyncUDPSocket> socket_;
|
|
};
|
|
|
|
} // namespace cricket
|
|
|
|
#endif // P2P_BASE_STUN_SERVER_H_
|