1//===- MCAssembler.h - Object File Generation -------------------*- C++ -*-===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#ifndef LLVM_MC_MCASSEMBLER_H
11#define LLVM_MC_MCASSEMBLER_H
12
13#include "llvm/ADT/SmallPtrSet.h"
14#include "llvm/ADT/ilist.h"
15#include "llvm/ADT/ilist_node.h"
16#include "llvm/ADT/iterator.h"
17#include "llvm/MC/MCDirectives.h"
18#include "llvm/MC/MCDwarf.h"
19#include "llvm/MC/MCFixup.h"
20#include "llvm/MC/MCInst.h"
21#include "llvm/MC/MCLinkerOptimizationHint.h"
22#include "llvm/MC/MCSubtargetInfo.h"
23
24namespace llvm {
25class raw_ostream;
26class MCAsmLayout;
27class MCAssembler;
28class MCContext;
29class MCCodeEmitter;
30class MCExpr;
31class MCFragment;
32class MCObjectWriter;
33class MCSection;
34class MCSubtargetInfo;
35class MCValue;
36class MCAsmBackend;
37
38class MCFragment : public ilist_node_with_parent<MCFragment, MCSection> {
39  friend class MCAsmLayout;
40
41  MCFragment(const MCFragment &) = delete;
42  void operator=(const MCFragment &) = delete;
43
44public:
45  enum FragmentType : uint8_t {
46    FT_Align,
47    FT_Data,
48    FT_CompactEncodedInst,
49    FT_Fill,
50    FT_Relaxable,
51    FT_Org,
52    FT_Dwarf,
53    FT_DwarfFrame,
54    FT_LEB,
55    FT_SafeSEH,
56    FT_Dummy
57  };
58
59private:
60  FragmentType Kind;
61
62protected:
63  bool HasInstructions;
64
65private:
66  /// \brief Should this fragment be aligned to the end of a bundle?
67  bool AlignToBundleEnd;
68
69  uint8_t BundlePadding;
70
71  /// LayoutOrder - The layout order of this fragment.
72  unsigned LayoutOrder;
73
74  /// The data for the section this fragment is in.
75  MCSection *Parent;
76
77  /// Atom - The atom this fragment is in, as represented by it's defining
78  /// symbol.
79  const MCSymbol *Atom;
80
81  /// \name Assembler Backend Data
82  /// @{
83  //
84  // FIXME: This could all be kept private to the assembler implementation.
85
86  /// Offset - The offset of this fragment in its section. This is ~0 until
87  /// initialized.
88  uint64_t Offset;
89
90  /// @}
91
92protected:
93  MCFragment(FragmentType Kind, bool HasInstructions,
94             uint8_t BundlePadding, MCSection *Parent = nullptr);
95
96  ~MCFragment();
97private:
98
99  // This is a friend so that the sentinal can be created.
100  friend struct ilist_sentinel_traits<MCFragment>;
101  MCFragment();
102
103public:
104  /// Destroys the current fragment.
105  ///
106  /// This must be used instead of delete as MCFragment is non-virtual.
107  /// This method will dispatch to the appropriate subclass.
108  void destroy();
109
110  FragmentType getKind() const { return Kind; }
111
112  MCSection *getParent() const { return Parent; }
113  void setParent(MCSection *Value) { Parent = Value; }
114
115  const MCSymbol *getAtom() const { return Atom; }
116  void setAtom(const MCSymbol *Value) { Atom = Value; }
117
118  unsigned getLayoutOrder() const { return LayoutOrder; }
119  void setLayoutOrder(unsigned Value) { LayoutOrder = Value; }
120
121  /// \brief Does this fragment have instructions emitted into it? By default
122  /// this is false, but specific fragment types may set it to true.
123  bool hasInstructions() const { return HasInstructions; }
124
125  /// \brief Should this fragment be placed at the end of an aligned bundle?
126  bool alignToBundleEnd() const { return AlignToBundleEnd; }
127  void setAlignToBundleEnd(bool V) { AlignToBundleEnd = V; }
128
129  /// \brief Get the padding size that must be inserted before this fragment.
130  /// Used for bundling. By default, no padding is inserted.
131  /// Note that padding size is restricted to 8 bits. This is an optimization
132  /// to reduce the amount of space used for each fragment. In practice, larger
133  /// padding should never be required.
134  uint8_t getBundlePadding() const { return BundlePadding; }
135
136  /// \brief Set the padding size for this fragment. By default it's a no-op,
137  /// and only some fragments have a meaningful implementation.
138  void setBundlePadding(uint8_t N) { BundlePadding = N; }
139
140  /// \brief Return true if given frgment has FT_Dummy type.
141  bool isDummy() const { return Kind == FT_Dummy; }
142
143  void dump();
144};
145
146class MCDummyFragment : public MCFragment {
147public:
148  explicit MCDummyFragment(MCSection *Sec)
149      : MCFragment(FT_Dummy, false, 0, Sec){};
150  static bool classof(const MCFragment *F) { return F->getKind() == FT_Dummy; }
151};
152
153/// Interface implemented by fragments that contain encoded instructions and/or
154/// data.
155///
156class MCEncodedFragment : public MCFragment {
157protected:
158  MCEncodedFragment(MCFragment::FragmentType FType, bool HasInstructions,
159                    MCSection *Sec)
160      : MCFragment(FType, HasInstructions, 0, Sec) {}
161
162public:
163  static bool classof(const MCFragment *F) {
164    MCFragment::FragmentType Kind = F->getKind();
165    switch (Kind) {
166    default:
167      return false;
168    case MCFragment::FT_Relaxable:
169    case MCFragment::FT_CompactEncodedInst:
170    case MCFragment::FT_Data:
171      return true;
172    }
173  }
174};
175
176/// Interface implemented by fragments that contain encoded instructions and/or
177/// data.
178///
179template<unsigned ContentsSize>
180class MCEncodedFragmentWithContents : public MCEncodedFragment {
181  SmallVector<char, ContentsSize> Contents;
182
183protected:
184  MCEncodedFragmentWithContents(MCFragment::FragmentType FType,
185                                bool HasInstructions,
186                                MCSection *Sec)
187      : MCEncodedFragment(FType, HasInstructions, Sec) {}
188
189public:
190  SmallVectorImpl<char> &getContents() { return Contents; }
191  const SmallVectorImpl<char> &getContents() const { return Contents; }
192};
193
194/// Interface implemented by fragments that contain encoded instructions and/or
195/// data and also have fixups registered.
196///
197template<unsigned ContentsSize, unsigned FixupsSize>
198class MCEncodedFragmentWithFixups :
199  public MCEncodedFragmentWithContents<ContentsSize> {
200
201  /// Fixups - The list of fixups in this fragment.
202  SmallVector<MCFixup, FixupsSize> Fixups;
203
204protected:
205  MCEncodedFragmentWithFixups(MCFragment::FragmentType FType,
206                              bool HasInstructions,
207                              MCSection *Sec)
208      : MCEncodedFragmentWithContents<ContentsSize>(FType, HasInstructions,
209                                                    Sec) {}
210
211public:
212  typedef SmallVectorImpl<MCFixup>::const_iterator const_fixup_iterator;
213  typedef SmallVectorImpl<MCFixup>::iterator fixup_iterator;
214
215  SmallVectorImpl<MCFixup> &getFixups() { return Fixups; }
216  const SmallVectorImpl<MCFixup> &getFixups() const { return Fixups; }
217
218  fixup_iterator fixup_begin() { return Fixups.begin(); }
219  const_fixup_iterator fixup_begin() const { return Fixups.begin(); }
220
221  fixup_iterator fixup_end() { return Fixups.end(); }
222  const_fixup_iterator fixup_end() const { return Fixups.end(); }
223
224  static bool classof(const MCFragment *F) {
225    MCFragment::FragmentType Kind = F->getKind();
226    return Kind == MCFragment::FT_Relaxable || Kind == MCFragment::FT_Data;
227  }
228};
229
230/// Fragment for data and encoded instructions.
231///
232class MCDataFragment : public MCEncodedFragmentWithFixups<32, 4> {
233public:
234  MCDataFragment(MCSection *Sec = nullptr)
235      : MCEncodedFragmentWithFixups<32, 4>(FT_Data, false, Sec) {}
236
237  void setHasInstructions(bool V) { HasInstructions = V; }
238
239  static bool classof(const MCFragment *F) {
240    return F->getKind() == MCFragment::FT_Data;
241  }
242};
243
244/// This is a compact (memory-size-wise) fragment for holding an encoded
245/// instruction (non-relaxable) that has no fixups registered. When applicable,
246/// it can be used instead of MCDataFragment and lead to lower memory
247/// consumption.
248///
249class MCCompactEncodedInstFragment : public MCEncodedFragmentWithContents<4> {
250public:
251  MCCompactEncodedInstFragment(MCSection *Sec = nullptr)
252      : MCEncodedFragmentWithContents(FT_CompactEncodedInst, true, Sec) {
253  }
254
255  static bool classof(const MCFragment *F) {
256    return F->getKind() == MCFragment::FT_CompactEncodedInst;
257  }
258};
259
260/// A relaxable fragment holds on to its MCInst, since it may need to be
261/// relaxed during the assembler layout and relaxation stage.
262///
263class MCRelaxableFragment : public MCEncodedFragmentWithFixups<8, 1> {
264
265  /// Inst - The instruction this is a fragment for.
266  MCInst Inst;
267
268  /// STI - The MCSubtargetInfo in effect when the instruction was encoded.
269  const MCSubtargetInfo &STI;
270
271public:
272  MCRelaxableFragment(const MCInst &Inst, const MCSubtargetInfo &STI,
273                      MCSection *Sec = nullptr)
274      : MCEncodedFragmentWithFixups(FT_Relaxable, true, Sec),
275        Inst(Inst), STI(STI) {}
276
277  const MCInst &getInst() const { return Inst; }
278  void setInst(const MCInst &Value) { Inst = Value; }
279
280  const MCSubtargetInfo &getSubtargetInfo() { return STI; }
281
282  static bool classof(const MCFragment *F) {
283    return F->getKind() == MCFragment::FT_Relaxable;
284  }
285};
286
287class MCAlignFragment : public MCFragment {
288
289  /// Alignment - The alignment to ensure, in bytes.
290  unsigned Alignment;
291
292  /// EmitNops - Flag to indicate that (optimal) NOPs should be emitted instead
293  /// of using the provided value. The exact interpretation of this flag is
294  /// target dependent.
295  bool EmitNops : 1;
296
297  /// Value - Value to use for filling padding bytes.
298  int64_t Value;
299
300  /// ValueSize - The size of the integer (in bytes) of \p Value.
301  unsigned ValueSize;
302
303  /// MaxBytesToEmit - The maximum number of bytes to emit; if the alignment
304  /// cannot be satisfied in this width then this fragment is ignored.
305  unsigned MaxBytesToEmit;
306
307public:
308  MCAlignFragment(unsigned Alignment, int64_t Value, unsigned ValueSize,
309                  unsigned MaxBytesToEmit, MCSection *Sec = nullptr)
310      : MCFragment(FT_Align, false, 0, Sec), Alignment(Alignment),
311        EmitNops(false), Value(Value),
312        ValueSize(ValueSize), MaxBytesToEmit(MaxBytesToEmit) {}
313
314  /// \name Accessors
315  /// @{
316
317  unsigned getAlignment() const { return Alignment; }
318
319  int64_t getValue() const { return Value; }
320
321  unsigned getValueSize() const { return ValueSize; }
322
323  unsigned getMaxBytesToEmit() const { return MaxBytesToEmit; }
324
325  bool hasEmitNops() const { return EmitNops; }
326  void setEmitNops(bool Value) { EmitNops = Value; }
327
328  /// @}
329
330  static bool classof(const MCFragment *F) {
331    return F->getKind() == MCFragment::FT_Align;
332  }
333};
334
335class MCFillFragment : public MCFragment {
336
337  /// Value - Value to use for filling bytes.
338  int64_t Value;
339
340  /// ValueSize - The size (in bytes) of \p Value to use when filling, or 0 if
341  /// this is a virtual fill fragment.
342  unsigned ValueSize;
343
344  /// Size - The number of bytes to insert.
345  uint64_t Size;
346
347public:
348  MCFillFragment(int64_t Value, unsigned ValueSize, uint64_t Size,
349                 MCSection *Sec = nullptr)
350      : MCFragment(FT_Fill, false, 0, Sec), Value(Value), ValueSize(ValueSize),
351        Size(Size) {
352    assert((!ValueSize || (Size % ValueSize) == 0) &&
353           "Fill size must be a multiple of the value size!");
354  }
355
356  /// \name Accessors
357  /// @{
358
359  int64_t getValue() const { return Value; }
360
361  unsigned getValueSize() const { return ValueSize; }
362
363  uint64_t getSize() const { return Size; }
364
365  /// @}
366
367  static bool classof(const MCFragment *F) {
368    return F->getKind() == MCFragment::FT_Fill;
369  }
370};
371
372class MCOrgFragment : public MCFragment {
373
374  /// Offset - The offset this fragment should start at.
375  const MCExpr *Offset;
376
377  /// Value - Value to use for filling bytes.
378  int8_t Value;
379
380public:
381  MCOrgFragment(const MCExpr &Offset, int8_t Value, MCSection *Sec = nullptr)
382      : MCFragment(FT_Org, false, 0, Sec), Offset(&Offset), Value(Value) {}
383
384  /// \name Accessors
385  /// @{
386
387  const MCExpr &getOffset() const { return *Offset; }
388
389  uint8_t getValue() const { return Value; }
390
391  /// @}
392
393  static bool classof(const MCFragment *F) {
394    return F->getKind() == MCFragment::FT_Org;
395  }
396};
397
398class MCLEBFragment : public MCFragment {
399
400  /// Value - The value this fragment should contain.
401  const MCExpr *Value;
402
403  /// IsSigned - True if this is a sleb128, false if uleb128.
404  bool IsSigned;
405
406  SmallString<8> Contents;
407
408public:
409  MCLEBFragment(const MCExpr &Value_, bool IsSigned_, MCSection *Sec = nullptr)
410      : MCFragment(FT_LEB, false, 0, Sec), Value(&Value_), IsSigned(IsSigned_) {
411    Contents.push_back(0);
412  }
413
414  /// \name Accessors
415  /// @{
416
417  const MCExpr &getValue() const { return *Value; }
418
419  bool isSigned() const { return IsSigned; }
420
421  SmallString<8> &getContents() { return Contents; }
422  const SmallString<8> &getContents() const { return Contents; }
423
424  /// @}
425
426  static bool classof(const MCFragment *F) {
427    return F->getKind() == MCFragment::FT_LEB;
428  }
429};
430
431class MCDwarfLineAddrFragment : public MCFragment {
432
433  /// LineDelta - the value of the difference between the two line numbers
434  /// between two .loc dwarf directives.
435  int64_t LineDelta;
436
437  /// AddrDelta - The expression for the difference of the two symbols that
438  /// make up the address delta between two .loc dwarf directives.
439  const MCExpr *AddrDelta;
440
441  SmallString<8> Contents;
442
443public:
444  MCDwarfLineAddrFragment(int64_t LineDelta, const MCExpr &AddrDelta,
445                          MCSection *Sec = nullptr)
446      : MCFragment(FT_Dwarf, false, 0, Sec), LineDelta(LineDelta),
447        AddrDelta(&AddrDelta) {
448    Contents.push_back(0);
449  }
450
451  /// \name Accessors
452  /// @{
453
454  int64_t getLineDelta() const { return LineDelta; }
455
456  const MCExpr &getAddrDelta() const { return *AddrDelta; }
457
458  SmallString<8> &getContents() { return Contents; }
459  const SmallString<8> &getContents() const { return Contents; }
460
461  /// @}
462
463  static bool classof(const MCFragment *F) {
464    return F->getKind() == MCFragment::FT_Dwarf;
465  }
466};
467
468class MCDwarfCallFrameFragment : public MCFragment {
469
470  /// AddrDelta - The expression for the difference of the two symbols that
471  /// make up the address delta between two .cfi_* dwarf directives.
472  const MCExpr *AddrDelta;
473
474  SmallString<8> Contents;
475
476public:
477  MCDwarfCallFrameFragment(const MCExpr &AddrDelta, MCSection *Sec = nullptr)
478      : MCFragment(FT_DwarfFrame, false, 0, Sec), AddrDelta(&AddrDelta) {
479    Contents.push_back(0);
480  }
481
482  /// \name Accessors
483  /// @{
484
485  const MCExpr &getAddrDelta() const { return *AddrDelta; }
486
487  SmallString<8> &getContents() { return Contents; }
488  const SmallString<8> &getContents() const { return Contents; }
489
490  /// @}
491
492  static bool classof(const MCFragment *F) {
493    return F->getKind() == MCFragment::FT_DwarfFrame;
494  }
495};
496
497class MCSafeSEHFragment : public MCFragment {
498  const MCSymbol *Sym;
499
500public:
501  MCSafeSEHFragment(const MCSymbol *Sym, MCSection *Sec = nullptr)
502      : MCFragment(FT_SafeSEH, false, 0, Sec), Sym(Sym) {}
503
504  /// \name Accessors
505  /// @{
506
507  const MCSymbol *getSymbol() { return Sym; }
508  const MCSymbol *getSymbol() const { return Sym; }
509
510  /// @}
511
512  static bool classof(const MCFragment *F) {
513    return F->getKind() == MCFragment::FT_SafeSEH;
514  }
515};
516
517// FIXME: This really doesn't belong here. See comments below.
518struct IndirectSymbolData {
519  MCSymbol *Symbol;
520  MCSection *Section;
521};
522
523// FIXME: Ditto this. Purely so the Streamer and the ObjectWriter can talk
524// to one another.
525struct DataRegionData {
526  // This enum should be kept in sync w/ the mach-o definition in
527  // llvm/Object/MachOFormat.h.
528  enum KindTy { Data = 1, JumpTable8, JumpTable16, JumpTable32 } Kind;
529  MCSymbol *Start;
530  MCSymbol *End;
531};
532
533class MCAssembler {
534  friend class MCAsmLayout;
535
536public:
537  typedef std::vector<MCSection *> SectionListType;
538  typedef std::vector<const MCSymbol *> SymbolDataListType;
539
540  typedef pointee_iterator<SectionListType::const_iterator> const_iterator;
541  typedef pointee_iterator<SectionListType::iterator> iterator;
542
543  typedef pointee_iterator<SymbolDataListType::const_iterator>
544  const_symbol_iterator;
545  typedef pointee_iterator<SymbolDataListType::iterator> symbol_iterator;
546
547  typedef iterator_range<symbol_iterator> symbol_range;
548  typedef iterator_range<const_symbol_iterator> const_symbol_range;
549
550  typedef std::vector<IndirectSymbolData>::const_iterator
551      const_indirect_symbol_iterator;
552  typedef std::vector<IndirectSymbolData>::iterator indirect_symbol_iterator;
553
554  typedef std::vector<DataRegionData>::const_iterator
555      const_data_region_iterator;
556  typedef std::vector<DataRegionData>::iterator data_region_iterator;
557
558  /// MachO specific deployment target version info.
559  // A Major version of 0 indicates that no version information was supplied
560  // and so the corresponding load command should not be emitted.
561  typedef struct {
562    MCVersionMinType Kind;
563    unsigned Major;
564    unsigned Minor;
565    unsigned Update;
566  } VersionMinInfoType;
567
568private:
569  MCAssembler(const MCAssembler &) = delete;
570  void operator=(const MCAssembler &) = delete;
571
572  MCContext &Context;
573
574  MCAsmBackend &Backend;
575
576  MCCodeEmitter &Emitter;
577
578  MCObjectWriter &Writer;
579
580  SectionListType Sections;
581
582  SymbolDataListType Symbols;
583
584  std::vector<IndirectSymbolData> IndirectSymbols;
585
586  std::vector<DataRegionData> DataRegions;
587
588  /// The list of linker options to propagate into the object file.
589  std::vector<std::vector<std::string>> LinkerOptions;
590
591  /// List of declared file names
592  std::vector<std::string> FileNames;
593
594  MCDwarfLineTableParams LTParams;
595
596  /// The set of function symbols for which a .thumb_func directive has
597  /// been seen.
598  //
599  // FIXME: We really would like this in target specific code rather than
600  // here. Maybe when the relocation stuff moves to target specific,
601  // this can go with it? The streamer would need some target specific
602  // refactoring too.
603  mutable SmallPtrSet<const MCSymbol *, 64> ThumbFuncs;
604
605  /// \brief The bundle alignment size currently set in the assembler.
606  ///
607  /// By default it's 0, which means bundling is disabled.
608  unsigned BundleAlignSize;
609
610  unsigned RelaxAll : 1;
611  unsigned SubsectionsViaSymbols : 1;
612  unsigned IncrementalLinkerCompatible : 1;
613
614  /// ELF specific e_header flags
615  // It would be good if there were an MCELFAssembler class to hold this.
616  // ELF header flags are used both by the integrated and standalone assemblers.
617  // Access to the flags is necessary in cases where assembler directives affect
618  // which flags to be set.
619  unsigned ELFHeaderEFlags;
620
621  /// Used to communicate Linker Optimization Hint information between
622  /// the Streamer and the .o writer
623  MCLOHContainer LOHContainer;
624
625  VersionMinInfoType VersionMinInfo;
626
627private:
628  /// Evaluate a fixup to a relocatable expression and the value which should be
629  /// placed into the fixup.
630  ///
631  /// \param Layout The layout to use for evaluation.
632  /// \param Fixup The fixup to evaluate.
633  /// \param DF The fragment the fixup is inside.
634  /// \param Target [out] On return, the relocatable expression the fixup
635  /// evaluates to.
636  /// \param Value [out] On return, the value of the fixup as currently laid
637  /// out.
638  /// \return Whether the fixup value was fully resolved. This is true if the
639  /// \p Value result is fixed, otherwise the value may change due to
640  /// relocation.
641  bool evaluateFixup(const MCAsmLayout &Layout, const MCFixup &Fixup,
642                     const MCFragment *DF, MCValue &Target,
643                     uint64_t &Value) const;
644
645  /// Check whether a fixup can be satisfied, or whether it needs to be relaxed
646  /// (increased in size, in order to hold its value correctly).
647  bool fixupNeedsRelaxation(const MCFixup &Fixup, const MCRelaxableFragment *DF,
648                            const MCAsmLayout &Layout) const;
649
650  /// Check whether the given fragment needs relaxation.
651  bool fragmentNeedsRelaxation(const MCRelaxableFragment *IF,
652                               const MCAsmLayout &Layout) const;
653
654  /// \brief Perform one layout iteration and return true if any offsets
655  /// were adjusted.
656  bool layoutOnce(MCAsmLayout &Layout);
657
658  /// \brief Perform one layout iteration of the given section and return true
659  /// if any offsets were adjusted.
660  bool layoutSectionOnce(MCAsmLayout &Layout, MCSection &Sec);
661
662  bool relaxInstruction(MCAsmLayout &Layout, MCRelaxableFragment &IF);
663
664  bool relaxLEB(MCAsmLayout &Layout, MCLEBFragment &IF);
665
666  bool relaxDwarfLineAddr(MCAsmLayout &Layout, MCDwarfLineAddrFragment &DF);
667  bool relaxDwarfCallFrameFragment(MCAsmLayout &Layout,
668                                   MCDwarfCallFrameFragment &DF);
669
670  /// finishLayout - Finalize a layout, including fragment lowering.
671  void finishLayout(MCAsmLayout &Layout);
672
673  std::pair<uint64_t, bool> handleFixup(const MCAsmLayout &Layout,
674                                        MCFragment &F, const MCFixup &Fixup);
675
676public:
677  /// Compute the effective fragment size assuming it is laid out at the given
678  /// \p SectionAddress and \p FragmentOffset.
679  uint64_t computeFragmentSize(const MCAsmLayout &Layout,
680                               const MCFragment &F) const;
681
682  /// Find the symbol which defines the atom containing the given symbol, or
683  /// null if there is no such symbol.
684  const MCSymbol *getAtom(const MCSymbol &S) const;
685
686  /// Check whether a particular symbol is visible to the linker and is required
687  /// in the symbol table, or whether it can be discarded by the assembler. This
688  /// also effects whether the assembler treats the label as potentially
689  /// defining a separate atom.
690  bool isSymbolLinkerVisible(const MCSymbol &SD) const;
691
692  /// Emit the section contents using the given object writer.
693  void writeSectionData(const MCSection *Section,
694                        const MCAsmLayout &Layout) const;
695
696  /// Check whether a given symbol has been flagged with .thumb_func.
697  bool isThumbFunc(const MCSymbol *Func) const;
698
699  /// Flag a function symbol as the target of a .thumb_func directive.
700  void setIsThumbFunc(const MCSymbol *Func) { ThumbFuncs.insert(Func); }
701
702  /// ELF e_header flags
703  unsigned getELFHeaderEFlags() const { return ELFHeaderEFlags; }
704  void setELFHeaderEFlags(unsigned Flags) { ELFHeaderEFlags = Flags; }
705
706  /// MachO deployment target version information.
707  const VersionMinInfoType &getVersionMinInfo() const { return VersionMinInfo; }
708  void setVersionMinInfo(MCVersionMinType Kind, unsigned Major, unsigned Minor,
709                         unsigned Update) {
710    VersionMinInfo.Kind = Kind;
711    VersionMinInfo.Major = Major;
712    VersionMinInfo.Minor = Minor;
713    VersionMinInfo.Update = Update;
714  }
715
716public:
717  /// Construct a new assembler instance.
718  //
719  // FIXME: How are we going to parameterize this? Two obvious options are stay
720  // concrete and require clients to pass in a target like object. The other
721  // option is to make this abstract, and have targets provide concrete
722  // implementations as we do with AsmParser.
723  MCAssembler(MCContext &Context_, MCAsmBackend &Backend_,
724              MCCodeEmitter &Emitter_, MCObjectWriter &Writer_);
725  ~MCAssembler();
726
727  /// Reuse an assembler instance
728  ///
729  void reset();
730
731  MCContext &getContext() const { return Context; }
732
733  MCAsmBackend &getBackend() const { return Backend; }
734
735  MCCodeEmitter &getEmitter() const { return Emitter; }
736
737  MCObjectWriter &getWriter() const { return Writer; }
738
739  MCDwarfLineTableParams getDWARFLinetableParams() const { return LTParams; }
740  void setDWARFLinetableParams(MCDwarfLineTableParams P) { LTParams = P; }
741
742  /// Finish - Do final processing and write the object to the output stream.
743  /// \p Writer is used for custom object writer (as the MCJIT does),
744  /// if not specified it is automatically created from backend.
745  void Finish();
746
747  // Layout all section and prepare them for emission.
748  void layout(MCAsmLayout &Layout);
749
750  // FIXME: This does not belong here.
751  bool getSubsectionsViaSymbols() const { return SubsectionsViaSymbols; }
752  void setSubsectionsViaSymbols(bool Value) { SubsectionsViaSymbols = Value; }
753
754  bool isIncrementalLinkerCompatible() const {
755    return IncrementalLinkerCompatible;
756  }
757  void setIncrementalLinkerCompatible(bool Value) {
758    IncrementalLinkerCompatible = Value;
759  }
760
761  bool getRelaxAll() const { return RelaxAll; }
762  void setRelaxAll(bool Value) { RelaxAll = Value; }
763
764  bool isBundlingEnabled() const { return BundleAlignSize != 0; }
765
766  unsigned getBundleAlignSize() const { return BundleAlignSize; }
767
768  void setBundleAlignSize(unsigned Size) {
769    assert((Size == 0 || !(Size & (Size - 1))) &&
770           "Expect a power-of-two bundle align size");
771    BundleAlignSize = Size;
772  }
773
774  /// \name Section List Access
775  /// @{
776
777  iterator begin() { return Sections.begin(); }
778  const_iterator begin() const { return Sections.begin(); }
779
780  iterator end() { return Sections.end(); }
781  const_iterator end() const { return Sections.end(); }
782
783  size_t size() const { return Sections.size(); }
784
785  /// @}
786  /// \name Symbol List Access
787  /// @{
788  symbol_iterator symbol_begin() { return Symbols.begin(); }
789  const_symbol_iterator symbol_begin() const { return Symbols.begin(); }
790
791  symbol_iterator symbol_end() { return Symbols.end(); }
792  const_symbol_iterator symbol_end() const { return Symbols.end(); }
793
794  symbol_range symbols() { return make_range(symbol_begin(), symbol_end()); }
795  const_symbol_range symbols() const {
796    return make_range(symbol_begin(), symbol_end());
797  }
798
799  size_t symbol_size() const { return Symbols.size(); }
800
801  /// @}
802  /// \name Indirect Symbol List Access
803  /// @{
804
805  // FIXME: This is a total hack, this should not be here. Once things are
806  // factored so that the streamer has direct access to the .o writer, it can
807  // disappear.
808  std::vector<IndirectSymbolData> &getIndirectSymbols() {
809    return IndirectSymbols;
810  }
811
812  indirect_symbol_iterator indirect_symbol_begin() {
813    return IndirectSymbols.begin();
814  }
815  const_indirect_symbol_iterator indirect_symbol_begin() const {
816    return IndirectSymbols.begin();
817  }
818
819  indirect_symbol_iterator indirect_symbol_end() {
820    return IndirectSymbols.end();
821  }
822  const_indirect_symbol_iterator indirect_symbol_end() const {
823    return IndirectSymbols.end();
824  }
825
826  size_t indirect_symbol_size() const { return IndirectSymbols.size(); }
827
828  /// @}
829  /// \name Linker Option List Access
830  /// @{
831
832  std::vector<std::vector<std::string>> &getLinkerOptions() {
833    return LinkerOptions;
834  }
835
836  /// @}
837  /// \name Data Region List Access
838  /// @{
839
840  // FIXME: This is a total hack, this should not be here. Once things are
841  // factored so that the streamer has direct access to the .o writer, it can
842  // disappear.
843  std::vector<DataRegionData> &getDataRegions() { return DataRegions; }
844
845  data_region_iterator data_region_begin() { return DataRegions.begin(); }
846  const_data_region_iterator data_region_begin() const {
847    return DataRegions.begin();
848  }
849
850  data_region_iterator data_region_end() { return DataRegions.end(); }
851  const_data_region_iterator data_region_end() const {
852    return DataRegions.end();
853  }
854
855  size_t data_region_size() const { return DataRegions.size(); }
856
857  /// @}
858  /// \name Data Region List Access
859  /// @{
860
861  // FIXME: This is a total hack, this should not be here. Once things are
862  // factored so that the streamer has direct access to the .o writer, it can
863  // disappear.
864  MCLOHContainer &getLOHContainer() { return LOHContainer; }
865  const MCLOHContainer &getLOHContainer() const {
866    return const_cast<MCAssembler *>(this)->getLOHContainer();
867  }
868  /// @}
869  /// \name Backend Data Access
870  /// @{
871
872  bool registerSection(MCSection &Section);
873
874  void registerSymbol(const MCSymbol &Symbol, bool *Created = nullptr);
875
876  ArrayRef<std::string> getFileNames() { return FileNames; }
877
878  void addFileName(StringRef FileName) {
879    if (std::find(FileNames.begin(), FileNames.end(), FileName) ==
880        FileNames.end())
881      FileNames.push_back(FileName);
882  }
883
884  /// \brief Write the necessary bundle padding to the given object writer.
885  /// Expects a fragment \p F containing instructions and its size \p FSize.
886  void writeFragmentPadding(const MCFragment &F, uint64_t FSize,
887                            MCObjectWriter *OW) const;
888
889  /// @}
890
891  void dump();
892};
893
894/// \brief Compute the amount of padding required before the fragment \p F to
895/// obey bundling restrictions, where \p FOffset is the fragment's offset in
896/// its section and \p FSize is the fragment's size.
897uint64_t computeBundlePadding(const MCAssembler &Assembler, const MCFragment *F,
898                              uint64_t FOffset, uint64_t FSize);
899
900} // end namespace llvm
901
902#endif
903