Fix potential memory leak in BloomFilterWrapper (#833)

duckduckgo · Feb 16, 2021 · 9639784 · 9639784
1 parent dcace65
commit 9639784
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/Core/BloomFilterWrapper.h b/Core/BloomFilterWrapper.h
@@ -21,6 +21,7 @@
 @interface BloomFilterWrapper : NSObject
 - (instancetype)initFromPath:(NSString*)path withBitCount:(int)bitCount andTotalItems:(int)totalItems;
 - (instancetype)initWithTotalItems:(int)count errorRate:(double)errorRate;
+- (void)dealloc;
 - (void)add:(NSString*) entry;
 - (BOOL)contains:(NSString*) entry;
 @end
diff --git a/Core/BloomFilterWrapper.mm b/Core/BloomFilterWrapper.mm
@@ -44,6 +44,10 @@ - (instancetype)initWithTotalItems:(int)count errorRate:(double)errorRate {
     return self;
 }
 
+- (void)dealloc {
+	delete filter;
+}
+
 - (void)add:(NSString*)entry {
     if (filter != nil) {
         filter->add([entry UTF8String]);