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

Cover protocol stack RTMP. 3.0.63

Browse source
This commit is contained in:
winlin 2019-11-04 09:31:30 +08:00
parent a726a14b37
commit 9067786bd3
2 changed files with 64 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
trunk/src/kernel/srs_kernel_flv.cpp
View file

@ -284,6 +284,11 @@ int SrsSharedPtrMessage::count()
bool SrsSharedPtrMessage::check(int stream_id)
{
// Ignore error when message has no payload.
if (!ptr) {
return true;
}
// we donot use the complex basic header,
// ensure the basic header is 1bytes.
if (ptr->header.perfer_cid < 2) {

59
trunk/src/utest/srs_utest_protostack.cpp
View file

@ -183,5 +183,64 @@ VOID TEST(ProtoStackTest, SendZeroMessages)
SrsProtocol p(&io);
HELPER_EXPECT_SUCCESS(p.send_and_free_message(NULL, 0));
}
if (true) {
MockBufferIO io;
SrsProtocol p(&io);
SrsSharedPtrMessage* msg = new SrsSharedPtrMessage();
HELPER_EXPECT_SUCCESS(p.send_and_free_message(msg, 1));
}
if (true) {
MockBufferIO io;
SrsProtocol p(&io);
SrsSharedPtrMessage* msgs[1024];
for (int i = 0; i < 1024; i++) {
msgs[i] = new SrsSharedPtrMessage();
}
HELPER_EXPECT_SUCCESS(p.send_and_free_messages(msgs, 1024, 0));
}
}
VOID TEST(ProtoStackTest, HugeMessages)
{
srs_error_t err;
if (true) {
MockBufferIO io;
SrsProtocol p(&io);
SrsCommonMessage pkt;
pkt.header.initialize_audio(200, 1000, 1);
pkt.create_payload(256);
pkt.size = 256;
SrsSharedPtrMessage* msg = new SrsSharedPtrMessage();
msg->create(&pkt);
HELPER_EXPECT_SUCCESS(p.send_and_free_message(msg, 1));
EXPECT_EQ(269, io.out_buffer.length());
}
if (true) {
MockBufferIO io;
SrsProtocol p(&io);
SrsCommonMessage pkt;
pkt.header.initialize_audio(200, 1000, 1);
pkt.create_payload(256);
pkt.size = 256;
SrsSharedPtrMessage* msg = new SrsSharedPtrMessage();
msg->create(&pkt);
SrsAutoFree(SrsSharedPtrMessage, msg);
SrsSharedPtrMessage* msgs[1024];
for (int i = 0; i < 1024; i++) {
msgs[i] = msg->copy();
}
HELPER_EXPECT_SUCCESS(p.send_and_free_messages(msgs, 1024, 1));
EXPECT_EQ(269*1024, io.out_buffer.length());
}
}