いつきログ(2011-04-08)

2011-04-08 明日・明後日休みだー！

ひゃっほーい♪

◆ [日記] angelbeatsのタグ検索で18844件@pixiv

１日で６件減り。まーた減り始めた。

実際には１１件位増え。増え方は普通。

◆ [日記][tDiary][FreeBSD][Ruby] また日記（ここ）がDDoS受けた

「この方法まずいかなー」で対処したら見事にまずかったので、色々引き下げて再起動。

この休み中にパッチ作るか。

→とりあえず、Apacheでいじればいいと思われる箇所は発見したが、リコンパイルめんどくさいのと、特定条件下で無限ループ状態に陥ることが判明した（＝＝；）ので、tDiary側にも変更を加えた。

具体的には、RLimitNProc（RLIMIT_NPROC）で制限かかってる時に、IO.popen使うと無限ループに陥る。

Linuxの場合は分からないが、FreeBSD+Apache+suexecの場合、RLimitNProcで制限加えても、とりあえずスクリプトの実行までは行ってしまうのです（まぁsuexecがrootで実行されるからしゃーないが）。で、その状態でIO.popenで新しいプロセスを立ち上げようとすると、forkが失敗してしまうのです（ちなみにFreeBSDの場合、Apacheから起動されたプロセス数じゃなくて、システム全体のプロセス数とRLIMIT_NPROCが比較されます）

で、fork失敗した時には-1が返り、errnoにはEAGAINが設定されます(On-line Manual of "fork")

で、だ。

なんで戻ってこないねん！　って思って、rubyのソースコード見てみたわけですよ。

retry:
  switch ((pid = fork())) {
(中略)
    case -1:/* fork failed */
      if (errno == EAGAIN) {
        rb_thread_sleep(1);
        goto retry;
      }
(後略)

fork失敗して、errnoがEAGAINだったら、rb_thread_sleep(1);をコールした後、goto retry。で、retryってforkの前の行にあるので、またforkしに行って……あぁ、名前的にはrb_thread_sleepは1秒現在のスレッドお休みだろうから、１秒ごとにfork失敗して無限ループになってるのね（笑）

そりゃ無限ループするわ（笑）

というわけで、timeoutを使って特定時間反応がなかったら死ぬようにしました。こんな感じ。

class IKTimeoutException < Exception; end
begin
  require 'timeout'
  cgi_exec_count = 0
  begin
    timeout(2) do
      IO.popen("/bin/pgrep -G グループID -U ユーザID -f -x '/usr/local/bin/ruby 日記のスクリプトファイル名'",'r'){ |io|
        io.each_line { |l|
          cgi_exec_count += 1
        }
      }
    end
  rescue Timeout::Error
    raise IKTimeoutException
  end
  if cgi_exec_count > 最大プロセス数 then
    raise IKTimeoutException # めんどくさいので同じロジックへ飛ばす
  end
rescue IKTimeoutException
  print "Status: 503 Service Unavailable\n"
  print "Content-Type: text/text\n\n"
  print "503 Service Unavailable\n"
  exit
rescue
  print "Status: 500 Internal Server Error\n"
  print "Content-Type: text/text\n\n"
  print "500 Internal Server Error\n"
  exit
end