CBPatcher.c

//
//  CBPatcher.c
//  CBPatcher
//
//  Created by JonathanSeals on 11/18/18.
//  Copyright © 2018 JonathanSeals. All rights reserved.
//

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "CBPUtils.h"
#include "CBPatch.h"


int patch_kernel(char* infile, char* outfile, char* version);

#define MAIN
#ifdef MAIN
 int main(int argc, char **argv) {
    
     if (argc < 4 || 5 < argc) {
         printf("Missing args (cbpatcher infile outfile version [--nosb])\n");
         return -1;
     }
    
     int nukesb = 1;
    
     if (argc == 5) {
         if(!strcmp(argv[4], "--nosb")){
             nukesb = 0;
         }
     }
    
     char *fileName = argv[1];
    
     void *fileBuf = 0;
    
     size_t fileLen = 0;
    
     int ret = openFile(fileName, &fileLen, &fileBuf);
    
     if (ret) {
         printf("Failed to open %s\n", fileName);
         return -1;
     }
    
     char *versionNum = argv[3];
    
     ret = kernPat(fileBuf, fileLen, versionNum, nukesb);
    
     if (ret) {
         printf("Failed to patch kernel\n");
         free(fileBuf);
         return -1;
     }
    
     printf("Kernel patched successfully\n");
    
     FILE *outFile = fopen(argv[2], "w");
     fwrite(fileBuf, fileLen, 1, outFile);
     fflush(outFile);
     fclose(outFile);
    
     free(fileBuf);
    
     return 0;
 }
#endif
int patch_kernel(char* infile, char* outfile, char* version) {
   
    int nukesb = 1;
   
    char *fileName = infile;
   
    void *fileBuf = 0;
   
    size_t fileLen = 0;
   
    int ret = openFile(fileName, &fileLen, &fileBuf);
   
    if (ret) {
        printf("Failed to open %s\n", fileName);
        return -1;
    }
   
    char *versionNum = version;
   
    ret = kernPat(fileBuf, fileLen, versionNum, nukesb);
   
    if (ret) {
        printf("Failed to patch kernel\n");
        free(fileBuf);
        return -1;
    }
   
    printf("Kernel patched successfully\n");
   
    FILE *outFile = fopen(outfile, "w");
    fwrite(fileBuf, fileLen, 1, outFile);
    fflush(outFile);
    fclose(outFile);
   
    free(fileBuf);
   
    return 0;
}