Не получает передачу файла из TCP-соединения - PullRequest
0 голосов
/ 07 апреля 2019

Я пишу протокол TCP-клиента и сервера для школьного проекта.Клиент отправляет сообщение «GET \ r \ n», и сервер должен передать «+ OK \ r \ n», размер файла и файла, если он существует в каталоге сервера.Я заблокирован в передаче файлов

Я пытался решить ее небольшими шагами одновременно.Я установил соединение, отправил запрос от клиента и получил сообщение «ОК» с сервера.

Теперь я открыл файл на сервере и попытался отправить его клиенту по 128 байт за раз.Чтение файла работает и, по-видимому, также отправка буферов, но клиент ничего не получает ...

Вот мой сервер.c


#include    <stdlib.h>
#include    <string.h>
#include    <inttypes.h>

#include    "../errlib.h"
#include    "../sockwrap.h"

#define BUFLEN      128 /* Buffer length */
#define TIMEOUT     15  /* TIMEOUT */

/* FUNCTION PROTOTYPES */
void service(int s);

/* GLOBAL VARIABLES */
char *prog_name;

int main(int argc, char *argv[])
{
    int         conn_request_skt;   /* passive socket */
    uint16_t    lport_n, lport_h;   /* port used by server (net/host ord.) */
    int         bklog = 2;          /* listen backlog */
    int         s;                  /* connected socket */

    fd_set      cset;               // waiting for connection
    struct timeval  tval;           // timeout
    size_t      n;

    socklen_t   addrlen;
    struct sockaddr_in  saddr, caddr;   /* server and client addresses */

    prog_name = argv[0];

    if (argc != 2) {
        printf("Usage: %s <port number>\n", prog_name);
        exit(1);
    }

    /* get server port number */
    if (sscanf(argv[1], "%" SCNu16, &lport_h)!=1)
    err_sys("Invalid port number");
    lport_n = htons(lport_h);

    /* create the socket */
    printf("creating socket...\n");
    s = Socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    printf("done, socket number %u\n",s);

    /* bind the socket to any local IP address */
    bzero(&saddr, sizeof(saddr));
    saddr.sin_family      = AF_INET;
    saddr.sin_port        = lport_n;
    saddr.sin_addr.s_addr = INADDR_ANY;
    showAddr("Binding to address", &saddr);
    Bind(s, (struct sockaddr *) &saddr, sizeof(saddr));
    printf("done.\n");

    /* listen */
    printf ("Listening at socket %d with backlog = %d \n",s,bklog);
    Listen(s, bklog);
    printf("done.\n");

    conn_request_skt = s;

    /* main server loop */
    for ( ; ; )
    {
        printf("waiting for connection...\n");
        /* accept next connection */
        FD_ZERO(&cset);
        FD_SET(conn_request_skt, &cset);
        tval.tv_sec = TIMEOUT;
        tval.tv_usec = 0;
        n = Select(FD_SETSIZE, &cset, NULL, NULL, &tval);

        if ( n > 0 ){
            addrlen = sizeof(struct sockaddr_in);
            s = Accept(conn_request_skt, (struct sockaddr *) &caddr, &addrlen);
            showAddr("Accepted connection from", &caddr);
            printf("new socket: %u\n",s);

            /* serve the client on socket s */
            service(s);
        } else {
            printf("No connection request after %d seconds\n",TIMEOUT);

        }
    }
}

void service(int s) {
    char    buf[BUFLEN];        /* reception buffer */
    char    filename[BUFLEN];

    int     n;
    long    filesize;
    uint32_t fsize;

    FILE *fp;

    for ( ; ; )
    {
        n = recv(s, buf, BUFLEN, 0);
        if (n < 0) {
            printf("Read error\n");
            close(s);
            printf("Socket %d closed\n", s);
            break;
        } else if (n == 0) {
            printf("Connection closed by party on socket %d\n",s);
            close(s);
            break;
        } else {
            printf("Received request from socket %03d :\n", s);
            sscanf(buf, "GET %s\r\n", filename);
            strcpy(buf, "+OK\r\n");
            printf("%s",buf);
            if(writen(s, buf, strlen(buf)) != strlen(buf))
                printf("Write error while sending +OK\n");

            // open file
            fp = fopen(filename, "r");
            if( fp == NULL){
                //TODO close connection
            }
            // calculating dim of file
            fseek(fp, 0L, SEEK_END);
            filesize = ftell(fp);
            rewind(fp); // go back at beginning of file

            fsize = htonl(filesize); // size file in network byte order

            // sending file size
            if(writen(s, &fsize, 4) != 4)
                printf("Write error while sending file size\n");

            while(fread(buf, 1, BUFLEN - 1, fp) == BUFLEN - 1){
                printf("%s", buf);
                if(writen(s, buf, strlen(buf)) != strlen(buf))
                    printf("Write error while buf\n");
            }
            printf("%s", buf);
            printf("I am here\n");
        }
    }
}


Пока здесь мой клиент.c


#include     <stdlib.h>
#include     <string.h>
#include     <inttypes.h>

#include     "../errlib.h"
#include     "../sockwrap.h"

#define BUFLEN  128 /* BUFFER LENGTH */
#define TIMEOUT 15  /* TIMEOUT*/

/* GLOBAL VARIABLES */
char *prog_name;

int main(int argc, char *argv[])
{
    char        request[BUFLEN];        /* request buffer */
    char        rbuf[BUFLEN];       /* reception buffer */

    uint32_t        taddr_n;            /* server IP addr. (net/host ord) */
    uint16_t        tport_n, tport_h;   /* server port number (net/host ord) */

    int        s, len;
    int        result;
    struct sockaddr_in  saddr;      /* server address structure */
    struct in_addr  sIPaddr;        /* server IP addr. structure */

    fd_set      cset;               // variables for timeout
    struct timeval  tval;
    size_t n;

    prog_name = argv[0];

    if(argc < 4)
        err_sys("Wrong number of parameters!\n");

    // read address from first argument
    taddr_n = inet_addr(argv[1]);
    if (taddr_n == INADDR_NONE)
        err_sys("Invalid address");

    // read port number from second argument
    if (sscanf(argv[2], "%" SCNu16, &tport_h)!=1)
    err_sys("Invalid port number");
    tport_n = htons(tport_h);

    /* create the socket */
    printf("Creating socket\n");
    s = Socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    printf("done. Socket fd number: %d\n",s);

    /* prepare address structure */
    bzero(&saddr, sizeof(saddr));
    saddr.sin_family = AF_INET;
    saddr.sin_port   = tport_n;
    saddr.sin_addr   = sIPaddr;


    printf("trying to connect to the server...\n");

    /* connect */
    showAddr("Connecting to target address", &saddr);
    Connect(s, (struct sockaddr *) &saddr, sizeof(saddr));
    printf("done.\n");

    // loop to request files
    for (int i = 3 ; i < argc ; i++ ){ // i = 4 because the first file is the fourth argument
        // check if file name is too big
        if(strlen(argv[i]) >= BUFLEN - 6)
            err_sys("The file name is too big for the buffer request!\n");

        // create the string of bytes for the request
        strcpy(request, "GET ");
        strcat(request, argv[i]);
        strcat(request, "\r\n");

        len = strlen(request);
        if(writen(s, request, len) != len){
            printf("Write error\n");
            break;
        }

        printf("waiting for response...\n");
        // receive file from server
        n = recv(s, rbuf, BUFLEN, 0);
        if (n < 0) {
            printf("Read error\n");
            close(s);
            printf("Socket %d closed\n", s);
            break;
        } else if (n == 0) {
            printf("Connection closed by party on socket %d\n",s);
            close(s);
            break;
        } else {
            printf("Received reply from server\n");

            uint32_t fsize;

            printf("%s",rbuf);
            if(strcmp(rbuf, "+OK\r\n") == 0){
                n = recv(s, &fsize, 4, 0);
                if (n < 0) {
                    printf("Read error\n");
                    close(s);
                    printf("Socket %d closed\n", s);
                    break;
                } else if (n == 0) {
                    printf("Connection closed by party on socket %d\n",s);
                    close(s);
                    break;
                } else {
                    // received file dimension
                    fsize = ntohl(fsize);
                }

                while(fsize > 0){
                    printf("I am here1n\n");
                    // receive file
                    n = recv(s, rbuf, BUFLEN-1, 0);
                    if (n < 0) {
                        printf("Read error\n");
                        close(s);
                        printf("Socket %d closed\n", s);
                        break;
                    } else if (n == 0) {
                        printf("Connection closed by party on socket %d\n",s);
                        close(s);
                        break;
                    } else {
                        printf("I am here");
                        fsize -= n;
                    }
                }

            }
        }
    }

    printf("===========================================================\n");
    close(s);
    exit(0);
}

Recv в клиенте, где я должен получить файл, просто блокируется, ничего не получая.Я не понимаю, чего мне не хватает ...

1 Ответ

1 голос
/ 09 апреля 2019

Проблема здесь обычная: вы не осторожны с границами сообщений.

В вашем клиенте вы делаете recv и проверяете, больше ли число байтов, чем 0. Но тогда вы больше не будете проверять длину. Затем вы делаете strcmp для конкретной строки, которую ожидаете получить (+OK\r\n). Но вы, возможно, получили 3 байта (+OK) или могли получить 10: (+OK\r\nXXXXX) или более [в сторону: также, recv не гарантирует, что ваша строка байтов заканчивается нулем]. Ничто не мешает ядру на дальней стороне пакетировать преамбулу плюс последующих байтов в один пакет TCP. Аналогично, ничто не мешает локальной стороне объединять несколько пакетов TCP в один буфер.

Вы должны указать границы сообщения. Если вы ожидаете, что ваше следующее сообщение будет 5 байтами, то вы должны получить ровно 5 байтов (и повторите попытку, если получите меньше - будьте осторожны, чтобы также проверить EOF на случай, если другая сторона прервется раньше). Или, в качестве альтернативы, поместите слой буферизации перед логикой приема, чтобы он получал до некоторой большой суммы, возвращал вам желаемое количество байтов, а затем сохранял все, что больше, для последующего вызова «прием».

Чтобы переформулировать это другим способом: ваш сервер отправляет +OK\r\n, затем он отправляет четырехбайтовую длину, а затем начинает отправку файла. Но это означает, что ваш первый recv на стороне клиента может получить преамбулу, плюс длину плюс первые N байтов файла за один системный вызов.

TCP не уважает, не обеспечивает и не устанавливает границы сообщений.

...