ExilProductions/rtmp-cpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed conn limits and server shutdown

Browse Source
This commit is contained in:
Exil Productions
2025-12-27 00:00:58 +01:00
parent fad6425f44
commit 1ad0bccb2b
2 changed files with 104 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

96
example/main.c
View File

@@ -1,44 +1,106 @@
#include "../include/rtmp_capi.hpp"
#include <stdio.h>
#include <unistd.h>
#include <stdint.h>
#include <stdbool.h>
#include <sys/select.h>
#include <termios.h>
#include <fcntl.h>
#include <stdlib.h>
// Callbacks
static void on_connect_cb(const char* ip, void* data)
{
printf("Client connected: %s\n", ip);
}
static void on_publish_cb(const char* ip, const char* app, const char* key,
void *data)
static void on_publish_cb(const char* ip, const char* app, const char* key, void* data)
{
printf("Publish from %s: %s/%s\n", ip, app, key);
}
static void on_audio_cb(const char* app, const char* key, const uint8_t* data,
uint32_t len, uint32_t ts, void *ud)
static struct termios g_orig_termios;
static void restore_terminal(void)
{
printf("Audio data for %s/%s, len=%u, ts=%u\n", app, key, len, ts);
tcsetattr(STDIN_FILENO, TCSANOW, &g_orig_termios);
}
int main()
static int setup_nonblocking_stdin(void)
{
struct termios raw;
int flags;
if (tcgetattr(STDIN_FILENO, &g_orig_termios) != 0)
return 0;
raw = g_orig_termios;
raw.c_lflag &= ~(ICANON | ECHO);
if (tcsetattr(STDIN_FILENO, TCSANOW, &raw) != 0)
return 0;
flags = fcntl(STDIN_FILENO, F_GETFL, 0);
if (flags < 0)
return 0;
if (fcntl(STDIN_FILENO, F_SETFL, flags | O_NONBLOCK) < 0)
return 0;
atexit(restore_terminal);
return 1;
}
int main(void)
{
rtmp_logger_set_level(RTMP_LOG_INFO);
if (!setup_nonblocking_stdin()) {
printf("Failed to configure terminal input\n");
return 1;
}
RtmpServerHandle server = rtmp_server_create(1935);
if (!server) {
printf("Failed to create server\n");
return 1;
}
rtmp_server_set_on_connect(server, on_connect_cb, NULL);
rtmp_server_set_on_publish(server, on_publish_cb, NULL);
rtmp_server_set_on_audio_data(server, on_audio_cb, NULL);
rtmp_server_enable_gop_cache(server, true);
bool isRunning = false;
if (rtmp_server_start(server, &isRunning))
{
while (isRunning) {
sleep(1); // Block main Thread
if (!rtmp_server_start(server, &isRunning)) {
printf("Failed to start server\n");
rtmp_server_destroy(server);
return 1;
}
printf("RTMP server running. Press 'q' to stop.\n");
while (isRunning) {
fd_set readfds;
struct timeval tv;
FD_ZERO(&readfds);
FD_SET(STDIN_FILENO, &readfds);
tv.tv_sec = 1;
tv.tv_usec = 0;
int ret = select(STDIN_FILENO + 1, &readfds, NULL, NULL, &tv);
if (ret > 0 && FD_ISSET(STDIN_FILENO, &readfds)) {
char ch;
ssize_t n = read(STDIN_FILENO, &ch, 1);
if (n == 1 && (ch == 'q' || ch == 'Q')) {
printf("Shutting down...\n");
rtmp_server_stop(server);
break;
}
}
}
else
{
printf("Failed to start server\n");
}
rtmp_server_stop(server);
rtmp_server_destroy(server);
return 0;
}
}