Index: llvm/test/CMakeLists.txt =================================================================== --- llvm/test/CMakeLists.txt +++ llvm/test/CMakeLists.txt @@ -6,6 +6,7 @@ LLVM_ENABLE_FFI LLVM_ENABLE_THREADS LLVM_ENABLE_CURL + LLVM_ENABLE_HTTPLIB LLVM_ENABLE_ZLIB LLVM_ENABLE_LIBXML2 LLVM_INCLUDE_GO_TESTS @@ -75,6 +76,7 @@ llvm-cxxfilt llvm-cxxmap llvm-debuginfod-find + llvm-debuginfod llvm-diff llvm-dis llvm-dlltool Index: llvm/test/lit.cfg.py =================================================================== --- llvm/test/lit.cfg.py +++ llvm/test/lit.cfg.py @@ -158,7 +158,7 @@ tools.extend([ 'dsymutil', 'lli', 'lli-child-target', 'llvm-ar', 'llvm-as', 'llvm-addr2line', 'llvm-bcanalyzer', 'llvm-bitcode-strip', 'llvm-config', - 'llvm-cov', 'llvm-cxxdump', 'llvm-cvtres', 'llvm-debuginfod-find', + 'llvm-cov', 'llvm-cxxdump', 'llvm-cvtres', 'llvm-debuginfod-find', 'llvm-debuginfod', 'llvm-diff', 'llvm-dis', 'llvm-dwarfdump', 'llvm-dlltool', 'llvm-exegesis', 'llvm-extract', 'llvm-isel-fuzzer', 'llvm-ifs', 'llvm-install-name-tool', 'llvm-jitlink', 'llvm-opt-fuzzer', 'llvm-lib', @@ -401,6 +401,9 @@ if config.have_curl: config.available_features.add('curl') +if config.have_httplib: + config.available_features.add('httplib') + if config.have_opt_viewer_modules: config.available_features.add('have_opt_viewer_modules') @@ -427,4 +430,3 @@ if 'aix' in config.target_triple: for directory in ('/CodeGen/X86', '/DebugInfo', '/DebugInfo/X86', '/DebugInfo/Generic', '/LTO/X86', '/Linker'): exclude_unsupported_files_for_aix(config.test_source_root + directory) - Index: llvm/test/lit.site.cfg.py.in =================================================================== --- llvm/test/lit.site.cfg.py.in +++ llvm/test/lit.site.cfg.py.in @@ -39,6 +39,7 @@ config.have_libxar = @LLVM_HAVE_LIBXAR@ config.have_libxml2 = @LLVM_ENABLE_LIBXML2@ config.have_curl = @LLVM_ENABLE_CURL@ +config.have_httplib = @LLVM_ENABLE_HTTPLIB@ config.have_dia_sdk = @LLVM_ENABLE_DIA_SDK@ config.enable_ffi = @LLVM_ENABLE_FFI@ config.build_examples = @LLVM_BUILD_EXAMPLES@ Index: llvm/test/tools/llvm-debuginfod/llvm-debuginfod.test =================================================================== --- /dev/null +++ llvm/test/tools/llvm-debuginfod/llvm-debuginfod.test @@ -0,0 +1,110 @@ +# REQUIRES: curl, httplib, thread_support + +#int main () { +# int x = 1; +# return x; +#} +# +#Build as : clang -g main.c -o main-debug.exe +#Then run : cp main-debug.exe main.exe && strip main.exe +#resulting buildid: 2c39b7557c50162aaeb5a3148c9f76e6e46012e3 + +# RUN: rm -rf %t +# RUN: mkdir %t +# # Query the debuginfod server for artifacts +# RUN: DEBUGINFOD_CACHE_PATH=%t %python %s --server-cmd 'llvm-debuginfod -v -c 3 %S/Inputs' \ +# RUN: --tool-cmd 'llvm-debuginfod-find --dump --executable 2c39b7557c50162aaeb5a3148c9f76e6e46012e3' | \ +# RUN: diff - %S/Inputs/main.exe +# RUN: DEBUGINFOD_CACHE_PATH=%t %python %s --server-cmd 'llvm-debuginfod -v -c 3 %S/Inputs' \ +# RUN: --tool-cmd 'llvm-debuginfod-find --dump --debuginfo 2c39b7557c50162aaeb5a3148c9f76e6e46012e3' | \ +# RUN: diff - %S/Inputs/main-debug.exe +# Debuginfod server does not yet support source files + +# # The artifacts should still be present in the cache without needing to query +# # the server. +# RUN: DEBUGINFOD_CACHE_PATH=%t %python %s \ +# RUN: --tool-cmd 'llvm-debuginfod-find --dump --executable 2c39b7557c50162aaeb5a3148c9f76e6e46012e3' | \ +# RUN: diff - %S/Inputs/main.exe +# RUN: DEBUGINFOD_CACHE_PATH=%t %python %s \ +# RUN: --tool-cmd 'llvm-debuginfod-find --dump --debuginfo 2c39b7557c50162aaeb5a3148c9f76e6e46012e3' | \ +# RUN: diff - %S/Inputs/main-debug.exe + + + +# This script is used to test the debuginfod client within a host tool against +# the debuginfod server. +# It first stands up the debuginfod server and then executes the tool. +# This way the tool can make debuginfod HTTP requests to the debuginfod server. +import argparse +import threading +import subprocess +import sys +import os +import io + +# Starts the server and obtains the port number from the first line of stdout. +# Waits until the server has completed one full directory scan before returning. +def start_debuginfod_server(server_args): + env = os.environ + process = subprocess.Popen( + server_args, + env=env, + stdout=subprocess.PIPE) + port = -1 + # Obtain the port + stdout_reader = io.TextIOWrapper(process.stdout, encoding='ascii') + stdout_line = stdout_reader.readline() + port = int(stdout_line.split()[-1]) + # Wait until a directory scan is completed + while True: + stdout_line = stdout_reader.readline().strip() + print(stdout_line, file=sys.stderr) + if stdout_line == 'Updated collection': + break + return (process, port) + +# Starts the server with the specified args (if nonempty), then runs the tool +# with specified args. +# Sets the DEBUGINFOD_CACHE_PATH env var to point at the given cache_directory. +# Sets the DEBUGINFOD_URLS env var to point at the local server. +def test_tool(server_args, tool_args): + server_process = None + client_process = None + port = None + if server_args: + server_process, port = start_debuginfod_server(server_args) + try: + env = os.environ + if port is not None: + env['DEBUGINFOD_URLS'] = 'http://localhost:%s' % port + client_process = subprocess.Popen( + tool_args, env=env) + client_code = client_process.wait() + if client_code != 0: + print('nontrivial client return code %s' % client_code, file=sys.stderr) + return 1 + if server_process is not None: + server_process.terminate() + server_code = server_process.wait() + if server_code != -15: + print('nontrivial server return code %s' % server_code, file=sys.stderr) + return 1 + + finally: + if server_process is not None: + server_process.terminate() + if client_process is not None: + client_process.terminate() + return 0 + +def main(): + parser = argparse.ArgumentParser() + parser.add_argument('--server-cmd', default='', help='Command to start the server. If not present, no server is started.', type=str) + parser.add_argument('--tool-cmd', required=True, type=str) + args = parser.parse_args() + result = test_tool(args.server_cmd.split(), + args.tool_cmd.split()) + sys.exit(result) + +if __name__ == '__main__': + main() Index: llvm/tools/llvm-debuginfod/CMakeLists.txt =================================================================== --- /dev/null +++ llvm/tools/llvm-debuginfod/CMakeLists.txt @@ -0,0 +1,10 @@ +set(LLVM_LINK_COMPONENTS + Debuginfod + Support + ) +add_llvm_tool(llvm-debuginfod + llvm-debuginfod.cpp + ) +if(LLVM_INSTALL_BINUTILS_SYMLINKS) + add_llvm_tool_symlink(debuginfod llvm-debuginfod) +endif() Index: llvm/tools/llvm-debuginfod/llvm-debuginfod.cpp =================================================================== --- /dev/null +++ llvm/tools/llvm-debuginfod/llvm-debuginfod.cpp @@ -0,0 +1,98 @@ +//===-- llvm-debuginfod.cpp - federating debuginfod server ----------------===// +// +// 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 +// +//===----------------------------------------------------------------------===// +/// +/// \file +/// This file contains the llvm-debuginfod-find tool. This tool +/// queries the debuginfod servers in the DEBUGINFOD_URLS environment +/// variable (delimited by space (" ")) for the executable, +/// debuginfo, or specified source file of the binary matching the +/// given build-id. +/// +//===----------------------------------------------------------------------===// + +#include "llvm/Debuginfod/Debuginfod.h" +#include "llvm/Debuginfod/HTTPClient.h" +#include "llvm/Support/CommandLine.h" +#include "llvm/Support/InitLLVM.h" +#include "llvm/Support/ThreadPool.h" + +using namespace llvm; + +cl::OptionCategory DebuginfodCategory("llvm-debuginfod Options"); + +static cl::list ScanPaths(cl::Positional, + cl::desc(""), + cl::cat(DebuginfodCategory)); + +static cl::opt + Port("p", cl::init(0), + cl::desc("Port to listen on. Set to 0 to bind to any available port."), + cl::cat(DebuginfodCategory)); + +static cl::opt + HostInterface("i", cl::init("0.0.0.0"), + cl::desc("Host interface to bind to."), + cl::cat(DebuginfodCategory)); + +static cl::opt + ScanInterval("t", cl::init(10000), + cl::desc("Number of seconds to wait between subsequent scans " + "of the filesystem."), + cl::cat(DebuginfodCategory)); + +static cl::opt + MaxConcurrency("c", cl::init(0), + cl::desc("Maximum number of files to scan concurrently. If " + "0, use the hardware concurrency."), + cl::cat(DebuginfodCategory)); + +static cl::opt VerboseLogging("v", cl::init(false), + cl::desc("Enable verbose logging."), + cl::cat(DebuginfodCategory)); + +ExitOnError ExitOnErr; + +int main(int argc, char **argv) { + InitLLVM X(argc, argv); + HTTPClient::initialize(); + cl::HideUnrelatedOptions({&DebuginfodCategory}); + cl::ParseCommandLineOptions(argc, argv); + + SmallVector Paths; + for (const std::string &Path : ScanPaths) + Paths.push_back(Path); + + ThreadPool Pool(hardware_concurrency(MaxConcurrency)); + DebuginfodLog Log; + DebuginfodCollection Collection(Paths, Log, Pool); + DebuginfodServer Server(Log, Collection); + + if (!Port) + Port = ExitOnErr(Server.Server.bind(HostInterface.c_str())); + else + ExitOnErr(Server.Server.bind(Port, HostInterface.c_str())); + + Log.push("Listening on port " + Twine(Port).str()); + + Pool.async([&]() { ExitOnErr(Server.Server.listen()); }); + Pool.async([&]() { + while (1) { + Log.wait(); + if (Optional Entry = Log.pop()) { + if (VerboseLogging) { + outs() << Entry->Message << "\n"; + outs().flush(); + } + } + } + }); + if (Paths.size()) + ExitOnErr(Collection.updateForever( + std::chrono::milliseconds(static_cast(ScanInterval * 1000)))); + Pool.wait(); +}