updateSchema: Be resilient to invalid images

This commit is contained in:
Scott Nonnenberg 2022-07-05 17:28:00 -07:00 committed by GitHub
parent 5fcf97b43b
commit 064f3dd0e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 44 additions and 23 deletions

View File

@ -28,6 +28,7 @@ import { scaleImageToLevel } from '../util/scaleImageToLevel';
import * as GoogleChrome from '../util/GoogleChrome';
import { parseIntOrThrow } from '../util/parseIntOrThrow';
import { getValue } from '../RemoteConfig';
import { isRecord } from '../util/isRecord';
const MAX_WIDTH = 300;
const MAX_HEIGHT = MAX_WIDTH * 1.5;
@ -251,7 +252,7 @@ export function isValid(
// part of re-encoding the image:
export async function autoOrientJPEG(
attachment: AttachmentType,
_: unknown,
{ logger }: { logger: LoggerType },
{
sendHQImages = false,
isIncoming = false,
@ -280,6 +281,7 @@ export async function autoOrientJPEG(
const dataBlob = new Blob([attachment.data], {
type: attachment.contentType,
});
try {
const { blob: xcodedDataBlob } = await scaleImageToLevel(
dataBlob,
attachment.contentType,
@ -300,6 +302,16 @@ export async function autoOrientJPEG(
};
return xcodedAttachment;
} catch (error: unknown) {
const errorString =
isRecord(error) && 'stack' in error ? error.stack : error;
logger.error(
'autoOrientJPEG: Failed to rotate/scale attachment',
errorString
);
return attachment;
}
}
const UNICODE_LEFT_TO_RIGHT_OVERRIDE = '\u202D';

View File

@ -580,9 +580,13 @@ export const processNewAttachment = async (
throw new TypeError('context.logger is required');
}
const rotatedAttachment = await autoOrientJPEG(attachment, undefined, {
const rotatedAttachment = await autoOrientJPEG(
attachment,
{ logger },
{
isIncoming: true,
});
}
);
const onDiskAttachment = await migrateDataToFileSystem(rotatedAttachment, {
writeNewAttachmentData,
});

View File

@ -8,6 +8,7 @@ import { IMAGE_JPEG } from '../types/MIME';
import { canvasToBlob } from './canvasToBlob';
import { getValue } from '../RemoteConfig';
import { parseNumber } from './libphonenumberUtil';
import { isRecord } from './isRecord';
enum MediaQualityLevels {
One = 1,
@ -126,7 +127,11 @@ export async function scaleImageToLevel(
}
({ image } = data);
} catch (err) {
const error = new Error('scaleImageToLevel: Failed to process image');
const errorString = isRecord(err) && 'stack' in err ? err.stack : err;
const error = new Error(
'scaleImageToLevel: Failed to process image',
errorString
);
error.originalError = err;
throw error;
}