fixup! WIP: libcurl backend

ostreedev · Jan 19, 2017 · bda80cd · bda80cd
1 parent ffc0f5e
commit bda80cd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/libostree/ostree-fetcher-curl.c b/src/libostree/ostree-fetcher-curl.c
@@ -62,6 +62,7 @@ struct OstreeFetcher
   GHashTable *outstanding_requests; /* Set<GTask> */
   GHashTable *sockets; /* Set<SockInfo> */
 
+  guint64 bytes_transferred;
 };
 
 /* Information associated with a request */
@@ -199,6 +200,8 @@ _ostree_fetcher_init (OstreeFetcher *self)
   curl_multi_setopt (self->multi, CURLMOPT_SOCKETDATA, self);
   curl_multi_setopt (self->multi, CURLMOPT_TIMERFUNCTION, update_timeout_cb);
   curl_multi_setopt (self->multi, CURLMOPT_TIMERDATA, self);
+  /* Let's do something reasonable here. */
+  curl_multi_setopt (self->multi, CURLMOPT_MAX_TOTAL_CONNECTIONS, 8);
 }
 
 
@@ -546,6 +549,7 @@ write_cb (void *ptr, size_t size, size_t nmemb, void *data)
     }
 
   req->current_size += realsize;
+  req->fetcher->bytes_transferred += realsize;
 
   return realsize;
 }
@@ -879,5 +883,5 @@ _ostree_fetcher_request_to_membuf_finish (OstreeFetcher *self,
 guint64
 _ostree_fetcher_bytes_transferred (OstreeFetcher       *self)
 {
-  return 0;
+  return self->bytes_transferred;
 }