We intend to add signatures to FITs also, so rename this function so that
it is not specific to hashing. Also rename fit_image_set_hashes() and
make it static since it is not used outside this file.

Signed-off-by: Simon Glass <s...@chromium.org>
---
 include/image.h    |   10 +++-
 tools/fit_image.c  |    2 +-
 tools/image-host.c |  147 +++++++++++++++++++++++++---------------------------
 3 files changed, 79 insertions(+), 80 deletions(-)

diff --git a/include/image.h b/include/image.h
index d933bad..ca612f3 100644
--- a/include/image.h
+++ b/include/image.h
@@ -609,8 +609,14 @@ int fit_image_hash_get_ignore(const void *fit, int 
noffset, int *ignore);
 #endif
 
 int fit_set_timestamp(void *fit, int noffset, time_t timestamp);
-int fit_set_hashes(void *fit);
-int fit_image_set_hashes(void *fit, int image_noffset);
+
+/**
+ * fit_add_verification_data() - Calculate and add hashes to FIT
+ *
+ * @fit:       Fit image to process
+ * @return 0 if ok, <0 for error
+ */
+int fit_add_verification_data(void *fit);
 
 int fit_image_verify(const void *fit, int noffset);
 int fit_all_image_verify(const void *fit);
diff --git a/tools/fit_image.c b/tools/fit_image.c
index 76bbba1..8f51159 100644
--- a/tools/fit_image.c
+++ b/tools/fit_image.c
@@ -125,7 +125,7 @@ static int fit_handle_file (struct mkimage_params *params)
        }
 
        /* set hashes for images in the blob */
-       if (fit_set_hashes (ptr)) {
+       if (fit_add_verification_data(ptr)) {
                fprintf (stderr, "%s Can't add hashes to FIT blob",
                                params->cmdname);
                unlink (tmpfile);
diff --git a/tools/image-host.c b/tools/image-host.c
index d127fc1..4c589af 100644
--- a/tools/image-host.c
+++ b/tools/image-host.c
@@ -34,51 +34,6 @@
 #include <u-boot/md5.h>
 
 /**
- * fit_set_hashes - process FIT component image nodes and calculate hashes
- * @fit: pointer to the FIT format image header
- *
- * fit_set_hashes() adds hash values for all component images in the FIT blob.
- * Hashes are calculated for all component images which have hash subnodes
- * with algorithm property set to one of the supported hash algorithms.
- *
- * returns
- *     0, on success
- *     libfdt error code, on failure
- */
-int fit_set_hashes(void *fit)
-{
-       int images_noffset;
-       int noffset;
-       int ndepth;
-       int ret;
-
-       /* Find images parent node offset */
-       images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
-       if (images_noffset < 0) {
-               printf("Can't find images parent node '%s' (%s)\n",
-                       FIT_IMAGES_PATH, fdt_strerror(images_noffset));
-               return images_noffset;
-       }
-
-       /* Process its subnodes, print out component images details */
-       for (ndepth = 0, noffset = fdt_next_node(fit, images_noffset, &ndepth);
-            (noffset >= 0) && (ndepth > 0);
-            noffset = fdt_next_node(fit, noffset, &ndepth)) {
-               if (ndepth == 1) {
-                       /*
-                        * Direct child node of the images parent node,
-                        * i.e. component image node.
-                        */
-                       ret = fit_image_set_hashes(fit, noffset);
-                       if (ret)
-                               return ret;
-               }
-       }
-
-       return 0;
-}
-
-/**
  * fit_set_hash_value - set hash value in requested has node
  * @fit: pointer to the FIT format image header
  * @noffset: hash node offset
@@ -125,23 +80,16 @@ static int fit_image_process_hash(void *fit, const char 
*image_name,
                int noffset, const void *data, size_t size)
 {
        uint8_t value[FIT_MAX_HASH_LEN];
+       const char *node_name;
        int value_len;
        char *algo;
 
-       /*
-        * Check subnode name, must be equal to "hash".
-        * Multiple hash nodes require unique unit node
-        * names, e.g. hash@1, hash@2, etc.
-        */
-       if (strncmp(fit_get_name(fit, noffset, NULL),
-                               FIT_HASH_NODENAME,
-                               strlen(FIT_HASH_NODENAME)) != 0)
-               return 0;
+       node_name = fit_get_name(fit, noffset, NULL);
 
        if (fit_image_hash_get_algo(fit, noffset, &algo)) {
                printf("Can't get hash algo property for "
                        "'%s' hash node in '%s' image node\n",
-                       fit_get_name(fit, noffset, NULL), image_name);
+                       node_name, image_name);
                return -1;
        }
 
@@ -149,7 +97,7 @@ static int fit_image_process_hash(void *fit, const char 
*image_name,
                                &value_len)) {
                printf("Unsupported hash algorithm (%s) for "
                        "'%s' hash node in '%s' image node\n",
-                       algo, fit_get_name(fit, noffset, NULL), image_name);
+                       algo, node_name, image_name);
                return -1;
        }
 
@@ -157,7 +105,7 @@ static int fit_image_process_hash(void *fit, const char 
*image_name,
                                        value_len)) {
                printf("Can't set hash value for "
                        "'%s' hash node in '%s' image node\n",
-                       fit_get_name(fit, noffset, NULL), image_name);
+                       node_name, image_name);
                return -1;
        }
 
@@ -165,14 +113,13 @@ static int fit_image_process_hash(void *fit, const char 
*image_name,
 }
 
 /**
- * fit_image_set_hashes - calculate/set hashes for given component image node
- * @fit: pointer to the FIT format image header
- * @image_noffset: requested component image node
+ * fit_image_add_verification_data() - calculate/set hash data for image node
  *
- * fit_image_set_hashes() adds hash values for an component image node. All
- * existing hash subnodes are checked, if algorithm property is set to one of
- * the supported hash algorithms, hash value is computed and corresponding
- * hash node property is set, for example:
+ * This adds hash values for a component image node.
+ *
+ * All existing hash subnodes are checked, if algorithm property is set to
+ * one of the supported hash algorithms, hash value is computed and
+ * corresponding hash node property is set, for example:
  *
  * Input component image node structure:
  *
@@ -189,17 +136,19 @@ static int fit_image_process_hash(void *fit, const char 
*image_name,
  *     |- algo = "sha1"
  *     |- value = sha1(data)
  *
- * returns:
- *     0 on sucess
- *    <0 on failure
+ * For signature details, please see doc/uImage.FIT/signature.txt
+ *
+ * @fit:       Pointer to the FIT format image header
+ * @image_noffset: Requested component image node
+ * @return: 0 on success, <0 on failure
  */
-int fit_image_set_hashes(void *fit, int image_noffset)
+int fit_image_add_verification_data(void *fit, int image_noffset)
 {
+       const char *image_name;
        const void *data;
        size_t size;
        int noffset;
        int ndepth;
-       const char *image_name;
 
        /* Get image data and data length */
        if (fit_image_get_data(fit, image_noffset, &data, &size)) {
@@ -210,15 +159,59 @@ int fit_image_set_hashes(void *fit, int image_noffset)
        image_name = fit_get_name(fit, image_noffset, NULL);
 
        /* Process all hash subnodes of the component image node */
-       for (ndepth = 0, noffset = fdt_next_node(fit, image_noffset, &ndepth);
-                       (noffset >= 0) && (ndepth > 0);
-                       noffset = fdt_next_node(fit, noffset, &ndepth)) {
-               if (ndepth == 1) {
-                       /* Direct child node of the component image node */
-                       if (fit_image_process_hash(fit, image_name, noffset,
-                                       data, size))
-                               return -1;
+       for (ndepth = 0,
+                       noffset = fdt_next_subnode(fit, image_noffset, &ndepth);
+                       noffset >= 0;
+                       noffset = fdt_next_subnode(fit, noffset, &ndepth)) {
+               const char *node_name;
+               int ret = 0;
+
+               /*
+                * Check subnode name, must be equal to "hash" or "signature".
+                * Multiple hash nodes require unique unit node
+                * names, e.g. hash@1, hash@2, signature@1, etc.
+                */
+               node_name = fit_get_name(fit, noffset, NULL);
+               if (!strncmp(node_name, FIT_HASH_NODENAME,
+                               strlen(FIT_HASH_NODENAME))) {
+                       ret = fit_image_process_hash(fit, image_name, noffset,
+                                               data, size);
                }
+               if (ret)
+                       return -1;
+       }
+
+       return 0;
+}
+
+int fit_add_verification_data(void *fit)
+{
+       int images_noffset;
+       int noffset;
+       int ndepth;
+       int ret;
+
+       /* Find images parent node offset */
+       images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
+       if (images_noffset < 0) {
+               printf("Can't find images parent node '%s' (%s)\n",
+                       FIT_IMAGES_PATH, fdt_strerror(images_noffset));
+               return images_noffset;
+       }
+
+       /* Process its subnodes, print out component images details */
+       for (ndepth = 0,
+                       noffset = fdt_next_subnode(fit, images_noffset,
+                                                  &ndepth);
+                       noffset >= 0;
+                       noffset = fdt_next_subnode(fit, noffset, &ndepth)) {
+               /*
+                * Direct child node of the images parent node,
+                * i.e. component image node.
+                */
+               ret = fit_image_add_verification_data(fit, noffset);
+               if (ret)
+                       return ret;
        }
 
        return 0;
-- 
1.7.7.3

_______________________________________________
U-Boot mailing list
U-Boot@lists.denx.de
http://lists.denx.de/mailman/listinfo/u-boot

Reply via email to