模拟一个简单的基于tcp的远程关机程序

最近在学习unix网络编程，现在正在学习tcp的通信。其实，只要建立起了tcp通信，操作远端的计算机就不是什么问题了。正向telnet一样，也是基于tcp/IP协议的。所以这个实验，也算是对telnet功能的一种简单的模拟。

但是，值得注意的问题是关机涉及到系统权限，所以要给运行在Server端的程序以足够的权限，这样才可以在接收到Client端的关机请求时，执行关机。

将会模拟如下执行过程：

1． 执行 mytelnet 跟上参数telnet服务器 IP地址 127.0.0.1

2． 输入login 向服务器请求登录，随之服务器会要求输入密码

3． 输入一个错误的登录密码 123 

4． 服务器验证不通过，所以返回Fail to login ,please check your password

5． 再次输入密码，这次输入正确的密码123456

6． 服务器验证通过，此时客户端可以执行基于telnet的远程操控

7． 输入操控命令，例如关机命令shutdown 

8． 服务器将会执行shutdown 操作。

在编译时使用如下的命令：

Server:

sudo gcc server.c -o server //足够的权限

sudo chmod u+s server 

./server

Client:

gcc client.c -o mytelnet

./mytelnet 10.3.1.210

Client端：

#include < stdio.h >
#include < string .h >
#include < arpa / inet.h >
#include < netinet / in .h >
#include < sys / socket.h >
#define MAXLINE 4096
void clientWork(FILE * fp , int sockfd)
{
char sendline[MAXLINE],receiveline[MAXLINE];
int n;
while (fgets(sendline,MAXLINE,fp) != NULL)
{
write(sockfd,sendline,strlen(sendline));
n = read(sockfd,receiveline,MAXLINE);
receiveline[n] = ' \0 ' ;
fputs(receiveline,stdout);
if (strcmp(receiveline, " bye\n " ) == 0 ) break ;
}
}
int main( int argc , char * * argv )
{
int sockfd , n ;
char recvline[ MAXLINE + 1 ];
struct sockaddr_in servaddr;
if ( ( sockfd = socket( AF_INET , SOCK_STREAM , 0 ) ) < 0 ) {
printf( " socket error\n " );
exit( 1 );
}
memset( & servaddr , 0 , sizeof ( servaddr ) );
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons( 2000 ); // 指定Server端的端口号
char * serverAddress = argv[ 1 ];
// 判断指定的ip地址是否有错误
if ( inet_pton( AF_INET ,serverAddress , & servaddr.sin_addr ) <= 0 ) {
printf( " inet_pton error for %s\n " , serverAddress );
exit( 1 );
}
if ( connect( sockfd , ( struct sockaddr * ) & servaddr , sizeof ( servaddr ) ) < 0 ) {
printf( " connect error\n " );
exit( 1 );
}
// 从Terminal中读取用户输入的指令
clientWork(stdin,sockfd);
close(sockfd);
return 0 ;
}

Server端：

#include < stdio.h >
#include < string .h >
#include < arpa / inet.h >
#include < netinet / in .h >
#include < sys / socket.h >
#define MAXSIZE 1024
#define LOGIN 1 // 登录Server
#define BYE 2 // 离开Server
#define PASS 3 // 合法登录
#define DENY 4 // 非法登录
#define SHUTDOWN 5 // 执行关机
#define ERROR -1 // 非法信息
// 记录状态信息
static int STATE = 0 ;
int getClientChoice( char * clientmsg)
{
if (strcmp(clientmsg, " login\n " ) == 0 ) return LOGIN;
if (STATE > 0 )
{
if (strcmp(clientmsg, " bye\n " ) == 0 ) return BYE;
}
if (STATE == LOGIN)
{
// Default password is 123456
if (strcmp(clientmsg, " 123456\n " ) == 0 ) return PASS;
else return DENY;
}
if (STATE == PASS)
{
if (strcmp(clientmsg, " shutdown\n " ) == 0 ) return SHUTDOWN;
else ERROR;
}
return ERROR;
};
// 发送消息
void sendMsg( int sockfd, char * buffer, char * msg)
{
char buffer2[MAXSIZE];
snprintf(buffer2, sizeof (buffer2), " %s " ,msg);
write(sockfd,buffer2, sizeof (buffer2));
}
// 定义关机函数
void myshutdown()
{
// 系统在一分钟后关机
system( " shutdown -t 1 " );
}
int main( int argc , char * * argv )
{
char buffer[MAXSIZE];
int listenfd = socket(AF_INET,SOCK_STREAM, 0 ); // 定义socket，指向ipv4地址的字节流套接口

struct sockaddr_in serverAddr;
memset( & serverAddr, 0 , sizeof (serverAddr)); // sockAddr_in 进行初始化
serverAddr.sin_family = AF_INET;
serverAddr.sin_addr.s_addr = htonl(INADDR_ANY);
serverAddr.sin_port = htons( 2000 );
if (bind(listenfd,( struct sockaddr * ) & serverAddr, sizeof (serverAddr)) ==- 1 )
{
printf( " There is an error during binding\n " );
return - 1 ;
}
else
{
printf( " Bind successfully!!!\n " );
}
// 对listenfd进行监听，从最初建立时的主动套接口（用于进行connect的套接口）转化为被动套接口（接受连接）
listen(listenfd, 100 ); // 第二个参数为套接口排队的最大连接个数
int connectfd;
socklen_t addrlen;
struct sockaddr_in connectAddr;
memset( & connectAddr, 0 , sizeof (connectAddr));
printf( " Be ready to accept a connection!\n " );
while ( 1 )
{
connectfd = accept(listenfd,( struct sockAddr * ) & connectAddr, & addrlen); // 接受client端一个请求的socket
char receivebuffer[MAXSIZE];
int revbuflen;
while ( 1 )
{
revbuflen = read(connectfd,receivebuffer,MAXSIZE);
receivebuffer[revbuflen] = ' \0 ' ;
// printf("%s",receivebuffer);
int clientChoice = getClientChoice(receivebuffer);
// 模拟操作过程
if (clientChoice == LOGIN)
{
STATE = LOGIN;
sendMsg(connectfd,buffer, " Please input your password :\n " );
}
else if (clientChoice == PASS && STATE == LOGIN)
{
STATE = PASS;
sendMsg(connectfd,buffer, " Welcome to my Telnet Server...\n " );
}
else if (clientChoice == SHUTDOWN && STATE == PASS)
{
myshutdown();
sendMsg(connectfd,buffer, " Remote computer is going to shutdown...\n " );
}
else if (clientChoice == BYE)
{
sendMsg(connectfd,buffer, " bye\n " );
break ;
}
else if (clientChoice == DENY)
{
sendMsg(connectfd,buffer, " Fail to login ,please check your password\n " );
}
else if (clientChoice == ERROR)
{
sendMsg(connectfd,buffer, " wrong,Check your input...\n " );
}
}
close(connectfd);
}
close(listenfd); // 虽然因为上面有while（true），这行永远都执行不了，但是时刻注意关闭socket连接应该是个好习惯。
return 0 ;
}