2020-08-14 16:58:22 +00:00
|
|
|
/*
|
|
|
|
* 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 RTC_BASE_ASYNC_UDP_SOCKET_H_
|
|
|
|
#define RTC_BASE_ASYNC_UDP_SOCKET_H_
|
|
|
|
|
|
|
|
#include <stddef.h>
|
|
|
|
|
|
|
|
#include <memory>
|
|
|
|
|
|
|
|
#include "rtc_base/async_packet_socket.h"
|
|
|
|
#include "rtc_base/socket.h"
|
|
|
|
#include "rtc_base/socket_address.h"
|
|
|
|
#include "rtc_base/socket_factory.h"
|
|
|
|
|
|
|
|
namespace rtc {
|
|
|
|
|
|
|
|
// Provides the ability to receive packets asynchronously. Sends are not
|
|
|
|
// buffered since it is acceptable to drop packets under high load.
|
|
|
|
class AsyncUDPSocket : public AsyncPacketSocket {
|
|
|
|
public:
|
2022-03-11 16:49:54 +00:00
|
|
|
// Binds `socket` and creates AsyncUDPSocket for it. Takes ownership
|
|
|
|
// of `socket`. Returns null if bind() fails (`socket` is destroyed
|
2020-08-14 16:58:22 +00:00
|
|
|
// in that case).
|
2022-03-11 16:49:54 +00:00
|
|
|
static AsyncUDPSocket* Create(Socket* socket,
|
2020-08-14 16:58:22 +00:00
|
|
|
const SocketAddress& bind_address);
|
|
|
|
// Creates a new socket for sending asynchronous UDP packets using an
|
|
|
|
// asynchronous socket from the given factory.
|
|
|
|
static AsyncUDPSocket* Create(SocketFactory* factory,
|
|
|
|
const SocketAddress& bind_address);
|
2022-03-11 16:49:54 +00:00
|
|
|
explicit AsyncUDPSocket(Socket* socket);
|
2020-08-14 16:58:22 +00:00
|
|
|
~AsyncUDPSocket() override;
|
|
|
|
|
|
|
|
SocketAddress GetLocalAddress() const override;
|
|
|
|
SocketAddress GetRemoteAddress() const override;
|
|
|
|
int Send(const void* pv,
|
|
|
|
size_t cb,
|
|
|
|
const rtc::PacketOptions& options) override;
|
|
|
|
int SendTo(const void* pv,
|
|
|
|
size_t cb,
|
|
|
|
const SocketAddress& addr,
|
|
|
|
const rtc::PacketOptions& options) override;
|
|
|
|
int Close() override;
|
|
|
|
|
|
|
|
State GetState() const override;
|
|
|
|
int GetOption(Socket::Option opt, int* value) override;
|
|
|
|
int SetOption(Socket::Option opt, int value) override;
|
|
|
|
int GetError() const override;
|
|
|
|
void SetError(int error) override;
|
|
|
|
|
|
|
|
private:
|
|
|
|
// Called when the underlying socket is ready to be read from.
|
2022-03-11 16:49:54 +00:00
|
|
|
void OnReadEvent(Socket* socket);
|
2020-08-14 16:58:22 +00:00
|
|
|
// Called when the underlying socket is ready to send.
|
2022-03-11 16:49:54 +00:00
|
|
|
void OnWriteEvent(Socket* socket);
|
2020-08-14 16:58:22 +00:00
|
|
|
|
2022-03-11 16:49:54 +00:00
|
|
|
std::unique_ptr<Socket> socket_;
|
2020-08-14 16:58:22 +00:00
|
|
|
char* buf_;
|
|
|
|
size_t size_;
|
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace rtc
|
|
|
|
|
|
|
|
#endif // RTC_BASE_ASYNC_UDP_SOCKET_H_
|