forked from gpellicci/Secure-File-Storage
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathkeygen.h
More file actions
127 lines (108 loc) · 2.23 KB
/
keygen.h
File metadata and controls
127 lines (108 loc) · 2.23 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#include <openssl/rand.h>
bool keyGenToFile(int len, const char* fname){
int ret = RAND_poll();
if(ret == -1){
perror("RAND_poll. Error");
return false;
}
unsigned char* key = (unsigned char*)malloc(len);
if(key == NULL){
perror("Could not allocate memory. Error");
return false;
}
ret = RAND_bytes(key, len);
if(ret == -1){
perror("RAND_bytes. Error");
return false;
}
FILE* fd = fopen(fname, "w");
if(fd == NULL){
perror("Could not open the file. Error");
return false;
}
else{
int w_len = fwrite(key, sizeof(unsigned char), len, fd);
if(w_len != len || w_len == -1){
perror("fwrite. Error");
return false;
}
ret = fclose(fd);
if(ret == -1){
perror("fclose. Error");
return false;
}
return true;
}
}
bool keyGen(unsigned char* &pointer, int len){
int ret = RAND_poll();
if(ret == -1){
perror("RAND_poll. Error");
return false;
}
pointer = (unsigned char*)malloc(len);
if(pointer == NULL){
perror("Could not allocate memory. Error");
return false;
}
ret = RAND_bytes(pointer, len);
if(ret == -1){
perror("RAND_bytes. Error");
return false;
}
return true;
}
bool int64Gen(uint64_t* rand, int len){
int ret = RAND_poll();
if(ret == -1){
perror("RAND_poll. Error");
return false;
}
ret = RAND_bytes((unsigned char*)rand, len);
if(ret == -1){
perror("RAND_bytes. Error");
return false;
}
return true;
}
bool readKeyFromFile(unsigned char* &key, int len, const char* fname){
unsigned char* tmp = (unsigned char*)malloc(len);
if(tmp == NULL){
perror(". Error");
return false;
}
FILE* fd = fopen(fname, "r");
if(fd == NULL){
perror("fopen. Error");
return false;
}
else{
int r_len = fread(tmp, sizeof(unsigned char), len, fd);
if(r_len != len || r_len == -1){
perror("fread. Error");
return false;
}
int ret = fclose(fd);
if(ret == -1){
perror("fclose. Error");
return false;
}
key = tmp;
return true;
}
}
void printHexKey(unsigned char* key, int len){
if(key == NULL || len < 0)
return;
printf("Key: ");
for(int i=0; i<len; i++)
printf("%02x", key[i]);
printf("\n");
}
void printHex(unsigned char* key, int len){
if(key == NULL || len < 0)
return;
for(int i=0; i<len; i++)
printf("%02x", key[i]);
printf("\n");
}