From 00aae24a08a3935d408d211f13aced1c7aabc4bb Mon Sep 17 00:00:00 2001 From: Rafael Espindola Date: Tue, 5 Jul 2011 18:41:47 +0000 Subject: [PATCH] Compare all 4 bytes of the header. llvm-svn: 134427 --- llvm/lib/Object/COFFObjectFile.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/llvm/lib/Object/COFFObjectFile.cpp b/llvm/lib/Object/COFFObjectFile.cpp index 158263e5b108..00038071eadd 100644 --- a/llvm/lib/Object/COFFObjectFile.cpp +++ b/llvm/lib/Object/COFFObjectFile.cpp @@ -309,8 +309,9 @@ COFFObjectFile::COFFObjectFile(MemoryBuffer *Object, error_code &ec) if (!checkSize(Data, ec, 0x3c + 8)) return; HeaderStart += *reinterpret_cast(base() + 0x3c); // Check the PE header. ("PE\0\0") - if (StringRef(reinterpret_cast(base() + HeaderStart), 4) - != "PE\0\0") { + StringRef Actual(reinterpret_cast(base() + HeaderStart), 4); + StringRef Expcted("PE\0\0", 4); + if (Actual != Expcted) { ec = object_error::parse_failed; return; }