(注意)このブログは
本家のほうの文章部分のみの転載です．ソースコードの配布，画像などについては本家のほうを参照してください．文章中のリンク先は面倒なのですべて本家のほうに変換してしまっているのでご注意ください．
そろそろOSとしてもそこそこ動くようになってきたので，単なる実験用のサンプルプログラムでなく，ちょっとした実用的なそれっぽいサービスを動かしてみたい．ということで，今回は telnetd と httpd と，あと時計アプリ(１秒おきに時刻を表示するだけだけど)を動かしてみよう．
で，実装なのだけど，ソケット開いて bind() して accept() して単に select()するだけ，ってだけでは芸が無い．サーバアプリを２つも動かすのだから，ソケットの select() 部分はうまく共通化したい．
とりあえず簡単に思い付くのは，kz_timer()使って一定時間ごとに select() でソケットの状態をポーリングして，受信データがあるならば，当該のスレッドに kz_send() でメッセージを送る，というものだ．しかしポーリングするというのはいまいちなあ．．．ちなみに前回まででとくに言及はしなかったのだが，main.c の内部ではwhile (1) で待ってしまっていたため，CPU負荷がやたら高くなってしまう，という問題があった．なにもやることが無いときには while (1) に入ってきて，無限ループをえんえんと繰り返しているということだ．
せっかく select() を使うならば，ソケット待ちでスリープすることでCPU負荷を一気に下げたいところだ．ということで，こーいう実装はどうだろう．
- ソケット監視スレッド(extintr)を作成し，select()によってソケットを監視させる．
- 各スレッドは，extintr に見張ってほしいソケットをメッセージによってextintr に通知する．extintr は accept() 待ちと read() 待ちを select() することができる．
- extintr はタイムアウト無限で select() する．このため，受信データが無い場合にはプロセスはスリープしているのでCPU負荷は上がらない．
- データが到着したら select() から抜け，read() もしくは accept() を行い，当該のスレッドに受信データをメッセージで投げる．で，各スレッドは受信データに対して適切な処理を行う．
- extintr に無闇に処理が渡ってしまうと，select()無限待ちに入って他スレッドが固まってしまうことになる(ていうかKOZOS全体が固まる)．これを防ぐために，extintr は優先度を最低にしておく．(extintr に処理が渡るときは，他のスレッドはすべてスリープ状態にある場合なので，これで問題無い)
- タイマ割り込み発生時には，SIGALRMによって select() は抜ける．この場合，タイマを仕掛けているスレッドにKOZOSからメッセージが渡され，そのスレッドが動作を終えると，extintr は再び select()無限待ちに入ることになる．
ほんとは外部割り込みっぽく実装してみたいのだけど(このため extintr という名前になっている)，とりあえずサーバ動かしてみたいし，いちどにあれもこれも説明するのもたいへんなので，とりあえず今回はこれでいいとしよう．
で，修正後のソースは以下になる．今回は extintr.c, clock.c, telnetd.c, httpd.c が追加されている．
以下は前回からの差分．
ちなみに今回，システムコールとして kz_pending() というのを追加してある．受信メッセージがあるかどうかを調べるためのシステムコールだ．こんなふうにして使う．
if (kz_pending()) { /* 受信メッセージがあるならば，受信して処理する */
size = kz_recv(&id, &p);
...
で，各ソースを説明しよう．まずは extintr.c だ．
まず extintr.c は intrbuf という構造体と，setintrbuf() という関数を持つ．これらは各スレッドからソケットを通知された際に，それらのソケットをリンクリスト管理しておくためのものだ．まああまり難しくはないと思うので，ここでは説明しない．
extintr スレッドのメインループは，以下のようになっている．
int extintr_main(int argc, char *argv[])
{
fd_set fds;
char *p, *buffer;
int fd, id, ret, size, s, len;
intrbuf *ibp;
FD_ZERO(&readfds);
maxfd = 0;
while (1) {
while (kz_pending()) {
fd = kz_recv(&id, &p);
if (fd) { /* from thread */
setintrbuf(fd, id, (struct sockaddr *)p);
}
}
fds = readfds;
ret = select(maxfd + 1, &fds, NULL, NULL, NULL);
if (ret > 0) {
for (ibp = interrupts; ibp; ibp = ibp->next) {
if (FD_ISSET(ibp->fd, &fds)) {
switch (ibp->type) {
case INTR_TYPE_ACCEPT:
len = ibp->addr.sa_len;
s = accept(ibp->fd, &ibp->addr, &len);
if (s > 0)
kz_send(ibp->id, s, NULL);
break;
case INTR_TYPE_READ:
buffer = kz_memalloc(BUFFER_SIZE);
size = read(ibp->fd, buffer, BUFFER_SIZE);
if (size > 0)
kz_send(ibp->id, size, buffer);
break;
}
}
}
}
}
}
まず最初に，
while (1) {
while (kz_pending()) {
fd = kz_recv(&id, &p);
if (fd) { /* from thread */
setintrbuf(fd, id, (struct sockaddr *)p);
}
}
という部分で，各スレッドからのメッセージを受け付ける．各スレッドからは，見張ってほしいソケットをメッセージで送ってくるので，それを受信して setintrbuf() を呼び出し，構造体 intrbuf のリンクリストに追加する．ちなみにソケット情報は，
- accept()のためのソケット ... kz_send() のサイズの位置にソケット番号，データの位置に bind() したときのアドレス情報を入れてメッセージが送られる．
- read()のためのソケット ... kz_send() のサイズの位置にソケット番号，データの位置に NULL を入れてメッセージが送られる．
というようにして送られてくる．なので setintrbuf() では，データとしてポインタが送られてきたかNULLが送られてきたかで，accept() 用のソケットなのかread() 用のソケットなのかを判断している．
kz_send() の戻り値がサイズではなくソケット番号になっていることに注意．前回だか前々回だかに説明したが，kz_send() ではパラメータ名は size となっているが，実際には指定されたパラメータ値をそのまま送るだけなので，このようにソケット番号を送ることに利用してもいいわけだ(本当は通信用の専用の構造体を作るべきだろうが，面倒なのでまあこれでいいとする)．
さらに，select() によって無限待ちする．
fds = readfds;
ret = select(maxfd + 1, &fds, NULL, NULL, NULL);
select()から抜けるのは，ソケットがなんらかのデータを受信して読み込み可能になったか，もしくはシグナル(ここでは SIGALRM のみ)が発生したかだ．読み込むデータがある場合には select() の戻り値が正の数(読み込めるソケットの個数)になるので，受信処理を行う．
if (ret > 0) {
for (ibp = interrupts; ibp; ibp = ibp->next) {
if (FD_ISSET(ibp->fd, &fds)) {
switch (ibp->type) {
case INTR_TYPE_ACCEPT:
len = ibp->addr.sa_len;
s = accept(ibp->fd, &ibp->addr, &len);
if (s > 0)
kz_send(ibp->id, s, NULL);
break;
case INTR_TYPE_READ:
buffer = kz_memalloc(BUFFER_SIZE);
size = read(ibp->fd, buffer, BUFFER_SIZE);
if (size > 0)
kz_send(ibp->id, size, buffer);
break;
}
}
}
}
ソケットは accept() 用のものと，read() 用のものがある．これらの種類に応じて accept() もしくは read() を行い，実行結果を当該のスレッドにメッセージにして送っている．select()でチェックした後なので，accept(),read()は必ず成功するはずだ．にもかかわらず accept(),read() の戻り値チェックをしているのは，万が一これらと同時に SIGALRM が発生し，(そのようなことがあり得るのかどうかちょっと不明だけど)シグナル抜けしてしまったときの対処だ．
では次に，各種アプリについて説明しよう．まずは時計アプリである clock.c から．
int clock_main(int argc, char *argv[])
{
time_t t;
char *p;
while (1) {
kz_timer(1000);
kz_recv(NULL, NULL);
t = time(NULL);
p = kz_memalloc(128);
strcpy(p, ctime(&t));
kz_send(outlog_id, 0, p);
}
}
メインループでは，単にタイマを１秒おきでかけて時刻を表示しているだけ．表示は outlog スレッドにまかせている．
次に telnetd.c について．
int telnetd_main(int argc, char *argv[])
{
int sockt, s, ret;
char hostname[256];
struct hostent *host;
struct sockaddr_in address;
int backlog = 5;
#if 1
gethostname(hostname, sizeof(hostname));
#else
strcpy(hostname, "localhost");
#endif
host = gethostbyname(hostname);
if (host == NULL) {
fprintf(stderr, "gethostbyname() failed.\n");
exit (-1);
}
memset(&address, 0, sizeof(address));
address.sin_family = AF_INET;
address.sin_len = sizeof(address);
address.sin_port = htons(PORT);
memcpy(&(address.sin_addr), host->h_addr, host->h_length);
sockt = socket(AF_INET, SOCK_STREAM, 0);
if (sockt < 0) {
fprintf(stderr, "socket() failed.\n");
exit (-1);
}
ret = bind(sockt, (struct sockaddr *)&address, sizeof(address));
if (ret < 0) {
fprintf(stderr, "bind() failed.\n");
exit (-1);
}
ret = listen(sockt, backlog);
if (ret < 0) {
fprintf(stderr, "listen() failed.\n");
exit (-1);
}
ここまでは通常のネットワークアプリなので，とくに説明はない．ソケットを開いて各種準備をするだけだ．
で，問題は次からだ．
kz_send(extintr_id, sockt, (char *)&address);
while (1) {
s = kz_recv(NULL, NULL);
kz_run(command_main, "command", 7, s, NULL);
}
close(sockt);
return 0;
}
listen()の直後に kz_send() によって extintr にメッセージを送る．これにより，accept() 用のソケット番号とアドレス情報を extintr に通知し，あとは extintr が select() でソケットを見張って，accept() できる状態(telnetに対する接続要求が来たとき)になったらメッセージを投げてくる．
kz_send() の後は kz_recv() でメッセージ待ちに入る．extintr からのメッセージを待つわけだ．で，接続要求が来たら extintr のほうで accept() してソケット番号を(kz_send()の size パラメータを使って)送ってくるので，そのソケット番号をパラメータにしてコマンド処理用スレッドを起動する．ちなみに telnetd はこの後再び kz_recv() による accept() 待ちに入り，接続のたびにコマンドスレッドを新規に起動するので，複数の telnet 接続を受け付けることができる．
コマンドスレッドのメインループは以下のようになっている．
static int command_main(int s, char *argv[])
{
char *p;
char buffer[128];
int len, size;
kz_send(extintr_id, s, NULL);
len = 0;
write(s, "> ", 2);
while (1) {
size = kz_recv(NULL, &p);
memcpy(buffer + len, p, size);
kz_memfree(p);
len += size;
buffer[len] = '\0';
p = strchr(buffer, '\n');
if (p == NULL) continue;
if (!strncmp(buffer, "echo", 4)) {
write(s, buffer + 4, strlen(buffer + 4));
} else if (!strncmp(buffer, "date", 4)) {
time_t t;
t = time(NULL);
strcpy(buffer, ctime(&t));
write(s, buffer, strlen(buffer));
} else if (!strncmp(buffer, "threads", 7)) {
kz_thread *thp;
int i;
for (i = 0; i < THREAD_NUM; i++) {
thp = &threads[i];
if (!thp->id) break;
write(s, thp->name, strlen(thp->name));
write(s, "\n", 1);
}
} else if (!strncmp(buffer, "exit", 4)) {
break;
}
len = 0;
write(s, "OK\n> ", 5);
}
kz_send(extintr_id, s, NULL);
close(s);
return 0;
}
順番に説明していこう．まず，先頭で kz_send() により extintr にメッセージを送っている．
kz_send(extintr_id, s, NULL);
これはソケットを extintr に登録しておいて，受信データがあったときにやはりメッセージで知らせてもらうためだ．で，kz_recv() により受信データ待ちになる．
while (1) {
size = kz_recv(NULL, &p);
先程の kz_send() で extintr に対してソケット番号を通知してあるので，あとは extintr が select() して，受信データがあれば kz_send() でメッセージにして送ってくる．
で，メッセージを受信したら，受信データの内容に応じてコマンドを実行する．ちなみに受信データは extintr 側で kz_memalloc() によって獲得したメモリに格納して送られてくるので，コマンド処理側で kz_memfree() によって解放している．
memcpy(buffer + len, p, size);
kz_memfree(p);
len += size;
buffer[len] = '\0';
で，以下がコマンド処理部分になる．
if (!strncmp(buffer, "echo", 4)) {
write(s, buffer + 4, strlen(buffer + 4));
} else if (!strncmp(buffer, "date", 4)) {
time_t t;
t = time(NULL);
strcpy(buffer, ctime(&t));
write(s, buffer, strlen(buffer));
} else if (!strncmp(buffer, "threads", 7)) {
kz_thread *thp;
int i;
for (i = 0; i < THREAD_NUM; i++) {
thp = &threads[i];
if (!thp->id) break;
write(s, thp->name, strlen(thp->name));
write(s, "\n", 1);
}
} else if (!strncmp(buffer, "exit", 4)) {
break;
}
len = 0;
write(s, "OK\n> ", 5);
}
現在実装してあるコマンドは以下の４つだけだ．
- echo ... 指定した文字列をリピート表示する．
- date ... 時刻を表示する．
- threads ... 起動中のスレッド一覧を表示する．
- exit ... telnet接続を終了する．
exitが打たれると，以下の終了処理を行う．
kz_send(extintr_id, s, NULL);
close(s);
kz_send()により extintr に対してもう一度ソケット番号を通知することで，そのソケットを監視対象から外す(extintr はメッセージを受けるとソケットの監視を開始し，もう一度メッセージを受けると監視から外す)．で，ソケットをクローズしている．
さて次は httpd だ．まあ telnetd とあんまし変わらないのだが，異なる部分だけいちおう説明する．
httpdのメインループは，telnetdとほとんど同じだ．接続要求がくると処理用のスレッドとして以下を起動する．
static int http_proc(int s, char *argv[])
{
char *p;
char buffer[1024];
int len, size;
kz_send(extintr_id, s, NULL);
len = 0;
do {
size = kz_recv(NULL, &p);
if (len + size >= sizeof(buffer)) {
break;
}
memcpy(buffer + len, p, size);
kz_memfree(p);
len += size;
buffer[len] = '\0';
} while (strchr(buffer, '\n') == NULL);
write(s, mes, sizeof(mes));
kz_send(extintr_id, s, NULL);
close(s);
return 0;
}
HTTPのプロトコルでは，たしか要求の最後に改行が入っていたような．．．(あれ，空行だったかな？)，なので，てきとうに要求を受けた後にHTMLのメッセージを返すようになっている．メッセージは以下の簡単なものだ．
static char mes[] =
"<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 1.0//EN\">\n"
"<html>\n"
"<head><title>hello</title></head>\n"
"<body>\n"
"<center><h1>KOZOS</h1></center>\n"
"<p><h2>This is HTTP server running on KOZO-OS.</h2>\n"
"</body></html>\n";
メインの関数は以下になる．
ここで，各スレッドの起動部分について説明する．
int mainfunc(int argc, char *argv[])
{
extintr_id = kz_run(extintr_main, "extintr", 31, 0, NULL);
outlog_id = kz_run(outlog_main, "outlog", 1, 0, NULL);
clock_id = kz_run(clock_main, "clock", 7, 0, NULL);
telnetd_id = kz_run(telnetd_main, "telnetd", 8, 0, NULL);
httpd_id = kz_run(httpd_main, "httpd", 9, 0, NULL);
return 0;
}
extintr を一番最初に起動しているのだが，優先度は31と最低になっていることに注意してほしい．優先度が31になっているのはずっと上のほうで説明したように，内部で select() で待つので，他のスレッドが固まってしまうのを防止するためだ．まあ優先度が最低ということは，なにもやることがないときには extintr に処理が渡って select() でプロセス自体がスリープするということなので，これはけっこうまともな設計だと思う．(実際のOSでも，多くは idle スレッドみたいなものが最低の優先度で起動していて，なにもやることがない場合にはCPUを省電力モードにする，などの動作をする)
telnetd, httpd は extintr に対してメッセージを投げるので，telnetd, httpd が起動する前に extintr は存在していなければならない．なので extintr を一番最初に起動している(ただし優先度が最低なので，起動しただけで処理は何も行われておらず，存在しているだけの状態である)．outlog を２番目に起動しているのも，まあ似たような理由だ．clock は outlog に対してメッセージを送信するので，clock よりも前にoutlog が起動している必要がある．ちなみに前回や前々回とかでは，メインとなるスレッドは，他のスレッドを全部起動した後に自身の優先度を最低に変更して while(1) で無限ループしていた．つまり，最後に idle スレッドになっていた(しかし無限ループなのでCPU負荷が異常に高くなっていた)．しかし今回は extintr が select() で待つことでidleスレッドの役割をこなすので，メインスレッドはすべてのスレッドの起動後はそのまま終了してしまっている．(メインスレッドは優先度がゼロで一番高いので，メインスレッドの終了後に各スレッドが動作を開始することになる)
では，いよいよサンプルプログラムによって動作確認してみよう．で，動作確認の前に１点注意．今回は telnetd と httpd が gethostbyname() によってホスト情報を取得しているが，ここでネットワーク接続されていない端末(もしくは接続されていても)でホスト名がてきとうだったりすると，DNSから情報が得られなくてエラーになってしまう．対策としては，/etc/hosts にホスト名とIPアドレスの対応を書いて名前解決できるようにしておくか，もしくはtelnetd_main(), httpd_main()でgethostname()しているところを
-#if 1
+#if 0
gethostname(hostname, sizeof(hostname));
#else
strcpy(hostname, "localhost");
#endif
のように書き換えて localhost を使うようにするか，対処してほしい．
では起動してみよう．コンパイルしたら実行形式を起動する．
% ./koz
Tue Oct 23 22:32:08 2007
Tue Oct 23 22:32:09 2007
Tue Oct 23 22:32:10 2007
Tue Oct 23 22:32:11 2007
Tue Oct 23 22:32:12 2007
Tue Oct 23 22:32:13 2007
Tue Oct 23 22:32:14 2007
Tue Oct 23 22:32:15 2007
Tue Oct 23 22:32:16 2007
Tue Oct 23 22:32:17 2007
Tue Oct 23 22:32:18 2007
Tue Oct 23 22:32:19 2007
...
時刻が１秒おきに表示されている．
この状態で，telnet で接続してみよう．説明を忘れていたが，telnetd.c ではポート番号 20001 で待っているので，20001 番ポートに対して接続する．前述したようにDNSでうまく名前解決できなければ，localhost を使うかIPアドレス直書きでもいいだろう．ここではIPアドレス直書きで接続する．
% telnet 192.168.0.3 20001
Trying 192.168.0.3...
Connected to 192.168.0.3.
Escape character is '^]'.
>
おお，無事に接続できた．
echoコマンドを実行してみよう．
> echo test
test
OK
問題ないようだ．
> date
Tue Oct 23 22:32:50 2007
OK
> threads
command
extintr
outlog
clock
telnetd
httpd
OK
>
date, threads コマンドも問題無し．
前述したように，telnetd は接続ごとに処理用スレッドを起動するので，この状態(すでに接続されている状態)で，抜けないままでさらに別の接続をすることができる．試してみよう．
別の kterm 上から，telnet を起動する．
% telnet 192.168.0.3 20001
Trying 192.168.0.3...
Connected to 192.168.0.3.
Escape character is '^]'.
>
問題なく接続できた．
コマンドを実行してみよう．
> date
Tue Oct 23 22:33:11 2007
OK
> threads
command
extintr
outlog
clock
telnetd
httpd
command
OK
>
threads によるスレッド一覧表示で，command スレッドが２つ表示されていることに注目(さっきはひとつだけだった)．２箇所から同時接続して，コマンド処理スレッドが２つ起動しているので，このようなことになっている．
次に，ブラウザでHTTP接続してみよう．こちらはポート番号は 30001 (httpd.c参照)なので，
http://192.168.0.3:30001/
として接続する．
画像はこちらおー，問題無く表示された．ちょっと感動．この間も時刻は１秒ごとに表示されているし，telnetによるコマンド実行ももちろんできる．時刻表示やtelnetによるコマンド処理も行いながら，webサーバも同時にできているということに注目してほしい．つまり，スレッドのディスパッチがきちんと行われているということだ．
telnet接続は，exitで抜けることができる．
> exit
Connection closed by foreign host.
%
さて，ここまでで，各種サーバ機能と時刻表示を動かすことができたのだが，たとえばこれをスレッド構成にしないで，普通の作りにしたらどんな感じになるか考えてみてほしい．
まあ，いちばん最初に思い付くのは
signal(SIGALRM, alarm_handler);
...
while (1) {
...
r = select(...);
if (r < 0) {
/* シグナルによってselect()が終了した(※１) */
...
} else if (r == 0) {
/* select()がタイムアウトした */
...
} else {
/* ソケットに受信データが到着した */
for (i = 0; i <= maxfd; i++) {
if (FD_ISSET(i, &fds)) {
/* ソケットの受信処理(※２) */
...
}
みたいにして，select()で待ってその戻り値に応じて各種動作を行うというものだ．これでもたしかに，やりたいことはできるだろう．
しかしこの書き方だと，時刻表示とtelnetサーバとhttpサーバの処理を，完全に独立させて書くことはまあできない．それらはすべて select() から派生して処理が行われるような構造になるからだ．つまり，上記の(※１)や(※２)の位置から関数を呼び出すことで各種処理を行わなければならない．このため，関数の構造などは select() から呼び出せるようにしなければならず，それなりに制約を受ける．たとえば今回の telnetd.c のように，accept() したらコマンド処理してまたaccept() の無限ループ，のようなことはできない．無限ループしてしまっては，select() に処理が戻らずに，他の動作が固まってしまうからだ．
これに対して，今回の clock.c, telnetd.c, httpd.c の実装を見てほしい．まるでひとつひとつが，単なるアプリとなっていて，select() から派生して動くとかいったことは考える必要が無い．extintr にメッセージを投げておけば，あとは適切にメッセージが返ってくるので，それに対して処理を行うだけだ．telnetd.c の中では無限ループにより accept() を行っているが，無限ループのために他のサービスが固まってしまうだとかいったことを心配する必要は無い．OSがてきとうにスレッドディスパッチをして，処理を振り分けてくれるからだ．
ここで，たとえばスレッドを用いずに select() のみで処理するような実装で，telnet 接続したときに，あるコマンドを叩いたら複雑な時間のかかる処理を行うような場合を考えてみよう．たとえばソートとかだ．
ソートに10秒かかるとして，それをコマンドスレッドが何も考えずに実行してしまったら，10秒間 select() には戻ってこない．つまり，他の全てのサービスが固まってしまう．これを防ぐためには，ソート処理の途中でいったん抜けてselect() に戻り，しばらくしたらまたソート処理を呼んでもらってまた途中まで処理して，というような複雑なことを行わなければならない．このためにはソートの現在状態のセーブ／ロード処理を実装しなければならなくなる．これは面倒だし，なによりも本来のソート処理よりも，このような本筋ではない対応のほうに手間がかかってしまうことになる．場合によってはプログラムの構造自体を変更しなければならなくなる．
しかし今回のように，スレッド化してあれば簡単だ．簡単，というより，何も考えずにソート処理を行えばよい．とはいってもまあ実は現状のKOZOSの実装だと，コマンドで重い処理に入るとselect() が呼ばれずに同じようなことになってしまう．とりあえず重い処理をしたいならば，とりあえずはソート処理の要所に
kz_timer(10);
kz_recv(NULL, NULL);
とか入れておく必要がある．まあこれでもプログラム全体の構成を変える必要は無いわけで，とっても嬉しいともいえる．
この問題には，外部割り込み実装という解決策がある．まあこれはまた後で紹介しよう．
