#include <stdio.h>
#include <openssl/evp.h>
#include <openssl/sha.h>
#include <openssl/md5.h>
#include <openssl/rand.h>
#include <string.h>

void 哈希_sha256(const char *信息, char *输出);
void 哈希_md5(const char *信息, char *输出);
void 哈希(const char *算法, const char *信息, char *输出);
void 加密_解密(const unsigned char *明文, unsigned char *密文, unsigned char *解密文);

int main()
{
    const char *信息 = "Hello, World!";

    char sha256输出[SHA256_DIGEST_LENGTH * 2 + 1];  // 创建一个字符串来存储SHA-256哈希值
    char md5输出[MD5_DIGEST_LENGTH * 2 + 1];  // 创建一个字符串来存储MD5哈希值

    //因为是已经被弃用了的方法，所以这边放弃使用
    //哈希_sha256(信息, sha256输出);  // 调用SHA-256哈希函数
    //哈希_md5(信息, md5输出);  // 调用MD5哈希函数

    //改用推荐的EVP方法
    哈希("SHA256", 信息, sha256输出);
    哈希("MD5", 信息, md5输出);

    printf("SHA-256 Hash: %s\n", sha256输出);
    printf("MD5 Hash: %s\n", md5输出);

    char 用户输入[256];
    printf("输入你的密码 (最大 255 字符):");
    scanf("%255s", 用户输入); 
    char 用户输入sha256输出[SHA256_DIGEST_LENGTH * 2 + 1];
    char 用户输入md5输出[MD5_DIGEST_LENGTH * 2 + 1];
    哈希("SHA256", 用户输入, 用户输入sha256输出);
    哈希("MD5", 用户输入, 用户输入md5输出);
    printf("密码 SHA-256 Hash: %s\n", 用户输入sha256输出);
    printf("密码 MD5 Hash: %s\n", 用户输入md5输出);

    const unsigned char 明文[] = "Hello, World!";
    unsigned char 密文[128];
    unsigned char 解密文[128];

    加密_解密(明文, 密文, 解密文);

    // 输出结果
    printf("原始消息: %s\n", 明文);
    printf("加密消息: ");
    for (int i = 0; i < strlen((char *)密文); ++i) {
        printf("%02x", 密文[i]);
    }
    printf("\n");
    printf("解密消息: %s\n", 解密文);

    return 0;
}

void 哈希(const char *算法, const char *信息, char *输出) {
    EVP_MD_CTX *mdctx;
    const EVP_MD *md;
    unsigned char md值[EVP_MAX_MD_SIZE];
    unsigned int md长度, i;

    md = EVP_get_digestbyname(算法);
    if (md == NULL) {
        printf("未知的哈希算法: %s\n", 算法);
        return;
    }

    mdctx = EVP_MD_CTX_new();
    EVP_DigestInit_ex(mdctx, md, NULL);
    EVP_DigestUpdate(mdctx, 信息, strlen(信息));
    EVP_DigestFinal_ex(mdctx, md值, &md长度);
    EVP_MD_CTX_free(mdctx);

    for(i = 0; i < md长度; i++)
        sprintf(输出 + (i * 2), "%02x", md值[i]);
}

void 加密_解密(const unsigned char *明文, unsigned char *密文, unsigned char *解密文) {
    // 生成随机密钥和IV (初始化向量)
    unsigned char 密钥[EVP_MAX_KEY_LENGTH];
    unsigned char iv[EVP_MAX_IV_LENGTH];
    if (!RAND_bytes(密钥, EVP_MAX_KEY_LENGTH) || !RAND_bytes(iv, EVP_MAX_IV_LENGTH)) {
        fprintf(stderr, "生成密钥或IV失败。\n");
        return;
    }

    // 创建并初始化加密上下文
    EVP_CIPHER_CTX *ctx = EVP_CIPHER_CTX_new();
    if (!EVP_EncryptInit_ex(ctx, EVP_aes_256_cbc(), NULL, 密钥, iv)) {
        fprintf(stderr, "初始化加密上下文失败。\n");
        return;
    }

    // 加密明文
    int len, 密文长度;
    if (!EVP_EncryptUpdate(ctx, 密文, &len, 明文, strlen((char *)明文))) {
        fprintf(stderr, "加密失败。\n");
        return;
    }
    密文长度 = len;
    if (!EVP_EncryptFinal_ex(ctx, 密文 + len, &len)) {
        fprintf(stderr, "完成加密失败。\n");
        return;
    }
    密文长度 += len;
    EVP_CIPHER_CTX_free(ctx);  // 释放加密上下文

    // 创建并初始化解密上下文
    ctx = EVP_CIPHER_CTX_new();
    if (!EVP_DecryptInit_ex(ctx, EVP_aes_256_cbc(), NULL, 密钥, iv)) {
        fprintf(stderr, "初始化解密上下文失败。\n");
        return;
    }

    // 解密密文
    int 解密文长度;
    if (!EVP_DecryptUpdate(ctx, 解密文, &len, 密文, 密文长度)) {
        fprintf(stderr, "解密失败。\n");
        return;
    }
    解密文长度 = len;
    if (!EVP_DecryptFinal_ex(ctx, 解密文 + len, &len)) {
        fprintf(stderr, "完成解密失败。\n");
        return;
    }
    解密文长度 += len;
    解密文[解密文长度] = '\0';  // 在解密字符串末尾添加空字符
    EVP_CIPHER_CTX_free(ctx);  // 释放解密上下文
}