146 lines
3.6 KiB
C++
146 lines
3.6 KiB
C++
//
|
|
// Copyright (c) 2013-2025 The SRS Authors
|
|
//
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#ifndef SRS_UTEST_PROTOCOL_HPP
|
|
#define SRS_UTEST_PROTOCOL_HPP
|
|
|
|
/*
|
|
#include <srs_utest_protocol.hpp>
|
|
*/
|
|
#include <srs_utest.hpp>
|
|
|
|
#include <srs_protocol_utility.hpp>
|
|
#include <string>
|
|
|
|
#include <srs_kernel_kbps.hpp>
|
|
#include <srs_protocol_rtmp_handshake.hpp>
|
|
#include <srs_protocol_rtmp_stack.hpp>
|
|
#include <srs_protocol_stream.hpp>
|
|
|
|
using namespace srs_internal;
|
|
|
|
#include <srs_protocol_io.hpp>
|
|
|
|
class MockEmptyIO : public ISrsProtocolReadWriter
|
|
{
|
|
public:
|
|
MockEmptyIO();
|
|
virtual ~MockEmptyIO();
|
|
// for handshake.
|
|
public:
|
|
virtual srs_error_t read_fully(void *buf, size_t size, ssize_t *nread);
|
|
virtual srs_error_t write(void *buf, size_t size, ssize_t *nwrite);
|
|
// for protocol
|
|
public:
|
|
virtual void set_recv_timeout(srs_utime_t tm);
|
|
virtual srs_utime_t get_recv_timeout();
|
|
virtual int64_t get_recv_bytes();
|
|
// for protocol
|
|
public:
|
|
virtual void set_send_timeout(srs_utime_t tm);
|
|
virtual srs_utime_t get_send_timeout();
|
|
virtual int64_t get_send_bytes();
|
|
virtual srs_error_t writev(const iovec *iov, int iov_size, ssize_t *nwrite);
|
|
// for protocol/amf0/msg-codec
|
|
public:
|
|
virtual srs_error_t read(void *buf, size_t size, ssize_t *nread);
|
|
};
|
|
|
|
class MockBufferIO : public ISrsProtocolReadWriter
|
|
{
|
|
public:
|
|
// The send/recv timeout in srs_utime_t.
|
|
srs_utime_t rtm;
|
|
srs_utime_t stm;
|
|
// The send/recv data in bytes.
|
|
int64_t rbytes;
|
|
int64_t sbytes;
|
|
// data source for socket read.
|
|
SrsSimpleStream in_buffer;
|
|
// data buffer for socket send.
|
|
SrsSimpleStream out_buffer;
|
|
|
|
public:
|
|
// Mock error for io.
|
|
srs_error_t in_err;
|
|
srs_error_t out_err;
|
|
|
|
public:
|
|
MockBufferIO();
|
|
virtual ~MockBufferIO();
|
|
|
|
public:
|
|
virtual int length();
|
|
virtual MockBufferIO *append(std::string data);
|
|
virtual MockBufferIO *append(MockBufferIO *data);
|
|
virtual MockBufferIO *append(uint8_t *data, int size);
|
|
|
|
public:
|
|
virtual int out_length();
|
|
virtual MockBufferIO *out_append(std::string data);
|
|
virtual MockBufferIO *out_append(MockBufferIO *data);
|
|
virtual MockBufferIO *out_append(uint8_t *data, int size);
|
|
// for handshake.
|
|
public:
|
|
virtual srs_error_t read_fully(void *buf, size_t size, ssize_t *nread);
|
|
virtual srs_error_t write(void *buf, size_t size, ssize_t *nwrite);
|
|
// for protocol
|
|
public:
|
|
virtual void set_recv_timeout(srs_utime_t tm);
|
|
virtual srs_utime_t get_recv_timeout();
|
|
virtual int64_t get_recv_bytes();
|
|
// for protocol
|
|
public:
|
|
virtual void set_send_timeout(srs_utime_t tm);
|
|
virtual srs_utime_t get_send_timeout();
|
|
virtual int64_t get_send_bytes();
|
|
virtual srs_error_t writev(const iovec *iov, int iov_size, ssize_t *nwrite);
|
|
// for protocol/amf0/msg-codec
|
|
public:
|
|
virtual srs_error_t read(void *buf, size_t size, ssize_t *nread);
|
|
};
|
|
|
|
class MockStatistic : public ISrsProtocolStatistic
|
|
{
|
|
// clang-format off
|
|
SRS_DECLARE_PRIVATE: // clang-format on
|
|
int64_t in;
|
|
int64_t out;
|
|
|
|
public:
|
|
MockStatistic();
|
|
virtual ~MockStatistic();
|
|
|
|
public:
|
|
virtual int64_t get_recv_bytes();
|
|
virtual int64_t get_send_bytes();
|
|
|
|
public:
|
|
MockStatistic *set_in(int64_t v);
|
|
MockStatistic *set_out(int64_t v);
|
|
MockStatistic *add_in(int64_t v);
|
|
MockStatistic *add_out(int64_t v);
|
|
};
|
|
|
|
class MockWallClock : public SrsWallClock
|
|
{
|
|
// clang-format off
|
|
SRS_DECLARE_PRIVATE: // clang-format on
|
|
int64_t clock;
|
|
|
|
public:
|
|
MockWallClock();
|
|
virtual ~MockWallClock();
|
|
|
|
public:
|
|
virtual srs_utime_t now();
|
|
|
|
public:
|
|
virtual MockWallClock *set_clock(srs_utime_t v);
|
|
};
|
|
|
|
#endif
|