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);
|
2012-03-13 06:41:51 +01:00
|
|
|
static ngx_int_t ngx_rtmp_receive_message(ngx_rtmp_session_t *s,
|
|
|
|
ngx_rtmp_header_t *h, ngx_chain_t *in);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
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",
|
2012-03-09 22:49:09 +01:00
|
|
|
"abort",
|
|
|
|
"ack",
|
2012-03-14 17:38:15 +01:00
|
|
|
"user",
|
2012-03-09 22:49:09 +01:00
|
|
|
"ack_size",
|
|
|
|
"bandwidth",
|
|
|
|
"edge",
|
2012-03-08 23:39:15 +01:00
|
|
|
"audio",
|
|
|
|
"video",
|
|
|
|
"?",
|
|
|
|
"?",
|
|
|
|
"?",
|
|
|
|
"?",
|
|
|
|
"?",
|
2012-03-09 22:49:09 +01:00
|
|
|
"amf3_meta",
|
|
|
|
"amf3_shared",
|
|
|
|
"amd3_cmd",
|
|
|
|
"amf0_meta",
|
|
|
|
"amf0_shared",
|
|
|
|
"amf0_cmd",
|
|
|
|
"?",
|
|
|
|
"aggregate"
|
2012-03-08 23:39:15 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
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_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-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2012-03-12 18:58:00 +01:00
|
|
|
s->in_streams = ngx_pcalloc(c->pool, sizeof(ngx_rtmp_stream_t)
|
2012-03-13 06:41:51 +01:00
|
|
|
* cscf->max_streams);
|
2012-03-12 18:58:00 +01:00
|
|
|
if (s->in_streams == NULL) {
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-11 17:17:17 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
s->in_chunk_size = NGX_RTMP_DEFAULT_CHUNK_SIZE;
|
2012-03-11 17:17:17 +01:00
|
|
|
s->in_pool = ngx_create_pool(NGX_RTMP_HANDSHAKE_SIZE + 1
|
|
|
|
+ sizeof(ngx_pool_t), c->log);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
/* start handshake */
|
2012-03-08 16:21:22 +01:00
|
|
|
b = &s->buf;
|
|
|
|
size = NGX_RTMP_HANDSHAKE_SIZE + 1;
|
2012-03-11 17:17:17 +01:00
|
|
|
b->start = b->pos = b->last = ngx_pcalloc(s->in_pool, size);
|
2012-03-08 16:21:22 +01:00
|
|
|
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);
|
|
|
|
|
2012-03-13 14:51:41 +01:00
|
|
|
if (c->destroyed) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2012-03-08 16:21:22 +01:00
|
|
|
if (rev->timedout) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
|
|
|
|
c->timedout = 1;
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
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) {
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
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-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
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-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
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-11 17:17:17 +01:00
|
|
|
ngx_reset_pool(s->in_pool);
|
|
|
|
|
2012-03-08 16:21:22 +01:00
|
|
|
c->read->handler = ngx_rtmp_recv;
|
|
|
|
c->write->handler = ngx_rtmp_send;
|
2012-03-11 17:17:17 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
ngx_log_debug0(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP handshake done");
|
|
|
|
|
2012-03-08 16:21:22 +01:00
|
|
|
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);
|
|
|
|
|
2012-03-13 14:51:41 +01:00
|
|
|
if (c->destroyed) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2012-03-08 16:21:22 +01:00
|
|
|
if (wev->timedout) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT, "client timed out");
|
|
|
|
c->timedout = 1;
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
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) {
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (n == NGX_AGAIN) {
|
|
|
|
ngx_add_timer(c->write, cscf->timeout);
|
|
|
|
if (ngx_handle_write_event(c->write, 0) != NGX_OK) {
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
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)
|
|
|
|
{
|
2012-03-09 22:49:09 +01:00
|
|
|
ngx_int_t n;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
2012-03-13 06:41:51 +01:00
|
|
|
ngx_rtmp_header_t *h;
|
|
|
|
ngx_rtmp_stream_t *st, *st0;
|
2012-03-11 17:17:17 +01:00
|
|
|
ngx_chain_t *in, *head;
|
|
|
|
ngx_buf_t *b;
|
2012-03-14 17:38:15 +01:00
|
|
|
u_char *p, *pp, *old_pos;
|
|
|
|
size_t size, fsize, old_size;
|
|
|
|
uint8_t fmt;
|
|
|
|
uint32_t csid, timestamp;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
c = rev->data;
|
|
|
|
s = c->data;
|
2012-03-11 17:17:17 +01:00
|
|
|
b = NULL;
|
2012-03-14 17:38:15 +01:00
|
|
|
old_pos = NULL;
|
|
|
|
old_size = 0;
|
2012-03-08 16:21:22 +01:00
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
2012-03-13 14:51:41 +01:00
|
|
|
if (c->destroyed) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
for( ;; ) {
|
2012-03-11 17:17:17 +01:00
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
st = &s->in_streams[s->in_csid];
|
2012-03-11 17:17:17 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
/* allocate new buffer */
|
2012-03-11 17:17:17 +01:00
|
|
|
if (st->in == NULL) {
|
2012-03-14 17:38:15 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
if ((st->in = ngx_alloc_chain_link(s->in_pool)) == NULL
|
2012-03-13 06:41:51 +01:00
|
|
|
|| (st->in->buf = ngx_calloc_buf(s->in_pool)) == NULL)
|
2012-03-11 17:17:17 +01:00
|
|
|
{
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ERROR,
|
|
|
|
"chain alloc failed");
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-11 17:17:17 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
st->in->next = NULL;
|
2012-03-14 17:38:15 +01:00
|
|
|
b = st->in->buf;
|
2012-03-11 17:17:17 +01:00
|
|
|
size = s->in_chunk_size + NGX_RTMP_MAX_CHUNK_HEADER;
|
2012-03-14 17:38:15 +01:00
|
|
|
|
|
|
|
b->start = b->last = b->pos = ngx_palloc(s->in_pool, size);
|
|
|
|
if (b->start == NULL) {
|
2012-03-11 17:17:17 +01:00
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ERROR,
|
|
|
|
"buf alloc failed");
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-13 06:41:51 +01:00
|
|
|
return;
|
2012-03-11 17:17:17 +01:00
|
|
|
}
|
2012-03-14 17:38:15 +01:00
|
|
|
b->end = b->start + size;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
h = &st->hdr;
|
|
|
|
in = st->in;
|
2012-03-14 17:38:15 +01:00
|
|
|
b = in->buf;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
if (old_size) {
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"reusing formerly read data: %d", old_size);
|
|
|
|
#if 0
|
|
|
|
/* DEBUG! */
|
|
|
|
{
|
|
|
|
size_t i;
|
|
|
|
for(i = 0; i < 16 && i < old_size; ++i) {
|
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0, "reuse %d", (int)*(old_pos + i));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
b->pos = b->start;
|
|
|
|
b->last = ngx_movemem(b->pos, old_pos, old_size);
|
2012-03-11 17:17:17 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
} else {
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
if (old_pos) {
|
|
|
|
b->pos = b->last = b->start;
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
n = c->recv(c, b->last, b->end - b->last);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
if (n == NGX_ERROR || n == 0) {
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-14 17:38:15 +01:00
|
|
|
return;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
2012-03-14 17:38:15 +01:00
|
|
|
|
|
|
|
if (n == NGX_AGAIN) {
|
|
|
|
if (ngx_handle_read_event(c->read, 0) != NGX_OK) {
|
|
|
|
ngx_rtmp_close_connection(c);
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
b->last += n;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
old_pos = NULL;
|
|
|
|
old_size = 0;
|
2012-03-09 22:49:09 +01:00
|
|
|
|
|
|
|
/* parse headers */
|
|
|
|
if (b->pos == b->start) {
|
|
|
|
p = b->pos;
|
|
|
|
|
|
|
|
/* chunk basic header */
|
2012-03-11 17:17:17 +01:00
|
|
|
fmt = (*p >> 6) & 0x03;
|
|
|
|
csid = *p++ & 0x3f;
|
2012-03-09 22:49:09 +01:00
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
if (csid == 0) {
|
2012-03-09 22:49:09 +01:00
|
|
|
if (b->last - p < 1)
|
|
|
|
continue;
|
2012-03-11 17:17:17 +01:00
|
|
|
csid = 64;
|
|
|
|
csid += *(uint8_t*)p++;
|
2012-03-09 22:49:09 +01:00
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
} else if (csid == 1) {
|
2012-03-09 22:49:09 +01:00
|
|
|
if (b->last - p < 2)
|
|
|
|
continue;
|
2012-03-11 17:17:17 +01:00
|
|
|
csid = 64;
|
|
|
|
csid += *(uint8_t*)p++;
|
|
|
|
csid += (uint32_t)256 * (*(uint8_t*)p++);
|
2012-03-09 22:49:09 +01:00
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-09 22:49:09 +01:00
|
|
|
ngx_log_debug2(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP bheader fmt=%d csid=%D",
|
2012-03-11 17:17:17 +01:00
|
|
|
(int)fmt, csid);
|
|
|
|
|
|
|
|
if (csid >= cscf->max_streams) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ERROR,
|
|
|
|
"RTMP chunk stream too big: %D >= %D",
|
|
|
|
csid, cscf->max_streams);
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-11 17:17:17 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* link orphan */
|
2012-03-13 06:41:51 +01:00
|
|
|
if (s->in_csid == 0) {
|
2012-03-11 17:17:17 +01:00
|
|
|
|
|
|
|
/* unlink from stream #0 */
|
|
|
|
st->in = st->in->next;
|
|
|
|
|
|
|
|
/* link to new stream */
|
2012-03-13 06:41:51 +01:00
|
|
|
s->in_csid = csid;
|
|
|
|
st = &s->in_streams[csid];
|
2012-03-11 17:17:17 +01:00
|
|
|
if (st->in == NULL) {
|
|
|
|
in->next = in;
|
|
|
|
} else {
|
|
|
|
in->next = st->in->next;
|
|
|
|
st->in->next = in;
|
|
|
|
}
|
|
|
|
st->in = in;
|
|
|
|
h = &st->hdr;
|
|
|
|
h->csid = csid;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* get previous header to inherit data from */
|
|
|
|
timestamp = h->timestamp;
|
2012-03-09 22:49:09 +01:00
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
if (fmt <= 2 ) {
|
2012-03-09 22:49:09 +01:00
|
|
|
if (b->last - p < 3)
|
|
|
|
continue;
|
|
|
|
/* timestamp:
|
|
|
|
* big-endian 3b -> little-endian 4b */
|
|
|
|
pp = (u_char*)×tamp;
|
|
|
|
pp[2] = *p++;
|
|
|
|
pp[1] = *p++;
|
|
|
|
pp[0] = *p++;
|
|
|
|
pp[3] = 0;
|
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
if (fmt <= 1) {
|
2012-03-09 22:49:09 +01:00
|
|
|
if (b->last - p < 4)
|
|
|
|
continue;
|
|
|
|
/* size:
|
|
|
|
* big-endian 3b -> little-endian 4b
|
|
|
|
* type:
|
|
|
|
* 1b -> 1b*/
|
|
|
|
pp = (u_char*)&h->mlen;
|
|
|
|
pp[2] = *p++;
|
|
|
|
pp[1] = *p++;
|
|
|
|
pp[0] = *p++;
|
|
|
|
pp[3] = 0;
|
|
|
|
h->type = *(uint8_t*)p++;
|
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
if (fmt == 0) {
|
2012-03-09 22:49:09 +01:00
|
|
|
if (b->last - p < 4)
|
|
|
|
continue;
|
|
|
|
/* stream:
|
|
|
|
* little-endian 4b -> little-endian 4b */
|
|
|
|
pp = (u_char*)&h->msid;
|
|
|
|
pp[0] = *p++;
|
|
|
|
pp[1] = *p++;
|
|
|
|
pp[2] = *p++;
|
|
|
|
pp[3] = *p++;
|
|
|
|
}
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-09 22:49:09 +01:00
|
|
|
/* extended header */
|
|
|
|
if (timestamp == 0x00ffffff) {
|
|
|
|
if (b->last - p < 4)
|
|
|
|
continue;
|
|
|
|
pp = (u_char*)&h->timestamp;
|
|
|
|
pp[3] = *p++;
|
|
|
|
pp[2] = *p++;
|
|
|
|
pp[1] = *p++;
|
|
|
|
pp[0] = *p++;
|
2012-03-13 06:41:51 +01:00
|
|
|
} else if (fmt) {
|
2012-03-09 22:49:09 +01:00
|
|
|
h->timestamp += timestamp;
|
|
|
|
} else {
|
|
|
|
h->timestamp = timestamp;
|
|
|
|
}
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
ngx_log_debug6(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
2012-03-09 22:49:09 +01:00
|
|
|
"RTMP mheader %s (%d) "
|
2012-03-14 17:38:15 +01:00
|
|
|
"timestamp=%D mlen=%D len=%D msid=%D",
|
2012-03-09 22:49:09 +01:00
|
|
|
ngx_rtmp_packet_type(h->type), (int)h->type,
|
2012-03-14 17:38:15 +01:00
|
|
|
h->timestamp, h->mlen, st->len, h->msid);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-09 22:49:09 +01:00
|
|
|
/* header done */
|
|
|
|
b->pos = p;
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
size = b->last - b->pos;
|
2012-03-14 17:38:15 +01:00
|
|
|
fsize = h->mlen - st->len;
|
2012-03-11 17:17:17 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
if (size < ngx_min(fsize, s->in_chunk_size))
|
2012-03-08 16:21:22 +01:00
|
|
|
continue;
|
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
/* buffer is ready */
|
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
if (fsize > s->in_chunk_size) {
|
2012-03-11 17:17:17 +01:00
|
|
|
/* collect fragmented chunks */
|
2012-03-14 17:38:15 +01:00
|
|
|
st->len += s->in_chunk_size;
|
|
|
|
|
|
|
|
old_pos = b->pos + s->in_chunk_size;
|
|
|
|
old_size = size - s->in_chunk_size;
|
2012-03-11 17:17:17 +01:00
|
|
|
|
|
|
|
} else {
|
|
|
|
/* handle! */
|
|
|
|
head = st->in->next;
|
|
|
|
st->in->next = NULL;
|
2012-03-14 17:38:15 +01:00
|
|
|
old_pos = b->pos + fsize;
|
|
|
|
old_size = size - fsize;
|
|
|
|
st->len = 0;
|
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
if (ngx_rtmp_receive_message(s, h, head) != NGX_OK) {
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2012-03-11 17:17:17 +01:00
|
|
|
/* add used bufs to stream #0 */
|
2012-03-12 18:58:00 +01:00
|
|
|
st0 = &s->in_streams[0];
|
2012-03-14 07:06:47 +01:00
|
|
|
st->in->next = st0->in;
|
|
|
|
st0->in = head;
|
|
|
|
st->in = NULL;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
s->in_csid = 0;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
#define ngx_rtmp_buf_addref(b) \
|
2012-03-13 14:51:41 +01:00
|
|
|
(++*(int*)&((b)->tag))
|
2012-03-12 00:44:32 +01:00
|
|
|
|
|
|
|
|
|
|
|
#define ngx_rtmp_buf_release(b) \
|
2012-03-13 14:51:41 +01:00
|
|
|
(--*(int*)&((b)->tag))
|
2012-03-12 00:44:32 +01:00
|
|
|
|
|
|
|
|
|
|
|
static void
|
2012-03-08 16:21:22 +01:00
|
|
|
ngx_rtmp_send(ngx_event_t *wev)
|
|
|
|
{
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_connection_t *c;
|
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
2012-03-14 07:06:47 +01:00
|
|
|
ngx_chain_t *out, *l;
|
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);
|
|
|
|
|
2012-03-13 14:51:41 +01:00
|
|
|
if (c->destroyed) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2012-03-08 16:21:22 +01:00
|
|
|
if (wev->timedout) {
|
2012-03-09 22:49:09 +01:00
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, NGX_ETIMEDOUT,
|
|
|
|
"client timed out");
|
2012-03-08 16:21:22 +01:00
|
|
|
c->timedout = 1;
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (wev->timer_set) {
|
|
|
|
ngx_del_timer(wev);
|
|
|
|
}
|
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
while (s->out) {
|
2012-03-12 00:44:32 +01:00
|
|
|
out = c->send_chain(c, s->out, 0);
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
if (out == NGX_CHAIN_ERROR) {
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2012-03-14 07:06:47 +01:00
|
|
|
if (out == s->out
|
|
|
|
&& s->out->buf->pos == s->out->buf->last)
|
|
|
|
{
|
2012-03-08 16:21:22 +01:00
|
|
|
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) {
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_close_connection(c);
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
2012-03-09 22:49:09 +01:00
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
while (s->out) {
|
2012-03-12 00:44:32 +01:00
|
|
|
|
2012-03-14 07:06:47 +01:00
|
|
|
l = s->out;
|
|
|
|
if (l->buf->pos < l->buf->last) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
s->out = s->out->next;
|
2012-03-14 17:38:15 +01:00
|
|
|
l->next = NULL;
|
2012-03-14 07:06:47 +01:00
|
|
|
|
|
|
|
/* anyone still using this buffer? */
|
|
|
|
if (ngx_rtmp_buf_release(l->buf)) {
|
2012-03-14 17:38:15 +01:00
|
|
|
ngx_log_debug0(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"keeping shared buffer");
|
2012-03-14 07:06:47 +01:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* return buffer to core */
|
|
|
|
if (ngx_rtmp_release_shared_buf(s, l)) {
|
|
|
|
ngx_rtmp_close_connection(c);
|
|
|
|
return;
|
|
|
|
}
|
2012-03-12 00:44:32 +01:00
|
|
|
|
2012-03-14 07:06:47 +01:00
|
|
|
if (s->out == out) {
|
|
|
|
break;
|
2012-03-12 00:44:32 +01:00
|
|
|
}
|
2012-03-09 22:49:09 +01:00
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
ngx_del_event(wev, NGX_WRITE_EVENT, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
ngx_chain_t *
|
|
|
|
ngx_rtmp_alloc_shared_buf(ngx_rtmp_session_t *s)
|
2012-03-08 16:21:22 +01:00
|
|
|
{
|
2012-03-12 00:44:32 +01:00
|
|
|
ngx_chain_t *out;
|
|
|
|
ngx_buf_t *b;
|
|
|
|
size_t size;
|
2012-03-08 16:21:22 +01:00
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
2012-03-14 07:06:47 +01:00
|
|
|
ngx_connection_t *c;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-14 07:06:47 +01:00
|
|
|
c = s->connection;
|
2012-03-08 16:21:22 +01:00
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
2012-03-12 00:44:32 +01:00
|
|
|
|
2012-03-12 18:58:00 +01:00
|
|
|
if (cscf->out_free) {
|
|
|
|
out = cscf->out_free;
|
|
|
|
cscf->out_free = out->next;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-14 07:06:47 +01:00
|
|
|
ngx_log_debug0(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"reuse shared buf");
|
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
} else {
|
2012-03-14 07:06:47 +01:00
|
|
|
|
|
|
|
ngx_log_debug0(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"alloc shared buf");
|
|
|
|
|
2012-03-12 18:58:00 +01:00
|
|
|
out = ngx_alloc_chain_link(cscf->out_pool);
|
2012-03-12 00:44:32 +01:00
|
|
|
if (out == NULL) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 18:58:00 +01:00
|
|
|
out->buf = ngx_calloc_buf(cscf->out_pool);
|
2012-03-12 00:44:32 +01:00
|
|
|
if (out->buf == NULL) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2012-03-08 18:45:10 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
size = cscf->out_chunk_size + NGX_RTMP_MAX_CHUNK_HEADER;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
b = out->buf;
|
2012-03-12 18:58:00 +01:00
|
|
|
b->start = ngx_palloc(cscf->out_pool, size);
|
2012-03-12 00:44:32 +01:00
|
|
|
b->end = b->start + size;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
out->next = NULL;
|
|
|
|
b = out->buf;
|
|
|
|
b->pos = b->last = b->start + NGX_RTMP_MAX_CHUNK_HEADER;
|
|
|
|
b->tag = (ngx_buf_tag_t)0;
|
2012-03-13 14:51:41 +01:00
|
|
|
b->memory = 1;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
return out;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-03-14 07:06:47 +01:00
|
|
|
ngx_int_t
|
|
|
|
ngx_rtmp_release_shared_buf(ngx_rtmp_session_t *s,
|
|
|
|
ngx_chain_t *out)
|
|
|
|
{
|
|
|
|
ngx_rtmp_core_srv_conf_t *cscf;
|
|
|
|
size_t nbufs;
|
|
|
|
ngx_chain_t *cl;
|
|
|
|
|
|
|
|
cscf = ngx_rtmp_get_module_srv_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
for(nbufs = 1, cl = out; cl->next;
|
|
|
|
cl = cl->next, ++nbufs);
|
|
|
|
|
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, s->connection->log, 0,
|
|
|
|
"release %d shared bufs", nbufs);
|
|
|
|
|
|
|
|
cl->next = cscf->out_free;
|
|
|
|
cscf->out_free = out;
|
|
|
|
|
|
|
|
return NGX_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-03-08 16:21:22 +01:00
|
|
|
void
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_prepare_message(ngx_rtmp_session_t *s, ngx_rtmp_header_t *h,
|
|
|
|
ngx_chain_t *out, uint8_t fmt)
|
2012-03-08 16:21:22 +01:00
|
|
|
{
|
2012-03-12 00:44:32 +01:00
|
|
|
ngx_chain_t *l;
|
2012-03-13 06:41:51 +01:00
|
|
|
u_char *p, *pp;
|
2012-03-12 00:44:32 +01:00
|
|
|
ngx_int_t hsize, thsize, nbufs;
|
|
|
|
uint32_t mlen, timestamp, ext_timestamp;
|
|
|
|
static uint8_t hdrsize[] = { 12, 8, 4, 1 };
|
2012-03-14 17:38:15 +01:00
|
|
|
u_char th[3];
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-09 22:49:09 +01:00
|
|
|
/* detect packet size */
|
|
|
|
mlen = 0;
|
|
|
|
nbufs = 0;
|
2012-03-12 00:44:32 +01:00
|
|
|
for(l = out; l; l = l->next) {
|
2012-03-14 17:38:15 +01:00
|
|
|
mlen += (l->buf->last - l->buf->pos);
|
2012-03-09 22:49:09 +01:00
|
|
|
++nbufs;
|
|
|
|
}
|
2012-03-13 14:51:41 +01:00
|
|
|
|
2012-03-14 07:06:47 +01:00
|
|
|
ngx_log_debug8(NGX_LOG_DEBUG_RTMP, s->connection->log, 0,
|
|
|
|
"RTMP prep %s (%d) fmt=%d csid=%D timestamp=%D "
|
2012-03-09 22:49:09 +01:00
|
|
|
"mlen=%D msid=%D nbufs=%d",
|
2012-03-14 07:06:47 +01:00
|
|
|
ngx_rtmp_packet_type(h->type), (int)h->type, (int)fmt,
|
2012-03-09 22:49:09 +01:00
|
|
|
h->csid, h->timestamp, mlen, h->msid, nbufs);
|
2012-03-13 14:51:41 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
/* determine initial header size */
|
|
|
|
hsize = hdrsize[fmt];
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
if (h->timestamp >= 0x00ffffff) {
|
|
|
|
timestamp = 0x00ffffff;
|
|
|
|
ext_timestamp = h->timestamp;
|
|
|
|
hsize += 4;
|
|
|
|
} else {
|
|
|
|
timestamp = h->timestamp;
|
2012-03-09 22:49:09 +01:00
|
|
|
ext_timestamp = 0;
|
2012-03-12 00:44:32 +01:00
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
if (h->csid >= 64) {
|
|
|
|
++hsize;
|
|
|
|
if (h->csid >= 320) {
|
2012-03-09 22:49:09 +01:00
|
|
|
++hsize;
|
|
|
|
}
|
2012-03-12 00:44:32 +01:00
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
/* fill initial header */
|
|
|
|
out->buf->pos -= hsize;
|
|
|
|
p = out->buf->pos;
|
|
|
|
|
|
|
|
/* basic header */
|
|
|
|
*p = (fmt << 6);
|
|
|
|
if (h->csid >= 2 && h->csid <= 63) {
|
|
|
|
*p++ |= (((uint8_t)h->csid) & 0x3f);
|
|
|
|
} else if (h->csid >= 64 && h->csid < 320) {
|
|
|
|
++p;
|
|
|
|
*p++ = (uint8_t)(h->csid - 64);
|
|
|
|
} else {
|
|
|
|
*p++ |= 1;
|
|
|
|
*p++ = (uint8_t)(h->csid - 64);
|
|
|
|
*p++ = (uint8_t)((h->csid - 64) >> 8);
|
|
|
|
}
|
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
/* create fmt3 header for successive fragments */
|
2012-03-13 06:41:51 +01:00
|
|
|
thsize = p - out->buf->pos;
|
2012-03-14 17:38:15 +01:00
|
|
|
ngx_memcpy(th, out->buf->pos, thsize);
|
|
|
|
th[0] |= 0xc0;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
/* message header */
|
|
|
|
if (fmt <= 2) {
|
|
|
|
pp = (u_char*)×tamp;
|
|
|
|
*p++ = pp[2];
|
|
|
|
*p++ = pp[1];
|
|
|
|
*p++ = pp[0];
|
|
|
|
if (fmt <= 1) {
|
|
|
|
pp = (u_char*)&mlen;
|
2012-03-09 22:49:09 +01:00
|
|
|
*p++ = pp[2];
|
|
|
|
*p++ = pp[1];
|
|
|
|
*p++ = pp[0];
|
2012-03-12 00:44:32 +01:00
|
|
|
*p++ = h->type;
|
|
|
|
if (fmt == 0) {
|
|
|
|
pp = (u_char*)&h->msid;
|
2012-03-09 22:49:09 +01:00
|
|
|
*p++ = pp[0];
|
2012-03-12 00:44:32 +01:00
|
|
|
*p++ = pp[1];
|
|
|
|
*p++ = pp[2];
|
|
|
|
*p++ = pp[3];
|
2012-03-09 22:49:09 +01:00
|
|
|
}
|
|
|
|
}
|
2012-03-12 00:44:32 +01:00
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
/* extended header */
|
|
|
|
if (ext_timestamp) {
|
|
|
|
pp = (u_char*)&ext_timestamp;
|
|
|
|
*p++ = pp[3];
|
|
|
|
*p++ = pp[2];
|
|
|
|
*p++ = pp[1];
|
|
|
|
*p++ = pp[0];
|
|
|
|
}
|
|
|
|
|
2012-03-14 17:38:15 +01:00
|
|
|
/* append headers to successive fragments */
|
2012-03-12 00:44:32 +01:00
|
|
|
for(out = out->next; out; out = out->next) {
|
2012-03-13 14:51:41 +01:00
|
|
|
out->buf->pos -= thsize;
|
2012-03-14 17:38:15 +01:00
|
|
|
ngx_memcpy(out->buf->pos, th, thsize);
|
2012-03-12 00:44:32 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_int_t
|
2012-03-12 00:44:32 +01:00
|
|
|
ngx_rtmp_send_message(ngx_rtmp_session_t *s, ngx_chain_t *out)
|
|
|
|
{
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_chain_t *l, **ll;
|
|
|
|
size_t nbytes, nbufs;
|
|
|
|
ngx_connection_t *c;
|
|
|
|
|
|
|
|
c = s->connection;
|
|
|
|
nbytes = 0;
|
|
|
|
nbufs = 0;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
for(l = out; l; l = l->next) {
|
|
|
|
ngx_rtmp_buf_addref(l->buf);
|
2012-03-13 14:51:41 +01:00
|
|
|
nbytes += (l->buf->last - l->buf->pos);
|
|
|
|
++nbufs;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
2012-03-09 22:49:09 +01:00
|
|
|
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_log_debug2(NGX_LOG_DEBUG_RTMP, s->connection->log, 0,
|
|
|
|
"RTMP send nbytes=%d, nbufs=%d",
|
|
|
|
nbytes, nbufs);
|
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
/* TODO: optimize lookup */
|
|
|
|
/* TODO: implement dropper */
|
|
|
|
for(ll = &s->out; *ll; ll = &(*ll)->next);
|
|
|
|
*ll = out;
|
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
ngx_rtmp_send(s->connection->write);
|
2012-03-13 14:51:41 +01:00
|
|
|
|
|
|
|
return c->destroyed ? NGX_ERROR : NGX_OK;
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-03-12 00:44:32 +01:00
|
|
|
static ngx_int_t
|
|
|
|
ngx_rtmp_receive_message(ngx_rtmp_session_t *s,
|
2012-03-12 18:58:00 +01:00
|
|
|
ngx_rtmp_header_t *h, ngx_chain_t *in)
|
2012-03-08 16:21:22 +01:00
|
|
|
{
|
2012-03-12 18:58:00 +01:00
|
|
|
ngx_rtmp_core_main_conf_t *cmcf;
|
|
|
|
ngx_array_t *evhs;
|
|
|
|
size_t n;
|
2012-03-13 06:41:51 +01:00
|
|
|
ngx_rtmp_event_handler_pt *evh;
|
|
|
|
ngx_connection_t *c;
|
2012-03-09 22:49:09 +01:00
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
c = s->connection;
|
2012-03-12 18:58:00 +01:00
|
|
|
cmcf = ngx_rtmp_get_module_main_conf(s, ngx_rtmp_core_module);
|
2012-03-09 22:49:09 +01:00
|
|
|
|
2012-03-08 16:21:22 +01:00
|
|
|
#ifdef NGX_DEBUG
|
|
|
|
{
|
2012-03-09 22:49:09 +01:00
|
|
|
int nbufs;
|
2012-03-08 18:45:10 +01:00
|
|
|
ngx_chain_t *ch;
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
for(nbufs = 1, ch = in;
|
2012-03-09 22:49:09 +01:00
|
|
|
ch->next;
|
|
|
|
ch = ch->next, ++nbufs);
|
|
|
|
|
|
|
|
ngx_log_debug7(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"RTMP recv %s (%d) csid=%D timestamp=%D "
|
|
|
|
"mlen=%D msid=%D nbufs=%d",
|
|
|
|
ngx_rtmp_packet_type(h->type), (int)h->type,
|
|
|
|
h->csid, h->timestamp, h->mlen, h->msid, nbufs);
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2012-03-12 18:58:00 +01:00
|
|
|
if (h->type >= NGX_RTMP_MSG_MAX) {
|
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"unexpected RTMP message type: %d", (int)h->type);
|
|
|
|
return NGX_OK;
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-12 18:58:00 +01:00
|
|
|
evhs = &cmcf->events[h->type];
|
2012-03-13 06:41:51 +01:00
|
|
|
evh = evhs->elts;
|
2012-03-14 17:38:15 +01:00
|
|
|
|
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"nhandlers: %d", evhs->nelts);
|
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
for(n = 0; n < evhs->nelts; ++n, ++evh) {
|
2012-03-12 18:58:00 +01:00
|
|
|
if (!evh) {
|
|
|
|
continue;
|
|
|
|
}
|
2012-03-14 17:38:15 +01:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"calling handler %d", n);
|
2012-03-12 18:58:00 +01:00
|
|
|
|
2012-03-13 06:41:51 +01:00
|
|
|
if ((*evh)(s, h, in) != NGX_OK) {
|
2012-03-14 17:38:15 +01:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_RTMP, c->log, 0,
|
|
|
|
"handler %d failed", n);
|
2012-03-12 18:58:00 +01:00
|
|
|
return NGX_ERROR;
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
return NGX_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void
|
|
|
|
ngx_rtmp_close_connection(ngx_connection_t *c)
|
|
|
|
{
|
2012-03-13 14:51:41 +01:00
|
|
|
ngx_rtmp_session_t *s;
|
|
|
|
ngx_pool_t *pool;
|
|
|
|
ngx_rtmp_core_main_conf_t *cmcf;
|
|
|
|
ngx_rtmp_disconnect_handler_pt *h;
|
|
|
|
size_t n;
|
|
|
|
|
|
|
|
if (c->destroyed) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
s = c->data;
|
|
|
|
cmcf = ngx_rtmp_get_module_main_conf(s, ngx_rtmp_core_module);
|
|
|
|
|
|
|
|
ngx_log_debug0(NGX_LOG_DEBUG_RTMP, c->log, 0, "close connection");
|
2012-03-08 16:21:22 +01:00
|
|
|
|
2012-03-13 14:51:41 +01:00
|
|
|
if (s) {
|
|
|
|
h = cmcf->disconnect.elts;
|
|
|
|
for(n = 0; n < cmcf->disconnect.nelts; ++n, ++h) {
|
|
|
|
if (*h) {
|
|
|
|
(*h)(s);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (s->in_pool) {
|
|
|
|
ngx_destroy_pool(s->in_pool);
|
|
|
|
}
|
|
|
|
}
|
2012-03-08 16:21:22 +01:00
|
|
|
|
|
|
|
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;
|
|
|
|
}
|