ssn-installer/src/parseArguments.c

#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>

#include "decrypt.h"
#include "errorAndExit.h"
#include "parseArguments.h"

//Uses GNU's getopt_long_only(), see https://www.gnu.org/software/libc/manual/html_node/Getopt-Long-Options.html

static struct option long_options[] = {
  {"disk", required_argument, 0, 'd'},
  {"offset", required_argument, 0, 'o'},
  {"size", required_argument, 0, 's'},
  {"keys", required_argument, 0, 'k'},
  {"prev", required_argument, 0, 'p'},
};


struct arguments parseArguments(int argc, char *argv[]) {
  //Stores current state from command line arguments, initialized to zero
  struct arguments state = {};

  bool hasDisk = false;
  bool hasOffset = false;
  bool hasSize = false;

  while (1) {
    //In this variable, getopt_long stores the current long option that was found. Unused since we don't use flags in our long options
    int optionIndex = 0;

    int curOption = getopt_long_only(argc, argv, "d:o:s:k:p:", long_options, &optionIndex);

    //end of command line arguments reached
    if (curOption == -1) {
      break;
    }

    switch (curOption) {
      case 'd': //disk name
        state.diskName = optarg;
        hasDisk = true;
        break;
      case 'o': //offset
        state.diskOffset = atol(optarg);
        hasOffset = true;
        break;
      case 's': //size
        state.fileSize = atol(optarg);
        hasSize = true;
        break;
      case 'k': { //decryption keys
        //TODO: parse from optarg
        uint32_t key0 = atoi(argv[8]);
        uint32_t key1 = atoi(argv[9]);
        uint32_t key2 = atoi(argv[10]);
        //Initialize decryption (pass decryption keys)
        initDecryptor(key0, key1, key2);
        state.isEncrypted = true;
        break;
      }
      case 'p': //prev file for xdelta3
        //TODO
        break;
      case '?':
        errorAndExit();
        break;
      default:
        fprintf(stderr, "Unknown option '%c'.\n", (char)curOption);
        errorAndExit();
    }
  }

  if (!hasDisk) {
    fprintf(stderr, "Missing required argument --disk.\n");
    errorAndExit();
  }

  if (!hasOffset) {
    fprintf(stderr, "Missing required argument --offset.\n");
    errorAndExit();
  }

  if (!hasSize) {
    fprintf(stderr, "Missing required argument --size.\n");
    errorAndExit();
  }

  if (optind < argc) {
    fprintf(stderr, "Found %i arguments without an option. All arguments must be preceded by an option.\n", argc - optind);
    errorAndExit();
  }

  return state;
}