H5离线包下载安全改造

sf-common/src/main/java/com/sf/common/utils/openssl/AES256Util.java

@@ -0,0 +1,109 @@
+package com.sf.common.utils.openssl;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.nio.charset.StandardCharsets;
+
+import javax.crypto.Cipher;
+import javax.crypto.CipherInputStream;
+import javax.crypto.CipherOutputStream;
+import javax.crypto.spec.IvParameterSpec;
+import javax.crypto.spec.SecretKeySpec;
+
+public class AES256Util {
+    private static final String ALGORITHM = "AES/ECB/PKCS5Padding";
+
+    public static Cipher initAESCipher(String key, int cipherMode) throws Exception {
+        Cipher cipher = Cipher.getInstance(ALGORITHM);
+        SecretKeySpec secretKeySpec = new SecretKeySpec(key.getBytes(StandardCharsets.UTF_8), "AES");
+        cipher.init(cipherMode, secretKeySpec);
+
+        return cipher;
+    }
+
+    /**
+     * 加密文件
+     *
+     * @return
+     */
+    public static InputStream encryptFile(String key, InputStream inputStream) throws Exception {
+        Cipher cipher = initAESCipher(key, Cipher.ENCRYPT_MODE);
+        CipherInputStream cipherInputStream = new CipherInputStream(inputStream, cipher);
+
+        return cipherInputStream;
+    }
+
+    /**
+     * 解密文件
+     *
+     * @param encryptedFile
+     * @param sourceFile
+     */
+    public static String decryptFile(String key, File encryptedFile, File sourceFile) {
+        InputStream inputStream = null;
+        OutputStream outputStream = null;
+        CipherOutputStream cipherOutputStream = null;
+        try {
+            inputStream = new FileInputStream(encryptedFile);
+            outputStream = new FileOutputStream(sourceFile);
+            Cipher cipher = initAESCipher(key, Cipher.DECRYPT_MODE);
+            cipherOutputStream = new CipherOutputStream(outputStream, cipher);
+            byte[] buffer = new byte[1024];
+            int n = -1;
+            int count = 0;
+            while ((n = inputStream.read(buffer)) != -1) {
+                count += n;
+                cipherOutputStream.write(buffer, 0, n);
+                cipherOutputStream.flush();
+            }
+
+        } catch (Exception e) {
+            return e.getLocalizedMessage();
+        }
+        if (cipherOutputStream != null) {
+            try {
+                cipherOutputStream.close();
+            } catch (IOException e) {
+                return e.getLocalizedMessage();
+            }
+        }
+        if (outputStream != null) {
+            try {
+                outputStream.close();
+            } catch (IOException e) {
+                return e.getLocalizedMessage();
+            }
+        }
+        if (inputStream != null) {
+            try {
+                inputStream.close();
+            } catch (IOException e) {
+                return e.getLocalizedMessage();
+            }
+        }
+
+        return "";
+    }
+
+//    public static void main(String[] args) throws Exception {
+//        File sourceFile = new File("/Users/a1234/Downloads/dist.zip");
+//        InputStream encryptInputStream = encryptFile("efghijklmnopqrstuvwxyz0123456789", new FileInputStream(sourceFile));
+//        OutputStream outputStream = new FileOutputStream(new File("/Users/a1234/Downloads/x.zip"));
+//        byte[] buffer = new byte[1024];
+//        int n = -1;
+//        int count = 0;
+//        while ((n = encryptInputStream.read(buffer)) != -1) {
+//            count += n;
+//            outputStream.write(buffer, 0, n);
+//        }
+//        outputStream.flush();
+//        System.out.println(count);
+//
+//        decryptFile("efghijklmnopqrstuvwxyz0123456789", new File("/Users/a1234/Downloads/x.zip"), new File("/Users/a1234/Downloads/y.zip"));
+//    }
+}
+

sf-system/src/main/java/com/sf/system/deployment/controller/DeploymentModuleListController.java

@@ -1,14 +1,15 @@
 package com.sf.system.deployment.controller;
 
+import java.io.InputStream;
+import java.net.URL;
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
-
 import com.sf.common.enums.RequestHeaderEnums;
-import com.sf.common.utils.StringUtils;
 import com.sf.common.utils.http.RequestUtils;
 import com.sf.system.deployment.domain.rqs.ModuleListByCoreRequest;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.util.FileCopyUtils;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.PutMapping;
@@ -28,14 +29,13 @@ import com.sf.common.core.page.TableDataInfo;
 
 /**
  * H5模块包Controller
- * 
+ *
  * @author ztzh
  * @date 2024-04-22
  */
 @RestController
 @RequestMapping("/deployment/module")
-public class DeploymentModuleListController extends BaseController
+public class DeploymentModuleListController extends BaseController {
-{
     @Autowired
     private IDeploymentModuleListService deploymentModuleListService;
 
@@ -44,19 +44,18 @@ public class DeploymentModuleListController extends BaseController
      */
     @PreAuthorize("@ss.hasPermi('deployment:module:list')")
     @GetMapping("/list")
-    public TableDataInfo list(DeploymentModuleList deploymentModuleList)
+    public TableDataInfo list(DeploymentModuleList deploymentModuleList) {
-    {
         deploymentModuleList.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode()));
         startPage();
         List<DeploymentModuleList> list = deploymentModuleListService.selectDeploymentModuleListList(deploymentModuleList);
         return getDataTable(list);
     }
+
     /**
      * 查询H5模块包列表
      */
     @GetMapping("/qurey/list")
-    public AjaxResult queryByCore(ModuleListByCoreRequest request)
+    public AjaxResult queryByCore(ModuleListByCoreRequest request) {
-    {
         DeploymentModuleList deploymentModuleList = new DeploymentModuleList();
         deploymentModuleList.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode()));
         deploymentModuleList.setSysType(request.getSysType());
@@ -64,14 +63,36 @@ public class DeploymentModuleListController extends BaseController
         return success(list);
     }
 
+    /**
+     * 查询模块包列表
+     */
+    @GetMapping("/qurey/file")
+    public AjaxResult queryByCore(ModuleListByCoreRequest request, HttpServletResponse response) {
+        DeploymentModuleList deploymentModuleList = new DeploymentModuleList();
+        deploymentModuleList.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode()));
+        deploymentModuleList.setSysType(request.getSysType());
+        deploymentModuleList.setModuleType(request.getModuleType());
+        List<DeploymentModuleList> list = deploymentModuleListService.selectDeploymentModuleListList(deploymentModuleList);
+        if (list.isEmpty() && null == list.get(0)) {
+            throw new SecurityException("无可用模块包");
+        }
+
+        try {
+            InputStream inputStream = new URL(list.get(0).getModuleUrl()).openStream();
+            FileCopyUtils.copy(inputStream, response.getOutputStream());
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+        return success(null);
+    }
+
     /**
      * 导出H5模块包列表
      */
     @PreAuthorize("@ss.hasPermi('deployment:module:export')")
     @Log(title = "H5模块包", businessType = BusinessType.EXPORT)
     @PostMapping("/export")
-    public void export(HttpServletResponse response, DeploymentModuleList deploymentModuleList)
+    public void export(HttpServletResponse response, DeploymentModuleList deploymentModuleList) {
-    {
         List<DeploymentModuleList> list = deploymentModuleListService.selectDeploymentModuleListList(deploymentModuleList);
         ExcelUtil<DeploymentModuleList> util = new ExcelUtil<DeploymentModuleList>(DeploymentModuleList.class);
         util.exportExcel(response, list, "H5模块包数据");
@@ -82,8 +103,7 @@ public class DeploymentModuleListController extends BaseController
      */
     @PreAuthorize("@ss.hasPermi('deployment:module:query')")
     @GetMapping(value = "/{id}")
-    public AjaxResult getInfo(@PathVariable("id") Long id)
+    public AjaxResult getInfo(@PathVariable("id") Long id) {
-    {
         return success(deploymentModuleListService.selectDeploymentModuleListById(id));
     }
 
@@ -93,8 +113,7 @@ public class DeploymentModuleListController extends BaseController
     @PreAuthorize("@ss.hasPermi('deployment:module:add')")
     @Log(title = "H5模块包", businessType = BusinessType.INSERT)
     @PostMapping
-    public AjaxResult add(@RequestBody DeploymentModuleList deploymentModuleList)
+    public AjaxResult add(@RequestBody DeploymentModuleList deploymentModuleList) {
-    {
         deploymentModuleList.setAppCode(RequestUtils.getHeader(RequestHeaderEnums.APP_CODE.getCode()));
         return toAjax(deploymentModuleListService.insertDeploymentModuleList(deploymentModuleList));
     }
@@ -105,8 +124,7 @@ public class DeploymentModuleListController extends BaseController
     @PreAuthorize("@ss.hasPermi('deployment:module:edit')")
     @Log(title = "H5模块包", businessType = BusinessType.UPDATE)
     @PutMapping
-    public AjaxResult edit(@RequestBody DeploymentModuleList deploymentModuleList)
+    public AjaxResult edit(@RequestBody DeploymentModuleList deploymentModuleList) {
-    {
         return toAjax(deploymentModuleListService.updateDeploymentModuleList(deploymentModuleList));
     }
 
@@ -115,9 +133,8 @@ public class DeploymentModuleListController extends BaseController
      */
     @PreAuthorize("@ss.hasPermi('deployment:module:remove')")
     @Log(title = "H5模块包", businessType = BusinessType.DELETE)
-	@DeleteMapping("/{ids}")
+    @DeleteMapping("/{ids}")
-    public AjaxResult remove(@PathVariable Long[] ids)
+    public AjaxResult remove(@PathVariable Long[] ids) {
-    {
         return toAjax(deploymentModuleListService.deleteDeploymentModuleListByIds(ids));
     }
 }

sf-system/src/main/java/com/sf/system/deployment/domain/rqs/ModuleListByCoreRequest.java

@@ -13,4 +13,7 @@ public class ModuleListByCoreRequest {
 
     @NotBlank
     private String sysType;
+
+    @NotBlank
+    private String moduleType;
 }