-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCrypto.java
More file actions
121 lines (100 loc) · 3.59 KB
/
Copy pathCrypto.java
File metadata and controls
121 lines (100 loc) · 3.59 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
package com.ilcapo.framework.util;
import java.io.File;
import java.io.IOException;
import java.security.NoSuchAlgorithmException;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import javax.crypto.SecretKey;
public class Crypto {
public Crypto() {
}
/*
* ** OUT-DATED- Dont call mainMenu() method!
*/
public void mainMenu() throws NoSuchAlgorithmException, IOException {
AES aes = new AES();
SecretKey key = null, iv;
String filename, path;
File inputFile;
Scanner scan = new Scanner(System.in);
System.out.println("Action: ");
switch(scan.next()) {
case "keygen":
iv = aes.generateIV();
aes.writeKeyToFile(iv, "iv.dat");
System.out.println("[SUCCESS]You have been generated a new InitVector.");
break;
case "encrypt":
System.out.println("Key to encrypt: ");
key = aes.generateKey(scan.next());
System.out.println("Filename to encrypt: ");
filename = scan.next();
iv = aes.readKeyFromFile("iv.dat");
inputFile = new File(filename);
aes.encrypt(key, iv, inputFile, new File(filename+".dat"));
System.out.println("[SUCCESS]Your file has been encrypted.");
inputFile.delete();
break;
case "decrypt":
System.out.println("Key to decrypt: ");
key = aes.generateKey(scan.next());
System.out.println("Filename to decrypt: ");
filename = scan.next();
iv = aes.readKeyFromFile("iv.dat");
inputFile = new File(filename);
aes.decrypt(key, iv, new File(filename), new File(filename.replace(".dat", "")));
System.out.println("[SUCCESS]Your file has been decrypted.");
inputFile.delete();
break;
case "encryptfolder":
System.out.println("Key to encrypt: ");
key = aes.generateKey(scan.next());
System.out.println("Path to folder to encrypt all files: ");
path = scan.next();
iv = aes.readKeyFromFile("iv.dat");
encryptFiles(path, key, iv, aes, true);
break;
case "decryptfolder":
System.out.println("Key to decrypt: ");
key = aes.generateKey(scan.next());
System.out.println("Path to folder to decrypt all files: ");
path = scan.next();
iv = aes.readKeyFromFile("iv.dat");
decryptFiles(path, key, iv, aes, true);
break;
case "quit":
System.out.print("BB !");
return;
default:
System.err.println("[ERROR] Wrong Syntex.");
break;
}
}
public void encryptFiles(String dirPath, SecretKey key, SecretKey iv, AES aes, boolean deleteFiles) {
//File[] listOfFiles = getFilesInDirectory(dirPath);
DirectoriesCrawler crawler = new DirectoriesCrawler(dirPath);
crawler.scanDirectories();
File[] listOfFiles = crawler.getAllFiles();
for (int i = 0; i < listOfFiles.length; i++) {
File inputFile = listOfFiles[i];
System.out.println("Encrypting: "+inputFile);
aes.encrypt(key, iv, inputFile, new File(inputFile.getPath()+".dat"));
if (deleteFiles) inputFile.delete();
}
}
public void decryptFiles(String dirPath, SecretKey key, SecretKey iv, AES aes, boolean deleteFiles) {
//File[] listOfFiles = getFilesInDirectory(dirPath);
DirectoriesCrawler crawler = new DirectoriesCrawler(dirPath);
crawler.scanDirectories();
File[] listOfFiles = crawler.getAllFiles();
for (int i = 0; i < listOfFiles.length; i++) {
File inputFile = listOfFiles[i];
aes.decrypt(key, iv, inputFile, new File(inputFile.getPath().replace(".dat", "")));
if (aes.isDecryptionKeyBad() == true)
return;
System.out.println("Decrypting: "+inputFile);
if (deleteFiles) inputFile.delete();
}
}
}