Skip to content

Commit 5f01b02

Browse files
committed
Cleanup output of pe_bitmap_loader
1 parent f8214de commit 5f01b02

1 file changed

Lines changed: 9 additions & 11 deletions

File tree

pe_bitmap_loader/main.c

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,9 @@ int main(int argc, char ** argv) {
4444
if (bitmap_ids) {
4545
free(bitmap_ids);
4646
}
47+
if (bitmap_names) {
48+
free(bitmap_names);
49+
}
4750
if (languages) {
4851
free(languages);
4952
}
@@ -54,40 +57,35 @@ int main(int argc, char ** argv) {
5457
for (uint16_t li = 0; li < language_count; li++) {
5558
printf("Language with id %u:\n", languages[li]);
5659
for (uint16_t bi = 0; bi < bitmap_count; bi++) {
57-
printf("Found bitmap with id %u\n", bitmap_ids[bi]);
5860
uint32_t file_size = 0;
5961
void * data = PeResourceLoader_GetResource(loader, PRL_TYPE_BITMAP, languages[li], bitmap_ids[bi], &file_size);
60-
printf("File size is %u\n", file_size);
6162
if (file_size > 0) {
62-
char * file_name = calloc(34, sizeof(char));
63-
snprintf(file_name, 34, "%u_%u.bmp", bitmap_ids[bi], languages[li]);
64-
printf("Using file name %s\n", file_name);
63+
char * file_name = calloc(32 + 32 + 5, sizeof(char));
64+
snprintf(file_name, 32 + 32 + 5, "%u_%u.bmp", bitmap_ids[bi], languages[li]);
65+
printf("Exporting file %s\n", file_name);
6566
FILE * file = fopen(file_name, "wb");
6667
free(file_name);
6768
fwrite(data, 1, file_size, file);
6869
fclose(file);
6970
}
70-
if (data)
71+
if (data != NULL)
7172
free(data);
7273
}
7374
for (uint16_t bi = 0; bi < bitmap_name_count; bi++) {
74-
printf("Found bitmap with name %s\n", bitmap_names[bi].name);
7575
uint32_t file_size = 0;
7676
void * data = PeResourceLoader_GetResource(loader, PRL_TYPE_BITMAP, languages[li], bitmap_names[bi].offset, &file_size);
77-
printf("File size is %u\n", file_size);
7877
if (file_size > 0) {
7978
char * file_name = calloc(bitmap_names[bi].name_length + 32 + 4, sizeof(char));
8079
snprintf(file_name, bitmap_names[bi].name_length + 32 + 4, "%s_%u.bmp", bitmap_names[bi].name, languages[li]);
81-
printf("Using file name %s\n", file_name);
80+
printf("Exporting file %s\n", file_name);
8281
FILE * file = fopen(file_name, "wb");
8382
free(file_name);
8483
fwrite(data, 1, file_size, file);
8584
fclose(file);
8685
}
87-
if (data)
86+
if (data != NULL)
8887
free(data);
8988
}
90-
printf("\n");
9189
}
9290
free(languages);
9391
free(bitmap_ids);

0 commit comments

Comments
 (0)