Make text section 2MB aligned for hugepages.

chromeos/hugepage_text/hugepage_text.cc

Index: chromeos/hugepage_text/hugepage_text.cc
diff --git a/chromeos/hugepage_text/hugepage_text.cc b/chromeos/hugepage_text/hugepage_text.cc
index ccae53f723fdbd9e181f5421fd82c2fdc851062b..1509e0b8e452a2a01e17276ab1937aadee0e04d0 100644
--- a/chromeos/hugepage_text/hugepage_text.cc
+++ b/chromeos/hugepage_text/hugepage_text.cc
@@ -36,11 +36,6 @@ const int kMmapFlags = (MAP_ANONYMOUS | MAP_SHARED);
 const int kMmapHtlbFlags = (kMmapFlags | MAP_HUGETLB);
 const int kMremapFlags = (MREMAP_MAYMOVE | MREMAP_FIXED);
 
-// The number of hugepages we want to use to map chrome text section
-// to hugepages. With the help of AutoFDO, the hot functions are grouped
-// in to a small area of the binary.
-const int kNumHugePages = 15;
-
 // mremap syscall is always supported for small page segment on all kernels.
 // However, it is not the case for hugepage.
 // If not used carefully, mremap() a hugepage segment directly onto small page
@@ -172,15 +167,18 @@ static void MremapHugetlbText(void* vaddr, const size_t hsize) {
 // Effect: physical backing page changed from small page to hugepage. If there
 //         are error condition, the remaping operation is aborted.
 static void RemapHugetlbText(void* vaddr, const size_t segsize) {
-  int hsize = segsize;
-  if (segsize > kHpageSize * kNumHugePages)
-    hsize = kHpageSize * kNumHugePages;
+  // remove unaligned head regions
+  size_t head_gap = (kHpageSize - static_cast<size_t>(vaddr) % kHpageSize)
+                    % kHpageSize;
+  size_t addr = static_cast<size_t>(vaddr) + head_gap;
+
+  int hsize = segsize - head_gap;
   hsize = hsize & kHpageMask;
 
   if (hsize == 0)
     return;
 
-  MremapHugetlbText(vaddr, hsize);
+  MremapHugetlbText(static_cast<void *>(addr), hsize);
 }
 
 // For a given ELF program header descriptor, iterates over all segments within