sway/swaylock/main.c

306 lines
9.1 KiB
C
Raw Normal View History

2018-04-03 02:48:13 +00:00
#define _XOPEN_SOURCE 700
#define _POSIX_C_SOURCE 200112L
#include <assert.h>
#include <ctype.h>
#include <fcntl.h>
#include <getopt.h>
#include <stdbool.h>
2015-12-10 12:52:24 +00:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>
2018-04-03 02:48:13 +00:00
#include <sys/stat.h>
#include <time.h>
#include <unistd.h>
2018-04-03 02:48:13 +00:00
#include <wayland-client.h>
#include <wlr/util/log.h>
2018-04-03 03:14:37 +00:00
#include "swaylock/seat.h"
#include "swaylock/swaylock.h"
2018-04-03 02:48:13 +00:00
#include "background-image.h"
#include "pool-buffer.h"
#include "cairo.h"
#include "log.h"
#include "util.h"
#include "wlr-input-inhibitor-unstable-v1-client-protocol.h"
2018-04-03 02:48:13 +00:00
#include "wlr-layer-shell-unstable-v1-client-protocol.h"
void sway_terminate(int exit_code) {
exit(exit_code);
}
2018-04-03 02:48:13 +00:00
static void daemonize() {
int fds[2];
if (pipe(fds) != 0) {
wlr_log(L_ERROR, "Failed to pipe");
exit(1);
}
2018-04-03 02:48:13 +00:00
if (fork() == 0) {
close(fds[0]);
2018-04-03 02:48:13 +00:00
int devnull = open("/dev/null", O_RDWR);
dup2(STDOUT_FILENO, devnull);
dup2(STDERR_FILENO, devnull);
uint8_t success = 0;
if (chdir("/") != 0) {
write(fds[1], &success, 1);
exit(1);
}
success = 1;
if (write(fds[1], &success, 1) != 1) {
exit(1);
}
close(fds[1]);
2018-04-03 02:48:13 +00:00
} else {
close(fds[1]);
uint8_t success;
if (read(fds[0], &success, 1) != 1 || !success) {
wlr_log(L_ERROR, "Failed to daemonize");
exit(1);
}
close(fds[0]);
2018-04-03 02:48:13 +00:00
exit(0);
2016-04-02 22:33:15 +00:00
}
}
2018-04-03 02:48:13 +00:00
static void layer_surface_configure(void *data,
struct zwlr_layer_surface_v1 *layer_surface,
2018-04-03 02:48:13 +00:00
uint32_t serial, uint32_t width, uint32_t height) {
struct swaylock_surface *surface = data;
surface->width = width;
surface->height = height;
zwlr_layer_surface_v1_ack_configure(layer_surface, serial);
render_frame(surface);
}
2018-04-03 02:48:13 +00:00
static void layer_surface_closed(void *data,
struct zwlr_layer_surface_v1 *layer_surface) {
struct swaylock_surface *surface = data;
zwlr_layer_surface_v1_destroy(surface->layer_surface);
wl_surface_destroy(surface->surface);
surface->state->run_display = false;
}
2018-04-03 03:14:37 +00:00
static const struct zwlr_layer_surface_v1_listener layer_surface_listener = {
2018-04-03 02:48:13 +00:00
.configure = layer_surface_configure,
.closed = layer_surface_closed,
};
2018-04-03 23:15:14 +00:00
static void output_geometry(void *data, struct wl_output *output, int32_t x,
int32_t y, int32_t width_mm, int32_t height_mm, int32_t subpixel,
const char *make, const char *model, int32_t transform) {
// Who cares
}
static void output_mode(void *data, struct wl_output *output, uint32_t flags,
int32_t width, int32_t height, int32_t refresh) {
// Who cares
}
static void output_done(void *data, struct wl_output *output) {
// Who cares
}
static void output_scale(void *data, struct wl_output *output, int32_t factor) {
struct swaylock_surface *surface = data;
surface->scale = factor;
if (surface->state->run_display) {
render_frames(surface->state);
}
}
struct wl_output_listener output_listener = {
.geometry = output_geometry,
.mode = output_mode,
.done = output_done,
.scale = output_scale,
};
2018-04-03 02:48:13 +00:00
static void handle_global(void *data, struct wl_registry *registry,
uint32_t name, const char *interface, uint32_t version) {
struct swaylock_state *state = data;
if (strcmp(interface, wl_compositor_interface.name) == 0) {
state->compositor = wl_registry_bind(registry, name,
2018-04-03 23:15:14 +00:00
&wl_compositor_interface, 3);
2018-04-03 02:48:13 +00:00
} else if (strcmp(interface, wl_shm_interface.name) == 0) {
state->shm = wl_registry_bind(registry, name,
&wl_shm_interface, 1);
} else if (strcmp(interface, wl_seat_interface.name) == 0) {
struct wl_seat *seat = wl_registry_bind(
registry, name, &wl_seat_interface, 1);
2018-04-03 02:55:30 +00:00
wl_seat_add_listener(seat, &seat_listener, state);
2018-04-03 02:48:13 +00:00
} else if (strcmp(interface, zwlr_layer_shell_v1_interface.name) == 0) {
state->layer_shell = wl_registry_bind(
registry, name, &zwlr_layer_shell_v1_interface, 1);
} else if (strcmp(interface, zwlr_input_inhibit_manager_v1_interface.name) == 0) {
state->input_inhibit_manager = wl_registry_bind(
registry, name, &zwlr_input_inhibit_manager_v1_interface, 1);
2018-04-03 02:48:13 +00:00
} else if (strcmp(interface, wl_output_interface.name) == 0) {
struct swaylock_surface *surface =
calloc(1, sizeof(struct swaylock_surface));
surface->state = state;
surface->output = wl_registry_bind(registry, name,
2018-04-03 23:15:14 +00:00
&wl_output_interface, 3);
wl_output_add_listener(surface->output, &output_listener, surface);
wl_list_insert(&state->surfaces, &surface->link);
2016-01-25 19:16:00 +00:00
}
2018-04-03 02:48:13 +00:00
}
2016-01-25 19:16:00 +00:00
2018-04-03 02:48:13 +00:00
static void handle_global_remove(void *data, struct wl_registry *registry,
uint32_t name) {
// who cares
}
2018-04-03 02:48:13 +00:00
static const struct wl_registry_listener registry_listener = {
.global = handle_global,
.global_remove = handle_global_remove,
};
2018-04-03 02:48:13 +00:00
static struct swaylock_state state;
2016-01-25 19:16:00 +00:00
2015-12-10 12:52:24 +00:00
int main(int argc, char **argv) {
2016-01-23 12:58:43 +00:00
static struct option long_options[] = {
{"help", no_argument, NULL, 'h'},
2016-01-25 19:17:36 +00:00
{"color", required_argument, NULL, 'c'},
{"image", required_argument, NULL, 'i'},
2018-04-03 02:48:13 +00:00
{"scaling", required_argument, NULL, 's'},
{"tiling", no_argument, NULL, 't'},
2016-01-23 12:58:43 +00:00
{"version", no_argument, NULL, 'v'},
{"socket", required_argument, NULL, 'p'},
2016-05-26 15:16:34 +00:00
{"no-unlock-indicator", no_argument, NULL, 'u'},
{"daemonize", no_argument, NULL, 'f'},
2016-01-23 12:58:43 +00:00
{0, 0, 0, 0}
};
const char *usage =
"Usage: swaylock [options...]\n"
2016-01-23 12:58:43 +00:00
"\n"
" -h, --help Show help message and quit.\n"
" -c, --color <rrggbb[aa]> Turn the screen into the given color instead of white.\n"
2018-04-03 02:48:13 +00:00
" -s, --scaling Scaling mode: stretch, fill, fit, center, tile.\n"
" -t, --tiling Same as --scaling=tile.\n"
" -v, --version Show the version number and quit.\n"
" -i, --image [<output>:]<path> Display the given image.\n"
" -u, --no-unlock-indicator Disable the unlock indicator.\n"
2018-04-03 02:48:13 +00:00
" -f, --daemonize Detach from the controlling terminal.\n"
" --socket <socket> Use the specified socket.\n";
2018-04-03 02:48:13 +00:00
struct swaylock_args args = {
.mode = BACKGROUND_MODE_SOLID_COLOR,
.color = 0xFFFFFFFF,
.show_indicator = true,
};
2018-04-03 19:04:31 +00:00
cairo_surface_t *background_image = NULL;
2018-04-03 02:48:13 +00:00
state.args = args;
wlr_log_init(L_DEBUG, NULL);
2016-01-23 12:58:43 +00:00
int c;
while (1) {
int option_index = 0;
2018-04-03 02:48:13 +00:00
c = getopt_long(argc, argv, "hc:i:s:tvuf", long_options, &option_index);
2016-01-23 12:58:43 +00:00
if (c == -1) {
break;
}
switch (c) {
2018-04-03 02:48:13 +00:00
case 'c': {
state.args.color = parse_color(optarg);
state.args.mode = BACKGROUND_MODE_SOLID_COLOR;
2016-01-25 19:17:36 +00:00
break;
}
case 'i':
2018-04-03 19:04:31 +00:00
// TODO: Multiple background images (bleh)
background_image = load_background_image(optarg);
if (!background_image) {
return 1;
}
state.args.mode = BACKGROUND_MODE_FILL;
break;
2018-04-03 02:48:13 +00:00
case 's':
state.args.mode = parse_background_mode(optarg);
if (state.args.mode == BACKGROUND_MODE_INVALID) {
return 1;
}
break;
case 't':
2018-04-03 19:04:31 +00:00
state.args.mode = BACKGROUND_MODE_TILE;
break;
2016-01-23 12:58:43 +00:00
case 'v':
2018-04-03 02:48:13 +00:00
#if defined SWAY_GIT_VERSION && defined SWAY_GIT_BRANCH && defined SWAY_VERSION_DATE
fprintf(stdout, "swaylock version %s (%s, branch \"%s\")\n",
SWAY_GIT_VERSION, SWAY_VERSION_DATE, SWAY_GIT_BRANCH);
#else
fprintf(stdout, "version unknown\n");
#endif
return 0;
2016-05-26 15:16:34 +00:00
case 'u':
2018-04-03 02:48:13 +00:00
state.args.show_indicator = false;
break;
2018-04-03 02:48:13 +00:00
case 'f':
daemonize();
break;
2016-01-23 12:58:43 +00:00
default:
fprintf(stderr, "%s", usage);
2018-04-03 02:48:13 +00:00
return 1;
2016-01-23 12:58:43 +00:00
}
}
#ifdef __linux__
// Most non-linux platforms require root to mlock()
if (mlock(state.password.buffer, sizeof(state.password.buffer)) != 0) {
sway_abort("Unable to mlock() password memory.");
}
#endif
wl_list_init(&state.surfaces);
2018-04-03 03:07:43 +00:00
state.xkb.context = xkb_context_new(XKB_CONTEXT_NO_FLAGS);
2018-04-04 22:52:44 +00:00
state.display = wl_display_connect(NULL);
assert(state.display);
2018-04-03 02:48:13 +00:00
struct wl_registry *registry = wl_display_get_registry(state.display);
wl_registry_add_listener(registry, &registry_listener, &state);
wl_display_roundtrip(state.display);
assert(state.compositor && state.layer_shell && state.shm);
if (!state.input_inhibit_manager) {
wlr_log(L_ERROR, "Compositor does not support the input inhibitor "
"protocol, refusing to run insecurely");
}
if (wl_list_empty(&state.surfaces)) {
2018-04-03 02:48:13 +00:00
wlr_log(L_DEBUG, "Exiting - no outputs to show on.");
return 0;
}
struct swaylock_surface *surface;
wl_list_for_each(surface, &state.surfaces, link) {
2018-04-03 19:04:31 +00:00
surface->image = background_image;
2018-04-04 22:52:44 +00:00
surface->surface = wl_compositor_create_surface(state.compositor);
assert(surface->surface);
2018-04-03 02:48:13 +00:00
surface->layer_surface = zwlr_layer_shell_v1_get_layer_surface(
state.layer_shell, surface->surface, surface->output,
2018-04-03 02:48:13 +00:00
ZWLR_LAYER_SHELL_V1_LAYER_OVERLAY, "lockscreen");
assert(surface->layer_surface);
2018-04-03 02:48:13 +00:00
zwlr_layer_surface_v1_set_size(surface->layer_surface, 0, 0);
zwlr_layer_surface_v1_set_anchor(surface->layer_surface,
2018-04-03 02:48:13 +00:00
ZWLR_LAYER_SURFACE_V1_ANCHOR_TOP |
ZWLR_LAYER_SURFACE_V1_ANCHOR_RIGHT |
ZWLR_LAYER_SURFACE_V1_ANCHOR_BOTTOM |
ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT);
zwlr_layer_surface_v1_set_exclusive_zone(surface->layer_surface, -1);
2018-04-03 02:48:13 +00:00
zwlr_layer_surface_v1_set_keyboard_interactivity(
surface->layer_surface, true);
zwlr_layer_surface_v1_add_listener(surface->layer_surface,
&layer_surface_listener, surface);
wl_surface_commit(surface->surface);
2018-04-03 02:48:13 +00:00
wl_display_roundtrip(state.display);
}
zwlr_input_inhibit_manager_v1_get_inhibitor(state.input_inhibit_manager);
2018-04-03 02:48:13 +00:00
state.run_display = true;
while (wl_display_dispatch(state.display) != -1 && state.run_display) {
// This space intentionally left blank
}
2015-12-10 12:52:24 +00:00
return 0;
}