hyprwm/Hyprland
1
0
Fork 0
mirror of https://github.com/hyprwm/Hyprland synced 2024-11-05 06:25:59 +01:00
Code Issues Projects Releases Packages Wiki Activity

xwayland: Some readability improvements (#7807)

Browse source 
* Readability improvements xwayland server

* Made requested changes

* removed braces

* fix

* Ok this time is fixed

* Formatting
This commit is contained in:
Jasson 2024-09-18 13:12:26 -04:00 committed by GitHub
parent b248d59713
commit 94140e886e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 105 additions and 89 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

194
src/xwayland/Server.cpp
View file

@ -1,105 +1,115 @@
#include <cstdint>
#ifndef NO_XWAYLAND #ifndef NO_XWAYLAND
#include "Server.hpp" #include <format>
#include "../defines.hpp" #include <string>
#include "../Compositor.hpp"
#include "../managers/CursorManager.hpp"
#include "XWayland.hpp"
#include <errno.h> #include <errno.h>
#include <fcntl.h> #include <fcntl.h>
#include <stdio.h>
#include <cstring>
#include <signal.h> #include <signal.h>
#include <stddef.h> #include <stddef.h>
#include <stdint.h> #include <stdint.h>
#include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/un.h> #include <sys/un.h>
#include <unistd.h> #include <unistd.h>
#include <exception>
#include <filesystem> #include <filesystem>
#include <cstring> #include <sys/stat.h>
#include <sys/types.h>
#include <sys/socket.h>
// TODO: cleanup #include "Server.hpp"
static bool set_cloexec(int fd, bool cloexec) { #include "XWayland.hpp"
#include "debug/Log.hpp"
#include "../defines.hpp"
#include "../Compositor.hpp"
#include "../managers/CursorManager.hpp"
// Constants
constexpr int SOCKET_DIR_PERMISSIONS = 0755;
constexpr int SOCKET_BACKLOG = 1;
constexpr int MAX_SOCKET_RETRIES = 32;
constexpr int LOCK_FILE_MODE = 044;
static bool setCloseOnExec(int fd, bool cloexec) {
int flags = fcntl(fd, F_GETFD); int flags = fcntl(fd, F_GETFD);
if (flags == -1) { if (flags == -1) {
Debug::log(ERR, "fcntl failed"); Debug::log(ERR, "fcntl failed");
return false; return false;
} }
if (cloexec) {
if (cloexec)
flags = flags | FD_CLOEXEC; flags = flags | FD_CLOEXEC;
} else { else
flags = flags & ~FD_CLOEXEC; flags = flags & ~FD_CLOEXEC;
}
if (fcntl(fd, F_SETFD, flags) == -1) { if (fcntl(fd, F_SETFD, flags) == -1) {
Debug::log(ERR, "fcntl failed"); Debug::log(ERR, "fcntl failed");
return false; return false;
} }
return true; return true;
} }
static int openSocket(struct sockaddr_un* addr, size_t path_size) { void cleanUpSocket(int fd, const char* path) {
int fd, rc; close(fd);
socklen_t size = offsetof(struct sockaddr_un, sun_path) + path_size + 1; if (path[0])
unlink(path);
}
fd = socket(AF_UNIX, SOCK_STREAM, 0); static int createSocket(struct sockaddr_un* addr, size_t path_size) {
socklen_t size = offsetof(struct sockaddr_un, sun_path) + path_size + 1;
int fd = socket(AF_UNIX, SOCK_STREAM, 0);
if (fd < 0) { if (fd < 0) {
Debug::log(ERR, "failed to create socket {}{}", addr->sun_path[0] ? addr->sun_path[0] : '@', addr->sun_path + 1); Debug::log(ERR, "Failed to create socket {}{}", addr->sun_path[0] ? addr->sun_path[0] : '@', addr->sun_path + 1);
return -1; return -1;
} }
if (!set_cloexec(fd, true)) {
if (!setCloseOnExec(fd, true)) {
close(fd); close(fd);
return -1; return -1;
} }
if (addr->sun_path[0]) { if (addr->sun_path[0])
unlink(addr->sun_path); unlink(addr->sun_path);
}
if (bind(fd, (struct sockaddr*)addr, size) < 0) { if (bind(fd, (struct sockaddr*)addr, size) < 0) {
rc = errno; Debug::log(ERR, "Failed to bind socket {}{}", addr->sun_path[0] ? addr->sun_path[0] : '@', addr->sun_path + 1);
Debug::log(ERR, "failed to bind socket {}{}", addr->sun_path[0] ? addr->sun_path[0] : '@', addr->sun_path + 1); cleanUpSocket(fd, addr->sun_path);
goto cleanup; return -1;
} }
if (listen(fd, 1) < 0) {
rc = errno; if (listen(fd, SOCKET_BACKLOG) < 0) {
Debug::log(ERR, "failed to listen to socket {}{}", addr->sun_path[0] ? addr->sun_path[0] : '@', addr->sun_path + 1); Debug::log(ERR, "Failed to listen to socket {}{}", addr->sun_path[0] ? addr->sun_path[0] : '@', addr->sun_path + 1);
goto cleanup; cleanUpSocket(fd, addr->sun_path);
return -1;
} }
return fd; return fd;
cleanup:
close(fd);
if (addr->sun_path[0]) {
unlink(addr->sun_path);
}
errno = rc;
return -1;
} }
static bool checkPermissionsForSocketDir(void) { static bool checkPermissionsForSocketDir(void) {
struct stat buf; struct stat buf;
if (lstat("/tmp/.X11-unix", &buf)) { if (lstat("/tmp/.X11-unix", &buf)) {
Debug::log(ERR, "Failed statting X11 socket dir"); Debug::log(ERR, "Failed to stat X11 socket dir");
return false; return false;
} }
if (!(buf.st_mode & S_IFDIR)) { if (!(buf.st_mode & S_IFDIR)) {
Debug::log(ERR, "X11 socket dir is not a dir"); Debug::log(ERR, "X11 socket dir is not a directory");
return false; return false;
} }
if (!((buf.st_uid == 0) || (buf.st_uid == getuid()))) { if (!((buf.st_uid == 0) || (buf.st_uid == getuid()))) {
Debug::log(ERR, "X11 socket dir is not ours"); Debug::log(ERR, "X11 socket dir is not owned by root or current user");
return false; return false;
} }
if (!(buf.st_mode & S_ISVTX)) { if (!(buf.st_mode & S_ISVTX)) {
if ((buf.st_mode & (S_IWGRP | S_IWOTH))) { if ((buf.st_mode & (S_IWGRP | S_IWOTH))) {
Debug::log(ERR, "X11 socket dir is sticky by others"); Debug::log(ERR, "X11 socket dir is writable by others");
return false; return false;
} }
} }
@ -107,38 +117,51 @@ static bool checkPermissionsForSocketDir(void) {
return true; return true;
} }
static bool openSockets(std::array<int, 2>& sockets, int display) { static bool ensureSocketDirExists() {
auto ret = mkdir("/tmp/.X11-unix", 755); if (mkdir("/tmp/.X11-unix", SOCKET_DIR_PERMISSIONS) != 0) {
if (errno == EEXIST)
if (ret != 0) { return checkPermissionsForSocketDir();
if (errno == EEXIST) { else {
if (!checkPermissionsForSocketDir()) Debug::log(ERR, "XWayland: Couldn't create socket dir");
return false;
} else {
Debug::log(ERR, "XWayland: couldn't create socket dir");
return false; return false;
} }
} }
std::string path; return true;
}
static std::string getSocketPath(int display, bool isLinux) {
if (isLinux)
return std::format("/tmp/.X11-unix{}", display);
return std::format("/tmp/.X11-unix{}_", display);
}
static bool openSockets(std::array<int, 2>& sockets, int display) {
if (!ensureSocketDirExists())
return false;
sockaddr_un addr = {.sun_family = AF_UNIX}; sockaddr_un addr = {.sun_family = AF_UNIX};
std::string path;
#ifdef __linux__ #ifdef __linux__
// cursed... // cursed...
addr.sun_path[0] = 0; addr.sun_path[0] = 0;
path = std::format("/tmp/.X11-unix/X{}", display); path = getSocketPath(display, true);
strncpy(addr.sun_path + 1, path.c_str(), path.length() + 1); strncpy(addr.sun_path + 1, path.c_str(), path.length() + 1);
#else #else
path = std::format("/tmp/.X11-unix/X{}_", display); path = getSocketPath(display, false);
strncpy(addr.sun_path, path.c_str(), path.length() + 1); strncpy(addr.sun_path, path.c_str(), path.length() + 1);
#endif #endif
sockets[0] = openSocket(&addr, path.length());
sockets[0] = createSocket(&addr, path.length());
if (sockets[0] < 0) if (sockets[0] < 0)
return false; return false;
path = std::format("/tmp/.X11-unix/X{}", display); path = getSocketPath(display, true);
strncpy(addr.sun_path, path.c_str(), path.length() + 1); strncpy(addr.sun_path, path.c_str(), path.length() + 1);
sockets[1] = openSocket(&addr, path.length());
sockets[1] = createSocket(&addr, path.length());
if (sockets[1] < 0) { if (sockets[1] < 0) {
close(sockets[0]); close(sockets[0]);
sockets[0] = -1; sockets[0] = -1;
@ -160,39 +183,37 @@ static int xwaylandReady(int fd, uint32_t mask, void* data) {
static bool safeRemove(const std::string& path) { static bool safeRemove(const std::string& path) {
try { try {
return std::filesystem::remove(path); return std::filesystem::remove(path);
} catch (std::exception& e) { Debug::log(ERR, "[XWayland] failed to remove {}", path); } } catch (const std::exception& e) { Debug::log(ERR, "[XWayland] Failed to remove {}", path); }
return false; return false;
} }
bool CXWaylandServer::tryOpenSockets() { bool CXWaylandServer::tryOpenSockets() {
for (size_t i = 0; i <= 32; ++i) { for (size_t i = 0; i <= MAX_SOCKET_RETRIES; ++i) {
auto LOCK = std::format("/tmp/.X{}-lock", i); std::string lockPath = std::format("/tmp/.X{}-lock", i);
if (int fd = open(LOCK.c_str(), O_WRONLY | O_CREAT | O_EXCL | O_CLOEXEC, 0444); fd >= 0) { int fd = open(lockPath.c_str(), O_WRONLY | O_CREAT | O_EXCL | O_CLOEXEC, LOCK_FILE_MODE);
if (fd >= 0) {
// we managed to open the lock // we managed to open the lock
if (!openSockets(xFDs, i)) { if (!openSockets(xFDs, i)) {
safeRemove(LOCK); safeRemove(lockPath);
close(fd); close(fd);
continue; continue;
} }
const auto PIDSTR = std::format("{}", getpid()); const std::string pidStr = std::to_string(getpid());
if (write(fd, pidStr.c_str(), pidStr.length()) != (long)pidStr.length()) {
if (write(fd, PIDSTR.c_str(), PIDSTR.length()) != (long)PIDSTR.length()) { safeRemove(lockPath);
safeRemove(LOCK);
close(fd); close(fd);
continue; continue;
} }
close(fd); close(fd);
display = i; display = i;
displayName = std::format(":{}", display); displayName = std::format(":{}", display);
break; break;
} }
int fd = open(LOCK.c_str(), O_RDONLY | O_CLOEXEC); fd = open(lockPath.c_str(), O_RDONLY | O_CLOEXEC);
if (fd < 0) if (fd < 0)
continue; continue;
@ -201,21 +222,20 @@ bool CXWaylandServer::tryOpenSockets() {
read(fd, pidstr, sizeof(pidstr) - 1); read(fd, pidstr, sizeof(pidstr) - 1);
close(fd); close(fd);
uint64_t pid = 0; int32_t pid = 0;
try { try {
pid = std::stoi(std::string{pidstr, 11}); pid = std::stoi(std::string{pidstr, 11});
} catch (...) { continue; } } catch (...) { continue; }
if (kill(pid, 0) != 0 && errno == ESRCH) { if (kill(pid, 0) != 0 && errno == ESRCH) {
if (!safeRemove(LOCK)) if (!safeRemove(lockPath))
continue; continue;
i--; i--;
} }
} }
if (display < 0) { if (display < 0) {
Debug::log(ERR, "Failed to find a suitable socket for xwayland"); Debug::log(ERR, "Failed to find a suitable socket for XWayland");
return false; return false;
} }
@ -232,19 +252,17 @@ CXWaylandServer::~CXWaylandServer() {
if (display < 0) if (display < 0)
return; return;
if (xFDs[0]) close(xFDs[0]);
close(xFDs[0]); close(xFDs[1]);
if (xFDs[1])
close(xFDs[1]);
auto LOCK = std::format("/tmp/.X{}-lock", display); std::string lockPath = std::format("/tmp/.X{}-lock", display);
safeRemove(LOCK); safeRemove(lockPath);
std::string path; std::string path;
#ifdef __linux__ #ifdef __linux__
path = std::format("/tmp/.X11-unix/X{}", display); path = getSocketPath(display, true);
#else #else
path = std::format("/tmp/.X11-unix/X{}_", display); path = getSocketPath(display, false);
#endif #endif
safeRemove(path); safeRemove(path);
} }
@ -256,7 +274,6 @@ void CXWaylandServer::die() {
if (xFDReadEvents[0]) { if (xFDReadEvents[0]) {
wl_event_source_remove(xFDReadEvents[0]); wl_event_source_remove(xFDReadEvents[0]);
wl_event_source_remove(xFDReadEvents[1]); wl_event_source_remove(xFDReadEvents[1]);
xFDReadEvents = {nullptr, nullptr}; xFDReadEvents = {nullptr, nullptr};
} }
@ -298,7 +315,7 @@ bool CXWaylandServer::create() {
} }
void CXWaylandServer::runXWayland(int notifyFD) { void CXWaylandServer::runXWayland(int notifyFD) {
if (!set_cloexec(xFDs[0], false) || !set_cloexec(xFDs[1], false) || !set_cloexec(waylandFDs[1], false) || !set_cloexec(xwmFDs[1], false)) { if (!setCloseOnExec(xFDs[0], false) || !setCloseOnExec(xFDs[1], false) || !setCloseOnExec(waylandFDs[1], false) || !setCloseOnExec(xwmFDs[1], false)) {
Debug::log(ERR, "Failed to unset cloexec on fds"); Debug::log(ERR, "Failed to unset cloexec on fds");
_exit(EXIT_FAILURE); _exit(EXIT_FAILURE);
} }
@ -325,7 +342,7 @@ bool CXWaylandServer::start() {
return false; return false;
} }
if (!set_cloexec(waylandFDs[0], true) || !set_cloexec(waylandFDs[1], true)) { if (!setCloseOnExec(waylandFDs[0], true) || !setCloseOnExec(waylandFDs[1], true)) {
Debug::log(ERR, "set_cloexec failed (1)"); Debug::log(ERR, "set_cloexec failed (1)");
die(); die();
return false; return false;
@ -337,7 +354,7 @@ bool CXWaylandServer::start() {
return false; return false;
} }
if (!set_cloexec(xwmFDs[0], true) || !set_cloexec(xwmFDs[1], true)) { if (!setCloseOnExec(xwmFDs[0], true) || !setCloseOnExec(xwmFDs[1], true)) {
Debug::log(ERR, "set_cloexec failed (2)"); Debug::log(ERR, "set_cloexec failed (2)");
die(); die();
return false; return false;
@ -359,7 +376,7 @@ bool CXWaylandServer::start() {
return false; return false;
} }
if (!set_cloexec(notify[0], true)) { if (!setCloseOnExec(notify[0], true)) {
Debug::log(ERR, "set_cloexec failed (3)"); Debug::log(ERR, "set_cloexec failed (3)");
close(notify[0]); close(notify[0]);
close(notify[1]); close(notify[1]);
@ -382,9 +399,8 @@ bool CXWaylandServer::start() {
if (pid < 0) { if (pid < 0) {
Debug::log(ERR, "second fork failed"); Debug::log(ERR, "second fork failed");
_exit(1); _exit(1);
} else if (pid == 0) { } else if (pid == 0)
runXWayland(notify[1]); runXWayland(notify[1]);
}
_exit(0); _exit(0);
} }