From 3de06974e874f948e9b4edc014d027e3f64a49ca Mon Sep 17 00:00:00 2001
From: Charles Kerr <charles@transmissionbt.com>
Date: Fri, 29 Feb 2008 17:09:33 +0000
Subject: [PATCH] #727: Download speed setting also affects upload speed, peer
 informations in torrent inspector

---
 libtransmission/peer-msgs.c | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/libtransmission/peer-msgs.c b/libtransmission/peer-msgs.c
index 9d04ef9f8..b94553c35 100644
--- a/libtransmission/peer-msgs.c
+++ b/libtransmission/peer-msgs.c
@@ -1571,23 +1571,26 @@ static ReadState
 canRead( struct bufferevent * evin, void * vmsgs )
 {
     ReadState ret;
-    tr_peermsgs * msgs = (tr_peermsgs *) vmsgs;
-    struct evbuffer * inbuf = EVBUFFER_INPUT ( evin );
-    const size_t buflen = EVBUFFER_LENGTH( inbuf );
-    const size_t downloadMax = getDownloadMax( msgs );
-    const size_t n = MIN( buflen, downloadMax );
+    tr_peermsgs * msgs = vmsgs;
+    struct evbuffer * in = EVBUFFER_INPUT ( evin );
+    const size_t inlen = EVBUFFER_LENGTH( in );
 
-    if( !n )
+    if( !inlen )
     {
         ret = READ_DONE;
     }
+    else if( msgs->state == AWAITING_BT_PIECE )
+    {
+        const size_t downloadMax = getDownloadMax( msgs );
+        const size_t n = MIN( inlen, downloadMax );
+        ret = n ? readBtPiece( msgs, in, n ) : READ_DONE;
+    }
     else switch( msgs->state )
     {
-        case AWAITING_BT_LENGTH:  ret = readBtLength ( msgs, inbuf, n ); break;
-        case AWAITING_BT_ID:      ret = readBtId     ( msgs, inbuf, n ); break;
-        case AWAITING_BT_MESSAGE: ret = readBtMessage( msgs, inbuf, n ); break;
-        case AWAITING_BT_PIECE:   ret = readBtPiece  ( msgs, inbuf, n ); break;
-        default: assert( 0 );
+        case AWAITING_BT_LENGTH:  ret = readBtLength ( msgs, in, inlen ); break;
+        case AWAITING_BT_ID:      ret = readBtId     ( msgs, in, inlen ); break;
+        case AWAITING_BT_MESSAGE: ret = readBtMessage( msgs, in, inlen ); break;
+        default:                  assert( 0 );
     }
 
     return ret;