Processing: add --processing-timeout

This commit is contained in:
李通洲
2023-07-10 22:12:07 +08:00
parent d53b584ba2
commit 7dae59ee92
10 changed files with 23 additions and 8 deletions
+1
View File
@@ -37,6 +37,7 @@ static void defaultConfig(void)
ffStrbufInit(&instance.config.colorKeys);
ffStrbufInit(&instance.config.colorTitle);
ffStrbufInitS(&instance.config.keyValueSeparator, ": ");
instance.config.processingTimeout = 1000;
#if defined(__linux__) || defined(__FreeBSD__)
ffStrbufInit(&instance.config.playerName);
+2
View File
@@ -309,6 +309,8 @@ const char* ffParseGeneralJsonConfig(void)
config->escapeBedrock = yyjson_get_bool(val);
else if (ffStrEqualsIgnCase(key, "pipe"))
config->pipe = yyjson_get_bool(val);
else if (ffStrEqualsIgnCase(key, "processingTimeout"))
config->processingTimeout = (int32_t) yyjson_get_int(val);
#if defined(__linux__) || defined(__FreeBSD__)
else if (ffStrEqualsIgnCase(key, "playerName"))
-2
View File
@@ -5,8 +5,6 @@
#include "util/FFstrbuf.h"
#define FF_WAIT_TIMEOUT 1000
const char* ffProcessAppendOutput(FFstrbuf* buffer, char* const argv[], bool useStdErr);
static inline const char* ffProcessAppendStdOut(FFstrbuf* buffer, char* const argv[])
+6 -4
View File
@@ -16,15 +16,17 @@
int waitpid_timeout(pid_t pid, int* status)
{
if (FF_WAIT_TIMEOUT <= 0)
if (instance.config.processingTimeout <= 0)
return waitpid(pid, status, 0);
uint32_t timeout = (uint32_t) instance.config.processingTimeout;
#if defined(__linux__) && defined(SYS_pidfd_open) // musl don't define SYS_pidfd_open
FF_AUTO_CLOSE_FD int pidfd = (int) syscall(SYS_pidfd_open, pid, 0);
if (pidfd >= 0)
{
int res = poll(&(struct pollfd) { .events = POLLIN, .fd = pidfd }, 1, FF_WAIT_TIMEOUT);
int res = poll(&(struct pollfd) { .events = POLLIN, .fd = pidfd }, 1, (int) timeout);
if (res > 0)
return (int) waitpid(pid, status, WNOHANG);
else if (res == 0)
@@ -43,8 +45,8 @@ int waitpid_timeout(pid_t pid, int* status)
int res = (int) waitpid(pid, status, WNOHANG);
if (res != 0)
return res;
if (ffTimeGetTick() - start < FF_WAIT_TIMEOUT)
ffTimeSleep(FF_WAIT_TIMEOUT / 10);
if (ffTimeGetTick() - start < timeout)
ffTimeSleep(timeout / 10);
else
{
kill(pid, SIGTERM);
+2 -2
View File
@@ -58,9 +58,9 @@ const char* ffProcessAppendOutput(FFstrbuf* buffer, char* const argv[], bool use
if(!success)
return "CreateProcessA() failed";
if (FF_WAIT_TIMEOUT > 0)
if (instance.config.processingTimeout > 0)
{
DWORD ret = WaitForSingleObjectEx(piProcInfo.hProcess, FF_WAIT_TIMEOUT, TRUE);
DWORD ret = WaitForSingleObjectEx(piProcInfo.hProcess, (DWORD) instance.config.processingTimeout, TRUE);
if (ret == WAIT_TIMEOUT)
{
TerminateProcess(piProcInfo.hProcess, 1);