140 lines
4.0 KiB
C
140 lines
4.0 KiB
C
#include <fs/initrd.h>
|
|
|
|
int main(int argc, char **argv) {
|
|
|
|
if (argc < 2) {
|
|
fprintf(stderr, "No input files\n");
|
|
return 1;
|
|
}
|
|
|
|
fprintf (stderr, "Rozmiar struktury: %x\n", sizeof(initrd_header_t));
|
|
|
|
for (u32int i = 1; i < argc; i++) {
|
|
fprintf(stderr, "File nr %d to embed: '%s'\n",
|
|
i, argv[i]);
|
|
}
|
|
|
|
u32int nrFiles = 0;
|
|
|
|
FILE *f = fopen("initrd.img", "w+");
|
|
if (!f) {
|
|
fprintf(stderr, "I cannot open file 'initrd.img for write\n");
|
|
return 2;
|
|
}
|
|
|
|
u32int offset = 0;
|
|
u32int buffer[12];
|
|
memset(buffer, 0, sizeof(buffer));
|
|
fwrite(&nrFiles, 4, 1, f); // teraz tylko dla wyrównania danych
|
|
fwrite(buffer, 1, 12, f); // wyrównanie do 16
|
|
|
|
initrd_header_t *headers = (initrd_header_t *)
|
|
malloc(sizeof(initrd_header_t) * (argc - 1));
|
|
memset(headers, 0, sizeof(initrd_header_t) * (argc - 1));
|
|
|
|
fprintf(stderr, "Filling in headers\n");
|
|
|
|
for (u32int i = 1; i < argc; i++) {
|
|
fprintf (stderr, "writing file %s to initrd.img\n",
|
|
argv[i]);
|
|
|
|
FILE *fAttachment = fopen(argv[i], "r");
|
|
if (!fAttachment) {
|
|
fprintf (stderr, "cannot open attachment %s to read\n",
|
|
argv[i]);
|
|
continue;
|
|
}
|
|
|
|
if (fseek(fAttachment, 0, SEEK_END) < 0) {
|
|
fprintf (stderr, "cannot rewind attachment %s to the end\n",
|
|
argv[i]);
|
|
continue;
|
|
}
|
|
|
|
long size = ftell(fAttachment);
|
|
if (size < 0) {
|
|
fprintf (stderr, "cannot acquire current position in file %s\n",
|
|
argv[i]);
|
|
continue;
|
|
}
|
|
|
|
fprintf (stderr, "Rozmiar pliku: 0x%x\n", size);
|
|
|
|
if (fseek(fAttachment, 0, SEEK_SET) < 0) {
|
|
fprintf (stderr, "cannot rewind attachment %s to the start\n",
|
|
argv[i]);
|
|
continue;
|
|
}
|
|
|
|
fclose(fAttachment); // zamykamy, na co nam jest jeszcze potrzebny
|
|
|
|
headers[nrFiles].magic = INITRD_MAGIC;
|
|
char *bName = basename(argv[i]);
|
|
fprintf (stderr, "Base name: %s\n", bName);
|
|
strncpy(headers[nrFiles].name ,bName, INITRD_NAME_LENGTH);
|
|
headers[nrFiles].length = size;
|
|
headers[nrFiles].offset = offset;
|
|
offset += size;
|
|
|
|
fprintf (stderr, "[%d] header MAGIC: 0x%x\n", nrFiles,
|
|
headers[nrFiles].magic);
|
|
fprintf (stderr, "[%d] header name: '%s'\n", nrFiles,
|
|
headers[nrFiles].name);
|
|
fprintf (stderr, "[%d] header offset: 0x%x\n", nrFiles,
|
|
headers[nrFiles].offset);
|
|
fprintf (stderr, "[%d] header length: 0x%x\n", nrFiles,
|
|
headers[nrFiles].length);
|
|
|
|
nrFiles++;
|
|
// zwiększamy liczbę plików, które zapisaliśmy faktycznie
|
|
// do initrd.img
|
|
|
|
}
|
|
|
|
fwrite(headers, 1, sizeof(initrd_header_t) * nrFiles, f);
|
|
|
|
// zapisane nagłówki initrd.
|
|
|
|
fprintf(stderr, "Filling in file content\n");
|
|
|
|
u32int idx = 0;
|
|
|
|
for (u32int i = 1; i < argc; i++) {
|
|
fprintf (stderr, "writing file %s to initrd.img\n",
|
|
argv[i]);
|
|
|
|
FILE *fAttachment = fopen(argv[i], "r");
|
|
if (!fAttachment) {
|
|
fprintf (stderr, "cannot open attachment %s to read\n",
|
|
argv[i]);
|
|
continue;
|
|
}
|
|
|
|
char *content = (char *)malloc(sizeof(char) * headers[idx].length);
|
|
if (!content) {
|
|
fprintf (stderr, "cannot allocate buffer for file content\n");
|
|
continue;
|
|
}
|
|
int bytesRead = fread(content, 1, headers[idx].length, fAttachment);
|
|
if (bytesRead < 0) {
|
|
fprintf (stderr, "cannot read open attachment file: %s\n",
|
|
argv[i]);
|
|
continue;
|
|
}
|
|
|
|
fclose(fAttachment); // zamykamy, na co nam jest jeszcze potrzebny
|
|
|
|
fwrite(content, 1, headers[idx].length, f);
|
|
free(content);
|
|
idx++;
|
|
}
|
|
|
|
fseek(f, 0, SEEK_SET);
|
|
|
|
fwrite(&nrFiles, 4, 1, f); // nadpisujemy prawidłową liczbę plików
|
|
|
|
fclose(f);
|
|
free(headers);
|
|
return 0;
|
|
}
|