mirror of
https://github.com/ossrs/srs.git
synced 2025-02-12 11:21:52 +00:00
add ingest flv to RTMP over srs-librtmp
This commit is contained in:
winlin
parent
f17f5de7d1
commit
b1835ee2fe
2 changed files with 254 additions and 0 deletions
|
|
@ -28,9 +28,262 @@ gcc srs_ingest_flv.c ../../objs/lib/srs_librtmp.a -g -O0 -lstdc++ -o srs_ingest_
|
|||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <fcntl.h>
|
||||
|
||||
#include "../../objs/include/srs_librtmp.h"
|
||||
|
||||
#define trace(msg, ...) printf(msg, ##__VA_ARGS__);printf("\n")
|
||||
#define verbose(msg, ...) printf(msg, ##__VA_ARGS__);printf("\n")
|
||||
#if 1
|
||||
#undef verbose
|
||||
#define verbose(msg, ...) (void)0
|
||||
#endif
|
||||
|
||||
int proxy(int flv_fd, srs_rtmp_t ortmp);
|
||||
int connect_oc(srs_rtmp_t ortmp);
|
||||
|
||||
int open_flv_file(char* in_flv_file);
|
||||
void close_flv_file(int flv_fd);
|
||||
int flv_open_ic(int flv_fd);
|
||||
int flv_read_packet(int flv_fd, int* type, u_int32_t* timestamp, char** data, int* size);
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
// user option parse index.
|
||||
int opt = 0;
|
||||
// user options.
|
||||
char* in_flv_file; char* out_rtmp_url;
|
||||
// rtmp handler
|
||||
srs_rtmp_t ortmp;
|
||||
// flv handler
|
||||
int flv_fd;
|
||||
|
||||
if (argc <= 2) {
|
||||
printf("ingest flv file and publish to RTMP server\n"
|
||||
"Usage: %s <-i in_flv_file> <-y out_rtmp_url>\n"
|
||||
" in_flv_file input flv file, ingest from this file.\n"
|
||||
" out_rtmp_url output rtmp url, publish to this url.\n"
|
||||
"For example:\n"
|
||||
" %s -i ../../doc/source.200kbps.768x320.flv -y rtmp://127.0.0.1/live/demo\n",
|
||||
argv[0]);
|
||||
ret = 1;
|
||||
exit(ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// parse options in FFMPEG format.
|
||||
while ((opt = getopt(argc, argv, "i:y:")) != -1) {
|
||||
switch (opt) {
|
||||
case 'i':
|
||||
in_flv_file = optarg;
|
||||
break;
|
||||
case 'y':
|
||||
out_rtmp_url = optarg;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
trace("ingest flv file and publish to RTMP server like FFMPEG.");
|
||||
trace("srs(simple-rtmp-server) client librtmp library.");
|
||||
trace("version: %d.%d.%d", srs_version_major(), srs_version_minor(), srs_version_revision());
|
||||
trace("input: %s", in_flv_file);
|
||||
trace("output: %s", out_rtmp_url);
|
||||
|
||||
flv_fd = open_flv_file(in_flv_file);
|
||||
if (flv_fd <= 0) {
|
||||
ret = 2;
|
||||
trace("open flv file failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
ortmp = srs_rtmp_create(out_rtmp_url);
|
||||
|
||||
ret = proxy(flv_fd, ortmp);
|
||||
trace("proxy completed");
|
||||
|
||||
srs_rtmp_destroy(ortmp);
|
||||
close_flv_file(flv_fd);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int64_t re_create()
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
int64_t re_update(int64_t re, u_int32_t time)
|
||||
{
|
||||
if (time - re > 500) {
|
||||
usleep((time - re) * 1000);
|
||||
return time;
|
||||
}
|
||||
|
||||
return re;
|
||||
}
|
||||
|
||||
int proxy(int flv_fd, srs_rtmp_t ortmp)
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
// packet data
|
||||
int type, size;
|
||||
u_int32_t timestamp = 0;
|
||||
char* data = NULL;
|
||||
// re
|
||||
int64_t re = re_create();
|
||||
|
||||
if ((ret = flv_open_ic(flv_fd)) != 0) {
|
||||
return ret;
|
||||
}
|
||||
if ((ret = connect_oc(ortmp)) != 0) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
trace("start proxy RTMP stream");
|
||||
for (;;) {
|
||||
if ((ret = flv_read_packet(flv_fd, &type, ×tamp, &data, &size)) != 0) {
|
||||
trace("irtmp get packet failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
verbose("irtmp got packet: type=%s, time=%d, size=%d",
|
||||
srs_type2string(type), timestamp, size);
|
||||
|
||||
if ((ret = srs_write_packet(ortmp, type, timestamp, data, size)) != 0) {
|
||||
trace("irtmp get packet failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
verbose("ortmp sent packet: type=%s, time=%d, size=%d",
|
||||
srs_type2string(type), timestamp, size);
|
||||
|
||||
re = re_update(re, timestamp);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int connect_oc(srs_rtmp_t ortmp)
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
if ((ret = srs_simple_handshake(ortmp)) != 0) {
|
||||
trace("ortmp simple handshake failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
trace("ortmp simple handshake success");
|
||||
|
||||
if ((ret = srs_connect_app(ortmp)) != 0) {
|
||||
trace("ortmp connect vhost/app failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
trace("ortmp connect vhost/app success");
|
||||
|
||||
if ((ret = srs_publish_stream(ortmp)) != 0) {
|
||||
trace("ortmp publish stream failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
trace("ortmp publish stream success");
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int open_flv_file(char* in_flv_file)
|
||||
{
|
||||
return open(in_flv_file, O_RDONLY);
|
||||
}
|
||||
|
||||
void close_flv_file(int fd)
|
||||
{
|
||||
if (fd > 0) {
|
||||
close(fd);
|
||||
}
|
||||
}
|
||||
|
||||
int flv_open_ic(int flv_fd)
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
char h[13]; // 9+4
|
||||
|
||||
if (read(flv_fd, h, sizeof(h)) != sizeof(h)) {
|
||||
ret = -1;
|
||||
trace("read flv header failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
if (h[0] != 'F' || h[1] != 'L' || h[2] != 'V') {
|
||||
ret = -1;
|
||||
trace("input is not a flv file. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int flv_read_packet(int flv_fd, int* type, u_int32_t* timestamp, char** data, int* size)
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
char th[11]; // tag header
|
||||
char ts[4]; // tag size
|
||||
|
||||
u_int32_t data_size;
|
||||
u_int32_t time;
|
||||
|
||||
char* pp;
|
||||
|
||||
// read tag header
|
||||
if (read(flv_fd, th, sizeof(th)) != sizeof(th)) {
|
||||
ret = -1;
|
||||
trace("read flv tag header failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// Reserved UB [2]
|
||||
// Filter UB [1]
|
||||
// TagType UB [5]
|
||||
*type = (int)(th[0] & 0x1F);
|
||||
|
||||
// DataSize UI24
|
||||
pp = (char*)&data_size;
|
||||
pp[2] = th[1];
|
||||
pp[1] = th[2];
|
||||
pp[0] = th[3];
|
||||
|
||||
// Timestamp UI24
|
||||
pp = (char*)&time;
|
||||
pp[2] = th[4];
|
||||
pp[1] = th[5];
|
||||
pp[0] = th[6];
|
||||
|
||||
// TimestampExtended UI8
|
||||
pp[3] = th[7];
|
||||
|
||||
*timestamp = time;
|
||||
|
||||
if (data_size > 0) {
|
||||
*size = data_size;
|
||||
*data = (char*)malloc(data_size);
|
||||
|
||||
// read tag data
|
||||
if (read(flv_fd, *data, data_size) != data_size) {
|
||||
ret = -1;
|
||||
trace("read flv tag data failed. size=%d, ret=%d", data_size, ret);
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
|
||||
// ignore 4bytes tag size.
|
||||
if (read(flv_fd, ts, sizeof(ts)) != sizeof(ts)) {
|
||||
ret = -1;
|
||||
trace("read flv tag size failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -44,6 +44,7 @@ int proxy(srs_rtmp_t irtmp, srs_rtmp_t ortmp);
|
|||
int main(int argc, char** argv)
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
// user option parse index.
|
||||
int opt = 0;
|
||||
// user options.
|
||||
|
|
|
|||
Loading…
Reference in a new issue