web-dev-qa-db-ja.com

ソケットを使用したファイル転送サーバー/クライアント

サーバーとクライアントの間でファイル転送を行おうとしていますが、うまく機能しません。基本的に何が起こる必要があるかです:
1)クライアントがサーバーにtxtファイルを送信します(私は「quotidiani.txt」と呼んでいます)
2)サーバーはそれを別のtxtファイル( "receive.txt")に保存します
3)サーバーはサーバー上でスクリプトを実行し、スクリプトを変更して別の名前で保存します( "output.txt")
4)サーバーはファイルをクライアントに送り返し、クライアントはそれを(同じソケットに)名前(final.txt)で保存します。

問題は、最初のファイル(quotidiani.txt)がほんの少しだけ読み取られ、その後いくつかのエラーが発生することです。エラーを理解して修正していただけると助かります。

これが私のコードです:

client.c:

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/wait.h>
#include <sys/socket.h>
#include <signal.h>
#include <ctype.h>          
#include <arpa/inet.h>
#include <netdb.h>

#define PORT 20000
#define LENGTH 512 


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

int main(int argc, char *argv[])
{
    /* Variable Definition */
    int sockfd; 
    int nsockfd;
    char revbuf[LENGTH]; 
    struct sockaddr_in remote_addr;

    /* Get the Socket file descriptor */
    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
    {
        fprintf(stderr, "ERROR: Failed to obtain Socket Descriptor! (errno = %d)\n",errno);
        exit(1);
    }

    /* Fill the socket address struct */
    remote_addr.sin_family = AF_INET; 
    remote_addr.sin_port = htons(PORT); 
    inet_pton(AF_INET, "127.0.0.1", &remote_addr.sin_addr); 
    bzero(&(remote_addr.sin_zero), 8);

    /* Try to connect the remote */
    if (connect(sockfd, (struct sockaddr *)&remote_addr, sizeof(struct sockaddr)) == -1)
    {
        fprintf(stderr, "ERROR: Failed to connect to the Host! (errno = %d)\n",errno);
        exit(1);
    }
    else 
        printf("[Client] Connected to server at port %d...ok!\n", PORT);

    /* Send File to Server */
    //if(!fork())
    //{
        char* fs_name = "/home/aryan/Desktop/quotidiani.txt";
        char sdbuf[LENGTH]; 
        printf("[Client] Sending %s to the Server... ", fs_name);
        FILE *fs = fopen(fs_name, "r");
        if(fs == NULL)
        {
            printf("ERROR: File %s not found.\n", fs_name);
            exit(1);
        }

        bzero(sdbuf, LENGTH); 
        int fs_block_sz; 
        while((fs_block_sz = fread(sdbuf, sizeof(char), LENGTH, fs)) > 0)
        {
            if(send(sockfd, sdbuf, fs_block_sz, 0) < 0)
            {
                fprintf(stderr, "ERROR: Failed to send file %s. (errno = %d)\n", fs_name, errno);
                break;
            }
            bzero(sdbuf, LENGTH);
        }
        printf("Ok File %s from Client was Sent!\n", fs_name);
    //}

    /* Receive File from Server */
    printf("[Client] Receiveing file from Server and saving it as final.txt...");
    char* fr_name = "/home/aryan/Desktop/progetto/final.txt";
    FILE *fr = fopen(fr_name, "a");
    if(fr == NULL)
        printf("File %s Cannot be opened.\n", fr_name);
    else
    {
        bzero(revbuf, LENGTH); 
        int fr_block_sz = 0;
        while((fr_block_sz = recv(sockfd, revbuf, LENGTH, 0)) > 0)
        {
            int write_sz = fwrite(revbuf, sizeof(char), fr_block_sz, fr);
            if(write_sz < fr_block_sz)
            {
                error("File write failed.\n");
            }
            bzero(revbuf, LENGTH);
            if (fr_block_sz == 0 || fr_block_sz != 512) 
            {
                break;
            }
        }
        if(fr_block_sz < 0)
        {
            if (errno == EAGAIN)
            {
                printf("recv() timed out.\n");
            }
            else
            {
                fprintf(stderr, "recv() failed due to errno = %d\n", errno);
            }
        }
        printf("Ok received from server!\n");
        fclose(fr);
    }
    close (sockfd);
    printf("[Client] Connection lost.\n");
    return (0);
}

server.c

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/wait.h>
#include <sys/socket.h>
#include <signal.h>
#include <ctype.h>          
#include <arpa/inet.h>
#include <netdb.h>

#define PORT 20000 
#define BACKLOG 5
#define LENGTH 512 

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

int main ()
{
    /* Defining Variables */
    int sockfd; 
    int nsockfd; 
    int num;
    int sin_size; 
    struct sockaddr_in addr_local; /* client addr */
    struct sockaddr_in addr_remote; /* server addr */
    char revbuf[LENGTH]; // Receiver buffer

    /* Get the Socket file descriptor */
    if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    {
        fprintf(stderr, "ERROR: Failed to obtain Socket Descriptor. (errno = %d)\n", errno);
        exit(1);
    }
    else 
        printf("[Server] Obtaining socket descriptor successfully.\n");

    /* Fill the client socket address struct */
    addr_local.sin_family = AF_INET; // Protocol Family
    addr_local.sin_port = htons(PORT); // Port number
    addr_local.sin_addr.s_addr = INADDR_ANY; // AutoFill local address
    bzero(&(addr_local.sin_zero), 8); // Flush the rest of struct

    /* Bind a special Port */
    if( bind(sockfd, (struct sockaddr*)&addr_local, sizeof(struct sockaddr)) == -1 )
    {
        fprintf(stderr, "ERROR: Failed to bind Port. (errno = %d)\n", errno);
        exit(1);
    }
    else 
        printf("[Server] Binded tcp port %d in addr 127.0.0.1 sucessfully.\n",PORT);

    /* Listen remote connect/calling */
    if(listen(sockfd,BACKLOG) == -1)
    {
        fprintf(stderr, "ERROR: Failed to listen Port. (errno = %d)\n", errno);
        exit(1);
    }
    else
        printf ("[Server] Listening the port %d successfully.\n", PORT);

    int success = 0;
    while(success == 0)
    {
        sin_size = sizeof(struct sockaddr_in);

        /* Wait a connection, and obtain a new socket file despriptor for single connection */
        if ((nsockfd = accept(sockfd, (struct sockaddr *)&addr_remote, &sin_size)) == -1) 
        {
            fprintf(stderr, "ERROR: Obtaining new Socket Despcritor. (errno = %d)\n", errno);
            exit(1);
        }
        else 
            printf("[Server] Server has got connected from %s.\n", inet_ntoa(addr_remote.sin_addr));

        /*Receive File from Client */
        char* fr_name = "/home/aryan/Desktop/receive.txt";
        FILE *fr = fopen(fr_name, "a");
        if(fr == NULL)
            printf("File %s Cannot be opened file on server.\n", fr_name);
        else
        {
            bzero(revbuf, LENGTH); 
            int fr_block_sz = 0;
            while((fr_block_sz = recv(nsockfd, revbuf, LENGTH, 0)) > 0) 
            {
                int write_sz = fwrite(revbuf, sizeof(char), fr_block_sz, fr);
                if(write_sz < fr_block_sz)
                {
                    error("File write failed on server.\n");
                }
                bzero(revbuf, LENGTH);
                if (fr_block_sz == 0 || fr_block_sz != 512) 
                {
                    break;
                }
            }
            if(fr_block_sz < 0)
            {
                if (errno == EAGAIN)
                {
                    printf("recv() timed out.\n");
                }
                else
                {
                    fprintf(stderr, "recv() failed due to errno = %d\n", errno);
                    exit(1);
                }
            }
            printf("Ok received from client!\n");
            fclose(fr); 
        }

        /* Call the Script */
        system("cd ; chmod +x script.sh ; ./script.sh");

        /* Send File to Client */
        //if(!fork())
        //{
            char* fs_name = "/home/aryan/Desktop/output.txt";
            char sdbuf[LENGTH]; // Send buffer
            printf("[Server] Sending %s to the Client...", fs_name);
            FILE *fs = fopen(fs_name, "r");
            if(fs == NULL)
            {
                fprintf(stderr, "ERROR: File %s not found on server. (errno = %d)\n", fs_name, errno);
                exit(1);
            }

            bzero(sdbuf, LENGTH); 
            int fs_block_sz; 
            while((fs_block_sz = fread(sdbuf, sizeof(char), LENGTH, fs))>0)
            {
                if(send(nsockfd, sdbuf, fs_block_sz, 0) < 0)
                {
                    fprintf(stderr, "ERROR: Failed to send file %s. (errno = %d)\n", fs_name, errno);
                    exit(1);
                }
                bzero(sdbuf, LENGTH);
            }
            printf("Ok sent to client!\n");
            success = 1;
            close(nsockfd);
            printf("[Server] Connection with Client closed. Server will wait now...\n");
            while(waitpid(-1, NULL, WNOHANG) > 0);
        //}
    }
}
12
AscaL

順不同のコメント:

  • あなたは頻繁に正確なエラーを知る機会を逃しています:

    _if(listen(sockfd,BACKLOG) == -1)
    {
        printf("ERROR: Failed to listen Port %d.\n", PORT);
        return (0);
    }
    _

    このブロックには、必ずperror("listen")または類似のものを含める必要があります。エラーの詳細がerrnoを介して報告される場合は、すべてのエラー処理ブロックに常にperror()またはstrerror()を含めてください。正確な失敗の理由があると、プログラミングの時間を節約でき、将来的に期待どおりに機能しなくなったときのユーザーとユーザーの時間も節約できます。

  • エラー処理には、さらに標準化が必要です。

    _if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 )
    {
        printf("ERROR: Failed to obtain Socket Descriptor.\n");
        return (0);
    }
    _

    これはしない_return 0_にする必要があります。これは、プログラムがエラーなしで完了したことをシェルに通知するためです。異常終了を知らせるには_return 1_(または_EXIT_SUCCESS_と_EXIT_FAILURE_を使用)する必要があります。

    _ else 
        printf("[Server] Server has got connected from %s.\n", inet_ntoa(addr_remote.sin_addr));
    
     /*Receive File from Client */
    _

    この前のブロックでは、エラー状態が発生していますが、とにかく実行を続けています。これは、非常に望ましくない動作をすばやく取得する方法です。これにより、メインサーバーループが再起動するか、子プロセスまたは同様のものが終了します。 (マルチプロセスサーバーを保持するかどうかによって異なります。)

    _if(!fork())
    {
    _

    前のブロックはfork()failingの説明を忘れていました。 fork()は失敗する可能性があり、失敗します-特に大学で一般的な共有ホスティング環境では-完全で複雑なthreefork()から返される可能性のある値:失敗、子、親。

  • fork()を無差別に使用しているようです。クライアントとサーバーはどちらも非常にシンプルで、実行するように設計されているため、複数のクライアントに同時にサービスを提供することはできません。少なくともアルゴリズムが完全にデバッグされ、複数のクライアントを同時に実行する方法を見つけるまでは、おそらくプロセスごとに厳密に1つのプロセスに固執する必要があります。これが現在発生している問題の原因であると思います。

  • 詳細をカプセル化するには、関数を使用する必要があります。サーバーに接続する関数、ファイルを送信する関数、ファイルを書き込む関数などを記述します。複雑な部分的な書き込みを処理する関数を記述します。 (特にwriten関数を nix環境での高度なプログラミング 本のソースコードから盗むことをお勧めします。ファイル_lib/writen.c_。)関数を正しく記述すれば、再クライアントとサーバーの両方で使用します。 (それらを_utils.c_に配置し、_gcc -o server server.c utils.c_のようなプログラムをコンパイルするようなもの。)

    それぞれが1つの処理を実行する小さな関数を使用すると、一度に少量のコードに集中できそれぞれに小さなテストを記述して、絞り込みに役立ちますコードのどのセクションにまだ改善が必要かを調べます。

8
sarnold

ここには議論のポイントが一つ足りないようですので、ここで触れておきたいと思いました。

TCPのデータ転送をすぐに理解しましょう。 3つのステップがあります。a)接続の確立、b)データ転送、c)接続の終了

ここで、クライアントはTCPソケットを介してサーバーにファイルを送信しています。

サーバーはファイルに対して何らかの処理を行い、それをクライアントに送り返します。

ここで、3つのステップすべてを実行する必要があります。接続の確立は、connectを呼び出すことによって行われます。データの読み取り/書き込みはここでrecv/sendによって行われ、接続の終了はcloseによって行われます。

ここのサーバーは、recvを使用してループでデータを読み取っています。さて、ループが終わるとき? recvが0を返すか、エラーで0未満になる場合があります。 recvが0を返すとき? ->反対側が接続を閉じたとき。(TCP FINセグメントがこの側によって受信されたとき)。

したがって、このコードでは、クライアントがファイルの送信を終了したときに、シャットダウン機能を使用しました。これは、クライアント側からFINセグメントを送信し、サーバーのrecvは0を返し、プログラムは続行します(クライアントがその後もデータを読み取る必要があります)。

(理解のために、TCPの接続確立は3ウェイハンドシェイクであり、接続終了は4ウェイハンドシェイクであることに注意してください。)

同様に、サーバー側で接続ソケットを閉じるのを忘れると、クライアントのrecvも永久にブロックします。 ctrl cを使用してクライアントを停止する理由は、あなたが言及したときだったと思います。

あなたはplすることができます。 TCPの詳細については、標準のネットワーキングブックまたはrfc http://www.ietf.org/rfc/rfc793.txt を参照してください。

変更したコードを貼り付け、コメントも少し追加しました。

この説明がお役に立てば幸いです。

変更されたクライアントコード:

 while((fs_block_sz = fread(sdbuf, sizeof(char), LENGTH, fs)) > 0)
    {
        if(send(sockfd, sdbuf, fs_block_sz, 0) < 0)
        {
            fprintf(stderr, "ERROR: Failed to send file %s. (errno = %d)\n", fs_name, errno);
            exit(1);
        }
        bzero(sdbuf, LENGTH);
    }

 /*Now we have sent the File's data, what about server's recv?
 Recv is blocked  and waiting for data to arrive or if the protocol
 stack receives a TCP FIN segment ..then the recv will return 0 and
 the server code can continue */
 /*Sending the TCP FIN segment by shutdown and this is half way
 close, since the client also needs to read data subsequently*/

 shutdown(sockfd, SHUT_WR);
 printf("Ok File %s from Client was Sent!\n", fs_name);