#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 EINTR;
int EINVAL;
int /*<<< orphan*/ assert(int);
int write(int, char *, int);
int robust_write(int fd, char *buf, int buf_size) {
  int bytes_remain, bytes_written;
  int buf_offset;
  bytes_remain = buf_size;
  while (true) {
    buf_offset = buf_size - bytes_remain;
    bytes_written = write(fd, buf + buf_offset, bytes_remain);
    bytes_remain =
        (bytes_written > 0) ? bytes_remain - bytes_written : bytes_remain;
    if (bytes_written < 0 && bytes_written != -EINTR) {
      return bytes_written;
    } else if (bytes_written == 0) {
      return buf_size - bytes_remain;
    }
  }
  assert(false);
  return -EINVAL;
}


x86 assembly (clang, -O0):
<robust_write>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %edi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
mov    -0x14(%rbp),%eax
mov    %eax,-0x18(%rbp)
mov    -0x14(%rbp),%eax
sub    -0x18(%rbp),%eax
mov    %eax,-0x20(%rbp)
mov    -0x8(%rbp),%edi
mov    -0x10(%rbp),%rsi
movslq -0x20(%rbp),%rax
add    %rax,%rsi
mov    -0x18(%rbp),%edx
call   1030 <write@plt>
mov    %eax,-0x1c(%rbp)
cmpl   $0x0,-0x1c(%rbp)
jle    1162 <robust_write+0x52>
mov    -0x18(%rbp),%eax
sub    -0x1c(%rbp),%eax
mov    %eax,-0x24(%rbp)
jmp    1168 <robust_write+0x58>
mov    -0x18(%rbp),%eax
mov    %eax,-0x24(%rbp)
mov    -0x24(%rbp),%eax
mov    %eax,-0x18(%rbp)
cmpl   $0x0,-0x1c(%rbp)
jge    1199 <robust_write+0x89>
mov    -0x1c(%rbp),%eax
xor    %ecx,%ecx
mov    0x2e5c(%rip),%rdx
sub    (%rdx),%ecx
cmp    %ecx,%eax
je     1199 <robust_write+0x89>
mov    -0x1c(%rbp),%eax
mov    %eax,-0x4(%rbp)
jmp    11bb <robust_write+0xab>
cmpl   $0x0,-0x1c(%rbp)
jne    11b1 <robust_write+0xa1>
mov    -0x14(%rbp),%eax
sub    -0x18(%rbp),%eax
mov    %eax,-0x4(%rbp)
jmp    11bb <robust_write+0xab>
jmp    11b6 <robust_write+0xa6>
jmp    1128 <robust_write+0x18>
mov    -0x4(%rbp),%eax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<robust_write>:
push   %rbp
push   %r15
push   %r14
push   %r12
push   %rbx
mov    %edx,%ebp
mov    %rsi,%r14
mov    %edi,%r15d
mov    0x2eb9(%rip),%r12
mov    %edx,%ebx
jmp    1136 <robust_write+0x26>
nopl   0x0(%rax,%rax,1)
sub    %ecx,%ebx
test   %eax,%eax
je     1160 <robust_write+0x50>
mov    %ebp,%eax
sub    %ebx,%eax
movslq %eax,%rsi
add    %r14,%rsi
mov    %r15d,%edi
mov    %ebx,%edx
call   1030 <write@plt>
mov    $0x0,%ecx
test   %eax,%eax
cmovns %eax,%ecx
jns    1130 <robust_write+0x20>
mov    (%r12),%edx
add    %eax,%edx
je     1130 <robust_write+0x20>
jmp    1164 <robust_write+0x54>
sub    %ebx,%ebp
mov    %ebp,%eax
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
@EINTR = global i32 0
@EINVAL = global i32 0
define i32 @robust_write(i32 %0, i8* %1, i32 %2) {
%4 = alloca i32
%5 = alloca i32
%6 = alloca i8*
%7 = alloca i32
%8 = alloca i32
%9 = alloca i32
%10 = alloca i32
store i32 %0, i32* %5
store i8* %1, i8** %6
store i32 %2, i32* %7
%11 = load i32, i32* %7
store i32 %11, i32* %8
br label %12
12: ; preds = %3, %50
%13 = load i32, i32* %7
%14 = load i32, i32* %8
%15 = sub nsw i32 %13, %14
store i32 %15, i32* %10
%16 = load i32, i32* %5
%17 = load i8*, i8** %6
%18 = load i32, i32* %10
%19 = sext i32 %18 to i64
%20 = getelementptr inbounds i8, i8* %17, i64 %19
%21 = load i32, i32* %8
%22 = call i32 @write(i32 %16, i8* %20, i32 %21)
store i32 %22, i32* %9
%23 = load i32, i32* %9
%24 = icmp sgt i32 %23, 0
br i1 %24, label %25, label %29
25: ; preds = %12
%26 = load i32, i32* %8
%27 = load i32, i32* %9
%28 = sub nsw i32 %26, %27
br label %31
29: ; preds = %12
%30 = load i32, i32* %8
br label %31
31: ; preds = %29, %25
%32 = phi i32 [ %28, %25 ], [ %30, %29 ]
store i32 %32, i32* %8
%33 = load i32, i32* %9
%34 = icmp slt i32 %33, 0
br i1 %34, label %35, label %42
35: ; preds = %31
%36 = load i32, i32* %9
%37 = load i32, i32* @EINTR
%38 = sub nsw i32 0, %37
%39 = icmp ne i32 %36, %38
br i1 %39, label %40, label %42
40: ; preds = %35
%41 = load i32, i32* %9
store i32 %41, i32* %4
br label %51
42: ; preds = %35, %31
%43 = load i32, i32* %9
%44 = icmp eq i32 %43, 0
br i1 %44, label %45, label %49
45: ; preds = %42
%46 = load i32, i32* %7
%47 = load i32, i32* %8
%48 = sub nsw i32 %46, %47
store i32 %48, i32* %4
br label %51
49: ; preds = %42
br label %50
50: ; preds = %49
br label %12
51: ; preds = %45, %40
%52 = load i32, i32* %4
ret i32 %52
}

LLVM IR (-O2):
@EINTR = global i32 0
@EINVAL = global i32 0
define i32 @robust_write(i32 %0, i8* nocapture %1, i32 %2) {
br label %4
4: ; preds = %18, %3
%5 = phi i32 [ %2, %3 ], [ %12, %18 ]
%6 = sub nsw i32 %2, %5
%7 = sext i32 %6 to i64
%8 = getelementptr inbounds i8, i8* %1, i64 %7
%9 = tail call i32 @write(i32 %0, i8* %8, i32 %5)
%10 = icmp sgt i32 %9, 0
%11 = select i1 %10, i32 %9, i32 0
%12 = sub nsw i32 %5, %11
%13 = icmp slt i32 %9, 0
br i1 %13, label %14, label %18
14: ; preds = %4
%15 = load i32, i32* @EINTR
%16 = sub nsw i32 0, %15
%17 = icmp eq i32 %9, %16
br i1 %17, label %18, label %22
18: ; preds = %14, %4
%19 = icmp eq i32 %9, 0
br i1 %19, label %20, label %4
20: ; preds = %18
%21 = sub nsw i32 %2, %12
br label %22
22: ; preds = %14, %20
%23 = phi i32 [ %21, %20 ], [ %9, %14 ]
ret i32 %23
}