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 <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 
bootimg_set_cmdline(boot_img_hdr * h,const char * cmdline)35 void bootimg_set_cmdline(boot_img_hdr* h, const char* cmdline)
36 {
37     strcpy((char*) h->cmdline, cmdline);
38 }
39 
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,int64_t * bootimg_size)40 boot_img_hdr* mkbootimg(void* kernel, int64_t kernel_size, off_t kernel_offset,
41                         void* ramdisk, int64_t ramdisk_size, off_t ramdisk_offset,
42                         void* second, int64_t second_size, off_t second_offset,
43                         size_t page_size, size_t base, off_t tags_offset,
44                         int64_t* bootimg_size)
45 {
46     size_t page_mask = page_size - 1;
47 
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 
52     *bootimg_size = page_size + kernel_actual + ramdisk_actual + second_actual;
53 
54     boot_img_hdr* hdr = reinterpret_cast<boot_img_hdr*>(calloc(*bootimg_size, 1));
55     if (hdr == nullptr) {
56         return hdr;
57     }
58 
59     memcpy(hdr->magic, BOOT_MAGIC, BOOT_MAGIC_SIZE);
60 
61     hdr->kernel_size =  kernel_size;
62     hdr->ramdisk_size = ramdisk_size;
63     hdr->second_size =  second_size;
64 
65     hdr->kernel_addr =  base + kernel_offset;
66     hdr->ramdisk_addr = base + ramdisk_offset;
67     hdr->second_addr =  base + second_offset;
68     hdr->tags_addr =    base + tags_offset;
69 
70     hdr->page_size =    page_size;
71 
72     memcpy(hdr->magic + page_size, kernel, kernel_size);
73     memcpy(hdr->magic + page_size + kernel_actual, ramdisk, ramdisk_size);
74     memcpy(hdr->magic + page_size + kernel_actual + ramdisk_actual, second, second_size);
75 
76     return hdr;
77 }
78