#include <xf86drm.h>
#include <xf86drmMode.h>
+// Required for compatibility when building against libdrm < 2.4.83.
+#ifndef DRM_FORMAT_MOD_INVALID
+#define DRM_FORMAT_MOD_INVALID ((1ULL << 56) - 1)
+#endif
+
#include "libavutil/hwcontext.h"
#include "libavutil/hwcontext_drm.h"
#include "libavutil/internal.h"
AVBufferRef *device_ref;
AVHWDeviceContext *device;
AVDRMDeviceContext *hwctx;
+ int fb2_available;
AVBufferRef *frames_ref;
AVHWFramesContext *frames;
static void kmsgrab_free_desc(void *opaque, uint8_t *data)
{
AVDRMFrameDescriptor *desc = (AVDRMFrameDescriptor*)data;
+ int i;
- close(desc->objects[0].fd);
+ for (i = 0; i < desc->nb_objects; i++)
+ close(desc->objects[i].fd);
av_free(desc);
}
av_frame_free(&frame);
}
-static int kmsgrab_read_packet(AVFormatContext *avctx, AVPacket *pkt)
+static int kmsgrab_get_fb(AVFormatContext *avctx,
+ drmModePlane *plane,
+ AVDRMFrameDescriptor *desc)
{
KMSGrabContext *ctx = avctx->priv_data;
- drmModePlane *plane;
- drmModeFB *fb;
- AVDRMFrameDescriptor *desc;
- AVFrame *frame;
- int64_t now;
+ drmModeFB *fb = NULL;
int err, fd;
- now = av_gettime();
- if (ctx->frame_last) {
- int64_t delay;
- while (1) {
- delay = ctx->frame_last + ctx->frame_delay - now;
- if (delay <= 0)
- break;
- av_usleep(delay);
- now = av_gettime();
- }
- }
- ctx->frame_last = now;
-
- plane = drmModeGetPlane(ctx->hwctx->fd, ctx->plane_id);
- if (!plane) {
- av_log(avctx, AV_LOG_ERROR, "Failed to get plane "
- "%"PRIu32".\n", ctx->plane_id);
- return AVERROR(EIO);
- }
- if (!plane->fb_id) {
- av_log(avctx, AV_LOG_ERROR, "Plane %"PRIu32" no longer has "
- "an associated framebuffer.\n", ctx->plane_id);
- return AVERROR(EIO);
- }
-
fb = drmModeGetFB(ctx->hwctx->fd, plane->fb_id);
if (!fb) {
+ err = errno;
av_log(avctx, AV_LOG_ERROR, "Failed to get framebuffer "
- "%"PRIu32".\n", plane->fb_id);
- return AVERROR(EIO);
+ "%"PRIu32": %s.\n", plane->fb_id, strerror(err));
+ err = AVERROR(err);
+ goto fail;
}
if (fb->width != ctx->width || fb->height != ctx->height) {
av_log(avctx, AV_LOG_ERROR, "Plane %"PRIu32" framebuffer "
"dimensions changed: now %"PRIu32"x%"PRIu32".\n",
ctx->plane_id, fb->width, fb->height);
- return AVERROR(EIO);
+ err = AVERROR(EIO);
+ goto fail;
}
if (!fb->handle) {
av_log(avctx, AV_LOG_ERROR, "No handle set on framebuffer.\n");
- return AVERROR(EIO);
+ err = AVERROR(EIO);
+ goto fail;
}
err = drmPrimeHandleToFD(ctx->hwctx->fd, fb->handle, O_RDONLY, &fd);
if (err < 0) {
err = errno;
av_log(avctx, AV_LOG_ERROR, "Failed to get PRIME fd from "
- "framebuffer handle: %s.\n", strerror(errno));
- return AVERROR(err);
+ "framebuffer handle: %s.\n", strerror(err));
+ err = AVERROR(err);
+ goto fail;
}
- desc = av_mallocz(sizeof(*desc));
- if (!desc)
- return AVERROR(ENOMEM);
-
*desc = (AVDRMFrameDescriptor) {
.nb_objects = 1,
.objects[0] = {
},
};
+ err = 0;
+fail:
+ drmModeFreeFB(fb);
+ return err;
+}
+
+#if HAVE_LIBDRM_GETFB2
+static int kmsgrab_get_fb2(AVFormatContext *avctx,
+ drmModePlane *plane,
+ AVDRMFrameDescriptor *desc)
+{
+ KMSGrabContext *ctx = avctx->priv_data;
+ drmModeFB2 *fb;
+ int err, i, nb_objects;
+ uint64_t modifier = ctx->drm_format_modifier;
+
+ fb = drmModeGetFB2(ctx->hwctx->fd, plane->fb_id);
+ if (!fb) {
+ err = errno;
+ av_log(avctx, AV_LOG_ERROR, "Failed to get framebuffer "
+ "%"PRIu32": %s.\n", plane->fb_id, strerror(err));
+ return AVERROR(err);
+ }
+ if (fb->pixel_format != ctx->drm_format) {
+ av_log(avctx, AV_LOG_ERROR, "Plane %"PRIu32" framebuffer "
+ "format changed: now %"PRIx32".\n",
+ ctx->plane_id, fb->pixel_format);
+ err = AVERROR(EIO);
+ goto fail;
+ }
+ if (fb->width != ctx->width || fb->height != ctx->height) {
+ av_log(avctx, AV_LOG_ERROR, "Plane %"PRIu32" framebuffer "
+ "dimensions changed: now %"PRIu32"x%"PRIu32".\n",
+ ctx->plane_id, fb->width, fb->height);
+ err = AVERROR(EIO);
+ goto fail;
+ }
+ if (!fb->handles[0]) {
+ av_log(avctx, AV_LOG_ERROR, "No handle set on framebuffer.\n");
+ err = AVERROR(EIO);
+ goto fail;
+ }
+
+ if (fb->flags & DRM_MODE_FB_MODIFIERS)
+ modifier = fb->modifier;
+
+ *desc = (AVDRMFrameDescriptor) {
+ .nb_layers = 1,
+ .layers[0] = {
+ .format = ctx->drm_format,
+ },
+ };
+
+ nb_objects = 0;
+ for (i = 0; i < 4 && fb->handles[i]; i++) {
+ size_t size;
+ int dup = 0, j, obj;
+
+ size = fb->offsets[i] + fb->height * fb->pitches[i];
+
+ for (j = 0; j < i; j++) {
+ if (fb->handles[i] == fb->handles[j]) {
+ dup = 1;
+ break;
+ }
+ }
+ if (dup) {
+ obj = desc->layers[0].planes[j].object_index;
+
+ if (desc->objects[j].size < size)
+ desc->objects[j].size = size;
+
+ desc->layers[0].planes[i] = (AVDRMPlaneDescriptor) {
+ .object_index = obj,
+ .offset = fb->offsets[i],
+ .pitch = fb->pitches[i],
+ };
+
+ } else {
+ int fd;
+ err = drmPrimeHandleToFD(ctx->hwctx->fd, fb->handles[i],
+ O_RDONLY, &fd);
+ if (err < 0) {
+ err = errno;
+ av_log(avctx, AV_LOG_ERROR, "Failed to get PRIME fd from "
+ "framebuffer handle: %s.\n", strerror(err));
+ err = AVERROR(err);
+ goto fail;
+ }
+
+ obj = nb_objects++;
+ desc->objects[obj] = (AVDRMObjectDescriptor) {
+ .fd = fd,
+ .size = size,
+ .format_modifier = modifier,
+ };
+ desc->layers[0].planes[i] = (AVDRMPlaneDescriptor) {
+ .object_index = obj,
+ .offset = fb->offsets[i],
+ .pitch = fb->pitches[i],
+ };
+ }
+ }
+ desc->nb_objects = nb_objects;
+ desc->layers[0].nb_planes = i;
+
+ err = 0;
+fail:
+ drmModeFreeFB2(fb);
+ return err;
+}
+#endif
+
+static int kmsgrab_read_packet(AVFormatContext *avctx, AVPacket *pkt)
+{
+ KMSGrabContext *ctx = avctx->priv_data;
+ drmModePlane *plane = NULL;
+ AVDRMFrameDescriptor *desc = NULL;
+ AVFrame *frame = NULL;
+ int64_t now;
+ int err;
+
+ now = av_gettime_relative();
+ if (ctx->frame_last) {
+ int64_t delay;
+ while (1) {
+ delay = ctx->frame_last + ctx->frame_delay - now;
+ if (delay <= 0)
+ break;
+ av_usleep(delay);
+ now = av_gettime_relative();
+ }
+ }
+ ctx->frame_last = now;
+ now = av_gettime();
+
+ plane = drmModeGetPlane(ctx->hwctx->fd, ctx->plane_id);
+ if (!plane) {
+ err = errno;
+ av_log(avctx, AV_LOG_ERROR, "Failed to get plane "
+ "%"PRIu32": %s.\n", ctx->plane_id, strerror(err));
+ err = AVERROR(err);
+ goto fail;
+ }
+ if (!plane->fb_id) {
+ av_log(avctx, AV_LOG_ERROR, "Plane %"PRIu32" no longer has "
+ "an associated framebuffer.\n", ctx->plane_id);
+ err = AVERROR(EIO);
+ goto fail;
+ }
+
+ desc = av_mallocz(sizeof(*desc));
+ if (!desc) {
+ err = AVERROR(ENOMEM);
+ goto fail;
+ }
+
+#if HAVE_LIBDRM_GETFB2
+ if (ctx->fb2_available)
+ err = kmsgrab_get_fb2(avctx, plane, desc);
+ else
+#endif
+ err = kmsgrab_get_fb(avctx, plane, desc);
+ if (err < 0)
+ goto fail;
+
frame = av_frame_alloc();
- if (!frame)
- return AVERROR(ENOMEM);
+ if (!frame) {
+ err = AVERROR(ENOMEM);
+ goto fail;
+ }
frame->hw_frames_ctx = av_buffer_ref(ctx->frames_ref);
- if (!frame->hw_frames_ctx)
- return AVERROR(ENOMEM);
+ if (!frame->hw_frames_ctx) {
+ err = AVERROR(ENOMEM);
+ goto fail;
+ }
frame->buf[0] = av_buffer_create((uint8_t*)desc, sizeof(*desc),
&kmsgrab_free_desc, avctx, 0);
- if (!frame->buf[0])
- return AVERROR(ENOMEM);
+ if (!frame->buf[0]) {
+ err = AVERROR(ENOMEM);
+ goto fail;
+ }
frame->data[0] = (uint8_t*)desc;
frame->format = AV_PIX_FMT_DRM_PRIME;
- frame->width = fb->width;
- frame->height = fb->height;
+ frame->width = ctx->width;
+ frame->height = ctx->height;
- drmModeFreeFB(fb);
drmModeFreePlane(plane);
+ plane = NULL;
+ desc = NULL;
pkt->buf = av_buffer_create((uint8_t*)frame, sizeof(*frame),
&kmsgrab_free_frame, avctx, 0);
- if (!pkt->buf)
- return AVERROR(ENOMEM);
+ if (!pkt->buf) {
+ err = AVERROR(ENOMEM);
+ goto fail;
+ }
pkt->data = (uint8_t*)frame;
pkt->size = sizeof(*frame);
pkt->flags |= AV_PKT_FLAG_TRUSTED;
return 0;
+
+fail:
+ drmModeFreePlane(plane);
+ av_freep(&desc);
+ av_frame_free(&frame);
+ return err;
}
static const struct {
enum AVPixelFormat pixfmt;
uint32_t drm_format;
} kmsgrab_formats[] = {
+ // Monochrome.
#ifdef DRM_FORMAT_R8
{ AV_PIX_FMT_GRAY8, DRM_FORMAT_R8 },
#endif
{ AV_PIX_FMT_GRAY16LE, DRM_FORMAT_R16 },
{ AV_PIX_FMT_GRAY16BE, DRM_FORMAT_R16 | DRM_FORMAT_BIG_ENDIAN },
#endif
+ // <8-bit RGB.
{ AV_PIX_FMT_BGR8, DRM_FORMAT_BGR233 },
{ AV_PIX_FMT_RGB555LE, DRM_FORMAT_XRGB1555 },
{ AV_PIX_FMT_RGB555BE, DRM_FORMAT_XRGB1555 | DRM_FORMAT_BIG_ENDIAN },
{ AV_PIX_FMT_RGB565BE, DRM_FORMAT_RGB565 | DRM_FORMAT_BIG_ENDIAN },
{ AV_PIX_FMT_BGR565LE, DRM_FORMAT_BGR565 },
{ AV_PIX_FMT_BGR565BE, DRM_FORMAT_BGR565 | DRM_FORMAT_BIG_ENDIAN },
+ // 8-bit RGB.
{ AV_PIX_FMT_RGB24, DRM_FORMAT_RGB888 },
{ AV_PIX_FMT_BGR24, DRM_FORMAT_BGR888 },
{ AV_PIX_FMT_0RGB, DRM_FORMAT_BGRX8888 },
{ AV_PIX_FMT_ABGR, DRM_FORMAT_RGBA8888 },
{ AV_PIX_FMT_RGBA, DRM_FORMAT_ABGR8888 },
{ AV_PIX_FMT_BGRA, DRM_FORMAT_ARGB8888 },
+ // 10-bit RGB.
+ { AV_PIX_FMT_X2RGB10LE, DRM_FORMAT_XRGB2101010 },
+ { AV_PIX_FMT_X2RGB10BE, DRM_FORMAT_XRGB2101010 | DRM_FORMAT_BIG_ENDIAN },
+ // 8-bit YUV 4:2:0.
+ { AV_PIX_FMT_NV12, DRM_FORMAT_NV12 },
+ // 8-bit YUV 4:2:2.
{ AV_PIX_FMT_YUYV422, DRM_FORMAT_YUYV },
{ AV_PIX_FMT_YVYU422, DRM_FORMAT_YVYU },
{ AV_PIX_FMT_UYVY422, DRM_FORMAT_UYVY },
drmModePlaneRes *plane_res = NULL;
drmModePlane *plane = NULL;
drmModeFB *fb = NULL;
+#if HAVE_LIBDRM_GETFB2
+ drmModeFB2 *fb2 = NULL;
+#endif
AVStream *stream;
int err, i;
- for (i = 0; i < FF_ARRAY_ELEMS(kmsgrab_formats); i++) {
- if (kmsgrab_formats[i].pixfmt == ctx->format) {
- ctx->drm_format = kmsgrab_formats[i].drm_format;
- break;
- }
- }
- if (i >= FF_ARRAY_ELEMS(kmsgrab_formats)) {
- av_log(avctx, AV_LOG_ERROR, "Unsupported format %s.\n",
- av_get_pix_fmt_name(ctx->format));
- return AVERROR(EINVAL);
- }
-
err = av_hwdevice_ctx_create(&ctx->device_ref, AV_HWDEVICE_TYPE_DRM,
ctx->device_path, NULL, 0);
if (err < 0) {
} else {
plane_res = drmModeGetPlaneResources(ctx->hwctx->fd);
if (!plane_res) {
+ err = errno;
av_log(avctx, AV_LOG_ERROR, "Failed to get plane "
- "resources: %s.\n", strerror(errno));
- err = AVERROR(EINVAL);
+ "resources: %s.\n", strerror(err));
+ err = AVERROR(err);
goto fail;
}
ctx->plane_id = plane->plane_id;
- fb = drmModeGetFB(ctx->hwctx->fd, plane->fb_id);
- if (!fb) {
+#if HAVE_LIBDRM_GETFB2
+ fb2 = drmModeGetFB2(ctx->hwctx->fd, plane->fb_id);
+ if (!fb2 && errno == ENOSYS) {
+ av_log(avctx, AV_LOG_INFO, "GETFB2 not supported, "
+ "will try to use GETFB instead.\n");
+ } else if (!fb2) {
err = errno;
av_log(avctx, AV_LOG_ERROR, "Failed to get "
"framebuffer %"PRIu32": %s.\n",
plane->fb_id, strerror(err));
err = AVERROR(err);
goto fail;
+ } else {
+ av_log(avctx, AV_LOG_INFO, "Template framebuffer is "
+ "%"PRIu32": %"PRIu32"x%"PRIu32" "
+ "format %"PRIx32" modifier %"PRIx64" flags %"PRIx32".\n",
+ fb2->fb_id, fb2->width, fb2->height,
+ fb2->pixel_format, fb2->modifier, fb2->flags);
+
+ ctx->width = fb2->width;
+ ctx->height = fb2->height;
+
+ if (!fb2->handles[0]) {
+ av_log(avctx, AV_LOG_ERROR, "No handle set on framebuffer: "
+ "maybe you need some additional capabilities?\n");
+ err = AVERROR(EINVAL);
+ goto fail;
+ }
+
+ for (i = 0; i < FF_ARRAY_ELEMS(kmsgrab_formats); i++) {
+ if (kmsgrab_formats[i].drm_format == fb2->pixel_format) {
+ if (ctx->format != AV_PIX_FMT_NONE &&
+ ctx->format != kmsgrab_formats[i].pixfmt) {
+ av_log(avctx, AV_LOG_ERROR, "Framebuffer pixel format "
+ "%"PRIx32" does not match expected format.\n",
+ fb2->pixel_format);
+ err = AVERROR(EINVAL);
+ goto fail;
+ }
+ ctx->drm_format = fb2->pixel_format;
+ ctx->format = kmsgrab_formats[i].pixfmt;
+ break;
+ }
+ }
+ if (i == FF_ARRAY_ELEMS(kmsgrab_formats)) {
+ av_log(avctx, AV_LOG_ERROR, "Framebuffer pixel format "
+ "%"PRIx32" is not a known supported format.\n",
+ fb2->pixel_format);
+ err = AVERROR(EINVAL);
+ goto fail;
+ }
+
+ if (fb2->flags & DRM_MODE_FB_MODIFIERS) {
+ if (ctx->drm_format_modifier != DRM_FORMAT_MOD_INVALID &&
+ ctx->drm_format_modifier != fb2->modifier) {
+ av_log(avctx, AV_LOG_ERROR, "Framebuffer format modifier "
+ "%"PRIx64" does not match expected modifier.\n",
+ fb2->modifier);
+ err = AVERROR(EINVAL);
+ goto fail;
+ } else {
+ ctx->drm_format_modifier = fb2->modifier;
+ }
+ }
+ av_log(avctx, AV_LOG_VERBOSE, "Format is %s, from "
+ "DRM format %"PRIx32" modifier %"PRIx64".\n",
+ av_get_pix_fmt_name(ctx->format),
+ ctx->drm_format, ctx->drm_format_modifier);
+
+ ctx->fb2_available = 1;
}
+#endif
+
+ if (!ctx->fb2_available) {
+ if (ctx->format == AV_PIX_FMT_NONE) {
+ // Backward compatibility: assume BGR0 if no format supplied.
+ ctx->format = AV_PIX_FMT_BGR0;
+ }
+ for (i = 0; i < FF_ARRAY_ELEMS(kmsgrab_formats); i++) {
+ if (kmsgrab_formats[i].pixfmt == ctx->format) {
+ ctx->drm_format = kmsgrab_formats[i].drm_format;
+ break;
+ }
+ }
+ if (i >= FF_ARRAY_ELEMS(kmsgrab_formats)) {
+ av_log(avctx, AV_LOG_ERROR, "Unsupported format %s.\n",
+ av_get_pix_fmt_name(ctx->format));
+ return AVERROR(EINVAL);
+ }
+
+ fb = drmModeGetFB(ctx->hwctx->fd, plane->fb_id);
+ if (!fb) {
+ err = errno;
+ av_log(avctx, AV_LOG_ERROR, "Failed to get "
+ "framebuffer %"PRIu32": %s.\n",
+ plane->fb_id, strerror(err));
+ err = AVERROR(err);
+ goto fail;
+ }
- av_log(avctx, AV_LOG_INFO, "Template framebuffer is %"PRIu32": "
- "%"PRIu32"x%"PRIu32" %"PRIu32"bpp %"PRIu32"b depth.\n",
- fb->fb_id, fb->width, fb->height, fb->bpp, fb->depth);
+ av_log(avctx, AV_LOG_INFO, "Template framebuffer is %"PRIu32": "
+ "%"PRIu32"x%"PRIu32" %"PRIu32"bpp %"PRIu32"b depth.\n",
+ fb->fb_id, fb->width, fb->height, fb->bpp, fb->depth);
- ctx->width = fb->width;
- ctx->height = fb->height;
+ ctx->width = fb->width;
+ ctx->height = fb->height;
- if (!fb->handle) {
- av_log(avctx, AV_LOG_ERROR, "No handle set on framebuffer: "
- "maybe you need some additional capabilities?\n");
- err = AVERROR(EINVAL);
- goto fail;
+ if (!fb->handle) {
+ av_log(avctx, AV_LOG_ERROR, "No handle set on framebuffer: "
+ "maybe you need some additional capabilities?\n");
+ err = AVERROR(EINVAL);
+ goto fail;
+ }
}
stream = avformat_new_stream(avctx, NULL);
stream->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
stream->codecpar->codec_id = AV_CODEC_ID_WRAPPED_AVFRAME;
- stream->codecpar->width = fb->width;
- stream->codecpar->height = fb->height;
+ stream->codecpar->width = ctx->width;
+ stream->codecpar->height = ctx->height;
stream->codecpar->format = AV_PIX_FMT_DRM_PRIME;
avpriv_set_pts_info(stream, 64, 1, 1000000);
ctx->frames->format = AV_PIX_FMT_DRM_PRIME;
ctx->frames->sw_format = ctx->format,
- ctx->frames->width = fb->width;
- ctx->frames->height = fb->height;
+ ctx->frames->width = ctx->width;
+ ctx->frames->height = ctx->height;
err = av_hwframe_ctx_init(ctx->frames_ref);
if (err < 0) {
err = 0;
fail:
- if (plane_res)
- drmModeFreePlaneResources(plane_res);
- if (plane)
- drmModeFreePlane(plane);
- if (fb)
- drmModeFreeFB(fb);
-
+ drmModeFreePlaneResources(plane_res);
+ drmModeFreePlane(plane);
+ drmModeFreeFB(fb);
+#if HAVE_LIBDRM_GETFB2
+ drmModeFreeFB2(fb2);
+#endif
return err;
}
{ .str = "/dev/dri/card0" }, 0, 0, FLAGS },
{ "format", "Pixel format for framebuffer",
OFFSET(format), AV_OPT_TYPE_PIXEL_FMT,
- { .i64 = AV_PIX_FMT_BGR0 }, 0, UINT32_MAX, FLAGS },
+ { .i64 = AV_PIX_FMT_NONE }, -1, INT32_MAX, FLAGS },
{ "format_modifier", "DRM format modifier for framebuffer",
OFFSET(drm_format_modifier), AV_OPT_TYPE_INT64,
- { .i64 = DRM_FORMAT_MOD_NONE }, 0, INT64_MAX, FLAGS },
+ { .i64 = DRM_FORMAT_MOD_INVALID }, 0, INT64_MAX, FLAGS },
{ "crtc_id", "CRTC ID to define capture source",
OFFSET(source_crtc), AV_OPT_TYPE_INT64,
{ .i64 = 0 }, 0, UINT32_MAX, FLAGS },
.category = AV_CLASS_CATEGORY_DEVICE_VIDEO_INPUT,
};
-AVInputFormat ff_kmsgrab_demuxer = {
+const AVInputFormat ff_kmsgrab_demuxer = {
.name = "kmsgrab",
.long_name = NULL_IF_CONFIG_SMALL("KMS screen capture"),
.priv_data_size = sizeof(KMSGrabContext),
projects / ffmpeg / blobdiff