aboot: Add dtree support for flash targets.
Change-Id: I320a66851dbdb9b842f2e15894082480a257b7ef
diff --git a/platform/msm_shared/dev_tree.c b/platform/msm_shared/dev_tree.c
new file mode 100644
index 0000000..29cce69
--- /dev/null
+++ b/platform/msm_shared/dev_tree.c
@@ -0,0 +1,424 @@
+/* Copyright (c) 2012, The Linux Foundation. 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 The Linux Foundation 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 "AS IS" AND ANY EXPRESS OR IMPLIED
+ * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT
+ * 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 <libfdt.h>
+#include <dev_tree.h>
+#include <lib/ptable.h>
+#include <malloc.h>
+#include <qpic_nand.h>
+#include <stdlib.h>
+#include <string.h>
+#include <platform.h>
+#include <board.h>
+
+extern unsigned char *update_cmdline(const char * cmdline);
+extern int target_is_emmc_boot(void);
+extern uint32_t target_dev_tree_mem(void *fdt, uint32_t memory_node_offset);
+
+/* Function to return the pointer to the start of the correct device tree
+ * based on the platform data.
+ */
+struct dt_entry * dev_tree_get_entry_ptr(struct dt_table *table)
+{
+ uint32_t i;
+ struct dt_entry *dt_entry_ptr;
+
+ dt_entry_ptr = (struct dt_entry *)((char *)table + DEV_TREE_HEADER_SIZE);
+
+ for(i = 0; i < table->num_entries; i++)
+ {
+ /* TODO: Add support to pass soc rev correctly.
+ * Currently, the code does not support different soc revisions.
+ */
+ if((dt_entry_ptr->platform_id == board_platform_id()) &&
+ (dt_entry_ptr->variant_id == board_hardware_id()) &&
+ (dt_entry_ptr->soc_rev == 0))
+ {
+ return dt_entry_ptr;
+ }
+ dt_entry_ptr++;
+ }
+ return NULL;
+}
+
+/* Function to return the offset of flash node - "/qcom,mtd-partitions".
+ * The function creates this node if it does not exist.
+ */
+static uint32_t dev_tree_get_flash_node_offset(void *fdt)
+{
+ uint32_t offset;
+ int ret = DT_OP_SUCCESS;
+
+ /* Find the mtd node. */
+ ret = fdt_path_offset(fdt, "/qcom,mtd-partitions");
+
+ if (ret & FDT_ERR_NOTFOUND)
+ {
+ /* Node not found.
+ * Add node as sub node of root.
+ */
+ ret = fdt_path_offset(fdt, "/");
+
+ if (ret < 0)
+ {
+ dprintf(CRITICAL, "Unable to calculate root offset\n");
+ ret = DT_OP_FAILURE;
+ goto dev_tree_get_flash_node_offset_err;
+ }
+
+ offset = ret;
+ /* Add flash partition node. */
+ ret = fdt_add_subnode(fdt, offset, "qcom,mtd-partitions");
+
+ if (ret < 0)
+ {
+ dprintf(CRITICAL, "Unable to add partition node. \n");
+ ret = DT_OP_FAILURE;
+ goto dev_tree_get_flash_node_offset_err;
+ }
+
+ /* Save the offset of the added node. */
+ offset = fdt_path_offset(fdt, "/qcom,mtd-partitions");
+ }
+ else if (ret != 0)
+ {
+ dprintf(CRITICAL, "Unable to calculate partition node offset\n");
+ ret = DT_OP_FAILURE;
+ goto dev_tree_get_flash_node_offset_err;
+ }
+
+dev_tree_get_flash_node_offset_err:
+
+ return offset;
+}
+
+/* Function to add the individual nand partition nodes to the device tree.
+ * Pre-condition: The function assumes the presence of
+ * "/qcom,mtd-partitions" device node.
+ */
+static int dev_tree_add_ptable_nodes(void *fdt, uint32_t parent_offset)
+{
+ struct ptable *ptable;
+ int n;
+ unsigned char array[100];
+ unsigned char *ptn_name_array;
+ int dt_ret = DT_OP_SUCCESS;
+ int ret;
+ int i;
+ uint32_t node_offset;
+ uint32_t blk_size;
+
+ n = sizeof("partition@");
+
+ /* Allocate bytes to save partition name:
+ * Since address is of uint uint32_t,
+ * allocate twice this size for string
+ * as 1 digit occupies 1 byte is ASCII.
+ */
+ ptn_name_array = (unsigned char*) malloc(sizeof(uint32_t) * 2 + n + 1);
+
+ if (ptn_name_array == NULL)
+ {
+ dprintf(CRITICAL, "Failed to allocate memory for flash partition name\n");
+ return -1;
+ }
+
+ strcpy((char*)ptn_name_array, (const char*)"partition@");
+
+ /* Add ptable nodes. */
+ ptable = flash_get_ptable();
+ /* Get block size. */
+ blk_size = flash_block_size();
+
+ /* Need to add partitions in reverse order since libfdt adds
+ * new nodes on the top.
+ * Kernel looks to mount the partitions in the order specified in
+ * the partition.xml in the meta build.
+ */
+ for (i = (ptable->count - 1); i >= 0; i--)
+ {
+ /* Add the partition node. */
+ if (itoa(ptable->parts[i].start * blk_size, ptn_name_array + n - 1, sizeof(uint32_t) * 2 + 1, 16) < 0)
+ {
+ dprintf(CRITICAL, "String len exceeded for itoa\n");
+ return -1;
+ }
+
+ strcpy((char *)array, (const char*)"/qcom,mtd-partitions/");
+ strcat((char*)array, (const char*)ptn_name_array);
+
+ ret = fdt_add_subnode(fdt, parent_offset, (const char*)ptn_name_array);
+
+ if (ret < 0)
+ {
+ dprintf(CRITICAL, "Unable to add partition node: %s.\n",
+ ptn_name_array);
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_ptable_nodes_err;
+ }
+
+ dt_ret = fdt_path_offset(fdt, (const char*)array);
+
+ if (dt_ret < 0)
+ {
+ dprintf(CRITICAL, "Unable to calculate parition node offset: %s\n",
+ ptn_name_array);
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_ptable_nodes_err;
+ }
+
+ node_offset = dt_ret;
+
+ /* Add the partition name as label. */
+ ret = fdt_setprop_string(fdt, node_offset, (const char*)"label", (const char*)ptable->parts[i].name);
+
+ if (ret != 0)
+ {
+ dprintf(CRITICAL, "Unable to add label property: %s.\n",
+ ptable->parts[i].name);
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_ptable_nodes_err;
+ }
+
+ /* Add the reg values. */
+ ret = fdt_setprop_u32(fdt, node_offset, (const char*)"reg", ptable->parts[i].start * blk_size);
+
+ if (ret != 0)
+ {
+ dprintf(CRITICAL, "Unable to add reg property. %s\n",
+ ptable->parts[i].name);
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_ptable_nodes_err;
+ }
+
+ ret = fdt_appendprop_u32(fdt, node_offset, (const char*)"reg", ptable->parts[i].length * blk_size);
+
+ if (ret != 0)
+ {
+ dprintf(CRITICAL, "Unable to add reg property. %s\n",
+ ptable->parts[i].name);
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_ptable_nodes_err;
+ }
+
+ }
+
+dev_tree_add_ptable_nodes_err:
+ free(ptn_name_array);
+ return dt_ret;
+}
+
+/* Top level function to add flash ptable info to the device tree. */
+static int dev_tree_add_flash_ptable(void *fdt)
+{
+ uint32_t offset;
+ int ret;
+ int dt_ret = DT_OP_SUCCESS;
+
+ dt_ret = dev_tree_get_flash_node_offset(fdt);
+
+ if (dt_ret < 0)
+ {
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_flash_ptable_err;
+ }
+
+ offset = dt_ret;
+
+ /* Add address and size cell properties. */
+ ret = fdt_setprop_u32(fdt, offset, (const char*)"#address-cells", 1);
+
+ if (ret != 0)
+ {
+ dprintf(CRITICAL, "Unable to add #address-cells property. \n");
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_flash_ptable_err;
+ }
+
+ ret = fdt_setprop_u32(fdt, offset, (const char*)"#size-cells", 1);
+
+ if (ret != 0)
+ {
+ dprintf(CRITICAL, "Unable to add #size-cells property. \n");
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_flash_ptable_err;
+ }
+
+ ret = dev_tree_add_ptable_nodes(fdt, offset);
+
+ if (ret < 0)
+ {
+ dprintf(CRITICAL, "Unable to add #size-cells property. \n");
+ dt_ret = DT_OP_FAILURE;
+ goto dev_tree_add_flash_ptable_err;
+ }
+
+dev_tree_add_flash_ptable_err:
+ return dt_ret;
+}
+
+/* Function to add the first RAM partition info to the device tree.
+ * Note: The function replaces the reg property in the "/memory" node
+ * with the addr and size provided.
+ */
+int dev_tree_add_first_mem_info(uint32_t *fdt, uint32_t offset, uint32_t addr, uint32_t size)
+{
+ int ret;
+
+ ret = fdt_setprop_u32(fdt, offset, "reg", addr);
+
+ if (ret)
+ {
+ dprintf(CRITICAL, "Failed to add the memory information addr: %d\n",
+ ret);
+ }
+
+
+ ret = fdt_appendprop_u32(fdt, offset, "reg", size);
+
+ if (ret)
+ {
+ dprintf(CRITICAL, "Failed to add the memory information size: %d\n",
+ ret);
+ }
+
+ return ret;
+}
+
+/* Function to add the subsequent RAM partition info to the device tree. */
+int dev_tree_add_mem_info(void *fdt, uint32_t offset, uint32_t addr, uint32_t size)
+{
+ static int mem_info_cnt = 0;
+ int ret;
+
+ if (!mem_info_cnt)
+ {
+ /* Replace any other reg prop in the memory node. */
+ ret = fdt_setprop_u32(fdt, offset, "reg", addr);
+ mem_info_cnt = 1;
+ }
+ else
+ {
+ /* Append the mem info to the reg prop for subsequent nodes. */
+ ret = fdt_appendprop_u32(fdt, offset, "reg", addr);
+ }
+
+ if (ret)
+ {
+ dprintf(CRITICAL, "Failed to add the memory information addr: %d\n",
+ ret);
+ }
+
+
+ ret = fdt_appendprop_u32(fdt, offset, "reg", size);
+
+ if (ret)
+ {
+ dprintf(CRITICAL, "Failed to add the memory information size: %d\n",
+ ret);
+ }
+
+ return ret;
+}
+
+/* Top level function that updates the device tree. */
+int update_device_tree(void *fdt, const char *cmdline,
+ void *ramdisk, uint32_t ramdisk_size)
+{
+ int ret = 0;
+ uint32_t offset;
+ unsigned char *final_cmdline;
+
+ /* Check the device tree header */
+ ret = fdt_check_header(fdt);
+ if (ret)
+ {
+ dprintf(CRITICAL, "Invalid device tree header \n");
+ return ret;
+ }
+
+ /* Get offset of the memory node */
+ ret = fdt_path_offset(fdt, "/memory");
+ if (ret < 0)
+ {
+ dprintf(CRITICAL, "Could not find memory node.\n");
+ return ret;
+ }
+
+ offset = ret;
+
+ ret = target_dev_tree_mem(fdt, offset);
+ if(ret)
+ {
+ dprintf(CRITICAL, "ERROR: Cannot update memory node\n");
+ return ret;
+ }
+
+ /* Skip NAND partition nodes for eMMC boot */
+ if (!target_is_emmc_boot()){
+ dev_tree_add_flash_ptable(fdt);
+ }
+
+ /* Get offset of the chosen node */
+ ret = fdt_path_offset(fdt, "/chosen");
+ if (ret < 0)
+ {
+ dprintf(CRITICAL, "Could not find chosen node.\n");
+ return ret;
+ }
+
+ offset = ret;
+ /* Adding the cmdline to the chosen node */
+ final_cmdline = update_cmdline((const char*)cmdline);
+ ret = fdt_setprop_string(fdt, offset, (const char*)"bootargs", (const void*)final_cmdline);
+ if (ret)
+ {
+ dprintf(CRITICAL, "ERROR: Cannot update chosen node [bootargs]\n");
+ return ret;
+ }
+
+ /* Adding the initrd-start to the chosen node */
+ ret = fdt_setprop_u32(fdt, offset, "linux,initrd-start", (uint32_t)ramdisk);
+ if (ret)
+ {
+ dprintf(CRITICAL, "ERROR: Cannot update chosen node [linux,initrd-start]\n");
+ return ret;
+ }
+
+ /* Adding the initrd-end to the chosen node */
+ ret = fdt_setprop_u32(fdt, offset, "linux,initrd-end", ((uint32_t)ramdisk + ramdisk_size));
+ if (ret)
+ {
+ dprintf(CRITICAL, "ERROR: Cannot update chosen node [linux,initrd-end]\n");
+ return ret;
+ }
+
+ fdt_pack(fdt);
+
+ return ret;
+}