diff options
Diffstat (limited to 'cvs/cvsweb.cgi/code/nelns/admin_executor_service/admin_executor_service.cpp?rev=1.9&content-type=text/x-cvsweb-markup/index.html')
| -rw-r--r-- | cvs/cvsweb.cgi/code/nelns/admin_executor_service/admin_executor_service.cpp?rev=1.9&content-type=text/x-cvsweb-markup/index.html | 516 |
1 files changed, 516 insertions, 0 deletions
diff --git a/cvs/cvsweb.cgi/code/nelns/admin_executor_service/admin_executor_service.cpp?rev=1.9&content-type=text/x-cvsweb-markup/index.html b/cvs/cvsweb.cgi/code/nelns/admin_executor_service/admin_executor_service.cpp?rev=1.9&content-type=text/x-cvsweb-markup/index.html new file mode 100644 index 00000000..ee691d76 --- /dev/null +++ b/cvs/cvsweb.cgi/code/nelns/admin_executor_service/admin_executor_service.cpp?rev=1.9&content-type=text/x-cvsweb-markup/index.html @@ -0,0 +1,516 @@ +<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.0 Transitional//EN"><HTML> +<HEAD><style> A { color:black }</style> +<!-- hennerik CVSweb $Revision: 1.93 $ --> +<TITLE>code/nelns/admin_executor_service/admin_executor_service.cpp - view - 1.9</TITLE></HEAD> +<BODY BGCOLOR="#eeeeee"> +<table width="100%" border=0 cellspacing=0 cellpadding=1 bgcolor="#aaaaaa"><tr valign=bottom><td><a href="admin_executor_service.cpp"><IMG SRC="http://www.nevrax.org/inc/img/picto-up.gif" ALT="[BACK]" BORDER="0" WIDTH="14" HEIGHT="13"></a> <b>Return to <A HREF="admin_executor_service.cpp">admin_executor_service.cpp</A> + CVS log</b> <IMG SRC="http://www.nevrax.org/inc/img/picto-news.gif" ALT="[TXT]" BORDER="0" WIDTH="13" HEIGHT="15"></td><td align=right><IMG SRC="http://www.nevrax.org/inc/img/picto-dir.gif" ALT="[DIR]" BORDER="0" WIDTH="15" HEIGHT="13"> <b>Up to <a href="/cvs/cvsweb.cgi/">Nevrax</a> / <a href="/cvs/cvsweb.cgi/code/">code</a> / <a href="/cvs/cvsweb.cgi/code/nelns/">nelns</a> / <a href="/cvs/cvsweb.cgi/code/nelns/admin_executor_service/">admin_executor_service</a></b></td></tr></table><HR noshade><table width="100%"><tr><td bgcolor="#ffffff">File: <a href="/cvs/cvsweb.cgi/">Nevrax</a> / <a href="/cvs/cvsweb.cgi/code/">code</a> / <a href="/cvs/cvsweb.cgi/code/nelns/">nelns</a> / <a href="/cvs/cvsweb.cgi/code/nelns/admin_executor_service/">admin_executor_service</a> / <a href="/cvs/cvsweb.cgi/code/nelns/admin_executor_service/admin_executor_service.cpp">admin_executor_service.cpp</a> (<A HREF="/cvs/cvsweb.cgi/~checkout~/code/nelns/admin_executor_service/admin_executor_service.cpp?rev=1.9" target="cvs_checkout" onClick="window.open('/cvs/cvsweb.cgi/~checkout~/code/nelns/admin_executor_service/admin_executor_service.cpp?rev=1.9','cvs_checkout','resizeable,scrollbars');"><b>download</b></A>)<BR> +Revision <B>1.9</B>, <i>Mon Jun 18 14:54:06 2001 UTC</i> (13 months ago) by <i>lecroart</i> +<BR>Branch: <b>MAIN</b> +<BR>Changes since <b>1.8: +5 -5 + lines</b><PRE> +BUGFIX: linux compilation +</PRE> +</td></tr></table><HR noshade><PRE>/** \file admin_executor_service.cpp + * Admin Executor Service (AES) + * + * $Id: admin_executor_service.cpp,v 1.9 2001/06/18 14:54:06 lecroart Exp $ + * + */ + +/* Copyright, 2000 Nevrax Ltd. + * + * This file is part of NEVRAX NeL Network Services. + * NEVRAX NeL Network Services 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, or (at your option) + * any later version. + * + * NEVRAX NeL Network Services is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with NEVRAX NeL Network Services; see the file COPYING. If not, write to the + * Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, + * MA 02111-1307, USA. + */ + +#include <fcntl.h> +#include <sys/stat.h> + +#include "nel/misc/types_nl.h" + +#ifdef NL_OS_WINDOWS +#include <direct.h> +#else +#include <unistd.h> +#endif + +#include <string> +#include <list> + +#include "nel/misc/debug.h" +#include "nel/misc/config_file.h" +#include "nel/misc/thread.h" + +#include "nel/net/service.h" +#include "nel/net/net_manager.h" + +#ifdef NL_OS_WINDOWS +#define getcwd _getcwd +#define chdir _chdir +#endif + + +using namespace std; +using namespace NLMISC; +using namespace NLNET; + + + +struct CService +{ + CService(TSockId s) : SockId(s), Id(NextId++), Ready(false) { } + + TSockId SockId; /// connection to the service + uint32 Id; /// uint32 to identify the service + string AliasName; /// alias of the service used in the AES and AS to find him (unique per AES) + string ShortName; /// name of the service in short format ("NS" for example) + string LongName; /// name of the service in long format ("naming_service") + bool Ready; /// true if the service is ready + vector<string> Commands; + +private: + static uint32 NextId; +}; + +uint32 CService::NextId = 1; + +list<CService> Services; +typedef list<CService>::iterator SIT; + +SIT find (TSockId sid) +{ + SIT sit; + for (sit = Services.begin(); sit != Services.end(); sit++) + { + if ((*sit).SockId == sid) break; + } + return sit; +} + +SIT findService (uint32 sid, bool asrt = true) +{ + SIT sit; + for (sit = Services.begin(); sit != Services.end(); sit++) + { + if ((*sit).Id == sid) break; + } + if (asrt) + nlassert (sit != Services.end()); + return sit; +} + + +class CExecuteCommandThread : public IRunnable +{ +public: + string Command, Path; + + CExecuteCommandThread (string command, string path = "") : Command(command), Path(path) { } + + void run () + { + nlinfo ("start executing '%s' in '%s' directory", Command.c_str(), Path.c_str()); + + char oldpath[256]; + if (!Path.empty()) + { + getcwd(oldpath,256); + chdir(Path.c_str()); + } + + system (Command.c_str()); + + if (!Path.empty()) + chdir(oldpath); + + nlinfo ("end executing: %s", Command.c_str()); + } +}; + +class CExecuteServiceThread : public IRunnable +{ +public: + string ServiceAlias, ServiceCommand, ServicePath; + + CExecuteServiceThread (string serviceAlias, string serviceCommand, string servicePath = "") : + ServiceCommand(serviceCommand), ServicePath(servicePath), ServiceAlias(serviceAlias) { } + + void run () + { + nlinfo ("start service '%s' '%s' in '%s' directory", ServiceAlias.c_str(), ServiceCommand.c_str(), ServicePath.c_str()); + + char oldpath[256]; + if (!ServicePath.empty()) + { + getcwd(oldpath,256); + chdir(ServicePath.c_str()); + } + + system (ServiceCommand.c_str()); + + if (!ServicePath.empty()) + chdir(oldpath); + + nlinfo ("end service '%s' '%s' in '%s' directory", ServiceAlias.c_str(), ServiceCommand.c_str(), ServicePath.c_str()); + } +}; + +void executeCommand (string command, bool background) +{ + if (command.empty()) return; + +/* + nlinfo ("start executing: %s", command.c_str()); + if (command[command.size()-1] == '&') + { + command.resize(command.size()-2); + + if (spawnlp (_P_NOWAIT, cmd.c_str(), command.c_str(), NULL) == -1) + { + perror ("ca chie grave!!!!: "); + } + } + else + { + if (spawnlp (_P_WAIT, cmd.c_str(), command.c_str(), NULL) == -1) + { + perror ("ca chie grave!!!!: "); + } + } + nlinfo ("end executing: %s", command.c_str()); +*/ + + command += " >NUL:"; + if (background) + { + IThread *thread = IThread::create (new CExecuteCommandThread (command)); + thread->start (); + } + else + { + CExecuteCommandThread cmdt (command); + cmdt.run (); + } +} + + +/* +// execute without +void executeCommand (string command, TSockId from, CCallbackNetBase &netbase) +{ + if (command.empty()) return; + + #define STDOUT 1 + #define STDERR 2 + int nul, oldstdout, oldstderr; + char *tmpfilename = tmpnam (NULL); + nul = _open(tmpfilename, _O_RDWR | _O_CREAT | _O_TRUNC | _O_TEMPORARY | _O_SHORT_LIVED | _O_EXCL, _S_IREAD | _S_IWRITE); + oldstdout = _dup(STDOUT); + oldstderr = _dup(STDERR); + _dup2(nul, STDOUT); + _dup2(nul, STDERR); + system(command.c_str()); + _dup2(oldstdout, STDOUT); + _dup2(oldstderr, STDERR); + _close(oldstdout); + _close(oldstderr); + + _lseek (nul, 0L, SEEK_SET); + + while (!_eof(nul)) + { + uint8 buffer[10000]; + uint32 nbread = _read (nul, buffer, 10000); + + CMessage msgout (netbase.getSIDA(), "ESCR"); + msgout.serial (nbread); + msgout.serialBuffer (buffer, nbread); + netbase.send (msgout, from); + } + + _close(nul); +*/ +/* + FILE *fp = fopen ("test.txt", "r"); + do + { + char str[1024]; + fgets (str, 1024, fp); + if (feof(fp)) break; + result.push_back (str); + } + while (true); + fclose (fp); +// remove ("test.txt"); +*///} + +//////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// +////////////////// CONNECTION TO THE SERVICES ////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// + +static void cbServiceIdentification (CMessage& msgin, TSockId from, CCallbackNetBase &netbase) +{ + CService *s = (CService*) (uint) from->appId(); + + msgin.serial (s->AliasName, s->ShortName, s->LongName); + msgin.serialCont (s->Commands); + + nlinfo ("*:*:%d is identified to be '%s' '%s' '%s'", s->Id, s->AliasName.c_str(), s->ShortName.c_str(), s->LongName.c_str()); + + // broadcast the message to the admin service + CMessage msgout (CNetManager::getSIDA ("AESAS"), "SID"); + msgout.serial (s->Id, s->AliasName, s->ShortName, s->LongName); + msgout.serialCont (s->Commands); + CNetManager::send ("AESAS", msgout); +} + +static void cbServiceReady (CMessage& msgin, TSockId from, CCallbackNetBase &netbase) +{ + CService *s = (CService*) (uint) from->appId(); + + nlinfo ("*:*:%d is ready", s->Id); + s->Ready = true; + + // broadcast the message to the admin service + CMessage msgout (CNetManager::getSIDA ("AESAS"), "SR"); + msgout.serial (s->Id); + CNetManager::send ("AESAS", msgout); +} + +static void cbLog (CMessage& msgin, TSockId from, CCallbackNetBase &netbase) +{ + // received an answer for a command, give it to the AS + + // broadcast the message to the admin service + CMessage msgout (CNetManager::getSIDA ("AESAS"), "LOG"); + string log; + msgin.serial (log); + msgout.serial (log); + CNetManager::send ("AESAS", msgout); +} + +void serviceConnection (const string &serviceName, TSockId from, void *arg) +{ + Services.push_back (CService (from)); + CService *s = &(Services.back()); + from->setAppId ((uint64)(uint)s); + + nlinfo ("*:*:%d connected", s->Id); + + // broadcast the message to the admin service + CMessage msgout (CNetManager::getSIDA ("AESAS"), "SC"); + msgout.serial (s->Id); + CNetManager::send ("AESAS", msgout); +} + +void serviceDisconnection (const string &serviceName, TSockId from, void *arg) +{ + CService *s = (CService*) (uint) from->appId(); + + nlinfo ("*:*:%d disconnected", s->Id); + + // broadcast the message to the admin service + CMessage msgout (CNetManager::getSIDA ("AESAS"), "SD"); + msgout.serial (s->Id); + CNetManager::send ("AESAS", msgout); + + // remove the service from the list + Services.erase (findService(s->Id)); +} + + +/** Callback Array + */ +TCallbackItem ServicesCallbackArray[] = +{ + { "SID", cbServiceIdentification }, + { "SR", cbServiceReady }, + { "LOG", cbLog }, +}; + +//////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// +////////////////// CONNECTION TO THE AS //////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// + +void errorMessage(string message, TSockId from) +{ + CMessage msgout (CNetManager::getSIDA ("AESAS"), "ERR"); + msgout.serial (message); + CNetManager::send ("AESAS", msgout, from); +} + +static void cbExecuteSystemCommand (CMessage& msgin, TSockId from, CCallbackNetBase &netbase) +{ + string command; + + msgin.serial (command); + + IThread *thread = IThread::create (new CExecuteCommandThread (command)); + thread->start (); +} + +static void cbStartService (CMessage& msgin, TSockId from, CCallbackNetBase &netbase) +{ + string serviceAlias, command, path; + msgin.serial (serviceAlias); + + nlinfo ("Starting the service alias '%s'", serviceAlias.c_str()); + + try + { + path = IService::ConfigFile.getVar(serviceAlias).asString(0); + command = IService::ConfigFile.getVar(serviceAlias).asString(1); + } + catch(EConfigFile &e) + { + nlwarning ("error in serviceAlias '%s' in config file (%s)", serviceAlias.c_str(), e.what()); + return; + } + + // give the service alias to the service to forward it back when it will connected to the aes. + command += " -n"; + command += serviceAlias.c_str(); + + command += " >NUL:"; + + IThread *thread = IThread::create (new CExecuteServiceThread (serviceAlias, command, path)); + thread->start (); +} + +static void cbStopService (CMessage& msgin, TSockId from, CCallbackNetBase &netbase) +{ + uint32 sid; + + msgin.serial (sid); + + nlinfo ("I have to stop service '%s'"); + + SIT sit = findService (sid, false); + if (sit == Services.end()) + { + // don't find the aes, send an error message + errorMessage ("couldn't stop service, aes didn't find the service", from); + return; + } + + CMessage msgout (CNetManager::getSIDA("AES"), "STOPS"); + CNetManager::send ("AES", msgout, (*sit).SockId); +} + +static void cbExecCommand (CMessage& msgin, TSockId from, CCallbackNetBase &netbase) +{ + uint32 sid; + string command; + + msgin.serial (sid); + msgin.serial (command); + + SIT sit = findService (sid, false); + if (sit == Services.end()) + { + // don't find the aes, send an error message + errorMessage ("couldn't stop service, aes didn't find the service", from); + return; + } + + CMessage msgout (CNetManager::getSIDA("AES"), "EXEC_COMMAND"); + msgout.serial (command); + CNetManager::send ("AES", msgout, (*sit).SockId); +} + +void loadAndSendServicesAliasList (CConfigFile::CVar &var); + +void cbASServiceConnection (const string &serviceName, TSockId from, void *arg) +{ + // new admin service, send him all out info about services + + nlinfo ("AS %s is connected", from->asString().c_str()); + + CMessage msgout (CNetManager::getSIDA ("AESAS"), "SL"); + uint32 nbs = (uint32)Services.size(); + msgout.serial (nbs); + for (SIT sit = Services.begin(); sit != Services.end(); sit++) + { + msgout.serial ((*sit).Id, (*sit).AliasName, (*sit).ShortName, (*sit).LongName, (*sit).Ready); + msgout.serialCont ((*sit).Commands); + } + CNetManager::send ("AESAS", msgout, from); + + loadAndSendServicesAliasList (IService::ConfigFile.getVar ("Services")); +} + +TCallbackItem AESASCallbackArray[] = +{ + { "SYS", cbExecuteSystemCommand }, + { "STARTS", cbStartService }, + { "STOPS", cbStopService }, + { "EXEC_COMMAND", cbExecCommand }, +}; + +//////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// +////////////////// SERVICE IMPLEMENTATION ////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// +//////////////////////////////////////////////////////////////////////////////////////////////////////// + +void loadAndSendServicesAliasList (CConfigFile::CVar &services) +{ + vector<string> servicesaliaslist; + for (sint i = 0 ; i < services.size (); i++) + { + servicesaliaslist.push_back (services.asString(i)); + } + + CMessage msgout2 (CNetManager::getSIDA ("AESAS"), "SAL"); + msgout2.serialCont (servicesaliaslist); + CNetManager::send ("AESAS", msgout2, 0); +} + + +class CAdminExecutorService : public IService +{ +public: + + /// Init the service, load the universal time. + void init () + { + CNetManager::setConnectionCallback ("AES", serviceConnection, NULL); + CNetManager::setDisconnectionCallback ("AES", serviceDisconnection, NULL); + + // install the server for AS + CNetManager::setConnectionCallback ("AESAS", cbASServiceConnection, NULL); + CNetManager::addServer ("AESAS", 49996); + CNetManager::addCallbackArray ("AESAS", AESASCallbackArray, sizeof(AESASCallbackArray)/sizeof(AESASCallbackArray[0])); + + ConfigFile.setCallback ("Services", loadAndSendServicesAliasList); + loadAndSendServicesAliasList (IService::ConfigFile.getVar ("Services")); + } + + bool update () + { + return true; + } +}; + + +/// Naming Service +NLNET_SERVICE_MAIN (CAdminExecutorService, "AES", "admin_executor_service", 49997, ServicesCallbackArray); +</PRE>
\ No newline at end of file |