TCP/IP网络编程——计算器服务器/客户端

op_server.c

#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;
  int clnt_sock;
  
  struct sockaddr_in serv_addr;
  struct sockaddr_in clnt_addr;
  socklen_t clnt_addr_size;
  int result, opnd_cnt;
  int recv_cnt, recv_len;
  char opinfo[BUF_SIZE];
  
  if(argc!=2)
  {
    exit(1);
  }
  
  //TCP socket
  //1.安装电话机
  serv_sock=socket(PF_INET, SOCK_STREAM, 0);
  if(serv_sock == -1)
    error_handling("socket error!");
  //2.调用bind分配电话号码
  memset(&serv_addr, 0, sizeof(serv_addr));
  serv_addr.sin_family = AF_INET;	//IPV4协议族
  //INADDR_ANY表示自动获取服务器端IP地址，
  //如果计算机只有一个NIC（网络接口卡），则可以这么配置，如果有多个，则需要手动配置
  //服务器IP地址的数量与NIC数量有关
  serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);	//主机字节序（host）转换成网络字节序（net）（大端序）
  serv_addr.sin_port = htons(atoi(argv[1]));	//端口号
  //表示请把进入IP...、9190端口的数据传给我
  if(bind(serv_sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1)
    error_handling("bind error");
  //3.调用listen连接电话线（进入等待连接请求状态）
  //5表示连接请求等待队列为5
  if(listen(serv_sock, 5) == -1)
    error_handling("listen error");
  //4.调用accept拿起话筒（受理连接请求）
  clnt_addr_size = sizeof(clnt_addr);
  
  int i;
  for(i=0;i<5;i++)
  {
    opnd_cnt=0;
    clnt_sock=accept(serv_sock, (struct sockaddr*)&clnt_addr, &clnt_addr_size);
    //首先接受一个字符为待计算数的数量
    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);
      recv_len+=recv_cnt;
    }
    //计算结果并把结果发出去
    //直接把opinfo转成了int型
    result=calculate(opnd_cnt, (int*)opinfo, opinfo[recv_len-1]);
    write(clnt_sock, (char*)&result, sizeof(result));
    close(clnt_sock);
  }
  
  close(serv_sock);
  
  return 0;
}

int calculate(int opnum, int opnds[], char oprator)
{
  int result=opnds[0];
  int i;
  switch(oprator)
  {
    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;
  }
  return result;
}

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

op_client.c

#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;
  struct sockaddr_in serv_addr;
  char opmsg[BUF_SIZE];
  int result;
  
  if(argc!=3)
  {
    exit(1);
  }
  
  sock=socket(PF_INET, SOCK_STREAM, 0);
  if(sock == -1)
    error_handling("socket error!");
  
  memset(&serv_addr, 0, sizeof(serv_addr));
  serv_addr.sin_family = AF_INET;
  //基于字符串的IP地址初始化
  //也可以使用 int inet_aton(const char* string, struct in_addr* addr);
  //反向转换 char* inet_ntoa(struct in_addr adr);
  serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
  serv_addr.sin_port = htons(atoi(argv[2]));
  //向服务器发起连接请求
  //注意：客户端IP地址在调用connect时会自动分配
  if(connect(sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1)
    error_handling("connect error");
  else
    puts("Connected.....");
  
  //输入待算数的个数
  unsigned int opnd_cnt;
  fputs("Operand count: ", stdout);
  scanf("%d", &opnd_cnt);
  opmsg[0] = (char)opnd_cnt;
  //输入待计算的数
  int i;
  for(i=0;i<opnd_cnt;i++)
  {
    printf("Operand %d: ", i+1);
    //4字节int型数据要保存到char数组
    scanf("%d", (int *) &opmsg[ i*OPSZ + 1 ]);
  }
  fgetc(stdin);	//调用fgetc删掉缓冲中的\n
  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);
}