论坛徽章:: 0

电梯直达

1楼 [收藏(0)] [报告]

发表于 2008-05-07 00:05 |只看该作者 |倒序浏览

根据高人《在UNIX系统上记录进程输入/输出的方法》（注：见[url]http://bbs.chinaunix.net/thread-1090836-1-3.html[/url]）

在UNIX环境下用C语言和伪终端方法，实现同一操作的多屏幕输出（相当于现场转播功能）

命令用法：

vtp 空闲的伪终端号转播终端1 转播终端2 ...

例：vtp ttyp4 tty1a tty1b tty1c tty1d ... (其中ttyp4必须是空闲和有效的)

源程序如下：
/code

/* vtp.c */

#include "stdio.h"
#include "stdlib.h"
#include "unistd.h"
#include "fcntl.h"
#include "signal.h"
#include "tinfo.h"
#include "termio.h"
#include "sys/stream.h"

#define BUFSIZE 4096

int console_fd,pty_fd;
struct termio t_attr;
char pttyname[15]="/dev/pty";

/* 退出时,恢复原来的终端属性 */
void exitproc(int i){
ioctl(console_fd,TCSETAF,&t_attr);
exit(0);
}

void HELP(char *str){
printf("\n用法: %s 伪终端号 [转播终端1 转播终端2...]\n\n", str);
exit(0);
}

/* 设置终端的低层特性 */
static void set_console_raw(int fd){
struct termios buf;
if(tcgetattr(fd,&buf)<0) exit(-1);
buf.c_lflag&=~(ECHO|ICANON|IEXTEN|ISIG);
buf.c_iflag&=~(BRKINT|ICRNL|INPCK|ISTRIP|IXON);
buf.c_cflag&=~(CSIZE|PARENB);
buf.c_cflag|=CS8;
buf.c_oflag&=~(OPOST);
buf.c_cc[VMIN]=1;
buf.c_cc[VTIME]=0;
if(tcsetattr(fd,TCSAFLUSH,&buf)<0) exit(-1);
}

main(int argc,char *argv[]){
int fd_in,fd_out,nread,slave_fd,i;
int MAX,fd[50]; /* 允许的最大"转播"终端数为50个 */
char str[100];
fd_set rset,rrset;
static unsigned char buf[BUFSIZE];

if(argc<2) HELP(argv[0]);
/* 判断第一个参数是不是伪终端号 */
if(strncmp("ttyp",argv[1],4)!=0) HELP(argv[0]);
strcat(pttyname,argv[1]+3);
for(MAX=0;MAX<argc-2;MAX++){ /* 打开所有要转播的终端 */
sprintf(str,"/dev/%s",argv[MAX+2]);
fd[MAX]=open(str,O_WRONLY);
}
signal(SIGTERM,exitproc);
signal(SIGKILL,exitproc);
signal(SIGCLD,exitproc);
console_fd=open(ttyname(0),O_RDWR);
ioctl(console_fd,TCGETA,&t_attr);
set_console_raw(console_fd);
pty_fd=open(pttyname,O_RDWR);

/* 打开用于保存输入输出信息用的文件 */
fd_in=open("./vtp.in",O_CREAT|O_RDWR|O_TRUNC,0660);
fd_out=open("./vtp.out",O_CREAT|O_RDWR|O_TRUNC,0660);

switch(fork()){
case -1: exit(1);
case 0:
setsid();
/* 让子进程以指定的伪终端号运行 */
sprintf(str,"/dev/%s",argv[1]);
slave_fd=open(str,O_RDWR);
if(slave_fd==-1){
printf("打开%s出错\n\n",pttyname);
exit(-1);
}
ioctl(slave_fd,TCSETAF,&t_attr);
dup2(slave_fd,0);
dup2(slave_fd,1);
dup2(slave_fd,2);
close(slave_fd);
execlp("/bin/sh",NULL);
}
FD_ZERO(&rrset);
FD_SET(console_fd,&rrset);
FD_SET(pty_fd,&rrset);
while(1){
rset=rrset;
if(select(FD_SETSIZE,&rset,0,0,0)==-1) continue;
/* 通过伪终端,向子进程转递输入信息,并保存 */
if(FD_ISSET(console_fd,&rset)){
nread=read(console_fd,buf,BUFSIZE);
buf[nread]='\0';
if(nread>0){
write(pty_fd,buf,strlen(buf));
write(fd_in,buf,nread);
}
}
/* 向伪终端读取输出信息,并保存 */
if(FD_ISSET(pty_fd,&rset)){
nread=read(pty_fd,buf,BUFSIZE);
/* 向终端转播输出信息 */
for(i=0;i<MAX;i++) write(fd[i],buf,nread);
buf[nread]='\0';
if(nread>0){
write(console_fd,buf,strlen(buf));
write(fd_out,buf,nread);
}
}
}
}

/code

转播时子进程以伪终端号运行，并在当前目录下保存输入和输出信息文件 vtp.in vtp.out

要回放时只需cat vtp.out即可。

如果喜欢请回贴一下。谢谢。

[[i] 本帖最后由 melove 于 2008-5-7 00:11 编辑 [/i]]