1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3 * Copyright (c) 2017 Konsulko Group Inc. All rights reserved.
4 *
5 * Author:
6 * Pantelis Antoniou <pantelis.antoniou@konsulko.com>
7 */
8
9 #include <assert.h>
10 #include <ctype.h>
11 #include <getopt.h>
12 #include <stdio.h>
13 #include <stdlib.h>
14 #include <string.h>
15 #include <inttypes.h>
16
17 #include <libfdt.h>
18
19 #include "util.h"
20
21 #define BUF_INCREMENT 65536
22
23 /* Usage related data. */
24 static const char usage_synopsis[] =
25 "apply a number of overlays to a base blob\n"
26 " fdtoverlay <options> [<overlay.dtbo> [<overlay.dtbo>]]";
27 static const char usage_short_opts[] = "i:o:v" USAGE_COMMON_SHORT_OPTS;
28 static struct option const usage_long_opts[] = {
29 {"input", required_argument, NULL, 'i'},
30 {"output", required_argument, NULL, 'o'},
31 {"verbose", no_argument, NULL, 'v'},
32 USAGE_COMMON_LONG_OPTS,
33 };
34 static const char * const usage_opts_help[] = {
35 "Input base DT blob",
36 "Output DT blob",
37 "Verbose messages",
38 USAGE_COMMON_OPTS_HELP
39 };
40
41 int verbose = 0;
42
apply_one(char * base,const char * overlay,size_t * buf_len,const char * name)43 static void *apply_one(char *base, const char *overlay, size_t *buf_len,
44 const char *name)
45 {
46 char *tmp = NULL;
47 char *tmpo;
48 int ret;
49
50 /*
51 * We take copies first, because a failed apply can trash
52 * both the base blob and the overlay
53 */
54 tmpo = xmalloc(fdt_totalsize(overlay));
55
56 do {
57 tmp = xrealloc(tmp, *buf_len);
58 ret = fdt_open_into(base, tmp, *buf_len);
59 if (ret) {
60 fprintf(stderr,
61 "\nFailed to make temporary copy: %s\n",
62 fdt_strerror(ret));
63 goto fail;
64 }
65
66 memcpy(tmpo, overlay, fdt_totalsize(overlay));
67
68 ret = fdt_overlay_apply(tmp, tmpo);
69 if (ret == -FDT_ERR_NOSPACE) {
70 *buf_len += BUF_INCREMENT;
71 }
72 } while (ret == -FDT_ERR_NOSPACE);
73
74 if (ret) {
75 fprintf(stderr, "\nFailed to apply '%s': %s\n",
76 name, fdt_strerror(ret));
77 goto fail;
78 }
79
80 free(base);
81 free(tmpo);
82 return tmp;
83
84 fail:
85 free(tmpo);
86 if (tmp)
87 free(tmp);
88
89 return NULL;
90 }
do_fdtoverlay(const char * input_filename,const char * output_filename,int argc,char * argv[])91 static int do_fdtoverlay(const char *input_filename,
92 const char *output_filename,
93 int argc, char *argv[])
94 {
95 char *blob = NULL;
96 char **ovblob = NULL;
97 size_t buf_len;
98 int i, ret = -1;
99
100 blob = utilfdt_read(input_filename, &buf_len);
101 if (!blob) {
102 fprintf(stderr, "\nFailed to read '%s'\n", input_filename);
103 goto out_err;
104 }
105 if (fdt_totalsize(blob) > buf_len) {
106 fprintf(stderr,
107 "\nBase blob is incomplete (%lu / %" PRIu32 " bytes read)\n",
108 (unsigned long)buf_len, fdt_totalsize(blob));
109 goto out_err;
110 }
111
112 /* allocate blob pointer array */
113 ovblob = xmalloc(sizeof(*ovblob) * argc);
114 memset(ovblob, 0, sizeof(*ovblob) * argc);
115
116 /* read and keep track of the overlay blobs */
117 for (i = 0; i < argc; i++) {
118 size_t ov_len;
119 ovblob[i] = utilfdt_read(argv[i], &ov_len);
120 if (!ovblob[i]) {
121 fprintf(stderr, "\nFailed to read '%s'\n", argv[i]);
122 goto out_err;
123 }
124 if (fdt_totalsize(ovblob[i]) > ov_len) {
125 fprintf(stderr,
126 "\nOverlay '%s' is incomplete (%lu / %" PRIu32 " bytes read)\n",
127 argv[i], (unsigned long)ov_len,
128 fdt_totalsize(ovblob[i]));
129 goto out_err;
130 }
131 }
132
133 buf_len = fdt_totalsize(blob);
134
135 /* apply the overlays in sequence */
136 for (i = 0; i < argc; i++) {
137 blob = apply_one(blob, ovblob[i], &buf_len, argv[i]);
138 if (!blob)
139 goto out_err;
140 }
141
142 fdt_pack(blob);
143 ret = utilfdt_write(output_filename, blob);
144 if (ret)
145 fprintf(stderr, "\nFailed to write '%s'\n",
146 output_filename);
147
148 out_err:
149 if (ovblob) {
150 for (i = 0; i < argc; i++) {
151 if (ovblob[i])
152 free(ovblob[i]);
153 }
154 free(ovblob);
155 }
156 free(blob);
157
158 return ret;
159 }
160
main(int argc,char * argv[])161 int main(int argc, char *argv[])
162 {
163 int opt, i;
164 char *input_filename = NULL;
165 char *output_filename = NULL;
166
167 while ((opt = util_getopt_long()) != EOF) {
168 switch (opt) {
169 case_USAGE_COMMON_FLAGS
170
171 case 'i':
172 input_filename = optarg;
173 break;
174 case 'o':
175 output_filename = optarg;
176 break;
177 case 'v':
178 verbose = 1;
179 break;
180 }
181 }
182
183 if (!input_filename)
184 usage("missing input file");
185
186 if (!output_filename)
187 usage("missing output file");
188
189 argv += optind;
190 argc -= optind;
191
192 if (argc <= 0)
193 usage("missing overlay file(s)");
194
195 if (verbose) {
196 printf("input = %s\n", input_filename);
197 printf("output = %s\n", output_filename);
198 for (i = 0; i < argc; i++)
199 printf("overlay[%d] = %s\n", i, argv[i]);
200 }
201
202 if (do_fdtoverlay(input_filename, output_filename, argc, argv))
203 return 1;
204
205 return 0;
206 }
207