From 34f5c66ba8ea9e9f345c8847d08e58dfb3ca7d7c Mon Sep 17 00:00:00 2001
From: Carlin Bingham <carlin@theintersect.org>
Date: Mon, 26 Oct 2009 11:40:40 +1300
Subject: [PATCH] Change IO::Socket.recv() to accept a parameter specifying the number of bytes which will be received

---
 src/setting/IO/Socket.pm |   22 ++++++++++++++--------
 1 files changed, 14 insertions(+), 8 deletions(-)

diff --git a/src/setting/IO/Socket.pm b/src/setting/IO/Socket.pm
index 3b175f2..2373b4b 100644
--- a/src/setting/IO/Socket.pm
+++ b/src/setting/IO/Socket.pm
@@ -2,16 +2,22 @@ use v6;
 
 role IO::Socket {
     has $!PIO;
+    has $!buffer = '';
 
-    method recv () {
+    method recv (Int $bufsize = Inf) {
         fail('Socket not available') unless $!PIO;
-        my $received = $!PIO.recv();
-        my $len = $received.chars;
-        my $buf;
-        while $len > 0 {
-            $buf = $!PIO.recv();
-            $received ~= $buf;
-            $len = $buf.chars;
+        my $received;
+        while $bufsize > $!buffer.bytes {
+            $received = $!PIO.recv();
+            last unless $received.chars;
+            $!buffer ~= $received;
+        }
+        if $bufsize == Inf {
+            $received = $!buffer;
+            $!buffer = '';
+        } else {
+            $received = $!buffer.substr(0, $bufsize);
+            $!buffer .= substr($bufsize);
         }
         return $received;
     }
-- 
1.6.0.4