您的位置:首頁技術文章
文章詳情頁

JAVA 實現磁盤文件加解密操作的示例代碼

瀏覽:2日期:2022-08-25 09:32:12

簡單實現了下:

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
相關文章:
国产综合久久一区二区三区