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


x86 assembly (clang, -O0):
<count_blocks>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
movl   $0x4,-0x10(%rbp)
movl   $0x0,-0x18(%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     11d5 <count_blocks+0xb5>
cmpl   $0x0,-0xc(%rbp)
je     118c <count_blocks+0x6c>
mov    -0x8(%rbp),%rax
movslq -0x10(%rbp),%rcx
mov    (%rax,%rcx,4),%edi
call   1040 <get_use_bit@plt>
cmp    $0x0,%rax
je     1187 <count_blocks+0x67>
mov    -0x18(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x18(%rbp)
jmp    11b4 <count_blocks+0x94>
mov    -0x8(%rbp),%rax
movslq -0x10(%rbp),%rcx
mov    (%rax,%rcx,4),%edi
call   1040 <get_use_bit@plt>
cmp    $0x0,%rax
jne    11af <count_blocks+0x8f>
mov    -0x18(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x18(%rbp)
jmp    11b4 <count_blocks+0x94>
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    1150 <count_blocks+0x30>
mov    -0x18(%rbp),%eax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<count_blocks>:
push   %rbp
push   %r15
push   %r14
push   %r12
push   %rbx
mov    %esi,%r14d
mov    %rdi,%r15
mov    0x10(%rdi),%edi
call   1030 <get_size@plt>
test   %eax,%eax
je     118c <count_blocks+0x6c>
mov    %eax,%r12d
mov    $0x4,%ebx
xor    %ebp,%ebp
cs nopw 0x0(%rax,%rax,1)
xchg   %ax,%ax
movslq %ebx,%rbx
mov    (%r15,%rbx,4),%edi
call   1040 <get_use_bit@plt>
xor    %ecx,%ecx
xor    %edx,%edx
test   %rax,%rax
setne  %cl
sete   %dl
test   %r14d,%r14d
cmove  %edx,%ecx
movzbl %cl,%eax
add    %eax,%ebp
add    %r12d,%ebx
movslq %ebx,%rax
mov    (%r15,%rax,4),%edi
call   1030 <get_size@plt>
mov    %eax,%r12d
test   %eax,%eax
jne    1150 <count_blocks+0x30>
jmp    118e <count_blocks+0x6e>
xor    %ebp,%ebp
mov    %ebp,%eax
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define i32 @count_blocks(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 4, i32* %5
store i32 0, i32* %7
%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 = %44, %2
%15 = load i32, i32* %6
%16 = icmp ne i32 %15, 0
br i1 %16, label %17, label %54
17: ; preds = %14
%18 = load i32, i32* %4
%19 = icmp ne i32 %18, 0
br i1 %19, label %20, label %32
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 %31
28: ; preds = %20
%29 = load i32, i32* %7
%30 = add nsw i32 %29, 1
store i32 %30, i32* %7
br label %31
31: ; preds = %28, %20
br label %44
32: ; preds = %17
%33 = load i32*, i32** %3
%34 = load i32, i32* %5
%35 = sext i32 %34 to i64
%36 = getelementptr inbounds i32, i32* %33, i64 %35
%37 = load i32, i32* %36
%38 = call i64 @get_use_bit(i32 %37)
%39 = icmp ne i64 %38, 0
br i1 %39, label %43, label %40
40: ; preds = %32
%41 = load i32, i32* %7
%42 = add nsw i32 %41, 1
store i32 %42, i32* %7
br label %43
43: ; preds = %40, %32
br label %44
44: ; preds = %43, %31
%45 = load i32, i32* %6
%46 = load i32, i32* %5
%47 = add nsw i32 %46, %45
store i32 %47, i32* %5
%48 = load i32*, i32** %3
%49 = load i32, i32* %5
%50 = sext i32 %49 to i64
%51 = getelementptr inbounds i32, i32* %48, i64 %50
%52 = load i32, i32* %51
%53 = call i32 @get_size(i32 %52)
store i32 %53, i32* %6
br label %14
54: ; preds = %14
%55 = load i32, i32* %7
ret i32 %55
}

LLVM IR (-O2):
define i32 @count_blocks(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 ne i64 %16, 0
%18 = icmp eq i64 %16, 0
%19 = select i1 %8, i1 %18, i1 %17
%20 = zext i1 %19 to i32
%21 = add nuw nsw i32 %10, %20
%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
}