qemu/dtc/tests/dtb_reverse.c

   1 /*
   2  * libfdt - Flat Device Tree manipulation
   3  *      Tests if two given dtbs are structurally equal (including order)
   4  * Copyright (C) 2010 David Gibson, IBM Corporation.
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public License
   8  * as published by the Free Software Foundation; either version 2.1 of
   9  * the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 #include <stdlib.h>
  22 #include <stdio.h>
  23 #include <string.h>
  24 #include <stdint.h>
  25 #include <limits.h>
  26
  27 #include <libfdt.h>
  28
  29 #include "tests.h"
  30 #include "testdata.h"
  31
  32 #define CHECK(code) \
  33         { \
  34                 err = (code); \
  35                 if (err) \
  36                         FAIL(#code ": %s", fdt_strerror(err)); \
  37         }
  38
  39 static void reverse_reservemap(void *in, void *out, int n)
  40 {
  41         int err;
  42         uint64_t addr, size;
  43
  44         verbose_printf("reverse_reservemap(): %d/%d\n",
  45                        n, fdt_num_mem_rsv(in));
  46
  47         if (n < (fdt_num_mem_rsv(in)-1))
  48                 reverse_reservemap(in, out, n+1);
  49
  50         CHECK(fdt_get_mem_rsv(in, n, &addr, &size));
  51         CHECK(fdt_add_reservemap_entry(out, addr, size));
  52         verbose_printf("Added entry 0x%llx 0x%llx\n",
  53                        (unsigned long long)addr, (unsigned long long)size);
  54 }
  55
  56 static void reverse_properties(void *in, void *out, int offset)
  57 {
  58         int err;
  59         int len;
  60         const char *name;
  61         const void *data;
  62
  63         data = fdt_getprop_by_offset(in, offset, &name, &len);
  64         if (!data)
  65                 FAIL("fdt_getprop_by_offset(): %s\n", fdt_strerror(len));
  66
  67         verbose_printf("reverse_properties(): offset=%d  name=%s\n",
  68                        offset, name);
  69
  70         offset = fdt_next_property_offset(in, offset);
  71         if (offset >= 0)
  72                 reverse_properties(in, out, offset);
  73         else if (offset != -FDT_ERR_NOTFOUND)
  74                 FAIL("fdt_next_property_offset(): %s\n", fdt_strerror(offset));
  75
  76         CHECK(fdt_property(out, name, data, len));
  77         verbose_printf("  -> output property %s\n", name);
  78 }
  79
  80 static void reverse_node(void *in, void *out, int nodeoffset);
  81
  82 static void reverse_children(void *in, void *out, int offset)
  83 {
  84         int err;
  85         int nextoffset = offset;
  86         int depth = 1;
  87
  88         do {
  89                 char path[PATH_MAX];
  90
  91                 CHECK(fdt_get_path(in, nextoffset, path, sizeof(path)));
  92                 verbose_printf("reverse_children() offset=%d nextoffset=%d [%s]"
  93                                " depth=%d\n", offset, nextoffset, path, depth);
  94
  95                 nextoffset = fdt_next_node(in, nextoffset, &depth);
  96         } while ((depth >= 0) && (depth != 1));
  97
  98         if (depth == 1)
  99                 reverse_children(in, out, nextoffset);
 100
 101         reverse_node(in, out, offset);
 102 }
 103
 104 static void reverse_node(void *in, void *out, int nodeoffset)
 105 {
 106         const char *name = fdt_get_name(in, nodeoffset, NULL);
 107         char path[PATH_MAX];
 108         int err;
 109         int offset;
 110         int depth = 0;
 111
 112         CHECK(fdt_get_path(in, nodeoffset, path, sizeof(path)));
 113         verbose_printf("reverse_node(): nodeoffset=%d [%s]\n",
 114                        nodeoffset, path);
 115
 116         CHECK(fdt_begin_node(out, name));
 117
 118         offset = fdt_first_property_offset(in, nodeoffset);
 119         if (offset >= 0)
 120                 reverse_properties(in, out, offset);
 121         else if (offset != -FDT_ERR_NOTFOUND)
 122                 FAIL("fdt_first_property(): %s\n", fdt_strerror(offset));
 123
 124         offset = fdt_next_node(in, nodeoffset, &depth);
 125
 126         if (depth == 1)
 127                 reverse_children(in, out, offset);
 128
 129         CHECK(fdt_end_node(out));
 130 }
 131
 132 int main(int argc, char *argv[])
 133 {
 134         void *in, *out;
 135         char outname[PATH_MAX];
 136         int bufsize;
 137         int err;
 138
 139         test_init(argc, argv);
 140         if (argc != 2)
 141                 CONFIG("Usage: %s <dtb file>", argv[0]);
 142
 143         in = load_blob(argv[1]);
 144         sprintf(outname, "%s.reversed.test.dtb", argv[1]);
 145
 146         bufsize = fdt_totalsize(in);
 147         out = xmalloc(bufsize);
 148
 149         CHECK(fdt_create(out, bufsize));
 150
 151         fdt_set_boot_cpuid_phys(out, fdt_boot_cpuid_phys(in));
 152
 153         reverse_reservemap(in, out, 0);
 154         CHECK(fdt_finish_reservemap(out));
 155
 156         reverse_node(in, out, 0);
 157
 158         CHECK(fdt_finish(out));
 159
 160         save_blob(outname, out);
 161
 162         PASS();
 163 }