java实现磁盘分配显示,JAVA 实现磁盘文件加解密操作的示例代码

简单实现了下：

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 文件加解密的资料请关注我们其它相关文章！

时间： 2020-09-09