b2a253855f
Allow to have a name for workers in case the fuzzed code is itself using threads. Reviewers: vitalybuka Reviewed-By: vitalybuka Differential Revision: https://reviews.llvm.org/D155754
54 lines
1.5 KiB
C++
54 lines
1.5 KiB
C++
//===- FuzzerUtilLinux.cpp - Misc utils for Linux. ------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
// Misc utils for Linux.
|
|
//===----------------------------------------------------------------------===//
|
|
#include "FuzzerPlatform.h"
|
|
#if LIBFUZZER_LINUX || LIBFUZZER_NETBSD || LIBFUZZER_FREEBSD || \
|
|
LIBFUZZER_EMSCRIPTEN
|
|
#include "FuzzerCommand.h"
|
|
#include "FuzzerInternal.h"
|
|
|
|
#include <signal.h>
|
|
#include <stdlib.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
|
|
|
|
namespace fuzzer {
|
|
|
|
int ExecuteCommand(const Command &Cmd) {
|
|
std::string CmdLine = Cmd.toString();
|
|
int exit_code = system(CmdLine.c_str());
|
|
if (WIFEXITED(exit_code))
|
|
return WEXITSTATUS(exit_code);
|
|
if (WIFSIGNALED(exit_code) && WTERMSIG(exit_code) == SIGINT)
|
|
return Fuzzer::InterruptExitCode();
|
|
return exit_code;
|
|
}
|
|
|
|
void DiscardOutput(int Fd) {
|
|
FILE* Temp = fopen("/dev/null", "w");
|
|
if (!Temp)
|
|
return;
|
|
dup2(fileno(Temp), Fd);
|
|
fclose(Temp);
|
|
}
|
|
|
|
void SetThreadName(std::thread &thread, const std::string &name) {
|
|
#if LIBFUZZER_LINUX || LIBFUZZER_FREEBSD
|
|
(void)pthread_setname_np(thread.native_handle(), name.c_str());
|
|
#elif LIBFUZZER_NETBSD
|
|
(void)pthread_set_name_np(thread.native_handle(), "%s", name.c_str());
|
|
#endif
|
|
}
|
|
|
|
} // namespace fuzzer
|
|
|
|
#endif
|