Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 //===- NaClBitcodeHeader.cpp ----------------------------------------------===// | 1 //===- NaClBitcodeHeader.cpp ----------------------------------------------===// |
| 2 // PNaCl bitcode header reader. | 2 // PNaCl bitcode header reader. |
| 3 // | 3 // |
| 4 // The LLVM Compiler Infrastructure | 4 // The LLVM Compiler Infrastructure |
| 5 // | 5 // |
| 6 // This file is distributed under the University of Illinois Open Source | 6 // This file is distributed under the University of Illinois Open Source |
| 7 // License. See LICENSE.TXT for details. | 7 // License. See LICENSE.TXT for details. |
| 8 // | 8 // |
| 9 //===----------------------------------------------------------------------===// | 9 //===----------------------------------------------------------------------===// |
| 10 | 10 |
| 11 #include "llvm/Bitcode/NaCl/NaClBitcodeHeader.h" | 11 #include "llvm/Bitcode/NaCl/NaClBitcodeHeader.h" |
| 12 #include "llvm/Bitcode/NaCl/NaClReaderWriter.h" | 12 #include "llvm/Bitcode/NaCl/NaClReaderWriter.h" |
| 13 #include "llvm/Bitcode/ReaderWriter.h" | |
| 13 #include "llvm/Support/ErrorHandling.h" | 14 #include "llvm/Support/ErrorHandling.h" |
| 14 #include "llvm/Support/Format.h" | 15 #include "llvm/Support/Format.h" |
| 15 #include "llvm/Support/raw_ostream.h" | 16 #include "llvm/Support/raw_ostream.h" |
| 16 #include "llvm/Support/StreamableMemoryObject.h" | 17 #include "llvm/Support/StreamableMemoryObject.h" |
| 17 | 18 |
| 18 #include <limits> | 19 #include <limits> |
| 19 #include <cstring> | 20 #include <cstring> |
| 20 #include <iomanip> | 21 #include <iomanip> |
| 21 | 22 |
| 22 using namespace llvm; | 23 using namespace llvm; |
| (...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 111 case kBufferType: | 112 case kBufferType: |
| 112 ss << "["; | 113 ss << "["; |
| 113 for (size_t i = 0; i < Len; ++i) { | 114 for (size_t i = 0; i < Len; ++i) { |
| 114 if (i) | 115 if (i) |
| 115 ss << " "; | 116 ss << " "; |
| 116 ss << format("%02x", Data[i]); | 117 ss << format("%02x", Data[i]); |
| 117 } | 118 } |
| 118 ss << "]"; | 119 ss << "]"; |
| 119 break; | 120 break; |
| 120 default: | 121 default: |
| 121 report_fatal_error("PNaCL bitcode file contains unknown field type"); | 122 report_fatal_error("PNaCl bitcode file contains unknown field type"); |
| 122 } | 123 } |
| 123 return ss.str(); | 124 return ss.str(); |
| 124 } | 125 } |
| 125 | 126 |
| 126 NaClBitcodeHeader::NaClBitcodeHeader() | 127 NaClBitcodeHeader::NaClBitcodeHeader() |
| 127 : HeaderSize(0), UnsupportedMessage(), IsSupportedFlag(false), | 128 : HeaderSize(0), UnsupportedMessage(), IsSupportedFlag(false), |
| 128 IsReadableFlag(false), PNaClVersion(0) {} | 129 IsReadableFlag(false), PNaClVersion(0) {} |
| 129 | 130 |
| 130 NaClBitcodeHeader::~NaClBitcodeHeader() { | 131 NaClBitcodeHeader::~NaClBitcodeHeader() { |
| 131 for (std::vector<NaClBitcodeHeaderField *>::const_iterator | 132 for (std::vector<NaClBitcodeHeaderField *>::const_iterator |
| 132 Iter = Fields.begin(), | 133 Iter = Fields.begin(), |
| 133 IterEnd = Fields.end(); | 134 IterEnd = Fields.end(); |
| 134 Iter != IterEnd; ++Iter) { | 135 Iter != IterEnd; ++Iter) { |
| 135 delete *Iter; | 136 delete *Iter; |
| 136 } | 137 } |
| 137 } | 138 } |
| 138 | 139 |
| 139 bool NaClBitcodeHeader::ReadPrefix(const unsigned char *BufPtr, | 140 bool NaClBitcodeHeader::ReadPrefix(const unsigned char *BufPtr, |
| 140 const unsigned char *BufEnd, | 141 const unsigned char *BufEnd, |
| 141 unsigned &NumFields, unsigned &NumBytes) { | 142 unsigned &NumFields, unsigned &NumBytes) { |
| 142 // Must contain PEXE. | 143 // Must contain PEXE. |
| 143 if (!isNaClBitcode(BufPtr, BufEnd)) | 144 if (!isNaClBitcode(BufPtr, BufEnd)) { |
| 145 UnsupportedMessage = "Invalid PNaCl bitcode header"; | |
| 146 if (isBitcode(BufPtr, BufEnd)) { | |
| 147 UnsupportedMessage += " (to run in Chrome, bitcode files must be " | |
| 148 "finalized using pnacl-finalize)"; | |
| 149 } | |
| 144 return true; | 150 return true; |
| 151 } | |
| 145 BufPtr += WordSize; | 152 BufPtr += WordSize; |
| 146 | 153 |
| 147 // Read #Fields and number of bytes needed for the header. | 154 // Read #Fields and number of bytes needed for the header. |
| 148 if (BufPtr + WordSize > BufEnd) | 155 if (BufPtr + WordSize > BufEnd) |
| 149 return true; | 156 return UnsupportedError("Bitcode read failure"); |
| 150 NumFields = static_cast<unsigned>(BufPtr[0]) | | 157 NumFields = static_cast<unsigned>(BufPtr[0]) | |
| 151 (static_cast<unsigned>(BufPtr[1]) << 8); | 158 (static_cast<unsigned>(BufPtr[1]) << 8); |
| 152 NumBytes = static_cast<unsigned>(BufPtr[2]) | | 159 NumBytes = static_cast<unsigned>(BufPtr[2]) | |
| 153 (static_cast<unsigned>(BufPtr[3]) << 8); | 160 (static_cast<unsigned>(BufPtr[3]) << 8); |
| 154 BufPtr += WordSize; | 161 BufPtr += WordSize; |
| 155 return false; | 162 return false; |
| 156 } | 163 } |
| 157 | 164 |
| 158 bool NaClBitcodeHeader::ReadFields(const unsigned char *BufPtr, | 165 bool NaClBitcodeHeader::ReadFields(const unsigned char *BufPtr, |
| 159 const unsigned char *BufEnd, | 166 const unsigned char *BufEnd, |
| 160 unsigned NumFields, unsigned NumBytes) { | 167 unsigned NumFields, unsigned NumBytes) { |
| 161 HeaderSize = NumBytes + (2 * WordSize); | 168 HeaderSize = NumBytes + (2 * WordSize); |
| 162 | 169 |
| 163 // Read in each field. | 170 // Read in each field. |
| 164 for (size_t i = 0; i < NumFields; ++i) { | 171 for (size_t i = 0; i < NumFields; ++i) { |
| 165 NaClBitcodeHeaderField *Field = new NaClBitcodeHeaderField(); | 172 NaClBitcodeHeaderField *Field = new NaClBitcodeHeaderField(); |
| 166 Fields.push_back(Field); | 173 Fields.push_back(Field); |
| 167 if (!Field->Read(BufPtr, BufEnd - BufPtr)) | 174 if (!Field->Read(BufPtr, BufEnd - BufPtr)) |
| 168 return true; | 175 return UnsupportedError("Bitcode read failure"); |
| 169 size_t FieldSize = Field->GetTotalSize(); | 176 size_t FieldSize = Field->GetTotalSize(); |
| 170 BufPtr += FieldSize; | 177 BufPtr += FieldSize; |
| 171 } | 178 } |
| 172 return false; | 179 return false; |
| 173 } | 180 } |
| 174 | 181 |
| 175 bool NaClBitcodeHeader::Read(const unsigned char *&BufPtr, | 182 bool NaClBitcodeHeader::Read(const unsigned char *&BufPtr, |
| 176 const unsigned char *&BufEnd) { | 183 const unsigned char *&BufEnd) { |
| 177 unsigned NumFields; | 184 unsigned NumFields; |
| 178 unsigned NumBytes; | 185 unsigned NumBytes; |
| 179 if (ReadPrefix(BufPtr, BufEnd, NumFields, NumBytes)) | 186 if (ReadPrefix(BufPtr, BufEnd, NumFields, NumBytes)) |
| 180 return true; | 187 return true; // ReadPrefix sets UnsupportedMessage |
| 181 BufPtr += 2 * WordSize; | 188 BufPtr += 2 * WordSize; |
| 182 | 189 |
| 183 if (ReadFields(BufPtr, BufEnd, NumFields, NumBytes)) | 190 if (ReadFields(BufPtr, BufEnd, NumFields, NumBytes)) |
| 184 return true; | 191 return true; // ReadFields sets UnsupportedMessage |
| 185 BufPtr += NumBytes; | 192 BufPtr += NumBytes; |
| 186 InstallFields(); | 193 InstallFields(); |
| 187 return false; | 194 return false; |
| 188 } | 195 } |
| 189 | 196 |
| 190 bool NaClBitcodeHeader::Read(StreamableMemoryObject *Bytes) { | 197 bool NaClBitcodeHeader::Read(StreamableMemoryObject *Bytes) { |
| 191 unsigned NumFields; | 198 unsigned NumFields; |
| 192 unsigned NumBytes; | 199 unsigned NumBytes; |
| 193 { | 200 { |
| 194 unsigned char Buffer[2 * WordSize]; | 201 unsigned char Buffer[2 * WordSize]; |
| 195 if (Bytes->readBytes(0, sizeof(Buffer), Buffer, NULL) || | 202 if (Bytes->readBytes(0, sizeof(Buffer), Buffer, NULL) || |
| 196 ReadPrefix(Buffer, Buffer + sizeof(Buffer), NumFields, NumBytes)) | 203 ReadPrefix(Buffer, Buffer + sizeof(Buffer), NumFields, NumBytes)) |
| 197 return true; | 204 // ReadPrefix sets UnsupportedMessage but readBytes does not |
|
jvoung (off chromium)
2013/09/13 20:20:23
readBytes could happen first w/ a chance to set th
Derek Schuff
2013/09/13 20:33:15
I mean that if If readBytes fails, ReadPrefix will
jvoung (off chromium)
2013/09/13 20:35:25
but is there a case where readBytes doesn't fail,
| |
| 205 return UnsupportedError("Bitcode read failure"); | |
| 198 } | 206 } |
| 199 uint8_t *Header = new uint8_t[NumBytes]; | 207 uint8_t *Header = new uint8_t[NumBytes]; |
| 200 bool failed = | 208 bool failed = |
| 201 Bytes->readBytes(2 * WordSize, NumBytes, Header, NULL) || | 209 Bytes->readBytes(2 * WordSize, NumBytes, Header, NULL) || |
| 202 ReadFields(Header, Header + NumBytes, NumFields, NumBytes); | 210 ReadFields(Header, Header + NumBytes, NumFields, NumBytes); |
| 203 delete[] Header; | 211 delete[] Header; |
| 204 if (failed) | 212 if (failed) |
| 205 return true; | 213 return UnsupportedError("Bitcode read failure"); |
| 206 InstallFields(); | 214 InstallFields(); |
| 207 return false; | 215 return false; |
| 208 } | 216 } |
| 209 | 217 |
| 210 NaClBitcodeHeaderField * | 218 NaClBitcodeHeaderField * |
| 211 NaClBitcodeHeader::GetTaggedField(NaClBitcodeHeaderField::Tag ID) const { | 219 NaClBitcodeHeader::GetTaggedField(NaClBitcodeHeaderField::Tag ID) const { |
| 212 for (std::vector<NaClBitcodeHeaderField *>::const_iterator | 220 for (std::vector<NaClBitcodeHeaderField *>::const_iterator |
| 213 Iter = Fields.begin(), | 221 Iter = Fields.begin(), |
| 214 IterEnd = Fields.end(); | 222 IterEnd = Fields.end(); |
| 215 Iter != IterEnd; ++Iter) { | 223 Iter != IterEnd; ++Iter) { |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 256 << PNaClVersion << "\n"; | 264 << PNaClVersion << "\n"; |
| 257 UnsupportedStream.flush(); | 265 UnsupportedStream.flush(); |
| 258 } | 266 } |
| 259 if (Fields.size() != 1) { | 267 if (Fields.size() != 1) { |
| 260 IsSupportedFlag = false; | 268 IsSupportedFlag = false; |
| 261 IsReadableFlag = false; | 269 IsReadableFlag = false; |
| 262 if (!UpdatedUnsupportedMessage) | 270 if (!UpdatedUnsupportedMessage) |
| 263 UnsupportedMessage = "Unknown header field(s) found"; | 271 UnsupportedMessage = "Unknown header field(s) found"; |
| 264 } | 272 } |
| 265 } | 273 } |
| OLD | NEW |