imx-flashconf.c

   1 // SPDX-License-Identifier: (GPL-2.0 or MIT)
   2 /*
   3  * Copyright (C) 2020 Jesse Taube <[email protected]>
   4  */
   5
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <stdint.h>
   9 #include <string.h>
  10 #include <errno.h>
  11 #include <unistd.h>
  12 #include <stdbool.h>
  13 #include "flash-defs.h"
  14 #include "flash_confs.h"
  15
  16 size_t get_name_match(const flash_configs * , const char *);
  17 void help(int);
  18 void prepend(FILE *, size_t);
  19 void list(const flash_configs *);
  20 int main(int argc, char *argv[]){
  21
  22         char name[32] = {};
  23         const char * fname;
  24         int opt;
  25         size_t index = 0;
  26         size_t padding = 0x1000; // default offset for nor
  27         size_t bin = 0;
  28         FILE * fnum;
  29         const flash_configs * configs = get_flash_confs();
  30         const flexspi_nor_config_t * config;
  31         while ((opt = getopt(argc, argv, "p:n:o:b:lh")) != -1)
  32         {
  33                 errno = 0;
  34                 switch (opt)
  35                 {
  36                         case 'p':
  37                                 padding = strtol(optarg, NULL, 0);
  38                                 if (errno) {
  39                                         printf("Invalid value for -%c.\n", opt);
  40                                         help(-3);
  41                                 }
  42                                 break;
  43                         case 'n':
  44                                 strncpy(name,optarg,32);
  45                                 break;
  46                         case 'o':
  47                                 fname = optarg;
  48                                 break;
  49                         case 'b':
  50                                 bin = strtol(optarg, NULL, 0);
  51                                 if (errno) {
  52                                         printf("Invalid value for -%c.\n", opt);
  53                                         help(-3);
  54                                 }
  55                                 break;
  56                         case 'h':
  57                                 help(0);
  58                                 break;
  59                         case 'l':
  60                                 list(configs);
  61                                 break;
  62                         default:
  63                                 help(-3);
  64                                 break;
  65                 }
  66         }
  67
  68         if(!strnlen(fname,32))
  69                 puts("Need an output file name.");
  70         if(!strnlen(name,32))
  71                 puts("Need a flash config name.");
  72         if(strnlen(name,32) == 32)
  73                 puts("Flash config name too long.");
  74         if(!strnlen(name,32) || !strnlen(fname,32) || strnlen(name,32) == 32)
  75                 help(-3);
  76
  77         index = get_name_match(configs,name);
  78         if(index == -1){
  79                 printf("No match found for %s!\nExiting...\n",name);
  80                 exit(-2);
  81         }
  82         printf("Match found using %s flash config.\n",configs[index].name);
  83
  84         config = &configs[index].config;
  85
  86         fnum = fopen(fname,"wb+");
  87         if(!fnum){
  88                 printf("Can't open %s for writing!\nExiting...\n",fname);
  89                 exit(-1);
  90         }
  91
  92         fwrite(config,sizeof(flexspi_nor_config_t),1,fnum);
  93         for(size_t i = sizeof(flexspi_nor_config_t); i < padding;i++)
  94                 fputc(0xff,fnum);
  95         if(bin)
  96                 prepend(fnum,bin);
  97         fclose(fnum);
  98         puts("Done!");
  99         return 0;
 100 }
 101 void list(const flash_configs * configs){
 102         for(size_t i = 0; *configs[i].name;i++)
 103                 printf("%lu: %s\n",i,configs[i].name);
 104         exit(0);
 105 }
 106
 107 size_t get_name_match(const flash_configs * configs, const char * name){
 108         size_t i = 0;
 109         while(*configs[i].name){
 110                 if(strncmp(configs[i].name,name,32) == 0)
 111                         return i;
 112                 i++;
 113         }
 114         return -1;
 115 }
 116 void prepend(FILE * fno, size_t offset){
 117         FILE * SPL, * uboot;
 118         char * buf;
 119         size_t sz;
 120         uboot = fopen("u-boot.img","r");
 121         if(!uboot){
 122                 puts("Can't open u-boot.img for reading!\nExiting...");
 123                 fclose(fno);
 124                 exit(-1);
 125         }
 126         SPL = fopen("SPL","r");
 127         if(!SPL){
 128                 puts("Can't open SPL for reading!\nExiting...");
 129                 fclose(uboot);
 130                 fclose(fno);
 131                 exit(-1);
 132         }
 133         fseek(SPL, 0, SEEK_END);
 134         sz = ftell(SPL);
 135         buf = malloc(sz);
 136         rewind(SPL);
 137         fread(buf,sz,1,SPL);
 138         fwrite(buf,sz,1,fno);
 139
 140         for(size_t i = ftell(fno); i < offset;i++)
 141                 fputc(0xff,fno);
 142
 143         fseek(uboot, 0, SEEK_END);
 144         sz = ftell(uboot);
 145         buf = realloc(buf, sz);
 146         rewind(uboot);
 147         fread(buf,sz,1,uboot);
 148         fwrite(buf,sz,1,fno);
 149         free(buf);
 150
 151         fclose(SPL);
 152         fclose(uboot);
 153 }
 154
 155 void help(int code){
 156         puts("+--------------------------------------------------------------+");
 157         puts("| -h - This message.                                           |");
 158         puts("| -o - Output file name.                                       |");
 159         puts("| -n - Name of flash config to use, defined in flash_confs.c . |");
 160         puts("| -p - Padding offset, other than default (0x1000).            |");
 161         puts("| -b - Create binary image, and offset to put u-boot.img at.   |");
 162         puts("| -l - List names of flash defined in flash_confs.c .          |");
 163         puts("+--------------------------------------------------------------+");
 164         exit(code);
 165 }