mirror of
https://github.com/samsonjs/arq_restore.git
synced 2026-03-25 09:25:53 +00:00
240 lines
9.3 KiB
Objective-C
240 lines
9.3 KiB
Objective-C
//
|
|
// Node.m
|
|
// s3print
|
|
//
|
|
// Created by Stefan Reitshamer on 4/10/09.
|
|
// Copyright 2009 PhotoMinds LLC. All rights reserved.
|
|
//
|
|
|
|
#include <sys/stat.h>
|
|
#import "Node.h"
|
|
#import "BooleanIO.h"
|
|
#import "IntegerIO.h"
|
|
#import "StringIO.h"
|
|
#import "BufferedInputStream.h"
|
|
#import "BlobKey.h"
|
|
#import "NSObject_extra.h"
|
|
#import "Tree.h"
|
|
#import "BlobKeyIO.h"
|
|
|
|
|
|
@implementation Node
|
|
@synthesize isTree, treeContainsMissingItems, uncompressedDataSize, xattrsBlobKey, xattrsSize, aclBlobKey, uid, gid, mode, mtime_sec, mtime_nsec, flags, finderFlags, extendedFinderFlags, finderFileType, finderFileCreator, isFileExtensionHidden, st_dev, treeVersion, st_rdev;
|
|
@synthesize ctime_sec, ctime_nsec, createTime_sec, createTime_nsec, st_nlink, st_ino, st_blocks, st_blksize;
|
|
@dynamic treeBlobKey, dataBlobKeys;
|
|
@dynamic aggregateGlacierArchiveSize;
|
|
|
|
|
|
- (id)initWithInputStream:(BufferedInputStream *)is treeVersion:(int)theTreeVersion error:(NSError **)error {
|
|
if (self = [super init]) {
|
|
treeVersion = theTreeVersion;
|
|
dataBlobKeys = [[NSMutableArray alloc] init];
|
|
|
|
if (![BooleanIO read:&isTree from:is error:error]) {
|
|
[self release];
|
|
return nil;
|
|
}
|
|
|
|
if (theTreeVersion >= 18) {
|
|
if (![BooleanIO read:&treeContainsMissingItems from:is error:error]) {
|
|
[self release];
|
|
return nil;
|
|
}
|
|
}
|
|
|
|
BOOL dataAreCompressed = NO;
|
|
BOOL xattrsAreCompressed = NO;
|
|
BOOL aclIsCompressed = NO;
|
|
if (treeVersion >= 12) {
|
|
if (![BooleanIO read:&dataAreCompressed from:is error:error]
|
|
|| ![BooleanIO read:&xattrsAreCompressed from:is error:error]
|
|
|| ![BooleanIO read:&aclIsCompressed from:is error:error]) {
|
|
[self release];
|
|
return nil;
|
|
}
|
|
}
|
|
|
|
int dataBlobKeysCount;
|
|
if (![IntegerIO readInt32:&dataBlobKeysCount from:is error:error]) {
|
|
[self release];
|
|
return nil;
|
|
}
|
|
for (int i = 0; i < dataBlobKeysCount; i++) {
|
|
BlobKey *dataBlobKey = nil;
|
|
if (![BlobKeyIO read:&dataBlobKey from:is treeVersion:treeVersion compressed:dataAreCompressed error:error]) {
|
|
[self release];
|
|
return nil;
|
|
}
|
|
[dataBlobKeys addObject:dataBlobKey];
|
|
}
|
|
if (![IntegerIO readUInt64:&uncompressedDataSize from:is error:error]) {
|
|
[self release];
|
|
return nil;
|
|
}
|
|
|
|
// As of Tree version 18 thumbnailBlobKey and previewBlobKey have been removed. They were never used.
|
|
if (theTreeVersion < 18) {
|
|
BlobKey *theThumbnailBlobKey = nil;
|
|
BlobKey *thePreviewBlobKey = nil;
|
|
if (![BlobKeyIO read:&theThumbnailBlobKey from:is treeVersion:treeVersion compressed:NO error:error]
|
|
|| ![BlobKeyIO read:&thePreviewBlobKey from:is treeVersion:treeVersion compressed:NO error:error]) {
|
|
[self release];
|
|
return nil;
|
|
}
|
|
}
|
|
|
|
BOOL ret = [BlobKeyIO read:&xattrsBlobKey from:is treeVersion:treeVersion compressed:xattrsAreCompressed error:error]
|
|
&& [IntegerIO readUInt64:&xattrsSize from:is error:error]
|
|
&& [BlobKeyIO read:&aclBlobKey from:is treeVersion:treeVersion compressed:aclIsCompressed error:error]
|
|
&& [IntegerIO readInt32:&uid from:is error:error]
|
|
&& [IntegerIO readInt32:&gid from:is error:error]
|
|
&& [IntegerIO readInt32:&mode from:is error:error]
|
|
&& [IntegerIO readInt64:&mtime_sec from:is error:error]
|
|
&& [IntegerIO readInt64:&mtime_nsec from:is error:error]
|
|
&& [IntegerIO readInt64:&flags from:is error:error]
|
|
&& [IntegerIO readInt32:&finderFlags from:is error:error]
|
|
&& [IntegerIO readInt32:&extendedFinderFlags from:is error:error]
|
|
&& [StringIO read:&finderFileType from:is error:error]
|
|
&& [StringIO read:&finderFileCreator from:is error:error]
|
|
&& [BooleanIO read:&isFileExtensionHidden from:is error:error]
|
|
&& [IntegerIO readInt32:&st_dev from:is error:error]
|
|
&& [IntegerIO readInt32:&st_ino from:is error:error]
|
|
&& [IntegerIO readUInt32:&st_nlink from:is error:error]
|
|
&& [IntegerIO readInt32:&st_rdev from:is error:error]
|
|
&& [IntegerIO readInt64:&ctime_sec from:is error:error]
|
|
&& [IntegerIO readInt64:&ctime_nsec from:is error:error]
|
|
&& [IntegerIO readInt64:&createTime_sec from:is error:error]
|
|
&& [IntegerIO readInt64:&createTime_nsec from:is error:error]
|
|
&& [IntegerIO readInt64:&st_blocks from:is error:error]
|
|
&& [IntegerIO readUInt32:&st_blksize from:is error:error];
|
|
[xattrsBlobKey retain];
|
|
[aclBlobKey retain];
|
|
[finderFileType retain];
|
|
[finderFileCreator retain];
|
|
if (!ret) {
|
|
[self release];
|
|
return nil;
|
|
}
|
|
|
|
// If any BlobKey has a nil sha1, drop it.
|
|
|
|
if ([xattrsBlobKey sha1] == nil) {
|
|
[xattrsBlobKey release];
|
|
xattrsBlobKey = nil;
|
|
}
|
|
if ([aclBlobKey sha1] == nil) {
|
|
[aclBlobKey release];
|
|
aclBlobKey = nil;
|
|
}
|
|
}
|
|
return self;
|
|
}
|
|
- (void)dealloc {
|
|
[dataBlobKeys release];
|
|
[xattrsBlobKey release];
|
|
[aclBlobKey release];
|
|
[finderFileType release];
|
|
[finderFileCreator release];
|
|
[super dealloc];
|
|
}
|
|
- (BlobKey *)treeBlobKey {
|
|
NSAssert(isTree, @"must be a Tree");
|
|
return [dataBlobKeys objectAtIndex:0];
|
|
}
|
|
- (NSArray *)dataBlobKeys {
|
|
return dataBlobKeys;
|
|
}
|
|
- (uint64_t)aggregateGlacierArchiveSize {
|
|
uint64_t ret = 0;
|
|
ret += [aclBlobKey archiveSize];
|
|
ret += [xattrsBlobKey archiveSize];
|
|
for (BlobKey *dataBlobKey in dataBlobKeys) {
|
|
ret += [dataBlobKey archiveSize];
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
- (BOOL)dataMatchesStat:(struct stat *)st {
|
|
return st->st_mtimespec.tv_sec == mtime_sec && st->st_mtimespec.tv_nsec == mtime_nsec && st->st_size == uncompressedDataSize;
|
|
}
|
|
- (BOOL)ctimeMatchesStat:(struct stat *)st {
|
|
return st->st_ctimespec.tv_sec == ctime_sec && st->st_ctimespec.tv_nsec == ctime_nsec;
|
|
}
|
|
- (void)writeToData:(NSMutableData *)data {
|
|
[BooleanIO write:isTree to:data];
|
|
[BooleanIO write:treeContainsMissingItems to:data];
|
|
BOOL dataAreCompressed = [dataBlobKeys count] == 0 ? NO : [[dataBlobKeys objectAtIndex:0] compressed];
|
|
[BooleanIO write:dataAreCompressed to:data];
|
|
[BooleanIO write:[xattrsBlobKey compressed] to:data];
|
|
[BooleanIO write:[aclBlobKey compressed] to:data];
|
|
[IntegerIO writeInt32:(int32_t)[dataBlobKeys count] to:data];
|
|
for (BlobKey *dataBlobKey in dataBlobKeys) {
|
|
NSAssert([dataBlobKey compressed] == dataAreCompressed, @"all dataBlobKeys must have same compressed flag value");
|
|
[BlobKeyIO write:dataBlobKey to:data];
|
|
}
|
|
[IntegerIO writeUInt64:uncompressedDataSize to:data];
|
|
[BlobKeyIO write:xattrsBlobKey to:data];
|
|
[IntegerIO writeUInt64:xattrsSize to:data];
|
|
[BlobKeyIO write:aclBlobKey to:data];
|
|
[IntegerIO writeInt32:uid to:data];
|
|
[IntegerIO writeInt32:gid to:data];
|
|
[IntegerIO writeInt32:mode to:data];
|
|
[IntegerIO writeInt64:mtime_sec to:data];
|
|
[IntegerIO writeInt64:mtime_nsec to:data];
|
|
[IntegerIO writeInt64:flags to:data];
|
|
[IntegerIO writeInt32:finderFlags to:data];
|
|
[IntegerIO writeInt32:extendedFinderFlags to:data];
|
|
[StringIO write:finderFileType to:data];
|
|
[StringIO write:finderFileCreator to:data];
|
|
[BooleanIO write:isFileExtensionHidden to:data];
|
|
[IntegerIO writeInt32:st_dev to:data];
|
|
[IntegerIO writeInt32:st_ino to:data];
|
|
[IntegerIO writeUInt32:st_nlink to:data];
|
|
[IntegerIO writeInt32:st_rdev to:data];
|
|
[IntegerIO writeInt64:ctime_sec to:data];
|
|
[IntegerIO writeInt64:ctime_nsec to:data];
|
|
[IntegerIO writeInt64:createTime_sec to:data];
|
|
[IntegerIO writeInt64:createTime_nsec to:data];
|
|
[IntegerIO writeInt64:st_blocks to:data];
|
|
[IntegerIO writeUInt32:st_blksize to:data];
|
|
}
|
|
|
|
|
|
#pragma mark NSObject
|
|
- (BOOL)isEqual:(id)object {
|
|
if (![object isKindOfClass:[Node class]]) {
|
|
return NO;
|
|
}
|
|
Node *other = (Node *)object;
|
|
return treeVersion == [other treeVersion]
|
|
&& isTree == [other isTree]
|
|
&& uncompressedDataSize == [other uncompressedDataSize]
|
|
&& [dataBlobKeys isEqualToArray:[other dataBlobKeys]]
|
|
&& [NSObject equalObjects:xattrsBlobKey and:[other xattrsBlobKey]]
|
|
&& xattrsSize == [other xattrsSize]
|
|
&& [NSObject equalObjects:aclBlobKey and:[other aclBlobKey]]
|
|
&& uid == [other uid]
|
|
&& gid == [other gid]
|
|
&& mode == [other mode]
|
|
&& mtime_sec == [other mtime_sec]
|
|
&& mtime_nsec == [other mtime_nsec]
|
|
&& flags == [other flags]
|
|
&& finderFlags == [other finderFlags]
|
|
&& extendedFinderFlags == [other extendedFinderFlags]
|
|
&& [NSObject equalObjects:finderFileType and:[other finderFileType]]
|
|
&& [NSObject equalObjects:finderFileCreator and:[other finderFileCreator]]
|
|
&& st_dev == [other st_dev]
|
|
&& st_ino == [other st_ino]
|
|
&& st_nlink == [other st_nlink]
|
|
&& st_rdev == [other st_rdev]
|
|
&& ctime_sec == [other ctime_sec]
|
|
&& ctime_nsec == [other ctime_nsec]
|
|
&& createTime_sec == [other createTime_sec]
|
|
&& createTime_nsec == [other createTime_nsec]
|
|
&& st_blocks == [other st_blocks]
|
|
&& st_blksize == [other st_blksize];
|
|
}
|
|
- (NSUInteger)hash {
|
|
return (NSUInteger)treeVersion + [dataBlobKeys hash];
|
|
}
|
|
@end
|