summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJari Vetoniemi <mailroxas@gmail.com>2017-01-22 06:19:05 +0200
committerJari Vetoniemi <mailroxas@gmail.com>2017-01-22 20:10:35 +0200
commit2d8e13e2a1fbd5363c30fa6ebb39385f95ab0af2 (patch)
treeee1e04b6bf2e4dae316ee133f35a50f9a57ddebe
Initial commit
-rw-r--r--glcapture.c475
1 files changed, 475 insertions, 0 deletions
diff --git a/glcapture.c b/glcapture.c
new file mode 100644
index 0000000..60ac1be
--- /dev/null
+++ b/glcapture.c
@@ -0,0 +1,475 @@
+/* gcc -std=c99 -fPIC -shared -Wl,-soname,glcapture.so glcapture.c -o glcapture.so
+ * gcc -m32 -std=c99 -fPIC -shared -Wl,-soname,glcapture.so glcapture.c -o glcapture.so (for 32bit)
+ *
+ * Capture OpenGL framebuffer and ALSA audio and pipe them to ffmpeg
+ * Usage: LD_PRELOAD="/path/to/glcapture.so" ./program
+ */
+
+#define _GNU_SOURCE
+#define GL_GLEXT_PROTOTYPES
+#include <dlfcn.h>
+#include <GL/glx.h>
+#include <alsa/asoundlib.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdio.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <unistd.h>
+#include <time.h>
+#include <err.h>
+#include <pthread.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+
+static void* (*_dlsym)(void*, const char*) = NULL;
+static void (*_glXSwapBuffers)(Display*, GLXDrawable) = NULL;
+static void* (*_glXGetProcAddress)(const GLubyte*) = NULL;
+static void* (*_glXGetProcAddressARB)(const GLubyte*) = NULL;
+static void* (*_glXGetProcAddressEXT)(const GLubyte*) = NULL;
+static snd_pcm_sframes_t (*_snd_pcm_writei)(snd_pcm_t *pcm, const void *buffer, snd_pcm_uframes_t size) = NULL;
+static snd_pcm_sframes_t (*_snd_pcm_writen)(snd_pcm_t *pcm, void **bufs, snd_pcm_uframes_t size) = NULL;
+static snd_pcm_sframes_t (*_snd_pcm_mmap_writei)(snd_pcm_t *pcm, const void *buffer, snd_pcm_uframes_t size) = NULL;
+static snd_pcm_sframes_t (*_snd_pcm_mmap_writen)(snd_pcm_t *pcm, void **bufs, snd_pcm_uframes_t size) = NULL;
+static int (*_clock_gettime)(clockid_t clk_id, struct timespec *tp) = NULL;
+static void* store_real_symbol_and_return_fake_symbol(const char *symbol, void *ret);
+
+#define WARN(x, ...) do { warn("glcapture: "x, ##__VA_ARGS__); } while (0)
+#define WARNX(x, ...) do { warnx("glcapture: "x, ##__VA_ARGS__); } while (0)
+#define WARN_ONCE(x, ...) do { static bool o = false; if (!o) { WARNX(x, ##__VA_ARGS__); o = true; } } while (0)
+#define HOOK_DLSYM(x, y) if (!x) { if ((x = dlvsym(RTLD_NEXT, __func__, "GLIBC_2.0"))) WARNX("HOOK dlsym"); }
+#define HOOK(x) if (!x) if ((x = _dlsym(RTLD_NEXT, __func__))) { WARNX("HOOK %s", __func__); }
+#define SET_IF_NOT_HOOKED(x, y) if (!x) { x = y; WARNX("SET %s", #x); }
+
+// Some tunables
+// XXX: Make these configurable
+#define NUM_PBOS 2
+static double FPS = 60.0;
+static double SPEED_HACK = 1.0;
+static const char *FIFO_PATH = "/tmp/glcapture.fifo";
+
+enum stream {
+ STREAM_VIDEO,
+ STREAM_AUDIO,
+ STREAM_LAST,
+};
+
+struct pbo {
+ uint64_t ts;
+ GLuint obj;
+ bool written;
+};
+
+struct gl {
+ GLint view[4];
+ struct pbo pbo[NUM_PBOS];
+ uint8_t active; // pbo
+};
+
+struct frame_info {
+ union {
+ struct {
+ const char *format;
+ uint32_t width, height, fps;
+ } video;
+ struct {
+ const char *format;
+ uint32_t rate;
+ uint8_t channels;
+ } audio;
+ };
+ uint64_t ts;
+ enum stream stream;
+};
+
+struct fifo {
+ struct {
+ struct frame_info info;
+ uint64_t base;
+ bool ready;
+ } stream[STREAM_LAST];
+ size_t size;
+ int fd;
+ bool created;
+};
+
+static uint64_t get_time_ns(void)
+{
+ struct timespec ts;
+ _clock_gettime(CLOCK_MONOTONIC, &ts);
+ return (uint64_t)ts.tv_sec * (uint64_t)1e9 + (uint64_t)ts.tv_nsec;
+}
+
+static void
+reset_fifo(struct fifo *fifo)
+{
+ close(fifo->fd);
+ memset(fifo, 0, sizeof(*fifo));
+ fifo->fd = -1;
+ WARNX("reseting fifo");
+}
+
+static void
+write_data_unsafe(struct fifo *fifo, const struct frame_info *info, const void *buffer, const size_t size)
+{
+ fifo->stream[info->stream].info = *info;
+ fifo->stream[info->stream].ready = true;
+
+ for (enum stream i = 0; i < STREAM_LAST; ++i) {
+ if (!fifo->stream[i].ready)
+ return;
+ }
+
+ if (memcmp(info, &fifo->stream[info->stream].info, sizeof(*info)))
+ reset_fifo(fifo);
+
+ if (!fifo->created) {
+ remove(FIFO_PATH);
+
+ if (!(fifo->created = !mkfifo(FIFO_PATH, 0666)))
+ return;
+
+ fifo->created = true;
+ }
+
+ if (fifo->fd < 0) {
+ if ((fifo->fd = open(FIFO_PATH, O_WRONLY | O_NONBLOCK | O_CLOEXEC)) < 0)
+ return;
+
+ const int flags = fcntl(fifo->fd, F_GETFL);
+ fcntl(fifo->fd, F_SETFL, flags & ~O_NONBLOCK);
+
+ uint8_t header[255] = { 'r', 'a', 'w', 'm', 'u', 'x' };
+
+ if (strlen(info->video.format) + strlen(info->audio.format) + 31 > sizeof(header)) {
+ warnx("something went wrong");
+ reset_fifo(fifo);
+ return;
+ }
+
+ uint8_t *p = header + 6;
+
+ {
+ const struct frame_info *info = &fifo->stream[STREAM_VIDEO].info;
+ memcpy(p, (uint8_t[]){1}, sizeof(uint8_t)); p += 1;
+ memcpy(p, info->video.format, strlen(info->video.format)); p += strlen(info->video.format) + 1;
+ memcpy(p, (uint32_t[]){1}, sizeof(uint32_t)); p += 4;
+ memcpy(p, (uint32_t[]){info->video.fps}, sizeof(uint32_t)); p += 4;
+ memcpy(p, &info->video.width, sizeof(uint32_t)); p += 4;
+ memcpy(p, &info->video.height, sizeof(uint32_t)); p += 4;
+ }
+
+ {
+ const struct frame_info *info = &fifo->stream[STREAM_AUDIO].info;
+ memcpy(p, (uint8_t[]){2}, sizeof(uint8_t)); p += 1;
+ memcpy(p, info->audio.format, strlen(info->audio.format)); p += strlen(info->audio.format) + 1;
+ memcpy(p, &info->audio.rate, sizeof(info->audio.rate)); p += 4;
+ memcpy(p, &info->audio.channels, sizeof(info->audio.channels)); p += 1;
+ }
+
+ WARNX("stream ready, writing headers");
+ write(fifo->fd, header, (p + 1) - header);
+ }
+
+ if (!fifo->stream[info->stream].base)
+ fifo->stream[info->stream].base = info->ts;
+
+ const uint64_t rate = 1e9 / (info->stream == STREAM_VIDEO ? info->video.fps : info->audio.rate);
+ const uint64_t pts = (info->ts - fifo->stream[info->stream].base) / rate;
+
+ uint8_t frame[] = {
+ info->stream,
+ 0, 0, 0, 0,
+ 0, 0, 0, 0, 0, 0, 0, 0
+ };
+
+ memcpy(frame + 1, (uint32_t[]){size}, sizeof(uint32_t));
+ memcpy(frame + 1 + 4, (uint64_t[]){pts}, sizeof(uint64_t));
+
+ if (fifo->size < size) {
+ fcntl(fifo->fd, F_SETPIPE_SZ, 15 * (size + sizeof(frame)));
+ fifo->size = size;
+ }
+
+ errno = 0;
+ ssize_t ret;
+ if ((ret = write(fifo->fd, frame, sizeof(frame)) != (ssize_t)sizeof(frame)) ||
+ ((ret = write(fifo->fd, buffer, size)) != (ssize_t)size)) {
+ WARN("write() == %zu", ret);
+ reset_fifo(fifo);
+ }
+}
+
+static void
+write_data(const struct frame_info *info, const void *buffer, const size_t size)
+{
+ // we need to protect our fifo structure, since games usually output audio on another thread and so
+ static struct fifo fifo = { .fd = -1 };
+ static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
+ pthread_mutex_lock(&mutex);
+ write_data_unsafe(&fifo, info, buffer, size);
+ pthread_mutex_unlock(&mutex);
+}
+
+static const char*
+alsa_get_format(const snd_pcm_format_t format)
+{
+ switch (format) {
+ case SND_PCM_FORMAT_FLOAT64_LE: return "f64le";
+ case SND_PCM_FORMAT_FLOAT64_BE: return "f64be";
+ case SND_PCM_FORMAT_FLOAT_LE: return "f32le";
+ case SND_PCM_FORMAT_FLOAT_BE: return "f32be";
+ case SND_PCM_FORMAT_S32_LE: return "s32le";
+ case SND_PCM_FORMAT_S32_BE: return "s32be";
+ case SND_PCM_FORMAT_S24_LE: return "s24le";
+ case SND_PCM_FORMAT_S24_BE: return "s24be";
+ case SND_PCM_FORMAT_S16_LE: return "s16le";
+ case SND_PCM_FORMAT_S16_BE: return "s16be";
+ default: break;
+ }
+
+ WARN_ONCE("can't convert alsa format: %u", format);
+ return NULL;
+}
+
+static void
+alsa_get_frame_info(snd_pcm_t *pcm, struct frame_info *out_info, const char *caller)
+{
+ snd_pcm_format_t format;
+ unsigned int channels, rate;
+ snd_pcm_hw_params_t *params = alloca(snd_pcm_hw_params_sizeof());
+ snd_pcm_hw_params_current(pcm, params);
+ snd_pcm_hw_params_get_format(params, &format);
+ snd_pcm_hw_params_get_channels(params, &channels);
+ snd_pcm_hw_params_get_rate(params, &rate, NULL);
+ WARN_ONCE("%s (%s:%u:%u)", caller, snd_pcm_format_name(format), rate, channels);
+ out_info->ts = get_time_ns();
+ out_info->stream = STREAM_AUDIO;
+ out_info->audio.format = alsa_get_format(format);
+ out_info->audio.rate = rate;
+ out_info->audio.channels = channels;
+}
+
+static void
+alsa_writei(snd_pcm_t *pcm, const void *buffer, const snd_pcm_uframes_t size, const char *caller)
+{
+ struct frame_info info;
+ alsa_get_frame_info(pcm, &info, caller);
+
+ if (!info.audio.format)
+ return;
+
+ write_data(&info, buffer, snd_pcm_frames_to_bytes(pcm, size));
+}
+
+static void
+reset_capture(struct gl *gl)
+{
+ for (size_t i = 0; i < NUM_PBOS; ++i)
+ glDeleteBuffers(1, &gl->pbo[i].obj);
+
+ memset(gl, 0, sizeof(*gl));
+ WARNX("capture reset");
+}
+
+static void
+capture(struct gl *gl, const GLint view[4])
+{
+ if (memcmp(gl->view, view, sizeof(gl->view))) {
+ WARNX("resolution change to: %d,%d+%ux%u", view[0], view[1], view[2], view[3]);
+ reset_capture(gl);
+ return;
+ }
+
+ {
+ if (!gl->pbo[gl->active].obj) {
+ glGenBuffers(1, &gl->pbo[gl->active].obj);
+ glBindBuffer(GL_PIXEL_PACK_BUFFER, gl->pbo[gl->active].obj);
+ glBufferData(GL_PIXEL_PACK_BUFFER, view[2] * view[3] * 3, NULL, GL_STREAM_READ);
+ WARNX("create pbo %u", gl->active);
+ } else {
+ glBindBuffer(GL_PIXEL_PACK_BUFFER, gl->pbo[gl->active].obj);
+ }
+
+ glPixelStorei(GL_PACK_ALIGNMENT, 1);
+ glReadPixels(view[0], view[1], view[2], view[3], GL_RGB, GL_UNSIGNED_BYTE, NULL);
+ gl->pbo[gl->active].ts = get_time_ns();
+ gl->pbo[gl->active].written = true;
+ }
+
+ gl->active = (gl->active + 1) % NUM_PBOS;
+
+ if (gl->pbo[gl->active].written) {
+ const void *buf;
+ glBindBuffer(GL_PIXEL_PACK_BUFFER, gl->pbo[gl->active].obj);
+ if ((buf = glMapBuffer(GL_PIXEL_PACK_BUFFER, GL_READ_ONLY))) {
+ const struct frame_info info = {
+ .ts = gl->pbo[gl->active].ts,
+ .stream = STREAM_VIDEO,
+ .video.format = "rgb24",
+ .video.width = view[2],
+ .video.height = view[3],
+ .video.fps = FPS,
+ };
+
+ write_data(&info, buf, info.video.width * info.video.height * 3);
+ glUnmapBuffer(GL_PIXEL_PACK_BUFFER);
+ gl->pbo[gl->active].written = false;
+ }
+ }
+}
+
+static void
+draw_indicator(const GLint view[4])
+{
+ const uint32_t size = (view[3] / 75 > 10 ? view[3] / 75 : 10);
+ glEnable(GL_SCISSOR_TEST);
+ glScissor(size / 2 - 1, view[3] - size - size / 2 - 1, size + 2, size + 2);
+ glClearColor(0.0f, 0.0f, 0.0f, 0.0f);
+ glClear(GL_COLOR_BUFFER_BIT);
+ glScissor(size / 2, view[3] - size - size / 2, size, size);
+ glClearColor(1.0f, 0.0f, 0.0f, 0.0f);
+ glClear(GL_COLOR_BUFFER_BIT);
+ glDisable(GL_SCISSOR_TEST);
+}
+
+void
+glXSwapBuffers(Display *dpy, GLXDrawable drawable)
+{
+ HOOK(_glXSwapBuffers);
+
+ GLint view[4] = {0};
+ static __thread struct gl gl;
+ const GLenum error0 = glGetError();
+ glGetIntegerv(GL_VIEWPORT, view);
+ glPushAttrib(GL_ALL_ATTRIB_BITS);
+ capture(&gl, view);
+ draw_indicator(view);
+ glPopAttrib();
+ memcpy(gl.view, view, sizeof(gl.view));
+
+ if (error0 != glGetError()) {
+ WARNX("glError occured");
+ reset_capture(&gl);
+ }
+
+ _glXSwapBuffers(dpy, drawable);
+}
+
+__GLXextFuncPtr
+glXGetProcAddressEXT(const GLubyte *procname)
+{
+ HOOK(_glXGetProcAddressEXT);
+ return (_glXGetProcAddressEXT ? store_real_symbol_and_return_fake_symbol((const char*)procname, _glXGetProcAddressEXT(procname)) : NULL);
+}
+
+__GLXextFuncPtr
+glXGetProcAddressARB(const GLubyte *procname)
+{
+ HOOK(_glXGetProcAddressARB);
+ return (_glXGetProcAddressARB ? store_real_symbol_and_return_fake_symbol((const char*)procname, _glXGetProcAddressARB(procname)) : NULL);
+}
+
+__GLXextFuncPtr
+glXGetProcAddress(const GLubyte *procname)
+{
+ HOOK(_glXGetProcAddress);
+ return (_glXGetProcAddress ? store_real_symbol_and_return_fake_symbol((const char*)procname, _glXGetProcAddress(procname)) : NULL);
+}
+
+snd_pcm_sframes_t
+snd_pcm_writei(snd_pcm_t *pcm, const void *buffer, snd_pcm_uframes_t size)
+{
+ HOOK(_snd_pcm_writei);
+ alsa_writei(pcm, buffer, size, __func__);
+ return _snd_pcm_writei(pcm, buffer, size);
+}
+
+snd_pcm_sframes_t
+snd_pcm_writen(snd_pcm_t *pcm, void **bufs, snd_pcm_uframes_t size)
+{
+ HOOK(_snd_pcm_writen);
+ return _snd_pcm_writen(pcm, bufs, size);
+}
+
+snd_pcm_sframes_t
+snd_pcm_mmap_writei(snd_pcm_t *pcm, const void *buffer, snd_pcm_uframes_t size)
+{
+ HOOK(_snd_pcm_mmap_writei);
+ alsa_writei(pcm, buffer, size, __func__);
+ return _snd_pcm_mmap_writei(pcm, buffer, size);
+}
+
+snd_pcm_sframes_t
+snd_pcm_mmap_writen(snd_pcm_t *pcm, void **bufs, snd_pcm_uframes_t size)
+{
+ HOOK(_snd_pcm_mmap_writen);
+ return _snd_pcm_mmap_writen(pcm, bufs, size);
+}
+
+int
+clock_gettime(clockid_t clk_id, struct timespec *tp)
+{
+ HOOK(_clock_gettime);
+
+ if ((clk_id == CLOCK_MONOTONIC || clk_id == CLOCK_MONOTONIC_RAW)) {
+ static __thread uint64_t base;
+ const uint64_t current = get_time_ns();
+ if (!base) base = current;
+ const uint64_t fake = base + (current - base) * SPEED_HACK;
+ tp->tv_sec = fake / (uint64_t)1e9;
+ tp->tv_nsec = (fake % (uint64_t)1e9);
+ return 0;
+ }
+
+ return _clock_gettime(clk_id, tp);
+}
+
+void*
+store_real_symbol_and_return_fake_symbol(const char *symbol, void *ret)
+{
+ if (!ret || !symbol)
+ return ret;
+
+ if (!strcmp(symbol, "glXSwapBuffers")) {
+ SET_IF_NOT_HOOKED(_glXSwapBuffers, ret);
+ return glXSwapBuffers;
+ } else if (!strcmp(symbol, "glXGetProcAddressEXT")) {
+ SET_IF_NOT_HOOKED(_glXGetProcAddressEXT, ret);
+ return glXGetProcAddressEXT;
+ } else if (!strcmp(symbol, "glXGetProcAddressARB")) {
+ SET_IF_NOT_HOOKED(_glXGetProcAddressARB, ret);
+ return glXGetProcAddressARB;
+ } else if (!strcmp(symbol, "glXGetProcAddress")) {
+ SET_IF_NOT_HOOKED(_glXGetProcAddress, ret);
+ return glXGetProcAddress;
+ } else if (!strcmp(symbol, "snd_pcm_writei")) {
+ SET_IF_NOT_HOOKED(_snd_pcm_writei, ret);
+ return snd_pcm_writei;
+ } else if (!strcmp(symbol, "snd_pcm_writen")) {
+ SET_IF_NOT_HOOKED(_snd_pcm_writen, ret);
+ return snd_pcm_writen;
+ } else if (!strcmp(symbol, "snd_pcm_mmap_writei")) {
+ SET_IF_NOT_HOOKED(_snd_pcm_mmap_writei, ret);
+ return snd_pcm_mmap_writei;
+ } else if (!strcmp(symbol, "snd_pcm_mmap_writen")) {
+ SET_IF_NOT_HOOKED(_snd_pcm_mmap_writen, ret);
+ return snd_pcm_mmap_writen;
+ } else if (!strcmp(symbol, "clock_gettime")) {
+ SET_IF_NOT_HOOKED(_clock_gettime, ret);
+ return clock_gettime;
+ }
+
+ return ret;
+}
+
+void*
+dlsym(void *handle, const char *symbol)
+{
+ HOOK_DLSYM(_dlsym, dlsym);
+
+ if (!strcmp(symbol, "dlsym"))
+ return dlsym;
+
+ return store_real_symbol_and_return_fake_symbol(symbol, _dlsym(handle, symbol));
+}