add .gitignore

[swift-upb.git] / transfer.cpp

diff --git a/transfer.cpp b/transfer.cpp
index 827b2cb..ad121f3 100644 (file)
--- a/transfer.cpp
+++ b/transfer.cpp
@@ -40,11 +40,13 @@ void    Channel::CloseTransfer (FileTransfer* trans) {
 }
 
 
-void swift::AddProgressCallback (int transfer, TransferProgressCallback cb) {
+void swift::AddProgressCallback (int transfer,ProgressCallback cb,uint8_t agg) {
     FileTransfer* trans = FileTransfer::file(transfer);
     if (!trans)
         return;
-    trans->callbacks[trans->cb_installed++] = cb;
+    trans->cb_agg[trans->cb_installed] = agg;
+    trans->callbacks[trans->cb_installed] = cb;
+    trans->cb_installed++;
 }
 
 
@@ -56,12 +58,12 @@ void swift::ExternallyRetrieved (int transfer,bin64_t piece) {
 }
 
 
-void swift::RemoveProgressCallback (int transfer, TransferProgressCallback cb) {
+void swift::RemoveProgressCallback (int transfer, ProgressCallback cb) {
     FileTransfer* trans = FileTransfer::file(transfer);
     if (!trans)
         return;
     for(int i=0; i<trans->cb_installed; i++)
-        if (trans->callbacks[i].cb==cb.cb)
+        if (trans->callbacks[i]==cb)
             trans->callbacks[i]=trans->callbacks[--trans->cb_installed];
 }