#include<stdio.h>
#include<ctype.h>
#include<string.h>
#define MAX 80

char b[MAX],code[MAX],source_file[MAX],dest_file[MAX];
int scelta,ok=0,a,i,j,lcode,alpha[26];
FILE *fp,*fp2;

int acquisizione(void);
int cript(void);
int decript(void);
int test(void);

int main()
{
   printf("[1]codifica documento\n[2]decodifica documento\n[3]esci\n");
   scanf("%d",&scelta);
   switch(scelta){
   		case 1:	cript();
                  break;
	      case 2:	decript();
         			break;
         case 3:  return(1);
         default:	printf("Opzione errata\n");
	}
	return 0;
}

int acquisizione(void)
{
   int alphac[26],codlett='Z';
	printf("Inserisci il nome del file sorgente: ");
   scanf("%s",&source_file);
   printf("Inserisci il nome del file di destinazione:");
   scanf("%s",&dest_file);
   do{
   	printf("Inserisci il codice di cifratura: ");
   	scanf("%s",&code);
      ok=test();
   }while(ok);
   for(i=0;i<26;i++)
   	alphac[i]=0;
   for(i=0;i<lcode;i++)
   {
   	alpha[i]=code[i];
      if(islower(code[i]))
      	alphac[code[i]-'a']=1;
      else
      	alphac[code[i]-'A']=1;
   }
   for(i=lcode;i<26;i++)
   {
   	while(alphac[codlett-'A'])
      	codlett--;
      alpha[i]=codlett;
	  codlett--;
   }
   return 0;
}

int cript()
{
   acquisizione();
	if((fp=fopen(source_file,"r"))==NULL)
   {
   	printf("unable to open %s",source_file);
      return 1;
   }
   fp2=fopen(dest_file,"w");
   while((a=fgetc(fp))!=EOF)
   {
   	if(isalpha(a))
      {
			if(isupper(a))
				a=alpha[a-65];
			else
				a=alpha[a-97];
      }
      fprintf(fp2,"%c",a);
   }
   fclose(fp);
   fclose(fp2);
   return 0;
}

int decript()
{
	acquisizione();
	if((fp=fopen(source_file,"r"))==NULL)
   {
   	printf("unable to open %s",source_file);
      return 1;
   }
   fp2=fopen(dest_file,"w");
   while((a=fgetc(fp))!=EOF)
   {
   	if(isalpha(a))
      {
       if(isupper(a))
				for(i=0;i<26;i++)
            {
            	if(a==alpha[i])
               	fprintf(fp2,"%c",i+65);

            }
       else
       		for(i=0;i<26;i++)
					if(a==alpha[i])
               	fprintf(fp2,"%c",i+97);
      }
      else
      	fprintf(fp2,"%c",a);
   }
   fclose(fp);
   fclose(fp2);
   return 0;
}

int test(void)
{
    int j,count=0;
    lcode=strlen(code);
    for(i=0;i<lcode;i++)
    {
    	count=0;
    	for(j=0;j<lcode;j++)
      	if(code[i]==code[j])
         	count++;
         if(count>1)
         	return 1;
    }
    return 0;
}