Added support to save and load Attachemnts on KeePass2 Databases.

Support for Keepass1 DBs is still missing
2025-12-23 07:59:27 +00:00 · 2013-07-05 20:09:17 +02:00
parent 7feb2517dd
commit 9c3a62bef6
14 changed files with 272 additions and 94 deletions
--- a/MacPass/MPDocument+Attachments.m
+++ b/MacPass/MPDocument+Attachments.m
@@ -0,0 +1,135 @@
+//
+//  MPDocument+Attachments.m
+//  MacPass
+//
+//  Created by Michael Starke on 05.07.13.
+//  Copyright (c) 2013 HicknHack Software GmbH. All rights reserved.
+//
+
+#import "MPDocument.h"
+
+#import "NSMutableData+Base64.h"
+#import "NSData+Gzip.h"
+
+#import "Kdb3Node.h"
+#import "Kdb4Node.h"
+#import "Kdb4Entry+KVOAdditions.h"
+
+@implementation MPDocument (Attachments)
+
+- (void)addAttachment:(NSURL *)location toEntry:(KdbEntry *)anEntry {
+  NSError *error = nil;
+  NSString *fileName = [location lastPathComponent];
+  if([anEntry isKindOfClass:[Kdb3Entry class]]) {
+    Kdb3Entry *entry = (Kdb3Entry *)anEntry;
+    NSData *binaryData = [NSData dataWithContentsOfURL:location options:NSDataReadingUncached error:&error];
+    if(!binaryData) {
+      [NSApp presentError:error];
+      binaryData = nil;
+      error = nil;
+      return; // failed
+    }
+    entry.binary = binaryData;
+    entry.binaryDesc = fileName;
+  }
+  if( [anEntry isKindOfClass:[Kdb4Entry class]]) {
+    Kdb4Entry *entry = (Kdb4Entry *)anEntry;
+    NSStringEncoding encoding;
+    NSString *fileContents = [NSString stringWithContentsOfURL:location usedEncoding:&encoding error:&error];
+    if(!fileContents) {
+      [NSApp presentError:error];
+      fileContents = nil;
+      error = nil;
+      return; // failed
+    }
+    Binary *binary = [[Binary alloc] init];
+    NSUInteger nextId = [self nextBinaryId];
+    if(nextId == NSNotFound) {
+      binary = nil;
+      return; // No id found. Something went wrong
+    }
+    binary.binaryId = nextId;
+    binary.compressed = (self.treeV4.compressionAlgorithm != KPLCompressionNone);
+    NSData *encodedData;
+    NSData *fileData = [fileContents dataUsingEncoding:encoding];
+    if(binary.compressed) {
+      switch(self.treeV4.compressionAlgorithm) {
+        case KPLCompressionGzip: {
+          NSData *compressedData = [fileData gzipDeflate];
+          encodedData = [NSMutableData mutableDataWithBase64EncodedData:compressedData];
+          break;
+        }
+        default:
+          NSAssert(NO, @"Unsupported Compression Algorithm");
+          binary = nil;
+          encodedData = nil;
+          fileData = nil;
+          return;
+      }
+    }
+    else {
+      encodedData = fileData;
+    }
+    binary.data = [[NSString alloc] initWithData:encodedData encoding:NSASCIIStringEncoding];
+    
+    [self.treeV4.binaries addObject:binary];
+    BinaryRef *ref = [[BinaryRef alloc] init];
+    ref.key = fileName;
+    ref.ref = binary.binaryId;
+    [entry insertObject:ref inBinariesAtIndex:[entry.binaries count]];
+  }
+}
+
+- (Binary *)findBinary:(BinaryRef *)reference {
+  if(self.version != MPDatabaseVersion4) {
+    return nil;
+  }
+  NSPredicate *filterPredicate = [NSPredicate predicateWithBlock:^BOOL(id evaluatedObject, NSDictionary *bindings) {
+    Binary *binaryFile = evaluatedObject;
+    return (binaryFile.binaryId == reference.ref);
+  }];
+  NSArray *filteredBinary = [self.treeV4.binaries filteredArrayUsingPredicate:filterPredicate];
+  return [filteredBinary lastObject];
+}
+
+- (void)saveAttachment:(BinaryRef *)reference toLocation:(NSURL *)location {
+  Binary *binary = [self findBinary:reference];
+  NSData *rawData = nil;
+  if(binary) {
+    if(binary.compressed) {
+      rawData = [NSMutableData mutableDataWithBase64DecodedData:[binary.data dataUsingEncoding:NSASCIIStringEncoding]];
+      rawData = [rawData gzipInflate];
+    }
+    else {
+      rawData = [NSMutableData mutableDataWithBase64DecodedData:[binary.data dataUsingEncoding:NSASCIIStringEncoding]];
+    }
+    NSError *error = nil;
+    if( ![rawData writeToURL:location options:0 error:&error] ) {
+      [NSApp presentError:error];
+    }
+  }
+}
+
+- (void)saveAttachmentFromEntry:(KdbEntry *)anEntry toLocation:(NSURL *)location {
+  if([anEntry isKindOfClass:[Kdb3Entry class]]) {
+    Kdb3Entry *entry = (Kdb3Entry *)anEntry;
+    NSError *error = nil;
+    if(! [entry.binary writeToURL:location options:NSDataWritingWithoutOverwriting error:&error] ) {
+      [NSApp presentError:error];
+    }
+  }
+  return; //
+}
+
+- (NSUInteger)nextBinaryId {
+  if(self.version != MPDatabaseVersion4) {
+    return NSNotFound;
+  }
+  NSUInteger maxKey = 0;
+  for(Binary *binary in self.treeV4.binaries) {
+    maxKey = MAX(binary.binaryId, maxKey);
+  }
+  return (maxKey + 1);
+}
+
+@end