1//===- LTO.h ----------------------------------------------------*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file provides a way to combine bitcode files into one COFF
10// file by compiling them using LLVM.
11//
12// If LTO is in use, your input files are not in regular COFF files
13// but instead LLVM bitcode files. In that case, the linker has to
14// convert bitcode files into the native format so that we can create
15// a COFF file that contains native code. This file provides that
16// functionality.
17//
18//===----------------------------------------------------------------------===//
19
20#ifndef LLD_COFF_LTO_H
21#define LLD_COFF_LTO_H
22
23#include "lld/Common/LLVM.h"
24#include "llvm/ADT/DenseSet.h"
25#include "llvm/ADT/SmallString.h"
26#include "llvm/Support/raw_ostream.h"
27#include <memory>
28#include <vector>
29
30namespace llvm::lto {
31struct Config;
32class LTO;
33}
34
35namespace lld::coff {
36
37class BitcodeFile;
38class InputFile;
39class COFFLinkerContext;
40
41class BitcodeCompiler {
42public:
43 BitcodeCompiler(COFFLinkerContext &ctx);
44 ~BitcodeCompiler();
45
46 void add(BitcodeFile &f);
47 std::vector<InputFile *> compile();
48
49private:
50 std::unique_ptr<llvm::lto::LTO> ltoObj;
51 std::vector<std::pair<std::string, SmallString<0>>> buf;
52 std::vector<std::unique_ptr<MemoryBuffer>> files;
53 std::vector<std::string> file_names;
54 std::unique_ptr<llvm::raw_fd_ostream> indexFile;
55 llvm::DenseSet<StringRef> thinIndices;
56
57 std::string getThinLTOOutputFile(StringRef path);
58 llvm::lto::Config createConfig();
59
60 COFFLinkerContext &ctx;
61};
62}
63
64#endif
65