[PATCH] media: v4l2: Fix the _MPLANE format check in v4l_enum_fmt()

[Boris Brezillon]

CAP_M2M_MPLANE means the device supports _MPLANE formats for both
capture and output. Adjust the check to avoid EINVAL errors on
such devices.

Fixes: 366c719d6479 ("media: v4l2: Get rid of 
->vidioc_enum_fmt_vid_{cap,out}_mplane")
Reported-by: Maxime Jourdan <mjour...@baylibre.com>
Signed-off-by: Boris Brezillon <boris.brezil...@collabora.com>
---
 drivers/media/v4l2-core/v4l2-ioctl.c | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/drivers/media/v4l2-core/v4l2-ioctl.c 
b/drivers/media/v4l2-core/v4l2-ioctl.c
index b4c73e8f23c5..ace9b9761bed 100644
--- a/drivers/media/v4l2-core/v4l2-ioctl.c
+++ b/drivers/media/v4l2-core/v4l2-ioctl.c
@@ -1385,6 +1385,7 @@ static int v4l_enum_fmt(const struct v4l2_ioctl_ops *ops,
        struct video_device *vdev = video_devdata(file);
        struct v4l2_fmtdesc *p = arg;
        int ret = check_fmt(file, p->type);
+       u32 cap_mask;
 
        if (ret)
                return ret;
@@ -1393,7 +1394,9 @@ static int v4l_enum_fmt(const struct v4l2_ioctl_ops *ops,
        switch (p->type) {
        case V4L2_BUF_TYPE_VIDEO_CAPTURE:
        case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:
-               if (!!(vdev->device_caps & V4L2_CAP_VIDEO_CAPTURE_MPLANE) !=
+               cap_mask = V4L2_CAP_VIDEO_CAPTURE_MPLANE |
+                          V4L2_CAP_VIDEO_M2M_MPLANE;
+               if (!!(vdev->device_caps & cap_mask) !=
                    (p->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE))
                        break;
 
@@ -1408,7 +1411,9 @@ static int v4l_enum_fmt(const struct v4l2_ioctl_ops *ops,
                break;
        case V4L2_BUF_TYPE_VIDEO_OUTPUT:
        case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:
-               if (!!(vdev->device_caps & V4L2_CAP_VIDEO_OUTPUT_MPLANE) !=
+               cap_mask = V4L2_CAP_VIDEO_OUTPUT_MPLANE |
+                          V4L2_CAP_VIDEO_M2M_MPLANE;
+               if (!!(vdev->device_caps & cap_mask) !=
                    (p->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE))
                        break;
 
-- 
2.20.1