mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-12 21:30:45 +01:00
14630a7e37
StunServer is updated to ensure registring for receiving packet from the socket is happening on the same thread as where the packets are recevied. Bug: webrtc:15368, webrtc:11943 Change-Id: I94cc3a47278d5489de7f170c8d43015d1551c437 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/328120 Commit-Queue: Per Kjellander <perkj@webrtc.org> Reviewed-by: Jonas Oreland <jonaso@webrtc.org> Reviewed-by: Harald Alvestrand <hta@webrtc.org> Cr-Commit-Position: refs/heads/main@{#41219}
68 lines
2.3 KiB
C++
68 lines
2.3 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 <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"
|
|
|
|
namespace cricket {
|
|
|
|
const int STUN_SERVER_PORT = 3478;
|
|
|
|
class StunServer {
|
|
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.
|
|
virtual ~StunServer();
|
|
|
|
protected:
|
|
// Callback for packets from socket.
|
|
void OnPacket(rtc::AsyncPacketSocket* socket,
|
|
const rtc::ReceivedPacket& packet);
|
|
|
|
// 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:
|
|
webrtc::SequenceChecker sequence_checker_;
|
|
std::unique_ptr<rtc::AsyncUDPSocket> socket_;
|
|
};
|
|
|
|
} // namespace cricket
|
|
|
|
#endif // P2P_BASE_STUN_SERVER_H_
|