#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 int /*<<< orphan*/ AES_KEY;

/* Variables and functions */
scalar_t__ AES_set_decrypt_key(unsigned char const *, int,
                               int /*<<< orphan*/ *);
scalar_t__ AES_set_encrypt_key(unsigned char const *, int,
                               int /*<<< orphan*/ *);
int AES_unwrap_key(int /*<<< orphan*/ *, unsigned char const *, unsigned char *,
                   unsigned char *, int);
int AES_wrap_key(int /*<<< orphan*/ *, unsigned char const *, unsigned char *,
                 unsigned char const *, int);
int /*<<< orphan*/ free(unsigned char *);
unsigned char *malloc(int);
scalar_t__ memcmp(unsigned char const *, unsigned char *, int);
int AES_wrap_unwrap_test(const unsigned char *kek, int keybits,
                         const unsigned char *iv, const unsigned char *eout,
                         const unsigned char *key, int keylen) {
  unsigned char *otmp = NULL, *ptmp = NULL;
  int r, ret = 0;
  AES_KEY wctx;
  otmp = malloc(keylen + 8);
  ptmp = malloc(keylen);
  if (!otmp || !ptmp)
    return 0;
  if (AES_set_encrypt_key(kek, keybits, &wctx))
    goto err;
  r = AES_wrap_key(&wctx, iv, otmp, key, keylen);
  if (r <= 0)
    goto err;
  if (eout && memcmp(eout, otmp, keylen))
    goto err;
  if (AES_set_decrypt_key(kek, keybits, &wctx))
    goto err;
  r = AES_unwrap_key(&wctx, iv, ptmp, otmp, r);
  if (memcmp(key, ptmp, keylen))
    goto err;
  ret = 1;
err:
  free(otmp);
  free(ptmp);
  return ret;
}


x86 assembly (clang, -O0):
<AES_wrap_unwrap_test>:
push   %rbp
mov    %rsp,%rbp
sub    $0x60,%rsp
mov    %rdi,-0x10(%rbp)
mov    %esi,-0x14(%rbp)
mov    %rdx,-0x20(%rbp)
mov    %rcx,-0x28(%rbp)
mov    %r8,-0x30(%rbp)
mov    %r9d,-0x34(%rbp)
movq   $0x0,-0x40(%rbp)
movq   $0x0,-0x48(%rbp)
movl   $0x0,-0x50(%rbp)
mov    -0x34(%rbp),%edi
add    $0x8,%edi
call   1060 <malloc@plt>
mov    %rax,-0x40(%rbp)
mov    -0x34(%rbp),%edi
call   1060 <malloc@plt>
mov    %rax,-0x48(%rbp)
cmpq   $0x0,-0x40(%rbp)
je     11d7 <AES_wrap_unwrap_test+0x67>
cmpq   $0x0,-0x48(%rbp)
jne    11e3 <AES_wrap_unwrap_test+0x73>
movl   $0x0,-0x4(%rbp)
jmp    12d0 <AES_wrap_unwrap_test+0x160>
mov    -0x10(%rbp),%rdi
mov    -0x14(%rbp),%esi
lea    -0x54(%rbp),%rdx
call   1040 <AES_set_encrypt_key@plt>
cmp    $0x0,%rax
je     1202 <AES_wrap_unwrap_test+0x92>
jmp    12b8 <AES_wrap_unwrap_test+0x148>
mov    -0x20(%rbp),%rsi
mov    -0x40(%rbp),%rdx
mov    -0x30(%rbp),%rcx
mov    -0x34(%rbp),%r8d
lea    -0x54(%rbp),%rdi
call   1090 <AES_wrap_key@plt>
mov    %eax,-0x4c(%rbp)
cmpl   $0x0,-0x4c(%rbp)
jg     122d <AES_wrap_unwrap_test+0xbd>
jmp    12b8 <AES_wrap_unwrap_test+0x148>
cmpq   $0x0,-0x28(%rbp)
je     1257 <AES_wrap_unwrap_test+0xe7>
mov    -0x28(%rbp),%rdi
mov    -0x40(%rbp),%rsi
mov    -0x34(%rbp),%edx
call   1050 <memcmp@plt>
cmp    $0x0,%rax
je     1257 <AES_wrap_unwrap_test+0xe7>
jmp    12b8 <AES_wrap_unwrap_test+0x148>
mov    -0x10(%rbp),%rdi
mov    -0x14(%rbp),%esi
lea    -0x54(%rbp),%rdx
call   1070 <AES_set_decrypt_key@plt>
cmp    $0x0,%rax
je     1276 <AES_wrap_unwrap_test+0x106>
jmp    12b8 <AES_wrap_unwrap_test+0x148>
mov    -0x20(%rbp),%rsi
mov    -0x48(%rbp),%rdx
mov    -0x40(%rbp),%rcx
mov    -0x4c(%rbp),%r8d
lea    -0x54(%rbp),%rdi
call   1080 <AES_unwrap_key@plt>
mov    %eax,-0x4c(%rbp)
mov    -0x30(%rbp),%rdi
mov    -0x48(%rbp),%rsi
mov    -0x34(%rbp),%edx
call   1050 <memcmp@plt>
cmp    $0x0,%rax
je     12b1 <AES_wrap_unwrap_test+0x141>
jmp    12b8 <AES_wrap_unwrap_test+0x148>
movl   $0x1,-0x50(%rbp)
mov    -0x40(%rbp),%rdi
call   1030 <free@plt>
mov    -0x48(%rbp),%rdi
call   1030 <free@plt>
mov    -0x50(%rbp),%eax
mov    %eax,-0x4(%rbp)
mov    -0x4(%rbp),%eax
add    $0x60,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<AES_wrap_unwrap_test>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
sub    $0x28,%rsp
mov    %r9d,%ebx
mov    %r8,0x18(%rsp)
mov    %rcx,0x20(%rsp)
mov    %rdx,0x10(%rsp)
mov    %esi,%r12d
mov    %rdi,%r13
lea    0x8(%rbx),%edi
call   1060 <malloc@plt>
mov    %rax,%r15
mov    %ebx,%edi
call   1060 <malloc@plt>
xor    %ebp,%ebp
test   %r15,%r15
je     11de <AES_wrap_unwrap_test+0x6e>
mov    %rax,%r14
test   %rax,%rax
je     11de <AES_wrap_unwrap_test+0x6e>
lea    0x8(%rsp),%rdx
mov    %r13,%rdi
mov    %r12d,%esi
call   1040 <AES_set_encrypt_key@plt>
xor    %ebp,%ebp
test   %rax,%rax
je     11ef <AES_wrap_unwrap_test+0x7f>
mov    %r15,%rdi
call   1030 <free@plt>
mov    %r14,%rdi
call   1030 <free@plt>
mov    %ebp,%eax
add    $0x28,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret
lea    0x8(%rsp),%rdi
mov    0x10(%rsp),%rsi
mov    %r15,%rdx
mov    0x18(%rsp),%rcx
mov    %ebx,%r8d
call   1090 <AES_wrap_key@plt>
test   %eax,%eax
jle    11ce <AES_wrap_unwrap_test+0x5e>
mov    %eax,0xc(%rsp)
mov    0x20(%rsp),%rdi
test   %rdi,%rdi
je     122a <AES_wrap_unwrap_test+0xba>
mov    %r15,%rsi
mov    %ebx,%edx
call   1050 <memcmp@plt>
test   %rax,%rax
jne    11ce <AES_wrap_unwrap_test+0x5e>
lea    0x8(%rsp),%rdx
mov    %r13,%rdi
mov    %r12d,%esi
call   1070 <AES_set_decrypt_key@plt>
test   %rax,%rax
jne    11ce <AES_wrap_unwrap_test+0x5e>
lea    0x8(%rsp),%rdi
mov    0x10(%rsp),%rsi
mov    %r14,%rdx
mov    %r15,%rcx
mov    0xc(%rsp),%r8d
call   1080 <AES_unwrap_key@plt>
mov    0x18(%rsp),%rdi
mov    %r14,%rsi
mov    %ebx,%edx
call   1050 <memcmp@plt>
xor    %ebp,%ebp
test   %rax,%rax
sete   %bpl
jmp    11ce <AES_wrap_unwrap_test+0x5e>

LLVM IR (-O0):
define i32 @AES_wrap_unwrap_test(i8* %0, i32 %1, i8* %2, i8* %3, i8* %4, i32 %5) {
%7 = alloca i32
%8 = alloca i8*
%9 = alloca i32
%10 = alloca i8*
%11 = alloca i8*
%12 = alloca i8*
%13 = alloca i32
%14 = alloca i8*
%15 = alloca i8*
%16 = alloca i32
%17 = alloca i32
%18 = alloca i32
store i8* %0, i8** %8
store i32 %1, i32* %9
store i8* %2, i8** %10
store i8* %3, i8** %11
store i8* %4, i8** %12
store i32 %5, i32* %13
store i8* null, i8** %14
store i8* null, i8** %15
store i32 0, i32* %17
%19 = load i32, i32* %13
%20 = add nsw i32 %19, 8
%21 = call i8* @malloc(i32 %20)
store i8* %21, i8** %14
%22 = load i32, i32* %13
%23 = call i8* @malloc(i32 %22)
store i8* %23, i8** %15
%24 = load i8*, i8** %14
%25 = icmp ne i8* %24, null
br i1 %25, label %26, label %29
26: ; preds = %6
%27 = load i8*, i8** %15
%28 = icmp ne i8* %27, null
br i1 %28, label %30, label %29
29: ; preds = %26, %6
store i32 0, i32* %7
br label %80
30: ; preds = %26
%31 = load i8*, i8** %8
%32 = load i32, i32* %9
%33 = call i64 @AES_set_encrypt_key(i8* %31, i32 %32, i32* %18)
%34 = icmp ne i64 %33, 0
br i1 %34, label %35, label %36
35: ; preds = %30
br label %74
36: ; preds = %30
%37 = load i8*, i8** %10
%38 = load i8*, i8** %14
%39 = load i8*, i8** %12
%40 = load i32, i32* %13
%41 = call i32 @AES_wrap_key(i32* %18, i8* %37, i8* %38, i8* %39, i32 %40)
store i32 %41, i32* %16
%42 = load i32, i32* %16
%43 = icmp sle i32 %42, 0
br i1 %43, label %44, label %45
44: ; preds = %36
br label %74
45: ; preds = %36
%46 = load i8*, i8** %11
%47 = icmp ne i8* %46, null
br i1 %47, label %48, label %55
48: ; preds = %45
%49 = load i8*, i8** %11
%50 = load i8*, i8** %14
%51 = load i32, i32* %13
%52 = call i64 @memcmp(i8* %49, i8* %50, i32 %51)
%53 = icmp ne i64 %52, 0
br i1 %53, label %54, label %55
54: ; preds = %48
br label %74
55: ; preds = %48, %45
%56 = load i8*, i8** %8
%57 = load i32, i32* %9
%58 = call i64 @AES_set_decrypt_key(i8* %56, i32 %57, i32* %18)
%59 = icmp ne i64 %58, 0
br i1 %59, label %60, label %61
60: ; preds = %55
br label %74
61: ; preds = %55
%62 = load i8*, i8** %10
%63 = load i8*, i8** %15
%64 = load i8*, i8** %14
%65 = load i32, i32* %16
%66 = call i32 @AES_unwrap_key(i32* %18, i8* %62, i8* %63, i8* %64, i32 %65)
store i32 %66, i32* %16
%67 = load i8*, i8** %12
%68 = load i8*, i8** %15
%69 = load i32, i32* %13
%70 = call i64 @memcmp(i8* %67, i8* %68, i32 %69)
%71 = icmp ne i64 %70, 0
br i1 %71, label %72, label %73
72: ; preds = %61
br label %74
73: ; preds = %61
store i32 1, i32* %17
br label %74
74: ; preds = %73, %72, %60, %54, %44, %35
%75 = load i8*, i8** %14
%76 = call i32 @free(i8* %75)
%77 = load i8*, i8** %15
%78 = call i32 @free(i8* %77)
%79 = load i32, i32* %17
store i32 %79, i32* %7
br label %80
80: ; preds = %74, %29
%81 = load i32, i32* %7
ret i32 %81
}

LLVM IR (-O2):
define i32 @AES_wrap_unwrap_test(i8* %0, i32 %1, i8* %2, i8* %3, i8* %4, i32 %5) {
%7 = alloca i32
%8 = bitcast i32* %7 to i8*
%9 = add nsw i32 %5, 8
%10 = tail call i8* @malloc(i32 %9)
%11 = tail call i8* @malloc(i32 %5)
%12 = icmp ne i8* %10, null
%13 = icmp ne i8* %11, null
%14 = and i1 %12, %13
br i1 %14, label %15, label %38
15: ; preds = %6
%16 = call i64 @AES_set_encrypt_key(i8* %0, i32 %1, i32* %7)
%17 = icmp eq i64 %16, 0
br i1 %17, label %18, label %34
18: ; preds = %15
%19 = call i32 @AES_wrap_key(i32* %7, i8* %2, i8* %10, i8* %4, i32 %5)
%20 = icmp slt i32 %19, 1
br i1 %20, label %34, label %21
21: ; preds = %18
%22 = icmp eq i8* %3, null
br i1 %22, label %26, label %23
23: ; preds = %21
%24 = call i64 @memcmp(i8* %3, i8* %10, i32 %5)
%25 = icmp eq i64 %24, 0
br i1 %25, label %26, label %34
26: ; preds = %23, %21
%27 = call i64 @AES_set_decrypt_key(i8* %0, i32 %1, i32* %7)
%28 = icmp eq i64 %27, 0
br i1 %28, label %29, label %34
29: ; preds = %26
%30 = call i32 @AES_unwrap_key(i32* %7, i8* %2, i8* %11, i8* %10, i32 %19)
%31 = call i64 @memcmp(i8* %4, i8* %11, i32 %5)
%32 = icmp eq i64 %31, 0
%33 = zext i1 %32 to i32
br label %34
34: ; preds = %29, %26, %23, %18, %15
%35 = phi i32 [ 0, %15 ], [ 0, %18 ], [ 0, %23 ], [ 0, %26 ], [ %33, %29 ]
%36 = call i32 @free(i8* %10)
%37 = call i32 @free(i8* %11)
br label %38
38: ; preds = %6, %34
%39 = phi i32 [ %35, %34 ], [ 0, %6 ]
ret i32 %39
}