Alexandre Ganea 6f2e92c10c Re-land [LLD] Allow usage of LLD as a library
This reverts commit aa495214b39d475bab24b468de7a7c676ce9e366.

As discussed in https://github.com/llvm/llvm-project/issues/53475 this patch
allows for using LLD-as-a-lib. It also lets clients link only the drivers that
they want (see unit tests).

This also adds the unit test infra as in the other LLVM projects. Among the
test coverage, I've added the original issue from @krzysz00, see:
https://github.com/ROCmSoftwarePlatform/D108850-lld-bug-reproduction

Important note: this doesn't allow (yet) linking in parallel. This will come a
bit later hopefully, in subsequent patches, for COFF at least.

Differential revision: https://reviews.llvm.org/D119049
2023-06-19 07:35:11 -04:00

36 lines
1.3 KiB
C++

//===- AllDrivers.cpp -------------------------------------------*- C++ -*-===//
//
// This file is licensed 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
//
//===----------------------------------------------------------------------===//
// This test shows a typical case where all LLD drivers are linked into the
// application binary. This is very similar to how lld.exe binary is linked,
// except that here we cleanup the internal LLD memory context after each call.
//===----------------------------------------------------------------------===//
#include "lld/Common/Driver.h"
#include "gmock/gmock.h"
LLD_HAS_DRIVER(coff)
LLD_HAS_DRIVER(elf)
LLD_HAS_DRIVER(mingw)
LLD_HAS_DRIVER(macho)
LLD_HAS_DRIVER(wasm)
static bool lldInvoke(std::vector<const char *> args) {
args.push_back("--version");
lld::Result r =
lld::lldMain(args, llvm::outs(), llvm::errs(), LLD_ALL_DRIVERS);
return !r.retCode && r.canRunAgain;
}
TEST(AsLib, AllDrivers) {
EXPECT_TRUE(lldInvoke({"ld.lld"}));
EXPECT_TRUE(lldInvoke({"ld64.lld"}));
EXPECT_TRUE(lldInvoke({"ld", "-m", "i386pe"})); // MinGW
EXPECT_TRUE(lldInvoke({"lld-link"}));
EXPECT_TRUE(lldInvoke({"wasm-ld"}));
}