Fix types and precompute blob sizes to avoid overflows

pull/21833/head
Jaiwanth 2021-07-19 13:17:19 +05:30
parent ed93bf4c77
commit f07402d234
5 changed files with 31 additions and 21 deletions

View File

@ -179,7 +179,7 @@ export default class MImageBody extends React.Component<IProps, IState> {
}; };
protected getContentUrl(): string { protected getContentUrl(): string {
const content = this.props.mxEvent.getContent(); const content: IMediaEventContent= this.props.mxEvent.getContent();
if (this.props.forExport) return content.url || content.file.url; if (this.props.forExport) return content.url || content.file.url;
const media = mediaFromContent(content); const media = mediaFromContent(content);
if (media.isEncrypted) { if (media.isEncrypted) {

View File

@ -306,13 +306,17 @@ export default class HTMLExporter extends Exporter {
if (this.exportOptions.attachmentsIncluded) { if (this.exportOptions.attachmentsIncluded) {
try { try {
const blob = await this.getMediaBlob(mxEv); const blob = await this.getMediaBlob(mxEv);
if (this.totalSize + blob.size > this.exportOptions.maxSize) {
eventTile = await this.getEventTile(this.createModifiedEvent(this.mediaOmitText, mxEv), joined);
} else {
this.totalSize += blob.size; this.totalSize += blob.size;
const filePath = this.getFilePath(mxEv); const filePath = this.getFilePath(mxEv);
eventTile = await this.getEventTile(mxEv, joined, filePath); eventTile = await this.getEventTile(mxEv, joined, filePath);
if (this.totalSize > this.exportOptions.maxSize - 1024 * 512) { if (this.totalSize == this.exportOptions.maxSize) {
this.exportOptions.attachmentsIncluded = false; this.exportOptions.attachmentsIncluded = false;
} }
this.addFile(filePath, blob); this.addFile(filePath, blob);
}
} catch (e) { } catch (e) {
console.log("Error while fetching file" + e); console.log("Error while fetching file" + e);
eventTile = await this.getEventTile( eventTile = await this.getEventTile(
@ -339,7 +343,7 @@ export default class HTMLExporter extends Exporter {
content += this._wantsDateSeparator(event, prevEvent) ? this.getDateSeparator(event) : ""; content += this._wantsDateSeparator(event, prevEvent) ? this.getDateSeparator(event) : "";
const shouldBeJoined = !this._wantsDateSeparator(event, prevEvent) const shouldBeJoined = !this._wantsDateSeparator(event, prevEvent)
&& shouldFormContinuation(prevEvent, event); && shouldFormContinuation(prevEvent, event, false);
const body = await this.createMessageBody(event, shouldBeJoined); const body = await this.createMessageBody(event, shouldBeJoined);
this.totalSize += Buffer.byteLength(body); this.totalSize += Buffer.byteLength(body);
content += body; content += body;

View File

@ -45,12 +45,14 @@ export default class JSONExporter extends Exporter {
if (this.exportOptions.attachmentsIncluded && this.isAttachment(mxEv)) { if (this.exportOptions.attachmentsIncluded && this.isAttachment(mxEv)) {
try { try {
const blob = await this.getMediaBlob(mxEv); const blob = await this.getMediaBlob(mxEv);
if (this.totalSize + blob.size < this.exportOptions.maxSize) {
this.totalSize += blob.size; this.totalSize += blob.size;
const filePath = this.getFilePath(mxEv); const filePath = this.getFilePath(mxEv);
this.addFile(filePath, blob); if (this.totalSize == this.exportOptions.maxSize) {
if (this.totalSize > this.exportOptions.maxSize - 1024 * 1024) {
this.exportOptions.attachmentsIncluded = false; this.exportOptions.attachmentsIncluded = false;
} }
this.addFile(filePath, blob);
}
} catch (err) { } catch (err) {
console.log("Error fetching file: " + err); console.log("Error fetching file: " + err);
} }

View File

@ -66,13 +66,17 @@ export default class PlainTextExporter extends Exporter {
if (this.exportOptions.attachmentsIncluded) { if (this.exportOptions.attachmentsIncluded) {
try { try {
const blob = await this.getMediaBlob(mxEv); const blob = await this.getMediaBlob(mxEv);
if (this.totalSize + blob.size > this.exportOptions.maxSize) {
mediaText = ` (${this.mediaOmitText})`;
} else {
this.totalSize += blob.size; this.totalSize += blob.size;
const filePath = this.getFilePath(mxEv); const filePath = this.getFilePath(mxEv);
mediaText = " (" + _t("File Attached") + ")"; mediaText = " (" + _t("File Attached") + ")";
this.addFile(filePath, blob); this.addFile(filePath, blob);
if (this.totalSize > this.exportOptions.maxSize - 1024 * 1024) { if (this.totalSize == this.exportOptions.maxSize) {
this.exportOptions.attachmentsIncluded = false; this.exportOptions.attachmentsIncluded = false;
} }
}
} catch (error) { } catch (error) {
mediaText = " (" + _t("Error fetching file") + ")"; mediaText = " (" + _t("Error fetching file") + ")";
console.log("Error fetching file " + error); console.log("Error fetching file " + error);

View File

@ -223,8 +223,8 @@ export default function streamToZIP(underlyingSource: UnderlyingSource) {
function closeZip() { function closeZip() {
let length = 0; let length = 0;
let index = 0; let index = 0;
let indexFilename; let indexFilename: number;
let file; let file: any;
for (indexFilename = 0; indexFilename < filenames.length; indexFilename++) { for (indexFilename = 0; indexFilename < filenames.length; indexFilename++) {
file = files[filenames[indexFilename]]; file = files[filenames[indexFilename]];