/*
* File:        AESKey.cpp
*
* Copyright (c) Freescale Semiconductor, Inc. All rights reserved.
* See included license file for license details.
*/

#include "AESKey.h"
#include <stdexcept>
#include "smart_ptr.h"
#include "HexValues.h"
#include <ctype.h>

//! The data from the stream is expected to be hex encoded. Each two characters
//! from the stream encode a single result byte. All non-hexadecimal characters
//! are ignored, including newlines. Every two hexadecimal characters form
//! an encoded byte. This is true even if the two characters representing the
//! upper and lower nibbles are separated by whitespace or other characters.
//!
//! \post The stream read head is left pointing just after the last encoded byte.
//!
//! \param stream Input stream to read from.
//! \param bytes Number of encoded bytes to read. This is the number of \e
//!             result bytes, not the count of bytes to read from the stream.
//! \param[out] buffer Pointer to the buffer where decoded data is written.
//!
//! \exception std::runtime_error This exception will be thrown if less
//!             data than required is available from \a stream, or if some other
//!             error occurs while reading from \a stream.
void AESKeyBase::_readFromStream(std::istream & stream, unsigned bytes, uint8_t * buffer)
{
       char temp[2];
       char c;
       char n = 0;

       while (bytes)
       {
               if (stream.get(c).fail())
               {
                       throw std::runtime_error("not enough data in stream");
               }

               if (isHexDigit(c))
               {
                       temp[n++] = c;
                       if (n == 2)
                       {
                               *buffer++ = hexByteToInt(temp);
                               bytes--;
                               n = 0;
                       }
               }
       }
}

//! Key data is written to \a stream as a sequence of hex encoded octets, each two
//! characters long. No spaces or newlines are inserted between the encoded octets
//! or at the end of the sequence.
//!
//! \exception std::runtime_error Thrown if the \a stream reports an error while
//!             writing the key data.
void AESKeyBase::_writeToStream(std::ostream & stream, unsigned bytes, const uint8_t * buffer)
{
       const char hexChars[] = "0123456789ABCDEF";
       while (bytes--)
       {
               uint8_t thisByte = *buffer++;
               char byteString[2];
               byteString[0] = hexChars[(thisByte & 0xf0) >> 4];
               byteString[1] = hexChars[thisByte & 0x0f];
               if (stream.write(byteString, 2).bad())
               {
                       throw std::runtime_error("error while writing to stream");
               }
       }
}