cdrmcrypt.cpp

aes的原代码

// CDrmCrypt.cpp : 定义控制台应用程序的入口点。
// 测试程序

#include "stdafx.h"
#include "decrypt.h"
#include "stdio.h"
#include "DeDistBuffer.h"
#include "EnDistBuffer.h"
#include "Header.h"
#include <iostream>
#include <fstream>
#define KEYLEN 128
int main(int argc, TCHAR* argv[])
{
	unsigned char key[] = {0x11,0xff,0x9d,0x88,0x77,0x78,0xad,0xfc,0x9a,0x66,0x23,0x39,0x41,0xab,0x4d,0x13};
	unsigned char IV[] = {0x11,0xff,0x9d,0x88,0x77,0x78,0xad,0xfc,0x9a,0x66,0x23,0x39,0x41,0xab,0x4d,0x13};
	unsigned char pcID[] = {0xaa,0xcc,0x9f,0xa8,0xe7,0xb8,0xab,0x5b,0x78,0x17,0x39,0xa8,0xc7,0x7c,0x9f,0xef};
	char debuf[DEC_SEC_LEN];
	FILE * fp,*fp1,*fp2,*fp3;
	char buf[DEC_SEC_LEN];
	int len,count=0;

	fp2 = fopen("test.txt","rb");
	fp3 = fopen("encrpyt.txt","wb");
	EnDistBuffer * eb = new EnDistBuffer(pcID,DEC_SEC_LEN,0);
	printf("正在加扰。。。\n");
	len=fread(buf,1,160,fp2);
	fwrite(buf,1,len,fp3);
	fseek(fp2,160,SEEK_SET);
	fseek(fp3,160,SEEK_SET);
	while(!feof(fp2))
	{
		len = fread(buf,1,DEC_SEC_LEN,fp2);
		count+=len;
		printf("加扰 count=%d len=%d  字节\n",count,len);
		eb->EnBuffer((unsigned char*)((void *)buf),len);						//加扰测试
		fwrite(buf,1,len,fp3);
	}
	fclose(fp2);
	fclose(fp3);
	delete(eb);
	printf("加扰完成！\n\n");

	fp=fopen("encrpyt.txt","rb");//d:\\data.txt
	if(fp==NULL){printf("can not open file!\n"); return 0;}
	fp1=fopen("b.txt","wb");
	if(fp1==NULL){printf("can not open file!\n"); return 0;}
	
	
	Decrypt *dc = new Decrypt(key,IV,KEYLEN);
	DeDistBuffer *db = new DeDistBuffer(pcID,DEC_SEC_LEN,0,0);
	//Header * h=new Header(fp);
	count=0;
	fseek(fp,160,SEEK_SET);
	while(!feof(fp))
	{
		len=fread(buf,1,DEC_SEC_LEN, fp);
		count+=len;
		db->DeBuffer((unsigned char*)((void *)buf),len);					//解扰测试
		try{
			len=dc->DecryptBuffer((unsigned char*)((void *)buf),(unsigned char*)((void *)debuf),len);	//解密测试
		}catch(exception e)
		{
			printf("%s\n",e);
		}
		printf("count=%d len=%d  字节\n",count,len);
		fwrite(debuf,1,len,fp1);
	}
	fclose(fp);
	fclose(fp1);
	delete(dc);
	delete(db);
	//delete(h);
	printf("完成！\n");
	getchar();
	return 0;
}