网络版shell之网络编程练习篇--telnet服务端

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include "my_type.h" //外部头文件 申明了一个函数体 exec_cmd()
int main(void)
{
char buf[2048];//为了保持程序简短，便于阅读，省去了错误判断。
int socket_fd;
struct sockaddr_in netaddr;
unsigned short int port=8000;
socket_fd=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
bzero(buf,2048);
bzero(&netaddr,sizeof(struct sockaddr_in));//为了保持程序简短，便于阅读，省去了错误判断。
netaddr.sin_family=AF_INET;
netaddr.sin_port=htons(port);
netaddr.sin_addr.s_addr=htonl(INADDR_ANY);
bind(socket_fd,(struct sockaddr *)&netaddr,sizeof(struct sockaddr));//在实际代码中，已加入了错误判断
listen(socket_fd,3);
while(1)
{
int connfd;
int i=0;
char ch=0;
char os_chose[4];
int len=sizeof(struct sockaddr);
struct sockaddr_in cliaddr;
char buf_in[2048];
bzero(os_chose,4);
bzero(&cliaddr,sizeof(struct sockaddr_in));
printf("port = %d \n",ntohs(netaddr.sin_port));
start:
connfd=accept(socket_fd,(struct sockaddr *)&cliaddr,&len);//监听端口 等待客户端连接
printf(" accept is over, remote ip = \n ");
while(1)
{
strcpy(buf," chose your system : \r\n");
write(connfd,buf,30);
strcpy(buf,"\r [1] windons \n");
write(connfd,buf,60);
strcpy(buf,"\r [2] linux \n");
write(connfd,buf,60);
read(connfd,os_chose,4);

/*window 与linux 有一些小区别，linux中，换行会自动回车，

而window只会换行而不会自动回车，所以如果对于从linux中获取的命令输出，

不做处理的话，在window终端下，会呈现梯形显示 */
if( strpbrk(os_chose,"1")!=NULL )
{
ch='1';
printf("window \n");
break;
}
else if( strpbrk(os_chose,"2")!=NULL )
{
ch='2';
printf(" linux \n");
break;
}
// window的telnet程序，按每个字符发送，而首字符往往不是输入的字符，可能是\0,没有去深究，

直接捕获输入的前4个字符，

然后检查是否1 或者2 来判断登陆者选择的系统类型。
}
strcpy(buf,"\r\ntelnet_shell>>");
write(connfd,buf,20);   //输入提示符
bzero(&buf,2048);
while(1)
{
char tmp_ch=0;
if(connfd>=0)
{
if(ch=='2')//linux的telnet客户端，以回车符作为命令的结束，与window不同，

要想去别对待，这样服务端获取的就是一个字符串
{
bzero(&buf,2048);
read(connfd,buf,2048);
if( exec_cmd(buf,connfd)==10 )//my_type.h  申明的函数，主要解释输入的字符串命令，执行命令后，将标准输出和错误输出都重定向到网络中。
goto start;
bzero(&buf,2048);
strcpy(buf,"\rtelnet_shell>>");
write(connfd,buf,20);
bzero(&buf,2048);
}
if(ch=='1')
{ // window的telnet客户端，按单个字符发送命令，所以要对输入的单个字符重新组合。
read(connfd,&tmp_ch,1);
if(tmp_ch=='\0'&&i==0)
{}
else if(tmp_ch!='\n')
buf_in[i++]=tmp_ch;
else if(tmp_ch=='\n'&&i!=0)
{
printf("%s \n",buf_in);
buf_in[i]=' ';
buf_in[i+1]=' ';
buf_in[i+2]='\n';
if( exec_cmd(buf_in,connfd)==10 )
goto start;
bzero(&buf_in,2048);
strcpy(buf_in,"\rtelnet_shell>>");
write(connfd,buf_in,20);
bzero(&buf_in,2048);
i=0;
}
}
}
}
}
return 0;
}

#include "my_type.h"
int exec_cmd(char *p,int fd)
{
char *argv[5];
char cmd_data[5][10];
char tmp_p[200];
char *token;
char *p_tmp;
pid_t pid;
int id=0,i=0,j=0;
int pipe_fd[2];
char buf_cmd[2048];
memset(buf_cmd,'\0',2048);
if( (pipe(pipe_fd))==-1 )
{
perror("pipe init error");
return -1;
}
memset(tmp_p,'\0',200);
memset(cmd_data,'\0',10*5);
memset(argv,'\0',5);
strcpy(tmp_p,p);
p_tmp=tmp_p;
while( (*p_tmp)!='\n' )
{
if( ((*p_tmp)=='\r') )
{
*p_tmp=' ';
*(p_tmp+1)='\r';
*(p_tmp+2)='\n';
break;
}
++p_tmp;
}    //以空格符为字符串分割符有一个缺点，就是会把字符串的末尾符号\r\n也认为是字符串，而在exec时，无法正确执行命令，所以不管字符串末尾是否有空格，都统一在\r\n前加入空格符。
token=strtok(tmp_p," ");
while(token!=NULL)
{
if( (*token!=' ')&&(*token!='\n')&&(*token!='\0') )
strcpy(cmd_data[id++],token);
token=strtok(NULL," ");
}
cmd_data[id-1][strlen(cmd_data[id-1])-1]='\0';
for(j=0;j<id-1;j++)
{
argv[j]=cmd_data[j];
}
argv[j]=(char *)NULL;
if( strcmp(argv[0],"cd")==0)
{
puts("cmd is cd ");
chdir(argv[1]);
return 0 ;
}
else if( strcmp(argv[0],"exit")==0 )
{
puts("cmd is exit ");
puts("bye bye ! \n");
close(fd);
return 10;
}
if( (pid=fork())==-1)
{
perror("fork error");
exit(1);
}
if( pid==0 )
{
close(pipe_fd[0]);//通过无名管道对stdou、stderr输出重定向
dup2(pipe_fd[1],fileno(stderr));
dup2(pipe_fd[1],fileno(stdout));
execvp(argv[0],argv);
close(pipe_fd[1]);
exit(0);
}
else
{
char *buf_p=0,buf_ch=0;
int buf_i=0;
char tmp[2048];
close(pipe_fd[1]);
memset(buf_cmd,'\0',2048);
memset(tmp,'\0',2048);//读取子进程写入到管道中的信息
read(pipe_fd[0],buf_cmd,2048);
buf_p=buf_cmd;
puts(buf_cmd);//linux和window都统一起来，虽然linux对于换行符的处理是换行和自动回车，这里为了方便起见，都统一加入回车符。这样就可以在window下完美显示了。
while( buf_cmd[buf_i]!='\0')
{
tmp[buf_i]=buf_cmd[buf_i++];
if( buf_cmd[buf_i]=='\n' )
{
tmp[buf_i-1]='\r';
tmp[buf_i++]='\n';
}
}
write(fd,tmp,2048);
waitpid(pid,NULL,NULL);
return 0;
}

}

#ifndef __my_type__
#define __my_type__
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/wait.h>
extern int exec_cmd(char *p,int fd);
#endif