From de5c84a52cad824ba5ec1a0a8b8def7971c5917d Mon Sep 17 00:00:00 2001
From: Charles Kerr <charles@transmissionbt.com>
Date: Thu, 13 Dec 2007 21:08:05 +0000
Subject: [PATCH] allocate more sockets to the tracker during shutdown to help
 get all the `stopped' commands done

---
 libtransmission/tracker.c | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/libtransmission/tracker.c b/libtransmission/tracker.c
index 956806223..f6d90f56b 100644
--- a/libtransmission/tracker.c
+++ b/libtransmission/tracker.c
@@ -257,10 +257,13 @@ publishNewPeers( tr_tracker * t, int count, uint8_t * peers )
 ****
 ***/
 
+static void onReqDone( tr_handle * handle );
+
 static void
-onStoppedResponse( struct evhttp_request * req UNUSED, void * handle UNUSED )
+onStoppedResponse( struct evhttp_request * req UNUSED, void * handle )
 {
     dbgmsg( NULL, "got a response to some `stop' message" );
+    onReqDone( handle );
 }
 
 static int
@@ -379,8 +382,6 @@ parseOldPeers( benc_val_t * bePeers, int * setmePeerCount )
     return compact;
 }
 
-static void onReqDone( tr_handle * handle );
-
 static void
 onTrackerResponse( struct evhttp_request * req, void * vhash )
 {
@@ -756,6 +757,7 @@ createScrape( tr_handle * handle, const tr_tracker * tracker )
 struct tr_tracker_handle
 {
     int socketCount;
+    unsigned int isShuttingDown : 1;
     tr_timer * pulseTimer;
     tr_list * requestQueue;
     tr_list * scrapeQueue;
@@ -783,9 +785,13 @@ freeRequest2( void * req )
 void
 tr_trackerShuttingDown( tr_handle * handle )
 {
-    /* since we're shutting down, we don't need to scrape anymore... */
     if( handle->tracker )
+    {
+        /* since we're shutting down, we don't need to scrape anymore... */
         tr_list_free( &handle->tracker->scrapeQueue, freeRequest2 );
+
+        handle->tracker->isShuttingDown = 1;
+    }
 }
 
 static int
@@ -864,7 +870,10 @@ invokeNextInQueue( tr_handle * handle, tr_list ** list )
 static int
 socketIsAvailable( tr_handle * handle )
 {
-    return handle->tracker->socketCount < MAX_TRACKER_SOCKETS;
+    int max = MAX_TRACKER_SOCKETS;
+    if( handle->tracker->isShuttingDown )
+        max *= 2;
+    return handle->tracker->socketCount < max;
 }
 
 static void ensureGlobalsExist( tr_handle * );