1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
#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);
}
|
