testing of Lua
This commit is contained in:
Markus Makela
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);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Reference in New Issue
Block a user