/*
 *
 *    Copyright (c) 2013-2017 Nest Labs, Inc.
 *    All rights reserved.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

/**
 *    @file
 *      This file defines a template object for doing counter (CTR) mode
 *      block ciphers and specialized objects for CTR mode AES-128 and
 *      AES-256.
 *
 */

#include <Weave/Support/NLDLLUtil.h>

#include "AESBlockCipher.h"

#ifndef CTRMODE_H_
#define CTRMODE_H_

namespace nl {
namespace Weave {
namespace Crypto {

template <class BlockCipher>
class NL_DLL_EXPORT CTRMode
{
public:
    enum
    {
        kKeyLength      = BlockCipher::kKeyLength,
        kCounterLength  = BlockCipher::kBlockLength
    };

    CTRMode(void);
    ~CTRMode(void);

    // The CTR-mode encryption counter. The algorithm assumes the counter is in big-endian
    // form and increments it once for each encrypted block.
    uint8_t Counter[kCounterLength];

    void SetKey(const uint8_t *key);
    void SetCounter(const uint8_t *counter);
    void SetWeaveMessageCounter(uint64_t sendingNodeId, uint32_t msgId);
    void EncryptData(const uint8_t *inData, uint16_t dataLen, uint8_t *outData);

    void Reset(void);

private:
    BlockCipher mBlockCipher;
    uint32_t mMsgIndex;
    uint8_t mEncryptedCounter[kCounterLength];
};

typedef CTRMode<Platform::Security::AES128BlockCipherEnc> AES128CTRMode;
typedef CTRMode<Platform::Security::AES256BlockCipherEnc> AES256CTRMode;

} /* namespace Crypto */
} /* namespace Weave */
} /* namespace nl */

#endif /* CTRMODE_H_ */