From 5a54774474f43ae29716751d6415563a59b92c7d Mon Sep 17 00:00:00 2001 From: pk33 Date: Mon, 11 Nov 2024 18:06:18 +0100 Subject: tidying up and basic input --- src/display33.c | 575 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 575 insertions(+) create mode 100644 src/display33.c (limited to 'src/display33.c') diff --git a/src/display33.c b/src/display33.c new file mode 100644 index 0000000..928d4e0 --- /dev/null +++ b/src/display33.c @@ -0,0 +1,575 @@ +#include "string.h" +#include "stdlib.h" +#include "signal.h" +#include "fcntl.h" +#include "errno.h" +#include "termios.h" +#include "unistd.h" +#include "poll.h" +#include "dirent.h" +#include "termio.h" +#include "stdio.h" +#include "libdrm/drm_mode.h" +#include "sys/vt.h" +#include "sys/kd.h" +#include "sys/ioctl.h" +#include "sys/mman.h" + +#include "display33.h" +#include "system33.h" +#include "logger33.h" +#include "options33.h" + + +#define REL_SIGNAL SIGUSR1 +#define ACQ_SIGNAL SIGUSR2 +#define REL_EVENT 1 +#define ACQ_EVENT 2 + + +Display33 __display = {0,}; + + +typedef struct { + struct drm_mode_card_res drmRes; + __u64 *drmRes_fbs; + __u32 *drmRes_conns; + __u64 *drmRes_encs; + __u64 *drmRes_crtcs; + + __u64 connProps[ 16 ]; + __u64 connPropVals[ 16 ]; + __u64 connEncs[ 16 ]; + struct drm_mode_modeinfo connModes[ 32 ]; + struct drm_mode_get_encoder *encoder; + + struct drm_mode_create_dumb drmCreateDumb[2]; + struct drm_mode_map_dumb drmMapDumb[2]; + struct drm_mode_fb_cmd drmFBCmd[2]; +} DRMData_t; + +static DRMData_t drmData = {0,}; + +static int vtPipe[2] = { -1, -1 }; + +static struct termios termOldConfig; +static struct vt_mode vtModeOld = {0,}; + +static Error _open_graphics_device( void ); +static Error _get_drm_resources( void ); +static Error _get_drm_connector( void ); +static Error _get_drm_crtc( void ); +static Error _get_drm_framebuffer(void ); + +static Size _init_vt_switch( void ); +static void _vt_release( void ); +static void _vt_acquire( void ); +static void _vt_switch_sighandler(int sig); + + +Error display33_init( void ) +{ + struct termios termConfig; + + + if(!isatty(STDIN_FILENO)) { + LOGF( "stdin is not a terminal." ); + return E33_EXIT_FAILURE; + } + + if( tcgetattr(STDIN_FILENO, &termOldConfig) == -1 ) { + LOGF( "Could not get terminal attributes." ); + return E33_EXIT_FAILURE; + } + + termConfig = termOldConfig; + termConfig.c_iflag &= ~(IGNBRK | BRKINT | PARMRK); + termConfig.c_lflag &= ~(ICANON | ECHO | IEXTEN | TOSTOP); + termConfig.c_lflag |= ISIG; + termConfig.c_cc[VTIME] = 0; + termConfig.c_cc[VMIN] = 0; + termConfig.c_cc[VSTART] = 0; + termConfig.c_cc[VSTOP] = 0; + + if( tcsetattr(STDIN_FILENO, TCSANOW, &termConfig) == -1 ) { + LOGF( "Could not set terminal attributes." ); + return E33_EXIT_FAILURE; + } + + if( _init_vt_switch() ) { + LOGW( "Failed to initialize VT switcher. VT switching will not be available." ); + } + + + if( _open_graphics_device() ) { + LOGF( "Failed to find a suitable graphics device." ); + return E33_EXIT_FAILURE; + } + + if( e33_ioctl( __display.devFd, DRM_IOCTL_SET_MASTER, 0 ) == -1 ) + { + LOGF( "Failed to set DRM master." ); + return E33_EXIT_FAILURE; + } + + if( _get_drm_resources() ) { + LOGF( "Failed to get resources." ); + return E33_EXIT_FAILURE; + } + if( _get_drm_connector() ) { + LOGF( "Failed to set connector." ); + return E33_EXIT_FAILURE; + } + if( _get_drm_framebuffer() ) { + LOGF( "Failed to set framebuffer." ); + return E33_EXIT_FAILURE; + } + if( _get_drm_crtc() ) { + LOGF( "Failed to set crtc." ); + return E33_EXIT_FAILURE; + } + + + __display.surface.w = __display.mode.hdisplay; + __display.surface.h = __display.mode.vdisplay; + __display.fb.size = (Size)drmData.drmCreateDumb[0].size; + __display.surface.data = __display.fb.map[0]; + __display.active = E33_TRUE; + + free( drmData.drmRes_fbs ); + free( drmData.drmRes_encs ); + free( drmData.drmRes_conns ); + + return E33_EXIT_SUCCESS; +} + +void display33_flip( void ) +{ + static u8 i_s = 1; + + struct drm_mode_crtc_page_flip flip = { 0, }; + + __display.surface.data = __display.fb.map[i_s]; + i_s ^= 1; + __display.crtc.fb_id = __display.fb.id[i_s]; + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_SETCRTC, &__display.crtc ) == -1 ) { + LOGW( "Failed to set CRTC for page flip prep. Skipping." ); + return; + } + + flip.fb_id = __display.fb.id[i_s]; + flip.crtc_id = __display.crtc.crtc_id; + flip.user_data = ((__u64)(&__display.crtc.crtc_id)); + flip.flags = DRM_MODE_PAGE_FLIP_EVENT; + + e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_PAGE_FLIP, &flip ); + + /* TODO remove later */ + memset( __display.surface.data, 0, (uSize)__display.fb.size ); +} + +void display33_term( void ) +{ + if( e33_ioctl( __display.devFd, DRM_IOCTL_DROP_MASTER, 0 ) == -1 ) { + LOGW( "Failed to drop drm master." ); + } + + if( munmap( __display.fb.map[0], (uSize)__display.fb.size ) == -1 ) { + LOGW( "Failed to unmap framebuffer[0]." ); + } + if( munmap( __display.fb.map[1], (uSize)__display.fb.size ) == -1 ) { + LOGW( "Failed to unmap framebuffer[1]." ); + } + + if( close( __display.devFd ) ) { + LOGW( "Failed to close graphics device." ); + } + + free( drmData.drmRes_crtcs ); + + if( ioctl(__display.ttyFd, VT_SETMODE, &vtModeOld) < 0 ) { + LOGW( "Could not reset VT mode." ); + } + + if( tcsetattr(STDIN_FILENO, TCSAFLUSH, &termOldConfig) == -1 ) { + LOGW( "Failed to set stdin attributes." ); + } + + if( close( __display.ttyFd ) == -1 ) { + LOGW( "Failed to close TTY." ); + } +} + + +void display33_vtswitcher_poll( int timeout ) +{ + struct pollfd fds[1]; + unsigned char event; + + fds[0].fd = vtPipe[0]; + fds[0].events = POLLIN; + fds[0].revents = 0; + + poll( fds, 1, timeout ); + if( !fds[0].revents ) { + return; + } + + if( read(fds[0].fd, &event, sizeof(event)) != sizeof(event) ) { + LOGW( "Invalid VT switch events read."); + return; + } + + switch(event) { + case REL_EVENT: _vt_release(); + break; + case ACQ_EVENT: _vt_acquire(); + break; + } +} + + + + + +static Error _open_graphics_device( void ) +{ + struct drm_get_cap cap = {0,}; + + + if( (__display.devFd = open( E33_DRM_DEVICE, O_RDWR )) == -1 ) { + LOGE( "Failed to open DRM device '%s'.", E33_DRM_DEVICE ); + return E33_EXIT_FAILURE; + } + + cap.capability = DRM_CAP_DUMB_BUFFER; + + if( e33_ioctl( __display.devFd, DRM_IOCTL_GET_CAP, &cap ) == -1 ) { + LOGW( "Failed to get DRM device capabilities." ); + return E33_EXIT_FAILURE; + } + + if( cap.value ) { + return E33_EXIT_SUCCESS; + } + + return E33_EXIT_FAILURE; +} + +static Size _get_drm_resources( void ) +{ + if( e33_ioctl(__display.devFd, (int)DRM_IOCTL_MODE_GETRESOURCES, &drmData.drmRes) == -1 ) { + LOGE( "Failed to init drm resources." ); + return E33_EXIT_FAILURE; + } + + drmData.drmRes_fbs = malloc( sizeof(__u64) * drmData.drmRes.count_fbs ); + drmData.drmRes_crtcs = malloc( sizeof(__u64) * drmData.drmRes.count_crtcs ); + drmData.drmRes_encs = malloc( sizeof(__u64) * drmData.drmRes.count_encoders ); + drmData.drmRes_conns = malloc( sizeof(__u32) * drmData.drmRes.count_connectors ); + + /* DRM resource objects must also be zero'd before second call */ + memset( drmData.drmRes_fbs, 0, sizeof(__u64) * drmData.drmRes.count_fbs ); + memset( drmData.drmRes_crtcs, 0, sizeof(__u64) * drmData.drmRes.count_crtcs ); + memset( drmData.drmRes_encs, 0, sizeof(__u64) * drmData.drmRes.count_encoders ); + memset( drmData.drmRes_conns, 0, sizeof(__u32) * drmData.drmRes.count_connectors ); + + drmData.drmRes.fb_id_ptr = (__u64)drmData.drmRes_fbs; + drmData.drmRes.crtc_id_ptr = (__u64)drmData.drmRes_crtcs; + drmData.drmRes.connector_id_ptr = (__u64)drmData.drmRes_conns; + drmData.drmRes.encoder_id_ptr = (__u64)drmData.drmRes_encs; + + /* Second call writes drm resource data using the given pointers */ + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_GETRESOURCES, &drmData.drmRes ) == -1 ) { + LOGE( "Failed to get drm resources." ); + return E33_EXIT_FAILURE; + } + + return E33_EXIT_SUCCESS; +} + + +static Size _get_drm_connector( void ) +{ + u16 i, j; + + for( i = 0; i < drmData.drmRes.count_connectors; ++i ) + { + struct drm_mode_get_connector *connector = &__display.connector; + + + connector->connector_id = drmData.drmRes_conns[i]; + + if( connector->connector_id <= 0 ) { + continue; + } + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_GETCONNECTOR, connector ) == -1 ) { + continue; + } + + if( connector->connection == 1 && + connector->count_modes > 0 && + connector->count_encoders > 0 + ) { + connector->modes_ptr = (__u64)drmData.connModes; + connector->props_ptr = (__u64)drmData.connProps; + connector->prop_values_ptr = (__u64)drmData.connPropVals; + connector->encoders_ptr = (__u64)drmData.connEncs; + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_GETCONNECTOR, connector) == -1 ) { + continue; + } + + for( j = 0; j < connector->count_modes; ++j ) + { + struct drm_mode_modeinfo *mode = &drmData.connModes[j]; + + if( e33_strcmp( mode->name, E33_DRM_MODE ) && (mode->vrefresh == E33_DRM_RATE) ) + { + /* TODO there has to be a better way */ + if( mode->hdisplay == 1366 ) { + mode->hdisplay += 10; + } + + __display.mode = *mode; + return E33_EXIT_SUCCESS; + } + } + } + } + + LOGE( "Failed to find a valid connector." ); + return E33_EXIT_FAILURE; +} + +static Size _get_drm_crtc( void ) +{ + struct drm_mode_get_encoder encoder = {0,}; + struct drm_mode_crtc *crtc = &__display.crtc; + + encoder.encoder_id = __display.connector.encoder_id; + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_GETENCODER, &encoder ) == -1 ) { + LOGE( "Failed to get encoder." ); + return E33_EXIT_FAILURE; + } + + crtc->crtc_id = encoder.crtc_id; + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_GETCRTC, crtc ) == -1 ) { + LOGE( "Failed to get CRTC." ); + return E33_EXIT_FAILURE; + } + + crtc->fb_id = drmData.drmFBCmd[0].fb_id; + crtc->set_connectors_ptr = (__u64)drmData.drmRes_conns; + crtc->count_connectors = 1; + crtc->mode = __display.mode; + crtc->mode_valid = 1; + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_SETCRTC, crtc ) == -1 ) { + LOGE( "Could not set CRTC." ); + return E33_EXIT_FAILURE; + } + + return E33_EXIT_SUCCESS; +} + +static Size _get_drm_framebuffer( void ) +{ + struct drm_mode_create_dumb *drmCreateDumb = drmData.drmCreateDumb; + struct drm_mode_map_dumb *drmMapDumb = drmData.drmMapDumb; + struct drm_mode_fb_cmd *drmFBCmd = drmData.drmFBCmd; + struct drm_mode_modeinfo *mode = &__display.mode; + u8 i; + + for( i = 0; i < 2; ++i ) + { + drmCreateDumb[i].width = mode->hdisplay; + drmCreateDumb[i].height = mode->vdisplay; + drmCreateDumb[i].bpp = 32; + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_CREATE_DUMB, + drmCreateDumb + i ) == -1 ) + { + LOGE( "Failed to create dumb buffer." ); + return E33_EXIT_FAILURE; + } + + drmFBCmd[i].width = drmCreateDumb[i].width; + drmFBCmd[i].height = drmCreateDumb[i].height; + drmFBCmd[i].bpp = drmCreateDumb[i].bpp; + drmFBCmd[i].pitch = drmCreateDumb[i].pitch; + drmFBCmd[i].depth = 24; + drmFBCmd[i].handle = drmCreateDumb[i].handle; + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_ADDFB, drmFBCmd + i ) == -1 ) { + LOGE( "Failed to add framebuffer." ); + return E33_EXIT_FAILURE; + } + + __display.fb.id[i] = drmFBCmd[i].fb_id; + + drmMapDumb[i].handle = drmCreateDumb[i].handle; + + if( e33_ioctl( __display.devFd, (int)DRM_IOCTL_MODE_MAP_DUMB, drmMapDumb + i ) == -1 ) + { + LOGE( "Failed to map dumb buffer." ); + return E33_EXIT_FAILURE; + } + + __display.fb.map[i] = + mmap( 0, drmCreateDumb[i].size, PROT_READ | PROT_WRITE, + MAP_SHARED, __display.devFd, drmMapDumb[i].offset ); + + if( __display.fb.map[i] == MAP_FAILED ) { + LOGE( "Failed to map framebuffer." ); + return E33_EXIT_FAILURE; + } + } + + return E33_EXIT_SUCCESS; +} + +static void _vt_release( void ) +{ + LOGI( "VT release signal."); + + if( ioctl( __display.devFd, DRM_IOCTL_DROP_MASTER, 0 ) ) { + LOGW( "Could not drop DRM master." ); + } + + if( ioctl(__display.ttyFd, VT_RELDISP, 1) < 0 ) { + LOGW( "Failed to release VT."); + } + + __display.active = 0; +} + +static void _vt_acquire( void ) +{ + LOGI( "VT acquire signal"); + + if( ioctl( __display.devFd, DRM_IOCTL_SET_MASTER, 0 ) == -1 ) { + LOGW( "Could not become DRM master." ); + } + + if( ioctl(__display.ttyFd, VT_RELDISP, VT_ACKACQ) < 0 ) { + LOGW( "(vt_acquire) Failed to acquire VT."); + } + + if( ioctl( __display.devFd, (int)DRM_IOCTL_MODE_SETCRTC, &__display.crtc ) == -1 ) { + LOGW( "Failed to set crtc on VT switch." ); + } + + __display.active = 1; +} + + +static int _set_pipe(int fd) +{ + if (fd >= 0) + { + int flags = fcntl(fd, F_GETFD); + + if (flags == -1) { + LOGW( "Could not get flags for VT switcher pipe."); + return E33_EXIT_FAILURE; + } + + if (fcntl(fd, F_SETFD, flags | FD_CLOEXEC) == -1) { + LOGW( "Could not set flags for VT switcher pipe."); + return E33_EXIT_FAILURE; + } + } + + return E33_EXIT_SUCCESS; +} + + +static void _vt_switch_sighandler(int sig) +{ + unsigned char event = REL_EVENT; + + if( sig == ACQ_SIGNAL ) { + event = ACQ_EVENT; + } + + (void)write( vtPipe[1], &event, sizeof(event) ); +} + +static Size _init_vt_switch( void ) +{ + struct vt_mode vtMode = {0,}; + sigset_t set; + + + __display.ttyFd = -1; + + if( pipe2(vtPipe, 0) == -1 ) + { + vtPipe[0] = vtPipe[1] = -1; + LOGW( "Could not set VT pipes."); + return E33_EXIT_FAILURE; + } + + if( _set_pipe(vtPipe[0]) ) { + LOGW( "Could not set VT read pipe." ); + return E33_EXIT_FAILURE; + } + if( _set_pipe(vtPipe[1]) ) { + LOGW( "Could not set VT write pipe." ); + return E33_EXIT_FAILURE; + } + + + if( __display.ttyFd = open("/dev/tty", O_RDWR ) < 0 ) { + LOGW( "Could not open TTY for VT control." ); + return E33_EXIT_FAILURE; + } + + + if( e33_has_signal(REL_SIGNAL) ) { + LOGW( "VT release signal is already in use."); + return E33_EXIT_FAILURE; + } + if( e33_has_signal(ACQ_SIGNAL) ) { + LOGW( "VT acquire signal is already in use."); + return E33_EXIT_FAILURE; + } + + if( e33_set_signal( REL_SIGNAL, _vt_switch_sighandler ) ) { + LOGW( "Could not set relese signal handler."); + return E33_EXIT_FAILURE; + } + if( e33_set_signal( ACQ_SIGNAL, _vt_switch_sighandler ) ) { + LOGW( "Could not set acquire signal handler."); + return E33_EXIT_FAILURE; + } + + + if( ioctl(__display.ttyFd, VT_GETMODE, &vtModeOld) < 0 ) { + LOGW( "Could not get VT mode." ); + return E33_EXIT_FAILURE; + } + + vtMode = vtModeOld; + vtMode.mode = VT_PROCESS; + vtMode.relsig = REL_SIGNAL; + vtMode.acqsig = ACQ_SIGNAL; + vtMode.frsig = SIGIO; + + if( ioctl(__display.ttyFd, VT_SETMODE, &vtMode) < 0 ) { + LOGW( "Could not set VT mode." ); + return E33_EXIT_FAILURE; + } + + return E33_EXIT_SUCCESS; +} + + + + + -- cgit v1.2.3