libtdevnc/vncterm/VNCommand.c

#ifdef __STRICT_ANSI__
#define _BSD_SOURCE
#define _POSIX_SOURCE
#endif
#include "VNConsole.h"
#include "vga.h"
#ifdef LIBVNCSERVER_HAVE_FCNTL_H
#include <fcntl.h>
#endif
#ifdef LIBVNCSERVER_HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
#ifdef LIBVNCSERVER_HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef LIBVNCSERVER_HAVE_SYS_WAIT_H
#include <sys/wait.h>
#endif
#include <errno.h>


int main(int argc, char **argv)
{
  rfbBool interactive=FALSE,sendOnlyWholeLines=TRUE;
  int serverArgc,programArg0;
  for(serverArgc=1;serverArgc<argc
	&& argv[serverArgc][0]=='-' && argv[serverArgc][1]!='-';serverArgc++)
    if(!strcmp(argv[serverArgc],"-interactive")) {
      interactive=TRUE;
      sendOnlyWholeLines=FALSE;

      rfbPurgeArguments(&argc,&serverArgc,1,argv);
    }
  programArg0=serverArgc;
  if(programArg0<argc && argv[programArg0][0]=='-' && argv[programArg0][1]=='-')
    programArg0++;
  argv[argc]=0;

  if(programArg0<argc) {
    int in[2],out[2],err[2],pid;
    if(pipe(in)<0 || pipe(out)<0 || pipe(err)<0) {
      rfbErr("Couldn't make pipes!");
      return(1);
    }

    pid=fork();
    if(!pid) {
      dup2(in[0],0);
      dup2(out[1],1);
      dup2(err[1],2);
      /*setbuf(stdin,NULL);*/
      execvp(argv[programArg0],argv+programArg0);
    }

    {
      char buffer[1024];
      fd_set fs,fs1/*,ifs,ifs1*/;
      struct timeval tv,tv1;
      int i,c=1,num_fds,max_fd=out[0],status;
      FILE *input_pipe;
      vncConsolePtr console=vcGetConsole(&serverArgc,argv,80,25,&vgaFont,FALSE);
      if(interactive)
	console->doEcho = FALSE;

      if(max_fd<err[0])
	max_fd=err[0];
      FD_ZERO(&fs);
      FD_SET(out[0],&fs);
      FD_SET(err[0],&fs);
      /*FD_SET(0,&fs);*/
      tv.tv_sec=0; tv.tv_usec=5000;

      input_pipe=fdopen(in[1],"w");
      setbuf(input_pipe,NULL);
      while(c || waitpid(pid,&status,WNOHANG)==0) {
	/* event loop */
	vcProcessEvents(console);

	/* get input */
	if(console->inputCount) {
	  if(sendOnlyWholeLines) {
	    for(i=0;i<console->inputCount;i++)
	      if(console->inputBuffer[i]=='\n') {
		i++;
		fwrite(console->inputBuffer,i,1,input_pipe);
		fflush(input_pipe);
		/* fwrite(console->inputBuffer,i,1,stderr); */
		if(console->inputCount>i)
		  memmove(console->inputBuffer,console->inputBuffer+i,console->inputCount-i);
		console->inputCount-=i;
		i=0;
	      }
	  } else {
	    fwrite(console->inputBuffer,console->inputCount,1,input_pipe);
	    fflush(input_pipe);
	    /* fwrite(console->inputBuffer,console->inputCount,1,stderr); */
	    console->inputCount=0;
	  }
	}
	/* process output */
	fs1=fs; tv1=tv;
	num_fds=select(max_fd+1,&fs1,NULL,NULL,&tv1);
	if(num_fds>0) {
	  /*
	  if(FD_ISSET(0,&fs1)) {
	    ch=getchar();
	    fputc(ch,f);
	  }
	  */
	  if(FD_ISSET(out[0],&fs1)) {
	    c=read(out[0],buffer,1024);
	    for(i=0;i<c;i++)
	      vcPutChar(console,buffer[i]);
	  }
	  if(FD_ISSET(err[0],&fs1)) {
	    c=read(err[0],buffer,1024);
	    for(i=0;i<c;i++)
	      vcPutChar(console,buffer[i]);
	  }		
	} else
	  c=0;
      }
    }
  }
  rfbLog("exit\n");
  return(0);
}