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_v3(boot_img_hdr_v3 * h,const std::string & cmdline)37 static void bootimg_set_cmdline_v3(boot_img_hdr_v3* 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
bootimg_set_cmdline(boot_img_hdr_v2 * h,const std::string & cmdline)42 void bootimg_set_cmdline(boot_img_hdr_v2* h, const std::string& cmdline) {
43 if (h->header_version == 3) {
44 return bootimg_set_cmdline_v3(reinterpret_cast<boot_img_hdr_v3*>(h), cmdline);
45 }
46 if (cmdline.size() >= sizeof(h->cmdline)) die("command line too large: %zu", cmdline.size());
47 strcpy(reinterpret_cast<char*>(h->cmdline), cmdline.c_str());
48 }
49
mkbootimg_v3(const std::vector<char> & kernel,const std::vector<char> & ramdisk,const boot_img_hdr_v2 & src,std::vector<char> * out)50 static boot_img_hdr_v3* mkbootimg_v3(const std::vector<char>& kernel,
51 const std::vector<char>& ramdisk, const boot_img_hdr_v2& src,
52 std::vector<char>* out) {
53 #define V3_PAGE_SIZE 4096
54 const size_t page_mask = V3_PAGE_SIZE - 1;
55 int64_t kernel_actual = (kernel.size() + page_mask) & (~page_mask);
56 int64_t ramdisk_actual = (ramdisk.size() + page_mask) & (~page_mask);
57
58 int64_t bootimg_size = V3_PAGE_SIZE + kernel_actual + ramdisk_actual;
59 out->resize(bootimg_size);
60
61 boot_img_hdr_v3* hdr = reinterpret_cast<boot_img_hdr_v3*>(out->data());
62
63 memcpy(hdr->magic, BOOT_MAGIC, BOOT_MAGIC_SIZE);
64 hdr->kernel_size = kernel.size();
65 hdr->ramdisk_size = ramdisk.size();
66 hdr->os_version = src.os_version;
67 hdr->header_size = sizeof(boot_img_hdr_v3);
68 hdr->header_version = 3;
69
70 memcpy(hdr->magic + V3_PAGE_SIZE, kernel.data(), kernel.size());
71 memcpy(hdr->magic + V3_PAGE_SIZE + kernel_actual, ramdisk.data(), ramdisk.size());
72
73 return hdr;
74 }
75
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)76 boot_img_hdr_v2* mkbootimg(const std::vector<char>& kernel, const std::vector<char>& ramdisk,
77 const std::vector<char>& second, const std::vector<char>& dtb,
78 size_t base, const boot_img_hdr_v2& src, std::vector<char>* out) {
79 if (src.header_version == 3) {
80 if (!second.empty() || !dtb.empty()) {
81 die("Second stage bootloader and dtb not supported in v3 boot image\n");
82 }
83 return reinterpret_cast<boot_img_hdr_v2*>(mkbootimg_v3(kernel, ramdisk, src, out));
84 }
85 const size_t page_mask = src.page_size - 1;
86
87 int64_t header_actual = (sizeof(boot_img_hdr_v1) + page_mask) & (~page_mask);
88 int64_t kernel_actual = (kernel.size() + page_mask) & (~page_mask);
89 int64_t ramdisk_actual = (ramdisk.size() + page_mask) & (~page_mask);
90 int64_t second_actual = (second.size() + page_mask) & (~page_mask);
91 int64_t dtb_actual = (dtb.size() + page_mask) & (~page_mask);
92
93 int64_t bootimg_size =
94 header_actual + kernel_actual + ramdisk_actual + second_actual + dtb_actual;
95 out->resize(bootimg_size);
96
97 boot_img_hdr_v2* hdr = reinterpret_cast<boot_img_hdr_v2*>(out->data());
98
99 *hdr = src;
100 memcpy(hdr->magic, BOOT_MAGIC, BOOT_MAGIC_SIZE);
101
102 hdr->kernel_size = kernel.size();
103 hdr->ramdisk_size = ramdisk.size();
104 hdr->second_size = second.size();
105
106 hdr->kernel_addr += base;
107 hdr->ramdisk_addr += base;
108 hdr->second_addr += base;
109 hdr->tags_addr += base;
110
111 if (hdr->header_version == 1) {
112 hdr->header_size = sizeof(boot_img_hdr_v1);
113 } else if (hdr->header_version == 2) {
114 hdr->header_size = sizeof(boot_img_hdr_v2);
115 hdr->dtb_size = dtb.size();
116 hdr->dtb_addr += base;
117 }
118
119 memcpy(hdr->magic + hdr->page_size, kernel.data(), kernel.size());
120 memcpy(hdr->magic + hdr->page_size + kernel_actual, ramdisk.data(), ramdisk.size());
121 memcpy(hdr->magic + hdr->page_size + kernel_actual + ramdisk_actual, second.data(),
122 second.size());
123 memcpy(hdr->magic + hdr->page_size + kernel_actual + ramdisk_actual + second_actual, dtb.data(),
124 dtb.size());
125 return hdr;
126 }
127