YAMLTraits.cpp revision 296417
1//===- lib/Support/YAMLTraits.cpp -----------------------------------------===//
2//
3//                             The LLVM Linker
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 "llvm/Support/YAMLTraits.h"
11#include "llvm/ADT/SmallString.h"
12#include "llvm/ADT/Twine.h"
13#include "llvm/Support/Casting.h"
14#include "llvm/Support/Errc.h"
15#include "llvm/Support/ErrorHandling.h"
16#include "llvm/Support/Format.h"
17#include "llvm/Support/LineIterator.h"
18#include "llvm/Support/YAMLParser.h"
19#include "llvm/Support/raw_ostream.h"
20#include <cctype>
21#include <cstring>
22using namespace llvm;
23using namespace yaml;
24
25//===----------------------------------------------------------------------===//
26//  IO
27//===----------------------------------------------------------------------===//
28
29IO::IO(void *Context) : Ctxt(Context) {
30}
31
32IO::~IO() {
33}
34
35void *IO::getContext() {
36  return Ctxt;
37}
38
39void IO::setContext(void *Context) {
40  Ctxt = Context;
41}
42
43//===----------------------------------------------------------------------===//
44//  Input
45//===----------------------------------------------------------------------===//
46
47Input::Input(StringRef InputContent,
48             void *Ctxt,
49             SourceMgr::DiagHandlerTy DiagHandler,
50             void *DiagHandlerCtxt)
51  : IO(Ctxt),
52    Strm(new Stream(InputContent, SrcMgr)),
53    CurrentNode(nullptr) {
54  if (DiagHandler)
55    SrcMgr.setDiagHandler(DiagHandler, DiagHandlerCtxt);
56  DocIterator = Strm->begin();
57}
58
59Input::~Input() {
60}
61
62std::error_code Input::error() { return EC; }
63
64// Pin the vtables to this file.
65void Input::HNode::anchor() {}
66void Input::EmptyHNode::anchor() {}
67void Input::ScalarHNode::anchor() {}
68void Input::MapHNode::anchor() {}
69void Input::SequenceHNode::anchor() {}
70
71bool Input::outputting() {
72  return false;
73}
74
75bool Input::setCurrentDocument() {
76  if (DocIterator != Strm->end()) {
77    Node *N = DocIterator->getRoot();
78    if (!N) {
79      assert(Strm->failed() && "Root is NULL iff parsing failed");
80      EC = make_error_code(errc::invalid_argument);
81      return false;
82    }
83
84    if (isa<NullNode>(N)) {
85      // Empty files are allowed and ignored
86      ++DocIterator;
87      return setCurrentDocument();
88    }
89    TopNode = this->createHNodes(N);
90    CurrentNode = TopNode.get();
91    return true;
92  }
93  return false;
94}
95
96bool Input::nextDocument() {
97  return ++DocIterator != Strm->end();
98}
99
100const Node *Input::getCurrentNode() const {
101  return CurrentNode ? CurrentNode->_node : nullptr;
102}
103
104bool Input::mapTag(StringRef Tag, bool Default) {
105  std::string foundTag = CurrentNode->_node->getVerbatimTag();
106  if (foundTag.empty()) {
107    // If no tag found and 'Tag' is the default, say it was found.
108    return Default;
109  }
110  // Return true iff found tag matches supplied tag.
111  return Tag.equals(foundTag);
112}
113
114void Input::beginMapping() {
115  if (EC)
116    return;
117  // CurrentNode can be null if the document is empty.
118  MapHNode *MN = dyn_cast_or_null<MapHNode>(CurrentNode);
119  if (MN) {
120    MN->ValidKeys.clear();
121  }
122}
123
124bool Input::preflightKey(const char *Key, bool Required, bool, bool &UseDefault,
125                         void *&SaveInfo) {
126  UseDefault = false;
127  if (EC)
128    return false;
129
130  // CurrentNode is null for empty documents, which is an error in case required
131  // nodes are present.
132  if (!CurrentNode) {
133    if (Required)
134      EC = make_error_code(errc::invalid_argument);
135    return false;
136  }
137
138  MapHNode *MN = dyn_cast<MapHNode>(CurrentNode);
139  if (!MN) {
140    setError(CurrentNode, "not a mapping");
141    return false;
142  }
143  MN->ValidKeys.push_back(Key);
144  HNode *Value = MN->Mapping[Key].get();
145  if (!Value) {
146    if (Required)
147      setError(CurrentNode, Twine("missing required key '") + Key + "'");
148    else
149      UseDefault = true;
150    return false;
151  }
152  SaveInfo = CurrentNode;
153  CurrentNode = Value;
154  return true;
155}
156
157void Input::postflightKey(void *saveInfo) {
158  CurrentNode = reinterpret_cast<HNode *>(saveInfo);
159}
160
161void Input::endMapping() {
162  if (EC)
163    return;
164  // CurrentNode can be null if the document is empty.
165  MapHNode *MN = dyn_cast_or_null<MapHNode>(CurrentNode);
166  if (!MN)
167    return;
168  for (const auto &NN : MN->Mapping) {
169    if (!MN->isValidKey(NN.first())) {
170      setError(NN.second.get(), Twine("unknown key '") + NN.first() + "'");
171      break;
172    }
173  }
174}
175
176void Input::beginFlowMapping() { beginMapping(); }
177
178void Input::endFlowMapping() { endMapping(); }
179
180unsigned Input::beginSequence() {
181  if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode))
182    return SQ->Entries.size();
183  if (isa<EmptyHNode>(CurrentNode))
184    return 0;
185  // Treat case where there's a scalar "null" value as an empty sequence.
186  if (ScalarHNode *SN = dyn_cast<ScalarHNode>(CurrentNode)) {
187    if (isNull(SN->value()))
188      return 0;
189  }
190  // Any other type of HNode is an error.
191  setError(CurrentNode, "not a sequence");
192  return 0;
193}
194
195void Input::endSequence() {
196}
197
198bool Input::preflightElement(unsigned Index, void *&SaveInfo) {
199  if (EC)
200    return false;
201  if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
202    SaveInfo = CurrentNode;
203    CurrentNode = SQ->Entries[Index].get();
204    return true;
205  }
206  return false;
207}
208
209void Input::postflightElement(void *SaveInfo) {
210  CurrentNode = reinterpret_cast<HNode *>(SaveInfo);
211}
212
213unsigned Input::beginFlowSequence() { return beginSequence(); }
214
215bool Input::preflightFlowElement(unsigned index, void *&SaveInfo) {
216  if (EC)
217    return false;
218  if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
219    SaveInfo = CurrentNode;
220    CurrentNode = SQ->Entries[index].get();
221    return true;
222  }
223  return false;
224}
225
226void Input::postflightFlowElement(void *SaveInfo) {
227  CurrentNode = reinterpret_cast<HNode *>(SaveInfo);
228}
229
230void Input::endFlowSequence() {
231}
232
233void Input::beginEnumScalar() {
234  ScalarMatchFound = false;
235}
236
237bool Input::matchEnumScalar(const char *Str, bool) {
238  if (ScalarMatchFound)
239    return false;
240  if (ScalarHNode *SN = dyn_cast<ScalarHNode>(CurrentNode)) {
241    if (SN->value().equals(Str)) {
242      ScalarMatchFound = true;
243      return true;
244    }
245  }
246  return false;
247}
248
249bool Input::matchEnumFallback() {
250  if (ScalarMatchFound)
251    return false;
252  ScalarMatchFound = true;
253  return true;
254}
255
256void Input::endEnumScalar() {
257  if (!ScalarMatchFound) {
258    setError(CurrentNode, "unknown enumerated scalar");
259  }
260}
261
262bool Input::beginBitSetScalar(bool &DoClear) {
263  BitValuesUsed.clear();
264  if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
265    BitValuesUsed.insert(BitValuesUsed.begin(), SQ->Entries.size(), false);
266  } else {
267    setError(CurrentNode, "expected sequence of bit values");
268  }
269  DoClear = true;
270  return true;
271}
272
273bool Input::bitSetMatch(const char *Str, bool) {
274  if (EC)
275    return false;
276  if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
277    unsigned Index = 0;
278    for (auto &N : SQ->Entries) {
279      if (ScalarHNode *SN = dyn_cast<ScalarHNode>(N.get())) {
280        if (SN->value().equals(Str)) {
281          BitValuesUsed[Index] = true;
282          return true;
283        }
284      } else {
285        setError(CurrentNode, "unexpected scalar in sequence of bit values");
286      }
287      ++Index;
288    }
289  } else {
290    setError(CurrentNode, "expected sequence of bit values");
291  }
292  return false;
293}
294
295void Input::endBitSetScalar() {
296  if (EC)
297    return;
298  if (SequenceHNode *SQ = dyn_cast<SequenceHNode>(CurrentNode)) {
299    assert(BitValuesUsed.size() == SQ->Entries.size());
300    for (unsigned i = 0; i < SQ->Entries.size(); ++i) {
301      if (!BitValuesUsed[i]) {
302        setError(SQ->Entries[i].get(), "unknown bit value");
303        return;
304      }
305    }
306  }
307}
308
309void Input::scalarString(StringRef &S, bool) {
310  if (ScalarHNode *SN = dyn_cast<ScalarHNode>(CurrentNode)) {
311    S = SN->value();
312  } else {
313    setError(CurrentNode, "unexpected scalar");
314  }
315}
316
317void Input::blockScalarString(StringRef &S) { scalarString(S, false); }
318
319void Input::setError(HNode *hnode, const Twine &message) {
320  assert(hnode && "HNode must not be NULL");
321  this->setError(hnode->_node, message);
322}
323
324void Input::setError(Node *node, const Twine &message) {
325  Strm->printError(node, message);
326  EC = make_error_code(errc::invalid_argument);
327}
328
329std::unique_ptr<Input::HNode> Input::createHNodes(Node *N) {
330  SmallString<128> StringStorage;
331  if (ScalarNode *SN = dyn_cast<ScalarNode>(N)) {
332    StringRef KeyStr = SN->getValue(StringStorage);
333    if (!StringStorage.empty()) {
334      // Copy string to permanent storage
335      KeyStr = StringStorage.str().copy(StringAllocator);
336    }
337    return llvm::make_unique<ScalarHNode>(N, KeyStr);
338  } else if (BlockScalarNode *BSN = dyn_cast<BlockScalarNode>(N)) {
339    StringRef ValueCopy = BSN->getValue().copy(StringAllocator);
340    return llvm::make_unique<ScalarHNode>(N, ValueCopy);
341  } else if (SequenceNode *SQ = dyn_cast<SequenceNode>(N)) {
342    auto SQHNode = llvm::make_unique<SequenceHNode>(N);
343    for (Node &SN : *SQ) {
344      auto Entry = this->createHNodes(&SN);
345      if (EC)
346        break;
347      SQHNode->Entries.push_back(std::move(Entry));
348    }
349    return std::move(SQHNode);
350  } else if (MappingNode *Map = dyn_cast<MappingNode>(N)) {
351    auto mapHNode = llvm::make_unique<MapHNode>(N);
352    for (KeyValueNode &KVN : *Map) {
353      Node *KeyNode = KVN.getKey();
354      ScalarNode *KeyScalar = dyn_cast<ScalarNode>(KeyNode);
355      if (!KeyScalar) {
356        setError(KeyNode, "Map key must be a scalar");
357        break;
358      }
359      StringStorage.clear();
360      StringRef KeyStr = KeyScalar->getValue(StringStorage);
361      if (!StringStorage.empty()) {
362        // Copy string to permanent storage
363        KeyStr = StringStorage.str().copy(StringAllocator);
364      }
365      auto ValueHNode = this->createHNodes(KVN.getValue());
366      if (EC)
367        break;
368      mapHNode->Mapping[KeyStr] = std::move(ValueHNode);
369    }
370    return std::move(mapHNode);
371  } else if (isa<NullNode>(N)) {
372    return llvm::make_unique<EmptyHNode>(N);
373  } else {
374    setError(N, "unknown node kind");
375    return nullptr;
376  }
377}
378
379bool Input::MapHNode::isValidKey(StringRef Key) {
380  for (const char *K : ValidKeys) {
381    if (Key.equals(K))
382      return true;
383  }
384  return false;
385}
386
387void Input::setError(const Twine &Message) {
388  this->setError(CurrentNode, Message);
389}
390
391bool Input::canElideEmptySequence() {
392  return false;
393}
394
395//===----------------------------------------------------------------------===//
396//  Output
397//===----------------------------------------------------------------------===//
398
399Output::Output(raw_ostream &yout, void *context, int WrapColumn)
400    : IO(context),
401      Out(yout),
402      WrapColumn(WrapColumn),
403      Column(0),
404      ColumnAtFlowStart(0),
405      ColumnAtMapFlowStart(0),
406      NeedBitValueComma(false),
407      NeedFlowSequenceComma(false),
408      EnumerationMatchFound(false),
409      NeedsNewLine(false) {
410}
411
412Output::~Output() {
413}
414
415bool Output::outputting() {
416  return true;
417}
418
419void Output::beginMapping() {
420  StateStack.push_back(inMapFirstKey);
421  NeedsNewLine = true;
422}
423
424bool Output::mapTag(StringRef Tag, bool Use) {
425  if (Use) {
426    this->output(" ");
427    this->output(Tag);
428  }
429  return Use;
430}
431
432void Output::endMapping() {
433  StateStack.pop_back();
434}
435
436bool Output::preflightKey(const char *Key, bool Required, bool SameAsDefault,
437                          bool &UseDefault, void *&) {
438  UseDefault = false;
439  if (Required || !SameAsDefault) {
440    auto State = StateStack.back();
441    if (State == inFlowMapFirstKey || State == inFlowMapOtherKey) {
442      flowKey(Key);
443    } else {
444      this->newLineCheck();
445      this->paddedKey(Key);
446    }
447    return true;
448  }
449  return false;
450}
451
452void Output::postflightKey(void *) {
453  if (StateStack.back() == inMapFirstKey) {
454    StateStack.pop_back();
455    StateStack.push_back(inMapOtherKey);
456  } else if (StateStack.back() == inFlowMapFirstKey) {
457    StateStack.pop_back();
458    StateStack.push_back(inFlowMapOtherKey);
459  }
460}
461
462void Output::beginFlowMapping() {
463  StateStack.push_back(inFlowMapFirstKey);
464  this->newLineCheck();
465  ColumnAtMapFlowStart = Column;
466  output("{ ");
467}
468
469void Output::endFlowMapping() {
470  StateStack.pop_back();
471  this->outputUpToEndOfLine(" }");
472}
473
474void Output::beginDocuments() {
475  this->outputUpToEndOfLine("---");
476}
477
478bool Output::preflightDocument(unsigned index) {
479  if (index > 0)
480    this->outputUpToEndOfLine("\n---");
481  return true;
482}
483
484void Output::postflightDocument() {
485}
486
487void Output::endDocuments() {
488  output("\n...\n");
489}
490
491unsigned Output::beginSequence() {
492  StateStack.push_back(inSeq);
493  NeedsNewLine = true;
494  return 0;
495}
496
497void Output::endSequence() {
498  StateStack.pop_back();
499}
500
501bool Output::preflightElement(unsigned, void *&) {
502  return true;
503}
504
505void Output::postflightElement(void *) {
506}
507
508unsigned Output::beginFlowSequence() {
509  StateStack.push_back(inFlowSeq);
510  this->newLineCheck();
511  ColumnAtFlowStart = Column;
512  output("[ ");
513  NeedFlowSequenceComma = false;
514  return 0;
515}
516
517void Output::endFlowSequence() {
518  StateStack.pop_back();
519  this->outputUpToEndOfLine(" ]");
520}
521
522bool Output::preflightFlowElement(unsigned, void *&) {
523  if (NeedFlowSequenceComma)
524    output(", ");
525  if (WrapColumn && Column > WrapColumn) {
526    output("\n");
527    for (int i = 0; i < ColumnAtFlowStart; ++i)
528      output(" ");
529    Column = ColumnAtFlowStart;
530    output("  ");
531  }
532  return true;
533}
534
535void Output::postflightFlowElement(void *) {
536  NeedFlowSequenceComma = true;
537}
538
539void Output::beginEnumScalar() {
540  EnumerationMatchFound = false;
541}
542
543bool Output::matchEnumScalar(const char *Str, bool Match) {
544  if (Match && !EnumerationMatchFound) {
545    this->newLineCheck();
546    this->outputUpToEndOfLine(Str);
547    EnumerationMatchFound = true;
548  }
549  return false;
550}
551
552bool Output::matchEnumFallback() {
553  if (EnumerationMatchFound)
554    return false;
555  EnumerationMatchFound = true;
556  return true;
557}
558
559void Output::endEnumScalar() {
560  if (!EnumerationMatchFound)
561    llvm_unreachable("bad runtime enum value");
562}
563
564bool Output::beginBitSetScalar(bool &DoClear) {
565  this->newLineCheck();
566  output("[ ");
567  NeedBitValueComma = false;
568  DoClear = false;
569  return true;
570}
571
572bool Output::bitSetMatch(const char *Str, bool Matches) {
573  if (Matches) {
574    if (NeedBitValueComma)
575      output(", ");
576    this->output(Str);
577    NeedBitValueComma = true;
578  }
579  return false;
580}
581
582void Output::endBitSetScalar() {
583  this->outputUpToEndOfLine(" ]");
584}
585
586void Output::scalarString(StringRef &S, bool MustQuote) {
587  this->newLineCheck();
588  if (S.empty()) {
589    // Print '' for the empty string because leaving the field empty is not
590    // allowed.
591    this->outputUpToEndOfLine("''");
592    return;
593  }
594  if (!MustQuote) {
595    // Only quote if we must.
596    this->outputUpToEndOfLine(S);
597    return;
598  }
599  unsigned i = 0;
600  unsigned j = 0;
601  unsigned End = S.size();
602  output("'"); // Starting single quote.
603  const char *Base = S.data();
604  while (j < End) {
605    // Escape a single quote by doubling it.
606    if (S[j] == '\'') {
607      output(StringRef(&Base[i], j - i + 1));
608      output("'");
609      i = j + 1;
610    }
611    ++j;
612  }
613  output(StringRef(&Base[i], j - i));
614  this->outputUpToEndOfLine("'"); // Ending single quote.
615}
616
617void Output::blockScalarString(StringRef &S) {
618  if (!StateStack.empty())
619    newLineCheck();
620  output(" |");
621  outputNewLine();
622
623  unsigned Indent = StateStack.empty() ? 1 : StateStack.size();
624
625  auto Buffer = MemoryBuffer::getMemBuffer(S, "", false);
626  for (line_iterator Lines(*Buffer, false); !Lines.is_at_end(); ++Lines) {
627    for (unsigned I = 0; I < Indent; ++I) {
628      output("  ");
629    }
630    output(*Lines);
631    outputNewLine();
632  }
633}
634
635void Output::setError(const Twine &message) {
636}
637
638bool Output::canElideEmptySequence() {
639  // Normally, with an optional key/value where the value is an empty sequence,
640  // the whole key/value can be not written.  But, that produces wrong yaml
641  // if the key/value is the only thing in the map and the map is used in
642  // a sequence.  This detects if the this sequence is the first key/value
643  // in map that itself is embedded in a sequnce.
644  if (StateStack.size() < 2)
645    return true;
646  if (StateStack.back() != inMapFirstKey)
647    return true;
648  return (StateStack[StateStack.size()-2] != inSeq);
649}
650
651void Output::output(StringRef s) {
652  Column += s.size();
653  Out << s;
654}
655
656void Output::outputUpToEndOfLine(StringRef s) {
657  this->output(s);
658  if (StateStack.empty() || (StateStack.back() != inFlowSeq &&
659                             StateStack.back() != inFlowMapFirstKey &&
660                             StateStack.back() != inFlowMapOtherKey))
661    NeedsNewLine = true;
662}
663
664void Output::outputNewLine() {
665  Out << "\n";
666  Column = 0;
667}
668
669// if seq at top, indent as if map, then add "- "
670// if seq in middle, use "- " if firstKey, else use "  "
671//
672
673void Output::newLineCheck() {
674  if (!NeedsNewLine)
675    return;
676  NeedsNewLine = false;
677
678  this->outputNewLine();
679
680  assert(StateStack.size() > 0);
681  unsigned Indent = StateStack.size() - 1;
682  bool OutputDash = false;
683
684  if (StateStack.back() == inSeq) {
685    OutputDash = true;
686  } else if ((StateStack.size() > 1) && ((StateStack.back() == inMapFirstKey) ||
687             (StateStack.back() == inFlowSeq) ||
688             (StateStack.back() == inFlowMapFirstKey)) &&
689             (StateStack[StateStack.size() - 2] == inSeq)) {
690    --Indent;
691    OutputDash = true;
692  }
693
694  for (unsigned i = 0; i < Indent; ++i) {
695    output("  ");
696  }
697  if (OutputDash) {
698    output("- ");
699  }
700
701}
702
703void Output::paddedKey(StringRef key) {
704  output(key);
705  output(":");
706  const char *spaces = "                ";
707  if (key.size() < strlen(spaces))
708    output(&spaces[key.size()]);
709  else
710    output(" ");
711}
712
713void Output::flowKey(StringRef Key) {
714  if (StateStack.back() == inFlowMapOtherKey)
715    output(", ");
716  if (WrapColumn && Column > WrapColumn) {
717    output("\n");
718    for (int I = 0; I < ColumnAtMapFlowStart; ++I)
719      output(" ");
720    Column = ColumnAtMapFlowStart;
721    output("  ");
722  }
723  output(Key);
724  output(": ");
725}
726
727//===----------------------------------------------------------------------===//
728//  traits for built-in types
729//===----------------------------------------------------------------------===//
730
731void ScalarTraits<bool>::output(const bool &Val, void *, raw_ostream &Out) {
732  Out << (Val ? "true" : "false");
733}
734
735StringRef ScalarTraits<bool>::input(StringRef Scalar, void *, bool &Val) {
736  if (Scalar.equals("true")) {
737    Val = true;
738    return StringRef();
739  } else if (Scalar.equals("false")) {
740    Val = false;
741    return StringRef();
742  }
743  return "invalid boolean";
744}
745
746void ScalarTraits<StringRef>::output(const StringRef &Val, void *,
747                                     raw_ostream &Out) {
748  Out << Val;
749}
750
751StringRef ScalarTraits<StringRef>::input(StringRef Scalar, void *,
752                                         StringRef &Val) {
753  Val = Scalar;
754  return StringRef();
755}
756
757void ScalarTraits<std::string>::output(const std::string &Val, void *,
758                                     raw_ostream &Out) {
759  Out << Val;
760}
761
762StringRef ScalarTraits<std::string>::input(StringRef Scalar, void *,
763                                         std::string &Val) {
764  Val = Scalar.str();
765  return StringRef();
766}
767
768void ScalarTraits<uint8_t>::output(const uint8_t &Val, void *,
769                                   raw_ostream &Out) {
770  // use temp uin32_t because ostream thinks uint8_t is a character
771  uint32_t Num = Val;
772  Out << Num;
773}
774
775StringRef ScalarTraits<uint8_t>::input(StringRef Scalar, void *, uint8_t &Val) {
776  unsigned long long n;
777  if (getAsUnsignedInteger(Scalar, 0, n))
778    return "invalid number";
779  if (n > 0xFF)
780    return "out of range number";
781  Val = n;
782  return StringRef();
783}
784
785void ScalarTraits<uint16_t>::output(const uint16_t &Val, void *,
786                                    raw_ostream &Out) {
787  Out << Val;
788}
789
790StringRef ScalarTraits<uint16_t>::input(StringRef Scalar, void *,
791                                        uint16_t &Val) {
792  unsigned long long n;
793  if (getAsUnsignedInteger(Scalar, 0, n))
794    return "invalid number";
795  if (n > 0xFFFF)
796    return "out of range number";
797  Val = n;
798  return StringRef();
799}
800
801void ScalarTraits<uint32_t>::output(const uint32_t &Val, void *,
802                                    raw_ostream &Out) {
803  Out << Val;
804}
805
806StringRef ScalarTraits<uint32_t>::input(StringRef Scalar, void *,
807                                        uint32_t &Val) {
808  unsigned long long n;
809  if (getAsUnsignedInteger(Scalar, 0, n))
810    return "invalid number";
811  if (n > 0xFFFFFFFFUL)
812    return "out of range number";
813  Val = n;
814  return StringRef();
815}
816
817void ScalarTraits<uint64_t>::output(const uint64_t &Val, void *,
818                                    raw_ostream &Out) {
819  Out << Val;
820}
821
822StringRef ScalarTraits<uint64_t>::input(StringRef Scalar, void *,
823                                        uint64_t &Val) {
824  unsigned long long N;
825  if (getAsUnsignedInteger(Scalar, 0, N))
826    return "invalid number";
827  Val = N;
828  return StringRef();
829}
830
831void ScalarTraits<int8_t>::output(const int8_t &Val, void *, raw_ostream &Out) {
832  // use temp in32_t because ostream thinks int8_t is a character
833  int32_t Num = Val;
834  Out << Num;
835}
836
837StringRef ScalarTraits<int8_t>::input(StringRef Scalar, void *, int8_t &Val) {
838  long long N;
839  if (getAsSignedInteger(Scalar, 0, N))
840    return "invalid number";
841  if ((N > 127) || (N < -128))
842    return "out of range number";
843  Val = N;
844  return StringRef();
845}
846
847void ScalarTraits<int16_t>::output(const int16_t &Val, void *,
848                                   raw_ostream &Out) {
849  Out << Val;
850}
851
852StringRef ScalarTraits<int16_t>::input(StringRef Scalar, void *, int16_t &Val) {
853  long long N;
854  if (getAsSignedInteger(Scalar, 0, N))
855    return "invalid number";
856  if ((N > INT16_MAX) || (N < INT16_MIN))
857    return "out of range number";
858  Val = N;
859  return StringRef();
860}
861
862void ScalarTraits<int32_t>::output(const int32_t &Val, void *,
863                                   raw_ostream &Out) {
864  Out << Val;
865}
866
867StringRef ScalarTraits<int32_t>::input(StringRef Scalar, void *, int32_t &Val) {
868  long long N;
869  if (getAsSignedInteger(Scalar, 0, N))
870    return "invalid number";
871  if ((N > INT32_MAX) || (N < INT32_MIN))
872    return "out of range number";
873  Val = N;
874  return StringRef();
875}
876
877void ScalarTraits<int64_t>::output(const int64_t &Val, void *,
878                                   raw_ostream &Out) {
879  Out << Val;
880}
881
882StringRef ScalarTraits<int64_t>::input(StringRef Scalar, void *, int64_t &Val) {
883  long long N;
884  if (getAsSignedInteger(Scalar, 0, N))
885    return "invalid number";
886  Val = N;
887  return StringRef();
888}
889
890void ScalarTraits<double>::output(const double &Val, void *, raw_ostream &Out) {
891  Out << format("%g", Val);
892}
893
894StringRef ScalarTraits<double>::input(StringRef Scalar, void *, double &Val) {
895  SmallString<32> buff(Scalar.begin(), Scalar.end());
896  char *end;
897  Val = strtod(buff.c_str(), &end);
898  if (*end != '\0')
899    return "invalid floating point number";
900  return StringRef();
901}
902
903void ScalarTraits<float>::output(const float &Val, void *, raw_ostream &Out) {
904  Out << format("%g", Val);
905}
906
907StringRef ScalarTraits<float>::input(StringRef Scalar, void *, float &Val) {
908  SmallString<32> buff(Scalar.begin(), Scalar.end());
909  char *end;
910  Val = strtod(buff.c_str(), &end);
911  if (*end != '\0')
912    return "invalid floating point number";
913  return StringRef();
914}
915
916void ScalarTraits<Hex8>::output(const Hex8 &Val, void *, raw_ostream &Out) {
917  uint8_t Num = Val;
918  Out << format("0x%02X", Num);
919}
920
921StringRef ScalarTraits<Hex8>::input(StringRef Scalar, void *, Hex8 &Val) {
922  unsigned long long n;
923  if (getAsUnsignedInteger(Scalar, 0, n))
924    return "invalid hex8 number";
925  if (n > 0xFF)
926    return "out of range hex8 number";
927  Val = n;
928  return StringRef();
929}
930
931void ScalarTraits<Hex16>::output(const Hex16 &Val, void *, raw_ostream &Out) {
932  uint16_t Num = Val;
933  Out << format("0x%04X", Num);
934}
935
936StringRef ScalarTraits<Hex16>::input(StringRef Scalar, void *, Hex16 &Val) {
937  unsigned long long n;
938  if (getAsUnsignedInteger(Scalar, 0, n))
939    return "invalid hex16 number";
940  if (n > 0xFFFF)
941    return "out of range hex16 number";
942  Val = n;
943  return StringRef();
944}
945
946void ScalarTraits<Hex32>::output(const Hex32 &Val, void *, raw_ostream &Out) {
947  uint32_t Num = Val;
948  Out << format("0x%08X", Num);
949}
950
951StringRef ScalarTraits<Hex32>::input(StringRef Scalar, void *, Hex32 &Val) {
952  unsigned long long n;
953  if (getAsUnsignedInteger(Scalar, 0, n))
954    return "invalid hex32 number";
955  if (n > 0xFFFFFFFFUL)
956    return "out of range hex32 number";
957  Val = n;
958  return StringRef();
959}
960
961void ScalarTraits<Hex64>::output(const Hex64 &Val, void *, raw_ostream &Out) {
962  uint64_t Num = Val;
963  Out << format("0x%016llX", Num);
964}
965
966StringRef ScalarTraits<Hex64>::input(StringRef Scalar, void *, Hex64 &Val) {
967  unsigned long long Num;
968  if (getAsUnsignedInteger(Scalar, 0, Num))
969    return "invalid hex64 number";
970  Val = Num;
971  return StringRef();
972}
973