summaryrefslogtreecommitdiff
path: root/include/llvm/Linker.h
blob: 3667b8521d90c5c8e482e9f3f4f84e4f9e4e764a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
//===- llvm/Linker.h - Module Linker Interface ------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_LINKER_H
#define LLVM_LINKER_H

#include "llvm/ADT/SmallPtrSet.h"
#include <string>

namespace llvm {

class Module;
class StringRef;
class StructType;

/// This class provides the core functionality of linking in LLVM. It keeps a
/// pointer to the merged module so far. It doesn't take ownership of the
/// module since it is assumed that the user of this class will want to do
/// something with it after the linking.
class Linker {
  public:
    enum LinkerMode {
      DestroySource = 0, // Allow source module to be destroyed.
      PreserveSource = 1 // Preserve the source module.
    };

    Linker(Module *M);
    ~Linker();
    Module *getModule() const { return Composite; }

    /// \brief Link \p Src into the composite. The source is destroyed if
    /// \p Mode is DestroySource and preserved if it is PreserveSource.
    /// If \p ErrorMsg is not null, information about any error is written
    /// to it.
    /// Returns true on error.
    bool linkInModule(Module *Src, unsigned Mode, std::string *ErrorMsg);
    bool linkInModule(Module *Src, std::string *ErrorMsg) {
      return linkInModule(Src, Linker::DestroySource, ErrorMsg);
    }

    static bool LinkModules(Module *Dest, Module *Src, unsigned Mode,
                            std::string *ErrorMsg);

  private:
    Module *Composite;
    SmallPtrSet<StructType*, 32> IdentifiedStructTypes;
};

} // End llvm namespace

#endif