138 lines
2.2 KiB
C
138 lines
2.2 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/*
|
|
* Copyright (C) 2020-2024 Microsoft Corporation. All rights reserved.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <stddef.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
|
|
static void usage(const char *const name)
|
|
{
|
|
printf("Usage: %s OutputFile (PolicyFile)\n", name);
|
|
exit(EINVAL);
|
|
}
|
|
|
|
static int policy_to_buffer(const char *pathname, char **buffer, size_t *size)
|
|
{
|
|
size_t fsize;
|
|
size_t read;
|
|
char *lbuf;
|
|
int rc = 0;
|
|
FILE *fd;
|
|
|
|
fd = fopen(pathname, "r");
|
|
if (!fd) {
|
|
rc = errno;
|
|
goto out;
|
|
}
|
|
|
|
fseek(fd, 0, SEEK_END);
|
|
fsize = ftell(fd);
|
|
rewind(fd);
|
|
|
|
lbuf = malloc(fsize);
|
|
if (!lbuf) {
|
|
rc = ENOMEM;
|
|
goto out_close;
|
|
}
|
|
|
|
read = fread((void *)lbuf, sizeof(*lbuf), fsize, fd);
|
|
if (read != fsize) {
|
|
rc = -1;
|
|
goto out_free;
|
|
}
|
|
|
|
*buffer = lbuf;
|
|
*size = fsize;
|
|
fclose(fd);
|
|
|
|
return rc;
|
|
|
|
out_free:
|
|
free(lbuf);
|
|
out_close:
|
|
fclose(fd);
|
|
out:
|
|
return rc;
|
|
}
|
|
|
|
static int write_boot_policy(const char *pathname, const char *buf, size_t size)
|
|
{
|
|
FILE *fd;
|
|
size_t i;
|
|
|
|
fd = fopen(pathname, "w");
|
|
if (!fd)
|
|
return errno;
|
|
|
|
fprintf(fd, "/* This file is automatically generated.");
|
|
fprintf(fd, " Do not edit. */\n");
|
|
fprintf(fd, "#include <linux/stddef.h>\n");
|
|
fprintf(fd, "\nextern const char *const ipe_boot_policy;\n\n");
|
|
fprintf(fd, "const char *const ipe_boot_policy =\n");
|
|
|
|
if (!buf || size == 0) {
|
|
fprintf(fd, "\tNULL;\n");
|
|
fclose(fd);
|
|
return 0;
|
|
}
|
|
|
|
fprintf(fd, "\t\"");
|
|
|
|
for (i = 0; i < size; ++i) {
|
|
switch (buf[i]) {
|
|
case '"':
|
|
fprintf(fd, "\\\"");
|
|
break;
|
|
case '\'':
|
|
fprintf(fd, "'");
|
|
break;
|
|
case '\n':
|
|
fprintf(fd, "\\n\"\n\t\"");
|
|
break;
|
|
case '\\':
|
|
fprintf(fd, "\\\\");
|
|
break;
|
|
case '\t':
|
|
fprintf(fd, "\\t");
|
|
break;
|
|
case '\?':
|
|
fprintf(fd, "\\?");
|
|
break;
|
|
default:
|
|
fprintf(fd, "%c", buf[i]);
|
|
}
|
|
}
|
|
fprintf(fd, "\";\n");
|
|
fclose(fd);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, const char *const argv[])
|
|
{
|
|
char *policy = NULL;
|
|
size_t len = 0;
|
|
int rc = 0;
|
|
|
|
if (argc < 2)
|
|
usage(argv[0]);
|
|
|
|
if (argc > 2) {
|
|
rc = policy_to_buffer(argv[2], &policy, &len);
|
|
if (rc != 0)
|
|
goto cleanup;
|
|
}
|
|
|
|
rc = write_boot_policy(argv[1], policy, len);
|
|
cleanup:
|
|
if (policy)
|
|
free(policy);
|
|
if (rc != 0)
|
|
perror("An error occurred during policy conversion: ");
|
|
return rc;
|
|
}
|