|
#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;
|
|
}
|