#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 char *buffer, size_t count) {
  ssize_t offset = 0;
  while (count > 0) {
    ssize_t block = write(fd, &buffer[offset], count);
    if (block < 0 && errno == EINTR)
      continue;
    if (block <= 0)
      return offset ? offset : block;
    offset += block;
    count -= block;
  }
  return offset;
}


x86 assembly (clang, -O0):
<safe_write>:
push   %rbp
mov    %rsp,%rbp
sub    $0x40,%rsp
mov    %edi,-0xc(%rbp)
mov    %rsi,-0x18(%rbp)
mov    %rdx,-0x20(%rbp)
movq   $0x0,-0x28(%rbp)
cmpq   $0x0,-0x20(%rbp)
jbe    11d0 <safe_write+0xc0>
mov    -0xc(%rbp),%edi
mov    -0x18(%rbp),%rsi
add    -0x28(%rbp),%rsi
mov    -0x20(%rbp),%rdx
call   1030 <write@plt>
mov    %rax,-0x30(%rbp)
cmpq   $0x0,-0x30(%rbp)
jae    1178 <safe_write+0x68>
mov    0x2e80(%rip),%rax
mov    (%rax),%rax
mov    0x2e6e(%rip),%rcx
cmp    (%rcx),%rax
jne    1178 <safe_write+0x68>
jmp    112b <safe_write+0x1b>
cmpq   $0x0,-0x30(%rbp)
ja     11b0 <safe_write+0xa0>
cmpq   $0x0,-0x28(%rbp)
je     119b <safe_write+0x8b>
mov    -0x28(%rbp),%rax
mov    %rax,-0x38(%rbp)
jmp    11a3 <safe_write+0x93>
mov    -0x30(%rbp),%rax
mov    %rax,-0x38(%rbp)
mov    -0x38(%rbp),%rax
mov    %rax,-0x8(%rbp)
jmp    11d8 <safe_write+0xc8>
mov    -0x30(%rbp),%rax
add    -0x28(%rbp),%rax
mov    %rax,-0x28(%rbp)
mov    -0x30(%rbp),%rcx
mov    -0x20(%rbp),%rax
sub    %rcx,%rax
mov    %rax,-0x20(%rbp)
jmp    112b <safe_write+0x1b>
mov    -0x28(%rbp),%rax
mov    %rax,-0x8(%rbp)
mov    -0x8(%rbp),%rax
add    $0x40,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<safe_write>:
push   %rbp
push   %r15
push   %r14
push   %rbx
push   %rax
test   %rdx,%rdx
je     114d <safe_write+0x3d>
mov    %rdx,%rbx
mov    %rsi,%r15
mov    %edi,%ebp
xor    %r14d,%r14d
nopw   0x0(%rax,%rax,1)
lea    (%r15,%r14,1),%rsi
mov    %ebp,%edi
mov    %rbx,%rdx
call   1030 <write@plt>
test   %rax,%rax
je     1150 <safe_write+0x40>
add    %rax,%r14
sub    %rax,%rbx
jne    1130 <safe_write+0x20>
jmp    1150 <safe_write+0x40>
xor    %r14d,%r14d
mov    %r14,%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 = %39, %26, %3
%11 = load i64, i64* %7
%12 = icmp ugt i64 %11, 0
br i1 %12, label %13, label %46
13: ; preds = %10
%14 = load i32, i32* %5
%15 = load i8*, i8** %6
%16 = load i64, i64* %8
%17 = getelementptr inbounds i8, i8* %15, i64 %16
%18 = load i64, i64* %7
%19 = call i64 @write(i32 %14, i8* %17, i64 %18)
store i64 %19, i64* %9
%20 = load i64, i64* %9
%21 = icmp ult i64 %20, 0
br i1 %21, label %22, label %27
22: ; preds = %13
%23 = load i64, i64* @errno
%24 = load i64, i64* @EINTR
%25 = icmp eq i64 %23, %24
br i1 %25, label %26, label %27
26: ; preds = %22
br label %10
27: ; preds = %22, %13
%28 = load i64, i64* %9
%29 = icmp ule i64 %28, 0
br i1 %29, label %30, label %39
30: ; preds = %27
%31 = load i64, i64* %8
%32 = icmp ne i64 %31, 0
br i1 %32, label %33, label %35
33: ; preds = %30
%34 = load i64, i64* %8
br label %37
35: ; preds = %30
%36 = load i64, i64* %9
br label %37
37: ; preds = %35, %33
%38 = phi i64 [ %34, %33 ], [ %36, %35 ]
store i64 %38, i64* %4
br label %48
39: ; preds = %27
%40 = load i64, i64* %9
%41 = load i64, i64* %8
%42 = add i64 %41, %40
store i64 %42, i64* %8
%43 = load i64, i64* %9
%44 = load i64, i64* %7
%45 = sub i64 %44, %43
store i64 %45, i64* %7
br label %10
46: ; preds = %10
%47 = load i64, i64* %8
store i64 %47, i64* %4
br label %48
48: ; preds = %46, %37
%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) {
%4 = icmp eq i64 %2, 0
br i1 %4, label %15, label %5
5: ; preds = %3, %11
%6 = phi i64 [ %12, %11 ], [ 0, %3 ]
%7 = phi i64 [ %13, %11 ], [ %2, %3 ]
%8 = getelementptr inbounds i8, i8* %1, i64 %6
%9 = tail call i64 @write(i32 %0, i8* %8, i64 %7)
%10 = icmp eq i64 %9, 0
br i1 %10, label %15, label %11
11: ; preds = %5
%12 = add i64 %9, %6
%13 = sub i64 %7, %9
%14 = icmp eq i64 %13, 0
br i1 %14, label %15, label %5
15: ; preds = %11, %5, %3
%16 = phi i64 [ 0, %3 ], [ %6, %5 ], [ %12, %11 ]
ret i64 %16
}