mirror of https://github.com/SuperBFG7/ympd
95 lines
2.4 KiB
C
95 lines
2.4 KiB
C
/* This program is used to embed arbitrary data into a C binary. It takes
|
|
* a list of files as an input, and produces a .c data file that contains
|
|
* contents of all these files as collection of char arrays.
|
|
*
|
|
* Usage: ./mkdata <this_file> <file1> [file2, ...] > embedded_data.c
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <err.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
|
|
const char* header =
|
|
"#include <stddef.h>\n"
|
|
"#include <string.h>\n"
|
|
"#include <sys/types.h>\n"
|
|
"#include \"src/http_server.h\"\n"
|
|
"\n"
|
|
"static const struct embedded_file embedded_files[] = {\n";
|
|
|
|
const char* footer =
|
|
" {NULL, NULL, NULL, 0}\n"
|
|
"};\n"
|
|
"\n"
|
|
"const struct embedded_file *find_embedded_file(const char *name) {\n"
|
|
" const struct embedded_file *p;\n"
|
|
" for (p = embedded_files; p->name != NULL; p++)\n"
|
|
" if (!strcmp(p->name, name))\n"
|
|
" return p;\n"
|
|
" return NULL;\n"
|
|
"}\n";
|
|
|
|
static const char* get_mime(char* filename)
|
|
{
|
|
const char *extension = strrchr(filename, '.');
|
|
if(!strcmp(extension, ".js"))
|
|
return "application/javascript";
|
|
if(!strcmp(extension, ".css"))
|
|
return "text/css";
|
|
if(!strcmp(extension, ".ico"))
|
|
return "image/vnd.microsoft.icon";
|
|
if(!strcmp(extension, ".woff"))
|
|
return "application/font-woff";
|
|
if(!strcmp(extension, ".ttf"))
|
|
return "application/x-font-ttf";
|
|
if(!strcmp(extension, ".eot"))
|
|
return "application/octet-stream";
|
|
if(!strcmp(extension, ".svg"))
|
|
return "image/svg+xml";
|
|
if(!strcmp(extension, ".html"))
|
|
return "text/html";
|
|
return "text/plain";
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
int i, j, buf;
|
|
FILE *fd;
|
|
|
|
if(argc <= 1)
|
|
err(EXIT_FAILURE, "Usage: ./%s <this_file> <file1> [file2, ...] > embedded_data.c", argv[0]);
|
|
|
|
for(i = 1; i < argc; i++)
|
|
{
|
|
fd = fopen(argv[i], "r");
|
|
if(!fd)
|
|
err(EXIT_FAILURE, "%s", argv[i]);
|
|
|
|
printf("static const unsigned char v%d[] = {", i);
|
|
|
|
j = 0;
|
|
while((buf = fgetc(fd)) != EOF)
|
|
{
|
|
if(!(j % 12))
|
|
putchar('\n');
|
|
|
|
printf(" %#04x, ", buf);
|
|
j++;
|
|
}
|
|
printf(" 0x00\n};\n\n");
|
|
fclose(fd);
|
|
}
|
|
fputs(header, stdout);
|
|
|
|
for(i = 1; i < argc; i++)
|
|
{
|
|
printf(" {\"%s\", v%d, \"%s\", sizeof(v%d) - 1}, \n",
|
|
argv[i]+6, i, get_mime(argv[i]), i);
|
|
}
|
|
|
|
fputs(footer, stdout);
|
|
return EXIT_SUCCESS;
|
|
}
|