322 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			322 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * Functions to help device tree manipulation using libfdt.
 | |
|  * It also provides functions to read entries from device tree proc
 | |
|  * interface.
 | |
|  *
 | |
|  * Copyright 2008 IBM Corporation.
 | |
|  * Authors: Jerone Young <jyoung5@us.ibm.com>
 | |
|  *          Hollis Blanchard <hollisb@us.ibm.com>
 | |
|  *
 | |
|  * This work is licensed under the GNU GPL license version 2 or later.
 | |
|  *
 | |
|  */
 | |
| 
 | |
| #include <stdio.h>
 | |
| #include <sys/types.h>
 | |
| #include <sys/stat.h>
 | |
| #include <fcntl.h>
 | |
| #include <unistd.h>
 | |
| #include <stdlib.h>
 | |
| 
 | |
| #include "config.h"
 | |
| #include "qemu-common.h"
 | |
| #include "sysemu/device_tree.h"
 | |
| #include "hw/loader.h"
 | |
| #include "qemu/option.h"
 | |
| #include "qemu/config-file.h"
 | |
| 
 | |
| #include <libfdt.h>
 | |
| 
 | |
| #define FDT_MAX_SIZE  0x10000
 | |
| 
 | |
| void *create_device_tree(int *sizep)
 | |
| {
 | |
|     void *fdt;
 | |
|     int ret;
 | |
| 
 | |
|     *sizep = FDT_MAX_SIZE;
 | |
|     fdt = g_malloc0(FDT_MAX_SIZE);
 | |
|     ret = fdt_create(fdt, FDT_MAX_SIZE);
 | |
|     if (ret < 0) {
 | |
|         goto fail;
 | |
|     }
 | |
|     ret = fdt_begin_node(fdt, "");
 | |
|     if (ret < 0) {
 | |
|         goto fail;
 | |
|     }
 | |
|     ret = fdt_end_node(fdt);
 | |
|     if (ret < 0) {
 | |
|         goto fail;
 | |
|     }
 | |
|     ret = fdt_finish(fdt);
 | |
|     if (ret < 0) {
 | |
|         goto fail;
 | |
|     }
 | |
|     ret = fdt_open_into(fdt, fdt, *sizep);
 | |
|     if (ret) {
 | |
|         fprintf(stderr, "Unable to copy device tree in memory\n");
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     return fdt;
 | |
| fail:
 | |
|     fprintf(stderr, "%s Couldn't create dt: %s\n", __func__, fdt_strerror(ret));
 | |
|     exit(1);
 | |
| }
 | |
| 
 | |
| void *load_device_tree(const char *filename_path, int *sizep)
 | |
| {
 | |
|     int dt_size;
 | |
|     int dt_file_load_size;
 | |
|     int ret;
 | |
|     void *fdt = NULL;
 | |
| 
 | |
|     *sizep = 0;
 | |
|     dt_size = get_image_size(filename_path);
 | |
|     if (dt_size < 0) {
 | |
|         printf("Unable to get size of device tree file '%s'\n",
 | |
|             filename_path);
 | |
|         goto fail;
 | |
|     }
 | |
| 
 | |
|     /* Expand to 2x size to give enough room for manipulation.  */
 | |
|     dt_size += 10000;
 | |
|     dt_size *= 2;
 | |
|     /* First allocate space in qemu for device tree */
 | |
|     fdt = g_malloc0(dt_size);
 | |
| 
 | |
|     dt_file_load_size = load_image(filename_path, fdt);
 | |
|     if (dt_file_load_size < 0) {
 | |
|         printf("Unable to open device tree file '%s'\n",
 | |
|                filename_path);
 | |
|         goto fail;
 | |
|     }
 | |
| 
 | |
|     ret = fdt_open_into(fdt, fdt, dt_size);
 | |
|     if (ret) {
 | |
|         printf("Unable to copy device tree in memory\n");
 | |
|         goto fail;
 | |
|     }
 | |
| 
 | |
|     /* Check sanity of device tree */
 | |
|     if (fdt_check_header(fdt)) {
 | |
|         printf ("Device tree file loaded into memory is invalid: %s\n",
 | |
|             filename_path);
 | |
|         goto fail;
 | |
|     }
 | |
|     *sizep = dt_size;
 | |
|     return fdt;
 | |
| 
 | |
| fail:
 | |
|     g_free(fdt);
 | |
|     return NULL;
 | |
| }
 | |
| 
 | |
| static int findnode_nofail(void *fdt, const char *node_path)
 | |
| {
 | |
|     int offset;
 | |
| 
 | |
|     offset = fdt_path_offset(fdt, node_path);
 | |
|     if (offset < 0) {
 | |
|         fprintf(stderr, "%s Couldn't find node %s: %s\n", __func__, node_path,
 | |
|                 fdt_strerror(offset));
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     return offset;
 | |
| }
 | |
| 
 | |
| int qemu_devtree_setprop(void *fdt, const char *node_path,
 | |
|                          const char *property, const void *val_array, int size)
 | |
| {
 | |
|     int r;
 | |
| 
 | |
|     r = fdt_setprop(fdt, findnode_nofail(fdt, node_path), property, val_array, size);
 | |
|     if (r < 0) {
 | |
|         fprintf(stderr, "%s: Couldn't set %s/%s: %s\n", __func__, node_path,
 | |
|                 property, fdt_strerror(r));
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     return r;
 | |
| }
 | |
| 
 | |
| int qemu_devtree_setprop_cell(void *fdt, const char *node_path,
 | |
|                               const char *property, uint32_t val)
 | |
| {
 | |
|     int r;
 | |
| 
 | |
|     r = fdt_setprop_cell(fdt, findnode_nofail(fdt, node_path), property, val);
 | |
|     if (r < 0) {
 | |
|         fprintf(stderr, "%s: Couldn't set %s/%s = %#08x: %s\n", __func__,
 | |
|                 node_path, property, val, fdt_strerror(r));
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     return r;
 | |
| }
 | |
| 
 | |
| int qemu_devtree_setprop_u64(void *fdt, const char *node_path,
 | |
|                              const char *property, uint64_t val)
 | |
| {
 | |
|     val = cpu_to_be64(val);
 | |
|     return qemu_devtree_setprop(fdt, node_path, property, &val, sizeof(val));
 | |
| }
 | |
| 
 | |
| int qemu_devtree_setprop_string(void *fdt, const char *node_path,
 | |
|                                 const char *property, const char *string)
 | |
| {
 | |
|     int r;
 | |
| 
 | |
|     r = fdt_setprop_string(fdt, findnode_nofail(fdt, node_path), property, string);
 | |
|     if (r < 0) {
 | |
|         fprintf(stderr, "%s: Couldn't set %s/%s = %s: %s\n", __func__,
 | |
|                 node_path, property, string, fdt_strerror(r));
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     return r;
 | |
| }
 | |
| 
 | |
| const void *qemu_devtree_getprop(void *fdt, const char *node_path,
 | |
|                                  const char *property, int *lenp)
 | |
| {
 | |
|     int len;
 | |
|     const void *r;
 | |
|     if (!lenp) {
 | |
|         lenp = &len;
 | |
|     }
 | |
|     r = fdt_getprop(fdt, findnode_nofail(fdt, node_path), property, lenp);
 | |
|     if (!r) {
 | |
|         fprintf(stderr, "%s: Couldn't get %s/%s: %s\n", __func__,
 | |
|                 node_path, property, fdt_strerror(*lenp));
 | |
|         exit(1);
 | |
|     }
 | |
|     return r;
 | |
| }
 | |
| 
 | |
| uint32_t qemu_devtree_getprop_cell(void *fdt, const char *node_path,
 | |
|                                    const char *property)
 | |
| {
 | |
|     int len;
 | |
|     const uint32_t *p = qemu_devtree_getprop(fdt, node_path, property, &len);
 | |
|     if (len != 4) {
 | |
|         fprintf(stderr, "%s: %s/%s not 4 bytes long (not a cell?)\n",
 | |
|                 __func__, node_path, property);
 | |
|         exit(1);
 | |
|     }
 | |
|     return be32_to_cpu(*p);
 | |
| }
 | |
| 
 | |
| uint32_t qemu_devtree_get_phandle(void *fdt, const char *path)
 | |
| {
 | |
|     uint32_t r;
 | |
| 
 | |
|     r = fdt_get_phandle(fdt, findnode_nofail(fdt, path));
 | |
|     if (r <= 0) {
 | |
|         fprintf(stderr, "%s: Couldn't get phandle for %s: %s\n", __func__,
 | |
|                 path, fdt_strerror(r));
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     return r;
 | |
| }
 | |
| 
 | |
| int qemu_devtree_setprop_phandle(void *fdt, const char *node_path,
 | |
|                                  const char *property,
 | |
|                                  const char *target_node_path)
 | |
| {
 | |
|     uint32_t phandle = qemu_devtree_get_phandle(fdt, target_node_path);
 | |
|     return qemu_devtree_setprop_cell(fdt, node_path, property, phandle);
 | |
| }
 | |
| 
 | |
| uint32_t qemu_devtree_alloc_phandle(void *fdt)
 | |
| {
 | |
|     static int phandle = 0x0;
 | |
| 
 | |
|     /*
 | |
|      * We need to find out if the user gave us special instruction at
 | |
|      * which phandle id to start allocting phandles.
 | |
|      */
 | |
|     if (!phandle) {
 | |
|         QemuOpts *machine_opts;
 | |
|         machine_opts = qemu_opts_find(qemu_find_opts("machine"), 0);
 | |
|         if (machine_opts) {
 | |
|             const char *phandle_start;
 | |
|             phandle_start = qemu_opt_get(machine_opts, "phandle_start");
 | |
|             if (phandle_start) {
 | |
|                 phandle = strtoul(phandle_start, NULL, 0);
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     if (!phandle) {
 | |
|         /*
 | |
|          * None or invalid phandle given on the command line, so fall back to
 | |
|          * default starting point.
 | |
|          */
 | |
|         phandle = 0x8000;
 | |
|     }
 | |
| 
 | |
|     return phandle++;
 | |
| }
 | |
| 
 | |
| int qemu_devtree_nop_node(void *fdt, const char *node_path)
 | |
| {
 | |
|     int r;
 | |
| 
 | |
|     r = fdt_nop_node(fdt, findnode_nofail(fdt, node_path));
 | |
|     if (r < 0) {
 | |
|         fprintf(stderr, "%s: Couldn't nop node %s: %s\n", __func__, node_path,
 | |
|                 fdt_strerror(r));
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     return r;
 | |
| }
 | |
| 
 | |
| int qemu_devtree_add_subnode(void *fdt, const char *name)
 | |
| {
 | |
|     char *dupname = g_strdup(name);
 | |
|     char *basename = strrchr(dupname, '/');
 | |
|     int retval;
 | |
|     int parent = 0;
 | |
| 
 | |
|     if (!basename) {
 | |
|         g_free(dupname);
 | |
|         return -1;
 | |
|     }
 | |
| 
 | |
|     basename[0] = '\0';
 | |
|     basename++;
 | |
| 
 | |
|     if (dupname[0]) {
 | |
|         parent = findnode_nofail(fdt, dupname);
 | |
|     }
 | |
| 
 | |
|     retval = fdt_add_subnode(fdt, parent, basename);
 | |
|     if (retval < 0) {
 | |
|         fprintf(stderr, "FDT: Failed to create subnode %s: %s\n", name,
 | |
|                 fdt_strerror(retval));
 | |
|         exit(1);
 | |
|     }
 | |
| 
 | |
|     g_free(dupname);
 | |
|     return retval;
 | |
| }
 | |
| 
 | |
| void qemu_devtree_dumpdtb(void *fdt, int size)
 | |
| {
 | |
|     QemuOpts *machine_opts;
 | |
| 
 | |
|     machine_opts = qemu_opts_find(qemu_find_opts("machine"), 0);
 | |
|     if (machine_opts) {
 | |
|         const char *dumpdtb = qemu_opt_get(machine_opts, "dumpdtb");
 | |
|         if (dumpdtb) {
 | |
|             /* Dump the dtb to a file and quit */
 | |
|             exit(g_file_set_contents(dumpdtb, fdt, size, NULL) ? 0 : 1);
 | |
|         }
 | |
|     }
 | |
| 
 | |
| }
 | 
