You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.5 KiB
C
88 lines
2.5 KiB
C
#include <stdio.h>
|
|
#include "gcrypt.h"
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <stdint.h>
|
|
#include <strings.h>
|
|
#include <inttypes.h>
|
|
|
|
#include "cryptwrapper.h"
|
|
|
|
int main(int argc, char const *argv[]) {
|
|
if (argc < 3){
|
|
printf("please select file and pw\n");
|
|
return 1;
|
|
}
|
|
FILE * fptr = fopen(argv[1],"rb");
|
|
if (fptr==NULL){
|
|
printf("could not open file %s\n",argv[1]);
|
|
return 1;
|
|
}
|
|
char * new_file_name = malloc(strlen(argv[1]+5));
|
|
strcpy(new_file_name,argv[1]);
|
|
strcat(new_file_name,".encry");
|
|
FILE * tfptr = fopen(new_file_name,"wb");
|
|
if (fseek(fptr,0,SEEK_END)){
|
|
printf("could not ssek end\n");
|
|
return 1;
|
|
};
|
|
int file_len = ftell(fptr);
|
|
rewind(fptr);// sets pointer to beginning of file
|
|
//printf("file is %d long\n", file_len);
|
|
char * buf_out = malloc(file_len + 32); // for sha256 check sum 32*8=256 lol
|
|
if (buf_out == NULL){
|
|
printf("could not allocate memory, maybe file is too large\n");
|
|
return 1;
|
|
}
|
|
|
|
if (file_len!=fread(buf_out,sizeof(char),file_len,fptr)){
|
|
printf("could not read all data from file\nfile_len=%d",file_len);
|
|
return 1;
|
|
}
|
|
|
|
//calculating sha256 hash to append to file
|
|
uint8_t * pw_hash = malloc(gcry_md_get_algo_dlen(algo));
|
|
uint8_t * file_hash = malloc(gcry_md_get_algo_dlen(algo));
|
|
|
|
|
|
mycryptwrapper_init();
|
|
gcry_md_hash_buffer(algo,pw_hash,argv[2],strlen(argv[2]));
|
|
gcry_md_hash_buffer(algo,file_hash,buf_out,file_len);
|
|
|
|
|
|
//appending sha256 to end of buffer
|
|
for (size_t i = 0; i<+32; i++){
|
|
buf_out[i+file_len] = file_hash[i];
|
|
}
|
|
|
|
gcry_cipher_hd_t hd;
|
|
if (gcry_cipher_open(&hd,cipher,GCRY_CIPHER_MODE_CFB,0)){
|
|
perror("could not open handle for cipher\n");
|
|
return 1;
|
|
}
|
|
if(gcry_cipher_setkey(hd,pw_hash,8)){
|
|
printf("could not set key for encryption");
|
|
return 1;
|
|
}
|
|
size_t len = 8;
|
|
void * iv = malloc(len);
|
|
if (gcry_cipher_setiv(hd, iv , len)){
|
|
printf("could not init init vector");
|
|
return 1;
|
|
}
|
|
if (gcry_cipher_encrypt(hd,buf_out,file_len+32,NULL,0)){
|
|
perror("could not encrypt data\n");
|
|
return 1;
|
|
}
|
|
if (fwrite(buf_out, sizeof(char),file_len+32,tfptr)!= file_len+32){
|
|
perror("could not write all data\n");
|
|
return 1;
|
|
}
|
|
|
|
gcry_cipher_close(hd);
|
|
|
|
fclose(fptr);
|
|
fclose(tfptr);
|
|
printf("successfully encrypted file %s with sha256 check sum and stored in file %s\n",argv[1],new_file_name);
|
|
return 0;
|
|
} |