common/image.c

   1 /*
   2  * (C) Copyright 2008 Semihalf
   3  *
   4  * (C) Copyright 2000-2006
   5  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
   6  *
   7  * See file CREDITS for list of people who contributed to this
   8  * project.
   9  *
  10  * This program is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU General Public License as
  12  * published by the Free Software Foundation; either version 2 of
  13  * the License, or (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
  23  * MA 02111-1307 USA
  24  */
  25 #ifndef USE_HOSTCC
  26 # include <common.h>
  27 # include <watchdog.h>
  28 #else
  29 # include "mkimage.h"
  30 #endif
  31
  32 #include <image.h>
  33
  34 unsigned long crc32 (unsigned long, const unsigned char *, unsigned int);
  35
  36 int image_check_hcrc (image_header_t *hdr)
  37 {
  38         ulong hcrc;
  39         ulong len = image_get_header_size ();
  40         image_header_t header;
  41
  42         /* Copy header so we can blank CRC field for re-calculation */
  43         memmove (&header, (char *)hdr, image_get_header_size ());
  44         image_set_hcrc (&header, 0);
  45
  46         hcrc = crc32 (0, (unsigned char *)&header, len);
  47
  48         return (hcrc == image_get_hcrc (hdr));
  49 }
  50
  51 int image_check_dcrc (image_header_t *hdr)
  52 {
  53         ulong data = image_get_data (hdr);
  54         ulong len = image_get_data_size (hdr);
  55         ulong dcrc = crc32 (0, (unsigned char *)data, len);
  56
  57         return (dcrc == image_get_dcrc (hdr));
  58 }
  59
  60 #ifndef USE_HOSTCC
  61 int image_check_dcrc_wd (image_header_t *hdr, ulong chunksz)
  62 {
  63         ulong dcrc = 0;
  64         ulong len = image_get_data_size (hdr);
  65         ulong data = image_get_data (hdr);
  66
  67 #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
  68         ulong cdata = data;
  69         ulong edata = cdata + len;
  70
  71         while (cdata < edata) {
  72                 ulong chunk = edata - cdata;
  73
  74                 if (chunk > chunksz)
  75                         chunk = chunksz;
  76                 dcrc = crc32 (dcrc, (unsigned char *)cdata, chunk);
  77                 cdata += chunk;
  78
  79                 WATCHDOG_RESET ();
  80         }
  81 #else
  82         dcrc = crc32 (0, (unsigned char *)data, len);
  83 #endif
  84
  85         return (dcrc == image_get_dcrc (hdr));
  86 }
  87
  88 int getenv_verify (void)
  89 {
  90         char *s = getenv ("verify");
  91         return (s && (*s == 'n')) ? 0 : 1;
  92 }
  93
  94 void memmove_wd (void *to, void *from, size_t len, ulong chunksz)
  95 {
  96 #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
  97         while (len > 0) {
  98                 size_t tail = (len > chunksz) ? chunksz : len;
  99                 WATCHDOG_RESET ();
 100                 memmove (to, from, tail);
 101                 to += tail;
 102                 from += tail;
 103                 len -= tail;
 104         }
 105 #else   /* !(CONFIG_HW_WATCHDOG || CONFIG_WATCHDOG) */
 106         memmove (to, from, len);
 107 #endif  /* CONFIG_HW_WATCHDOG || CONFIG_WATCHDOG */
 108 }
 109 #endif /* USE_HOSTCC */
 110
 111 /**
 112  * image_multi_count - get component (sub-image) count
 113  * @hdr: pointer to the header of the multi component image
 114  *
 115  * image_multi_count() returns number of components in a multi
 116  * component image.
 117  *
 118  * Note: no checking of the image type is done, caller must pass
 119  * a valid multi component image.
 120  *
 121  * returns:
 122  *     number of components
 123  */
 124 ulong image_multi_count (image_header_t *hdr)
 125 {
 126         ulong i, count = 0;
 127         ulong *size;
 128
 129         /* get start of the image payload, which in case of multi
 130          * component images that points to a table of component sizes */
 131         size = (ulong *)image_get_data (hdr);
 132
 133         /* count non empty slots */
 134         for (i = 0; size[i]; ++i)
 135                 count++;
 136
 137         return count;
 138 }
 139
 140 /**
 141  * image_multi_getimg - get component data address and size
 142  * @hdr: pointer to the header of the multi component image
 143  * @idx: index of the requested component
 144  * @data: pointer to a ulong variable, will hold component data address
 145  * @len: pointer to a ulong variable, will hold component size
 146  *
 147  * image_multi_getimg() returns size and data address for the requested
 148  * component in a multi component image.
 149  *
 150  * Note: no checking of the image type is done, caller must pass
 151  * a valid multi component image.
 152  *
 153  * returns:
 154  *     data address and size of the component, if idx is valid
 155  *     0 in data and len, if idx is out of range
 156  */
 157 void image_multi_getimg (image_header_t *hdr, ulong idx,
 158                         ulong *data, ulong *len)
 159 {
 160         int i;
 161         ulong *size;
 162         ulong offset, tail, count, img_data;
 163
 164         /* get number of component */
 165         count = image_multi_count (hdr);
 166
 167         /* get start of the image payload, which in case of multi
 168          * component images that points to a table of component sizes */
 169         size = (ulong *)image_get_data (hdr);
 170
 171         /* get address of the proper component data start, which means
 172          * skipping sizes table (add 1 for last, null entry) */
 173         img_data = image_get_data (hdr) + (count + 1) * sizeof (ulong);
 174
 175         if (idx < count) {
 176                 *len = size[idx];
 177                 offset = 0;
 178                 tail = 0;
 179
 180                 /* go over all indices preceding requested component idx */
 181                 for (i = 0; i < idx; i++) {
 182                         /* add up i-th component size */
 183                         offset += size[i];
 184
 185                         /* add up alignment for i-th component */
 186                         tail += (4 - size[i] % 4);
 187                 }
 188
 189                 /* calculate idx-th component data address */
 190                 *data = img_data + offset + tail;
 191         } else {
 192                 *len = 0;
 193                 *data = 0;
 194         }
 195 }