mirror of
https://github.com/ossrs/srs.git
synced 2025-02-15 04:42:04 +00:00
1009 lines
23 KiB
C++
1009 lines
23 KiB
C++
/*
|
|
The MIT License (MIT)
|
|
|
|
Copyright (c) 2013-2014 winlin
|
|
|
|
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_librtmp.hpp>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string>
|
|
#include <sstream>
|
|
using namespace std;
|
|
|
|
#include <srs_kernel_error.hpp>
|
|
#include <srs_protocol_rtmp.hpp>
|
|
#include <srs_lib_simple_socket.hpp>
|
|
#include <srs_kernel_log.hpp>
|
|
#include <srs_protocol_utility.hpp>
|
|
#include <srs_core_autofree.hpp>
|
|
#include <srs_protocol_stack.hpp>
|
|
#include <srs_kernel_utility.hpp>
|
|
#include <srs_kernel_stream.hpp>
|
|
#include <srs_protocol_amf0.hpp>
|
|
#include <srs_kernel_flv.hpp>
|
|
#include <srs_kernel_codec.hpp>
|
|
#include <srs_kernel_file.hpp>
|
|
#include <srs_lib_bandwidth.hpp>
|
|
|
|
// if user want to define log, define the folowing macro.
|
|
#ifndef SRS_RTMP_USER_DEFINED_LOG
|
|
// kernel module.
|
|
ISrsLog* _srs_log = new ISrsLog();
|
|
ISrsThreadContext* _srs_context = new ISrsThreadContext();
|
|
#endif
|
|
|
|
/**
|
|
* export runtime context.
|
|
*/
|
|
struct Context
|
|
{
|
|
std::string url;
|
|
std::string tcUrl;
|
|
std::string host;
|
|
std::string ip;
|
|
std::string port;
|
|
std::string vhost;
|
|
std::string app;
|
|
std::string stream;
|
|
std::string param;
|
|
|
|
SrsRtmpClient* rtmp;
|
|
SimpleSocketStream* skt;
|
|
int stream_id;
|
|
|
|
Context() {
|
|
rtmp = NULL;
|
|
skt = NULL;
|
|
stream_id = 0;
|
|
}
|
|
virtual ~Context() {
|
|
srs_freep(rtmp);
|
|
srs_freep(skt);
|
|
}
|
|
};
|
|
|
|
int srs_librtmp_context_parse_uri(Context* context)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
// parse uri
|
|
size_t pos = string::npos;
|
|
string uri = context->url;
|
|
// tcUrl, stream
|
|
if ((pos = uri.rfind("/")) != string::npos) {
|
|
context->stream = uri.substr(pos + 1);
|
|
context->tcUrl = uri = uri.substr(0, pos);
|
|
}
|
|
|
|
std::string schema;
|
|
srs_discovery_tc_url(context->tcUrl,
|
|
schema, context->host, context->vhost, context->app, context->port,
|
|
context->param);
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_librtmp_context_resolve_host(Context* context)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
// create socket
|
|
srs_freep(context->skt);
|
|
context->skt = new SimpleSocketStream();
|
|
|
|
if ((ret = context->skt->create_socket()) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
// connect to server:port
|
|
context->ip = srs_dns_resolve(context->host);
|
|
if (context->ip.empty()) {
|
|
return -1;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_librtmp_context_connect(Context* context)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(context->skt);
|
|
|
|
std::string ip = context->ip;
|
|
int port = ::atoi(context->port.c_str());
|
|
|
|
if ((ret = context->skt->connect(ip.c_str(), port)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
extern "C"{
|
|
#endif
|
|
|
|
srs_rtmp_t srs_rtmp_create(const char* url)
|
|
{
|
|
Context* context = new Context();
|
|
context->url = url;
|
|
return context;
|
|
}
|
|
|
|
srs_rtmp_t srs_rtmp_create2(const char* url)
|
|
{
|
|
Context* context = new Context();
|
|
|
|
// use url as tcUrl.
|
|
context->url = url;
|
|
// auto append stream.
|
|
context->url += "/livestream";
|
|
|
|
return context;
|
|
}
|
|
|
|
void srs_rtmp_destroy(srs_rtmp_t rtmp)
|
|
{
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
srs_freep(context);
|
|
}
|
|
|
|
int srs_simple_handshake(srs_rtmp_t rtmp)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
if ((ret = __srs_dns_resolve(rtmp)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
if ((ret = __srs_connect_server(rtmp)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
if ((ret = __srs_do_simple_handshake(rtmp)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int __srs_dns_resolve(srs_rtmp_t rtmp)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
// parse uri
|
|
if ((ret = srs_librtmp_context_parse_uri(context)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
// resolve host
|
|
if ((ret = srs_librtmp_context_resolve_host(context)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int __srs_connect_server(srs_rtmp_t rtmp)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
if ((ret = srs_librtmp_context_connect(context)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int __srs_do_simple_handshake(srs_rtmp_t rtmp)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
srs_assert(context->skt != NULL);
|
|
|
|
// simple handshake
|
|
srs_freep(context->rtmp);
|
|
context->rtmp = new SrsRtmpClient(context->skt);
|
|
|
|
if ((ret = context->rtmp->simple_handshake()) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_connect_app(srs_rtmp_t rtmp)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
string tcUrl = srs_generate_tc_url(
|
|
context->ip, context->vhost, context->app, context->port,
|
|
context->param
|
|
);
|
|
|
|
if ((ret = context->rtmp->connect_app(
|
|
context->app, tcUrl, NULL, true)) != ERROR_SUCCESS)
|
|
{
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_connect_app2(srs_rtmp_t rtmp,
|
|
char srs_server_ip[128],char srs_server[128], char srs_primary_authors[128],
|
|
char srs_version[32], int* srs_id, int* srs_pid
|
|
) {
|
|
srs_server_ip[0] = 0;
|
|
srs_server[0] = 0;
|
|
srs_primary_authors[0] = 0;
|
|
srs_version[0] = 0;
|
|
*srs_id = 0;
|
|
*srs_pid = 0;
|
|
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
string tcUrl = srs_generate_tc_url(
|
|
context->ip, context->vhost, context->app, context->port,
|
|
context->param
|
|
);
|
|
|
|
std::string sip, sserver, sauthors, sversion;
|
|
|
|
if ((ret = context->rtmp->connect_app2(context->app, tcUrl, NULL, true,
|
|
sip, sserver, sauthors, sversion, *srs_id, *srs_pid)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
snprintf(srs_server_ip, 128, "%s", sip.c_str());
|
|
snprintf(srs_server, 128, "%s", sserver.c_str());
|
|
snprintf(srs_primary_authors, 128, "%s", sauthors.c_str());
|
|
snprintf(srs_version, 32, "%s", sversion.c_str());
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_play_stream(srs_rtmp_t rtmp)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
if ((ret = context->rtmp->create_stream(context->stream_id)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
if ((ret = context->rtmp->play(context->stream, context->stream_id)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_publish_stream(srs_rtmp_t rtmp)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
if ((ret = context->rtmp->fmle_publish(context->stream, context->stream_id)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
const char* srs_type2string(int type)
|
|
{
|
|
static const char* audio = "Audio";
|
|
static const char* video = "Video";
|
|
static const char* data = "Data";
|
|
static const char* unknown = "Unknown";
|
|
|
|
switch (type) {
|
|
case SRS_RTMP_TYPE_AUDIO: return audio;
|
|
case SRS_RTMP_TYPE_VIDEO: return video;
|
|
case SRS_RTMP_TYPE_SCRIPT: return data;
|
|
default: return unknown;
|
|
}
|
|
|
|
return unknown;
|
|
}
|
|
|
|
int srs_bandwidth_check(srs_rtmp_t rtmp,
|
|
int64_t* start_time, int64_t* end_time,
|
|
int* play_kbps, int* publish_kbps,
|
|
int* play_bytes, int* publish_bytes,
|
|
int* play_duration, int* publish_duration
|
|
) {
|
|
*start_time = 0;
|
|
*end_time = 0;
|
|
*play_kbps = 0;
|
|
*publish_kbps = 0;
|
|
*play_bytes = 0;
|
|
*publish_bytes = 0;
|
|
*play_duration = 0;
|
|
*publish_duration = 0;
|
|
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
SrsBandwidthClient client;
|
|
|
|
if ((ret = client.initialize(context->rtmp)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
if ((ret = client.bandwidth_check(
|
|
start_time, end_time, play_kbps, publish_kbps,
|
|
play_bytes, publish_bytes, play_duration, publish_duration)) != ERROR_SUCCESS
|
|
) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_read_packet(srs_rtmp_t rtmp, int* type, u_int32_t* timestamp, char** data, int* size)
|
|
{
|
|
*type = 0;
|
|
*timestamp = 0;
|
|
*data = NULL;
|
|
*size = 0;
|
|
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
for (;;) {
|
|
SrsMessage* msg = NULL;
|
|
if ((ret = context->rtmp->recv_message(&msg)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
if (!msg) {
|
|
continue;
|
|
}
|
|
|
|
SrsAutoFree(SrsMessage, msg);
|
|
|
|
if (msg->header.is_audio()) {
|
|
*type = SRS_RTMP_TYPE_AUDIO;
|
|
*timestamp = (u_int32_t)msg->header.timestamp;
|
|
*data = (char*)msg->payload;
|
|
*size = (int)msg->size;
|
|
// detach bytes from packet.
|
|
msg->payload = NULL;
|
|
} else if (msg->header.is_video()) {
|
|
*type = SRS_RTMP_TYPE_VIDEO;
|
|
*timestamp = (u_int32_t)msg->header.timestamp;
|
|
*data = (char*)msg->payload;
|
|
*size = (int)msg->size;
|
|
// detach bytes from packet.
|
|
msg->payload = NULL;
|
|
} else if (msg->header.is_amf0_data() || msg->header.is_amf3_data()) {
|
|
*type = SRS_RTMP_TYPE_SCRIPT;
|
|
*data = (char*)msg->payload;
|
|
*size = (int)msg->size;
|
|
// detach bytes from packet.
|
|
msg->payload = NULL;
|
|
} else {
|
|
// ignore and continue
|
|
continue;
|
|
}
|
|
|
|
// got expected message.
|
|
break;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_write_packet(srs_rtmp_t rtmp, int type, u_int32_t timestamp, char* data, int size)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
|
|
SrsSharedPtrMessage* msg = NULL;
|
|
|
|
if (type == SRS_RTMP_TYPE_AUDIO) {
|
|
SrsMessageHeader header;
|
|
header.initialize_audio(size, timestamp, context->stream_id);
|
|
|
|
msg = new SrsSharedPtrMessage();
|
|
if ((ret = msg->create(&header, data, size)) != ERROR_SUCCESS) {
|
|
srs_freep(data);
|
|
return ret;
|
|
}
|
|
} else if (type == SRS_RTMP_TYPE_VIDEO) {
|
|
SrsMessageHeader header;
|
|
header.initialize_video(size, timestamp, context->stream_id);
|
|
|
|
msg = new SrsSharedPtrMessage();
|
|
if ((ret = msg->create(&header, data, size)) != ERROR_SUCCESS) {
|
|
srs_freep(data);
|
|
return ret;
|
|
}
|
|
} else if (type == SRS_RTMP_TYPE_SCRIPT) {
|
|
SrsMessageHeader header;
|
|
header.initialize_amf0_script(size, context->stream_id);
|
|
|
|
msg = new SrsSharedPtrMessage();
|
|
if ((ret = msg->create(&header, data, size)) != ERROR_SUCCESS) {
|
|
srs_freep(data);
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
if (msg) {
|
|
// send out encoded msg.
|
|
if ((ret = context->rtmp->send_and_free_message(msg, context->stream_id)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
} else {
|
|
// directly free data if not sent out.
|
|
srs_freep(data);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_version_major()
|
|
{
|
|
return VERSION_MAJOR;
|
|
}
|
|
|
|
int srs_version_minor()
|
|
{
|
|
return VERSION_MINOR;
|
|
}
|
|
|
|
int srs_version_revision()
|
|
{
|
|
return VERSION_REVISION;
|
|
}
|
|
|
|
int64_t srs_get_time_ms()
|
|
{
|
|
srs_update_system_time_ms();
|
|
return srs_get_system_time_ms();
|
|
}
|
|
|
|
int64_t srs_get_nsend_bytes(srs_rtmp_t rtmp)
|
|
{
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
return context->rtmp->get_send_bytes();
|
|
}
|
|
|
|
int64_t srs_get_nrecv_bytes(srs_rtmp_t rtmp)
|
|
{
|
|
srs_assert(rtmp != NULL);
|
|
Context* context = (Context*)rtmp;
|
|
return context->rtmp->get_recv_bytes();
|
|
}
|
|
|
|
struct FlvContext
|
|
{
|
|
SrsFileReader reader;
|
|
SrsFileWriter writer;
|
|
SrsFlvEncoder enc;
|
|
SrsFlvDecoder dec;
|
|
};
|
|
|
|
srs_flv_t srs_flv_open_read(const char* file)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
FlvContext* flv = new FlvContext();
|
|
|
|
if ((ret = flv->reader.open(file)) != ERROR_SUCCESS) {
|
|
srs_freep(flv);
|
|
return NULL;
|
|
}
|
|
|
|
if ((ret = flv->dec.initialize(&flv->reader)) != ERROR_SUCCESS) {
|
|
srs_freep(flv);
|
|
return NULL;
|
|
}
|
|
|
|
return flv;
|
|
}
|
|
|
|
srs_flv_t srs_flv_open_write(const char* file)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
FlvContext* flv = new FlvContext();
|
|
|
|
if ((ret = flv->writer.open(file)) != ERROR_SUCCESS) {
|
|
srs_freep(flv);
|
|
return NULL;
|
|
}
|
|
|
|
if ((ret = flv->enc.initialize(&flv->writer)) != ERROR_SUCCESS) {
|
|
srs_freep(flv);
|
|
return NULL;
|
|
}
|
|
|
|
return flv;
|
|
}
|
|
|
|
void srs_flv_close(srs_flv_t flv)
|
|
{
|
|
FlvContext* context = (FlvContext*)flv;
|
|
srs_freep(context);
|
|
}
|
|
|
|
int srs_flv_read_header(srs_flv_t flv, char header[9])
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
FlvContext* context = (FlvContext*)flv;
|
|
|
|
if (!context->reader.is_open()) {
|
|
return ERROR_SYSTEM_IO_INVALID;
|
|
}
|
|
|
|
if ((ret = context->dec.read_header(header)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
char ts[4]; // tag size
|
|
if ((ret = context->dec.read_previous_tag_size(ts)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_flv_read_tag_header(srs_flv_t flv, char* ptype, int32_t* pdata_size, u_int32_t* ptime)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
FlvContext* context = (FlvContext*)flv;
|
|
|
|
if (!context->reader.is_open()) {
|
|
return ERROR_SYSTEM_IO_INVALID;
|
|
}
|
|
|
|
if ((ret = context->dec.read_tag_header(ptype, pdata_size, ptime)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_flv_read_tag_data(srs_flv_t flv, char* data, int32_t size)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
FlvContext* context = (FlvContext*)flv;
|
|
|
|
if (!context->reader.is_open()) {
|
|
return ERROR_SYSTEM_IO_INVALID;
|
|
}
|
|
|
|
if ((ret = context->dec.read_tag_data(data, size)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
char ts[4]; // tag size
|
|
if ((ret = context->dec.read_previous_tag_size(ts)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_flv_write_header(srs_flv_t flv, char header[9])
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
FlvContext* context = (FlvContext*)flv;
|
|
|
|
if (!context->writer.is_open()) {
|
|
return ERROR_SYSTEM_IO_INVALID;
|
|
}
|
|
|
|
if ((ret = context->enc.write_header(header)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_flv_write_tag(srs_flv_t flv, char type, int32_t time, char* data, int size)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
FlvContext* context = (FlvContext*)flv;
|
|
|
|
if (!context->writer.is_open()) {
|
|
return ERROR_SYSTEM_IO_INVALID;
|
|
}
|
|
|
|
if (type == SRS_RTMP_TYPE_AUDIO) {
|
|
return context->enc.write_audio(time, data, size);
|
|
} else if (type == SRS_RTMP_TYPE_VIDEO) {
|
|
return context->enc.write_video(time, data, size);
|
|
} else {
|
|
return context->enc.write_metadata(data, size);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int srs_flv_size_tag(int data_size)
|
|
{
|
|
return SrsFlvEncoder::size_tag(data_size);
|
|
}
|
|
|
|
int64_t srs_flv_tellg(srs_flv_t flv)
|
|
{
|
|
FlvContext* context = (FlvContext*)flv;
|
|
return context->reader.tellg();
|
|
}
|
|
|
|
void srs_flv_lseek(srs_flv_t flv, int64_t offset)
|
|
{
|
|
FlvContext* context = (FlvContext*)flv;
|
|
context->reader.lseek(offset);
|
|
}
|
|
|
|
flv_bool srs_flv_is_eof(int error_code)
|
|
{
|
|
return error_code == ERROR_SYSTEM_FILE_EOF;
|
|
}
|
|
|
|
flv_bool srs_flv_is_sequence_header(char* data, int32_t size)
|
|
{
|
|
return SrsFlvCodec::video_is_sequence_header(data, (int)size);
|
|
}
|
|
|
|
flv_bool srs_flv_is_keyframe(char* data, int32_t size)
|
|
{
|
|
return SrsFlvCodec::video_is_keyframe(data, (int)size);
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_parse(char* data, int size, int* nparsed)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
srs_amf0_t amf0 = NULL;
|
|
|
|
SrsStream stream;
|
|
if ((ret = stream.initialize(data, size)) != ERROR_SUCCESS) {
|
|
return amf0;
|
|
}
|
|
|
|
SrsAmf0Any* any = NULL;
|
|
if ((ret = SrsAmf0Any::discovery(&stream, &any)) != ERROR_SUCCESS) {
|
|
return amf0;
|
|
}
|
|
|
|
stream.skip(-1 * stream.pos());
|
|
if ((ret = any->read(&stream)) != ERROR_SUCCESS) {
|
|
srs_freep(any);
|
|
return amf0;
|
|
}
|
|
|
|
*nparsed = stream.pos();
|
|
amf0 = (srs_amf0_t)any;
|
|
|
|
return amf0;
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_create_number(srs_amf0_number value)
|
|
{
|
|
return SrsAmf0Any::number(value);
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_create_ecma_array()
|
|
{
|
|
return SrsAmf0Any::ecma_array();
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_create_strict_array()
|
|
{
|
|
return SrsAmf0Any::strict_array();
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_create_object()
|
|
{
|
|
return SrsAmf0Any::object();
|
|
}
|
|
|
|
void srs_amf0_free(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_freep(any);
|
|
}
|
|
|
|
void srs_amf0_free_bytes(char* data)
|
|
{
|
|
srs_freep(data);
|
|
}
|
|
|
|
int srs_amf0_size(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->total_size();
|
|
}
|
|
|
|
int srs_amf0_serialize(srs_amf0_t amf0, char* data, int size)
|
|
{
|
|
int ret = ERROR_SUCCESS;
|
|
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
|
|
SrsStream stream;
|
|
if ((ret = stream.initialize(data, size)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
if ((ret = any->write(&stream)) != ERROR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
srs_amf0_bool srs_amf0_is_string(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->is_string();
|
|
}
|
|
|
|
srs_amf0_bool srs_amf0_is_boolean(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->is_boolean();
|
|
}
|
|
|
|
srs_amf0_bool srs_amf0_is_number(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->is_number();
|
|
}
|
|
|
|
srs_amf0_bool srs_amf0_is_null(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->is_null();
|
|
}
|
|
|
|
srs_amf0_bool srs_amf0_is_object(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->is_object();
|
|
}
|
|
|
|
srs_amf0_bool srs_amf0_is_ecma_array(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->is_ecma_array();
|
|
}
|
|
|
|
srs_amf0_bool srs_amf0_is_strict_array(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->is_strict_array();
|
|
}
|
|
|
|
const char* srs_amf0_to_string(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->to_str_raw();
|
|
}
|
|
|
|
srs_amf0_bool srs_amf0_to_boolean(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->to_boolean();
|
|
}
|
|
|
|
srs_amf0_number srs_amf0_to_number(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
return any->to_number();
|
|
}
|
|
|
|
void srs_amf0_set_number(srs_amf0_t amf0, srs_amf0_number value)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
any->set_number(value);
|
|
}
|
|
|
|
int srs_amf0_object_property_count(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_object());
|
|
|
|
SrsAmf0Object* obj = (SrsAmf0Object*)amf0;
|
|
return obj->count();
|
|
}
|
|
|
|
const char* srs_amf0_object_property_name_at(srs_amf0_t amf0, int index)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_object());
|
|
|
|
SrsAmf0Object* obj = (SrsAmf0Object*)amf0;
|
|
return obj->key_raw_at(index);
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_object_property_value_at(srs_amf0_t amf0, int index)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_object());
|
|
|
|
SrsAmf0Object* obj = (SrsAmf0Object*)amf0;
|
|
return (srs_amf0_t)obj->value_at(index);
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_object_property(srs_amf0_t amf0, const char* name)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_object());
|
|
|
|
SrsAmf0Object* obj = (SrsAmf0Object*)amf0;
|
|
return (srs_amf0_t)obj->get_property(name);
|
|
}
|
|
|
|
void srs_amf0_object_property_set(srs_amf0_t amf0, const char* name, srs_amf0_t value)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_object());
|
|
|
|
SrsAmf0Object* obj = (SrsAmf0Object*)amf0;
|
|
any = (SrsAmf0Any*)value;
|
|
obj->set(name, any);
|
|
}
|
|
|
|
void srs_amf0_object_clear(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_object());
|
|
|
|
SrsAmf0Object* obj = (SrsAmf0Object*)amf0;
|
|
obj->clear();
|
|
}
|
|
|
|
int srs_amf0_ecma_array_property_count(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_ecma_array());
|
|
|
|
SrsAmf0EcmaArray * obj = (SrsAmf0EcmaArray*)amf0;
|
|
return obj->count();
|
|
}
|
|
|
|
const char* srs_amf0_ecma_array_property_name_at(srs_amf0_t amf0, int index)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_ecma_array());
|
|
|
|
SrsAmf0EcmaArray* obj = (SrsAmf0EcmaArray*)amf0;
|
|
return obj->key_raw_at(index);
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_ecma_array_property_value_at(srs_amf0_t amf0, int index)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_ecma_array());
|
|
|
|
SrsAmf0EcmaArray* obj = (SrsAmf0EcmaArray*)amf0;
|
|
return (srs_amf0_t)obj->value_at(index);
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_ecma_array_property(srs_amf0_t amf0, const char* name)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_ecma_array());
|
|
|
|
SrsAmf0EcmaArray* obj = (SrsAmf0EcmaArray*)amf0;
|
|
return (srs_amf0_t)obj->get_property(name);
|
|
}
|
|
|
|
void srs_amf0_ecma_array_property_set(srs_amf0_t amf0, const char* name, srs_amf0_t value)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_ecma_array());
|
|
|
|
SrsAmf0EcmaArray* obj = (SrsAmf0EcmaArray*)amf0;
|
|
any = (SrsAmf0Any*)value;
|
|
obj->set(name, any);
|
|
}
|
|
|
|
int srs_amf0_strict_array_property_count(srs_amf0_t amf0)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_strict_array());
|
|
|
|
SrsAmf0StrictArray * obj = (SrsAmf0StrictArray*)amf0;
|
|
return obj->count();
|
|
}
|
|
|
|
srs_amf0_t srs_amf0_strict_array_property_at(srs_amf0_t amf0, int index)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_strict_array());
|
|
|
|
SrsAmf0StrictArray* obj = (SrsAmf0StrictArray*)amf0;
|
|
return (srs_amf0_t)obj->at(index);
|
|
}
|
|
|
|
void srs_amf0_strict_array_append(srs_amf0_t amf0, srs_amf0_t value)
|
|
{
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
srs_assert(any->is_strict_array());
|
|
|
|
SrsAmf0StrictArray* obj = (SrsAmf0StrictArray*)amf0;
|
|
any = (SrsAmf0Any*)value;
|
|
obj->append(any);
|
|
}
|
|
|
|
char* srs_amf0_human_print(srs_amf0_t amf0, char** pdata, int* psize)
|
|
{
|
|
if (!amf0) {
|
|
return NULL;
|
|
}
|
|
|
|
SrsAmf0Any* any = (SrsAmf0Any*)amf0;
|
|
|
|
return any->human_print(pdata, psize);
|
|
}
|
|
|
|
int srs_h264_to_rtmp(char* h264_raw_data, int h264_raw_size, u_int32_t dts, u_int32_t pts, char** prtmp_data, int* prtmp_size, u_int32_t* ptimestamp)
|
|
{
|
|
*prtmp_data = h264_raw_data;
|
|
*prtmp_size = h264_raw_size;
|
|
return 0;
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|