2 files changed, 11 insertions, 0 deletions
diff --git a/pica-client/dialogs/filetransferdialog.cpp b/pica-client/dialogs/filetransferdialog.cpp
index bd442c8..41ebcdb 100644
--- a/pica-client/dialogs/filetransferdialog.cpp
+++ b/pica-client/dialogs/filetransferdialog.cpp
@@ -1,6 +1,7 @@
 #include "filetransferdialog.h"
 #include <QVBoxLayout>
 #include <QHBoxLayout>
+#include <QMessageBox>
 #include <stdlib.h>
 
 FileTransferDialog::FileTransferDialog(QByteArray peer_id, QString filename, quint64 size,
@@ -188,13 +189,21 @@ void FileTransferDialog::rightbuttonclick()
         if (rightbutton->text() == tr("Deny"))
         {
             emit deniedFile(peer_id_);
+            setTransferStatus(DENIED);
             return;
         }
     }
 
     if (rightbutton->text() == tr("Cancel"))
     {
+
+        if (QMessageBox::No == QMessageBox::question(this, tr("Cancel File Transfer"),
+            tr("Are you sure you want to cancel file transfer?"),
+                QMessageBox::Yes | QMessageBox::No, QMessageBox::No))
+             return;
+
         emit cancelledFile(peer_id_, this);
+        setTransferStatus(CANCELLED);
     }
 }
 
diff --git a/pica-client/dialogs/filetransferdialog.h b/pica-client/dialogs/filetransferdialog.h
index 2511651..2f42e6d 100644
--- a/pica-client/dialogs/filetransferdialog.h
+++ b/pica-client/dialogs/filetransferdialog.h
@@ -26,6 +26,8 @@ public:
     RECEIVINGFILE,
     PAUSED,
     CANCELLED,
+    PEERDISCONNECTED,
+    IOERROR,
     FINISHED
     };