文件上传加密/下载解密

￡神魔★判官ぃ 2023-07-10 05:24 8阅读 0赞

# 文件上传加密/下载解密 #

项目中有些文件需要上传服务器，或者上传到ftp，或者minio，而对于一些敏感文件，需要加密上传，下载也只能通过特定的程序下载，否则无法看到文件内容，即使登录服务器或者mino手动下载下来的文件都是无法打开的。以下是一个简单的文件加密方式，可直接测试。

import com.design.mode.service.UploadFile;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.RestController;
    import javax.annotation.Resource;
    import javax.servlet.http.HttpServletResponse;
    /**
    * 通过接口的方式调用
    */
    @RestController
    @RequestMapping("upload")
    public class UploadFileController {
        
    
        @Resource
        private UploadFile uploadFile;
       
        /**
         * 上传文件到服务器
         * @return 返回是否成功信息
         */
        @RequestMapping("file")
        public String testUpload(){
        
            return uploadFile.testUpload();
        }
    
        /**
         * 页面下载文件
         * @param response 响应
         */
        @RequestMapping("download")
        public void testDownload(HttpServletResponse response){
        
            uploadFile.testDownload(response);
        }
    }

import lombok.extern.slf4j.Slf4j;
    import org.apache.commons.io.IOUtils;
    import org.springframework.stereotype.Service;
    import javax.servlet.http.HttpServletResponse;
    import java.io.*;
    import java.nio.charset.StandardCharsets;
    
    @Service
    @Slf4j
    public class UploadFile {
        
    
        /**
         * 加密/解密文件的密钥
         */
        private static final int CRYPTO_SECRET_KEY = 0x99;
    
        /**
         * 测试文件上传加密
         *
         * @return 返回结果
         */
        public String testUpload() {
        
            //准备上传文件
            File srcFile = new File("G:\\test\\src\\SRC.pdf");
            //目标文件，可以是服务器上的文件保存地址
            File dscFile = new File("G:\\test\\dsc\\DSC.pdf");
            return dealFile(srcFile, dscFile);
        }
    
        /**
         * 加密/解密 文件
         *
         * @param srcFile 原文件
         * @param desFile 加密/解密后的文件
         */
        private String dealFile(File srcFile, File desFile) {
        
            int fileData;
            if (!desFile.exists() && desFile.mkdirs()) {
        
                log.error("创建目标文件失败");
                return "加密上传文件失败";
            }
            try (InputStream inputStream = new FileInputStream(srcFile);
                 OutputStream outputStream = new FileOutputStream(desFile)){
        
                while ((fileData = inputStream.read()) > -1) {
        
                    outputStream.write(fileData ^ CRYPTO_SECRET_KEY);
                }
                outputStream.flush();
            }catch (Exception e){
        
                log.error("加密上传文件失败",e);
                return "加密上传文件失败";
            }
            return "加密上传文件成功";
        }
    
        /**
         * 下载加密文件
         * @param response 响应
         */
        public void testDownload(HttpServletResponse response) {
        
            int fileData;
            //要下载的加密文件
            File srcFile = new File("G:\\test\\src\\SRC.pdf");
            OutputStream os = null;
            try (InputStream inputStream = new FileInputStream(srcFile)) {
        
                String fileName = "DSC.pdf";
                response.setContentType("multipart/form-data");
                response.setCharacterEncoding(StandardCharsets.UTF_8.name());
                response.setHeader("Content-Disposition", "attachment; filename=" +
                        java.net.URLEncoder.encode(fileName, StandardCharsets.UTF_8.name()));
                os = response.getOutputStream();
                while ((fileData = inputStream.read()) > -1) {
        
                    os.write(fileData ^ CRYPTO_SECRET_KEY);
                }
                os.flush();
            } catch (Exception e) {
        
                log.error("文件解密下载异常",e);
            } finally {
        
                IOUtils.closeQuietly(os);
            }
        }
    }