#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*/ uint8_t;
typedef int uint64_t;

/* Variables and functions */
int /*<<< orphan*/ AESEncrypt(unsigned int const *, unsigned int *);
void jambu_tag_generation(const unsigned int *rk, uint64_t msglen, uint8_t *c,
                          unsigned int *stateS, unsigned int *stateR) {
  AESEncrypt(rk, stateS);
  stateS[0] ^= stateR[0] ^ 0x03;
  stateS[1] ^= stateR[1];
  stateR[0] ^= stateS[2];
  stateR[1] ^= stateS[3];
  AESEncrypt(rk, stateS);
  ((unsigned int *)(c + msglen))[0] = stateS[0] ^ stateS[2] ^ stateR[0];
  ((unsigned int *)(c + msglen))[1] = stateS[1] ^ stateS[3] ^ stateR[1];
}


x86 assembly (clang, -O0):
<jambu_tag_generation>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %rcx,-0x20(%rbp)
mov    %r8,-0x28(%rbp)
mov    -0x8(%rbp),%rdi
mov    -0x20(%rbp),%rsi
call   1030 <AESEncrypt@plt>
mov    -0x28(%rbp),%rax
mov    (%rax),%ecx
xor    $0x3,%ecx
mov    -0x20(%rbp),%rax
xor    (%rax),%ecx
mov    %ecx,(%rax)
mov    -0x28(%rbp),%rax
mov    0x4(%rax),%ecx
mov    -0x20(%rbp),%rax
xor    0x4(%rax),%ecx
mov    %ecx,0x4(%rax)
mov    -0x20(%rbp),%rax
mov    0x8(%rax),%ecx
mov    -0x28(%rbp),%rax
xor    (%rax),%ecx
mov    %ecx,(%rax)
mov    -0x20(%rbp),%rax
mov    0xc(%rax),%ecx
mov    -0x28(%rbp),%rax
xor    0x4(%rax),%ecx
mov    %ecx,0x4(%rax)
mov    -0x8(%rbp),%rdi
mov    -0x20(%rbp),%rsi
call   1030 <AESEncrypt@plt>
mov    -0x20(%rbp),%rax
mov    (%rax),%edx
mov    -0x20(%rbp),%rax
xor    0x8(%rax),%edx
mov    -0x28(%rbp),%rax
xor    (%rax),%edx
mov    -0x18(%rbp),%rax
movslq -0xc(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x20(%rbp),%rax
mov    0x4(%rax),%edx
mov    -0x20(%rbp),%rax
xor    0xc(%rax),%edx
mov    -0x28(%rbp),%rax
xor    0x4(%rax),%edx
mov    -0x18(%rbp),%rax
movslq -0xc(%rbp),%rcx
mov    %edx,0x4(%rax,%rcx,4)
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<jambu_tag_generation>:
push   %rbp
push   %r15
push   %r14
push   %r12
push   %rbx
mov    %r8,%r15
mov    %rcx,%rbx
mov    %rdx,%r14
mov    %esi,%r12d
mov    %rdi,%rbp
mov    %rcx,%rsi
call   1030 <AESEncrypt@plt>
mov    (%r15),%eax
xor    (%rbx),%eax
xor    $0x3,%eax
mov    %eax,(%rbx)
mov    0x4(%r15),%eax
xor    %eax,0x4(%rbx)
mov    0x8(%rbx),%eax
xor    %eax,(%r15)
mov    0xc(%rbx),%eax
xor    %eax,0x4(%r15)
mov    %rbp,%rdi
mov    %rbx,%rsi
call   1030 <AESEncrypt@plt>
mov    0x8(%rbx),%eax
xor    (%rbx),%eax
xor    (%r15),%eax
movslq %r12d,%rcx
mov    %eax,(%r14,%rcx,4)
mov    0xc(%rbx),%eax
xor    0x4(%rbx),%eax
xor    0x4(%r15),%eax
mov    %eax,0x4(%r14,%rcx,4)
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define void @jambu_tag_generation(i32* %0, i32 %1, i32* %2, i32* %3, i32* %4) {
%6 = alloca i32*
%7 = alloca i32
%8 = alloca i32*
%9 = alloca i32*
%10 = alloca i32*
store i32* %0, i32** %6
store i32 %1, i32* %7
store i32* %2, i32** %8
store i32* %3, i32** %9
store i32* %4, i32** %10
%11 = load i32*, i32** %6
%12 = load i32*, i32** %9
%13 = call i32 @AESEncrypt(i32* %11, i32* %12)
%14 = load i32*, i32** %10
%15 = getelementptr inbounds i32, i32* %14, i64 0
%16 = load i32, i32* %15
%17 = xor i32 %16, 3
%18 = load i32*, i32** %9
%19 = getelementptr inbounds i32, i32* %18, i64 0
%20 = load i32, i32* %19
%21 = xor i32 %20, %17
store i32 %21, i32* %19
%22 = load i32*, i32** %10
%23 = getelementptr inbounds i32, i32* %22, i64 1
%24 = load i32, i32* %23
%25 = load i32*, i32** %9
%26 = getelementptr inbounds i32, i32* %25, i64 1
%27 = load i32, i32* %26
%28 = xor i32 %27, %24
store i32 %28, i32* %26
%29 = load i32*, i32** %9
%30 = getelementptr inbounds i32, i32* %29, i64 2
%31 = load i32, i32* %30
%32 = load i32*, i32** %10
%33 = getelementptr inbounds i32, i32* %32, i64 0
%34 = load i32, i32* %33
%35 = xor i32 %34, %31
store i32 %35, i32* %33
%36 = load i32*, i32** %9
%37 = getelementptr inbounds i32, i32* %36, i64 3
%38 = load i32, i32* %37
%39 = load i32*, i32** %10
%40 = getelementptr inbounds i32, i32* %39, i64 1
%41 = load i32, i32* %40
%42 = xor i32 %41, %38
store i32 %42, i32* %40
%43 = load i32*, i32** %6
%44 = load i32*, i32** %9
%45 = call i32 @AESEncrypt(i32* %43, i32* %44)
%46 = load i32*, i32** %9
%47 = getelementptr inbounds i32, i32* %46, i64 0
%48 = load i32, i32* %47
%49 = load i32*, i32** %9
%50 = getelementptr inbounds i32, i32* %49, i64 2
%51 = load i32, i32* %50
%52 = xor i32 %48, %51
%53 = load i32*, i32** %10
%54 = getelementptr inbounds i32, i32* %53, i64 0
%55 = load i32, i32* %54
%56 = xor i32 %52, %55
%57 = load i32*, i32** %8
%58 = load i32, i32* %7
%59 = sext i32 %58 to i64
%60 = getelementptr inbounds i32, i32* %57, i64 %59
%61 = getelementptr inbounds i32, i32* %60, i64 0
store i32 %56, i32* %61
%62 = load i32*, i32** %9
%63 = getelementptr inbounds i32, i32* %62, i64 1
%64 = load i32, i32* %63
%65 = load i32*, i32** %9
%66 = getelementptr inbounds i32, i32* %65, i64 3
%67 = load i32, i32* %66
%68 = xor i32 %64, %67
%69 = load i32*, i32** %10
%70 = getelementptr inbounds i32, i32* %69, i64 1
%71 = load i32, i32* %70
%72 = xor i32 %68, %71
%73 = load i32*, i32** %8
%74 = load i32, i32* %7
%75 = sext i32 %74 to i64
%76 = getelementptr inbounds i32, i32* %73, i64 %75
%77 = getelementptr inbounds i32, i32* %76, i64 1
store i32 %72, i32* %77
ret void
}

LLVM IR (-O2):
define void @jambu_tag_generation(i32* %0, i32 %1, i32* nocapture writeonly %2, i32* %3, i32* nocapture %4) {
%6 = tail call i32 @AESEncrypt(i32* %0, i32* %3)
%7 = load i32, i32* %4
%8 = load i32, i32* %3
%9 = xor i32 %7, %8
%10 = xor i32 %9, 3
store i32 %10, i32* %3
%11 = getelementptr inbounds i32, i32* %4, i64 1
%12 = load i32, i32* %11
%13 = getelementptr inbounds i32, i32* %3, i64 1
%14 = load i32, i32* %13
%15 = xor i32 %14, %12
store i32 %15, i32* %13
%16 = getelementptr inbounds i32, i32* %3, i64 2
%17 = load i32, i32* %16
%18 = load i32, i32* %4
%19 = xor i32 %18, %17
store i32 %19, i32* %4
%20 = getelementptr inbounds i32, i32* %3, i64 3
%21 = load i32, i32* %20
%22 = load i32, i32* %11
%23 = xor i32 %22, %21
store i32 %23, i32* %11
%24 = tail call i32 @AESEncrypt(i32* %0, i32* %3)
%25 = load i32, i32* %3
%26 = load i32, i32* %16
%27 = xor i32 %26, %25
%28 = load i32, i32* %4
%29 = xor i32 %27, %28
%30 = sext i32 %1 to i64
%31 = getelementptr inbounds i32, i32* %2, i64 %30
store i32 %29, i32* %31
%32 = load i32, i32* %13
%33 = load i32, i32* %20
%34 = xor i32 %33, %32
%35 = load i32, i32* %11
%36 = xor i32 %34, %35
%37 = getelementptr inbounds i32, i32* %31, i64 1
store i32 %36, i32* %37
ret void
}