RubyでシンプルなHTTPサーバを作る

RubyでHTTPクライアントを書いてみる の兄弟記事です。Rubyのsocketライブラリを使って、シンプルなHTTPサーバを書いてみましょう。

HTTPサーバの前に超シンプルなTCPサーバを書いてみる

以下のようなスクリプトを作成し、実行してみましょう(あるいは、irb上に貼り付けてみましょう)。

require 'socket'

server = TCPServer.new 2000
i = 0
loop do
  client = server.accept
  p client
  client.puts "Hello #{Time.now} counter:#{i}"
  client.close
  i += 1
end

別の端末を立ち上げて、以下のコマンドを実行してみましょう。複数回実行すると、最後のカウンタが更新されていくことも確認できるかと思います。

$ nc 127.0.0.1 2000

このままでは、クライアントからのメッセージをサーバで確認することができないので、もう少し改造します。

シンプルなTCPサーバをRubyで書いてみる

TCPサーバのコードを以下のように変更してみましょう (while buffer = client.gets あたりを追加しました)

require 'socket'

server = TCPServer.new 2000
i = 0
loop do
  client = server.accept
  p client
  while buffer = client.gets
    puts buffer
  end
  client.puts "Hello #{Time.now} counter:#{i}"
  client.close
  i += 1
end

そして、クライアント側の実行も、以下のように変更してみます。

$ echo hello | nc 127.0.0.1 2000

このようにして、TCPリクエストに応答するシンプルなTCPサーバを作ることができました。

シンプルなHTTPサーバをRubyで書いてみる

ここまではTCPのリクエスト/レスポンスの話でした。このままではHTTPクライアントの求めるHTTPサーバの振る舞いになっていないので、もう少し改造を進めてみます。

require 'socket'

server = TCPServer.new 2000
loop do
  client = server.accept
  headers = []
  while header = client.gets
    break if header.chomp.empty?
    headers << header.chomp
  end
  p headers

  client.puts "HTTP/1.0 200 OK"
  client.puts "Content-Type: text/plain"
  client.puts
  client.puts "message body"
  client.close
end

クライアント側は、 curl コマンドを使ってみましょう。

$ curl -v http://127.0.0.1:2000/
*   Trying 127.0.0.1...
* Connected to 127.0.0.1 (127.0.0.1) port 2000 (#0)
> GET / HTTP/1.1
> Host: 127.0.0.1:2000
> User-Agent: curl/7.43.0
> Accept: */*
>
* HTTP 1.0, assume close after body
< HTTP/1.0 200 OK
< Content-Type: text/plain
<
message body
* Closing connection 0

その時のサーバ側の出力も確認してみましょう。

$ ruby server.rb
["GET / HTTP/1.1", "Host: 127.0.0.1:2000", "User-Agent: curl/7.43.0", "Accept: */*"]

curl コマンドのHTTPリクエストの内容が、RubyのHTTPサーバで受け取れていることが確認できました。

シンプルなHTTPサーバに対するDoS攻撃してみる

端末を3つ用意して、以下のコマンドを順に実行してみてください。

terminal-1$ ruby server.rb

terminal-2$ nc 127.0.0.1 2000

terminal-3$ curl -v http://127.0.0.1:2000/

nc コマンドによりTCPセッションを接続したまま放置すると、シンプルなHTTPサーバは while header = client.gets の行で待ち続けるため、次の接続 (curl によるリクエスト) を受け付けることができません。実際のインターネットサービス環境では、このような問題に対処するために、 selectを使う/Threadを使う/forkする、等の対策が行われます。ここではThreadを使う方法でこの問題への対処を行ってみましょう。

HTTPリクエストに並列で応えられるHTTPサーバをRubyで作ってみる

require 'socket'

server = TCPServer.new 2000
pids = []
loop do
  Thread.start(server.accept) do |client|
    p [Thread.current]
    headers = []
    while header = client.gets
      break if header.chomp.empty?
      headers << header.chomp
    end
    p [Thread.current, headers]

    client.puts "HTTP/1.0 200 OK"
    client.puts "Content-Type: text/plain"
    client.puts
    client.puts "message body"
    client.close
  end
end

このプログラムを使って、さきほどと同じように3つの端末でコマンドを実行してみましょう。

terminal-1$ ruby server.rb

terminal-2$ nc 127.0.0.1 2000

terminal-3$ curl -v http://127.0.0.1:2000/

すると、 server.rb を実行した端末の出力は以下のようになるはずです。

terminal-1$ ruby server.rb
[#<Thread:[e-mail]:6 run>]
[#<Thread:[e-mail]:6 run>]
[#<Thread:[e-mail]:6 run>, ["GET / HTTP/1.1", "Host: 127.0.0.1:2000", "User-Agent: curl/7.43.0", "Accept: */*"]]

最初の #<Thread:0x007fb8e184dea0> は、HTTPヘッダを受信することなく待ち受け続けています。その後にcurlによって新たなスレッド#<Thread:0x007fb8e184da40>が作成され、HTTPヘッダを受け取った後に応答を返しています。

まとめ

あとで書く