Skip to content

Commit

Permalink
plugins: exec: Windows compatibility for exit code checks
Browse files Browse the repository at this point in the history
Add a compatibility shim to handle win32's lack of POSIX
sys/wait.h macros like WIFEXITED, so fluent-bit's in_exec
plugin can use them on platforms where they're supported,
but fall back to a simple exit code extraction on windows.

Signed-off-by: Craig Ringer <[email protected]>
  • Loading branch information
ringerc committed May 24, 2023
1 parent a686c42 commit b4cdb89
Show file tree
Hide file tree
Showing 4 changed files with 115 additions and 11 deletions.
1 change: 1 addition & 0 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,7 @@ include(ExternalProject)
include(cmake/FindJournald.cmake)
include(cmake/FindMonkey.cmake)
include(cmake/macros.cmake)
include(cmake/platform_feature_checks.cmake)
set(CMAKE_MODULE_PATH "${PROJECT_SOURCE_DIR}/cmake/sanitizers-cmake/cmake" ${CMAKE_MODULE_PATH})
find_package(Sanitizers)

Expand Down
8 changes: 8 additions & 0 deletions cmake/platform_feature_checks.cmake
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# Feature tests for various platform and compiler capabilities,
# system headers, etc.

include(CheckIncludeFile)
CHECK_INCLUDE_FILE("sys/wait.h" FLB_HAVE_SYS_WAIT_H)
if (FLB_HAVE_SYS_WAIT_H)
FLB_DEFINITION(FLB_HAVE_SYS_WAIT_H)
endif()
23 changes: 12 additions & 11 deletions plugins/in_exec/in_exec.c
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/wait.h>
#include "in_exec_win32_compat.h"

#include "in_exec.h"

Expand Down Expand Up @@ -59,7 +59,7 @@ static int in_exec_collect(struct flb_input_instance *ins,
}
}

cmdp = popen(ctx->cmd, "r");
cmdp = flb_popen(ctx->cmd, "r");
if (cmdp == NULL) {
flb_plg_debug(ctx->ins, "command %s failed", ctx->cmd);
goto collect_end;
Expand Down Expand Up @@ -177,9 +177,8 @@ static int in_exec_collect(struct flb_input_instance *ins,
* https://www.gnu.org/software/bash/manual/html_node/Exit-Status.html
* and
* https://skarnet.org/software/execline/exitcodes.html
*
*/
cmdret = pclose(cmdp);
cmdret = flb_pclose(cmdp);
if (cmdret == -1) {
flb_errno();
flb_plg_debug(ctx->ins,
Expand All @@ -191,24 +190,26 @@ static int in_exec_collect(struct flb_input_instance *ins,
* returned through a shell by a real child command.
*/
flb_exit_code = 128;
} else if (WIFEXITED(cmdret)) {
flb_plg_debug(ctx->ins, "command %s exited with code %d", ctx->cmd, WEXITSTATUS(cmdret));
} else if (FLB_WIFEXITED(cmdret)) {
flb_plg_debug(ctx->ins, "command %s exited with code %d",
ctx->cmd, FLB_WEXITSTATUS(cmdret));
/*
* Propagate shell exit code, which may encode a normal or signal
* exit for the real child process, directly to the caller. This
* could be greater than 127 if the shell encoded a signal exit
* status from the child process into its own return code.
*/
flb_exit_code = WEXITSTATUS(cmdret);
} else if (WIFSIGNALED(cmdret)) {
flb_plg_debug(ctx->ins, "command %s exited with signal %d", ctx->cmd, WTERMSIG(cmdret));
flb_exit_code = FLB_WEXITSTATUS(cmdret);
} else if (FLB_WIFSIGNALED(cmdret)) {
flb_plg_debug(ctx->ins, "command %s exited with signal %d",
ctx->cmd, FLB_WTERMSIG(cmdret));
/*
* Follow the shell convention of returning 128+signo for signal
* exits. The consumer of fluent-bit's exit code will be unable to
* differentiate between the shell exiting on a signal and the
* process called by the shell exiting on a signal.
*/
flb_exit_code = 128 + WTERMSIG(cmdret);
flb_exit_code = 128 + FLB_WTERMSIG(cmdret);
} else {
flb_plg_debug(ctx->ins, "command %s exited with unknown status",
ctx->cmd);
Expand All @@ -227,7 +228,7 @@ static int in_exec_collect(struct flb_input_instance *ins,
if (ctx->propagate_exit_code == FLB_TRUE) {
config->exit_status_code = flb_exit_code;
}
flb_plg_debug(ctx->ins,
flb_plg_info(ctx->ins,
"one-shot command exited, terminating fluent-bit");
flb_engine_exit(config);
} else {
Expand Down
94 changes: 94 additions & 0 deletions plugins/in_exec/in_exec_win32_compat.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,94 @@
/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */

/* Fluent Bit
* ==========
* Copyright (C) 2015-2022 The Fluent Bit Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef FLB_IN_EXEC_WIN32_COMPAT_H
#define FLB_IN_EXEC_WIN32_COMPAT_H

#include <stdio.h>
#include <fluent-bit/flb_info.h>

/*
* Work around lack of sys/wait.h and POSIX exit status macros from waitpid()
* in win32's _popen() and _pclose() implementation, since fluent-bit uses
* these in the in_exec plugin.
*
* On POSIX-like OSes this'll just use the standard macros with a name alias.
*
* On windows, where the concept of a signal exit does not exist, it defines
* dummy macros to indicate that the process exited normally and extract the
* exit code.
*
* These macros are for use with flb_pclose() only. Do not use them with
* other APIs that may differ in return value semantics.
*/
#ifdef FLB_HAVE_SYS_WAIT_H
#include <sys/wait.h>
#define FLB_WIFEXITED(status) WIFEXITED((status))
#define FLB_WEXITSTATUS(status) WEXITSTATUS((status))
#define FLB_WIFSIGNALED(status) WIFSIGNALED((status))
#define FLB_WTERMSIG(status) WTERMSIG((status))
#else
#define FLB_WIFEXITED(status) (1)
#define FLB_WEXITSTATUS(status) ((status) & 0x00ff)
#define FLB_WIFSIGNALED(status) (0)
#define FLB_WTERMSIG(status) (-1)
#endif

/*
* Because Windows has to do everything differently, call _popen() and
* _pclose() instead of the POSIX popen() and pclose() functions.
*
* flb_pclose() has different return value semantics on Windows vs non-windows
* targets because it propagates the pclose() or _pclose() return value
* directly. You MUST use the FLB_WIFEXITED(), FLB_WEXITSTATUS(),
* FLB_WIFSIGNALED() and FLB_WTERMSIG() macros to consume the return value,
* rather than the underlying POSIX macros or manual bit-shifts.
*/
#if !defined(FLB_SYSTEM_WINDOWS)
static inline FILE* flb_popen(const char *command, const char *type) {
return popen(command, type);
}
static inline int flb_pclose(FILE *stream) {
return pclose(stream);
}
#define FLB_PCLOSE pclose
#else
static inline FILE* flb_popen(const char *command, const char *type) {
return _popen(command, type);
}
/*
* flb_pclose() has the same return value on Windows as win32 _pclose(), rather
* than posix pclose(). The process exit code is not bit-shifted to the high
* byte.
*
* The MSVC docs for _pclose() at
* https://learn.microsoft.com/en-us/cpp/c-runtime-library/reference/pclose?view=msvc-170
* are misleading; they say that "The format of the return value is the same as
* for _cwait, except the low-order and high-order bytes are swapped." But
* _cwait isn't documented as having any meaningful return on success, the
* process exit code is meant to be in its "termstat" out parameter per
* https://learn.microsoft.com/en-us/cpp/c-runtime-library/reference/cwait?view=msvc-170
* The return code of _pclose() actually appears to be the process exit code
* without the bit-shift that waitpid() applies.
*/
static inline int flb_pclose(FILE *stream) {
return _pclose(stream);
}
#endif

#endif /* FLB_IN_EXEC_WIN32_COMPAT_H */

0 comments on commit b4cdb89

Please sign in to comment.