Project

General

Profile

Feature #1657 » binary_streaming.1.5.0-svn-r2146.mod_flv_streaming.c

snailfly, 2008-05-04 09:33

 
#include <ctype.h>
#include <stdlib.h>
#include <string.h>

#include "base.h"
#include "log.h"
#include "buffer.h"
#include "response.h"
#include "stat_cache.h"

#include "plugin.h"

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

/* plugin config for all request/connections */

typedef struct {
array *extensions;
unsigned short binary_streaming;
} plugin_config;

typedef struct {
PLUGIN_DATA;

buffer *query_str;
array *get_params;

plugin_config **config_storage;

plugin_config conf;
} plugin_data;

/* init the plugin data */
INIT_FUNC(mod_flv_streaming_init) {
plugin_data *p;

UNUSED(srv);

p = calloc(1, sizeof(*p));

p->query_str = buffer_init();
p->get_params = array_init();

return p;
}

/* detroy the plugin data */
FREE_FUNC(mod_flv_streaming_free) {
plugin_data *p = p_d;

UNUSED(srv);

if (!p) return HANDLER_GO_ON;

if (p->config_storage) {
size_t i;

for (i = 0; i < srv->config_context->used; i++) {
plugin_config *s = p->config_storage[i];

if (!s) continue;

array_free(s->extensions);

free(s);
}
free(p->config_storage);
}

buffer_free(p->query_str);
array_free(p->get_params);

free(p);

return HANDLER_GO_ON;
}

/* handle plugin config and check values */

SETDEFAULTS_FUNC(mod_flv_streaming_set_defaults) {
plugin_data *p = p_d;
size_t i = 0;

config_values_t cv[] = {
{ "flv-streaming.extensions", NULL, T_CONFIG_ARRAY, T_CONFIG_SCOPE_CONNECTION }, /* 0 */
{ "flv-streaming.binary-streaming", NULL, T_CONFIG_BOOLEAN, T_CONFIG_SCOPE_CONNECTION }, /* 1 */
{ NULL, NULL, T_CONFIG_UNSET, T_CONFIG_SCOPE_UNSET }
};

if (!p) return HANDLER_ERROR;

p->config_storage = calloc(1, srv->config_context->used * sizeof(specific_config *));

for (i = 0; i < srv->config_context->used; i++) {
plugin_config *s;

s = calloc(1, sizeof(plugin_config));
s->extensions = array_init();
s->binary_streaming = 0;

cv[0].destination = s->extensions;
cv[1].destination = &(s->binary_streaming);

p->config_storage[i] = s;

if (0 != config_insert_values_global(srv, ((data_config *)srv->config_context->data[i])->value, cv)) {
return HANDLER_ERROR;
}
}

return HANDLER_GO_ON;
}

static int mod_flv_streaming_patch_connection(server *srv, connection *con, plugin_data *p) {
size_t i, j;
plugin_config *s = p->config_storage[0];

PATCH_OPTION(extensions);
PATCH_OPTION(binary_streaming);

/* skip the first, the global context */
for (i = 1; i < srv->config_context->used; i++) {
data_config *dc = (data_config *)srv->config_context->data[i];
s = p->config_storage[i];

/* condition didn't match */
if (!config_check_cond(srv, con, dc)) continue;

/* merge config */
for (j = 0; j < dc->value->used; j++) {
data_unset *du = dc->value->data[j];

if (buffer_is_equal_string(du->key, CONST_STR_LEN("flv-streaming.extensions"))) {
PATCH_OPTION(extensions);
}else if (buffer_is_equal_string(du->key, CONST_STR_LEN("flv-streaming.binary-streaming"))) {
PATCH_OPTION(binary_streaming);
}
}
}

return 0;
}

static int split_get_params(array *get_params, buffer *qrystr) {
size_t is_key = 1;
size_t i;
char *key = NULL, *val = NULL;

key = qrystr->ptr;

/* we need the \0 */
for (i = 0; i < qrystr->used; i++) {
switch(qrystr->ptr[i]) {
case '=':
if (is_key) {
val = qrystr->ptr + i + 1;

qrystr->ptr[i] = '\0';

is_key = 0;
}

break;
case '&':
case '\0': /* fin symbol */
if (!is_key) {
data_string *ds;
/* we need at least a = since the last & */

/* terminate the value */
qrystr->ptr[i] = '\0';

if (NULL == (ds = (data_string *)array_get_unused_element(get_params, TYPE_STRING))) {
ds = data_string_init();
}
buffer_copy_string_len(ds->key, key, strlen(key));
buffer_copy_string_len(ds->value, val, strlen(val));

array_insert_unique(get_params, (data_unset *)ds);
}

key = qrystr->ptr + i + 1;
val = NULL;
is_key = 1;
break;
}
}

return 0;
}

URIHANDLER_FUNC(mod_flv_streaming_path_handler) {
plugin_data *p = p_d;
int s_len;
size_t k;

UNUSED(srv);

if (buffer_is_empty(con->physical.path)) return HANDLER_GO_ON;

if (con->conf.log_request_handling) {
TRACE("-- handling %s in mod_flv_streaming", BUF_STR(con->physical.path));
}

mod_flv_streaming_patch_connection(srv, con, p);

s_len = con->physical.path->used - 1;

for (k = 0; k < p->conf.extensions->used; k++) {
data_string *ds = (data_string *)p->conf.extensions->data[k];
int ct_len = ds->value->used - 1;

if (ct_len > s_len) continue;
if (ds->value->used == 0) continue;

if (0 == strncmp(con->physical.path->ptr + s_len - ct_len, ds->value->ptr, ct_len)) {
data_string *get_param;
stat_cache_entry *sce = NULL;
buffer *b;
long start;
char *err = NULL;
/* if there is a start=[0-9]+ in the header use it as start,
* otherwise send the full file */

array_reset(p->get_params);
buffer_copy_string_buffer(p->query_str, con->uri.query);
split_get_params(p->get_params, p->query_str);

if (NULL == (get_param = (data_string *)array_get_element(p->get_params, CONST_STR_LEN("start")))) {
if (con->conf.log_request_handling) {
TRACE("start=... not found, skipping %s", BUF_STR(con->physical.path));
}

return HANDLER_GO_ON;
}

/* too short */
if (get_param->value->used < 2) {
if (con->conf.log_request_handling) {
TRACE("start=... found, but empty, skipping %s", BUF_STR(con->physical.path));
}

return HANDLER_GO_ON;
}

/* check if it is a number */
start = strtol(get_param->value->ptr, &err, 10);
if (*err != '\0') {
if (con->conf.log_request_handling) {
TRACE("parsing start '%s' as number failed, skipping %s",
BUF_STR(get_param->value), BUF_STR(con->physical.path));
}

return HANDLER_GO_ON;
}

if (start <= 0) {
if (con->conf.log_request_handling) {
TRACE("start is <= 0, skipping %s", BUF_STR(con->physical.path));
}

return HANDLER_GO_ON;
}

/* check if start is > filesize */
if (HANDLER_GO_ON != stat_cache_get_entry(srv, con, con->physical.path, &sce)) {
if (con->conf.log_request_handling) {
TRACE("stat() for %s failed", BUF_STR(con->physical.path));
}

return HANDLER_GO_ON;
}

if (start > sce->st.st_size) {
if (con->conf.log_request_handling) {
TRACE("start > file-size, skipping %s", BUF_STR(con->physical.path));
}

return HANDLER_GO_ON;
}

if(p->conf.binary_streaming){
/* we are safe now, let's build a Range request header */
//request: Range: bytes=217404-
//response: Content-Range: bytes 217404-3043664/3043665
data_string *hdr;
if (NULL == (hdr = (data_string *)array_get_unused_element(con->request.headers, TYPE_STRING))) {
hdr = data_string_init();
}
buffer_copy_string(hdr->key, "Range");
buffer_copy_string(hdr->value, "bytes=");
buffer_append_long(hdr->value, start);
buffer_append_string(hdr->value, "-");
array_insert_unique(con->request.headers, (data_unset *)hdr);
sendlen = sce->st.st_size - start;
if (con->conf.log_request_handling) {
TRACE("sending %s from position %d, size %d ( %d to %d ).", con->physical.path->ptr, start, sendlen, start, sce->st.st_size);
}
return HANDLER_GO_ON;
}else{
/* we are safe now, let's build a flv header */
b = chunkqueue_get_append_buffer(con->send);
BUFFER_COPY_STRING_CONST(b, "FLV\x1\x1\0\0\0\x9\0\0\0\x9");
chunkqueue_append_file(con->send, con->physical.path, start, sce->st.st_size - start);
response_header_overwrite(srv, con, CONST_STR_LEN("Content-Type"), CONST_STR_LEN("video/x-flv"));
con->send->is_closed = 1;
if (con->conf.log_request_handling) {
TRACE("sending %s from position %ld", BUF_STR(con->physical.path), start);
}
return HANDLER_FINISHED;
}
}
}

if (con->conf.log_request_handling) {
TRACE("none of the extensions matched %s, leaving", BUF_STR(con->physical.path));
}

/* not found */
return HANDLER_GO_ON;
}

/* this function is called at dlopen() time and inits the callbacks */

LI_EXPORT int mod_flv_streaming_plugin_init(plugin *p) {
p->version = LIGHTTPD_VERSION_ID;
p->name = buffer_init_string("flv_streaming");

p->init = mod_flv_streaming_init;
p->handle_physical = mod_flv_streaming_path_handler;
p->set_defaults = mod_flv_streaming_set_defaults;
p->cleanup = mod_flv_streaming_free;

p->data = NULL;

return 0;
}
(1-1/4)