static void *rtmp_stream_create(obs_data_t *settings, obs_output_t *output)
{
struct rtmp_stream *stream = bzalloc(sizeof(struct rtmp_stream));
stream->output = output;
pthread_mutex_init_value(&stream->packets_mutex);
RTMP_Init(&stream->rtmp);
RTMP_LogSetCallback(log_rtmp);
RTMP_LogSetLevel(RTMP_LOGWARNING);
if (pthread_mutex_init(&stream->packets_mutex, NULL) != 0)
goto fail;
if (os_event_init(&stream->stop_event, OS_EVENT_TYPE_MANUAL) != 0)
goto fail;
if (pthread_mutex_init(&stream->write_buf_mutex, NULL) != 0) {
warn("Failed to initialize write buffer mutex");
goto fail;
}
if (os_event_init(&stream->buffer_space_available_event,
OS_EVENT_TYPE_AUTO) != 0) {
warn("Failed to initialize write buffer event");
goto fail;
}
if (os_event_init(&stream->buffer_has_data_event,
OS_EVENT_TYPE_AUTO) != 0) {
warn("Failed to initialize data buffer event");
goto fail;
}
if (os_event_init(&stream->socket_available_event,
OS_EVENT_TYPE_AUTO) != 0) {
warn("Failed to initialize socket buffer event");
goto fail;
}
if (os_event_init(&stream->send_thread_signaled_exit,
OS_EVENT_TYPE_MANUAL) != 0) {
warn("Failed to initialize socket exit event");
goto fail;
}
UNUSED_PARAMETER(settings);
return stream;
fail:
rtmp_stream_destroy(stream);
return NULL;
}
RTMPWriter::RTMPWriter()
:rtmp(NULL)
,state(RW_STATE_NONE)
,cb_disconnect(NULL)
,cb_user(NULL)
{
//#if !defined(NDEBUG)
RTMP_LogSetLevel(RTMP_LOGDEBUG);
RTMP_LogSetOutput(stderr);
//#endif
signal(SIGPIPE, rtmp_sigpipe_handler);
// signal(SIGPIPE, SIG_IGN);
}
/**
* Open RTMP connection and verify that the stream can be played.
*
* URL syntax: rtmp://server[:port][/app][/playpath][ keyword=value]...
* where 'app' is first one or two directories in the path
* (e.g. /ondemand/, /flash/live/, etc.)
* and 'playpath' is a file name (the rest of the path,
* may be prefixed with "mp4:")
*
* Additional RTMP library options may be appended as
* space-separated key-value pairs.
*/
static int rtmp_open(URLContext *s, const char *uri, int flags)
{
RTMP *r;
int rc;
r = av_mallocz(sizeof(RTMP));
if (!r)
return AVERROR(ENOMEM);
switch (av_log_get_level()) {
default:
case AV_LOG_FATAL: rc = RTMP_LOGCRIT; break;
case AV_LOG_ERROR: rc = RTMP_LOGERROR; break;
case AV_LOG_WARNING: rc = RTMP_LOGWARNING; break;
case AV_LOG_INFO: rc = RTMP_LOGINFO; break;
case AV_LOG_VERBOSE: rc = RTMP_LOGDEBUG; break;
case AV_LOG_DEBUG: rc = RTMP_LOGDEBUG2; break;
}
RTMP_LogSetLevel(rc);
RTMP_LogSetCallback(rtmp_log);
RTMP_Init(r);
if (!RTMP_SetupURL(r, s->filename)) {
rc = -1;
goto fail;
}
if (flags & AVIO_WRONLY)
RTMP_EnableWrite(r);
if (!RTMP_Connect(r, NULL) || !RTMP_ConnectStream(r, 0)) {
rc = -1;
goto fail;
}
s->priv_data = r;
s->is_streamed = 1;
return 0;
fail:
av_free(r);
return rc;
}
LibRtmp::LibRtmp(bool isNeedLog, bool isNeedRecord)
{
if (isNeedLog)
{
flog_ = fopen("librtmp.log", "w");
RTMP_LogSetLevel(RTMP_LOGDEBUG2);
RTMP_LogSetOutput(flog_);
}
else
{
flog_ = NULL;
}
rtmp_ = RTMP_Alloc();
RTMP_Init(rtmp_);
RTMP_SetBufferMS(rtmp_, 300);
streming_url_ = NULL;
is_need_record_ = isNeedRecord;
}
static void *rtmp_stream_create(obs_data_t *settings, obs_output_t *output)
{
struct rtmp_stream *stream = bzalloc(sizeof(struct rtmp_stream));
stream->output = output;
pthread_mutex_init_value(&stream->packets_mutex);
RTMP_Init(&stream->rtmp);
RTMP_LogSetCallback(log_rtmp);
RTMP_LogSetLevel(RTMP_LOGWARNING);
if (pthread_mutex_init(&stream->packets_mutex, NULL) != 0)
goto fail;
if (os_event_init(&stream->stop_event, OS_EVENT_TYPE_MANUAL) != 0)
goto fail;
UNUSED_PARAMETER(settings);
return stream;
fail:
rtmp_stream_destroy(stream);
return NULL;
}
/**
* Open RTMP connection and verify that the stream can be played.
*
* URL syntax: rtmp://server[:port][/app][/playpath][ keyword=value]...
* where 'app' is first one or two directories in the path
* (e.g. /ondemand/, /flash/live/, etc.)
* and 'playpath' is a file name (the rest of the path,
* may be prefixed with "mp4:")
*
* Additional RTMP library options may be appended as
* space-separated key-value pairs.
*/
static int rtmp_open(URLContext *s, const char *uri, int flags)
{
LibRTMPContext *ctx = s->priv_data;
RTMP *r = &ctx->rtmp;
int rc = 0, level;
char *filename = s->filename;
int len = strlen(s->filename) + 1;
switch (av_log_get_level()) {
default:
case AV_LOG_FATAL: level = RTMP_LOGCRIT; break;
case AV_LOG_ERROR: level = RTMP_LOGERROR; break;
case AV_LOG_WARNING: level = RTMP_LOGWARNING; break;
case AV_LOG_INFO: level = RTMP_LOGINFO; break;
case AV_LOG_VERBOSE: level = RTMP_LOGDEBUG; break;
case AV_LOG_DEBUG: level = RTMP_LOGDEBUG2; break;
}
RTMP_LogSetLevel(level);
RTMP_LogSetCallback(rtmp_log);
if (ctx->app) len += strlen(ctx->app) + sizeof(" app=");
if (ctx->tcurl) len += strlen(ctx->tcurl) + sizeof(" tcUrl=");
if (ctx->pageurl) len += strlen(ctx->pageurl) + sizeof(" pageUrl=");
if (ctx->flashver) len += strlen(ctx->flashver) + sizeof(" flashver=");
if (ctx->conn) {
char *sep, *p = ctx->conn;
int options = 0;
while (p) {
options++;
p += strspn(p, " ");
if (!*p)
break;
sep = strchr(p, ' ');
if (sep)
p = sep + 1;
else
break;
}
len += options * sizeof(" conn=");
len += strlen(ctx->conn);
}
if (ctx->playpath)
len += strlen(ctx->playpath) + sizeof(" playpath=");
if (ctx->live)
len += sizeof(" live=1");
if (ctx->subscribe)
len += strlen(ctx->subscribe) + sizeof(" subscribe=");
if (ctx->client_buffer_time)
len += strlen(ctx->client_buffer_time) + sizeof(" buffer=");
if (ctx->swfurl || ctx->swfverify) {
len += sizeof(" swfUrl=");
if (ctx->swfverify)
len += strlen(ctx->swfverify) + sizeof(" swfVfy=1");
else
len += strlen(ctx->swfurl);
}
if (!(ctx->temp_filename = filename = av_malloc(len)))
return AVERROR(ENOMEM);
av_strlcpy(filename, s->filename, len);
if (ctx->app) {
av_strlcat(filename, " app=", len);
av_strlcat(filename, ctx->app, len);
}
if (ctx->tcurl) {
av_strlcat(filename, " tcUrl=", len);
av_strlcat(filename, ctx->tcurl, len);
}
if (ctx->pageurl) {
av_strlcat(filename, " pageUrl=", len);
av_strlcat(filename, ctx->pageurl, len);
}
if (ctx->swfurl) {
av_strlcat(filename, " swfUrl=", len);
av_strlcat(filename, ctx->swfurl, len);
}
if (ctx->flashver) {
av_strlcat(filename, " flashVer=", len);
av_strlcat(filename, ctx->flashver, len);
}
if (ctx->conn) {
char *sep, *p = ctx->conn;
while (p) {
av_strlcat(filename, " conn=", len);
p += strspn(p, " ");
if (!*p)
break;
sep = strchr(p, ' ');
if (sep)
*sep = '\0';
av_strlcat(filename, p, len);
if (sep)
p = sep + 1;
else
break;
}
}
if (ctx->playpath) {
av_strlcat(filename, " playpath=", len);
av_strlcat(filename, ctx->playpath, len);
}
if (ctx->live)
av_strlcat(filename, " live=1", len);
if (ctx->subscribe) {
av_strlcat(filename, " subscribe=", len);
av_strlcat(filename, ctx->subscribe, len);
}
if (ctx->client_buffer_time) {
av_strlcat(filename, " buffer=", len);
av_strlcat(filename, ctx->client_buffer_time, len);
}
if (ctx->swfurl || ctx->swfverify) {
av_strlcat(filename, " swfUrl=", len);
if (ctx->swfverify) {
av_strlcat(filename, ctx->swfverify, len);
av_strlcat(filename, " swfVfy=1", len);
} else {
av_strlcat(filename, ctx->swfurl, len);
}
}
RTMP_Init(r);
if (!RTMP_SetupURL(r, filename)) {
rc = AVERROR_UNKNOWN;
goto fail;
}
if (flags & AVIO_FLAG_WRITE)
RTMP_EnableWrite(r);
if (!RTMP_Connect(r, NULL) || !RTMP_ConnectStream(r, 0)) {
rc = AVERROR_UNKNOWN;
goto fail;
}
#if CONFIG_NETWORK
if (ctx->buffer_size >= 0 && (flags & AVIO_FLAG_WRITE)) {
int tmp = ctx->buffer_size;
setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_SNDBUF, &tmp, sizeof(tmp));
}
#endif
s->is_streamed = 1;
return 0;
fail:
av_freep(&ctx->temp_filename);
if (rc)
RTMP_Close(r);
return rc;
}
void QRtmp::setLogLevel(RTMP_LogLevel level)
{
RTMP_LogSetLevel((::RTMP_LogLevel)(static_cast<int>(level)));
}
static event_t *
rtmp_playvideo(const char *url0, media_pipe_t *mp,
char *errbuf, size_t errlen,
video_queue_t *vq, struct vsource_list *vsl,
const video_args_t *va0)
{
video_args_t va = *va0;
rtmp_t r = {0};
event_t *e;
char *url = mystrdupa(url0);
mp_set_url(mp, va0->canonical_url, va0->parent_url, va0->parent_title);
usage_event("Play video", 1, USAGE_SEG("format", "RTMP"));
prop_set(mp->mp_prop_metadata, "format", PROP_SET_STRING, "RTMP");
prop_set(mp->mp_prop_root, "loading", PROP_SET_INT, 1);
va.flags |= BACKEND_VIDEO_NO_FS_SCAN;
rtmp_log_level = RTMP_LOGINFO;
RTMP_LogSetLevel(rtmp_log_level);
r.r = RTMP_Alloc();
RTMP_Init(r.r, mp->mp_cancellable);
int64_t start = playinfo_get_restartpos(va.canonical_url, va.title, va.resume_mode);
if(!RTMP_SetupURL(r.r, url)) {
snprintf(errbuf, errlen, "Unable to setup RTMP-session");
rtmp_free(&r);
return NULL;
}
r.r->Link.lFlags |= RTMP_LF_SWFV;
if(!RTMP_Connect(r.r, NULL, errbuf, errlen, 5000)) {
rtmp_free(&r);
return NULL;
}
if(!RTMP_ConnectStream(r.r, 0)) {
snprintf(errbuf, errlen, "Unable to connect RTMP-stream");
rtmp_free(&r);
return NULL;
}
if(start)
RTMP_SendSeek(r.r, start);
r.mp = mp;
mp->mp_audio.mq_stream = 0;
mp->mp_video.mq_stream = 0;
if(start > 0) {
r.seekpos_video = start * 1000;
r.seekpos_audio = start * 1000;
mp->mp_seek_base = r.seekpos_video;
mp->mp_video.mq_seektarget = r.seekpos_video;
mp->mp_audio.mq_seektarget = r.seekpos_video;
} else {
mp->mp_video.mq_seektarget = AV_NOPTS_VALUE;
mp->mp_audio.mq_seektarget = AV_NOPTS_VALUE;
mp->mp_seek_base = 0;
r.seekpos_audio = AV_NOPTS_VALUE;
r.seekpos_video = AV_NOPTS_VALUE;
}
mp_configure(mp, MP_CAN_PAUSE, MP_BUFFER_DEEP, 0, "video");
mp->mp_max_realtime_delay = (r.r->Link.timeout - 1) * 1000000;
mp_become_primary(mp);
playinfo_register_play(va.canonical_url, 0);
r.canonical_url = va.canonical_url;
r.restartpos_last = -1;
r.url = url;
r.va = &va;
r.is_loading = 1;
e = rtmp_loop(&r, mp, url, errbuf, errlen);
if(r.ss)
sub_scanner_destroy(r.ss);
if(r.total_duration) {
int p = mp->mp_seek_base / (r.total_duration * 10);
if(p >= video_settings.played_threshold) {
TRACE(TRACE_DEBUG, "RTMP", "Playback reached %d%%, counting as played",
p);
playinfo_register_play(va.canonical_url, 1);
playinfo_set_restartpos(va.canonical_url, -1, 0);
} else {
playinfo_set_restartpos(va.canonical_url, mp->mp_seek_base / 1000, 0);
}
}
mp_shutdown(mp);
TRACE(TRACE_DEBUG, "RTMP", "End of playback");
rtmp_free(&r);
return e;
}
static event_t *
rtmp_playvideo(const char *url0, media_pipe_t *mp,
char *errbuf, size_t errlen,
video_queue_t *vq, struct vsource_list *vsl,
const video_args_t *va0)
{
video_args_t va = *va0;
rtmp_t r = {0};
event_t *e;
char *url = mystrdupa(url0);
va.flags |= BACKEND_VIDEO_NO_FS_SCAN;
prop_set_string(mp->mp_prop_type, "video");
rtmp_log_level = RTMP_LOGINFO;
RTMP_LogSetLevel(rtmp_log_level);
r.r = RTMP_Alloc();
RTMP_Init(r.r);
int64_t start = 0;
if(va.flags & BACKEND_VIDEO_RESUME ||
(video_settings.resume_mode == VIDEO_RESUME_YES &&
!(va.flags & BACKEND_VIDEO_START_FROM_BEGINNING)))
start = video_get_restartpos(va.canonical_url);
if(!RTMP_SetupURL(r.r, url)) {
snprintf(errbuf, errlen, "Unable to setup RTMP-session");
rtmp_free(&r);
return NULL;
}
r.r->Link.lFlags |= RTMP_LF_SWFV;
if(!RTMP_Connect(r.r, NULL)) {
snprintf(errbuf, errlen, "Unable to connect RTMP-session");
rtmp_free(&r);
return NULL;
}
if(!RTMP_ConnectStream(r.r, 0)) {
snprintf(errbuf, errlen, "Unable to connect RTMP-stream");
rtmp_free(&r);
return NULL;
}
if(start)
RTMP_SendSeek(r.r, start);
r.mp = mp;
mp->mp_audio.mq_stream = 0;
mp->mp_video.mq_stream = 0;
if(start > 0) {
r.seekpos_video = start * 1000;
r.seekpos_audio = start * 1000;
mp->mp_seek_base = r.seekpos_video;
mp->mp_video.mq_seektarget = r.seekpos_video;
mp->mp_audio.mq_seektarget = r.seekpos_video;
} else {
mp->mp_video.mq_seektarget = AV_NOPTS_VALUE;
mp->mp_audio.mq_seektarget = AV_NOPTS_VALUE;
mp->mp_seek_base = 0;
r.seekpos_audio = AV_NOPTS_VALUE;
r.seekpos_video = AV_NOPTS_VALUE;
}
mp_configure(mp, MP_PLAY_CAPS_PAUSE, MP_BUFFER_DEEP, 0);
mp->mp_max_realtime_delay = (r.r->Link.timeout - 1) * 1000000;
mp_become_primary(mp);
metadb_register_play(va.canonical_url, 0, CONTENT_VIDEO);
r.canonical_url = va.canonical_url;
r.restartpos_last = -1;
sub_scanner_t *ss =
sub_scanner_create(url, mp->mp_prop_subtitle_tracks, &va, 0);
e = rtmp_loop(&r, mp, url, errbuf, errlen);
sub_scanner_destroy(ss);
if(r.total_duration) {
int p = mp->mp_seek_base / (r.total_duration * 10);
if(p >= video_settings.played_threshold) {
TRACE(TRACE_DEBUG, "RTMP", "Playback reached %d%%, counting as played",
p);
metadb_register_play(va.canonical_url, 1, CONTENT_VIDEO);
metadb_set_video_restartpos(va.canonical_url, -1);
}
}
mp_flush(mp, 0);
mp_shutdown(mp);
TRACE(TRACE_DEBUG, "RTMP", "End of stream");
rtmp_free(&r);
return e;
}
static event_t *
rtmp_playvideo(const char *url0, media_pipe_t *mp,
int flags, int priority,
char *errbuf, size_t errlen,
const char *mimetype,
const char *canonical_url)
{
rtmp_t r = {0};
event_t *e;
char *url = mystrdupa(url0);
prop_set_string(mp->mp_prop_type, "video");
RTMP_LogSetLevel(RTMP_LOGINFO);
r.r = RTMP_Alloc();
RTMP_Init(r.r);
int64_t start = video_get_restartpos(canonical_url);
if(!RTMP_SetupURL(r.r, url)) {
snprintf(errbuf, errlen, "Unable to setup RTMP-session");
rtmp_free(&r);
return NULL;
}
if(!RTMP_Connect(r.r, NULL)) {
snprintf(errbuf, errlen, "Unable to connect RTMP-session");
rtmp_free(&r);
return NULL;
}
if(!RTMP_ConnectStream(r.r, start)) {
snprintf(errbuf, errlen, "Unable to connect RTMP-stream");
rtmp_free(&r);
return NULL;
}
r.seek_origin = start;
r.mp = mp;
r.hold = 0;
r.lost_focus = 0;
r.epoch = 1;
mp->mp_audio.mq_stream = 0;
mp->mp_video.mq_stream = 0;
if(start > 0) {
r.seekpos = start * 1000;
r.seekbase = r.seekpos;
mp->mp_video.mq_seektarget = r.seekpos;
mp->mp_audio.mq_seektarget = r.seekpos;
} else {
mp->mp_video.mq_seektarget = AV_NOPTS_VALUE;
mp->mp_audio.mq_seektarget = AV_NOPTS_VALUE;
r.seekbase = AV_NOPTS_VALUE;
r.seekpos = AV_NOPTS_VALUE;
}
mp_configure(mp, MP_PLAY_CAPS_PAUSE, MP_BUFFER_DEEP);
mp->mp_max_realtime_delay = (r.r->Link.timeout - 1) * 1000000;
mp_become_primary(mp);
metadb_register_play(canonical_url, 0, CONTENT_VIDEO);
r.canonical_url = canonical_url;
r.restartpos_last = -1;
e = rtmp_loop(&r, mp, url, errbuf, errlen);
if(r.total_duration) {
int p = r.seekbase / (r.total_duration * 10);
if(p >= video_settings.played_threshold) {
TRACE(TRACE_DEBUG, "RTMP", "Playback reached %d%%, counting as played",
p);
metadb_register_play(canonical_url, 1, CONTENT_VIDEO);
metadb_set_video_restartpos(canonical_url, -1);
}
}
mp_flush(mp, 0);
mp_shutdown(mp);
TRACE(TRACE_DEBUG, "RTMP", "End of stream");
rtmp_free(&r);
return e;
}
int main(int argc, char* argv[])
{
InitSockets();
double duration = -1;
int nRead;
//is live stream ?
bool bLiveStream = TRUE;
int bufsize = 1024 * 1024 * 10;
char *buf = (char*)malloc(bufsize);
memset(buf, 0, bufsize);
long countbufsize = 0;
FILE *fp = fopen("receive.flv", "wb");
if (!fp)
{
RTMP_LogPrintf("Open File Error.\n");
CleanupSockets();
return -1;
}
/* set log level */
RTMP_LogLevel loglvl = RTMP_LOGDEBUG;
RTMP_LogSetLevel(loglvl);
RTMP *rtmp = RTMP_Alloc();
RTMP_Init(rtmp);
//set connection timeout,default 30s
rtmp->Link.timeout = 10;
char* rtmpFilePath = "d:\\rtmp.raw";
rtmp->m_pRTMPFile = fopen(rtmpFilePath,"rb");
if (!rtmp->m_pRTMPFile)
{
RTMP_LogPrintf("Failed to open File :%s\n", rtmpFilePath);
return FALSE;
}
// HKS's live URL
if(!RTMP_SetupURL(rtmp, "rtmp://live.hkstv.hk.lxdns.com/live/hks"))
{
RTMP_Log(RTMP_LOGERROR, "SetupURL Err\n");
RTMP_Free(rtmp);
CleanupSockets();
return -1;
}
if (bLiveStream)
{
rtmp->Link.lFlags |= RTMP_LF_LIVE;
}
//1hour
RTMP_SetBufferMS(rtmp, 3600 * 1000);
if(!RTMP_Connect(rtmp, NULL))
{
RTMP_Log(RTMP_LOGERROR, "Connect Err\n");
RTMP_Free(rtmp);
CleanupSockets();
return -1;
}
if(!RTMP_ConnectStream(rtmp, 0))
{
RTMP_Log(RTMP_LOGERROR, "ConnectStream Err\n");
RTMP_Close(rtmp);
RTMP_Free(rtmp);
CleanupSockets();
return -1;
}
while(nRead = RTMP_Read(rtmp, buf, bufsize))
{
fwrite(buf, 1, nRead, fp);
countbufsize += nRead;
RTMP_LogPrintf("Receive: %5dByte, Total: %5.2fkB\n", nRead, countbufsize * 1.0 / 1024);
}
if(fp)
fclose(fp);
if(fpPcap)
fclose(fpPcap);
if (rtmp->m_pRTMPFile)
{
fclose(rtmp->m_pRTMPFile);
}
if(buf)
{
free(buf);
}
if(rtmp)
{
RTMP_Close(rtmp);
RTMP_Free(rtmp);
CleanupSockets();
rtmp = NULL;
}
return 0;
}
/**
* Open RTMP connection and verify that the stream can be played.
*
* URL syntax: rtmp://server[:port][/app][/playpath][ keyword=value]...
* where 'app' is first one or two directories in the path
* (e.g. /ondemand/, /flash/live/, etc.)
* and 'playpath' is a file name (the rest of the path,
* may be prefixed with "mp4:")
*
* Additional RTMP library options may be appended as
* space-separated key-value pairs.
*/
static int rtmp_open(URLContext *s, const char *uri, int flags)
{
LibRTMPContext *ctx = s->priv_data;
RTMP *r = &ctx->rtmp;
int rc = 0, level;
char *filename = s->filename;
switch (av_log_get_level()) {
default:
case AV_LOG_FATAL: level = RTMP_LOGCRIT; break;
case AV_LOG_ERROR: level = RTMP_LOGERROR; break;
case AV_LOG_WARNING: level = RTMP_LOGWARNING; break;
case AV_LOG_INFO: level = RTMP_LOGINFO; break;
case AV_LOG_VERBOSE: level = RTMP_LOGDEBUG; break;
case AV_LOG_DEBUG: level = RTMP_LOGDEBUG2; break;
}
RTMP_LogSetLevel(level);
RTMP_LogSetCallback(rtmp_log);
if (ctx->app || ctx->playpath) {
int len = strlen(s->filename) + 1;
if (ctx->app) len += strlen(ctx->app) + sizeof(" app=");
if (ctx->playpath) len += strlen(ctx->playpath) + sizeof(" playpath=");
if (!(ctx->temp_filename = filename = av_malloc(len)))
return AVERROR(ENOMEM);
av_strlcpy(filename, s->filename, len);
if (ctx->app) {
av_strlcat(filename, " app=", len);
av_strlcat(filename, ctx->app, len);
}
if (ctx->playpath) {
av_strlcat(filename, " playpath=", len);
av_strlcat(filename, ctx->playpath, len);
}
}
RTMP_Init(r);
if (!RTMP_SetupURL(r, filename)) {
rc = AVERROR_UNKNOWN;
goto fail;
}
if (flags & AVIO_FLAG_WRITE)
RTMP_EnableWrite(r);
if (!RTMP_Connect(r, NULL) || !RTMP_ConnectStream(r, 0)) {
rc = AVERROR_UNKNOWN;
goto fail;
}
s->is_streamed = 1;
return 0;
fail:
av_freep(&ctx->temp_filename);
if (rc)
RTMP_Close(r);
return rc;
}
int rtmp_open(LibRTMPContext *ctx , int flags)
{
RTMP *r = &ctx->rtmp;
int rc = 0;
char *filename = NULL;
if(ctx->filename==NULL)
return -1;
int len = strlen(ctx->filename) + 1;
RTMP_LogSetLevel(RTMP_LOGERROR);
RTMP_LogSetCallback(rtmp_log);
if (ctx->app) len += strlen(ctx->app) + sizeof(" app=");
if (ctx->tcurl) len += strlen(ctx->tcurl) + sizeof(" tcUrl=");
if (ctx->pageurl) len += strlen(ctx->pageurl) + sizeof(" pageUrl=");
if (ctx->flashver) len += strlen(ctx->flashver) + sizeof(" flashver=");
if (ctx->conn) {
char *sep, *p = ctx->conn;
int options = 0;
while (p) {
options++;
p += strspn(p, " ");
if (!*p)
break;
sep = strchr(p, ' ');
if (sep)
p = sep + 1;
else
break;
}
len += options * sizeof(" conn=");
len += strlen(ctx->conn);
}
if (ctx->playpath)
len += strlen(ctx->playpath) + sizeof(" playpath=");
if (ctx->live)
len += sizeof(" live=1");
if (ctx->subscribe)
len += strlen(ctx->subscribe) + sizeof(" subscribe=");
if (ctx->client_buffer_time)
len += strlen(ctx->client_buffer_time) + sizeof(" buffer=");
if (ctx->swfurl || ctx->swfverify) {
len += sizeof(" swfUrl=");
if (ctx->swfverify)
len += strlen(ctx->swfverify) + sizeof(" swfVfy=1");
else
len += strlen(ctx->swfurl);
}
if (!(ctx->temp_filename = filename = malloc(len)))
return ERROR_MEM;
strlcpy(filename, ctx->filename, len);
if (ctx->app) {
strlcat(filename, " app=", len);
strlcat(filename, ctx->app, len);
}
if (ctx->tcurl) {
strlcat(filename, " tcUrl=", len);
strlcat(filename, ctx->tcurl, len);
}
if (ctx->pageurl) {
strlcat(filename, " pageUrl=", len);
strlcat(filename, ctx->pageurl, len);
}
if (ctx->swfurl) {
strlcat(filename, " swfUrl=", len);
strlcat(filename, ctx->swfurl, len);
}
if (ctx->flashver) {
strlcat(filename, " flashVer=", len);
strlcat(filename, ctx->flashver, len);
}
if (ctx->conn) {
char *sep, *p = ctx->conn;
while (p) {
strlcat(filename, " conn=", len);
p += strspn(p, " ");
if (!*p)
break;
sep = strchr(p, ' ');
if (sep)
*sep = '\0';
strlcat(filename, p, len);
if (sep)
p = sep + 1;
}
}
if (ctx->playpath) {
strlcat(filename, " playpath=", len);
strlcat(filename, ctx->playpath, len);
}
if (ctx->live)
strlcat(filename, " live=1", len);
if (ctx->subscribe) {
strlcat(filename, " subscribe=", len);
strlcat(filename, ctx->subscribe, len);
}
if (ctx->client_buffer_time) {
strlcat(filename, " buffer=", len);
strlcat(filename, ctx->client_buffer_time, len);
}
if (ctx->swfurl || ctx->swfverify) {
strlcat(filename, " swfUrl=", len);
if (ctx->swfverify) {
strlcat(filename, ctx->swfverify, len);
strlcat(filename, " swfVfy=1", len);
} else {
strlcat(filename, ctx->swfurl, len);
}
}
RTMP_Init(r);
if (!RTMP_SetupURL(r, filename)) {
rc = ERROR_SETUP;
goto fail;
}
// if (flags & WRITE_FLAGS)
RTMP_EnableWrite(r);
if (!RTMP_Connect(r, NULL) || !RTMP_ConnectStream(r, 0)) {
rc = ERROR_CONNECT;
goto fail;
}
/*
if (ctx->buffer_size >= 0 && (flags & WRITE_FLAGS)) {
int tmp = ctx->buffer_size;
setsockopt(r->m_sb.sb_socket, SOL_SOCKET, SO_SNDBUF, &tmp, sizeof(tmp));
}
*/
return 0;
fail:
//free(ctx->temp_filename);
if (rc)
RTMP_Close(r);
return rc;
}
