external/srs
1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-03-09 15:49:59 +00:00
Code Issues Releases Wiki Activity

refine protocol, add comments. add utest for protocol bytes and timeout.

Browse source
This commit is contained in:
winlin 2014-07-09 12:04:40 +08:00
parent 1e73ae4d47
commit 7027a125f0
3 changed files with 188 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

38
trunk/src/utest/srs_utest_protocol.hpp
View file

@ -34,6 +34,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#include <srs_protocol_rtmp.hpp>
#include <srs_protocol_handshake.hpp>
#include <srs_kernel_buffer.hpp>
#ifdef SRS_AUTO_SSL
using namespace _srs_internal;
@ -69,4 +70,41 @@ public:
virtual int read(void* buf, size_t size, ssize_t* nread);
};
class MockBufferIO : public ISrsProtocolReaderWriter
{
public:
int64_t recv_timeout;
int64_t send_timeout;
int64_t recv_bytes;
int64_t send_bytes;
// data source for socket read.
SrsBuffer in_buffer;
// data buffer for socket send.
SrsBuffer out_buffer;
public:
MockBufferIO();
virtual ~MockBufferIO();
// for protocol
public:
virtual bool is_never_timeout(int64_t timeout_us);
// for handshake.
public:
virtual int read_fully(void* buf, size_t size, ssize_t* nread);
virtual int write(void* buf, size_t size, ssize_t* nwrite);
// for protocol
public:
virtual void set_recv_timeout(int64_t timeout_us);
virtual int64_t get_recv_timeout();
virtual int64_t get_recv_bytes();
// for protocol
public:
virtual void set_send_timeout(int64_t timeout_us);
virtual int64_t get_send_timeout();
virtual int64_t get_send_bytes();
virtual int writev(const iovec *iov, int iov_size, ssize_t* nwrite);
// for protocol/amf0/msg-codec
public:
virtual int read(void* buf, size_t size, ssize_t* nread);
};
#endif