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 "fastboot.h"
32
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36
bootimg_set_cmdline(boot_img_hdr_v1 * h,const char * cmdline)37 void bootimg_set_cmdline(boot_img_hdr_v1* h, const char* cmdline) {
38 if (strlen(cmdline) >= sizeof(h->cmdline)) die("command line too large: %zu", strlen(cmdline));
39 strcpy(reinterpret_cast<char*>(h->cmdline), cmdline);
40 }
41
mkbootimg(void * kernel,int64_t kernel_size,off_t kernel_offset,void * ramdisk,int64_t ramdisk_size,off_t ramdisk_offset,void * second,int64_t second_size,off_t second_offset,size_t page_size,size_t base,off_t tags_offset,uint32_t header_version,int64_t * bootimg_size)42 boot_img_hdr_v1* mkbootimg(void* kernel, int64_t kernel_size, off_t kernel_offset, void* ramdisk,
43 int64_t ramdisk_size, off_t ramdisk_offset, void* second,
44 int64_t second_size, off_t second_offset, size_t page_size, size_t base,
45 off_t tags_offset, uint32_t header_version, int64_t* bootimg_size) {
46 size_t page_mask = page_size - 1;
47
48 int64_t header_actual = (sizeof(boot_img_hdr_v1) + page_mask) & (~page_mask);
49 int64_t kernel_actual = (kernel_size + page_mask) & (~page_mask);
50 int64_t ramdisk_actual = (ramdisk_size + page_mask) & (~page_mask);
51 int64_t second_actual = (second_size + page_mask) & (~page_mask);
52
53 *bootimg_size = header_actual + kernel_actual + ramdisk_actual + second_actual;
54
55 boot_img_hdr_v1* hdr = reinterpret_cast<boot_img_hdr_v1*>(calloc(*bootimg_size, 1));
56 if (hdr == nullptr) {
57 return hdr;
58 }
59
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 + kernel_offset;
67 hdr->ramdisk_addr = base + ramdisk_offset;
68 hdr->second_addr = base + second_offset;
69 hdr->tags_addr = base + tags_offset;
70
71 hdr->page_size = page_size;
72
73 if (header_version) {
74 hdr->header_version = header_version;
75 hdr->header_size = sizeof(boot_img_hdr_v1);
76 }
77
78 memcpy(hdr->magic + page_size, kernel, kernel_size);
79 memcpy(hdr->magic + page_size + kernel_actual, ramdisk, ramdisk_size);
80 memcpy(hdr->magic + page_size + kernel_actual + ramdisk_actual, second, second_size);
81 return hdr;
82 }
83