wrapper/FFmpeg
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

libavutil/hwcontext_amf: add format validation in transfer_data functions

Browse source
This commit is contained in:
Dmitrii Ovchinnikov 2025-03-12 16:15:53 +01:00
parent e5d62e20c8
commit 5b460bde8b
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
libavutil/hwcontext_amf.c
View file

@ -277,6 +277,9 @@ static int amf_transfer_data_to(AVHWFramesContext *ctx, AVFrame *dst,
int w = FFMIN(dst->width, src->width);
int h = FFMIN(dst->height, src->height);
if (dst->hw_frames_ctx->data != (uint8_t *)ctx || src->format != ctx->sw_format)
return AVERROR(EINVAL);
if (!surface) {
AVHWDeviceContext *hwdev_ctx = ctx->device_ctx;
AVAMFDeviceContext *amf_device_ctx = (AVAMFDeviceContext *)hwdev_ctx->hwctx;
@ -319,6 +322,9 @@ static int amf_transfer_data_from(AVHWFramesContext *ctx, AVFrame *dst,
int h = FFMIN(dst->height, src->height);
int ret;
if (src->hw_frames_ctx->data != (uint8_t *)ctx || dst->format != ctx->sw_format)
return AVERROR(EINVAL);
ret = surface->pVtbl->Convert(surface, AMF_MEMORY_HOST);
AMF_RETURN_IF_FALSE(ctx, ret == AMF_OK, AVERROR_UNKNOWN, "Convert(amf::AMF_MEMORY_HOST) failed with error %d\n", AVERROR_UNKNOWN);