#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 /*<<< orphan*/ assert(int /*<<< orphan*/);
scalar_t__ memcmp(char const *, char const *, size_t const);
size_t strlen(char const *);
char *strnstr(const char *haystack, const char *needle, size_t len) {
  assert(needle != NULL);
  const size_t i = strlen(needle);
  if (i == 0)
    return (char *)haystack;
  if (len < i)
    return NULL;
  size_t count = len - i;
  do {
    if (memcmp(haystack, needle, i))
      return (char *)haystack;
    haystack++;
  } while (count--);
  return NULL;
}


x86 assembly (clang, -O0):
<strnstr>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %rdi,-0x10(%rbp)
mov    %rsi,-0x18(%rbp)
mov    %rdx,-0x20(%rbp)
cmpq   $0x0,-0x18(%rbp)
setne  %al
and    $0x1,%al
movzbl %al,%edi
call   1050 <assert@plt>
mov    -0x18(%rbp),%rdi
call   1030 <strlen@plt>
mov    %rax,-0x28(%rbp)
cmpq   $0x0,-0x28(%rbp)
jne    117b <strnstr+0x4b>
mov    -0x10(%rbp),%rax
mov    %rax,-0x8(%rbp)
jmp    11f7 <strnstr+0xc7>
mov    -0x20(%rbp),%rax
cmp    -0x28(%rbp),%rax
jae    1196 <strnstr+0x66>
movq   $0x0,-0x8(%rbp)
jmp    11f7 <strnstr+0xc7>
mov    -0x20(%rbp),%rax
sub    -0x28(%rbp),%rax
mov    %rax,-0x30(%rbp)
mov    -0x10(%rbp),%rdi
mov    -0x18(%rbp),%rsi
mov    -0x28(%rbp),%rdx
call   1040 <memcmp@plt>
cmp    $0x0,%rax
je     11ca <strnstr+0x9a>
mov    -0x10(%rbp),%rax
mov    %rax,-0x8(%rbp)
jmp    11f7 <strnstr+0xc7>
mov    -0x10(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x10(%rbp)
mov    -0x30(%rbp),%rax
mov    %rax,%rcx
add    $0xffffffffffffffff,%rcx
mov    %rcx,-0x30(%rbp)
cmp    $0x0,%rax
jne    11a2 <strnstr+0x72>
movq   $0x0,-0x8(%rbp)
mov    -0x8(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<strnstr>:
push   %r15
push   %r14
push   %r12
push   %rbx
push   %rax
mov    %rdx,%rbx
mov    %rsi,%r15
mov    %rdi,%r14
xor    %edi,%edi
test   %rsi,%rsi
setne  %dil
call   1050 <assert@plt>
mov    %r15,%rdi
call   1030 <strlen@plt>
test   %rax,%rax
je     1193 <strnstr+0x63>
mov    %rax,%r12
cmp    %rbx,%rax
ja     1190 <strnstr+0x60>
add    $0x1,%rbx
nopl   0x0(%rax,%rax,1)
mov    %r14,%rdi
mov    %r15,%rsi
mov    %r12,%rdx
call   1040 <memcmp@plt>
test   %rax,%rax
jne    1193 <strnstr+0x63>
add    $0x1,%r14
add    $0xffffffffffffffff,%rbx
cmp    %rbx,%r12
jne    1170 <strnstr+0x40>
xor    %r14d,%r14d
mov    %r14,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
ret

LLVM IR (-O0):
define i8* @strnstr(i8* %0, i8* %1, i64 %2) {
%4 = alloca i8*
%5 = alloca i8*
%6 = alloca i8*
%7 = alloca i64
%8 = alloca i64
%9 = alloca i64
store i8* %0, i8** %5
store i8* %1, i8** %6
store i64 %2, i64* %7
%10 = load i8*, i8** %6
%11 = icmp ne i8* %10, null
%12 = zext i1 %11 to i32
%13 = call i32 @assert(i32 %12)
%14 = load i8*, i8** %6
%15 = call i64 @strlen(i8* %14)
store i64 %15, i64* %8
%16 = load i64, i64* %8
%17 = icmp eq i64 %16, 0
br i1 %17, label %18, label %20
18: ; preds = %3
%19 = load i8*, i8** %5
store i8* %19, i8** %4
br label %45
20: ; preds = %3
%21 = load i64, i64* %7
%22 = load i64, i64* %8
%23 = icmp ult i64 %21, %22
br i1 %23, label %24, label %25
24: ; preds = %20
store i8* null, i8** %4
br label %45
25: ; preds = %20
%26 = load i64, i64* %7
%27 = load i64, i64* %8
%28 = sub i64 %26, %27
store i64 %28, i64* %9
br label %29
29: ; preds = %40, %25
%30 = load i8*, i8** %5
%31 = load i8*, i8** %6
%32 = load i64, i64* %8
%33 = call i64 @memcmp(i8* %30, i8* %31, i64 %32)
%34 = icmp ne i64 %33, 0
br i1 %34, label %35, label %37
35: ; preds = %29
%36 = load i8*, i8** %5
store i8* %36, i8** %4
br label %45
37: ; preds = %29
%38 = load i8*, i8** %5
%39 = getelementptr inbounds i8, i8* %38, i32 1
store i8* %39, i8** %5
br label %40
40: ; preds = %37
%41 = load i64, i64* %9
%42 = add i64 %41, -1
store i64 %42, i64* %9
%43 = icmp ne i64 %41, 0
br i1 %43, label %29, label %44
44: ; preds = %40
store i8* null, i8** %4
br label %45
45: ; preds = %44, %35, %24, %18
%46 = load i8*, i8** %4
ret i8* %46
}

LLVM IR (-O2):
define i8* @strnstr(i8* %0, i8* %1, i64 %2) {
%4 = icmp ne i8* %1, null
%5 = zext i1 %4 to i32
%6 = tail call i32 @assert(i32 %5)
%7 = tail call i64 @strlen(i8* dereferenceable(1) %1)
%8 = icmp eq i64 %7, 0
br i1 %8, label %22, label %9
9: ; preds = %3
%10 = icmp ugt i64 %7, %2
br i1 %10, label %22, label %11
11: ; preds = %9
%12 = sub i64 %2, %7
br label %13
13: ; preds = %18, %11
%14 = phi i8* [ %0, %11 ], [ %19, %18 ]
%15 = phi i64 [ %12, %11 ], [ %20, %18 ]
%16 = tail call i64 @memcmp(i8* %14, i8* %1, i64 %7)
%17 = icmp eq i64 %16, 0
br i1 %17, label %18, label %22
18: ; preds = %13
%19 = getelementptr inbounds i8, i8* %14, i64 1
%20 = add i64 %15, -1
%21 = icmp eq i64 %15, 0
br i1 %21, label %22, label %13
22: ; preds = %13, %18, %9, %3
%23 = phi i8* [ %0, %3 ], [ null, %9 ], [ null, %18 ], [ %14, %13 ]
ret i8* %23
}