Node.m

//
//  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