您的位置:首页 > 编程语言 > C语言/C++

VC++使用DES算法进行文件加密和解密

2014-12-10 15:18 519 查看
头文件:

extern "C"
{
BOOL xEncrypt(LPCTSTR lpszSource, LPCTSTR lpszDestination, LPCTSTR lpszPassword);
BOOL xDecrypt(LPCTSTR lpszSource, LPCTSTR lpszDestination, LPCTSTR lpszPassword);
}

cpp文件:
#include "stdafx.h"
#include "xEncrypt.h"

#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <wincrypt.h>

#define BLOCK_SIZE 1000
#define BUFFER_SIZE 1008

// 加密文件
BOOL xEncrypt(LPCTSTR lpszSource, LPCTSTR lpszDestination, LPCTSTR lpszPassword)
{
FILE *hSrcFile = NULL, *hDestFile = NULL;

HCRYPTPROV hProv = 0;
HCRYPTHASH hHash = 0;
HCRYPTKEY hKey = 0, hXchgKey = 0;
PBYTE pbBuffer = NULL, pbKeyBlob = NULL;
BOOL bEOF = FALSE, bReturn = FALSE;
DWORD dwCount = 0, dwKeyBlobLen = 0;

if (_wfopen_s(&hSrcFile, lpszSource, TEXT("rb")) != 0)
{
goto exit;
}

if (_wfopen_s(&hDestFile, lpszDestination, TEXT("wb")) != 0)
{
goto exit;
}

if (!CryptAcquireContext(&hProv, NULL, NULL, PROV_RSA_FULL, 0))
{
DWORD ret = GetLastError();
if (ret != NTE_BAD_KEYSET)
{
goto exit;
}
else
{
if (!CryptAcquireContext(&hProv, NULL, NULL, PROV_RSA_FULL, CRYPT_NEWKEYSET))
{
goto exit;
}
}
}

if (lpszPassword == NULL)
{
if (!CryptGenKey(hProv, CALG_RC2, CRYPT_EXPORTABLE, &hKey))
{
goto exit;
}

if (!CryptGetUserKey(hProv, AT_KEYEXCHANGE, &hXchgKey))
{
DWORD ret = GetLastError();
if (ret != NTE_NO_KEY)
{
goto exit;
}
if (!CryptGenKey(hProv,AT_KEYEXCHANGE,NULL,&hXchgKey))
{
goto exit;
}
}

if (!CryptExportKey(hKey, hXchgKey, SIMPLEBLOB, 0, NULL, &dwKeyBlobLen))
{
goto exit;
}

if ((pbKeyBlob = (PBYTE)malloc(dwKeyBlobLen)) == NULL)
{
goto exit;
}

if (!CryptExportKey (hKey, hXchgKey, SIMPLEBLOB, 0, pbKeyBlob, &dwKeyBlobLen))
{
goto exit;
}

fwrite (&dwKeyBlobLen, sizeof (DWORD), 1, hDestFile);

if (ferror (hDestFile))
{
goto exit;
}

fwrite (pbKeyBlob, 1, dwKeyBlobLen, hDestFile);

if (ferror(hDestFile))
{
goto exit;
}
}
else
{
if (!CryptCreateHash(hProv, CALG_MD5, 0, 0, &hHash))
{
goto exit;
}

if (!CryptHashData(hHash, (PBYTE)lpszPassword, wcslen (lpszPassword), 0))
{
goto exit;
}

if (!CryptDeriveKey(hProv, CALG_RC2, hHash, 0, &hKey))
{
goto exit;
}
}

if ((pbBuffer=(PBYTE)malloc(BUFFER_SIZE)) == NULL)
{
goto exit;
}

do
{
dwCount = fread(pbBuffer, 1, BLOCK_SIZE, hSrcFile);

if (ferror(hSrcFile))
{
goto exit;
}
bEOF = feof(hSrcFile);

if (!CryptEncrypt(hKey, 0, bEOF, 0, pbBuffer, &dwCount, BUFFER_SIZE))
{
goto exit;
}

fwrite(pbBuffer, 1, dwCount, hDestFile);
if (ferror(hDestFile))
{
goto exit;
}

} while (!bEOF);

bReturn = TRUE;

exit:
if (hSrcFile) fclose(hSrcFile);
if (hDestFile) fclose(hDestFile);
if (pbKeyBlob) free(pbKeyBlob);
if (pbBuffer) free(pbBuffer);
if (hKey) CryptDestroyKey(hKey);
if (hXchgKey) CryptDestroyKey(hXchgKey);
if (hHash) CryptDestroyHash(hHash);
if (hProv) CryptReleaseContext(hProv, 0);

return bReturn;
}

// 解密文件
BOOL xDecrypt (LPCTSTR lpszSource, LPCTSTR lpszDestination, LPCTSTR lpszPassword)
{
FILE *hSrcFile = NULL, *hDestFile = NULL;

HCRYPTPROV hProv = 0;
HCRYPTHASH hHash = 0;
HCRYPTKEY hKey = 0;

PBYTE pbBuffer = NULL, pbKeyBlob = NULL;
BOOL bEOF = 0, bReturn = FALSE;
DWORD dwCount = 0, dwKeyBlobLen = 0;

if (_wfopen_s(&hSrcFile, lpszSource, TEXT("rb")) != 0)
{
goto exit;
}

if (_wfopen_s(&hDestFile, lpszDestination, TEXT("wb")) != 0)
{
goto exit;
}

if (!CryptAcquireContext(&hProv, NULL, NULL, PROV_RSA_FULL, 0))
{
DWORD ret = GetLastError();
if (ret != NTE_BAD_KEYSET)
{
goto exit;
}
else
{
if (!CryptAcquireContext(&hProv, NULL, NULL, PROV_RSA_FULL, CRYPT_NEWKEYSET))
{
goto exit;
}
}
}

if (lpszPassword == NULL)
{
fread(&dwKeyBlobLen, sizeof (DWORD), 1, hSrcFile);
if (ferror(hSrcFile) || feof (hSrcFile))
{
goto exit;
}

if ((pbKeyBlob = (PBYTE)malloc(dwKeyBlobLen)) == NULL)
{
goto exit;
}

fread(pbKeyBlob, 1, dwKeyBlobLen, hSrcFile);

if (ferror(hSrcFile) || feof(hSrcFile))
{
goto exit;
}

if (!CryptImportKey(hProv, pbKeyBlob, dwKeyBlobLen, 0, 0, &hKey))
{
goto exit;
}
}
else
{
if (!CryptCreateHash(hProv, CALG_MD5, 0, 0, &hHash))
{
goto exit;
}

if (!CryptHashData(hHash, (PBYTE)lpszPassword, wcslen(lpszPassword), 0))
{
goto exit;
}

if (!CryptDeriveKey(hProv, CALG_RC2, hHash, 0, &hKey))
{
goto exit;
}
}

if ((pbBuffer = (PBYTE)malloc(BUFFER_SIZE)) == NULL)
{
goto exit;
}

do
{
dwCount = fread(pbBuffer, 1, BLOCK_SIZE, hSrcFile);

if (ferror(hSrcFile))
{
goto exit;
}

bEOF = feof(hSrcFile);

if (!CryptDecrypt(hKey, 0, bEOF, 0, pbBuffer, &dwCount))
{
goto exit;
}

fwrite(pbBuffer, 1, dwCount, hDestFile);
if (ferror(hDestFile))
{
goto exit;
}

} while (!bEOF);

bReturn = TRUE;
exit:
if (hSrcFile) fclose(hSrcFile);
if (hDestFile) fclose(hDestFile);
if (pbKeyBlob) free(pbKeyBlob);
if (pbBuffer) free(pbBuffer);
if (hKey) CryptDestroyKey(hKey);
if (hHash) CryptDestroyHash(hHash);
if (hProv) CryptReleaseContext (hProv, 0);

return bReturn;

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  加密 wincrypt DES 解密 VC