Skip to content

Commit

Permalink
Use shm_open instead of mkstemp
Browse files Browse the repository at this point in the history
shm_open is more reliable because it does not require
a writeable filesystem folder, unlike mkstemp.
  • Loading branch information
mstoeckl authored and emersion committed Nov 24, 2022
1 parent 7623292 commit e2bc886
Showing 1 changed file with 31 additions and 35 deletions.
66 changes: 31 additions & 35 deletions client/pool-buffer.c
Original file line number Diff line number Diff line change
@@ -1,50 +1,43 @@
#define _POSIX_C_SOURCE 200809
#include <assert.h>
#include <cairo.h>
#include <errno.h>
#include <fcntl.h>
#include <pango/pangocairo.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>
#include <time.h>
#include <unistd.h>
#include <wayland-client.h>
#include "config.h"
#include "pool-buffer.h"
#include "util.h"

static int create_pool_file(size_t size, char **name) {
static const char template[] = "sway-client-XXXXXX";
const char *path = getenv("XDG_RUNTIME_DIR");
if (path == NULL) {
fprintf(stderr, "XDG_RUNTIME_DIR is not set\n");
return -1;
}

size_t name_size = strlen(template) + 1 + strlen(path) + 1;
*name = malloc(name_size);
if (*name == NULL) {
fprintf(stderr, "allocation failed\n");
return -1;
}
snprintf(*name, name_size, "%s/%s", path, template);

int fd = mkstemp(*name);
if (fd < 0) {
return -1;
}

if (!sway_set_cloexec(fd, true)) {
close(fd);
return -1;
}
static int anonymous_shm_open(void) {
int retries = 100;

do {
// try a probably-unique name
struct timespec ts;
clock_gettime(CLOCK_MONOTONIC, &ts);
pid_t pid = getpid();
char name[50];
snprintf(name, sizeof(name), "/sway-%x-%x",
(unsigned int)pid, (unsigned int)ts.tv_nsec);

// shm_open guarantees that O_CLOEXEC is set
int fd = shm_open(name, O_RDWR | O_CREAT | O_EXCL, 0600);
if (fd >= 0) {
shm_unlink(name);
return fd;
}

if (ftruncate(fd, size) < 0) {
close(fd);
return -1;
}
--retries;
} while (retries > 0 && errno == EEXIST);

return fd;
return -1;
}

static void buffer_release(void *data, struct wl_buffer *wl_buffer) {
Expand All @@ -62,17 +55,20 @@ static struct pool_buffer *create_buffer(struct wl_shm *shm,
uint32_t stride = width * 4;
size_t size = stride * height;

char *name;
int fd = create_pool_file(size, &name);
assert(fd != -1);
int fd = anonymous_shm_open();
if (fd == -1) {
return NULL;
}
if (ftruncate(fd, size) < 0) {
close(fd);
return NULL;
}
void *data = mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
struct wl_shm_pool *pool = wl_shm_create_pool(shm, fd, size);
buf->buffer = wl_shm_pool_create_buffer(pool, 0,
width, height, stride, format);
wl_shm_pool_destroy(pool);
close(fd);
unlink(name);
free(name);

buf->size = size;
buf->width = width;
Expand Down

0 comments on commit e2bc886

Please sign in to comment.