X-Git-Url: http://code.vuplus.com/gitweb/?a=blobdiff_plain;f=lib%2Fbase%2Fconsole.cpp;h=6dc21d2ecf99be30a191b3e16d6a7b6c501546e5;hb=654bb0a0f8583b7cbc47b32f7e9b6921dffc4e03;hp=75c6162f2b86c52b82e5d1f9c70baf59fc1c2bf6;hpb=94b67b79cf83ec356b05b31a51c3849ee8953e6a;p=vuplus_dvbapp diff --git a/lib/base/console.cpp b/lib/base/console.cpp index 75c6162..6dc21d2 100644 --- a/lib/base/console.cpp +++ b/lib/base/console.cpp @@ -5,8 +5,11 @@ #include #include #include +#include +#include +#include -int bidirpipe(int pfd[], char *cmd , char *argv[]) +int bidirpipe(int pfd[], const char *cmd , const char * const argv[], const char *cwd ) { int pfdin[2]; /* from child to parent */ int pfdout[2]; /* from parent to child */ @@ -16,7 +19,7 @@ int bidirpipe(int pfd[], char *cmd , char *argv[]) if ( pipe(pfdin) == -1 || pipe(pfdout) == -1 || pipe(pfderr) == -1) return(-1); - if ( ( pid = fork() ) == -1 ) + if ( ( pid = vfork() ) == -1 ) return(-1); else if (pid == 0) /* child process */ { @@ -35,7 +38,11 @@ int bidirpipe(int pfd[], char *cmd , char *argv[]) for (unsigned int i=3; i < 90; ++i ) close(i); - execvp(cmd,argv); + if (cwd) + chdir(cwd); + + execvp(cmd, (char * const *)argv); + /* the vfork will actually suspend the parent thread until execvp is called. thus it's ok to use the shared arg/cmdline pointers here. */ _exit(0); } if (close(pfdout[0]) == -1 || close(pfdin[1]) == -1 || close(pfderr[1]) == -1) @@ -52,165 +59,166 @@ eConsoleAppContainer::eConsoleAppContainer() :pid(-1), killstate(0), in(0), out(0), err(0) { for (int i=0; i < 3; ++i) + { fd[i]=-1; + filefd[i]=-1; + } } -int eConsoleAppContainer::execute( const std::string &cmd ) +static char brakets[][2] = { + { '\'','\'' }, + {'"','"'}, + {'`','`'}, + {'(',')'}, + {'{','}'}, + {'[',']'}, + {'<','>'} +}; + +static char *find_bracket(char ch) { - if (running()) - return -1; - pid=-1; - killstate=0; -// eDebug("cmd = %s", cmd.c_str() ); - int cnt=2; // path to app + terminated 0 - std::string str(cmd.length()?cmd:""); + size_t idx=0; + while (idx < sizeof(brakets)/2) { + if (brakets[idx][0] == ch) + return &brakets[idx][0]; + ++idx; + } + return NULL; +} - // kill spaces at beginning - unsigned int pos = str.find_first_not_of(' '); - if (pos != std::string::npos && pos) - str = str.substr(pos); +int eConsoleAppContainer::setCWD( const char *path ) +{ + struct stat dir_stat; - // kill spaces at the end - pos = str.find_last_not_of(' '); - if (pos != std::string::npos && (pos+1) < str.length()) - str = str.erase(pos+1); + if (stat(path, &dir_stat) == -1) + return -1; - unsigned int slen=str.length(); - if (!slen) + if (!S_ISDIR(dir_stat.st_mode)) return -2; - std::map brackets; - brackets.insert(std::pair('\'','\'')); - brackets.insert(std::pair('"','"')); - brackets.insert(std::pair('`','`')); - brackets.insert(std::pair('(',')')); - brackets.insert(std::pair('{','}')); - brackets.insert(std::pair('[',']')); - brackets.insert(std::pair('<','>')); - - unsigned int idx=str.find(' '); - std::string path = str.substr(0, idx != std::string::npos ? idx : slen ); -// eDebug("path = %s", path.c_str() ); - unsigned int plen = path.length(); - - std::string cmds = slen > plen ? str.substr( plen+1 ) : ""; - unsigned int clen = cmds.length(); -// eDebug("cmds = %s", cmds.c_str() ); - - idx = 0; - std::map::iterator it = brackets.find(cmds[idx]); - while ( (idx = cmds.find(' ',idx) ) != std::string::npos ) // count args - { - if (it != brackets.end()) - { - if (cmds[idx-1] == it->second) - it = brackets.end(); - } - if (it == brackets.end()) - { - cnt++; - it = brackets.find(cmds[idx+1]); - } - idx++; - } + m_cwd = path; + return 0; +} -// eDebug("idx = %d, %d counted spaces", idx, cnt-2); +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); - if ( clen ) - { - cnt++; -// eDebug("increase cnt"); +// printf("cmd = %s, len %d\n", cmd, slen); + + // kill spaces at beginning + while(path[0] == ' ') { + ++path; + ++cmds; + --slen; } -// eDebug("%d args", cnt-2); - char **argv = new char*[cnt]; // min two args... path and terminating 0 -// eDebug("%d args", cnt); - argv[0] = new char[ plen+1 ]; -// eDebug("new argv[0] %d bytes (%s)", plen+1, path.c_str()); - strcpy( argv[0], path.c_str() ); - argv[cnt-1] = 0; // set terminating null + // kill spaces at the end + while(slen && path[slen-1] == ' ') { + path[slen-1] = 0; + --slen; + } - if ( cnt > 2 ) // more then default args? - { - cnt=1; // do not overwrite path in argv[0] + if (!slen) + return -2; - it = brackets.find(cmds[0]); - idx=0; - while ( (idx = cmds.find(' ',idx)) != std::string::npos ) // parse all args.. - { - bool bracketClosed=false; - if ( it != brackets.end() ) - { - if (cmds[idx-1]==it->second) - { - it = brackets.end(); - bracketClosed=true; - } + 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 == brackets.end() ) - { - std::string tmp = cmds.substr(0, idx); - if (bracketClosed) - { - tmp.erase(0,1); - tmp.erase(tmp.length()-1, 1); - bracketClosed=false; - } -// eDebug("new argv[%d] %d bytes (%s)", cnt, tmp.length()+1, tmp.c_str()); - argv[cnt] = new char[ tmp.length()+1 ]; -// eDebug("idx=%d, arg = %s", idx, tmp.c_str() ); - strcpy( argv[cnt++], tmp.c_str() ); - cmds.erase(0, idx+1); -// eDebug("str = %s", cmds.c_str() ); - it = brackets.find(cmds[0]); - idx=0; + if (!it) { // end of arg + *tmp = 0; + argv[cnt++] = argb; + argb=0; // reset arg begin } - else - idx++; - } - if ( it != brackets.end() ) - { - cmds.erase(0,1); - cmds.erase(cmds.length()-1, 1); + cmds = tmp+1; + while (*cmds && *cmds == ' ') + ++cmds; } - // store the last arg -// eDebug("new argv[%d] %d bytes (%s)", cnt, cmds.length()+1, cmds.c_str()); - argv[cnt] = new char[ cmds.length()+1 ]; - strcpy( argv[cnt], cmds.c_str() ); + argv[cnt++] = argb ? argb : cmds; + if (it) + *argv[cnt-1] = it[0]; // set old char for open braket } - else - cnt=1; - - // get one read ,one write and the err pipe to the prog.. // int tmp=0; // while(argv[tmp]) // eDebug("%d is %s", tmp, argv[tmp++]); - - pid = bidirpipe(fd, argv[0], argv); + return execute(argv[0], argv); +} + +int eConsoleAppContainer::execute(const char *cmdline, const char * const argv[]) +{ + if (running()) + return -1; + + pid=-1; + killstate=0; + + // get one read ,one write and the err pipe to the prog.. + pid = bidirpipe(fd, cmdline, argv, m_cwd.length() ? m_cwd.c_str() : 0); - while ( cnt >= 0 ) // release heap memory - { -// eDebug("delete argv[%d]", cnt); - delete [] argv[cnt--]; - } -// eDebug("delete argv"); - delete [] argv; - if ( pid == -1 ) return -3; // eDebug("pipe in = %d, out = %d, err = %d", fd[0], fd[1], fd[2]); + ::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 ); CONNECT(in->activated, eConsoleAppContainer::readyRead); CONNECT(out->activated, eConsoleAppContainer::readyWrite); CONNECT(err->activated, eConsoleAppContainer::readyErrRead); + 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(); @@ -222,7 +230,11 @@ void eConsoleAppContainer::kill() { eDebug("user kill(SIGKILL) console App"); killstate=-1; - ::kill(pid, SIGKILL); + /* + * Use a negative pid value, to signal the whole process group + * ('pid' might not even be running anymore at this point) + */ + ::kill(-pid, SIGKILL); closePipes(); } while( outbuf.size() ) // cleanup out buffer @@ -235,6 +247,12 @@ void eConsoleAppContainer::kill() delete out; delete err; in=out=err=0; + + for (int i=0; i < 3; ++i) + { + if ( filefd[i] > 0 ) + close(filefd[i]); + } } void eConsoleAppContainer::sendCtrlC() @@ -242,7 +260,11 @@ void eConsoleAppContainer::sendCtrlC() if ( killstate != -1 && pid != -1 ) { eDebug("user send SIGINT(Ctrl-C) to console App"); - ::kill(pid, SIGINT); + /* + * Use a negative pid value, to signal the whole process group + * ('pid' might not even be running anymore at this point) + */ + ::kill(-pid, SIGINT); } } @@ -281,24 +303,42 @@ void eConsoleAppContainer::closePipes() void eConsoleAppContainer::readyRead(int what) { + bool hungup = what & eSocketNotifier::Hungup; if (what & (eSocketNotifier::Priority|eSocketNotifier::Read)) { // eDebug("what = %d"); - char buf[2048]; + char buf[2049]; int rd; - while((rd = read(fd[0], buf, 2047)) > 0) + while((rd = read(fd[0], buf, 2048)) > 0) { -/* for ( int i = 0; i < rd; i++ ) - eDebug("%d = %c (%02x)", i, buf[i], buf[i] );*/ buf[rd]=0; /*emit*/ dataAvail(buf); + stdoutAvail(buf); + if ( filefd[1] > 0 ) + ::write(filefd[1], buf, rd); + if (!hungup) + break; } } - if (what & eSocketNotifier::Hungup) + readyErrRead(eSocketNotifier::Priority|eSocketNotifier::Read); /* be sure to flush all data which might be already written */ + if (hungup) { eDebug("child has terminated"); closePipes(); - /*emit*/ appClosed(killstate); + int childstatus; + int retval = killstate; + /* + * We have to call 'wait' on the child process, in order to avoid zombies. + * Also, this gives us the chance to provide better exit status info to appClosed. + */ + if (::waitpid(pid, &childstatus, 0) > 0) + { + if (WIFEXITED(childstatus)) + { + retval = WEXITSTATUS(childstatus); + } + } + /*emit*/ appClosed(retval); } } @@ -307,44 +347,94 @@ void eConsoleAppContainer::readyErrRead(int what) if (what & (eSocketNotifier::Priority|eSocketNotifier::Read)) { // eDebug("what = %d"); - char buf[2048]; + char buf[2049]; int rd; - while((rd = read(fd[2], buf, 2047)) > 0) + while((rd = read(fd[2], buf, 2048)) > 0) { /* for ( int i = 0; i < rd; i++ ) eDebug("%d = %c (%02x)", i, buf[i], buf[i] );*/ buf[rd]=0; /*emit*/ dataAvail(buf); + stderrAvail(buf); } } } +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]; memcpy(tmp, data, len); outbuf.push(queue_data(tmp,len)); - out->start(); + if (out) + out->start(); +} + +void eConsoleAppContainer::write( PyObject *data ) +{ + char *buffer; + int 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() ) { - queue_data d = outbuf.front(); - outbuf.pop(); - if ( ::write( fd[1], d.data, d.len ) != d.len ) - { - /* emit */ dataSent(-1); -// eDebug("writeError"); - } + queue_data &d = outbuf.front(); + int wr = ::write( fd[1], d.data+d.dataSent, d.len-d.dataSent ); + if (wr < 0) + eDebug("eConsoleAppContainer write failed (%m)"); else + d.dataSent += wr; + if (d.dataSent == d.len) { + outbuf.pop(); + delete [] d.data; + if ( filefd[0] == -1 ) /* emit */ dataSent(0); -// eDebug("write ok"); } - delete [] d.data; } if ( !outbuf.size() ) - out->stop(); + { + if ( filefd[0] > 0 ) + { + char readbuf[32*1024]; + int rsize = read(filefd[0], readbuf, 32*1024); + if ( rsize > 0 ) + write(readbuf, rsize); + else + { + close(filefd[0]); + filefd[0] = -1; + ::close(fd[1]); + eDebug("readFromFile done - closing eConsoleAppContainer stdin pipe"); + fd[1]=-1; + dataSent(0); + out->stop(); + } + } + else + out->stop(); + } }