#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 get_size(int const);
scalar_t__ get_use_bit(int const);
int total_block_size(const int *mem, int sum_allocs) {
  int k, current_size, size_sum;
  current_size = size_sum = 0;
  k = 4;
  current_size = get_size(mem[k]);
  while (current_size != 0) {
    if (sum_allocs) {
      if (get_use_bit(mem[k])) {
        size_sum += current_size;
      }
    } else {
      if (!get_use_bit(mem[k])) {
        size_sum += current_size;
      }
    }
    k += current_size;
    current_size = get_size(mem[k]);
  }
  return size_sum;
}


x86 assembly (clang, -O0):
<total_block_size>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
movl   $0x0,-0x18(%rbp)
movl   $0x0,-0x14(%rbp)
movl   $0x4,-0x10(%rbp)
mov    -0x8(%rbp),%rax
movslq -0x10(%rbp),%rcx
mov    (%rax,%rcx,4),%edi
call   1030 <get_size@plt>
mov    %eax,-0x14(%rbp)
cmpl   $0x0,-0x14(%rbp)
je     11dc <total_block_size+0xbc>
cmpl   $0x0,-0xc(%rbp)
je     1193 <total_block_size+0x73>
mov    -0x8(%rbp),%rax
movslq -0x10(%rbp),%rcx
mov    (%rax,%rcx,4),%edi
call   1040 <get_use_bit@plt>
cmp    $0x0,%rax
je     118e <total_block_size+0x6e>
mov    -0x14(%rbp),%eax
add    -0x18(%rbp),%eax
mov    %eax,-0x18(%rbp)
jmp    11bb <total_block_size+0x9b>
mov    -0x8(%rbp),%rax
movslq -0x10(%rbp),%rcx
mov    (%rax,%rcx,4),%edi
call   1040 <get_use_bit@plt>
cmp    $0x0,%rax
jne    11b6 <total_block_size+0x96>
mov    -0x14(%rbp),%eax
add    -0x18(%rbp),%eax
mov    %eax,-0x18(%rbp)
jmp    11bb <total_block_size+0x9b>
mov    -0x14(%rbp),%eax
add    -0x10(%rbp),%eax
mov    %eax,-0x10(%rbp)
mov    -0x8(%rbp),%rax
movslq -0x10(%rbp),%rcx
mov    (%rax,%rcx,4),%edi
call   1030 <get_size@plt>
mov    %eax,-0x14(%rbp)
jmp    1157 <total_block_size+0x37>
mov    -0x18(%rbp),%eax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<total_block_size>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
push   %rax
mov    %esi,%r14d
mov    %rdi,%r15
mov    0x10(%rdi),%edi
call   1030 <get_size@plt>
test   %eax,%eax
je     118c <total_block_size+0x6c>
mov    %eax,%ebx
mov    $0x4,%ebp
xor    %r13d,%r13d
xor    %r12d,%r12d
nopw   0x0(%rax,%rax,1)
movslq %ebp,%rbp
mov    (%r15,%rbp,4),%edi
call   1040 <get_use_bit@plt>
test   %rax,%rax
mov    %ebx,%eax
cmove  %r13d,%eax
mov    $0x0,%ecx
cmove  %ebx,%ecx
test   %r14d,%r14d
cmovne %eax,%ecx
add    %ecx,%r12d
add    %ebx,%ebp
movslq %ebp,%rax
mov    (%r15,%rax,4),%edi
call   1030 <get_size@plt>
mov    %eax,%ebx
test   %eax,%eax
jne    1150 <total_block_size+0x30>
jmp    118f <total_block_size+0x6f>
xor    %r12d,%r12d
mov    %r12d,%eax
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define i32 @total_block_size(i32* %0, i32 %1) {
%3 = alloca i32*
%4 = alloca i32
%5 = alloca i32
%6 = alloca i32
%7 = alloca i32
store i32* %0, i32** %3
store i32 %1, i32* %4
store i32 0, i32* %7
store i32 0, i32* %6
store i32 4, i32* %5
%8 = load i32*, i32** %3
%9 = load i32, i32* %5
%10 = sext i32 %9 to i64
%11 = getelementptr inbounds i32, i32* %8, i64 %10
%12 = load i32, i32* %11
%13 = call i32 @get_size(i32 %12)
store i32 %13, i32* %6
br label %14
14: ; preds = %46, %2
%15 = load i32, i32* %6
%16 = icmp ne i32 %15, 0
br i1 %16, label %17, label %56
17: ; preds = %14
%18 = load i32, i32* %4
%19 = icmp ne i32 %18, 0
br i1 %19, label %20, label %33
20: ; preds = %17
%21 = load i32*, i32** %3
%22 = load i32, i32* %5
%23 = sext i32 %22 to i64
%24 = getelementptr inbounds i32, i32* %21, i64 %23
%25 = load i32, i32* %24
%26 = call i64 @get_use_bit(i32 %25)
%27 = icmp ne i64 %26, 0
br i1 %27, label %28, label %32
28: ; preds = %20
%29 = load i32, i32* %6
%30 = load i32, i32* %7
%31 = add nsw i32 %30, %29
store i32 %31, i32* %7
br label %32
32: ; preds = %28, %20
br label %46
33: ; preds = %17
%34 = load i32*, i32** %3
%35 = load i32, i32* %5
%36 = sext i32 %35 to i64
%37 = getelementptr inbounds i32, i32* %34, i64 %36
%38 = load i32, i32* %37
%39 = call i64 @get_use_bit(i32 %38)
%40 = icmp ne i64 %39, 0
br i1 %40, label %45, label %41
41: ; preds = %33
%42 = load i32, i32* %6
%43 = load i32, i32* %7
%44 = add nsw i32 %43, %42
store i32 %44, i32* %7
br label %45
45: ; preds = %41, %33
br label %46
46: ; preds = %45, %32
%47 = load i32, i32* %6
%48 = load i32, i32* %5
%49 = add nsw i32 %48, %47
store i32 %49, i32* %5
%50 = load i32*, i32** %3
%51 = load i32, i32* %5
%52 = sext i32 %51 to i64
%53 = getelementptr inbounds i32, i32* %50, i64 %52
%54 = load i32, i32* %53
%55 = call i32 @get_size(i32 %54)
store i32 %55, i32* %6
br label %14
56: ; preds = %14
%57 = load i32, i32* %7
ret i32 %57
}

LLVM IR (-O2):
define i32 @total_block_size(i32* nocapture %0, i32 %1) {
%3 = getelementptr inbounds i32, i32* %0, i64 4
%4 = load i32, i32* %3
%5 = tail call i32 @get_size(i32 %4)
%6 = icmp eq i32 %5, 0
br i1 %6, label %28, label %7
7: ; preds = %2
%8 = icmp eq i32 %1, 0
br label %9
9: ; preds = %7, %9
%10 = phi i32 [ 0, %7 ], [ %21, %9 ]
%11 = phi i32 [ %5, %7 ], [ %26, %9 ]
%12 = phi i32 [ 4, %7 ], [ %22, %9 ]
%13 = sext i32 %12 to i64
%14 = getelementptr inbounds i32, i32* %0, i64 %13
%15 = load i32, i32* %14
%16 = tail call i64 @get_use_bit(i32 %15)
%17 = icmp eq i64 %16, 0
%18 = select i1 %17, i32 0, i32 %11
%19 = select i1 %17, i32 %11, i32 0
%20 = select i1 %8, i32 %19, i32 %18
%21 = add nsw i32 %20, %10
%22 = add nsw i32 %11, %12
%23 = sext i32 %22 to i64
%24 = getelementptr inbounds i32, i32* %0, i64 %23
%25 = load i32, i32* %24
%26 = tail call i32 @get_size(i32 %25)
%27 = icmp eq i32 %26, 0
br i1 %27, label %28, label %9
28: ; preds = %9, %2
%29 = phi i32 [ 0, %2 ], [ %21, %9 ]
ret i32 %29
}