[TCP/IP] タイムアウト

unsigned int arlarm(unsigned int secs)

alermはタイマーを開始する
タイマーが切れると、SIGALARMシグナルがプロセスに送られ、SIGALRMハンドラ関数がある場合はそれが実行される

#include <stdio.h> /* printf(), fprintf()に必要 */
#include <sys/socket.h> /* socket(), bind(), connect()に必要 */
#include <arpa/inet.h> /* sockaddr_in, inet_ntoaに必要 */
#include <stdlib.h> /* atoi()に必要 */
#include <string.h> /* memset()に必要 */
#include <unistd.h> /* close()に必要 */
#include <fcntl.h> /* fcntl()に必要 */
#include <sys/file.h> /* O_NONBLOCK, FASYNCに必要 */
#include <signal.h> /* signal(),SIGLRMに必要 */
#include <error.h> /* errnoに必要 */

#define ECHOMAX 255 /* エコー文字列の最大長 */
#define TIMEOUT_SECS 2 /* 再送信までの秒数 */
#define MAXTRIES 5 /* 最大試行回数 */

int tries=0; /* エコー文字列の最大長 */

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

int main(int argc, char *argv[]){
    int sock; /* ソケットディスクリプタ */
    struct sockaddr_in echoServAddr; /* エコーサーバアドレス */
    struct sockaddr_in fromAddr; /* エコー送信元のアドレス */
    unsigned short echoServPort; /* エコーサーバポート */
    unsigned int fromSize; /* recvfrom()アドレスの入出力サイズ */
    struct sigaction myAction; /* シグナルハンドラの設定用 */
    char *servIP;  /* サーバのIP */
    char *echoString; /* 送信する文字列 */
    char echoBuffer[ECHOMAX+1]; /* エコー文字列受信用バッファ */
    int echoStringLen; /* エコー文字列の長さ */
    int respStringLen; /* 受信した応答の長さ */

    /* 引数が正しいか確認 */
    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((echoStringLen = strlen(echoString)) > ECHOMAX) /* 入力した文字列の長さをチェック */
        DieWithError("Echo word too long");

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

    /* UDPデータグラムソケット作成 */
    if((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0)
        DieWithError("socket() failed");

    /* アラームシグナル用シグナルハンドラのセット */
    myAction.sa_handler = CatchAlarm;
    if(sigfillset(&myAction.sa_mask) < 0) /* ハンドラ内では全てをブロック */
        DieWithError("sigfillset() failed");
    myAction.sa_flags = 0;

    if(sigaction(SIGALRM, &myAction, 0) < 0) /* ハンドラ内では全てをブロック */
        DieWithError("sigaction() failed for SIGALRM");

    /* サーバのアドレス構造体 */
    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 (sendto(sock, echoString, echoStringLen, 0, (struct sockaddr *) &echoServAddr, sizeof(echoServAddr)) != echoStringLen)
        DieWithError("sendto() sent a different number of bytes than expected");        

    /* 応答を受信 */
    fromSize = sizeof(fromAddr);
    alarm(TIMEOUT_SECS);
    while((respStringLen = recvfrom(sock, echoBuffer, ECHOMAX, 0,
        (struct sockaddr *) &fromAddr, &fromSize)) < 0)
        if (errno = EINTR){
            if (tries < MAXTRIES) {
                printf("timed out, %d more tries..\n", MATRIES - tries);
                if(sendto(sock, echoString, echoStringLen, 0, (struct sockaddr *) &echoServAddr, sizeof(echoServAddr)) != echoStringLen)
                    DieWithError("sendto() failed");
                alarm(TIMEOUT_SECS);
            }
            else
                DieWithError("No Response");
        }
        else
            DieWithError("recvfrom() failed");
        alarm(0);

        /* 受信データをNULL文字で終端させる */
        echoBuffer[respStringLen] = '\0';
        printf("Received: %s\n", echoBuffer);

        close(sock);
        exit(0);
}

void CatchAlarm(int ignored) {
    tries += 1;
}