testing of Lua
This commit is contained in:
Markus Makela
parent
1d4779215d
commit
2a2fdb2cc6
362
server/modules/filter/luafilter.c
Normal file
362
server/modules/filter/luafilter.c
Normal file
@ -0,0 +1,362 @@
|
||||
/*
|
||||
* This file is distributed as part of MaxScale by MariaDB Corporation. It is free
|
||||
* software: you can redistribute it and/or modify it under the terms of the
|
||||
* GNU General Public License as published by the Free Software Foundation,
|
||||
* version 2.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but WITHOUT
|
||||
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
||||
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
||||
* details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License along with
|
||||
* this program; if not, write to the Free Software Foundation, Inc., 51
|
||||
* Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
||||
*
|
||||
* Copyright MariaDB Corporation Ab 2014
|
||||
*/
|
||||
|
||||
/**
|
||||
* @file luafilter.c
|
||||
* Lua Filter
|
||||
*
|
||||
* A filter that calls a set of Lua script functions.
|
||||
*
|
||||
* This filter calls the createInstance, newSession, closeSession, routeQuery and clientReply functions in the Lua script.
|
||||
* Out of these functions the newSession, closeSession, routeQuery and clientReply functions are
|
||||
* In addition to being called with the service-specific global Lua state created during createInstance, the following functions
|
||||
* are called with a Lua state bound to the client session: newSession, closeSession, routeQuery and clientReply
|
||||
*/
|
||||
|
||||
#include <my_config.h>
|
||||
#include <ctype.h>
|
||||
#include <stdio.h>
|
||||
#include <fcntl.h>
|
||||
#include <filter.h>
|
||||
#include <string.h>
|
||||
#include <atomic.h>
|
||||
#include <modutil.h>
|
||||
#include <log_manager.h>
|
||||
#include <spinlock.h>
|
||||
#include <session.h>
|
||||
#include <plugin.h>
|
||||
#include <skygw_types.h>
|
||||
|
||||
extern "C"{
|
||||
#include <lua.h>
|
||||
#include <lualib.h>
|
||||
#include <lauxlib.h>
|
||||
}
|
||||
|
||||
MODULE_INFO info = {
|
||||
MODULE_API_FILTER,
|
||||
MODULE_ALPHA_RELEASE,
|
||||
FILTER_VERSION,
|
||||
"Lua Filter"
|
||||
};
|
||||
|
||||
static char *version_str = "V1.0.0";
|
||||
|
||||
/*
|
||||
* The filter entry points
|
||||
*/
|
||||
static FILTER *createInstance(char **options, FILTER_PARAMETER **);
|
||||
static void *newSession(FILTER *instance, SESSION *session);
|
||||
static void closeSession(FILTER *instance, void *session);
|
||||
static void freeSession(FILTER *instance, void *session);
|
||||
static void setDownstream(FILTER *instance, void *fsession, DOWNSTREAM *downstream);
|
||||
static void setUpstream(FILTER *instance, void *fsession, UPSTREAM *upstream);
|
||||
static int routeQuery(FILTER *instance, void *fsession, GWBUF *queue);
|
||||
static int clientReply(FILTER *instance, void *fsession, GWBUF *queue);
|
||||
static void diagnostic(FILTER *instance, void *fsession, DCB *dcb);
|
||||
|
||||
static FILTER_OBJECT MyObject = {
|
||||
createInstance,
|
||||
newSession,
|
||||
closeSession,
|
||||
freeSession,
|
||||
setDownstream,
|
||||
setUpstream,
|
||||
routeQuery,
|
||||
clientReply,
|
||||
diagnostic,
|
||||
};
|
||||
|
||||
/**
|
||||
* The Lua filter instance.
|
||||
*/
|
||||
typedef struct {
|
||||
lua_State* global_state;
|
||||
char* global_script;
|
||||
char* session_script;
|
||||
} LUA_INSTANCE;
|
||||
|
||||
/**
|
||||
* The session structure for Firewall filter.
|
||||
*/
|
||||
typedef struct {
|
||||
SESSION* session;
|
||||
lua_State* state;
|
||||
DOWNSTREAM down;
|
||||
UPSTREAM up;
|
||||
} LUA_SESSION;
|
||||
|
||||
/**
|
||||
* Implementation of the mandatory version entry point
|
||||
*
|
||||
* @return version string of the module
|
||||
*/
|
||||
char *
|
||||
version()
|
||||
{
|
||||
return version_str;
|
||||
}
|
||||
|
||||
/**
|
||||
* The module initialisation routine, called when the module
|
||||
* is first loaded.
|
||||
*/
|
||||
void
|
||||
ModuleInit()
|
||||
{
|
||||
}
|
||||
|
||||
/**
|
||||
* The module entry point routine. It is this routine that
|
||||
* must populate the structure that is referred to as the
|
||||
* "module object", this is a structure with the set of
|
||||
* external entry points for this module.
|
||||
*
|
||||
* @return The module object
|
||||
*/
|
||||
FILTER_OBJECT *
|
||||
GetModuleObject()
|
||||
{
|
||||
return &MyObject;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Create an instance of the filter for a particular service
|
||||
* within MaxScale.
|
||||
*
|
||||
* @param options The options for this filter
|
||||
*
|
||||
* @return The instance data for this new instance
|
||||
*/
|
||||
static FILTER *
|
||||
createInstance(char **options, FILTER_PARAMETER **params)
|
||||
{
|
||||
LUA_INSTANCE *my_instance;
|
||||
int i;
|
||||
if ((my_instance = calloc(1, sizeof(LUA_INSTANCE))) == NULL){
|
||||
return NULL;
|
||||
}
|
||||
|
||||
for(i = 0;params[i];i++){
|
||||
|
||||
if(strcmp(params[i]->name, "global_script") == 0){
|
||||
my_instance->global_script = strdup(params[i]->value);
|
||||
}else if(strcmp(params[i]->name, "session_script") == 0){
|
||||
my_instance->session_script = strdup(params[i]->value);
|
||||
}
|
||||
}
|
||||
|
||||
if(my_instance->global_script)
|
||||
{
|
||||
my_instance->global_state = luaL_newstate();
|
||||
if(my_instance->global_state == NULL){
|
||||
skygw_log_write(LOGFILE_ERROR, "Unable to initialize new Lua state.");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
luaL_openlibs(my_instance->global_state);
|
||||
|
||||
if(luaL_dofile(my_instance->global_state,my_instance->global_script)){
|
||||
skygw_log_write(LOGFILE_ERROR, "luafilter: Failed to execute global script at '%s'.",my_instance->global_script);
|
||||
}
|
||||
|
||||
lua_getglobal(my_instance->global_state,"createInstance");
|
||||
lua_pcall(my_instance->global_state,0,0,0);
|
||||
|
||||
}
|
||||
|
||||
return (FILTER *)my_instance;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* Associate a new session with this instance of the filter and opens
|
||||
* a connection to the server and prepares the exchange and the queue for use.
|
||||
*
|
||||
*
|
||||
* @param instance The filter instance data
|
||||
* @param session The session itself
|
||||
* @return Session specific data for this session
|
||||
*/
|
||||
static void *
|
||||
newSession(FILTER *instance, SESSION *session)
|
||||
{
|
||||
LUA_SESSION *my_session;
|
||||
LUA_INSTANCE *my_instance = (LUA_INSTANCE*)instance;
|
||||
|
||||
if ((my_session = calloc(1, sizeof(LUA_SESSION))) == NULL){
|
||||
return NULL;
|
||||
}
|
||||
my_session->session = session;
|
||||
my_session->state = luaL_newstate();
|
||||
luaL_openlibs(my_session->state);
|
||||
if(luaL_dofile(my_session->state,my_instance->session_script)){
|
||||
skygw_log_write(LOGFILE_ERROR, "luafilter: Failed to execute session script at '%s'.",my_instance->session_script);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
lua_getglobal(my_session->state,"newSession");
|
||||
lua_pcall(my_session->state,0,0,0);
|
||||
|
||||
return my_session;
|
||||
}
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* Close a session with the filter, this is the mechanism
|
||||
* by which a filter may cleanup data structure etc.
|
||||
*
|
||||
* @param instance The filter instance data
|
||||
* @param session The session being closed
|
||||
*/
|
||||
static void
|
||||
closeSession(FILTER *instance, void *session)
|
||||
{
|
||||
LUA_SESSION *my_session = (LUA_SESSION *)session;
|
||||
LUA_INSTANCE *my_instance = (LUA_INSTANCE*)instance;
|
||||
|
||||
lua_getglobal(my_session->state,"closeSession");
|
||||
lua_pcall(my_session->state,0,0,0);
|
||||
|
||||
lua_close(my_session->state);
|
||||
}
|
||||
|
||||
/**
|
||||
* Free the memory associated with the session
|
||||
*
|
||||
* @param instance The filter instance
|
||||
* @param session The filter session
|
||||
*/
|
||||
static void
|
||||
freeSession(FILTER *instance, void *session)
|
||||
{
|
||||
LUA_SESSION *my_session = (LUA_SESSION *)session;
|
||||
free(my_session);
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the downstream filter or router to which queries will be
|
||||
* passed from this filter.
|
||||
*
|
||||
* @param instance The filter instance data
|
||||
* @param session The filter session
|
||||
* @param downstream The downstream filter or router.
|
||||
*/
|
||||
static void
|
||||
setDownstream(FILTER *instance, void *session, DOWNSTREAM *downstream)
|
||||
{
|
||||
LUA_SESSION *my_session = (LUA_SESSION *)session;
|
||||
my_session->down = *downstream;
|
||||
}
|
||||
|
||||
static void setUpstream(FILTER *instance, void *session, UPSTREAM *upstream)
|
||||
{
|
||||
LUA_SESSION *my_session = (LUA_SESSION *)session;
|
||||
my_session->up = *upstream;
|
||||
}
|
||||
|
||||
|
||||
static int clientReply(FILTER *instance, void *session, GWBUF *queue)
|
||||
{
|
||||
LUA_SESSION *my_session = (LUA_SESSION *)session;
|
||||
LUA_INSTANCE *my_instance = (LUA_INSTANCE *)instance;
|
||||
char *fullquery;
|
||||
int qsize;
|
||||
|
||||
qsize = (unsigned char*)queue->end - (unsigned char*)(queue->start + 5);
|
||||
fullquery = calloc(qsize + 1,sizeof(char));
|
||||
memcpy(fullquery,(queue->start + 5),qsize);
|
||||
|
||||
lua_getglobal(my_session->state,"clientReply");
|
||||
lua_pushlstring(my_session->state,fullquery,qsize + 1);
|
||||
lua_pcall(my_session->state,1,0,0);
|
||||
|
||||
lua_getglobal(my_instance->global_state,"clientReply");
|
||||
lua_pushlstring(my_instance->global_state,fullquery,strlen(fullquery));
|
||||
lua_pcall(my_instance->global_state,1,0,0);
|
||||
|
||||
return my_session->up.clientReply(my_session->up.instance,
|
||||
my_session->up.session, queue);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* The routeQuery entry point. This is passed the query buffer
|
||||
* to which the filter should be applied. Once processed the
|
||||
* query is passed to the downstream component
|
||||
* (filter or router) in the filter chain.
|
||||
*
|
||||
* @param instance The filter instance data
|
||||
* @param session The filter session
|
||||
* @param queue The query data
|
||||
*/
|
||||
static int
|
||||
routeQuery(FILTER *instance, void *session, GWBUF *queue)
|
||||
{
|
||||
LUA_SESSION *my_session = (LUA_SESSION *)session;
|
||||
LUA_INSTANCE *my_instance = (LUA_INSTANCE *)instance;
|
||||
char *fullquery,*ptr;
|
||||
int qlen;
|
||||
|
||||
if(modutil_is_SQL(queue)){
|
||||
|
||||
modutil_extract_SQL(queue, &ptr, &qlen);
|
||||
fullquery = malloc((qlen + 1) * sizeof(char));
|
||||
memcpy(fullquery,ptr,qlen);
|
||||
memset(fullquery + qlen,0,1);
|
||||
|
||||
lua_getglobal(my_session->state,"routeQuery");
|
||||
lua_pushlstring(my_session->state,fullquery,strlen(fullquery));
|
||||
lua_pcall(my_session->state,1,0,0);
|
||||
|
||||
lua_getglobal(my_instance->global_state,"routeQuery");
|
||||
lua_pushlstring(my_instance->global_state,fullquery,strlen(fullquery));
|
||||
lua_pcall(my_instance->global_state,1,0,0);
|
||||
|
||||
free(fullquery);
|
||||
}
|
||||
|
||||
return my_session->down.routeQuery(my_session->down.instance,
|
||||
my_session->down.session, queue);
|
||||
}
|
||||
|
||||
/**
|
||||
* Diagnostics routine
|
||||
*
|
||||
* Prints the connection details and the names of the exchange,
|
||||
* queue and the routing key.
|
||||
*
|
||||
* @param instance The filter instance
|
||||
* @param fsession Filter session, may be NULL
|
||||
* @param dcb The DCB for diagnostic output
|
||||
*/
|
||||
static void
|
||||
diagnostic(FILTER *instance, void *fsession, DCB *dcb)
|
||||
{
|
||||
LUA_INSTANCE *my_instance = (LUA_INSTANCE *)instance;
|
||||
|
||||
if (my_instance)
|
||||
{
|
||||
dcb_printf(dcb, "\t\tLua Filter\nglobal script: %s\nsession script: %s\n",my_instance->global_script,my_instance->session_script);
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user