这是一个简单工具,用于远程执行在服务器上的脚本或进程。 原理是在服务器上将socket重定向到所运行的进程的标准输入输出流,这样进程的标准输出(STDOUT)会自动发送到客户端,客户端写到socket上的数据自动输出到进程的标准输入(STDIN),这样就实现了远程交互,类似telnet服务. 代码如下: //句柄复制 from qmail tcpserver int fd_copy(int to,int from) { if (to == from) return 0; if (fcntl(from,F_GETFL,0) == -1) return -1; close(to); if (fcntl(from,F_DUPFD,to) == -1) return -1; return 0; } //句柄复制,并关闭源句柄 from from qmail tcpserver int fd_move(int to,int from) { if (to == from) return 0; if (fd_copy(to,from) == -1) return -1; close(from); return 0; } //处理客户端连接的主循环 void clientloop(CONNECT_INFO *pConn) { int pid ; int stat = 0; int res = 0; char cmd[100]=""; char Prompt[]="->"; //定义提示符 int i; SendData(pConn->socket,Prompt,strlen(Prompt)); //输出提示符 while(1) //主循环 { memset(cmd,0,sizeof(cmd)); if(RecvData(pConn->socket,cmd,sizeof(cmd),1000 )>0) //从socket上获取指令(主进程) { chomp(cmd); //去掉回车换行,TAB,空格 printf("Get cmd:[%s]\n",cmd); //在服务器端打印出收到的命令 if(strcmp(cmd,"quit")==0) //收到quit则退出 { break; } pid = fork(); //创建子进程 if(pid < 0) //错误 { perror("fork error"); break; } if(pid > 0) //主进程 { waitpid(pid,&stat,WCONTINUED); //等待子进程退出 kill(pid,9); //再kill一次防止僵死进程 SendData(pConn->socket,Prompt,strlen(Prompt)); //发送提示符号 }else{ fd_copy(0,pConn->socket); //将socket重定向到标准输入 fd_move(1,pConn->socket); //将标准输出重定向到socket printf("[%d]run cmd %s\n",getpid(),cmd); //打印当前命令 res = execl(cmd,"",(char*)0); //使用execl运行指令,子进程被指令创建的进程取代,该进程的输入输出已经被重定向 if (errno == ENOEXEC) //这是处理某些没有指定默认shell的脚本 execl ("/bin/sh", "sh", "-c", cmd, (char *)0); else printf("run res = %d,%s\n",res,strerror(errno)); //错误输出到客户端 exit(0); }//pid }//recvdata }//while }
//主程序入口 main(){ /*建立监听socket,获得客户端连接,对新连接fork进程,调用clientloop*/ }
目前这个代码还比较粗糙,能实现基本的运行和交互,最需要优化的主要是execl的调用方法,这样的写法只能输入完整路径名,而且不能带参数。可以 1:根据当前的PATH环境变量,实现类似whereis功能, 2:对输入的命令行进行解析,获取命令参数,构造成char **字符数组传递给execl. 还有一种方式是用popen,用popen的效果也满不错,比起用execl来,避免了以上1,2问题,还可以对输出的信息先进行处理再发给客户端 int runcmdpipe(const char *sCmd) { char pline [1024]; FILE *fp = NULL; if ((fp = popen (sCmd, "r")) == NULL) return -1; while (fgets (pline, sizeof(pline)-1, fp)!= NULL) { printf(">> %s",pline); //输出到客户端 } pclose (fp); return ncount; }
客户端可以用telnet程序,但unix和windows的telnet客户端有小小差异,unix的telnet输入回车才发送,而windows的telnet一击键即刻发送,所以程序应该判断是否得到回车换行,才认为是一个完整的指令。 还有一些unix命令如su,top,vi在这种模式下运行不太顺畅,但此工具已经满足要求了,以后有空再研究如何解决。 
|