我是否正确创建了协议？

Question

我正在尝试制作一个协议（建立在 TCP 之上），可以将字符串从客户端发送到端口 457 上的服务器。这是我目前所拥有的：

服务器.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h> 
#include <sys/socket.h>
#include <netinet/in.h>

void error(const char *msg)
{
    perror(msg);
    exit(1);
}

int main(int argc, char *argv[])
{
    int sockfd, newsockfd, portno;
    socklen_t clilen;
    char buffer[256];
    struct sockaddr_in serv_addr, cli_addr;
    int n;
    if (argc < 2) {
        fprintf(stderr,"ERROR, no port provided\n");
        exit(1);
    }
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0) 
        error("ERROR opening socket");
    bzero((char *) &serv_addr, sizeof(serv_addr));
    portno = atoi(argv[1]);
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = INADDR_ANY;
    serv_addr.sin_port = htons(portno);
    if (bind(sockfd, (struct sockaddr *) &serv_addr,
             sizeof(serv_addr)) < 0) 
        error("ERROR on binding");
    listen(sockfd,5);
    clilen = sizeof(cli_addr);
    newsockfd = accept(sockfd, 
                       (struct sockaddr *) &cli_addr, 
                       &clilen);
    if (newsockfd < 0) 
        error("ERROR on accept");
    bzero(buffer,256);
    n = read(newsockfd,buffer,255);
    if (n < 0) error("ERROR reading from socket");
    printf("Here is the message: %s\n",buffer);
    n = write(newsockfd,"U got your messaze",18);
    if (n < 0) error("ERROR writing to socket");
    close(newsockfd);
    close(sockfd);
    return 0; 
}

客户端.c

    #include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h> 

void error(const char *msg)
{
    perror(msg);
    exit(0);
}

int main(int argc, char *argv[])
{
    int sockfd, portno, n;
    struct sockaddr_in serv_addr;
    struct hostent *server;

    char buffer[256];
    if (argc < 3) {
        fprintf(stderr,"usage %s hostname port\n", argv[0]);
        exit(0);
    }
    portno = atoi(argv[2]);
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0) 
        error("ERROR opening socket");
    server = gethostbyname(argv[1]);
    if (server == NULL) {
        fprintf(stderr,"ERROR, no such host\n");
        exit(0);
    }
    bzero((char *) &serv_addr, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    bcopy((char *)server->h_addr, 
          (char *)&serv_addr.sin_addr.s_addr,
          server->h_length);
    serv_addr.sin_port = htons(portno);
    if (connect(sockfd,(struct sockaddr *) &serv_addr,sizeof(serv_addr)) < 0) 
        error("ERROR connecting");
    printf("Please enter the message: ");
    bzero(buffer,256);
    fgets(buffer,255,stdin);
    n = write(sockfd,buffer,strlen(buffer));
    if (n < 0) 
        error("ERROR writing to socket");
    bzero(buffer,256);
    n = read(sockfd,buffer,255);
    if (n < 0) 
        error("ERROR reading from socket");
    printf("%s\n",buffer);
    close(sockfd);
    return 0;
}

我感觉好像我没有正确设置协议。我是吗？

Answer 1

您应该首先为您的协议编写规范。然后，在审查它之后，您实施规范。您基本上是在要求我们对您的协议进行逆向工程。这是创建协议的一种倒退方法。在开始实施之前，您应该已经清楚地了解您的程序需要做什么。

您的服务器程序充当有限类型的 ECHO 服务器，即接受来自客户端的输入不超过 255 个字节。服务器能够读取的任何内容都会记录到控制台，并且消息 U got your messaze 作为响应发送到客户端。

您可能想要解决的一些问题：

• 虽然不太可能，但您的 read() 调用返回的数据可能少于客户端发送的数据，即使客户端发送的数据少于 256 个字节。例如，如果客户端一次发送一个字节的aaaaaaaaaa，您的服务器可能只会看到第一个a，并假定它是一条完整的消息。

• 您没有采取预防措施来防止写入已关闭的连接。这可能会生成SIGPIPE，并导致您的程序意外退出。

• 信号通常可能会中断您的系统调用。如果发生这种情况，您应该检测到这种情况并重新启动系统调用。

Answer 2

你有一些不正确的类型：

---

htons 期望 uint16_t 作为参数

htons((uint16_t)portno);

---

read() 和 write() 期望 ssize_t

ssize_t n;

---

bzero 已弃用，请使用

memset(buffer, 0, sizeof(buffer));

---

bcopy 已弃用，请使用

memmove(server->h_addr, &serv_addr.sin_addr.s_addr, server->h_length);

---

NUL 在使用 read() 时终止 buffer

/* bzero(buffer,256); Not needed */
n = read(newsockfd,buffer,255);
if (n < 0) error("ERROR reading from socket");
buffer[n]= '\0'; /* here */
printf("Here is the message: %s\n",buffer);

请注意，现代程序使用send() 和recv() 而不是read() 和write()

---

最后，不要使用幻数

fgets(buffer, 255, stdin);

改为：

fgets(buffer, sizeof(buffer), stdin); /* 256 */

为什么是 255？ fgets() 函数应该从流中读取字节到 s 指向的数组中，直到读取 n-1 个字节，所以 256 是正确的。