You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1605 lines
41 KiB
1605 lines
41 KiB
5 years ago
|
|
||
|
/*
|
||
|
* Copyright (C) Roman Arutyunyan
|
||
|
*/
|
||
|
|
||
|
|
||
|
#include <ngx_config.h>
|
||
|
#include <ngx_core.h>
|
||
|
#include "ngx_rtmp_cmd_module.h"
|
||
|
#include "ngx_rtmp_record_module.h"
|
||
|
#include "ngx_rtmp_eval.h"
|
||
|
#include <stdlib.h>
|
||
|
|
||
|
#ifdef NGX_LINUX
|
||
|
#include <unistd.h>
|
||
|
#endif
|
||
|
|
||
|
|
||
|
#if !(NGX_WIN32)
|
||
|
static ngx_rtmp_publish_pt next_publish;
|
||
|
static ngx_rtmp_play_pt next_play;
|
||
|
static ngx_rtmp_close_stream_pt next_close_stream;
|
||
|
static ngx_rtmp_record_done_pt next_record_done;
|
||
|
#endif
|
||
|
|
||
|
|
||
|
static ngx_int_t ngx_rtmp_exec_init_process(ngx_cycle_t *cycle);
|
||
|
static ngx_int_t ngx_rtmp_exec_postconfiguration(ngx_conf_t *cf);
|
||
|
static void * ngx_rtmp_exec_create_main_conf(ngx_conf_t *cf);
|
||
|
static char * ngx_rtmp_exec_init_main_conf(ngx_conf_t *cf, void *conf);
|
||
|
static void * ngx_rtmp_exec_create_app_conf(ngx_conf_t *cf);
|
||
|
static char * ngx_rtmp_exec_merge_app_conf(ngx_conf_t *cf,
|
||
|
void *parent, void *child);
|
||
|
/*static char * ngx_rtmp_exec_block(ngx_conf_t *cf, ngx_command_t *cmd,
|
||
|
void *conf);*/
|
||
|
static char * ngx_rtmp_exec_conf(ngx_conf_t *cf, ngx_command_t *cmd,
|
||
|
void *conf);
|
||
|
static char *ngx_rtmp_exec_kill_signal(ngx_conf_t *cf, ngx_command_t *cmd,
|
||
|
void *conf);
|
||
|
|
||
|
|
||
|
#define NGX_RTMP_EXEC_RESPAWN 0x01
|
||
|
#define NGX_RTMP_EXEC_KILL 0x02
|
||
|
|
||
|
|
||
|
#define NGX_RTMP_EXEC_PUBLISHING 0x01
|
||
|
#define NGX_RTMP_EXEC_PLAYING 0x02
|
||
|
|
||
|
|
||
|
enum {
|
||
|
NGX_RTMP_EXEC_PUSH,
|
||
|
NGX_RTMP_EXEC_PULL,
|
||
|
|
||
|
NGX_RTMP_EXEC_PUBLISH,
|
||
|
NGX_RTMP_EXEC_PUBLISH_DONE,
|
||
|
NGX_RTMP_EXEC_PLAY,
|
||
|
NGX_RTMP_EXEC_PLAY_DONE,
|
||
|
NGX_RTMP_EXEC_RECORD_DONE,
|
||
|
|
||
|
NGX_RTMP_EXEC_MAX,
|
||
|
|
||
|
NGX_RTMP_EXEC_STATIC
|
||
|
};
|
||
|
|
||
|
|
||
|
typedef struct {
|
||
|
ngx_str_t id;
|
||
|
ngx_uint_t type;
|
||
|
ngx_str_t cmd;
|
||
|
ngx_array_t args; /* ngx_str_t */
|
||
|
ngx_array_t names;
|
||
|
} ngx_rtmp_exec_conf_t;
|
||
|
|
||
|
|
||
|
typedef struct {
|
||
|
ngx_rtmp_exec_conf_t *conf;
|
||
|
ngx_log_t *log;
|
||
|
ngx_rtmp_eval_t **eval;
|
||
|
void *eval_ctx;
|
||
|
unsigned active:1;
|
||
|
unsigned managed:1;
|
||
|
ngx_pid_t pid;
|
||
|
ngx_pid_t *save_pid;
|
||
|
int pipefd;
|
||
|
ngx_connection_t dummy_conn; /*needed by ngx_xxx_event*/
|
||
|
ngx_event_t read_evt, write_evt;
|
||
|
ngx_event_t respawn_evt;
|
||
|
ngx_msec_t respawn_timeout;
|
||
|
ngx_int_t kill_signal;
|
||
|
} ngx_rtmp_exec_t;
|
||
|
|
||
|
|
||
|
typedef struct {
|
||
|
ngx_array_t static_conf; /* ngx_rtmp_exec_conf_t */
|
||
|
ngx_array_t static_exec; /* ngx_rtmp_exec_t */
|
||
|
ngx_msec_t respawn_timeout;
|
||
|
ngx_int_t kill_signal;
|
||
|
ngx_log_t *log;
|
||
|
} ngx_rtmp_exec_main_conf_t;
|
||
|
|
||
|
|
||
|
typedef struct ngx_rtmp_exec_pull_ctx_s ngx_rtmp_exec_pull_ctx_t;
|
||
|
|
||
|
struct ngx_rtmp_exec_pull_ctx_s {
|
||
|
ngx_pool_t *pool;
|
||
|
ngx_uint_t counter;
|
||
|
ngx_str_t name;
|
||
|
ngx_str_t app;
|
||
|
ngx_array_t pull_exec; /* ngx_rtmp_exec_t */
|
||
|
ngx_rtmp_exec_pull_ctx_t *next;
|
||
|
};
|
||
|
|
||
|
|
||
|
typedef struct {
|
||
|
ngx_int_t active;
|
||
|
ngx_array_t conf[NGX_RTMP_EXEC_MAX];
|
||
|
/* ngx_rtmp_exec_conf_t */
|
||
|
ngx_flag_t respawn;
|
||
|
ngx_flag_t options;
|
||
|
ngx_uint_t nbuckets;
|
||
|
ngx_rtmp_exec_pull_ctx_t **pull;
|
||
|
} ngx_rtmp_exec_app_conf_t;
|
||
|
|
||
|
|
||
|
typedef struct {
|
||
|
ngx_uint_t flags;
|
||
|
ngx_str_t path; /* /tmp/rec/myfile-123.flv */
|
||
|
ngx_str_t filename; /* myfile-123.flv */
|
||
|
ngx_str_t basename; /* myfile-123 */
|
||
|
ngx_str_t dirname; /* /tmp/rec */
|
||
|
ngx_str_t recorder;
|
||
|
u_char name[NGX_RTMP_MAX_NAME];
|
||
|
u_char args[NGX_RTMP_MAX_ARGS];
|
||
|
ngx_array_t push_exec; /* ngx_rtmp_exec_t */
|
||
|
ngx_rtmp_exec_pull_ctx_t *pull;
|
||
|
} ngx_rtmp_exec_ctx_t;
|
||
|
|
||
|
|
||
|
#if !(NGX_WIN32)
|
||
|
static void ngx_rtmp_exec_respawn(ngx_event_t *ev);
|
||
|
static ngx_int_t ngx_rtmp_exec_kill(ngx_rtmp_exec_t *e, ngx_int_t kill_signal);
|
||
|
static ngx_int_t ngx_rtmp_exec_run(ngx_rtmp_exec_t *e);
|
||
|
#endif
|
||
|
|
||
|
|
||
|
static ngx_command_t ngx_rtmp_exec_commands[] = {
|
||
|
/*
|
||
|
{ ngx_string("exec_block"),
|
||
|
NGX_RTMP_APP_CONF|NGX_CONF_BLOCK|NGX_CONF_NOARGS|NGX_CONF_TAKE1,
|
||
|
ngx_rtmp_exec_block,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
0,
|
||
|
NULL },
|
||
|
*/
|
||
|
{ ngx_string("exec"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, conf) +
|
||
|
NGX_RTMP_EXEC_PUSH * sizeof(ngx_array_t),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_push"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, conf) +
|
||
|
NGX_RTMP_EXEC_PUSH * sizeof(ngx_array_t),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_pull"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, conf) +
|
||
|
NGX_RTMP_EXEC_PULL * sizeof(ngx_array_t),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_publish"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, conf) +
|
||
|
NGX_RTMP_EXEC_PUBLISH * sizeof(ngx_array_t),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_publish_done"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, conf) +
|
||
|
NGX_RTMP_EXEC_PUBLISH_DONE * sizeof(ngx_array_t),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_play"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, conf) +
|
||
|
NGX_RTMP_EXEC_PLAY * sizeof(ngx_array_t),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_play_done"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, conf) +
|
||
|
NGX_RTMP_EXEC_PLAY_DONE * sizeof(ngx_array_t),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_record_done"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_RTMP_REC_CONF|
|
||
|
NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, conf) +
|
||
|
NGX_RTMP_EXEC_RECORD_DONE * sizeof(ngx_array_t),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_static"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_1MORE,
|
||
|
ngx_rtmp_exec_conf,
|
||
|
NGX_RTMP_MAIN_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_main_conf_t, static_conf),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("respawn"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_TAKE1,
|
||
|
ngx_conf_set_flag_slot,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, respawn),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("respawn_timeout"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_TAKE1,
|
||
|
ngx_conf_set_msec_slot,
|
||
|
NGX_RTMP_MAIN_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_main_conf_t, respawn_timeout),
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_kill_signal"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_TAKE1,
|
||
|
ngx_rtmp_exec_kill_signal,
|
||
|
NGX_RTMP_MAIN_CONF_OFFSET,
|
||
|
0,
|
||
|
NULL },
|
||
|
|
||
|
{ ngx_string("exec_options"),
|
||
|
NGX_RTMP_MAIN_CONF|NGX_RTMP_SRV_CONF|NGX_RTMP_APP_CONF|NGX_CONF_TAKE1,
|
||
|
ngx_conf_set_flag_slot,
|
||
|
NGX_RTMP_APP_CONF_OFFSET,
|
||
|
offsetof(ngx_rtmp_exec_app_conf_t, options),
|
||
|
NULL },
|
||
|
|
||
|
ngx_null_command
|
||
|
};
|
||
|
|
||
|
|
||
|
static ngx_rtmp_module_t ngx_rtmp_exec_module_ctx = {
|
||
|
NULL, /* preconfiguration */
|
||
|
ngx_rtmp_exec_postconfiguration, /* postconfiguration */
|
||
|
ngx_rtmp_exec_create_main_conf, /* create main configuration */
|
||
|
ngx_rtmp_exec_init_main_conf, /* init main configuration */
|
||
|
NULL, /* create server configuration */
|
||
|
NULL, /* merge server configuration */
|
||
|
ngx_rtmp_exec_create_app_conf, /* create app configuration */
|
||
|
ngx_rtmp_exec_merge_app_conf /* merge app configuration */
|
||
|
};
|
||
|
|
||
|
|
||
|
ngx_module_t ngx_rtmp_exec_module = {
|
||
|
NGX_MODULE_V1,
|
||
|
&ngx_rtmp_exec_module_ctx, /* module context */
|
||
|
ngx_rtmp_exec_commands, /* module directives */
|
||
|
NGX_RTMP_MODULE, /* module type */
|
||
|
NULL, /* init master */
|
||
|
NULL, /* init module */
|
||
|
ngx_rtmp_exec_init_process, /* init process */
|
||
|
NULL, /* init thread */
|
||
|
NULL, /* exit thread */
|
||
|
NULL, /* exit process */
|
||
|
NULL, /* exit master */
|
||
|
NGX_MODULE_V1_PADDING
|
||
|
};
|
||
|
|
||
|
|
||
|
static void
|
||
|
ngx_rtmp_exec_eval_ctx_cstr(void *sctx, ngx_rtmp_eval_t *e, ngx_str_t *ret)
|
||
|
{
|
||
|
ngx_rtmp_session_t *s = sctx;
|
||
|
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
if (ctx == NULL) {
|
||
|
ret->len = 0;
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
ret->data = (u_char *) ctx + e->offset;
|
||
|
ret->len = ngx_strlen(ret->data);
|
||
|
}
|
||
|
|
||
|
|
||
|
static void
|
||
|
ngx_rtmp_exec_eval_ctx_str(void *sctx, ngx_rtmp_eval_t *e, ngx_str_t *ret)
|
||
|
{
|
||
|
ngx_rtmp_session_t *s = sctx;
|
||
|
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
if (ctx == NULL) {
|
||
|
ret->len = 0;
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
*ret = * (ngx_str_t *) ((u_char *) ctx + e->offset);
|
||
|
}
|
||
|
|
||
|
|
||
|
static void
|
||
|
ngx_rtmp_exec_eval_pctx_str(void *ctx, ngx_rtmp_eval_t *e, ngx_str_t *ret)
|
||
|
{
|
||
|
*ret = *(ngx_str_t *) ((u_char *) ctx + e->offset);
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_rtmp_eval_t ngx_rtmp_exec_push_specific_eval[] = {
|
||
|
|
||
|
{ ngx_string("name"),
|
||
|
ngx_rtmp_exec_eval_ctx_cstr,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, name) },
|
||
|
|
||
|
{ ngx_string("args"),
|
||
|
ngx_rtmp_exec_eval_ctx_cstr,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, args) },
|
||
|
|
||
|
ngx_rtmp_null_eval
|
||
|
};
|
||
|
|
||
|
|
||
|
static ngx_rtmp_eval_t * ngx_rtmp_exec_push_eval[] = {
|
||
|
ngx_rtmp_eval_session,
|
||
|
ngx_rtmp_exec_push_specific_eval,
|
||
|
NULL
|
||
|
};
|
||
|
|
||
|
|
||
|
static ngx_rtmp_eval_t ngx_rtmp_exec_pull_specific_eval[] = {
|
||
|
|
||
|
{ ngx_string("name"),
|
||
|
ngx_rtmp_exec_eval_pctx_str,
|
||
|
offsetof(ngx_rtmp_exec_pull_ctx_t, name) },
|
||
|
|
||
|
{ ngx_string("app"),
|
||
|
ngx_rtmp_exec_eval_pctx_str,
|
||
|
offsetof(ngx_rtmp_exec_pull_ctx_t, app) },
|
||
|
|
||
|
ngx_rtmp_null_eval
|
||
|
};
|
||
|
|
||
|
|
||
|
static ngx_rtmp_eval_t * ngx_rtmp_exec_pull_eval[] = {
|
||
|
ngx_rtmp_exec_pull_specific_eval,
|
||
|
NULL
|
||
|
};
|
||
|
|
||
|
|
||
|
static ngx_rtmp_eval_t ngx_rtmp_exec_event_specific_eval[] = {
|
||
|
|
||
|
{ ngx_string("name"),
|
||
|
ngx_rtmp_exec_eval_ctx_cstr,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, name) },
|
||
|
|
||
|
{ ngx_string("args"),
|
||
|
ngx_rtmp_exec_eval_ctx_cstr,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, args) },
|
||
|
|
||
|
{ ngx_string("path"),
|
||
|
ngx_rtmp_exec_eval_ctx_str,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, path) },
|
||
|
|
||
|
{ ngx_string("filename"),
|
||
|
ngx_rtmp_exec_eval_ctx_str,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, filename) },
|
||
|
|
||
|
{ ngx_string("basename"),
|
||
|
ngx_rtmp_exec_eval_ctx_str,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, basename) },
|
||
|
|
||
|
{ ngx_string("dirname"),
|
||
|
ngx_rtmp_exec_eval_ctx_str,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, dirname) },
|
||
|
|
||
|
{ ngx_string("recorder"),
|
||
|
ngx_rtmp_exec_eval_ctx_str,
|
||
|
offsetof(ngx_rtmp_exec_ctx_t, recorder) },
|
||
|
|
||
|
ngx_rtmp_null_eval
|
||
|
};
|
||
|
|
||
|
|
||
|
static ngx_rtmp_eval_t * ngx_rtmp_exec_event_eval[] = {
|
||
|
ngx_rtmp_eval_session,
|
||
|
ngx_rtmp_exec_event_specific_eval,
|
||
|
NULL
|
||
|
};
|
||
|
|
||
|
|
||
|
static void *
|
||
|
ngx_rtmp_exec_create_main_conf(ngx_conf_t *cf)
|
||
|
{
|
||
|
ngx_rtmp_exec_main_conf_t *emcf;
|
||
|
|
||
|
emcf = ngx_pcalloc(cf->pool, sizeof(ngx_rtmp_exec_main_conf_t));
|
||
|
if (emcf == NULL) {
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
emcf->respawn_timeout = NGX_CONF_UNSET_MSEC;
|
||
|
emcf->kill_signal = NGX_CONF_UNSET;
|
||
|
|
||
|
if (ngx_array_init(&emcf->static_conf, cf->pool, 1,
|
||
|
sizeof(ngx_rtmp_exec_conf_t)) != NGX_OK)
|
||
|
{
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
return emcf;
|
||
|
}
|
||
|
|
||
|
|
||
|
static char *
|
||
|
ngx_rtmp_exec_init_main_conf(ngx_conf_t *cf, void *conf)
|
||
|
{
|
||
|
ngx_rtmp_exec_main_conf_t *emcf = conf;
|
||
|
ngx_rtmp_exec_conf_t *ec;
|
||
|
ngx_rtmp_exec_t *e;
|
||
|
ngx_uint_t n;
|
||
|
|
||
|
if (emcf->respawn_timeout == NGX_CONF_UNSET_MSEC) {
|
||
|
emcf->respawn_timeout = 5000;
|
||
|
}
|
||
|
|
||
|
#if !(NGX_WIN32)
|
||
|
if (emcf->kill_signal == NGX_CONF_UNSET) {
|
||
|
emcf->kill_signal = SIGKILL;
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
if (ngx_array_init(&emcf->static_exec, cf->pool,
|
||
|
emcf->static_conf.nelts,
|
||
|
sizeof(ngx_rtmp_exec_t)) != NGX_OK)
|
||
|
{
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
e = ngx_array_push_n(&emcf->static_exec, emcf->static_conf.nelts);
|
||
|
if (e == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
emcf->log = &cf->cycle->new_log;
|
||
|
|
||
|
ec = emcf->static_conf.elts;
|
||
|
|
||
|
for (n = 0; n < emcf->static_conf.nelts; n++, e++, ec++) {
|
||
|
ngx_memzero(e, sizeof(*e));
|
||
|
e->conf = ec;
|
||
|
e->managed = 1;
|
||
|
e->log = emcf->log;
|
||
|
e->respawn_timeout = emcf->respawn_timeout;
|
||
|
e->kill_signal = emcf->kill_signal;
|
||
|
}
|
||
|
|
||
|
return NGX_CONF_OK;
|
||
|
}
|
||
|
|
||
|
|
||
|
static void *
|
||
|
ngx_rtmp_exec_create_app_conf(ngx_conf_t *cf)
|
||
|
{
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
|
||
|
eacf = ngx_pcalloc(cf->pool, sizeof(ngx_rtmp_exec_app_conf_t));
|
||
|
if (eacf == NULL) {
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
eacf->respawn = NGX_CONF_UNSET;
|
||
|
eacf->options = NGX_CONF_UNSET;
|
||
|
eacf->nbuckets = NGX_CONF_UNSET_UINT;
|
||
|
|
||
|
return eacf;
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_merge_confs(ngx_array_t *conf, ngx_array_t *prev)
|
||
|
{
|
||
|
size_t n;
|
||
|
ngx_rtmp_exec_conf_t *ec, *pec;
|
||
|
|
||
|
if (prev->nelts == 0) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
if (conf->nelts == 0) {
|
||
|
*conf = *prev;
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
ec = ngx_array_push_n(conf, prev->nelts);
|
||
|
if (ec == NULL) {
|
||
|
return NGX_ERROR;
|
||
|
}
|
||
|
|
||
|
pec = prev->elts;
|
||
|
for (n = 0; n < prev->nelts; n++, ec++, pec++) {
|
||
|
*ec = *pec;
|
||
|
}
|
||
|
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
|
||
|
static char *
|
||
|
ngx_rtmp_exec_merge_app_conf(ngx_conf_t *cf, void *parent, void *child)
|
||
|
{
|
||
|
ngx_rtmp_exec_app_conf_t *prev = parent;
|
||
|
ngx_rtmp_exec_app_conf_t *conf = child;
|
||
|
|
||
|
ngx_uint_t n;
|
||
|
|
||
|
ngx_conf_merge_value(conf->respawn, prev->respawn, 1);
|
||
|
ngx_conf_merge_uint_value(conf->nbuckets, prev->nbuckets, 1024);
|
||
|
|
||
|
for (n = 0; n < NGX_RTMP_EXEC_MAX; n++) {
|
||
|
if (ngx_rtmp_exec_merge_confs(&conf->conf[n], &prev->conf[n]) != NGX_OK)
|
||
|
{
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
if (conf->conf[n].nelts) {
|
||
|
conf->active = 1;
|
||
|
prev->active = 1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (conf->conf[NGX_RTMP_EXEC_PULL].nelts > 0) {
|
||
|
conf->pull = ngx_pcalloc(cf->pool, sizeof(void *) * conf->nbuckets);
|
||
|
if (conf->pull == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return NGX_CONF_OK;
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_init_process(ngx_cycle_t *cycle)
|
||
|
{
|
||
|
#if !(NGX_WIN32)
|
||
|
ngx_rtmp_core_main_conf_t *cmcf = ngx_rtmp_core_main_conf;
|
||
|
ngx_rtmp_core_srv_conf_t **cscf;
|
||
|
ngx_rtmp_conf_ctx_t *cctx;
|
||
|
ngx_rtmp_exec_main_conf_t *emcf;
|
||
|
ngx_rtmp_exec_t *e;
|
||
|
ngx_uint_t n;
|
||
|
|
||
|
if (cmcf == NULL || cmcf->servers.nelts == 0) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
/* execs are always started by the first worker */
|
||
|
if (ngx_process_slot) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
cscf = cmcf->servers.elts;
|
||
|
cctx = (*cscf)->ctx;
|
||
|
emcf = cctx->main_conf[ngx_rtmp_exec_module.ctx_index];
|
||
|
|
||
|
/* FreeBSD note:
|
||
|
* When worker is restarted, child process (ffmpeg) will
|
||
|
* not be terminated if it's connected to another
|
||
|
* (still alive) worker. That leads to starting
|
||
|
* another instance of exec_static process.
|
||
|
* Need to kill previously started processes.
|
||
|
*
|
||
|
* On Linux "prctl" syscall is used to kill child
|
||
|
* when nginx worker is terminated.
|
||
|
*/
|
||
|
|
||
|
e = emcf->static_exec.elts;
|
||
|
for (n = 0; n < emcf->static_exec.nelts; ++n, ++e) {
|
||
|
e->respawn_evt.data = e;
|
||
|
e->respawn_evt.log = e->log;
|
||
|
e->respawn_evt.handler = ngx_rtmp_exec_respawn;
|
||
|
ngx_post_event((&e->respawn_evt), &ngx_rtmp_init_queue);
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
|
||
|
#if !(NGX_WIN32)
|
||
|
static void
|
||
|
ngx_rtmp_exec_respawn(ngx_event_t *ev)
|
||
|
{
|
||
|
ngx_rtmp_exec_run((ngx_rtmp_exec_t *) ev->data);
|
||
|
}
|
||
|
|
||
|
|
||
|
static void
|
||
|
ngx_rtmp_exec_child_dead(ngx_event_t *ev)
|
||
|
{
|
||
|
ngx_connection_t *dummy_conn = ev->data;
|
||
|
ngx_rtmp_exec_t *e;
|
||
|
|
||
|
e = dummy_conn->data;
|
||
|
|
||
|
ngx_log_error(NGX_LOG_INFO, e->log, 0,
|
||
|
"exec: child %ui exited; %s", (ngx_int_t) e->pid,
|
||
|
e->respawn_timeout == NGX_CONF_UNSET_MSEC ? "respawning" :
|
||
|
"ignoring");
|
||
|
|
||
|
ngx_rtmp_exec_kill(e, 0);
|
||
|
|
||
|
if (e->respawn_timeout == NGX_CONF_UNSET_MSEC) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (e->respawn_timeout == 0) {
|
||
|
ngx_rtmp_exec_run(e);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, e->log, 0,
|
||
|
"exec: shedule respawn %Mmsec", e->respawn_timeout);
|
||
|
|
||
|
e->respawn_evt.data = e;
|
||
|
e->respawn_evt.log = e->log;
|
||
|
e->respawn_evt.handler = ngx_rtmp_exec_respawn;
|
||
|
|
||
|
ngx_add_timer(&e->respawn_evt, e->respawn_timeout);
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_kill(ngx_rtmp_exec_t *e, ngx_int_t kill_signal)
|
||
|
{
|
||
|
if (e->respawn_evt.timer_set) {
|
||
|
ngx_del_timer(&e->respawn_evt);
|
||
|
}
|
||
|
|
||
|
if (e->read_evt.active) {
|
||
|
ngx_del_event(&e->read_evt, NGX_READ_EVENT, 0);
|
||
|
}
|
||
|
|
||
|
if (e->active == 0) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
ngx_log_error(NGX_LOG_INFO, e->log, 0,
|
||
|
"exec: terminating child %ui", (ngx_int_t) e->pid);
|
||
|
|
||
|
e->active = 0;
|
||
|
close(e->pipefd);
|
||
|
if (e->save_pid) {
|
||
|
*e->save_pid = NGX_INVALID_PID;
|
||
|
}
|
||
|
|
||
|
if (kill_signal == 0) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
if (kill(e->pid, kill_signal) == -1) {
|
||
|
ngx_log_error(NGX_LOG_INFO, e->log, ngx_errno,
|
||
|
"exec: kill failed pid=%i", (ngx_int_t) e->pid);
|
||
|
} else {
|
||
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, e->log, 0,
|
||
|
"exec: killed pid=%i", (ngx_int_t) e->pid);
|
||
|
}
|
||
|
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_run(ngx_rtmp_exec_t *e)
|
||
|
{
|
||
|
int fd, ret, maxfd, pipefd[2];
|
||
|
char **args, **arg_out;
|
||
|
ngx_pid_t pid;
|
||
|
ngx_str_t *arg_in, a;
|
||
|
ngx_uint_t n;
|
||
|
ngx_rtmp_exec_conf_t *ec;
|
||
|
|
||
|
ec = e->conf;
|
||
|
|
||
|
ngx_log_error(NGX_LOG_INFO, e->log, 0,
|
||
|
"exec: starting %s child '%V'",
|
||
|
e->managed ? "managed" : "unmanaged", &ec->cmd);
|
||
|
|
||
|
pipefd[0] = -1;
|
||
|
pipefd[1] = -1;
|
||
|
|
||
|
if (e->managed) {
|
||
|
|
||
|
if (e->active) {
|
||
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, e->log, 0,
|
||
|
"exec: already active '%V'", &ec->cmd);
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
if (pipe(pipefd) == -1) {
|
||
|
ngx_log_error(NGX_LOG_INFO, e->log, ngx_errno,
|
||
|
"exec: pipe failed");
|
||
|
return NGX_ERROR;
|
||
|
}
|
||
|
|
||
|
/* make pipe write end survive through exec */
|
||
|
|
||
|
ret = fcntl(pipefd[1], F_GETFD);
|
||
|
|
||
|
if (ret != -1) {
|
||
|
ret &= ~FD_CLOEXEC;
|
||
|
ret = fcntl(pipefd[1], F_SETFD, ret);
|
||
|
}
|
||
|
|
||
|
if (ret == -1) {
|
||
|
|
||
|
close(pipefd[0]);
|
||
|
close(pipefd[1]);
|
||
|
|
||
|
ngx_log_error(NGX_LOG_INFO, e->log, ngx_errno,
|
||
|
"exec: fcntl failed");
|
||
|
|
||
|
return NGX_ERROR;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pid = fork();
|
||
|
|
||
|
switch (pid) {
|
||
|
|
||
|
case -1:
|
||
|
|
||
|
/* failure */
|
||
|
|
||
|
if (pipefd[0] != -1) {
|
||
|
close(pipefd[0]);
|
||
|
}
|
||
|
|
||
|
if (pipefd[1] != -1) {
|
||
|
close(pipefd[1]);
|
||
|
}
|
||
|
|
||
|
ngx_log_error(NGX_LOG_INFO, e->log, ngx_errno,
|
||
|
"exec: fork failed");
|
||
|
|
||
|
return NGX_ERROR;
|
||
|
|
||
|
case 0:
|
||
|
|
||
|
/* child */
|
||
|
|
||
|
#if (NGX_LINUX)
|
||
|
if (e->managed) {
|
||
|
prctl(PR_SET_PDEATHSIG, e->kill_signal, 0, 0, 0);
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
/* close all descriptors but pipe write end */
|
||
|
|
||
|
maxfd = sysconf(_SC_OPEN_MAX);
|
||
|
for (fd = 0; fd < maxfd; ++fd) {
|
||
|
if (fd == pipefd[1]) {
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
close(fd);
|
||
|
}
|
||
|
|
||
|
fd = open("/dev/null", O_RDWR);
|
||
|
|
||
|
dup2(fd, STDIN_FILENO);
|
||
|
dup2(fd, STDOUT_FILENO);
|
||
|
dup2(fd, STDERR_FILENO);
|
||
|
|
||
|
args = ngx_alloc((ec->args.nelts + 2) * sizeof(char *), e->log);
|
||
|
if (args == NULL) {
|
||
|
exit(1);
|
||
|
}
|
||
|
|
||
|
arg_in = ec->args.elts;
|
||
|
arg_out = args;
|
||
|
*arg_out++ = (char *) ec->cmd.data;
|
||
|
|
||
|
for (n = 0; n < ec->args.nelts; n++, ++arg_in) {
|
||
|
|
||
|
if (e->eval == NULL) {
|
||
|
a = *arg_in;
|
||
|
} else {
|
||
|
ngx_rtmp_eval(e->eval_ctx, arg_in, e->eval, &a, e->log);
|
||
|
}
|
||
|
|
||
|
if (ngx_rtmp_eval_streams(&a) != NGX_DONE) {
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
*arg_out++ = (char *) a.data;
|
||
|
}
|
||
|
|
||
|
*arg_out = NULL;
|
||
|
|
||
|
#if (NGX_DEBUG)
|
||
|
{
|
||
|
char **p;
|
||
|
|
||
|
for (p = args; *p; p++) {
|
||
|
ngx_write_fd(STDERR_FILENO, "'", 1);
|
||
|
ngx_write_fd(STDERR_FILENO, *p, strlen(*p));
|
||
|
ngx_write_fd(STDERR_FILENO, "' ", 2);
|
||
|
}
|
||
|
|
||
|
ngx_write_fd(STDERR_FILENO, "\n", 1);
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
if (execvp((char *) ec->cmd.data, args) == -1) {
|
||
|
char *msg;
|
||
|
|
||
|
msg = strerror(errno);
|
||
|
|
||
|
ngx_write_fd(STDERR_FILENO, "execvp error: ", 14);
|
||
|
ngx_write_fd(STDERR_FILENO, msg, strlen(msg));
|
||
|
ngx_write_fd(STDERR_FILENO, "\n", 1);
|
||
|
|
||
|
exit(1);
|
||
|
}
|
||
|
|
||
|
break;
|
||
|
|
||
|
default:
|
||
|
|
||
|
/* parent */
|
||
|
|
||
|
if (pipefd[1] != -1) {
|
||
|
close(pipefd[1]);
|
||
|
}
|
||
|
|
||
|
if (pipefd[0] != -1) {
|
||
|
|
||
|
e->active = 1;
|
||
|
e->pid = pid;
|
||
|
e->pipefd = pipefd[0];
|
||
|
|
||
|
if (e->save_pid) {
|
||
|
*e->save_pid = pid;
|
||
|
}
|
||
|
|
||
|
e->dummy_conn.fd = e->pipefd;
|
||
|
e->dummy_conn.data = e;
|
||
|
e->dummy_conn.read = &e->read_evt;
|
||
|
e->dummy_conn.write = &e->write_evt;
|
||
|
e->read_evt.data = &e->dummy_conn;
|
||
|
e->write_evt.data = &e->dummy_conn;
|
||
|
|
||
|
e->read_evt.log = e->log;
|
||
|
e->read_evt.handler = ngx_rtmp_exec_child_dead;
|
||
|
|
||
|
if (ngx_add_event(&e->read_evt, NGX_READ_EVENT, 0) != NGX_OK) {
|
||
|
ngx_log_error(NGX_LOG_INFO, e->log, ngx_errno,
|
||
|
"exec: failed to add child control event");
|
||
|
}
|
||
|
}
|
||
|
|
||
|
ngx_log_debug2(NGX_LOG_DEBUG_RTMP, e->log, 0,
|
||
|
"exec: child '%V' started pid=%i",
|
||
|
&ec->cmd, (ngx_int_t) pid);
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_init_ctx(ngx_rtmp_session_t *s, u_char name[NGX_RTMP_MAX_NAME],
|
||
|
u_char args[NGX_RTMP_MAX_ARGS], ngx_uint_t flags)
|
||
|
{
|
||
|
ngx_uint_t n;
|
||
|
ngx_array_t *push_conf;
|
||
|
ngx_rtmp_exec_t *e;
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
ngx_rtmp_exec_conf_t *ec;
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
ngx_rtmp_exec_main_conf_t *emcf;
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
if (ctx != NULL) {
|
||
|
goto done;
|
||
|
}
|
||
|
|
||
|
ctx = ngx_pcalloc(s->connection->pool, sizeof(ngx_rtmp_exec_ctx_t));
|
||
|
|
||
|
if (ctx == NULL) {
|
||
|
return NGX_ERROR;
|
||
|
}
|
||
|
|
||
|
ngx_rtmp_set_ctx(s, ctx, ngx_rtmp_exec_module);
|
||
|
|
||
|
eacf = ngx_rtmp_get_module_app_conf(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
emcf = ngx_rtmp_get_module_main_conf(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
push_conf = &eacf->conf[NGX_RTMP_EXEC_PUSH];
|
||
|
|
||
|
if (push_conf->nelts > 0) {
|
||
|
|
||
|
if (ngx_array_init(&ctx->push_exec, s->connection->pool,
|
||
|
push_conf->nelts,
|
||
|
sizeof(ngx_rtmp_exec_t)) != NGX_OK)
|
||
|
{
|
||
|
return NGX_ERROR;
|
||
|
}
|
||
|
|
||
|
e = ngx_array_push_n(&ctx->push_exec, push_conf->nelts);
|
||
|
|
||
|
if (e == NULL) {
|
||
|
return NGX_ERROR;
|
||
|
}
|
||
|
|
||
|
ec = push_conf->elts;
|
||
|
|
||
|
for (n = 0; n < push_conf->nelts; n++, e++, ec++) {
|
||
|
ngx_memzero(e, sizeof(*e));
|
||
|
e->conf = ec;
|
||
|
e->managed = 1;
|
||
|
e->log = s->connection->log;
|
||
|
e->eval = ngx_rtmp_exec_push_eval;
|
||
|
e->eval_ctx = s;
|
||
|
e->kill_signal = emcf->kill_signal;
|
||
|
e->respawn_timeout = (eacf->respawn ? emcf->respawn_timeout :
|
||
|
NGX_CONF_UNSET_MSEC);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
done:
|
||
|
|
||
|
ngx_memcpy(ctx->name, name, NGX_RTMP_MAX_NAME);
|
||
|
ngx_memcpy(ctx->args, args, NGX_RTMP_MAX_ARGS);
|
||
|
|
||
|
ctx->flags |= flags;
|
||
|
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_init_pull_ctx(ngx_rtmp_session_t *s,
|
||
|
u_char name[NGX_RTMP_MAX_NAME])
|
||
|
{
|
||
|
size_t len;
|
||
|
ngx_uint_t n;
|
||
|
ngx_pool_t *pool;
|
||
|
ngx_array_t *pull_conf;
|
||
|
ngx_rtmp_exec_t *e;
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
ngx_rtmp_exec_conf_t *ec;
|
||
|
ngx_rtmp_exec_pull_ctx_t *pctx, **ppctx;
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
ngx_rtmp_exec_main_conf_t *emcf;
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
if (ctx->pull != NULL) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
eacf = ngx_rtmp_get_module_app_conf(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
pull_conf = &eacf->conf[NGX_RTMP_EXEC_PULL];
|
||
|
|
||
|
if (pull_conf->nelts == 0) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
emcf = ngx_rtmp_get_module_main_conf(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
len = ngx_strlen(name);
|
||
|
|
||
|
ppctx = &eacf->pull[ngx_hash_key(name, len) % eacf->nbuckets];
|
||
|
|
||
|
for (; *ppctx; ppctx = &(*ppctx)->next) {
|
||
|
pctx = *ppctx;
|
||
|
|
||
|
if (pctx->name.len == len &&
|
||
|
ngx_strncmp(name, pctx->name.data, len) == 0)
|
||
|
{
|
||
|
goto done;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pool = ngx_create_pool(4096, emcf->log);
|
||
|
if (pool == NULL) {
|
||
|
return NGX_ERROR;
|
||
|
}
|
||
|
|
||
|
pctx = ngx_pcalloc(pool, sizeof(ngx_rtmp_exec_pull_ctx_t));
|
||
|
if (pctx == NULL) {
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
pctx->pool = pool;
|
||
|
pctx->name.len = len;
|
||
|
pctx->name.data = ngx_palloc(pool, len);
|
||
|
|
||
|
if (pctx->name.data == NULL) {
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
ngx_memcpy(pctx->name.data, name, len);
|
||
|
|
||
|
pctx->app.len = s->app.len;
|
||
|
pctx->app.data = ngx_palloc(pool, s->app.len);
|
||
|
|
||
|
if (pctx->app.data == NULL) {
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
ngx_memcpy(pctx->app.data, s->app.data, s->app.len);
|
||
|
|
||
|
if (ngx_array_init(&pctx->pull_exec, pool, pull_conf->nelts,
|
||
|
sizeof(ngx_rtmp_exec_t)) != NGX_OK)
|
||
|
{
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
e = ngx_array_push_n(&pctx->pull_exec, pull_conf->nelts);
|
||
|
if (e == NULL) {
|
||
|
goto error;
|
||
|
}
|
||
|
|
||
|
ec = pull_conf->elts;
|
||
|
for (n = 0; n < pull_conf->nelts; n++, e++, ec++) {
|
||
|
ngx_memzero(e, sizeof(*e));
|
||
|
e->conf = ec;
|
||
|
e->managed = 1;
|
||
|
e->log = emcf->log;
|
||
|
e->eval = ngx_rtmp_exec_pull_eval;
|
||
|
e->eval_ctx = pctx;
|
||
|
e->kill_signal = emcf->kill_signal;
|
||
|
e->respawn_timeout = (eacf->respawn ? emcf->respawn_timeout :
|
||
|
NGX_CONF_UNSET_MSEC);
|
||
|
}
|
||
|
|
||
|
*ppctx = pctx;
|
||
|
|
||
|
done:
|
||
|
|
||
|
ctx->pull = pctx;
|
||
|
ctx->pull->counter++;
|
||
|
|
||
|
return NGX_OK;
|
||
|
|
||
|
error:
|
||
|
|
||
|
ngx_destroy_pool(pool);
|
||
|
|
||
|
return NGX_ERROR;
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_filter(ngx_rtmp_session_t *s, ngx_rtmp_exec_conf_t *ec)
|
||
|
{
|
||
|
size_t len;
|
||
|
ngx_str_t *v;
|
||
|
ngx_uint_t n;
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
|
||
|
if (ec->names.nelts == 0) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
len = ngx_strlen(ctx->name);
|
||
|
|
||
|
v = ec->names.elts;
|
||
|
for (n = 0; n < ec->names.nelts; n++, s++) {
|
||
|
if (v->len == len && ngx_strncmp(v->data, ctx->name, len) == 0) {
|
||
|
return NGX_OK;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return NGX_DECLINED;
|
||
|
}
|
||
|
|
||
|
|
||
|
static void
|
||
|
ngx_rtmp_exec_unmanaged(ngx_rtmp_session_t *s, ngx_array_t *e, const char *op)
|
||
|
{
|
||
|
ngx_uint_t n;
|
||
|
ngx_rtmp_exec_t en;
|
||
|
ngx_rtmp_exec_conf_t *ec;
|
||
|
|
||
|
if (e->nelts == 0) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
ngx_log_debug2(NGX_LOG_DEBUG_RTMP, s->connection->log, 0,
|
||
|
"exec: %s %uz unmanaged command(s)", op, e->nelts);
|
||
|
|
||
|
ec = e->elts;
|
||
|
for (n = 0; n < e->nelts; n++, ec++) {
|
||
|
if (ngx_rtmp_exec_filter(s, ec) != NGX_OK) {
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
ngx_memzero(&en, sizeof(ngx_rtmp_exec_t));
|
||
|
|
||
|
en.conf = ec;
|
||
|
en.eval = ngx_rtmp_exec_event_eval;
|
||
|
en.eval_ctx = s;
|
||
|
en.log = s->connection->log;
|
||
|
|
||
|
ngx_rtmp_exec_run(&en);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
static void
|
||
|
ngx_rtmp_exec_managed(ngx_rtmp_session_t *s, ngx_array_t *e, const char *op)
|
||
|
{
|
||
|
ngx_uint_t n;
|
||
|
ngx_rtmp_exec_t *en;
|
||
|
|
||
|
if (e->nelts == 0) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
ngx_log_debug2(NGX_LOG_DEBUG_RTMP, s->connection->log, 0,
|
||
|
"exec: %s %uz managed command(s)", op, e->nelts);
|
||
|
|
||
|
en = e->elts;
|
||
|
for (n = 0; n < e->nelts; n++, en++) {
|
||
|
if (ngx_rtmp_exec_filter(s, en->conf) == NGX_OK) {
|
||
|
ngx_rtmp_exec_run(en);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_publish(ngx_rtmp_session_t *s, ngx_rtmp_publish_t *v)
|
||
|
{
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
|
||
|
eacf = ngx_rtmp_get_module_app_conf(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
if (eacf == NULL || !eacf->active) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
if (s->auto_pushed) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
if (ngx_rtmp_exec_init_ctx(s, v->name, v->args, NGX_RTMP_EXEC_PUBLISHING)
|
||
|
!= NGX_OK)
|
||
|
{
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
ngx_rtmp_exec_unmanaged(s, &eacf->conf[NGX_RTMP_EXEC_PUBLISH], "publish");
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
ngx_rtmp_exec_managed(s, &ctx->push_exec, "push");
|
||
|
|
||
|
next:
|
||
|
return next_publish(s, v);
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_play(ngx_rtmp_session_t *s, ngx_rtmp_play_t *v)
|
||
|
{
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
ngx_rtmp_exec_pull_ctx_t *pctx;
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
|
||
|
eacf = ngx_rtmp_get_module_app_conf(s, ngx_rtmp_exec_module);
|
||
|
|
||
|
if (eacf == NULL || !eacf->active) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
if (ngx_rtmp_exec_init_ctx(s, v->name, v->args, NGX_RTMP_EXEC_PLAYING)
|
||
|
!= NGX_OK)
|
||
|
{
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
ngx_rtmp_exec_unmanaged(s, &eacf->conf[NGX_RTMP_EXEC_PLAY], "play");
|
||
|
|
||
|
if (ngx_rtmp_exec_init_pull_ctx(s, v->name) != NGX_OK) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
pctx = ctx->pull;
|
||
|
|
||
|
if (pctx && pctx->counter == 1) {
|
||
|
ngx_rtmp_exec_managed(s, &pctx->pull_exec, "pull");
|
||
|
}
|
||
|
|
||
|
next:
|
||
|
return next_play(s, v);
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_close_stream(ngx_rtmp_session_t *s, ngx_rtmp_close_stream_t *v)
|
||
|
{
|
||
|
size_t n;
|
||
|
ngx_rtmp_exec_t *e;
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
ngx_rtmp_exec_pull_ctx_t *pctx, **ppctx;
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
|
||
|
eacf = ngx_rtmp_get_module_app_conf(s, ngx_rtmp_exec_module);
|
||
|
if (eacf == NULL) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
if (ctx == NULL) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
if (ctx->flags & NGX_RTMP_EXEC_PUBLISHING) {
|
||
|
ngx_rtmp_exec_unmanaged(s, &eacf->conf[NGX_RTMP_EXEC_PUBLISH_DONE],
|
||
|
"publish_done");
|
||
|
}
|
||
|
|
||
|
if (ctx->flags & NGX_RTMP_EXEC_PLAYING) {
|
||
|
ngx_rtmp_exec_unmanaged(s, &eacf->conf[NGX_RTMP_EXEC_PLAY_DONE],
|
||
|
"play_done");
|
||
|
}
|
||
|
|
||
|
ctx->flags = 0;
|
||
|
|
||
|
if (ctx->push_exec.nelts > 0) {
|
||
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, s->connection->log, 0,
|
||
|
"exec: delete %uz push command(s)",
|
||
|
ctx->push_exec.nelts);
|
||
|
|
||
|
e = ctx->push_exec.elts;
|
||
|
for (n = 0; n < ctx->push_exec.nelts; n++, e++) {
|
||
|
ngx_rtmp_exec_kill(e, e->kill_signal);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pctx = ctx->pull;
|
||
|
|
||
|
if (pctx && --pctx->counter == 0) {
|
||
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, s->connection->log, 0,
|
||
|
"exec: delete %uz pull command(s)",
|
||
|
pctx->pull_exec.nelts);
|
||
|
|
||
|
e = pctx->pull_exec.elts;
|
||
|
for (n = 0; n < pctx->pull_exec.nelts; n++, e++) {
|
||
|
ngx_rtmp_exec_kill(e, e->kill_signal);
|
||
|
}
|
||
|
|
||
|
ppctx = &eacf->pull[ngx_hash_key(pctx->name.data, pctx->name.len) %
|
||
|
eacf->nbuckets];
|
||
|
|
||
|
for (; *ppctx; ppctx = &(*ppctx)->next) {
|
||
|
if (pctx == *ppctx) {
|
||
|
*ppctx = pctx->next;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
ngx_destroy_pool(pctx->pool);
|
||
|
}
|
||
|
|
||
|
ctx->pull = NULL;
|
||
|
|
||
|
next:
|
||
|
return next_close_stream(s, v);
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_record_done(ngx_rtmp_session_t *s, ngx_rtmp_record_done_t *v)
|
||
|
{
|
||
|
u_char c;
|
||
|
ngx_uint_t ext, dir;
|
||
|
ngx_rtmp_exec_ctx_t *ctx;
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
|
||
|
if (s->auto_pushed) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
eacf = ngx_rtmp_get_module_app_conf(s, ngx_rtmp_exec_module);
|
||
|
if (eacf == NULL || !eacf->active) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_exec_module);
|
||
|
if (ctx == NULL) {
|
||
|
goto next;
|
||
|
}
|
||
|
|
||
|
ctx->recorder = v->recorder;
|
||
|
ctx->path = v->path;
|
||
|
|
||
|
ctx->dirname.data = ctx->path.data;
|
||
|
ctx->dirname.len = 0;
|
||
|
|
||
|
for (dir = ctx->path.len; dir > 0; dir--) {
|
||
|
c = ctx->path.data[dir - 1];
|
||
|
if (c == '/' || c == '\\') {
|
||
|
ctx->dirname.len = dir - 1;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
ctx->filename.data = ctx->path.data + dir;
|
||
|
ctx->filename.len = ctx->path.len - dir;
|
||
|
|
||
|
ctx->basename = ctx->filename;
|
||
|
|
||
|
for (ext = ctx->filename.len; ext > 0; ext--) {
|
||
|
if (ctx->filename.data[ext - 1] == '.') {
|
||
|
ctx->basename.len = ext - 1;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
ngx_rtmp_exec_unmanaged(s, &eacf->conf[NGX_RTMP_EXEC_RECORD_DONE],
|
||
|
"record_done");
|
||
|
|
||
|
ngx_str_null(&v->recorder);
|
||
|
ngx_str_null(&v->path);
|
||
|
|
||
|
next:
|
||
|
return next_record_done(s, v);
|
||
|
}
|
||
|
#endif /* NGX_WIN32 */
|
||
|
|
||
|
|
||
|
static char *
|
||
|
ngx_rtmp_exec_conf(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
||
|
{
|
||
|
char *p = conf;
|
||
|
|
||
|
size_t n, nargs;
|
||
|
ngx_str_t *s, *value, v;
|
||
|
ngx_array_t *confs;
|
||
|
ngx_rtmp_exec_conf_t *ec;
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
|
||
|
confs = (ngx_array_t *) (p + cmd->offset);
|
||
|
|
||
|
eacf = ngx_rtmp_conf_get_module_app_conf(cf, ngx_rtmp_exec_module);
|
||
|
|
||
|
if (confs->nalloc == 0 && ngx_array_init(confs, cf->pool, 1,
|
||
|
sizeof(ngx_rtmp_exec_conf_t))
|
||
|
!= NGX_OK)
|
||
|
{
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
value = cf->args->elts;
|
||
|
|
||
|
ec = ngx_array_push(confs);
|
||
|
if (ec == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
ngx_memzero(ec, sizeof(ngx_rtmp_exec_conf_t));
|
||
|
|
||
|
/* type is undefined for explicit execs */
|
||
|
|
||
|
ec->type = NGX_CONF_UNSET_UINT;
|
||
|
ec->cmd = value[1];
|
||
|
|
||
|
if (ngx_array_init(&ec->names, cf->pool, 1, sizeof(ngx_str_t)) != NGX_OK) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
if (cf->args->nelts == 2) {
|
||
|
return NGX_CONF_OK;
|
||
|
}
|
||
|
|
||
|
nargs = cf->args->nelts - 2;
|
||
|
if (ngx_array_init(&ec->args, cf->pool, nargs, sizeof(ngx_str_t)) != NGX_OK)
|
||
|
{
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
for (n = 2; n < cf->args->nelts; n++) {
|
||
|
|
||
|
v = value[n];
|
||
|
|
||
|
if (eacf->options == 1) {
|
||
|
|
||
|
if (v.len >= 5 && ngx_strncmp(v.data, "name=", 5) == 0) {
|
||
|
|
||
|
s = ngx_array_push(&ec->names);
|
||
|
if (s == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
v.data += 5;
|
||
|
v.len -= 5;
|
||
|
|
||
|
*s = v;
|
||
|
|
||
|
continue;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
s = ngx_array_push(&ec->args);
|
||
|
if (s == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
*s = v;
|
||
|
}
|
||
|
|
||
|
return NGX_CONF_OK;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
static char *
|
||
|
ngx_rtmp_exec_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
||
|
{
|
||
|
char *rv;
|
||
|
ngx_str_t *value;
|
||
|
ngx_conf_t save;
|
||
|
ngx_array_t *confs;
|
||
|
ngx_rtmp_conf_ctx_t *ctx, *pctx;
|
||
|
ngx_rtmp_exec_conf_t *ec, *eec;
|
||
|
ngx_rtmp_exec_app_conf_t *eacf;
|
||
|
ngx_rtmp_exec_main_conf_t *emcf;
|
||
|
|
||
|
value = cf->args->elts;
|
||
|
|
||
|
eacf = ngx_rtmp_conf_get_module_app_conf(cf, ngx_rtmp_exec_module);
|
||
|
|
||
|
emcf = ngx_rtmp_conf_get_module_main_conf(cf, ngx_rtmp_exec_module);
|
||
|
|
||
|
ctx = ngx_pcalloc(cf->pool, sizeof(ngx_rtmp_conf_ctx_t));
|
||
|
if (ctx == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
pctx = cf->ctx;
|
||
|
|
||
|
ctx->main_conf = pctx->main_conf;
|
||
|
ctx->srv_conf = pctx->srv_conf;
|
||
|
|
||
|
ctx->app_conf = ngx_pcalloc(cf->pool, sizeof(void *) * ngx_rtmp_max_module);
|
||
|
if (ctx->app_conf == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
ec = ngx_pcalloc(cf->pool, sizeof(ngx_rtmp_exec_conf_t));
|
||
|
if (ec == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
ec->id = value[1];
|
||
|
ec->type = NGX_CONF_UNSET_UINT;
|
||
|
|
||
|
ctx->app_conf[ngx_rtmp_exec_module.ctx_index] = ec;
|
||
|
|
||
|
save = *cf;
|
||
|
|
||
|
cf->ctx = ctx;
|
||
|
cf->cmd_type = NGX_RTMP_EXEC_CONF;
|
||
|
|
||
|
rv = ngx_conf_parse(cf, NULL);
|
||
|
*cf= save;
|
||
|
|
||
|
switch (ec->type) {
|
||
|
|
||
|
case NGX_RTMP_EXEC_STATIC:
|
||
|
confs = &emcf->static_conf;
|
||
|
break;
|
||
|
|
||
|
case NGX_CONF_UNSET_UINT:
|
||
|
return "unspecified exec type";
|
||
|
|
||
|
default:
|
||
|
confs = &eacf->conf[ec->type];
|
||
|
}
|
||
|
|
||
|
if (confs->nalloc == 0 && ngx_array_init(confs, cf->pool, 1,
|
||
|
sizeof(ngx_rtmp_exec_conf_t))
|
||
|
!= NGX_OK)
|
||
|
{
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
eec = ngx_array_push(confs);
|
||
|
if (eec == NULL) {
|
||
|
return NGX_CONF_ERROR;
|
||
|
}
|
||
|
|
||
|
*eec = *ec;
|
||
|
|
||
|
return rv;
|
||
|
}
|
||
|
*/
|
||
|
|
||
|
static char *
|
||
|
ngx_rtmp_exec_kill_signal(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
||
|
{
|
||
|
ngx_rtmp_exec_main_conf_t *emcf = conf;
|
||
|
|
||
|
ngx_str_t *value;
|
||
|
|
||
|
value = cf->args->elts;
|
||
|
value++;
|
||
|
|
||
|
emcf->kill_signal = ngx_atoi(value->data, value->len);
|
||
|
if (emcf->kill_signal != NGX_ERROR) {
|
||
|
return NGX_CONF_OK;
|
||
|
}
|
||
|
|
||
|
#define NGX_RMTP_EXEC_SIGNAL(name) \
|
||
|
if (value->len == sizeof(#name) - 1 && \
|
||
|
ngx_strncasecmp(value->data, (u_char *) #name, value->len) == 0) \
|
||
|
{ \
|
||
|
emcf->kill_signal = SIG##name; \
|
||
|
return NGX_CONF_OK; \
|
||
|
}
|
||
|
|
||
|
/* POSIX.1-1990 signals */
|
||
|
|
||
|
#if !(NGX_WIN32)
|
||
|
NGX_RMTP_EXEC_SIGNAL(HUP);
|
||
|
NGX_RMTP_EXEC_SIGNAL(INT);
|
||
|
NGX_RMTP_EXEC_SIGNAL(QUIT);
|
||
|
NGX_RMTP_EXEC_SIGNAL(ILL);
|
||
|
NGX_RMTP_EXEC_SIGNAL(ABRT);
|
||
|
NGX_RMTP_EXEC_SIGNAL(FPE);
|
||
|
NGX_RMTP_EXEC_SIGNAL(KILL);
|
||
|
NGX_RMTP_EXEC_SIGNAL(SEGV);
|
||
|
NGX_RMTP_EXEC_SIGNAL(PIPE);
|
||
|
NGX_RMTP_EXEC_SIGNAL(ALRM);
|
||
|
NGX_RMTP_EXEC_SIGNAL(TERM);
|
||
|
NGX_RMTP_EXEC_SIGNAL(USR1);
|
||
|
NGX_RMTP_EXEC_SIGNAL(USR2);
|
||
|
NGX_RMTP_EXEC_SIGNAL(CHLD);
|
||
|
NGX_RMTP_EXEC_SIGNAL(CONT);
|
||
|
NGX_RMTP_EXEC_SIGNAL(STOP);
|
||
|
NGX_RMTP_EXEC_SIGNAL(TSTP);
|
||
|
NGX_RMTP_EXEC_SIGNAL(TTIN);
|
||
|
NGX_RMTP_EXEC_SIGNAL(TTOU);
|
||
|
#endif
|
||
|
|
||
|
#undef NGX_RMTP_EXEC_SIGNAL
|
||
|
|
||
|
return "unknown signal";
|
||
|
}
|
||
|
|
||
|
|
||
|
static ngx_int_t
|
||
|
ngx_rtmp_exec_postconfiguration(ngx_conf_t *cf)
|
||
|
{
|
||
|
#if !(NGX_WIN32)
|
||
|
|
||
|
next_publish = ngx_rtmp_publish;
|
||
|
ngx_rtmp_publish = ngx_rtmp_exec_publish;
|
||
|
|
||
|
next_play = ngx_rtmp_play;
|
||
|
ngx_rtmp_play = ngx_rtmp_exec_play;
|
||
|
|
||
|
next_close_stream = ngx_rtmp_close_stream;
|
||
|
ngx_rtmp_close_stream = ngx_rtmp_exec_close_stream;
|
||
|
|
||
|
next_record_done = ngx_rtmp_record_done;
|
||
|
ngx_rtmp_record_done = ngx_rtmp_exec_record_done;
|
||
|
|
||
|
#endif /* NGX_WIN32 */
|
||
|
|
||
|
return NGX_OK;
|
||
|
}
|