veridlg.cpp

该程序能实现将数字签名嵌入到BMP图片中形成数字印章

// VeriDlg.cpp : implementation file
//

#include "stdafx.h"
#include "PKI.h"
#include "PKIDlg.h"
#include "VeriDlg.h"
#include "SetDlg.h"
#include <stdio.h>
#define _WIN32_WINNT 0x0400
#include <windows.h>
#include <wincrypt.h>

#define ENCRYPT_BLOCK_SIZE 8 

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CVeriDlg dialog


CVeriDlg::CVeriDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CVeriDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CVeriDlg)
	m_file_veri = _T("");
	m_pkey_veri = _T("");
	m_signed_veri = _T("");
	m_state_veri = _T("");
	//}}AFX_DATA_INIT

	m_alg_id_veri=CALG_RC4;
	m_hash_veri=CALG_MD5;
	m_prov_veri=PROV_RSA_FULL;
}


void CVeriDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CVeriDlg)
	DDX_Text(pDX, IDC_FILE_VERI_EDIT, m_file_veri);
	DDX_Text(pDX, IDC_PKEY_VERI_EDIT, m_pkey_veri);
	DDX_Text(pDX, IDC_SIGNED_VERI_EDIT, m_signed_veri);
	DDX_Text(pDX, IDC_STATE_VERY_STATIC, m_state_veri);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CVeriDlg, CDialog)
	//{{AFX_MSG_MAP(CVeriDlg)
	ON_BN_CLICKED(IDC_VERI_BUTTON, OnVeriButton)
	ON_BN_CLICKED(IDC_PUB_KEY_VERI_BUTTON, OnPubKeyVeriButton)
	ON_BN_CLICKED(IDC_SIGN_FILE_VERI_BUTTON, OnSignFileVeriButton)
	ON_BN_CLICKED(IDC_FILE_VERI_BUTTON, OnFileVeriButton)
	ON_BN_CLICKED(IDC_YUANWEN_FILE_BUTTON, OnYuanwenFileButton)
//	ON_BN_CLICKED(IDC_LET_BUTTON, OnLetButton)
	ON_BN_CLICKED(IDC_SET_VERI_BUTTON, OnSetVeriButton)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CVeriDlg message handlers

CString MyHandleError2(char *s)
{
	CString str;
	str.Format("程序运行时发生错误. \n错误代码： %x.\n", GetLastError());
	str+=s;
	AfxMessageBox(str);
	return str;
}

void CVeriDlg::OnVeriButton() 
{
	// TODO: Add your control notification handler code her

	
	HCRYPTPROV hProv;
	HCRYPTKEY hPubKey;
	BYTE *pbKeyBlob;
	DWORD dwBlobLen;
	HCRYPTHASH hHash;
	BYTE *pbSignature;			//数字签名
	DWORD dwSigLen;
	LPTSTR szDescription = "";

	UpdateData(TRUE);
	m_state_veri="";
	//--------------------------------------------------------------------
	// 获得CSP句柄,密钥容器名为登陆用户名
	if(CryptAcquireContext(
		&hProv, 
		NULL, 
		NULL, 
		PROV_RSA_FULL, 
		0))
	{
		m_state_veri+="已获取CSP,秘钥生成算法："+GetProvType(m_prov_veri)+"\n";
	}
	else
		//密钥容器不存在创建之
	{
		if(CryptAcquireContext(
			&hProv, 
			NULL, 
			NULL, 
			PROV_RSA_FULL, 
			CRYPT_NEWKEYSET)) 
			m_state_veri+="已创建一个新的密钥容器秘钥生成算法："+GetProvType(m_prov_veri)+"\n";
		else
		{	m_state_veri+=MyHandleError2("在获取密钥容器时发生错误，退出\n");UpdateData(FALSE);return;}
	}




	CFile signdatafile,yuanwenfile,pubkeyfile;
	if(m_pkey_veri==""||!pubkeyfile.Open(m_pkey_veri,CFile::modeReadWrite))
	{
		MessageBox("请选择正确的公钥文件！");return;
	}
	if(m_file_veri==""||!yuanwenfile.Open(m_file_veri,CFile::modeReadWrite))
	{
		MessageBox("请选择正确的原文件！");return;
	}	
	if(m_signed_veri==""||!signdatafile.Open(m_signed_veri,CFile::modeReadWrite))
	{
		MessageBox("请选择正确的签名文件！");return;
	}


	dwBlobLen=pubkeyfile.GetLength();
	
	pbKeyBlob=(BYTE *)malloc(dwBlobLen);
	pubkeyfile.Read(pbKeyBlob,dwBlobLen);


	if(CryptImportKey(
		hProv,
		pbKeyBlob,
		dwBlobLen,
		0,
		0,
		&hPubKey))
		m_state_veri+="公钥已经成功导入！\n\n";
	else
	{
		m_state_veri+=MyHandleError2("公钥导出出错.\n");UpdateData(FALSE);return;
	}

	//--------------------------------------------------------------------
	// 创建哈希对象
	if(CryptCreateHash(
		hProv, 
		m_hash_veri,//CALG_MD5, 
		0, 
		0, 
		&hHash)) 
		m_state_veri+="已经获取hash对象,hash算法"+GetHashType(m_hash_veri)+"\n\n";
	else
	{
		m_state_veri+=MyHandleError2("创建hash对象时出错，退出");UpdateData(FALSE);return;
	}

	//--------------------------------------------------------------------
	// 跟生成时一样对数据进行hash运算
	BYTE *pbBuffer;

	pbBuffer=(BYTE *)malloc(yuanwenfile.GetLength());
	yuanwenfile.Read(pbBuffer,yuanwenfile.GetLength());

	

	DWORD dwBufferLen = yuanwenfile.GetLength();

	
	pbSignature=(BYTE *)malloc(signdatafile.GetLength());
	signdatafile.Read(pbSignature,signdatafile.GetLength());

	dwSigLen = signdatafile.GetLength();
	if(CryptHashData(
		hHash, 
		pbBuffer, 
		dwBufferLen, 
		0)) 
		m_state_veri+="对数据hash运算成功!\n\n";
	else
	{
		m_state_veri+=MyHandleError2("对数据进行hash运算出错，退出");UpdateData(FALSE);return;
	}

	//--------------------------------------------------------------------
	// 验证数字签名
	if(CryptVerifySignature(
		hHash, 
		pbSignature,	//数字签名数据
		dwSigLen, 
		hPubKey,		//签名者的公钥
		szDescription, 
		0)) 
	{
		MessageBox("恭喜：是正确的数字签名！");
		m_state_veri+="恭喜：是正确的数字签名！";
	}
	else
	{
		MessageBox("错误：签名是错误的\n");
		m_state_veri+="错误：签名是错误的\n请检查参数是否设置正确,若正确则请联系发送方";
	}


	UpdateData(FALSE);
	//--------------------------------------------------------------------
	// Free memory to be used to store signature.
	if(pbSignature)
		free(pbSignature);

	if(pbKeyBlob)
		free(pbKeyBlob);
	
	//--------------------------------------------------------------------
	// Destroy the hash object.
	if(hHash) 
		CryptDestroyHash(hHash);
	
	//--------------------------------------------------------------------
	// Release the provider handle.
	if(hProv) 
		CryptReleaseContext(hProv, 0);
	signdatafile.Close(),yuanwenfile.Close(),pubkeyfile.Close();
}

void CVeriDlg::OnPubKeyVeriButton() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg(TRUE,NULL,".\\公钥",OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,NULL,NULL);
	if(dlg.DoModal()==IDOK)
	{
		m_pkey_veri=dlg.GetPathName();
	}
	else m_pkey_veri="";
	UpdateData(FALSE);
}

void CVeriDlg::OnSignFileVeriButton() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg(TRUE,NULL,".\\签名后的文件",OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,NULL,NULL);
	if(dlg.DoModal()==IDOK)
	{
		m_signed_veri=dlg.GetPathName();
	}
	else m_signed_veri="";
	UpdateData(FALSE);
}

void CVeriDlg::OnFileVeriButton() 
{
	// TODO: Add your control notification handler code here
	
}

void CVeriDlg::OnYuanwenFileButton() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg(TRUE,NULL,".\\原文.txt",OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,NULL,NULL);
	if(dlg.DoModal()==IDOK)
	{
		m_file_veri=dlg.GetPathName();
	}
	else m_file_veri="";
	UpdateData(FALSE);
}

void CVeriDlg::OnSetVeriButton() 
{
	// TODO: Add your control notification handler code here
	CSetDlg dlg;

	GetProvType(m_prov_veri);
	dlg.m_prov=m_prov_type;

	GetHashType(m_hash_veri);
	dlg.m_hash=m_hash_type;

	GetSessionType(m_alg_id_veri);
	dlg.m_alg_id=m_session_type;

	if(IDOK==dlg.DoModal())
	{
		switch(dlg.m_alg_id)
		{
		case 0:
			m_alg_id_veri=CALG_RC2;break;
		case 1:
			m_alg_id_veri=CALG_RC4;break;
		default:m_alg_id_veri=CALG_RC2;break;
		}

		switch(dlg.m_hash)
		{
		case 0:
			m_hash_veri=CALG_HMAC;break;
		case 1:
			m_hash_veri=CALG_MAC;break;
		case 2:
			m_hash_veri=CALG_MD2;break;
		case 3:
			m_hash_veri=CALG_MD5;break;
		case 4:
			m_hash_veri=CALG_SHA;break;
		default:m_hash_veri=CALG_MD5;break;
		}

		switch(dlg.m_prov)
		{
		case 0:
			m_prov_veri=PROV_RSA_FULL;break;
		case 1:
			m_prov_veri=PROV_RSA_SIG;break;
		case 2:
			m_prov_veri=PROV_DSS;break;
		case 3:
			m_prov_veri=PROV_DSS_DH;break;
		case 4:
			m_prov_veri=PROV_FORTEZZA;break;
		case 5:
			m_prov_veri=PROV_MS_EXCHANGE;break;
		case 6:
			m_prov_veri=PROV_RSA_SCHANNEL;break;
		case 7:
			m_prov_veri=PROV_SSL;break;
		default:
			m_prov_veri=PROV_RSA_FULL;break;
		}
	}
}


CString CVeriDlg::GetHashType(unsigned int hash)
{
		CString str;
		switch(hash)
		{
		case CALG_HMAC:
			str="HMAC";m_hash_type=0;break;
		case CALG_MAC:
			str="MAC";m_hash_type=1;break;
		case CALG_MD2:
			str="MD2";m_hash_type=2;break;
		case CALG_MD5:
			str="MD5";m_hash_type=3;break;
		case CALG_SHA:
			str="SHA";m_hash_type=4;break;
		default:str="MD5";m_hash_type=3;break;
		}
		return str;
}

CString CVeriDlg::GetSessionType(unsigned int session)
{
		CString str;
		switch(session)
		{
		case CALG_RC2:
			str="RC2";m_session_type=0;break;
		case CALG_RC4:
			str="RC4";m_session_type=1;break;
		default:str="RC4";m_session_type=1;break;
		}
		return str;
}

CString CVeriDlg::GetProvType(unsigned int prov)
{

		CString str;
		switch(prov)
		{
		case PROV_RSA_FULL:
			m_prov_type=0;str="RSA_FULL";break;
		case PROV_RSA_SIG:
			m_prov_type=1;str="RSA_SIG";break;
		case PROV_DSS:
			m_prov_type=2;str="DSS";break;
		case PROV_DSS_DH:
			m_prov_type=3;str="DSS_DH";break;
		case PROV_FORTEZZA:
			m_prov_type=4;str="FORTEZZA";break;
		case PROV_MS_EXCHANGE:
			m_prov_type=5;str="MS_EXCHANGE";break;
		case PROV_RSA_SCHANNEL:
			m_prov_type=6;str="RSA_SCHANNEL";break;
		case PROV_SSL:
			m_prov_type=7;str="SSL";break;
		default:
			m_prov_type=0;str="RSA_FULL";break;
		}
		return str;
}

void CVeriDlg::OnCancel() 
{
	// TODO: Add extra cleanup here
	CPKIDlg dlg;
	dlg.DoModal();
	CDialog::OnCancel();
}