Почему фактический размер файла намного меньше данных, полученных через TCP-сокет в C? - PullRequest
Новая
       9

Почему фактический размер файла намного меньше данных, полученных через TCP-сокет в C?

1 голос
/ 26 мая 2020

Я пишу функцию сокета C для передачи файлов между системами (запущенными windows) в общей сети. Я мог бы передать все данные с сервера клиенту (в виде двоичных данных) и записать их в нужный файл. Но когда я смотрю на свойства файла, его размер намного меньше того, что я получил, и я не мог его открыть.

Это мой серверный код: 

#include<io.h>
#include<stdio.h>
#include<winsock2.h>
#include<string.h>
#include<stdint.h>

#pragma comment(lib,"ws2_32.lib") //Winsock Library

int main(int argc , char *argv[])
{
    WSADATA wsa;
    SOCKET s , new_socket;
    struct sockaddr_in server , client;
    int c;
    char message[10240];
    char filename[100];
    char response[50];
    int k = 0;

    printf("Enter filename :");
    scanf("%s",&filename);

    FILE *f = fopen(filename, "rb");
    fseek(f, 0, SEEK_END);
    long fsize = ftell(f);
    fseek(f, 0, SEEK_SET);  /* same as rewind(f); */

    uint8_t *string = malloc((1024 * sizeof(uint8_t)));


    printf("\nInitialising Winsock...");
    if (WSAStartup(MAKEWORD(2,2),&wsa) != 0)
    {
        printf("Failed. Error Code : %d",WSAGetLastError());
        return 1;
    }

    printf("Initialised.\n");

    //Create a socket
    if((s = socket(AF_INET , SOCK_STREAM , 0 )) == INVALID_SOCKET)
    {
        printf("Could not create socket : %d" , WSAGetLastError());
    }

    printf("Socket created.\n");

    //Prepare the sockaddr_in structure
    server.sin_family = AF_INET;
    server.sin_addr.s_addr = INADDR_ANY;
    server.sin_port = htons( 8888 );

    //Bind
    if( bind(s ,(struct sockaddr *)&server , sizeof(server)) == SOCKET_ERROR)
    {
        printf("Bind failed with error code : %d" , WSAGetLastError());
    }

    puts("Bind done");

    //Listen to incoming connections
    listen(s , 3);

    //Accept and incoming connection
    puts("Waiting for incoming connections...");

    c = sizeof(struct sockaddr_in);
    new_socket = accept(s , (struct sockaddr *)&client, &c);
    if (new_socket == INVALID_SOCKET)
    {
        printf("accept failed with error code : %d" , WSAGetLastError());
    }

    puts("Connection accepted");

    //Reply to client
    // message = "Hello Client , I have received your connection. But I have to go now, bye\n";
    send(new_socket , filename , strlen(filename) , 0);

    int len = recv(new_socket, response, 50, 0);
    response[len] = '\0';

    if(strcmp(response, "OKAY") == 0) printf("SUCCESS");
    else printf("ERROR");

    int i = 0;
    char cn = 'a';
    // getchar();
    getchar(); getchar();

    while(i < fsize){
        printf("%d\n",i);
        len = fread(string, 1, 1024, f);
        i += len;
        send(new_socket, string, len, 0);
    }
    getchar(); getchar();
    /*
    send(new_socket, "...END...CONNECTION...", strlen("...END...CONNECTION..."), 0);
    printf("sent end connection");
    recv(new_socket, response, strlen(response), 0);

    if(strcmp(response, "...END...CONNECTION...") != 0)
        send(new_socket, "...END...CONNECTION...", strlen("...END...CONNECTION..."), 0);
    */ 
    printf("SUCCESS");
    closesocket(s);
    fclose(f);
    WSACleanup();

    return 0;
}

и это мой клиентский код 

#include<stdio.h>
#include <stdint.h>
#include<winsock2.h>

#pragma comment(lib,&quot;ws2_32.lib&quot;) //Winsock Library

int main(int argc , char *argv[])
{
    WSADATA wsa;
    SOCKET s;
    struct sockaddr_in server;
    uint8_t server_reply[1024];
    int recv_size;

    printf("\nInitialising Winsock...");
    if (WSAStartup(MAKEWORD(2,2),&wsa) != 0)
    {
        printf(";Failed. Error Code : %d",WSAGetLastError());
        return 1;
    }

    printf("Initialised.\n");

    //Create a socket
    if((s = socket(AF_INET , SOCK_STREAM , 0 )) == INVALID_SOCKET)
    {
        printf("Could not create socket : %d" , WSAGetLastError());
    }

    printf("Socket created.\n");


    server.sin_addr.s_addr = inet_addr("127.0.0.1");
    server.sin_family = AF_INET;
    server.sin_port = htons( 8888 );

    //Connect to remote server
    if (connect(s , (struct sockaddr *)&server , sizeof(server)) < 0)
    {
        puts("connect error");
        return 1;
    }

    puts("Connected");

    if((recv_size = recv(s , server_reply , 1024 , 0)) == SOCKET_ERROR)
    {
        puts("recv failed");
    }

    puts("filename : ");
    server_reply[recv_size] = '\0';
    puts(server_reply);

    FILE* fptr = fopen(server_reply, "wb");

    send(s , "OKAY" , strlen("OKAY") , 0);
    int i = 0;
    while(1){
        if((recv_size = recv(s , server_reply , 1024, 0)) == SOCKET_ERROR)
        {
            puts("recv failed"); break;
        }
        fputs(server_reply, fptr);
        i += recv_size;
        printf("%d\n",i);
    }

    // send(s, "...END...CONNECTION...", strlen("...END...CONNECTION..."), 0);
    printf("sent : end connection");
    fclose(fptr);
    puts("File received\n");

    return 0;
}

Пожалуйста, укажите, где я ошибаюсь.

1 Ответ

2 голосов
/ 26 мая 2020

Сразу после вопроса я сам придумал ответ. Проблема заключалась в том, что в моей клиентской программе я записал полученные двоичные данные в файл, используя функцию fputs(), которая фактически принимает только тип данных char*.

Я изменил его на fwrite() и теперь он работает отлично.

это моя исправленная программа (только то, что пока l oop входит в клиентскую программу) 

int i = 0;
while(1){
    if((recv_size = recv(s , server_reply , 1024, 0)) == SOCKET_ERROR)
    {
        puts("recv failed"); break;
    }
    //fputs(server_reply, fptr);
    fwrite(server_reply, 1, recv_size, fptr);
    i += recv_size;
    printf("%d\n",i);
}
...