RijndaelCBCMAC.h

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

#include "AESKey.h"
#include <string.h>

class RijndaelCBCMAC
{
public:
    enum
    {
        BLOCK_SIZE = 16 
    };

    typedef uint8_t block_t[BLOCK_SIZE];

public:
    RijndaelCBCMAC() {}
    RijndaelCBCMAC(const AESKey<128> &key, const uint8_t *iv = 0);

    void update(const uint8_t *data, unsigned length);

    void finalize();

    const block_t &getMAC() const { return m_mac; }
    RijndaelCBCMAC &operator=(const RijndaelCBCMAC &other)
    {
        m_key = other.m_key;
        memcpy(m_mac, other.m_mac, sizeof(m_mac));
        return *this;
    }

protected:
    AESKey<128> m_key; 
    block_t m_mac;     

    void updateOneBlock(const uint8_t *data);
};

#endif // _RijndaelCBCMAC_h_