init

[imx_flashconf.git] / main.c

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

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

	char name[32];
	char fname[32];
	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:")) != -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':
				strncpy(fname,optarg,32);
				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;
			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) || !strnlen(fname,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;
}

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("help");
	exit(code);
}