268 lines
9.0 KiB
C
268 lines
9.0 KiB
C
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdbool.h>
|
|
#include <stdlib.h>
|
|
|
|
// the program writes generated text to stdout
|
|
// other messages must be sent to stderr
|
|
#define printf USE_FPRINTF!!!
|
|
|
|
#define exit_with_error(FORMAT, ARGS...) ({ \
|
|
fprintf(stderr, "\e[91m\nerror: "FORMAT"\n", ## ARGS); \
|
|
exit(4); \
|
|
})
|
|
|
|
#define argis(STR) streq(arg, STR) || \
|
|
streq(arg, "-"STR) || \
|
|
streq(arg, "--"STR)
|
|
|
|
////////////////////////////////////////////////
|
|
// STR FUNCTIONS //
|
|
////////////////////////////////////////////////
|
|
|
|
bool streq(const char* key0,const char* key1){
|
|
if(!key0) return key1 ? false : true;
|
|
if(!key1) return false;
|
|
while(*key0&&*key1)
|
|
if(*key0++ != *key1++)
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
char* basedir_exclude(const char* path, const char* basename){
|
|
int len=strlen(path);
|
|
int base_len=strlen(basename);
|
|
int i=0;
|
|
if(len>0 && base_len>0){
|
|
if(basename[base_len-1]=='/' || basename[base_len-1]=='\\')
|
|
base_len--; // remove trailing path separator
|
|
if(len>base_len){
|
|
while(i<base_len){
|
|
if(path[i]!=basename[i])
|
|
break;
|
|
i++;
|
|
}
|
|
if(i<base_len || basename[i]!='/' || basename[i]!='\\')
|
|
i=0;
|
|
else i++;
|
|
}
|
|
}
|
|
int rzlt_len=len-i;
|
|
char* rzlt=malloc(rzlt_len+1);
|
|
memcpy(rzlt, path+i, rzlt_len);
|
|
return rzlt;
|
|
}
|
|
|
|
///@returns ptr to string of <count> spaces
|
|
char* mult_space(int count){
|
|
char* _mult_space_buf=malloc(count+1);
|
|
_mult_space_buf[count]=0;
|
|
memset(_mult_space_buf, ' ', count);
|
|
return _mult_space_buf;
|
|
}
|
|
|
|
void print_line(bool print_lines, bool start_new_line){
|
|
if(!print_lines)
|
|
return;
|
|
fflush(stdout);
|
|
if(start_new_line)
|
|
fprintf(stderr, "\n");
|
|
fprintf(stderr, "------------------------------------------------------------\n");
|
|
fflush(stderr);
|
|
}
|
|
|
|
////////////////////////////////////////////////
|
|
// WRITE CODE //
|
|
////////////////////////////////////////////////
|
|
|
|
bool _first_file_processed=false;
|
|
unsigned int _input_files_n=0;
|
|
|
|
void process_file(const char* input_file_path,
|
|
const char* embedded_file_path,
|
|
FILE* out_file,
|
|
const char* out_file_path)
|
|
{
|
|
bool print_lines=streq(out_file_path, "this_file");
|
|
|
|
if(!_first_file_processed){
|
|
print_line(print_lines, false);
|
|
|
|
int onml=strlen(out_file_path);
|
|
if(onml>43) onml=0;
|
|
char* _sp1=mult_space(43-onml);
|
|
fprintf(out_file,
|
|
// "////////////////////////////////////////////////////////////\n"
|
|
// "// //\n"
|
|
// "//%s%s%s//\n"
|
|
// "// //\n"
|
|
"////////////////////////////////////////////////////////////\n"
|
|
"// This file has been generated by resource_embedder //\n"
|
|
"// src: https://github.com/Timerix22/resource_embedder //\n"
|
|
"////////////////////////////////////////////////////////////\n"
|
|
"// USAGE: //\n"
|
|
"// Put inside any .c file //\n"
|
|
"// #define EMBEDDED_RESOURCE_POSTFIX your_postfix //\n"
|
|
"// #include \"%s\"%s//\n"
|
|
"// //\n"
|
|
"// Then you can access embedded files through //\n"
|
|
"// EmbeddedResourceFile_table_your_postfix. You can put //\n"
|
|
"// table content into a hashtable or get it by index. //\n"
|
|
"////////////////////////////////////////////////////////////\n"
|
|
"\n",
|
|
// mult_space(28-onml/2), outname, mult_space(28-onml/2+onml%2),
|
|
out_file_path, _sp1);
|
|
free(_sp1);
|
|
fprintf(out_file,
|
|
"\n"
|
|
"typedef struct {\n"
|
|
" char* path;\n"
|
|
" char* data;\n"
|
|
" unsigned long long size;\n"
|
|
"} EmbeddedResourceFile;\n"
|
|
"\n"
|
|
"#define RSCAT(A,B,C...) A##B##C"
|
|
"\n"
|
|
"#ifdef EMBEDDED_RESOURCE_POSTFIX\n"
|
|
" #define _EmbeddedResourceFile_table(P) \\\n"
|
|
" RSCAT(EmbeddedResourceFile_table_, P)\n"
|
|
" #define _EmbeddedResourceFile_table_count(P) \\\n"
|
|
" RSCAT(EmbeddedResourceFile_table_, P, _count)\n"
|
|
"#else\n"
|
|
" #define _EmbeddedResourceFile_table(P) \\\n"
|
|
" EmbeddedResourceFile_table\n"
|
|
" #define _EmbeddedResourceFile_table_count(P) \\\n"
|
|
" EmbeddedResourceFile_table_count\n"
|
|
"#endif\n"
|
|
"extern EmbeddedResourceFile _EmbeddedResourceFile_table(EMBEDDED_RESOURCE_POSTFIX)[];\n"
|
|
"extern unsigned int _EmbeddedResourceFile_table_count(EMBEDDED_RESOURCE_POSTFIX);"
|
|
"\n"
|
|
"\n"
|
|
"EmbeddedResourceFile _EmbeddedResourceFile_table(EMBEDDED_RESOURCE_POSTFIX)[]={\n");
|
|
print_line(print_lines, false);
|
|
}
|
|
|
|
fprintf(stderr, "reading file %s\n", input_file_path);
|
|
print_line(print_lines, false);
|
|
|
|
|
|
FILE* input_file=fopen(input_file_path, "rb");
|
|
if(input_file==NULL)
|
|
exit_with_error("can't open file %s", input_file_path);
|
|
|
|
if(_first_file_processed)
|
|
fprintf(out_file, ",\n");
|
|
// writing input file code header
|
|
fprintf(out_file,
|
|
" {\n"
|
|
" .path=(char*)\"%s\",\n"
|
|
" .data=(char[]){",
|
|
embedded_file_path);
|
|
|
|
// writing input file content
|
|
int byte=0;
|
|
unsigned long long size=0;
|
|
while( (byte=fgetc(input_file)) != EOF ){
|
|
if(size%16==0)
|
|
fprintf(out_file, "\n ");
|
|
fprintf(out_file, "0x%02X,", byte);
|
|
size+=1;
|
|
}
|
|
// trailing zero to use file.data as cstring
|
|
fprintf(out_file, "0x0");
|
|
|
|
fprintf(out_file,
|
|
"\n },\n"
|
|
" .size=%llu\n"
|
|
" }",
|
|
size);
|
|
fflush(out_file);
|
|
|
|
print_line(print_lines,true);
|
|
if(ferror(input_file)){
|
|
perror("unexpected end of stream: ");
|
|
exit_with_error("input read error");
|
|
}
|
|
fclose(input_file);
|
|
|
|
fprintf(stderr, "read %llu bytes\n", size);
|
|
|
|
_first_file_processed=true;
|
|
_input_files_n++;
|
|
}
|
|
|
|
|
|
|
|
////////////////////////////////////////////////
|
|
// MAIN //
|
|
////////////////////////////////////////////////
|
|
|
|
int main(const int argc, const char * const* argv){
|
|
FILE* out_file=NULL;
|
|
const char* out_file_path=NULL;
|
|
const char* basedir="";
|
|
|
|
for(int i=1; i<argc; i++){
|
|
const char* arg=argv[i];
|
|
print_line(true, false);
|
|
fprintf(stderr, "arg: %s\n", arg);
|
|
|
|
if(argis("h") || argis("help")){
|
|
fprintf(stderr, "Usage:\n"
|
|
" resource_embedder [options] [file_options] -i input_file [[file_options] -i input_file]...\n"
|
|
"Options:\n"
|
|
" -o output file path (default=stdout)\n"
|
|
"File options:\n"
|
|
" -d base dir path that will be removed from compiled strings\n");
|
|
return 0;
|
|
}
|
|
|
|
else if(argis("o")){
|
|
const char* file_path=argv[++i];
|
|
if(out_file!=NULL)
|
|
exit_with_error("can't set out_file to %s, because it is already set to %s",
|
|
file_path, out_file_path);
|
|
out_file_path=file_path;
|
|
out_file=fopen(out_file_path, "wb");
|
|
if(out_file==NULL)
|
|
exit_with_error("can't open file %s", out_file_path);
|
|
fprintf(stderr, "output file has been set to %s\n", out_file_path);
|
|
}
|
|
|
|
else if(argis("i")){
|
|
const char* input_file_path=argv[++i];
|
|
if(out_file==NULL){
|
|
out_file=stdout;
|
|
out_file_path="this_file";
|
|
}
|
|
char* embedded_path=basedir_exclude(input_file_path, basedir);
|
|
fprintf(stderr, "input file path: %s\n", arg);
|
|
fprintf(stderr, "embedded path: %s\n", embedded_path);
|
|
process_file(input_file_path, embedded_path, out_file, out_file_path);
|
|
free(embedded_path);
|
|
}
|
|
|
|
else exit_with_error("unknown argument: %s", arg);
|
|
}
|
|
|
|
if(out_file_path!=NULL){
|
|
bool out_file_set=!streq(out_file_path, "this_file");
|
|
if(!out_file_set)
|
|
print_line(true, false);
|
|
fprintf(out_file,
|
|
"\n};\n\n"
|
|
"unsigned int _EmbeddedResourceFile_table_count(EMBEDDED_RESOURCE_POSTFIX)=%u;\n"
|
|
"\n"
|
|
"#undef _EmbeddedResourceFile_table\n"
|
|
"#undef _EmbeddedResourceFile_table_count\n"
|
|
"#undef EMBEDDED_RESOURCE_POSTFIX\n"
|
|
"#undef RSCAT\n",
|
|
_input_files_n);
|
|
if(!out_file_set)
|
|
fclose(out_file);
|
|
print_line(true, false);
|
|
}
|
|
|
|
return 0;
|
|
}
|