1 //===- DarwinAsmParser.cpp - Darwin (Mach-O) Assembly Parser --------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/MC/MCParser/MCAsmParserExtension.h"
11 #include "llvm/ADT/StringRef.h"
12 #include "llvm/ADT/StringSwitch.h"
13 #include "llvm/ADT/Twine.h"
14 #include "llvm/MC/MCContext.h"
15 #include "llvm/MC/MCParser/MCAsmLexer.h"
16 #include "llvm/MC/MCParser/MCAsmParser.h"
17 #include "llvm/MC/MCSectionMachO.h"
18 #include "llvm/MC/MCStreamer.h"
19 #include "llvm/MC/MCSymbol.h"
20 #include "llvm/Support/MemoryBuffer.h"
21 #include "llvm/Support/SourceMgr.h"
26 /// \brief Implementation of directive handling which is shared across all
28 class DarwinAsmParser : public MCAsmParserExtension {
29 template<bool (DarwinAsmParser::*HandlerMethod)(StringRef, SMLoc)>
30 void addDirectiveHandler(StringRef Directive) {
31 MCAsmParser::ExtensionDirectiveHandler Handler = std::make_pair(
32 this, HandleDirective<DarwinAsmParser, HandlerMethod>);
33 getParser().addDirectiveHandler(Directive, Handler);
36 bool ParseSectionSwitch(const char *Segment, const char *Section,
37 unsigned TAA = 0, unsigned ImplicitAlign = 0,
38 unsigned StubSize = 0);
43 virtual void Initialize(MCAsmParser &Parser) {
44 // Call the base implementation.
45 this->MCAsmParserExtension::Initialize(Parser);
47 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveDesc>(".desc");
48 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveIndirectSymbol>(
50 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveLsym>(".lsym");
51 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveSubsectionsViaSymbols>(
52 ".subsections_via_symbols");
53 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveDumpOrLoad>(".dump");
54 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveDumpOrLoad>(".load");
55 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveSection>(".section");
56 addDirectiveHandler<&DarwinAsmParser::ParseDirectivePushSection>(
58 addDirectiveHandler<&DarwinAsmParser::ParseDirectivePopSection>(
60 addDirectiveHandler<&DarwinAsmParser::ParseDirectivePrevious>(".previous");
61 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveSecureLogUnique>(
62 ".secure_log_unique");
63 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveSecureLogReset>(
65 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveTBSS>(".tbss");
66 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveZerofill>(".zerofill");
68 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveDataRegion>(
70 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveDataRegionEnd>(
73 // Special section directives.
74 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveBss>(".bss");
75 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveConst>(".const");
76 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveConstData>(
78 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveConstructor>(
80 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveCString>(
82 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveData>(".data");
83 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveDestructor>(
85 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveDyld>(".dyld");
86 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveFVMLibInit0>(
88 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveFVMLibInit1>(
91 &DarwinAsmParser::ParseSectionDirectiveLazySymbolPointers>(
92 ".lazy_symbol_pointer");
93 addDirectiveHandler<&DarwinAsmParser::ParseDirectiveLinkerOption>(
95 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveLiteral16>(
97 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveLiteral4>(
99 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveLiteral8>(
101 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveModInitFunc>(
103 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveModTermFunc>(
106 &DarwinAsmParser::ParseSectionDirectiveNonLazySymbolPointers>(
107 ".non_lazy_symbol_pointer");
108 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCCatClsMeth>(
109 ".objc_cat_cls_meth");
110 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCCatInstMeth>(
111 ".objc_cat_inst_meth");
112 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCCategory>(
114 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCClass>(
116 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCClassNames>(
117 ".objc_class_names");
118 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCClassVars>(
120 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCClsMeth>(
122 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCClsRefs>(
124 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCInstMeth>(
127 &DarwinAsmParser::ParseSectionDirectiveObjCInstanceVars>(
128 ".objc_instance_vars");
129 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCMessageRefs>(
130 ".objc_message_refs");
131 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCMetaClass>(
134 &DarwinAsmParser::ParseSectionDirectiveObjCMethVarNames>(
135 ".objc_meth_var_names");
137 &DarwinAsmParser::ParseSectionDirectiveObjCMethVarTypes>(
138 ".objc_meth_var_types");
139 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCModuleInfo>(
140 ".objc_module_info");
141 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCProtocol>(
144 &DarwinAsmParser::ParseSectionDirectiveObjCSelectorStrs>(
145 ".objc_selector_strs");
147 &DarwinAsmParser::ParseSectionDirectiveObjCStringObject>(
148 ".objc_string_object");
149 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveObjCSymbols>(
151 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectivePICSymbolStub>(
153 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveStaticConst>(
155 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveStaticData>(
157 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveSymbolStub>(
159 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveTData>(".tdata");
160 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveText>(".text");
161 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveThreadInitFunc>(
162 ".thread_init_func");
163 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveTLV>(".tlv");
165 addDirectiveHandler<&DarwinAsmParser::ParseSectionDirectiveIdent>(".ident");
168 bool ParseDirectiveDesc(StringRef, SMLoc);
169 bool ParseDirectiveIndirectSymbol(StringRef, SMLoc);
170 bool ParseDirectiveDumpOrLoad(StringRef, SMLoc);
171 bool ParseDirectiveLsym(StringRef, SMLoc);
172 bool ParseDirectiveLinkerOption(StringRef, SMLoc);
173 bool ParseDirectiveSection(StringRef, SMLoc);
174 bool ParseDirectivePushSection(StringRef, SMLoc);
175 bool ParseDirectivePopSection(StringRef, SMLoc);
176 bool ParseDirectivePrevious(StringRef, SMLoc);
177 bool ParseDirectiveSecureLogReset(StringRef, SMLoc);
178 bool ParseDirectiveSecureLogUnique(StringRef, SMLoc);
179 bool ParseDirectiveSubsectionsViaSymbols(StringRef, SMLoc);
180 bool ParseDirectiveTBSS(StringRef, SMLoc);
181 bool ParseDirectiveZerofill(StringRef, SMLoc);
182 bool ParseDirectiveDataRegion(StringRef, SMLoc);
183 bool ParseDirectiveDataRegionEnd(StringRef, SMLoc);
185 // Named Section Directive
186 bool ParseSectionDirectiveBss(StringRef, SMLoc) {
187 return ParseSectionSwitch("__DATA", "__bss");
190 bool ParseSectionDirectiveConst(StringRef, SMLoc) {
191 return ParseSectionSwitch("__TEXT", "__const");
193 bool ParseSectionDirectiveStaticConst(StringRef, SMLoc) {
194 return ParseSectionSwitch("__TEXT", "__static_const");
196 bool ParseSectionDirectiveCString(StringRef, SMLoc) {
197 return ParseSectionSwitch("__TEXT","__cstring",
198 MCSectionMachO::S_CSTRING_LITERALS);
200 bool ParseSectionDirectiveLiteral4(StringRef, SMLoc) {
201 return ParseSectionSwitch("__TEXT", "__literal4",
202 MCSectionMachO::S_4BYTE_LITERALS, 4);
204 bool ParseSectionDirectiveLiteral8(StringRef, SMLoc) {
205 return ParseSectionSwitch("__TEXT", "__literal8",
206 MCSectionMachO::S_8BYTE_LITERALS, 8);
208 bool ParseSectionDirectiveLiteral16(StringRef, SMLoc) {
209 return ParseSectionSwitch("__TEXT","__literal16",
210 MCSectionMachO::S_16BYTE_LITERALS, 16);
212 bool ParseSectionDirectiveConstructor(StringRef, SMLoc) {
213 return ParseSectionSwitch("__TEXT","__constructor");
215 bool ParseSectionDirectiveDestructor(StringRef, SMLoc) {
216 return ParseSectionSwitch("__TEXT","__destructor");
218 bool ParseSectionDirectiveFVMLibInit0(StringRef, SMLoc) {
219 return ParseSectionSwitch("__TEXT","__fvmlib_init0");
221 bool ParseSectionDirectiveFVMLibInit1(StringRef, SMLoc) {
222 return ParseSectionSwitch("__TEXT","__fvmlib_init1");
224 bool ParseSectionDirectiveSymbolStub(StringRef, SMLoc) {
225 return ParseSectionSwitch("__TEXT","__symbol_stub",
226 MCSectionMachO::S_SYMBOL_STUBS |
227 MCSectionMachO::S_ATTR_PURE_INSTRUCTIONS,
228 // FIXME: Different on PPC and ARM.
231 bool ParseSectionDirectivePICSymbolStub(StringRef, SMLoc) {
232 return ParseSectionSwitch("__TEXT","__picsymbol_stub",
233 MCSectionMachO::S_SYMBOL_STUBS |
234 MCSectionMachO::S_ATTR_PURE_INSTRUCTIONS, 0, 26);
236 bool ParseSectionDirectiveData(StringRef, SMLoc) {
237 return ParseSectionSwitch("__DATA", "__data");
239 bool ParseSectionDirectiveStaticData(StringRef, SMLoc) {
240 return ParseSectionSwitch("__DATA", "__static_data");
242 bool ParseSectionDirectiveNonLazySymbolPointers(StringRef, SMLoc) {
243 return ParseSectionSwitch("__DATA", "__nl_symbol_ptr",
244 MCSectionMachO::S_NON_LAZY_SYMBOL_POINTERS, 4);
246 bool ParseSectionDirectiveLazySymbolPointers(StringRef, SMLoc) {
247 return ParseSectionSwitch("__DATA", "__la_symbol_ptr",
248 MCSectionMachO::S_LAZY_SYMBOL_POINTERS, 4);
250 bool ParseSectionDirectiveDyld(StringRef, SMLoc) {
251 return ParseSectionSwitch("__DATA", "__dyld");
253 bool ParseSectionDirectiveModInitFunc(StringRef, SMLoc) {
254 return ParseSectionSwitch("__DATA", "__mod_init_func",
255 MCSectionMachO::S_MOD_INIT_FUNC_POINTERS, 4);
257 bool ParseSectionDirectiveModTermFunc(StringRef, SMLoc) {
258 return ParseSectionSwitch("__DATA", "__mod_term_func",
259 MCSectionMachO::S_MOD_TERM_FUNC_POINTERS, 4);
261 bool ParseSectionDirectiveConstData(StringRef, SMLoc) {
262 return ParseSectionSwitch("__DATA", "__const");
264 bool ParseSectionDirectiveObjCClass(StringRef, SMLoc) {
265 return ParseSectionSwitch("__OBJC", "__class",
266 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
268 bool ParseSectionDirectiveObjCMetaClass(StringRef, SMLoc) {
269 return ParseSectionSwitch("__OBJC", "__meta_class",
270 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
272 bool ParseSectionDirectiveObjCCatClsMeth(StringRef, SMLoc) {
273 return ParseSectionSwitch("__OBJC", "__cat_cls_meth",
274 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
276 bool ParseSectionDirectiveObjCCatInstMeth(StringRef, SMLoc) {
277 return ParseSectionSwitch("__OBJC", "__cat_inst_meth",
278 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
280 bool ParseSectionDirectiveObjCProtocol(StringRef, SMLoc) {
281 return ParseSectionSwitch("__OBJC", "__protocol",
282 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
284 bool ParseSectionDirectiveObjCStringObject(StringRef, SMLoc) {
285 return ParseSectionSwitch("__OBJC", "__string_object",
286 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
288 bool ParseSectionDirectiveObjCClsMeth(StringRef, SMLoc) {
289 return ParseSectionSwitch("__OBJC", "__cls_meth",
290 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
292 bool ParseSectionDirectiveObjCInstMeth(StringRef, SMLoc) {
293 return ParseSectionSwitch("__OBJC", "__inst_meth",
294 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
296 bool ParseSectionDirectiveObjCClsRefs(StringRef, SMLoc) {
297 return ParseSectionSwitch("__OBJC", "__cls_refs",
298 MCSectionMachO::S_ATTR_NO_DEAD_STRIP |
299 MCSectionMachO::S_LITERAL_POINTERS, 4);
301 bool ParseSectionDirectiveObjCMessageRefs(StringRef, SMLoc) {
302 return ParseSectionSwitch("__OBJC", "__message_refs",
303 MCSectionMachO::S_ATTR_NO_DEAD_STRIP |
304 MCSectionMachO::S_LITERAL_POINTERS, 4);
306 bool ParseSectionDirectiveObjCSymbols(StringRef, SMLoc) {
307 return ParseSectionSwitch("__OBJC", "__symbols",
308 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
310 bool ParseSectionDirectiveObjCCategory(StringRef, SMLoc) {
311 return ParseSectionSwitch("__OBJC", "__category",
312 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
314 bool ParseSectionDirectiveObjCClassVars(StringRef, SMLoc) {
315 return ParseSectionSwitch("__OBJC", "__class_vars",
316 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
318 bool ParseSectionDirectiveObjCInstanceVars(StringRef, SMLoc) {
319 return ParseSectionSwitch("__OBJC", "__instance_vars",
320 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
322 bool ParseSectionDirectiveObjCModuleInfo(StringRef, SMLoc) {
323 return ParseSectionSwitch("__OBJC", "__module_info",
324 MCSectionMachO::S_ATTR_NO_DEAD_STRIP);
326 bool ParseSectionDirectiveObjCClassNames(StringRef, SMLoc) {
327 return ParseSectionSwitch("__TEXT", "__cstring",
328 MCSectionMachO::S_CSTRING_LITERALS);
330 bool ParseSectionDirectiveObjCMethVarTypes(StringRef, SMLoc) {
331 return ParseSectionSwitch("__TEXT", "__cstring",
332 MCSectionMachO::S_CSTRING_LITERALS);
334 bool ParseSectionDirectiveObjCMethVarNames(StringRef, SMLoc) {
335 return ParseSectionSwitch("__TEXT", "__cstring",
336 MCSectionMachO::S_CSTRING_LITERALS);
338 bool ParseSectionDirectiveObjCSelectorStrs(StringRef, SMLoc) {
339 return ParseSectionSwitch("__OBJC", "__selector_strs",
340 MCSectionMachO::S_CSTRING_LITERALS);
342 bool ParseSectionDirectiveTData(StringRef, SMLoc) {
343 return ParseSectionSwitch("__DATA", "__thread_data",
344 MCSectionMachO::S_THREAD_LOCAL_REGULAR);
346 bool ParseSectionDirectiveText(StringRef, SMLoc) {
347 return ParseSectionSwitch("__TEXT", "__text",
348 MCSectionMachO::S_ATTR_PURE_INSTRUCTIONS);
350 bool ParseSectionDirectiveTLV(StringRef, SMLoc) {
351 return ParseSectionSwitch("__DATA", "__thread_vars",
352 MCSectionMachO::S_THREAD_LOCAL_VARIABLES);
354 bool ParseSectionDirectiveIdent(StringRef, SMLoc) {
355 // Darwin silently ignores the .ident directive.
356 getParser().eatToEndOfStatement();
359 bool ParseSectionDirectiveThreadInitFunc(StringRef, SMLoc) {
360 return ParseSectionSwitch("__DATA", "__thread_init",
361 MCSectionMachO::S_THREAD_LOCAL_INIT_FUNCTION_POINTERS);
366 } // end anonymous namespace
368 bool DarwinAsmParser::ParseSectionSwitch(const char *Segment,
370 unsigned TAA, unsigned Align,
372 if (getLexer().isNot(AsmToken::EndOfStatement))
373 return TokError("unexpected token in section switching directive");
376 // FIXME: Arch specific.
377 bool isText = TAA & MCSectionMachO::S_ATTR_PURE_INSTRUCTIONS;
378 getStreamer().SwitchSection(getContext().getMachOSection(
379 Segment, Section, TAA, StubSize,
380 isText ? SectionKind::getText()
381 : SectionKind::getDataRel()));
383 // Set the implicit alignment, if any.
385 // FIXME: This isn't really what 'as' does; I think it just uses the implicit
386 // alignment on the section (e.g., if one manually inserts bytes into the
387 // section, then just issuing the section switch directive will not realign
388 // the section. However, this is arguably more reasonable behavior, and there
389 // is no good reason for someone to intentionally emit incorrectly sized
390 // values into the implicitly aligned sections.
392 getStreamer().EmitValueToAlignment(Align, 0, 1, 0);
397 /// ParseDirectiveDesc
398 /// ::= .desc identifier , expression
399 bool DarwinAsmParser::ParseDirectiveDesc(StringRef, SMLoc) {
401 if (getParser().parseIdentifier(Name))
402 return TokError("expected identifier in directive");
404 // Handle the identifier as the key symbol.
405 MCSymbol *Sym = getContext().GetOrCreateSymbol(Name);
407 if (getLexer().isNot(AsmToken::Comma))
408 return TokError("unexpected token in '.desc' directive");
412 if (getParser().parseAbsoluteExpression(DescValue))
415 if (getLexer().isNot(AsmToken::EndOfStatement))
416 return TokError("unexpected token in '.desc' directive");
420 // Set the n_desc field of this Symbol to this DescValue
421 getStreamer().EmitSymbolDesc(Sym, DescValue);
426 /// ParseDirectiveIndirectSymbol
427 /// ::= .indirect_symbol identifier
428 bool DarwinAsmParser::ParseDirectiveIndirectSymbol(StringRef, SMLoc Loc) {
429 const MCSectionMachO *Current = static_cast<const MCSectionMachO*>(
430 getStreamer().getCurrentSection().first);
431 unsigned SectionType = Current->getType();
432 if (SectionType != MCSectionMachO::S_NON_LAZY_SYMBOL_POINTERS &&
433 SectionType != MCSectionMachO::S_LAZY_SYMBOL_POINTERS &&
434 SectionType != MCSectionMachO::S_SYMBOL_STUBS)
435 return Error(Loc, "indirect symbol not in a symbol pointer or stub "
439 if (getParser().parseIdentifier(Name))
440 return TokError("expected identifier in .indirect_symbol directive");
442 MCSymbol *Sym = getContext().GetOrCreateSymbol(Name);
444 // Assembler local symbols don't make any sense here. Complain loudly.
445 if (Sym->isTemporary())
446 return TokError("non-local symbol required in directive");
448 if (!getStreamer().EmitSymbolAttribute(Sym, MCSA_IndirectSymbol))
449 return TokError("unable to emit indirect symbol attribute for: " + Name);
451 if (getLexer().isNot(AsmToken::EndOfStatement))
452 return TokError("unexpected token in '.indirect_symbol' directive");
459 /// ParseDirectiveDumpOrLoad
460 /// ::= ( .dump | .load ) "filename"
461 bool DarwinAsmParser::ParseDirectiveDumpOrLoad(StringRef Directive,
463 bool IsDump = Directive == ".dump";
464 if (getLexer().isNot(AsmToken::String))
465 return TokError("expected string in '.dump' or '.load' directive");
469 if (getLexer().isNot(AsmToken::EndOfStatement))
470 return TokError("unexpected token in '.dump' or '.load' directive");
474 // FIXME: If/when .dump and .load are implemented they will be done in the
475 // the assembly parser and not have any need for an MCStreamer API.
477 return Warning(IDLoc, "ignoring directive .dump for now");
479 return Warning(IDLoc, "ignoring directive .load for now");
482 /// ParseDirectiveLinkerOption
483 /// ::= .linker_option "string" ( , "string" )*
484 bool DarwinAsmParser::ParseDirectiveLinkerOption(StringRef IDVal, SMLoc) {
485 SmallVector<std::string, 4> Args;
487 if (getLexer().isNot(AsmToken::String))
488 return TokError("expected string in '" + Twine(IDVal) + "' directive");
491 if (getParser().parseEscapedString(Data))
494 Args.push_back(Data);
497 if (getLexer().is(AsmToken::EndOfStatement))
500 if (getLexer().isNot(AsmToken::Comma))
501 return TokError("unexpected token in '" + Twine(IDVal) + "' directive");
505 getStreamer().EmitLinkerOptions(Args);
509 /// ParseDirectiveLsym
510 /// ::= .lsym identifier , expression
511 bool DarwinAsmParser::ParseDirectiveLsym(StringRef, SMLoc) {
513 if (getParser().parseIdentifier(Name))
514 return TokError("expected identifier in directive");
516 // Handle the identifier as the key symbol.
517 MCSymbol *Sym = getContext().GetOrCreateSymbol(Name);
519 if (getLexer().isNot(AsmToken::Comma))
520 return TokError("unexpected token in '.lsym' directive");
524 if (getParser().parseExpression(Value))
527 if (getLexer().isNot(AsmToken::EndOfStatement))
528 return TokError("unexpected token in '.lsym' directive");
532 // We don't currently support this directive.
534 // FIXME: Diagnostic location!
536 return TokError("directive '.lsym' is unsupported");
539 /// ParseDirectiveSection:
540 /// ::= .section identifier (',' identifier)*
541 bool DarwinAsmParser::ParseDirectiveSection(StringRef, SMLoc) {
542 SMLoc Loc = getLexer().getLoc();
544 StringRef SectionName;
545 if (getParser().parseIdentifier(SectionName))
546 return Error(Loc, "expected identifier after '.section' directive");
548 // Verify there is a following comma.
549 if (!getLexer().is(AsmToken::Comma))
550 return TokError("unexpected token in '.section' directive");
552 std::string SectionSpec = SectionName;
555 // Add all the tokens until the end of the line, ParseSectionSpecifier will
557 StringRef EOL = getLexer().LexUntilEndOfStatement();
558 SectionSpec.append(EOL.begin(), EOL.end());
561 if (getLexer().isNot(AsmToken::EndOfStatement))
562 return TokError("unexpected token in '.section' directive");
566 StringRef Segment, Section;
570 std::string ErrorStr =
571 MCSectionMachO::ParseSectionSpecifier(SectionSpec, Segment, Section,
572 TAA, TAAParsed, StubSize);
574 if (!ErrorStr.empty())
575 return Error(Loc, ErrorStr.c_str());
577 // FIXME: Arch specific.
578 bool isText = Segment == "__TEXT"; // FIXME: Hack.
579 getStreamer().SwitchSection(getContext().getMachOSection(
580 Segment, Section, TAA, StubSize,
581 isText ? SectionKind::getText()
582 : SectionKind::getDataRel()));
586 /// ParseDirectivePushSection:
587 /// ::= .pushsection identifier (',' identifier)*
588 bool DarwinAsmParser::ParseDirectivePushSection(StringRef S, SMLoc Loc) {
589 getStreamer().PushSection();
591 if (ParseDirectiveSection(S, Loc)) {
592 getStreamer().PopSection();
599 /// ParseDirectivePopSection:
601 bool DarwinAsmParser::ParseDirectivePopSection(StringRef, SMLoc) {
602 if (!getStreamer().PopSection())
603 return TokError(".popsection without corresponding .pushsection");
607 /// ParseDirectivePrevious:
609 bool DarwinAsmParser::ParseDirectivePrevious(StringRef DirName, SMLoc) {
610 MCSectionSubPair PreviousSection = getStreamer().getPreviousSection();
611 if (PreviousSection.first == NULL)
612 return TokError(".previous without corresponding .section");
613 getStreamer().SwitchSection(PreviousSection.first, PreviousSection.second);
617 /// ParseDirectiveSecureLogUnique
618 /// ::= .secure_log_unique ... message ...
619 bool DarwinAsmParser::ParseDirectiveSecureLogUnique(StringRef, SMLoc IDLoc) {
620 StringRef LogMessage = getParser().parseStringToEndOfStatement();
621 if (getLexer().isNot(AsmToken::EndOfStatement))
622 return TokError("unexpected token in '.secure_log_unique' directive");
624 if (getContext().getSecureLogUsed() != false)
625 return Error(IDLoc, ".secure_log_unique specified multiple times");
627 // Get the secure log path.
628 const char *SecureLogFile = getContext().getSecureLogFile();
629 if (SecureLogFile == NULL)
630 return Error(IDLoc, ".secure_log_unique used but AS_SECURE_LOG_FILE "
631 "environment variable unset.");
633 // Open the secure log file if we haven't already.
634 raw_ostream *OS = getContext().getSecureLog();
637 OS = new raw_fd_ostream(SecureLogFile, Err, sys::fs::F_Append);
640 return Error(IDLoc, Twine("can't open secure log file: ") +
641 SecureLogFile + " (" + Err + ")");
643 getContext().setSecureLog(OS);
646 // Write the message.
647 int CurBuf = getSourceManager().FindBufferContainingLoc(IDLoc);
648 *OS << getSourceManager().getBufferInfo(CurBuf).Buffer->getBufferIdentifier()
649 << ":" << getSourceManager().FindLineNumber(IDLoc, CurBuf) << ":"
650 << LogMessage + "\n";
652 getContext().setSecureLogUsed(true);
657 /// ParseDirectiveSecureLogReset
658 /// ::= .secure_log_reset
659 bool DarwinAsmParser::ParseDirectiveSecureLogReset(StringRef, SMLoc IDLoc) {
660 if (getLexer().isNot(AsmToken::EndOfStatement))
661 return TokError("unexpected token in '.secure_log_reset' directive");
665 getContext().setSecureLogUsed(false);
670 /// ParseDirectiveSubsectionsViaSymbols
671 /// ::= .subsections_via_symbols
672 bool DarwinAsmParser::ParseDirectiveSubsectionsViaSymbols(StringRef, SMLoc) {
673 if (getLexer().isNot(AsmToken::EndOfStatement))
674 return TokError("unexpected token in '.subsections_via_symbols' directive");
678 getStreamer().EmitAssemblerFlag(MCAF_SubsectionsViaSymbols);
683 /// ParseDirectiveTBSS
684 /// ::= .tbss identifier, size, align
685 bool DarwinAsmParser::ParseDirectiveTBSS(StringRef, SMLoc) {
686 SMLoc IDLoc = getLexer().getLoc();
688 if (getParser().parseIdentifier(Name))
689 return TokError("expected identifier in directive");
691 // Handle the identifier as the key symbol.
692 MCSymbol *Sym = getContext().GetOrCreateSymbol(Name);
694 if (getLexer().isNot(AsmToken::Comma))
695 return TokError("unexpected token in directive");
699 SMLoc SizeLoc = getLexer().getLoc();
700 if (getParser().parseAbsoluteExpression(Size))
703 int64_t Pow2Alignment = 0;
704 SMLoc Pow2AlignmentLoc;
705 if (getLexer().is(AsmToken::Comma)) {
707 Pow2AlignmentLoc = getLexer().getLoc();
708 if (getParser().parseAbsoluteExpression(Pow2Alignment))
712 if (getLexer().isNot(AsmToken::EndOfStatement))
713 return TokError("unexpected token in '.tbss' directive");
718 return Error(SizeLoc, "invalid '.tbss' directive size, can't be less than"
721 // FIXME: Diagnose overflow.
722 if (Pow2Alignment < 0)
723 return Error(Pow2AlignmentLoc, "invalid '.tbss' alignment, can't be less"
726 if (!Sym->isUndefined())
727 return Error(IDLoc, "invalid symbol redefinition");
729 getStreamer().EmitTBSSSymbol(getContext().getMachOSection(
730 "__DATA", "__thread_bss",
731 MCSectionMachO::S_THREAD_LOCAL_ZEROFILL,
732 0, SectionKind::getThreadBSS()),
733 Sym, Size, 1 << Pow2Alignment);
738 /// ParseDirectiveZerofill
739 /// ::= .zerofill segname , sectname [, identifier , size_expression [
740 /// , align_expression ]]
741 bool DarwinAsmParser::ParseDirectiveZerofill(StringRef, SMLoc) {
743 if (getParser().parseIdentifier(Segment))
744 return TokError("expected segment name after '.zerofill' directive");
746 if (getLexer().isNot(AsmToken::Comma))
747 return TokError("unexpected token in directive");
751 if (getParser().parseIdentifier(Section))
752 return TokError("expected section name after comma in '.zerofill' "
755 // If this is the end of the line all that was wanted was to create the
756 // the section but with no symbol.
757 if (getLexer().is(AsmToken::EndOfStatement)) {
758 // Create the zerofill section but no symbol
759 getStreamer().EmitZerofill(getContext().getMachOSection(
760 Segment, Section, MCSectionMachO::S_ZEROFILL,
761 0, SectionKind::getBSS()));
765 if (getLexer().isNot(AsmToken::Comma))
766 return TokError("unexpected token in directive");
769 SMLoc IDLoc = getLexer().getLoc();
771 if (getParser().parseIdentifier(IDStr))
772 return TokError("expected identifier in directive");
774 // handle the identifier as the key symbol.
775 MCSymbol *Sym = getContext().GetOrCreateSymbol(IDStr);
777 if (getLexer().isNot(AsmToken::Comma))
778 return TokError("unexpected token in directive");
782 SMLoc SizeLoc = getLexer().getLoc();
783 if (getParser().parseAbsoluteExpression(Size))
786 int64_t Pow2Alignment = 0;
787 SMLoc Pow2AlignmentLoc;
788 if (getLexer().is(AsmToken::Comma)) {
790 Pow2AlignmentLoc = getLexer().getLoc();
791 if (getParser().parseAbsoluteExpression(Pow2Alignment))
795 if (getLexer().isNot(AsmToken::EndOfStatement))
796 return TokError("unexpected token in '.zerofill' directive");
801 return Error(SizeLoc, "invalid '.zerofill' directive size, can't be less "
804 // NOTE: The alignment in the directive is a power of 2 value, the assembler
805 // may internally end up wanting an alignment in bytes.
806 // FIXME: Diagnose overflow.
807 if (Pow2Alignment < 0)
808 return Error(Pow2AlignmentLoc, "invalid '.zerofill' directive alignment, "
809 "can't be less than zero");
811 if (!Sym->isUndefined())
812 return Error(IDLoc, "invalid symbol redefinition");
814 // Create the zerofill Symbol with Size and Pow2Alignment
816 // FIXME: Arch specific.
817 getStreamer().EmitZerofill(getContext().getMachOSection(
818 Segment, Section, MCSectionMachO::S_ZEROFILL,
819 0, SectionKind::getBSS()),
820 Sym, Size, 1 << Pow2Alignment);
825 /// ParseDirectiveDataRegion
826 /// ::= .data_region [ ( jt8 | jt16 | jt32 ) ]
827 bool DarwinAsmParser::ParseDirectiveDataRegion(StringRef, SMLoc) {
828 if (getLexer().is(AsmToken::EndOfStatement)) {
830 getStreamer().EmitDataRegion(MCDR_DataRegion);
833 StringRef RegionType;
834 SMLoc Loc = getParser().getTok().getLoc();
835 if (getParser().parseIdentifier(RegionType))
836 return TokError("expected region type after '.data_region' directive");
837 int Kind = StringSwitch<int>(RegionType)
838 .Case("jt8", MCDR_DataRegionJT8)
839 .Case("jt16", MCDR_DataRegionJT16)
840 .Case("jt32", MCDR_DataRegionJT32)
843 return Error(Loc, "unknown region type in '.data_region' directive");
846 getStreamer().EmitDataRegion((MCDataRegionType)Kind);
850 /// ParseDirectiveDataRegionEnd
851 /// ::= .end_data_region
852 bool DarwinAsmParser::ParseDirectiveDataRegionEnd(StringRef, SMLoc) {
853 if (getLexer().isNot(AsmToken::EndOfStatement))
854 return TokError("unexpected token in '.end_data_region' directive");
857 getStreamer().EmitDataRegion(MCDR_DataRegionEnd);
863 MCAsmParserExtension *createDarwinAsmParser() {
864 return new DarwinAsmParser;
867 } // end llvm namespace