Changeset 4920

Timestamp:

Feb 4, 2008, 7:52:00 PM (14 years ago)

Author:

charles

Message:

make the stats code a little more difficult to corrupt

Location:

branches/1.0x/libtransmission

Files:

• stats.c (modified) (9 diffs)
• transmission.c (modified) (2 diffs)

branches/1.0x/libtransmission/stats.c

@@ -25 +25 @@
 {
     tr_session_stats single;
-    tr_session_stats cumulative;
+    tr_session_stats old;
     time_t startTime;
 };
@@ -116 +116 @@
 {
     struct tr_stats_handle * stats = tr_new0( struct tr_stats_handle, 1 );
-    loadCumulativeStats( &stats->cumulative );
-    stats->cumulative.sessionCount++;
-    stats->startTime = time(NULL);
+    loadCumulativeStats( &stats->old );
+    stats->single.sessionCount = 1;
+    stats->startTime = time( NULL );
     handle->sessionStats = stats;
 }
@@ -125 +125 @@
 tr_statsClose( tr_handle * handle )
 {
-    tr_session_stats tmp;
-    tr_getCumulativeSessionStats( handle, &tmp );
-    saveCumulativeStats( &tmp );
+    tr_session_stats cumulative;
+    tr_getCumulativeSessionStats( handle, &cumulative );
+    saveCumulativeStats( &cumulative );
 
     tr_free( handle->sessionStats );
@@ -133 +133 @@
 }
 
+static struct tr_stats_handle *
+getStats( const tr_handle * handle )
+{
+    static struct tr_stats_handle nullObject;
+
+    return handle && handle->sessionStats
+        ? handle->sessionStats
+        : &nullObject;
+}
+
+/***
+****
+***/
+
 static void
 updateRatio( tr_session_stats * setme )
 {
-    setme->ratio = tr_getRatio( setme->uploadedBytes, setme->downloadedBytes );
+    setme->ratio = tr_getRatio( setme->uploadedBytes,
+                                setme->downloadedBytes );
+}
+
+static void
+addStats( tr_session_stats       * setme,
+          const tr_session_stats * a,
+          const tr_session_stats * b )
+{
+    setme->uploadedBytes   = a->uploadedBytes   + b->uploadedBytes;
+    setme->downloadedBytes = a->downloadedBytes + b->downloadedBytes;
+    setme->filesAdded      = a->filesAdded      + b->filesAdded;
+    setme->sessionCount    = a->sessionCount    + b->sessionCount;
+    setme->secondsActive   = a->secondsActive   + b->secondsActive;
+    updateRatio( setme );
 }
 
@@ -143 +171 @@
                     tr_session_stats  * setme )
 {
-    const struct tr_stats_handle * stats = handle->sessionStats;
+    const struct tr_stats_handle * stats = getStats( handle );
     *setme = stats->single;
-    setme->secondsActive += ( time(NULL) - stats->startTime );
+    setme->secondsActive = time( NULL ) - stats->startTime;
     updateRatio( setme );
 }
@@ -153 +181 @@
                               tr_session_stats  * setme )
 {
-    const struct tr_stats_handle * stats = handle->sessionStats;
-    *setme = stats->cumulative;
-    setme->secondsActive += ( time(NULL) - stats->startTime );
-    updateRatio( setme );
+    tr_session_stats current;
+    tr_getSessionStats( handle, &current );
+    addStats( setme, &getStats(handle)->old, &current );
 }
 
@@ -166 +193 @@
 tr_statsAddUploaded( tr_handle * handle, uint32_t bytes )
 {
-    struct tr_stats_handle * stats = handle->sessionStats;
-    stats->single.uploadedBytes += bytes;
-    stats->cumulative.uploadedBytes += bytes;
+    getStats(handle)->single.uploadedBytes += bytes;
 }
 
@@ -174 +199 @@
 tr_statsAddDownloaded( tr_handle * handle, uint32_t bytes )
 {
-    struct tr_stats_handle * stats = handle->sessionStats;
-    stats->single.downloadedBytes += bytes;
-    stats->cumulative.downloadedBytes += bytes;
+    getStats(handle)->single.downloadedBytes += bytes;
 }
 
@@ -182 +205 @@
 tr_statsFileCreated( tr_handle * handle )
 {
-    struct tr_stats_handle * stats = handle->sessionStats;
-    ++stats->cumulative.filesAdded;
-    ++stats->single.filesAdded;
-}
+    getStats(handle)->single.filesAdded++;
+}

branches/1.0x/libtransmission/transmission.c

@@ -393 +393 @@
     const uint64_t deadline = tr_date( ) + maxwait_msec;
 
+    tr_statsClose( h );
+
     tr_runInEventThread( h, tr_closeImpl, h );
     while( !h->isClosed && !deadlineReached( deadline ) )
@@ -402 +404 @@
 
     tr_fdClose( );
-    tr_statsClose( h );
     tr_lockFree( h->lock );
     free( h->tag );