1 /*
2 * Copyright (C) 2008 The Android Open Source Project
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in
12 * the documentation and/or other materials provided with the
13 * distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include "bootimg_utils.h"
30
31 #include "util.h"
32
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36
bootimg_set_cmdline(boot_img_hdr_v2 * h,const std::string & cmdline)37 void bootimg_set_cmdline(boot_img_hdr_v2* h, const std::string& cmdline) {
38 if (cmdline.size() >= sizeof(h->cmdline)) die("command line too large: %zu", cmdline.size());
39 strcpy(reinterpret_cast<char*>(h->cmdline), cmdline.c_str());
40 }
41
mkbootimg(const std::vector<char> & kernel,const std::vector<char> & ramdisk,const std::vector<char> & second,const std::vector<char> & dtb,size_t base,const boot_img_hdr_v2 & src,std::vector<char> * out)42 boot_img_hdr_v2* mkbootimg(const std::vector<char>& kernel, const std::vector<char>& ramdisk,
43 const std::vector<char>& second, const std::vector<char>& dtb,
44 size_t base, const boot_img_hdr_v2& src, std::vector<char>* out) {
45 const size_t page_mask = src.page_size - 1;
46
47 int64_t header_actual = (sizeof(boot_img_hdr_v1) + page_mask) & (~page_mask);
48 int64_t kernel_actual = (kernel.size() + page_mask) & (~page_mask);
49 int64_t ramdisk_actual = (ramdisk.size() + page_mask) & (~page_mask);
50 int64_t second_actual = (second.size() + page_mask) & (~page_mask);
51 int64_t dtb_actual = (dtb.size() + page_mask) & (~page_mask);
52
53 int64_t bootimg_size =
54 header_actual + kernel_actual + ramdisk_actual + second_actual + dtb_actual;
55 out->resize(bootimg_size);
56
57 boot_img_hdr_v2* hdr = reinterpret_cast<boot_img_hdr_v2*>(out->data());
58
59 *hdr = src;
60 memcpy(hdr->magic, BOOT_MAGIC, BOOT_MAGIC_SIZE);
61
62 hdr->kernel_size = kernel.size();
63 hdr->ramdisk_size = ramdisk.size();
64 hdr->second_size = second.size();
65
66 hdr->kernel_addr += base;
67 hdr->ramdisk_addr += base;
68 hdr->second_addr += base;
69 hdr->tags_addr += base;
70
71 if (hdr->header_version == 1) {
72 hdr->header_size = sizeof(boot_img_hdr_v1);
73 } else if (hdr->header_version == 2) {
74 hdr->header_size = sizeof(boot_img_hdr_v2);
75 hdr->dtb_size = dtb.size();
76 hdr->dtb_addr += base;
77 }
78
79 memcpy(hdr->magic + hdr->page_size, kernel.data(), kernel.size());
80 memcpy(hdr->magic + hdr->page_size + kernel_actual, ramdisk.data(), ramdisk.size());
81 memcpy(hdr->magic + hdr->page_size + kernel_actual + ramdisk_actual, second.data(),
82 second.size());
83 memcpy(hdr->magic + hdr->page_size + kernel_actual + ramdisk_actual + second_actual, dtb.data(),
84 dtb.size());
85 return hdr;
86 }
87