better solution to add possibility to delete eSocketNotifiers,
[vuplus_dvbapp] / lib / base / console.cpp
index d8820ee..a12cb5e 100644 (file)
@@ -55,8 +55,10 @@ int bidirpipe(int pfd[], const char *cmd , const char * const argv[], const char
        return(pid);
 }
 
+DEFINE_REF(eConsoleAppContainer);
+
 eConsoleAppContainer::eConsoleAppContainer()
-:pid(-1), killstate(0), in(0), out(0), err(0)
+:pid(-1), killstate(0)
 {
        for (int i=0; i < 3; ++i)
        {
@@ -65,27 +67,6 @@ eConsoleAppContainer::eConsoleAppContainer()
        }
 }
 
-static char brakets[][2] = {
-       { '\'','\'' },
-       {'"','"'},
-       {'`','`'},
-       {'(',')'},
-       {'{','}'},
-       {'[',']'},
-       {'<','>'}
-};
-
-static char *find_bracket(char ch)
-{
-       size_t idx=0;
-       while (idx < sizeof(brakets)/2) {
-               if (brakets[idx][0] == ch)
-                       return &brakets[idx][0];
-               ++idx;
-       }
-       return NULL;
-}
-
 int eConsoleAppContainer::setCWD( const char *path )
 {
        struct stat dir_stat;
@@ -102,70 +83,13 @@ int eConsoleAppContainer::setCWD( const char *path )
 
 int eConsoleAppContainer::execute( const char *cmd )
 {
-       int cnt=0, slen=strlen(cmd);
-       char buf[slen+1];
-       char *tmp=0, *argv[64], *path=buf, *cmds = buf;
-       memcpy(buf, cmd, slen+1);
-
-//     printf("cmd = %s, len %d\n", cmd, slen);
-
-       // kill spaces at beginning
-       while(path[0] == ' ') {
-               ++path;
-               ++cmds;
-               --slen;
-       }
-
-       // kill spaces at the end
-       while(slen && path[slen-1] == ' ') {
-               path[slen-1] = 0;
-               --slen;
-       }
-
-       if (!slen)
-               return -2;
-
-       tmp = strchr(path, ' ');
-       if (tmp) {
-               *tmp = 0;
-               cmds = tmp+1;
-               while(*cmds && *cmds == ' ')
-                       ++cmds;
-       }
-       else
-               cmds = path+slen;
-
-       memset(argv, 0, sizeof(argv));
-       argv[cnt++] = path;
-
-       if (*cmds) {
-               char *argb=NULL, *it=NULL;
-               while ( (tmp = strchr(cmds, ' ')) ) {
-                       if (!it && *cmds && (it = find_bracket(*cmds)) )
-                               *cmds = 'X'; // replace open braket...
-                       if (!argb) // not arg begin
-                               argb = cmds;
-                       if (it && *(tmp-1) == it[1]) {
-                               *argb = it[0]; // set old char for open braket
-                               it = 0;
-                       }
-                       if (!it) { // end of arg
-                               *tmp = 0;
-                               argv[cnt++] = argb;
-                               argb=0; // reset arg begin
-                       }
-                       cmds = tmp+1;
-                       while (*cmds && *cmds == ' ')
-                               ++cmds;
-               }
-               argv[cnt++] = argb ? argb : cmds;
-               if (it)
-                   *argv[cnt-1] = it[0]; // set old char for open braket
-       }
+       int argc = 3;
+       const char *argv[argc + 1];
+       argv[0] = "/bin/sh";
+       argv[1] = "-c";
+       argv[2] = cmd;
+       argv[argc] = NULL;
 
-//     int tmp=0;
-//     while(argv[tmp])
-//             eDebug("%d is %s", tmp, argv[tmp++]);
        return execute(argv[0], argv);
 }
 
@@ -187,38 +111,19 @@ int eConsoleAppContainer::execute(const char *cmdline, const char * const argv[]
 
        ::fcntl(fd[1], F_SETFL, O_NONBLOCK);
        ::fcntl(fd[2], F_SETFL, O_NONBLOCK);
-       in = new eSocketNotifier(eApp, fd[0], eSocketNotifier::Read|eSocketNotifier::Priority|eSocketNotifier::Hungup );
-       out = new eSocketNotifier(eApp, fd[1], eSocketNotifier::Write, false);  
-       err = new eSocketNotifier(eApp, fd[2], eSocketNotifier::Read|eSocketNotifier::Priority );
+       in = eSocketNotifier::create(eApp, fd[0], eSocketNotifier::Read|eSocketNotifier::Priority|eSocketNotifier::Hungup );
+       out = eSocketNotifier::create(eApp, fd[1], eSocketNotifier::Write, false);  
+       err = eSocketNotifier::create(eApp, fd[2], eSocketNotifier::Read|eSocketNotifier::Priority );
        CONNECT(in->activated, eConsoleAppContainer::readyRead);
        CONNECT(out->activated, eConsoleAppContainer::readyWrite);
        CONNECT(err->activated, eConsoleAppContainer::readyErrRead);
+       in->m_clients.push_back(this);
+       out->m_clients.push_back(this);
+       err->m_clients.push_back(this);
 
        return 0;
 }
 
-int eConsoleAppContainer::execute( PyObject *cmdline, PyObject *args )
-{
-       if (!PyString_Check(cmdline))
-               return -1;
-       if (!PyList_Check(args))
-               return -1;
-       const char *argv[PyList_Size(args) + 1];
-       int i;
-       for (i = 0; i < PyList_Size(args); ++i)
-       {
-               PyObject *arg = PyList_GetItem(args, i); /* borrowed ref */
-               if (!arg)
-                       return -1;
-               if (!PyString_Check(arg))
-                       return -1;
-               argv[i] = PyString_AsString(arg); /* borrowed pointer */
-       }
-       argv[i] = 0;
-
-       return execute(PyString_AsString(cmdline), argv); /* borrowed pointer */
-}
-
 eConsoleAppContainer::~eConsoleAppContainer()
 {
        kill();
@@ -243,10 +148,9 @@ void eConsoleAppContainer::kill()
                outbuf.pop();
                delete [] d.data;
        }
-       delete in;
-       delete out;
-       delete err;
-       in=out=err=0;
+       in = 0;
+       out = 0;
+       err = 0;
 
        for (int i=0; i < 3; ++i)
        {
@@ -309,6 +213,7 @@ void eConsoleAppContainer::closePipes()
                outbuf.pop();
                delete [] d.data;
        }
+       in = 0; out = 0; err = 0;
        pid = -1;
 }
 
@@ -371,24 +276,6 @@ void eConsoleAppContainer::readyErrRead(int what)
        }
 }
 
-void eConsoleAppContainer::dumpToFile( PyObject *py_filename )
-{
-       char *filename = PyString_AsString(py_filename);
-       filefd[1] = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0644);
-       eDebug("eConsoleAppContainer::dumpToFile open(%s, O_WRONLY|O_CREAT|O_TRUNC, 0644)=%i", filename, filefd[1]);
-}
-
-void eConsoleAppContainer::readFromFile( PyObject *py_filename )
-{
-       char *filename = PyString_AsString(py_filename);
-       char readbuf[32*1024];
-       filefd[0] = open(filename, O_RDONLY);
-       int rsize = read(filefd[0], readbuf, 32*1024);
-       eDebug("eConsoleAppContainer::readFromFile open(%s, O_RDONLY)=%i, read: %i", filename, filefd[0], rsize);
-       if ( filefd[0] > 0 && rsize > 0 )
-               write(readbuf, rsize);
-}
-
 void eConsoleAppContainer::write( const char *data, int len )
 {
        char *tmp = new char[len];
@@ -398,16 +285,6 @@ void eConsoleAppContainer::write( const char *data, int len )
                out->start();
 }
 
-void eConsoleAppContainer::write( PyObject *data )
-{
-       char *buffer;
-       Py_ssize_t length;
-       if (PyString_AsStringAndSize(data, &buffer, &length))
-               return;
-       if (buffer && length)
-               write(buffer, length);
-}
-
 void eConsoleAppContainer::readyWrite(int what)
 {
        if (what&eSocketNotifier::Write && outbuf.size() )
@@ -449,3 +326,371 @@ void eConsoleAppContainer::readyWrite(int what)
                        out->stop();
        }
 }
+
+#include "structmember.h"
+
+extern "C" {
+
+struct eConsolePy
+{
+       PyObject_HEAD
+       eConsoleAppContainer *cont;
+       PyObject *in_weakreflist; /* List of weak references */
+};
+
+static PyObject *
+eConsolePy_dataAvail(eConsolePy *self, void *closure)
+{
+       return self->cont->dataAvail.get();
+}
+
+static PyObject *
+eConsolePy_stdoutAvail(eConsolePy *self, void *closure)
+{
+       return self->cont->stdoutAvail.get();
+}
+
+static PyObject *
+eConsolePy_stderrAvail(eConsolePy *self, void *closure)
+{
+       return self->cont->stderrAvail.get();
+}
+
+static PyObject *
+eConsolePy_dataSent(eConsolePy *self, void *closure)
+{
+       return self->cont->dataSent.get();
+}
+
+static PyObject *
+eConsolePy_appClosed(eConsolePy *self, void *closure)
+{
+       return self->cont->appClosed.get();
+}
+
+static PyGetSetDef eConsolePy_getseters[] = {
+       {"dataAvail",
+        (getter)eConsolePy_dataAvail, (setter)0,
+        "dataAvail callback list",
+        NULL},
+       {"stdoutAvail",
+        (getter)eConsolePy_stdoutAvail, (setter)0,
+        "stdoutAvail callback list",
+        NULL},
+       {"stderrAvail",
+        (getter)eConsolePy_stderrAvail, (setter)0,
+        "stderrAvail callback list",
+        NULL},
+       {"dataSent",
+        (getter)eConsolePy_dataSent, (setter)0,
+        "dataSent callback list",
+        NULL},
+       {"appClosed",
+        (getter)eConsolePy_appClosed, (setter)0,
+        "appClosed callback list",
+        NULL},
+       {NULL} /* Sentinel */
+};
+
+static int
+eConsolePy_traverse(eConsolePy *self, visitproc visit, void *arg)
+{
+       PyObject *obj = self->cont->dataAvail.getSteal();
+       if (obj) {
+               Py_VISIT(obj);
+       }
+       obj = self->cont->stdoutAvail.getSteal();
+       if (obj) {
+               Py_VISIT(obj);
+       }
+       obj = self->cont->stderrAvail.getSteal();
+       if (obj) {
+               Py_VISIT(obj);
+       }
+       obj = self->cont->dataSent.getSteal();
+       if (obj) {
+               Py_VISIT(obj);
+       }
+       obj = self->cont->appClosed.getSteal();
+       if (obj) {
+               Py_VISIT(obj);
+       }
+       return 0;
+}
+
+static int
+eConsolePy_clear(eConsolePy *self)
+{
+       PyObject *obj = self->cont->dataAvail.getSteal(true);
+       if (obj) {
+               Py_CLEAR(obj);
+       }
+       obj = self->cont->stdoutAvail.getSteal(true);
+       if (obj) {
+               Py_CLEAR(obj);
+       }
+       obj = self->cont->stderrAvail.getSteal(true);
+       if (obj) {
+               Py_CLEAR(obj);
+       }
+       obj = self->cont->dataSent.getSteal(true);
+       if (obj) {
+               Py_CLEAR(obj);
+       }
+       obj = self->cont->appClosed.getSteal(true);
+       if (obj) {
+               Py_CLEAR(obj);
+       }
+       return 0;
+}
+
+static void
+eConsolePy_dealloc(eConsolePy* self)
+{
+       if (self->in_weakreflist != NULL)
+               PyObject_ClearWeakRefs((PyObject *) self);
+       eConsolePy_clear(self);
+       self->cont->Release();
+       self->ob_type->tp_free((PyObject*)self);
+}
+
+static PyObject *
+eConsolePy_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
+{
+       eConsolePy *self = (eConsolePy *)type->tp_alloc(type, 0);
+       self->cont = new eConsoleAppContainer();
+       self->cont->AddRef();
+       self->in_weakreflist = NULL;
+       return (PyObject *)self;
+}
+
+static PyObject *
+eConsolePy_running(eConsolePy* self)
+{
+       PyObject *ret = NULL;
+       ret = self->cont->running() ? Py_True : Py_False;
+       Org_Py_INCREF(ret);
+       return ret;
+}
+
+static PyObject *
+eConsolePy_execute(eConsolePy* self, PyObject *argt)
+{
+       const char *str;
+       if (PyArg_ParseTuple(argt, "s", &str))
+               return PyInt_FromLong(self->cont->execute(str));
+       PyErr_SetString(PyExc_TypeError,
+               "argument is not a string");
+       return NULL;
+}
+
+static PyObject *
+eConsolePy_write(eConsolePy* self, PyObject *args)
+{
+       int len;
+       char *data;
+       if (PyArg_ParseTuple(args, "si", &data, &len))
+               ;
+       else
+       {
+               PyObject *ob;
+               if (!PyArg_ParseTuple(args, "O", &ob) || !PyString_Check(ob))
+               {
+                       PyErr_SetString(PyExc_TypeError,
+                               "1st arg must be a string, optionaly 2nd arg can be the string length");
+                       return NULL;
+               }
+               else
+               {
+                       Py_ssize_t length;
+                       if (!PyString_AsStringAndSize(ob, &data, &length))
+                               len = length;
+                       else
+                               len = 0;
+               }
+       }
+       self->cont->write(data, len);
+       Py_RETURN_NONE;
+}
+
+static PyObject *
+eConsolePy_getPID(eConsolePy* self)
+{
+       return PyInt_FromLong(self->cont->getPID());
+}
+
+static PyObject *
+eConsolePy_setCWD(eConsolePy* self, PyObject *args)
+{
+       const char *path=0;
+       if (!PyArg_ParseTuple(args, "s", &path))
+               return NULL;
+       self->cont->setCWD(path);
+       Py_RETURN_NONE;
+}
+
+static PyObject *
+eConsolePy_kill(eConsolePy* self)
+{
+       self->cont->kill();
+       Py_RETURN_NONE;
+}
+
+static PyObject *
+eConsolePy_sendCtrlC(eConsolePy* self)
+{
+       self->cont->sendCtrlC();
+       Py_RETURN_NONE;
+}
+
+static PyObject *
+eConsolePy_sendEOF(eConsolePy* self)
+{
+       self->cont->sendEOF();
+       Py_RETURN_NONE;
+}
+
+static PyObject *
+eConsolePy_dumpToFile(eConsolePy* self, PyObject *args)
+{
+       char *filename;
+       if (!PyArg_ParseTuple(args, "s", &filename))
+       {
+               PyErr_SetString(PyExc_TypeError,
+                       "arg must be a string (filename)");
+               return NULL;
+       }
+       else
+       {
+               int fd = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0644);
+               self->cont->setFileFD(1, fd);
+               eDebug("eConsoleAppContainer::dumpToFile open(%s, O_WRONLY|O_CREAT|O_TRUNC, 0644)=%d", filename, fd);
+       }
+       Py_RETURN_NONE;
+}
+
+static PyObject *
+eConsolePy_readFromFile(eConsolePy* self, PyObject *args)
+{
+       char *filename;
+       if (!PyArg_ParseTuple(args, "s", &filename))
+       {
+               PyErr_SetString(PyExc_TypeError,
+                       "arg must be a string (filename)");
+               return NULL;
+       }
+       else
+       {
+               int fd = open(filename, O_RDONLY);
+               if (fd >= 0)
+               {
+                       char readbuf[32*1024];
+                       int rsize = read(fd, readbuf, 32*1024);
+                       self->cont->setFileFD(0, fd);
+                       eDebug("eConsoleAppContainer::readFromFile open(%s, O_RDONLY)=%d, read: %d", filename, fd, rsize);
+                       self->cont->write(readbuf, rsize);
+               }
+               else
+               {
+                       eDebug("eConsoleAppContainer::readFromFile %s not exist!", filename);
+                       self->cont->setFileFD(0, -1);
+               }
+       }
+       Py_RETURN_NONE;
+}
+
+static PyMethodDef eConsolePy_methods[] = {
+       {"setCWD", (PyCFunction)eConsolePy_setCWD, METH_VARARGS,
+        "set working dir"
+       },
+       {"execute", (PyCFunction)eConsolePy_execute, METH_VARARGS,
+        "execute command"
+       },
+       {"dumpToFile", (PyCFunction)eConsolePy_dumpToFile, METH_VARARGS,
+        "set output file"
+       },
+       {"readFromFile", (PyCFunction)eConsolePy_readFromFile, METH_VARARGS,
+        "set input file"
+       },
+       {"getPID", (PyCFunction)eConsolePy_getPID, METH_NOARGS,
+        "execute command"
+       },
+       {"kill", (PyCFunction)eConsolePy_kill, METH_NOARGS,
+        "kill application"
+       },
+       {"sendCtrlC", (PyCFunction)eConsolePy_sendCtrlC, METH_NOARGS,
+        "send Ctrl-C to application"
+       },
+       {"sendEOF", (PyCFunction)eConsolePy_sendEOF, METH_NOARGS,
+        "send EOF to application"
+       },
+       {"write", (PyCFunction)eConsolePy_write, METH_VARARGS,
+        "write data to application"
+       },
+       {"running", (PyCFunction)eConsolePy_running, METH_NOARGS,
+        "returns the running state"
+       },
+       {NULL}  /* Sentinel */
+};
+
+static PyTypeObject eConsolePyType = {
+       PyObject_HEAD_INIT(NULL)
+       0, /*ob_size*/
+       "eConsoleImpl.eConsoleAppContainer", /*tp_name*/
+       sizeof(eConsolePy), /*tp_basicsize*/
+       0, /*tp_itemsize*/
+       (destructor)eConsolePy_dealloc, /*tp_dealloc*/
+       0, /*tp_print*/
+       0, /*tp_getattr*/
+       0, /*tp_setattr*/
+       0, /*tp_compare*/
+       0, /*tp_repr*/
+       0, /*tp_as_number*/
+       0, /*tp_as_sequence*/
+       0, /*tp_as_mapping*/
+       0, /*tp_hash */
+       0, /*tp_call*/
+       0, /*tp_str*/
+       0, /*tp_getattro*/
+       0, /*tp_setattro*/
+       0, /*tp_as_buffer*/
+       Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /*tp_flags*/
+       "eConsoleAppContainer objects", /* tp_doc */
+       (traverseproc)eConsolePy_traverse, /* tp_traverse */
+       (inquiry)eConsolePy_clear, /* tp_clear */
+       0, /* tp_richcompare */
+       offsetof(eConsolePy, in_weakreflist), /* tp_weaklistoffset */
+       0, /* tp_iter */
+       0, /* tp_iternext */
+       eConsolePy_methods, /* tp_methods */
+       0, /* tp_members */
+       eConsolePy_getseters, /* tp_getset */
+       0, /* tp_base */
+       0, /* tp_dict */
+       0, /* tp_descr_get */
+       0, /* tp_descr_set */
+       0, /* tp_dictoffset */
+       0, /* tp_init */
+       0, /* tp_alloc */
+       eConsolePy_new, /* tp_new */
+};
+
+static PyMethodDef module_methods[] = {
+       {NULL}  /* Sentinel */
+};
+
+void eConsoleInit(void)
+{
+       PyObject* m = Py_InitModule3("eConsoleImpl", module_methods,
+               "Module that implements eConsoleAppContainer with working cyclic garbage collection.");
+
+       if (m == NULL)
+               return;
+
+       if (!PyType_Ready(&eConsolePyType))
+       {
+               Org_Py_INCREF((PyObject*)&eConsolePyType);
+               PyModule_AddObject(m, "eConsoleAppContainer", (PyObject*)&eConsolePyType);
+       }
+}
+}