File Manager

003 File Manager

Current Path: /usr/src/contrib/llvm-project/compiler-rt/lib/fuzzer

usr / src / contrib / llvm-project / compiler-rt / lib / fuzzer /

📁 ..
📄 FuzzerBuiltins.h(1.25 KB)
📄 FuzzerBuiltinsMsvc.h(2.33 KB)
📄 FuzzerCommand.h(5.95 KB)
📄 FuzzerCorpus.h(16.93 KB)
📄 FuzzerCrossOver.cpp(1.85 KB)
📄 FuzzerDataFlowTrace.cpp(9.93 KB)
📄 FuzzerDataFlowTrace.h(3.98 KB)
📄 FuzzerDefs.h(1.99 KB)
📄 FuzzerDictionary.h(3.32 KB)
📄 FuzzerDriver.cpp(28.35 KB)
📄 FuzzerExtFunctions.def(2.5 KB)
📄 FuzzerExtFunctions.h(1.04 KB)
📄 FuzzerExtFunctionsDlsym.cpp(1.63 KB)
📄 FuzzerExtFunctionsWeak.cpp(1.9 KB)
📄 FuzzerExtFunctionsWindows.cpp(2.79 KB)
📄 FuzzerExtraCounters.cpp(1.55 KB)
📄 FuzzerFlags.def(10.02 KB)
📄 FuzzerFork.cpp(12.53 KB)
📄 FuzzerFork.h(778 B)
📄 FuzzerIO.cpp(4.47 KB)
📄 FuzzerIO.h(3.31 KB)
📄 FuzzerIOPosix.cpp(4.47 KB)
📄 FuzzerIOWindows.cpp(11.59 KB)
📄 FuzzerInterface.h(3.03 KB)
📄 FuzzerInternal.h(5.33 KB)
📄 FuzzerLoop.cpp(28.54 KB)
📄 FuzzerMain.cpp(791 B)
📄 FuzzerMerge.cpp(13.75 KB)
📄 FuzzerMerge.h(3.31 KB)
📄 FuzzerMutate.cpp(20.02 KB)
📄 FuzzerMutate.h(6.57 KB)
📄 FuzzerOptions.h(2.36 KB)
📄 FuzzerPlatform.h(4.69 KB)
📄 FuzzerRandom.h(1.21 KB)
📄 FuzzerSHA1.cpp(5.56 KB)
📄 FuzzerSHA1.h(978 B)
📄 FuzzerTracePC.cpp(22.56 KB)
📄 FuzzerTracePC.h(8.87 KB)
📄 FuzzerUtil.cpp(6.75 KB)
📄 FuzzerUtil.h(3.25 KB)
📄 FuzzerUtilDarwin.cpp(5.87 KB)
📄 FuzzerUtilFuchsia.cpp(20.16 KB)
📄 FuzzerUtilLinux.cpp(1.12 KB)
📄 FuzzerUtilPosix.cpp(5.07 KB)
📄 FuzzerUtilWindows.cpp(5.93 KB)
📄 FuzzerValueBitMap.h(2.24 KB)
📄 README.txt(40 B)

Editing: FuzzerUtilPosix.cpp

//===- FuzzerUtilPosix.cpp - Misc utils for Posix. ------------------------===//
//
// 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 implementation using Posix API.
//===----------------------------------------------------------------------===//
#include "FuzzerPlatform.h"
#if LIBFUZZER_POSIX
#include "FuzzerIO.h"
#include "FuzzerInternal.h"
#include "FuzzerTracePC.h"
#include <cassert>
#include <chrono>
#include <cstring>
#include <errno.h>
#include <iomanip>
#include <signal.h>
#include <stdio.h>
#include <sys/mman.h>
#include <sys/resource.h>
#include <sys/syscall.h>
#include <sys/time.h>
#include <sys/types.h>
#include <thread>
#include <unistd.h>

namespace fuzzer {

static void AlarmHandler(int, siginfo_t *, void *) {
  Fuzzer::StaticAlarmCallback();
}

static void (*upstream_segv_handler)(int, siginfo_t *, void *);

static void SegvHandler(int sig, siginfo_t *si, void *ucontext) {
  assert(si->si_signo == SIGSEGV);
  if (upstream_segv_handler)
    return upstream_segv_handler(sig, si, ucontext);
  Fuzzer::StaticCrashSignalCallback();
}

static void CrashHandler(int, siginfo_t *, void *) {
  Fuzzer::StaticCrashSignalCallback();
}

static void InterruptHandler(int, siginfo_t *, void *) {
  Fuzzer::StaticInterruptCallback();
}

static void GracefulExitHandler(int, siginfo_t *, void *) {
  Fuzzer::StaticGracefulExitCallback();
}

static void FileSizeExceedHandler(int, siginfo_t *, void *) {
  Fuzzer::StaticFileSizeExceedCallback();
}

static void SetSigaction(int signum,
                         void (*callback)(int, siginfo_t *, void *)) {
  struct sigaction sigact = {};
  if (sigaction(signum, nullptr, &sigact)) {
    Printf("libFuzzer: sigaction failed with %d\n", errno);
    exit(1);
  }
  if (sigact.sa_flags & SA_SIGINFO) {
    if (sigact.sa_sigaction) {
      if (signum != SIGSEGV)
        return;
      upstream_segv_handler = sigact.sa_sigaction;
    }
  } else {
    if (sigact.sa_handler != SIG_DFL && sigact.sa_handler != SIG_IGN &&
        sigact.sa_handler != SIG_ERR)
      return;
  }

sigact = {};
  sigact.sa_flags = SA_SIGINFO;
  sigact.sa_sigaction = callback;
  if (sigaction(signum, &sigact, 0)) {
    Printf("libFuzzer: sigaction failed with %d\n", errno);
    exit(1);
  }
}

// Return true on success, false otherwise.
bool ExecuteCommand(const Command &Cmd, std::string *CmdOutput) {
  FILE *Pipe = popen(Cmd.toString().c_str(), "r");
  if (!Pipe)
    return false;

if (CmdOutput) {
    char TmpBuffer[128];
    while (fgets(TmpBuffer, sizeof(TmpBuffer), Pipe))
      CmdOutput->append(TmpBuffer);
  }
  return pclose(Pipe) == 0;
}

void SetTimer(int Seconds) {
  struct itimerval T {
    {Seconds, 0}, { Seconds, 0 }
  };
  if (setitimer(ITIMER_REAL, &T, nullptr)) {
    Printf("libFuzzer: setitimer failed with %d\n", errno);
    exit(1);
  }
  SetSigaction(SIGALRM, AlarmHandler);
}

void SetSignalHandler(const FuzzingOptions& Options) {
  // setitimer is not implemented in emscripten.
  if (Options.UnitTimeoutSec > 0 && !LIBFUZZER_EMSCRIPTEN)
    SetTimer(Options.UnitTimeoutSec / 2 + 1);
  if (Options.HandleInt)
    SetSigaction(SIGINT, InterruptHandler);
  if (Options.HandleTerm)
    SetSigaction(SIGTERM, InterruptHandler);
  if (Options.HandleSegv)
    SetSigaction(SIGSEGV, SegvHandler);
  if (Options.HandleBus)
    SetSigaction(SIGBUS, CrashHandler);
  if (Options.HandleAbrt)
    SetSigaction(SIGABRT, CrashHandler);
  if (Options.HandleIll)
    SetSigaction(SIGILL, CrashHandler);
  if (Options.HandleFpe)
    SetSigaction(SIGFPE, CrashHandler);
  if (Options.HandleXfsz)
    SetSigaction(SIGXFSZ, FileSizeExceedHandler);
  if (Options.HandleUsr1)
    SetSigaction(SIGUSR1, GracefulExitHandler);
  if (Options.HandleUsr2)
    SetSigaction(SIGUSR2, GracefulExitHandler);
}

void SleepSeconds(int Seconds) {
  sleep(Seconds); // Use C API to avoid coverage from instrumented libc++.
}

unsigned long GetPid() { return (unsigned long)getpid(); }

size_t GetPeakRSSMb() {
  struct rusage usage;
  if (getrusage(RUSAGE_SELF, &usage))
    return 0;
  if (LIBFUZZER_LINUX || LIBFUZZER_FREEBSD || LIBFUZZER_NETBSD ||
      LIBFUZZER_OPENBSD || LIBFUZZER_EMSCRIPTEN) {
    // ru_maxrss is in KiB
    return usage.ru_maxrss >> 10;
  } else if (LIBFUZZER_APPLE) {
    // ru_maxrss is in bytes
    return usage.ru_maxrss >> 20;
  }
  assert(0 && "GetPeakRSSMb() is not implemented for your platform");
  return 0;
}

FILE *OpenProcessPipe(const char *Command, const char *Mode) {
  return popen(Command, Mode);
}

int CloseProcessPipe(FILE *F) {
  return pclose(F);
}

const void *SearchMemory(const void *Data, size_t DataLen, const void *Patt,
                         size_t PattLen) {
  return memmem(Data, DataLen, Patt, PattLen);
}

std::string DisassembleCmd(const std::string &FileName) {
  return "objdump -d " + FileName;
}

std::string SearchRegexCmd(const std::string &Regex) {
  return "grep '" + Regex + "'";
}

}  // namespace fuzzer

#endif // LIBFUZZER_POSIX

003 File Manager

Editing: FuzzerUtilPosix.cpp

Upload File

Create Folder