/*
 * FIPS 180-2 SHA-224/256/384/512 implementation
 * Last update: 02/02/2007
 * Issue date:  04/30/2005
 *
 * Copyright (C) 2005, 2007 Olivier Gay <olivier.gay@a3.epfl.ch>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the project nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#ifndef SHA2_H
#define SHA2_H

#define SHA224_DIGEST_SIZE ( 224 / 8)
#define SHA256_DIGEST_SIZE ( 256 / 8)
#define SHA384_DIGEST_SIZE ( 384 / 8)
#define SHA512_DIGEST_SIZE ( 512 / 8)

#define SHA256_BLOCK_SIZE  ( 512 / 8)
#define SHA512_BLOCK_SIZE  (1024 / 8)
#define SHA384_BLOCK_SIZE  SHA512_BLOCK_SIZE
#define SHA224_BLOCK_SIZE  SHA256_BLOCK_SIZE

#ifndef SHA2_TYPES
#define SHA2_TYPES
typedef unsigned char uint8;
typedef unsigned int  uint32;
typedef unsigned long long uint64;
#endif

#ifdef __cplusplus
extern "C" {
#endif

    typedef struct {
        unsigned int tot_len;
        unsigned int len;
        unsigned char block[2 * SHA256_BLOCK_SIZE];
        uint32 h[8];
    } sha256_ctx;

    typedef struct {
        unsigned int tot_len;
        unsigned int len;
        unsigned char block[2 * SHA512_BLOCK_SIZE];
        uint64 h[8];
    } sha512_ctx;

    typedef sha512_ctx sha384_ctx;
    typedef sha256_ctx sha224_ctx;

    void sha224_init(sha224_ctx* ctx);
    void sha224_update(sha224_ctx* ctx, const unsigned char* message,
        unsigned int len);
    void sha224_final(sha224_ctx* ctx, unsigned char* digest);
    void sha224(const unsigned char* message, unsigned int len,
        unsigned char* digest);

    void sha256_init(sha256_ctx* ctx);
    void sha256_update(sha256_ctx* ctx, const unsigned char* message,
        unsigned int len);
    void sha256_final(sha256_ctx* ctx, unsigned char* digest);
    void sha256(const unsigned char* message, unsigned int len,
        unsigned char* digest);

    void sha384_init(sha384_ctx* ctx);
    void sha384_update(sha384_ctx* ctx, const unsigned char* message,
        unsigned int len);
    void sha384_final(sha384_ctx* ctx, unsigned char* digest);
    void sha384(const unsigned char* message, unsigned int len,
        unsigned char* digest);

    void sha512_init(sha512_ctx* ctx);
    void sha512_update(sha512_ctx* ctx, const unsigned char* message,
        unsigned int len);
    void sha512_final(sha512_ctx* ctx, unsigned char* digest);
    void sha512(const unsigned char* message, unsigned int len,
        unsigned char* digest);

#ifdef __cplusplus
}
#endif
#include <cpps/cpps.h>
#include <string>

class SHA224
{
public:
    SHA224() {
        sha224_init(&ctx);
    }
public:
    void        update(std::string input) {
		if (input.empty()) return;

        sha224_update(&ctx, (const unsigned char*)input.c_str(), (unsigned int)input.size());
    }
    std::string digest(){
        std::string ret;
		unsigned char digest[SHA224_DIGEST_SIZE];
        sha224_final(&ctx, digest);
        ret.append((char *)digest, SHA224_DIGEST_SIZE);
        return ret;
    }
    std::string hex_digest()
    {
		unsigned char digest[SHA224_DIGEST_SIZE];
		sha224_final(&ctx, digest);

		char s[SHA224_DIGEST_SIZE*2 + 1];
		::memset(s, 0, SHA224_DIGEST_SIZE * 2+1);

		for (int i = 0; i < SHA224_DIGEST_SIZE; i++)
			sprintf(s + i * 2, "%02x", digest[i]);

		s[SHA224_DIGEST_SIZE * 2] = '\0';

        return s;
    }
    void        update_file(FILE* file)
    {
        if (!file) return;

		unsigned char buffer[1024];
		size_t len;

		while ((len = fread(buffer, 1, 1024, file)) != 0)
            sha224_update(&ctx,buffer, (unsigned int)len);

		fclose(file);
    }

public:
    sha224_ctx ctx;
};

class SHA256
{
public:
	SHA256() {
		sha256_init(&ctx);
	}
public:
	void        update(std::string input) {
		if (input.empty()) return;

		sha256_update(&ctx, (const unsigned char*)input.c_str(), (unsigned int)input.size());
	}
	std::string digest() {
		std::string ret;
		unsigned char digest[SHA256_DIGEST_SIZE];
		sha256_final(&ctx, digest);
		ret.append((char*)digest, SHA256_DIGEST_SIZE);
		return ret;
	}
	std::string hex_digest()
	{
		unsigned char digest[SHA256_DIGEST_SIZE];
		sha256_final(&ctx, digest);

		char s[SHA256_DIGEST_SIZE * 2 + 1];
		::memset(s, 0, SHA256_DIGEST_SIZE * 2 + 1);

		for (int i = 0; i < SHA256_DIGEST_SIZE; i++)
			sprintf(s + i * 2, "%02x", digest[i]);

		s[SHA256_DIGEST_SIZE * 2] = '\0';

		return s;
	}
	void        update_file(FILE* file)
	{
		if (!file) return;

		unsigned char buffer[1024];
		size_t len;

		while ((len = fread(buffer, 1, 1024, file)) != 0)
			sha256_update(&ctx, buffer, (unsigned int)len);

		fclose(file);
	}

public:
	sha256_ctx ctx;
};

class SHA384
{
public:
	SHA384() {
		sha384_init(&ctx);
	}
public:
	void        update(std::string input) {
		if (input.empty()) return;

		sha384_update(&ctx, (const unsigned char*)input.c_str(), (unsigned int)input.size());
	}
	std::string digest() {
		std::string ret;
		unsigned char digest[SHA384_DIGEST_SIZE];
		sha384_final(&ctx, digest);
		ret.append((char*)digest, SHA384_DIGEST_SIZE);
		return ret;
	}
	std::string hex_digest()
	{
		unsigned char digest[SHA384_DIGEST_SIZE];
		sha384_final(&ctx, digest);

		char s[SHA384_DIGEST_SIZE * 2 + 1];
		::memset(s, 0, SHA384_DIGEST_SIZE * 2 + 1);

		for (int i = 0; i < SHA384_DIGEST_SIZE; i++)
			sprintf(s + i * 2, "%02x", digest[i]);

		s[SHA384_DIGEST_SIZE * 2] = '\0';

		return s;
	}
	void        update_file(FILE* file)
	{
		if (!file) return;

		unsigned char buffer[1024];
		size_t len;

		while ((len = fread(buffer, 1, 1024, file)) != 0)
			sha384_update(&ctx, buffer, (unsigned int)len);

		fclose(file);
	}

public:
	sha384_ctx ctx;
};

class SHA512
{
public:
	SHA512() {
		sha512_init(&ctx);
	}
public:
	void        update(std::string input) {
		if (input.empty()) return;
		sha512_update(&ctx, (const unsigned char*)input.c_str(), (unsigned int)input.size());
	}
	std::string digest() {
		std::string ret;
		unsigned char digest[SHA512_DIGEST_SIZE];
		sha512_final(&ctx, digest);
		ret.append((char*)digest, SHA512_DIGEST_SIZE);
		return ret;
	}

	std::string hex_digest()
	{
		unsigned char digest[SHA512_DIGEST_SIZE];
		sha512_final(&ctx, digest);

		char s[SHA512_DIGEST_SIZE * 2 + 1];
		::memset(s, 0, SHA512_DIGEST_SIZE * 2 + 1);

		for (int i = 0; i < SHA512_DIGEST_SIZE; i++)
			sprintf(s + i * 2, "%02x", digest[i]);

		s[SHA512_DIGEST_SIZE * 2] = '\0';

		return s;
	}
	void        update_file(FILE* file)
	{
		if (!file) return;

		unsigned char buffer[1024];
		size_t len;

		while ((len = fread(buffer, 1, 1024, file)) != 0)
			sha512_update(&ctx, buffer, (unsigned int)len);

		fclose(file);
	}

public:
	sha512_ctx ctx;
};

#endif /* !SHA2_H */