|
Packit Service |
87beb7 |
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
Packit Service |
87beb7 |
/*
|
|
Packit Service |
87beb7 |
* Copyright (c) 2017 Konsulko Group Inc. All rights reserved.
|
|
Packit Service |
87beb7 |
*
|
|
Packit Service |
87beb7 |
* Author:
|
|
Packit Service |
87beb7 |
* Pantelis Antoniou <pantelis.antoniou@konsulko.com>
|
|
Packit Service |
87beb7 |
*/
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
#include <assert.h>
|
|
Packit Service |
87beb7 |
#include <ctype.h>
|
|
Packit Service |
87beb7 |
#include <getopt.h>
|
|
Packit Service |
87beb7 |
#include <stdio.h>
|
|
Packit Service |
87beb7 |
#include <stdlib.h>
|
|
Packit Service |
87beb7 |
#include <string.h>
|
|
Packit Service |
87beb7 |
#include <inttypes.h>
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
#include <libfdt.h>
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
#include "util.h"
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
#define BUF_INCREMENT 65536
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
/* Usage related data. */
|
|
Packit Service |
87beb7 |
static const char usage_synopsis[] =
|
|
Packit Service |
87beb7 |
"apply a number of overlays to a base blob\n"
|
|
Packit Service |
87beb7 |
" fdtoverlay <options> [<overlay.dtbo> [<overlay.dtbo>]]\n"
|
|
Packit Service |
87beb7 |
"\n"
|
|
Packit Service |
87beb7 |
USAGE_TYPE_MSG;
|
|
Packit Service |
87beb7 |
static const char usage_short_opts[] = "i:o:v" USAGE_COMMON_SHORT_OPTS;
|
|
Packit Service |
87beb7 |
static struct option const usage_long_opts[] = {
|
|
Packit Service |
87beb7 |
{"input", required_argument, NULL, 'i'},
|
|
Packit Service |
87beb7 |
{"output", required_argument, NULL, 'o'},
|
|
Packit Service |
87beb7 |
{"verbose", no_argument, NULL, 'v'},
|
|
Packit Service |
87beb7 |
USAGE_COMMON_LONG_OPTS,
|
|
Packit Service |
87beb7 |
};
|
|
Packit Service |
87beb7 |
static const char * const usage_opts_help[] = {
|
|
Packit Service |
87beb7 |
"Input base DT blob",
|
|
Packit Service |
87beb7 |
"Output DT blob",
|
|
Packit Service |
87beb7 |
"Verbose messages",
|
|
Packit Service |
87beb7 |
USAGE_COMMON_OPTS_HELP
|
|
Packit Service |
87beb7 |
};
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
int verbose = 0;
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
static void *apply_one(char *base, const char *overlay, size_t *buf_len,
|
|
Packit Service |
87beb7 |
const char *name)
|
|
Packit Service |
87beb7 |
{
|
|
Packit Service |
87beb7 |
char *tmp = NULL;
|
|
Packit Service |
87beb7 |
char *tmpo;
|
|
Packit Service |
87beb7 |
int ret;
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
/*
|
|
Packit Service |
87beb7 |
* We take a copies first, because a a failed apply can trash
|
|
Packit Service |
87beb7 |
* both the base blob and the overlay
|
|
Packit Service |
87beb7 |
*/
|
|
Packit Service |
87beb7 |
tmpo = xmalloc(fdt_totalsize(overlay));
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
do {
|
|
Packit Service |
87beb7 |
tmp = xrealloc(tmp, *buf_len);
|
|
Packit Service |
87beb7 |
ret = fdt_open_into(base, tmp, *buf_len);
|
|
Packit Service |
87beb7 |
if (ret) {
|
|
Packit Service |
87beb7 |
fprintf(stderr,
|
|
Packit Service |
87beb7 |
"\nFailed to make temporary copy: %s\n",
|
|
Packit Service |
87beb7 |
fdt_strerror(ret));
|
|
Packit Service |
87beb7 |
goto fail;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
memcpy(tmpo, overlay, fdt_totalsize(overlay));
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
ret = fdt_overlay_apply(tmp, tmpo);
|
|
Packit Service |
87beb7 |
if (ret == -FDT_ERR_NOSPACE) {
|
|
Packit Service |
87beb7 |
*buf_len += BUF_INCREMENT;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
} while (ret == -FDT_ERR_NOSPACE);
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
if (ret) {
|
|
Packit Service |
87beb7 |
fprintf(stderr, "\nFailed to apply '%s': %s\n",
|
|
Packit Service |
87beb7 |
name, fdt_strerror(ret));
|
|
Packit Service |
87beb7 |
goto fail;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
free(base);
|
|
Packit Service |
87beb7 |
free(tmpo);
|
|
Packit Service |
87beb7 |
return tmp;
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
fail:
|
|
Packit Service |
87beb7 |
free(tmpo);
|
|
Packit Service |
87beb7 |
if (tmp)
|
|
Packit Service |
87beb7 |
free(tmp);
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
return NULL;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
static int do_fdtoverlay(const char *input_filename,
|
|
Packit Service |
87beb7 |
const char *output_filename,
|
|
Packit Service |
87beb7 |
int argc, char *argv[])
|
|
Packit Service |
87beb7 |
{
|
|
Packit Service |
87beb7 |
char *blob = NULL;
|
|
Packit Service |
87beb7 |
char **ovblob = NULL;
|
|
Packit Service |
87beb7 |
size_t buf_len;
|
|
Packit Service |
87beb7 |
int i, ret = -1;
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
blob = utilfdt_read(input_filename, &buf_len);
|
|
Packit Service |
87beb7 |
if (!blob) {
|
|
Packit Service |
87beb7 |
fprintf(stderr, "\nFailed to read '%s'\n", input_filename);
|
|
Packit Service |
87beb7 |
goto out_err;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
if (fdt_totalsize(blob) > buf_len) {
|
|
Packit Service |
87beb7 |
fprintf(stderr,
|
|
Packit Service |
87beb7 |
"\nBase blob is incomplete (%lu / %" PRIu32 " bytes read)\n",
|
|
Packit Service |
87beb7 |
(unsigned long)buf_len, fdt_totalsize(blob));
|
|
Packit Service |
87beb7 |
goto out_err;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
/* allocate blob pointer array */
|
|
Packit Service |
87beb7 |
ovblob = xmalloc(sizeof(*ovblob) * argc);
|
|
Packit Service |
87beb7 |
memset(ovblob, 0, sizeof(*ovblob) * argc);
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
/* read and keep track of the overlay blobs */
|
|
Packit Service |
87beb7 |
for (i = 0; i < argc; i++) {
|
|
Packit Service |
87beb7 |
size_t ov_len;
|
|
Packit Service |
87beb7 |
ovblob[i] = utilfdt_read(argv[i], &ov_len);
|
|
Packit Service |
87beb7 |
if (!ovblob[i]) {
|
|
Packit Service |
87beb7 |
fprintf(stderr, "\nFailed to read '%s'\n", argv[i]);
|
|
Packit Service |
87beb7 |
goto out_err;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
if (fdt_totalsize(ovblob[i]) > ov_len) {
|
|
Packit Service |
87beb7 |
fprintf(stderr,
|
|
Packit Service |
87beb7 |
"\nOverlay '%s' is incomplete (%lu / %" PRIu32 " bytes read)\n",
|
|
Packit Service |
87beb7 |
argv[i], (unsigned long)ov_len,
|
|
Packit Service |
87beb7 |
fdt_totalsize(ovblob[i]));
|
|
Packit Service |
87beb7 |
goto out_err;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
buf_len = fdt_totalsize(blob);
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
/* apply the overlays in sequence */
|
|
Packit Service |
87beb7 |
for (i = 0; i < argc; i++) {
|
|
Packit Service |
87beb7 |
blob = apply_one(blob, ovblob[i], &buf_len, argv[i]);
|
|
Packit Service |
87beb7 |
if (!blob)
|
|
Packit Service |
87beb7 |
goto out_err;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
fdt_pack(blob);
|
|
Packit Service |
87beb7 |
ret = utilfdt_write(output_filename, blob);
|
|
Packit Service |
87beb7 |
if (ret)
|
|
Packit Service |
87beb7 |
fprintf(stderr, "\nFailed to write '%s'\n",
|
|
Packit Service |
87beb7 |
output_filename);
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
out_err:
|
|
Packit Service |
87beb7 |
if (ovblob) {
|
|
Packit Service |
87beb7 |
for (i = 0; i < argc; i++) {
|
|
Packit Service |
87beb7 |
if (ovblob[i])
|
|
Packit Service |
87beb7 |
free(ovblob[i]);
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
free(ovblob);
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
free(blob);
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
return ret;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
int main(int argc, char *argv[])
|
|
Packit Service |
87beb7 |
{
|
|
Packit Service |
87beb7 |
int opt, i;
|
|
Packit Service |
87beb7 |
char *input_filename = NULL;
|
|
Packit Service |
87beb7 |
char *output_filename = NULL;
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
while ((opt = util_getopt_long()) != EOF) {
|
|
Packit Service |
87beb7 |
switch (opt) {
|
|
Packit Service |
87beb7 |
case_USAGE_COMMON_FLAGS
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
case 'i':
|
|
Packit Service |
87beb7 |
input_filename = optarg;
|
|
Packit Service |
87beb7 |
break;
|
|
Packit Service |
87beb7 |
case 'o':
|
|
Packit Service |
87beb7 |
output_filename = optarg;
|
|
Packit Service |
87beb7 |
break;
|
|
Packit Service |
87beb7 |
case 'v':
|
|
Packit Service |
87beb7 |
verbose = 1;
|
|
Packit Service |
87beb7 |
break;
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
if (!input_filename)
|
|
Packit Service |
87beb7 |
usage("missing input file");
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
if (!output_filename)
|
|
Packit Service |
87beb7 |
usage("missing output file");
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
argv += optind;
|
|
Packit Service |
87beb7 |
argc -= optind;
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
if (argc <= 0)
|
|
Packit Service |
87beb7 |
usage("missing overlay file(s)");
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
if (verbose) {
|
|
Packit Service |
87beb7 |
printf("input = %s\n", input_filename);
|
|
Packit Service |
87beb7 |
printf("output = %s\n", output_filename);
|
|
Packit Service |
87beb7 |
for (i = 0; i < argc; i++)
|
|
Packit Service |
87beb7 |
printf("overlay[%d] = %s\n", i, argv[i]);
|
|
Packit Service |
87beb7 |
}
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
if (do_fdtoverlay(input_filename, output_filename, argc, argv))
|
|
Packit Service |
87beb7 |
return 1;
|
|
Packit Service |
87beb7 |
|
|
Packit Service |
87beb7 |
return 0;
|
|
Packit Service |
87beb7 |
}
|