776 lines
19 KiB
C++
776 lines
19 KiB
C++
|
#include "pattern_replace.h"
|
||
|
|
#include "http_lua.h"
|
||
|
|
|
||
|
|
#include <tfe_utils.h>
|
||
|
|
#include <tfe_http.h>
|
||
|
|
|
||
|
|
#include <string.h>
|
||
|
|
#include <sys/types.h>
|
||
|
|
#include <sys/stat.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <assert.h>
|
||
|
|
|
||
|
|
static const char __attribute__((__unused__))*http_lua_log_get_url(struct tsg_script_ctx *tsg_ctx)
|
||
|
|
{
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
if(events < EV_HTTP_REQ_HDR)
|
||
|
|
{
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
struct tfe_http_half * in_req_half = session->req;
|
||
|
|
if(in_req_half == NULL)
|
||
|
|
return NULL;
|
||
|
|
struct tfe_http_req_spec * in_req_spec = &in_req_half->req_spec;
|
||
|
|
if(in_req_spec == NULL)
|
||
|
|
return NULL;
|
||
|
|
|
||
|
|
if (in_req_spec->url != NULL)
|
||
|
|
{
|
||
|
|
return in_req_spec->url;
|
||
|
|
}
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_get_param_from_lua(struct elua_vm *vm, struct elua_data **out_lua_argv)
|
||
|
|
{
|
||
|
|
int i=0, out_lua_argc=0;
|
||
|
|
struct elua_data *lua_argv=NULL;
|
||
|
|
|
||
|
|
out_lua_argc = elua_cbinding_get_input_params_num(vm);
|
||
|
|
if(out_lua_argc <= 0)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
lua_argv = ALLOC(struct elua_data, out_lua_argc);
|
||
|
|
for (i = 0; i < out_lua_argc; i++)
|
||
|
|
{
|
||
|
|
elua_cbinding_get_input_param(vm, i+1, &lua_argv[i]);
|
||
|
|
}
|
||
|
|
*out_lua_argv = lua_argv;
|
||
|
|
|
||
|
|
return out_lua_argc;
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_free_params(struct elua_data *out_lua_argv)
|
||
|
|
{
|
||
|
|
if(out_lua_argv)
|
||
|
|
FREE(&out_lua_argv);
|
||
|
|
}
|
||
|
|
|
||
|
|
int http_set_string_to_lua(struct elua_vm *vm, const char *buff, size_t len)
|
||
|
|
{
|
||
|
|
struct elua_data edata;
|
||
|
|
edata.type = STRING;
|
||
|
|
edata.buff = (char *)buff;
|
||
|
|
edata.len = len;
|
||
|
|
elua_cbinding_append_output_params(vm, &edata, 1);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
int http_set_num_to_lua(struct elua_vm *vm, long num)
|
||
|
|
{
|
||
|
|
struct elua_data retvalue;
|
||
|
|
retvalue.type = INTEGER;
|
||
|
|
retvalue.len = sizeof(long);
|
||
|
|
retvalue.integer = num;
|
||
|
|
elua_cbinding_append_output_params(vm, &retvalue, 1);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
int http_set_table_to_lua(struct elua_vm *vm, struct elua_table *table)
|
||
|
|
{
|
||
|
|
struct elua_data retvalue;
|
||
|
|
retvalue.type = TABLE;
|
||
|
|
retvalue.table = table;
|
||
|
|
retvalue.len = sizeof(struct elua_data);
|
||
|
|
elua_cbinding_append_output_params(vm, &retvalue, 1);
|
||
|
|
elua_destroy_table(table);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_log_debug(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
int out_lua_argc=0, i=0;
|
||
|
|
struct elua_data *out_lua_argv=NULL;
|
||
|
|
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
out_lua_argc = http_get_param_from_lua(vm, &out_lua_argv);
|
||
|
|
if(out_lua_argc < 1 || out_lua_argv == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char buff[TFE_STRING_MAX]={0}, *p = NULL;
|
||
|
|
p = buff;
|
||
|
|
for(i=0; i<out_lua_argc; i++)
|
||
|
|
{
|
||
|
|
char *in=out_lua_argv[i].buff;
|
||
|
|
size_t in_sz=out_lua_argv[i].len;
|
||
|
|
if(in == NULL || in_sz ==0 || (sizeof(buff) - (p - buff)) <= 0)
|
||
|
|
{
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
p += snprintf(p, sizeof(buff) - (p - buff), " %s", in);
|
||
|
|
}
|
||
|
|
|
||
|
|
TFE_LOG_DEBUG(tsg_ctx->local_logger, "policy_id:%d, profile_id:%d, message:%20s", tsg_ctx->config_id, tsg_ctx->profile_id, buff);
|
||
|
|
|
||
|
|
http_free_params(out_lua_argv);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_log_info(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
int out_lua_argc=0, i=0;
|
||
|
|
struct elua_data *out_lua_argv=NULL;
|
||
|
|
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
out_lua_argc = http_get_param_from_lua(vm, &out_lua_argv);
|
||
|
|
if(out_lua_argc < 1 || out_lua_argv == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char buff[TFE_STRING_MAX]={0}, *p = NULL;
|
||
|
|
p = buff;
|
||
|
|
for(i=0; i<out_lua_argc; i++)
|
||
|
|
{
|
||
|
|
char *in=out_lua_argv[i].buff;
|
||
|
|
size_t in_sz=out_lua_argv[i].len;
|
||
|
|
if(in == NULL || in_sz ==0 )
|
||
|
|
{
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
p += snprintf(p, sizeof(buff) - (p - buff), " %s", in);
|
||
|
|
}
|
||
|
|
|
||
|
|
TFE_LOG_INFO(tsg_ctx->local_logger, "policy_id:%d, profile_id:%d, message:%s", tsg_ctx->config_id, tsg_ctx->profile_id, buff);
|
||
|
|
|
||
|
|
http_free_params(out_lua_argv);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_log_error(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
int out_lua_argc=0, i=0;
|
||
|
|
struct elua_data *out_lua_argv=NULL;
|
||
|
|
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
out_lua_argc = http_get_param_from_lua(vm, &out_lua_argv);
|
||
|
|
if(out_lua_argc < 1 || out_lua_argv == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char buff[TFE_STRING_MAX]={0}, *p = NULL;
|
||
|
|
p = buff;
|
||
|
|
for(i=0; i<out_lua_argc; i++)
|
||
|
|
{
|
||
|
|
char *in=out_lua_argv[i].buff;
|
||
|
|
size_t in_sz=out_lua_argv[i].len;
|
||
|
|
if(in == NULL || in_sz ==0 )
|
||
|
|
{
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
p += snprintf(p, sizeof(buff) - (p - buff), " %s", in);
|
||
|
|
}
|
||
|
|
|
||
|
|
TFE_LOG_ERROR(tsg_ctx->local_logger, "policy_id:%d, profile_id:%d, message:%s", tsg_ctx->config_id, tsg_ctx->profile_id, buff);
|
||
|
|
|
||
|
|
http_free_params(out_lua_argv);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_get_method(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
if(tfe_http_in_response(events))
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct tfe_http_half * in_req_half = session->req;
|
||
|
|
struct tfe_http_req_spec * in_req_spec = &in_req_half->req_spec;
|
||
|
|
|
||
|
|
const char *method_str = http_std_method_to_string(in_req_spec->method);
|
||
|
|
if(method_str != NULL)
|
||
|
|
{
|
||
|
|
http_set_string_to_lua(vm, method_str, strlen(method_str));
|
||
|
|
}
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_get_uri(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
if(tfe_http_in_response(events) || !tsg_ctx->http_req_uri)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct tfe_http_half * in_req_half = session->req;
|
||
|
|
struct tfe_http_req_spec * in_req_spec = &in_req_half->req_spec;
|
||
|
|
if (in_req_spec->uri != NULL)
|
||
|
|
{
|
||
|
|
http_set_string_to_lua(vm, in_req_spec->uri, strlen(in_req_spec->uri));
|
||
|
|
}
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_set_uri(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
int out_lua_argc = 0;
|
||
|
|
struct elua_data *out_lua_argv = NULL;
|
||
|
|
|
||
|
|
out_lua_argc = http_get_param_from_lua(vm, &out_lua_argv);
|
||
|
|
if(out_lua_argc != 1 || out_lua_argv == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char * rewrite_uri = out_lua_argv[0].buff;
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
if(tfe_http_in_response(events) || !tsg_ctx->http_req_uri)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct tfe_http_half * in_req_half = session->req;
|
||
|
|
struct tfe_http_req_spec * in_req_spec = &in_req_half->req_spec;
|
||
|
|
if (in_req_spec->uri != NULL)
|
||
|
|
{
|
||
|
|
tsg_ctx->execut_lua_sucess=1;
|
||
|
|
tsg_ctx->rewrite_uri = tfe_strdup(rewrite_uri);
|
||
|
|
}
|
||
|
|
http_free_params(out_lua_argv);
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_get_status_code(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
if(tfe_http_in_request(events))
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
struct tfe_http_half * in_resp_half = session->resp;
|
||
|
|
struct tfe_http_resp_spec * in_resp_spec = &in_resp_half->resp_spec;
|
||
|
|
|
||
|
|
if (in_resp_spec != NULL)
|
||
|
|
{
|
||
|
|
http_set_num_to_lua(vm, in_resp_spec->resp_code);
|
||
|
|
}
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_set_status_code(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
int out_lua_argc = 0;
|
||
|
|
struct elua_data *out_lua_argv = NULL;
|
||
|
|
|
||
|
|
out_lua_argc = http_get_param_from_lua(vm, &out_lua_argv);
|
||
|
|
if(out_lua_argc != 1 || out_lua_argv == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int lua_resp_code = out_lua_argv[0].integer;
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
if(tfe_http_in_request(events))
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct tfe_http_half * in_resp_half = session->resp;
|
||
|
|
struct tfe_http_resp_spec * in_resp_spec = &in_resp_half->resp_spec;
|
||
|
|
|
||
|
|
if(lua_resp_code != 0)
|
||
|
|
{
|
||
|
|
in_resp_spec->resp_code = lua_resp_code;
|
||
|
|
}
|
||
|
|
http_free_params(out_lua_argv);
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_exit(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
int out_lua_argc = 0;
|
||
|
|
struct elua_data *out_lua_argv = NULL;
|
||
|
|
|
||
|
|
out_lua_argc = http_get_param_from_lua(vm, &out_lua_argv);
|
||
|
|
if(out_lua_argc != 1 || out_lua_argv == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int exit_code = out_lua_argv[0].integer;
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
if(exit_code == 0)
|
||
|
|
{
|
||
|
|
tsg_ctx->events = (enum tfe_http_event )0;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
http_free_params(out_lua_argv);
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_get_current_stage(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
const char * stage_name[] = {"http_req_uri", "http_req_header","http_req_body","http_resp_header","http_resp_body","http_stage_test"};
|
||
|
|
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
|
||
|
|
//For test_http_lua
|
||
|
|
if(tsg_ctx->http_req_uri==2)
|
||
|
|
{
|
||
|
|
http_set_string_to_lua(vm, stage_name[5], strlen(stage_name[5]));
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(events & EV_HTTP_REQ_HDR)
|
||
|
|
{
|
||
|
|
if(tsg_ctx->http_req_uri)
|
||
|
|
{
|
||
|
|
http_set_string_to_lua(vm, stage_name[0], strlen(stage_name[0]));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
http_set_string_to_lua(vm, stage_name[1], strlen(stage_name[1]));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if((events & EV_HTTP_REQ_BODY_BEGIN) | (events & EV_HTTP_REQ_BODY_END) | (events & EV_HTTP_REQ_BODY_CONT))
|
||
|
|
{
|
||
|
|
http_set_string_to_lua(vm, stage_name[2], strlen(stage_name[2]));
|
||
|
|
}
|
||
|
|
|
||
|
|
if(events & EV_HTTP_RESP_HDR)
|
||
|
|
{
|
||
|
|
http_set_string_to_lua(vm, stage_name[3], strlen(stage_name[3]));
|
||
|
|
}
|
||
|
|
|
||
|
|
if((events & EV_HTTP_RESP_BODY_BEGIN) | (events & EV_HTTP_RESP_BODY_CONT) | (events & EV_HTTP_RESP_BODY_END))
|
||
|
|
{
|
||
|
|
http_set_string_to_lua(vm, stage_name[4], strlen(stage_name[4]));
|
||
|
|
}
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_get_headers(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
|
||
|
|
struct tfe_http_half * in_req_half = session->req;
|
||
|
|
struct tfe_http_half * in_resp_half = session->resp;
|
||
|
|
|
||
|
|
struct tfe_http_half * in_half = tfe_http_in_request(events) ? in_req_half : in_resp_half;
|
||
|
|
if(in_half == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct http_field_name in_header_field{};
|
||
|
|
const char * in_header_value = NULL;
|
||
|
|
void * iterator = NULL;
|
||
|
|
|
||
|
|
struct elua_data in_field_name, in_field_value;
|
||
|
|
struct elua_table *table = elua_create_table(vm);
|
||
|
|
in_field_name.type = STRING;
|
||
|
|
in_field_value.type = STRING;
|
||
|
|
|
||
|
|
while (true)
|
||
|
|
{
|
||
|
|
if ((in_header_value = tfe_http_field_iterate(in_half, &iterator, &in_header_field)) == NULL)
|
||
|
|
{
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
in_field_name.buff = (char *)http_field_name_to_string(&in_header_field);
|
||
|
|
in_field_name.len = strlen(http_field_name_to_string(&in_header_field));
|
||
|
|
|
||
|
|
in_field_value.buff = (char *)in_header_value;
|
||
|
|
in_field_value.len = strlen(in_header_value);
|
||
|
|
elua_add_table(table, &in_field_name, &in_field_value);
|
||
|
|
}
|
||
|
|
|
||
|
|
http_set_table_to_lua(vm, table);
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_set_headers(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
int actually_write=0;
|
||
|
|
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const struct tfe_http_session * session = tsg_ctx->session;
|
||
|
|
if(session == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
enum tfe_http_event events = tsg_ctx->events;
|
||
|
|
|
||
|
|
struct tfe_http_half * in_req_half = session->req;
|
||
|
|
struct tfe_http_half * in_resp_half = session->resp;
|
||
|
|
|
||
|
|
struct tfe_http_half * in_half = tfe_http_in_request(events) ? in_req_half : in_resp_half;
|
||
|
|
if(in_half == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct http_field_name in_header_field{};
|
||
|
|
const char * in_header_value = NULL;
|
||
|
|
void * iterator = NULL;
|
||
|
|
|
||
|
|
int out_lua_argc = 0;
|
||
|
|
struct elua_data *out_lua_argv = NULL;
|
||
|
|
|
||
|
|
const char *in_field_name;
|
||
|
|
|
||
|
|
out_lua_argc = http_get_param_from_lua(vm, &out_lua_argv);
|
||
|
|
if(out_lua_argc != 2 || out_lua_argv == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char *field_name=out_lua_argv[0].buff, *field_value=out_lua_argv[1].buff;
|
||
|
|
|
||
|
|
if(field_name == NULL || field_value == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
while (true)
|
||
|
|
{
|
||
|
|
if ((in_header_value = tfe_http_field_iterate(in_half, &iterator, &in_header_field)) == NULL)
|
||
|
|
{
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
in_field_name = http_field_name_to_string(&in_header_field);
|
||
|
|
if(strcasecmp(in_field_name, field_name)==0 && strcasecmp(field_value, "nil")!=0)
|
||
|
|
{
|
||
|
|
if(in_header_field.field_id == TFE_HTTP_UNKNOWN_FIELD)
|
||
|
|
{
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
tfe_http_std_field_write(in_half, in_header_field.field_id, NULL);
|
||
|
|
tfe_http_std_field_write(in_half, in_header_field.field_id, field_value);
|
||
|
|
actually_write=1;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(strcasecmp(http_field_name_to_string(&in_header_field), field_name)==0 && strcasecmp(field_value, "nil")==0)
|
||
|
|
{
|
||
|
|
if(in_header_field.field_id == TFE_HTTP_UNKNOWN_FIELD)
|
||
|
|
{
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
tfe_http_std_field_write(in_half, in_header_field.field_id, NULL);
|
||
|
|
actually_write=1;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if(actually_write ==1)
|
||
|
|
{
|
||
|
|
tsg_ctx->execut_lua_sucess=1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(actually_write!=1 && strcasecmp(field_value, "nil")!=0)
|
||
|
|
{
|
||
|
|
tfe_http_nonstd_field_write(in_half, field_name, field_value);
|
||
|
|
tsg_ctx->execut_lua_sucess=1;
|
||
|
|
}
|
||
|
|
|
||
|
|
http_free_params(out_lua_argv);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_get_body(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
struct tsg_script_ctx *tsg_ctx = (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(tsg_ctx->http_body == NULL)
|
||
|
|
{
|
||
|
|
TFE_LOG_ERROR(tsg_ctx->local_logger, "policy_id:%d, profile_id:%d, message:%s", tsg_ctx->config_id, tsg_ctx->profile_id, "Can't to get req/resp body data");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char * __http_body = (char *) evbuffer_pullup(tsg_ctx->http_body, -1);
|
||
|
|
size_t __http_body_len = evbuffer_get_length(tsg_ctx->http_body);
|
||
|
|
|
||
|
|
http_set_string_to_lua(vm, __http_body, __http_body_len);
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int http_lua_set_body(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
int i=0;
|
||
|
|
struct tsg_script_ctx *tsg_ctx= (struct tsg_script_ctx *)elua_get_execute_userdata(vm);
|
||
|
|
if(tsg_ctx == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int out_lua_argc = 0;
|
||
|
|
struct elua_data *out_lua_argv = NULL;
|
||
|
|
|
||
|
|
out_lua_argc=http_get_param_from_lua(vm, &out_lua_argv);
|
||
|
|
if(out_lua_argc < 1 || out_lua_argv == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
tsg_ctx->http_lua_body = evbuffer_new();
|
||
|
|
for(i=0; i<out_lua_argc; i++)
|
||
|
|
{
|
||
|
|
char *in=out_lua_argv[i].buff;
|
||
|
|
size_t in_sz=strlen(out_lua_argv[i].buff);
|
||
|
|
if(in == NULL || in_sz ==0 )
|
||
|
|
{
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
evbuffer_add(tsg_ctx->http_lua_body, in, in_sz);
|
||
|
|
}
|
||
|
|
http_free_params(out_lua_argv);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct elua_context *http_lua_ctx_new(struct tsg_lua_script *lua_script, unsigned int thread_id)
|
||
|
|
{
|
||
|
|
return elua_create_context(lua_script->http_lua_handle[thread_id], "context");
|
||
|
|
}
|
||
|
|
|
||
|
|
struct elua_script *http_lua_map_cache_script(struct elua_vm *vm, const char *script, size_t script_len, size_t timeout_ms)
|
||
|
|
{
|
||
|
|
return elua_cache_script(vm, script, script_len, timeout_ms);
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_lua_ctx_free(struct tsg_lua_script *lua_script, unsigned int thread_id, struct elua_context *elua_ctx)
|
||
|
|
{
|
||
|
|
if(elua_ctx)
|
||
|
|
elua_destroy_context(elua_ctx);
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_lua_inject_http_consts(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
elua_register_cbinding(vm, NULL, "get_current_stage", http_lua_get_current_stage);
|
||
|
|
elua_register_cbinding(vm, NULL, "log_debug", http_lua_log_debug);
|
||
|
|
elua_register_cbinding(vm, NULL, "log_info", http_lua_log_info);
|
||
|
|
elua_register_cbinding(vm, NULL, "log_error", http_lua_log_error);
|
||
|
|
elua_register_cbinding(vm, NULL, "exit", http_lua_exit);
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_lua_inject_req_header_api(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
elua_register_cbinding(vm, "req", "get_method", http_lua_get_method);
|
||
|
|
elua_register_cbinding(vm, "req", "get_uri", http_lua_get_uri);
|
||
|
|
elua_register_cbinding(vm, "req", "set_uri", http_lua_set_uri);
|
||
|
|
elua_register_cbinding(vm, "req", "get_headers", http_lua_get_headers);
|
||
|
|
elua_register_cbinding(vm, "req", "set_header", http_lua_set_headers);
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_lua_inject_req_body_api(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
elua_register_cbinding(vm, "req", "get_body_data", http_lua_get_body);
|
||
|
|
elua_register_cbinding(vm, "req", "set_body_data", http_lua_set_body);
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_lua_inject_resp_header_api(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
elua_register_cbinding(vm, "resp", "get_status_code", http_lua_get_status_code);
|
||
|
|
elua_register_cbinding(vm, "resp", "set_status_code", http_lua_set_status_code);
|
||
|
|
|
||
|
|
elua_register_cbinding(vm, "resp", "get_headers", http_lua_get_headers);
|
||
|
|
elua_register_cbinding(vm, "resp", "set_header", http_lua_set_headers);
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_lua_inject_resp_body_api(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
elua_register_cbinding(vm, "resp", "get_body_data", http_lua_get_body);
|
||
|
|
elua_register_cbinding(vm, "resp", "set_body_data", http_lua_set_body);
|
||
|
|
}
|
||
|
|
|
||
|
|
void http_lua_inject_api(struct elua_vm *vm)
|
||
|
|
{
|
||
|
|
http_lua_inject_http_consts(vm);
|
||
|
|
http_lua_inject_req_header_api(vm);
|
||
|
|
http_lua_inject_req_body_api(vm);
|
||
|
|
http_lua_inject_resp_header_api(vm);
|
||
|
|
http_lua_inject_resp_body_api(vm);
|
||
|
|
}
|
||
|
|
|
||
|
|
int http_lua_handle_create(struct tsg_lua_script *lua_script, int thread_num, const char *name_space)
|
||
|
|
{
|
||
|
|
int id=0;
|
||
|
|
struct elua_vm **http_lua_handle = NULL;
|
||
|
|
|
||
|
|
http_lua_handle = ALLOC(struct elua_vm *, thread_num);
|
||
|
|
if(http_lua_handle == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
for(id=0; id < thread_num; id++)
|
||
|
|
{
|
||
|
|
http_lua_handle[id] = elua_create_vm(name_space);
|
||
|
|
if(http_lua_handle[id] == NULL)
|
||
|
|
{
|
||
|
|
goto finish;
|
||
|
|
}
|
||
|
|
http_lua_inject_api(http_lua_handle[id]);
|
||
|
|
}
|
||
|
|
lua_script->http_lua_handle=http_lua_handle;
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
finish:
|
||
|
|
for(id=0; id < thread_num; id++)
|
||
|
|
{
|
||
|
|
if(http_lua_handle[id])
|
||
|
|
FREE(&http_lua_handle[id])
|
||
|
|
}
|
||
|
|
free(http_lua_handle);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t execute_lua_script_rule(struct tsg_lua_script *lua_script, int profile_id, struct elua_context *elua_ctx, unsigned int thread_id, void *user_data)
|
||
|
|
{
|
||
|
|
int ret=0;
|
||
|
|
struct elua_script **escript=NULL;
|
||
|
|
|
||
|
|
ret=lua_script->http_lua_profile(profile_id, &escript);
|
||
|
|
if(ret<0)
|
||
|
|
{
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
const char *input="tfe";
|
||
|
|
size_t input_len=strlen(input);
|
||
|
|
|
||
|
|
ret=elua_execute_script(escript[thread_id], input, input_len, user_data, elua_ctx, NULL);
|
||
|
|
if(ret !=0 )
|
||
|
|
{
|
||
|
|
struct tsg_script_ctx *tsg_ctx= (struct tsg_script_ctx *)user_data;
|
||
|
|
if(tsg_ctx != NULL && tsg_ctx->local_logger != NULL)
|
||
|
|
{
|
||
|
|
TFE_LOG_ERROR(tsg_ctx->local_logger, "policy_id:%d, profile_id:%d, error_code:%d, error: %s", tsg_ctx->config_id, tsg_ctx->profile_id, ret, elua_get_last_error_string(lua_script->http_lua_handle[thread_id]));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|