#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 */
scalar_t__ MemAlloc(int);
char *wstrndup(char *str, int len) {
  char *s, *p;
  if (str == 0 || len < 0)
    return 0;
  s = (char *)MemAlloc(sizeof(char) * (1 + len));
  p = s;
  while (len-- > 0 && (*p++ = *str++))
    ;
  *p = '\0';
  return s;
}


x86 assembly (clang, -O0):
<wstrndup>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %rdi,-0x10(%rbp)
mov    %esi,-0x14(%rbp)
cmpq   $0x0,-0x10(%rbp)
je     1134 <wstrndup+0x24>
cmpl   $0x0,-0x14(%rbp)
jge    1141 <wstrndup+0x31>
movq   $0x0,-0x8(%rbp)
jmp    11cb <wstrndup+0xbb>
mov    -0x14(%rbp),%eax
add    $0x1,%eax
cltq
shl    $0x0,%rax
mov    %eax,%edi
call   1030 <MemAlloc@plt>
mov    %rax,-0x20(%rbp)
mov    -0x20(%rbp),%rax
mov    %rax,-0x28(%rbp)
mov    -0x14(%rbp),%ecx
mov    %ecx,%eax
add    $0xffffffff,%eax
mov    %eax,-0x14(%rbp)
xor    %eax,%eax
cmp    $0x0,%ecx
mov    %al,-0x29(%rbp)
jle    11a7 <wstrndup+0x97>
mov    -0x10(%rbp),%rax
mov    %rax,%rcx
add    $0x1,%rcx
mov    %rcx,-0x10(%rbp)
mov    (%rax),%al
mov    -0x28(%rbp),%rcx
mov    %rcx,%rdx
add    $0x1,%rdx
mov    %rdx,-0x28(%rbp)
mov    %al,(%rcx)
movsbl %al,%eax
cmp    $0x0,%eax
setne  %al
mov    %al,-0x29(%rbp)
mov    -0x29(%rbp),%al
test   $0x1,%al
jne    11b7 <wstrndup+0xa7>
jmp    11bc <wstrndup+0xac>
jmp    1160 <wstrndup+0x50>
mov    -0x28(%rbp),%rax
movb   $0x0,(%rax)
mov    -0x20(%rbp),%rax
mov    %rax,-0x8(%rbp)
mov    -0x8(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<wstrndup>:
push   %r14
push   %rbx
push   %rax
xor    %eax,%eax
test   %rdi,%rdi
je     1150 <wstrndup+0x40>
mov    %esi,%ebx
test   %esi,%esi
js     1150 <wstrndup+0x40>
mov    %rdi,%r14
lea    0x1(%rbx),%edi
call   1030 <MemAlloc@plt>
xor    %ecx,%ecx
xchg   %ax,%ax
test   %ebx,%ebx
jle    1147 <wstrndup+0x37>
add    $0xffffffff,%ebx
movzbl (%r14,%rcx,1),%edx
mov    %dl,(%rax,%rcx,1)
add    $0x1,%rcx
test   %dl,%dl
jne    1130 <wstrndup+0x20>
mov    %rax,%rdx
add    %rcx,%rdx
movb   $0x0,(%rdx)
add    $0x8,%rsp
pop    %rbx
pop    %r14
ret

LLVM IR (-O0):
define i8* @wstrndup(i8* %0, i32 %1) {
%3 = alloca i8*
%4 = alloca i8*
%5 = alloca i32
%6 = alloca i8*
%7 = alloca i8*
store i8* %0, i8** %4
store i32 %1, i32* %5
%8 = load i8*, i8** %4
%9 = icmp eq i8* %8, null
br i1 %9, label %13, label %10
10: ; preds = %2
%11 = load i32, i32* %5
%12 = icmp slt i32 %11, 0
br i1 %12, label %13, label %14
13: ; preds = %10, %2
store i8* null, i8** %3
br label %41
14: ; preds = %10
%15 = load i32, i32* %5
%16 = add nsw i32 1, %15
%17 = sext i32 %16 to i64
%18 = mul i64 1, %17
%19 = trunc i64 %18 to i32
%20 = call i64 @MemAlloc(i32 %19)
%21 = inttoptr i64 %20 to i8*
store i8* %21, i8** %6
%22 = load i8*, i8** %6
store i8* %22, i8** %7
br label %23
23: ; preds = %37, %14
%24 = load i32, i32* %5
%25 = add nsw i32 %24, -1
store i32 %25, i32* %5
%26 = icmp sgt i32 %24, 0
br i1 %26, label %27, label %35
27: ; preds = %23
%28 = load i8*, i8** %4
%29 = getelementptr inbounds i8, i8* %28, i32 1
store i8* %29, i8** %4
%30 = load i8, i8* %28
%31 = load i8*, i8** %7
%32 = getelementptr inbounds i8, i8* %31, i32 1
store i8* %32, i8** %7
store i8 %30, i8* %31
%33 = sext i8 %30 to i32
%34 = icmp ne i32 %33, 0
br label %35
35: ; preds = %27, %23
%36 = phi i1 [ false, %23 ], [ %34, %27 ]
br i1 %36, label %37, label %38
37: ; preds = %35
br label %23
38: ; preds = %35
%39 = load i8*, i8** %7
store i8 0, i8* %39
%40 = load i8*, i8** %6
store i8* %40, i8** %3
br label %41
41: ; preds = %38, %13
%42 = load i8*, i8** %3
ret i8* %42
}

LLVM IR (-O2):
define i8* @wstrndup(i8* %0, i32 %1) {
%3 = icmp eq i8* %0, null
%4 = icmp slt i32 %1, 0
%5 = or i1 %3, %4
br i1 %5, label %23, label %6
6: ; preds = %2
%7 = add nuw nsw i32 %1, 1
%8 = tail call i64 @MemAlloc(i32 %7)
%9 = inttoptr i64 %8 to i8*
br label %10
10: ; preds = %15, %6
%11 = phi i8* [ %0, %6 ], [ %17, %15 ]
%12 = phi i32 [ %1, %6 ], [ %16, %15 ]
%13 = phi i8* [ %9, %6 ], [ %19, %15 ]
%14 = icmp sgt i32 %12, 0
br i1 %14, label %15, label %21
15: ; preds = %10
%16 = add nsw i32 %12, -1
%17 = getelementptr inbounds i8, i8* %11, i64 1
%18 = load i8, i8* %11
%19 = getelementptr inbounds i8, i8* %13, i64 1
store i8 %18, i8* %13
%20 = icmp eq i8 %18, 0
br i1 %20, label %21, label %10
21: ; preds = %10, %15
%22 = phi i8* [ %19, %15 ], [ %13, %10 ]
store i8 0, i8* %22
br label %23
23: ; preds = %2, %21
%24 = phi i8* [ %9, %21 ], [ null, %2 ]
ret i8* %24
}