回声服务器和客户端的实现

客户端

#include<stdio.h>

#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#define BUF_SIZE 1024
#define RLT_SIZE 4
#define OPSZ 4


void error_handling(char *message);


int main(int argc, char * argv[])
{
int sock;        //建立套接字 
int result, opnd_cnt, i;  //
char opmsg[BUF_SIZE];    //缓冲区 
struct sockaddr_in serv_adr;   //声明地址结构 
    if(argc != 3)
{
printf("Usage: %s <IP> <port>\n",argv[0]);
exit(1);
}


sock = socket(PF_INET, SOCK_STREAM, 0);  //初始化套接字 
if(sock == -1)
{
error_handling("socket() error\n");
}


memset(&serv_adr, 0, sizeof(serv_adr));  //初始化为0 
    serv_adr.sin_family = AF_INET;   //使用IPv4 
serv_adr.sin_addr.s_addr = inet_addr(argv[1]);  //第一个字节为IP地址 
serv_adr.sin_port = htons(atoi(argv[2]));  //第二个为端口号 


if(connect(sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)   //链接套接字和地址 
error_handling( "connect() error! " );
else
puts("connected.....");
fputs("OPerand count: ", stdout);
scanf("%d", &opnd_cnt);
opmsg[0] = (char)opnd_cnt; //将一个 个数 按照字节模式输入到缓冲区 


for(i = 0; i < opnd_cnt; i++)
{
      printf("operand %d :", i+1);
 scanf("%d", (int *)&opmsg[i*OPSZ+1]);   //向缓冲区写入其他字节 
}


fgetc(stdin);
fputs("operator:", stdout);
scanf("%c", &opmsg[opnd_cnt*OPSZ+1]);
write(sock, opmsg, opnd_cnt*OPSZ+2);   //发送到服务器 
read(sock, &result,RLT_SIZE);  //得到结果 


printf("result %d\n",result);
close(sock);
return 0;
}


void error_handling(char *message)
{
fputs(message,stderr);
fputc('\n',stderr);
exit(1);

}

服务器：

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#define BUF_SIZE 1024
#define OPSZ 4
void error_handling(char * message);   //设置出错函数 
int calculate(int opnum, int opnds[], char oprator);  //计算结果函数 


int main(int argc, char * argv[])
{
int serv_sock, clnt_sock;   //初始化 服务器和客户端套接字 
char opinfo[BUF_SIZE];      //设立缓冲区 
int result, opnd_cnt, i;    // 
int recv_cnt, recv_len;     // 
struct sockaddr_in serv_adr, clnt_adr;  //声明协议 
socklen_t clnt_adr_sz;      //


if(argc != 2){
printf("Usage: %s <port>\n", argv[0]);
exit(1);
}
   
    serv_sock=socket(AF_INET,SOCK_STREAM,0);  //初始化套接字 
memset(&serv_adr, 0, sizeof(serv_adr));   //将结构体清零 
serv_adr.sin_family = AF_INET;            //设定协议族 使用IPv4 
serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);  //接受任意地址 
serv_adr.sin_port = htons(atoi(argv[1]));   //第二个参数为端口号 


if(bind(serv_sock, (struct sockaddr*)&serv_adr, sizeof(serv_adr))== -1)  //绑定地址 
{ 
error_handling("bind() error!");
}


if(listen(serv_sock, 5) == -1)    //监听 队列为5 
error_handling("listen() error!"); 
 
clnt_adr_sz = sizeof(clnt_adr);   


for(i = 0; i < 5; i++)
{
opnd_cnt = 0;     
clnt_sock = accept(serv_sock,(struct sockaddr *)&clnt_adr,&clnt_adr_sz);  //链接成功 产生用于I/O链接的套接字 
read(clnt_sock, &opnd_cnt, 1);   //读取第一个字节  第一个为个数 
 
recv_len = 0;
while((opnd_cnt*OPSZ+1) > recv_len)
{
recv_cnt = read(clnt_sock, &opinfo[recv_len],BUF_SIZE-1); //把剩下的全部信息放在 opinfo中 
recv_len += recv_cnt;
}
result = calculate(opnd_cnt, (int*)opinfo, opinfo[recv_len-1]); // 第一个为 数字个数  数组 符号 
printf("result = %d\n", result);   //输出结果 
write(clnt_sock, (char*)&result, sizeof(result));   //向客户端写结果 
  close(clnt_sock);  //关闭链接的套接字 
} 
close(serv_sock);  //关闭链接客户端的套接字 
return 0;
}


int calculate(int opnum, int opnds[], char op)
{
int result = opnds[0], i;
switch(op)
{
case '+':
for(i = 1; i < opnum; i++)
{
result+= opnds[i];
}
break;
case '-':
for(i = 1; i < opnum; i++)
result-= opnds[i];
break;
case'*':
for(i = 1; i < opnum; i++)
   result*=opnds[i];
break;
//case'/':
// for(i = 1; i < opnum; i++)
// result/=opnum[i];
// break;
}
return result;
}


void error_handling(char *message)
{
fputs(message,stderr);
fputc('\n',stderr);
exit(1);
}