1 files changed, 11 insertions, 5 deletions
diff --git a/src/async/imap/MCIMAPCustomCommandOperation.cpp b/src/async/imap/MCIMAPCustomCommandOperation.cpp
index 692c8f7d..61e0e655 100644
--- a/src/async/imap/MCIMAPCustomCommandOperation.cpp
+++ b/src/async/imap/MCIMAPCustomCommandOperation.cpp
@@ -16,21 +16,27 @@ using namespace mailcore;
 IMAPCustomCommandOperation::IMAPCustomCommandOperation()
 {
     mCustomCommand = NULL;
+    mResponse = NULL;
 }
 
 IMAPCustomCommandOperation::~IMAPCustomCommandOperation()
 {
     MC_SAFE_RELEASE(mCustomCommand);
+    MC_SAFE_RELEASE(mResponse);
 }
 
-void IMAPCustomCommandOperation::setCustomCommand(mailcore::String * command)
+void IMAPCustomCommandOperation::setCustomCommand(String * command)
 {
-    mCustomCommand = command;
+    MC_SAFE_REPLACE_COPY(String, mCustomCommand, command);
 }
 
-void IMAPCustomCommandOperation::main()
+String * IMAPCustomCommandOperation::response()
 {
+    return mResponse;
+}
 
-    session()->session()->sendCustomCommand(mCustomCommand);
-
+void IMAPCustomCommandOperation::main()
+{
+    mResponse = session()->session()->sendCustomCommand(mCustomCommand);
+    MC_SAFE_RETAIN(mResponse);
 }