YAML.cpp revision 263508 
1//===- YAML.cpp - YAMLIO utilities for object files -----------------------===//
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// This file defines utility classes for handling the YAML representation of
11// object files.
12//
13//===----------------------------------------------------------------------===//
14
15#include "llvm/Object/YAML.h"
16#include "llvm/ADT/StringExtras.h"
17#include "llvm/Support/raw_ostream.h"
18#include <cctype>
19
20using namespace llvm;
21using namespace object::yaml;
22
23void yaml::ScalarTraits<object::yaml::BinaryRef>::output(
24    const object::yaml::BinaryRef &Val, void *, llvm::raw_ostream &Out) {
25  Val.writeAsHex(Out);
26}
27
28StringRef yaml::ScalarTraits<object::yaml::BinaryRef>::input(
29    StringRef Scalar, void *, object::yaml::BinaryRef &Val) {
30  if (Scalar.size() % 2 != 0)
31    return "BinaryRef hex string must contain an even number of nybbles.";
32  // TODO: Can we improve YAMLIO to permit a more accurate diagnostic here?
33  // (e.g. a caret pointing to the offending character).
34  for (unsigned I = 0, N = Scalar.size(); I != N; ++I)
35    if (!isxdigit(Scalar[I]))
36      return "BinaryRef hex string must contain only hex digits.";
37  Val = object::yaml::BinaryRef(Scalar);
38  return StringRef();
39}
40
41void BinaryRef::writeAsBinary(raw_ostream &OS) const {
42  if (!DataIsHexString) {
43    OS.write((const char *)Data.data(), Data.size());
44    return;
45  }
46  for (unsigned I = 0, N = Data.size(); I != N; I += 2) {
47    uint8_t Byte;
48    StringRef((const char *)&Data[I],  2).getAsInteger(16, Byte);
49    OS.write(Byte);
50  }
51}
52
53void BinaryRef::writeAsHex(raw_ostream &OS) const {
54  if (binary_size() == 0) {
55    OS << "\"\"";
56    return;
57  }
58  if (DataIsHexString) {
59    OS.write((const char *)Data.data(), Data.size());
60    return;
61  }
62  for (ArrayRef<uint8_t>::iterator I = Data.begin(), E = Data.end(); I != E;
63       ++I) {
64    uint8_t Byte = *I;
65    OS << hexdigit(Byte >> 4);
66    OS << hexdigit(Byte & 0xf);
67  }
68}
69