[TCP/IP] TCPクライアントをCで実装

TCPEchoClient.c

#include <stdio.h> /* printf(), fprintf()に必要 */
#include <sys/socket.h> /* socket(), connect(), send(), recv() */
#include <arpa/inet.h> /* sockaddr_in, inet_addr() */
#include <stdlib.h> /* atoi() */
#include <string.h> /* memset() */
#include <unistd.h> /* close() */

#define RCVBUFSIZE 32 /* 受信バッファのサイズ */

void DieWithError(char *errorMessage); /* エラー処理関数 */

int main(int argc, char *argv[]){
    int sock; /* ソケットディスクリプタ */
    struct sockaddr_in echoServAddr; /* エコーサーバアドレス */
    unsigned short echoServPort; /* エコーサーバポート */
    char *servIP;  /* サーバのIP */
    char *echoString; /* 送信する文字列 */
    char echoBuffer[RCVBUFSIZE]; /* エコー文字列用バッファ */
    unsigned int echoStringLen; /* エコーする文字列サイズ */
    int bytesRcvd, totalBytesRcvd; /* 1回のrecv()で読み取られるバイト数と全バイト数 */

    if((argc < 3) || (argc > 4)){
        fprintf(stderr, "Usage: %s <Server IP><Echo Word>[<Echo Port>]\n", argv[0]);
        exit(1);
    }

    servIP = argv[1]; /* 1つ目の引数: サーバのIP */
    echoString = argv[2]; /* 2つ目の引数: エコー文字列 */

    if (argc == 4)
        echoServPort = atoi(argv[3]); /* 指定ポートがあれば使用 */
    else
        echoServPort = 7; /* 7はエコーサービスのwellknown port */

    /* ストリームソケット作成 */
    if((sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
        DieWithError("socket() failed");

    /* サーバのアドレス構造体 */
    memset(&echoServAddr, 0, sizeof(echoServAddr)); /* 構造体に0埋め */
    echoServAddr.sin_family = AF_INET; /* インターネットアドレスファミリ */
    echoServAddr.sin_addr.s_addr = inet_addr(servIP); /* サーバIP */
    echoServAddr.sin_port = htons(echoServPort); /* サーバポート */

    /* サーバへの接続 */
    if(connect(sock, (struct sockaddr *) &echoServAddr, sizeof(echoServAdd)) < 0)
        DieWithError("connect() failed");

    echoStringLen = strlen(echoString); /* 入力データの長さ */

    /* 文字列をサーバに送信 */
    if (send(sock, echoString, echoStringLen, 0) != echoStringLen)
        DieWithError("send () sent a different number of bytes than expected");

    /* 同じ文字列をサーバから受信 */
    totalBytesRcvd = 0;
    printf("Received: ");

    while (totalBytesRcvd < echoStringLen) {
        /* バッファサイズに達するまでデータ受信 */
        if((bytesRcvd = recv(sock, echoBuffer, RCVBUFSIZE - 1, 0)) <= 0)
            DieWithError("recv() failed or connection closed prematurely");
        totalBytesRcvd += bytesRcvd; /* バイト数集計 */
        echoBuffer[bytesRcvd] = '\0'; /* 文字列終了 */
        printf(echoBuffer); /* バッファの表示 */
    }

    printf("\n");

    close(sock);
    exit(0);
}

DieWithError.c

#include <stdio.h> /* perror()に必要 */
#include <stdlib.h> /* exit()に必要 */

void DieWithError(char *errorMessage) {
    perror(errorMessage);
    exit(1);
}

$ gcc -o TCPEchoClient TCPEchoClient.c DieWithError.c