【TCP/IP】简易代数计算服务器/客户端的实现

今天有点失眠 T T，用这些时间整理了下之前写的一个demo。

功能是通过客户端发出欲计算的数据，然后在服务器端计算完成后返回给客户端，最后结果在客户端接收到后输出出来。

代码参见

CalClient:

#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <memory.h>
#include <stdlib.h>
#include <stdio.h>

#define BUF_SIZE 1024
#define BYTEUNIT 4
#define RLT_SIZE 4

void error_handling(char *message)
{
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}
int main()
{
    int opnd_cnt;
    char opmsg[BUF_SIZE];

    struct sockaddr_in serv_addr;

    int clnt_sock = socket(PF_INET, SOCK_STREAM, 0);

    if (clnt_sock == -1)
    {
        error_handling((char*)"socket() error");
    }

    memset(&serv_addr, 0, sizeof(serv_addr));

    serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(9190);

    if (connect(clnt_sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) == -1)
    {
        error_handling( (char*)"connect() error!");
    }
    else
    {
        puts("Connected....");
    }

    fputs("欲计算的数字数量:", stdout);
    scanf("%d", &opnd_cnt);
    opmsg[0] = (char)opnd_cnt;

    for (int i = 0; i < opnd_cnt; i++)
    {
        printf("操作数 %d: ", i + 1);
        //每个整数型占用4个字节
        scanf("%d", (int *)&opmsg[i * BYTEUNIT + 1]);
    }
    fgetc(stdin);
    fputs("操作符: ", stdout);
    scanf("%c", &opmsg[opnd_cnt * BYTEUNIT + 1]);
    write(clnt_sock, opmsg, opnd_cnt * BYTEUNIT + 2);
    int result;
    read(clnt_sock,&result,RLT_SIZE);

    printf("Operation result: %d \n",result);
    close(clnt_sock);
    return 0;
}

CalServer:

#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <memory.h>
#include <stdlib.h>
#include <stdio.h>

#define BUF_SIZE 1024
#define BYTEUNIT 4

void error_handling(char *message);

int calculate(int opnum, int opnds[], char op);

int main()
{
    int serv_sock, clnt_sock;
    struct sockaddr_in serv_adr, clnt_adr;
    socklen_t clnt_adr_sz;
    int result, opnd_cnt, i;
    int recv_cnt, recv_len;
    char opinfo[BUF_SIZE];

    serv_sock = socket(PF_INET, SOCK_STREAM, 0);
    if (serv_sock == -1)
    {
        error_handling((char*)"socket() error");
    }

    memset(&serv_adr, 0, sizeof(serv_adr));
    serv_adr.sin_family = AF_INET;
    serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_adr.sin_port = htons(9190);

    if (bind(serv_sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)
    {
        error_handling((char*)"bind() error");
    }

    if (listen(serv_sock, 5) == -1)
    {
        error_handling((char*)"listen() error");
    }

    clnt_adr_sz = sizeof(clnt_adr);

    for (int i = 0; i < 5; i++)
    {
        opnd_cnt = 0;
        clnt_sock = accept(serv_sock, (struct sockaddr *)&clnt_adr, &clnt_adr_sz);
        read(clnt_sock, &opnd_cnt, 1);

        recv_len = 0;

        while ((BYTEUNIT * opnd_cnt + 1) > recv_len)
        {
            recv_cnt = read(clnt_sock, &opinfo[recv_len], BUF_SIZE - 1);
            recv_len += recv_cnt;
        }

        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;
}

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

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

    case '-':
    {
        for (int i = 1; i < opnum; i++)
        {
            result -= opnds[i];
            break;
        }
    }

    case '*':
    {
        for (int i = 1; i < opnum; i++)
        {
            result *= opnds[i];
            break;
        }
    }
    }
    return result;
}