Клиент сокета C

Question

#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");

    while (1) {
        printf("Please enter the message: ");
        bzero(buffer, 256);
        fgets(buffer, 255, stdin);
        // n = write(sockfd, buffer, strlen(buffer));
        n = send(sockfd, buffer, strlen(buffer), 0);
        if (n < 0) error("ERROR writing to socket");

        bzero(buffer, 256);
        // n = read(sockfd, buffer, 255);
        n = recv(sockfd, buffer, 255, 0);
        if (n < 0) error("ERROR reading from socket");

        printf("%s\n", buffer);
    }
    close(sockfd);

    return 0;
}

Приведенный выше код работает только на первой итерации.Однако на второй итерации он блокируется на recv() или read().

. Может кто-нибудь объяснить причину такого поведения?

Обновление: вот код сервера (не тот, который я использую, но те же принципы), и он работает как тот, который я упомянул выше:

/* A simple server in the internet domain using TCP
   The port number is passed as an argument */
#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");
    memset((char*) &serv_addr, 0, 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);
    while (1) {
        newsockfd = accept(sockfd,
                (struct sockaddr *) &cli_addr,
                &clilen);
        if (newsockfd < 0)
            error("ERROR on accept");
        memset(buffer, 0, sizeof(buffer));
        n = read(newsockfd, buffer, 255);
        if (n < 0) error("ERROR reading from socket");
        printf("Here is the message: %s\n", buffer);
        n = write(newsockfd, "I got your message", 18);
        if (n < 0) error("ERROR writing to socket");
    }
    close(newsockfd);
    close(sockfd);
    return 0;
}

Answer 1

Ваш while цикл в коде сервера находится не в том месте.Его следует изменить следующим образом:

        memset(buffer, 0, sizeof(buffer));
        n = read(newsockfd, buffer, 255);
        if (n < 0) error("ERROR reading from socket");
        printf("Here is the message: %s\n", buffer);
        n = write(newsockfd, "I got your message", 18);
        if (n < 0) error("ERROR writing to socket");

В вашем коде вы accept() устанавливаете новое соединение в каждой итерации, а затем listen() для read() на нем.С другой стороны, клиент не устанавливает новое соединение, он выполняет write() для одного и того же сокета.

Либо ваш клиент должен каждый раз закрывать соединение и переподключаться, либо сервер должен разрешать несколькоread() / write() для связи.

Answer 2

Это одна из тех вещей с сокетами, которые я не совсем понимаю, но если вы делаете это так, она должна работать. По крайней мере, для меня это так.

while(recv(sockfd, buffer, 255, 0) > 0) {
    ...
}

Если кто-нибудь знает, почему эта работа, пожалуйста, оставьте мне комментарий ниже:)