1//===-- RecordStreamer.cpp - Record asm definde and used symbols ----------===//
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#include "RecordStreamer.h"
11#include "llvm/MC/MCSymbol.h"
12using namespace llvm;
13
14void RecordStreamer::markDefined(const MCSymbol &Symbol) {
15  State &S = Symbols[Symbol.getName()];
16  switch (S) {
17  case DefinedGlobal:
18  case Global:
19    S = DefinedGlobal;
20    break;
21  case NeverSeen:
22  case Defined:
23  case Used:
24    S = Defined;
25    break;
26  case GlobalWeak:
27    break;
28  }
29}
30
31void RecordStreamer::markGlobal(const MCSymbol &Symbol,
32                                MCSymbolAttr Attribute) {
33  State &S = Symbols[Symbol.getName()];
34  switch (S) {
35  case DefinedGlobal:
36  case Defined:
37    S = (Attribute == MCSA_Weak) ? GlobalWeak : DefinedGlobal;
38    break;
39
40  case NeverSeen:
41  case Global:
42  case Used:
43    S = (Attribute == MCSA_Weak) ? GlobalWeak : Global;
44    break;
45  case GlobalWeak:
46    break;
47  }
48}
49
50void RecordStreamer::markUsed(const MCSymbol &Symbol) {
51  State &S = Symbols[Symbol.getName()];
52  switch (S) {
53  case DefinedGlobal:
54  case Defined:
55  case Global:
56  case GlobalWeak:
57    break;
58
59  case NeverSeen:
60  case Used:
61    S = Used;
62    break;
63  }
64}
65
66void RecordStreamer::visitUsedSymbol(const MCSymbol &Sym) { markUsed(Sym); }
67
68RecordStreamer::const_iterator RecordStreamer::begin() {
69  return Symbols.begin();
70}
71
72RecordStreamer::const_iterator RecordStreamer::end() { return Symbols.end(); }
73
74RecordStreamer::RecordStreamer(MCContext &Context) : MCStreamer(Context) {}
75
76void RecordStreamer::EmitInstruction(const MCInst &Inst,
77                                     const MCSubtargetInfo &STI) {
78  MCStreamer::EmitInstruction(Inst, STI);
79}
80
81void RecordStreamer::EmitLabel(MCSymbol *Symbol) {
82  MCStreamer::EmitLabel(Symbol);
83  markDefined(*Symbol);
84}
85
86void RecordStreamer::EmitAssignment(MCSymbol *Symbol, const MCExpr *Value) {
87  markDefined(*Symbol);
88  MCStreamer::EmitAssignment(Symbol, Value);
89}
90
91bool RecordStreamer::EmitSymbolAttribute(MCSymbol *Symbol,
92                                         MCSymbolAttr Attribute) {
93  if (Attribute == MCSA_Global || Attribute == MCSA_Weak)
94    markGlobal(*Symbol, Attribute);
95  return true;
96}
97
98void RecordStreamer::EmitZerofill(MCSection *Section, MCSymbol *Symbol,
99                                  uint64_t Size, unsigned ByteAlignment) {
100  markDefined(*Symbol);
101}
102
103void RecordStreamer::EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
104                                      unsigned ByteAlignment) {
105  markDefined(*Symbol);
106}
107