#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 */
#define EINTR 128
int errno;
scalar_t__ write(int, void const *, size_t);
ssize_t safe_write(int fd, const void *buf, size_t count) {
  ssize_t res, sum = 0;
  for (;;) {
    res = write(fd, buf, count);
    if (res >= 0) {
      sum += res;
      if (res == 0 || (size_t)res == count)
        return sum;
      count -= res;
      buf = (const char *)buf + res;
      continue;
    }
    switch (errno) {
    case EINTR:
      continue;
    default:
      return -1;
    }
  }
}


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,-0x30(%rbp)
mov    -0xc(%rbp),%edi
mov    -0x18(%rbp),%rsi
mov    -0x20(%rbp),%rdx
call   1030 <write@plt>
mov    %rax,-0x28(%rbp)
cmpq   $0x0,-0x28(%rbp)
jl     119c <safe_write+0x8c>
mov    -0x28(%rbp),%rax
add    -0x30(%rbp),%rax
mov    %rax,-0x30(%rbp)
cmpq   $0x0,-0x28(%rbp)
je     116f <safe_write+0x5f>
mov    -0x28(%rbp),%rax
cmp    -0x20(%rbp),%rax
jne    117c <safe_write+0x6c>
mov    -0x30(%rbp),%rax
mov    %rax,-0x8(%rbp)
jmp    11c2 <safe_write+0xb2>
mov    -0x28(%rbp),%rcx
mov    -0x20(%rbp),%rax
sub    %rcx,%rax
mov    %rax,-0x20(%rbp)
mov    -0x18(%rbp),%rax
add    -0x28(%rbp),%rax
mov    %rax,-0x18(%rbp)
jmp    112b <safe_write+0x1b>
mov    0x2e3d(%rip),%rax
mov    (%rax),%eax
sub    $0x80,%eax
jne    11ba <safe_write+0xaa>
jmp    11b5 <safe_write+0xa5>
jmp    112b <safe_write+0x1b>
movq   $0xffffffffffffffff,-0x8(%rbp)
ff
mov    -0x8(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<safe_write>:
push   %rbp
push   %r15
push   %r14
push   %r12
push   %rbx
mov    %rdx,%rbx
mov    %rsi,%r15
mov    %edi,%ebp
xor    %r14d,%r14d
mov    0x2eb6(%rip),%r12
nopw   0x0(%rax,%rax,1)
mov    %ebp,%edi
mov    %r15,%rsi
mov    %rbx,%rdx
call   1030 <write@plt>
test   %rax,%rax
jns    114e <safe_write+0x3e>
cmpl   $0x80,(%r12)
je     1130 <safe_write+0x20>
jmp    1160 <safe_write+0x50>
add    %rax,%r14
test   %rax,%rax
je     1167 <safe_write+0x57>
sub    %rax,%rbx
je     1167 <safe_write+0x57>
add    %rax,%r15
jmp    1130 <safe_write+0x20>
mov    $0xffffffffffffffff,%r14
mov    %r14,%rax
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
pop    %rbp
ret

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

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