X-Git-Url: https://jfr.im/git/irc/evilnet/x3.git/blobdiff_plain/e0f76584a52189fdf040ec65fb45ad17bc21db71..3b7fa78b1de8f9ee8718cba3da3b2db522b70620:/src/mod-python.c diff --git a/src/mod-python.c b/src/mod-python.c index 55cbded..f563cfa 100644 --- a/src/mod-python.c +++ b/src/mod-python.c @@ -23,32 +23,43 @@ #ifdef WITH_PYTHON /* just disable this file if python doesnt exist */ -#include "Python.h" +#include #include "chanserv.h" #include "conf.h" #include "modcmd.h" #include "nickserv.h" #include "opserv.h" #include "saxdb.h" -#include "sendmail.h" +#include "mail.h" #include "timeq.h" +#include "compat.h" /* TODO notes * * - Impliment most of proto-p10 irc_* commands for calling from scripts * - Impliment functions to look up whois, channel, account, and reg-channel info for scripts * - Impliment x3.conf settings for python variables like include path, etc. - * - kod-python.py calls for everything you can reg_ a handler for in x3 + * - modpython.py calls for everything you can reg_ a handler for in x3 * - Some kind of system for getting needed binds bound automagicaly to make it easier * to run peoples scripts and mod-python in general. + * - An interface to reading/writing data to x3.db. Maybe generic, or attached to account or channel reg records? */ static const struct message_entry msgtab[] = { { "PYMSG_RELOAD_SUCCESS", "Reloaded Python scripts successfully." }, { "PYMSG_RELOAD_FAILED", "Error reloading Python scripts." }, + { "PYMSG_RUN_UNKNOWN_EXCEPTION", "Error running python: unknown exception." }, + { "PYMSG_RUN_EXCEPTION", "Error running python: %s." }, { NULL, NULL } /* sentenal */ }; +#define MODPYTHON_CONF_NAME "modules/python" + +static +struct { + char const* scripts_dir; +} modpython_conf; + static struct log_type *PY_LOG; const char *python_module_deps[] = { NULL }; static struct module *python_module; @@ -56,13 +67,26 @@ static struct module *python_module; PyObject *base_module = NULL; /* Base python handling library */ PyObject *handler_object = NULL; /* instanciation of handler class */ + +extern struct userNode *global, *chanserv, *opserv, *nickserv, *spamserv; + +/* ---------------------------------------------------------------------- * + Some hooks you can call from modpython.py to interact with the + service, and IRC. These emb_* functions are available as svc.* + in python. + */ + static PyObject* -emb_dump(PyObject *self, PyObject *args) +emb_dump(UNUSED_ARG(PyObject *self), PyObject *args) { + /* Dump a raw string into the socket + usage: svc.dump() + */ char *buf; int ret = 0; char linedup[MAXLEN]; + if(!PyArg_ParseTuple(args, "s:dump", &buf )) return NULL; safestrncpy(linedup, buf, sizeof(linedup)); @@ -74,8 +98,11 @@ emb_dump(PyObject *self, PyObject *args) } static PyObject* -emb_send_target_privmsg(PyObject *self, PyObject *args) +emb_send_target_privmsg(UNUSED_ARG(PyObject *self), PyObject *args) { + /* Send a privmsg + usage: svc.send_target_privmsg(, , ) + */ int ret = 0; char *servicenick; char *channel; @@ -83,6 +110,7 @@ emb_send_target_privmsg(PyObject *self, PyObject *args) struct service *service; + if(!PyArg_ParseTuple(args, "sss:reply", &servicenick, &channel, &buf )) return NULL; if(!(service = service_find(servicenick))) { @@ -94,8 +122,11 @@ emb_send_target_privmsg(PyObject *self, PyObject *args) } static PyObject* -emb_send_target_notice(PyObject *self, PyObject *args) +emb_send_target_notice(UNUSED_ARG(PyObject *self), PyObject *args) { + /* send a notice + usage: svc.send_target_notice(, , ) + */ int ret = 0; char *servicenick; char *target; @@ -103,6 +134,7 @@ emb_send_target_notice(PyObject *self, PyObject *args) struct service *service; + if(!PyArg_ParseTuple(args, "sss:reply", &servicenick, &target, &buf )) return NULL; if(!(service = service_find(servicenick))) { @@ -113,15 +145,19 @@ emb_send_target_notice(PyObject *self, PyObject *args) return Py_BuildValue("i", ret); } - static PyObject* -emb_get_user(PyObject *self, PyObject *args) +emb_get_user(UNUSED_ARG(PyObject *self), PyObject *args) { + /* Get a python object containing everything x3 knows about a user, by nick. + usage: svc.get_user() + */ char *nick; struct userNode *user; struct modeNode *mn; unsigned int n; PyObject* pChanList; + + if(!PyArg_ParseTuple(args, "s", &nick)) return NULL; if(!(user = GetUserH(nick))) { @@ -159,8 +195,11 @@ emb_get_user(PyObject *self, PyObject *args) } static PyObject* -emb_get_channel(PyObject *self, PyObject *args) +emb_get_channel(UNUSED_ARG(PyObject *self), PyObject *args) { + /* Returns a python dict object with all sorts of info about a channel. + usage: svc.get_channel() + */ char *name; struct chanNode *channel; unsigned int n; @@ -168,6 +207,7 @@ emb_get_channel(PyObject *self, PyObject *args) PyObject *pChannelBans; PyObject *pChannelExempts; + if(!PyArg_ParseTuple(args, "s", &name)) return NULL; if(!(channel = GetChannel(name))) { @@ -194,7 +234,6 @@ emb_get_channel(PyObject *self, PyObject *args) ); } - /* build tuple of exempts */ pChannelExempts = PyTuple_New(channel->exemptlist.used); for(n=0; n < channel->exemptlist.used;n++) { @@ -207,8 +246,6 @@ emb_get_channel(PyObject *self, PyObject *args) ); } - - return Py_BuildValue("{s:s,s:s,s:s,s:i" ",s:i,s:i,s:O,s:O,s:O}", @@ -225,103 +262,163 @@ emb_get_channel(PyObject *self, PyObject *args) ); } -/* static PyObject* -emb_get_account(PyObject *self, PyObject *args) +emb_get_account(UNUSED_ARG(PyObject *self), PyObject *args) { + /* Returns a python dict object with all sorts of info about an account. + usage: svc.get_account() + */ char *name; + struct handle_info *hi; + + if(!PyArg_ParseTuple(args, "s", &name)) return NULL; + + hi = get_handle_info(name); + if(!hi) { + return NULL; + } + return Py_BuildValue("{s:s,s:i,s:s,s:s,s:s" + ",s:s,s:s}", + + "account", hi->handle, + "registered", hi->registered, + "last_seen", hi->lastseen, + "infoline", hi->infoline ? hi->infoline : "", + "email", hi->email_addr ? hi->email_addr : "", + + "fakehost", hi->fakehost ? hi->fakehost : "", + "last_quit_host", hi->last_quit_host + + /* TODO: */ + /* users online authed to this account */ + /* cookies */ + /* nicks (nickserv nets only?) */ + /* masks */ + /* ignores */ + /* channels */ + ); } -*/ +static PyObject* +emb_get_info(UNUSED_ARG(PyObject *self), UNUSED_ARG(PyObject *args)) +{ + /* return some info about the general setup + * of X3, such as what the chanserv's nickname + * is. + */ + + + return Py_BuildValue("{s:s,s:s,s:s,s:s,s:s}", + "chanserv", chanserv? chanserv->nick : "ChanServ", + "nickserv", nickserv?nickserv->nick : "NickServ", + "opserv", opserv?opserv->nick : "OpServ", + "global", global?global->nick : "Global", + "spamserv", spamserv?spamserv->nick : "SpamServ"); +} + +static PyObject* +emb_log_module(UNUSED_ARG(PyObject *self), PyObject *args) +{ + /* a gateway to standard X3 logging subsystem. + * level is a value 0 to 9 as defined by the log_severity enum in log.h. + * LOG_INFO is 3, LOG_WARNING is 6, LOG_ERROR is 7. + * + * for now, all logs go to the PY_LOG log. In the future this will change. + */ + char *message; + int ret = 0; + int level; + + + if(!PyArg_ParseTuple(args, "is", &level, &message)) + return NULL; + + log_module(PY_LOG, level, "%s", message); + + return Py_BuildValue("i", ret); +} static PyMethodDef EmbMethods[] = { + /* Communication methods */ {"dump", emb_dump, METH_VARARGS, "Dump raw P10 line to server"}, {"send_target_privmsg", emb_send_target_privmsg, METH_VARARGS, "Send a message to somewhere"}, {"send_target_notice", emb_send_target_notice, METH_VARARGS, "Send a notice to somewhere"}, + {"log_module", emb_log_module, METH_VARARGS, "Log something using the X3 log subsystem"}, +//TODO: {"exec_cmd", emb_exec_cmd, METH_VARARGS, "execute x3 command provided"}, +// This should use environment from "python command" call to pass in, if available +//TODO: {"kill" +//TODO: {"shun" +//TODO: {"unshun" +//TODO: {"gline", emb_gline, METH_VARARGS, "gline a mask"}, +//TODO: {"ungline", emb_ungline, METH_VARARGS, "remove a gline"}, +//TODO: {"kick", emb_kick, METH_VARARGS, "kick someone from a channel"}, +//TODO: {"channel_mode", emb_channel_mode, METH_VARARGS, "set modes on a channel"}, +//TODO: {"user_mode", emb_user_mode, METH_VARARGS, "Have x3 set usermodes on one of its own nicks"}, +// +//TODO: {"get_config", emb_get_config, METH_VARARGS, "get x3.conf settings into a nested dict"}, +//TODO: {"config_set", emb_config_set, METH_VARARGS, "change a config setting 'on-the-fly'."}, +// +//TODO: {"timeq_add", emb_timeq_new, METH_VARARGS, "some kind of interface to the timed event system."}, +//TODO: {"timeq_del", emb_timeq_new, METH_VARARGS, "some kind of interface to the timed event system."}, + /* Information gathering methods */ {"get_user", emb_get_user, METH_VARARGS, "Get details about a nickname"}, {"get_channel", emb_get_channel, METH_VARARGS, "Get details about a channel"}, + {"get_account", emb_get_account, METH_VARARGS, "Get details about an account"}, + {"get_info", emb_get_info, METH_VARARGS, "Get various misc info about x3"}, + /* null terminator */ {NULL, NULL, 0, NULL} }; +/* ------------------------------------------------------------------------------------------------ * + Thes functions set up the embedded environment for us to call out to modpython.py class + methods. + */ -/* This is just a hack-job for testing. It'll go away. */ -int python_call_func_real(char *function, char *args[], size_t argc) { - /* TODO: get arguments, pass through to python function */ - PyObject *pFunc, *pValue; - PyObject *pArgs = NULL; - if(base_module != NULL) { - log_module(PY_LOG, LOG_INFO, "Attempting to run python function %s", function); - pFunc = PyObject_GetAttrString(base_module, function); - /* pFunc is a new reference */ - if(pFunc && PyCallable_Check(pFunc)) { - size_t i; - if(args && argc) { - pArgs = PyTuple_New(argc); - for(i = 0; i< argc; ++i) { - pValue = PyString_FromString(args[i]); - if(!pValue) { - Py_DECREF(pArgs); - Py_DECREF(pFunc); - log_module(PY_LOG, LOG_INFO, "Unable to convert '%s' to python string", args[i]); - return 0; - } - PyTuple_SetItem(pArgs, i, pValue); - } - } +void python_log_module() { + /* Attempt to convert python errors to x3 log system */ + PyObject *exc, *tb, *value, *tmp; + char *str_exc = "NONE"; + char *str_value = "NONE"; + char *str_tb = "NONE"; - pValue = PyObject_CallObject(pFunc, pArgs); - if(pArgs != NULL) { - Py_DECREF(pArgs); - } - if(pValue != NULL) { - int ret; - ret = PyInt_AsLong(pValue); - if(ret == -1 && PyErr_Occurred()) { - PyErr_Print(); - log_module(PY_LOG, LOG_INFO, "error converting return value of %s to type long. ", function); - ret = 0; - } - log_module(PY_LOG, LOG_INFO, "%s was run successfully, returned %d.", function, ret); - /* TODO: convert pValue to c int, return it below */ - Py_DECREF(pValue); - return ret; - } - else { - Py_DECREF(pFunc); - /* TODO: instead of print errors, get them as strings - * and deal with them with normal x3 log system. */ - PyErr_Print(); - log_module(PY_LOG, LOG_WARNING, "call to %s failed", function); - return 0; - } - } - else { - if(PyErr_Occurred()) - PyErr_Print(); - log_module(PY_LOG, LOG_WARNING, "function %s not found or uncallable", function); - return 0; - } - } - else { - return 0; + PyErr_Fetch(&exc, &value, &tb); + + if(exc) { + if((tmp = PyObject_Str(exc))) + str_exc = PyString_AsString(tmp); + } + if(value) { + if((tmp = PyObject_Str(value))) + str_value = PyString_AsString(tmp); + } + if(tb) { + if((tmp = PyObject_Str(tb))) + str_tb = PyString_AsString(tmp); } -} -/* This is just a hack-job for testing. It will go away */ -int python_call_func(char *function, char *args[], size_t argc, char *command_caller, char *command_target, char *command_service) { - char *setargs[] = {command_caller?command_caller:"", - command_target?command_target:"", - command_service?command_service:""}; - python_call_func_real("command_set", setargs, 3); - python_call_func_real(function, args, argc); - python_call_func_real("command_clear", NULL, 0); - return 0; + /* Now restore it so we can print it (just in case) + * (should we do this only when running in debug mode?) */ + PyErr_Restore(exc, value, tb); + PyErr_Print(); /* which of course, clears it again.. */ + + log_module(PY_LOG, LOG_WARNING, "PYTHON error: %s, value: %s", str_exc, str_value); + + /* TODO: get the traceback using the traceback module via C api so we can add it to the X3 logs. See + * http://mail.python.org/pipermail/python-list/2003-March/192226.html */ + // (this doesnt work, str_tb is just an object hashid) log_module(PY_LOG, LOG_INFO, "PYTHON error, traceback: %s", str_tb); } + PyObject *python_build_handler_args(size_t argc, char *args[], PyObject *pIrcObj) { + /* Sets up a python tuple with passed in arguments, prefixed by the Irc instance + which handlers use to interact with c. + argc = number of args + args = array of args + pIrcObj = instance of the irc class + */ size_t i = 0, n; PyObject *pArgs = NULL; @@ -345,6 +442,10 @@ PyObject *python_build_handler_args(size_t argc, char *args[], PyObject *pIrcObj } PyObject *python_build_args(size_t argc, char *args[]) { + /* Builds the passed in arguments into a python argument tuple. + argc = number of args + args = array of args + */ size_t i; PyObject *pArgs = NULL; @@ -366,11 +467,18 @@ PyObject *python_build_args(size_t argc, char *args[]) { PyObject *new_irc_object(char *command_service, char *command_caller, char *command_target) { + /* Creates a new instance of the irc class (from modpython.py) which is initalized + with current environment details like which service were talking to. + command_service = which service we are talking to, or empty string if none + command_caller = nick of user generating message, or empty string if none + command_target = If were reacting to something on a channel, this will + be set to the name of the channel. Otherwise empty + */ PyObject *pIrcArgs = NULL; PyObject *pIrcClass; PyObject *pIrcObj; - log_module(PY_LOG, LOG_INFO, "Attempting to instanciate irc class"); + log_module(PY_LOG, LOG_INFO, "Attempting to instanciate irc class; %s %s %s", command_service, command_caller, command_target); pIrcClass = PyObject_GetAttrString(base_module, "irc"); /* pIrcClass is a new reference */ if(pIrcClass && PyCallable_Check(pIrcClass)) { @@ -379,48 +487,29 @@ PyObject *new_irc_object(char *command_service, char *command_caller, char *comm //PyObject *pValue; pIrcArgs = python_build_args(3, ircargs); - /* - pIrcArgs = PyTuple_New(sizeof(ircargs)); - for(i = 0; i< ircargc; ++i) { - if(ircargs[i]) { - pValue = PyString_FromString(ircargs[i]); - if(!pValue) { - Py_DECREF(pIrcArgs); - log_module(PY_LOG, LOG_ERROR, "Unable to convert '%s' to python string", ircargs[i]); - return 0; - } - } - else { - pValue = Py_None; - Py_INCREF(Py_None); - } - PyTuple_SetItem(pIrcArgs, i, pValue); - } - */ pIrcObj = PyObject_CallObject(pIrcClass, pIrcArgs); if(!pIrcObj) { log_module(PY_LOG, LOG_ERROR, "IRC Class failed to load"); - PyErr_Print(); + python_log_module(); + //PyErr_Print(); } if(pIrcArgs != NULL) { Py_DECREF(pIrcArgs); } + Py_DECREF(pIrcClass); return pIrcObj; } else { + /* need to free pIrcClass here if it WAS found but was NOT callable? */ log_module(PY_LOG, LOG_ERROR, "Unable to find irc class"); return NULL; } } - int python_call_handler(char *handler, char *args[], size_t argc, char *command_service, char *command_caller, char *command_target) { - /* TODO: - * - Instanciate class 'irc' with command-* arguments and save it. - * - get/find handler class instance - * - call handler. passing in proper args - * - destroy irc instance - * - return something useful? + /* This is how we talk to modpython.c. First a new instance of the irc class is created using these + arguments to setup the current environment. Then the named method of the handler object is called + with the givin arguments. */ PyObject *pIrcObj; PyObject *pArgs; @@ -428,7 +517,7 @@ int python_call_handler(char *handler, char *args[], size_t argc, char *command_ PyObject *pValue; log_module(PY_LOG, LOG_INFO, "attempting to call handler %s.", handler); - if(base_module != NULL) { + if(base_module != NULL && handler_object != NULL) { pIrcObj = new_irc_object(command_service, command_caller, command_target); if(!pIrcObj) { log_module(PY_LOG, LOG_INFO, "Can't get irc object. Bailing."); @@ -438,6 +527,7 @@ int python_call_handler(char *handler, char *args[], size_t argc, char *command_ pArgs = python_build_handler_args(argc, args, pIrcObj); pMethod = PyObject_GetAttrString(handler_object, handler); if(pMethod && PyCallable_Check(pMethod)) { + /* Call the method, with the arguments */ pValue = PyObject_CallObject(pMethod, pArgs); if(pArgs) { Py_DECREF(pArgs); @@ -446,26 +536,31 @@ int python_call_handler(char *handler, char *args[], size_t argc, char *command_ int ret; ret = PyInt_AsLong(pValue); if(ret == -1 && PyErr_Occurred()) { - PyErr_Print(); + //PyErr_Print(); log_module(PY_LOG, LOG_INFO, "error converting return value of handler %s to type long. ", handler); + python_log_module(); ret = 0; } log_module(PY_LOG, LOG_INFO, "handler %s was run successfully, returned %d.", handler, ret); - /* TODO: convert pValue to c int, return it below */ Py_DECREF(pValue); + Py_DECREF(pIrcObj); + Py_DECREF(pMethod); return ret; } else { - Py_DECREF(pIrcObj); - Py_DECREF(pMethod); /* TODO: instead of print errors, get them as strings * and deal with them with normal x3 log system. */ - PyErr_Print(); log_module(PY_LOG, LOG_WARNING, "call to handler %s failed", handler); + //PyErr_Print(); + python_log_module(); + Py_DECREF(pIrcObj); + Py_DECREF(pMethod); return 0; } } else { /* couldn't find handler methed */ + Py_DECREF(pArgs); + /* Free pMethod if it was found but not callable? */ log_module(PY_LOG, LOG_ERROR, "Cannot find handler %s.", handler); return 0; @@ -478,6 +573,10 @@ int python_call_handler(char *handler, char *args[], size_t argc, char *command_ } PyObject *python_new_handler_object() { + /* Create a new instance of the handler class. + This is called during python initilization (or reload) + and the result is saved and re-used. + */ PyObject *pHandlerClass, *pHandlerObj; log_module(PY_LOG, LOG_INFO, "Attempting to instanciate python class handler"); @@ -487,28 +586,48 @@ PyObject *python_new_handler_object() { /*PyObject *pValue; */ pHandlerObj = PyObject_CallObject(pHandlerClass, NULL); - return pHandlerObj; + if(pHandlerObj != NULL) { + log_module(PY_LOG, LOG_INFO, "Created new python handler object."); + return pHandlerObj; + } + else { + log_module(PY_LOG, LOG_ERROR, "Unable to instanciate handler object"); + //PyErr_Print(); + python_log_module(); + return NULL; + } } else { log_module(PY_LOG, LOG_ERROR, "Unable to find handler class"); + //PyErr_Print(); + python_log_module(); + if(pHandlerClass) { + Py_DECREF(pHandlerClass); + } return NULL; } } -/* debate: do we just register these and check them in python - * for every one (slow?) or actually work out if a plugin needs - * it first? We will start by doing it every time. +/* ------------------------------------------------------------------------------- * + Some gateway functions to convert x3 callbacks into modpython.py callbacks. + Mostly we just build relevant args and call the proper handler method + + debate: do we just register these and check them in python + for every one (slow?) or actually work out if a plugin needs + it first? We will start by doing it every time. */ static int python_handle_join(struct modeNode *mNode) { + /* callback for handle_join events. + */ struct userNode *user = mNode->user; struct chanNode *channel = mNode->channel; log_module(PY_LOG, LOG_INFO, "python module handle_join"); if(!channel||!user) { - log_module(PY_LOG, LOG_WARNING, "Join without channel or user?"); + log_module(PY_LOG, LOG_WARNING, "Python code got join without channel or user!"); return 0; } else { @@ -517,14 +636,73 @@ python_handle_join(struct modeNode *mNode) } } +static int +python_handle_server_link(struct server *server) +{ + log_module(PY_LOG, LOG_INFO, "python module handle_server_link"); + if(!server) { + log_module(PY_LOG, LOG_WARNING, "Python code got server link without server!"); + return 0; + } + else { + char *args[] = {server->name, server->description}; + return python_call_handler("server_link", args, 2, "", "", ""); + } +} + +static int +python_handle_new_user(struct userNode *user) +{ + log_module(PY_LOG, LOG_INFO, "Python module handle_new_user"); + if(!user) { + log_module(PY_LOG, LOG_WARNING, "Python code got new_user without the user"); + return 0; + } + else { + char *args[] = {user->nick, user->ident, user->hostname, user->info}; + return python_call_handler("new_user", args, 4, "", "", ""); + } +} + +static void +python_handle_nick_change(struct userNode *user, const char *old_nick) +{ + log_module(PY_LOG, LOG_INFO, "Python module handle_nick_change"); + if(!user) { + log_module(PY_LOG, LOG_WARNING, "Python code got nick_change without the user!"); + } + else { + char *args[] = {user->nick, (char *)old_nick}; + python_call_handler("nick_change", args, 2, "", "", ""); + } +} + +/* ----------------------------------------------------------------------------- */ + int python_load() { + /* Init the python engine and do init work on modpython.py + This is called during x3 startup, and on a python reload + */ PyObject *pName; + char* buffer; + char* env = getenv("PYTHONPATH"); + + if (env) + env = strdup(env); + + if (!env) + setenv("PYTHONPATH", modpython_conf.scripts_dir, 1); + else if (!strstr(env, modpython_conf.scripts_dir)) { + buffer = (char*)malloc(strlen(env) + strlen(modpython_conf.scripts_dir) + 2); + sprintf(buffer, "%s:%s", modpython_conf.scripts_dir, env); + setenv("PYTHONPATH", buffer, 1); + free(buffer); + free(env); + } - setenv("PYTHONPATH", "/home/rubin/afternet/services/x3/x3-run/", 1); Py_Initialize(); Py_InitModule("svc", EmbMethods); - //PyRun_SimpleString("import svc"); /* TODO: get "modpython" from x3.conf */ pName = PyString_FromString("modpython"); base_module = PyImport_Import(pName); @@ -542,16 +720,21 @@ int python_load() { } } else { - PyErr_Print(); + //PyErr_Print(); + python_log_module(); log_module(PY_LOG, LOG_WARNING, "Failed to load modpython.py"); return 0; } return 0; } -/* Called after X3 is fully up and running */ int python_finalize(void) { + /* Called after X3 is fully up and running. + Code can be put here that needs to run to init things, but + which is sensitive to everything else in x3 being up and ready + to go. + */ PyRun_SimpleString("print 'Hello, World of Python!'"); log_module(PY_LOG, LOG_INFO, "python module finalize"); @@ -559,16 +742,24 @@ python_finalize(void) { return 1; } -/* Called on shutdown of the module */ static void -python_cleanup(void) -{ +python_cleanup(void) { + /* Called on shutdown of the python module (or before reloading) + */ + log_module(PY_LOG, LOG_INFO, "python module cleanup"); + if (PyErr_Occurred()) + PyErr_Clear(); Py_Finalize(); /* Shut down python enterpriter */ return; } +/* ---------------------------------------------------------------------------------- * + Python module command handlers. +*/ static MODCMD_FUNC(cmd_reload) { + /* reload the python system completely + */ log_module(PY_LOG, LOG_INFO, "Shutting python down"); python_cleanup(); log_module(PY_LOG, LOG_INFO, "Loading python stuff"); @@ -581,21 +772,151 @@ static MODCMD_FUNC(cmd_reload) { return 1; } +static char* format_python_error(int space_nls) { + PyObject* extype = NULL, *exvalue = NULL, *extraceback = NULL; + PyObject* pextypestr = NULL, *pexvaluestr = NULL; + char* extypestr = NULL, *exvaluestr = NULL; + size_t retvallen = 0; + char* retval = NULL, *tmp; + + PyErr_Fetch(&extype, &exvalue, &extraceback); + if (!extype) + goto cleanup; + + pextypestr = PyObject_Str(extype); + if (!pextypestr) + goto cleanup; + extypestr = PyString_AsString(pextypestr); + if (!extypestr) + goto cleanup; + + pexvaluestr = PyObject_Str(exvalue); + if (pexvaluestr) + exvaluestr = PyString_AsString(pexvaluestr); + + retvallen = strlen(extypestr) + (exvaluestr ? strlen(exvaluestr) + 2 : 0) + 1; + retval = (char*)malloc(retvallen); + if (exvaluestr) + snprintf(retval, retvallen, "%s: %s", extypestr, exvaluestr); + else + strncpy(retval, extypestr, retvallen); + + if (space_nls) { + tmp = retval; + while (*tmp) { + if (*tmp == '\n') + *tmp = ' '; + ++tmp; + } + } + +cleanup: + if (PyErr_Occurred()) + PyErr_Clear(); /* ignore errors caused by formatting */ + Py_XDECREF(extype); + Py_XDECREF(exvalue); + Py_XDECREF(extraceback); + Py_XDECREF(pextypestr); + Py_XDECREF(pexvaluestr); + + if (retval) + return retval; + + return strdup("unknown exception"); +} + +static int python_run_statements(char const* msg, char** err) { + PyObject* o; + char* exmsg = NULL; + PyObject* py_main_module = NULL; + PyObject* py_globals; + + py_main_module = PyImport_AddModule("__main__"); + if (!py_main_module) { + exmsg = format_python_error(1); + if (exmsg) + *err = exmsg; + else + *err = strdup("unknown exception"); + return 1; + } + + py_globals = PyModule_GetDict(py_main_module); + + o = PyRun_String(msg, Py_file_input, py_globals, py_globals); + if (o == NULL) { + exmsg = format_python_error(1); + if (exmsg) + *err = exmsg; + else + *err = strdup("unknown exception"); + return 1; + } + else { + Py_DECREF(o); + } + + return 0; +} + static MODCMD_FUNC(cmd_run) { - - char *msg; + /* this method allows running arbitrary python commands. + * use with care. + */ + char* msg; + char* exmsg = NULL; + msg = unsplit_string(argv + 1, argc - 1, NULL); - /* PyRun_SimpleString(msg); */ - char *args[] = {msg}; - python_call_func("run", args, 1, user?user->nick:"", channel?channel->name:"", cmd->parent->bot->nick); + + if (python_run_statements(msg, &exmsg)) { + if (exmsg) { + reply("PYMSG_RUN_EXCEPTION", exmsg); + free(exmsg); + } else + reply("PYMSG_RUN_UNKNOWN_EXCEPTION"); + } + return 1; } -/* Called on init of the module during startup */ +#define numstrargs(X) sizeof(X) / sizeof(*X) +static MODCMD_FUNC(cmd_command) { + char *plugin = argv[1]; + char *command = argv[2]; + char *msg; /* args */ + if(argc > 3) { + msg = unsplit_string(argv + 3, argc - 3, NULL); + } + else { + msg = ""; + } + char *args[] = {plugin, command, msg}; + python_call_handler("cmd_command", args, numstrargs(args), cmd->parent->bot->nick, user?user->nick:"", channel?channel->name:""); + return 1; +} + +static void modpython_conf_read(void) { + dict_t conf_node; + char const* str; + + if (!(conf_node = conf_get_data(MODPYTHON_CONF_NAME, RECDB_OBJECT))) { + log_module(PY_LOG, LOG_ERROR, "config node '%s' is missing or has wrong type", MODPYTHON_CONF_NAME); + return; + } + + str = database_get_data(conf_node, "scripts_dir", RECDB_QSTRING); + modpython_conf.scripts_dir = str ? str : "./"; +} + int python_init(void) { + /* X3 calls this function on init of the module during startup. We use it to + do all our setup tasks and bindings + */ PY_LOG = log_register_type("Python", "file:python.log"); python_module = module_register("python", PY_LOG, "mod-python.help", NULL); + conf_register_reload(modpython_conf_read); + log_module(PY_LOG, LOG_INFO, "python module init"); message_register_table(msgtab); @@ -609,7 +930,36 @@ int python_init(void) { */ modcmd_register(python_module, "reload", cmd_reload, 1, MODCMD_REQUIRE_AUTHED, "flags", "+oper", NULL); modcmd_register(python_module, "run", cmd_run, 2, MODCMD_REQUIRE_AUTHED, "flags", "+oper", NULL); + modcmd_register(python_module, "command", cmd_command, 3, MODCMD_REQUIRE_STAFF, NULL); + +// Please help us by implimenting any of the callbacks listed as TODO below. They already exist +// in x3, they just need handle_ bridges implimented. (see python_handle_join for an example) + reg_server_link_func(python_handle_server_link); + reg_new_user_func(python_handle_new_user); + reg_nick_change_func(python_handle_nick_change); +//TODO: reg_del_user_func(python_handle_del_user); +//TODO: reg_account_func(python_handle_account); /* stamping of account name to the ircd */ +//TODO: reg_handle_rename_func(python_handle_handle_rename); /* handle used to ALSO mean account name */ +//TODO: reg_failpw_func(python_handle_failpw); +//TODO: reg_allowauth_func(python_handle_allowauth); +//TODO: reg_handle_merge_func(python_handle_merge); +// +//TODO: reg_oper_func(python_handle_oper); +//TODO: reg_new_channel_func(python_handle_new_channel); reg_join_func(python_handle_join); +//TODO: reg_del_channel_func(python_handle_del_channel); +//TODO: reg_part_func(python_handle_part); +//TODO: reg_kick_func(python_handle_kick); +//TODO: reg_topic_func(python_handle_topic); +//TODO: reg_channel_mode_func(python_handle_channel_mode); + +//TODO: reg_privmsg_func(python_handle_privmsg); +//TODO: reg_notice_func +//TODO: reg_svccmd_unbind_func(python_handle_svccmd_unbind); +//TODO: reg_chanmsg_func(python_handle_chanmsg); +//TODO: reg_allchanmsg_func +//TODO: reg_user_mode_func + reg_exit_func(python_cleanup); python_load();