CONTRIB: move spoa_server out of the tree
As previously mentioned SPOA code has nothing to do in the haproxy core
since they're not dependent on haproxy's version. This one was moved to
its own repository here with complete history:
https://github.com/haproxy/spoa-server
diff --git a/contrib/spoa_server/Makefile b/contrib/spoa_server/Makefile
deleted file mode 100644
index 5de6135..0000000
--- a/contrib/spoa_server/Makefile
+++ /dev/null
@@ -1,77 +0,0 @@
-DESTDIR =
-PREFIX = /usr/local
-BINDIR = $(PREFIX)/bin
-
-CC = gcc
-LD = $(CC)
-
-CFLAGS = -g -O2 -Wall -Werror -pthread
-LDFLAGS = -lpthread
-
-OBJS = spoa.o
-
-ifneq ($(USE_LUA),)
-OBJS += ps_lua.o
-ifneq ($(LUA_INC),)
-CFLAGS += -I$(LUA_INC)
-endif
-ifneq ($(LUA_LIB),)
-LDLIBS += -L$(LUA_LIB)
-endif
-LDLIBS += -ldl -Wl,--export-dynamic -llua -lm -Wl,--no-export-dynamic
-endif
-
-ifneq ($(USE_PYTHON),)
-OBJS += ps_python.o
-
-# "--embed" flag is supported (and required) only from python 3.8+
-check_python_config := $(shell if python3-config --embed > /dev/null 2>&1 ; then echo "python3.8+"; \
-elif hash python3-config > /dev/null 2>&1 ; then echo "python3"; \
-elif hash python-config > /dev/null 2>&1 ; then echo "python2"; fi)
-
-ifeq ($(check_python_config), python3.8+)
-PYTHON_DEFAULT_INC := $(shell python3-config --includes)
-PYTHON_DEFAULT_LIB := $(shell python3-config --libs --embed)
-else ifeq ($(check_python_config), python3)
-PYTHON_DEFAULT_INC := $(shell python3-config --includes)
-PYTHON_DEFAULT_LIB := $(shell python3-config --libs)
-else ifeq ($(check_python_config), python2)
-PYTHON_DEFAULT_INC := $(shell python-config --includes)
-PYTHON_DEFAULT_LIB := $(shell python-config --libs)
-endif
-
-
-# Add default path
-ifneq ($(PYTHON_DEFAULT_INC),)
-CFLAGS += $(PYTHON_DEFAULT_INC)
-else
-CFLAGS += -I/usr/include/python2.7
-endif
-ifneq ($(PYTHON_DEFAULT_LIB),)
-LDLIBS += $(PYTHON_DEFAULT_LIB)
-else
-LDLIBS += -lpython2.7
-endif
-
-# Add user additional paths if any
-ifneq ($(PYTHON_INC),)
-CFLAGS += -I$(PYTHON_INC)
-endif
-ifneq ($(PYTHON_LIB),)
-LDLIBS += -L$(PYTHON_LIB)
-endif
-
-LDLIBS +=-Wl,--export-dynamic
-endif
-
-spoa: $(OBJS)
- $(LD) $(LDFLAGS) -o $@ $^ $(LDLIBS)
-
-install: spoa
- install spoa $(DESTDIR)$(BINDIR)
-
-clean:
- rm -f spoa $(OBJS)
-
-%.o: %.c
- $(CC) $(CFLAGS) -c -o $@ $<
diff --git a/contrib/spoa_server/README b/contrib/spoa_server/README
deleted file mode 100644
index a6c696c..0000000
--- a/contrib/spoa_server/README
+++ /dev/null
@@ -1,92 +0,0 @@
-Multi script langyage Stream Processing Offload Agent
------------------------------------------------------
-
-This agent receive SPOP message and process it with script languages. The
-language register callback with a message. Each callback receive the list
-of arguments with types according with the language capabilities. The
-callback write variables which are sent as response when the processing
-is done.
-
-
- Prerequirement
-----------------
-
-You have to install the development packages, either from the
-distribution repositories or from the source.
-
-CentOS/RHEL: sudo yum install python3-devel
-
-The current minimal python version compatible with this library is 2.7.
-It's recommended to use python version 3 where possible due to python 2 deprecation.
-
-
- Compilation
----------------
-
-The server currently supports Lua and Python. Type "make" with the options:
-USE_LUA=1 and/or USE_PYTHON=1.
-
-You can add LUA_INC=.. LUA_LIB=.. to the make command to set the paths to
-the lua header files and lua libraries.
-
-Similarly, you can add PYTHON_INC=.. PYTHON_LIB=.. to the make command to set the paths to
-the python header files and python libraries.
-By default, it will try to compile by detecting the default python 3 parameters.
-It will fall back to python 2 if python 3 is not available.
-
- Start the service
----------------------
-
-After you have compiled it, to start the service, you just need to use "spoa"
-binary:
-
- $> ./spoa -h
- Usage: ./spoa [-h] [-d] [-p <port>] [-n <num-workers>]
- -h Print this message
- -d Enable the debug mode
- -p <port> Specify the port to listen on (default: 12345)
- -n <num-workers> Specify the number of workers (default: 5)
- -f <file> Load script according with the supported languages
-
-The file processor is recognized using the extension. .lua or .luac for lua and
-.py for python. Start example:
-
- $> ./spoa -d -f ps_lua.lua
-
- $> ./spoa -d -f ps_python.py
-
-
- Configure
--------------
-
-Sample configuration are join to this server:
-
- spoa-server.conf : The HAProxy configuration file using SPOE server
- spoa-server.spoe.conf : The SPOP description file used by HAProxy
- ps_lua.lua : Processing Lua example
- ps_python.py : Processing Python example
-
-
- Considerations
-------------------
-
-This server is a beta version. It works fine, but some improvement will be
-welcome:
-
-Main process:
-
- * Improve log management: Today the log are sent on stdout.
- * Improve process management: The dead process are ignored.
- * Implement systemd integration.
- * Implement threads: It would be fine to implement thread working. Shared
- memory is welcome for managing database connection pool and something like
- that.
- * Add PHP support and some other languages.
-
-Python:
-
- * Improve reporting: Catch python error message and report it in the right
- place. Today the error are dumped on stdout. How using syslog for logging
- stack traces ?
-
-Maybe some other things...
diff --git a/contrib/spoa_server/print_r.lua b/contrib/spoa_server/print_r.lua
deleted file mode 100644
index 2fa57e7..0000000
--- a/contrib/spoa_server/print_r.lua
+++ /dev/null
@@ -1,68 +0,0 @@
-function color(index, str)
- return "\x1b[" .. index .. "m" .. str .. "\x1b[00m"
-end
-
-function nocolor(index, str)
- return str
-end
-
-function sp(count)
- local spaces = ""
- while count > 0 do
- spaces = spaces .. " "
- count = count - 1
- end
- return spaces
-end
-
-function print_rr(p, indent, c, wr)
- local i = 0
- local nl = ""
-
- if type(p) == "table" then
- wr(c("33", "(table)") .. " " .. c("34", tostring(p)) .. " [")
-
- mt = getmetatable(p)
- if mt ~= nil then
- wr("\n" .. sp(indent+1) .. c("31", "METATABLE") .. ": ")
- print_rr(mt, indent+1, c, wr)
- end
-
- for k,v in pairs(p) do
- if i > 0 then
- nl = "\n"
- else
- wr("\n")
- end
- wr(nl .. sp(indent+1))
- if type(k) == "number" then
- wr(c("32", tostring(k)))
- else
- wr("\"" .. c("32", tostring(k)) .. "\"")
- end
- wr(": ")
- print_rr(v, indent+1, c, wr)
- i = i + 1
- end
- if i == 0 then
- wr(" " .. c("35", "/* empty */") .. " ]")
- else
- wr("\n" .. sp(indent) .. "]")
- end
- elseif type(p) == "string" then
- wr(c("33", "(string)") .. " \"" .. c("34", p) .. "\"")
- else
- wr(c("33", "(" .. type(p) .. ")") .. " " .. c("34", tostring(p)))
- end
-end
-
-function print_r(p, col, wr)
- if col == nil then col = true end
- if wr == nil then wr = function(msg) io.stdout:write(msg) end end
- if col == true then
- print_rr(p, 0, color, wr)
- else
- print_rr(p, 0, nocolor, wr)
- end
- wr("\n")
-end
diff --git a/contrib/spoa_server/ps_lua.c b/contrib/spoa_server/ps_lua.c
deleted file mode 100644
index 2d37d9b..0000000
--- a/contrib/spoa_server/ps_lua.c
+++ /dev/null
@@ -1,509 +0,0 @@
-/* spoa-server: processing Lua
- *
- * Copyright 2018 OZON / Thierry Fournier <thierry.fournier@ozon.io>
- *
- * This program 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; either version
- * 2 of the License, or (at your option) any later version.
- *
- */
-
-#include <arpa/inet.h>
-
-#include <errno.h>
-#include <string.h>
-
-#include <lauxlib.h>
-#include <lua.h>
-#include <lualib.h>
-
-#include "spoa.h"
-
-static lua_State *L = NULL;
-static struct worker *worker;
-
-static int ps_lua_start_worker(struct worker *w);
-static int ps_lua_load_file(struct worker *w, const char *file);
-static int ps_lua_exec_message(struct worker *w, void *ref, int nargs, struct spoe_kv *args);
-
-static struct ps ps_lua_bindings_1 = {
- .init_worker = ps_lua_start_worker,
- .load_file = ps_lua_load_file,
- .exec_message = ps_lua_exec_message,
- .ext = ".lua",
-};
-
-static struct ps ps_lua_bindings_2 = {
- .init_worker = ps_lua_start_worker,
- .load_file = ps_lua_load_file,
- .exec_message = ps_lua_exec_message,
- .ext = ".luac",
-};
-
-/* Imported from Lua-5.3.4 */
-static int typeerror (lua_State *L, int arg, const char *tname)
-{
- const char *msg;
- const char *typearg; /* name for the type of the actual argument */
- if (luaL_getmetafield(L, arg, "__name") == LUA_TSTRING)
- typearg = lua_tostring(L, -1); /* use the given type name */
- else if (lua_type(L, arg) == LUA_TLIGHTUSERDATA)
- typearg = "light userdata"; /* special name for messages */
- else
- typearg = luaL_typename(L, arg); /* standard name */
- msg = lua_pushfstring(L, "%s expected, got %s", tname, typearg);
- return luaL_argerror(L, arg, msg);
-}
-
-/* Imported from Lua-5.3.4 */
-static void tag_error (lua_State *L, int arg, int tag) {
- typeerror(L, arg, lua_typename(L, tag));
-}
-
-#ifndef luaL_checkboolean
-static int luaL_checkboolean(lua_State *L, int index)
-{
- if (!lua_isboolean(L, index)) {
- tag_error(L, index, LUA_TBOOLEAN);
- }
- return lua_toboolean(L, index);
-}
-#endif
-
-static int ps_lua_register_message(lua_State *L)
-{
- const char *name;
- long ref;
-
- /* First argument is a message name */
- name = luaL_checkstring(L, 1);
-
- /* Second argument is a function */
- if (!lua_isfunction(L, 2)) {
- const char *msg = lua_pushfstring(L, "function expected, got %s", luaL_typename(L, 2));
- luaL_argerror(L, 2, msg);
- }
- lua_pushvalue(L, 2);
- ref = luaL_ref(L, LUA_REGISTRYINDEX);
-
- /* Register the message processor */
- ps_register_message(&ps_lua_bindings_1, name, (void *)ref);
-
- return 1;
-}
-
-static int ps_lua_set_var_null(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
-
- if (!set_var_null(worker, name, name_len, scope)) {
- luaL_error(L, "No space left available");
- }
- return 0;
-}
-
-static int ps_lua_set_var_boolean(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- int64_t value;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checkboolean(L, 3);
-
- if (!set_var_bool(worker, name, name_len, scope, value))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_set_var_uint32(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- int64_t value;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checkinteger(L, 3);
-
- if (value < 0 || value > UINT_MAX)
- luaL_error(L, "Integer '%lld' out of range for 'uint32' type", value);
-
- if (!set_var_uint32(worker, name, name_len, scope, value))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_set_var_int32(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- int64_t value;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checkinteger(L, 3);
-
- if (value < INT_MIN || value > INT_MAX)
- luaL_error(L, "Integer '%lld' out of range for 'int32' type", value);
-
- if (!set_var_int32(worker, name, name_len, scope, value))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_set_var_uint64(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- int64_t value;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checkinteger(L, 3);
-
- if (value < 0)
- luaL_error(L, "Integer '%lld' out of range for 'uint64' type", value);
-
- if (!set_var_uint64(worker, name, name_len, scope, value))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_set_var_int64(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- int64_t value;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checkinteger(L, 3);
-
- if (!set_var_int64(worker, name, name_len, scope, value))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_set_var_ipv4(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- const char *value;
- struct in_addr ipv4;
- int ret;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checkstring(L, 3);
-
- ret = inet_pton(AF_INET, value, &ipv4);
- if (ret == 0)
- luaL_error(L, "IPv4 '%s': invalid format", value);
- if (ret == -1)
- luaL_error(L, "IPv4 '%s': %s", value, strerror(errno));
-
- if (!set_var_ipv4(worker, name, name_len, scope, &ipv4))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_set_var_ipv6(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- const char *value;
- struct in6_addr ipv6;
- int ret;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checkstring(L, 3);
-
- ret = inet_pton(AF_INET6, value, &ipv6);
- if (ret == 0)
- luaL_error(L, "IPv6 '%s': invalid format", value);
- if (ret == -1)
- luaL_error(L, "IPv6 '%s': %s", value, strerror(errno));
-
- if (!set_var_ipv6(worker, name, name_len, scope, &ipv6))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_set_var_str(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- const char *value;
- size_t value_len;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checklstring(L, 3, &value_len);
-
- if (!set_var_string(worker, name, name_len, scope, value, value_len))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_set_var_bin(lua_State *L)
-{
- const char *name;
- size_t name_len;
- unsigned char scope;
- const char *value;
- size_t value_len;
-
- name = luaL_checklstring(L, 1, &name_len);
- scope = (unsigned char)luaL_checkinteger(L, 2);
- value = luaL_checklstring(L, 3, &value_len);
-
- if (!set_var_bin(worker, name, name_len, scope, value, value_len))
- luaL_error(L, "No space left available");
- return 0;
-}
-
-static int ps_lua_start_worker(struct worker *w)
-{
- if (L != NULL)
- return 1;
-
- worker = w;
-
- L = luaL_newstate();
- luaL_openlibs(L);
-
- lua_newtable(L);
-
- lua_pushstring(L, "register_message");
- lua_pushcclosure(L, ps_lua_register_message, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_null");
- lua_pushcclosure(L, ps_lua_set_var_null, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_boolean");
- lua_pushcclosure(L, ps_lua_set_var_boolean, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_uint32");
- lua_pushcclosure(L, ps_lua_set_var_uint32, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_int32");
- lua_pushcclosure(L, ps_lua_set_var_int32, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_uint64");
- lua_pushcclosure(L, ps_lua_set_var_uint64, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_int64");
- lua_pushcclosure(L, ps_lua_set_var_int64, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_ipv4");
- lua_pushcclosure(L, ps_lua_set_var_ipv4, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_ipv6");
- lua_pushcclosure(L, ps_lua_set_var_ipv6, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_str");
- lua_pushcclosure(L, ps_lua_set_var_str, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "set_var_bin");
- lua_pushcclosure(L, ps_lua_set_var_bin, 0);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "scope");
- lua_newtable(L);
-
- lua_pushstring(L, "proc");
- lua_pushinteger(L, SPOE_SCOPE_PROC);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "sess");
- lua_pushinteger(L, SPOE_SCOPE_SESS);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "txn");
- lua_pushinteger(L, SPOE_SCOPE_TXN);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "req");
- lua_pushinteger(L, SPOE_SCOPE_REQ);
- lua_rawset(L, -3);
-
- lua_pushstring(L, "res");
- lua_pushinteger(L, SPOE_SCOPE_RES);
- lua_rawset(L, -3);
-
- lua_rawset(L, -3); /* scope */
-
- lua_setglobal(L, "spoa");
- return 1;
-}
-
-static int ps_lua_load_file(struct worker *w, const char *file)
-{
- int error;
-
- /* Load the file and check syntax */
- error = luaL_loadfile(L, file);
- if (error) {
- fprintf(stderr, "lua syntax error: %s\n", lua_tostring(L, -1));
- return 0;
- }
-
- /* If no syntax error where detected, execute the code. */
- error = lua_pcall(L, 0, LUA_MULTRET, 0);
- switch (error) {
- case LUA_OK:
- break;
- case LUA_ERRRUN:
- fprintf(stderr, "lua runtime error: %s\n", lua_tostring(L, -1));
- lua_pop(L, 1);
- return 0;
- case LUA_ERRMEM:
- fprintf(stderr, "lua out of memory error\n");
- return 0;
- case LUA_ERRERR:
- fprintf(stderr, "lua message handler error: %s\n", lua_tostring(L, 0));
- lua_pop(L, 1);
- return 0;
- case LUA_ERRGCMM:
- fprintf(stderr, "lua garbage collector error: %s\n", lua_tostring(L, 0));
- lua_pop(L, 1);
- return 0;
- default:
- fprintf(stderr, "lua unknown error: %s\n", lua_tostring(L, 0));
- lua_pop(L, 1);
- return 0;
- }
- return 1;
-}
-
-static int ps_lua_exec_message(struct worker *w, void *ref, int nargs, struct spoe_kv *args)
-{
- long lua_ref = (long)ref;
- int ret;
- char *msg_fmt = NULL;
- const char *msg;
- int i;
- char ipbuf[64];
-
- /* Restore function in the stack */
- lua_rawgeti(L, LUA_REGISTRYINDEX, lua_ref);
-
- /* convert args in lua mode */
- lua_newtable(L);
- for (i = 0; i < nargs; i++) {
- lua_newtable(L);
- lua_pushstring(L, "name");
- lua_pushlstring(L, args[i].name.str, args[i].name.len);
- lua_rawset(L, -3); /* Push name */
- lua_pushstring(L, "value");
- switch (args[i].value.type) {
- case SPOE_DATA_T_NULL:
- lua_pushnil(L);
- break;
- case SPOE_DATA_T_BOOL:
- lua_pushboolean(L, args[i].value.u.boolean);
- break;
- case SPOE_DATA_T_INT32:
- lua_pushinteger(L, args[i].value.u.sint32);
- break;
- case SPOE_DATA_T_UINT32:
- lua_pushinteger(L, args[i].value.u.uint32);
- break;
- case SPOE_DATA_T_INT64:
- lua_pushinteger(L, args[i].value.u.sint64);
- break;
- case SPOE_DATA_T_UINT64:
- if (args[i].value.u.uint64 > LLONG_MAX)
- lua_pushnil(L);
- else
- lua_pushinteger(L, args[i].value.u.uint64);
- break;
- case SPOE_DATA_T_IPV4:
- if (inet_ntop(AF_INET, &args[i].value.u.ipv4, ipbuf, 64) == NULL)
- lua_pushnil(L);
- else
- lua_pushstring(L, ipbuf);
- break;
- case SPOE_DATA_T_IPV6:
- if (inet_ntop(AF_INET6, &args[i].value.u.ipv4, ipbuf, 64) == NULL)
- lua_pushnil(L);
- else
- lua_pushstring(L, ipbuf);
- break;
- case SPOE_DATA_T_STR:
- case SPOE_DATA_T_BIN:
- lua_pushlstring(L, args[i].value.u.buffer.str, args[i].value.u.buffer.len);
- break;
- default:
- lua_pushnil(L);
- break;
- }
- lua_rawset(L, -3); /* Push name */
- lua_rawseti(L, -2, i + 1); /* Pusg table in globale table */
- }
-
- /* execute lua function */
- while (1) {
- ret = lua_resume(L, L, 1);
- switch (ret) {
- case LUA_OK:
- return 1;
- case LUA_YIELD:
- DEBUG("Lua yield");
- continue;
- case LUA_ERRMEM:
- LOG("Lua: Out of memory error");
- return 0;
- case LUA_ERRRUN:
- msg_fmt = "Lua runtime error";
- case LUA_ERRGCMM:
- msg_fmt = msg_fmt ? msg_fmt : "Lua garbage collector error";
- case LUA_ERRERR:
- msg_fmt = msg_fmt ? msg_fmt : "Lua message handler error";
- default:
- msg_fmt = msg_fmt ? msg_fmt : "Lua unknown error";
- msg = lua_tostring(L, -1);
- if (msg == NULL)
- msg = "Unknown error";
- LOG("%s: %s", msg_fmt, msg);
- lua_settop(L, 0);
- return 0;
- }
- }
-
- return 1;
-}
-
-__attribute__((constructor))
-static void __ps_lua_init(void)
-{
- ps_register(&ps_lua_bindings_1);
- ps_register(&ps_lua_bindings_2);
-}
diff --git a/contrib/spoa_server/ps_lua.lua b/contrib/spoa_server/ps_lua.lua
deleted file mode 100644
index 4d1fc52..0000000
--- a/contrib/spoa_server/ps_lua.lua
+++ /dev/null
@@ -1,19 +0,0 @@
-require("print_r")
-require("math")
-
-print_r("Load lua message processors")
-
-spoa.register_message("check-client-ip", function(args)
- print_r(args)
- spoa.set_var_null("null", spoa.scope.txn)
- spoa.set_var_boolean("boolean", spoa.scope.txn, true)
- spoa.set_var_int32("int32", spoa.scope.txn, 1234)
- spoa.set_var_uint32("uint32", spoa.scope.txn, 1234)
- spoa.set_var_int64("int64", spoa.scope.txn, 1234)
- spoa.set_var_uint64("uint64", spoa.scope.txn, 1234)
- spoa.set_var_ipv4("ipv4", spoa.scope.txn, "127.0.0.1")
- spoa.set_var_ipv6("ipv6", spoa.scope.txn, "1::f")
- spoa.set_var_str("str", spoa.scope.txn, "1::f")
- spoa.set_var_bin("bin", spoa.scope.txn, "1::f")
- spoa.set_var_int32("ip_score", spoa.scope.sess, math.random(100))
-end)
diff --git a/contrib/spoa_server/ps_python.c b/contrib/spoa_server/ps_python.c
deleted file mode 100644
index 04b21f1..0000000
--- a/contrib/spoa_server/ps_python.c
+++ /dev/null
@@ -1,800 +0,0 @@
-/* spoa-server: processing Python
- *
- * Copyright 2018 OZON / Thierry Fournier <thierry.fournier@ozon.io>
- * Copyright (C) 2020 Gilchrist Dadaglo <gilchrist@dadaglo.com>
- *
- * This program 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; either version
- * 2 of the License, or (at your option) any later version.
- *
- * This program is provided 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.
- *
- */
-
-/*
- * Define PY_SSIZE_T_CLEAN before including Python.h
- * as per https://docs.python.org/3/c-api/arg.html and https://docs.python.org/2/c-api/arg.html
- */
-#define PY_SSIZE_T_CLEAN
-
-#include <Python.h>
-
-#include <arpa/inet.h>
-
-#include <errno.h>
-#include <string.h>
-#include <limits.h>
-
-#include "spoa.h"
-#include "ps_python.h"
-
-/* Embedding python documentation:
- *
- * https://docs.python.org/2/extending/embedding.html
- * https://docs.python.org/2/extending/extending.html#extending-python-with-c-or-c
- * https://docs.python.org/2/extending/extending.html#calling-python-functions-from-c
- */
-
-static PyObject *module_ipaddress;
-static PyObject *ipv4_address;
-static PyObject *ipv6_address;
-static PyObject *spoa_error;
-static PyObject *empty_tuple;
-static struct worker *worker;
-
-static int ps_python_start_worker(struct worker *w);
-static int ps_python_load_file(struct worker *w, const char *file);
-static int ps_python_exec_message(struct worker *w, void *ref, int nargs, struct spoe_kv *args);
-
-static struct ps ps_python_bindings = {
- .init_worker = ps_python_start_worker,
- .load_file = ps_python_load_file,
- .exec_message = ps_python_exec_message,
- .ext = ".py",
-};
-
-static int ps_python_check_overflow(Py_ssize_t len)
-{
- /* There might be an overflow when converting from Py_ssize_t to int.
- * This function will catch those cases.
- * Also, spoa "struct chunk" is limited to int size.
- * We should not send data bigger than it can handle.
- */
- if (len >= (Py_ssize_t)INT_MAX) {
- PyErr_Format(spoa_error,
- "%zd is over 2GB. Please split in smaller pieces.", \
- len);
- return -1;
- } else {
- return Py_SAFE_DOWNCAST(len, Py_ssize_t, int);
- }
-}
-
-#if IS_PYTHON_3K
-static PyObject *module_spoa;
-static PyObject *PyInit_spoa_module(void);
-#endif /* IS_PYTHON_3K */
-
-static PyObject *ps_python_register_message(PyObject *self, PyObject *args)
-{
- const char *name;
- PyObject *ref;
-
- if (!PyArg_ParseTuple(args, "sO!", &name, &PyFunction_Type, &ref))
- return NULL;
- Py_XINCREF(ref); /* because the function is internally referenced */
-
- ps_register_message(&ps_python_bindings, name, (void *)ref);
-
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_null(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int name_len_i;
- int scope;
-
- if (!PyArg_ParseTuple(args, "s#i", &name, &name_len, &scope))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- if (name_len_i == -1)
- return NULL;
- if (!set_var_null(worker, name, name_len_i, scope)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_boolean(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- int value;
- int name_len_i;
-
- if (!PyArg_ParseTuple(args, "s#ii", &name, &name_len, &scope, &value))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- if (name_len_i == -1)
- return NULL;
- if (!set_var_bool(worker, name, name_len_i, scope, value)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_int32(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- int32_t value;
- int name_len_i;
-
- if (!PyArg_ParseTuple(args, "s#ii", &name, &name_len, &scope, &value))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- if (name_len_i == -1)
- return NULL;
- if (!set_var_int32(worker, name, name_len_i, scope, value)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_uint32(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- uint32_t value;
- int name_len_i;
-
- if (!PyArg_ParseTuple(args, "s#iI", &name, &name_len, &scope, &value))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- if (name_len_i == -1)
- return NULL;
- if (!set_var_uint32(worker, name, name_len_i, scope, value)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_int64(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- int64_t value;
- int name_len_i;
-
- if (!PyArg_ParseTuple(args, "s#il", &name, &name_len, &scope, &value))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- if (name_len_i == -1)
- return NULL;
- if (!set_var_int64(worker, name, name_len_i, scope, value)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_uint64(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- uint64_t value;
- int name_len_i;
-
- if (!PyArg_ParseTuple(args, "s#ik", &name, &name_len, &scope, &value))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- if (name_len_i == -1)
- return NULL;
- if (!set_var_uint64(worker, name, name_len_i, scope, value)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_ipv4(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- PyObject *ipv4;
- PyObject *value;
- struct in_addr ip;
- int name_len_i;
-
- if (!PyArg_ParseTuple(args, "s#iO", &name, &name_len, &scope, &ipv4))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- if (name_len_i == -1)
- return NULL;
- if (!PyObject_IsInstance(ipv4, ipv4_address)) {
- PyErr_Format(spoa_error, "must be 'IPv4Address', not '%s'", ipv4->ob_type->tp_name);
- return NULL;
- }
- /* Execute packed ... I think .. */
- value = PyObject_GetAttrString(ipv4, "packed");
- if (value == NULL)
- return NULL;
- if (PY_STRING_GET_SIZE(value) != sizeof(ip)) {
- PyErr_Format(spoa_error, "IPv4 manipulation internal error");
- return NULL;
- }
- memcpy(&ip, PY_STRING_AS_STRING(value), PY_STRING_GET_SIZE(value));
- if (!set_var_ipv4(worker, name, name_len_i, scope, &ip)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- /* Once we set the IP value in the worker, we don't need it anymore... */
- Py_XDECREF(value);
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_ipv6(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- PyObject *ipv6;
- PyObject *value;
- struct in6_addr ip;
- int name_len_i;
-
- if (!PyArg_ParseTuple(args, "s#iO", &name, &name_len, &scope, &ipv6))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- if (name_len_i == -1)
- return NULL;
- if (!PyObject_IsInstance(ipv6, ipv6_address)) {
- PyErr_Format(spoa_error, "must be 'IPv6Address', not '%s'", ipv6->ob_type->tp_name);
- return NULL;
- }
- /* Execute packed ... I think .. */
- value = PyObject_GetAttrString(ipv6, "packed");
- if (value == NULL)
- return NULL;
- if (PY_STRING_GET_SIZE(value) != sizeof(ip)) {
- PyErr_Format(spoa_error, "IPv6 manipulation internal error");
- return NULL;
- }
- memcpy(&ip, PY_STRING_AS_STRING(value), PY_STRING_GET_SIZE(value));
- if (!set_var_ipv6(worker, name, name_len_i, scope, &ip)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- /* Once we set the IP value in the worker, we don't need it anymore... */
- Py_XDECREF(value);
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_str(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- const char *value;
- Py_ssize_t value_len;
- int name_len_i;
- int value_len_i;
-
- if (!PyArg_ParseTuple(args, "s#is#", &name, &name_len, &scope, &value, &value_len))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- value_len_i = ps_python_check_overflow(value_len);
- if (name_len_i == -1 || value_len_i == -1)
- return NULL;
- if (!set_var_string(worker, name, name_len_i, scope, value, value_len_i)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- Py_RETURN_NONE;
-}
-
-static PyObject *ps_python_set_var_bin(PyObject *self, PyObject *args)
-{
- const char *name;
- Py_ssize_t name_len;
- int scope;
- const char *value;
- Py_ssize_t value_len;
- int name_len_i;
- int value_len_i;
-
- if (!PyArg_ParseTuple(args, "s#is#", &name, &name_len, &scope, &value, &value_len))
- return NULL;
- name_len_i = ps_python_check_overflow(name_len);
- value_len_i = ps_python_check_overflow(value_len);
- if (name_len_i == -1 || value_len_i == -1)
- return NULL;
- if (!set_var_bin(worker, name, name_len_i, scope, value, value_len_i)) {
- PyErr_SetString(spoa_error, "No more memory space available");
- return NULL;
- }
- Py_RETURN_NONE;
-}
-
-
-static PyMethodDef spoa_methods[] = {
- {"register_message", ps_python_register_message, METH_VARARGS,
- "Register binding for SPOA message."},
- {"set_var_null", ps_python_set_var_null, METH_VARARGS,
- "Set SPOA NULL variable"},
- {"set_var_boolean", ps_python_set_var_boolean, METH_VARARGS,
- "Set SPOA boolean variable"},
- {"set_var_int32", ps_python_set_var_int32, METH_VARARGS,
- "Set SPOA int32 variable"},
- {"set_var_uint32", ps_python_set_var_uint32, METH_VARARGS,
- "Set SPOA uint32 variable"},
- {"set_var_int64", ps_python_set_var_int64, METH_VARARGS,
- "Set SPOA int64 variable"},
- {"set_var_uint64", ps_python_set_var_uint64, METH_VARARGS,
- "Set SPOA uint64 variable"},
- {"set_var_ipv4", ps_python_set_var_ipv4, METH_VARARGS,
- "Set SPOA ipv4 variable"},
- {"set_var_ipv6", ps_python_set_var_ipv6, METH_VARARGS,
- "Set SPOA ipv6 variable"},
- {"set_var_str", ps_python_set_var_str, METH_VARARGS,
- "Set SPOA str variable"},
- {"set_var_bin", ps_python_set_var_bin, METH_VARARGS,
- "Set SPOA bin variable"},
- { /* end */ }
-};
-
-#if IS_PYTHON_3K
-static struct PyModuleDef spoa_module_definition = {
- PyModuleDef_HEAD_INIT, /* m_base */
- "spoa", /* m_name */
- "HAProxy SPOA module for python", /* m_doc */
- -1, /* m_size */
- spoa_methods, /* m_methods */
- NULL, /* m_slots */
- NULL, /* m_traverse */
- NULL, /* m_clear */
- NULL /* m_free */
-};
-
-static PyObject *PyInit_spoa_module(void)
-{
- return module_spoa;
-}
-#endif /* IS_PYTHON_3K */
-
-static int ps_python_start_worker(struct worker *w)
-{
- PyObject *m;
- PyObject *module_name;
- PyObject *value;
- int ret;
-
-#if IS_PYTHON_27
- Py_SetProgramName("spoa-server");
-#endif /* IS_PYTHON_27 */
-#if IS_PYTHON_3K
- Py_SetProgramName(Py_DecodeLocale("spoa-server", NULL));
- PyImport_AppendInittab("spoa", &PyInit_spoa_module);
-#endif /* IS_PYTHON_3K */
-
- Py_Initialize();
-
- module_name = PY_STRING_FROM_STRING("ipaddress");
- if (module_name == NULL) {
- PyErr_Print();
- return 0;
- }
-
- module_ipaddress = PyImport_Import(module_name);
- Py_DECREF(module_name);
- if (module_ipaddress == NULL) {
- PyErr_Print();
- return 0;
- }
-
- ipv4_address = PyObject_GetAttrString(module_ipaddress, "IPv4Address");
- if (ipv4_address == NULL) {
- Py_DECREF(module_ipaddress);
- PyErr_Print();
- return 0;
- }
-
- ipv6_address = PyObject_GetAttrString(module_ipaddress, "IPv6Address");
- if (ipv6_address == NULL) {
- Py_DECREF(ipv4_address);
- Py_DECREF(module_ipaddress);
- PyErr_Print();
- return 0;
- }
-
- PY_INIT_MODULE(m, "spoa", spoa_methods, &spoa_module_definition);
- if (m == NULL) {
- Py_DECREF(ipv4_address);
- Py_DECREF(ipv6_address);
- Py_DECREF(module_ipaddress);
- PyErr_Print();
- return 0;
- }
-
- spoa_error = PyErr_NewException("spoa.error", NULL, NULL);
- /* PyModule_AddObject will steal the reference to spoa_error
- * in case of success only
- * We need to increment the counters to continue using it
- * but cleanup in case of failure
- */
- Py_INCREF(spoa_error);
- ret = PyModule_AddObject(m, "error", spoa_error);
- if (ret == -1) {
- Py_DECREF(m);
- Py_DECREF(spoa_error);
- PyErr_Print();
- return 0;
- }
-
-
- value = PyLong_FromLong(SPOE_SCOPE_PROC);
- if (value == NULL) {
- PyErr_Print();
- return 0;
- }
-
- ret = PyModule_AddObject(m, "scope_proc", value);
- if (ret == -1) {
- Py_DECREF(m);
- Py_DECREF(value);
- PyErr_Print();
- return 0;
- }
-
- value = PyLong_FromLong(SPOE_SCOPE_SESS);
- if (value == NULL) {
- Py_DECREF(m);
- PyErr_Print();
- return 0;
- }
-
- ret = PyModule_AddObject(m, "scope_sess", value);
- if (ret == -1) {
- Py_DECREF(m);
- Py_DECREF(value);
- PyErr_Print();
- return 0;
- }
-
- value = PyLong_FromLong(SPOE_SCOPE_TXN);
- if (value == NULL) {
- Py_DECREF(m);
- PyErr_Print();
- return 0;
- }
-
- ret = PyModule_AddObject(m, "scope_txn", value);
- if (ret == -1) {
- Py_DECREF(m);
- Py_DECREF(value);
- PyErr_Print();
- return 0;
- }
-
- value = PyLong_FromLong(SPOE_SCOPE_REQ);
- if (value == NULL) {
- Py_DECREF(m);
- PyErr_Print();
- return 0;
- }
-
- ret = PyModule_AddObject(m, "scope_req", value);
- if (ret == -1) {
- Py_DECREF(m);
- Py_DECREF(value);
- PyErr_Print();
- return 0;
- }
-
- value = PyLong_FromLong(SPOE_SCOPE_RES);
- if (value == NULL) {
- Py_DECREF(m);
- PyErr_Print();
- return 0;
- }
-
- ret = PyModule_AddObject(m, "scope_res", value);
- if (ret == -1) {
- Py_DECREF(m);
- Py_DECREF(value);
- PyErr_Print();
- return 0;
- }
-
- empty_tuple = PyTuple_New(0);
- if (empty_tuple == NULL) {
- PyErr_Print();
- return 0;
- }
-
-#if IS_PYTHON_3K
- module_spoa = m;
-#endif /* IS_PYTHON_3K */
- worker = w;
- return 1;
-}
-
-static int ps_python_load_file(struct worker *w, const char *file)
-{
- FILE *fp;
- int ret;
-
- fp = fopen(file, "r");
- if (fp == NULL) {
- LOG("python: Cannot read file \"%s\": %s", file, strerror(errno));
- return 0;
- }
-
- ret = PyRun_SimpleFile(fp, file);
- fclose(fp);
- if (ret != 0) {
- PyErr_Print();
- return 0;
- }
-
- return 1;
-}
-
-static int ps_python_exec_message(struct worker *w, void *ref, int nargs, struct spoe_kv *args)
-{
- int i;
- PyObject *python_ref = ref;
- PyObject *fkw;
- PyObject *kw_args;
- PyObject *result;
- PyObject *ent;
- PyObject *key;
- PyObject *value;
- PyObject *func;
- int ret;
- char ipbuf[64];
- const char *p;
- PyObject *ip_dict;
- PyObject *ip_name;
- PyObject *ip_value;
-
- /* Dict containing arguments */
-
- kw_args = PyList_New(0);
- if (kw_args == NULL) {
- PyErr_Print();
- return 0;
- }
-
- for (i = 0; i < nargs; i++) {
-
- /* New dict containing one argument */
-
- ent = PyDict_New();
- if (ent == NULL) {
- Py_DECREF(kw_args);
- PyErr_Print();
- return 0;
- }
-
- /* Create the name entry */
-
- key = PY_STRING_FROM_STRING("name");
- if (key == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- PyErr_Print();
- return 0;
- }
-
- value = PY_STRING_FROM_STRING_AND_SIZE(args[i].name.str, args[i].name.len);
- if (value == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- Py_DECREF(key);
- PyErr_Print();
- return 0;
- }
-
- ret = PyDict_SetItem(ent, key, value);
- Py_DECREF(key);
- Py_DECREF(value);
- if (ret == -1) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- PyErr_Print();
- return 0;
- }
-
- /* Create the value entry */
-
- key = PY_STRING_FROM_STRING("value");
- if (key == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- PyErr_Print();
- return 0;
- }
-
- switch (args[i].value.type) {
- case SPOE_DATA_T_NULL:
- Py_INCREF(Py_None);
- value = Py_None;
- break;
- case SPOE_DATA_T_BOOL:
- value = PyBool_FromLong(args[i].value.u.boolean);
- break;
- case SPOE_DATA_T_INT32:
- value = PyLong_FromLong(args[i].value.u.sint32);
- break;
- case SPOE_DATA_T_UINT32:
- value = PyLong_FromLong(args[i].value.u.uint32);
- break;
- case SPOE_DATA_T_INT64:
- value = PyLong_FromLong(args[i].value.u.sint64);
- break;
- case SPOE_DATA_T_UINT64:
- value = PyLong_FromUnsignedLong(args[i].value.u.uint64);
- break;
- case SPOE_DATA_T_IPV4:
- case SPOE_DATA_T_IPV6:
- if (args[i].value.type == SPOE_DATA_T_IPV4)
- p = inet_ntop(AF_INET, &args[i].value.u.ipv4, ipbuf, 64);
- else
- p = inet_ntop(AF_INET6, &args[i].value.u.ipv6, ipbuf, 64);
- if (!p)
- strcpy(ipbuf, "0.0.0.0");
-
- func = PyObject_GetAttrString(module_ipaddress, "ip_address");
- if (func == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- Py_DECREF(key);
- PyErr_Print();
- return 0;
- }
- ip_dict = PyDict_New();
- if (ip_dict == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- Py_DECREF(key);
- Py_DECREF(func);
- PyErr_Print();
- return 0;
- }
- ip_name = PY_STRING_FROM_STRING("address");
- if (ip_name == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- Py_DECREF(key);
- Py_DECREF(func);
- Py_DECREF(ip_dict);
- PyErr_Print();
- return 0;
- }
- ip_value = PyUnicode_FromString(ipbuf);
- if (ip_value == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- Py_DECREF(key);
- Py_DECREF(func);
- Py_DECREF(ip_dict);
- Py_DECREF(ip_name);
- PyErr_Print();
- return 0;
- }
- ret = PyDict_SetItem(ip_dict, ip_name, ip_value);
- Py_DECREF(ip_name);
- Py_DECREF(ip_value);
- if (ret == -1) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- Py_DECREF(key);
- Py_DECREF(func);
- Py_DECREF(ip_dict);
- PyErr_Print();
- return 0;
- }
- value = PyObject_Call(func, empty_tuple, ip_dict);
- Py_DECREF(func);
- Py_DECREF(ip_dict);
- break;
-
- case SPOE_DATA_T_STR:
- value = PY_STRING_FROM_STRING_AND_SIZE(args[i].value.u.buffer.str, args[i].value.u.buffer.len);
- break;
- case SPOE_DATA_T_BIN:
- value = PY_BYTES_FROM_STRING_AND_SIZE(args[i].value.u.buffer.str, args[i].value.u.buffer.len);
- break;
- default:
- Py_INCREF(Py_None);
- value = Py_None;
- break;
- }
- if (value == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- Py_DECREF(key);
- PyErr_Print();
- return 0;
- }
-
- ret = PyDict_SetItem(ent, key, value);
- Py_DECREF(key);
- Py_DECREF(value);
- if (ret == -1) {
- Py_DECREF(kw_args);
- Py_DECREF(ent);
- PyErr_Print();
- return 0;
- }
-
- /* Add dict to the list */
-
- ret = PyList_Append(kw_args, ent);
- Py_DECREF(ent);
- if (ret == -1) {
- Py_DECREF(kw_args);
- PyErr_Print();
- return 0;
- }
- }
-
- /* Dictionary { args = <list-of-args> } for the function */
-
- fkw = PyDict_New();
- if (fkw == NULL) {
- Py_DECREF(kw_args);
- PyErr_Print();
- return 0;
- }
-
- key = PY_STRING_FROM_STRING("args");
- if (key == NULL) {
- Py_DECREF(kw_args);
- Py_DECREF(fkw);
- PyErr_Print();
- return 0;
- }
-
- ret = PyDict_SetItem(fkw, key, kw_args);
- Py_DECREF(kw_args);
- Py_DECREF(key);
- if (ret == -1) {
- Py_DECREF(fkw);
- PyErr_Print();
- return 0;
- }
-
- result = PyObject_Call(python_ref, empty_tuple, fkw);
- Py_DECREF(fkw);
- if (result == NULL) {
- PyErr_Print();
- return 0;
- }
- Py_DECREF(result);
-
- return 1;
-}
-
-__attribute__((constructor))
-static void __ps_python_init(void)
-{
- ps_register(&ps_python_bindings);
-}
diff --git a/contrib/spoa_server/ps_python.h b/contrib/spoa_server/ps_python.h
deleted file mode 100644
index 069c51d..0000000
--- a/contrib/spoa_server/ps_python.h
+++ /dev/null
@@ -1,52 +0,0 @@
-/* ps_python.h: SPOA Python processing includes
- *
- * Copyright (C) 2020 Gilchrist Dadaglo <gilchrist@dadaglo.com>
- *
- * This program 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; either version
- * 2 of the License, or (at your option) any later version.
- *
- * This program is provided 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.
- *
- */
-
-#ifndef __PS_PYTHON_H__
-#define __PS_PYTHON_H__
-
-#include <Python.h>
-
-#if PY_MAJOR_VERSION >= 3
- #define IS_PYTHON_3K 1
- #define IS_PYTHON_27 0
-#elif PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION == 7
- #define IS_PYTHON_3K 0
- #define IS_PYTHON_27 1
-#else
- #error "Unsupported Python Version - Please use Python 3"
-#endif /* PY_MAJOR_VERSION */
-
-#if IS_PYTHON_3K
- #define PY_INIT_MODULE(instance, name, methods, moduledef) \
- (instance = PyModule_Create(moduledef))
- #define PY_STRING_FROM_STRING PyUnicode_FromString
- #define PY_STRING_FROM_STRING_AND_SIZE PyUnicode_FromStringAndSize
- #define PY_BYTES_FROM_STRING_AND_SIZE PyBytes_FromStringAndSize
- #define PY_STRING_GET_SIZE PyBytes_Size
- #define PY_STRING_AS_STRING PyBytes_AsString
-#elif IS_PYTHON_27
- #define PY_INIT_MODULE(instance, name, methods, moduledef) \
- (instance = Py_InitModule(name, methods))
- #define PY_STRING_FROM_STRING PyString_FromString
- #define PY_STRING_FROM_STRING_AND_SIZE PyString_FromStringAndSize
- #define PY_BYTES_FROM_STRING_AND_SIZE PyString_FromStringAndSize
- #define PY_STRING_GET_SIZE PyString_GET_SIZE
- #define PY_STRING_AS_STRING PyString_AS_STRING
-#endif /* IS_PYTHON_3K */
-
-#endif /* __PS_PYTHON_H__ */
-
-/* EOF */
diff --git a/contrib/spoa_server/ps_python.py b/contrib/spoa_server/ps_python.py
deleted file mode 100644
index 37bda5e..0000000
--- a/contrib/spoa_server/ps_python.py
+++ /dev/null
@@ -1,22 +0,0 @@
-from pprint import pprint
-import spoa
-import ipaddress
-import random
-
-def check_client_ip(args):
- pprint(args)
- spoa.set_var_null("null", spoa.scope_txn)
- spoa.set_var_boolean("boolean", spoa.scope_txn, True)
- spoa.set_var_int32("int32", spoa.scope_txn, 1234)
- spoa.set_var_uint32("uint32", spoa.scope_txn, 1234)
- spoa.set_var_int64("int64", spoa.scope_txn, 1234)
- spoa.set_var_uint64("uint64", spoa.scope_txn, 1234)
- spoa.set_var_ipv4("ipv4", spoa.scope_txn, ipaddress.IPv4Address(u"127.0.0.1"))
- spoa.set_var_ipv6("ipv6", spoa.scope_txn, ipaddress.IPv6Address(u"1::f"))
- spoa.set_var_str("str", spoa.scope_txn, "1::f")
- spoa.set_var_bin("bin", spoa.scope_txn, "1:\x01:\x02f\x00\x00")
- spoa.set_var_int32("ip_score", spoa.scope_sess, random.randint(1,100))
- return
-
-
-spoa.register_message("check-client-ip", check_client_ip)
diff --git a/contrib/spoa_server/spoa-server.conf b/contrib/spoa_server/spoa-server.conf
deleted file mode 100644
index 13bd126..0000000
--- a/contrib/spoa_server/spoa-server.conf
+++ /dev/null
@@ -1,33 +0,0 @@
-global
- debug
-
-defaults
- mode http
- option httplog
- option dontlognull
- timeout connect 5000
- timeout client 5000
- timeout server 5000
-
-listen test
- mode http
- bind :10001
- filter spoe engine spoa-server config spoa-server.spoe.conf
- http-request set-var(req.a) var(txn.iprep.null),debug
- http-request set-var(req.a) var(txn.iprep.boolean),debug
- http-request set-var(req.a) var(txn.iprep.int32),debug
- http-request set-var(req.a) var(txn.iprep.uint32),debug
- http-request set-var(req.a) var(txn.iprep.int64),debug
- http-request set-var(req.a) var(txn.iprep.uint64),debug
- http-request set-var(req.a) var(txn.iprep.ipv4),debug
- http-request set-var(req.a) var(txn.iprep.ipv6),debug
- http-request set-var(req.a) var(txn.iprep.str),debug
- http-request set-var(req.a) var(txn.iprep.bin),debug
- http-request redirect location /%[var(sess.iprep.ip_score)]
-
-backend spoe-server
- mode tcp
- balance roundrobin
- timeout connect 5s
- timeout server 3m
- server spoe-server 127.0.0.1:12345
diff --git a/contrib/spoa_server/spoa-server.spoe.conf b/contrib/spoa_server/spoa-server.spoe.conf
deleted file mode 100644
index dab4e5a..0000000
--- a/contrib/spoa_server/spoa-server.spoe.conf
+++ /dev/null
@@ -1,13 +0,0 @@
-[spoa-server]
-
-spoe-agent spoa-server
- messages check-client-ip
- option var-prefix iprep
- timeout hello 100ms
- timeout idle 30s
- timeout processing 15ms
- use-backend spoe-server
-
-spoe-message check-client-ip
- args always_true int(1234) src ipv6(::55) req.fhdr(host)
- event on-frontend-http-request
diff --git a/contrib/spoa_server/spoa.c b/contrib/spoa_server/spoa.c
deleted file mode 100644
index 9a8ffd0..0000000
--- a/contrib/spoa_server/spoa.c
+++ /dev/null
@@ -1,1316 +0,0 @@
-/*
- * A Random IP reputation service acting as a Stream Processing Offload Agent
- *
- * This is a very simple service that implement a "random" ip reputation
- * service. It will return random scores for all checked IP addresses. It only
- * shows you how to implement a ip reputation service or such kind of services
- * using the SPOE.
- *
- * Copyright 2016 HAProxy Technologies, Christopher Faulet <cfaulet@haproxy.com>
- * Copyright 2018 OZON / Thierry Fournier <thierry.fournier@ozon.io>
- *
- * This program 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; either version
- * 2 of the License, or (at your option) any later version.
- *
- */
-#include <limits.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <stdbool.h>
-#include <unistd.h>
-#include <signal.h>
-#include <pthread.h>
-#include <sys/time.h>
-#include <sys/types.h>
-#include <sys/wait.h>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <netinet/tcp.h>
-#include <arpa/inet.h>
-
-#include "spoa.h"
-
-#define DEFAULT_PORT 12345
-#define NUM_WORKERS 5
-
-#define SLEN(str) (sizeof(str)-1)
-
-/* Frame Types sent by HAProxy and by agents */
-enum spoe_frame_type {
- /* Frames sent by HAProxy */
- SPOE_FRM_T_HAPROXY_HELLO = 1,
- SPOE_FRM_T_HAPROXY_DISCON,
- SPOE_FRM_T_HAPROXY_NOTIFY,
-
- /* Frames sent by the agents */
- SPOE_FRM_T_AGENT_HELLO = 101,
- SPOE_FRM_T_AGENT_DISCON,
- SPOE_FRM_T_AGENT_ACK
-};
-
-/* Errors triggered by SPOE applet */
-enum spoe_frame_error {
- SPOE_FRM_ERR_NONE = 0,
- SPOE_FRM_ERR_IO,
- SPOE_FRM_ERR_TOUT,
- SPOE_FRM_ERR_TOO_BIG,
- SPOE_FRM_ERR_INVALID,
- SPOE_FRM_ERR_NO_VSN,
- SPOE_FRM_ERR_NO_FRAME_SIZE,
- SPOE_FRM_ERR_NO_CAP,
- SPOE_FRM_ERR_BAD_VSN,
- SPOE_FRM_ERR_BAD_FRAME_SIZE,
- SPOE_FRM_ERR_UNKNOWN = 99,
- SPOE_FRM_ERRS,
-};
-
-/* All supported SPOE actions */
-enum spoe_action_type {
- SPOE_ACT_T_SET_VAR = 1,
- SPOE_ACT_T_UNSET_VAR,
- SPOE_ACT_TYPES,
-};
-
-
-/* Masks to get data type or flags value */
-#define SPOE_DATA_T_MASK 0x0F
-#define SPOE_DATA_FL_MASK 0xF0
-
-/* Flags to set Boolean values */
-#define SPOE_DATA_FL_FALSE 0x00
-#define SPOE_DATA_FL_TRUE 0x10
-static const char *spoe_frm_err_reasons[SPOE_FRM_ERRS] = {
- [SPOE_FRM_ERR_NONE] = "normal",
- [SPOE_FRM_ERR_IO] = "I/O error",
- [SPOE_FRM_ERR_TOUT] = "a timeout occurred",
- [SPOE_FRM_ERR_TOO_BIG] = "frame is too big",
- [SPOE_FRM_ERR_INVALID] = "invalid frame received",
- [SPOE_FRM_ERR_NO_VSN] = "version value not found",
- [SPOE_FRM_ERR_NO_FRAME_SIZE] = "max-frame-size value not found",
- [SPOE_FRM_ERR_NO_CAP] = "capabilities value not found",
- [SPOE_FRM_ERR_BAD_VSN] = "unsupported version",
- [SPOE_FRM_ERR_BAD_FRAME_SIZE] = "max-frame-size too big or too small",
- [SPOE_FRM_ERR_UNKNOWN] = "an unknown error occurred",
-};
-
-bool debug = false;
-pthread_key_t worker_id;
-static struct ps *ps_list = NULL;
-static struct ps_message *ps_messages = NULL;
-static int nfiles = 0;
-static char **files = NULL;
-
-static inline void add_file(const char *file)
-{
- nfiles++;
- files = realloc(files, sizeof(*files) * nfiles);
- if (files == NULL) {
- fprintf(stderr, "Out of memory error\n");
- exit(EXIT_FAILURE);
- }
- files[nfiles - 1] = strdup(file);
- if (files[nfiles - 1] == NULL) {
- fprintf(stderr, "Out of memory error\n");
- exit(EXIT_FAILURE);
- }
-}
-
-void ps_register(struct ps *ps)
-{
- ps->next = ps_list;
- ps_list = ps;
-}
-
-void ps_register_message(struct ps *ps, const char *name, void *ref)
-{
- struct ps_message *msg;
-
- /* Look for already registered name */
- for (msg = ps_messages; msg; msg = msg->next) {
- if (strcmp(name, msg->name) == 0) {
- LOG("Message \"%s\" already registered\n", name);
- exit(EXIT_FAILURE);
- }
- }
-
- msg = calloc(1, sizeof(*msg));
- if (msg == NULL) {
- LOG("Out of memory error\n");
- exit(EXIT_FAILURE);
- }
-
- msg->next = ps_messages;
- ps_messages = msg;
- msg->name = strdup(name);
- if (msg->name == NULL) {
- LOG("Out of memory error\n");
- exit(EXIT_FAILURE);
- }
- msg->ref = ref;
- msg->ps = ps;
-}
-
-static int
-do_read(int sock, void *buf, int read_len)
-{
- fd_set readfds;
- int n = 0, total = 0, bytesleft = read_len;
-
- FD_ZERO(&readfds);
- FD_SET(sock, &readfds);
-
- while (total < read_len) {
- if (select(FD_SETSIZE, &readfds, NULL, NULL, NULL) == -1)
- return -1;
- if (!FD_ISSET(sock, &readfds))
- return -1;
-
- n = read(sock, buf + total, bytesleft);
- if (n <= 0)
- break;
-
- total += n;
- bytesleft -= n;
- }
-
- return (n == -1) ? -1 : total;
-}
-
-static int
-do_write(int sock, void *buf, int write_len)
-{
- fd_set writefds;
- int n = 0, total = 0, bytesleft = write_len;
-
- FD_ZERO(&writefds);
- FD_SET(sock, &writefds);
-
- while (total < write_len) {
- if (select(FD_SETSIZE, NULL, &writefds, NULL, NULL) == -1)
- return -1;
- if (!FD_ISSET(sock, &writefds))
- return -1;
-
- n = write(sock, buf + total, bytesleft);
- if (n <= 0)
- break;
-
- total += n;
- bytesleft -= n;
- }
-
- return (n == -1) ? -1 : total;
-}
-
-/* Receive a frame sent by HAProxy. It returns -1 if an error occurred,
- * otherwise the number of read bytes.*/
-static int
-read_frame(int sock, struct worker *w)
-{
- uint32_t netint;
- unsigned int framesz;
-
- /* Read the frame size, on 4 bytes */
- if (do_read(sock, &netint, sizeof(netint)) != 4) {
- w->status_code = SPOE_FRM_ERR_IO;
- return -1;
- }
-
- /* Check it against the max size */
- framesz = ntohl(netint);
- if (framesz > w->size) {
- w->status_code = SPOE_FRM_ERR_TOO_BIG;
- return -1;
- }
-
- /* Read the frame */
- if (do_read(sock, w->buf, framesz) != framesz) {
- w->status_code = SPOE_FRM_ERR_IO;
- return -1;
- }
-
- w->len = framesz;
- return framesz;
-}
-
-/* Send a frame to HAProxy. It returns -1 if an error occurred, otherwise the
- * number of written bytes. */
-static int
-write_frame(int sock, struct worker *w)
-{
- uint32_t netint;
-
- /* Write the frame size, on 4 bytes */
- netint = htonl(w->len);
- if (do_write(sock, &netint, sizeof(netint)) != 4) {
- w->status_code = SPOE_FRM_ERR_IO;
- return -1;
- }
-
- /* Write the frame */
- if (do_write(sock, w->buf, w->len) != w->len) {
- w->status_code = SPOE_FRM_ERR_IO;
- return -1;
- }
- return w->len;
-}
-
-/* Encode a variable-length integer. This function never fails and returns the
- * number of written bytes. */
-static int
-encode_spoe_varint(uint64_t i, char *buf)
-{
- int idx;
-
- if (i < 240) {
- buf[0] = (unsigned char)i;
- return 1;
- }
-
- buf[0] = (unsigned char)i | 240;
- i = (i - 240) >> 4;
- for (idx = 1; i >= 128; ++idx) {
- buf[idx] = (unsigned char)i | 128;
- i = (i - 128) >> 7;
- }
- buf[idx++] = (unsigned char)i;
- return idx;
-}
-
-/* Decode a varable-length integer. If the decoding fails, -1 is returned. This
- * happens when the buffer's end in reached. On success, the number of read
- * bytes is returned. */
-static int
-decode_spoe_varint(char *buf, char *end, uint64_t *i)
-{
- unsigned char *msg = (unsigned char *)buf;
- int idx = 0;
-
- if (msg > (unsigned char *)end)
- return -1;
-
- if (msg[0] < 240) {
- *i = msg[0];
- return 1;
- }
- *i = msg[0];
- do {
- ++idx;
- if (msg+idx > (unsigned char *)end)
- return -1;
- *i += (uint64_t)msg[idx] << (4 + 7 * (idx-1));
- } while (msg[idx] >= 128);
- return (idx + 1);
-}
-
-/* Encode a string. The string will be prefix by its length, encoded as a
- * variable-length integer. This function never fails and returns the number of
- * written bytes. */
-static int
-encode_spoe_string(const char *str, size_t len, char *dst)
-{
- int idx = 0;
-
- if (!len) {
- dst[0] = 0;
- return 1;
- }
-
- idx += encode_spoe_varint(len, dst);
- memcpy(dst+idx, str, len);
- return (idx + len);
-}
-
-/* Decode a string. Its length is decoded first as a variable-length integer. If
- * it succeeds, and if the string length is valid, the begin of the string is
- * saved in <*str>, its length is saved in <*len> and the total numbre of bytes
- * read is returned. If an error occurred, -1 is returned and <*str> remains
- * NULL. */
-static int
-decode_spoe_string(char *buf, char *end, char **str, uint64_t *len)
-{
- int r, idx = 0;
-
- *str = NULL;
- *len = 0;
-
- if ((r = decode_spoe_varint(buf, end, len)) == -1)
- goto error;
- idx += r;
- if (buf + idx + *len > end)
- goto error;
-
- *str = buf+idx;
- return (idx + *len);
-
-error:
- return -1;
-}
-
-/* Skip a typed data. If an error occurred, -1 is returned, otherwise the number
- * of bytes read is returned. A types data is composed of a type (1 byte) and
- * corresponding data:
- * - boolean: non additional data (0 bytes)
- * - integers: a variable-length integer (see decode_spoe_varint)
- * - ipv4: 4 bytes
- * - ipv6: 16 bytes
- * - binary and string: a buffer prefixed by its size, a variable-length
- * integer (see decode_spoe_string) */
-static int
-skip_spoe_data(char *frame, char *end)
-{
- uint64_t sz = 0;
- int r, idx = 0;
-
- if (frame > end)
- return -1;
-
- switch (frame[idx++] & SPOE_DATA_T_MASK) {
- case SPOE_DATA_T_BOOL:
- idx++;
- break;
- case SPOE_DATA_T_INT32:
- case SPOE_DATA_T_INT64:
- case SPOE_DATA_T_UINT32:
- case SPOE_DATA_T_UINT64:
- if ((r = decode_spoe_varint(frame+idx, end, &sz)) == -1)
- return -1;
- idx += r;
- break;
- case SPOE_DATA_T_IPV4:
- idx += 4;
- break;
- case SPOE_DATA_T_IPV6:
- idx += 16;
- break;
- case SPOE_DATA_T_STR:
- case SPOE_DATA_T_BIN:
- if ((r = decode_spoe_varint(frame+idx, end, &sz)) == -1)
- return -1;
- idx += r + sz;
- break;
- }
-
- if (frame+idx > end)
- return -1;
- return idx;
-}
-
-/* Decode a typed data. If an error occurred, -1 is returned, otherwise the
- * number of read bytes is returned. See skip_spoe_data for details. */
-static int
-decode_spoe_data(char *frame, char *end, struct spoe_data *data)
-{
- uint64_t sz = 0;
- int type, r, idx = 0;
-
- if (frame > end)
- return -1;
-
- type = frame[idx++];
- data->type = (type & SPOE_DATA_T_MASK);
- switch (data->type) {
- case SPOE_DATA_T_BOOL:
- data->u.boolean = ((type & SPOE_DATA_FL_TRUE) == SPOE_DATA_FL_TRUE);
- break;
- case SPOE_DATA_T_INT32:
- if ((r = decode_spoe_varint(frame+idx, end, &sz)) == -1)
- return -1;
- data->u.sint32 = sz;
- idx += r;
- break;
- case SPOE_DATA_T_INT64:
- if ((r = decode_spoe_varint(frame+idx, end, &sz)) == -1)
- return -1;
- data->u.uint32 = sz;
- idx += r;
- break;
- case SPOE_DATA_T_UINT32:
- if ((r = decode_spoe_varint(frame+idx, end, &sz)) == -1)
- return -1;
- data->u.sint64 = sz;
- idx += r;
- break;
- case SPOE_DATA_T_UINT64:
- if ((r = decode_spoe_varint(frame+idx, end, &sz)) == -1)
- return -1;
- data->u.uint64 = sz;
- idx += r;
- break;
- case SPOE_DATA_T_IPV4:
- if (frame+idx+4 > end)
- return -1;
- memcpy(&data->u.ipv4, frame+idx, 4);
- idx += 4;
- break;
- case SPOE_DATA_T_IPV6:
- if (frame+idx+16 > end)
- return -1;
- memcpy(&data->u.ipv6, frame+idx, 16);
- idx += 16;
- break;
- case SPOE_DATA_T_STR:
- if ((r = decode_spoe_varint(frame+idx, end, &sz)) == -1)
- return -1;
- idx += r;
- if (frame+idx+sz > end)
- return -1;
- data->u.buffer.str = frame+idx;
- data->u.buffer.len = sz;
- idx += sz;
- break;
- case SPOE_DATA_T_BIN:
- if ((r = decode_spoe_varint(frame+idx, end, &sz)) == -1)
- return -1;
- idx += r;
- if (frame+idx+sz > end)
- return -1;
- data->u.buffer.str = frame+idx;
- data->u.buffer.len = sz;
- idx += sz;
- break;
- default:
- break;
- }
-
- if (frame+idx > end)
- return -1;
- return idx;
-}
-
-
-/* Check the protocol version. It returns -1 if an error occurred, the number of
- * read bytes otherwise. */
-static int
-check_proto_version(struct worker *w, int idx)
-{
- char *str;
- uint64_t sz;
-
- /* Get the list of all supported versions by HAProxy */
- if ((w->buf[idx++] & SPOE_DATA_T_MASK) != SPOE_DATA_T_STR) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- return -1;
- }
- idx += decode_spoe_string(w->buf+idx, w->buf+w->len, &str, &sz);
- if (str == NULL) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- return -1;
- }
-
- /* TODO: Find the right version in supported ones */
-
- return idx;
-}
-
-/* Check max frame size value. It returns -1 if an error occurred, the number of
- * read bytes otherwise. */
-static int
-check_max_frame_size(struct worker *w, int idx)
-{
- uint64_t sz;
- int type, i;
-
- /* Get the max-frame-size value of HAProxy */
- type = w->buf[idx++];
- if ((type & SPOE_DATA_T_MASK) != SPOE_DATA_T_INT32 &&
- (type & SPOE_DATA_T_MASK) != SPOE_DATA_T_INT64 &&
- (type & SPOE_DATA_T_MASK) != SPOE_DATA_T_UINT32 &&
- (type & SPOE_DATA_T_MASK) != SPOE_DATA_T_UINT64) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- return -1;
- }
- if ((i = decode_spoe_varint(w->buf+idx, w->buf+w->len, &sz)) == -1) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- return -1;
- }
- idx += i;
-
- /* Keep the lower value */
- if (sz < w->size)
- w->size = sz;
-
- return idx;
-}
-
-/* Check healthcheck value. It returns -1 if an error occurred, the number of
- * read bytes otherwise. */
-static int
-check_healthcheck(struct worker *w, int idx)
-{
- int type;
-
- /* Get the "healthcheck" value of HAProxy */
- type = w->buf[idx++];
- if ((type & SPOE_DATA_T_MASK) != SPOE_DATA_T_BOOL) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- return -1;
- }
- w->healthcheck = ((type & SPOE_DATA_FL_TRUE) == SPOE_DATA_FL_TRUE);
- return idx;
-}
-
-
-/* Decode a HELLO frame received from HAProxy. It returns -1 if an error
- * occurred, 0 if the frame must be skipped, otherwise the number of read
- * bytes. */
-static int
-handle_hahello(struct worker *w)
-{
- char *end = w->buf+w->len;
- int i, idx = 0;
-
- /* Check frame type */
- if (w->buf[idx++] != SPOE_FRM_T_HAPROXY_HELLO)
- goto skip;
-
- /* Skip flags */
- idx += 4;
-
- /* stream-id and frame-id must be cleared */
- if (w->buf[idx] != 0 || w->buf[idx+1] != 0) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- idx += 2;
-
- /* Loop on K/V items */
- while (idx < w->len) {
- char *str;
- uint64_t sz;
-
- /* Decode the item name */
- idx += decode_spoe_string(w->buf+idx, end, &str, &sz);
- if (str == NULL) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
-
- /* Check "supported-versions" K/V item */
- if (!memcmp(str, "supported-versions", sz)) {
- if ((i = check_proto_version(w, idx)) == -1)
- goto error;
- idx = i;
- }
- /* Check "max-frame-size" K/V item "*/
- else if (!memcmp(str, "max-frame-size", sz)) {
- if ((i = check_max_frame_size(w, idx)) == -1)
- goto error;
- idx = i;
- }
- /* Check "healthcheck" K/V item "*/
- else if (!memcmp(str, "healthcheck", sz)) {
- if ((i = check_healthcheck(w, idx)) == -1)
- goto error;
- idx = i;
- }
- /* Skip "capabilities" K/V item for now */
- else {
- /* Silently ignore unknown item */
- if ((i = skip_spoe_data(w->buf+idx, end)) == -1) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- idx += i;
- }
- }
-
- return idx;
-skip:
- return 0;
-error:
- return -1;
-}
-
-/* Decode a DISCONNECT frame received from HAProxy. It returns -1 if an error
- * occurred, 0 if the frame must be skipped, otherwise the number of read
- * bytes. */
-static int
-handle_hadiscon(struct worker *w)
-{
- char *end = w->buf+w->len;
- int i, idx = 0;
-
- /* Check frame type */
- if (w->buf[idx++] != SPOE_FRM_T_HAPROXY_DISCON)
- goto skip;
-
- /* Skip flags */
- idx += 4;
-
- /* stream-id and frame-id must be cleared */
- if (w->buf[idx] != 0 || w->buf[idx+1] != 0) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- idx += 2;
-
- /* Loop on K/V items */
- while (idx < w->len) {
- char *str;
- uint64_t sz;
-
- /* Decode item key */
- idx += decode_spoe_string(w->buf+idx, end, &str, &sz);
- if (str == NULL) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- /* Silently ignore unknown item */
- if ((i = skip_spoe_data(w->buf+idx, end)) == -1) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- idx += i;
- }
-
- w->status_code = SPOE_FRM_ERR_NONE;
- return idx;
-skip:
- return 0;
-error:
- return -1;
-}
-
-/* Encode a ACK frame to send it to HAProxy. It returns -1 if an error occurred,
- * the number of written bytes otherwise. */
-static void prepare_agentack(struct worker *w)
-{
- unsigned int flags = 0;
-
- w->ack_len = 0;
-
- /* Frame type */
- w->ack[w->ack_len++] = SPOE_FRM_T_AGENT_ACK;
-
- /* Set flags */
- flags |= htonl(SPOE_FRM_FL_FIN);
- memcpy(w->ack + w->ack_len, &flags, 4);
- w->ack_len += 4;
-
- /* Set stream-id and frame-id for ACK frames */
- w->ack_len += encode_spoe_varint(w->stream_id, w->ack + w->ack_len);
- w->ack_len += encode_spoe_varint(w->frame_id, w->ack + w->ack_len);
-}
-
-static inline
-int set_var_name(struct worker *w, const char *name, int name_len, unsigned char scope)
-{
- w->ack[w->ack_len++] = SPOE_ACT_T_SET_VAR; /* Action type */
- w->ack[w->ack_len++] = 3; /* Number of args */
- w->ack[w->ack_len++] = scope; /* Arg 1: the scope */
- w->ack_len += encode_spoe_string(name, name_len, w->ack+w->ack_len); /* Arg 2: variable name */
- return 1;
-}
-
-int set_var_null(struct worker *w,
- const char *name, int name_len,
- unsigned char scope)
-{
- if (!set_var_name(w, name, name_len, scope))
- return 0;
- w->ack[w->ack_len++] = SPOE_DATA_T_NULL;
- return 1;
-}
-
-int set_var_bool(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, bool value)
-{
- if (!set_var_name(w, name, name_len, scope))
- return 0;
- w->ack[w->ack_len++] = SPOE_DATA_T_BOOL | (!!value << 4);
- return 1;
-}
-
-static inline
-int set_var_int(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, int type, uint64_t value)
-{
- if (!set_var_name(w, name, name_len, scope))
- return 0;
- w->ack[w->ack_len++] = SPOE_DATA_T_UINT32;
- w->ack_len += encode_spoe_varint(value, w->ack+w->ack_len); /* Arg 3: variable value */
- return 1;
-}
-
-int set_var_uint32(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, uint32_t value)
-{
- return set_var_int(w, name, name_len, scope, SPOE_DATA_T_UINT32, value);
-}
-
-int set_var_int32(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, int32_t value)
-{
- return set_var_int(w, name, name_len, scope, SPOE_DATA_T_INT32, value);
-}
-
-int set_var_uint64(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, uint64_t value)
-{
- return set_var_int(w, name, name_len, scope, SPOE_DATA_T_INT32, value);
-}
-
-int set_var_int64(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, int64_t value)
-{
- return set_var_int(w, name, name_len, scope, SPOE_DATA_T_INT32, value);
-}
-
-int set_var_ipv4(struct worker *w,
- const char *name, int name_len,
- unsigned char scope,
- struct in_addr *ipv4)
-{
- if (!set_var_name(w, name, name_len, scope))
- return 0;
- w->ack[w->ack_len++] = SPOE_DATA_T_IPV4;
- memcpy(w->ack+w->ack_len, ipv4, 4);
- w->ack_len += 4;
- return 1;
-}
-
-int set_var_ipv6(struct worker *w,
- const char *name, int name_len,
- unsigned char scope,
- struct in6_addr *ipv6)
-{
- if (!set_var_name(w, name, name_len, scope))
- return 0;
- w->ack[w->ack_len++] = SPOE_DATA_T_IPV6;
- memcpy(w->ack+w->ack_len, ipv6, 16);
- w->ack_len += 16;
- return 1;
-}
-
-static inline
-int set_var_buf(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, int type,
- const char *str, int str_len)
-{
- if (!set_var_name(w, name, name_len, scope))
- return 0;
- w->ack[w->ack_len++] = type;
- w->ack_len += encode_spoe_string(str, str_len, w->ack+w->ack_len);
- return 1;
-}
-
-int set_var_string(struct worker *w,
- const char *name, int name_len,
- unsigned char scope,
- const char *str, int strlen)
-{
- return set_var_buf(w, name, name_len, scope, SPOE_DATA_T_STR, str, strlen);
-}
-
-int set_var_bin(struct worker *w,
- const char *name, int name_len,
- unsigned char scope,
- const char *str, int strlen)
-{
- return set_var_buf(w, name, name_len, scope, SPOE_DATA_T_BIN, str, strlen);
-}
-
-/* This function is a little bit ugly,
- * TODO: improve the response without copying the buffer
- */
-static int commit_agentack(struct worker *w)
-{
- memcpy(w->buf, w->ack, w->ack_len);
- w->len = w->ack_len;
- return 1;
-}
-
-/* Decode a NOTIFY frame received from HAProxy. It returns -1 if an error
- * occurred, 0 if the frame must be skipped, otherwise the number of read
- * bytes. */
-static int
-handle_hanotify(struct worker *w)
-{
- char *end = w->buf+w->len;
- uint64_t stream_id, frame_id;
- int nbargs, i, idx = 0;
- int index;
- struct spoe_kv args[256];
- uint64_t length;
- struct ps_message *msg;
-
- /* Check frame type */
- if (w->buf[idx++] != SPOE_FRM_T_HAPROXY_NOTIFY)
- goto skip;
-
- /* Skip flags */
- idx += 4;
-
- /* Read the stream-id */
- if ((i = decode_spoe_varint(w->buf+idx, end, &stream_id)) == -1) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- idx += i;
-
- /* Read the frame-id */
- if ((i = decode_spoe_varint(w->buf+idx, end, &frame_id)) == -1) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- idx += i;
-
- w->stream_id = (unsigned int)stream_id;
- w->frame_id = (unsigned int)frame_id;
-
- DEBUG("Notify frame received: stream-id=%u - frame-id=%u",
- w->stream_id, w->frame_id);
-
- /* Prepare ack, if the processing fails the ack will be cancelled */
- prepare_agentack(w);
-
- /* Loop on messages */
- while (idx < w->len) {
- char *str;
- uint64_t sz;
-
- /* Decode the message name */
- idx += decode_spoe_string(w->buf+idx, end, &str, &sz);
- if (str == NULL) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- DEBUG(" Message '%.*s' received", (int)sz, str);
-
- /* Decode all SPOE data */
- nbargs = (unsigned char)w->buf[idx++];
- for (index = 0; index < nbargs; index++) {
-
- /* Read the key name */
- if ((i = decode_spoe_string(w->buf+idx, end,
- &args[index].name.str,
- &length)) == -1) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- if (length > INT_MAX) {
- w->status_code = SPOE_FRM_ERR_TOO_BIG;
- goto error;
- }
- args[index].name.len = length;
- idx += i;
-
- /* Read the value */
- memset(&args[index].value, 0, sizeof(args[index].value));
- if ((i = decode_spoe_data(w->buf+idx, end, &args[index].value)) == -1) {
- w->status_code = SPOE_FRM_ERR_INVALID;
- goto error;
- }
- idx += i;
- }
-
- /* Lookup for existing bindings. If no existing message
- * where found, does nothing.
- */
- for (msg = ps_messages; msg; msg = msg->next)
- if (sz == strlen(msg->name) && strncmp(str, msg->name, sz) == 0)
- break;
- if (msg == NULL || msg->ps->exec_message == NULL) {
- DEBUG(" Message '%.*s' have no bindings registered", (int)sz, str);
- continue;
- }
-
- /* Process the message */
- msg->ps->exec_message(w, msg->ref, nbargs, args);
- }
-
- return idx;
-skip:
- return 0;
-error:
- return -1;
-}
-
-/* Encode a HELLO frame to send it to HAProxy. It returns -1 if an error
- * occurred, the number of written bytes otherwise. */
-static int
-prepare_agenthello(struct worker *w)
-{
- int idx = 0;
- unsigned int flags = 0;
-
- /* Frame Type */
- w->buf[idx++] = SPOE_FRM_T_AGENT_HELLO;
-
- /* Set flags */
- flags |= htonl(SPOE_FRM_FL_FIN);
- memcpy(w->buf+idx, &flags, 4);
- idx += 4;
-
- /* No stream-id and frame-id for HELLO frames */
- w->buf[idx++] = 0;
- w->buf[idx++] = 0;
-
- /* "version" K/V item */
- idx += encode_spoe_string("version", 7, w->buf+idx);
- w->buf[idx++] = SPOE_DATA_T_STR;
- idx += encode_spoe_string(SPOP_VERSION, SLEN(SPOP_VERSION), w->buf+idx);
-
- /* "max-frame-size" K/V item */
- idx += encode_spoe_string("max-frame-size", 14, w->buf+idx);
- w->buf[idx++] = SPOE_DATA_T_UINT32;
- idx += encode_spoe_varint(w->size, w->buf+idx);
-
- /* "capabilities" K/V item */
- idx += encode_spoe_string("capabilities", 12, w->buf+idx);
- w->buf[idx++] = SPOE_DATA_T_STR;
- idx += encode_spoe_string(SPOA_CAPABILITIES, SLEN(SPOA_CAPABILITIES), w->buf+idx);
-
- w->len = idx;
- return idx;
-}
-
-/* Encode a DISCONNECT frame to send it to HAProxy. It returns -1 if an error
- * occurred, the number of written bytes otherwise. */
-static int
-prepare_agentdicon(struct worker *w)
-{
- const char *reason;
- int rlen, idx = 0;
- unsigned int flags = 0;
-
- if (w->status_code >= SPOE_FRM_ERRS)
- w->status_code = SPOE_FRM_ERR_UNKNOWN;
- reason = spoe_frm_err_reasons[w->status_code];
- rlen = strlen(reason);
-
- /* Frame type */
- w->buf[idx++] = SPOE_FRM_T_AGENT_DISCON;
-
- /* Set flags */
- flags |= htonl(SPOE_FRM_FL_FIN);
- memcpy(w->buf+idx, &flags, 4);
- idx += 4;
-
- /* No stream-id and frame-id for DISCONNECT frames */
- w->buf[idx++] = 0;
- w->buf[idx++] = 0;
-
- /* There are 2 mandatory items: "status-code" and "message" */
-
- /* "status-code" K/V item */
- idx += encode_spoe_string("status-code", 11, w->buf+idx);
- w->buf[idx++] = SPOE_DATA_T_UINT32;
- idx += encode_spoe_varint(w->status_code, w->buf+idx);
-
- /* "message" K/V item */
- idx += encode_spoe_string("message", 7, w->buf+idx);
- w->buf[idx++] = SPOE_DATA_T_STR;
- idx += encode_spoe_string(reason, rlen, w->buf+idx);
-
- w->len = idx;
- return idx;
-}
-
-static int
-hello_handshake(int sock, struct worker *w)
-{
- if (read_frame(sock, w) < 0) {
- LOG("Failed to read Haproxy HELLO frame");
- goto error;
- }
- if (handle_hahello(w) < 0) {
- LOG("Failed to handle Haproxy HELLO frame");
- goto error;
- }
- if (prepare_agenthello(w) < 0) {
- LOG("Failed to prepare Agent HELLO frame");
- goto error;
- }
- if (write_frame(sock, w) < 0) {
- LOG("Failed to write Agent frame");
- goto error;
- }
- DEBUG("Hello handshake done: version=%s - max-frame-size=%u - healthcheck=%s",
- SPOP_VERSION, w->size, (w->healthcheck ? "true" : "false"));
- return 0;
-error:
- return -1;
-}
-
-static int
-notify_ack_roundtip(int sock, struct worker *w)
-{
- if (read_frame(sock, w) < 0) {
- LOG("Failed to read Haproxy NOTIFY frame");
- goto error_or_quit;
- }
- if (handle_hadiscon(w) != 0) {
- if (w->status_code != SPOE_FRM_ERR_NONE)
- LOG("Failed to handle Haproxy DISCONNECT frame");
- DEBUG("Disconnect frame received: reason=%s",
- spoe_frm_err_reasons[w->status_code]);
- goto error_or_quit;
- }
- if (handle_hanotify(w) < 0) {
- LOG("Failed to handle Haproxy NOTIFY frame");
- goto error_or_quit;
- }
- if (commit_agentack(w) < 0) {
- LOG("Failed to prepare Agent ACK frame");
- goto error_or_quit;
- }
- if (write_frame(sock, w) < 0) {
- LOG("Failed to write Agent ACK frame");
- goto error_or_quit;
- }
- DEBUG("Ack frame sent: stream-id=%u - frame-id=%u",
- w->stream_id, w->frame_id);
- return 0;
-error_or_quit:
- return -1;
-}
-
-static void *
-spoa_worker(void *data)
-{
- struct worker w;
- struct sockaddr_in client;
- int *info = (int *)data;
- int csock, lsock = info[0];
- struct ps *ps;
- int i;
- int len;
-
- signal(SIGPIPE, SIG_IGN);
- pthread_setspecific(worker_id, &info[1]);
-
- /* Init registered processors */
- for (ps = ps_list; ps != NULL; ps = ps->next)
- ps->init_worker(&w);
-
- /* Load files */
- for (i = 0; i < nfiles; i++) {
- len = strlen(files[i]);
- for (ps = ps_list; ps != NULL; ps = ps->next)
- if (strcmp(files[i] + len - strlen(ps->ext), ps->ext) == 0)
- break;
- if (ps == NULL) {
- LOG("Can't load file \"%s\"\n", files[i]);
- goto out;
- }
- if (!ps->load_file(&w, files[i]))
- goto out;
- }
-
- while (1) {
- socklen_t sz = sizeof(client);
-
- if ((csock = accept(lsock, (struct sockaddr *)&client, &sz)) < 0) {
- LOG("Failed to accept client connection: %m");
- goto out;
- }
- memset(&w, 0, sizeof(w));
- w.id = info[1];
- w.size = MAX_FRAME_SIZE;
-
- DEBUG("New connection from HAProxy accepted");
-
- if (hello_handshake(csock, &w) < 0)
- goto disconnect;
- if (w.healthcheck == true)
- goto close;
- while (1) {
- if (notify_ack_roundtip(csock, &w) < 0)
- break;
- }
-
- disconnect:
- if (w.status_code == SPOE_FRM_ERR_IO) {
- LOG("Close the client socket because of I/O errors");
- goto close;
- }
- if (prepare_agentdicon(&w) < 0) {
- LOG("Failed to prepare Agent DISCONNECT frame");
- goto close;
- }
- if (write_frame(csock, &w) < 0) {
- LOG("Failed to write Agent DISCONNECT frame");
- goto close;
- }
- DEBUG("Disconnect frame sent: reason=%s",
- spoe_frm_err_reasons[w.status_code]);
-
- close:
- close(csock);
- }
-
-out:
- free(info);
-#if 0
- pthread_exit(NULL);
-#endif
- return NULL;
-}
-
-int process_create(pid_t *pid, void *(*ps)(void *), void *data)
-{
- if (debug) {
- ps(data);
- exit(EXIT_SUCCESS);
- }
- *pid = fork();
- if (*pid == -1)
- return -1;
- if (*pid > 0)
- return 0;
- ps(data);
- return 0;
-}
-
-static void
-usage(char *prog)
-{
- fprintf(stderr, "Usage: %s [-h] [-d] [-p <port>] [-n <num-workers>] -f <file>\n", prog);
- fprintf(stderr, " -h Print this message\n");
- fprintf(stderr, " -d Enable the debug mode\n");
- fprintf(stderr, " -p <port> Specify the port to listen on (default: 12345)\n");
- fprintf(stderr, " -n <num-workers> Specify the number of workers (default: 5)\n");
- fprintf(stderr, " -f <file> Specify the file whoch contains the processing code.\n");
- fprintf(stderr, " This argument can specified more than once.\n");
-}
-
-int
-main(int argc, char **argv)
-{
-#if 0
- pthread_t *ts = NULL;
-#endif
- pid_t *pids;
- struct sockaddr_in server;
- int i, sock, opt, nbworkers, port;
- int status;
-
- nbworkers = NUM_WORKERS;
- port = DEFAULT_PORT;
- while ((opt = getopt(argc, argv, "hdn:p:f:")) != -1) {
- switch (opt) {
- case 'h':
- usage(argv[0]);
- return EXIT_SUCCESS;
- case 'd':
- debug = true;
- break;
- case 'n':
- nbworkers = atoi(optarg);
- break;
- case 'p':
- port = atoi(optarg);
- break;
- case 'f':
- add_file(optarg);
- break;
- default:
- usage(argv[0]);
- return EXIT_FAILURE;
- }
- }
-
- if (nbworkers <= 0) {
- fprintf(stderr, "%s: Invalid number of workers '%d'\n",
- argv[0], nbworkers);
- goto error;
- }
- if (port <= 0) {
- fprintf(stderr, "%s: Invalid port '%d'\n", argv[0], port);
- goto error;
- }
-
- if((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
- fprintf(stderr, "Failed creating socket: %m\n");
- goto error;
- }
-
- setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (int []){1}, sizeof(int));
- setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, (int []){1}, sizeof(int));
-
- memset(&server, 0, sizeof(server));
- server.sin_family = AF_INET;
- server.sin_addr.s_addr = INADDR_ANY;
- server.sin_port = htons(port);
-
- if (bind(sock, (struct sockaddr *)&server, sizeof(server)) < 0) {
- fprintf(stderr, "Failed to bind the socket: %m\n");
- goto error;
- }
-
- if (listen(sock , 10) < 0) {
- fprintf(stderr, "Failed to listen on the socket: %m\n");
- goto error;
- }
- fprintf(stderr, "SPOA is listening on port %d\n", port);
-
- pthread_key_create(&worker_id, NULL);
-
- /* Initialise the server in thread mode. This code is commented
- * out and not deleted, because later I expect to work with
- * process ansd threads. This first version just support processes.
- */
-#if 0
- ts = calloc(nbworkers, sizeof(*ts));
- for (i = 0; i < nbworkers; i++) {
- int *info = calloc(2, sizeof(*info));
-
- info[0] = sock;
- info[1] = i+1;
-
- if (pthread_create(&ts[i], NULL, spoa_worker, info) < 0) {
- fprintf(stderr, "Failed to create thread %d: %m\n", i+1);
- goto error;
- }
- fprintf(stderr, "SPOA worker %02d started\n", i+1);
- }
-
- for (i = 0; i < nbworkers; i++) {
- pthread_join(ts[i], NULL);
- fprintf(stderr, "SPOA worker %02d stopped\n", i+1);
- }
- free(ts);
-#endif
-
- /* Start processes */
- pids = calloc(nbworkers, sizeof(*pids));
- if (!pids) {
- fprintf(stderr, "Out of memory error\n");
- goto error;
- }
- for (i = 0; i < nbworkers; i++) {
- int *info = calloc(2, sizeof(*info));
-
- info[0] = sock;
- info[1] = i+1;
-
- if (process_create(&pids[i], spoa_worker, info) == -1) {
- fprintf(stderr, "SPOA worker %02d started\n", i+1);
- goto error;
- }
- fprintf(stderr, "SPOA worker %02d started\n", i+1);
- }
- for (i = 0; i < nbworkers; i++) {
- waitpid(pids[0], &status, 0);
- fprintf(stderr, "SPOA worker %02d stopped\n", i+1);
- }
-
- close(sock);
- pthread_key_delete(worker_id);
- return EXIT_SUCCESS;
-
-error:
- return EXIT_FAILURE;
-}
diff --git a/contrib/spoa_server/spoa.h b/contrib/spoa_server/spoa.h
deleted file mode 100644
index 8d6d4be..0000000
--- a/contrib/spoa_server/spoa.h
+++ /dev/null
@@ -1,167 +0,0 @@
-/* Main SPOA server includes
- *
- * Copyright 2016 HAProxy Technologies, Christopher Faulet <cfaulet@haproxy.com>
- * Copyright 2018 OZON / Thierry Fournier <thierry.fournier@ozon.io>
- *
- * This program 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; either version
- * 2 of the License, or (at your option) any later version.
- */
-#ifndef __SPOA_H__
-#define __SPOA_H__
-
-#include <pthread.h>
-#include <stdbool.h>
-#include <stdint.h>
-#include <netinet/in.h>
-#include <sys/time.h>
-
-#ifndef MAX_FRAME_SIZE
-#define MAX_FRAME_SIZE 16384
-#endif
-
-#define SPOP_VERSION "2.0"
-#define SPOA_CAPABILITIES ""
-
-/* Flags set on the SPOE frame */
-#define SPOE_FRM_FL_FIN 0x00000001
-
-/* All supported data types */
-enum spoe_data_type {
- SPOE_DATA_T_NULL = 0,
- SPOE_DATA_T_BOOL,
- SPOE_DATA_T_INT32,
- SPOE_DATA_T_UINT32,
- SPOE_DATA_T_INT64,
- SPOE_DATA_T_UINT64,
- SPOE_DATA_T_IPV4,
- SPOE_DATA_T_IPV6,
- SPOE_DATA_T_STR,
- SPOE_DATA_T_BIN,
- SPOE_DATA_TYPES
-};
-
-/* Scopes used for variables set by agents. It is a way to be agnotic to vars
- * scope. */
-enum spoe_vars_scope {
- SPOE_SCOPE_PROC = 0, /* <=> SCOPE_PROC */
- SPOE_SCOPE_SESS, /* <=> SCOPE_SESS */
- SPOE_SCOPE_TXN, /* <=> SCOPE_TXN */
- SPOE_SCOPE_REQ, /* <=> SCOPE_REQ */
- SPOE_SCOPE_RES, /* <=> SCOPE_RES */
-};
-
-struct worker {
- unsigned int id;
- char buf[MAX_FRAME_SIZE];
- unsigned int len;
- unsigned int size;
- int status_code;
- unsigned int stream_id;
- unsigned int frame_id;
- bool healthcheck;
- char ack[MAX_FRAME_SIZE];
- unsigned int ack_len;
-};
-
-struct chunk {
- char *str; /* beginning of the string itself. Might not be 0-terminated */
- int len; /* current size of the string from first to last char */
-};
-
-union spoe_value {
- bool boolean; /* use for boolean */
- int32_t sint32; /* used for signed 32bits integers */
- uint32_t uint32; /* used for signed 32bits integers */
- int32_t sint64; /* used for signed 64bits integers */
- uint32_t uint64; /* used for signed 64bits integers */
- struct in_addr ipv4; /* used for ipv4 addresses */
- struct in6_addr ipv6; /* used for ipv6 addresses */
- struct chunk buffer; /* used for char strings or buffers */
-};
-
-/* Used to store sample constant */
-struct spoe_data {
- enum spoe_data_type type; /* SPOE_DATA_T_* */
- union spoe_value u; /* spoe data value */
-};
-
-struct spoe_kv {
- struct chunk name;
- struct spoe_data value;
-};
-
-struct ps {
- struct ps *next;
- char *ext;
- int (*init_worker)(struct worker *w);
- int (*exec_message)(struct worker *w, void *ref, int nargs, struct spoe_kv *args);
- int (*load_file)(struct worker *w, const char *file);
-};
-
-struct ps_message {
- struct ps_message *next;
- const char *name;
- struct ps *ps;
- void *ref;
-};
-
-extern bool debug;
-extern pthread_key_t worker_id;
-
-void ps_register(struct ps *ps);
-void ps_register_message(struct ps *ps, const char *name, void *ref);
-
-int set_var_null(struct worker *w,
- const char *name, int name_len,
- unsigned char scope);
-int set_var_bool(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, bool value);
-int set_var_uint32(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, uint32_t value);
-int set_var_int32(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, int32_t value);
-int set_var_uint64(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, uint64_t value);
-int set_var_int64(struct worker *w,
- const char *name, int name_len,
- unsigned char scope, int64_t value);
-int set_var_ipv4(struct worker *w,
- const char *name, int name_len,
- unsigned char scope,
- struct in_addr *ipv4);
-int set_var_ipv6(struct worker *w,
- const char *name, int name_len,
- unsigned char scope,
- struct in6_addr *ipv6);
-int set_var_string(struct worker *w,
- const char *name, int name_len,
- unsigned char scope,
- const char *str, int strlen);
-int set_var_bin(struct worker *w,
- const char *name, int name_len,
- unsigned char scope,
- const char *str, int strlen);
-
-#define LOG(fmt, args...) \
- do { \
- struct timeval now; \
- int wid = *((int*)pthread_getspecific(worker_id)); \
- \
- gettimeofday(&now, NULL); \
- fprintf(stderr, "%ld.%06ld [%02d] " fmt "\n", \
- now.tv_sec, now.tv_usec, wid, ##args); \
- } while (0)
-
-#define DEBUG(x...) \
- do { \
- if (debug) \
- LOG(x); \
- } while (0)
-
-#endif /* __SPOA_H__ */