From 9d2e33cd506b5d3ff859c47c5cb9507fd00fb09a Mon Sep 17 00:00:00 2001
From: Dhinak G <17605561+dhinakg@users.noreply.github.com>
Date: Sun, 18 Aug 2024 22:32:37 -0400
Subject: [PATCH] Fix leak

---
 src/extract_standalone.m | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/extract_standalone.m b/src/extract_standalone.m
index 34601a0..c89a709 100644
--- a/src/extract_standalone.m
+++ b/src/extract_standalone.m
@@ -235,19 +235,18 @@ int extractAssetStandalone(AAByteStream byteStream, ExtractionConfiguration* con
                     break;
                 }
 
+                CFTypeRef configCopy = CFBridgingRetain([config copyWithFunction:(yop == AA_YOP_TYPE_DST_FIXUP ? @"VERIFY" : @"EXTRACT")]);
+
                 // TODO: What is the difference between these two?
                 // TODO: Magic constant
-                // TODO: CFBridgingRetain will leak
                 AAArchiveStream extractStream =
                     yop == AA_YOP_TYPE_DST_FIXUP
-                        ? AAVerifyDirectoryArchiveOutputStreamOpen(config.outputPath.UTF8String, keySet,
-                                                                   (void*)CFBridgingRetain([config copyWithFunction:@"VERIFY"]),
-                                                                   aa_callback, UINT64_C(1) << 53, 0)
-                        : AAExtractArchiveOutputStreamOpen(config.outputPath.UTF8String,
-                                                           (void*)CFBridgingRetain([config copyWithFunction:@"EXTRACT"]), aa_callback, 0,
-                                                           0);
+                        ? AAVerifyDirectoryArchiveOutputStreamOpen(config.outputPath.UTF8String, keySet, (void*)configCopy, aa_callback,
+                                                                   UINT64_C(1) << 53, 0)
+                        : AAExtractArchiveOutputStreamOpen(config.outputPath.UTF8String, (void*)configCopy, aa_callback, 0, 0);
                 if (!extractStream) {
                     ERRLOG(@"Failed to open extract stream");
+                    CFRelease(configCopy);
                     AAArchiveStreamClose(innerDecodeStream);
                     AAByteStreamClose(decompressStream);
                     AAByteStreamClose(datStream);
@@ -258,6 +257,7 @@ int extractAssetStandalone(AAByteStream byteStream, ExtractionConfiguration* con
                                            aa_callback, 0, 0) < 0) {
                     ERRLOG(@"Failed to process archive stream");
                     AAArchiveStreamClose(extractStream);
+                    CFRelease(configCopy);
                     AAArchiveStreamClose(innerDecodeStream);
                     AAByteStreamClose(decompressStream);
                     AAByteStreamClose(datStream);
@@ -265,6 +265,7 @@ int extractAssetStandalone(AAByteStream byteStream, ExtractionConfiguration* con
                 }
 
                 AAArchiveStreamClose(extractStream);
+                CFRelease(configCopy);
                 AAArchiveStreamClose(innerDecodeStream);
                 AAByteStreamClose(decompressStream);
                 AAByteStreamClose(datStream);