Adds error code to IMAPSession::customCommand. Renames sendCustomCommand to customCommand.

1 files changed, 9 insertions, 2 deletions

diff --git a/src/core/imap/MCIMAPSession.cpp b/src/core/imap/MCIMAPSession.cpp
index 69d80cc7..cbd94a2b 100755
--- a/src/core/imap/MCIMAPSession.cpp
+++ b/src/core/imap/MCIMAPSession.cpp
@@ -996,9 +996,16 @@ static uint64_t get_mod_sequence_value(mailimap * session)
     return mod_sequence_value;
 }
 
-String * IMAPSession::sendCustomCommand(String * command)
+String * IMAPSession::customCommand(String * command, ErrorCode * pError)
 {
-    mailimap_custom_command(mImap, MCUTF8(command));
+    int r;
+    
+    r = mailimap_custom_command(mImap, MCUTF8(command));
+    if (r == MAILIMAP_ERROR_CUSTOM_COMMAND) {
+        * pError = ErrorCustomCommand;
+        return NULL;
+    }
+    
     String *response = String::stringWithUTF8Characters(mImap->imap_response);
     return response;
 }