#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 size_t ssize_t;

/* Variables and functions */
scalar_t__ EINTR;
scalar_t__ errno;
size_t write(int, char const *, size_t);
ssize_t safe_write(int fd, const void *buf, size_t count) {
  ssize_t size = 0, rtn;
repeat:
  rtn = write(fd, (const char *)buf + size, count - size);
  if (rtn > 0) {
    size += rtn;
    if (size < count) {
      goto repeat;
    }
  } else if (rtn < 0) {
    if (errno == EINTR) {
      goto repeat;
    } else if (size == 0) {
      return rtn;
    }
  }
  return size;
}


x86 assembly (clang, -O0):
<safe_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    -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)
jbe    1176 <safe_write+0x66>
mov    -0x30(%rbp),%rax
add    -0x28(%rbp),%rax
mov    %rax,-0x28(%rbp)
mov    -0x28(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    1171 <safe_write+0x61>
jmp    112b <safe_write+0x1b>
jmp    11c7 <safe_write+0xb7>
cmpq   $0x0,-0x30(%rbp)
jae    11c2 <safe_write+0xb2>
mov    0x2e58(%rip),%rax
mov    (%rax),%rax
mov    0x2e46(%rip),%rcx
cmp    (%rcx),%rax
jne    11a0 <safe_write+0x90>
jmp    112b <safe_write+0x1b>
cmpq   $0x0,-0x28(%rbp)
jne    11b8 <safe_write+0xa8>
mov    -0x30(%rbp),%rax
mov    %rax,-0x8(%rbp)
jmp    11cf <safe_write+0xbf>
jmp    11bd <safe_write+0xad>
jmp    11c2 <safe_write+0xb2>
jmp    11c7 <safe_write+0xb7>
mov    -0x28(%rbp),%rax
mov    %rax,-0x8(%rbp)
mov    -0x8(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<safe_write>:
push   %rbp
push   %r15
push   %r14
push   %rbx
push   %rax
mov    %rdx,%r14
mov    %rsi,%r15
mov    %edi,%ebp
xor    %ebx,%ebx
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax,%rax,1)
lea    (%r15,%rbx,1),%rsi
mov    %r14,%rdx
sub    %rbx,%rdx
mov    %ebp,%edi
call   1030 <write@plt>
test   %rax,%rax
je     114e <safe_write+0x3e>
add    %rax,%rbx
cmp    %r14,%rbx
jb     1130 <safe_write+0x20>
mov    %rbx,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
@errno = global i64 0
@EINTR = global i64 0
define i64 @safe_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 = %37, %28, %3
%11 = load i32, i32* %5
%12 = load i8*, i8** %6
%13 = load i64, i64* %8
%14 = getelementptr inbounds i8, i8* %12, i64 %13
%15 = load i64, i64* %7
%16 = load i64, i64* %8
%17 = sub i64 %15, %16
%18 = call i64 @write(i32 %11, i8* %14, i64 %17)
store i64 %18, i64* %9
%19 = load i64, i64* %9
%20 = icmp ugt i64 %19, 0
br i1 %20, label %21, label %30
21: ; preds = %10
%22 = load i64, i64* %9
%23 = load i64, i64* %8
%24 = add i64 %23, %22
store i64 %24, i64* %8
%25 = load i64, i64* %8
%26 = load i64, i64* %7
%27 = icmp ult i64 %25, %26
br i1 %27, label %28, label %29
28: ; preds = %21
br label %10
29: ; preds = %21
br label %46
30: ; preds = %10
%31 = load i64, i64* %9
%32 = icmp ult i64 %31, 0
br i1 %32, label %33, label %45
33: ; preds = %30
%34 = load i64, i64* @errno
%35 = load i64, i64* @EINTR
%36 = icmp eq i64 %34, %35
br i1 %36, label %37, label %38
37: ; preds = %33
br label %10
38: ; preds = %33
%39 = load i64, i64* %8
%40 = icmp eq i64 %39, 0
br i1 %40, label %41, label %43
41: ; preds = %38
%42 = load i64, i64* %9
store i64 %42, i64* %4
br label %48
43: ; preds = %38
br label %44
44: ; preds = %43
br label %45
45: ; preds = %44, %30
br label %46
46: ; preds = %45, %29
%47 = load i64, i64* %8
store i64 %47, i64* %4
br label %48
48: ; preds = %46, %41
%49 = load i64, i64* %4
ret i64 %49
}

LLVM IR (-O2):
@errno = global i64 0
@EINTR = global i64 0
define i64 @safe_write(i32 %0, i8* nocapture %1, i64 %2) {
br label %4
4: ; preds = %10, %3
%5 = phi i64 [ 0, %3 ], [ %11, %10 ]
%6 = getelementptr inbounds i8, i8* %1, i64 %5
%7 = sub i64 %2, %5
%8 = tail call i64 @write(i32 %0, i8* %6, i64 %7)
%9 = icmp eq i64 %8, 0
br i1 %9, label %13, label %10
10: ; preds = %4
%11 = add i64 %8, %5
%12 = icmp ult i64 %11, %2
br i1 %12, label %4, label %13
13: ; preds = %10, %4
%14 = phi i64 [ %11, %10 ], [ %5, %4 ]
ret i64 %14
}