PapaTutuWawa/gpu-screen-recorder
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cap_sys_nice, set caps on gsr-kms-server and gpu-screen-recorder

Browse Source
This commit is contained in:
dec05eba 2023-07-26 05:14:01 +02:00
parent 4ad0118f35
commit 15a7c2f1b0
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
install.sh
View File

@ -9,6 +9,8 @@ cd "$script_dir"
strip gsr-kms-server strip gsr-kms-server
strip gpu-screen-recorder strip gpu-screen-recorder
rm -f "/usr/local/bin/gpu-screen-recorder" rm -f "/usr/local/bin/gpu-screen-recorder"
setcap cap_sys_admin+ep gsr-kms-server
setcap cap_sys_nice+ep gpu-screen-recorder
install -Dm755 "gsr-kms-server" "/usr/bin/gsr-kms-server" install -Dm755 "gsr-kms-server" "/usr/bin/gsr-kms-server"
install -Dm755 "gpu-screen-recorder" "/usr/bin/gpu-screen-recorder" install -Dm755 "gpu-screen-recorder" "/usr/bin/gpu-screen-recorder"

17
src/egl.c
View File

@ -10,6 +10,7 @@
#include <wayland-egl.h> #include <wayland-egl.h>
#include "../external/wlr-export-dmabuf-unstable-v1-client-protocol.h" #include "../external/wlr-export-dmabuf-unstable-v1-client-protocol.h"
#include <unistd.h> #include <unistd.h>
#include <sys/capability.h>
// Move this shit to a separate wayland file, and have a separate file for x11. // Move this shit to a separate wayland file, and have a separate file for x11.
@ -206,6 +207,18 @@ static gsr_wayland_output* get_wayland_output_by_name(gsr_egl *egl, const char *
return NULL; return NULL;
} }
static void reset_cap_nice(void) {
cap_t caps = cap_get_proc();
if(!caps)
return;
const cap_value_t cap_to_remove = CAP_SYS_NICE;
cap_set_flag(caps, CAP_EFFECTIVE, 1, &cap_to_remove, CAP_CLEAR);
cap_set_flag(caps, CAP_PERMITTED, 1, &cap_to_remove, CAP_CLEAR);
cap_set_proc(caps);
cap_free(caps);
}
// TODO: Create egl context without surface (in other words, x11/wayland agnostic, doesn't require x11/wayland dependency) // TODO: Create egl context without surface (in other words, x11/wayland agnostic, doesn't require x11/wayland dependency)
static bool gsr_egl_create_window(gsr_egl *self, bool wayland) { static bool gsr_egl_create_window(gsr_egl *self, bool wayland) {
EGLConfig ecfg; EGLConfig ecfg;
@ -219,7 +232,7 @@ static bool gsr_egl_create_window(gsr_egl *self, bool wayland) {
const int32_t ctxattr[] = { const int32_t ctxattr[] = {
EGL_CONTEXT_CLIENT_VERSION, 2, EGL_CONTEXT_CLIENT_VERSION, 2,
EGL_CONTEXT_PRIORITY_LEVEL_IMG, EGL_CONTEXT_PRIORITY_HIGH_IMG, EGL_CONTEXT_PRIORITY_LEVEL_IMG, EGL_CONTEXT_PRIORITY_HIGH_IMG, /* requires cap_sys_nice, ignored otherwise */
EGL_NONE, EGL_NONE EGL_NONE, EGL_NONE
}; };
@ -294,9 +307,11 @@ static bool gsr_egl_create_window(gsr_egl *self, bool wayland) {
goto fail; goto fail;
} }
reset_cap_nice();
return true; return true;
fail: fail:
reset_cap_nice();
gsr_egl_unload(self); gsr_egl_unload(self);
return false; return false;
} }