bootstrapTool/bootstrapTool/Classes/InputPackage.m

#import "InputPackage.h"
#import "ErrorReturn.h"
#import "HelperClass.h"

@implementation InputPackage

- (NSString*) description {
    
    NSString *orig = [super description];
    NSMutableDictionary *details = [NSMutableDictionary new];
    NSArray *props = [self properties];
    [props enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        
        NSString *cv = [self valueForKey:obj];
        if (cv){
            details[obj] = cv;
        }
        
    }];
    return [NSString stringWithFormat:@"%@ = %@", orig, details];
    
}

- (NSString *)listfile {
    
    __block NSMutableArray *outFiles = [NSMutableArray new];
    [self.files enumerateObjectsUsingBlock:^(InputPackageFile * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        
        if ([obj.fileType isEqualToString:@"link"]){ //does this need to handle things differently?
            [outFiles addObject:obj.path];
        } else {
            [outFiles addObject:obj.path];
        }
    }];
    return [outFiles componentsJoinedByString:@"\n"];
    
}

- (int)installToBootstrapPath:(NSString *)bootstrapPath {
    
    DLog(@"\nFound package: '%@' at version: '%@'...\n", self.packageName, self.version );
    
    NSString *statusFile = [bootstrapPath stringByAppendingPathComponent:@"Library/dpkg/status"];
    NSArray *installedPackages = [HelperClass statusInstalledPackagesFromFile:statusFile];
    
    //DLog(@"installedPackages: %@", installedPackages);
    
    
    StatusPackageModel *model = [[installedPackages filteredArrayUsingPredicate:[NSPredicate predicateWithFormat:@"package == %@", self.packageName]] lastObject];
    
    if (model) {
        
        DLog(@"\n [WARNING] This package has already been installed! %@", model);
        NSComparisonResult result = [self.version compare:model.version options:NSNumericSearch];
        DLog(@"\n [WARNING] Comparing version numbers...");
        switch (result) {
            case NSOrderedSame:
                DLog(@"\n [WARNING] %@ and %@ match!", self.version, model.version);
                break;
                
            case NSOrderedAscending:
                DLog(@"\n [WARNING] Package version %@ is less than installed version %@!", self.version, model.version);
                break;
                
            case NSOrderedDescending:
                DLog(@"\n [WARNING] Package version %@ is greater than installed version %@!", self.version, model.version);
                break;
                
            default:
                break;
        }
        
        
        
    }
    
    ErrorReturn *safePackage = [self errorReturnForBootstrap:bootstrapPath];
    
    if (safePackage.returnStatus != 0) { //zero is success
        
        if (safePackage.returnStatus == 1) //ovewrwrites, just warnings!
        {
            
            NSString *error = [NSString stringWithFormat:@" [WARNING] %@ will overwrite the following files: \n\n\t%@\n\n", self.path.lastPathComponent, [safePackage.overwriteFiles componentsJoinedByString:@"\n\t"]];
            
            if(![HelperClass shouldContinueWithError:error]){
                
                return -1;
            }
            
        } else if (safePackage.returnStatus == 2) //bail!!"
        {
            return 2;
        }
    }
    
    NSString *pwd = [[HelperClass returnForProcess:@"/bin/pwd"] componentsJoinedByString:@"\n"];
    
    NSFileManager *man = [NSFileManager defaultManager];
    NSString *tmpPath = [pwd stringByAppendingPathComponent:self.packageName];
    NSString *debian = [tmpPath stringByAppendingPathComponent:@"DEBIAN"];
    [man createDirectoryAtPath:tmpPath withIntermediateDirectories:TRUE attributes:nil error:nil];
    DLog(@"\nExtracting deb for processing...\n");
    [HelperClass returnForProcess:[NSString stringWithFormat:@"/usr/local/bin/dpkg -x %@ %@", self.path, tmpPath]];
    
    NSString *bootstrapInfoPath = [bootstrapPath stringByAppendingPathComponent:@"Library/dpkg/info"];
    
    NSString *listFile = [bootstrapInfoPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.list", self.packageName]];
    NSString *md5s = [bootstrapInfoPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.md5sums", self.packageName]];
    
    DLog(@"\nCreating list file '%@'...\n", listFile);
    
    NSString *listString = [NSString stringWithFormat:@"/usr/bin/find %@ -type f -not -path \"*.DS_Store*\" | /usr/bin/sed \"s|%@||g\" >> \"%@\"", tmpPath, tmpPath, listFile];
    NSString *listOutput = [HelperClass singleLineReturnForProcess:listString];
    
    //DLog(@"list output: %@", listOutput);
    
    //[listOutput writeToFile:listFile atomically:TRUE encoding:NSASCIIStringEncoding error:nil];
    
    //find "$BUILD_ROOT/$WORKING_DIR/deb" -type f -not -path "$BUILD_ROOT/$WORKING_DIR/deb/DEBIAN/*" | sed "s|$BUILD_ROOT/$WORKING_DIR/deb||g" >> "$SKEL_PREFIX/var/lib/dpkg/info/$PKG_NAME.list"
    
    //find "$BUILD_ROOT/$WORKING_DIR/deb" -type f -not -path "$BUILD_ROOT/$WORKING_DIR/deb/DEBIAN/*" -exec $MD5 {} \; |  awk '{ print $1 "  " $2 }' | sed "s|$BUILD_ROOT/$WORKING_DIR/deb/||g" >> "$SKEL_PREFIX/var/lib/dpkg/info/$PKG_NAME.md5sums"
    
    //[self.listfile writeToFile:listFile atomically:TRUE encoding:NSASCIIStringEncoding error:nil];
    
    DLog(@"\nGenerating md5sums...\n");
    NSString *runString = [NSString stringWithFormat:@"/usr/bin/find %@ -type f -not -path \"*.DS_Store*\" -exec /sbin/md5 -r {} \\; |  /usr/bin/awk '{ print $1 \"  \" $2 }' | /usr/bin/sed \"s|%@||g\" >> \"%@\"", tmpPath, tmpPath, md5s];
    NSString *outputs = [[HelperClass returnForProcess:runString] componentsJoinedByString:@"\n"];
    
    DLog(@"\nCreating md5sum file '%@'...\n", md5s);
    
    //[outputs writeToFile:md5s atomically:TRUE encoding:NSASCIIStringEncoding error:nil];
    
    [man createDirectoryAtPath:debian withIntermediateDirectories:TRUE attributes:nil error:nil];
    
    DLog(@"\nExtracting DEBIAN files for processing...\n");
    
    [HelperClass returnForProcess:[NSString stringWithFormat:@"/usr/local/bin/dpkg -e %@ %@", self.path, debian]];
    //NSString *nextPath = [tmpPath stringByAppendingPathComponent:@"DEBIAN"];
    
    DLog(@"\nCopying any necessary DEBIAN files to new locations...\n\n");
    
    __block NSString *postInstFile = nil;
    
    NSArray *files = [man contentsOfDirectoryAtPath:debian error:nil];
    [files enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        
        NSString *fullPath = [debian stringByAppendingPathComponent:obj];
        if (![obj isEqualToString:@"control"]){
            
            if ([obj isEqualToString:@"postinst"]){
                
                postInstFile = fullPath;
                
            }
            DLog(@"Copying file... %@\n\n", fullPath);
            NSString *fileName = [NSString stringWithFormat:@"%@.%@", self.packageName, obj];
            NSString *newPath = [bootstrapPath stringByAppendingPathComponent:@"Library/dpkg/info"];
            newPath = [newPath stringByAppendingPathComponent:fileName];
            //DLog(@"newPath: %@", newPath);
            [man copyItemAtPath:fullPath toPath:newPath error:nil];
            
        } else {
            
            NSMutableString *controlFile = [[NSMutableString alloc] initWithContentsOfFile:fullPath encoding:NSASCIIStringEncoding error:nil];
            
            [controlFile replaceOccurrencesOfString:@"iphoneos-arm" withString:@"appletvos-arm64" options:NSLiteralSearch range:NSMakeRange(0, [controlFile length])];
            
            [controlFile appendString:@"Status: install ok installed"];
            
            //DLog(@"control file: -%@-\n", controlFile);
            
            NSString *statusFile = [bootstrapPath stringByAppendingPathComponent:@"Library/dpkg/status"];
            
            [FM copyItemAtPath:statusFile toPath:[statusFile stringByAppendingPathExtension:@"bak"] error:nil];
            
            NSMutableString *statusContents = [[NSMutableString alloc] initWithContentsOfFile:statusFile encoding:NSASCIIStringEncoding error:nil];
            
            if (model) { //we found the package model, just replace the old string with our new one
                
                
                DLog(@"Updating status file for new package version...\n\n");
                
                [statusContents replaceOccurrencesOfString:model.rawString withString:controlFile options:NSLiteralSearch range:NSMakeRange(0, [statusContents length])];
                
                //[statusContents writeToFileWithoutAttributes:statusFile];
                [statusContents writeToFile:statusFile atomically:TRUE encoding:NSASCIIStringEncoding error:nil];
                
                
            } else {
                
                
                
                DLog(@"Updating status file for new package...\n\n");
                
                //DLog(@"down here??: -%@-\n", controlFile);
                
                [statusContents appendFormat:@"%@\n", controlFile];
                
                //DLog(@"\nnew status file: %@\n", statusContents);
                
                //[statusContents writeToFileWithoutAttributes:statusFile];
                
                [statusContents writeToFile:statusFile atomically:TRUE encoding:NSASCIIStringEncoding error:nil];
                
            }
            
            
        }
        
        
    }];
    
    //finally actually install the package onto the bootstrap
    
    DLog(@"Extracting package onto bootstrap folder...\n");
    
    [HelperClass returnForProcess:[NSString stringWithFormat:@"/usr/local/bin/dpkg -x %@ %@", self.path, bootstrapPath]];
    
    if (postInstFile) {
        
        DLog(@"\n [WARNING] We found a postinst file, will not run this due to potential unexpected consequences in your run environment! Displaying contents so you can determine if any additional steps are necessary. Contents will be delimited by a line -----\n");
        
        DLog(@"\n%@ Contents:\n\n---------------------\n\n%@\n\n---------------------\n\n", postInstFile, [NSString stringWithContentsOfFile:postInstFile encoding:NSASCIIStringEncoding error:nil]);
        
        
    }
    
    [FM removeItemAtPath:tmpPath error:nil];
    
    DLog(@"Done!\n\n");
    return 0;
    
    
}

- (void)codesignIfNecessary:(NSString *)file {
    
    if (![[[file pathExtension] lowercaseString] isEqualToString:@"dylib"]){
        return;
    }
    
    NSString *jtp = [HelperClass singleLineReturnForProcess:@"/usr/bin/which jtool"];
    if (jtp){
        
        NSString *proc = [[HelperClass arrayReturnForTask:jtp withArguments:@[@"--sig", file]] componentsJoinedByString:@"\n"];
        
        NSLog(@"proc: %@", proc);
        
        if ([proc containsString:@"No Code Signing blob detected in this file"]){
        
            NSString *runCommand = [NSString stringWithFormat:@"%@ --sign platform %@ --inplace", jtp, file];
            
            NSLog(@"running codesign command: %@", runCommand);
            
           NSString *returnValue = [HelperClass singleLineReturnForProcess:runCommand];
            
            NSLog(@"returnValue: %@", returnValue);
        }
    }
}

- (void)validateSignaturesInPath:(NSString *)thePath {
    
    [self.files enumerateObjectsUsingBlock:^(InputPackageFile * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
       
        NSString *file = [thePath stringByAppendingPathComponent:obj.path];
        
        NSLog(@"check sig file: %@", file);
        [self codesignIfNecessary:file];
        
        
    }];
    
}

- (void)bumpVersionInCurrentDirectory {
    
    NSString *fakeRoot = [HelperClass singleLineReturnForProcess:@"/usr/bin/which fakeroot"];
    NSString *pwd = [HelperClass singleLineReturnForProcess:@"/bin/pwd"];
    DLog(@"\nProcessing file: %@\n", self.path);
    InputPackage *output = self;
    
    DLog(@"\nFound package: '%@' at version: '%@'...\n", output.packageName, output.version );
    
    NSString *tmpPath = [pwd stringByAppendingPathComponent:output.packageName];
    NSString *debian = [tmpPath stringByAppendingPathComponent:@"DEBIAN"];
    [FM createDirectoryAtPath:tmpPath withIntermediateDirectories:TRUE attributes:nil error:nil];
    DLog(@"\nExtracting package contents for processing...\n");
    [HelperClass returnForProcess:[NSString stringWithFormat:@"/usr/local/bin/dpkg -x %@ %@", self.path, tmpPath]];
    [FM createDirectoryAtPath:debian withIntermediateDirectories:TRUE attributes:nil error:nil];
    DLog(@"\nExtracting DEBIAN files for processing...\n");
    [HelperClass returnForProcess:[NSString stringWithFormat:@"/usr/local/bin/dpkg -e %@ %@", self.path, debian]];
    
    
    NSString *controlPath = [debian stringByAppendingPathComponent:@"control"];
    
    NSMutableString *controlFile = [[NSMutableString alloc] initWithContentsOfFile:controlPath encoding:NSASCIIStringEncoding error:nil];
    //@"appletvos-arm64"
    [controlFile replaceOccurrencesOfString:self.version withString:[self.version nextVersionNumber] options:NSLiteralSearch range:NSMakeRange(0, [controlFile length])];
    
    [controlFile writeToFile:controlPath atomically:TRUE];
    
    //at this point we have the files extracted, time to determine what needs to be changed
    
    NSArray *ignoreFiles = @[@".fauxsu", @".DS_Store"];
    NSArray *forbiddenRoots = @[@"etc", @"var", @"tmp"];
    
    [self validateSignaturesInPath:tmpPath];
    
    //__block NSMutableArray *_overwriteArray = [NSMutableArray new];
    [self.files enumerateObjectsUsingBlock:^(InputPackageFile * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        
        if (obj.type == BSPackageFileTypeFile || obj.type == BSPackageFileTypeDirectory){
            
            //DLog(@"processing path: %@", obj.path);
            if ([obj.path isEqualToString:@"/private/var/mobile/Applications/"]){
                
                NSString *badPath = [tmpPath stringByAppendingPathComponent:obj.path];
                NSString *newPath = [tmpPath stringByAppendingPathComponent:@"Applications"];
                DLog(@"\n [INFO] Moving %@ to %@...", badPath, newPath);
                [FM moveItemAtPath:badPath toPath:newPath error:nil];
                [FM removeItemAtPath:[tmpPath stringByAppendingPathComponent:@"private"] error:nil];
                *stop = TRUE;
            }
            
            NSString *fullPath = [tmpPath stringByAppendingPathComponent:obj.path];
            
            if ([ignoreFiles containsObject:obj.path.lastPathComponent]){
                
                DLog(@"in ignore file list, purge");
                [FM removeItemAtPath:fullPath error:nil];
                
            }
            
            NSArray *pathComponents = [obj.path pathComponents];
            if ([pathComponents count] > 1)
            {
                
                NSString *rootPath = [pathComponents objectAtIndex:1];
                //DLog(@"\n Checking root path: %@ for file %@\n", rootPath, obj.path);
                if ([forbiddenRoots containsObject:rootPath])
                {
                    DLog(@"\n [WARNING] package file: '%@' would overwrite symbolic link at '%@'\n", obj.path, rootPath);
                    NSString *privateDir = [tmpPath stringByAppendingPathComponent:@"private"];
                    if (![FM fileExistsAtPath:privateDir]){
                        [FM createDirectoryAtPath:privateDir withIntermediateDirectories:TRUE attributes:nil error:nil];
                        
                    }
                    //take <package_name>/[rootPath] (could be etc, var, tmp) and move to <package_name>/private/[rootPath]
                    NSString *badPath = [tmpPath stringByAppendingPathComponent:rootPath];
                    NSString *newPath = [privateDir stringByAppendingPathComponent:rootPath];
                    DLog(@"\n [INFO] Moving %@ to %@...", badPath, newPath);
                    [FM moveItemAtPath:badPath toPath:newPath error:nil];
                    
                    
                }
            }
            
            
        }
        
        
    }];
    
    
    NSString *depArchiveInfo = [NSString stringWithFormat:@"/usr/local/bin/dpkg -b %@", self.packageName];
    
    if (fakeRoot) {
        
        depArchiveInfo = [NSString stringWithFormat:@"%@ /usr/local/bin/dpkg -b %@", fakeRoot, self.packageName];
        
        
    }
    
    [[HelperClass returnForProcess:depArchiveInfo] componentsJoinedByString:@"\n"];
    
    
    
    DLog(@"\nDone!\n\n");
    
    //return er;
}

- (void)repackageInCurrentDirectoryWithArch:(NSString *)newArch {
    
    NSString *fakeRoot = [HelperClass singleLineReturnForProcess:@"/usr/bin/which fakeroot"];
    NSString *pwd = [HelperClass singleLineReturnForProcess:@"/bin/pwd"];
    DLog(@"\nProcessing file: %@\n", self.path);
    InputPackage *output = self;
    
    DLog(@"\nFound package: '%@' at version: '%@'...\n", output.packageName, output.version );
    
    NSString *tmpPath = [pwd stringByAppendingPathComponent:output.packageName];
    NSString *debian = [tmpPath stringByAppendingPathComponent:@"DEBIAN"];
    [FM createDirectoryAtPath:tmpPath withIntermediateDirectories:TRUE attributes:nil error:nil];
    DLog(@"\nExtracting package contents for processing...\n");
    [HelperClass returnForProcess:[NSString stringWithFormat:@"/usr/local/bin/dpkg -x %@ %@", self.path, tmpPath]];
    [FM createDirectoryAtPath:debian withIntermediateDirectories:TRUE attributes:nil error:nil];
    DLog(@"\nExtracting DEBIAN files for processing...\n");
    [HelperClass returnForProcess:[NSString stringWithFormat:@"/usr/local/bin/dpkg -e %@ %@", self.path, debian]];
    
    
    if (newArch != nil) {
        
        NSString *controlPath = [debian stringByAppendingPathComponent:@"control"];
        
        NSMutableString *controlFile = [[NSMutableString alloc] initWithContentsOfFile:controlPath encoding:NSASCIIStringEncoding error:nil];
        //@"appletvos-arm64"
        [controlFile replaceOccurrencesOfString:@"iphoneos-arm" withString:newArch options:NSLiteralSearch range:NSMakeRange(0, [controlFile length])];
        
        [controlFile writeToFile:controlPath atomically:TRUE];
    }
    
    //at this point we have the files extracted, time to determine what needs to be changed
    
    NSArray *ignoreFiles = @[@".fauxsu", @".DS_Store"];
    NSArray *forbiddenRoots = @[@"etc", @"var", @"tmp"];
    //__block NSMutableArray *_overwriteArray = [NSMutableArray new];
    [self.files enumerateObjectsUsingBlock:^(InputPackageFile * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        
        if (obj.type == BSPackageFileTypeFile || obj.type == BSPackageFileTypeDirectory){
            
            //DLog(@"processing path: %@", obj.path);
            if ([obj.path isEqualToString:@"/private/var/mobile/Applications/"]){
                
                NSString *badPath = [tmpPath stringByAppendingPathComponent:obj.path];
                NSString *newPath = [tmpPath stringByAppendingPathComponent:@"Applications"];
                DLog(@"\n [INFO] Moving %@ to %@...", badPath, newPath);
                [FM moveItemAtPath:badPath toPath:newPath error:nil];
                [FM removeItemAtPath:[tmpPath stringByAppendingPathComponent:@"private"] error:nil];
                *stop = TRUE;
            }
            
            NSString *fullPath = [tmpPath stringByAppendingPathComponent:obj.path];
            
            if ([ignoreFiles containsObject:obj.path.lastPathComponent]){
                
                DLog(@"in ignore file list, purge");
                [FM removeItemAtPath:fullPath error:nil];
                
            }
            
            NSArray *pathComponents = [obj.path pathComponents];
            if ([pathComponents count] > 1)
            {
                
                NSString *rootPath = [pathComponents objectAtIndex:1];
                //DLog(@"\n Checking root path: %@ for file %@\n", rootPath, obj.path);
                if ([forbiddenRoots containsObject:rootPath])
                {
                    DLog(@"\n [WARNING] package file: '%@' would overwrite symbolic link at '%@'\n", obj.path, rootPath);
                    NSString *privateDir = [tmpPath stringByAppendingPathComponent:@"private"];
                    if (![FM fileExistsAtPath:privateDir]){
                        [FM createDirectoryAtPath:privateDir withIntermediateDirectories:TRUE attributes:nil error:nil];
                        
                    }
                    //take <package_name>/[rootPath] (could be etc, var, tmp) and move to <package_name>/private/[rootPath]
                    NSString *badPath = [tmpPath stringByAppendingPathComponent:rootPath];
                    NSString *newPath = [privateDir stringByAppendingPathComponent:rootPath];
                    DLog(@"\n [INFO] Moving %@ to %@...", badPath, newPath);
                    [FM moveItemAtPath:badPath toPath:newPath error:nil];
                    
                    
                }
            }
            
            
        }
        
        
    }];
    
    
    NSString *depArchiveInfo = [NSString stringWithFormat:@"/usr/local/bin/dpkg -b %@", self.packageName];
    
    if (fakeRoot) {
        
        depArchiveInfo = [NSString stringWithFormat:@"%@ /usr/local/bin/dpkg -b %@", fakeRoot, self.packageName];
        
        
    }
    
    [[HelperClass returnForProcess:depArchiveInfo] componentsJoinedByString:@"\n"];
    
    
    
    DLog(@"\nDone!\n\n");
    
    //return er;
}

- (ErrorReturn *)errorReturnForBootstrap:(NSString *)bootstrapPath
{
    NSArray *ignoreFiles = @[@".fauxsu", @".DS_Store"];
    NSArray *forbiddenRoots = @[@"etc", @"var", @"tmp"];
    NSFileManager *man = [NSFileManager defaultManager];
    __block NSInteger returnValue = 0; //0 = good to go 1 = over write warning, 2 = no go
    __block NSMutableArray *_overwriteArray = [NSMutableArray new];
    [self.files enumerateObjectsUsingBlock:^(InputPackageFile * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        
        if ([obj.fileType isEqualToString:@"file"]){
            
            NSString *fullPath = [bootstrapPath stringByAppendingPathComponent:obj.path];
            if ([man fileExistsAtPath:fullPath] && ![ignoreFiles containsObject:obj.basename]){
                
                //DLog(@"[WARNING] overwriting a file that already exists and isn't DS_Store or .fauxsu: %@", fullPath);
                [_overwriteArray addObject:obj.path];
                //*stop = TRUE;//return FALSE;
                returnValue = 1;
            }
            
            NSArray *pathComponents = [obj.path pathComponents];
            if ([pathComponents count] > 1)
            {
                NSString *rootPath = [pathComponents objectAtIndex:1];
                if ([forbiddenRoots containsObject:rootPath])
                {
                    DLog(@"\n [ERROR] package file: '%@' would overwrite symbolic link at '%@'! Exiting!\n\n", obj.path, rootPath);
                    *stop = TRUE;
                    returnValue = 2;
                }
            }
            
            
        }
        
        
    }];
    
    ErrorReturn *er = [ErrorReturn new];
    er.returnStatus = returnValue;
    er.overwriteFiles = _overwriteArray;
    
    return er;
}


@end