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
|
#include "rc4.h"
char* rc4_getPassword(void) {
unsigned char* pass = malloc(PASS_MAX + 1);
pointer_alloc_check(pass);
initscr();
noecho(); // Doesn't echo chars when typed
unsigned int c, i;
do {
printw("Enter Password (32 character max): ");
memset(pass, '\0', PASS_MAX + 1);
i = 0, c = 0;
while (c != '\n' && (c = (unsigned) getch())) {
if (c != BACKSPACE) {
pass[i] = (char) c;
i++;
if (i > PASS_MAX) { // Breaks loop if more than max
c = '\n';
printw("\n");
}
} else if (i > 0) // If backspace
pass[i - 1] = '\0';
}
} while (i > PASS_MAX);
clear();
endwin();
return (char*) pass;
}
void rc4_key_exchange(int keySchedule[KEY_SCHEDULE_LENGTH], char* key) {
size_t len = strlen(key);
int j = 0;
for (int i = 0; i < KEY_SCHEDULE_LENGTH; i++)
keySchedule[i] = i;
int temp;
for (int i = 0; i < KEY_SCHEDULE_LENGTH; i++) {
j = (j + keySchedule[i] + key[i % len]) % KEY_SCHEDULE_LENGTH;
temp = keySchedule[i];
keySchedule[i] = keySchedule[j];
keySchedule[j] = temp;
}
}
char* rc4_prga(int keySchedule[KEY_SCHEDULE_LENGTH], size_t len) {
int i = 0, j = 0, temp;
char* output = malloc(len);
pointer_alloc_check(output);
for (int k = 0; k < (int) len; k++) {
i = (i + 1) % KEY_SCHEDULE_LENGTH;
j = (j + keySchedule[i]) % KEY_SCHEDULE_LENGTH;
temp = keySchedule[i];
keySchedule[i] = keySchedule[j];
keySchedule[j] = temp;
output[k] = (unsigned char) keySchedule[(keySchedule[i] + keySchedule[j]) % KEY_SCHEDULE_LENGTH];
}
return output;
}
void rc4_execute(char* output, String* pString) {
for (int i = 0; i < (int) pString->len; i++)
output[i] ^= pString->data[i];
}
char* rc4_encode_string(String* pString, char* password) {
int keySchedule[256];
rc4_key_exchange(keySchedule, password);
char* output = rc4_prga(keySchedule, pString->len);
rc4_execute(output, pString);
return output;
}
String* rc4_get_crypted_string(String* input_pString, char* password, int option) {
String* output_pString = NULL;
int free_pw = (password == NULL);
Json* jobj = json_tokener_parse(input_pString->data);
if (option == ENCRYPT && jobj == NULL) // If trying to encrypt an encrypted portfolio
printf("Your portfolio is already encrypted.\n");
else if (option == DECRYPT && jobj != NULL) // If trying to decrypt an unencrypted portfolio
printf("Your portfolio isn't encrypted.\n");
else {
if (password == NULL) { // Get password if not provided
password = rc4_getPassword();
if (option == ENCRYPT) { // When encrypting, ask to enter pw twice to make sure
printf("You will be asked to enter your password again to make sure the entries match.\n");
sleep(2);
char* passwordCheck = rc4_getPassword();
if (strcmp(password, passwordCheck) != 0) {
printf("Passwords do not match!\n");
free(password);
free(passwordCheck);
json_object_put(jobj);
return NULL;
}
free(passwordCheck);
}
}
output_pString = string_init(); // Set ouput string to en/decrypted data
free(output_pString->data);
output_pString->data = rc4_encode_string(input_pString, password);
output_pString->len = input_pString->len;
if (option == DECRYPT) {
json_object_put(jobj);
// Realloc to add null terminator for json parsing
output_pString->data = realloc(output_pString->data, output_pString->len + 1);
pointer_alloc_check(output_pString->data); // Dealing with len is annoying so it's easier to just realloc
output_pString->data[output_pString->len] = '\0'; // Null terminate string for parsing
jobj = json_tokener_parse(output_pString->data);
if (jobj == NULL) { // If after decrypting, the portfolio is not JSON formatted,
printf("Wrong password!\n"); // then it's the wrong password
free(password);
string_destroy(&output_pString);
return NULL;
}
}
if (free_pw) { // If being called from main
free(password);
if (option == ENCRYPT)
printf("Successfully encrypted your portfolio.\n");
else printf("Successfully decrypted your portfolio.\n");
}
}
json_object_put(jobj);
return output_pString;
}
|