1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-02-15 04:42:04 +00:00
srs/trunk/src/kernel/srs_kernel_flv.cpp

942 lines
25 KiB
C++
Raw Normal View History

2017-03-25 09:21:39 +00:00
/**
* The MIT License (MIT)
*
2019-12-30 02:10:35 +00:00
* Copyright (c) 2013-2020 Winlin
2017-03-25 09:21:39 +00:00
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
* the Software, and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#include <srs_kernel_flv.hpp>
2015-11-11 02:37:50 +00:00
// for srs-librtmp, @see https://github.com/ossrs/srs/issues/213
#ifndef _WIN32
2014-06-14 08:24:15 +00:00
#include <unistd.h>
#endif
#include <fcntl.h>
#include <sstream>
using namespace std;
#include <srs_kernel_log.hpp>
#include <srs_kernel_error.hpp>
2015-09-22 00:48:55 +00:00
#include <srs_kernel_buffer.hpp>
#include <srs_kernel_file.hpp>
#include <srs_kernel_codec.hpp>
#include <srs_kernel_utility.hpp>
#include <srs_core_autofree.hpp>
2021-02-26 13:35:51 +00:00
#include <srs_kernel_rtc_rtp.hpp>
2021-02-26 12:38:50 +00:00
#include <srs_kernel_kbps.hpp>
SrsPps* _srs_pps_objs_msgs = new SrsPps();
SrsMessageHeader::SrsMessageHeader()
{
message_type = 0;
payload_length = 0;
timestamp_delta = 0;
stream_id = 0;
timestamp = 0;
// we always use the connection chunk-id
perfer_cid = RTMP_CID_OverConnection;
}
SrsMessageHeader::~SrsMessageHeader()
{
}
bool SrsMessageHeader::is_audio()
{
return message_type == RTMP_MSG_AudioMessage;
}
bool SrsMessageHeader::is_video()
{
return message_type == RTMP_MSG_VideoMessage;
}
bool SrsMessageHeader::is_amf0_command()
{
return message_type == RTMP_MSG_AMF0CommandMessage;
}
bool SrsMessageHeader::is_amf0_data()
{
return message_type == RTMP_MSG_AMF0DataMessage;
}
bool SrsMessageHeader::is_amf3_command()
{
return message_type == RTMP_MSG_AMF3CommandMessage;
}
bool SrsMessageHeader::is_amf3_data()
{
return message_type == RTMP_MSG_AMF3DataMessage;
}
bool SrsMessageHeader::is_window_ackledgement_size()
{
return message_type == RTMP_MSG_WindowAcknowledgementSize;
}
bool SrsMessageHeader::is_ackledgement()
{
return message_type == RTMP_MSG_Acknowledgement;
}
bool SrsMessageHeader::is_set_chunk_size()
{
return message_type == RTMP_MSG_SetChunkSize;
}
bool SrsMessageHeader::is_user_control_message()
{
return message_type == RTMP_MSG_UserControlMessage;
}
bool SrsMessageHeader::is_set_peer_bandwidth()
{
return message_type == RTMP_MSG_SetPeerBandwidth;
}
bool SrsMessageHeader::is_aggregate()
{
return message_type == RTMP_MSG_AggregateMessage;
}
void SrsMessageHeader::initialize_amf0_script(int size, int stream)
{
message_type = RTMP_MSG_AMF0DataMessage;
payload_length = (int32_t)size;
timestamp_delta = (int32_t)0;
timestamp = (int64_t)0;
stream_id = (int32_t)stream;
// amf0 script use connection2 chunk-id
perfer_cid = RTMP_CID_OverConnection2;
}
void SrsMessageHeader::initialize_audio(int size, uint32_t time, int stream)
{
message_type = RTMP_MSG_AudioMessage;
payload_length = (int32_t)size;
timestamp_delta = (int32_t)time;
timestamp = (int64_t)time;
stream_id = (int32_t)stream;
// audio chunk-id
perfer_cid = RTMP_CID_Audio;
}
void SrsMessageHeader::initialize_video(int size, uint32_t time, int stream)
{
message_type = RTMP_MSG_VideoMessage;
payload_length = (int32_t)size;
timestamp_delta = (int32_t)time;
timestamp = (int64_t)time;
stream_id = (int32_t)stream;
// video chunk-id
perfer_cid = RTMP_CID_Video;
}
SrsCommonMessage::SrsCommonMessage()
{
payload = NULL;
size = 0;
}
SrsCommonMessage::~SrsCommonMessage()
{
2015-11-02 03:05:39 +00:00
srs_freepa(payload);
}
void SrsCommonMessage::create_payload(int size)
{
2015-11-02 03:05:39 +00:00
srs_freepa(payload);
payload = new char[size];
srs_verbose("create payload for RTMP message. size=%d", size);
}
srs_error_t SrsCommonMessage::create(SrsMessageHeader* pheader, char* body, int size)
2015-12-29 10:33:02 +00:00
{
// drop previous payload.
srs_freepa(payload);
this->header = *pheader;
this->payload = body;
this->size = size;
return srs_success;
2015-12-29 10:33:02 +00:00
}
SrsSharedMessageHeader::SrsSharedMessageHeader()
2017-06-13 08:10:46 +00:00
{
payload_length = 0;
message_type = 0;
perfer_cid = 0;
2017-06-13 08:10:46 +00:00
}
SrsSharedMessageHeader::~SrsSharedMessageHeader()
{
}
SrsSharedPtrMessage::SrsSharedPtrPayload::SrsSharedPtrPayload()
{
payload = NULL;
size = 0;
shared_count = 0;
}
SrsSharedPtrMessage::SrsSharedPtrPayload::~SrsSharedPtrPayload()
{
2015-11-02 03:05:39 +00:00
srs_freepa(payload);
}
2020-03-13 12:34:40 +00:00
SrsSharedPtrMessage::SrsSharedPtrMessage() : timestamp(0), stream_id(0), size(0), payload(NULL)
{
ptr = NULL;
2021-02-26 12:38:50 +00:00
++ _srs_pps_objs_msgs->sugar;
}
SrsSharedPtrMessage::~SrsSharedPtrMessage()
{
if (ptr) {
if (ptr->shared_count == 0) {
srs_freep(ptr);
} else {
ptr->shared_count--;
}
}
}
srs_error_t SrsSharedPtrMessage::create(SrsCommonMessage* msg)
{
srs_error_t err = srs_success;
if ((err = create(&msg->header, msg->payload, msg->size)) != srs_success) {
return srs_error_wrap(err, "create message");
}
// to prevent double free of payload:
// initialize already attach the payload of msg,
// detach the payload to transfer the owner to shared ptr.
msg->payload = NULL;
msg->size = 0;
return err;
}
srs_error_t SrsSharedPtrMessage::create(SrsMessageHeader* pheader, char* payload, int size)
{
srs_error_t err = srs_success;
2019-05-20 00:45:42 +00:00
if (size < 0) {
return srs_error_new(ERROR_RTMP_MESSAGE_CREATE, "create message size=%d", size);
}
2019-05-20 00:45:42 +00:00
srs_assert(!ptr);
ptr = new SrsSharedPtrPayload();
// direct attach the data.
if (pheader) {
ptr->header.message_type = pheader->message_type;
ptr->header.payload_length = size;
ptr->header.perfer_cid = pheader->perfer_cid;
this->timestamp = pheader->timestamp;
this->stream_id = pheader->stream_id;
}
ptr->payload = payload;
ptr->size = size;
// message can access it.
this->payload = ptr->payload;
this->size = ptr->size;
return err;
}
2020-05-14 06:26:19 +00:00
void SrsSharedPtrMessage::wrap(char* payload, int size)
{
srs_assert(!ptr);
ptr = new SrsSharedPtrPayload();
ptr->payload = payload;
ptr->size = size;
this->payload = ptr->payload;
this->size = ptr->size;
}
2021-02-26 13:35:51 +00:00
void SrsSharedPtrMessage::unwrap()
{
if (ptr) {
2021-02-27 10:39:09 +00:00
if (ptr->shared_count > 0) {
2021-02-26 13:35:51 +00:00
ptr->shared_count--;
ptr = NULL;
2021-02-27 10:39:09 +00:00
} else {
srs_freep(ptr);
2021-02-26 13:35:51 +00:00
}
}
payload = NULL;
size = 0;
}
int SrsSharedPtrMessage::count()
{
2021-02-26 13:35:51 +00:00
return ptr? ptr->shared_count : 0;
}
bool SrsSharedPtrMessage::check(int stream_id)
{
2019-11-04 01:31:30 +00:00
// 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 || ptr->header.perfer_cid > 63) {
srs_info("change the chunk_id=%d to default=%d", ptr->header.perfer_cid, RTMP_CID_ProtocolControl);
ptr->header.perfer_cid = RTMP_CID_ProtocolControl;
}
// we assume that the stream_id in a group must be the same.
if (this->stream_id == stream_id) {
return true;
}
this->stream_id = stream_id;
return false;
}
bool SrsSharedPtrMessage::is_av()
{
return ptr->header.message_type == RTMP_MSG_AudioMessage
2017-03-25 09:21:39 +00:00
|| ptr->header.message_type == RTMP_MSG_VideoMessage;
}
bool SrsSharedPtrMessage::is_audio()
{
return ptr->header.message_type == RTMP_MSG_AudioMessage;
}
bool SrsSharedPtrMessage::is_video()
{
return ptr->header.message_type == RTMP_MSG_VideoMessage;
}
int SrsSharedPtrMessage::chunk_header(char* cache, int nb_cache, bool c0)
{
if (c0) {
return srs_chunk_header_c0(ptr->header.perfer_cid, (uint32_t)timestamp,
ptr->header.payload_length, ptr->header.message_type, stream_id, cache, nb_cache);
} else {
return srs_chunk_header_c3(ptr->header.perfer_cid, (uint32_t)timestamp,
cache, nb_cache);
}
}
SrsSharedPtrMessage* SrsSharedPtrMessage::copy()
{
srs_assert(ptr);
SrsSharedPtrMessage* copy = _srs_rtp_msg_cache_objs->allocate();
2021-02-26 14:55:27 +00:00
// We got an object from cache, the ptr might exists, so unwrap it.
copy->unwrap();
// Reference to this message instead.
copy->ptr = ptr;
ptr->shared_count++;
copy->timestamp = timestamp;
copy->stream_id = stream_id;
copy->payload = ptr->payload;
copy->size = ptr->size;
2020-03-13 12:34:40 +00:00
return copy;
}
SrsFlvTransmuxer::SrsFlvTransmuxer()
{
writer = NULL;
nb_tag_headers = 0;
tag_headers = NULL;
nb_iovss_cache = 0;
iovss_cache = NULL;
nb_ppts = 0;
ppts = NULL;
}
SrsFlvTransmuxer::~SrsFlvTransmuxer()
{
2015-11-02 03:05:39 +00:00
srs_freepa(tag_headers);
srs_freepa(iovss_cache);
srs_freepa(ppts);
}
srs_error_t SrsFlvTransmuxer::initialize(ISrsWriter* fw)
{
srs_assert(fw);
writer = fw;
return srs_success;
}
srs_error_t SrsFlvTransmuxer::write_header(bool has_video, bool has_audio)
{
srs_error_t err = srs_success;
uint8_t av_flag = 0;
av_flag += (has_audio? 4:0);
av_flag += (has_video? 1:0);
// 9bytes header and 4bytes first previous-tag-size
char flv_header[] = {
'F', 'L', 'V', // Signatures "FLV"
(char)0x01, // File version (for example, 0x01 for FLV version 1)
(char)av_flag, // 4, audio; 1, video; 5 audio+video.
2014-05-28 11:01:47 +00:00
(char)0x00, (char)0x00, (char)0x00, (char)0x09 // DataOffset UI32 The length of this header in bytes
};
// flv specification should set the audio and video flag,
// actually in practise, application generally ignore this flag,
// so we generally set the audio/video to 0.
2014-05-28 11:01:47 +00:00
// write 9bytes header.
if ((err = write_header(flv_header)) != srs_success) {
return srs_error_wrap(err, "write header");
2014-05-28 11:01:47 +00:00
}
return err;
2014-05-28 11:01:47 +00:00
}
srs_error_t SrsFlvTransmuxer::write_header(char flv_header[9])
2014-05-28 11:01:47 +00:00
{
srs_error_t err = srs_success;
2014-05-28 11:01:47 +00:00
// write data.
if ((err = writer->write(flv_header, 9, NULL)) != srs_success) {
return srs_error_wrap(err, "write flv header failed");
}
// previous tag size.
char pts[] = { (char)0x00, (char)0x00, (char)0x00, (char)0x00 };
if ((err = writer->write(pts, 4, NULL)) != srs_success) {
return srs_error_wrap(err, "write pts");
2014-05-28 11:01:47 +00:00
}
return err;
}
srs_error_t SrsFlvTransmuxer::write_metadata(char type, char* data, int size)
{
srs_error_t err = srs_success;
2019-05-21 00:20:04 +00:00
if (size > 0) {
cache_metadata(type, data, size, tag_header);
}
if ((err = write_tag(tag_header, sizeof(tag_header), data, size)) != srs_success) {
return srs_error_wrap(err, "write tag");
}
return err;
}
srs_error_t SrsFlvTransmuxer::write_audio(int64_t timestamp, char* data, int size)
{
srs_error_t err = srs_success;
2019-05-21 00:20:04 +00:00
if (size > 0) {
cache_audio(timestamp, data, size, tag_header);
}
if ((err = write_tag(tag_header, sizeof(tag_header), data, size)) != srs_success) {
return srs_error_wrap(err, "write tag");
}
return err;
}
srs_error_t SrsFlvTransmuxer::write_video(int64_t timestamp, char* data, int size)
{
srs_error_t err = srs_success;
2019-05-21 00:20:04 +00:00
if (size > 0) {
cache_video(timestamp, data, size, tag_header);
}
if ((err = write_tag(tag_header, sizeof(tag_header), data, size)) != srs_success) {
return srs_error_wrap(err, "write flv video tag failed");
}
return err;
}
int SrsFlvTransmuxer::size_tag(int data_size)
{
srs_assert(data_size >= 0);
return SRS_FLV_TAG_HEADER_SIZE + data_size + SRS_FLV_PREVIOUS_TAG_SIZE;
}
srs_error_t SrsFlvTransmuxer::write_tags(SrsSharedPtrMessage** msgs, int count)
{
srs_error_t err = srs_success;
// realloc the iovss.
int nb_iovss = 3 * count;
iovec* iovss = iovss_cache;
if (nb_iovss_cache < nb_iovss) {
2015-11-02 03:05:39 +00:00
srs_freepa(iovss_cache);
nb_iovss_cache = nb_iovss;
iovss = iovss_cache = new iovec[nb_iovss];
}
// realloc the tag headers.
char* cache = tag_headers;
if (nb_tag_headers < count) {
2015-11-02 03:05:39 +00:00
srs_freepa(tag_headers);
nb_tag_headers = count;
cache = tag_headers = new char[SRS_FLV_TAG_HEADER_SIZE * count];
}
// realloc the pts.
char* pts = ppts;
if (nb_ppts < count) {
2015-11-02 03:05:39 +00:00
srs_freepa(ppts);
nb_ppts = count;
pts = ppts = new char[SRS_FLV_PREVIOUS_TAG_SIZE * count];
}
// the cache is ok, write each messages.
iovec* iovs = iovss;
for (int i = 0; i < count; i++) {
SrsSharedPtrMessage* msg = msgs[i];
// cache all flv header.
if (msg->is_audio()) {
2019-05-21 00:20:04 +00:00
cache_audio(msg->timestamp, msg->payload, msg->size, cache);
} else if (msg->is_video()) {
2019-05-21 00:20:04 +00:00
cache_video(msg->timestamp, msg->payload, msg->size, cache);
} else {
2019-05-21 00:20:04 +00:00
cache_metadata(SrsFrameTypeScript, msg->payload, msg->size, cache);
}
// cache all pts.
2019-05-21 00:20:04 +00:00
cache_pts(SRS_FLV_TAG_HEADER_SIZE + msg->size, pts);
// all ioves.
iovs[0].iov_base = cache;
iovs[0].iov_len = SRS_FLV_TAG_HEADER_SIZE;
iovs[1].iov_base = msg->payload;
iovs[1].iov_len = msg->size;
iovs[2].iov_base = pts;
iovs[2].iov_len = SRS_FLV_PREVIOUS_TAG_SIZE;
// move next.
cache += SRS_FLV_TAG_HEADER_SIZE;
pts += SRS_FLV_PREVIOUS_TAG_SIZE;
iovs += 3;
}
if ((err = writer->writev(iovss, nb_iovss, NULL)) != srs_success) {
return srs_error_wrap(err, "write flv tags failed");
}
return err;
}
2019-05-21 00:20:04 +00:00
void SrsFlvTransmuxer::cache_metadata(char type, char* data, int size, char* cache)
{
srs_assert(data);
// 11 bytes tag header
/*char tag_header[] = {
(char)type, // TagType UB [5], 18 = script data
(char)0x00, (char)0x00, (char)0x00, // DataSize UI24 Length of the message.
(char)0x00, (char)0x00, (char)0x00, // Timestamp UI24 Time in milliseconds at which the data in this tag applies.
(char)0x00, // TimestampExtended UI8
(char)0x00, (char)0x00, (char)0x00, // StreamID UI24 Always 0.
};*/
SrsBuffer* tag_stream = new SrsBuffer(cache, 11);
SrsAutoFree(SrsBuffer, tag_stream);
// write data size.
tag_stream->write_1bytes(type);
tag_stream->write_3bytes(size);
tag_stream->write_3bytes(0x00);
tag_stream->write_1bytes(0x00);
tag_stream->write_3bytes(0x00);
}
2019-05-21 00:20:04 +00:00
void SrsFlvTransmuxer::cache_audio(int64_t timestamp, char* data, int size, char* cache)
{
srs_assert(data);
timestamp &= 0x7fffffff;
// 11bytes tag header
2015-05-24 15:16:56 +00:00
/*char tag_header[] = {
2017-02-12 12:38:39 +00:00
(char)SrsFrameTypeAudio, // TagType UB [5], 8 = audio
(char)0x00, (char)0x00, (char)0x00, // DataSize UI24 Length of the message.
(char)0x00, (char)0x00, (char)0x00, // Timestamp UI24 Time in milliseconds at which the data in this tag applies.
(char)0x00, // TimestampExtended UI8
(char)0x00, (char)0x00, (char)0x00, // StreamID UI24 Always 0.
};*/
SrsBuffer* tag_stream = new SrsBuffer(cache, 11);
SrsAutoFree(SrsBuffer, tag_stream);
// write data size.
2017-02-12 12:38:39 +00:00
tag_stream->write_1bytes(SrsFrameTypeAudio);
tag_stream->write_3bytes(size);
tag_stream->write_3bytes((int32_t)timestamp);
// default to little-endian
tag_stream->write_1bytes((timestamp >> 24) & 0xFF);
tag_stream->write_3bytes(0x00);
}
2019-05-21 00:20:04 +00:00
void SrsFlvTransmuxer::cache_video(int64_t timestamp, char* data, int size, char* cache)
{
2014-07-05 09:30:13 +00:00
srs_assert(data);
timestamp &= 0x7fffffff;
// 11bytes tag header
2015-05-24 15:16:56 +00:00
/*char tag_header[] = {
2017-02-12 12:38:39 +00:00
(char)SrsFrameTypeVideo, // TagType UB [5], 9 = video
(char)0x00, (char)0x00, (char)0x00, // DataSize UI24 Length of the message.
(char)0x00, (char)0x00, (char)0x00, // Timestamp UI24 Time in milliseconds at which the data in this tag applies.
(char)0x00, // TimestampExtended UI8
(char)0x00, (char)0x00, (char)0x00, // StreamID UI24 Always 0.
};*/
SrsBuffer* tag_stream = new SrsBuffer(cache, 11);
SrsAutoFree(SrsBuffer, tag_stream);
// write data size.
2017-02-12 12:38:39 +00:00
tag_stream->write_1bytes(SrsFrameTypeVideo);
tag_stream->write_3bytes(size);
tag_stream->write_3bytes((int32_t)timestamp);
// default to little-endian
tag_stream->write_1bytes((timestamp >> 24) & 0xFF);
tag_stream->write_3bytes(0x00);
}
2019-05-21 00:20:04 +00:00
void SrsFlvTransmuxer::cache_pts(int size, char* cache)
{
SrsBuffer* tag_stream = new SrsBuffer(cache, 11);
SrsAutoFree(SrsBuffer, tag_stream);
tag_stream->write_4bytes(size);
}
srs_error_t SrsFlvTransmuxer::write_tag(char* header, int header_size, char* tag, int tag_size)
{
srs_error_t err = srs_success;
// PreviousTagSizeN UI32 Size of last tag, including its header, in bytes.
char pre_size[SRS_FLV_PREVIOUS_TAG_SIZE];
2019-05-21 00:20:04 +00:00
cache_pts(tag_size + header_size, pre_size);
iovec iovs[3];
iovs[0].iov_base = header;
iovs[0].iov_len = header_size;
iovs[1].iov_base = tag;
iovs[1].iov_len = tag_size;
iovs[2].iov_base = pre_size;
iovs[2].iov_len = SRS_FLV_PREVIOUS_TAG_SIZE;
if ((err = writer->writev(iovs, 3, NULL)) != srs_success) {
return srs_error_wrap(err, "write flv tag failed");
}
return err;
}
SrsFlvDecoder::SrsFlvDecoder()
{
2015-07-28 09:56:50 +00:00
reader = NULL;
}
SrsFlvDecoder::~SrsFlvDecoder()
{
}
srs_error_t SrsFlvDecoder::initialize(ISrsReader* fr)
{
2015-07-28 09:56:50 +00:00
srs_assert(fr);
reader = fr;
return srs_success;
}
srs_error_t SrsFlvDecoder::read_header(char header[9])
{
srs_error_t err = srs_success;
2017-03-25 09:21:39 +00:00
2014-07-05 09:30:13 +00:00
srs_assert(header);
2017-02-01 13:57:32 +00:00
// TODO: FIXME: Should use readfully.
if ((err = reader->read(header, 9, NULL)) != srs_success) {
return srs_error_wrap(err, "read header");
}
char* h = header;
if (h[0] != 'F' || h[1] != 'L' || h[2] != 'V') {
return srs_error_new(ERROR_KERNEL_FLV_HEADER, "flv header must start with FLV");
}
return err;
}
srs_error_t SrsFlvDecoder::read_tag_header(char* ptype, int32_t* pdata_size, uint32_t* ptime)
{
srs_error_t err = srs_success;
2017-03-25 09:21:39 +00:00
2014-07-05 09:30:13 +00:00
srs_assert(ptype);
srs_assert(pdata_size);
srs_assert(ptime);
2017-03-25 09:21:39 +00:00
char th[11]; // tag header
// read tag header
2017-02-01 13:57:32 +00:00
// TODO: FIXME: Should use readfully.
if ((err = reader->read(th, 11, NULL)) != srs_success) {
return srs_error_wrap(err, "read flv tag header failed");
}
// Reserved UB [2]
// Filter UB [1]
// TagType UB [5]
*ptype = (th[0] & 0x1F);
// DataSize UI24
char* pp = (char*)pdata_size;
pp[3] = 0;
pp[2] = th[1];
pp[1] = th[2];
pp[0] = th[3];
// Timestamp UI24
pp = (char*)ptime;
pp[2] = th[4];
pp[1] = th[5];
pp[0] = th[6];
// TimestampExtended UI8
pp[3] = th[7];
2017-03-25 09:21:39 +00:00
return err;
}
srs_error_t SrsFlvDecoder::read_tag_data(char* data, int32_t size)
{
srs_error_t err = srs_success;
2017-03-25 09:21:39 +00:00
2014-07-05 09:30:13 +00:00
srs_assert(data);
2017-02-01 13:57:32 +00:00
// TODO: FIXME: Should use readfully.
if ((err = reader->read(data, size, NULL)) != srs_success) {
return srs_error_wrap(err, "read flv tag header failed");
}
return err;
2017-03-25 09:21:39 +00:00
}
srs_error_t SrsFlvDecoder::read_previous_tag_size(char previous_tag_size[4])
{
srs_error_t err = srs_success;
2017-03-25 09:21:39 +00:00
2014-07-05 09:30:13 +00:00
srs_assert(previous_tag_size);
// ignore 4bytes tag size.
2017-02-01 13:57:32 +00:00
// TODO: FIXME: Should use readfully.
if ((err = reader->read(previous_tag_size, 4, NULL)) != srs_success) {
return srs_error_wrap(err, "read flv previous tag size failed");
}
return err;
}
SrsFlvVodStreamDecoder::SrsFlvVodStreamDecoder()
{
2015-07-28 09:56:50 +00:00
reader = NULL;
}
SrsFlvVodStreamDecoder::~SrsFlvVodStreamDecoder()
{
}
srs_error_t SrsFlvVodStreamDecoder::initialize(ISrsReader* fr)
{
srs_error_t err = srs_success;
2015-07-28 09:56:50 +00:00
srs_assert(fr);
reader = dynamic_cast<SrsFileReader*>(fr);
if (!reader) {
return srs_error_new(ERROR_EXPECT_FILE_IO, "stream is not file io");
}
2014-07-05 09:30:13 +00:00
if (!reader->is_open()) {
return srs_error_new(ERROR_KERNEL_FLV_STREAM_CLOSED, "stream is not open for decoder");
}
return err;
}
srs_error_t SrsFlvVodStreamDecoder::read_header_ext(char header[13])
{
srs_error_t err = srs_success;
2017-03-25 09:21:39 +00:00
2014-07-05 09:30:13 +00:00
srs_assert(header);
// @remark, always false, for sizeof(char[13]) equals to sizeof(char*)
//srs_assert(13 == sizeof(header));
// 9bytes header and 4bytes first previous-tag-size
int size = 13;
if ((err = reader->read(header, size, NULL)) != srs_success) {
return srs_error_wrap(err, "read header");
}
return err;
}
srs_error_t SrsFlvVodStreamDecoder::read_sequence_header_summary(int64_t* pstart, int* psize)
{
srs_error_t err = srs_success;
2017-03-25 09:21:39 +00:00
2014-07-05 09:30:13 +00:00
srs_assert(pstart);
srs_assert(psize);
// simply, the first video/audio must be the sequence header.
// and must be a sequence video and audio.
// 11bytes tag header
char tag_header[] = {
(char)0x00, // TagType UB [5], 9 = video, 8 = audio, 18 = script data
(char)0x00, (char)0x00, (char)0x00, // DataSize UI24 Length of the message.
(char)0x00, (char)0x00, (char)0x00, // Timestamp UI24 Time in milliseconds at which the data in this tag applies.
(char)0x00, // TimestampExtended UI8
(char)0x00, (char)0x00, (char)0x00, // StreamID UI24 Always 0.
};
// discovery the sequence header video and audio.
// @remark, maybe no video or no audio.
bool got_video = false;
bool got_audio = false;
// audio/video sequence and data offset.
int64_t av_sequence_offset_start = -1;
int64_t av_sequence_offset_end = -1;
for (;;) {
if ((err = reader->read(tag_header, SRS_FLV_TAG_HEADER_SIZE, NULL)) != srs_success) {
return srs_error_wrap(err, "read tag header");
}
SrsBuffer* tag_stream = new SrsBuffer(tag_header, SRS_FLV_TAG_HEADER_SIZE);
SrsAutoFree(SrsBuffer, tag_stream);
int8_t tag_type = tag_stream->read_1bytes();
int32_t data_size = tag_stream->read_3bytes();
bool is_video = tag_type == 0x09;
bool is_audio = tag_type == 0x08;
bool is_not_av = !is_video && !is_audio;
if (is_not_av) {
// skip body and tag size.
2015-07-28 09:56:50 +00:00
reader->skip(data_size + SRS_FLV_PREVIOUS_TAG_SIZE);
continue;
}
2019-05-21 00:43:20 +00:00
// if video duplicated, no audio
if (is_video && got_video) {
break;
}
// if audio duplicated, no video
if (is_audio && got_audio) {
break;
}
// video
if (is_video) {
srs_assert(!got_video);
got_video = true;
if (av_sequence_offset_start < 0) {
2015-07-28 09:56:50 +00:00
av_sequence_offset_start = reader->tellg() - SRS_FLV_TAG_HEADER_SIZE;
}
2015-07-28 09:56:50 +00:00
av_sequence_offset_end = reader->tellg() + data_size + SRS_FLV_PREVIOUS_TAG_SIZE;
reader->skip(data_size + SRS_FLV_PREVIOUS_TAG_SIZE);
}
// audio
if (is_audio) {
srs_assert(!got_audio);
got_audio = true;
2019-05-21 00:43:20 +00:00
if (av_sequence_offset_start < 0) {
2015-07-28 09:56:50 +00:00
av_sequence_offset_start = reader->tellg() - SRS_FLV_TAG_HEADER_SIZE;
}
2015-07-28 09:56:50 +00:00
av_sequence_offset_end = reader->tellg() + data_size + SRS_FLV_PREVIOUS_TAG_SIZE;
reader->skip(data_size + SRS_FLV_PREVIOUS_TAG_SIZE);
}
2019-02-07 14:49:41 +00:00
if (got_audio && got_video) {
break;
}
}
// seek to the sequence header start offset.
if (av_sequence_offset_start > 0) {
reader->seek2(av_sequence_offset_start);
*pstart = av_sequence_offset_start;
*psize = (int)(av_sequence_offset_end - av_sequence_offset_start);
}
return err;
}
srs_error_t SrsFlvVodStreamDecoder::seek2(int64_t offset)
{
srs_error_t err = srs_success;
2015-07-28 09:56:50 +00:00
if (offset >= reader->filesize()) {
return srs_error_new(ERROR_SYSTEM_FILE_EOF, "flv fast decoder seek overflow file, size=%d, offset=%d", (int)reader->filesize(), (int)offset);
}
if (reader->seek2(offset) < 0) {
return srs_error_new(ERROR_SYSTEM_FILE_SEEK, "flv fast decoder seek error, size=%d, offset=%d", (int)reader->filesize(), (int)offset);
}
return err;
}
2014-08-02 14:18:39 +00:00