[ruby-dev:26540] Re: WEBrick DoS vulnerability

In article <[e-mail]>,
  GOTOU Yuuzou <[e-mail]> writes:

> sysreadを使っているnet/protocol.rbやnet/telnet.rbなどに修正
> が必要になりそうですね。添付ライブラリはどうとでもなるのでしょ
> うが、ユーザのコードはそうではないので、デフォルトで
> non-blocking modeにするのはよしたほうがいいと思います。

前田さんの [ruby-dev:26492] の案で IO#read 以外のメソッドにも手を入れ
ることにすればデフォルトで non-blockingでもユーザのコードからはその違
いが見えないようにすることは可能でしょ。問題は、洩れなく完璧にできるの
かというところと、そこまでするのはやりすぎな感じがする、というところで
すが。

私は 1.8 でデフォルトで non-blocking にするのはお薦めしないのですが、
それでもできればデフォルトで write がブロックしないようにできればいい
とは思っているわけです。で、思い付いて [e-mail] のほうに
送ったのが、write を PIPE_BUF 毎に分割することによって non-blocking に
しなくても、環境によってはブロックしなくなるかもしれないという案ですが、
反応がありませんねぇ。

Index: io.c
===================================================================
RCS file: /src/ruby/io.c,v
retrieving revision 1.246.2.74
diff -u -p -r1.246.2.74 io.c
--- io.c	12 May 2005 08:12:39 -0000	1.246.2.74
+++ io.c	15 Jul 2005 00:03:44 -0000
@@ -394,7 +394,7 @@ io_fwrite(str, fptr)
     VALUE str;
     OpenFile *fptr;
 {
-    long len, n, r, offset = 0;
+    long len, n, r, l, offset = 0;
     FILE *f = GetWriteFile(fptr);
 
     len = RSTRING(str)->len;
@@ -404,8 +404,24 @@ io_fwrite(str, fptr)
 	if (!rb_thread_fd_writable(fileno(f))) {
 	    rb_io_check_closed(fptr);
 	}
+        if (!(fptr->mode & FMODE_WSPLIT_INITIALIZED)) {
+            struct stat buf;
+            if (fstat(fileno(f), &buf) == 0 &&
+                !(S_ISREG(buf.st_mode) ||
+                  S_ISDIR(buf.st_mode)) &&
+                (r = fcntl(fileno(f), F_GETFL)) != -1 &&
+                !(r & O_NONBLOCK)) {
+                fptr->mode |= FMODE_WSPLIT;
+            }
+            fptr->mode |= FMODE_WSPLIT_INITIALIZED;
+        }
       retry:
-	r = write(fileno(f), RSTRING(str)->ptr+offset, n);
+        l = n;
+        if (!rb_thread_critical && !rb_thread_alone() &&
+            (fptr->mode & FMODE_WSPLIT) && PIPE_BUF < l) {
+            l = PIPE_BUF;
+        }
+	r = write(fileno(f), RSTRING(str)->ptr+offset, l);
         if (r == n) return len;
         if (0 <= r) {
             offset += r;
Index: rubyio.h
===================================================================
RCS file: /src/ruby/rubyio.h,v
retrieving revision 1.23.2.6
diff -u -p -r1.23.2.6 rubyio.h
--- rubyio.h	21 Jan 2005 00:31:16 -0000	1.23.2.6
+++ rubyio.h	15 Jul 2005 00:03:44 -0000
@@ -39,6 +39,8 @@ typedef struct OpenFile {
 #define FMODE_SYNC      8
 #define FMODE_WBUF     16
 #define FMODE_RBUF     32
+#define FMODE_WSPLIT  0x200
+#define FMODE_WSPLIT_INITIALIZED  0x400
 
 #define GetOpenFile(obj,fp) rb_io_check_closed((fp) = RFILE(rb_io_taint_check(obj))->fptr)

> もとの話題であるWEBrickがwriteでブロック問題については、
> WEBrick側でソケットをnon-blocking modoにすることで対処しよう
> と思います。

それはそれでいいんじゃないかと思います。
(仮に上のパッチが入っても、non-blocking mode になっていれば write を分
割しなくて済むし、socket が select で writable になっても PIPE_BUF 未
満しか書き込めない環境があったとしても対応できるので)

>> % ./ruby -v
>> ruby 1.9.0 (2005-07-10) [i686-linux]
>> % ./ruby -rsocket -rio/nonblock -e '
>> s1, s2 = UNIXSocket.pair
>> s1.nonblock = true         
>> p s1.send("a" * 1000000, 0)
>> '
>> 65280
>
> バッファがいっぱいの状態でもう一度sendするとブロックするよう
> ですね。

ふむ。それは知りませんでした。
-- 
[田中 哲][たなか あきら][Tanaka Akira]