Remove allowing only 1 process of gpu-screen-recorder at a time
This commit is contained in:
parent
1a7fd28a3f
commit
912515303b
60
src/main.cpp
60
src/main.cpp
@ -1033,70 +1033,11 @@ static void save_replay_async(AVCodecContext *video_codec_context, int video_str
|
||||
});
|
||||
}
|
||||
|
||||
static bool is_process_running_program(pid_t pid, const char *program_name) {
|
||||
char filepath[256];
|
||||
snprintf(filepath, sizeof(filepath), "/proc/%ld/exe", (long)pid);
|
||||
|
||||
char resolved_path[PATH_MAX];
|
||||
const ssize_t resolved_path_len = readlink(filepath, resolved_path, sizeof(resolved_path) - 1);
|
||||
if(resolved_path_len == -1)
|
||||
return false;
|
||||
|
||||
resolved_path[resolved_path_len] = '\0';
|
||||
|
||||
const int program_name_len = strlen(program_name);
|
||||
return resolved_path_len >= program_name_len && memcmp(resolved_path + resolved_path_len - program_name_len, program_name, program_name_len) == 0;
|
||||
}
|
||||
|
||||
static void handle_existing_pid_file() {
|
||||
char buffer[256];
|
||||
int fd = open(pid_file, O_RDONLY);
|
||||
if(fd == -1)
|
||||
return;
|
||||
|
||||
ssize_t bytes_read = read(fd, buffer, sizeof(buffer) - 1);
|
||||
if(bytes_read < 0) {
|
||||
perror("failed to read gpu-screen-recorder pid file");
|
||||
exit(1);
|
||||
}
|
||||
buffer[bytes_read] = '\0';
|
||||
close(fd);
|
||||
|
||||
long pid = 0;
|
||||
if(sscanf(buffer, "%ld %120s", &pid, buffer) == 2) {
|
||||
if(is_process_running_program(pid, "gpu-screen-recorder")) {
|
||||
fprintf(stderr, "Error: gpu-screen-recorder is already running\n");
|
||||
exit(1);
|
||||
}
|
||||
} else {
|
||||
fprintf(stderr, "Warning: gpu-screen-recorder pid file is in incorrect format, it's possible that its corrupt. Replacing file and continuing...\n");
|
||||
}
|
||||
unlink(pid_file);
|
||||
}
|
||||
|
||||
static void handle_new_pid_file(const char *mode) {
|
||||
int fd = open(pid_file, O_WRONLY|O_CREAT|O_TRUNC, 0777);
|
||||
if(fd == -1) {
|
||||
perror("failed to create gpu-screen-recorder pid file");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
char buffer[256];
|
||||
const int buffer_size = snprintf(buffer, sizeof(buffer), "%ld %s", (long)getpid(), mode);
|
||||
if(write(fd, buffer, buffer_size) == -1) {
|
||||
perror("failed to write gpu-screen-recorder pid file");
|
||||
exit(1);
|
||||
}
|
||||
close(fd);
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
signal(SIGTERM, term_handler);
|
||||
signal(SIGINT, int_handler);
|
||||
signal(SIGUSR1, save_replay_handler);
|
||||
|
||||
handle_existing_pid_file();
|
||||
|
||||
std::map<std::string, Arg> args = {
|
||||
{ "-w", Arg { {}, false, false } },
|
||||
//{ "-s", Arg { nullptr, true } },
|
||||
@ -1601,7 +1542,6 @@ int main(int argc, char **argv) {
|
||||
}, av_format_context, &write_output_mutex);
|
||||
}
|
||||
|
||||
handle_new_pid_file(replay_buffer_size_secs == -1 ? "record" : "replay");
|
||||
started = 1;
|
||||
|
||||
// Set update_fps to 24 to test if duplicate/delayed frames cause video/audio desync or too fast/slow video.
|
||||
|
Loading…
Reference in New Issue
Block a user