mirror of
				https://gitlab.freedesktop.org/wlroots/wlroots.git
				synced 2025-11-03 09:01:40 -05:00 
			
		
		
		
	Previous logic could lead wlr_wl_backend.drm_render_name being written to twice, causing a memory leak. This commit fixes the race condition.
		
			
				
	
	
		
			716 lines
		
	
	
	
		
			22 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			716 lines
		
	
	
	
		
			22 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
#include <assert.h>
 | 
						|
#include <fcntl.h>
 | 
						|
#include <limits.h>
 | 
						|
#include <stdint.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#include <sys/mman.h>
 | 
						|
#include <sys/types.h>
 | 
						|
#include <unistd.h>
 | 
						|
 | 
						|
#include <drm_fourcc.h>
 | 
						|
#include <wayland-server-core.h>
 | 
						|
#include <xf86drm.h>
 | 
						|
 | 
						|
#include <wlr/backend/interface.h>
 | 
						|
#include <wlr/interfaces/wlr_output.h>
 | 
						|
#include <wlr/util/log.h>
 | 
						|
 | 
						|
#include "backend/wayland.h"
 | 
						|
#include "render/drm_format_set.h"
 | 
						|
#include "render/pixel_format.h"
 | 
						|
 | 
						|
#include "drm-client-protocol.h"
 | 
						|
#include "linux-dmabuf-v1-client-protocol.h"
 | 
						|
#include "pointer-gestures-unstable-v1-client-protocol.h"
 | 
						|
#include "presentation-time-client-protocol.h"
 | 
						|
#include "xdg-activation-v1-client-protocol.h"
 | 
						|
#include "xdg-decoration-unstable-v1-client-protocol.h"
 | 
						|
#include "xdg-shell-client-protocol.h"
 | 
						|
#include "tablet-v2-client-protocol.h"
 | 
						|
#include "relative-pointer-unstable-v1-client-protocol.h"
 | 
						|
#include "viewporter-client-protocol.h"
 | 
						|
 | 
						|
struct wlr_wl_linux_dmabuf_feedback_v1 {
 | 
						|
	struct wlr_wl_backend *backend;
 | 
						|
	dev_t main_device_id;
 | 
						|
	struct wlr_wl_linux_dmabuf_v1_table_entry *format_table;
 | 
						|
	size_t format_table_size;
 | 
						|
 | 
						|
	dev_t tranche_target_device_id;
 | 
						|
};
 | 
						|
 | 
						|
struct wlr_wl_linux_dmabuf_v1_table_entry {
 | 
						|
	uint32_t format;
 | 
						|
	uint32_t pad; /* unused */
 | 
						|
	uint64_t modifier;
 | 
						|
};
 | 
						|
 | 
						|
struct wlr_wl_backend *get_wl_backend_from_backend(struct wlr_backend *wlr_backend) {
 | 
						|
	assert(wlr_backend_is_wl(wlr_backend));
 | 
						|
	struct wlr_wl_backend *backend = wl_container_of(wlr_backend, backend, backend);
 | 
						|
	return backend;
 | 
						|
}
 | 
						|
 | 
						|
static int dispatch_events(int fd, uint32_t mask, void *data) {
 | 
						|
	struct wlr_wl_backend *wl = data;
 | 
						|
 | 
						|
	if ((mask & WL_EVENT_HANGUP) || (mask & WL_EVENT_ERROR)) {
 | 
						|
		if (mask & WL_EVENT_ERROR) {
 | 
						|
			wlr_log(WLR_ERROR, "Failed to read from remote Wayland display");
 | 
						|
		}
 | 
						|
		wlr_backend_destroy(&wl->backend);
 | 
						|
		return 0;
 | 
						|
	}
 | 
						|
 | 
						|
	int count = 0;
 | 
						|
	if (mask & WL_EVENT_READABLE) {
 | 
						|
		count = wl_display_dispatch(wl->remote_display);
 | 
						|
	}
 | 
						|
	if (mask & WL_EVENT_WRITABLE) {
 | 
						|
		wl_display_flush(wl->remote_display);
 | 
						|
	}
 | 
						|
	if (mask == 0) {
 | 
						|
		count = wl_display_dispatch_pending(wl->remote_display);
 | 
						|
		wl_display_flush(wl->remote_display);
 | 
						|
	}
 | 
						|
 | 
						|
	if (count < 0) {
 | 
						|
		wlr_log(WLR_ERROR, "Failed to dispatch remote Wayland display");
 | 
						|
		wlr_backend_destroy(&wl->backend);
 | 
						|
		return 0;
 | 
						|
	}
 | 
						|
	return count;
 | 
						|
}
 | 
						|
 | 
						|
static void xdg_wm_base_handle_ping(void *data,
 | 
						|
		struct xdg_wm_base *base, uint32_t serial) {
 | 
						|
	xdg_wm_base_pong(base, serial);
 | 
						|
}
 | 
						|
 | 
						|
static const struct xdg_wm_base_listener xdg_wm_base_listener = {
 | 
						|
	xdg_wm_base_handle_ping,
 | 
						|
};
 | 
						|
 | 
						|
static void presentation_handle_clock_id(void *data,
 | 
						|
		struct wp_presentation *presentation, uint32_t clock) {
 | 
						|
	struct wlr_wl_backend *wl = data;
 | 
						|
 | 
						|
	if (clock != CLOCK_MONOTONIC) {
 | 
						|
		wp_presentation_destroy(wl->presentation);
 | 
						|
		wl->presentation = NULL;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static const struct wp_presentation_listener presentation_listener = {
 | 
						|
	.clock_id = presentation_handle_clock_id,
 | 
						|
};
 | 
						|
 | 
						|
static void linux_dmabuf_v1_handle_format(void *data,
 | 
						|
		struct zwp_linux_dmabuf_v1 *linux_dmabuf_v1, uint32_t format) {
 | 
						|
	// Note, this event is deprecated
 | 
						|
	struct wlr_wl_backend *wl = data;
 | 
						|
 | 
						|
	wlr_drm_format_set_add(&wl->linux_dmabuf_v1_formats, format,
 | 
						|
		DRM_FORMAT_MOD_INVALID);
 | 
						|
}
 | 
						|
 | 
						|
static void linux_dmabuf_v1_handle_modifier(void *data,
 | 
						|
		struct zwp_linux_dmabuf_v1 *linux_dmabuf_v1, uint32_t format,
 | 
						|
		uint32_t modifier_hi, uint32_t modifier_lo) {
 | 
						|
	struct wlr_wl_backend *wl = data;
 | 
						|
 | 
						|
	uint64_t modifier = ((uint64_t)modifier_hi << 32) | modifier_lo;
 | 
						|
	wlr_drm_format_set_add(&wl->linux_dmabuf_v1_formats, format, modifier);
 | 
						|
}
 | 
						|
 | 
						|
static const struct zwp_linux_dmabuf_v1_listener linux_dmabuf_v1_listener = {
 | 
						|
	.format = linux_dmabuf_v1_handle_format,
 | 
						|
	.modifier = linux_dmabuf_v1_handle_modifier,
 | 
						|
};
 | 
						|
 | 
						|
static void linux_dmabuf_feedback_v1_handle_done(void *data,
 | 
						|
		struct zwp_linux_dmabuf_feedback_v1 *feedback) {
 | 
						|
	// This space is intentionally left blank
 | 
						|
}
 | 
						|
 | 
						|
static void linux_dmabuf_feedback_v1_handle_format_table(void *data,
 | 
						|
		struct zwp_linux_dmabuf_feedback_v1 *feedback, int fd, uint32_t size) {
 | 
						|
	struct wlr_wl_linux_dmabuf_feedback_v1 *feedback_data = data;
 | 
						|
 | 
						|
	feedback_data->format_table = NULL;
 | 
						|
 | 
						|
	void *table_data = mmap(NULL, size, PROT_READ, MAP_PRIVATE, fd, 0);
 | 
						|
	if (table_data == MAP_FAILED) {
 | 
						|
		wlr_log_errno(WLR_ERROR, "failed to mmap DMA-BUF format table");
 | 
						|
	} else {
 | 
						|
		feedback_data->format_table = table_data;
 | 
						|
		feedback_data->format_table_size = size;
 | 
						|
	}
 | 
						|
	close(fd);
 | 
						|
}
 | 
						|
 | 
						|
static void linux_dmabuf_feedback_v1_handle_main_device(void *data,
 | 
						|
		struct zwp_linux_dmabuf_feedback_v1 *feedback,
 | 
						|
		struct wl_array *dev_id_arr) {
 | 
						|
	struct wlr_wl_linux_dmabuf_feedback_v1 *feedback_data = data;
 | 
						|
 | 
						|
	dev_t dev_id;
 | 
						|
	assert(dev_id_arr->size == sizeof(dev_id));
 | 
						|
	memcpy(&dev_id, dev_id_arr->data, sizeof(dev_id));
 | 
						|
 | 
						|
	feedback_data->main_device_id = dev_id;
 | 
						|
 | 
						|
	drmDevice *device = NULL;
 | 
						|
	if (drmGetDeviceFromDevId(dev_id, 0, &device) != 0) {
 | 
						|
		wlr_log_errno(WLR_ERROR, "drmGetDeviceFromDevId failed");
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	const char *name = NULL;
 | 
						|
	if (device->available_nodes & (1 << DRM_NODE_RENDER)) {
 | 
						|
		name = device->nodes[DRM_NODE_RENDER];
 | 
						|
	} else {
 | 
						|
		// Likely a split display/render setup. Pick the primary node and hope
 | 
						|
		// Mesa will open the right render node under-the-hood.
 | 
						|
		assert(device->available_nodes & (1 << DRM_NODE_PRIMARY));
 | 
						|
		name = device->nodes[DRM_NODE_PRIMARY];
 | 
						|
		wlr_log(WLR_DEBUG, "DRM device %s has no render node, "
 | 
						|
			"falling back to primary node", name);
 | 
						|
	}
 | 
						|
 | 
						|
	struct wlr_wl_backend *wl = feedback_data->backend;
 | 
						|
	assert(wl->drm_render_name == NULL);
 | 
						|
	wl->drm_render_name = strdup(name);
 | 
						|
 | 
						|
	drmFreeDevice(&device);
 | 
						|
}
 | 
						|
 | 
						|
static void linux_dmabuf_feedback_v1_handle_tranche_done(void *data,
 | 
						|
		struct zwp_linux_dmabuf_feedback_v1 *feedback) {
 | 
						|
	struct wlr_wl_linux_dmabuf_feedback_v1 *feedback_data = data;
 | 
						|
	feedback_data->tranche_target_device_id = 0;
 | 
						|
}
 | 
						|
 | 
						|
static void linux_dmabuf_feedback_v1_handle_tranche_target_device(void *data,
 | 
						|
		struct zwp_linux_dmabuf_feedback_v1 *feedback,
 | 
						|
		struct wl_array *dev_id_arr) {
 | 
						|
	struct wlr_wl_linux_dmabuf_feedback_v1 *feedback_data = data;
 | 
						|
 | 
						|
	dev_t dev_id;
 | 
						|
	assert(dev_id_arr->size == sizeof(dev_id));
 | 
						|
	memcpy(&dev_id, dev_id_arr->data, sizeof(dev_id));
 | 
						|
 | 
						|
	feedback_data->tranche_target_device_id = dev_id;
 | 
						|
}
 | 
						|
 | 
						|
static void linux_dmabuf_feedback_v1_handle_tranche_formats(void *data,
 | 
						|
		struct zwp_linux_dmabuf_feedback_v1 *feedback,
 | 
						|
		struct wl_array *indices_arr) {
 | 
						|
	struct wlr_wl_linux_dmabuf_feedback_v1 *feedback_data = data;
 | 
						|
 | 
						|
	if (feedback_data->format_table == NULL) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	if (feedback_data->tranche_target_device_id != feedback_data->main_device_id) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	size_t table_cap = feedback_data->format_table_size /
 | 
						|
		sizeof(struct wlr_wl_linux_dmabuf_v1_table_entry);
 | 
						|
	uint16_t *index_ptr;
 | 
						|
	wl_array_for_each(index_ptr, indices_arr) {
 | 
						|
		assert(*index_ptr < table_cap);
 | 
						|
		const struct wlr_wl_linux_dmabuf_v1_table_entry *entry =
 | 
						|
			&feedback_data->format_table[*index_ptr];
 | 
						|
		wlr_drm_format_set_add(&feedback_data->backend->linux_dmabuf_v1_formats,
 | 
						|
			entry->format, entry->modifier);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void linux_dmabuf_feedback_v1_handle_tranche_flags(void *data,
 | 
						|
		struct zwp_linux_dmabuf_feedback_v1 *feedback, uint32_t flags) {
 | 
						|
	// TODO: handle SCANOUT flag
 | 
						|
}
 | 
						|
 | 
						|
static const struct zwp_linux_dmabuf_feedback_v1_listener
 | 
						|
		linux_dmabuf_feedback_v1_listener = {
 | 
						|
	.done = linux_dmabuf_feedback_v1_handle_done,
 | 
						|
	.format_table = linux_dmabuf_feedback_v1_handle_format_table,
 | 
						|
	.main_device = linux_dmabuf_feedback_v1_handle_main_device,
 | 
						|
	.tranche_done = linux_dmabuf_feedback_v1_handle_tranche_done,
 | 
						|
	.tranche_target_device = linux_dmabuf_feedback_v1_handle_tranche_target_device,
 | 
						|
	.tranche_formats = linux_dmabuf_feedback_v1_handle_tranche_formats,
 | 
						|
	.tranche_flags = linux_dmabuf_feedback_v1_handle_tranche_flags,
 | 
						|
};
 | 
						|
 | 
						|
static bool device_has_name(const drmDevice *device, const char *name) {
 | 
						|
	for (size_t i = 0; i < DRM_NODE_MAX; i++) {
 | 
						|
		if (!(device->available_nodes & (1 << i))) {
 | 
						|
			continue;
 | 
						|
		}
 | 
						|
		if (strcmp(device->nodes[i], name) == 0) {
 | 
						|
			return true;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return false;
 | 
						|
}
 | 
						|
 | 
						|
static char *get_render_name(const char *name) {
 | 
						|
	uint32_t flags = 0;
 | 
						|
	int devices_len = drmGetDevices2(flags, NULL, 0);
 | 
						|
	if (devices_len < 0) {
 | 
						|
		wlr_log(WLR_ERROR, "drmGetDevices2 failed: %s", strerror(-devices_len));
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
	drmDevice **devices = calloc(devices_len, sizeof(*devices));
 | 
						|
	if (devices == NULL) {
 | 
						|
		wlr_log_errno(WLR_ERROR, "Allocation failed");
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
	devices_len = drmGetDevices2(flags, devices, devices_len);
 | 
						|
	if (devices_len < 0) {
 | 
						|
		free(devices);
 | 
						|
		wlr_log(WLR_ERROR, "drmGetDevices2 failed: %s", strerror(-devices_len));
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	const drmDevice *match = NULL;
 | 
						|
	for (int i = 0; i < devices_len; i++) {
 | 
						|
		if (device_has_name(devices[i], name)) {
 | 
						|
			match = devices[i];
 | 
						|
			break;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	char *render_name = NULL;
 | 
						|
	if (match == NULL) {
 | 
						|
		wlr_log(WLR_ERROR, "Cannot find DRM device %s", name);
 | 
						|
	} else if (!(match->available_nodes & (1 << DRM_NODE_RENDER))) {
 | 
						|
		// Likely a split display/render setup. Pick the primary node and hope
 | 
						|
		// Mesa will open the right render node under-the-hood.
 | 
						|
		wlr_log(WLR_DEBUG, "DRM device %s has no render node, "
 | 
						|
			"falling back to primary node", name);
 | 
						|
		assert(match->available_nodes & (1 << DRM_NODE_PRIMARY));
 | 
						|
		render_name = strdup(match->nodes[DRM_NODE_PRIMARY]);
 | 
						|
	} else {
 | 
						|
		render_name = strdup(match->nodes[DRM_NODE_RENDER]);
 | 
						|
	}
 | 
						|
 | 
						|
	for (int i = 0; i < devices_len; i++) {
 | 
						|
		drmFreeDevice(&devices[i]);
 | 
						|
	}
 | 
						|
	free(devices);
 | 
						|
 | 
						|
	return render_name;
 | 
						|
}
 | 
						|
 | 
						|
static void legacy_drm_handle_device(void *data, struct wl_drm *drm,
 | 
						|
		const char *name) {
 | 
						|
	struct wlr_wl_backend *wl = data;
 | 
						|
	assert(wl->drm_render_name == NULL);
 | 
						|
	wl->drm_render_name = get_render_name(name);
 | 
						|
}
 | 
						|
 | 
						|
static void legacy_drm_handle_format(void *data, struct wl_drm *drm,
 | 
						|
		uint32_t format) {
 | 
						|
	// This space is intentionally left blank
 | 
						|
}
 | 
						|
 | 
						|
static void legacy_drm_handle_authenticated(void *data, struct wl_drm *drm) {
 | 
						|
	// This space is intentionally left blank
 | 
						|
}
 | 
						|
 | 
						|
static void legacy_drm_handle_capabilities(void *data, struct wl_drm *drm,
 | 
						|
		uint32_t caps) {
 | 
						|
	// This space is intentionally left blank
 | 
						|
}
 | 
						|
 | 
						|
static const struct wl_drm_listener legacy_drm_listener = {
 | 
						|
	.device = legacy_drm_handle_device,
 | 
						|
	.format = legacy_drm_handle_format,
 | 
						|
	.authenticated = legacy_drm_handle_authenticated,
 | 
						|
	.capabilities = legacy_drm_handle_capabilities,
 | 
						|
};
 | 
						|
 | 
						|
static void shm_handle_format(void *data, struct wl_shm *shm,
 | 
						|
		uint32_t shm_format) {
 | 
						|
	struct wlr_wl_backend *wl = data;
 | 
						|
	uint32_t drm_format = convert_wl_shm_format_to_drm(shm_format);
 | 
						|
	wlr_drm_format_set_add(&wl->shm_formats, drm_format, DRM_FORMAT_MOD_INVALID);
 | 
						|
}
 | 
						|
 | 
						|
static const struct wl_shm_listener shm_listener = {
 | 
						|
	.format = shm_handle_format,
 | 
						|
};
 | 
						|
 | 
						|
static void registry_global(void *data, struct wl_registry *registry,
 | 
						|
		uint32_t name, const char *iface, uint32_t version) {
 | 
						|
	struct wlr_wl_backend *wl = data;
 | 
						|
 | 
						|
	wlr_log(WLR_DEBUG, "Remote wayland global: %s v%" PRIu32, iface, version);
 | 
						|
 | 
						|
	if (strcmp(iface, wl_compositor_interface.name) == 0) {
 | 
						|
		wl->compositor = wl_registry_bind(registry, name,
 | 
						|
			&wl_compositor_interface, 4);
 | 
						|
	} else if (strcmp(iface, wl_seat_interface.name) == 0) {
 | 
						|
		uint32_t target_version = version;
 | 
						|
		if (version < 5) {
 | 
						|
			target_version = 5;
 | 
						|
		}
 | 
						|
		if (version > 9) {
 | 
						|
			target_version = 9;
 | 
						|
		}
 | 
						|
		struct wl_seat *wl_seat = wl_registry_bind(registry, name,
 | 
						|
			&wl_seat_interface, target_version);
 | 
						|
		if (!create_wl_seat(wl_seat, wl, name)) {
 | 
						|
			wl_seat_destroy(wl_seat);
 | 
						|
		}
 | 
						|
	} else if (strcmp(iface, xdg_wm_base_interface.name) == 0) {
 | 
						|
		wl->xdg_wm_base = wl_registry_bind(registry, name,
 | 
						|
			&xdg_wm_base_interface, 1);
 | 
						|
		xdg_wm_base_add_listener(wl->xdg_wm_base, &xdg_wm_base_listener, NULL);
 | 
						|
	} else if (strcmp(iface, zxdg_decoration_manager_v1_interface.name) == 0) {
 | 
						|
		wl->zxdg_decoration_manager_v1 = wl_registry_bind(registry, name,
 | 
						|
			&zxdg_decoration_manager_v1_interface, 1);
 | 
						|
	} else if (strcmp(iface, zwp_pointer_gestures_v1_interface.name) == 0) {
 | 
						|
		wl->zwp_pointer_gestures_v1 = wl_registry_bind(registry, name,
 | 
						|
			&zwp_pointer_gestures_v1_interface, version < 3 ? version : 3);
 | 
						|
	} else if (strcmp(iface, wp_presentation_interface.name) == 0) {
 | 
						|
		wl->presentation = wl_registry_bind(registry, name,
 | 
						|
			&wp_presentation_interface, 1);
 | 
						|
		wp_presentation_add_listener(wl->presentation,
 | 
						|
			&presentation_listener, wl);
 | 
						|
	} else if (strcmp(iface, zwp_tablet_manager_v2_interface.name) == 0) {
 | 
						|
		wl->tablet_manager = wl_registry_bind(registry, name,
 | 
						|
			&zwp_tablet_manager_v2_interface, 1);
 | 
						|
	} else if (strcmp(iface, zwp_linux_dmabuf_v1_interface.name) == 0 &&
 | 
						|
			version >= 3) {
 | 
						|
		wl->zwp_linux_dmabuf_v1 = wl_registry_bind(registry, name,
 | 
						|
			&zwp_linux_dmabuf_v1_interface, version >= 4 ? 4 : version);
 | 
						|
		zwp_linux_dmabuf_v1_add_listener(wl->zwp_linux_dmabuf_v1,
 | 
						|
			&linux_dmabuf_v1_listener, wl);
 | 
						|
	} else if (strcmp(iface, zwp_relative_pointer_manager_v1_interface.name) == 0) {
 | 
						|
		wl->zwp_relative_pointer_manager_v1 = wl_registry_bind(registry, name,
 | 
						|
			&zwp_relative_pointer_manager_v1_interface, 1);
 | 
						|
	} else if (strcmp(iface, wl_drm_interface.name) == 0) {
 | 
						|
		wl->legacy_drm = wl_registry_bind(registry, name, &wl_drm_interface, 1);
 | 
						|
		wl_drm_add_listener(wl->legacy_drm, &legacy_drm_listener, wl);
 | 
						|
	} else if (strcmp(iface, wl_shm_interface.name) == 0) {
 | 
						|
		uint32_t target_version = version;
 | 
						|
		if (version > 2) {
 | 
						|
			target_version = 2;
 | 
						|
		}
 | 
						|
		wl->shm = wl_registry_bind(registry, name, &wl_shm_interface, target_version);
 | 
						|
		wl_shm_add_listener(wl->shm, &shm_listener, wl);
 | 
						|
	} else if (strcmp(iface, xdg_activation_v1_interface.name) == 0) {
 | 
						|
		wl->activation_v1 = wl_registry_bind(registry, name,
 | 
						|
			&xdg_activation_v1_interface, 1);
 | 
						|
	} else if (strcmp(iface, wl_subcompositor_interface.name) == 0) {
 | 
						|
		wl->subcompositor = wl_registry_bind(registry, name,
 | 
						|
			&wl_subcompositor_interface, 1);
 | 
						|
	} else if (strcmp(iface, wp_viewporter_interface.name) == 0) {
 | 
						|
		wl->viewporter = wl_registry_bind(registry, name,
 | 
						|
			&wp_viewporter_interface, 1);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void registry_global_remove(void *data, struct wl_registry *registry,
 | 
						|
		uint32_t name) {
 | 
						|
	struct wlr_wl_backend *wl = data;
 | 
						|
 | 
						|
	struct wlr_wl_seat *seat;
 | 
						|
	wl_list_for_each(seat, &wl->seats, link) {
 | 
						|
		if (seat->global_name == name) {
 | 
						|
			destroy_wl_seat(seat);
 | 
						|
			break;
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static const struct wl_registry_listener registry_listener = {
 | 
						|
	.global = registry_global,
 | 
						|
	.global_remove = registry_global_remove
 | 
						|
};
 | 
						|
 | 
						|
/*
 | 
						|
 * Initializes the wayland backend. Opens a connection to a remote wayland
 | 
						|
 * compositor and creates surfaces for each output, then registers globals on
 | 
						|
 * the specified display.
 | 
						|
 */
 | 
						|
static bool backend_start(struct wlr_backend *backend) {
 | 
						|
	struct wlr_wl_backend *wl = get_wl_backend_from_backend(backend);
 | 
						|
	wlr_log(WLR_INFO, "Starting Wayland backend");
 | 
						|
 | 
						|
	wl->started = true;
 | 
						|
 | 
						|
	struct wlr_wl_seat *seat;
 | 
						|
	wl_list_for_each(seat, &wl->seats, link) {
 | 
						|
		if (seat->wl_keyboard) {
 | 
						|
			init_seat_keyboard(seat);
 | 
						|
		}
 | 
						|
 | 
						|
		if (seat->wl_touch) {
 | 
						|
			init_seat_touch(seat);
 | 
						|
		}
 | 
						|
 | 
						|
		if (wl->tablet_manager) {
 | 
						|
			init_seat_tablet(seat);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	for (size_t i = 0; i < wl->requested_outputs; ++i) {
 | 
						|
		wlr_wl_output_create(&wl->backend);
 | 
						|
	}
 | 
						|
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
static void backend_destroy(struct wlr_backend *backend) {
 | 
						|
	if (!backend) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	struct wlr_wl_backend *wl = get_wl_backend_from_backend(backend);
 | 
						|
 | 
						|
	struct wlr_wl_output *output, *tmp_output;
 | 
						|
	wl_list_for_each_safe(output, tmp_output, &wl->outputs, link) {
 | 
						|
		wlr_output_destroy(&output->wlr_output);
 | 
						|
	}
 | 
						|
 | 
						|
	// Avoid using wl_list_for_each_safe() here: destroying a buffer may
 | 
						|
	// have the side-effect of destroying the next one in the list
 | 
						|
	while (!wl_list_empty(&wl->buffers)) {
 | 
						|
		struct wlr_wl_buffer *buffer = wl_container_of(wl->buffers.next, buffer, link);
 | 
						|
		destroy_wl_buffer(buffer);
 | 
						|
	}
 | 
						|
 | 
						|
	wlr_backend_finish(backend);
 | 
						|
 | 
						|
	wl_list_remove(&wl->event_loop_destroy.link);
 | 
						|
 | 
						|
	wl_event_source_remove(wl->remote_display_src);
 | 
						|
 | 
						|
	close(wl->drm_fd);
 | 
						|
 | 
						|
	wlr_drm_format_set_finish(&wl->shm_formats);
 | 
						|
	wlr_drm_format_set_finish(&wl->linux_dmabuf_v1_formats);
 | 
						|
 | 
						|
	struct wlr_wl_seat *seat, *tmp_seat;
 | 
						|
	wl_list_for_each_safe(seat, tmp_seat, &wl->seats, link) {
 | 
						|
		destroy_wl_seat(seat);
 | 
						|
	}
 | 
						|
 | 
						|
	if (wl->activation_v1) {
 | 
						|
		xdg_activation_v1_destroy(wl->activation_v1);
 | 
						|
	}
 | 
						|
	if (wl->zxdg_decoration_manager_v1) {
 | 
						|
		zxdg_decoration_manager_v1_destroy(wl->zxdg_decoration_manager_v1);
 | 
						|
	}
 | 
						|
	if (wl->zwp_pointer_gestures_v1) {
 | 
						|
		zwp_pointer_gestures_v1_destroy(wl->zwp_pointer_gestures_v1);
 | 
						|
	}
 | 
						|
	if (wl->tablet_manager) {
 | 
						|
		zwp_tablet_manager_v2_destroy(wl->tablet_manager);
 | 
						|
	}
 | 
						|
	if (wl->presentation) {
 | 
						|
		wp_presentation_destroy(wl->presentation);
 | 
						|
	}
 | 
						|
	if (wl->zwp_linux_dmabuf_v1) {
 | 
						|
		zwp_linux_dmabuf_v1_destroy(wl->zwp_linux_dmabuf_v1);
 | 
						|
	}
 | 
						|
	if (wl->legacy_drm != NULL) {
 | 
						|
		wl_drm_destroy(wl->legacy_drm);
 | 
						|
	}
 | 
						|
	if (wl->shm) {
 | 
						|
		if (wl_shm_get_version(wl->shm) >= WL_SHM_RELEASE_SINCE_VERSION) {
 | 
						|
			wl_shm_release(wl->shm);
 | 
						|
		} else {
 | 
						|
			wl_shm_destroy(wl->shm);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if (wl->zwp_relative_pointer_manager_v1) {
 | 
						|
		zwp_relative_pointer_manager_v1_destroy(wl->zwp_relative_pointer_manager_v1);
 | 
						|
	}
 | 
						|
	if (wl->subcompositor) {
 | 
						|
		wl_subcompositor_destroy(wl->subcompositor);
 | 
						|
	}
 | 
						|
	if (wl->viewporter) {
 | 
						|
		wp_viewporter_destroy(wl->viewporter);
 | 
						|
	}
 | 
						|
	free(wl->drm_render_name);
 | 
						|
	free(wl->activation_token);
 | 
						|
	xdg_wm_base_destroy(wl->xdg_wm_base);
 | 
						|
	wl_compositor_destroy(wl->compositor);
 | 
						|
	wl_registry_destroy(wl->registry);
 | 
						|
	wl_display_flush(wl->remote_display);
 | 
						|
	if (wl->own_remote_display) {
 | 
						|
		wl_display_disconnect(wl->remote_display);
 | 
						|
	}
 | 
						|
	free(wl);
 | 
						|
}
 | 
						|
 | 
						|
static int backend_get_drm_fd(struct wlr_backend *backend) {
 | 
						|
	struct wlr_wl_backend *wl = get_wl_backend_from_backend(backend);
 | 
						|
	return wl->drm_fd;
 | 
						|
}
 | 
						|
 | 
						|
static uint32_t get_buffer_caps(struct wlr_backend *backend) {
 | 
						|
	struct wlr_wl_backend *wl = get_wl_backend_from_backend(backend);
 | 
						|
	return (wl->zwp_linux_dmabuf_v1 ? WLR_BUFFER_CAP_DMABUF : 0)
 | 
						|
		| (wl->shm ? WLR_BUFFER_CAP_SHM : 0);
 | 
						|
}
 | 
						|
 | 
						|
static const struct wlr_backend_impl backend_impl = {
 | 
						|
	.start = backend_start,
 | 
						|
	.destroy = backend_destroy,
 | 
						|
	.get_drm_fd = backend_get_drm_fd,
 | 
						|
	.get_buffer_caps = get_buffer_caps,
 | 
						|
};
 | 
						|
 | 
						|
bool wlr_backend_is_wl(struct wlr_backend *b) {
 | 
						|
	return b->impl == &backend_impl;
 | 
						|
}
 | 
						|
 | 
						|
static void handle_event_loop_destroy(struct wl_listener *listener, void *data) {
 | 
						|
	struct wlr_wl_backend *wl = wl_container_of(listener, wl, event_loop_destroy);
 | 
						|
	backend_destroy(&wl->backend);
 | 
						|
}
 | 
						|
 | 
						|
struct wlr_backend *wlr_wl_backend_create(struct wl_event_loop *loop,
 | 
						|
		struct wl_display *remote_display) {
 | 
						|
	wlr_log(WLR_INFO, "Creating wayland backend");
 | 
						|
 | 
						|
	struct wlr_wl_backend *wl = calloc(1, sizeof(*wl));
 | 
						|
	if (!wl) {
 | 
						|
		wlr_log_errno(WLR_ERROR, "Allocation failed");
 | 
						|
		return NULL;
 | 
						|
	}
 | 
						|
 | 
						|
	wlr_backend_init(&wl->backend, &backend_impl);
 | 
						|
 | 
						|
	wl->event_loop = loop;
 | 
						|
	wl_list_init(&wl->outputs);
 | 
						|
	wl_list_init(&wl->seats);
 | 
						|
	wl_list_init(&wl->buffers);
 | 
						|
 | 
						|
	if (remote_display != NULL) {
 | 
						|
		wl->remote_display = remote_display;
 | 
						|
	} else {
 | 
						|
		wl->remote_display = wl_display_connect(NULL);
 | 
						|
		if (!wl->remote_display) {
 | 
						|
			wlr_log_errno(WLR_ERROR, "Could not connect to remote display");
 | 
						|
			goto error_wl;
 | 
						|
		}
 | 
						|
		wl->own_remote_display = true;
 | 
						|
	}
 | 
						|
 | 
						|
	wl->registry = wl_display_get_registry(wl->remote_display);
 | 
						|
	if (!wl->registry) {
 | 
						|
		wlr_log_errno(WLR_ERROR, "Could not obtain reference to remote registry");
 | 
						|
		goto error_display;
 | 
						|
	}
 | 
						|
	wl_registry_add_listener(wl->registry, ®istry_listener, wl);
 | 
						|
 | 
						|
	wl_display_roundtrip(wl->remote_display); // get globals
 | 
						|
 | 
						|
	if (!wl->compositor) {
 | 
						|
		wlr_log(WLR_ERROR,
 | 
						|
			"Remote Wayland compositor does not support wl_compositor");
 | 
						|
		goto error_registry;
 | 
						|
	}
 | 
						|
	if (!wl->xdg_wm_base) {
 | 
						|
		wlr_log(WLR_ERROR,
 | 
						|
			"Remote Wayland compositor does not support xdg-shell");
 | 
						|
		goto error_registry;
 | 
						|
	}
 | 
						|
 | 
						|
	wl_display_roundtrip(wl->remote_display); // process initial event bursts
 | 
						|
 | 
						|
	struct zwp_linux_dmabuf_feedback_v1 *linux_dmabuf_feedback_v1 = NULL;
 | 
						|
	struct wlr_wl_linux_dmabuf_feedback_v1 feedback_data = { .backend = wl };
 | 
						|
	if (wl->zwp_linux_dmabuf_v1 != NULL &&
 | 
						|
			zwp_linux_dmabuf_v1_get_version(wl->zwp_linux_dmabuf_v1) >=
 | 
						|
			ZWP_LINUX_DMABUF_V1_GET_DEFAULT_FEEDBACK_SINCE_VERSION) {
 | 
						|
		linux_dmabuf_feedback_v1 =
 | 
						|
			zwp_linux_dmabuf_v1_get_default_feedback(wl->zwp_linux_dmabuf_v1);
 | 
						|
		if (linux_dmabuf_feedback_v1 == NULL) {
 | 
						|
			wlr_log(WLR_ERROR, "Allocation failed");
 | 
						|
			goto error_registry;
 | 
						|
		}
 | 
						|
		zwp_linux_dmabuf_feedback_v1_add_listener(linux_dmabuf_feedback_v1,
 | 
						|
			&linux_dmabuf_feedback_v1_listener, &feedback_data);
 | 
						|
 | 
						|
		if (wl->legacy_drm != NULL) {
 | 
						|
			wl_drm_destroy(wl->legacy_drm);
 | 
						|
			wl->legacy_drm = NULL;
 | 
						|
 | 
						|
			free(wl->drm_render_name);
 | 
						|
			wl->drm_render_name = NULL;
 | 
						|
		}
 | 
						|
 | 
						|
		wl_display_roundtrip(wl->remote_display); // get linux-dmabuf feedback events
 | 
						|
 | 
						|
		if (feedback_data.format_table != NULL) {
 | 
						|
			munmap(feedback_data.format_table, feedback_data.format_table_size);
 | 
						|
		}
 | 
						|
 | 
						|
		zwp_linux_dmabuf_feedback_v1_destroy(linux_dmabuf_feedback_v1);
 | 
						|
	}
 | 
						|
 | 
						|
	int fd = wl_display_get_fd(wl->remote_display);
 | 
						|
	wl->remote_display_src = wl_event_loop_add_fd(loop, fd, WL_EVENT_READABLE,
 | 
						|
		dispatch_events, wl);
 | 
						|
	if (!wl->remote_display_src) {
 | 
						|
		wlr_log(WLR_ERROR, "Failed to create event source");
 | 
						|
		goto error_registry;
 | 
						|
	}
 | 
						|
	wl_event_source_check(wl->remote_display_src);
 | 
						|
 | 
						|
	if (wl->drm_render_name != NULL) {
 | 
						|
		wlr_log(WLR_DEBUG, "Opening DRM render node %s", wl->drm_render_name);
 | 
						|
		wl->drm_fd = open(wl->drm_render_name, O_RDWR | O_NONBLOCK | O_CLOEXEC);
 | 
						|
		if (wl->drm_fd < 0) {
 | 
						|
			wlr_log_errno(WLR_ERROR, "Failed to open DRM render node %s",
 | 
						|
				wl->drm_render_name);
 | 
						|
			goto error_remote_display_src;
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		wl->drm_fd = -1;
 | 
						|
	}
 | 
						|
 | 
						|
	wl->event_loop_destroy.notify = handle_event_loop_destroy;
 | 
						|
	wl_event_loop_add_destroy_listener(loop, &wl->event_loop_destroy);
 | 
						|
 | 
						|
	const char *token = getenv("XDG_ACTIVATION_TOKEN");
 | 
						|
	if (token != NULL) {
 | 
						|
		wl->activation_token = strdup(token);
 | 
						|
		unsetenv("XDG_ACTIVATION_TOKEN");
 | 
						|
	}
 | 
						|
 | 
						|
	return &wl->backend;
 | 
						|
 | 
						|
error_remote_display_src:
 | 
						|
	wl_event_source_remove(wl->remote_display_src);
 | 
						|
error_registry:
 | 
						|
	free(wl->drm_render_name);
 | 
						|
	if (wl->compositor) {
 | 
						|
		wl_compositor_destroy(wl->compositor);
 | 
						|
	}
 | 
						|
	if (wl->xdg_wm_base) {
 | 
						|
		xdg_wm_base_destroy(wl->xdg_wm_base);
 | 
						|
	}
 | 
						|
	wl_registry_destroy(wl->registry);
 | 
						|
error_display:
 | 
						|
	if (wl->own_remote_display) {
 | 
						|
		wl_display_disconnect(wl->remote_display);
 | 
						|
	}
 | 
						|
error_wl:
 | 
						|
	wlr_backend_finish(&wl->backend);
 | 
						|
	free(wl);
 | 
						|
	return NULL;
 | 
						|
}
 | 
						|
 | 
						|
struct wl_display *wlr_wl_backend_get_remote_display(struct wlr_backend *backend) {
 | 
						|
	struct wlr_wl_backend *wl = get_wl_backend_from_backend(backend);
 | 
						|
	return wl->remote_display;
 | 
						|
}
 |