Fixed memory leak on cancellation in ObjC API

1 files changed, 11 insertions, 2 deletions

diff --git a/src/objc/imap/MCOIMAPMessageRenderingOperation.mm b/src/objc/imap/MCOIMAPMessageRenderingOperation.mm
index 983b6cac..4fb49bdb 100644
--- a/src/objc/imap/MCOIMAPMessageRenderingOperation.mm
+++ b/src/objc/imap/MCOIMAPMessageRenderingOperation.mm
@@ -39,12 +39,21 @@ typedef void (^CompletionType)(NSString * htmlString, NSError * error);
     [super dealloc];
 }
 
-- (void)start:(void (^)(NSString * htmlString, NSError * error))completionBlock {
+- (void) start:(void (^)(NSString * htmlString, NSError * error))completionBlock
+{
     _completionBlock = [completionBlock copy];
     [self start];
 }
 
-- (void)operationCompleted {
+- (void) cancel
+{
+  [_completionBlock release];
+  _completionBlock = nil;
+  [super cancel];
+}
+
+- (void) operationCompleted
+{
     if (_completionBlock == NULL)
         return;