1
0
mirror of https://github.com/CTCaer/hekate.git synced 2024-12-25 15:17:19 +00:00

Update bin2c.c

Fix behaviour under Windows if compiled with MinGW / TDM-GCC.
This commit is contained in:
Pablo Curiel 2021-02-08 14:10:38 -04:00
parent 4e7e5081a7
commit be23d1fa15

View File

@ -9,7 +9,6 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <fcntl.h>
#include <ctype.h> #include <ctype.h>
#include <unistd.h> #include <unistd.h>
@ -34,32 +33,39 @@ main ( int argc, char* argv[] )
{ {
unsigned char buf[BUFSIZ]; unsigned char buf[BUFSIZ];
char* ident; char* ident;
int fd, i, total, rd, need_comma; FILE *fd;
size_t size, i, total, blksize = BUFSIZ;
int need_comma = 0;
if ( argc < 2 ) if ( argc != 2 )
{ {
fprintf ( stderr, "Usage: %s binary_file > output_file\n", argv[0] ); fprintf ( stderr, "Usage: %s binary_file > output_file\n", argv[0] );
return -1; return -1;
} }
fd = open ( argv[1], O_RDONLY ); fd = fopen ( argv[1], "rb" );
if ( fd == -1 ) if ( fd == NULL )
{ {
fprintf ( stderr, "%s: can't open %s for reading\n", argv[0], argv[1] ); fprintf ( stderr, "%s: can't open %s for reading\n", argv[0], argv[1] );
return -1; return -1;
} }
fseek(fd, 0, SEEK_END);
size = ftell(fd);
rewind(fd);
ident = make_ident ( argv[1] ); ident = make_ident ( argv[1] );
printf ( "static const unsigned char __attribute__((section (\"._%s\"))) %s[] = {", ident, ident ); printf ( "static const unsigned char __attribute__((section (\"._%s\"))) %s[] = {", ident, ident );
for ( total = 0, need_comma = 0; ( rd = read ( fd, buf, BUFSIZ ) ) != 0; ) for ( total = 0; total < size; )
{ {
if ( rd == -1 ) if ( size - total < blksize ) blksize = size - total;
if ( fread ( buf, 1, blksize, fd ) != blksize )
{ {
fprintf ( stderr, "%s: file read error\n", argv[0] ); fprintf ( stderr, "%s: file read error\n", argv[0] );
return -1; return -1;
} }
for ( i = 0; i < rd; i++ ) for ( i = 0; i < blksize; i++ )
{ {
if ( need_comma ) printf ( ", " ); if ( need_comma ) printf ( ", " );
else need_comma = 1; else need_comma = 1;
@ -70,7 +76,7 @@ main ( int argc, char* argv[] )
} }
printf ( "\n};\n" ); printf ( "\n};\n" );
close ( fd ); fclose ( fd );
free ( ident ); free ( ident );
return 0; return 0;