Delphi AES加密

2018-10-31

(**************************************************************)
(*    Advanced Encryption Standard (AES)    *)
(*    Interface Unit v1.3    *)
(*    *)
(*    Copyright (c) 2002 Jorlen Young    *)
(*    *)
(* 说明:    *)
(*    基于 ElASE.pas 单元封装    *)
(*    *)
(*    这是一个 AES 加密算法的标准接口。    *)
(* 调用示例:    *)
(* if not EncryptStream(src, key, TStream(Dest), keybit) then *)
(*   showmessage('encrypt error');    *)
(*    *)
(* if not DecryptStream(src, key, TStream(Dest), keybit) then *)
(*   showmessage('encrypt error');    *)
(*    *)
(* *** 一定要对Dest进行TStream(Dest) ***    *)
(* ========================================================== *)
(*    *)
(*   支持 128 / 192 / 256 位的密匙    *)
(*   默认情况下按照 128 位密匙操作    *)
(*    *)
(**************************************************************)

unit AES;   

interface

{$IFDEF VER210}
  {$WARN IMPLICIT_STRING_CAST OFF} //关闭警告
  {$WARN IMPLICIT_STRING_CAST_LOSS OFF}
{$ENDIF}
uses
  SysUtils, Classes, Math, ElAES;

const
  SDestStreamNotCreated = 'Dest stream not created.';
  SEncryptStreamError = 'Encrypt stream error.';
  SDecryptStreamError = 'Decrypt stream error.';

type
  TKeyBit = (kb128, kb192, kb256);

function StrToHex(Const str: AnsiString): AnsiString;
function HexToStr(const Str: AnsiString): AnsiString;

function EncryptString(Value: AnsiString; Key: AnsiString;
  KeyBit: TKeyBit = kb128): AnsiString;
function DecryptString(Value: AnsiString; Key: AnsiString;
  KeyBit: TKeyBit = kb128): AnsiString;

function EncryptStream(Src: TStream; Key: AnsiString;
  var Dest: TStream; KeyBit: TKeyBit = kb128): Boolean;
function DecryptStream(Src: TStream; Key: AnsiString;
  var Dest: TStream; KeyBit: TKeyBit = kb128): Boolean;

procedure EncryptFile(SourceFile, DestFile: String;
  Key: AnsiString; KeyBit: TKeyBit = kb128);
procedure DecryptFile(SourceFile, DestFile: String;
  Key: AnsiString; KeyBit: TKeyBit = kb128);

implementation

function StrToHex(Const str: Ansistring): Ansistring;
asm
   push ebx
   push esi
   push edi
   test eax,eax
   jz   @@Exit
   mov  esi,edx    //保存edx值,用来产生新字符串的地址
   mov  edi,eax    //保存原字符串
   mov  edx,[eax-4]  //获得字符串长度
   test edx,edx    //检查长度
   je   @@Exit    {Length(S) = 0}
   mov  ecx,edx    //保存长度
   Push ecx
   shl  edx,1
   mov  eax,esi
   {$IFDEF VER210}
   movzx ecx, word ptr [edi-12] {需要设置CodePage}
   {$ENDIF}
   call System.@LStrSetLength //设置新串长度
   mov  eax,esi    //新字符串地址
   Call UniqueString  //产生一个唯一的新字符串,串位置在eax中
   Pop   ecx
  @@SetHex:
   xor  edx,edx    //清空edx
   mov  dl, [edi]    //Str字符串字符
   mov  ebx,edx    //保存当前的字符
   shr  edx,4    //右移4字节,得到高8位
   mov  dl,byte ptr[edx+@@HexChar] //转换成字符
   mov  [eax],dl    //将字符串输入到新建串中存放
   and  ebx,$0F    //获得低8位
   mov  dl,byte ptr[ebx+@@HexChar] //转换成字符
   inc  eax    //移动一个字节,存放低位
   mov  [eax],dl
   inc  edi
   inc  eax
   loop @@SetHex
  @@Exit:
   pop  edi
   pop  esi
   pop  ebx
   ret
  @@HexChar: db '0123456789ABCDEF'
end;

function HexToStr(const Str: AnsiString): AnsiString;
asm
  push ebx
  push edi
  push esi
  test eax,eax //为空串
  jz   @@Exit
  mov  edi,eax
  mov  esi,edx
  mov  edx,[eax-4]
  test edx,edx
  je   @@Exit
  mov  ecx,edx
  push ecx
  shr  edx,1
  mov  eax,esi //开始构造字符串
  {$IFDEF VER210}
  movzx ecx, word ptr [edi-12] {需要设置CodePage}
  {$ENDIF}
  call System.@LStrSetLength //设置新串长度
  mov  eax,esi    //新字符串地址
  Call UniqueString  //产生一个唯一的新字符串,串位置在eax中
  Pop   ecx
  xor  ebx,ebx
  xor  esi,esi
@@CharFromHex:
  xor  edx,edx
  mov  dl, [edi]    //Str字符串字符
  cmp  dl, '0'  //查看是否在0到f之间的字符
  JB   @@Exit   //小于0,退出
  cmp  dl,'9'   //小于=9
  ja  @@DoChar//CompOkNum
  sub  dl,'0'
  jmp  @@DoConvert
@@DoChar:
  //先转成大写字符
  and  dl,$DF
  cmp  dl,'F'
  ja   @@Exit  //大于F退出
  add  dl,10
  sub  dl,'A'
@@DoConvert: //转化
  inc  ebx
  cmp  ebx,2
  je   @@Num1
  xor  esi,esi
  shl  edx,4
  mov  esi,edx
  jmp  @@Num2
@@Num1:
  add  esi,edx
  mov  edx,esi
  mov  [eax],dl
  xor  ebx,ebx
  inc  eax
@@Num2:
  dec  ecx
  inc  edi
  test ecx,ecx
  jnz  @@CharFromHex
@@Exit:
  pop  esi
  pop  edi
  pop  ebx
end;

{  --  字符串加密函数 默认按照 128 位密匙加密 --  }
function EncryptString(Value: AnsiString; Key: AnsiString;
  KeyBit: TKeyBit = kb128): AnsiString;
var
  {$IFDEF VER210}
  SS,DS: TMemoryStream;
  {$ELSE}
  SS, DS: TStringStream;
  {$ENDIF}
  Size: Int64;
  AESKey128: TAESKey128;
  AESKey192: TAESKey192;
  AESKey256: TAESKey256;
  st: AnsiString;
begin
  Result := '';
  {$IFDEF VER210}
   ss := TMemoryStream.Create;
   SS.WriteBuffer(PAnsiChar(Value)^,Length(Value));
   DS := TMemoryStream.Create;
  {$ELSE}
   SS := TStringStream.Create(Value);
   DS := TStringStream.Create('');
  {$ENDIF}
  try
   Size := SS.Size;
   DS.WriteBuffer(Size, SizeOf(Size));
   {  --  128 位密匙最大长度为 16 个字符 --  }
   if KeyBit = kb128 then
   begin
   FillChar(AESKey128, SizeOf(AESKey128), 0 );
   Move(PAnsiChar(Key)^, AESKey128, Min(SizeOf(AESKey128), Length(Key)));
   EncryptAESStreamECB(SS, 0, AESKey128, DS);
   end;
   {  --  192 位密匙最大长度为 24 个字符 --  }
   if KeyBit = kb192 then
   begin
   FillChar(AESKey192, SizeOf(AESKey192), 0 );
   Move(PAnsiChar(Key)^, AESKey192, Min(SizeOf(AESKey192), Length(Key)));
   EncryptAESStreamECB(SS, 0, AESKey192, DS);
   end;
   {  --  256 位密匙最大长度为 32 个字符 --  }
   if KeyBit = kb256 then
   begin
   FillChar(AESKey256, SizeOf(AESKey256), 0 );
   Move(PAnsiChar(Key)^, AESKey256, Min(SizeOf(AESKey256), Length(Key)));
   EncryptAESStreamECB(SS, 0, AESKey256, DS);
   end;
   {$IFDEF VER210}
   SetLength(st,Ds.Size);
   DS.Position := 0;
   DS.ReadBuffer(PAnsiChar(st)^,DS.Size);
   Result := StrToHex(st);
   {$ELSE}
   Result := StrToHex(DS.DataString);
   {$ENDIF}
  finally
   SS.Free;
   DS.Free;
  end;
end;

{  --  字符串解密函数 默认按照 128 位密匙解密 --  }
function DecryptString(Value: AnsiString; Key: AnsiString;
  KeyBit: TKeyBit = kb128): AnsiString;
var
  SS, DS: TStringStream;
  Size: Int64;
  AESKey128: TAESKey128;
  AESKey192: TAESKey192;
  AESKey256: TAESKey256;
begin
  Result := '';
  SS := TStringStream.Create(HexToStr(Value));
  DS := TStringStream.Create('');
  try
   Size := SS.Size;
   SS.ReadBuffer(Size, SizeOf(Size));
   {  --  128 位密匙最大长度为 16 个字符 --  }
   if KeyBit = kb128 then
   begin
   FillChar(AESKey128, SizeOf(AESKey128), 0 );
   Move(PAnsiChar(Key)^, AESKey128, Min(SizeOf(AESKey128), Length(Key)));
   DecryptAESStreamECB(SS, SS.Size - SS.Position, AESKey128, DS);
   end;
   {  --  192 位密匙最大长度为 24 个字符 --  }
   if KeyBit = kb192 then
   begin
   FillChar(AESKey192, SizeOf(AESKey192), 0 );
   Move(PAnsiChar(Key)^, AESKey192, Min(SizeOf(AESKey192), Length(Key)));
   DecryptAESStreamECB(SS, SS.Size - SS.Position, AESKey192, DS);
   end;
   {  --  256 位密匙最大长度为 32 个字符 --  }
   if KeyBit = kb256 then
   begin
   FillChar(AESKey256, SizeOf(AESKey256), 0 );
   Move(PAnsiChar(Key)^, AESKey256, Min(SizeOf(AESKey256), Length(Key)));
   DecryptAESStreamECB(SS, SS.Size - SS.Position, AESKey256, DS);
   end;
   Result := DS.DataString;
  finally
   SS.Free;
   DS.Free;
  end;
end;

{ 流加密函数, default keybit: 128bit }
function EncryptStream(Src: TStream; Key: AnsiString;
  var Dest: TStream; KeyBit: TKeyBit = kb128): Boolean;
var
  Count: Int64;
  AESKey128: TAESKey128;
  AESKey192: TAESKey192;
  AESKey256: TAESKey256;
begin
  if Dest = nil then
  begin
   raise Exception.Create(SDestStreamNotCreated);
   Result:= False;
   Exit;
  end;

  try
   Src.Position:= 0;
   Count:= Src.Size;
   Dest.Write(Count, SizeOf(Count));
   {  --  128 位密匙最大长度为 16 个字符 --  }
   if KeyBit = kb128 then
   begin
   FillChar(AESKey128, SizeOf(AESKey128), 0 );
   Move(PAnsiChar(Key)^, AESKey128, Min(SizeOf(AESKey128), Length(Key)));
   EncryptAESStreamECB(Src, 0, AESKey128, Dest);
   end;
   {  --  192 位密匙最大长度为 24 个字符 --  }
   if KeyBit = kb192 then
   begin
   FillChar(AESKey192, SizeOf(AESKey192), 0 );
   Move(PAnsiChar(Key)^, AESKey192, Min(SizeOf(AESKey192), Length(Key)));
   EncryptAESStreamECB(Src, 0, AESKey192, Dest);
   end;
   {  --  256 位密匙最大长度为 32 个字符 --  }
   if KeyBit = kb256 then
   begin
   FillChar(AESKey256, SizeOf(AESKey256), 0 );
   Move(PAnsiChar(Key)^, AESKey256, Min(SizeOf(AESKey256), Length(Key)));
   EncryptAESStreamECB(Src, 0, AESKey256, Dest);
   end;

   Result := True;
  except
   raise Exception.Create(SEncryptStreamError);
   Result:= False;
  end;
end;

{ 流解密函数, default keybit: 128bit }
function DecryptStream(Src: TStream; Key: AnsiString;
  var Dest: TStream; KeyBit: TKeyBit = kb128): Boolean;
var
  Count, OutPos: Int64;
  AESKey128: TAESKey128;
  AESKey192: TAESKey192;
  AESKey256: TAESKey256;
begin
  if Dest = nil then
  begin
   raise Exception.Create(SDestStreamNotCreated);
   Result:= False;
   Exit;
  end;

  try
   Src.Position:= 0;
   OutPos:= Dest.Position;
   Src.ReadBuffer(Count, SizeOf(Count));
   {  --  128 位密匙最大长度为 16 个字符 --  }
   if KeyBit = kb128 then
   begin
   FillChar(AESKey128, SizeOf(AESKey128), 0 );
   Move(PAnsiChar(Key)^, AESKey128, Min(SizeOf(AESKey128), Length(Key)));
   DecryptAESStreamECB(Src, Src.Size - Src.Position,
   AESKey128, Dest);
   end;
   {  --  192 位密匙最大长度为 24 个字符 --  }
   if KeyBit = kb192 then
   begin
   FillChar(AESKey192, SizeOf(AESKey192), 0 );
   Move(PAnsiChar(Key)^, AESKey192, Min(SizeOf(AESKey192), Length(Key)));
   DecryptAESStreamECB(Src, Src.Size - Src.Position,
   AESKey192, Dest);
   end;
   {  --  256 位密匙最大长度为 32 个字符 --  }
   if KeyBit = kb256 then
   begin
   FillChar(AESKey256, SizeOf(AESKey256), 0 );
   Move(PAnsiChar(Key)^, AESKey256, Min(SizeOf(AESKey256), Length(Key)));
   DecryptAESStreamECB(Src, Src.Size - Src.Position,
   AESKey256, Dest);
   end;
   Dest.Size := OutPos + Count;
   Dest.Position := OutPos;

   Result := True;
  except
   raise Exception.Create(SDecryptStreamError);
   Result:= False;
  end;
end;

{  --  文件加密函数 默认按照 128 位密匙解密 --  }
procedure EncryptFile(SourceFile, DestFile: String;
  Key: AnsiString; KeyBit: TKeyBit = kb128);
var
  SFS, DFS: TFileStream;
  Size: Int64;
  AESKey128: TAESKey128;
  AESKey192: TAESKey192;
  AESKey256: TAESKey256;
begin
  SFS := TFileStream.Create(SourceFile, fmOpenRead);
  try
   DFS := TFileStream.Create(DestFile, fmCreate);
   try
   Size := SFS.Size;
   DFS.WriteBuffer(Size, SizeOf(Size));
   {  --  128 位密匙最大长度为 16 个字符 --  }
   if KeyBit = kb128 then
   begin
   FillChar(AESKey128, SizeOf(AESKey128), 0 );
   Move(PAnsiChar(Key)^, AESKey128, Min(SizeOf(AESKey128), Length(Key)));
   EncryptAESStreamECB(SFS, 0, AESKey128, DFS);
   end;
   {  --  192 位密匙最大长度为 24 个字符 --  }
   if KeyBit = kb192 then
   begin
   FillChar(AESKey192, SizeOf(AESKey192), 0 );
   Move(PAnsiChar(Key)^, AESKey192, Min(SizeOf(AESKey192), Length(Key)));
   EncryptAESStreamECB(SFS, 0, AESKey192, DFS);
   end;
   {  --  256 位密匙最大长度为 32 个字符 --  }
   if KeyBit = kb256 then
   begin
   FillChar(AESKey256, SizeOf(AESKey256), 0 );
   Move(PAnsiChar(Key)^, AESKey256, Min(SizeOf(AESKey256), Length(Key)));
   EncryptAESStreamECB(SFS, 0, AESKey256, DFS);
   end;
   finally
   DFS.Free;
   end;
  finally
   SFS.Free;
  end;
end;

{  --  文件解密函数 默认按照 128 位密匙解密 --  }
procedure DecryptFile(SourceFile, DestFile: String;
  Key: AnsiString; KeyBit: TKeyBit = kb128);
var
  SFS, DFS: TFileStream;
  Size: Int64;
  AESKey128: TAESKey128;
  AESKey192: TAESKey192;
  AESKey256: TAESKey256;
begin
  SFS := TFileStream.Create(SourceFile, fmOpenRead);
  try
   SFS.ReadBuffer(Size, SizeOf(Size));
   DFS := TFileStream.Create(DestFile, fmCreate);
   try
   {  --  128 位密匙最大长度为 16 个字符 --  }
   if KeyBit = kb128 then
   begin
   FillChar(AESKey128, SizeOf(AESKey128), 0 );
   Move(PAnsiChar(Key)^, AESKey128, Min(SizeOf(AESKey128), Length(Key)));
   DecryptAESStreamECB(SFS, SFS.Size - SFS.Position, AESKey128, DFS);
   end;
   {  --  192 位密匙最大长度为 24 个字符 --  }
   if KeyBit = kb192 then
   begin
   FillChar(AESKey192, SizeOf(AESKey192), 0 );
   Move(PAnsiChar(Key)^, AESKey192, Min(SizeOf(AESKey192), Length(Key)));
   DecryptAESStreamECB(SFS, SFS.Size - SFS.Position, AESKey192, DFS);
   end;
   {  --  256 位密匙最大长度为 32 个字符 --  }
   if KeyBit = kb256 then
   begin
   FillChar(AESKey256, SizeOf(AESKey256), 0 );
   Move(PAnsiChar(Key)^, AESKey256, Min(SizeOf(AESKey256), Length(Key)));
   DecryptAESStreamECB(SFS, SFS.Size - SFS.Position, AESKey256, DFS);
   end;
   DFS.Size := Size;
   finally
   DFS.Free;
   end;
  finally
   SFS.Free;
  end;
end;
end.


阅读196