1/* 2 * Copyright (C) 2016 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#include <stdio.h> 18#include <stdlib.h> 19#include <time.h> 20 21#include "ufdt_overlay.h" 22#include "libufdt_sysdeps.h" 23 24#include "util.h" 25 26 27int apply_ovleray_files(const char *out_filename, 28 const char *base_filename, 29 const char *overlay_filename) { 30 int ret = 1; 31 char *base_buf = NULL; 32 char *overlay_buf = NULL; 33 struct fdt_header *new_blob = NULL; 34 35 size_t blob_len; 36 base_buf = load_file(base_filename, &blob_len); 37 if (!base_buf) { 38 fprintf(stderr, "Can not load base file: %s\n", base_filename); 39 goto end; 40 } 41 42 size_t overlay_len; 43 overlay_buf = load_file(overlay_filename, &overlay_len); 44 if (!overlay_buf) { 45 fprintf(stderr, "Can not load overlay file: %s\n", overlay_filename); 46 goto end; 47 } 48 49 struct fdt_header *blob = ufdt_install_blob(base_buf, blob_len); 50 if (!blob) { 51 fprintf(stderr, "ufdt_install_blob() returns null\n"); 52 goto end; 53 } 54 55 clock_t start = clock(); 56 new_blob = ufdt_apply_overlay(blob, blob_len, overlay_buf, overlay_len); 57 clock_t end = clock(); 58 59 if (write_fdt_to_file(out_filename, new_blob) != 0) { 60 fprintf(stderr, "Write file error: %s\n", out_filename); 61 goto end; 62 } 63 64 // Outputs the used time. 65 double cpu_time_used = ((double)(end - start)) / CLOCKS_PER_SEC; 66 printf("ufdt_apply_overlay: took %.9f secs\n", cpu_time_used); 67 ret = 0; 68 69end: 70 // Do not dto_free(blob) - it's the same as base_buf. 71 72 if (new_blob) dto_free(new_blob); 73 if (overlay_buf) dto_free(overlay_buf); 74 if (base_buf) dto_free(base_buf); 75 76 return ret; 77} 78 79int main(int argc, char **argv) { 80 if (argc < 4) { 81 fprintf(stderr, "Usage: %s <base_file> <overlay_file> <out_file>\n", argv[0]); 82 return 1; 83 } 84 85 const char *base_file = argv[1]; 86 const char *overlay_file = argv[2]; 87 const char *out_file = argv[3]; 88 int ret = apply_ovleray_files(out_file, base_file, overlay_file); 89 90 return ret; 91} 92