mirror of
https://github.com/zotanmew/nginx-rtmp-module.git
synced 2024-06-29 08:18:58 +02:00
189 lines
4.3 KiB
C
189 lines
4.3 KiB
C
/*
|
|
* Copyright (c) 2012 Roman Arutyunyan
|
|
*/
|
|
|
|
|
|
#include "ngx_rtmp_eval.h"
|
|
|
|
|
|
#define NGX_RTMP_EVAL_BUFLEN 16
|
|
|
|
|
|
static void
|
|
ngx_rtmp_eval_session_str(ngx_rtmp_session_t *s, ngx_rtmp_eval_t *e,
|
|
ngx_str_t *ret)
|
|
{
|
|
*ret = *(ngx_str_t *) ((u_char *) s + e->offset);
|
|
}
|
|
|
|
|
|
static void
|
|
ngx_rtmp_eval_connection_str(ngx_rtmp_session_t *s, ngx_rtmp_eval_t *e,
|
|
ngx_str_t *ret)
|
|
{
|
|
*ret = *(ngx_str_t *) ((u_char *) s->connection + e->offset);
|
|
}
|
|
|
|
|
|
ngx_rtmp_eval_t ngx_rtmp_eval_session[] = {
|
|
|
|
{ ngx_string("app"),
|
|
ngx_rtmp_eval_session_str,
|
|
offsetof(ngx_rtmp_session_t, app) },
|
|
|
|
{ ngx_string("flashver"),
|
|
ngx_rtmp_eval_session_str,
|
|
offsetof(ngx_rtmp_session_t, flashver) },
|
|
|
|
{ ngx_string("swfurl"),
|
|
ngx_rtmp_eval_session_str,
|
|
offsetof(ngx_rtmp_session_t, swf_url) },
|
|
|
|
{ ngx_string("tcurl"),
|
|
ngx_rtmp_eval_session_str,
|
|
offsetof(ngx_rtmp_session_t, tc_url) },
|
|
|
|
{ ngx_string("pageurl"),
|
|
ngx_rtmp_eval_session_str,
|
|
offsetof(ngx_rtmp_session_t, page_url) },
|
|
|
|
{ ngx_string("addr"),
|
|
ngx_rtmp_eval_connection_str,
|
|
offsetof(ngx_connection_t, addr_text) },
|
|
|
|
ngx_rtmp_null_eval
|
|
};
|
|
|
|
|
|
static void
|
|
ngx_rtmp_eval_append(ngx_rtmp_session_t *s, ngx_buf_t *b,
|
|
void *data, size_t len)
|
|
{
|
|
size_t buf_len;
|
|
|
|
if (b->last + len > b->end) {
|
|
buf_len = 2 * (b->last - b->pos) + len;
|
|
|
|
b->start = ngx_palloc(s->connection->pool, buf_len);
|
|
if (b->start == NULL) {
|
|
return;
|
|
}
|
|
|
|
b->last = ngx_cpymem(b->start, b->pos, b->last - b->pos);
|
|
b->pos = b->start;
|
|
b->end = b->start + buf_len;
|
|
}
|
|
|
|
b->last = ngx_cpymem(b->last, data, len);
|
|
}
|
|
|
|
|
|
static void
|
|
ngx_rtmp_eval_append_var(ngx_rtmp_session_t *s, ngx_buf_t *b,
|
|
ngx_rtmp_eval_t **e, ngx_str_t *name)
|
|
{
|
|
ngx_uint_t k;
|
|
ngx_str_t v;
|
|
ngx_rtmp_eval_t *ee;
|
|
|
|
for (; *e; ++e) {
|
|
for (k = 0, ee = *e; ee->handler; ++k, ++ee) {
|
|
if (ee->name.len == name->len &&
|
|
ngx_memcmp(ee->name.data, name->data, name->len) == 0)
|
|
{
|
|
ee->handler(s, ee, &v);
|
|
ngx_rtmp_eval_append(s, b, v.data, v.len);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
ngx_int_t
|
|
ngx_rtmp_eval(ngx_rtmp_session_t *s, ngx_str_t *in, ngx_rtmp_eval_t **e,
|
|
ngx_str_t *out)
|
|
{
|
|
u_char c, *p;;
|
|
ngx_str_t name;
|
|
ngx_buf_t b;
|
|
ngx_uint_t n;
|
|
|
|
enum {
|
|
NORMAL,
|
|
ESCAPE,
|
|
NAME,
|
|
SNAME
|
|
} state = NORMAL;
|
|
|
|
b.pos = b.last = b.start = ngx_palloc(s->connection->pool,
|
|
NGX_RTMP_EVAL_BUFLEN);
|
|
if (b.pos == NULL) {
|
|
return NGX_ERROR;
|
|
}
|
|
|
|
b.end = b.pos + NGX_RTMP_EVAL_BUFLEN;
|
|
|
|
for (n = 0; n < in->len; ++n) {
|
|
p = &in->data[n];
|
|
c = *p;
|
|
|
|
switch (state) {
|
|
case SNAME:
|
|
if (c != '}') {
|
|
continue;
|
|
}
|
|
|
|
name.len = p - name.data;
|
|
ngx_rtmp_eval_append_var(s, &b, e, &name);
|
|
|
|
state = NORMAL;
|
|
|
|
continue;
|
|
|
|
case NAME:
|
|
if (c == '{' && name.data == p) {
|
|
++name.data;
|
|
state = SNAME;
|
|
continue;
|
|
}
|
|
if ((c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z')) {
|
|
continue;
|
|
}
|
|
|
|
name.len = p - name.data;
|
|
ngx_rtmp_eval_append_var(s, &b, e, &name);
|
|
|
|
case NORMAL:
|
|
switch (c) {
|
|
case '$':
|
|
name.data = p + 1;
|
|
state = NAME;
|
|
continue;
|
|
case '\\':
|
|
state = ESCAPE;
|
|
continue;
|
|
}
|
|
|
|
case ESCAPE:
|
|
ngx_rtmp_eval_append(s, &b, &c, 1);
|
|
state = NORMAL;
|
|
break;
|
|
|
|
}
|
|
}
|
|
|
|
if (state == NAME) {
|
|
p = &in->data[n];
|
|
name.len = p - name.data;
|
|
ngx_rtmp_eval_append_var(s, &b, e, &name);
|
|
}
|
|
|
|
c = 0;
|
|
ngx_rtmp_eval_append(s, &b, &c, 1);
|
|
|
|
out->data = b.pos;
|
|
out->len = b.last - b.pos - 1;
|
|
|
|
return NGX_OK;
|
|
}
|