LXR uboot/board/xilinx/zynq/bootimg.c

   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * Copyright (C) 2018 Xilinx, Inc.
   4 */
   5
   6#include <common.h>
   7#include <log.h>
   8#include <part.h>
   9#include <asm/global_data.h>
  10#include <asm/io.h>
  11#include <asm/arch/hardware.h>
  12#include <asm/arch/sys_proto.h>
  13#include <u-boot/md5.h>
  14#include <zynq_bootimg.h>
  15
  16DECLARE_GLOBAL_DATA_PTR;
  17
  18#define ZYNQ_IMAGE_PHDR_OFFSET          0x09C
  19#define ZYNQ_IMAGE_FSBL_LEN_OFFSET      0x040
  20#define ZYNQ_PART_HDR_CHKSUM_WORD_COUNT 0x0F
  21#define ZYNQ_PART_HDR_WORD_COUNT        0x10
  22#define ZYNQ_MAXIMUM_IMAGE_WORD_LEN     0x40000000
  23#define MD5_CHECKSUM_SIZE       16
  24
  25struct headerarray {
  26        u32 fields[16];
  27};
  28
  29/*
  30 * Check whether the given partition is last partition or not
  31 */
  32static int zynq_islastpartition(struct headerarray *head)
  33{
  34        int index;
  35
  36        debug("%s\n", __func__);
  37        if (head->fields[ZYNQ_PART_HDR_CHKSUM_WORD_COUNT] != 0xFFFFFFFF)
  38                return -1;
  39
  40        for (index = 0; index < ZYNQ_PART_HDR_WORD_COUNT - 1; index++) {
  41                if (head->fields[index] != 0x0)
  42                        return -1;
  43        }
  44
  45        return 0;
  46}
  47
  48/*
  49 * Get the partition count from the partition header
  50 */
  51int zynq_get_part_count(struct partition_hdr *part_hdr_info)
  52{
  53        u32 count;
  54        struct headerarray *hap;
  55
  56        debug("%s\n", __func__);
  57
  58        for (count = 0; count < ZYNQ_MAX_PARTITION_NUMBER; count++) {
  59                hap = (struct headerarray *)&part_hdr_info[count];
  60                if (zynq_islastpartition(hap) != -1)
  61                        break;
  62        }
  63
  64        return count;
  65}
  66
  67/*
  68 * Get the partition info of all the partitions available.
  69 */
  70int zynq_get_partition_info(u32 image_base_addr, u32 *fsbl_len,
  71                            struct partition_hdr *part_hdr)
  72{
  73        u32 parthdroffset;
  74
  75        *fsbl_len = *((u32 *)(image_base_addr + ZYNQ_IMAGE_FSBL_LEN_OFFSET));
  76
  77        parthdroffset = *((u32 *)(image_base_addr + ZYNQ_IMAGE_PHDR_OFFSET));
  78
  79        parthdroffset += image_base_addr;
  80
  81        memcpy(part_hdr, (u32 *)parthdroffset,
  82               (sizeof(struct partition_hdr) * ZYNQ_MAX_PARTITION_NUMBER));
  83
  84        return 0;
  85}
  86
  87/*
  88 * Check whether the partition header is valid or not
  89 */
  90int zynq_validate_hdr(struct partition_hdr *header)
  91{
  92        struct headerarray *hap;
  93        u32 index;
  94        u32 checksum;
  95
  96        debug("%s\n", __func__);
  97
  98        hap = (struct headerarray *)header;
  99
 100        for (index = 0; index < ZYNQ_PART_HDR_WORD_COUNT; index++) {
 101                if (hap->fields[index])
 102                        break;
 103        }
 104        if (index == ZYNQ_PART_HDR_WORD_COUNT)
 105                return -1;
 106
 107        checksum = 0;
 108        for (index = 0; index < ZYNQ_PART_HDR_CHKSUM_WORD_COUNT; index++)
 109                checksum += hap->fields[index];
 110
 111        checksum ^= 0xFFFFFFFF;
 112
 113        if (hap->fields[ZYNQ_PART_HDR_CHKSUM_WORD_COUNT] != checksum) {
 114                printf("Error: Checksum 0x%8.8x != 0x%8.8x\n",
 115                       checksum, hap->fields[ZYNQ_PART_HDR_CHKSUM_WORD_COUNT]);
 116                return -1;
 117        }
 118
 119        if (header->imagewordlen > ZYNQ_MAXIMUM_IMAGE_WORD_LEN) {
 120                printf("INVALID_PARTITION_LENGTH\n");
 121                return -1;
 122        }
 123
 124        return 0;
 125}
 126
 127/*
 128 * Validate the partition by calculationg the md5 checksum for the
 129 * partition and compare with checksum present in checksum offset of
 130 * partition
 131 */
 132int zynq_validate_partition(u32 start_addr, u32 len, u32 chksum_off)
 133{
 134        u8 checksum[MD5_CHECKSUM_SIZE];
 135        u8 calchecksum[MD5_CHECKSUM_SIZE];
 136
 137        memcpy(&checksum[0], (u32 *)chksum_off, MD5_CHECKSUM_SIZE);
 138
 139        md5_wd((u8 *)start_addr, len, &calchecksum[0], 0x10000);
 140
 141        if (!memcmp(checksum, calchecksum, MD5_CHECKSUM_SIZE))
 142                return 0;
 143
 144        printf("Error: Partition DataChecksum\n");
 145        return -1;
 146}
 147