linux下AES-CBC128的加密程序测试源码

/*
    From :http://voip01.cublog.cn
    Author :Andy_yang
    E-mail :[e-mail]
    Date :2009/01/23
    Descriptions: AES CBC 128 加密算法。可以批量加密路径里面所以文件。其中不满16byte的用0x00填充。最好是加密txt文件，否则pad的位会出问题。如果加密二进制文件的 话重新写下 PadData（）函数。例如pad最后一位添上需要pad的长度，然后解密时依照最后一位的值来还原加密前的长度。
    Require : OpenSSL lib
    Compile CLI ：gcc -o aes_cbc_128 aes_cbc_128.c -lcrypto -static
*/
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <dirent.h>
#include <fcntl.h>
#include <dirent.h>
#include <openssl/aes.h>
#include <openssl/crypto.h>
#include <openssl/x509.h>
#include <openssl/pem.h>
#include <openssl/ssl.h>
#include <openssl/bio.h>
#include <openssl/evp.h>
#include <openssl/err.h>
unsigned int PadData (unsigned char *ibuf, unsigned int ilen, int blksize)
{
  unsigned int i; /* loop counter*/
  unsigned char pad; /* pad character (calculated)*/
  unsigned char *p; /*pointer to end of data*/

  /* calculate pad character*/
  pad = (unsigned char) (blksize - (ilen % blksize));

  /* append pad to end of string*/
  p = ibuf + ilen;
  for (i = 0; i < (int) pad; i++) {
    *p = 0x00;
 ++p;
  }

  return (ilen + pad);
}
unsigned int NoPadLen (unsigned char *ibuf, unsigned int ilen)
{
  unsigned int i; /* adjusted length*/
  unsigned char *p; /* pointer to last character*/

  p = ibuf + (ilen - 1);
  i = ilen - (unsigned int) *p;
  return (i);
}
void myfree(void *a)
{
 if(a)free(a);
 a = NULL;
}
int OpenReadInFile(char *fullname,unsigned char *OutBuf,int *rtn_size)
{
 int ifd;
 struct stat istat;
 unsigned int inSize, bit4,count;
 unsigned char *srcPtr = NULL;
 *rtn_size = 0;
 ifd = open(fullname,O_RDONLY,S_IREAD);
 if( ifd < 0 )
 {
  printf("Error: Unable to open %s\n", fullname);
  return(0);
 }
 /* Get the input file properties*/
 fstat(ifd,&istat);
 /* Create the input buffer*/
 inSize = istat.st_size;
 bit4 = inSize%4;
 if(inSize+4-bit4 > 0x16000)
 {
  printf("File is too big.Max is 0x16000\n");
  close(ifd);
  return 0;
 }
 srcPtr = (unsigned char *) malloc(inSize+(4-bit4));
 if( srcPtr == NULL )
 {
  printf("Error: Unable to create input buffer size of %d\n", inSize);
  close(ifd);
  return(0);
 }
 memset(srcPtr,0x0,inSize+(4-bit4));

 /* Read the buffer from file to local buffer*/
 count = read(ifd, (void *)srcPtr, istat.st_size);
 if( count != istat.st_size )
 {
  printf("Debug: Need to handle multi-reads/error to buffer\n");
  printf("Debug: Read returned 0x%0x\n", (unsigned int) count);
  free(srcPtr);
  close(ifd);
  return(0);
 }
 memcpy(OutBuf,srcPtr,count);
 *rtn_size = count;
 close(ifd);
 return 1;
}

int OpenWriteFile(char *fullname,char *Buffer,int size)
{
 int ifd;
 struct stat istat;
 unsigned int inSize, bit4,count;
 unsigned char *srcPtr = Buffer;
 ifd = creat(fullname, S_IRUSR | S_IWUSR);
 if(ifd < 0)
 {
  printf("Error: Unable to create %s\n", fullname );
  return(0);
 }
 count = write(ifd, srcPtr, size);
 if( size )
 {
  close(ifd);
  printf("Error: Write size error!\n", fullname );
  return(0);
 }
 close(ifd);
 return 1;
}
void hexdump(const char *title,const unsigned char *s, int len)
{
 int i;

 printf("\n%s\n",title);
 for(i = 0 ; i < len ; i++)
 {
  if((i % 16)==0)
   printf("\n%04x",i);
  printf(" %02x",s[i]);
 }
 printf("\n");
}
int Encrypty(char *Key,char *InputBuf,int in_length,unsigned char *outputBuf,int *out_length)
{
 int result=0;
 unsigned char iv[16];
 unsigned char *Output = NULL;
 unsigned char *InBuf_ptr = InputBuf;
 AES_KEY ass_key;
 int length_in_pad=0;
 /***********************
 printf("#####Key =%s\n",Key);
 printf("#####input len =%d\n",in_length);*/
 length_in_pad=PadData(InBuf_ptr,in_length,16);
 /*printf("#####length_in_pad =%d\n",length_in_pad);
 *encrypt text*/
 memset(&ass_key,0x00,sizeof(AES_KEY));
 result=AES_set_encrypt_key((const char *)Key,128,&ass_key);
 if(result<0)
 {
  printf("!!!!!!!!AES_set_encrypt_key Error!!!!!!!!!!\n ");
  return 0;
 }
 memset(iv,0,sizeof(iv));
 if((Output = (unsigned char *)malloc(length_in_pad+4)) == NULL)
 {
  printf("malloc error!\n");
  return 0;
 }
 memset(Output,0x00,length_in_pad+4);
 AES_cbc_encrypt(InBuf_ptr,Output,length_in_pad,&ass_key,iv,AES_ENCRYPT);
 memcpy(outputBuf,Output,length_in_pad);
 *out_length=length_in_pad;
 myfree(Output);
 /**outputBuf = Output;*/
 /*printf("#####encrypt out len =%d\n",*out_length);*/
 return 1;
}

int main(int argc,char *argv[])
{
 unsigned char Input[0x16000];
 unsigned char Output[0x16000];
 int OutputSize;
 char ifname[256];
 char fullname[256];
 char ofname[256];
 char dirname[256];
 char dst_dirname[256];
 char Key[17];/*16+1 : key length is 16 and 128 bits*/
 struct dirent *ptr = NULL;
 DIR * dir;
 int pathlen;
 int len;
 if(argc != 4)
 {
     printf("%s \n",argv[0]);
     return 0;
 }
 memset(Key,0,sizeof(Key));
 strcpy(Key,argv[1]);
 if(strlen(Key) != 16)
 {
  printf("Length of key must be 16 char!\n");
  return 0;
 }
 memset(Input,0x00,0x16000);
 memset(Output,0x00,0x16000);
 dir =opendir(argv[2]);
 strcpy(dirname,argv[2]);
 if(dir == NULL)
 {
  /*printf("open dir %s error!Not directory!\n",argv[2]);*/
  if(OpenReadInFile(argv[2],Input,&len) == 0)
  {
   printf("Openfile %s error!\n",argv[2]);
   return 0;
  }
  /*printf("len:%d,Inputfile:\n%s\n",len,Input);*/
  if(Encrypty(Key,Input,len,Output,&OutputSize) == 0)
  {
   printf("Encrypt file %s fail!\n",argv[2]);
   return 0;
  }
  /*printf("outlen:%d\n",OutputSize);*/
  /*hexdump("\nOutputfile:\n",Output,OutputSize);*/
  if(OpenWriteFile(argv[3],Output,OutputSize) == 0)
  {
   printf("Creat encrypt file %s fail\n",argv[3]);
   return 0;
  }
  printf("Encrypt file %s succesfull!\n",argv[2]);
  return 1;
 }
 strcpy(dst_dirname,argv[3]);
 pathlen = strlen(dst_dirname);
 if(dst_dirname[pathlen] != '/')
  strcat(dst_dirname,"/");
 while((ptr = readdir(dir))!=NULL)
 {
  memset(Input,0x00,0x16000);
  memset(Output,0x00,0x16000);
  strcpy(ifname, ptr->d_name);
  if(strcmp(ifname,".") == 0 || strcmp(ifname,"..") == 0 || strstr(ifname,".txt") == NULL)
   continue;
  snprintf(fullname,256,"%s%s",dirname,ifname);
  /*printf("open full path file %s !\n",fullname);*/
  snprintf(ofname, 256,"%s%s.ser",dst_dirname,ifname);
  /*printf("open file %s !\n",ifname);*/
  if(OpenReadInFile(fullname,Input,&len) == 0)
  {
   printf("Open file %s fail!\n",fullname);
   continue;
  }
  if(Encrypty(Key,Input,len,Output,&OutputSize) == 0)
  {
   printf("Encrypt file %s fail!\n",fullname);
   continue;
  }
  if(OpenWriteFile(ofname,Output,OutputSize) == 0)
  {
   printf("Creat encrypt file %s fail!\n",fullname);
   continue;
  }
  printf("Encrypt file %s succesfull!\n",fullname);
 }
 closedir(dir);
 return 1;
}