#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 */
typedef scalar_t__ ssize_t;

/* Variables and functions */
scalar_t__ write(int, void const *, size_t);
ssize_t ensure_write(int fd, const void *buf, size_t count) {
  size_t total = 0;
  while (count > total) {
    ssize_t processed = write(fd, buf + total, count - total);
    if (processed < 0)
      return processed;
    else if (processed == 0)
      break;
    total += (size_t)processed;
  }
  return (ssize_t)total;
}


x86 assembly (clang, -O0):
<ensure_write>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %edi,-0xc(%rbp)
mov    %rsi,-0x18(%rbp)
mov    %rdx,-0x20(%rbp)
movq   $0x0,-0x28(%rbp)
mov    -0x20(%rbp),%rax
cmp    -0x28(%rbp),%rax
jbe    1193 <ensure_write+0x83>
mov    -0xc(%rbp),%edi
mov    -0x18(%rbp),%rsi
add    -0x28(%rbp),%rsi
mov    -0x20(%rbp),%rdx
sub    -0x28(%rbp),%rdx
call   1030 <write@plt>
mov    %rax,-0x30(%rbp)
cmpq   $0x0,-0x30(%rbp)
jge    116d <ensure_write+0x5d>
mov    -0x30(%rbp),%rax
mov    %rax,-0x8(%rbp)
jmp    119b <ensure_write+0x8b>
cmpq   $0x0,-0x30(%rbp)
jne    117d <ensure_write+0x6d>
jmp    1193 <ensure_write+0x83>
jmp    1182 <ensure_write+0x72>
mov    -0x30(%rbp),%rax
add    -0x28(%rbp),%rax
mov    %rax,-0x28(%rbp)
jmp    112b <ensure_write+0x1b>
mov    -0x28(%rbp),%rax
mov    %rax,-0x8(%rbp)
mov    -0x8(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<ensure_write>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
push   %rax
mov    %rdx,%r14
mov    %rsi,%r15
mov    %edi,%r12d
xor    %ebx,%ebx
mov    $0x1,%r13d
nopl   0x0(%rax)
cmp    %r14,%rbx
jae    1172 <ensure_write+0x62>
lea    (%r15,%rbx,1),%rsi
mov    %r14,%rdx
sub    %rbx,%rdx
mov    %r12d,%edi
call   1030 <write@plt>
xor    %ecx,%ecx
test   %rax,%rax
sete   %cl
lea    (%rcx,%rcx,2),%ecx
mov    $0x0,%edx
cmovns %rax,%rdx
cmovs  %rax,%rbp
cmovs  %r13d,%ecx
add    %rdx,%rbx
test   %ecx,%ecx
je     1130 <ensure_write+0x20>
cmp    $0x3,%ecx
je     1172 <ensure_write+0x62>
mov    %rbp,%rbx
mov    %rbx,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define i64 @ensure_write(i32 %0, i8* %1, i64 %2) {
%4 = alloca i64
%5 = alloca i32
%6 = alloca i8*
%7 = alloca i64
%8 = alloca i64
%9 = alloca i64
store i32 %0, i32* %5
store i8* %1, i8** %6
store i64 %2, i64* %7
store i64 0, i64* %8
br label %10
10: ; preds = %32, %3
%11 = load i64, i64* %7
%12 = load i64, i64* %8
%13 = icmp ugt i64 %11, %12
br i1 %13, label %14, label %36
14: ; preds = %10
%15 = load i32, i32* %5
%16 = load i8*, i8** %6
%17 = load i64, i64* %8
%18 = getelementptr i8, i8* %16, i64 %17
%19 = load i64, i64* %7
%20 = load i64, i64* %8
%21 = sub i64 %19, %20
%22 = call i64 @write(i32 %15, i8* %18, i64 %21)
store i64 %22, i64* %9
%23 = load i64, i64* %9
%24 = icmp slt i64 %23, 0
br i1 %24, label %25, label %27
25: ; preds = %14
%26 = load i64, i64* %9
store i64 %26, i64* %4
br label %38
27: ; preds = %14
%28 = load i64, i64* %9
%29 = icmp eq i64 %28, 0
br i1 %29, label %30, label %31
30: ; preds = %27
br label %36
31: ; preds = %27
br label %32
32: ; preds = %31
%33 = load i64, i64* %9
%34 = load i64, i64* %8
%35 = add i64 %34, %33
store i64 %35, i64* %8
br label %10
36: ; preds = %30, %10
%37 = load i64, i64* %8
store i64 %37, i64* %4
br label %38
38: ; preds = %36, %25
%39 = load i64, i64* %4
ret i64 %39
}

LLVM IR (-O2):
define i64 @ensure_write(i32 %0, i8* nocapture %1, i64 %2) {
br label %4
4: ; preds = %8, %3
%5 = phi i64 [ 0, %3 ], [ %16, %8 ]
%6 = phi i64 [ undef, %3 ], [ %17, %8 ]
%7 = icmp ult i64 %5, %2
br i1 %7, label %8, label %20
8: ; preds = %4
%9 = getelementptr i8, i8* %1, i64 %5
%10 = sub i64 %2, %5
%11 = tail call i64 @write(i32 %0, i8* %9, i64 %10)
%12 = icmp slt i64 %11, 0
%13 = icmp eq i64 %11, 0
%14 = select i1 %13, i32 3, i32 0
%15 = select i1 %12, i64 0, i64 %11
%16 = add i64 %15, %5
%17 = select i1 %12, i64 %11, i64 %6
%18 = select i1 %12, i32 1, i32 %14
switch i32 %18, label %19 [
i32 0, label %4
i32 3, label %20
]
19: ; preds = %8
br label %20
20: ; preds = %4, %8, %19
%21 = phi i64 [ %17, %19 ], [ %16, %8 ], [ %5, %4 ]
ret i64 %21
}