Home | History | Annotate | Download | only in fastboot
      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 
     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 
     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