[Writer][ELF] Use correct data types for sizes. Fixes integer overflow bug.

This only happens when the section header count is > 1024.

llvm-svn: 176747

1 files changed, 3 insertions, 3 deletions

diff --git a/lld/lib/ReaderWriter/ELF/HeaderChunks.h b/lld/lib/ReaderWriter/ELF/HeaderChunks.h
index 4c81b12d681..742f828e167 100644
--- a/lld/lib/ReaderWriter/ELF/HeaderChunks.h
+++ b/lld/lib/ReaderWriter/ELF/HeaderChunks.h
@@ -259,13 +259,13 @@ public:
 
   void finalize() {}
 
-  inline uint16_t fileSize() { return sizeof(Elf_Shdr) * _sectionInfo.size(); }
+  inline uint64_t fileSize() { return sizeof(Elf_Shdr) * _sectionInfo.size(); }
 
-  inline int64_t entsize() {
+  inline uint64_t entsize() {
     return sizeof(Elf_Shdr);
   }
   
-  inline int64_t numHeaders() {
+  inline uint64_t numHeaders() {
     return _sectionInfo.size();
   }