2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Copyright (c) 2012 Roman Arutyunyan
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
|
|
#include <ngx_config.h>
|
|
|
|
#include <ngx_core.h>
|
|
|
|
#include <ngx_event.h>
|
|
|
|
#include <strings.h>
|
|
|
|
|
|
|
|
#include "ngx_rtmp.h"
|
|
|
|
#include "ngx_rtmp_amf0.h"
|
|
|
|
|
|
|
|
|
|
|
|
static void ngx_rtmp_init_session(ngx_connection_t *c);
|
|
|
|
|
|
|
|
static void ngx_rtmp_handshake_recv(ngx_event_t *rev);
|
|
|
|
static void ngx_rtmp_handshake_send(ngx_event_t *rev);
|
|
|
|
|
|
|
|
static void ngx_rtmp_recv(ngx_event_t *rev);
|
|
|
|
static void ngx_rtmp_send(ngx_event_t *rev);
|
|
|
|
|
|
|
|
static void ngx_rtmp_close_connection(ngx_connection_t *c);
|
|
|
|
|
|
|
|
static size_t hdrsizes[] = { 12, 8, 4, 1 };
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
#ifdef NGX_DEBUG
|
|
|
|
static char*
|
|
|
|
ngx_rtmp_packet_type(uint8_t type) {
|
|
|
|
static char* types[] = {
|
|
|
|
"?",
|
|
|
|
"chunk_size",
|
|
|
|
"?",
|
|
|
|
"bytes_read",
|
|
|
|
"ping",
|
|
|
|
"server_bw",
|
|
|
|
"client_bw",
|
|
|
|
"?",
|
|
|
|
"audio",
|
|
|
|
"video",
|
|
|
|
"?",
|
|
|
|
"?",
|
|
|
|
"?",
|
|
|
|
"?",
|
|
|
|
"?",
|
|
|
|
"flex",
|
|
|
|
"flex_so",
|
|
|
|
"flex_msg",
|
|
|
|
"notify",
|
|
|
|
"so",
|
|
|
|
"invoke"
|
|
|
|
};
|
|
|
|
|
|
|
|
return type < sizeof(types) / sizeof(types[0])
|
|
|
|
? types[type]
|
|
|
|
: "?";
|
|
|
|
}
|
|
|
|
#endif
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_init_connection(ngx_connection_t *c)
|
|
|
|
{
|
|
|
|
ngx_uint_t i;
|
|
|
|
ngx_rtmp_port_t *port;
|
|
|
|
struct sockaddr *sa;
|
|
|
|
struct sockaddr_in *sin;
|
|
|
|
ngx_rtmp_log_ctx_t *ctx;
|
|
|
|
ngx_rtmp_in_addr_t *addr;
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_addr_conf_t *addr_conf;
|
|
|
|
#if (NGX_HAVE_INET6)
|
|
|
|
struct sockaddr_in6 *sin6;
|
|
|
|
ngx_rtmp_in6_addr_t *addr6;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
/* find the server configuration for the address:port */
|
|
|
|
|
|
|
|
/* AF_INET only */
|
|
|
|
|
|
|
|
port = c->listening->servers;
|
|
|
|
|
|
|
|
if (port->naddrs > 1) {
|
|
|
|
|
|
|
|
/*
|
|
|
|
* There are several addresses on this port and one of them
|
|
|
|
* is the "*:port" wildcard so getsockname() is needed to determine
|
|
|
|
* the server address.
|
|
|
|
*
|
|
|
|
* AcceptEx() already gave this address.
|
|
|
|
*/
|
|
|
|
|
|
|
|
if (ngx_connection_local_sockaddr(c, NULL, 0) != NGX_OK) {
|
|
|
|
ngx_rtmp_close_connection(c);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
sa = c->local_sockaddr;
|
|
|
|
|
|
|
|
switch (sa->sa_family) {
|
|
|
|
|
|
|
|
#if (NGX_HAVE_INET6)
|
|
|
|
case AF_INET6:
|
|
|
|
sin6 = (struct sockaddr_in6 *) sa;
|
|
|
|
|
|
|
|
addr6 = port->addrs;
|
|
|
|
|
|
|
|
/* the last address is "*" */
|
|
|
|
|
|
|
|
for (i = 0; i < port->naddrs - 1; i++) {
|
|
|
|
if (ngx_memcmp(&addr6[i].addr6, &sin6->sin6_addr, 16) == 0) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
addr_conf = &addr6[i].conf;
|
|
|
|
|
|
|
|
break;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
default: /* AF_INET */
|
|
|
|
sin = (struct sockaddr_in *) sa;
|
|
|
|
|
|
|
|
addr = port->addrs;
|
|
|
|
|
|
|
|
/* the last address is "*" */
|
|
|
|
|
|
|
|
for (i = 0; i < port->naddrs - 1; i++) {
|
|
|
|
if (addr[i].addr == sin->sin_addr.s_addr) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
addr_conf = &addr[i].conf;
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
switch (c->local_sockaddr->sa_family) {
|
|
|
|
|
|
|
|
#if (NGX_HAVE_INET6)
|
|
|
|
case AF_INET6:
|
|
|
|
addr6 = port->addrs;
|
|
|
|
addr_conf = &addr6[0].conf;
|
|
|
|
break;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
default: /* AF_INET */
|
|
|
|
addr = port->addrs;
|
|
|
|
addr_conf = &addr[0].conf;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
s = ngx_pcalloc(c->pool, sizeof(ngx_rtmp_session_t));
|
|
|
|
if (s == NULL) {
|
|
|
|
ngx_rtmp_close_connection(c);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
s->main_conf = addr_conf->ctx->main_conf;
|
|
|
|
s->srv_conf = addr_conf->ctx->srv_conf;
|
|
|
|
|
|
|
|
s->addr_text = &addr_conf->addr_text;
|
|
|
|
|
|
|
|
c->data = s;
|
|
|
|
s->connection = c;
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, 0, "*%ui client connected",
|
|
|
|
c->number, &c->addr_text);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
ctx = ngx_palloc(c->pool, sizeof(ngx_rtmp_log_ctx_t));
|
|
|
|
if (ctx == NULL) {
|
|
|
|
ngx_rtmp_close_connection(c);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
ctx->client = &c->addr_text;
|
|
|
|
ctx->session = s;
|
|
|
|
|
|
|
|
c->log->connection = c->number;
|
|
|
|
c->log->handler = ngx_rtmp_log_error;
|
|
|
|
c->log->data = ctx;
|
|
|
|
c->log->action = "sending client greeting line";
|
|
|
|
|
|
|
|
c->log_error = NGX_ERROR_INFO;
|
|
|
|
|
|
|
|
ngx_rtmp_init_session(c);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static void
|
|
|
|
ngx_rtmp_init_session(ngx_connection_t *c)
|
|
|
|
{
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
|
|
|
ngx_bufs_t bufs;
|
|
|
|
ngx_buf_t *b;
|
|
|
|
size_t size;
|
|
|
|
|
|
|
|
s = c->data;
|
|
|
|
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
s->ctx = ngx_pcalloc(c->pool, sizeof(void *) * ngx_rtmp_max_module);
|
|
|
|
if (s->ctx == NULL) {
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_close_session(s);
|
2012-03-08 16:21:22 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
s->chunk_size = NGX_RTMP_DEFAULT_CHUNK_SIZE;
|
|
|
|
|
|
|
|
/* TODO: we should move preallcation to a func to call on user request */
|
|
|
|
bufs.size = s->chunk_size + 14 /* + max header size */;
|
|
|
|
bufs.num = cscf->buffers;
|
|
|
|
|
|
|
|
s->free = ngx_create_chain_of_bufs(c->pool, &bufs);
|
|
|
|
|
|
|
|
b = &s->buf;
|
|
|
|
size = NGX_RTMP_HANDSHAKE_SIZE + 1;
|
|
|
|
b->start = b->pos = b->last = ngx_pcalloc(c->pool, size);
|
|
|
|
b->end = b->start + size;
|
|
|
|
b->temporary = 1;
|
|
|
|
|
|
|
|
c->write->handler = ngx_rtmp_handshake_send;
|
|
|
|
c->read->handler = ngx_rtmp_handshake_recv;
|
|
|
|
|
|
|
|
ngx_rtmp_handshake_recv(c->read);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_handshake_recv(ngx_event_t *rev)
|
|
|
|
{
|
|
|
|
ssize_t n;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
|
|
|
ngx_buf_t *b;
|
|
|
|
|
|
|
|
c = rev->data;
|
|
|
|
s = c->data;
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
if (rev->timedout) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
|
|
|
|
c->timedout = 1;
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (rev->timer_set) {
|
|
|
|
ngx_del_timer(rev);
|
|
|
|
}
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
b = &s->buf;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
while(b->last != b->end) {
|
|
|
|
|
|
|
|
n = c->recv(c, b->last, b->end - b->last);
|
|
|
|
|
|
|
|
if (n == NGX_ERROR || n == 0) {
|
|
|
|
ngx_rtmp_close_session(s);
|
2012-03-08 18:45:10 +01:00
|
|
|
return;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
if (n > 0) {
|
|
|
|
if (b->last == b->start
|
|
|
|
&& s->hs_stage == 0 && *b->last != '\x03')
|
|
|
|
{
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ERROR,
|
|
|
|
"invalid handshake signature");
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_close_session(s);
|
2012-03-08 16:21:22 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
b->last += n;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (n == NGX_AGAIN) {
|
|
|
|
ngx_add_timer(rev, cscf->timeout);
|
|
|
|
if (ngx_handle_read_event(c->read, 0) != NGX_OK) {
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_close_session(s);
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ngx_del_event(c->read, NGX_READ_EVENT, 0);
|
|
|
|
|
|
|
|
if (s->hs_stage++ == 0) {
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_handshake_send(c->write);
|
2012-03-08 16:21:22 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* handshake done */
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_pfree(c->pool, s->buf.start);
|
2012-03-08 16:21:22 +01:00
|
|
|
c->read->handler = ngx_rtmp_recv;
|
|
|
|
c->write->handler = ngx_rtmp_send;
|
|
|
|
ngx_rtmp_recv(rev);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_handshake_send(ngx_event_t *wev)
|
|
|
|
{
|
|
|
|
ngx_int_t n;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
|
|
|
ngx_buf_t *b;
|
|
|
|
|
|
|
|
c = wev->data;
|
|
|
|
s = c->data;
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
if (wev->timedout) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
|
|
|
|
c->timedout = 1;
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (wev->timer_set) {
|
|
|
|
ngx_del_timer(wev);
|
|
|
|
}
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
b = &s->buf;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
restart:
|
|
|
|
while(b->pos != b->last) {
|
|
|
|
n = c->send(c, b->pos, b->last - b->pos);
|
|
|
|
if (n > 0) {
|
|
|
|
b->pos += n;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (n == NGX_ERROR) {
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (n == NGX_AGAIN) {
|
|
|
|
ngx_add_timer(c->write, cscf->timeout);
|
|
|
|
if (ngx_handle_write_event(c->write, 0) != NGX_OK) {
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (s->hs_stage++ == 1) {
|
|
|
|
b->pos = b->start + 1;
|
|
|
|
goto restart;
|
|
|
|
}
|
|
|
|
|
|
|
|
b->pos = b->last = b->start + 1;
|
|
|
|
ngx_del_event(wev, NGX_WRITE_EVENT, 0);
|
|
|
|
ngx_rtmp_handshake_recv(c->read);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_recv(ngx_event_t *rev)
|
|
|
|
{
|
|
|
|
ngx_int_t n;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
|
|
|
ngx_buf_t *b, *bb;
|
2012-03-08 23:39:15 +01:00
|
|
|
u_char h, *p, *pp;
|
2012-03-08 16:21:22 +01:00
|
|
|
ngx_chain_t *lin;
|
|
|
|
|
|
|
|
c = rev->data;
|
|
|
|
s = c->data;
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
if (s->in == NULL) {
|
|
|
|
if (s->free == NULL) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ERROR, "no free buffers");
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
s->in = s->free;
|
|
|
|
s->free = s->free->next;
|
|
|
|
s->in->next = NULL;
|
|
|
|
b = s->in->buf;
|
2012-03-08 18:45:10 +01:00
|
|
|
b->pos = b->last = b->start;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
for(;;) {
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
/* find the last buf */
|
2012-03-08 16:21:22 +01:00
|
|
|
for(lin = s->in; lin->next; lin = lin->next);
|
|
|
|
b = lin->buf;
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
if (b->last == b->end) {
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
n = c->recv(c, b->last, b->end - b->last);
|
|
|
|
|
|
|
|
if (n == NGX_ERROR || n == 0) {
|
|
|
|
ngx_rtmp_close_session(s);
|
2012-03-08 18:45:10 +01:00
|
|
|
return;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
if (n == NGX_AGAIN) {
|
|
|
|
if (ngx_handle_read_event(c->read, 0) != NGX_OK) {
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_close_session(s);
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* first byte of a packet? */
|
|
|
|
if (b->last == b->start) {
|
|
|
|
h = *b->last;
|
|
|
|
s->in_hdr.hsize = hdrsizes[(h >> 6) & 0x03];
|
|
|
|
s->in_hdr.channel = h & 0x3f;
|
2012-03-08 23:39:15 +01:00
|
|
|
|
|
|
|
ngx_log_debug3(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP start %d hd=%d ch=%d",
|
|
|
|
(int)h,
|
|
|
|
(int)s->in_hdr.hsize,
|
|
|
|
(int)s->in_hdr.channel);
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
b->last += n;
|
|
|
|
if (b->last - b->pos < s->in_hdr.hsize)
|
|
|
|
continue;
|
2012-03-08 23:39:15 +01:00
|
|
|
p = b->start + 1;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
/* basic header */
|
|
|
|
do {
|
|
|
|
if (s->in_hdr.hsize < 4)
|
|
|
|
break;
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
/* FIXME: is this fix really needed?
|
2012-03-08 16:21:22 +01:00
|
|
|
if (s->in_hdr.channel == 1) {
|
|
|
|
p += 2;
|
2012-03-08 18:45:10 +01:00
|
|
|
}*/
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
/* timer:
|
|
|
|
* big-endian 3b -> little-endian 4b */
|
|
|
|
pp = (u_char*)&s->in_hdr.timer;
|
|
|
|
pp[0] = p[2];
|
|
|
|
pp[1] = p[1];
|
|
|
|
pp[2] = p[0];
|
|
|
|
pp[3] = 0;
|
2012-03-08 16:21:22 +01:00
|
|
|
if (s->in_hdr.hsize < 8)
|
|
|
|
break;
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
/* size:
|
|
|
|
* big-endian 3b -> little-endian 4b
|
|
|
|
* type:
|
|
|
|
* 1b -> 1b*/
|
2012-03-08 16:21:22 +01:00
|
|
|
p += 3;
|
2012-03-08 23:39:15 +01:00
|
|
|
pp = (u_char*)&s->in_hdr.size;
|
|
|
|
pp[0] = p[2];
|
|
|
|
pp[1] = p[1];
|
|
|
|
pp[2] = p[0];
|
|
|
|
pp[3] = 0;
|
2012-03-08 16:21:22 +01:00
|
|
|
p += 3;
|
2012-03-08 23:39:15 +01:00
|
|
|
pp = &s->in_hdr.type;
|
|
|
|
*pp = *p;
|
2012-03-08 16:21:22 +01:00
|
|
|
if (s->in_hdr.hsize < 12)
|
|
|
|
break;
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
/* stream:
|
|
|
|
* little-endian 4b -> little-endian 4b */
|
2012-03-08 16:21:22 +01:00
|
|
|
++p;
|
|
|
|
ngx_memcpy(&s->in_hdr.stream, p, 4);
|
|
|
|
p += 4;
|
|
|
|
|
|
|
|
} while(0);
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
ngx_log_debug7(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP header %s (%d) ch=%d hd=%d "
|
|
|
|
"sz=%D tm=%D st=%D",
|
|
|
|
ngx_rtmp_packet_type(s->in_hdr.type),
|
|
|
|
(int)s->in_hdr.type,
|
|
|
|
(int)s->in_hdr.channel,
|
|
|
|
(int)s->in_hdr.hsize,
|
|
|
|
s->in_hdr.size,
|
|
|
|
s->in_hdr.timer,
|
|
|
|
s->in_hdr.stream);
|
|
|
|
|
|
|
|
if (b->last < p + ngx_min(s->in_hdr.size, s->chunk_size))
|
2012-03-08 16:21:22 +01:00
|
|
|
continue;
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
b->pos = p;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
/* if fragmented then wait for more fragments */
|
2012-03-08 16:21:22 +01:00
|
|
|
if (s->in_hdr.size > s->chunk_size) {
|
|
|
|
if (s->free == NULL) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log,
|
|
|
|
NGX_ERROR, "no free buffers");
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
lin->next = s->free;
|
|
|
|
s->free = s->free->next;
|
|
|
|
lin = lin->next;
|
|
|
|
lin->next = NULL;
|
|
|
|
s->in_hdr.size -= s->chunk_size;
|
2012-03-08 18:45:10 +01:00
|
|
|
bb = lin->buf;
|
|
|
|
bb->pos = bb->last = bb->start;
|
2012-03-08 16:21:22 +01:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* handle packet! */
|
|
|
|
if (ngx_rtmp_receive_packet(s, &s->in_hdr, s->in) != NGX_OK) {
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
bb = s->in->buf;
|
2012-03-08 18:45:10 +01:00
|
|
|
bb->pos = bb->last = bb->start;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
/* copy remained data to first buffer */
|
|
|
|
if (s->in_hdr.size < b->last - b->pos) {
|
2012-03-08 18:45:10 +01:00
|
|
|
bb->last = ngx_movemem(bb->start,
|
2012-03-08 16:21:22 +01:00
|
|
|
b->pos + s->in_hdr.size,
|
|
|
|
b->last - b->pos - s->in_hdr.size);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* free all but one input buffer */
|
|
|
|
if (s->in->next) {
|
|
|
|
s->in->next->next = s->free;
|
|
|
|
s->free = s->in->next;
|
|
|
|
s->in->next = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_send(ngx_event_t *wev)
|
|
|
|
{
|
|
|
|
ngx_int_t n;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_chain_t *l, *ll;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
c = wev->data;
|
|
|
|
s = c->data;
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
if (wev->timedout) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
|
|
|
|
c->timedout = 1;
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (wev->timer_set) {
|
|
|
|
ngx_del_timer(wev);
|
|
|
|
}
|
|
|
|
|
|
|
|
while(s->out) {
|
|
|
|
|
|
|
|
l = c->send_chain(c, s->out, 0);
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
if (l == NGX_CHAIN_ERROR) {
|
2012-03-08 16:21:22 +01:00
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
n = 0;
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
if (l != s->out) {
|
|
|
|
for(ll = s->out; ll->next && ll->next != l; ll = ll->next);
|
2012-03-08 16:21:22 +01:00
|
|
|
ll->next = s->free;
|
|
|
|
s->out = l;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (l != NULL) {
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
ngx_add_timer(c->write, cscf->timeout);
|
|
|
|
if (ngx_handle_write_event(c->write, 0) != NGX_OK) {
|
|
|
|
ngx_rtmp_close_session(s);
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ngx_del_event(wev, NGX_WRITE_EVENT, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_session_t**
|
2012-03-08 16:21:22 +01:00
|
|
|
ngx_rtmp_get_session_head(ngx_rtmp_session_t *s)
|
|
|
|
{
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
|
|
|
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
return &cscf->sessions[
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_hash_key(s->name.data, s->name.len)
|
2012-03-08 16:21:22 +01:00
|
|
|
% NGX_RTMP_SESSION_HASH_SIZE];
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_join(ngx_rtmp_session_t *s, ngx_str_t *name, ngx_uint_t flags)
|
|
|
|
{
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_session_t **ps;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
|
|
|
|
c = s->connection;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
if (s->name.len) {
|
|
|
|
ngx_log_debug0(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"already joined");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP join '%V'",
|
|
|
|
&name);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
s->name = *name;
|
|
|
|
ps = ngx_rtmp_get_session_head(s);
|
2012-03-08 16:21:22 +01:00
|
|
|
s->next = *ps;
|
|
|
|
s->flags = flags;
|
|
|
|
*ps = s;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_leave(ngx_rtmp_session_t *s)
|
|
|
|
{
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_session_t **ps;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
|
|
|
|
c = s->connection;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
if (!s->name.len)
|
|
|
|
return;
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP leave '%V'",
|
|
|
|
&s->name);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
ps = ngx_rtmp_get_session_head(s);
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_str_null(&s->name);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
for(; *ps; ps = &(*ps)->next) {
|
|
|
|
if (*ps == s) {
|
|
|
|
*ps = (*ps)->next;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_send_packet(ngx_rtmp_session_t *s, ngx_rtmp_packet_hdr_t *h,
|
|
|
|
ngx_chain_t *l)
|
|
|
|
{
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_packet_hdr_t *lh;
|
|
|
|
size_t hsel, hsize, size;
|
|
|
|
ngx_chain_t *ll, **pl;
|
|
|
|
ngx_buf_t *b, *bb;
|
|
|
|
u_char *p, *pp;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
if (l == NULL)
|
|
|
|
return;
|
|
|
|
|
|
|
|
b = l->buf;
|
|
|
|
p = b->pos;
|
|
|
|
|
|
|
|
while(l) {
|
|
|
|
|
|
|
|
if (s->free == NULL) {
|
|
|
|
/* TODO: implement proper packet dropper */
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* add new output chunk */
|
|
|
|
ll = s->free;
|
|
|
|
s->free = s->free->next;
|
|
|
|
ll->next = NULL;
|
|
|
|
|
|
|
|
for(pl = &s->out; *pl; pl = &(*pl)->next);
|
|
|
|
*pl = ll;
|
|
|
|
|
|
|
|
/* put payload at the end; leave space for header */
|
|
|
|
bb = ll->buf;
|
|
|
|
bb->pos = bb->last = bb->end - s->chunk_size;
|
|
|
|
|
|
|
|
/* fill new chunk payload */
|
|
|
|
while(l && bb->pos != bb->last) {
|
|
|
|
size = ngx_min(bb->last - bb->pos, b->last - p);
|
|
|
|
bb->last = ngx_cpymem(bb->last, p, size);
|
|
|
|
p += size;
|
|
|
|
if (p != b->last)
|
|
|
|
continue;
|
|
|
|
l = l->next;
|
|
|
|
if (l) {
|
|
|
|
b = l->buf;
|
|
|
|
p = b->pos;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
lh = &s->out_hdr;
|
|
|
|
|
|
|
|
hsel = 0;
|
|
|
|
|
|
|
|
/* choose header size */
|
|
|
|
if (lh->hsize) {
|
|
|
|
|
|
|
|
if (lh->stream == h->stream)
|
|
|
|
++hsel;
|
|
|
|
|
|
|
|
if (lh->type == h->type && lh->size == h->size)
|
|
|
|
++hsel;
|
|
|
|
|
|
|
|
if (lh->timer == h->timer)
|
|
|
|
++hsel;
|
|
|
|
}
|
|
|
|
|
|
|
|
hsize = hdrsizes[hsel];
|
|
|
|
|
|
|
|
bb->pos -= hsize;
|
|
|
|
|
|
|
|
*lh = *h;
|
|
|
|
|
|
|
|
/* fill header: bb->pos..bb->pos+hsize */
|
|
|
|
|
|
|
|
pp = bb->pos;
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
*pp++ = (((uint8_t)hsel & 0x03) << 6) | (h->channel & 0x3f);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
if (hsize == 1)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
/* TODO: watch endians */
|
|
|
|
|
|
|
|
pp = ngx_cpymem(pp, &h->timer, 3);
|
|
|
|
|
|
|
|
if (hsize == 4)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
pp = ngx_cpymem(pp, &h->size, 3);
|
|
|
|
*pp++ = h->type;
|
|
|
|
|
|
|
|
if (hsize == 8)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
ngx_memcpy(pp, &h->stream, 4);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_int_t ngx_rtmp_receive_packet(ngx_rtmp_session_t *s,
|
2012-03-08 16:21:22 +01:00
|
|
|
ngx_rtmp_packet_hdr_t *h, ngx_chain_t *l)
|
|
|
|
{
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
ngx_buf_t *b;
|
|
|
|
struct {
|
|
|
|
uint16_t *v1;
|
|
|
|
uint16_t *v2;
|
|
|
|
uint16_t *v3;
|
|
|
|
} ping;
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_rtmp_session_t *ss;
|
2012-03-08 16:21:22 +01:00
|
|
|
static char invoke_name[64];
|
2012-03-08 18:45:10 +01:00
|
|
|
static ngx_rtmp_amf0_elt_t invoke_name_elt = {
|
2012-03-08 16:21:22 +01:00
|
|
|
NGX_RTMP_AMF0_STRING,
|
2012-03-08 18:45:10 +01:00
|
|
|
NULL,
|
2012-03-08 16:21:22 +01:00
|
|
|
invoke_name,
|
|
|
|
sizeof(invoke_name)
|
|
|
|
};
|
2012-03-08 23:39:15 +01:00
|
|
|
ngx_rtmp_amf0_ctx_t amf_ctx;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
c = s->connection;
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
if (l == NULL) {
|
2012-03-08 16:21:22 +01:00
|
|
|
return NGX_ERROR;
|
2012-03-08 23:39:15 +01:00
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
#ifdef NGX_DEBUG
|
|
|
|
{
|
|
|
|
int nch;
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_chain_t *ch;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
for(nch = 1, ch = l; ch->next; ch = ch->next, ++nch);
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_log_debug8(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
2012-03-08 23:39:15 +01:00
|
|
|
"RTMP packet %s (%d) ch=%d hd=%d "
|
|
|
|
"sz=%d tm=%D st=%D nbfs=%d",
|
|
|
|
ngx_rtmp_packet_type(h->type),
|
2012-03-08 18:45:10 +01:00
|
|
|
(int)h->type,
|
|
|
|
(int)h->channel,
|
|
|
|
(int)h->hsize,
|
|
|
|
(int)h->size,
|
2012-03-08 16:21:22 +01:00
|
|
|
h->timer,
|
|
|
|
h->stream,
|
2012-03-08 23:39:15 +01:00
|
|
|
nch);
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
b = l->buf;
|
|
|
|
|
|
|
|
switch(h->type) {
|
|
|
|
case NGX_RTMP_PACKET_CHUNK_SIZE:
|
|
|
|
if (b->last - b->pos < 4)
|
|
|
|
return NGX_ERROR;
|
2012-03-08 18:45:10 +01:00
|
|
|
/*ngx_rtmp_set_chunk_size(s, *(uint32_t*)(b->pos));*/
|
2012-03-08 16:21:22 +01:00
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_BYTES_READ:
|
|
|
|
if (b->last - b->pos < 4)
|
|
|
|
return NGX_ERROR;
|
2012-03-08 18:45:10 +01:00
|
|
|
/*ngx_rtmp_set_bytes_read(s, *(uint32_t*)(b->pos));*/
|
2012-03-08 16:21:22 +01:00
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_PING:
|
|
|
|
if (b->last - b->pos < 6)
|
|
|
|
return NGX_ERROR;
|
|
|
|
|
|
|
|
ping.v1 = (uint16_t*)(b->pos);
|
|
|
|
ping.v2 = (uint16_t*)(b->pos + 2);
|
|
|
|
ping.v3 = (uint16_t*)(b->pos + 4);
|
|
|
|
|
|
|
|
switch(*ping.v1) {
|
|
|
|
|
|
|
|
case NGX_RMTP_PING_CLEAR_STEAM:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RMTP_PING_CLEAR_BUFFER:
|
|
|
|
/*ngx_rtmp_clear_buffer(s);*/
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RMTP_PING_CLIENT_TIME:
|
|
|
|
/*ngx_rtmp_set_client_buffer_time(s, *ping.v3);*/
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RMTP_PING_RESET_STREAM:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RMTP_PING_PING:
|
|
|
|
/* ping client from server */
|
|
|
|
*ping.v1 = NGX_RMTP_PING_PONG;
|
|
|
|
ngx_rtmp_send_packet(s, h, l);
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RMTP_PING_PONG:
|
|
|
|
/* TODO: wtf the arg? */
|
|
|
|
/*ngx_rtmp_set_ping_time(s, *ping.v2);*/
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_SERVER_BW:
|
|
|
|
if (b->last - b->pos < 4)
|
|
|
|
return NGX_ERROR;
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
/*ngx_rtmp_set_server_bw(s, *(uint32_t*)b->pos,
|
2012-03-08 16:21:22 +01:00
|
|
|
b->last - b->pos >= 5
|
|
|
|
? *(uint8_t*)(b->pos + 4)
|
2012-03-08 18:45:10 +01:00
|
|
|
: 0);*/
|
2012-03-08 16:21:22 +01:00
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_CLIENT_BW:
|
|
|
|
if (b->last - b->pos < 4)
|
|
|
|
return NGX_ERROR;
|
|
|
|
|
2012-03-08 18:45:10 +01:00
|
|
|
/*ngx_rtmp_set_client_bw(s, *(uint32_t*)b->pos,
|
2012-03-08 16:21:22 +01:00
|
|
|
b->last - b->pos >= 5
|
|
|
|
? *(uint8_t*)(b->pos + 4)
|
2012-03-08 18:45:10 +01:00
|
|
|
: 0);*/
|
2012-03-08 16:21:22 +01:00
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_AUDIO:
|
|
|
|
case NGX_RTMP_PACKET_VIDEO:
|
|
|
|
if (!(s->flags & NGX_RTMP_PUBLISHER)) {
|
2012-03-08 23:39:15 +01:00
|
|
|
ngx_log_debug0(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"received audio/video from non-publisher");
|
2012-03-08 16:21:22 +01:00
|
|
|
return NGX_ERROR;
|
|
|
|
}
|
|
|
|
|
|
|
|
for(ss = *ngx_rtmp_get_session_head(s);
|
|
|
|
ss; ss = ss->next)
|
|
|
|
{
|
|
|
|
if (s != ss
|
|
|
|
&& ss->flags & NGX_RTMP_SUBSCRIBER
|
|
|
|
&& s->name.len == ss->name.len
|
|
|
|
&& !ngx_strncmp(s->name.data, ss->name.data,
|
|
|
|
s->name.len))
|
|
|
|
{
|
|
|
|
ngx_rtmp_send_packet(ss, h, l);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_SO:
|
|
|
|
/* TODO: implement
|
|
|
|
* plain: name, version, persistent; + lots of amf key-values
|
|
|
|
* ignore so far */
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_NOTIFY:
|
|
|
|
/* TODO: Implement HTTP callbacks on such packets
|
|
|
|
* with AMF fields converted to HTTP
|
|
|
|
* GET vars*/
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_INVOKE:
|
2012-03-08 23:39:15 +01:00
|
|
|
amf_ctx.link = &l;
|
|
|
|
amf_ctx.free = &s->free;
|
|
|
|
amf_ctx.log = c->log;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
memset(invoke_name, 0, sizeof(invoke_name));
|
2012-03-08 23:39:15 +01:00
|
|
|
if (ngx_rtmp_amf0_read(&amf_ctx, &invoke_name_elt, 1) != NGX_OK) {
|
|
|
|
ngx_log_debug0(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP invoke failed");
|
2012-03-08 16:21:22 +01:00
|
|
|
return NGX_ERROR;
|
|
|
|
}
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP invoke '%s'",
|
|
|
|
invoke_name);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
#define INVOKE_CALL(name) \
|
2012-03-08 18:45:10 +01:00
|
|
|
if (!strcasecmp(invoke_name, #name)) { \
|
2012-03-08 16:21:22 +01:00
|
|
|
return ngx_rtmp_##name(s, &l); \
|
|
|
|
}
|
|
|
|
|
|
|
|
/* NetConnection calls */
|
|
|
|
INVOKE_CALL(connect);
|
|
|
|
INVOKE_CALL(call);
|
|
|
|
INVOKE_CALL(close);
|
|
|
|
INVOKE_CALL(createstream);
|
|
|
|
|
|
|
|
/* NetStream calls */
|
|
|
|
INVOKE_CALL(play);
|
|
|
|
INVOKE_CALL(play2);
|
|
|
|
INVOKE_CALL(deletestream);
|
|
|
|
INVOKE_CALL(closestream);
|
|
|
|
INVOKE_CALL(receiveaudio);
|
|
|
|
INVOKE_CALL(receivevideo);
|
|
|
|
INVOKE_CALL(publish);
|
|
|
|
INVOKE_CALL(seek);
|
|
|
|
INVOKE_CALL(pause);
|
|
|
|
|
|
|
|
#undef INVOKE_CALL
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case NGX_RTMP_PACKET_FLEX:
|
|
|
|
case NGX_RTMP_PACKET_FLEX_SO:
|
|
|
|
case NGX_RTMP_PACKET_FLEX_MSG:
|
2012-03-08 23:39:15 +01:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"flex packets are not supported %d",
|
|
|
|
(int)h->type);
|
2012-03-08 16:21:22 +01:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
2012-03-08 23:39:15 +01:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"unsupported packet type %d",
|
|
|
|
(int)h->type);
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
return NGX_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_close_session(ngx_rtmp_session_t *s)
|
|
|
|
{
|
|
|
|
ngx_rtmp_leave(s);
|
|
|
|
ngx_rtmp_close_connection(s->connection);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_close_connection(ngx_connection_t *c)
|
|
|
|
{
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_pool_t *pool;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"close connection: %d", c->fd);
|
|
|
|
|
|
|
|
c->destroyed = 1;
|
|
|
|
pool = c->pool;
|
|
|
|
ngx_close_connection(c);
|
|
|
|
ngx_destroy_pool(pool);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
u_char *
|
|
|
|
ngx_rtmp_log_error(ngx_log_t *log, u_char *buf, size_t len)
|
|
|
|
{
|
|
|
|
u_char *p;
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_log_ctx_t *ctx;
|
|
|
|
|
|
|
|
if (log->action) {
|
|
|
|
p = ngx_snprintf(buf, len, " while %s", log->action);
|
|
|
|
len -= p - buf;
|
|
|
|
buf = p;
|
|
|
|
}
|
|
|
|
|
|
|
|
ctx = log->data;
|
|
|
|
|
|
|
|
p = ngx_snprintf(buf, len, ", client: %V", ctx->client);
|
|
|
|
len -= p - buf;
|
|
|
|
buf = p;
|
|
|
|
|
|
|
|
s = ctx->session;
|
|
|
|
|
|
|
|
if (s == NULL) {
|
|
|
|
return p;
|
|
|
|
}
|
|
|
|
|
2012-03-08 23:39:15 +01:00
|
|
|
p = ngx_snprintf(buf, len, ", server: %V", s->addr_text);
|
2012-03-08 16:21:22 +01:00
|
|
|
len -= p - buf;
|
|
|
|
buf = p;
|
|
|
|
|
|
|
|
return p;
|
|
|
|
}
|
|
|
|
|