1 //=== OutputBuffer.h - Output Buffer ----------------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file was developed by Bill Wendling and is distributed under the
6 // University of Illinois Open Source License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // Methods to output values to a data buffer.
12 //===----------------------------------------------------------------------===//
14 #ifndef LLVM_SUPPORT_OUTPUTBUFFER_H
15 #define LLVM_SUPPORT_OUTPUTBUFFER_H
24 std::vector<unsigned char> &Output;
26 /// is64Bit/isLittleEndian - This information is inferred from the target
27 /// machine directly, indicating what header values and flags to set.
28 bool is64Bit, isLittleEndian;
30 OutputBuffer(std::vector<unsigned char> &Out,
31 bool is64bit, bool le)
32 : Output(Out), is64Bit(is64bit), isLittleEndian(le) {}
34 // align - Emit padding into the file until the current output position is
35 // aligned to the specified power of two boundary.
36 void align(unsigned Boundary) {
37 assert(Boundary && (Boundary & (Boundary - 1)) == 0 &&
38 "Must align to 2^k boundary");
39 size_t Size = Output.size();
41 if (Size & (Boundary - 1)) {
42 // Add padding to get alignment to the correct place.
43 size_t Pad = Boundary - (Size & (Boundary - 1));
44 Output.resize(Size + Pad);
48 //===------------------------------------------------------------------===//
49 // Out Functions - Output the specified value to the data buffer.
51 void outbyte(unsigned char X) {
54 void outhalf(unsigned short X) {
56 Output.push_back(X & 255);
57 Output.push_back(X >> 8);
59 Output.push_back(X >> 8);
60 Output.push_back(X & 255);
63 void outword(unsigned X) {
65 Output.push_back((X >> 0) & 255);
66 Output.push_back((X >> 8) & 255);
67 Output.push_back((X >> 16) & 255);
68 Output.push_back((X >> 24) & 255);
70 Output.push_back((X >> 24) & 255);
71 Output.push_back((X >> 16) & 255);
72 Output.push_back((X >> 8) & 255);
73 Output.push_back((X >> 0) & 255);
76 void outxword(uint64_t X) {
78 Output.push_back(unsigned(X >> 0) & 255);
79 Output.push_back(unsigned(X >> 8) & 255);
80 Output.push_back(unsigned(X >> 16) & 255);
81 Output.push_back(unsigned(X >> 24) & 255);
82 Output.push_back(unsigned(X >> 32) & 255);
83 Output.push_back(unsigned(X >> 40) & 255);
84 Output.push_back(unsigned(X >> 48) & 255);
85 Output.push_back(unsigned(X >> 56) & 255);
87 Output.push_back(unsigned(X >> 56) & 255);
88 Output.push_back(unsigned(X >> 48) & 255);
89 Output.push_back(unsigned(X >> 40) & 255);
90 Output.push_back(unsigned(X >> 32) & 255);
91 Output.push_back(unsigned(X >> 24) & 255);
92 Output.push_back(unsigned(X >> 16) & 255);
93 Output.push_back(unsigned(X >> 8) & 255);
94 Output.push_back(unsigned(X >> 0) & 255);
97 void outaddr32(unsigned X) {
100 void outaddr64(uint64_t X) {
103 void outaddr(uint64_t X) {
105 outword((unsigned)X);
109 void outstring(const std::string &S, unsigned Length) {
110 unsigned len_to_copy = S.length() < Length ? S.length() : Length;
111 unsigned len_to_fill = S.length() < Length ? Length - S.length() : 0;
113 for (unsigned i = 0; i < len_to_copy; ++i)
116 for (unsigned i = 0; i < len_to_fill; ++i)
120 //===------------------------------------------------------------------===//
121 // Fix Functions - Replace an existing entry at an offset.
123 void fixhalf(unsigned short X, unsigned Offset) {
124 unsigned char *P = &Output[Offset];
125 P[0] = (X >> (isLittleEndian ? 0 : 8)) & 255;
126 P[1] = (X >> (isLittleEndian ? 8 : 0)) & 255;
128 void fixword(unsigned X, unsigned Offset) {
129 unsigned char *P = &Output[Offset];
130 P[0] = (X >> (isLittleEndian ? 0 : 24)) & 255;
131 P[1] = (X >> (isLittleEndian ? 8 : 16)) & 255;
132 P[2] = (X >> (isLittleEndian ? 16 : 8)) & 255;
133 P[3] = (X >> (isLittleEndian ? 24 : 0)) & 255;
135 void fixaddr(uint64_t X, unsigned Offset) {
137 fixword((unsigned)X, Offset);
139 assert(0 && "Emission of 64-bit data not implemented yet!");
142 unsigned char &operator[](unsigned Index) {
143 return Output[Index];
145 const unsigned char &operator[](unsigned Index) const {
146 return Output[Index];
150 } // end llvm namespace
152 #endif // LLVM_SUPPORT_OUTPUTBUFFER_H