selectを使う(タイムアウト付き)

ここでは、selectをタイムアウトさせる方法を説明します。

selectを使うサンプルコード(タイムアウト付き)

下記サンプルコードでは、UDPソケットを2つ作成しています。 作成した2つのUDPソケットは、selectに登録してselectはブロッキング状態に入ります。 UDPソケットにデータが到着するとrecvを行い、受信した内容を表示します。

以前のサンプルとこのサンプルの違いは、タイムアウトを設定している事です。 このサンプルでは、selectの4つめの引数はNULLではなくstruct timevalへのポインタを渡しています。 struct timevalには、10秒を表す値を代入してあります。 このサンプルでは、selectは最大10秒までしかブロッキングしません。 selectがブロッキングを開始してから10秒の時間が経過するとselectは0を返して終了します。 select以後のif文では、selectの返り値が0であればwhileループを抜けるようにしてあります。


#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/time.h>

int
main()
{
 int sock1, sock2;
 struct sockaddr_in addr1, addr2;
 fd_set fds, readfds;
 char buf[2048];
 int maxfd;
 int n;
 struct timeval tv;

 /* 受信ソケットを2つ作ります */
 sock1 = socket(AF_INET, SOCK_DGRAM, 0);
 sock2 = socket(AF_INET, SOCK_DGRAM, 0);

 addr1.sin_family = AF_INET;
 addr2.sin_family = AF_INET;

 addr1.sin_addr.s_addr = inet_addr("127.0.0.1");
 addr2.sin_addr.s_addr = inet_addr("127.0.0.1");

 /* 2つの別々のポートで待つために別のポート番号をそれぞれ設定します */
 addr1.sin_port = htons(11111);
 addr2.sin_port = htons(22222);

 addr1.sin_len = sizeof(addr1);
 addr2.sin_len = sizeof(addr2);

 /* 2つの別々のポートで待つようにbindします */
 bind(sock1, (struct sockaddr *)&addr1, sizeof(addr1));
 bind(sock2, (struct sockaddr *)&addr2, sizeof(addr2));

 /* fd_setの初期化します */
 FD_ZERO(&readfds);

 /* selectで待つ読み込みソケットとしてsock1を登録します */
 FD_SET(sock1, &readfds);
 /* selectで待つ読み込みソケットとしてsock2を登録します */
 FD_SET(sock2, &readfds);

 /* 10秒でタイムアウトするようにします */
 tv.tv_sec = 10;
 tv.tv_usec = 0;


 if (sock1 > sock2) {
    maxfd = sock1;
 } else {
    maxfd = sock2;
 }

 /* このサンプルでは、10秒間データを受信しないとループを抜けます */
 while (1) {
	/*
           読み込み用fd_setの初期化
	   selectが毎回内容を上書きしてしまうので、毎回初期化します
        */
	memcpy(&fds, &readfds, sizeof(fd_set));

	/* fdsに設定されたソケットが読み込み可能になるまで待ちます */
	n = select(maxfd+1, &fds, NULL, NULL, &tv);

	/* タイムアウトの場合にselectは0を返します */
	if (n == 0) {
		/* ループから抜けます */
		printf("timeout\n");
		break;
	}

	/* sock1に読み込み可能データがある場合 */
	if (FD_ISSET(sock1, &fds)) {
		/* sock1からデータを受信して表示します */
		memset(buf, 0, sizeof(buf));
		recv(sock1, buf, sizeof(buf), 0);
		printf("%s\n", buf);
	}

	/* sock2に読み込み可能データがある場合 */
	if (FD_ISSET(sock2, &fds)) {
		/* sock2からデータを受信して表示します */
		memset(buf, 0, sizeof(buf));
		recv(sock2, buf, sizeof(buf), 0);
		printf("%s\n", buf);
	}
 }

 close(sock1);
 close(sock2);

 return 0;
}

上記サンプルへのUDPパケット送信は、タイムアウトを利用しないselectの説明で利用したUDPを送るプログラムをご利用ください。