[PATCH v2 04/10] pylibfdt: Add support for reading the memory reserve map

[Simon Glass <sjg@xxxxxxxxxxxx>]

Add a way to access this information from Python.

Signed-off-by: Simon Glass <sjg@xxxxxxxxxxxx>
Reviewed-by: David Gibson <david@xxxxxxxxxxxxxxxxxxxxx>
---

Changes in v2: None

 pylibfdt/libfdt.i       | 34 ++++++++++++++++++++++++++++++++++
 tests/pylibfdt_tests.py |  8 ++++++++
 2 files changed, 42 insertions(+)

diff --git a/pylibfdt/libfdt.i b/pylibfdt/libfdt.i
index f33e2ab..b8732dc 100644
--- a/pylibfdt/libfdt.i
+++ b/pylibfdt/libfdt.i
@@ -333,6 +333,25 @@ class Fdt:
         """
         return fdt_size_dt_struct(self._fdt)
 
+    def num_mem_rsv(self, quiet=()):
+        """Return the number of memory reserve-map records
+
+        Returns:
+            Number of memory reserve-map records
+        """
+        return check_err(fdt_num_mem_rsv(self._fdt), quiet)
+
+    def get_mem_rsv(self, index, quiet=()):
+        """Return the indexed memory reserve-map record
+
+        Args:
+            index: Record to return (0=first)
+
+        Returns:
+            Number of memory reserve-map records
+        """
+        return check_err(fdt_get_mem_rsv(self._fdt, index), quiet)
+
     def subnode_offset(self, parentoffset, name, quiet=()):
         """Get the offset of a named subnode
 
@@ -605,6 +624,21 @@ typedef int fdt32_t;
 
 %apply int *depth { int *depth };
 
+/* typemaps for fdt_get_mem_rsv */
+%typemap(in, numinputs=0) uint64_t * (uint64_t temp) {
+   $1 = &temp;
+}
+
+%typemap(argout) uint64_t * {
+        PyObject *val = PyLong_FromUnsignedLong(*arg$argnum);
+        if (!result) {
+           if (PyTuple_GET_SIZE(resultobj) == 0)
+              resultobj = val;
+           else
+              resultobj = SWIG_Python_AppendOutput(resultobj, val);
+        }
+}
+
 /* We have both struct fdt_property and a function fdt_property() */
 %warnfilter(302) fdt_property;
 
diff --git a/tests/pylibfdt_tests.py b/tests/pylibfdt_tests.py
index 17bf30b..458bc9a 100644
--- a/tests/pylibfdt_tests.py
+++ b/tests/pylibfdt_tests.py
@@ -370,5 +370,13 @@ class PyLibfdtTests(unittest.TestCase):
                           libfdt.as_uint64(self.get_prop("prop-uint64")))
         self.assertEquals(-2, libfdt.as_int64(self.get_prop("prop-int64")))
 
+    def testReserveMap(self):
+        """Test that we can access the memory reserve map"""
+        self.assertEquals(2, self.fdt.num_mem_rsv())
+        self.assertEquals([ 0xdeadbeef00000000, 0x100000],
+                          self.fdt.get_mem_rsv(0))
+        self.assertEquals([123456789, 010000], self.fdt.get_mem_rsv(1))
+
+
 if __name__ == "__main__":
     unittest.main()
-- 
2.17.1.1185.g55be947832-goog

--
To unsubscribe from this list: send the line "unsubscribe devicetree-compiler" in
the body of a message to majordomo@xxxxxxxxxxxxxxx
More majordomo info at  http://vger.kernel.org/majordomo-info.html