#define NULL ((void *)0)
typedef unsigned long size_t; // Customize by platform.
typedef long intptr_t;
typedef unsigned long uintptr_t;
typedef long scalar_t__; // Either arithmetic or pointer type.
/* By default, we understand bool (as a convenience). */
typedef int bool;
#define false 0
#define true 1

/* Forward declarations */

/* Type definitions */

/* Variables and functions */
int /*<<< orphan*/ PROT_NONE;
void *alloc_pages(int);
int /*<<< orphan*/ assert(int);
int getpagesize();
int /*<<< orphan*/ mprotect(void *, int, int /*<<< orphan*/);
void *alloc_offset_guard(unsigned long valid_size,
                         unsigned long overflow_size) {
  int valid_pages, overflow_pages;
  int page_size;
  void *result;
  assert(overflow_size > 0);
  page_size = getpagesize();
  valid_pages = (valid_size + page_size - 1) / page_size;
  overflow_pages = (overflow_size + page_size - 1) / page_size;
  result = alloc_pages(valid_pages + overflow_pages);
  mprotect(result + valid_pages * page_size, overflow_pages * page_size,
           PROT_NONE);
  unsigned long offset = page_size - valid_size % page_size;
  if (offset == (unsigned long)page_size)
    offset = 0;
  return result + offset;
}


x86 assembly (clang, -O0):
<alloc_offset_guard>:
push   %rbp
mov    %rsp,%rbp
sub    $0x40,%rsp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
cmpq   $0x0,-0x10(%rbp)
seta   %al
and    $0x1,%al
movzbl %al,%edi
call   1030 <assert@plt>
mov    $0x0,%al
call   1050 <getpagesize@plt>
mov    %eax,-0x1c(%rbp)
mov    -0x8(%rbp),%rax
movslq -0x1c(%rbp),%rcx
add    %rcx,%rax
sub    $0x1,%rax
movslq -0x1c(%rbp),%rcx
xor    %edx,%edx
div    %rcx
mov    %eax,-0x14(%rbp)
mov    -0x10(%rbp),%rax
movslq -0x1c(%rbp),%rcx
add    %rcx,%rax
sub    $0x1,%rax
movslq -0x1c(%rbp),%rcx
xor    %edx,%edx
div    %rcx
mov    %eax,-0x18(%rbp)
mov    -0x14(%rbp),%edi
add    -0x18(%rbp),%edi
call   1060 <alloc_pages@plt>
mov    %rax,-0x28(%rbp)
mov    -0x28(%rbp),%rdi
mov    -0x14(%rbp),%eax
imul   -0x1c(%rbp),%eax
cltq
add    %rax,%rdi
mov    -0x18(%rbp),%esi
imul   -0x1c(%rbp),%esi
mov    0x2e21(%rip),%rax
mov    (%rax),%edx
call   1040 <mprotect@plt>
movslq -0x1c(%rbp),%rax
mov    %rax,-0x38(%rbp)
mov    -0x8(%rbp),%rax
movslq -0x1c(%rbp),%rcx
xor    %edx,%edx
div    %rcx
mov    -0x38(%rbp),%rax
sub    %rdx,%rax
mov    %rax,-0x30(%rbp)
mov    -0x30(%rbp),%rax
movslq -0x1c(%rbp),%rcx
cmp    %rcx,%rax
jne    120f <alloc_offset_guard+0xcf>
movq   $0x0,-0x30(%rbp)
mov    -0x28(%rbp),%rax
add    -0x30(%rbp),%rax
add    $0x40,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<alloc_offset_guard>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
push   %rax
mov    %rsi,%r15
mov    %rdi,%rbx
xor    %edi,%edi
test   %rsi,%rsi
setne  %dil
call   1030 <assert@plt>
xor    %eax,%eax
call   1050 <getpagesize@plt>
mov    %eax,%r12d
movslq %eax,%r14
lea    -0x1(%r14),%rcx
mov    %rbx,(%rsp)
lea    (%r14,%rbx,1),%rax
add    $0xffffffffffffffff,%rax
mov    %rax,%rdx
or     %r14,%rdx
shr    $0x20,%rdx
je     1192 <alloc_offset_guard+0x52>
xor    %edx,%edx
div    %r14
mov    %rax,%r13
jmp    119a <alloc_offset_guard+0x5a>
xor    %edx,%edx
div    %r12d
mov    %eax,%r13d
add    %r15,%rcx
mov    %rcx,%rax
or     %r14,%rax
shr    $0x20,%rax
je     11b6 <alloc_offset_guard+0x76>
mov    %rcx,%rax
xor    %edx,%edx
div    %r14
mov    %rax,%rbx
jmp    11bf <alloc_offset_guard+0x7f>
mov    %ecx,%eax
xor    %edx,%edx
div    %r12d
mov    %eax,%ebx
lea    (%rbx,%r13,1),%edi
call   1060 <alloc_pages@plt>
mov    %rax,%rbp
imul   %r12d,%r13d
movslq %r13d,%rdi
add    %rax,%rdi
imul   %r12d,%ebx
mov    0x2e10(%rip),%rax
mov    (%rax),%edx
mov    %ebx,%esi
call   1040 <mprotect@plt>
mov    (%rsp),%rax
mov    %rax,%rcx
or     %r14,%rcx
shr    $0x20,%rcx
je     1200 <alloc_offset_guard+0xc0>
xor    %edx,%edx
div    %r14
jmp    1205 <alloc_offset_guard+0xc5>
xor    %edx,%edx
div    %r12d
sub    %rdx,%r14
test   %rdx,%rdx
cmove  %rdx,%r14
add    %r14,%rbp
mov    %rbp,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
@PROT_NONE = global i32 0
define i8* @alloc_offset_guard(i64 %0, i64 %1) {
%3 = alloca i64
%4 = alloca i64
%5 = alloca i32
%6 = alloca i32
%7 = alloca i32
%8 = alloca i8*
%9 = alloca i64
store i64 %0, i64* %3
store i64 %1, i64* %4
%10 = load i64, i64* %4
%11 = icmp ugt i64 %10, 0
%12 = zext i1 %11 to i32
%13 = call i32 @assert(i32 %12)
%14 = call i32 (...) @getpagesize()
store i32 %14, i32* %7
%15 = load i64, i64* %3
%16 = load i32, i32* %7
%17 = sext i32 %16 to i64
%18 = add i64 %15, %17
%19 = sub i64 %18, 1
%20 = load i32, i32* %7
%21 = sext i32 %20 to i64
%22 = udiv i64 %19, %21
%23 = trunc i64 %22 to i32
store i32 %23, i32* %5
%24 = load i64, i64* %4
%25 = load i32, i32* %7
%26 = sext i32 %25 to i64
%27 = add i64 %24, %26
%28 = sub i64 %27, 1
%29 = load i32, i32* %7
%30 = sext i32 %29 to i64
%31 = udiv i64 %28, %30
%32 = trunc i64 %31 to i32
store i32 %32, i32* %6
%33 = load i32, i32* %5
%34 = load i32, i32* %6
%35 = add nsw i32 %33, %34
%36 = call i8* @alloc_pages(i32 %35)
store i8* %36, i8** %8
%37 = load i8*, i8** %8
%38 = load i32, i32* %5
%39 = load i32, i32* %7
%40 = mul nsw i32 %38, %39
%41 = sext i32 %40 to i64
%42 = getelementptr i8, i8* %37, i64 %41
%43 = load i32, i32* %6
%44 = load i32, i32* %7
%45 = mul nsw i32 %43, %44
%46 = load i32, i32* @PROT_NONE
%47 = call i32 @mprotect(i8* %42, i32 %45, i32 %46)
%48 = load i32, i32* %7
%49 = sext i32 %48 to i64
%50 = load i64, i64* %3
%51 = load i32, i32* %7
%52 = sext i32 %51 to i64
%53 = urem i64 %50, %52
%54 = sub i64 %49, %53
store i64 %54, i64* %9
%55 = load i64, i64* %9
%56 = load i32, i32* %7
%57 = sext i32 %56 to i64
%58 = icmp eq i64 %55, %57
br i1 %58, label %59, label %60
59: ; preds = %2
store i64 0, i64* %9
br label %60
60: ; preds = %59, %2
%61 = load i8*, i8** %8
%62 = load i64, i64* %9
%63 = getelementptr i8, i8* %61, i64 %62
ret i8* %63
}

LLVM IR (-O2):
@PROT_NONE = global i32 0
define i8* @alloc_offset_guard(i64 %0, i64 %1) {
%3 = icmp ne i64 %1, 0
%4 = zext i1 %3 to i32
%5 = tail call i32 @assert(i32 %4)
%6 = tail call i32 (...) @getpagesize()
%7 = sext i32 %6 to i64
%8 = add nsw i64 %7, -1
%9 = add i64 %8, %0
%10 = udiv i64 %9, %7
%11 = trunc i64 %10 to i32
%12 = add i64 %8, %1
%13 = udiv i64 %12, %7
%14 = trunc i64 %13 to i32
%15 = add nsw i32 %14, %11
%16 = tail call i8* @alloc_pages(i32 %15)
%17 = mul nsw i32 %6, %11
%18 = sext i32 %17 to i64
%19 = getelementptr i8, i8* %16, i64 %18
%20 = mul nsw i32 %6, %14
%21 = load i32, i32* @PROT_NONE
%22 = tail call i32 @mprotect(i8* %19, i32 %20, i32 %21)
%23 = urem i64 %0, %7
%24 = sub i64 %7, %23
%25 = icmp eq i64 %23, 0
%26 = select i1 %25, i64 0, i64 %24
%27 = getelementptr i8, i8* %16, i64 %26
ret i8* %27
}