lib/base/console.cpp

   1 /*
   2  * console.cpp
   3  *
   4  * Copyright (C) 2002 Felix Domke <tmbinc@tuxbox.org>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  19  *
  20  * $Id: console.cpp,v 1.1 2003-10-17 15:35:47 tmbinc Exp $
  21  */
  22
  23 #include <lib/base/console.h>
  24
  25 #include <lib/base/estring.h>
  26 #include <sys/vfs.h> // for statfs
  27 #include <unistd.h>
  28 #include <signal.h>
  29 #include <errno.h>
  30
  31 int bidirpipe(int pfd[], char *cmd , char *argv[])
  32 {
  33         int pfdin[2];  /* from child to parent */
  34         int pfdout[2]; /* from parent to child */
  35         int pfderr[2]; /* stderr from child to parent */
  36         int pid;       /* child's pid */
  37
  38         if ( pipe(pfdin) == -1 || pipe(pfdout) == -1 || pipe(pfderr) == -1)
  39                 return(-1);
  40
  41         if ( ( pid = fork() ) == -1 )
  42                 return(-1);
  43         else if (pid == 0) /* child process */
  44         {
  45                 if ( close(0) == -1 || close(1) == -1 || close(2) == -1 )
  46                         _exit(0);
  47
  48                 if (dup(pfdout[0]) != 0 || dup(pfdin[1]) != 1 || dup(pfderr[1]) != 2 )
  49                         _exit(0);
  50
  51                 if (close(pfdout[0]) == -1 || close(pfdout[1]) == -1 ||
  52                                 close(pfdin[0]) == -1 || close(pfdin[1]) == -1 ||
  53                                 close(pfderr[0]) == -1 || close(pfderr[1]) == -1 )
  54                         _exit(0);
  55
  56                 execv(cmd,argv);
  57                 _exit(0);
  58         }
  59         if (close(pfdout[0]) == -1 || close(pfdin[1]) == -1 || close(pfderr[1]) == -1)
  60                         return(-1);
  61
  62         pfd[0] = pfdin[0];
  63         pfd[1] = pfdout[1];
  64         pfd[2] = pfderr[0];
  65
  66         return(pid);
  67 }
  68
  69 eConsoleAppContainer::eConsoleAppContainer( const eString &cmd )
  70 :pid(-1), killstate(0), outbuf(0)
  71 {
  72 //      eDebug("cmd = %s", cmd.c_str() );
  73         memset(fd, 0, sizeof(fd) );
  74         int cnt=2; // path to app + terminated 0
  75         eString str(cmd?cmd:"");
  76
  77         while( str.length() && str[0] == ' ' )  // kill spaces at beginning
  78                 str = str.mid(1);
  79
  80         while( str.length() && str[str.length()-1] == ' ' )  // kill spaces at the end
  81                 str = str.left( str.length() - 1 );
  82
  83         if (!str.length())
  84                 return;
  85
  86         unsigned int idx=0;
  87         eString path = str.left( (idx = str.find(' ')) != eString::npos ? idx : str.length() );
  88 //      eDebug("path = %s", path.c_str() );
  89
  90         eString cmds = str.mid( path.length()+1 );
  91 //      eDebug("cmds = %s", cmds.c_str() );
  92
  93         idx = 0;
  94         while ( (idx = cmds.find(' ',idx) ) != eString::npos )  // count args
  95         {
  96                 cnt++;
  97                 idx++;
  98         }
  99
 100 //      eDebug("idx = %d, %d counted spaces", idx, cnt-2);
 101
 102         if ( cmds.length() )
 103         {
 104                 cnt++;
 105 //              eDebug("increase cnt");
 106         }
 107
 108 //      eDebug("%d args", cnt-2);
 109         char **argv = new char*[cnt];  // min two args... path and terminating 0
 110         argv[0] = new char[ path.length() ];
 111         strcpy( argv[0], path.c_str() );
 112         argv[cnt-1] = 0;               // set terminating null
 113
 114         if ( cnt > 2 )  // more then default args?
 115         {
 116                 cnt=1;  // do not overwrite path in argv[0]
 117
 118                 while ( (idx = cmds.find(' ')) != eString::npos )  // parse all args..
 119                 {
 120                         argv[cnt] = new char[ idx ];
 121 //                      eDebug("idx=%d, arg = %s", idx, cmds.left(idx).c_str() );
 122                         strcpy( argv[cnt++], cmds.left( idx ).c_str() );
 123                         cmds = cmds.mid(idx+1);
 124 //                      eDebug("str = %s", cmds.c_str() );
 125                 }
 126                 // store the last arg
 127                 argv[cnt] = new char[ cmds.length() ];
 128                 strcpy( argv[cnt], cmds.c_str() );
 129         }
 130
 131   // get one read ,one write and the err pipe to the prog..
 132
 133         if ( (pid = bidirpipe(fd, argv[0], argv)) == -1 )
 134         {
 135                 while ( cnt-- > 0 )
 136                         delete [] argv[cnt];
 137                 delete [] argv;
 138                 return;
 139         }
 140
 141         while ( cnt-- > 0 )  // release heap memory
 142                 delete [] argv[cnt];
 143         delete [] argv;
 144
 145         eDebug("pipe in = %d, out = %d, err = %d", fd[0], fd[1], fd[2]);
 146
 147         in = new eSocketNotifier(eApp, fd[0], 19 );  // 19 = POLLIN, POLLPRI, POLLHUP
 148         out = new eSocketNotifier(eApp, fd[1], eSocketNotifier::Write);  // POLLOUT
 149         err = new eSocketNotifier(eApp, fd[2], 19 );  // 19 = POLLIN, POLLPRI, POLLHUP
 150         CONNECT(in->activated, eConsoleAppContainer::readyRead);
 151         CONNECT(out->activated, eConsoleAppContainer::readyWrite);
 152         CONNECT(err->activated, eConsoleAppContainer::readyErrRead);
 153         signal(SIGCHLD, SIG_IGN);   // no zombie when child killed
 154 }
 155
 156 eConsoleAppContainer::~eConsoleAppContainer()
 157 {
 158         if ( running() )
 159         {
 160                 killstate=-1;
 161                 kill();
 162         }
 163         if ( outbuf )
 164                 delete [] outbuf;
 165 }
 166
 167 void eConsoleAppContainer::kill()
 168 {
 169         killstate=-1;
 170         system( eString().sprintf("kill %d", pid).c_str() );
 171         eDebug("user kill console App");
 172 }
 173
 174 void eConsoleAppContainer::closePipes()
 175 {
 176         in->stop();
 177         out->stop();
 178         err->stop();
 179         ::close(fd[0]);
 180         fd[0]=0;
 181         ::close(fd[1]);
 182         fd[1]=0;
 183         ::close(fd[2]);
 184         fd[2]=0;
 185         eDebug("pipes closed");
 186 }
 187
 188 void eConsoleAppContainer::readyRead(int what)
 189 {
 190         if (what & POLLPRI|POLLIN)
 191         {
 192                 eDebug("what = %d");
 193                 char buf[2048];
 194                 int readed = read(fd[0], buf, 2048);
 195                 eDebug("%d bytes read", readed);
 196                 if ( readed != -1 && readed )
 197                         /*emit*/ dataAvail( eString( buf ) );
 198                 else if (readed == -1)
 199                         eDebug("readerror %d", errno);
 200         }
 201         if (what & eSocketNotifier::Hungup)
 202         {
 203                 eDebug("child has terminated");
 204                 closePipes();
 205                 /*emit*/ appClosed(killstate);
 206         }
 207 }
 208
 209 void eConsoleAppContainer::readyErrRead(int what)
 210 {
 211         if (what & POLLPRI|POLLIN)
 212         {
 213                 eDebug("what = %d");
 214                 char buf[2048];
 215                 int readed = read(fd[2], buf, 2048);
 216                 eDebug("%d bytes read", readed);
 217                 if ( readed != -1 && readed )
 218                         /*emit*/ dataAvail( eString( buf ) );
 219                 else if (readed == -1)
 220                         eDebug("readerror %d", errno);
 221         }
 222 }
 223
 224 void eConsoleAppContainer::write( const eString & str )
 225 {
 226         outbuf = new char[ str.length()];
 227         strcpy( outbuf, str.c_str() );
 228 }
 229
 230 void eConsoleAppContainer::readyWrite(int what)
 231 {
 232         if (what == 4 && outbuf)
 233         {
 234                 if ( ::write( fd[1], outbuf, strlen(outbuf) ) != (int) strlen(outbuf) )
 235                 {
 236                         /* emit */ dataSent(-1);
 237                         eDebug("writeError");
 238                 }
 239                 else
 240                 {
 241                         /* emit */ dataSent(0);
 242                         eDebug("write ok");
 243                 }
 244
 245                 delete outbuf;
 246                 outbuf=0;
 247         }
 248 }