Inital commit of libotd.
This commit is contained in:
		
							parent
							
								
									1aed987301
								
							
						
					
					
						commit
						aca13320b3
					
				|  | @ -0,0 +1,466 @@ | ||||||
|  | #include "otd.h" | ||||||
|  | #include "drm.h" | ||||||
|  | #include "event.h" | ||||||
|  | 
 | ||||||
|  | #include <assert.h> | ||||||
|  | #include <stdio.h> | ||||||
|  | #include <stdlib.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <inttypes.h> | ||||||
|  | #include <errno.h> | ||||||
|  | #include <xf86drm.h> | ||||||
|  | #include <xf86drmMode.h> | ||||||
|  | #include <drm_mode.h> | ||||||
|  | #include <EGL/egl.h> | ||||||
|  | #include <EGL/eglext.h> | ||||||
|  | #include <gbm.h> | ||||||
|  | #include <GLES3/gl3.h> | ||||||
|  | 
 | ||||||
|  | static const char *conn_name[] = { | ||||||
|  | 	[DRM_MODE_CONNECTOR_Unknown]     = "Unknown", | ||||||
|  | 	[DRM_MODE_CONNECTOR_VGA]         = "VGA", | ||||||
|  | 	[DRM_MODE_CONNECTOR_DVII]        = "DVI-I", | ||||||
|  | 	[DRM_MODE_CONNECTOR_DVID]        = "DVI-D", | ||||||
|  | 	[DRM_MODE_CONNECTOR_DVIA]        = "DVI-A", | ||||||
|  | 	[DRM_MODE_CONNECTOR_Composite]   = "Composite", | ||||||
|  | 	[DRM_MODE_CONNECTOR_SVIDEO]      = "SVIDEO", | ||||||
|  | 	[DRM_MODE_CONNECTOR_LVDS]        = "LVDS", | ||||||
|  | 	[DRM_MODE_CONNECTOR_Component]   = "Component", | ||||||
|  | 	[DRM_MODE_CONNECTOR_9PinDIN]     = "DIN", | ||||||
|  | 	[DRM_MODE_CONNECTOR_DisplayPort] = "DP", | ||||||
|  | 	[DRM_MODE_CONNECTOR_HDMIA]       = "HDMI-A", | ||||||
|  | 	[DRM_MODE_CONNECTOR_HDMIB]       = "HDMI-B", | ||||||
|  | 	[DRM_MODE_CONNECTOR_TV]          = "TV", | ||||||
|  | 	[DRM_MODE_CONNECTOR_eDP]         = "eDP", | ||||||
|  | 	[DRM_MODE_CONNECTOR_VIRTUAL]     = "Virtual", | ||||||
|  | 	[DRM_MODE_CONNECTOR_DSI]         = "DSI", | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | // EGL extensions
 | ||||||
|  | PFNEGLGETPLATFORMDISPLAYEXTPROC get_platform_display; | ||||||
|  | PFNEGLCREATEPLATFORMWINDOWSURFACEEXTPROC create_platform_window_surface; | ||||||
|  | 
 | ||||||
|  | static bool egl_exts() | ||||||
|  | { | ||||||
|  | 	get_platform_display = (PFNEGLGETPLATFORMDISPLAYEXTPROC) | ||||||
|  | 		eglGetProcAddress("eglGetPlatformDisplayEXT"); | ||||||
|  | 
 | ||||||
|  | 	create_platform_window_surface = (PFNEGLCREATEPLATFORMWINDOWSURFACEEXTPROC) | ||||||
|  | 		eglGetProcAddress("eglCreatePlatformWindowSurfaceEXT"); | ||||||
|  | 
 | ||||||
|  | 	return get_platform_display && create_platform_window_surface; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | bool egl_get_config(EGLDisplay disp, EGLConfig *out) | ||||||
|  | { | ||||||
|  | 	EGLint count = 0, matched = 0, ret; | ||||||
|  | 
 | ||||||
|  | 	ret = eglGetConfigs(disp, NULL, 0, &count); | ||||||
|  | 	if (ret == EGL_FALSE || count == 0) { | ||||||
|  | 		return false; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	EGLConfig configs[count]; | ||||||
|  | 
 | ||||||
|  | 	ret = eglChooseConfig(disp, NULL, configs, count, &matched); | ||||||
|  | 	if (ret == EGL_FALSE) { | ||||||
|  | 		return false; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	for (int i = 0; i < matched; ++i) { | ||||||
|  | 		EGLint gbm_format; | ||||||
|  | 
 | ||||||
|  | 		if (!eglGetConfigAttrib(disp, | ||||||
|  | 					configs[i], | ||||||
|  | 					EGL_NATIVE_VISUAL_ID, | ||||||
|  | 					&gbm_format)) | ||||||
|  | 			continue; | ||||||
|  | 
 | ||||||
|  | 		if (gbm_format == GBM_FORMAT_XRGB8888) { | ||||||
|  | 			*out = configs[i]; | ||||||
|  | 			return true; | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return false; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | bool init_renderer(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	if (!egl_exts()) { | ||||||
|  | 		fprintf(stderr, "Could not get EGL extensions\n"); | ||||||
|  | 		return false; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	otd->gbm = gbm_create_device(otd->fd); | ||||||
|  | 	if (!otd->gbm) { | ||||||
|  | 		fprintf(stderr, "Could not create gbm device: %s\n", strerror(errno)); | ||||||
|  | 		return false; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (eglBindAPI(EGL_OPENGL_ES_API) == EGL_FALSE) { | ||||||
|  | 		fprintf(stderr, "Could not bind GLES3 API\n"); | ||||||
|  | 		goto error_gbm; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	otd->egl.disp = get_platform_display(EGL_PLATFORM_GBM_MESA, otd->gbm, NULL); | ||||||
|  | 	if (otd->egl.disp == EGL_NO_DISPLAY) { | ||||||
|  | 		fprintf(stderr, "Could not create EGL display\n"); | ||||||
|  | 		goto error_gbm; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (eglInitialize(otd->egl.disp, NULL, NULL) == EGL_FALSE) { | ||||||
|  | 		fprintf(stderr, "Could not initalise EGL\n"); | ||||||
|  | 		goto error_egl; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (!egl_get_config(otd->egl.disp, &otd->egl.conf)) { | ||||||
|  | 		fprintf(stderr, "Could not get EGL config\n"); | ||||||
|  | 		goto error_egl; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	static const EGLint attribs[] = {EGL_CONTEXT_CLIENT_VERSION, 3, EGL_NONE}; | ||||||
|  | 	otd->egl.context = eglCreateContext(otd->egl.disp, otd->egl.conf, EGL_NO_CONTEXT, attribs); | ||||||
|  | 	if (otd->egl.context == EGL_NO_CONTEXT) { | ||||||
|  | 		fprintf(stderr, "Could not create EGL context\n"); | ||||||
|  | 		goto error_egl; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return true; | ||||||
|  | 
 | ||||||
|  | error_egl: | ||||||
|  | 	eglTerminate(otd->egl.disp); | ||||||
|  | 	eglReleaseThread(); | ||||||
|  | 	eglMakeCurrent(EGL_NO_DISPLAY, EGL_NO_SURFACE, EGL_NO_SURFACE, EGL_NO_CONTEXT); | ||||||
|  | 
 | ||||||
|  | error_gbm: | ||||||
|  | 	gbm_device_destroy(otd->gbm); | ||||||
|  | 
 | ||||||
|  | 	return false; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void destroy_renderer(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	if (!otd) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	eglDestroyContext(otd->egl.disp, otd->egl.context); | ||||||
|  | 	eglTerminate(otd->egl.disp); | ||||||
|  | 	eglReleaseThread(); | ||||||
|  | 	eglMakeCurrent(EGL_NO_DISPLAY, EGL_NO_SURFACE, EGL_NO_SURFACE, EGL_NO_CONTEXT); | ||||||
|  | 
 | ||||||
|  | 	gbm_device_destroy(otd->gbm); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void scan_connectors(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	drmModeRes *res = drmModeGetResources(otd->fd); | ||||||
|  | 	if (!res) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	// I don't know if this needs to be allocated with realloc like this,
 | ||||||
|  | 	// as it may not even be possible for the number of connectors to change.
 | ||||||
|  | 	// I'll just have to see how DisplayPort MST works with DRM.
 | ||||||
|  | 	if ((size_t)res->count_connectors > otd->display_len) { | ||||||
|  | 		struct otd_display *new = realloc(otd->displays, sizeof *new * res->count_connectors); | ||||||
|  | 		if (!new) | ||||||
|  | 			goto error; | ||||||
|  | 
 | ||||||
|  | 		for (int i = otd->display_len; i < res->count_connectors; ++i) { | ||||||
|  | 			new[i] = (struct otd_display) { | ||||||
|  | 				.state = OTD_DISP_INVALID, | ||||||
|  | 				.otd = otd, | ||||||
|  | 			}; | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		otd->display_len = res->count_connectors; | ||||||
|  | 		otd->displays = new; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	for (int i = 0; i < res->count_connectors; ++i) { | ||||||
|  | 		struct otd_display *disp = &otd->displays[i]; | ||||||
|  | 		drmModeConnector *conn = drmModeGetConnector(otd->fd, res->connectors[i]); | ||||||
|  | 		if (!conn) | ||||||
|  | 			continue; | ||||||
|  | 
 | ||||||
|  | 		if (otd->displays[i].state == OTD_DISP_INVALID) { | ||||||
|  | 			disp->state = OTD_DISP_DISCONNECTED; | ||||||
|  | 			disp->connector = res->connectors[i]; | ||||||
|  | 			snprintf(disp->name, sizeof disp->name, "%s-%"PRIu32, | ||||||
|  | 				 conn_name[conn->connector_type], | ||||||
|  | 				 conn->connector_type_id); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		if (disp->state == OTD_DISP_DISCONNECTED && | ||||||
|  | 		    conn->connection == DRM_MODE_CONNECTED) { | ||||||
|  | 			disp->state = OTD_DISP_NEEDS_MODESET; | ||||||
|  | 			event_add(otd, disp, OTD_EV_DISPLAY_ADD); | ||||||
|  | 
 | ||||||
|  | 		} else if (disp->state == OTD_DISP_CONNECTED && | ||||||
|  | 		    conn->connection != DRM_MODE_CONNECTED) { | ||||||
|  | 			disp->state = OTD_DISP_DISCONNECTED; | ||||||
|  | 			event_add(otd, disp, OTD_EV_DISPLAY_REM); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		drmModeFreeConnector(conn); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | error: | ||||||
|  | 	drmModeFreeResources(res); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static void free_fb(struct gbm_bo *bo, void *data) | ||||||
|  | { | ||||||
|  | 	uint32_t *id = data; | ||||||
|  | 
 | ||||||
|  | 	if (id && *id) | ||||||
|  | 		drmModeRmFB(gbm_bo_get_fd(bo), *id); | ||||||
|  | 
 | ||||||
|  | 	free(id); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static uint32_t get_fb_for_bo(int fd, struct gbm_bo *bo) | ||||||
|  | { | ||||||
|  | 	uint32_t *id = gbm_bo_get_user_data(bo); | ||||||
|  | 
 | ||||||
|  | 	if (id) | ||||||
|  | 		return *id; | ||||||
|  | 
 | ||||||
|  | 	id = calloc(1, sizeof *id); | ||||||
|  | 	if (!id) | ||||||
|  | 		return 0; | ||||||
|  | 
 | ||||||
|  | 	drmModeAddFB(fd, gbm_bo_get_width(bo), gbm_bo_get_height(bo), 24, 32, | ||||||
|  | 		     gbm_bo_get_stride(bo), gbm_bo_get_handle(bo).u32, id); | ||||||
|  | 
 | ||||||
|  | 	gbm_bo_set_user_data(bo, id, free_fb); | ||||||
|  | 
 | ||||||
|  | 	return *id; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static void init_display_renderer(struct otd *otd, struct otd_display *disp) | ||||||
|  | { | ||||||
|  | 	disp->gbm = gbm_surface_create(otd->gbm, | ||||||
|  | 				       disp->width, disp->height, | ||||||
|  | 				       GBM_FORMAT_XRGB8888, | ||||||
|  | 				       GBM_BO_USE_SCANOUT | GBM_BO_USE_RENDERING); | ||||||
|  | 	if (!disp->gbm) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	disp->egl = create_platform_window_surface(otd->egl.disp, otd->egl.conf, | ||||||
|  | 						   disp->gbm, NULL); | ||||||
|  | 	if (disp->egl == EGL_NO_SURFACE) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	// Render black frame
 | ||||||
|  | 
 | ||||||
|  | 	eglMakeCurrent(otd->egl.disp, disp->egl, disp->egl, otd->egl.context); | ||||||
|  | 
 | ||||||
|  | 	glViewport(0, 0, disp->width, disp->height); | ||||||
|  | 	glClearColor(0.0, 0.0, 0.0, 1.0); | ||||||
|  | 	glClear(GL_COLOR_BUFFER_BIT); | ||||||
|  | 
 | ||||||
|  | 	eglSwapBuffers(otd->egl.disp, disp->egl); | ||||||
|  | 
 | ||||||
|  | 	struct gbm_bo *bo = gbm_surface_lock_front_buffer(disp->gbm); | ||||||
|  | 	uint32_t fb_id = get_fb_for_bo(otd->fd, bo); | ||||||
|  | 
 | ||||||
|  | 	drmModeSetCrtc(otd->fd, disp->crtc, fb_id, 0, 0, | ||||||
|  | 		       &disp->connector, 1, disp->active_mode); | ||||||
|  | 	drmModePageFlip(otd->fd, disp->crtc, fb_id, DRM_MODE_PAGE_FLIP_EVENT, disp); | ||||||
|  | 
 | ||||||
|  | 	gbm_surface_release_buffer(disp->gbm, bo); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static drmModeModeInfo *select_mode(size_t num_modes, | ||||||
|  | 				    drmModeModeInfo modes[static num_modes], | ||||||
|  | 				    drmModeCrtc *old_crtc, | ||||||
|  | 				    const char *str) | ||||||
|  | { | ||||||
|  | 	if (strcmp(str, "preferred") == 0) | ||||||
|  | 		return &modes[0]; | ||||||
|  | 
 | ||||||
|  | 	if (strcmp(str, "current") == 0) { | ||||||
|  | 		if (!old_crtc) { | ||||||
|  | 			fprintf(stderr, "Display does not have currently configured mode\n"); | ||||||
|  | 			return NULL; | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		for (size_t i = 0; i < num_modes; ++i) { | ||||||
|  | 			if (memcmp(&modes[i], &old_crtc->mode, sizeof modes[0]) == 0) | ||||||
|  | 				return &modes[i]; | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// We should never get here
 | ||||||
|  | 		assert(0); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	unsigned width = 0; | ||||||
|  | 	unsigned height = 0; | ||||||
|  | 	unsigned rate = 0; | ||||||
|  | 	int ret; | ||||||
|  | 
 | ||||||
|  | 	if ((ret = sscanf(str, "%ux%u@%u", &width, &height, &rate)) != 2 && ret != 3) { | ||||||
|  | 		fprintf(stderr, "Invalid modesetting string\n"); | ||||||
|  | 		return NULL; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	for (size_t i = 0; i < num_modes; ++i) { | ||||||
|  | 		if (modes[i].hdisplay == width && | ||||||
|  | 		    modes[i].vdisplay == height && | ||||||
|  | 		    (!rate || modes[i].vrefresh == rate)) | ||||||
|  | 			return &modes[i]; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return NULL; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | bool modeset_str(struct otd *otd, struct otd_display *disp, const char *str) | ||||||
|  | { | ||||||
|  | 	drmModeConnector *conn = drmModeGetConnector(otd->fd, disp->connector); | ||||||
|  | 	if (!conn || conn->connection != DRM_MODE_CONNECTED || conn->count_modes == 0) | ||||||
|  | 		goto error; | ||||||
|  | 
 | ||||||
|  | 	disp->num_modes = conn->count_modes; | ||||||
|  | 	disp->modes = malloc(sizeof *disp->modes * disp->num_modes); | ||||||
|  | 	if (!disp->modes) | ||||||
|  | 		goto error; | ||||||
|  | 	memcpy(disp->modes, conn->modes, sizeof *disp->modes * disp->num_modes); | ||||||
|  | 
 | ||||||
|  | 	drmModeEncoder *curr_enc = drmModeGetEncoder(otd->fd, conn->encoder_id); | ||||||
|  | 	if (curr_enc) { | ||||||
|  | 		disp->old_crtc = drmModeGetCrtc(otd->fd, curr_enc->crtc_id); | ||||||
|  | 		free(curr_enc); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	disp->active_mode = select_mode(disp->num_modes, disp->modes, disp->old_crtc, str); | ||||||
|  | 	if (!disp->active_mode) { | ||||||
|  | 		fprintf(stderr, "Could not find mode '%s' for %s\n", str, disp->name); | ||||||
|  | 		goto error; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	fprintf(stderr, "Configuring %s with mode %ux%u@%u\n", | ||||||
|  | 		disp->name, disp->active_mode->hdisplay, disp->active_mode->vdisplay, | ||||||
|  | 		disp->active_mode->vrefresh); | ||||||
|  | 
 | ||||||
|  | 	drmModeRes *res = drmModeGetResources(otd->fd); | ||||||
|  | 	if (!res) | ||||||
|  | 		goto error; | ||||||
|  | 
 | ||||||
|  | 	bool success = false; | ||||||
|  | 	for (int i = 0; !success && i < conn->count_encoders; ++i) { | ||||||
|  | 		drmModeEncoder *enc = drmModeGetEncoder(otd->fd, conn->encoders[i]); | ||||||
|  | 		if (!enc) | ||||||
|  | 			continue; | ||||||
|  | 
 | ||||||
|  | 		for (int j = 0; j < res->count_crtcs; ++j) { | ||||||
|  | 			if ((enc->possible_crtcs & (1 << j)) == 0) | ||||||
|  | 				continue; | ||||||
|  | 
 | ||||||
|  | 			if ((otd->taken_crtcs & (1 << j)) == 0) { | ||||||
|  | 				otd->taken_crtcs |= 1 << j; | ||||||
|  | 				disp->crtc = res->crtcs[j]; | ||||||
|  | 
 | ||||||
|  | 				success = true; | ||||||
|  | 				break; | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		drmModeFreeEncoder(enc); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	drmModeFreeResources(res); | ||||||
|  | 
 | ||||||
|  | 	if (!success) | ||||||
|  | 		goto error; | ||||||
|  | 
 | ||||||
|  | 	disp->state = OTD_DISP_CONNECTED; | ||||||
|  | 	drmModeFreeConnector(conn); | ||||||
|  | 
 | ||||||
|  | 	disp->width = disp->active_mode->hdisplay; | ||||||
|  | 	disp->height = disp->active_mode->vdisplay; | ||||||
|  | 
 | ||||||
|  | 	init_display_renderer(otd, disp); | ||||||
|  | 
 | ||||||
|  | 	return true; | ||||||
|  | error: | ||||||
|  | 	disp->state = OTD_DISP_DISCONNECTED; | ||||||
|  | 	drmModeFreeConnector(conn); | ||||||
|  | 
 | ||||||
|  | 	event_add(otd, disp, OTD_EV_DISPLAY_REM); | ||||||
|  | 
 | ||||||
|  | 	return false; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | void page_flip_handler(int fd, | ||||||
|  | 		       unsigned seq, | ||||||
|  | 		       unsigned tv_sec, | ||||||
|  | 		       unsigned tv_usec, | ||||||
|  | 		       void *user) | ||||||
|  | { | ||||||
|  | 	struct otd_display *disp = user; | ||||||
|  | 
 | ||||||
|  | 	disp->pageflip_pending = true; | ||||||
|  | 	if (!disp->cleanup) | ||||||
|  | 		event_add(disp->otd, disp, OTD_EV_RENDER); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void destroy_display_renderer(struct otd *otd, struct otd_display *disp) | ||||||
|  | { | ||||||
|  | 	if (!otd || !disp || disp->state != OTD_DISP_CONNECTED) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	drmModeCrtc *crtc = disp->old_crtc; | ||||||
|  | 	if (crtc) { | ||||||
|  | 		drmEventContext event = { | ||||||
|  | 			.version = DRM_EVENT_CONTEXT_VERSION, | ||||||
|  | 			.page_flip_handler = page_flip_handler, | ||||||
|  | 		}; | ||||||
|  | 
 | ||||||
|  | 		disp->cleanup = true; | ||||||
|  | 		while (disp->pageflip_pending) | ||||||
|  | 			drmHandleEvent(otd->fd, &event); | ||||||
|  | 
 | ||||||
|  | 		drmModeSetCrtc(otd->fd, crtc->crtc_id, crtc->buffer_id, | ||||||
|  | 			       crtc->x, crtc->y, &disp->connector, | ||||||
|  | 			       1, &crtc->mode); | ||||||
|  | 		drmModeFreeCrtc(crtc); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	eglDestroySurface(otd->egl.disp, disp->egl); | ||||||
|  | 	gbm_surface_destroy(disp->gbm); | ||||||
|  | 
 | ||||||
|  | 	free(disp->modes); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void get_drm_event(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	drmEventContext event = { | ||||||
|  | 		.version = DRM_EVENT_CONTEXT_VERSION, | ||||||
|  | 		.page_flip_handler = page_flip_handler, | ||||||
|  | 	}; | ||||||
|  | 
 | ||||||
|  | 	drmHandleEvent(otd->fd, &event); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void rendering_begin(struct otd_display *disp) | ||||||
|  | { | ||||||
|  | 	struct otd *otd = disp->otd; | ||||||
|  | 	eglMakeCurrent(otd->egl.disp, disp->egl, disp->egl, otd->egl.context); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void rendering_end(struct otd_display *disp) | ||||||
|  | { | ||||||
|  | 	struct otd *otd = disp->otd; | ||||||
|  | 	eglSwapBuffers(otd->egl.disp, disp->egl); | ||||||
|  | 
 | ||||||
|  | 	struct gbm_bo *bo = gbm_surface_lock_front_buffer(disp->gbm); | ||||||
|  | 	uint32_t fb_id = get_fb_for_bo(otd->fd, bo); | ||||||
|  | 
 | ||||||
|  | 	drmModePageFlip(otd->fd, disp->crtc, fb_id, DRM_MODE_PAGE_FLIP_EVENT, disp); | ||||||
|  | 
 | ||||||
|  | 	gbm_surface_release_buffer(disp->gbm, bo); | ||||||
|  | 
 | ||||||
|  | 	disp->pageflip_pending = false; | ||||||
|  | } | ||||||
|  | @ -0,0 +1,54 @@ | ||||||
|  | #ifndef DRM_H | ||||||
|  | #define DRM_H | ||||||
|  | 
 | ||||||
|  | #include <stdbool.h> | ||||||
|  | #include <stdint.h> | ||||||
|  | #include <xf86drmMode.h> | ||||||
|  | #include <EGL/egl.h> | ||||||
|  | #include <gbm.h> | ||||||
|  | 
 | ||||||
|  | enum otd_display_state { | ||||||
|  | 	OTD_DISP_INVALID, | ||||||
|  | 	OTD_DISP_DISCONNECTED, | ||||||
|  | 	OTD_DISP_NEEDS_MODESET, | ||||||
|  | 	OTD_DISP_CONNECTED, | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | struct otd_display { | ||||||
|  | 	struct otd *otd; | ||||||
|  | 
 | ||||||
|  | 	enum otd_display_state state; | ||||||
|  | 	uint32_t connector; | ||||||
|  | 	char name[16]; | ||||||
|  | 
 | ||||||
|  | 	size_t num_modes; | ||||||
|  | 	drmModeModeInfo *modes; | ||||||
|  | 	drmModeModeInfo *active_mode; | ||||||
|  | 
 | ||||||
|  | 	uint32_t width; | ||||||
|  | 	uint32_t height; | ||||||
|  | 
 | ||||||
|  | 	uint32_t crtc; | ||||||
|  | 	drmModeCrtc *old_crtc; | ||||||
|  | 
 | ||||||
|  | 	struct gbm_surface *gbm; | ||||||
|  | 	EGLSurface *egl; | ||||||
|  | 	uint32_t fb_id; | ||||||
|  | 
 | ||||||
|  | 	bool pageflip_pending; | ||||||
|  | 	bool cleanup; | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | bool init_renderer(struct otd *otd); | ||||||
|  | void destroy_renderer(struct otd *otd); | ||||||
|  | 
 | ||||||
|  | void scan_connectors(struct otd *otd); | ||||||
|  | bool modeset_str(struct otd *otd, struct otd_display *disp, const char *str); | ||||||
|  | void destroy_display_renderer(struct otd *otd, struct otd_display *disp); | ||||||
|  | 
 | ||||||
|  | void get_drm_event(struct otd *otd); | ||||||
|  | 
 | ||||||
|  | void rendering_begin(struct otd_display *disp); | ||||||
|  | void rendering_end(struct otd_display *disp); | ||||||
|  | 
 | ||||||
|  | #endif | ||||||
|  | @ -0,0 +1,91 @@ | ||||||
|  | #include "otd.h" | ||||||
|  | #include "event.h" | ||||||
|  | #include "drm.h" | ||||||
|  | #include "udev.h" | ||||||
|  | 
 | ||||||
|  | #include <stdbool.h> | ||||||
|  | #include <stdlib.h> | ||||||
|  | #include <poll.h> | ||||||
|  | 
 | ||||||
|  | static inline void event_swap(struct otd_event *a, struct otd_event *b) | ||||||
|  | { | ||||||
|  | 	struct otd_event tmp = *a; | ||||||
|  | 	*a = *b; | ||||||
|  | 	*b = tmp; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | bool otd_get_event(struct otd *otd, struct otd_event *restrict ret) | ||||||
|  | { | ||||||
|  | 	struct pollfd fds[] = { | ||||||
|  | 		{ .fd = otd->fd, .events = POLLIN }, | ||||||
|  | 		{ .fd = otd->udev_fd, .events = POLLIN }, | ||||||
|  | 	}; | ||||||
|  | 
 | ||||||
|  | 	while (poll(fds, 2, 0) > 0) { | ||||||
|  | 		if (fds[0].revents) | ||||||
|  | 			get_drm_event(otd); | ||||||
|  | 		if (fds[1].revents) | ||||||
|  | 			otd_udev_event(otd); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (otd->event_len == 0) { | ||||||
|  | 		ret->type = OTD_EV_NONE; | ||||||
|  | 		ret->display = NULL; | ||||||
|  | 		return false; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	struct otd_event *ev = otd->events; | ||||||
|  | 
 | ||||||
|  | 	// Downheap
 | ||||||
|  | 	*ret = ev[0]; | ||||||
|  | 	ev[0] = ev[--otd->event_len]; | ||||||
|  | 
 | ||||||
|  | 	size_t i = 0; | ||||||
|  | 	while (i < otd->event_len / 2) { | ||||||
|  | 		size_t left = i * 2 + 1; | ||||||
|  | 		size_t right = i * 2 + 2; | ||||||
|  | 		size_t max = (ev[left].type > ev[right].type) ? left : right; | ||||||
|  | 
 | ||||||
|  | 		if (ev[i].type <= ev[max].type) { | ||||||
|  | 			event_swap(&ev[i], &ev[max]); | ||||||
|  | 			i = max; | ||||||
|  | 		} else { | ||||||
|  | 			break; | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return true; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | bool event_add(struct otd *otd, struct otd_display *disp, enum otd_event_type type) | ||||||
|  | { | ||||||
|  | 	if (type == OTD_EV_NONE) | ||||||
|  | 		return true; | ||||||
|  | 
 | ||||||
|  | 	if (otd->event_len == otd->event_cap) { | ||||||
|  | 		size_t new_size = (otd->event_cap == 0) ? 8 : otd->event_cap * 2; | ||||||
|  | 
 | ||||||
|  | 		struct otd_event *new = realloc(otd->events, sizeof *new * new_size); | ||||||
|  | 		if (!new) { | ||||||
|  | 			return false; | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		otd->event_cap = new_size; | ||||||
|  | 		otd->events = new; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	struct otd_event *ev = otd->events; | ||||||
|  | 
 | ||||||
|  | 	// Upheap
 | ||||||
|  | 	size_t i = otd->event_len++; | ||||||
|  | 	ev[i].type = type; | ||||||
|  | 	ev[i].display = disp; | ||||||
|  | 
 | ||||||
|  | 	size_t j; | ||||||
|  | 	while (i > 0 && ev[i].type > ev[(j = (i - 1) / 2)].type) { | ||||||
|  | 		event_swap(&ev[i], &ev[j]); | ||||||
|  | 		i = j; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return true; | ||||||
|  | } | ||||||
|  | @ -0,0 +1,21 @@ | ||||||
|  | #ifndef EVENT_H | ||||||
|  | #define EVENT_H | ||||||
|  | 
 | ||||||
|  | #include <stdbool.h> | ||||||
|  | 
 | ||||||
|  | enum otd_event_type { | ||||||
|  | 	OTD_EV_NONE, | ||||||
|  | 	OTD_EV_RENDER, | ||||||
|  | 	OTD_EV_DISPLAY_REM, | ||||||
|  | 	OTD_EV_DISPLAY_ADD, | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | struct otd_event { | ||||||
|  | 	enum otd_event_type type; | ||||||
|  | 	struct otd_display *display; | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | bool otd_get_event(struct otd *otd, struct otd_event *restrict ret); | ||||||
|  | bool event_add(struct otd *otd, struct otd_display *disp, enum otd_event_type type); | ||||||
|  | 
 | ||||||
|  | #endif | ||||||
|  | @ -0,0 +1,160 @@ | ||||||
|  | #include <stdio.h> | ||||||
|  | #include <stdbool.h> | ||||||
|  | #include <inttypes.h> | ||||||
|  | #include <time.h> | ||||||
|  | #include <GLES3/gl3.h> | ||||||
|  | #include <unistd.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <stdlib.h> | ||||||
|  | #include <stdnoreturn.h> | ||||||
|  | 
 | ||||||
|  | #include "otd.h" | ||||||
|  | #include "drm.h" | ||||||
|  | #include "event.h" | ||||||
|  | 
 | ||||||
|  | // I know this is pretty crappy,
 | ||||||
|  | // but it's just for the example's sake
 | ||||||
|  | struct { | ||||||
|  | 	char *name; | ||||||
|  | 	char *mode; | ||||||
|  | } displays[8]; | ||||||
|  | int num_displays = 0; | ||||||
|  | 
 | ||||||
|  | noreturn void usage(const char *name, int ret) | ||||||
|  | { | ||||||
|  | 	fprintf(stderr, "usage: %s [-o <name> [-m <mode>]]*\n" | ||||||
|  | 			"\n" | ||||||
|  | 			"  -h       \tDisplay this help text\n" | ||||||
|  | 			"  -o <name>\tWhich diplay to use. e.g. DVI-I-1.\n" | ||||||
|  | 			"  -m <mode>\tWhich mode to use. It must come after an -o option.\n" | ||||||
|  | 			"           \tMust be either 'preferred', 'current', widthxheight\n" | ||||||
|  | 			"           \tor widthxheight@rate. Defaults to 'preferred'.\n" | ||||||
|  | 			"\n" | ||||||
|  | 			"example: %s -o DVI-I-1 -m 1920x1080@60 -o DP-1 -m 1920x1200\n", | ||||||
|  | 			name, name); | ||||||
|  | 
 | ||||||
|  | 	exit(ret); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void parse_args(int argc, char *argv[]) | ||||||
|  | { | ||||||
|  | 	int c; | ||||||
|  | 	int i = -1; | ||||||
|  | 
 | ||||||
|  | 	while ((c = getopt(argc, argv, "ho:m:")) != -1) { | ||||||
|  | 		switch (c) { | ||||||
|  | 		case 'h': | ||||||
|  | 			usage(argv[0], 0); | ||||||
|  | 		case 'o': | ||||||
|  | 			i = num_displays++; | ||||||
|  | 
 | ||||||
|  | 			if (num_displays == 8) { | ||||||
|  | 				fprintf(stderr, "Too many displays\n"); | ||||||
|  | 				exit(1); | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			displays[i].name = strdup(optarg); | ||||||
|  | 			break; | ||||||
|  | 		case 'm': | ||||||
|  | 			if (i == -1) { | ||||||
|  | 				fprintf(stderr, "A display is required first\n"); | ||||||
|  | 				exit(1); | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			fprintf(stderr, "Assigned '%s' to '%s'\n", optarg, displays[i].name); | ||||||
|  | 			displays[i].mode = optarg; | ||||||
|  | 			i = -1; | ||||||
|  | 			break; | ||||||
|  | 		default: | ||||||
|  | 			usage(argv[0], 1); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Trailing args
 | ||||||
|  | 	if (optind != argc) { | ||||||
|  | 		usage(argv[0], 1); | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | int main(int argc, char *argv[]) | ||||||
|  | { | ||||||
|  | 	parse_args(argc, argv); | ||||||
|  | 
 | ||||||
|  | 	struct otd *otd = otd_start(); | ||||||
|  | 	if (!otd) | ||||||
|  | 		return 1; | ||||||
|  | 
 | ||||||
|  | 	float colour[3] = {1.0, 0.0, 0.0}; | ||||||
|  | 	int dec = 0; | ||||||
|  | 
 | ||||||
|  | 	struct timespec start, now; | ||||||
|  | 	clock_gettime(CLOCK_MONOTONIC, &start); | ||||||
|  | 	struct timespec last = start; | ||||||
|  | 
 | ||||||
|  | 	while (clock_gettime(CLOCK_MONOTONIC, &now) == 0 && | ||||||
|  | 	       now.tv_sec < start.tv_sec + 10) { | ||||||
|  | 
 | ||||||
|  | 		struct otd_event ev; | ||||||
|  | 		if (!otd_get_event(otd, &ev)) | ||||||
|  | 			continue; | ||||||
|  | 
 | ||||||
|  | 		struct otd_display *disp = ev.display; | ||||||
|  | 
 | ||||||
|  | 		switch (ev.type) { | ||||||
|  | 		case OTD_EV_RENDER: | ||||||
|  | 			rendering_begin(disp); | ||||||
|  | 
 | ||||||
|  | 			// This is all just calculating the colours.
 | ||||||
|  | 			// It's not particularly important.
 | ||||||
|  | 
 | ||||||
|  | 			long ms = (now.tv_sec - last.tv_sec) * 1000 + | ||||||
|  | 				(now.tv_nsec - last.tv_nsec) / 1000000; | ||||||
|  | 
 | ||||||
|  | 			int inc = (dec + 1) % 3; | ||||||
|  | 
 | ||||||
|  | 			colour[dec] -= ms / 2000.0f; | ||||||
|  | 			colour[inc] += ms / 2000.0f; | ||||||
|  | 
 | ||||||
|  | 			if (colour[dec] < 0.0f) { | ||||||
|  | 				colour[dec] = 0.0f; | ||||||
|  | 				colour[inc] = 1.0f; | ||||||
|  | 
 | ||||||
|  | 				dec = (dec + 1) % 3; | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			last = now; | ||||||
|  | 
 | ||||||
|  | 			glViewport(0, 0, disp->width, disp->height); | ||||||
|  | 			glClearColor(colour[0], colour[1], colour[2], 1.0f); | ||||||
|  | 			glClear(GL_COLOR_BUFFER_BIT); | ||||||
|  | 
 | ||||||
|  | 			rendering_end(disp); | ||||||
|  | 			break; | ||||||
|  | 		case OTD_EV_DISPLAY_REM: | ||||||
|  | 			printf("%s removed\n", disp->name); | ||||||
|  | 			break; | ||||||
|  | 		case OTD_EV_DISPLAY_ADD: | ||||||
|  | 			printf("%s added\n", disp->name); | ||||||
|  | 
 | ||||||
|  | 			const char *mode = NULL; | ||||||
|  | 			for (int i = 0; i < num_displays; ++i) { | ||||||
|  | 				if (strcmp(disp->name, displays[i].name) == 0) { | ||||||
|  | 					mode = displays[i].mode; | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			if (!mode) | ||||||
|  | 				mode = "preferred"; | ||||||
|  | 
 | ||||||
|  | 			if (!modeset_str(otd, ev.display, mode)) { | ||||||
|  | 				fprintf(stderr, "Modesetting %s failed\n", disp->name); | ||||||
|  | 				goto out; | ||||||
|  | 			} | ||||||
|  | 			break; | ||||||
|  | 		default: | ||||||
|  | 			break; | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | out: | ||||||
|  | 	otd_finish(otd); | ||||||
|  | } | ||||||
|  | @ -0,0 +1,71 @@ | ||||||
|  | #include <unistd.h> | ||||||
|  | #include <stdlib.h> | ||||||
|  | #include <stdio.h> | ||||||
|  | 
 | ||||||
|  | #include "otd.h" | ||||||
|  | #include "drm.h" | ||||||
|  | #include "event.h" | ||||||
|  | #include "session.h" | ||||||
|  | #include "udev.h" | ||||||
|  | 
 | ||||||
|  | struct otd *otd_start(void) | ||||||
|  | { | ||||||
|  | 	struct otd *otd = calloc(1, sizeof *otd); | ||||||
|  | 	if (!otd) | ||||||
|  | 		return NULL; | ||||||
|  | 
 | ||||||
|  | 	if (!otd_new_session(otd)) { | ||||||
|  | 		fprintf(stderr, "Could not create session\n"); | ||||||
|  | 		goto error; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (!otd_udev_start(otd)) { | ||||||
|  | 		fprintf(stderr, "Could not start udev\n"); | ||||||
|  | 		goto error_session; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	otd_udev_find_gpu(otd); | ||||||
|  | 	if (otd->fd == -1) { | ||||||
|  | 		fprintf(stderr, "Could not open GPU\n"); | ||||||
|  | 		goto error_udev; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (!init_renderer(otd)) { | ||||||
|  | 		fprintf(stderr, "Could not initalise renderer\n"); | ||||||
|  | 		goto error_fd; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	scan_connectors(otd); | ||||||
|  | 
 | ||||||
|  | 	return otd; | ||||||
|  | 
 | ||||||
|  | error_fd: | ||||||
|  | 	release_device(otd, otd->fd); | ||||||
|  | error_udev: | ||||||
|  | 	otd_udev_finish(otd); | ||||||
|  | error_session: | ||||||
|  | 	otd_close_session(otd); | ||||||
|  | error: | ||||||
|  | 	free(otd); | ||||||
|  | 	return NULL; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void otd_finish(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	if (!otd) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	for (size_t i = 0; i < otd->display_len; ++i) { | ||||||
|  | 		destroy_display_renderer(otd, &otd->displays[i]); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	destroy_renderer(otd); | ||||||
|  | 	otd_close_session(otd); | ||||||
|  | 	otd_udev_finish(otd); | ||||||
|  | 
 | ||||||
|  | 	close(otd->fd); | ||||||
|  | 	free(otd->events); | ||||||
|  | 	free(otd->displays); | ||||||
|  | 	free(otd); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | @ -0,0 +1,44 @@ | ||||||
|  | #ifndef LIBOTD_H | ||||||
|  | #define LIBOTD_H | ||||||
|  | 
 | ||||||
|  | #include <stdbool.h> | ||||||
|  | #include <stddef.h> | ||||||
|  | #include <EGL/egl.h> | ||||||
|  | #include <gbm.h> | ||||||
|  | #include <libudev.h> | ||||||
|  | 
 | ||||||
|  | #include "session.h" | ||||||
|  | 
 | ||||||
|  | struct otd { | ||||||
|  | 	int fd; | ||||||
|  | 	bool paused; | ||||||
|  | 
 | ||||||
|  | 	// Priority Queue (Max-heap)
 | ||||||
|  | 	size_t event_cap; | ||||||
|  | 	size_t event_len; | ||||||
|  | 	struct otd_event *events; | ||||||
|  | 
 | ||||||
|  | 	size_t display_len; | ||||||
|  | 	struct otd_display *displays; | ||||||
|  | 
 | ||||||
|  | 	uint32_t taken_crtcs; | ||||||
|  | 
 | ||||||
|  | 	struct gbm_device *gbm; | ||||||
|  | 	struct { | ||||||
|  | 		EGLDisplay disp; | ||||||
|  | 		EGLConfig conf; | ||||||
|  | 		EGLContext context; | ||||||
|  | 	} egl; | ||||||
|  | 
 | ||||||
|  | 	struct otd_session session; | ||||||
|  | 
 | ||||||
|  | 	struct udev *udev; | ||||||
|  | 	struct udev_monitor *mon; | ||||||
|  | 	int udev_fd; | ||||||
|  | 	char *drm_path; | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | struct otd *otd_start(void); | ||||||
|  | void otd_finish(struct otd *otd); | ||||||
|  | 
 | ||||||
|  | #endif | ||||||
|  | @ -0,0 +1,224 @@ | ||||||
|  | #include <stdio.h> | ||||||
|  | #include <stdbool.h> | ||||||
|  | #include <stdlib.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <systemd/sd-bus.h> | ||||||
|  | #include <systemd/sd-login.h> | ||||||
|  | #include <unistd.h> | ||||||
|  | #include <sys/sysmacros.h> | ||||||
|  | #include <sys/stat.h> | ||||||
|  | #include <fcntl.h> | ||||||
|  | 
 | ||||||
|  | #include "session.h" | ||||||
|  | #include "otd.h" | ||||||
|  | 
 | ||||||
|  | int take_device(struct otd *restrict otd, | ||||||
|  | 		       const char *restrict path, | ||||||
|  | 		       bool *restrict paused_out) | ||||||
|  | { | ||||||
|  | 	int ret; | ||||||
|  | 	int fd = -1; | ||||||
|  | 	sd_bus_message *msg = NULL; | ||||||
|  | 	sd_bus_error error = SD_BUS_ERROR_NULL; | ||||||
|  | 	struct otd_session *s = &otd->session; | ||||||
|  | 
 | ||||||
|  | 	struct stat st; | ||||||
|  | 	if (stat(path, &st) < 0) { | ||||||
|  | 		fprintf(stderr, "Failed to stat '%s'\n", path); | ||||||
|  | 		return -1; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	ret = sd_bus_call_method(s->bus, | ||||||
|  | 				 "org.freedesktop.login1", | ||||||
|  | 				 s->path, | ||||||
|  | 				 "org.freedesktop.login1.Session", | ||||||
|  | 				 "TakeDevice", | ||||||
|  | 				 &error, &msg, | ||||||
|  | 				 "uu", major(st.st_rdev), minor(st.st_rdev)); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		fprintf(stderr, "%s\n", error.message); | ||||||
|  | 		goto error; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	int paused = 0; | ||||||
|  | 	ret = sd_bus_message_read(msg, "hb", &fd, &paused); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		fprintf(stderr, "%s\n", strerror(-ret)); | ||||||
|  | 		goto error; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// The original fd seem to be closed when the message is freed
 | ||||||
|  | 	// so we just clone it.
 | ||||||
|  | 	fd = fcntl(fd, F_DUPFD_CLOEXEC, 0); | ||||||
|  | 
 | ||||||
|  | 	if (paused_out) | ||||||
|  | 		*paused_out = paused; | ||||||
|  | 
 | ||||||
|  | error: | ||||||
|  | 	sd_bus_error_free(&error); | ||||||
|  | 	sd_bus_message_unref(msg); | ||||||
|  | 	return fd; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void release_device(struct otd *otd, int fd) | ||||||
|  | { | ||||||
|  | 	int ret; | ||||||
|  | 	sd_bus_message *msg = NULL; | ||||||
|  | 	sd_bus_error error = SD_BUS_ERROR_NULL; | ||||||
|  | 	struct otd_session *s = &otd->session; | ||||||
|  | 
 | ||||||
|  | 	struct stat st; | ||||||
|  | 	if (fstat(fd, &st) < 0) { | ||||||
|  | 		fprintf(stderr, "Could not stat fd %d\n", fd); | ||||||
|  | 		return; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	ret = sd_bus_call_method(s->bus, | ||||||
|  | 				 "org.freedesktop.login1", | ||||||
|  | 				 s->path, | ||||||
|  | 				 "org.freedesktop.login1.Session", | ||||||
|  | 				 "ReleaseDevice", | ||||||
|  | 				 &error, &msg, | ||||||
|  | 				 "uu", major(st.st_rdev), minor(st.st_rdev)); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		/* Log something */; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	sd_bus_error_free(&error); | ||||||
|  | 	sd_bus_message_unref(msg); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static bool session_activate(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	int ret; | ||||||
|  | 	sd_bus_message *msg = NULL; | ||||||
|  | 	sd_bus_error error = SD_BUS_ERROR_NULL; | ||||||
|  | 	struct otd_session *s = &otd->session; | ||||||
|  | 
 | ||||||
|  | 	ret = sd_bus_call_method(s->bus, | ||||||
|  | 				 "org.freedesktop.login1", | ||||||
|  | 				 s->path, | ||||||
|  | 				 "org.freedesktop.login1.Session", | ||||||
|  | 				 "Activate", | ||||||
|  | 				 &error, &msg, | ||||||
|  | 				 ""); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		fprintf(stderr, "%s\n", error.message); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	sd_bus_error_free(&error); | ||||||
|  | 	sd_bus_message_unref(msg); | ||||||
|  | 	return ret >= 0; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static bool take_control(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	int ret; | ||||||
|  | 	sd_bus_message *msg = NULL; | ||||||
|  | 	sd_bus_error error = SD_BUS_ERROR_NULL; | ||||||
|  | 	struct otd_session *s = &otd->session; | ||||||
|  | 
 | ||||||
|  | 	ret = sd_bus_call_method(s->bus, | ||||||
|  | 				 "org.freedesktop.login1", | ||||||
|  | 				 s->path, | ||||||
|  | 				 "org.freedesktop.login1.Session", | ||||||
|  | 				 "TakeControl", | ||||||
|  | 				 &error, &msg, | ||||||
|  | 				 "b", false); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		/* Log something */; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	sd_bus_error_free(&error); | ||||||
|  | 	sd_bus_message_unref(msg); | ||||||
|  | 	return ret >= 0; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static void release_control(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	int ret; | ||||||
|  | 	sd_bus_message *msg = NULL; | ||||||
|  | 	sd_bus_error error = SD_BUS_ERROR_NULL; | ||||||
|  | 	struct otd_session *s = &otd->session; | ||||||
|  | 
 | ||||||
|  | 	ret = sd_bus_call_method(s->bus, | ||||||
|  | 				 "org.freedesktop.login1", | ||||||
|  | 				 s->path, | ||||||
|  | 				 "org.freedesktop.login1.Session", | ||||||
|  | 				 "ReleaseControl", | ||||||
|  | 				 &error, &msg, | ||||||
|  | 				 ""); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		/* Log something */; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	sd_bus_error_free(&error); | ||||||
|  | 	sd_bus_message_unref(msg); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void otd_close_session(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	release_device(otd, otd->fd); | ||||||
|  | 	release_control(otd); | ||||||
|  | 
 | ||||||
|  | 	sd_bus_unref(otd->session.bus); | ||||||
|  | 	free(otd->session.id); | ||||||
|  | 	free(otd->session.path); | ||||||
|  | 	free(otd->session.seat); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | bool otd_new_session(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	int ret; | ||||||
|  | 	struct otd_session *s = &otd->session; | ||||||
|  | 
 | ||||||
|  | 	ret = sd_pid_get_session(getpid(), &s->id); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		fprintf(stderr, "Could not get session\n"); | ||||||
|  | 		goto error; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	ret = sd_session_get_seat(s->id, &s->seat); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		fprintf(stderr, "Could not get seat\n"); | ||||||
|  | 		goto error; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// This could be done using asprintf, but I don't want to define _GNU_SOURCE
 | ||||||
|  | 
 | ||||||
|  | 	const char *fmt = "/org/freedesktop/login1/session/%s"; | ||||||
|  | 	int len = snprintf(NULL, 0, fmt, s->id); | ||||||
|  | 
 | ||||||
|  | 	s->path = malloc(len + 1); | ||||||
|  | 	if (!s->path) | ||||||
|  | 		goto error; | ||||||
|  | 
 | ||||||
|  | 	sprintf(s->path, fmt, s->id); | ||||||
|  | 
 | ||||||
|  | 	ret = sd_bus_open_system(&s->bus); | ||||||
|  | 	if (ret < 0) { | ||||||
|  | 		fprintf(stderr, "Could not open bus\n"); | ||||||
|  | 		goto error; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (!session_activate(otd)) { | ||||||
|  | 		fprintf(stderr, "Could not activate session\n"); | ||||||
|  | 		goto error_bus; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (!take_control(otd)) { | ||||||
|  | 		fprintf(stderr, "Could not take control of session\n"); | ||||||
|  | 		goto error_bus; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return true; | ||||||
|  | 
 | ||||||
|  | error_bus: | ||||||
|  | 	sd_bus_unref(s->bus); | ||||||
|  | 
 | ||||||
|  | error: | ||||||
|  | 	free(s->path); | ||||||
|  | 	free(s->id); | ||||||
|  | 	free(s->seat); | ||||||
|  | 	return false; | ||||||
|  | } | ||||||
|  | @ -0,0 +1,25 @@ | ||||||
|  | #ifndef SESSION_H | ||||||
|  | #define SESSION_H | ||||||
|  | 
 | ||||||
|  | #include <systemd/sd-bus.h> | ||||||
|  | #include <stdbool.h> | ||||||
|  | 
 | ||||||
|  | struct otd_session { | ||||||
|  | 	char *id; | ||||||
|  | 	char *path; | ||||||
|  | 	char *seat; | ||||||
|  | 
 | ||||||
|  | 	sd_bus *bus; | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | struct otd; | ||||||
|  | bool otd_new_session(struct otd *otd); | ||||||
|  | void otd_close_session(struct otd *otd); | ||||||
|  | 
 | ||||||
|  | int take_device(struct otd *restrict otd, | ||||||
|  | 		       const char *restrict path, | ||||||
|  | 		       bool *restrict paused_out); | ||||||
|  | 
 | ||||||
|  | void release_device(struct otd *otd, int fd); | ||||||
|  | 
 | ||||||
|  | #endif | ||||||
|  | @ -0,0 +1,158 @@ | ||||||
|  | #include <libudev.h> | ||||||
|  | #include <stdbool.h> | ||||||
|  | #include <stdlib.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <stdint.h> | ||||||
|  | #include <xf86drm.h> | ||||||
|  | #include <xf86drmMode.h> | ||||||
|  | 
 | ||||||
|  | #include "otd.h" | ||||||
|  | #include "udev.h" | ||||||
|  | #include "session.h" | ||||||
|  | #include "drm.h" | ||||||
|  | 
 | ||||||
|  | static bool device_is_kms(struct otd *otd, struct udev_device *dev) | ||||||
|  | { | ||||||
|  | 	const char *path = udev_device_get_devnode(dev); | ||||||
|  | 	int fd; | ||||||
|  | 
 | ||||||
|  | 	if (!path) | ||||||
|  | 		return false; | ||||||
|  | 
 | ||||||
|  | 	fd = take_device(otd, path, &otd->paused); | ||||||
|  | 	if (fd < 0) | ||||||
|  | 		return false; | ||||||
|  | 
 | ||||||
|  | 	drmModeRes *res = drmModeGetResources(fd); | ||||||
|  | 	if (!res) | ||||||
|  | 		goto out_fd; | ||||||
|  | 
 | ||||||
|  | 	if (res->count_crtcs <= 0 || res->count_connectors <= 0 || | ||||||
|  | 	    res->count_encoders <= 0) | ||||||
|  | 		goto out_res; | ||||||
|  | 
 | ||||||
|  | 	if (otd->fd >= 0) { | ||||||
|  | 		release_device(otd, otd->fd); | ||||||
|  | 		free(otd->drm_path); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	otd->fd = fd; | ||||||
|  | 	otd->drm_path = strdup(path); | ||||||
|  | 
 | ||||||
|  | 	drmModeFreeResources(res); | ||||||
|  | 	return true; | ||||||
|  | 
 | ||||||
|  | out_res: | ||||||
|  | 	drmModeFreeResources(res); | ||||||
|  | out_fd: | ||||||
|  | 	release_device(otd, fd); | ||||||
|  | 	return false; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void otd_udev_find_gpu(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	struct udev *udev = otd->udev; | ||||||
|  | 	otd->fd = -1; | ||||||
|  | 
 | ||||||
|  | 	struct udev_enumerate *en = udev_enumerate_new(udev); | ||||||
|  | 	if (!en) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	udev_enumerate_add_match_subsystem(en, "drm"); | ||||||
|  | 	udev_enumerate_add_match_sysname(en, "card[0-9]*"); | ||||||
|  | 
 | ||||||
|  | 	udev_enumerate_scan_devices(en); | ||||||
|  | 	struct udev_list_entry *entry; | ||||||
|  | 	udev_list_entry_foreach(entry, udev_enumerate_get_list_entry(en)) { | ||||||
|  | 		bool is_boot_vga = false; | ||||||
|  | 
 | ||||||
|  | 		const char *path = udev_list_entry_get_name(entry); | ||||||
|  | 		struct udev_device *dev = udev_device_new_from_syspath(udev, path); | ||||||
|  | 		if (!dev) | ||||||
|  | 			continue; | ||||||
|  | 
 | ||||||
|  | 		const char *seat = udev_device_get_property_value(dev, "ID_SEAT"); | ||||||
|  | 		if (!seat) | ||||||
|  | 			seat = "seat0"; | ||||||
|  | 		if (strcmp(otd->session.seat, seat) != 0) { | ||||||
|  | 			udev_device_unref(dev); | ||||||
|  | 			continue; | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		struct udev_device *pci = | ||||||
|  | 			udev_device_get_parent_with_subsystem_devtype(dev, | ||||||
|  | 								      "pci", NULL); | ||||||
|  | 
 | ||||||
|  | 		if (pci) { | ||||||
|  | 			const char *id = udev_device_get_sysattr_value(pci, "boot_vga"); | ||||||
|  | 			if (id && strcmp(id, "1") == 0) | ||||||
|  | 				is_boot_vga = true; | ||||||
|  | 			udev_device_unref(pci); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		if (!is_boot_vga && otd->fd >= 0) { | ||||||
|  | 			udev_device_unref(dev); | ||||||
|  | 			continue; | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		if (!device_is_kms(otd, dev)) { | ||||||
|  | 			udev_device_unref(dev); | ||||||
|  | 			continue; | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		if (is_boot_vga) { | ||||||
|  | 			break; | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	udev_enumerate_unref(en); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | bool otd_udev_start(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	otd->udev = udev_new(); | ||||||
|  | 	if (!otd->udev) | ||||||
|  | 		return false; | ||||||
|  | 
 | ||||||
|  | 	otd->mon = udev_monitor_new_from_netlink(otd->udev, "udev"); | ||||||
|  | 	if (!otd->mon) { | ||||||
|  | 		udev_unref(otd->udev); | ||||||
|  | 		return false; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	udev_monitor_filter_add_match_subsystem_devtype(otd->mon, "drm", NULL); | ||||||
|  | 	udev_monitor_enable_receiving(otd->mon); | ||||||
|  | 
 | ||||||
|  | 	otd->udev_fd = udev_monitor_get_fd(otd->mon); | ||||||
|  | 
 | ||||||
|  | 	return true; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void otd_udev_finish(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	if (!otd) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	udev_monitor_unref(otd->mon); | ||||||
|  | 	udev_unref(otd->udev); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void otd_udev_event(struct otd *otd) | ||||||
|  | { | ||||||
|  | 	struct udev_device *dev = udev_monitor_receive_device(otd->mon); | ||||||
|  | 	if (!dev) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	const char *path = udev_device_get_devnode(dev); | ||||||
|  | 	if (!path || strcmp(path, otd->drm_path) != 0) | ||||||
|  | 		goto out; | ||||||
|  | 
 | ||||||
|  | 	const char *action = udev_device_get_action(dev); | ||||||
|  | 	if (!action || strcmp(action, "change") != 0) | ||||||
|  | 		goto out; | ||||||
|  | 
 | ||||||
|  | 	scan_connectors(otd); | ||||||
|  | 
 | ||||||
|  | out: | ||||||
|  | 	udev_device_unref(dev); | ||||||
|  | } | ||||||
|  | @ -0,0 +1,9 @@ | ||||||
|  | #ifndef UDEV_H | ||||||
|  | #define UDEV_H | ||||||
|  | 
 | ||||||
|  | bool otd_udev_start(struct otd *otd); | ||||||
|  | void otd_udev_finish(struct otd *otd); | ||||||
|  | void otd_udev_find_gpu(struct otd *otd); | ||||||
|  | void otd_udev_event(struct otd *otd); | ||||||
|  | 
 | ||||||
|  | #endif | ||||||
		Loading…
	
		Reference in New Issue