1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include "ezlive_config.h"
#include "srtserver.h"
#include "ringbuf.h"
#include "transcode_talker.h"
#include "s3_worker.h"
typedef struct {
RingBuffer *ringbuf;
TranscodeTalker transcode_talker;
} MainCtx;
void on_rtmp_start(void *ctx) {
MainCtx *main_ctx = ctx;
main_ctx->ringbuf = malloc(sizeof(RingBuffer));
RingBuffer_init(main_ctx->ringbuf, 4096);
RingBuffer *rb = main_ctx->ringbuf;
TranscodeTalker_new_stream(&main_ctx->transcode_talker, rb);
RingBuffer_write_char(rb, 'F');
RingBuffer_write_char(rb, 'L');
RingBuffer_write_char(rb, 'V');
RingBuffer_write_char(rb, 1);
RingBuffer_write_char(rb, 5);
RingBuffer_write_word32be(rb, 9);
RingBuffer_write_word32be(rb, 0);
}
void on_rtmp_stop(void *ctx) {
MainCtx *main_ctx = ctx;
RingBuffer_end(main_ctx->ringbuf);
}
void on_rtmp_video(void *ctx, int64_t timestamp, char *buf, size_t size) {
MainCtx *main_ctx = ctx;
RingBuffer *rb = main_ctx->ringbuf;
RingBuffer_write_char(rb, 9);
RingBuffer_write_word24be(rb, size);
RingBuffer_write_word24be(rb, timestamp);
RingBuffer_write_char(rb, timestamp >> 24);
RingBuffer_write_word24be(rb, 0);
RingBuffer_write(rb, (const uint8_t *)buf, size);
RingBuffer_write_word32be(rb, size + 11);
}
void on_rtmp_audio(void *ctx, int64_t timestamp, char *buf, size_t size) {
MainCtx *main_ctx = ctx;
RingBuffer *rb = main_ctx->ringbuf;
RingBuffer_write_char(rb, 8);
RingBuffer_write_word24be(rb, size);
RingBuffer_write_word24be(rb, timestamp);
RingBuffer_write_char(rb, timestamp >> 24);
RingBuffer_write_word24be(rb, 0);
RingBuffer_write(rb, (const uint8_t *)buf, size);
RingBuffer_write_word32be(rb, size + 11);
}
void on_srt_start(void *ctx) {
MainCtx *main_ctx = ctx;
main_ctx->ringbuf = malloc(sizeof(RingBuffer));
RingBuffer_init(main_ctx->ringbuf, 4096);
RingBuffer *rb = main_ctx->ringbuf;
TranscodeTalker_new_stream(&main_ctx->transcode_talker, rb);
}
void on_srt_stop(void *ctx) {
MainCtx *main_ctx = ctx;
RingBuffer_end(main_ctx->ringbuf);
}
void on_srt_data(void *ctx, char *buf, size_t size) {
MainCtx *main_ctx = ctx;
RingBuffer *rb = main_ctx->ringbuf;
RingBuffer_write(rb, (const uint8_t*)buf, size);
}
int main(int argc, char **argv) {
ezlive_config = malloc(sizeof(EZLiveConfig));
EZLiveConfig_init(ezlive_config);
bool succ;
if (argc == 1) {
succ = EZLiveConfig_load(ezlive_config, "./config");
#if defined(_WIN32)
if (!succ) {
succ = EZLiveConfig_load(ezlive_config, "./config.txt");
}
#endif
if (!succ) {
fprintf(stderr, "Failed to load config.\n");
return -1;
}
} else if (argc == 2) {
if (!EZLiveConfig_load(ezlive_config, argv[1])) {
fprintf(stderr, "Failed to load config.\n");
return -1;
}
} else {
fprintf(stderr, "wrong args.\n");
exit(-1);
}
int ret;
if ((ret = EZLiveConfig_validate(ezlive_config)) < 0) {
fprintf(stderr, "ezlive config error: %d.\n", ret);
exit(-1);
}
srand((unsigned) time(NULL));
MainCtx main_ctx;
SrtCallbacks srt_cbs = {
.on_data = &on_srt_data,
.on_stop = &on_srt_stop,
.on_start = &on_srt_start,
};
TranscodeTalker_init(&main_ctx.transcode_talker);
s3_worker_init();
s3_worker_push(s3_clear_task());
pthread_t transmux_thread;
pthread_create(&transmux_thread, NULL, &TranscodeTalker_main, &main_ctx.transcode_talker);
pthread_t s3worker_thread;
pthread_create(&s3worker_thread, NULL, &s3_worker_main, NULL);
// start_rtmpserver(rtmp_cbs, &main_ctx);
start_srt_server(srt_cbs, &main_ctx);
return 0;
}
|
