RC4 C語言實現

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define MAX 1024
FILE *fp;
char ch[1024],ch1,ch2;
char filename1[20],filename2[20];
typedef unsigned long ULONG;
void rc4_init(unsigned char *s, unsigned char *key, unsigned long Len) //初始化函數
{
int i =0, j = 0;
char k[256] = {0};
unsigned char tmp = 0;
for(i=0;i<256;i++)
{
s[i]=i;
k[i]=key[i%Len];
}
for (i=0; i<256; i++)
{
j=(j+s[i]+k[i])%6;
tmp = s[i];
s[i] = s[j]; //交換s[i]和s[j]
s[j] = tmp;
}
}
void rc4_crypt(unsigned char *s, unsigned char *Data, unsigned long Len) //加解密
{
int i = 0, j = 0, t = 0;
unsigned long k = 0;
unsigned char tmp;
for(k=0;k
{
i=(i+1)%6;
j=(j+s[i])%6;
tmp = s[i];
s[i] = s[j]; //交換s[x]和s[y]
s[j] = tmp;
t=(s[i]+s[j])%6;
Data[k] ^= s[t];
}
}//加解密函數//
puttext(char filename[],char *ch)
{
if((fp=fopen(filename2,"w"))==NULL)
    printf("打開錯誤!\n\n");
   fputs(ch,fp);
   fclose(fp);
   exit(0);
}
gettext(char filename[])
{
    if((fp=fopen(filename,"rt"))==NULL)
    printf("打開錯誤!\n\n");
    ch1=fgetc(fp);
    while(ch1!=EOF)
        ch1=fgetc(fp);
 printf("\n\n");
    fclose(fp);
 fp=fopen(filename,"r");
 fgets(ch,MAX,fp);
 fclose(fp);
    printf("\n");
    printf("文件打開完成！\n\n");
}//文件輸入輸出函數//
int main()
{
    unsigned char s[256] = {0},s2[256] = {0}; //S-box
    char key[256];
 int i,n;
    ULONG len;
    printf("請輸入密鑰：\n");
 gets(key);
    printf("請輸入數字選擇相應功能：\n");
    printf("**************************\n");
    printf("1------加密文件。\n");
    printf("2------解密文件。\n");
 printf("3------動態輸入。\n");
    printf("0------退出。\n");
 printf("**************************\n");
 scanf("%d",&n);
 if(n==0)
 {
  printf("謝謝使用！再見！\n");
        exit(0);
        system("pause");
 }
 else if(n>3)
 {
  printf("輸入錯誤！\n");
  exit(0);
 }
 printf("key = %s, length = %d\n\n",key,strlen(key));
    rc4_init(s,(unsigned char *)key,strlen(key)); //已經完成了初始化
 printf("完成對S[i]的初始化!");
    printf("\n\n");
    for(i=0;i<256;i++)//用s2[i]暫時保留通過初始化的s[i]
    s2[i]=s[i];
 printf("請輸入須要加（解）密文件的路徑及文件名:");
 scanf("%s",filename1);
    printf("\n\n");
    switch(n)
    {
 case 1:
  {
  gettext(filename1);
  len=strlen(ch);
  printf("文件內容爲%s;\n長度爲%d。\n\n",ch,len);
  printf("已經初始化，如今加密:\n\n");
        rc4_crypt(s,(unsigned char *)ch,len);//加密
  printf("加密後的內容爲%0x\n\n",ch);
  printf("請輸入輸出文件地址和名稱：");
  scanf("%s",filename2);
  puttext(filename2,ch);
     printf("文件輸出完畢！");
 }
 case 2:
  {
  gettext(filename1);
  len = strlen(ch);
        printf("如今解密:\n");
        rc4_init(s,(unsigned char *)key, strlen(key)); //初始化
        rc4_crypt(s2,(unsigned char *)ch,len);//解密
  printf("解密後的內容爲%s\n\n",ch);
        printf("請輸入輸出文件地址和名稱：");
  scanf("%s",filename2);
  puttext(filename2,ch);
  printf("文件輸出完畢！");
      }
 case 3:
  {
  printf("請輸入加密內容:\n");
  scanf("%s",ch);
  len=strlen(ch);
  printf("輸入內容爲%s;\n長度爲%d。\n\n",ch,len);
  printf("已經初始化，如今加密:\n\n");
        rc4_crypt(s,(unsigned char *)ch,len);//加密
  printf("加密後的內容爲%0x\n\n",ch);
        printf("如今解密:\n");
        rc4_init(s,(unsigned char *)key, strlen(key)); //初始化
        rc4_crypt(s2,(unsigned char *)ch,len);//解密
  printf("解密後的內容爲%s\n\n",ch);
  }
 }
  return 0;
}