JAVA 实现磁盘文件加解密操作的示例代码

时间:2022-06-24 07:39:49

简单实现了下:

?
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
import javax.crypto.BadPaddingException;
import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import java.io.*;
import java.security.GeneralSecurityException;
import java.security.SecureRandom;
 
/**
 * 文件/目录加解密相关
 * @author: Nemo
 * @date: 2019/3/19.
 */
public class FileCrote {
 
  /**
   * 加密后的文件后缀
   */
  private static final String SUBFIX = ".enc";
 
  /**
   * 执行路径
   */
  private static String path = "";
 
  /**
   * 执行模式 1加密 2解密
   */
  private static String mode = "1";
 
  /**
   * 执行密码
   */
  private static String pass = "";
 
  private static String currentFilePath = null;
 
  /**
   * 根据路径加密文件
   * 1、如果是目录,则找它下面的文件进行加密
   * 2、如果是文件,则直接加密
   * @param path
   * @throws IOException
   */
  private static void encrypt(String path) throws IOException, GeneralSecurityException {
    System.out.println("开始处理"+path);
    File file = new File(path);
    if(!file.exists()){
      System.out.println("需加密的路径不存在:"+path);
      return;
    }
    if(file.isDirectory()){
      //目录则遍历其下的文件
      File[] files = file.listFiles();
      if(files == null){
        return;
      }
      for (File subFile : files) {
        encrypt(subFile.getAbsolutePath());
      }
    }else{
      //文件则直接加密
      encrypt(file);
    }
  }
 
  /**
   * 文件加密
   * @param file
   * @throws IOException
   */
  private static void encrypt(File file) throws IOException, GeneralSecurityException {
    String path = file.getAbsolutePath();
    if(path.endsWith(SUBFIX)){
      System.out.println("已加密文件不需再次加密"+path);
      return;
    }
 
    String encFilePath = path + SUBFIX;
    File encFile = new File(encFilePath);
 
    System.out.println("开始加密文件"+path);
    encryptFile(file,encFile,Cipher.ENCRYPT_MODE);
  }
 
  /**
   * 加解密文件操作
   * @param srcFile
   * @param encFile
   * @throws IOException
   */
  private static void encryptFile(File srcFile, File encFile,int mode) throws IOException, GeneralSecurityException {
    if(!srcFile.exists()){
      System.out.println("source file not exixt");
      return;
    }
 
    currentFilePath = srcFile.getAbsolutePath();
 
    //密码
    Cipher cipher = getCipher(mode);
 
    FileInputStream fis = null;
    FileOutputStream fos = null;
    try {
      fis = new FileInputStream(srcFile);
      fos = new FileOutputStream(encFile);
 
      //真正处理
      crypt(fis, fos, cipher);
    } catch (IOException e) {
      e.printStackTrace();
    } finally {
      if (fis != null) {
        fis.close();
      }
      if (fos != null) {
        fos.close();
      }
    }
 
    //源文件删除
    srcFile.delete();
  }
 
  /**
   * 得到加解密Cipher
   * @param mode
   * @return
   * @throws GeneralSecurityException
   */
  private static Cipher getCipher(int mode) throws GeneralSecurityException {
    String type = "AES";
    Cipher cipher = Cipher.getInstance(type+"/ECB/PKCS5Padding");
    KeyGenerator kgen = KeyGenerator.getInstance(type);
    kgen.init(128, new SecureRandom(pass.getBytes()));
    SecretKey key = kgen.generateKey();
    cipher.init(mode,key);
    return cipher;
  }
 
  /**
   * 加密解密流
   * @param in    加密解密前的流
   * @param out    加密解密后的流
   * @param cipher  加密解密
   * @throws IOException
   * @throws GeneralSecurityException
   */
  private static void crypt(InputStream in, OutputStream out, Cipher cipher) throws IOException, GeneralSecurityException {
    int blockSize = cipher.getBlockSize() * 1000;
    int outputSize = cipher.getOutputSize(blockSize);
 
    byte[] inBytes = new byte[blockSize];
    byte[] outBytes = new byte[outputSize];
 
    int inLength = 0;
    boolean more = true;
    while (more) {
      inLength = in.read(inBytes);
      if (inLength == blockSize) {
        int outLength = cipher.update(inBytes, 0, blockSize, outBytes);
        out.write(outBytes, 0, outLength);
      } else {
        more = false;
      }
    }
    if (inLength > 0) {
      outBytes = cipher.doFinal(inBytes, 0, inLength);
    } else {
      outBytes = cipher.doFinal();
    }
    out.write(outBytes);
  }
 
  /**
   * 根据路径解密
   * 1、如果是目录,则找它下面的加密文件进行解密
   * 2、如果是文件,并且它是加密文件,则直接解密
   * @param path
   * @throws IOException
   */
  private static void decrypt(String path) throws IOException, GeneralSecurityException {
    System.out.println("开始处理"+path);
    File file = new File(path);
    if(!file.exists()){
      System.out.println("需解密的路径不存在:"+path);
      return;
    }
    if(file.isDirectory()){
      //目录则遍历其下的文件
      File[] files = file.listFiles();
      if(files == null){
        return;
      }
      for (File subFile : files) {
        decrypt(subFile.getAbsolutePath());
      }
    }else{
      decrypt(file);
    }
  }
 
  /**
   * 文件解密
   * @param file
   * @throws IOException
   */
  private static void decrypt(File file) throws IOException, GeneralSecurityException {
    String path = file.getAbsolutePath();
    if(!path.endsWith(SUBFIX)){
      System.out.println("非加密文件不需解密"+path);
      return;
    }
    System.out.println("开始解密文件"+path);
 
    String newPath = path.substring(0,path.length() - SUBFIX.length());
    encryptFile(file,new File(newPath),Cipher.DECRYPT_MODE);
  }
 
  /**
   * 字符串是否非空
   * @param s
   * @return
   */
  private static boolean isNotEmpty(String s){
    if (s == null || "".equals(s)) {
      return false;
    }
    return true;
  }
 
  /**
   * 开始处理
   * @throws IOException
   * @throws GeneralSecurityException
   */
  private static void deal() throws IOException, GeneralSecurityException {
    while (true) {
      print();
      BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
      String s = reader.readLine();
      if("path".equals(s)) {
        System.out.println("\r\n请输入执行路径\r\n");
        while (true) {
          reader = new BufferedReader(new InputStreamReader(System.in));
          s = reader.readLine();
          if (!isNotEmpty(s)) {
            System.out.println("\r\n请输入执行路径\r\n");
            continue;
          }else {
            File file = new File(s);
            if(!file.exists()){
              System.out.println("\r\n该路径不存在,请重新输入\r\n");
              continue;
            }
          }
          path = s;
          break;
        }
      } else if("pass".equals(s)) {
        System.out.println("\r\n请输入加/解密密码\r\n");
        while (true) {
          reader = new BufferedReader(new InputStreamReader(System.in));
          s = reader.readLine();
          if (!isNotEmpty(s)) {
            System.out.println("\r\n请输入加/解密密码\r\n");
            continue;
          }
          pass = s;
          break;
        }
      } else if ("1".equals(s)) {
        mode = s;
        System.out.println("\r\n当前已选模式:加密\n");
      } else if ("2".equals(s)) {
        mode = s;
        System.out.println("\r\n当前已选模式:解密\r\n");
      }else if("start".equals(s)){
        if(!isNotEmpty(path)) {
          System.out.println("\r\n请输入加/解密密码再开始\r\n");
          continue;
        }
 
        if(!isNotEmpty(pass)) {
          System.out.println("\r\n请输入执行路径后再开始\r\n");
          continue;
        }
 
        if ("1".equals(mode)) {
          encrypt(path);
          System.out.println("\r\n\r\n操作完成\r\n\r\n");
        } else {
          try {
            decrypt(path);
            System.out.println("\r\n\r\n操作完成\r\n\r\n");
          }catch (BadPaddingException e) {
            System.out.println("文件:"+currentFilePath+"解密失败,密码错误");
          }
        }
      }else if("quit".equals(s)){
        System.exit(-1);
      } else {
        System.out.println("\r\n输入错误\r\n");
      }
    }
  }
 
  /**
   * 输出提示语
   */
  private static void print(){
    System.out.println("\r\n" +
        "输入path开始选择执行路径\r\n" +
        "输入pass开始选择加/解密密码\r\n" +
        "输入如下数字以选择处理模式:1 加密 2 解密\r\n" +
        "输入start则开始执行已选操作\r\n" +
        "输入quit则退出程序\r\n" +
        "当前选择路径:"+path+"\r\n" +
        "当前选择模式:"+mode+"\r\n" +
        "当前选择密码:"+pass+"\r\n");
  }
 
  public static void main(String args[]) throws IOException, GeneralSecurityException {
    deal();
  }
 
}

以上就是JAVA 实现磁盘文件加解密操作的示例代码的详细内容,更多关于Java 文件加解密的资料请关注服务器之家其它相关文章!

原文链接:https://www.link-nemo.com/u/10001/post/80269