mirror of
https://github.com/swaywm/sway.git
synced 2024-09-08 20:16:58 +00:00
https://github.com/SirCmpwn/sway/issues/1350 is fixed with this. The commit that change swaygrab to use fork instead of Popen tried to write to the read end of the pipe in the child branch and exec in the parent branch. This commit fixes both of those and closes the write fd after writing, so convert actually exits.
299 lines
7.4 KiB
C
299 lines
7.4 KiB
C
#define _XOPEN_SOURCE 700
|
|
#define _POSIX_C_SOURCE 199309L
|
|
#include <stdio.h>
|
|
#include <stdbool.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <getopt.h>
|
|
#include <unistd.h>
|
|
#include <stdint.h>
|
|
#include <math.h>
|
|
#include <time.h>
|
|
#include <sys/wait.h>
|
|
#include <json-c/json.h>
|
|
#include "log.h"
|
|
#include "ipc-client.h"
|
|
#include "util.h"
|
|
#include "swaygrab/json.h"
|
|
|
|
void sway_terminate(int exit_code) {
|
|
exit(exit_code);
|
|
}
|
|
|
|
void grab_and_apply_magick(const char *file, const char *payload,
|
|
int socketfd, int raw) {
|
|
uint32_t len = strlen(payload);
|
|
char *pixels = ipc_single_command(socketfd,
|
|
IPC_SWAY_GET_PIXELS, payload, &len);
|
|
uint32_t *u32pixels = (uint32_t *)(pixels + 1);
|
|
uint32_t width = u32pixels[0];
|
|
uint32_t height = u32pixels[1];
|
|
len -= 9;
|
|
pixels += 9;
|
|
|
|
if (width == 0 || height == 0) {
|
|
// indicates geometry was clamped by WLC because it was outside of the output's area
|
|
json_object *obj = json_tokener_parse(payload);
|
|
json_object *output;
|
|
json_object_object_get_ex(obj, "output", &output);
|
|
const char *name = json_object_get_string(output);
|
|
json_object_put(obj);
|
|
sway_abort("Unknown output %s.", name);
|
|
}
|
|
|
|
if (raw) {
|
|
fwrite(pixels, 1, len, stdout);
|
|
fflush(stdout);
|
|
free(pixels - 9);
|
|
return;
|
|
}
|
|
|
|
char size[10 + 1 + 10 + 2 + 1]; // int32_t are max 10 digits
|
|
sprintf(size, "%dx%d+0", width, height);
|
|
|
|
pid_t child;
|
|
int fd[2];
|
|
pipe(fd);
|
|
|
|
if ((child = fork()) < 0) {
|
|
sway_log(L_ERROR, "Swaygrab failed to fork.");
|
|
exit(EXIT_FAILURE);
|
|
} else if (child != 0) {
|
|
close(fd[0]);
|
|
write(fd[1], pixels, len);
|
|
close(fd[1]);
|
|
free(pixels - 9);
|
|
waitpid(child, NULL, 0);
|
|
} else {
|
|
close(fd[1]);
|
|
if (dup2(fd[0], 0) != 0) {
|
|
sway_log(L_ERROR, "Could not fdup the pipe");
|
|
}
|
|
close(fd[0]);
|
|
execlp("convert", "convert", "-depth", "8", "-size", size, "rgba:-", "-flip", file, NULL);
|
|
sway_log(L_ERROR, "Swaygrab could not run convert.");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
void grab_and_apply_movie_magic(const char *file, const char *payload,
|
|
int socketfd, int raw, int framerate) {
|
|
if (raw) {
|
|
sway_log(L_ERROR, "Raw capture data is not yet supported. Proceeding with ffmpeg normally.");
|
|
}
|
|
|
|
uint32_t len = strlen(payload);
|
|
char *pixels = ipc_single_command(socketfd,
|
|
IPC_SWAY_GET_PIXELS, payload, &len);
|
|
uint32_t *u32pixels = (uint32_t *)(pixels + 1);
|
|
uint32_t width = u32pixels[0];
|
|
uint32_t height = u32pixels[1];
|
|
pixels += 9;
|
|
|
|
if (width == 0 || height == 0) {
|
|
// indicates geometry was clamped by WLC because it was outside of the output's area
|
|
json_object *obj = json_tokener_parse(payload);
|
|
json_object *output;
|
|
json_object_object_get_ex(obj, "output", &output);
|
|
const char *name = json_object_get_string(output);
|
|
json_object_put(obj);
|
|
sway_abort("Unknown output %s.", name);
|
|
}
|
|
|
|
char *ffmpeg_opts = getenv("SWAYGRAB_FFMPEG_OPTS");
|
|
if(!ffmpeg_opts) {
|
|
ffmpeg_opts = "";
|
|
}
|
|
|
|
const char *fmt = "ffmpeg %s -f rawvideo -framerate %d "
|
|
"-video_size %dx%d -pixel_format argb "
|
|
"-i pipe:0 -r %d -vf vflip %s";
|
|
char *cmd = malloc(strlen(fmt) - 8 /*args*/
|
|
+ strlen(ffmpeg_opts) + numlen(width) + numlen(height)
|
|
+ numlen(framerate) * 2 + strlen(file) + 1);
|
|
sprintf(cmd, fmt, ffmpeg_opts, framerate, width, height, framerate, file);
|
|
|
|
long ns = (long)(1000000000 * (1.0 / framerate));
|
|
struct timespec start, finish, ts;
|
|
ts.tv_sec = 0;
|
|
|
|
FILE *f = popen(cmd, "w");
|
|
fwrite(pixels, 1, len, f);
|
|
free(pixels - 9);
|
|
int sleep = 0;
|
|
while (sleep != -1) {
|
|
clock_gettime(CLOCK_MONOTONIC, &start);
|
|
len = strlen(payload);
|
|
pixels = ipc_single_command(socketfd,
|
|
IPC_SWAY_GET_PIXELS, payload, &len);
|
|
pixels += 9;
|
|
len -= 9;
|
|
|
|
fwrite(pixels, 1, len, f);
|
|
|
|
free(pixels - 9);
|
|
clock_gettime(CLOCK_MONOTONIC, &finish);
|
|
ts.tv_nsec = ns;
|
|
double fts = (double)finish.tv_sec + 1.0e-9*finish.tv_nsec;
|
|
double sts = (double)start.tv_sec + 1.0e-9*start.tv_nsec;
|
|
long diff = (fts - sts) * 1000000000;
|
|
ts.tv_nsec = ns - diff;
|
|
if (ts.tv_nsec < 0) {
|
|
ts.tv_nsec = 0;
|
|
}
|
|
sleep = nanosleep(&ts, NULL);
|
|
}
|
|
fflush(f);
|
|
|
|
fclose(f);
|
|
free(cmd);
|
|
}
|
|
|
|
char *default_filename(const char *extension) {
|
|
int ext_len = strlen(extension);
|
|
int len = 28 + ext_len; // format: "2015-12-17-180040_swaygrab.ext"
|
|
char *filename = malloc(len * sizeof(char));
|
|
time_t t = time(NULL);
|
|
|
|
struct tm *lt = localtime(&t);
|
|
strftime(filename, len, "%Y-%m-%d-%H%M%S_swaygrab.", lt);
|
|
strncat(filename, extension, ext_len);
|
|
|
|
return filename;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
static int capture = 0, raw = 0;
|
|
char *socket_path = NULL;
|
|
char *output = NULL;
|
|
int framerate = 30;
|
|
bool grab_focused = false;
|
|
|
|
init_log(L_INFO);
|
|
|
|
static struct option long_options[] = {
|
|
{"help", no_argument, NULL, 'h'},
|
|
{"capture", no_argument, NULL, 'c'},
|
|
{"output", required_argument, NULL, 'o'},
|
|
{"version", no_argument, NULL, 'v'},
|
|
{"socket", required_argument, NULL, 's'},
|
|
{"raw", no_argument, NULL, 'r'},
|
|
{"rate", required_argument, NULL, 'R'},
|
|
{"focused", no_argument, NULL, 'f'},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
const char *usage =
|
|
"Usage: swaygrab [options] [file]\n"
|
|
"\n"
|
|
" -h, --help Show help message and quit.\n"
|
|
" -c, --capture Capture video.\n"
|
|
" -o, --output <output> Output source.\n"
|
|
" -v, --version Show the version number and quit.\n"
|
|
" -s, --socket <socket> Use the specified socket.\n"
|
|
" -R, --rate <rate> Specify framerate (default: 30)\n"
|
|
" -r, --raw Write raw rgba data to stdout.\n"
|
|
" -f, --focused Grab the focused container.\n";
|
|
|
|
int c;
|
|
while (1) {
|
|
int option_index = 0;
|
|
c = getopt_long(argc, argv, "hco:vs:R:rf", long_options, &option_index);
|
|
if (c == -1) {
|
|
break;
|
|
}
|
|
switch (c) {
|
|
case 'f':
|
|
grab_focused = true;
|
|
break;
|
|
case 's': // Socket
|
|
socket_path = strdup(optarg);
|
|
break;
|
|
case 'r':
|
|
raw = 1;
|
|
break;
|
|
case 'o': // output
|
|
output = strdup(optarg);
|
|
break;
|
|
case 'c':
|
|
capture = 1;
|
|
break;
|
|
case 'R': // Frame rate
|
|
framerate = atoi(optarg);
|
|
break;
|
|
case 'v':
|
|
fprintf(stdout, "sway version " SWAY_VERSION "\n");
|
|
exit(EXIT_SUCCESS);
|
|
break;
|
|
default:
|
|
fprintf(stderr, "%s", usage);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
if (!socket_path) {
|
|
socket_path = get_socketpath();
|
|
if (!socket_path) {
|
|
sway_abort("Unable to retrieve socket path");
|
|
}
|
|
}
|
|
|
|
char *file = NULL;
|
|
if (raw) {
|
|
if (optind >= argc + 1) {
|
|
sway_abort("Invalid usage. See `man swaygrab` %d %d", argc, optind);
|
|
}
|
|
} else if (optind < argc) {
|
|
file = strdup(argv[optind]);
|
|
}
|
|
|
|
int socketfd = ipc_open_socket(socket_path);
|
|
free(socket_path);
|
|
|
|
init_json_tree(socketfd);
|
|
|
|
struct wlc_geometry *geo;
|
|
|
|
if (grab_focused) {
|
|
output = get_focused_output();
|
|
json_object *con = get_focused_container();
|
|
json_object *name;
|
|
json_object_object_get_ex(con, "name", &name);
|
|
geo = get_container_geometry(con);
|
|
free(con);
|
|
} else {
|
|
if (!output) {
|
|
output = get_focused_output();
|
|
}
|
|
geo = get_container_geometry(get_output_container(output));
|
|
// the geometry of the output in the get_tree response is relative to a global (0, 0).
|
|
// we need it to be relative to itself, so set origin to (0, 0) always.
|
|
geo->origin.x = 0;
|
|
geo->origin.y = 0;
|
|
}
|
|
|
|
const char *payload = create_payload(output, geo);
|
|
|
|
free(geo);
|
|
|
|
if (!file) {
|
|
if (!capture) {
|
|
file = default_filename("png");
|
|
} else {
|
|
file = default_filename("webm");
|
|
}
|
|
}
|
|
|
|
if (!capture) {
|
|
grab_and_apply_magick(file, payload, socketfd, raw);
|
|
} else {
|
|
grab_and_apply_movie_magic(file, payload, socketfd, raw, framerate);
|
|
}
|
|
|
|
free_json_tree();
|
|
free(output);
|
|
free(file);
|
|
close(socketfd);
|
|
return 0;
|
|
}
|