/*
 * $Id: echo-server-kqueue-1.c,v 1.3 2005年06月11日 20:25:10 68user Exp $
 *
 * kqueue・kevent を使ったマルチスレッド echo サーバ
 *
 * written by 68user http://X68000.q-e-d.net/~68user/
 */
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <netdb.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/event.h>
#define BUF_LEN 256 /* バッファのサイズ */
#define MAX_SOCK 256 /* 最大プロセス数 */
 /* クライアントの情報を保持する構造体 */
typedef struct CLIENT_INFO {
 char hostname[BUF_LEN]; /* ホスト名 */
 char ipaddr[BUF_LEN]; /* IP アドレス */
 int port; /* ポート番号 */
} CLIENT_INFO;
CLIENT_INFO client_info[MAX_SOCK+1];
/*-----------------------------------------------------
 引数でリスニングソケットを受け取り、accept し、
 client_info に新しいクライアントの情報を登録する。
 戻り値は新しいクライアントのソケットディスクリプタ。
 ただしエラー発生時は -1 を返す。
 -----------------------------------------------------*/
int
accept_new_client(int sock){
 int len;
 int new_socket;
 struct hostent *peer_host;
 struct sockaddr_in peer_sin;
 len = sizeof(peer_sin);
 new_socket = accept(sock, (struct sockaddr *)&peer_sin, &len);
 if ( new_socket == -1 ){
 perror("accept");
 exit(1);
 }
 /* ここから先はデバッグ用の情報取得 */
 peer_host = gethostbyaddr((char *)&peer_sin.sin_addr.s_addr,
 sizeof(peer_sin.sin_addr), AF_INET);
 /* ホスト名 */
 strncpy(client_info[new_socket].hostname, peer_host->h_name,
 sizeof client_info[new_socket].hostname);
 /* IP アドレス */
 strncpy(client_info[new_socket].ipaddr, inet_ntoa(peer_sin.sin_addr),
 sizeof client_info[new_socket].ipaddr);
 /* ポート番号 */
 client_info[new_socket].port = ntohs(peer_sin.sin_port);
 
 printf("接続: %s (%s) ポート %d ディスクリプタ %d 番\n",
 client_info[new_socket].hostname,
 client_info[new_socket].ipaddr,
 client_info[new_socket].port,
 new_socket);
 return new_socket;
}
/*-----------------------------------------------------
 引数でソケットディスクリプタを受け取り、そのソケットから
 read(2) で文字列を読み込み、文字列をそのままクライアントに
 送信する。
 -----------------------------------------------------*/
void
read_and_reply(int sock){
 int read_size;
 int ret;
 char buf[BUF_LEN];
 read_size = read(sock, buf, sizeof(buf)-1);
 if ( read_size == -1 ){
 perror("read");
 } else if ( read_size == 0 ){
 printf("%s (%s) ポート %d ディスクリプタ %d 番からの接続が切れました。\n",
 client_info[sock].hostname,
 client_info[sock].ipaddr,
 client_info[sock].port,
 sock);
 ret = close(sock);
 if ( ret == -1 ){
 perror("close");
 exit(1);
 }
 } else {
 printf("%s (%s) ポート %d ディスクリプタ %d 番からのメッセージ: %.*s", 
 client_info[sock].hostname,
 client_info[sock].ipaddr,
 client_info[sock].port,
 sock,
 read_size,
 buf);
 write(sock, buf, read_size);
 }
}
int
main(){
 int sock_optval = 1;
 int ret;
 int port = 5000;
 int listening_socket;
 struct sockaddr_in sin;
 /* リスニングソケットを作成 */
 listening_socket = socket(AF_INET, SOCK_STREAM, 0);
 /* ソケットオプション設定 */
 if ( setsockopt(listening_socket, SOL_SOCKET, SO_REUSEADDR,
 &sock_optval, sizeof(sock_optval)) == -1 ){
 perror("setsockopt");
 exit(1);
 }
 /* アドレスファミリ・ポート番号・IPアドレス設定 */
 sin.sin_family = AF_INET;
 sin.sin_port = htons(port);
 sin.sin_addr.s_addr = htonl(INADDR_ANY);
 ret = bind(listening_socket, (struct sockaddr *)&sin, sizeof(sin));
 if ( ret == -1 ){
 perror("bind");
 exit(1);
 }
 ret = listen(listening_socket, SOMAXCONN);
 if ( ret == -1 ){
 perror("listen");
 exit(1);
 }
 printf("ポート %d を見張ります。\n", port);
 int kq;
 struct kevent kev;
 kq = kqueue();
 if ( kq == -1 ){
 perror("kqueue");
 exit(1);
 }
 EV_SET(&kev, listening_socket, EVFILT_READ, EV_ADD, 0, 0, NULL);
 ret = kevent(kq, &kev, 1, NULL, 0, NULL);
 if ( ret == -1 ){
 perror("kevent");
 exit(1);
 }
 while (1){
 int n;
 struct timespec waitspec; /* kevent に待ち時間を指定するための構造体 */
 waitspec.tv_sec = 2; /* 待ち時間に 2.500 秒を指定 */
 waitspec.tv_nsec = 500000;
 n = kevent(kq, NULL, 0, &kev, 1, &waitspec);
 if ( n == -1 ){
 perror("kevent");
 exit(1);
 } else if ( n> 0 ){
 /* 読み込み可能なソケットが存在する */
 if ( kev.ident == listening_socket ){
 /* 新しいクライアントがやってきた */
 int new_sock = accept_new_client(kev.ident);
 if ( new_sock != -1 ){
 /* 監視対象に新たなソケットを追加 */
 EV_SET(&kev, new_sock, EVFILT_READ, EV_ADD, 0, 0, NULL);
 n = kevent(kq, &kev, 1, NULL, 0, NULL);
 if ( n == -1 ){
 perror("kevent");
 exit(1);
 }
 }
 } else {
 int sock = kev.ident;
 printf("ディスクリプタ %d 番が読み込み可能です。\n", sock);
 read_and_reply(sock);
 }
 }
 }
 
 close(listening_socket);
 return 0;
}
</div><div class="naked_ctrl">
<form action="/index.cgi/contrast" method="get" name="gate">
<p><a href="http://altstyle.alfasado.net">AltStyle</a> によって変換されたページ <a href="http://x68000.q-e-d.net/~68user/net/sample/echo-server-kqueue-1.c">(-&gt;オリジナル)</a>
/ <label>アドレス: <input type="text" name="naked_post_url" value="http://x68000.q-e-d.net/~68user/net/sample/echo-server-kqueue-1.c" size="22" /></label> <label>モード: <select name="naked_post_mode">
<option value="default">デフォルト</option>
<option value="speech">音声ブラウザ</option>
<option value="ruby">ルビ付き</option>
<option value="contrast" selected="selected">配色反転</option>
<option value="larger-text">文字拡大</option>
<option value="mobile">モバイル</option>
</select>
<input type="submit" value="表示" />
</p>
</form>
</div>