This simple utility allows writing of values into a device tree from the
command line.

Signed-off-by: Simon Glass <[email protected]>
---
 Makefile       |    7 ++
 Makefile.dtput |   14 ++++
 dtput.c        |  199 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 220 insertions(+), 0 deletions(-)
 create mode 100644 Makefile.dtput
 create mode 100644 dtput.c

diff --git a/Makefile b/Makefile
index 3401a65..e216006 100644
--- a/Makefile
+++ b/Makefile
@@ -107,11 +107,13 @@ include Makefile.convert-dtsv0
 include Makefile.dtc
 include Makefile.ftdump
 include Makefile.dtget
+include Makefile.dtput
 
 BIN += convert-dtsv0
 BIN += dtc
 BIN += ftdump
 BIN += dtget
+BIN += dtput
 
 SCRIPTS = dtdiff
 
@@ -123,6 +125,7 @@ ifneq ($(DEPTARGETS),)
 -include $(CONVERT_OBJS:%.o=%.d)
 -include $(FTDUMP_OBJS:%.o=%.d)
 -include $(DTGET_OBJS:%.o=%.d)
+-include $(DTPUT_OBJS:%.o=%.d)
 endif
 
 
@@ -187,6 +190,10 @@ dtget:     $(DTGET_OBJS) $(LIBFDT_archive)
        $(VECHO) LD $@
        $(LINK.c) -o $@ $^
 
+dtput: $(DTPUT_OBJS) $(LIBFDT_archive)
+       $(VECHO) LD $@
+       $(LINK.c) -o $@ $^
+
 
 #
 # Testsuite rules
diff --git a/Makefile.dtput b/Makefile.dtput
new file mode 100644
index 0000000..92eb054
--- /dev/null
+++ b/Makefile.dtput
@@ -0,0 +1,14 @@
+#
+# This is not a complete Makefile.
+# Instead, it is designed to be easily embeddable
+# into other systems of Makefiles.
+#
+
+DTPUT_SRCS = \
+       dtput.c \
+       util.c \
+       utilfdt.c
+
+DTPUT_GEN_SRCS =
+
+DTPUT_OBJS = $(DTPUT_SRCS:%.c=%.o) $(DTPUT_GEN_SRCS:%.c=%.o)
diff --git a/dtput.c b/dtput.c
new file mode 100644
index 0000000..48e3c75
--- /dev/null
+++ b/dtput.c
@@ -0,0 +1,199 @@
+/*
+ * Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
+ * Distributed under the terms of the GNU General Public License v2
+ *
+ * is_printable_string from ftdump.c
+ * Contributed by Pantelis Antoniou <pantelis.antoniou AT gmail.com>
+ */
+
+#include <ctype.h>
+#include <getopt.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <libfdt.h>
+
+#include "util.h"
+#include "utilfdt.h"
+
+int verbose;           /* verbose output */
+
+enum {
+       MAX_VALUE_SIZE = 1024
+};
+
+static void report_error(const char *where, int err)
+{
+       fprintf(stderr, "Error at '%s': %s\n", where, fdt_strerror(err));
+}
+
+/**
+ * Encode a series of arguments in a property value.
+ *
+ * @param arg  List of arguments from command line
+ * @param arg_count    Number of arguments (may be 0)
+ * @param format       sscanf() format string to use, NULL for default
+ * @param type         Value type (s=string, i=int, b=byte)
+ * @param buf          Buffer to place value into
+ * @param malen                Maximum length of buffer
+ * @param *value_len   Returns length of value encoded
+ */
+static int encode_value(char **arg, int arg_count, const char *format,
+               int type, char *buf, int maxlen, int *value_len)
+{
+       char value[MAX_VALUE_SIZE];     /* holding area for values */
+       int len;        /* length of this cell/string/byte */
+       int *iptr = (int *)value;
+       int ival;
+       int upto;       /* the number of bytes we have written to buf */
+
+       upto = 0;
+
+       if (verbose)
+               fprintf(stderr, "Decoding value:\n");
+       for (; arg_count > 0; arg++, arg_count--, upto += len) {
+               /* assume integer unless told otherwise */
+               if (type == 's') {
+                       sscanf(*arg, format ? format : "%s", value);
+                       len = strlen(value) + 1;
+                       if (verbose)
+                               fprintf(stderr, "\tstring: '%s'\n", value);
+               } else {
+                       len = type == 'b' ? 1 : 4;
+                       sscanf(*arg, format ? format : "%d", &ival);
+                       if (len == 4)
+                               *iptr = cpu_to_fdt32(ival);
+                       else
+                               *value = (uint8_t)ival;
+                       if (verbose) {
+                               fprintf(stderr, "\t%s: %d\n",
+                                       type == 'b' ? "byte" : "int", ival);
+                       }
+               }
+               if (upto + len > maxlen)
+                       return 8;
+               memcpy(buf + upto, value, len);
+       }
+       *value_len = upto;
+       if (verbose)
+               fprintf(stderr, "Value size %d\n", upto);
+       return 0;
+}
+
+static int store_key_value(void *blob, char *key, const char *buf, int len)
+{
+       char *prop;
+       int node, guess_node;
+       int err;
+
+       node = util_decode_key(blob, key, &prop, &guess_node);
+       if (node < 0) {
+               report_error(key, node);
+               return 7;
+       }
+
+       err = fdt_setprop(blob, node, prop, buf, len);
+       if (err) {
+               report_error(prop, len);
+               if (guess_node)
+                       fprintf(stderr, "(Node %s exists but you didn't "
+                               "specify a property to set)\n", key);
+               return 8;
+       }
+       return 0;
+}
+
+static int do_dtput(const char *filename, const char *format, int type,
+                   char **arg, int arg_count)
+{
+       char *blob;
+       int ret, len;
+       char buf[MAX_VALUE_SIZE];
+
+       blob = util_read_fdt(filename);
+       if (!blob)
+               return 10;
+
+       if (arg_count > 0) {
+               /* convert the value into binary */
+               ret = encode_value(arg + 1, arg_count - 1, format, type, buf,
+                               MAX_VALUE_SIZE, &len);
+               if (ret)
+                       return 8;
+
+               ret = store_key_value(blob, *arg, buf, len);
+               if (ret)
+                       return ret;
+       }
+       return util_write_fdt(blob, filename);
+}
+
+static void usage(const char *msg)
+{
+       if (msg)
+               fprintf(stderr, "Error: %s\n\n", msg);
+       fprintf(stderr, "dtput - write a property value to a device tree\n\n");
+       fprintf(stderr, "Usage: dtput <options> <dt file> <key> "
+               "[<value>...]\n");
+       fprintf(stderr, "Options:\n");
+       fprintf(stderr, "\t-f, --format <fmt>\tScanf format string to use "
+               "for value\n");
+       fprintf(stderr, "\t-t, --type <typechar>\tForce type to be string "
+               "(s), byte(b) or int(i)\n");
+       fprintf(stderr, "\t-h, --help\t\tPrint this help\n\n");
+       fprintf(stderr, "\t<key> is <node>/<property>, for example "
+               "/lcd/width for the width\n");
+       fprintf(stderr, "\tproperty in the LCD node\n");
+       exit(2);
+}
+
+int main(int argc, char *argv[])
+{
+       int c;
+       int ret;
+       char *format = NULL;
+       int type = 0;
+       char *filename = NULL;
+       static struct option long_options[] = {
+               {"format", 1, 0, 'f'},
+               {"help", 1, 0, 'h'},
+               {"type", 1, 0, 't'},
+               {"verbose", 1, 0, 'v'},
+               {0, 0, 0, 0}
+       };
+
+       for (;;) {
+               c = getopt_long(argc, argv, "f:ht:v", long_options, NULL);
+               if (c == -1)
+                       break;
+
+               switch (c) {
+               case 'h':
+               case '?':
+                       usage(NULL);
+
+               case 'f':
+                       format = optarg;
+                       break;
+
+               case 't':
+                       type = *optarg;
+                       break;
+
+               case 'v':
+                       verbose = 1;
+                       break;
+               }
+       }
+
+       if (optind < argc)
+               filename = argv[optind++];
+       if (!filename)
+               usage("Missing filename");
+       if (optind == argc)
+               usage("Missing key(s)");
+
+       ret = do_dtput(filename, format, type, argv + optind, argc - optind);
+       return ret;
+}
-- 
1.7.3.1

_______________________________________________
devicetree-discuss mailing list
[email protected]
https://lists.ozlabs.org/listinfo/devicetree-discuss

Reply via email to