128 lines
3.0 KiB
Plaintext
128 lines
3.0 KiB
Plaintext
#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);
|
|
}
|