#include "lib.h"

// Private Prototypes
void DisplayHelp(char *app_name);
void SetDefaults(user_settings *set);
int SetArgument(int argc, int i, char *argv[], user_settings *set);
int CheckArgumentCombination(user_settings *set);
void PrintNeedsArgument(char *arg);
void PrintArgumentInvalid(char *arg);
void PrintNeedsParam(char *arg);
void PrintNoNeedParam(char *arg);

int ParseArgs(int argc, char *argv[], user_settings *usr_settings)
{
	if(argv == NULL || usr_settings == NULL)
		return USR_PTR_PASS_FAIL;
		
	if(argc < 2){
		DisplayHelp(argv[0]);
		return USR_HELP;
	}
		
	// Detecting Help Requried
	for(int i = 1; i < argc; i++){
		if(strcmp(argv[i],"-help") == 0){
			DisplayHelp(argv[0]);
			return USR_HELP;
		}
	}
	
	// Allocating Memory for Content Path Ptrs
	usr_settings->common.contentPath = malloc(CIA_MAX_CONTENT*sizeof(char*));
	if(usr_settings->common.contentPath == NULL){
		fprintf(stderr,"[SETTING ERROR] Not Enough Memory\n");
		return USR_MEM_ERROR;
	}
	memset(usr_settings->common.contentPath,0,CIA_MAX_CONTENT*sizeof(char*));
	
	// Initialise Keys
	InitKeys(&usr_settings->common.keys);

	// Setting Defaults
	SetDefaults(usr_settings);
	
	// Parsing Arguments
#ifdef DEBUG
	fprintf(stdout,"[DEBUG] Parsing Args\n");
#endif
	int set_result;
	int i = 1;
	while(i < argc){
		set_result = SetArgument(argc,i,argv,usr_settings);
		if(set_result < 1){
			fprintf(stderr,"[RESULT] Invalid arguments, see '%s -help'\n",argv[0]);
			return set_result;
		}
		i += set_result;
	}
	
	// Checking arguments
#ifdef DEBUG
	fprintf(stdout,"[DEBUG] Checking Args\n");
#endif
	set_result = CheckArgumentCombination(usr_settings);
	if(set_result) return set_result;
	
	// Setting Keys
#ifdef DEBUG
	fprintf(stdout,"[DEBUG] Setting Keys\n");
#endif
	set_result = SetKeys(&usr_settings->common.keys);
	if(set_result) return set_result;

#ifdef DEBUG
	fprintf(stdout,"[DEBUG] Generating output path name if required\n");
#endif

	if(!usr_settings->common.outFileName){
		char *source_path = NULL;
		if(usr_settings->ncch.buildNcch0) source_path = usr_settings->common.rsfPath;
		else if(usr_settings->common.workingFileType == infile_ncsd || usr_settings->common.workingFileType == infile_srl) source_path = usr_settings->common.workingFilePath;
		else source_path = usr_settings->common.contentPath[0];
		u16 outfile_len = strlen(source_path) + 3;
		usr_settings->common.outFileName = malloc(outfile_len);
		if(!usr_settings->common.outFileName){
			fprintf(stderr,"[SETTING ERROR] Not Enough Memory\n");
			return USR_MEM_ERROR;
		}
		usr_settings->common.outFileName_mallocd = true;
		append_filextention(usr_settings->common.outFileName,outfile_len,source_path,(char*)&output_extention[usr_settings->common.outFormat-1]);
	}
	return 0;
}

void SetDefaults(user_settings *set)
{
	// Target Info
	set->common.keys.keyset = pki_TEST;

	// Build NCCH Info
	set->ncch.buildNcch0 = true;
	set->ncch.includeExefsLogo = false;
	set->common.outFormat = CXI;
	set->ncch.ncchType = format_not_set;

	// Yaml Settings
	set->common.rsfSet.Option.EnableCompress = true;
	set->common.rsfSet.Option.EnableCrypt = true;
	set->common.rsfSet.Option.UseOnSD = false;
	set->common.rsfSet.Option.FreeProductCode = false;

	// Working File Info
	set->common.workingFileType = infile_ncch;

	// CCI Info
	set->cci.useSDKStockData = false;

	// CIA Info
	set->cia.useDataTitleVer = false;
	set->cia.titleVersion[0] = 0xffff; // invalid for detection
	set->cia.encryptCia = true;
	set->cia.randomTitleKey = false;
	set->common.keys.aes.currentCommonKey = 0x100; // invalid for detection
	for(int i = 0; i < CIA_MAX_CONTENT; i++){
		set->cia.contentId[i] = 0x100000000;
	}
}

int SetArgument(int argc, int i, char *argv[], user_settings *set)
{
	bool IsLastArg = (i >= (argc -1));
	bool HasParam = false;
	if(!IsLastArg){
		if(argv[i+1][0] != '-') HasParam = true;
	}
	// Global Settings
	if(strcmp(argv[i],"-rsf") == 0){
		if(!HasParam){
			PrintNeedsParam("-rsf");
			return USR_ARG_REQ_PARAM;
		}
		set->common.rsfPath = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-f") == 0){
		if(!HasParam){
			PrintNeedsParam("-f");
			return USR_ARG_REQ_PARAM;
		}
		if(strcasecmp(argv[i+1],"cxi") == 0 || strcasecmp(argv[i+1],"exec") == 0 ) set->common.outFormat = CXI;
		else if(strcasecmp(argv[i+1],"cfa") == 0 || strcasecmp(argv[i+1],"data") == 0 ) set->common.outFormat = CFA;
		else if(strcasecmp(argv[i+1],"cci") == 0 || strcasecmp(argv[i+1],"card") == 0 ) set->common.outFormat = CCI;
		else if(strcasecmp(argv[i+1],"cia") == 0) set->common.outFormat = CIA;
		else {
			fprintf(stderr,"[-] Invalid output format '%s'\n",argv[i+1]);
			return USR_BAD_ARG;
		}		
		return 2;
	}
	else if(strcmp(argv[i],"-o") == 0){
		if(!HasParam){
			PrintNeedsParam("-o");
			return USR_ARG_REQ_PARAM;
		}
		set->common.outFileName = argv[i+1];
		set->common.outFileName_mallocd = false;
		return 2;
	}
	// Key Options
	else if(strcmp(argv[i],"-target") == 0){
		if(!HasParam){
			PrintNeedsParam("-target");
			return USR_ARG_REQ_PARAM;
		}
		if(strcasecmp(argv[i+1],"test") == 0 || strcasecmp(argv[i+1],"t") == 0)
			set->common.keys.keyset = pki_TEST;
		else if(strcasecmp(argv[i+1],"custom") == 0 || strcasecmp(argv[i+1],"c") == 0)
			set->common.keys.keyset = pki_CUSTOM;
		
#ifndef PUBLIC_BUILD
		else if(strcasecmp(argv[i+1],"debug") == 0 || strcasecmp(argv[i+1],"development") == 0 || strcasecmp(argv[i+1],"d") == 0)
			set->common.keys.keyset = pki_DEVELOPMENT;
		else if(strcasecmp(argv[i+1],"retail") == 0 || strcasecmp(argv[i+1],"production") == 0 || strcasecmp(argv[i+1],"p") == 0)
			set->common.keys.keyset = pki_PRODUCTION;
		/*
		else if(strcasecmp(argv[i+1],"beta") == 0 || strcasecmp(argv[i+1],"b") == 0)
			set->common.keys.keyset = pki_BETA;
		*/
#endif
		else{
			fprintf(stderr,"[SETTING ERROR] Unrecognised target '%s'\n",argv[i+1]);
			return USR_BAD_ARG;
		}
		return 2;
	}
	else if(strcmp(argv[i],"-keydir") == 0){
		if(!HasParam){
			PrintNeedsParam("-keydir");
			return USR_ARG_REQ_PARAM;
		}
		set->common.keys.keydir = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-ckeyID") == 0){
		if(!HasParam){
			PrintNeedsParam("-ckeyID");
			return USR_ARG_REQ_PARAM;
		}
		set->common.keys.aes.currentCommonKey = strtol(argv[i+1],NULL,0);
		if(set->common.keys.aes.currentCommonKey > 0xff)
		{
			fprintf(stderr,"[SETTING ERROR] Invalid Common Key ID: 0x%x\n",set->common.keys.aes.currentCommonKey);
			return USR_BAD_ARG;
		}
		return 2;
	}
	else if(strcmp(argv[i],"-showkeys") == 0){
		if(HasParam){
			PrintNoNeedParam("-showkeys");
			return USR_BAD_ARG;
		}
		set->common.keys.dumpkeys = true;
		return 1;
	}

	// Ncch Options
	else if(strcmp(argv[i],"-elf") == 0){
		if(!HasParam){
			PrintNeedsParam("-elf");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.elfPath = argv[i+1];
		return 2;
	}
	
	else if(strcmp(argv[i],"-icon") == 0){
		if(!HasParam){
			PrintNeedsParam("-icon");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.iconPath = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-banner") == 0){
		if(!HasParam){
			PrintNeedsParam("-banner");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.bannerPath = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-logo") == 0){
		if(!HasParam){
			PrintNeedsParam("-logo");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.logoPath = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-desc") == 0){
		if(!HasParam){
			PrintNeedsParam("-desc");
			return USR_ARG_REQ_PARAM;
		}
		char *tmp = argv[i+1];
		char *tmp2 = strstr(tmp,":");
		if(!tmp2){
			fprintf(stderr,"[SETTING ERROR] Bad argument '%s %s', correct format:\n",argv[i],argv[i+1]);
			fprintf(stderr,"	-desc <APP TYPE>:<TARGET FIRMWARE>\n");
		}
		if(strlen(tmp2) < 2){
			fprintf(stderr,"[SETTING ERROR] Bad argument '%s %s', correct format:\n",argv[i],argv[i+1]);
			fprintf(stderr,"	-desc <APP TYPE>:<TARGET FIRMWARE>\n");
		}

		u32 app_type_len = (u32)(tmp2-tmp);
		char *app_type = malloc(app_type_len+1);
		memset(app_type,0,app_type_len+1);
		memcpy(app_type,tmp,app_type_len);

		if(strcasecmp(app_type,"App") == 0 || strcasecmp(app_type,"SDApp") == 0) set->common.keys.accessDescSign.presetType = app;
		else if(strcasecmp(app_type,"ECApp") == 0) set->common.keys.accessDescSign.presetType = ec_app;
		else if(strcasecmp(app_type,"Demo") == 0) set->common.keys.accessDescSign.presetType = demo;
		else if(strcasecmp(app_type,"DlpChild") == 0 || strcasecmp(app_type,"Dlp") == 0) set->common.keys.accessDescSign.presetType = dlp;
		else{
			fprintf(stderr,"[SETTING ERROR] Accessdesc AppType preset '%s' not valid, please manually configure RSF\n",app_type);
			return USR_BAD_ARG;
		}


		char *target_firmware = (tmp2+1);
		set->common.keys.accessDescSign.targetFirmware = strtoul(target_firmware,NULL,10);
		return 2;
	}
	else if(strcmp(argv[i],"-exefslogo") == 0){
		if(HasParam){
			PrintNoNeedParam("-exefslogo");
			return USR_BAD_ARG;
		}
		set->ncch.includeExefsLogo = true;
		return 1;
	}
	else if(strcmp(argv[i],"-data") == 0){
		if(HasParam){
			PrintNoNeedParam("-data");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.ncchType = CFA;
		return 1;
	}

	// Ncch Rebuild Options
	else if(strcmp(argv[i],"-code") == 0){
		if(!HasParam){
			PrintNeedsParam("-code");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.codePath = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-exheader") == 0){
		if(!HasParam){
			PrintNeedsParam("-exheader");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.exheaderPath = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-plain-region") == 0){
		if(!HasParam){
			PrintNeedsParam("-plain-region");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.plainRegionPath = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-romfs") == 0){
		if(!HasParam){
			PrintNeedsParam("-romfs");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.romfsPath = argv[i+1];
		return 2;
	}
	// Cci Options
#ifndef PUBLIC_BUILD
	else if(strcmp(argv[i],"-devcardcci") == 0){
		if(HasParam){
			PrintNoNeedParam("-devcardcci");
			return USR_BAD_ARG;
		}
		set->cci.useSDKStockData = true;
		return 1;
	}
#endif
	// Cia Options
#ifndef PUBLIC_BUILD
	else if(strcmp(argv[i],"-cci") == 0){
		if(!HasParam){
			PrintNeedsParam("-cci");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.buildNcch0 = false;
		set->common.workingFileType = infile_ncsd;
		set->common.workingFilePath = argv[i+1];
		set->common.outFormat = CIA;
		return 2;
	}
	else if(strcmp(argv[i],"-srl") == 0){
		if(!HasParam){
			PrintNeedsParam("-srl");
			return USR_ARG_REQ_PARAM;
		}
		set->ncch.buildNcch0 = false;
		set->common.workingFileType = infile_srl;
		set->common.workingFilePath = argv[i+1];
		set->common.outFormat = CIA;
		return 2;

	}
	else if(strcmp(argv[i],"-dlc") == 0){
		if(HasParam){
			PrintNoNeedParam("-dlc");
			return USR_BAD_ARG;
		}
		set->cia.DlcContent = true;
		return 1;
	}
#endif
	else if(strcmp(argv[i],"-major") == 0){
		if(!HasParam){
			PrintNeedsParam("-major");
			return USR_ARG_REQ_PARAM;
		}
		set->cia.useNormTitleVer = true;
		u32 tmp = strtoul(argv[i+1],NULL,10);
		if(tmp > 63){
			fprintf(stderr,"[SETTING ERROR] Major version: '%d' is too large, max: '63'\n",tmp);
			return USR_BAD_ARG;
		}
		set->cia.titleVersion[0] = tmp;
		return 2;
	}
	else if(strcmp(argv[i],"-minor") == 0){
		if(!HasParam){
			PrintNeedsParam("-minor");
			return USR_ARG_REQ_PARAM;
		}
		set->cia.useNormTitleVer = true;
		u32 tmp = strtoul(argv[i+1],NULL,10);
		if(tmp > 63){
			fprintf(stderr,"[SETTING ERROR] Minor version: '%d' is too large, max: '63'\n",tmp);
			return USR_BAD_ARG;
		}
		set->cia.titleVersion[1] = tmp;
		return 2;
	}
	else if(strcmp(argv[i],"-micro") == 0){
		if(!HasParam){
			PrintNeedsParam("-micro");
			return USR_ARG_REQ_PARAM;
		}
		u32 tmp = strtoul(argv[i+1],NULL,10);
		if(tmp > 15){
			fprintf(stderr,"[SETTING ERROR] Micro version: '%d' is too large, max: '15'\n",tmp);
			return USR_BAD_ARG;
		}
		set->cia.titleVersion[2] = tmp;
		return 2;
	}
	else if(strcmp(argv[i],"-dver") == 0){
		if(!HasParam){
			PrintNeedsParam("-dver");
			return USR_ARG_REQ_PARAM;
		}
		set->cia.useDataTitleVer = true;
		u32 tmp = strtoul(argv[i+1],NULL,10);
		if(tmp > 4095){
			fprintf(stderr,"[SETTING ERROR] Data version: '%d' is too large, max: '4095'\n",tmp);
			return USR_BAD_ARG;
		}
		set->cia.titleVersion[0] = (tmp >> 6) & 63;
		set->cia.titleVersion[1] = tmp & 63;
		return 2;
	}
	else if(strcmp(argv[i],"-savesize") == 0){
		if(!HasParam){
			PrintNeedsParam("-savesize");
			return USR_ARG_REQ_PARAM;
		}
		set->cia.overideSaveDataSize = argv[i+1];
		return 2;
	}
	else if(strcmp(argv[i],"-rand") == 0){
		if(HasParam){
			PrintNoNeedParam("-rand");
			return USR_BAD_ARG;
		}
		set->cia.randomTitleKey = true;
		return 1;
	}
	else if(strcmp(argv[i],"-nocryptcia") == 0){
		if(HasParam){
			PrintNoNeedParam("-nocryptcia");
			return USR_BAD_ARG;
		}
		set->cia.encryptCia = false;
		return 1;
	}
	
	// Other Setting
	else if(strcmp(argv[i],"-content") == 0){
		if(!HasParam){
			PrintNeedsParam("-content");
			return USR_ARG_REQ_PARAM;
		}
		char *pos = strstr(argv[i+1],":");
		if(!pos){
			fprintf(stderr,"[SETTING ERROR] Bad argument '%s %s', correct format:\n",argv[i],argv[i+1]);
			fprintf(stderr,"	-content <CONTENT PATH>:<INDEX>\n");
			fprintf(stderr,"  If generating a CIA, then use the format:\n");
			fprintf(stderr,"	-content <CONTENT PATH>:<INDEX>:<ID>\n");
			return USR_BAD_ARG;
		}		
		if(strlen(pos) < 2){
			fprintf(stderr,"[SETTING ERROR] Bad argument '%s %s', correct format:\n",argv[i],argv[i+1]);
			fprintf(stderr,"	-content <CONTENT PATH>:<INDEX>\n");
			fprintf(stderr,"  If generating a CIA, then use the format:\n");
			fprintf(stderr,"	-content <CONTENT PATH>:<INDEX>:<ID>\n");
			return USR_BAD_ARG;
		}

		/* Getting Content Index */
		u16 content_index = strtol((char*)(pos+1),NULL,0);

		/* Storing Content Filepath */
		u32 path_len = (u32)(pos-argv[i+1])+1;
		
		if(content_index == 0) set->ncch.buildNcch0 = false;
		if(set->common.contentPath[content_index] != NULL){
			fprintf(stderr,"[SETTING ERROR] Content %d is already specified\n",content_index);
			return USR_BAD_ARG;
		}
		set->common.contentPath[content_index] = malloc(path_len);
		if(set->common.contentPath[content_index] == NULL){
			fprintf(stderr,"[SETTING ERROR] Not enough memory\n");
			return USR_MEM_ERROR;
		}
		memset(set->common.contentPath[content_index],0,path_len);
		strncpy(set->common.contentPath[content_index],argv[i+1],path_len-1);	

		/* Get ContentID for CIA gen */
		char *pos2 = strstr(pos+1,":"); 
		if(pos2) {
			set->cia.contentId[content_index] = strtoul((pos2+1),NULL,0);
		}
		
		/* Return Next Arg Pos*/
		return 2;
	}
	
	else if(strncmp(argv[i],"-D",2) == 0){
		if(HasParam){
			PrintNoNeedParam("-DNAME=VALUE");
			return USR_BAD_ARG;
		}
		if(set->dname.m_items == 0){
			set->dname.m_items = 10;
			set->dname.u_items = 0;
			set->dname.items = malloc(sizeof(dname_item)*set->dname.m_items);
			if(!set->dname.items){
				fprintf(stderr,"[SETTING ERROR] Not enough memory\n");
				return MEM_ERROR;
			}
			memset(set->dname.items,0,sizeof(dname_item)*set->dname.m_items);
		}
		else if(set->dname.m_items == set->dname.u_items){
			set->dname.m_items *= 2;
			dname_item *tmp = malloc(sizeof(dname_item)*set->dname.m_items);
			if(!tmp){
				fprintf(stderr,"[SETTING ERROR] Not enough memory\n");
				return MEM_ERROR;
			}
			memset(tmp,0,sizeof(dname_item)*set->dname.m_items);
			memcpy(tmp,set->dname.items,sizeof(dname_item)*set->dname.u_items);
			free(set->dname.items);
			set->dname.items = tmp;
		}

		char *name_pos = (char*)(argv[i]+2);
		u32 name_len = 0;
		char *val_pos = strstr(name_pos,"=");
		u32 val_len = 0;
		if(!val_pos){
			fprintf(stderr,"[SETTING ERROR] Format: '%s' is invalid\n",argv[i]);
			return USR_BAD_ARG;
		}
		if(strlen(val_pos) < 2){
			fprintf(stderr,"[SETTING ERROR] Format: '%s' is invalid\n",argv[i]);
			return USR_BAD_ARG;
		}
		val_pos = (val_pos + 1);
		name_len = (val_pos - 2 - name_pos);
		set->dname.items[set->dname.u_items].name = malloc(name_len+1);
		memset(set->dname.items[set->dname.u_items].name,0,name_len+1);
		memcpy(set->dname.items[set->dname.u_items].name,name_pos,name_len);

		val_len = strlen(val_pos);
		set->dname.items[set->dname.u_items].value = malloc(val_len+1);
		memset(set->dname.items[set->dname.u_items].value,0,val_len+1);
		memcpy(set->dname.items[set->dname.u_items].value,val_pos,val_len);

		set->dname.u_items++;

		return 1;
	}
	

	// If not a valid argument
	fprintf(stderr,"[SETTING ERROR] Unrecognised argument '%s'\n",argv[i]);
	return USR_UNK_ARG;
}

int CheckArgumentCombination(user_settings *set)
{
	for(int i = 0; i < CIA_MAX_CONTENT; i++){
		if( i > CCI_MAX_CONTENT-1 && set->common.contentPath[i] && set->common.outFormat == CCI){
			fprintf(stderr,"[SETTING ERROR] Content indexes > %d are invalid for CCI\n",CCI_MAX_CONTENT-1);
			return USR_BAD_ARG;
		}
		if(set->common.contentPath[i] && (set->common.outFormat == CXI || set->common.outFormat == CFA)){
			fprintf(stderr,"[SETTING ERROR] You cannot specify content while outputting CXI/CFA files\n");
			return USR_BAD_ARG;
		}
	}
	if((set->common.outFormat == CXI || set->common.outFormat == CFA) && set->ncch.ncchType != format_not_set){
		fprintf(stderr,"[SETTING ERROR] Arguments \"-f cxi|cfa\" and \"-ncch cxi|cfa\" are invalid\n");
		return USR_BAD_ARG;
	}
	if(set->ncch.ncchType != format_not_set && !set->ncch.buildNcch0){
		fprintf(stderr,"[SETTING ERROR] Arguments \"-content %s:0\" and \"-data\" cannot be used together\n",set->common.contentPath[0]);
		return USR_BAD_ARG;
	}

	if(set->cia.useDataTitleVer && set->cia.useNormTitleVer){
		fprintf(stderr,"[SETTING ERROR] Arguments \"-dver\" and \"-major\"/\"-minor\" cannot be used together\n");
		return USR_BAD_ARG;
	}

	if(set->ncch.elfPath && set->ncch.codePath){
		fprintf(stderr,"[SETTING ERROR] Arguments \"-elf\" and \"-code\" cannot be used together\n");
		return USR_BAD_ARG;
	}

	// Setting set->build_ncch_type if it isn't already set
	if(set->ncch.buildNcch0 && set->ncch.ncchType == format_not_set){
		if(set->common.outFormat == CCI || set->common.outFormat  == CIA) set->ncch.ncchType = CXI;
		else set->ncch.ncchType = set->common.outFormat;
	}

	bool buildCXI = set->ncch.ncchType == CXI;
	bool buildCFA = set->ncch.ncchType == CFA;
	// Detecting Required Arguments
	if(buildCXI && !set->ncch.elfPath && !set->ncch.codePath){
		PrintNeedsArgument("-elf");
		return USR_BAD_ARG;
	}
	if((buildCXI || buildCFA) && !set->common.rsfPath){
		PrintNeedsArgument("-rsf");
		return USR_BAD_ARG;
	}
	if(buildCXI && set->ncch.codePath && !set->ncch.exheaderPath){
		PrintNeedsArgument("-exheader");
		return USR_BAD_ARG;
	}

	// Reporting bad arguments
	if(!buildCXI && set->ncch.elfPath){
		PrintArgumentInvalid("-elf");
		return USR_BAD_ARG;
	}
	if(!buildCXI && set->ncch.codePath){
		PrintArgumentInvalid("-code");
		return USR_BAD_ARG;
	}
	if(!buildCXI && set->ncch.exheaderPath){
		PrintArgumentInvalid("-exheader");
		return USR_BAD_ARG;
	}
	if(!buildCXI && set->ncch.plainRegionPath){
		PrintArgumentInvalid("-plain-region");
		return USR_BAD_ARG;
	}
	if(!set->ncch.buildNcch0 && set->ncch.includeExefsLogo){
		PrintArgumentInvalid("-exefslogo");
		return USR_BAD_ARG;
	}
	if(!set->ncch.buildNcch0 && set->ncch.romfsPath){
		PrintArgumentInvalid("-romfs");
		return USR_BAD_ARG;
	}

	return 0;
}

void init_UserSettings(user_settings *usr_settings)
{
	memset(usr_settings,0,sizeof(user_settings));
}


void free_RsfSettings(rsf_settings *set)
{
	//Option
	free(set->Option.PageSize);
	for(u32 i = 0; i < set->Option.AppendSystemCallNum; i++){
		free(set->Option.AppendSystemCall[i]);
	}
	free(set->Option.AppendSystemCall);

	//AccessControlInfo
	free(set->AccessControlInfo.ProgramId);
	free(set->AccessControlInfo.IdealProcessor);
	free(set->AccessControlInfo.Priority);
	free(set->AccessControlInfo.MemoryType);
	free(set->AccessControlInfo.SystemMode);
	free(set->AccessControlInfo.CoreVersion);
	free(set->AccessControlInfo.HandleTableSize);
	free(set->AccessControlInfo.SystemSaveDataId1);
	free(set->AccessControlInfo.SystemSaveDataId2);
	free(set->AccessControlInfo.OtherUserSaveDataId1);
	free(set->AccessControlInfo.OtherUserSaveDataId2);
	free(set->AccessControlInfo.OtherUserSaveDataId3);
	free(set->AccessControlInfo.ExtSaveDataId);
	free(set->AccessControlInfo.SystemMode);	
	free(set->AccessControlInfo.AffinityMask);
	free(set->AccessControlInfo.DescVersion);
	free(set->AccessControlInfo.CryptoKey);
	free(set->AccessControlInfo.ResourceLimitCategory);
	free(set->AccessControlInfo.ReleaseKernelMajor);
	free(set->AccessControlInfo.ReleaseKernelMinor);
	free(set->AccessControlInfo.MaxCpu);

	for(u32 i = 0; i < set->AccessControlInfo.MemoryMappingNum; i++){
		free(set->AccessControlInfo.MemoryMapping[i]);
	}
	free(set->AccessControlInfo.MemoryMapping);
	
	for(u32 i = 0; i < set->AccessControlInfo.IORegisterMappingNum; i++){
		free(set->AccessControlInfo.IORegisterMapping[i]);
	}
	free(set->AccessControlInfo.IORegisterMapping);
	
	for(u32 i = 0; i < set->AccessControlInfo.FileSystemAccessNum; i++){
		free(set->AccessControlInfo.FileSystemAccess[i]);
	}
	free(set->AccessControlInfo.FileSystemAccess);
	
	for(u32 i = 0; i < set->AccessControlInfo.IoAccessControlNum; i++){
		free(set->AccessControlInfo.IoAccessControl[i]);
	}
	free(set->AccessControlInfo.IoAccessControl);
	
	for(u32 i = 0; i < set->AccessControlInfo.InterruptNumbersNum; i++){
		free(set->AccessControlInfo.InterruptNumbers[i]);
	}
	free(set->AccessControlInfo.InterruptNumbers);
	
	for(u32 i = 0; i < set->AccessControlInfo.SystemCallAccessNum; i++){
		free(set->AccessControlInfo.SystemCallAccess[i]);
	}
	free(set->AccessControlInfo.SystemCallAccess);
	
	for(u32 i = 0; i < set->AccessControlInfo.ServiceAccessControlNum; i++){
		free(set->AccessControlInfo.ServiceAccessControl[i]);
	}
	free(set->AccessControlInfo.ServiceAccessControl);
	
	for(u32 i = 0; i < set->AccessControlInfo.StorageIdNum; i++){
		free(set->AccessControlInfo.StorageId[i]);
	}
	free(set->AccessControlInfo.StorageId);

	for(u32 i = 0; i < set->AccessControlInfo.AccessibleSaveDataIdsNum; i++){
		free(set->AccessControlInfo.AccessibleSaveDataIds[i]);
	}
	free(set->AccessControlInfo.AccessibleSaveDataIds);
	
	//SystemControlInfo
	free(set->SystemControlInfo.AppType);
	free(set->SystemControlInfo.StackSize);
	free(set->SystemControlInfo.RemasterVersion);
	free(set->SystemControlInfo.JumpId);
	
	for(u32 i = 0; i < set->SystemControlInfo.DependencyNum; i++){
		free(set->SystemControlInfo.Dependency[i]);
	}
	free(set->SystemControlInfo.Dependency);
	
	//BasicInfo
	free(set->BasicInfo.Title);
	free(set->BasicInfo.CompanyCode);
	free(set->BasicInfo.ProductCode);
	free(set->BasicInfo.MediaSize);
	free(set->BasicInfo.ContentType);
	free(set->BasicInfo.Logo);
	free(set->BasicInfo.BackupMemoryType);
	free(set->BasicInfo.InitialCode);
	
	//Rom
	free(set->Rom.HostRoot);
	free(set->Rom.Padding);
	free(set->Rom.SaveDataSize);
	
	for(u32 i = 0; i < set->Rom.DefaultRejectNum; i++){
		free(set->Rom.DefaultReject[i]);
	}
	free(set->Rom.DefaultReject);
	
	for(u32 i = 0; i < set->Rom.RejectNum; i++){
		free(set->Rom.Reject[i]);
	}
	free(set->Rom.Reject);
	
	for(u32 i = 0; i < set->Rom.IncludeNum; i++){
		free(set->Rom.Include[i]);
	}
	free(set->Rom.Include);
	
	for(u32 i = 0; i < set->Rom.FileNum; i++){
		free(set->Rom.File[i]);
	}
	free(set->Rom.File);
	
	//ExeFs
	for(u32 i = 0; i < set->ExeFs.TextNum; i++){
		free(set->ExeFs.Text[i]);
	}
	free(set->ExeFs.Text);
	
	for(u32 i = 0; i < set->ExeFs.ReadOnlyNum; i++){
		free(set->ExeFs.ReadOnly[i]);
	}
	free(set->ExeFs.ReadOnly);
	
	for(u32 i = 0; i < set->ExeFs.ReadWriteNum; i++){
		free(set->ExeFs.ReadWrite[i]);
	}
	free(set->ExeFs.ReadWrite);
	
	//PlainRegion
	for(u32 i = 0; i < set->PlainRegionNum; i++){
		free(set->PlainRegion[i]);
	}
	free(set->PlainRegion);
	
	//TitleInfo
	free(set->TitleInfo.Platform);
	free(set->TitleInfo.Category);
	free(set->TitleInfo.UniqueId);
	free(set->TitleInfo.Version);
	free(set->TitleInfo.ContentsIndex);
	free(set->TitleInfo.Variation);
	free(set->TitleInfo.Use);
	free(set->TitleInfo.ChildIndex);
	free(set->TitleInfo.DemoIndex);
	free(set->TitleInfo.TargetCategory);
	
	for(u32 i = 0; i < set->TitleInfo.CategoryFlagsNum; i++){
		free(set->TitleInfo.CategoryFlags[i]);
	}
	free(set->TitleInfo.CategoryFlags);
	
	//CardInfo
	free(set->CardInfo.WritableAddress);
	free(set->CardInfo.CardType);
	free(set->CardInfo.CryptoType);
	free(set->CardInfo.CardDevice);
	free(set->CardInfo.MediaType);
	free(set->CardInfo.BackupWriteWaitTime);

	//CommonHeaderKey
	free(set->CommonHeaderKey.D);
	free(set->CommonHeaderKey.P);
	free(set->CommonHeaderKey.Q);
	free(set->CommonHeaderKey.DP);
	free(set->CommonHeaderKey.DQ);
	free(set->CommonHeaderKey.InverseQ);
	free(set->CommonHeaderKey.Modulus);
	free(set->CommonHeaderKey.Exponent);
	free(set->CommonHeaderKey.AccCtlDescSign);
	free(set->CommonHeaderKey.AccCtlDescBin);
}

void free_UserSettings(user_settings *usr_settings)
{
	// Free Content Paths
	if(usr_settings->common.contentPath){
		for(int i = 0; i < CIA_MAX_CONTENT; i++)
			free(usr_settings->common.contentPath[i]);
		free(usr_settings->common.contentPath);
	}
	
	// free -DNAME=VALUE
	for(u32 i = 0; i < usr_settings->dname.u_items; i++){
		free(usr_settings->dname.items[i].name);
		free(usr_settings->dname.items[i].value);
	}
	free(usr_settings->dname.items);

	// Free Spec File Setting
	free_RsfSettings(&usr_settings->common.rsfSet);

	// Free Key Data
	FreeKeys(&usr_settings->common.keys);
	
	// Free Working File
	free(usr_settings->common.workingFile.buffer);
		
	// Free outfile path, if malloc'd
	if(usr_settings->common.outFileName_mallocd) 
		free(usr_settings->common.outFileName);
	
	// Clear settings
	init_UserSettings(usr_settings);
	
	// Free
	free(usr_settings);
}

void PrintNeedsArgument(char *arg)
{
	fprintf(stderr,"[SETTING ERROR] Argument \"%s\" is required\n",arg);
}

void PrintArgumentInvalid(char *arg)
{
	fprintf(stderr,"[SETTING ERROR] Argument \"%s\" is invalid\n",arg);
}

void PrintNeedsParam(char *arg)
{
	fprintf(stderr,"[SETTING ERROR] \"%s\" requires a parameter\n",arg);
}

void PrintNoNeedParam(char *arg)
{
	fprintf(stderr,"[SETTING ERROR] \"%s\" does not take a parameter\n",arg);
}

void DisplayHelp(char *app_name)
{
	printf("CTR MAKEROM %d.%d",MAKEROM_VER_MAJOR,MAKEROM_VER_MINOR);
#ifndef PUBLIC_BUILD
	printf(" PRIVATE BUILD");
#endif
	printf("\n(C) 3DSGuy 2014\n");
	printf("Usage: %s [options... ]\n",app_name);
	printf("Option          Parameter           Explanation\n");
	printf("GLOBAL OPTIONS:\n");
	printf(" -help                              Display this text\n");
	printf(" -rsf           <file>              RSF File\n");
	printf(" -f             <out format>        Output Format (cxi|cfa|cci|cia)\n");
	printf(" -o             <file>              Output File\n");
	//printf(" -v                                 Verbose\n");
	printf(" -DNAME=VALUE                       Substitute values in Spec files\n");
	printf("KEY OPTIONS:\n");
#ifdef PUBLIC_BUILD
	printf(" -target        <t|c>               Target for crypto, defaults to 't'\n");
	printf("                                    't' Test(false) Keys & prod Certs\n");
	printf("                                    'c' User provides Keys & Certs\n");
#else
	printf(" -target        <t|d|p|c>           Target for crypto, defaults to 't'\n");
	printf("                                    't' Test(false) Keys & prod Certs\n");
	printf("                                    'd' Development Keys & Certs\n");
	printf("                                    'p' Production Keys & Certs\n");
	printf("                                    'c' User provides Keys & Certs\n");
#endif
	printf(" -keydir        <dir>               Key Directory (for use with \"-target c\")\n");
	printf(" -ckeyID        <u8 value>          Override the automatic commonKey selection\n");
	printf(" -showkeys                          Display the loaded keychain\n");
	printf("NCCH OPTIONS:\n");
	printf(" -elf           <file>              ELF File\n");
	printf(" -icon          <file>              Icon File\n");
	printf(" -banner        <file>              Banner File\n");
	printf(" -logo          <file>              Logo File (Overrides \"BasicInfo/Logo\" in RSF)\n");
	printf(" -desc          <apptype>:<fw>      Specify Access Descriptor Preset\n");
	printf(" -exefslogo                         Include Logo in ExeFs (Required for usage on <5.X Systems)\n");
	printf(" -data                              Specify if building a Data Archive when \"-f cia\"\n");
	printf("NCCH REBUILD OPTIONS:\n");
	printf(" -code          <code path>         Specify ExeFs code File\n");
	printf(" -exheader      <exhdr path>        ExHeader Template File\n");
	printf(" -plain-region  <pln region path>   PlainRegion File\n");
	printf(" -romfs         <romfs path>        RomFS File\n");	
	printf("CCI OPTIONS:\n");
#ifndef PUBLIC_BUILD
	printf(" -devcardcci                        Use SDK CardInfo Method\n");
#endif
	printf(" -content <filepath>:<index>        Specify content files\n");
	printf("CIA OPTIONS:\n");
#ifndef PUBLIC_BUILD
	printf(" -cci           <cci path>          Convert CCI to CIA\n");
	printf(" -srl           <srl path>          Use TWL SRL as Content0\n");
	printf(" -dlc                               Create DLC CIA\n");
#endif
	printf(" -content <filepath>:<index>:<id>   Specify content files\n");
	printf(" -major         <major version>     Specify Major Version\n");
	printf(" -minor         <minor version>     Specify Minor Version\n");
	printf(" -micro         <micro version>     Specify Micro Version\n");
	printf(" -dver          <datatitle ver>     Specify Data Title Version\n");
	printf(" -savesize      <size>              Savedata size\n");
	printf(" -rand                              Use a random title key\n");
	printf(" -nocryptcia                        Don't encrypt CIA contents\n");
}