rbutil/mkimxboot/main.c

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * $Id$
   9  *
  10  * Copyright (C) 2011 by Amaury Pouly
  11  *
  12  * This program is free software; you can redistribute it and/or
  13  * modify it under the terms of the GNU General Public License
  14  * as published by the Free Software Foundation; either version 2
  15  * of the License, or (at your option) any later version.
  16  *
  17  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  18  * KIND, either express or implied.
  19  *
  20  ****************************************************************************/
  21
  22 #include <getopt.h>
  23 #include <stdlib.h>
  24 #include <stdio.h>
  25 #include <string.h>
  26 #include "mkimxboot.h"
  27
  28 struct imx_variant_t
  29 {
  30     const char *name;
  31     enum imx_firmware_variant_t variant;
  32 };
  33
  34 struct imx_variant_t imx_variants[] =
  35 {
  36     { "default", VARIANT_DEFAULT },
  37     { "zenxfi2-recovery", VARIANT_ZENXFI2_RECOVERY },
  38     { "zenxfi2-nand", VARIANT_ZENXFI2_NAND },
  39     { "zenxfi2-sd", VARIANT_ZENXFI2_SD },
  40     { "zenxfistyle-recovery", VARIANT_ZENXFISTYLE_RECOVERY },
  41 };
  42
  43 #define NR_VARIANTS sizeof(imx_variants) / sizeof(imx_variants[0])
  44
  45 static void usage(void)
  46 {
  47     printf("Usage: elftosb [options | file]...\n");
  48     printf("Options:\n");
  49     printf("  -?/--help\tDisplay this message\n");
  50     printf("  -o <file>\tSet output file\n");
  51     printf("  -i <file>\tSet input file\n");
  52     printf("  -b <file>\tSet boot file\n");
  53     printf("  -d/--debug\tEnable debug output\n");
  54     printf("  -t <type>\tSet type (dualboot, singleboot, recovery)\n");
  55     printf("  -v <v>\tSet variant\n");
  56     printf("  -x\t\tDump device informations\n");
  57     printf("  -w\tExtract the original firmware\n");
  58     printf("Supported variants: (default is standard)\n");
  59     printf("  ");
  60     for(size_t i = 0; i < NR_VARIANTS; i++)
  61     {
  62         if(i != 0)
  63             printf(", ");
  64         printf("%s", imx_variants[i].name);
  65     }
  66     printf("\n");
  67     printf("By default a dualboot image is built\n");
  68     exit(1);
  69 }
  70
  71 int main(int argc, char *argv[])
  72 {
  73     char *infile = NULL;
  74     char *outfile = NULL;
  75     char *bootfile = NULL;
  76     enum imx_firmware_variant_t variant = VARIANT_DEFAULT;
  77     enum imx_output_type_t type = IMX_DUALBOOT;
  78     bool debug = false;
  79     bool extract_of = false;
  80
  81     if(argc == 1)
  82         usage();
  83
  84     while(1)
  85     {
  86         static struct option long_options[] =
  87         {
  88             {"help", no_argument, 0, '?'},
  89             {"in-file", no_argument, 0, 'i'},
  90             {"out-file", required_argument, 0, 'o'},
  91             {"boot-file", required_argument, 0, 'b'},
  92             {"debug", no_argument, 0, 'd'},
  93             {"type", required_argument, 0, 't'},
  94             {"variant", required_argument, 0, 'v'},
  95             {"dev-info", no_argument, 0, 'x'},
  96             {0, 0, 0, 0}
  97         };
  98
  99         int c = getopt_long(argc, argv, "?di:o:b:t:v:xw", long_options, NULL);
 100         if(c == -1)
 101             break;
 102         switch(c)
 103         {
 104             case 'd':
 105                 debug = true;
 106                 break;
 107             case '?':
 108                 usage();
 109                 break;
 110             case 'o':
 111                 outfile = optarg;
 112                 break;
 113             case 'i':
 114                 infile = optarg;
 115                 break;
 116             case 'b':
 117             {
 118                 bootfile = optarg;
 119                 break;
 120             }
 121             case 't':
 122                 if(strcmp(optarg, "dualboot") == 0)
 123                     type = IMX_DUALBOOT;
 124                 else if(strcmp(optarg, "singleboot") == 0)
 125                     type = IMX_SINGLEBOOT;
 126                 else if(strcmp(optarg, "recovery") == 0)
 127                     type = IMX_RECOVERY;
 128                 else
 129                 {
 130                     printf("Invalid boot type '%s'\n", optarg);
 131                     return 1;
 132                 }
 133                 break;
 134             case 'v':
 135             {
 136                 for(size_t i = 0; i < NR_VARIANTS; i++)
 137                 {
 138                     if(strcmp(optarg, imx_variants[i].name) == 0)
 139                     {
 140                         variant = imx_variants[i].variant;
 141                         goto Lok;
 142                     }
 143                 }
 144                 printf("Invalid variant '%s'\n", optarg);
 145                 return 1;
 146
 147                 Lok:
 148                 break;
 149             }
 150             case 'x':
 151                 dump_imx_dev_info("");
 152                 printf("variant mapping:\n");
 153                 for(int i = 0; i < sizeof(imx_variants) / sizeof(imx_variants[0]); i++)
 154                     printf("  %s -> variant=%d\n", imx_variants[i].name, imx_variants[i].variant);
 155                 break;
 156             case 'w':
 157                 extract_of = true;
 158                 break;
 159             default:
 160                 abort();
 161         }
 162     }
 163
 164     if(!infile)
 165     {
 166         printf("You must specify an input file\n");
 167         return 1;
 168     }
 169     if(!outfile)
 170     {
 171         printf("You must specify an output file\n");
 172         return 1;
 173     }
 174     if(!bootfile && !extract_of)
 175     {
 176         printf("You must specify an boot file\n");
 177         return 1;
 178     }
 179     if(optind != argc)
 180     {
 181         printf("Extra arguments on command line\n");
 182         return 1;
 183     }
 184
 185     if(extract_of)
 186     {
 187         enum imx_error_t err = extract_firmware(infile, variant, outfile);
 188         printf("Result: %d\n", err);
 189         return 0;
 190     }
 191
 192     struct imx_option_t opt;
 193     memset(&opt, 0, sizeof(opt));
 194     opt.debug = debug;
 195     opt.output = type;
 196     opt.fw_variant = variant;
 197     enum imx_error_t err = mkimxboot(infile, bootfile, outfile, opt);
 198     printf("Result: %d\n", err);
 199     return 0;
 200 }