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 };
  41
  42 #define NR_VARIANTS sizeof(imx_variants) / sizeof(imx_variants[0])
  43
  44 static void usage(void)
  45 {
  46     printf("Usage: elftosb [options | file]...\n");
  47     printf("Options:\n");
  48     printf("  -?/--help\tDisplay this message\n");
  49     printf("  -o <file>\tSet output file\n");
  50     printf("  -i <file>\tSet input file\n");
  51     printf("  -b <file>\tSet boot file\n");
  52     printf("  -d/--debug\tEnable debug output\n");
  53     printf("  -t <type>\tSet type (dualboot, singleboot, recovery)\n");
  54     printf("  -v <v>\tSet variant\n");
  55     printf("  -x\t\tDump device informations\n");
  56     printf("Supported variants: (default is standard)\n");
  57     printf("  ");
  58     for(size_t i = 0; i < NR_VARIANTS; i++)
  59     {
  60         if(i != 0)
  61             printf(", ");
  62         printf("%s", imx_variants[i].name);
  63     }
  64     printf("\n");
  65     printf("By default a dualboot image is built\n");
  66     exit(1);
  67 }
  68
  69 int main(int argc, char *argv[])
  70 {
  71     char *infile = NULL;
  72     char *outfile = NULL;
  73     char *bootfile = NULL;
  74     enum imx_firmware_variant_t variant = VARIANT_DEFAULT;
  75     enum imx_output_type_t type = IMX_DUALBOOT;
  76     bool debug = false;
  77
  78     if(argc == 1)
  79         usage();
  80
  81     while(1)
  82     {
  83         static struct option long_options[] =
  84         {
  85             {"help", no_argument, 0, '?'},
  86             {"in-file", no_argument, 0, 'i'},
  87             {"out-file", required_argument, 0, 'o'},
  88             {"boot-file", required_argument, 0, 'b'},
  89             {"debug", no_argument, 0, 'd'},
  90             {"type", required_argument, 0, 't'},
  91             {"variant", required_argument, 0, 'v'},
  92             {"dev-info", no_argument, 0, 'x'},
  93             {0, 0, 0, 0}
  94         };
  95
  96         int c = getopt_long(argc, argv, "?di:o:b:t:v:x", long_options, NULL);
  97         if(c == -1)
  98             break;
  99         switch(c)
 100         {
 101             case 'd':
 102                 debug = true;
 103                 break;
 104             case '?':
 105                 usage();
 106                 break;
 107             case 'o':
 108                 outfile = optarg;
 109                 break;
 110             case 'i':
 111                 infile = optarg;
 112                 break;
 113             case 'b':
 114             {
 115                 bootfile = optarg;
 116                 break;
 117             }
 118             case 't':
 119                 if(strcmp(optarg, "dualboot") == 0)
 120                     type = IMX_DUALBOOT;
 121                 else if(strcmp(optarg, "singleboot") == 0)
 122                     type = IMX_SINGLEBOOT;
 123                 else if(strcmp(optarg, "recovery") == 0)
 124                     type = IMX_RECOVERY;
 125                 else
 126                 {
 127                     printf("Invalid boot type '%s'\n", optarg);
 128                     return 1;
 129                 }
 130                 break;
 131             case 'v':
 132             {
 133                 for(size_t i = 0; i < NR_VARIANTS; i++)
 134                 {
 135                     if(strcmp(optarg, imx_variants[i].name) == 0)
 136                     {
 137                         variant = imx_variants[i].variant;
 138                         goto Lok;
 139                     }
 140                 }
 141                 printf("Invalid variant '%s'\n", optarg);
 142                 return 1;
 143
 144                 Lok:
 145                 break;
 146             }
 147             case 'x':
 148                 dump_imx_dev_info("");
 149                 printf("variant mapping:\n");
 150                 for(int i = 0; i < sizeof(imx_variants) / sizeof(imx_variants[0]); i++)
 151                     printf("  %s -> variant=%d\n", imx_variants[i].name, imx_variants[i].variant);
 152                 break;
 153             default:
 154                 abort();
 155         }
 156     }
 157
 158     if(!infile)
 159     {
 160         printf("You must specify an input file\n");
 161         return 1;
 162     }
 163     if(!outfile)
 164     {
 165         printf("You must specify an output file\n");
 166         return 1;
 167     }
 168     if(!bootfile)
 169     {
 170         printf("You must specify an boot file\n");
 171         return 1;
 172     }
 173     if(optind != argc)
 174     {
 175         printf("Extra arguments on command line\n");
 176         return 1;
 177     }
 178
 179     struct imx_option_t opt;
 180     memset(&opt, 0, sizeof(opt));
 181     opt.debug = debug;
 182     opt.output = type;
 183     opt.fw_variant = variant;
 184     enum imx_error_t err = mkimxboot(infile, bootfile, outfile, opt);
 185     printf("Result: %d\n", err);
 186     return 0;
 187 }