//      enc_rsa.c
//      
//      Copyright 2009 Kim Sung-tae <pchero21@gmail.com>
//      
//      This program is free software; you can redistribute it and/or modify
//      it under the terms of the GNU General Public License as published by
//      the Free Software Foundation; either version 2 of the License, or
//      (at your option) any later version.
//      
//      This program is distributed in the hope that it will be useful,
//      but WITHOUT ANY WARRANTY; without even the implied warranty of
//      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//      GNU General Public License for more details.
//      
//      You should have received a copy of the GNU General Public License
//      along with this program; if not, write to the Free Software
//      Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
//      MA 02110-1301, USA.

#include <stdio.h>
#include <string.h>

#include <openssl/bio.h>
#include <openssl/err.h>
#include <openssl/bn.h>
#include <openssl/rsa.h>
#include <openssl/pem.h>
#include <openssl/rand.h>

#define true 1
#define false 0

// 콜백 함수 정의
static void genrsa_cb(int p, int n, void *arb);

int main(int argc, char** argv)
{
    
    int keyLenInput = 512;    // 키 길이
    char outPublicKeyFile[50];    // 공개키를 저장
    char outPrivateKeyFile[50];    // 개인키를 저장
    
    int isPem = true;    // PEM 형식으로…
    
    BIO *publicOut = NULL;    // 공개키가 저장되는 파일 BIO
    BIO *privateOut = NULL;    // 개인키가 저장되는 파일 BIO
    BIO *bio_stdout = NULL;    // 화면에 출력할 stdout BIO
    
    RSA *rsa = NULL;    // RSA 구조체
    
    printf(“키 길이 입력 :”);    // 키 길이 입력
    scanf(“%d”, &keyLenInput);
    
    printf(“저장할 공개키 파일 이름 입력: “);    // 공개키를 저장할 파일명 입력
    scanf(“%s”, outPublicKeyFile);
    
    printf(“저장할 개인키 파일 이름 입력: “);    // 개인키를 저장할 파일명 입력
    scanf(“%s”, outPrivateKeyFile);
    
    // 표준 화면 출력 BIO 생성
    if((bio_stdout = BIO_new(BIO_s_file())) != NULL) {
        BIO_set_fp(bio_stdout, stdout, BIO_NOCLOSE | BIO_FP_TEXT);
    }
    
     //공개키를 저장할 파일 BIO 생성
     if((publicOut = BIO_new(BIO_s_file())) == NULL) {
         printf(“BIO 생성 에러. %s”, outPublicKeyFile);
         exit(1);
     }
    
     // 개인키를 저장할 파일 BIO 생성
     if((privateOut = BIO_new(BIO_s_file())) == NULL) {
         printf(“BIO 생성 에러. %s”, outPublicKeyFile);
         exit(1);
     }
    
     // 파일 BIO와 해당 공개키 저장할 파일 연결
     if(BIO_write_filename(publicOut, outPublicKeyFile) <= 0) {
         printf(“BIO 생성 에러. %s”, outPublicKeyFile);
         exit(1);
     }
    
     // 파일 BIO와 해당 개인키 저장할 파일 연결
     if(BIO_write_filename(privateOut, outPrivateKeyFile) <= 0) {
         printf(“BIO 생성 에러. %s”, outPrivateKeyFile);
         exit(1);
     }
    
     RAND_status();    // seed 생성, 공급
    
     // 키 생성
     rsa = RSA_generate_key(keyLenInput, RSA_F4, genrsa_cb, NULL);
    
     if(isPem) {
         // PEM 포멧으로 표준 화면 출력 BIO에 공개키 출력
         if(!PEM_write_bio_RSA_PUBKEY(bio_stdout, rsa)) {
             printf(“PEM 파일 생성 에러 %s”, outPrivateKeyFile);
             exit(1);
         }
         printf(“nn”);
        
         // PEM 포멧으로 표준 화면 출력 BIO에 개인키 출력
         if(!PEM_write_bio_RSAPrivateKey(bio_stdout, rsa, NULL, NULL, 0, NULL, NULL)) {
             printf(“PEM 파일 생성 에러 %s”, outPrivateKeyFile);
             exit(1);
         }
        
         // PEM 포멧으로 파일 BIO에 공개키 출력
         if(!PEM_write_bio_RSA_PUBKEY(publicOut, rsa)) {
             printf(“PEM 파일 생성 에러 %s”, outPrivateKeyFile);
             exit(1);
         }
        
         // PEM 포멧으로 파일 BIO에 개인키 출력
         if(!PEM_write_bio_RSAPrivateKey(privateOut, rsa, NULL, NULL, 0, NULL, NULL)) {
             printf(“PEM 파일 생성 에러 %s”, outPrivateKeyFile);
             exit(1);
         }
     }
     else {    // 만약 DEM 포멧으로 키 쌍을 출력 한다면..
        printf(“DEM PUBLIC KEYn”);
        
        // DEM 포멧으로 표준 화면 출력 BIO에 공개키 출력
        if(!i2d_RSA_PUBKEY_bio(bio_stdout, rsa)) {
            printf(“DEM 파일 생성 에러 %s”, outPrivateKeyFile);
            exit(1);
        }
        
        printf(“nn”);
        printf(“DER PRIVATE KEYn”);
        printf(“n”);
        
        // DEM 포멧으로 표준 화면 출력 BIO에 개인키 출력
        if(!i2d_RSAPrivateKey_bio(bio_stdout, rsa)) {
            printf(“DEM 파일 생성 에러 %s”, outPrivateKeyFile);
            exit(1);
        }
        
        // DEM 포멧으로 파일 BIO에 공개키 출력
        if(!i2d_RSA_PUBKEY_bio(publicOut, rsa)) {
            printf(“DEM 파일 생성 에러 %s”, outPrivateKeyFile);
            exit(1);
        }
        
        // DEM 포멧으로 파일 BIO에 개인키 출력
        if(!i2d_RSAPrivateKey_bio(privateOut, rsa)) {
            printf(“DEM 파일 생성 에러 %s”, outPrivateKeyFile);
            exit(1);
        }
    }
    
    // RSA 구조체 메모리에서 삭제
    if(rsa != NULL)
        RSA_free(rsa);
        
    // BIO를 메모리에서 삭제
    if(publicOut != NULL) BIO_free_all(publicOut);
    if(privateOut != NULL) BIO_free_all(privateOut);
    
    
    return 0;
}

// 콜백 함수
static void genrsa_cb(int p, int n, void *arg)
{
    char c = ‘*’;
    
    // prime number e 의 생성 과정을 표시
    if(p == 0) c = ‘.’;
    if(p == 1) c = ‘+’;
    if(p == 2) c = ‘*’;
    if(p == 3) c = ‘n’;
    
    printf(“%c”, c);
}

#include <stdio.h>

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

int main(int argc, char* argv[])
{
        int i;

        // salt bufer length = 8
        unsigned char salt[8];
        // EVP_CIPHER = Cipher structure
        const EVP_CIPHER *cipher = NULL;
        // password pointer. password = “aaaa”
        char *password = “aaaa”;

        // 키와 IV가 저장될 변수를 정의하고 길이는 OpenSSL에서 알아서 정해준다.
        unsigned char key[EVP_MAX_KEY_LENGTH], iv[EVP_MAX_IV_LENGTH];

        int ret = 0;

        // PRNG를 통해 랜덤 수를 만들고 그 값을 Salt에 저장한다. 길이는 8
        ret = RAND_pseudo_bytes(salt, 8);

        // PRNG에서 에러가 발생할 경우 에러 메시지를 출력하고 프로그램을 종료한다.
        if(ret < 0) {
                printf(“Can’t generate random number.n”);
                return 0;
        }

        // 암호화 구조체의 인스턴스를 생성. 여기서는 DES의 ECB 모드의 암호화 구조체 생성
        cipher = EVP_des_ecb();

        // 키와 IV를 생성함. 인자는 암호화 구조체, 다이제스트 구조체, salt 값, 패스워드

        // 카운트는 생성될 키와 IV를 저장할 변수

        // 다이제스트 구조체는 EVP_md5() 함수를 통해 생성. 카운트는 한번
        EVP_BytesToKey(cipher, EVP_md5(), salt, (unsigned char *)password, strlen(password), 1, key, iv);

        // display salt
        for(i = 0; i < sizeof(salt); i++) {
                printf(“%02X”, salt[i]);
        }

        // Display key
        if(cipher->key_len > 0) {
                for(i = 0; i < cipher->key_len; i++) {
                        printf(“%02X”, key[i]);
                }
        }

        // Display IV
        if(cipher->iv_len > 0) {
                for(i = 0; i < cipher->iv_len; i++) {
                        printf(“%02X”, iv[i]);
                }
        }

        return 0;
}