From 89c624b7550f8fed7b300b9ce22f7866b74d5ece Mon Sep 17 00:00:00 2001
From: Daniel Micay <danielmicay@gmail.com>
Date: Sun, 2 Sep 2018 02:08:44 -0400
Subject: [PATCH] mark page management error code paths as unlikely

---
 pages.c | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/pages.c b/pages.c
index e08c980..eb29256 100644
--- a/pages.c
+++ b/pages.c
@@ -14,11 +14,11 @@ void *allocate_pages(size_t usable_size, size_t guard_size, bool unprotect) {
         return NULL;
     }
     void *real = memory_map(real_size);
-    if (real == NULL) {
+    if (unlikely(real == NULL)) {
         return NULL;
     }
     void *usable = (char *)real + guard_size;
-    if (unprotect && memory_protect_rw(usable, usable_size)) {
+    if (unprotect && unlikely(memory_protect_rw(usable, usable_size))) {
         memory_unmap(real, real_size);
         return NULL;
     }
@@ -49,7 +49,7 @@ void *allocate_pages_aligned(size_t usable_size, size_t alignment, size_t guard_
     }
 
     void *real = memory_map(real_alloc_size);
-    if (real == NULL) {
+    if (unlikely(real == NULL)) {
         return NULL;
     }
 
@@ -59,20 +59,20 @@ void *allocate_pages_aligned(size_t usable_size, size_t alignment, size_t guard_
     size_t trail_size = alloc_size - lead_size - usable_size;
     void *base = (char *)usable + lead_size;
 
-    if (memory_protect_rw(base, usable_size)) {
+    if (unlikely(memory_protect_rw(base, usable_size))) {
         memory_unmap(real, real_alloc_size);
         return NULL;
     }
 
     if (lead_size) {
-        if (memory_unmap(real, lead_size)) {
+        if (unlikely(memory_unmap(real, lead_size))) {
             memory_unmap(real, real_alloc_size);
             return NULL;
         }
     }
 
     if (trail_size) {
-        if (memory_unmap((char *)base + usable_size + guard_size, trail_size)) {
+        if (unlikely(memory_unmap((char *)base + usable_size + guard_size, trail_size))) {
             memory_unmap(real, real_alloc_size);
             return NULL;
         }
@@ -80,4 +80,3 @@ void *allocate_pages_aligned(size_t usable_size, size_t alignment, size_t guard_
 
     return base;
 }
-