EncoreBootImageReader.h

/*
 * File:    EncoreBootImageReader.h
 *
 * Copyright (c) Freescale Semiconductor, Inc. All rights reserved.
 * See included license file for license details.
 */
#if !defined(_EncoreBootImageReader_h_)
#define _EncoreBootImageReader_h_

#include "EncoreBootImage.h"

namespace elftosb
{
class EncoreBootImageReader
{
public:
    class read_error : public std::runtime_error
    {
    public:
        read_error(const std::string &msg)
            : std::runtime_error(msg)
        {
        }
    };

    typedef std::vector<EncoreBootImage::section_header_t> section_array_t;

    typedef std::vector<EncoreBootImage::boot_command_t> boot_tag_array_t;

public:
    EncoreBootImageReader(std::istream &stream)
        : m_stream(stream)
    {
    }

    virtual ~EncoreBootImageReader() {}

    inline void setKey(const AESKey<128> &key) { m_dek = key; }
    inline const AESKey<128> &getKey() const { return m_dek; }


    void readImageHeader();

    void computeHeaderDigest(sha1_digest_t &digest);

    void readImageDigest();

    void computeImageDigest(sha1_digest_t &digest);

    void readSectionTable();

    bool readKeyDictionary(const AESKey<128> &kek);

    void readBootTags();

    EncoreBootImage::Section *readSection(unsigned index);


    inline bool isEncrypted() const { return m_header.m_keyCount > 0; }
    const EncoreBootImage::boot_image_header_t &getHeader() const { return m_header; }
    const sha1_digest_t &getDigest() const { return m_digest; }
    inline const section_array_t &getSections() const { return m_sections; }
    inline const boot_tag_array_t &getBootTags() const { return m_bootTags; }

protected:
    std::istream &m_stream;                        
    AESKey<128> m_dek;                             
    EncoreBootImage::boot_image_header_t m_header; 
    sha1_digest_t m_digest;                        
    section_array_t m_sections;                    
    boot_tag_array_t m_bootTags;                   

protected:
    uint8_t calculateCommandChecksum(EncoreBootImage::boot_command_t &header);
};

}; // namespace elftosb

#endif // _EncoreBootImageReader_h_