RageStudio/linebridge
1
0
Fork 0
mirror of https://github.com/ragestudio/linebridge.git synced 2025-06-09 10:34:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

improved mimetype check method

Browse Source
This commit is contained in:
srgooglo 2021-07-14 12:11:41 +02:00
parent 4dd10a538e
commit 7bdb555cd7
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/classes/Storage/drivers/filesystem/index.js
View File

@ -12,13 +12,16 @@ class FilesystemDriver {
this.defaultWriteMode = "0777" this.defaultWriteMode = "0777"
this.defaultWriteFlags = 'w+' this.defaultWriteFlags = 'w+'
this.matchMimetypes = String() this.strictMimetypeCheck = this.strictMimetypeCheck ?? false
this.checkMimetypes = this.params.checkMimetypes ?? false
this.mimetypes = String()
if (Array.isArray(this.params.allowedMimetypes)) { if (Array.isArray(this.params.allowedMimetypes)) {
this.checkMimetypes = true
this.params.allowedMimetypes.forEach((type, index) => { this.params.allowedMimetypes.forEach((type, index) => {
if (index == 0) { if (index == 0) {
return this.matchMimetypes = `${type}.*` return this.mimetypes = `${type}.*`
} }
return this.matchMimetypes = `${this.matchMimetypes}|${type}.*` return this.mimetypes = `${this.mimetypes}|${type}.*`
}) })
} }
@ -33,7 +36,10 @@ class FilesystemDriver {
} }
checkMimetype(type) { checkMimetype(type) {
return type.match(this.matchMimetypes) if (!this.checkMimetypes) {
return true
}
return type.match(this.mimetypes)
} }
stream = (dir, options) => { stream = (dir, options) => {
@ -49,12 +55,18 @@ class FilesystemDriver {
fs.mkdirSync(fileParent) fs.mkdirSync(fileParent)
} }
const validMIME = this.checkMimetype(file.mimetype) ? true : false if (typeof file.mimetype === "undefined" && this.strictMimetypeCheck) {
throw new Error(`Missing mimetype`)
if (!validMIME) {
throw new Error(`Invalid mimetype [${file.mimetype}]`)
} }
if (typeof file.mimetype !== "undefined") {
const validMIME = this.checkMimetype(file.mimetype) ? true : false
if (!validMIME && this.strictMimetypeCheck) {
throw new Error(`Invalid mimetype [${file.mimetype}]`)
}
}
const stream = this.stream(filePath, { mode: options?.mode ?? this.defaultWriteMode, flags: options?.flags ?? this.defaultWriteFlags }) const stream = this.stream(filePath, { mode: options?.mode ?? this.defaultWriteMode, flags: options?.flags ?? this.defaultWriteFlags })
stream.on("ready", () => { stream.on("ready", () => {