您的位置:首页技术文章
文章详情页

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

【字号: 日期:2022-08-25 09:32:12浏览:5作者:猪猪

简单实现了下:

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('rn请输入执行路径rn');while (true) { reader = new BufferedReader(new InputStreamReader(System.in)); s = reader.readLine(); if (!isNotEmpty(s)) { System.out.println('rn请输入执行路径rn'); continue; }else { File file = new File(s); if(!file.exists()){ System.out.println('rn该路径不存在,请重新输入rn'); continue; } } path = s; break;} } else if('pass'.equals(s)) {System.out.println('rn请输入加/解密密码rn');while (true) { reader = new BufferedReader(new InputStreamReader(System.in)); s = reader.readLine(); if (!isNotEmpty(s)) { System.out.println('rn请输入加/解密密码rn'); continue; } pass = s; break;} } else if ('1'.equals(s)) {mode = s;System.out.println('rn当前已选模式:加密n'); } else if ('2'.equals(s)) {mode = s;System.out.println('rn当前已选模式:解密rn'); }else if('start'.equals(s)){if(!isNotEmpty(path)) { System.out.println('rn请输入加/解密密码再开始rn'); continue;}if(!isNotEmpty(pass)) { System.out.println('rn请输入执行路径后再开始rn'); continue;}if ('1'.equals(mode)) { encrypt(path); System.out.println('rnrn操作完成rnrn');} else { try { decrypt(path); System.out.println('rnrn操作完成rnrn'); }catch (BadPaddingException e) { System.out.println('文件:'+currentFilePath+'解密失败,密码错误'); }} }else if('quit'.equals(s)){System.exit(-1); } else {System.out.println('rn输入错误rn'); } } } /** * 输出提示语 */ private static void print(){ System.out.println('rn' +'输入path开始选择执行路径rn' +'输入pass开始选择加/解密密码rn' +'输入如下数字以选择处理模式:1 加密 2 解密rn' +'输入start则开始执行已选操作rn' +'输入quit则退出程序rn' +'当前选择路径:'+path+'rn' +'当前选择模式:'+mode+'rn' +'当前选择密码:'+pass+'rn'); } public static void main(String args[]) throws IOException, GeneralSecurityException { deal(); }}

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

标签: Java
相关文章: