From c5e64eeb973ad041c4d00cd7ef83cd044c4fcc6b Mon Sep 17 00:00:00 2001
From: matz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Tue, 25 Sep 2007 12:45:03 +0000
Subject: * io.c (io_ungetc): reallocate internal buffer if pushing data  
 excess capacity.  [ruby-dev:31650]

git-svn-id: http://svn.ruby-lang.org/repos/ruby/trunk@13512 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 io.c | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

(limited to 'io.c')

diff --git a/io.c b/io.c
index 944a4f374..134b35c0b 100644
--- a/io.c
+++ b/io.c
@@ -296,20 +296,17 @@ io_ungetc(VALUE str, rb_io_t *fptr)
 	    fptr->rbuf_capa = 8192;
         fptr->rbuf = ALLOC_N(char, fptr->rbuf_capa);
     }
-    if (fptr->rbuf_off == 0) {
-        if (fptr->rbuf_len) {
-            MEMMOVE(fptr->rbuf+len, fptr->rbuf, char, fptr->rbuf_len);
-	}
-        fptr->rbuf_off = len;
-    }
-    else if (fptr->rbuf_off < len) {
+    if (fptr->rbuf_off < len) {
 	int capa = fptr->rbuf_len + len;
 	char *buf = ALLOC_N(char, capa);
 
         if (fptr->rbuf_len) {
             MEMMOVE(buf+len, fptr->rbuf+fptr->rbuf_off, char, fptr->rbuf_len);
 	}
+	fptr->rbuf_capa = capa;
 	fptr->rbuf_off = len;
+	free(fptr->rbuf);
+        fptr->rbuf = buf;
     }
     fptr->rbuf_off-=len;
     fptr->rbuf_len+=len;
-- 
cgit