diff --git a/fastbootd/Android.mk b/fastbootd/Android.mk deleted file mode 100644 index c74f906a6..000000000 --- a/fastbootd/Android.mk +++ /dev/null @@ -1,83 +0,0 @@ -# Copyright (C) 2013 Google Inc. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -LOCAL_PATH:= $(call my-dir) - -include $(CLEAR_VARS) - -LOCAL_SRC_FILES := \ - config.c \ - commands.c \ - commands/boot.c \ - commands/flash.c \ - commands/partitions.c \ - commands/virtual_partitions.c \ - fastbootd.c \ - protocol.c \ - network_discovery.c \ - socket_client.c \ - secure.c \ - transport.c \ - transport_socket.c \ - trigger.c \ - usb_linux_client.c \ - utils.c \ - -LOCAL_MODULE := fastbootd -LOCAL_MODULE_TAGS := optional -LOCAL_CFLAGS := -Wall -Werror -Wno-unused-parameter -Wno-deprecated-declarations -DFLASH_CERT - -LOCAL_STATIC_LIBRARIES := \ - libc \ - libcrypto_static \ - libcutils \ - libmdnssd \ - libsparse_static \ - libz - -LOCAL_HAL_STATIC_LIBRARIES := libvendortrigger - -LOCAL_FORCE_STATIC_EXECUTABLE := true - -include $(BUILD_EXECUTABLE) - -include $(CLEAR_VARS) - -LOCAL_SRC_FILES := \ - commands/partitions.c \ - other/gptedit.c \ - utils.c - -LOCAL_MODULE := gptedit -LOCAL_MODULE_TAGS := optional -LOCAL_CFLAGS := -Wall -Werror -Wno-unused-parameter - -LOCAL_STATIC_LIBRARIES := \ - libsparse_static \ - libc \ - libcutils \ - libz - -LOCAL_FORCE_STATIC_EXECUTABLE := true - -include $(BUILD_EXECUTABLE) - -# vendor trigger HAL -include $(CLEAR_VARS) -LOCAL_CFLAGS := -Wall -Werror -LOCAL_MODULE := libvendortrigger.default -LOCAL_MODULE_TAGS := optional -LOCAL_SRC_FILES := vendor_trigger_default.c -LOCAL_STATIC_LIBRARIES := libcutils -include $(BUILD_STATIC_LIBRARY) diff --git a/fastbootd/bootimg.h b/fastbootd/bootimg.h deleted file mode 100644 index 44fde9277..000000000 --- a/fastbootd/bootimg.h +++ /dev/null @@ -1,97 +0,0 @@ -/* - * Copyright (C) 2008 The Android Open Source Project - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#ifndef _BOOT_IMAGE_H_ -#define _BOOT_IMAGE_H_ - -typedef struct boot_img_hdr boot_img_hdr; - -#define BOOT_MAGIC "ANDROID!" -#define BOOT_MAGIC_SIZE 8 -#define BOOT_NAME_SIZE 16 -#define BOOT_ARGS_SIZE 512 - -struct boot_img_hdr -{ - unsigned char magic[BOOT_MAGIC_SIZE]; - - unsigned kernel_size; /* size in bytes */ - unsigned kernel_addr; /* physical load addr */ - - unsigned ramdisk_size; /* size in bytes */ - unsigned ramdisk_addr; /* physical load addr */ - - unsigned second_size; /* size in bytes */ - unsigned second_addr; /* physical load addr */ - - unsigned tags_addr; /* physical addr for kernel tags */ - unsigned page_size; /* flash page size we assume */ - unsigned unused[2]; /* future expansion: should be 0 */ - - unsigned char name[BOOT_NAME_SIZE]; /* asciiz product name */ - - unsigned char cmdline[BOOT_ARGS_SIZE]; - - unsigned id[8]; /* timestamp / checksum / sha1 / etc */ -}; - -/* -** +-----------------+ -** | boot header | 1 page -** +-----------------+ -** | kernel | n pages -** +-----------------+ -** | ramdisk | m pages -** +-----------------+ -** | second stage | o pages -** +-----------------+ -** -** n = (kernel_size + page_size - 1) / page_size -** m = (ramdisk_size + page_size - 1) / page_size -** o = (second_size + page_size - 1) / page_size -** -** 0. all entities are page_size aligned in flash -** 1. kernel and ramdisk are required (size != 0) -** 2. second is optional (second_size == 0 -> no second) -** 3. load each element (kernel, ramdisk, second) at -** the specified physical address (kernel_addr, etc) -** 4. prepare tags at tag_addr. kernel_args[] is -** appended to the kernel commandline in the tags. -** 5. r0 = 0, r1 = MACHINE_TYPE, r2 = tags_addr -** 6. if second_size != 0: jump to second_addr -** else: jump to kernel_addr -*/ - -boot_img_hdr *mkbootimg(void *kernel, unsigned kernel_size, - void *ramdisk, unsigned ramdisk_size, - void *second, unsigned second_size, - unsigned page_size, - unsigned *bootimg_size); - -void bootimg_set_cmdline(boot_img_hdr *hdr, const char *cmdline); -#endif diff --git a/fastbootd/commands.c b/fastbootd/commands.c deleted file mode 100644 index 9be96dc04..000000000 --- a/fastbootd/commands.c +++ /dev/null @@ -1,409 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#include -#include -#include -#include -#include -#include -#include -#include -#include - -#include "bootimg.h" -#include "commands/boot.h" -#include "commands/flash.h" -#include "commands/partitions.h" -#include "commands/virtual_partitions.h" -#include "debug.h" -#include "protocol.h" -#include "trigger.h" -#include "utils.h" - -#define ATAGS_LOCATION "/proc/atags" - -static void cmd_boot(struct protocol_handle *phandle, const char *arg) -{ - int sz, atags_sz, new_atags_sz; - int rv; - unsigned kernel_actual; - unsigned ramdisk_actual; - unsigned second_actual; - void *kernel_ptr; - void *ramdisk_ptr; - void *second_ptr; - struct boot_img_hdr *hdr; - char *ptr = NULL; - char *atags_ptr = NULL; - char *new_atags = NULL; - int data_fd = 0; - - D(DEBUG, "cmd_boot %s\n", arg); - - if (phandle->download_fd < 0) { - fastboot_fail(phandle, "no kernel file"); - return; - } - - atags_ptr = read_atags(ATAGS_LOCATION, &atags_sz); - if (atags_ptr == NULL) { - fastboot_fail(phandle, "atags read error"); - goto error; - } - - // TODO: With cms we can also verify partition name included as - // cms signed attribute - if (flash_validate_certificate(phandle->download_fd, &data_fd) != 1) { - fastboot_fail(phandle, "Access forbiden you need the certificate"); - return; - } - - sz = get_file_size(data_fd); - - ptr = (char *) mmap(NULL, sz, PROT_READ, - MAP_POPULATE | MAP_PRIVATE, data_fd, 0); - - hdr = (struct boot_img_hdr *) ptr; - - if (ptr == MAP_FAILED) { - fastboot_fail(phandle, "internal fastbootd error"); - goto error; - } - - if ((size_t) sz < sizeof(*hdr)) { - fastboot_fail(phandle, "invalid bootimage header"); - goto error; - } - - kernel_actual = ROUND_TO_PAGE(hdr->kernel_size, hdr->page_size); - ramdisk_actual = ROUND_TO_PAGE(hdr->ramdisk_size, hdr->page_size); - second_actual = ROUND_TO_PAGE(hdr->second_size, hdr->page_size); - - new_atags = (char *) create_atags((unsigned *) atags_ptr, atags_sz, hdr, &new_atags_sz); - - if (new_atags == NULL) { - fastboot_fail(phandle, "atags generate error"); - goto error; - } - if (new_atags_sz > 0x4000) { - fastboot_fail(phandle, "atags file to large"); - goto error; - } - - if ((int) (hdr->page_size + kernel_actual + ramdisk_actual) < sz) { - fastboot_fail(phandle, "incomplete bootimage"); - goto error; - } - - kernel_ptr = (void *)((uintptr_t) ptr + hdr->page_size); - ramdisk_ptr = (void *)((uintptr_t) kernel_ptr + kernel_actual); - second_ptr = (void *)((uintptr_t) ramdisk_ptr + ramdisk_actual); - - D(INFO, "preparing to boot"); - // Prepares boot physical address. Addresses from header are ignored - rv = prepare_boot_linux(hdr->kernel_addr, kernel_ptr, kernel_actual, - hdr->ramdisk_addr, ramdisk_ptr, ramdisk_actual, - hdr->second_addr, second_ptr, second_actual, - hdr->tags_addr, new_atags, ROUND_TO_PAGE(new_atags_sz, hdr->page_size)); - if (rv < 0) { - fastboot_fail(phandle, "kexec prepare failed"); - goto error; - } - - fastboot_okay(phandle, ""); - - free(atags_ptr); - munmap(ptr, sz); - free(new_atags); - close(data_fd); - - D(INFO, "Kexec going to reboot"); - reboot(LINUX_REBOOT_CMD_KEXEC); - - fastboot_fail(phandle, "reboot error"); - - return; - -error: - - if (atags_ptr != NULL) - free(atags_ptr); - if (ptr != NULL) - munmap(ptr, sz); - -} - -static void cmd_erase(struct protocol_handle *phandle, const char *arg) -{ - int partition_fd; - char path[PATH_MAX]; - D(DEBUG, "cmd_erase %s\n", arg); - - if (flash_find_entry(arg, path, PATH_MAX)) { - fastboot_fail(phandle, "partition table doesn't exist"); - return; - } - - if (!path[0]) { - fastboot_fail(phandle, "Couldn't find partition"); - return; - } - - partition_fd = flash_get_partiton(path); - if (partition_fd < 0) { - fastboot_fail(phandle, "partiton file does not exists"); - } - - if (flash_erase(partition_fd)) { - fastboot_fail(phandle, "failed to erase partition"); - flash_close(partition_fd); - return; - } - - if (flash_close(partition_fd) < 0) { - D(ERR, "could not close device %s", strerror(errno)); - fastboot_fail(phandle, "failed to erase partition"); - return; - } - fastboot_okay(phandle, ""); -} - -static int GPT_header_location() { - const char *location_str = fastboot_getvar("gpt_sector"); - char *str; - int location; - - if (!strcmp("", location_str)) { - D(INFO, "GPT location not specified using second sector"); - return 1; - } - else { - location = strtoul(location_str, &str, 10); - D(INFO, "GPT location specified as %d", location); - - if (*str != '\0') - return -1; - - return location - 1; - } -} - -static void cmd_gpt_layout(struct protocol_handle *phandle, const char *arg) { - struct GPT_entry_table *oldtable; - int location; - struct GPT_content content; - const char *device; - device = fastboot_getvar("blockdev"); - - if (!strcmp(device, "")) { - fastboot_fail(phandle, "blockdev not defined in config file"); - return; - } - - //TODO: add same verification as in cmd_flash - if (phandle->download_fd < 0) { - fastboot_fail(phandle, "no layout file"); - return; - } - - location = GPT_header_location(); - oldtable = GPT_get_device(device, location); - - GPT_default_content(&content, oldtable); - if (oldtable == NULL) - D(WARN, "Could not get old gpt table"); - else - GPT_release_device(oldtable); - - if (!GPT_parse_file(phandle->download_fd, &content)) { - fastboot_fail(phandle, "Could not parse partition config file"); - return; - } - - if (trigger_gpt_layout(&content)) { - fastboot_fail(phandle, "Vendor forbids this opperation"); - GPT_release_content(&content); - return; - } - - if (!GPT_write_content(device, &content)) { - fastboot_fail(phandle, "Unable to write gpt file"); - GPT_release_content(&content); - return; - } - - GPT_release_content(&content); - fastboot_okay(phandle, ""); -} - -static void cmd_flash(struct protocol_handle *phandle, const char *arg) -{ - int partition; - uint64_t sz; - char data[BOOT_MAGIC_SIZE]; - char path[PATH_MAX]; - ssize_t header_sz = 0; - int data_fd = 0; - - D(DEBUG, "cmd_flash %s\n", arg); - - if (try_handle_virtual_partition(phandle, arg)) { - return; - } - - if (phandle->download_fd < 0) { - fastboot_fail(phandle, "no kernel file"); - return; - } - - if (flash_find_entry(arg, path, PATH_MAX)) { - fastboot_fail(phandle, "partition table doesn't exist"); - return; - } - - if (flash_validate_certificate(phandle->download_fd, &data_fd) != 1) { - fastboot_fail(phandle, "Access forbiden you need certificate"); - return; - } - - // TODO: Maybe its goot idea to check whether the partition is bootable - if (!strcmp(arg, "boot") || !strcmp(arg, "recovery")) { - if (read_data_once(data_fd, data, BOOT_MAGIC_SIZE) < BOOT_MAGIC_SIZE) { - fastboot_fail(phandle, "incoming data read error, cannot read boot header"); - return; - } - if (memcmp((void *)data, BOOT_MAGIC, BOOT_MAGIC_SIZE)) { - fastboot_fail(phandle, "image is not a boot image"); - return; - } - } - - partition = flash_get_partiton(path); - - sz = get_file_size64(data_fd); - - sz -= header_sz; - - if (sz > get_file_size64(partition)) { - flash_close(partition); - D(WARN, "size of file too large"); - fastboot_fail(phandle, "size of file too large"); - return; - } - - D(INFO, "writing %"PRId64" bytes to '%s'\n", sz, arg); - - if (flash_write(partition, phandle->download_fd, sz, header_sz)) { - fastboot_fail(phandle, "flash write failure"); - return; - } - D(INFO, "partition '%s' updated\n", arg); - - flash_close(partition); - close(data_fd); - - fastboot_okay(phandle, ""); -} - -static void cmd_continue(struct protocol_handle *phandle, const char *arg) -{ - fastboot_okay(phandle, ""); -#if 0 - udc_stop(); - - boot_linux_from_flash(); -#endif -} - -static void cmd_getvar(struct protocol_handle *phandle, const char *arg) -{ - const char *value; - D(DEBUG, "cmd_getvar %s\n", arg); - - value = fastboot_getvar(arg); - - fastboot_okay(phandle, value); -} - -static void cmd_download(struct protocol_handle *phandle, const char *arg) -{ - unsigned len = strtoul(arg, NULL, 16); - int old_fd; - - if (len > 256 * 1024 * 1024) { - fastboot_fail(phandle, "data too large"); - return; - } - - fastboot_data(phandle, len); - - old_fd = protocol_get_download(phandle); - if (old_fd >= 0) { - off_t len = lseek(old_fd, 0, SEEK_END); - D(INFO, "disposing of unused fd %d, size %ld", old_fd, len); - close(old_fd); - } - - phandle->download_fd = protocol_handle_download(phandle, len); - if (phandle->download_fd < 0) { - fastboot_fail(phandle, "download failed"); - return; - } - - fastboot_okay(phandle, ""); -} - -static void cmd_oem(struct protocol_handle *phandle, const char *arg) { - const char *response = ""; - - //TODO: Maybe it should get download descriptor also - if (trigger_oem_cmd(arg, &response)) - fastboot_fail(phandle, response); - else - fastboot_okay(phandle, response); -} - -void commands_init() -{ - virtual_partition_register("partition-table", cmd_gpt_layout); - - fastboot_register("boot", cmd_boot); - fastboot_register("erase:", cmd_erase); - fastboot_register("flash:", cmd_flash); - fastboot_register("continue", cmd_continue); - fastboot_register("getvar:", cmd_getvar); - fastboot_register("download:", cmd_download); - fastboot_register("oem", cmd_oem); - //fastboot_publish("version", "0.5"); - //fastboot_publish("product", "swordfish"); - //fastboot_publish("kernel", "lk"); -} diff --git a/fastbootd/commands/boot.c b/fastbootd/commands/boot.c deleted file mode 100644 index bb939176d..000000000 --- a/fastbootd/commands/boot.c +++ /dev/null @@ -1,255 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#include -#include -#include -#include -#include -#include - -#include "boot.h" -#include "debug.h" -#include "utils.h" -#include "bootimg.h" - - -#define KEXEC_ARM_ATAGS_OFFSET 0x1000 -#define KEXEC_ARM_ZIMAGE_OFFSET 0x8000 - -#define MEMORY_SIZE 0x0800000 -#define START_ADDRESS 0x44000000 -#define KERNEL_START (START_ADDRESS + KEXEC_ARM_ZIMAGE_OFFSET) - -#define ATAG_NONE_TYPE 0x00000000 -#define ATAG_CORE_TYPE 0x54410001 -#define ATAG_RAMDISK_TYPE 0x54410004 -#define ATAG_INITRD2_TYPE 0x54420005 -#define ATAG_CMDLINE_TYPE 0x54410009 - -#define MAX_ATAG_SIZE 0x4000 - -struct atag_info { - unsigned size; - unsigned type; -}; - -struct atag_initrd2 { - unsigned start; - unsigned size; -}; - -struct atag_cmdline { - char cmdline[0]; -}; - -struct atag { - struct atag_info info; - union { - struct atag_initrd2 initrd2; - struct atag_cmdline cmdline; - } data; -}; - - -long kexec_load(unsigned int entry, unsigned long nr_segments, - struct kexec_segment *segment, unsigned long flags) { - return syscall(__NR_kexec_load, entry, nr_segments, segment, flags); -} - -/* - * Prepares arguments for kexec - * Kernel address is not set into kernel_phys - * Ramdisk is set to position relative to kernel - */ -int prepare_boot_linux(uintptr_t kernel_phys, void *kernel_addr, int kernel_size, - uintptr_t ramdisk_phys, void *ramdisk_addr, int ramdisk_size, - uintptr_t second_phys, void *second_addr, int second_size, - uintptr_t atags_phys, void *atags_addr, int atags_size) { - struct kexec_segment segment[4]; - int segment_count = 2; - unsigned entry = START_ADDRESS + KEXEC_ARM_ZIMAGE_OFFSET; - int rv; - int page_size = getpagesize(); - - segment[0].buf = kernel_addr; - segment[0].bufsz = kernel_size; - segment[0].mem = (void *) KERNEL_START; - segment[0].memsz = ROUND_TO_PAGE(kernel_size, page_size); - - if (kernel_size > MEMORY_SIZE - KEXEC_ARM_ZIMAGE_OFFSET) { - D(INFO, "Kernel image too big"); - return -1; - } - - segment[1].buf = atags_addr; - segment[1].bufsz = atags_size; - segment[1].mem = (void *) (START_ADDRESS + KEXEC_ARM_ATAGS_OFFSET); - segment[1].memsz = ROUND_TO_PAGE(atags_size, page_size); - - D(INFO, "Ramdisk size is %d", ramdisk_size); - - if (ramdisk_size != 0) { - segment[segment_count].buf = ramdisk_addr; - segment[segment_count].bufsz = ramdisk_size; - segment[segment_count].mem = (void *) (KERNEL_START + ramdisk_phys - kernel_phys); - segment[segment_count].memsz = ROUND_TO_PAGE(ramdisk_phys, page_size); - ++segment_count; - } - - D(INFO, "Ramdisk size is %d", ramdisk_size); - if (second_size != 0) { - segment[segment_count].buf = second_addr; - segment[segment_count].bufsz = second_size; - segment[segment_count].mem = (void *) (KERNEL_START + second_phys - kernel_phys); - segment[segment_count].memsz = ROUND_TO_PAGE(second_size, page_size); - entry = second_phys; - ++segment_count; - } - - rv = kexec_load(entry, segment_count, segment, KEXEC_ARCH_DEFAULT); - - if (rv != 0) { - D(INFO, "Kexec_load returned non-zero exit code: %s\n", strerror(errno)); - return -1; - } - - return 1; - -} - -unsigned *create_atags(unsigned *atags_position, int atag_size, const struct boot_img_hdr *hdr, int *size) { - struct atag *current_tag = (struct atag *) atags_position; - unsigned *current_tag_raw = atags_position; - unsigned *new_atags = malloc(ROUND_TO_PAGE(atag_size + BOOT_ARGS_SIZE * sizeof(char), - hdr->page_size)); - //This pointer will point into the beggining of buffer free space - unsigned *natags_raw_buff = new_atags; - int new_atags_size = 0; - int current_size; - int cmdl_length; - - // copy tags from current atag file - while (current_tag->info.type != ATAG_NONE_TYPE) { - switch (current_tag->info.type) { - case ATAG_CMDLINE_TYPE: - case ATAG_RAMDISK_TYPE: - case ATAG_INITRD2_TYPE: break; - default: - memcpy((void *)natags_raw_buff, (void *)current_tag_raw, current_tag->info.size * sizeof(unsigned)); - natags_raw_buff += current_tag->info.size; - new_atags_size += current_tag->info.size; - } - - current_tag_raw += current_tag->info.size; - current_tag = (struct atag *) current_tag_raw; - - if (current_tag_raw >= atags_position + atag_size) { - D(ERR, "Critical error in atags"); - return NULL; - } - } - - // set INITRD2 tag - if (hdr->ramdisk_size > 0) { - current_size = (sizeof(struct atag_info) + sizeof(struct atag_initrd2)) / sizeof(unsigned); - *((struct atag *) natags_raw_buff) = (struct atag) { - .info = { - .size = current_size, - .type = ATAG_INITRD2_TYPE - }, - .data = { - .initrd2 = (struct atag_initrd2) { - .start = hdr->ramdisk_addr, - .size = hdr->ramdisk_size - } - } - }; - - new_atags_size += current_size; - natags_raw_buff += current_size; - } - - // set ATAG_CMDLINE - cmdl_length = strnlen((char *) hdr->cmdline, BOOT_ARGS_SIZE - 1); - current_size = sizeof(struct atag_info) + (1 + cmdl_length); - current_size = (current_size + sizeof(unsigned) - 1) / sizeof(unsigned); - *((struct atag *) natags_raw_buff) = (struct atag) { - .info = { - .size = current_size, - .type = ATAG_CMDLINE_TYPE - }, - }; - - //copy cmdline and ensure that there is null character - memcpy(((struct atag *) natags_raw_buff)->data.cmdline.cmdline, - (char *) hdr->cmdline, cmdl_length); - ((struct atag *) natags_raw_buff)->data.cmdline.cmdline[cmdl_length] = '\0'; - - new_atags_size += current_size; - natags_raw_buff += current_size; - - // set ATAG_NONE - *((struct atag *) natags_raw_buff) = (struct atag) { - .info = { - .size = 0, - .type = ATAG_NONE_TYPE - }, - }; - new_atags_size += sizeof(struct atag_info) / sizeof(unsigned); - natags_raw_buff += sizeof(struct atag_info) / sizeof(unsigned); - - *size = new_atags_size * sizeof(unsigned); - return new_atags; -} - -char *read_atags(const char * path, int *atags_sz) { - int afd = -1; - char *atags_ptr = NULL; - - afd = open(path, O_RDONLY); - if (afd < 0) { - D(ERR, "wrong atags file"); - return 0; - } - - atags_ptr = (char *) malloc(MAX_ATAG_SIZE); - if (atags_ptr == NULL) { - D(ERR, "insufficient memory"); - return 0; - } - - *atags_sz = read(afd, atags_ptr, MAX_ATAG_SIZE); - - close(afd); - return atags_ptr; -} - diff --git a/fastbootd/commands/boot.h b/fastbootd/commands/boot.h deleted file mode 100644 index a5efd01a3..000000000 --- a/fastbootd/commands/boot.h +++ /dev/null @@ -1,50 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#ifndef __FASTBOOT_BOOT_H -#define __FASTBOOT_BOOT_H - -#include -#include - -#include "bootimg.h" - -#define KEXEC_TYPE_DEFAULT 0 -#define KEXEC_TYPE_CRASH 1 - -int prepare_boot_linux(uintptr_t, void *, int, uintptr_t, void *, int, - uintptr_t, void *, int, uintptr_t, void *, int); -unsigned *create_atags(unsigned *, int, const struct boot_img_hdr *, int *); -long kexec_load(unsigned int, unsigned long, struct kexec_segment *, unsigned long); -char *read_atags(const char *, int *); - -#endif /* _SYS_KEXEC_H */ - diff --git a/fastbootd/commands/flash.c b/fastbootd/commands/flash.c deleted file mode 100644 index 1eb4d1b33..000000000 --- a/fastbootd/commands/flash.c +++ /dev/null @@ -1,162 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#include -#include -#include -#include - -#include "flash.h" -#include "protocol.h" -#include "debug.h" -#include "utils.h" -#include "commands/partitions.h" - -#ifdef FLASH_CERT -#include "secure.h" -#endif - -#define ALLOWED_CHARS "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_-." -#define BUFFER_SIZE 1024 * 1024 -#define MIN(a, b) (a > b ? b : a) - - -int flash_find_entry(const char *name, char *out, size_t outlen) -{ -//TODO: Assumption: All the partitions has they unique name - - const char *path = fastboot_getvar("device-directory"); - size_t length; - if (strcmp(path, "") == 0) { - D(ERR, "device-directory: not defined in config file"); - return -1; - } - - length = strspn(name, ALLOWED_CHARS); - if (length != strlen(name)) { - D(ERR, "Not allowed char in name: %c", name[length]); - return -1; - } - - if (snprintf(out, outlen, "%s%s", path, name) >= (int) outlen) { - D(ERR, "Too long path to partition file"); - return -1; - } - - if (access(out, F_OK ) == -1) { - D(ERR, "could not find partition file %s", name); - return -1; - } - - return 0; -} - -int flash_erase(int fd) -{ - int64_t size; - size = get_block_device_size(fd); - D(DEBUG, "erase %"PRId64" data from %d\n", size, fd); - - return wipe_block_device(fd, size); -} - -int flash_write(int partition_fd, int data_fd, ssize_t size, ssize_t skip) -{ - ssize_t written = 0; - struct GPT_mapping input; - struct GPT_mapping output; - - while (written < size) { - int current_size = MIN(size - written, BUFFER_SIZE); - - if (gpt_mmap(&input, written + skip, current_size, data_fd)) { - D(ERR, "Error in writing data, unable to map data file %zd at %zd size %d", size, skip, current_size); - return -1; - } - if (gpt_mmap(&output, written, current_size, partition_fd)) { - D(ERR, "Error in writing data, unable to map output partition"); - return -1; - } - - memcpy(output.ptr, input.ptr, current_size); - - gpt_unmap(&input); - gpt_unmap(&output); - - written += current_size; - } - - return 0; -} - -#ifdef FLASH_CERT - -int flash_validate_certificate(int signed_fd, int *data_fd) { - int ret = 0; - const char *cert_path; - X509_STORE *store = NULL; - CMS_ContentInfo *content_info; - BIO *content; - - cert_path = fastboot_getvar("certificate-path"); - if (!strcmp(cert_path, "")) { - D(ERR, "could not find cert-key value in config file"); - goto finish; - } - - store = cert_store_from_path(cert_path); - if (store == NULL) { - D(ERR, "unable to create certification store"); - goto finish; - } - - if (cert_read(signed_fd, &content_info, &content)) { - D(ERR, "reading data failed"); - goto finish; - } - - ret = cert_verify(content, content_info, store, data_fd); - cert_release(content, content_info); - - return ret; - -finish: - if (store != NULL) - cert_release_store(store); - - return ret; -} - -#else -int flash_validate_certificate(int signed_fd, int *data_fd) { - return 1; -} -#endif diff --git a/fastbootd/commands/flash.h b/fastbootd/commands/flash.h deleted file mode 100644 index 5a64cabbc..000000000 --- a/fastbootd/commands/flash.h +++ /dev/null @@ -1,71 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#ifndef _FASTBOOTD_ERASE_H -#define _FASTBOOTD_ERASE_H - -#include -#include -#include -#include "debug.h" - -int flash_find_entry(const char *, char *, size_t); -int flash_erase(int fd); - -static inline int flash_get_partiton(const char *path) { - return open(path, O_RDWR); -} - -static inline int flash_close(int fd) { - return close(fd); -} - -int flash_write(int partition, int data, ssize_t size, ssize_t skip); - -static inline ssize_t read_data_once(int fd, char *buffer, ssize_t size) { - ssize_t readcount = 0; - ssize_t len; - - while ((len = TEMP_FAILURE_RETRY(read(fd, (void *) &buffer[readcount], size - readcount))) > 0) { - readcount += len; - } - if (len < 0) { - D(ERR, "Read error:%s", strerror(errno)); - return len; - } - - return readcount; -} - -int flash_validate_certificate(int signed_fd, int *data_fd); - -#endif - diff --git a/fastbootd/commands/partitions.c b/fastbootd/commands/partitions.c deleted file mode 100644 index 953c56a0f..000000000 --- a/fastbootd/commands/partitions.c +++ /dev/null @@ -1,775 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include - -#include "partitions.h" -#include "debug.h" -#include "utils.h" -#include "protocol.h" - -#define BLKRRPART _IO(0x12,95) /* re-read partition table */ -#define BLKSSZGET _IO(0x12,104) - -#define DIV_ROUND_UP(x, y) (((x) + (y) - 1)/(y)) -#define ALIGN(x, y) ((y) * DIV_ROUND_UP((x), (y))) -#define ALIGN_DOWN(x, y) ((y) * ((x) / (y))) - - -const uint8_t partition_type_uuid[16] = { - 0xa2, 0xa0, 0xd0, 0xeb, 0xe5, 0xb9, 0x33, 0x44, - 0x87, 0xc0, 0x68, 0xb6, 0xb7, 0x26, 0x99, 0xc7, -}; - -//TODO: There is assumption that we are using little endian - -static void GPT_entry_clear(struct GPT_entry_raw *entry) -{ - memset(entry, 0, sizeof(*entry)); -} - -/* - * returns mapped location to choosen area - * mapped_ptr is pointer to whole area mapped (it can be bigger then requested) - */ -int gpt_mmap(struct GPT_mapping *mapping, uint64_t location, int size, int fd) -{ - unsigned int location_diff = location & ~PAGE_MASK; - - mapping->size = ALIGN(size + location_diff, PAGE_SIZE); - - uint64_t sz = get_file_size64(fd); - if (sz < size + location) { - D(ERR, "the location of mapping area is outside of the device size %" PRId64, sz); - return 1; - } - location = ALIGN_DOWN(location, PAGE_SIZE); - - mapping->map_ptr = mmap64(NULL, mapping->size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, location); - - if (mapping->map_ptr == MAP_FAILED) { - mapping->ptr = MAP_FAILED; - D(ERR, "map failed: %s", strerror(errno)); - return 1; - } - - mapping->ptr = (void *)((char *) mapping->map_ptr + location_diff); - return 0; -} - -void gpt_unmap(struct GPT_mapping *mapping) { - munmap(mapping->map_ptr, mapping->size); -} - - -#define LBA_ADDR(table, value) ((uint64_t) (table)->sector_size * (value)) - -int GPT_map_from_content(struct GPT_entry_table *table, const struct GPT_content *content) -{ - - // Mapping header - if (gpt_mmap(&table->header_map, LBA_ADDR(table, content->header.current_lba), - table->sector_size, table->fd)) { - D(ERR, "unable to map header:%s\n", strerror(errno)); - goto error_header; - } - - table->header = (struct GPT_header *) table->header_map.ptr; - - table->partition_table_size = ROUND_UP(content->header.entries_count * sizeof(*table->entries), - table->sector_size); - - // Mapping entry table - if (gpt_mmap(&table->entries_map, LBA_ADDR(table, content->header.entries_lba), - table->partition_table_size, table->fd)) { - D(ERR, "unable to map entries"); - goto error_signature; - } - - table->entries = (struct GPT_entry_raw *) table->entries_map.ptr; - - // Mapping secondary header - if (gpt_mmap(&table->sec_header_map, LBA_ADDR(table, content->header.backup_lba), - table->sector_size, table->fd)) { - D(ERR, "unable to map backup gpt header"); - goto error_sec_header; - } - - // Mapping secondary entries table - if (gpt_mmap(&table->sec_entries_map, - LBA_ADDR(table, content->header.backup_lba) - table->partition_table_size, - table->partition_table_size, table->fd)) { - D(ERR, "unable to map secondary gpt table"); - goto error_sec_entries; - } - - table->second_header = (struct GPT_header *) table->sec_header_map.ptr; - table->second_entries = (struct GPT_entry_raw *) table->sec_entries_map.ptr; - table->second_valid = strcmp("EFI PART", (char *) table->second_header->signature) == 0; - - return 0; - -error_sec_entries: - gpt_unmap(&table->sec_header_map); -error_sec_header: - gpt_unmap(&table->entries_map); -error_signature: - gpt_unmap(&table->header_map); -error_header: - return 1; -} - -int GPT_map(struct GPT_entry_table *table, unsigned header_lba) -{ - struct GPT_content content; - struct GPT_mapping mapping; - struct GPT_header *header; - - if (gpt_mmap(&mapping, LBA_ADDR(table, header_lba), table->sector_size, table->fd)) { - D(ERR, "unable to map header: %s", strerror(errno)); - goto error_header; - } - - header = (struct GPT_header *) mapping.ptr; - - if (strcmp("EFI PART", (char *) header->signature)) { - D(ERR, "GPT entry not valid"); - goto error_signature; - } - - content.header = *header; - - gpt_unmap(&mapping); - - return GPT_map_from_content(table, &content); - -error_signature: - gpt_unmap(&table->header_map); -error_header: - return 1; -} - -struct GPT_entry_table* GPT_get_device(const char *path, unsigned header_lba) -{ - struct GPT_entry_table *table; - size_t sector_bytes; - - table = (struct GPT_entry_table *) malloc(sizeof(*table)); - table->fd = open(path, O_RDWR); - - if (table->fd < 0) { - D(ERR, "unable to open file %s:%s\n", path, strerror(errno)); - return NULL; - } - - if (!ioctl(table->fd, BLKSSZGET, §or_bytes)) { - table->sector_size = (unsigned) sector_bytes; - D(INFO, "Got sector size %d", table->sector_size); - } else { - D(WARN, "unable to get sector size, assuming 512"); - table->sector_size = 512; - } - - if (GPT_map(table, header_lba)) { - D(ERR, "Could not map gpt"); - return NULL; - } - - return table; -} - -static struct GPT_entry_table* GPT_get_from_content(const char *path, const struct GPT_content *content) -{ - struct GPT_entry_table *table; - size_t sector_bytes; - - table = (struct GPT_entry_table *) malloc(sizeof(*table)); - table->fd = open(path, O_RDWR); - - if (table->fd < 0) { - D(ERR, "unable to open file %s:%s\n", path, strerror(errno)); - return NULL; - } - - if (!ioctl(table->fd, BLKSSZGET, §or_bytes)) { - table->sector_size = (unsigned) sector_bytes; - D(INFO, "Got sector size %d", table->sector_size); - } else { - D(WARN, "unable to get sector size %s, assuming 512", strerror(errno)); - table->sector_size = 512; - } - - if (GPT_map_from_content(table, content)) { - D(ERR, "Could not map gpt"); - return NULL; - } - - return table; -} - - -void GPT_release_device(struct GPT_entry_table *table) -{ - gpt_unmap(&table->header_map); - gpt_unmap(&table->entries_map); - gpt_unmap(&table->sec_header_map); - gpt_unmap(&table->sec_entries_map); - close(table->fd); - free(table); -} - -static int GPT_check_overlap(struct GPT_entry_table *table, struct GPT_entry_raw *entry); -static int GPT_check_overlap_except(struct GPT_entry_table *table, - struct GPT_entry_raw *entry, - struct GPT_entry_raw *exclude); - -void GPT_edit_entry(struct GPT_entry_table *table, - struct GPT_entry_raw *old_entry, - struct GPT_entry_raw *new_entry) -{ - struct GPT_entry_raw *current_entry = GPT_get_pointer(table, old_entry); - - if (GPT_check_overlap_except(table, new_entry, current_entry)) { - D(ERR, "Couldn't add overlaping partition"); - return; - } - - if (current_entry == NULL) { - D(ERR, "Couldn't find entry"); - return; - } - - *current_entry = *new_entry; -} - -int GPT_delete_entry(struct GPT_entry_table *table, struct GPT_entry_raw *entry) -{ - struct GPT_entry_raw *raw = GPT_get_pointer(table, entry); - - if (raw == NULL) { - D(ERR, "could not find entry"); - return 1; - } - D(DEBUG, "Deleting gpt entry '%s'\n", raw->partition_guid); - - // Entry in the middle of table may become empty - GPT_entry_clear(raw); - - return 0; -} - -void GPT_add_entry(struct GPT_entry_table *table, struct GPT_entry_raw *entry) -{ - unsigned i; - int inserted = 0; - if (GPT_check_overlap(table, entry)) { - D(ERR, "Couldn't add overlaping partition"); - return; - } - - if (GPT_get_pointer(table, entry) != NULL) { - D(WARN, "Add entry fault, this entry already exists"); - return; - } - - struct GPT_entry_raw *entries = table->entries; - - for (i = 0; i < table->header->entries_count; ++i) { - if (!entries[i].type_guid[0]) { - inserted = 1; - D(DEBUG, "inserting"); - memcpy(&entries[i], entry, sizeof(entries[i])); - break; - } - } - - if (!inserted) { - D(ERR, "Unable to find empty partion entry"); - } -} - -struct GPT_entry_raw *GPT_get_pointer_by_UTFname(struct GPT_entry_table *table, const uint16_t *name); - -struct GPT_entry_raw *GPT_get_pointer(struct GPT_entry_table *table, struct GPT_entry_raw *entry) -{ - if (entry->partition_guid[0] != 0) - return GPT_get_pointer_by_guid(table, (const char *) entry->partition_guid); - else if (entry->name[0] != 0) - return GPT_get_pointer_by_UTFname(table, entry->name); - - D(WARN, "Name or guid needed to find entry"); - return NULL; -} - -struct GPT_entry_raw *GPT_get_pointer_by_guid(struct GPT_entry_table *table, const char *name) -{ - int current = (int) table->header->entries_count; - - for (current = current - 1; current >= 0; --current) { - if (strncmp((char *) name, - (char *) table->entries[current].partition_guid, 16) == 0) { - return &table->entries[current]; - } - } - - return NULL; -} - -int strncmp_UTF16_char(const uint16_t *s1, const char *s2, size_t n) -{ - if (n == 0) - return (0); - do { - if (((*s1) & 127) != *s2++) - return (((unsigned char) ((*s1) & 127)) - *(unsigned char *)--s2); - if (*s1++ == 0) - break; - } while (--n != 0); - return (0); -} - -int strncmp_UTF16(const uint16_t *s1, const uint16_t *s2, size_t n) -{ - if (n == 0) - return (0); - do { - if ((*s1) != *s2++) - return (*s1 - *--s2); - if (*s1++ == 0) - break; - } while (--n != 0); - return (0); -} - -struct GPT_entry_raw *GPT_get_pointer_by_name(struct GPT_entry_table *table, const char *name) -{ - int count = (int) table->header->entries_count; - int current; - - for (current = 0; current < count; ++current) { - if (strncmp_UTF16_char(table->entries[current].name, - (char *) name, 16) == 0) { - return &table->entries[current]; - } - } - - return NULL; -} - -struct GPT_entry_raw *GPT_get_pointer_by_UTFname(struct GPT_entry_table *table, const uint16_t *name) -{ - int count = (int) table->header->entries_count; - int current; - - for (current = 0; current < count; ++current) { - if (strncmp_UTF16(table->entries[current].name, - name, GPT_NAMELEN) == 0) { - return &table->entries[current]; - } - } - - return NULL; -} - -void GPT_sync(struct GPT_entry_table *table) -{ - uint32_t crc; - - //calculate crc32 - crc = crc32(0, Z_NULL, 0); - crc = crc32(crc, (void*) table->entries, table->header->entries_count * sizeof(*table->entries)); - table->header->partition_array_checksum = crc; - - table->header->header_checksum = 0; - crc = crc32(0, Z_NULL, 0); - crc = crc32(crc, (void*) table->header, table->header->header_size); - table->header->header_checksum = crc; - - //sync secondary partion - if (table->second_valid) { - memcpy((void *)table->second_entries, (void *) table->entries, table->partition_table_size); - memcpy((void *)table->second_header, (void *)table->header, sizeof(*table->header)); - } - - if(!ioctl(table->fd, BLKRRPART, NULL)) { - D(WARN, "Unable to force kernel to refresh partition table"); - } -} - -void GPT_to_UTF16(uint16_t *to, const char *from, int n) -{ - int i; - for (i = 0; i < (n - 1) && (to[i] = from[i]) != '\0'; ++i); - to[i] = '\0'; -} - -void GPT_from_UTF16(char *to, const uint16_t *from, int n) -{ - int i; - for (i = 0; i < (n - 1) && (to[i] = from[i] & 127) != '\0'; ++i); - to[i] = '\0'; -} - -static int GPT_check_overlap_except(struct GPT_entry_table *table, - struct GPT_entry_raw *entry, - struct GPT_entry_raw *exclude) { - int current = (int) table->header->entries_count; - int dontcheck; - struct GPT_entry_raw *current_entry; - if (entry->last_lba < entry->first_lba) { - D(WARN, "Start address have to be less than end address"); - return 1; - } - - for (current = current - 1; current >= 0; --current) { - current_entry = &table->entries[current]; - dontcheck = strncmp((char *) entry->partition_guid, - (char *) current_entry->partition_guid , 16) == 0; - dontcheck |= current_entry->type_guid[0] == 0; - dontcheck |= current_entry == exclude; - - if (!dontcheck && ((entry->last_lba >= current_entry->first_lba && - entry->first_lba < current_entry->last_lba ))) { - return 1; - } - } - - return 0; -} - -static int GPT_check_overlap(struct GPT_entry_table *table, struct GPT_entry_raw *entry) -{ - return GPT_check_overlap_except(table, entry, NULL); -} - -static char *get_key_value(char *ptr, char **key, char **value) -{ - *key = ptr; - ptr = strchr(ptr, '='); - - if (ptr == NULL) - return NULL; - - *ptr++ = '\0'; - *value = ptr; - ptr = strchr(ptr, ';'); - - if (ptr == NULL) - ptr = *value + strlen(*value); - else - *ptr = '\0'; - - *key = strip(*key); - *value = strip(*value); - - return ptr; -} - -//TODO: little endian? -static int add_key_value(const char *key, const char *value, struct GPT_entry_raw *entry) -{ - char *endptr; - if (!strcmp(key, "type")) { - strncpy((char *) entry->type_guid, value, 16); - entry->type_guid[15] = 0; - } - else if (!strcmp(key, "guid")) { - strncpy((char *) entry->partition_guid, value, 16); - entry->type_guid[15] = 0; - } - else if (!strcmp(key, "firstlba")) { - entry->first_lba = strtoul(value, &endptr, 10); - if (*endptr != '\0') goto error; - } - else if (!strcmp(key, "lastlba")) { - entry->last_lba = strtoul(value, &endptr, 10); - if (*endptr != '\0') goto error; - } - else if (!strcmp(key, "flags")) { - entry->flags = strtoul(value, &endptr, 16); - if (*endptr != '\0') goto error; - } - else if (!strcmp(key, "name")) { - GPT_to_UTF16(entry->name, value, GPT_NAMELEN); - } - else { - goto error; - } - - return 0; - -error: - D(ERR, "Could not find key or parse value: %s,%s", key, value); - return 1; -} - -int GPT_parse_entry(char *string, struct GPT_entry_raw *entry) -{ - char *ptr = string; - char *key = NULL; - char *value = NULL; - - while ((ptr = get_key_value(ptr, &key, &value)) != NULL) { - if (add_key_value(key, value, entry)) { - D(WARN, "key or value not valid: %s %s", key, value); - return 1; - } - } - - return 0; -} - -void entry_set_guid(int n, uint8_t *guid) -{ - int fd; - fd = open("/dev/urandom", O_RDONLY); - read(fd, guid, 16); - close(fd); - - //rfc4122 - guid[8] = (guid[8] & 0x3F) | 0x80; - guid[7] = (guid[7] & 0x0F) | 0x40; -} - -void GPT_default_content(struct GPT_content *content, struct GPT_entry_table *table) -{ - if (table != NULL) { - memcpy(&content->header, table->header, sizeof(content->header)); - content->header.header_size = sizeof(content->header); - content->header.entry_size = sizeof(struct GPT_entry_raw); - } - else { - D(WARN, "Could not locate old gpt table, using default values"); - memset(&content->header, 0, sizeof(content->header) / sizeof(int)); - content->header = (struct GPT_header) { - .revision = 0x10000, - .header_size = sizeof(content->header), - .header_checksum = 0, - .reserved_zeros = 0, - .current_lba = 1, - .backup_lba = 1, - .entry_size = sizeof(struct GPT_entry_raw), - .partition_array_checksum = 0 - }; - strncpy((char *)content->header.signature, "EFI PART", 8); - strncpy((char *)content->header.disk_guid, "ANDROID MMC DISK", 16); - } -} - -static int get_config_uint64(cnode *node, uint64_t *ptr, const char *name) -{ - const char *tmp; - uint64_t val; - char *endptr; - if ((tmp = config_str(node, name, NULL))) { - val = strtoull(tmp, &endptr, 10); - if (*endptr != '\0') { - D(WARN, "Value for %s is not a number: %s", name, tmp); - return 1; - } - *ptr = val; - return 0; - } - return 1; -} - -static int get_config_string(cnode *node, char *ptr, int max_len, const char *name) -{ - size_t begin, end; - const char *value = config_str(node, name, NULL); - if (!value) - return -1; - - begin = strcspn(value, "\"") + 1; - end = strcspn(&value[begin], "\""); - - if ((int) end > max_len) { - D(WARN, "Identifier \"%s\" too long", value); - return -1; - } - - strncpy(ptr, &value[begin], end); - if((int) end < max_len) - ptr[end] = 0; - return 0; -} - -static void GPT_parse_header(cnode *node, struct GPT_content *content) -{ - get_config_uint64(node, &content->header.current_lba, "header_lba"); - get_config_uint64(node, &content->header.backup_lba, "backup_lba"); - get_config_uint64(node, &content->header.first_usable_lba, "first_lba"); - get_config_uint64(node, &content->header.last_usable_lba, "last_lba"); - get_config_uint64(node, &content->header.entries_lba, "entries_lba"); - get_config_string(node, (char *) content->header.disk_guid, 16, "guid"); -} - -static int GPT_parse_partitions(cnode *node, struct GPT_content *content) -{ - cnode *current; - int i; - uint64_t partition_size; - struct GPT_entry_raw *entry; - for (i = 0, current = node->first_child; current; current = current->next, ++i) { - entry = &content->entries[i]; - entry_set_guid(i, content->entries[i].partition_guid); - memcpy(&content->entries[i].type_guid, partition_type_uuid, 16); - if (get_config_uint64(current, &entry->first_lba, "first_lba")) { - D(ERR, "first_lba not specified"); - return 1; - } - if (get_config_uint64(current, &partition_size, "partition_size")) { - D(ERR, "partition_size not specified"); - return 1; - } - if (config_str(current, "system", NULL)) { - entry->flags |= GPT_FLAG_SYSTEM; - } - if (config_str(current, "bootable", NULL)) { - entry->flags |= GPT_FLAG_BOOTABLE; - } - if (config_str(current, "readonly", NULL)) { - entry->flags |= GPT_FLAG_READONLY; - } - if (config_str(current, "automount", NULL)) { - entry->flags |= GPT_FLAG_DOAUTOMOUNT; - } - - get_config_uint64(current, &content->entries[i].flags, "flags"); - content->entries[i].last_lba = content->entries[i].first_lba + partition_size - 1; - GPT_to_UTF16(content->entries[i].name, current->name, 16); - } - return 0; -} - -static inline int cnode_count(cnode *node) -{ - int i; - cnode *current; - for (i = 0, current = node->first_child; current; current = current->next, ++i) - ; - return i; -} - - -static int GPT_parse_cnode(cnode *root, struct GPT_content *content) -{ - cnode *partnode; - - if (!(partnode = config_find(root, "partitions"))) { - D(ERR, "Could not find partition table"); - return 0; - } - - GPT_parse_header(root, content); - - content->header.entries_count = cnode_count(partnode); - content->entries = malloc(content->header.entries_count * sizeof(struct GPT_entry_raw)); - - if (GPT_parse_partitions(partnode, content)) { - D(ERR, "Could not parse partitions"); - return 0; - } - - return 1; -} - -int GPT_parse_file(int fd, struct GPT_content *content) -{ - char *data; - int size; - int ret; - cnode *root = config_node("", ""); - - size = get_file_size(fd); - data = (char *) mmap(NULL, size + 1, PROT_READ | PROT_WRITE, MAP_PRIVATE, fd, 0); - - if (data == NULL) { - if (size == 0) - D(ERR, "config file empty"); - else - D(ERR, "Out of memory"); - return 0; - } - - data[size - 1] = 0; - config_load(root, data); - - if (root->first_child == NULL) { - D(ERR, "Could not read config file"); - return 0; - } - - ret = GPT_parse_cnode(root, content); - munmap(data, size); - return ret; -} - -void GPT_release_content(struct GPT_content *content) -{ - free(content->entries); -} - -int GPT_write_content(const char *device, struct GPT_content *content) -{ - struct GPT_entry_table *maptable; - - maptable = GPT_get_from_content(device, content); - if (maptable == NULL) { - D(ERR, "could not map device"); - return 0; - } - - memcpy(maptable->header, &content->header, sizeof(*maptable->header)); - memcpy(maptable->entries, content->entries, - content->header.entries_count * sizeof(*maptable->entries)); - - GPT_sync(maptable); - GPT_release_device(maptable); - - return 1; -} - diff --git a/fastbootd/commands/partitions.h b/fastbootd/commands/partitions.h deleted file mode 100644 index 9a6a88d54..000000000 --- a/fastbootd/commands/partitions.h +++ /dev/null @@ -1,140 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - - -#ifndef __FASTBOOTD_PATITIONS_ -#define __FASTBOOTD_PATITIONS_ - -#include - -#define GPT_ENTRIES 128 -#define GPT_NAMELEN 36 - -#define GPT_FLAG_SYSTEM (1ULL << 0) -#define GPT_FLAG_BOOTABLE (1ULL << 2) -#define GPT_FLAG_READONLY (1ULL << 60) -#define GPT_FLAG_DOAUTOMOUNT (1ULL << 63) - -// it should be passed in little endian order -struct GPT_entry_raw { - uint8_t type_guid[16]; - uint8_t partition_guid[16]; - uint64_t first_lba; // little endian - uint64_t last_lba; - uint64_t flags; - uint16_t name[GPT_NAMELEN]; // UTF-16 LE -}; - -struct GPT_mapping { - void *map_ptr; - void *ptr; - unsigned size; -}; - -struct GPT_entry_table { - int fd; - - struct GPT_mapping header_map; - struct GPT_mapping entries_map; - struct GPT_mapping sec_header_map; - struct GPT_mapping sec_entries_map; - - struct GPT_header *header; - struct GPT_entry_raw *entries; - struct GPT_header *second_header; - struct GPT_entry_raw *second_entries; - - unsigned sector_size; - unsigned partition_table_size; - int second_valid; -}; - -struct GPT_header { - uint8_t signature[8]; - uint32_t revision; - uint32_t header_size; - uint32_t header_checksum; - uint32_t reserved_zeros; - uint64_t current_lba; - uint64_t backup_lba; - uint64_t first_usable_lba; - uint64_t last_usable_lba; - uint8_t disk_guid[16]; - uint64_t entries_lba; - uint32_t entries_count; - uint32_t entry_size; - uint32_t partition_array_checksum; - // the rest should be filled with zeros -} __attribute__((packed)); - -struct GPT_content { - struct GPT_header header; - struct GPT_entry_raw *entries; -}; - - -struct GPT_entry_table* GPT_get_device(const char *, unsigned lba); - -void GPT_release_device(struct GPT_entry_table *); - -void GPT_edit_entry(struct GPT_entry_table *table, - struct GPT_entry_raw *old_entry, - struct GPT_entry_raw *new_entry); - -int GPT_delete_entry(struct GPT_entry_table *table, struct GPT_entry_raw *entry); - -void GPT_add_entry(struct GPT_entry_table *table, struct GPT_entry_raw *entry); - -struct GPT_entry_raw *GPT_get_pointer(struct GPT_entry_table *table, struct GPT_entry_raw *entry); -struct GPT_entry_raw *GPT_get_pointer_by_guid(struct GPT_entry_table *, const char *); -struct GPT_entry_raw *GPT_get_pointer_by_name(struct GPT_entry_table *, const char *); - -//Use after every edit operation -void GPT_sync(); - -void GPT_to_UTF16(uint16_t *, const char *, int ); -void GPT_from_UTF16(char *, const uint16_t *, int); - -int GPT_parse_entry(char *string, struct GPT_entry_raw *entry); - -void GPT_default_content(struct GPT_content *content, struct GPT_entry_table *table); - -void GPT_release_content(struct GPT_content *content); - -int GPT_parse_file(int fd, struct GPT_content *content); - -int GPT_write_content(const char *device, struct GPT_content *content); - -int gpt_mmap(struct GPT_mapping *mapping, uint64_t location, int size, int fd); - -void gpt_unmap(struct GPT_mapping *mapping); - -#endif diff --git a/fastbootd/commands/virtual_partitions.c b/fastbootd/commands/virtual_partitions.c deleted file mode 100644 index b72d5b63c..000000000 --- a/fastbootd/commands/virtual_partitions.c +++ /dev/null @@ -1,69 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#include "commands/virtual_partitions.h" - -#include -#include - -#include "debug.h" - -static struct virtual_partition *partitions = NULL; - -int try_handle_virtual_partition(struct protocol_handle *handle, const char *arg) -{ - struct virtual_partition *current; - - for (current = partitions; current != NULL; current = current->next) { - if (!strcmp(current->name, arg)) { - current->handler(handle, arg); - } - } - - return 0; -} - -void virtual_partition_register( - const char * name, - void (*handler)(struct protocol_handle *phandle, const char *arg)) -{ - struct virtual_partition *new; - new = malloc(sizeof(*new)); - if (new) { - new->name = name; - new->handler = handler; - new->next = partitions; - partitions = new; - } - else { - D(ERR, "Out of memory"); - } -} diff --git a/fastbootd/commands/virtual_partitions.h b/fastbootd/commands/virtual_partitions.h deleted file mode 100644 index 88df71e95..000000000 --- a/fastbootd/commands/virtual_partitions.h +++ /dev/null @@ -1,49 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#ifndef FASTBOOTD_VIRTUAL_PARTITIONS_H -#define FASTBOOTD_VIRTUAL_PARTITIONS_H - -#include "protocol.h" - -struct virtual_partition { - struct virtual_partition *next; - const char *name; - void (*handler)(struct protocol_handle *phandle, const char *arg); -}; - -int try_handle_virtual_partition(struct protocol_handle *handle, const char *arg); - -void virtual_partition_register( - const char * name, - void (*handler)(struct protocol_handle *phandle, const char *arg)); - -#endif diff --git a/fastbootd/config.c b/fastbootd/config.c deleted file mode 100644 index daac3fd10..000000000 --- a/fastbootd/config.c +++ /dev/null @@ -1,147 +0,0 @@ -/* - * Copyright (c) 2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#include -#include -#include -#include -#include -#include - -#include "protocol.h" -#include "utils.h" - -#include "debug.h" - -// TODO: change config path -#define CONFIG_PATH "/data/fastboot.cfg" - -static int config_parse_line(char *line) -{ - char *c; - char *key; - char *value; - - c = strchr(line, '#'); - if (c) - *c = '\0'; - - if (strspn(line, " \t") == strlen(line)) - return 0; - - c = strchr(line, '='); - if (c == NULL) - return -1; - - key = line; - *c = '\0'; - value = c + 1; - - key = strip(key); - value = strip(value); - - key = strdup(key); - value = strdup(value); - - fastboot_publish(key, value); - - return 0; -} - -static void config_parse(char *buffer) -{ - char *saveptr; - char *str = buffer; - char *line = buffer; - int c; - int ret; - - for (c = 1; line != NULL; c++) { - line = strtok_r(str, "\r\n", &saveptr); - if (line != NULL) { - D(VERBOSE, "'%s'", line); - ret = config_parse_line(line); - if (ret < 0) { - D(WARN, "error parsing " CONFIG_PATH " line %d", c); - } - } - str = NULL; - } -} - -void config_init() -{ - int fd; - off_t len; - ssize_t ret; - size_t count = 0; - char *buffer; - - fd = open(CONFIG_PATH, O_RDONLY); - if (fd < 0) { - D(ERR, "failed to open " CONFIG_PATH); - return; - } - - len = lseek(fd, 0, SEEK_END); - if (len < 0) { - D(ERR, "failed to seek to end of " CONFIG_PATH); - return; - } - - lseek(fd, 0, SEEK_SET); - - buffer = malloc(len + 1); - if (buffer == NULL) { - D(ERR, "failed to allocate %ld bytes", len); - return; - } - - while (count < (size_t)len) { - ret = read(fd, buffer + count, len - count); - if (ret < 0 && errno != EINTR) { - D(ERR, "failed to read " CONFIG_PATH ": %d %s", errno, strerror(errno)); - return; - } - if (ret == 0) { - D(ERR, "early EOF reading " CONFIG_PATH); - return; - } - - count += ret; - } - - buffer[len] = '\0'; - - config_parse(buffer); - - free(buffer); -} diff --git a/fastbootd/debug.h b/fastbootd/debug.h deleted file mode 100644 index 74620b829..000000000 --- a/fastbootd/debug.h +++ /dev/null @@ -1,42 +0,0 @@ -/* - * Copyright (C) 2013 The Android Open Source Project - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#ifndef _FASTBOOTD_DEBUG_H_ -#define _FASTBOOTD_DEBUG_H_ - -#include - -#include - -#define ERR 0 -#define WARN 1 -#define INFO 2 -#define VERBOSE 3 -#define DEBUG 4 - -extern unsigned int debug_level; - -//#define DLOG(fmt, ...) printf(fmt, ##__VA_ARGS__) -#define DLOG(fmt, ...) KLOG_INFO("fastbootd", fmt, ##__VA_ARGS__) - -#define D(level, fmt, ...) \ - do { \ - if (debug_level == level || debug_level > level) { \ - DLOG("%s:%d " fmt "\n", __BASE_FILE__, __LINE__, ##__VA_ARGS__); \ - } \ - } while (0) - -#endif diff --git a/fastbootd/fastbootd.c b/fastbootd/fastbootd.c deleted file mode 100644 index 2b51b330e..000000000 --- a/fastbootd/fastbootd.c +++ /dev/null @@ -1,100 +0,0 @@ -/* - * Copyright (C) 2013 The Android Open Source Project - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#include -#include -#include -#include -#include - -#include "debug.h" -#include "trigger.h" -#include "socket_client.h" -#include "secure.h" - -unsigned int debug_level = DEBUG; - -void commands_init(); -void usb_init(); -void config_init(); -int transport_socket_init(); -int network_discovery_init(); -void ssh_server_start(); - -int main(int argc, char **argv) -{ - int socket_client = 0; - int c; - int network = 1; - - klog_init(); - klog_set_level(6); - - const struct option longopts[] = { - {"socket", no_argument, 0, 'S'}, - {"nonetwork", no_argument, 0, 'n'}, - {0, 0, 0, 0} - }; - - while (1) { - c = getopt_long(argc, argv, "Sn", longopts, NULL); - /* Alphabetical cases */ - if (c < 0) - break; - switch (c) { - case 'S': - socket_client = 1; - break; - case 'n': - network = 0; - break; - case '?': - return 1; - default: - return 0; - } - } - - (void)argc; - (void)argv; - - klog_init(); - klog_set_level(6); - - if (socket_client) { - //TODO: Shouldn't we change current tty into raw mode? - run_socket_client(); - } - else { - cert_init_crypto(); - config_init(); - load_trigger(); - commands_init(); - usb_init(); - - if (network) { - if (!transport_socket_init()) - exit(1); - ssh_server_start(); - network_discovery_init(); - } - - while (1) { - sleep(1); - } - } - return 0; -} diff --git a/fastbootd/network_discovery.c b/fastbootd/network_discovery.c deleted file mode 100644 index 1cd3e4809..000000000 --- a/fastbootd/network_discovery.c +++ /dev/null @@ -1,118 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#include -#include -#include -#include -#include - -#include "debug.h" -#include "network_discovery.h" -#include "utils.h" - -#define MDNS_SERVICE_NAME "mdnsd" -#define MDNS_SERVICE_STATUS "init.svc.mdnsd" -#define FASTBOOTD_TYPE "_fastbootd._tcp" -#define FASTBOOTD_DOMAIN "local." -#define FASTBOOTD_NAME "fastbootd" - - -static void reg_reply(DNSServiceRef sdref, const DNSServiceFlags flags, DNSServiceErrorType errorCode, - const char *name, const char *regtype, const char *domain, void *context) -{ - (void)sdref; // Unused - (void)flags; // Unused - (void)context; // Unused - if (errorCode == kDNSServiceErr_ServiceNotRunning) { - fprintf(stderr, "Error code %d\n", errorCode); - } - - - printf("Got a reply for service %s.%s%s: ", name, regtype, domain); - - if (errorCode == kDNSServiceErr_NoError) - { - if (flags & kDNSServiceFlagsAdd) - printf("Name now registered and active\n"); - else - printf("Name registration removed\n"); - if (errorCode == kDNSServiceErr_NameConflict) - printf("Name in use, please choose another\n"); - else - printf("Error %d\n", errorCode); - - if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout); - } -} - -static int register_service() { - DNSServiceRef sdref = NULL; - const char *domain = FASTBOOTD_DOMAIN; - const char *type = FASTBOOTD_TYPE; - const char *host = NULL; - char name[PROP_VALUE_MAX]; - uint16_t port = 22; - int flags = 0; - DNSServiceErrorType result; - property_get("ro.serialno", name, ""); - if (!strcmp(name, "")) { - D(ERR, "No property serialno"); - return -1; - } - - result = DNSServiceRegister(&sdref, flags, kDNSServiceInterfaceIndexAny, - name, type, domain, host, port, - 0, NULL, reg_reply, NULL); - if (result != kDNSServiceErr_NoError) { - D(ERR, "Unable to register service"); - return -1; - } - return 0; -} - - -int network_discovery_init() -{ - D(INFO, "Starting discovery"); - if (service_start(MDNS_SERVICE_NAME)) { - D(ERR, "Unable to start discovery"); - return -1; - } - - if (register_service()) { - D(ERR, "Unable to register service"); - return -1; - } - - return 0; -} - diff --git a/fastbootd/network_discovery.h b/fastbootd/network_discovery.h deleted file mode 100644 index 75fda6383..000000000 --- a/fastbootd/network_discovery.h +++ /dev/null @@ -1,37 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#ifndef _FASTBOOTD_NETWORK_DISCOVERY_H -#define _FASTBOOTD_NETWORK_DISCOVERY_H - -int network_discovery_init(); - -#endif diff --git a/fastbootd/other/gptedit.c b/fastbootd/other/gptedit.c deleted file mode 100644 index 2b65a012d..000000000 --- a/fastbootd/other/gptedit.c +++ /dev/null @@ -1,304 +0,0 @@ -/* - * Copyright (c) 2009-2013, Google Inc. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in - * the documentation and/or other materials provided with the - * distribution. - * * Neither the name of Google, Inc. nor the names of its contributors - * may be used to endorse or promote products derived from this - * software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS - * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE - * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, - * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, - * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS - * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED - * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, - * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT - * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - */ - -#include -#include -#include -#include -#include -#include - -#include - -#include "commands/partitions.h" -#include "debug.h" - -unsigned int debug_level = DEBUG; -//TODO: add tool to generate config file - -void usage() { - fprintf(stderr, - "usage: test_gpt [