zhangyuheng/MCSM-Daemon
1
0
Fork 0
forked from mirror/MCSM-Daemon
Code Pull Requests Activity

Feat: 新增文件解压缩上限限制

Browse Source
This commit is contained in:
Suwings 2022-07-16 18:08:02 +08:00
parent 3522ddf4bd
commit 61f644f7b0
2 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/entity/config.ts
View File

@ -36,6 +36,7 @@ class Config {
public port = 24444; public port = 24444;
public key = builderPassword(); public key = builderPassword();
public maxFileTask = 2; public maxFileTask = 2;
public maxZipFileSize = 60;
} }
// 守护进程配置类 // 守护进程配置类

21
src/service/system_file.ts
View File

@ -24,6 +24,7 @@ import fs from "fs-extra";
import { compress, decompress } from "../common/compress"; import { compress, decompress } from "../common/compress";
import os from "os"; import os from "os";
import iconv from "iconv-lite"; import iconv from "iconv-lite";
import { globalConfiguration } from "../entity/config";
const ERROR_MSG_01 = "非法访问路径"; const ERROR_MSG_01 = "非法访问路径";
const MAX_EDIT_SIZE = 1024 * 1024 * 4; const MAX_EDIT_SIZE = 1024 * 1024 * 4;
@ -158,20 +159,38 @@ export default class FileManager {
await fs.move(targetPath, destPath); await fs.move(targetPath, destPath);
} }
private zipFileCheck(path: string) {
const fileInfo = fs.statSync(path);
const MAX_ZIP_GB = globalConfiguration.config.maxZipFileSize;
if (fileInfo.size > 1024 * 1024 * 1024 * MAX_ZIP_GB)
throw new Error(`文件解压缩只支持最大 ${MAX_ZIP_GB}GB 文件的解压缩，如需改变上限请前往 data/Config/global.json 文件`);
}
async unzip(sourceZip: string, destDir: string, code?: string) { async unzip(sourceZip: string, destDir: string, code?: string) {
if (!code) code = this.fileCode; if (!code) code = this.fileCode;
if (!this.check(sourceZip) || !this.checkPath(destDir)) throw new Error(ERROR_MSG_01); if (!this.check(sourceZip) || !this.checkPath(destDir)) throw new Error(ERROR_MSG_01);
this.zipFileCheck(this.toAbsolutePath(sourceZip));
return await decompress(this.toAbsolutePath(sourceZip), this.toAbsolutePath(destDir), code); return await decompress(this.toAbsolutePath(sourceZip), this.toAbsolutePath(destDir), code);
} }
async zip(sourceZip: string, files: string[], code?: string) { async zip(sourceZip: string, files: string[], code?: string) {
if (!code) code = this.fileCode; if (!code) code = this.fileCode;
if (!this.checkPath(sourceZip)) throw new Error(ERROR_MSG_01); if (!this.checkPath(sourceZip)) throw new Error(ERROR_MSG_01);
const MAX_ZIP_GB = globalConfiguration.config.maxZipFileSize;
const MAX_TOTAL_FIELS_SIZE = 1024 * 1024 * 1024 * MAX_ZIP_GB;
const sourceZipPath = this.toAbsolutePath(sourceZip); const sourceZipPath = this.toAbsolutePath(sourceZip);
const filesPath = []; const filesPath = [];
let totalSize = 0;
for (const iterator of files) { for (const iterator of files) {
if (this.check(iterator)) filesPath.push(this.toAbsolutePath(iterator)); if (this.check(iterator)) {
filesPath.push(this.toAbsolutePath(iterator));
try {
totalSize += fs.statSync(this.toAbsolutePath(iterator))?.size;
} catch (error) {}
} }
}
if (totalSize > MAX_TOTAL_FIELS_SIZE)
throw new Error(`文件解压缩只支持所有文件大小和 ${MAX_ZIP_GB}GB 的压缩，如需改变上限请前往 data/Config/global.json 文件`);
return await compress(sourceZipPath, filesPath, code); return await compress(sourceZipPath, filesPath, code);
} }