error checking

[imx_flashconf.git] / imx-flashconf.c

// SPDX-License-Identifier: (GPL-2.0 or MIT)
/*
 * Copyright (C) 2020 Jesse Taube <[email protected]>
 */

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <stdbool.h>
#include "flash-defs.h"
#include "flash_confs.h"

size_t get_name_match(const flash_configs * , const char *);
void help(int);
void prepend(FILE *, size_t);
void list(const flash_configs *);
int main(int argc, char *argv[]){

 char name[32] = {};
 const char * fname;
 int opt;
 size_t index = 0;
 size_t padding = 0x1000; // default offset for nor
 size_t bin = 0;
 FILE * fnum;
 const flash_configs * configs = get_flash_confs();
 const flexspi_nor_config_t * config;
 while ((opt = getopt(argc, argv, "p:n:o:b:lh")) != -1)
 {
 errno = 0;
 switch (opt)
 {
 case 'p':
 padding = strtol(optarg, NULL, 0);
 if (errno) {
 printf("Invalid value for -%c.\n", opt);
 help(-3);
 }
 break;
 case 'n':
 strncpy(name,optarg,32);
 break;
 case 'o':
 fname = optarg;
 break;
 case 'b':
 bin = strtol(optarg, NULL, 0);
 if (errno) {
 printf("Invalid value for -%c.\n", opt);
 help(-3);
 }
 break;
 case 'h':
 help(0);
 break;
 case 'l':
 list(configs);
 break;
 default:
 help(-3);
 break;
 }
 }

 if(!strnlen(fname,32))
 puts("Need an output file name.");
 if(!strnlen(name,32))
 puts("Need a flash config name.");
 if(strnlen(name,32) == 32)
 puts("Flash config name too long.");
 if(!strnlen(name,32) || !strnlen(fname,32) || strnlen(name,32) == 32)
 help(-3);

 index = get_name_match(configs,name);
 if(index == -1){
 printf("No match found for %s!\nExiting...\n",name);
 exit(-2);
 }
 printf("Match found using %s flash config.\n",configs[index].name);

 config = &configs[index].config;

 fnum = fopen(fname,"wb+");
 if(!fnum){
 printf("Can't open %s for writing!\nExiting...\n",fname);
 exit(-1);
 }

 fwrite(config,sizeof(flexspi_nor_config_t),1,fnum);
 for(size_t i = sizeof(flexspi_nor_config_t); i < padding;i++)
 fputc(0xff,fnum);
 if(bin)
 prepend(fnum,bin);
 fclose(fnum);
 puts("Done!");
 return 0;
}
void list(const flash_configs * configs){
 for(size_t i = 0; *configs[i].name;i++)
 printf("%lu: %s\n",i,configs[i].name);
 exit(0);
}

size_t get_name_match(const flash_configs * configs, const char * name){
 size_t i = 0;
 while(*configs[i].name){
 if(strncmp(configs[i].name,name,32) == 0)
 return i;
 i++;
 }
 return -1;
}
void prepend(FILE * fno, size_t offset){
 FILE * SPL, * uboot;
 char * buf;
 size_t sz;
 uboot = fopen("u-boot.img","r");
 if(!uboot){
 puts("Can't open u-boot.img for reading!\nExiting...");
 fclose(fno);
 exit(-1);
 }
 SPL = fopen("SPL","r");
 if(!SPL){
 puts("Can't open SPL for reading!\nExiting...");
 fclose(uboot);
 fclose(fno);
 exit(-1);
 }
 fseek(SPL, 0, SEEK_END);
 sz = ftell(SPL);
 buf = malloc(sz);
 rewind(SPL);
 fread(buf,sz,1,SPL);
 fwrite(buf,sz,1,fno);

 for(size_t i = ftell(fno); i < offset;i++)
 fputc(0xff,fno);

 fseek(uboot, 0, SEEK_END);
 sz = ftell(uboot);
 buf = realloc(buf, sz);
 rewind(uboot);
 fread(buf,sz,1,uboot);
 fwrite(buf,sz,1,fno);
 free(buf);

 fclose(SPL);
 fclose(uboot);
}

void help(int code){
 puts("+--------------------------------------------------------------+");
 puts("| -h - This message. |");
 puts("| -o - Output file name. |");
 puts("| -n - Name of flash config to use, defined in flash_confs.c . |");
 puts("| -p - Padding offset, other than default (0x1000). |");
 puts("| -b - Create binary image, and offset to put u-boot.img at. |");
 puts("| -l - List names of flash defined in flash_confs.c . |");
 puts("+--------------------------------------------------------------+");
 exit(code);
}