// File: /workspace/github_c_makefile_star2_O2/mjosaarinen[P]brutus/crypto_aead/aes128poetv2aes128ni/ref/poet.c
// Function: encrypt_final
void encrypt_final(poet_ctx_t *ctx,
                   const unsigned char *plaintext,
                   unsigned long long plen,
                   unsigned char *ciphertext,
                   unsigned long long* clen, 
                   unsigned char tag[TAGLEN])
{   
    __m128i x = aes_encrypt(ctx->x, ctx->aes_axu);
    __m128i y = ctx->y;
    __m128i z;
    const uint64_t plaintext_len = plen;

    if (clen) {
        *clen = plen;
    }
    
    if (plen >= BLOCKLEN) {
        z = loadu(plaintext);
        x = vxor(x, z);
        z = x;
        xor_3blocks(x, y, z, ctx->aes_axu[0], ctx->aes_axu[0], ctx->aes_enc[0]);
        plaintext += BLOCKLEN;
    }

    __m128i tmp_x;
    __m128i tmp_z;
    __m128i tmp_y;

    while (plen > 2*BLOCKLEN) {
        aes_encrypt_3blocks(x, y, z, ctx->aes_enc, ctx->aes_axu, plaintext, ciphertext, tmp_x, tmp_y, tmp_z);
        plaintext += BLOCKLEN;
        ciphertext += BLOCKLEN;
        plen -= BLOCKLEN;
    }

    const __m128i s = aes_encrypt(encode_length(plaintext_len), ctx->aes_enc);

    if (plen == 0) { // Empty message
        z = vxor3(x, s, ctx->tau);
        aes_encrypt_2blocks(y, z, ctx->aes_axu, ctx->aes_enc);
        z = vxor(z, y);

        z = vxor(z, s); 
        storeu(tag, z);
    } else if (plen % BLOCKLEN == 0) { // No tag splitting
        z = vxor(z, s);
        x = vxor(x, s);
        aes_encrypt_3blocks_no_whitening(x, y, z, ctx->aes_enc, ctx->aes_axu);
        storeu(ciphertext, vxor3(z, y, s));

        // Generate tag
        y = z;
        z = vxor(x, ctx->tau);
        aes_encrypt_2blocks(y, z, ctx->aes_axu, ctx->aes_enc);
        z = vxor3(z, y, ctx->tau);
        storeu(tag, z);
    } else { // Tag splitting
        if (plen > BLOCKLEN) {
            aes_encrypt_3blocks_no_whitening(x, y, z, ctx->aes_enc, ctx->aes_axu);
            storeu(ciphertext, vxor(z, y));
            ciphertext += BLOCKLEN;
            y = z; 
        }

        plen = plen % BLOCKLEN;
        z = zero_pad(load_n_bytes(plaintext, plen), BLOCKLEN - plen);
        z = vxor3(x, s, vor(z, shift_left(ctx->tau, plen)));
        x = z;

        xor_3blocks(x, y, z, ctx->aes_axu[0], ctx->aes_axu[0], ctx->aes_enc[0]);
        aes_encrypt_3blocks_no_whitening(x, y, z, ctx->aes_enc, ctx->aes_axu);

        tmp_z = vxor3(z, s, y);
        store_partial_left(ciphertext, tmp_z, plen);
        store_partial_right(tag, tmp_z, BLOCKLEN - plen);

        // Generate tag
        y = z; 
        z = vxor(x, ctx->tau);
        xor_3blocks(x, y, z, ctx->aes_axu[0], ctx->aes_axu[0], ctx->aes_enc[0]);
        aes_encrypt_3blocks_no_whitening(x, y, z, ctx->aes_enc, ctx->aes_axu);

        z = vxor3(z, ctx->tau, y);
        store_partial_from_left_to_right(tag, z, plen);
    }
}

// Assembly:
pushq %rbp
pushq %r15
pushq %r14
pushq %r13
pushq %r12
pushq %rbx
subq $0xf8, %rsp
movq %r9, %rbx
movq %r8, %rbp
movq %rcx, %r15
movq %rdx, %r13
movq %rsi, %r12
movq %rdi, %r14
vmovdqa 0x220(%rdi), %xmm0
addq $0x160, %rdi            # imm = 0x160
callq 0x1724
vmovdqa 0x230(%r14), %xmm5
testq %rbp, %rbp
je 0x1bc5
movq %r13, (%rbp)
cmpq $0x10, %r13
jb 0x1bed
vpxor (%r12), %xmm0, %xmm2
vmovdqa 0x160(%r14), %xmm1
vpxor %xmm1, %xmm2, %xmm0
vpxor %xmm5, %xmm1, %xmm5
vpxor (%r14), %xmm2, %xmm8
addq $0x10, %r12
jmp 0x1bed
movq %r13, %rbp
cmpq $0x21, %rbp
jb 0x1d38
vmovdqa 0x160(%r14), %xmm7
vmovdqa 0x170(%r14), %xmm1
vmovdqa 0x180(%r14), %xmm2
vmovdqa 0x190(%r14), %xmm3
vaesenc %xmm1, %xmm0, %xmm4
vaesenc %xmm1, %xmm5, %xmm1
vaesenc 0x10(%r14), %xmm8, %xmm5
vaesenc %xmm2, %xmm4, %xmm4
vaesenc 0x20(%r14), %xmm5, %xmm5
vaesenc %xmm2, %xmm1, %xmm1
vaesenc %xmm3, %xmm4, %xmm2
vaesenc 0x30(%r14), %xmm5, %xmm4
vmovdqa 0x1a0(%r14), %xmm5
vaesenc %xmm3, %xmm1, %xmm1
vaesenc %xmm5, %xmm2, %xmm2
vaesenc 0x40(%r14), %xmm4, %xmm3
vmovdqa 0x1b0(%r14), %xmm4
vaesenc %xmm5, %xmm1, %xmm1
vaesenc %xmm4, %xmm2, %xmm2
vaesenc 0x50(%r14), %xmm3, %xmm3
vmovdqa 0x1c0(%r14), %xmm5
vaesenc %xmm4, %xmm1, %xmm1
vaesenc %xmm5, %xmm2, %xmm2
vaesenc 0x60(%r14), %xmm3, %xmm3
vmovdqa 0x1d0(%r14), %xmm4
vaesenc %xmm5, %xmm1, %xmm1
vaesenc %xmm4, %xmm2, %xmm2
vaesenc 0x70(%r14), %xmm3, %xmm3
vmovdqa 0x1e0(%r14), %xmm5
vaesenc %xmm4, %xmm1, %xmm1
vaesenc %xmm5, %xmm2, %xmm2
vaesenc 0x80(%r14), %xmm3, %xmm3
vpxor 0x200(%r14), %xmm7, %xmm4
vpxor (%r12), %xmm4, %xmm6
vpxor 0xa0(%r14), %xmm7, %xmm0
vmovdqa 0x1f0(%r14), %xmm7
vaesenc %xmm5, %xmm1, %xmm1
vaesenc %xmm7, %xmm2, %xmm2
vaesenc %xmm7, %xmm1, %xmm1
vaesenc 0x90(%r14), %xmm3, %xmm3
vaesenclast %xmm0, %xmm3, %xmm5
vaesenclast %xmm4, %xmm1, %xmm0
vpxor %xmm0, %xmm5, %xmm0
vmovdqu %xmm0, (%r15)
vmovdqa (%r14), %xmm8
vaesenclast %xmm6, %xmm2, %xmm0
vpternlogq $0x96, 0x160(%r14), %xmm0, %xmm8
addq $0x10, %r12
addq $0x10, %r15
addq $-0x10, %rbp
jmp 0x1bf0
vmovdqa %xmm8, 0x20(%rsp)
vmovdqa %xmm0, (%rsp)
vmovdqa %xmm5, 0x10(%rsp)
shlq $0x3, %r13
vmovq %r13, %xmm0
movq %r14, %rdi
callq 0x1724
vmovdqa %xmm0, %xmm6
testq %rbp, %rbp
je 0x21cc
movq %rbp, %r13
andq $0xf, %r13
je 0x22a0
cmpq $0x11, %rbp
vmovdqa 0x20(%rsp), %xmm7
vmovdqa %xmm6, 0x30(%rsp)
jb 0x1eb0
vmovdqa 0x170(%r14), %xmm0
vmovdqa 0x180(%r14), %xmm1
vmovdqa 0x190(%r14), %xmm2
vmovdqa 0x1a0(%r14), %xmm3
vmovdqa (%rsp), %xmm4
vaesenc %xmm0, %xmm4, %xmm4
vmovdqa 0x10(%rsp), %xmm5
vaesenc %xmm0, %xmm5, %xmm0
vaesenc 0x10(%r14), %xmm7, %xmm5
vaesenc %xmm1, %xmm4, %xmm4
vaesenc %xmm1, %xmm0, %xmm0
vaesenc 0x20(%r14), %xmm5, %xmm1
vaesenc %xmm2, %xmm4, %xmm4
vaesenc %xmm2, %xmm0, %xmm0
vaesenc 0x30(%r14), %xmm1, %xmm1
vaesenc %xmm3, %xmm4, %xmm2
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x40(%r14), %xmm1, %xmm1
vmovdqa 0x1b0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x50(%r14), %xmm1, %xmm1
vmovdqa 0x1c0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x60(%r14), %xmm1, %xmm1
vmovdqa 0x1d0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x70(%r14), %xmm1, %xmm1
vmovdqa 0x1e0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x80(%r14), %xmm1, %xmm1
vmovdqa 0x1f0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x90(%r14), %xmm1, %xmm1
vmovdqa 0x200(%r14), %xmm3
vaesenclast %xmm3, %xmm2, %xmm2
vmovdqa %xmm2, (%rsp)
vaesenclast %xmm3, %xmm0, %xmm0
vaesenclast 0xa0(%r14), %xmm1, %xmm1
vmovdqa %xmm1, 0x10(%rsp)
vpxor %xmm0, %xmm1, %xmm0
vmovdqu %xmm0, (%r15)
addq $0x10, %r15
movq %r12, %rdi
movq %r13, %rsi
callq 0x1793
vmovdqa %xmm0, 0x20(%rsp)
pushq $0x10
popq %r12
subq %r13, %r12
leaq 0x1161(%rip), %rbp      # 0x3030
vmovdqa 0x240(%r14), %xmm0
movq %r13, %rdi
callq 0x2495
vmovdqa 0x20(%rsp), %xmm1
vpternlogq $0xf8, (%r12,%rbp), %xmm1, %xmm0
vmovdqa 0x30(%rsp), %xmm2
vpternlogq $0x96, (%rsp), %xmm2, %xmm0
vmovdqa 0x160(%r14), %xmm1
vmovdqa 0x170(%r14), %xmm4
vmovdqa 0x180(%r14), %xmm5
vmovdqa 0x190(%r14), %xmm6
vmovdqa %xmm1, (%rsp)
vpxor 0x10(%rsp), %xmm1, %xmm3
vmovdqa %xmm0, 0x10(%rsp)
vpxor (%r14), %xmm0, %xmm1
vmovdqa %xmm4, 0x20(%rsp)
vaesenc %xmm4, %xmm3, %xmm0
vaesenc 0x10(%r14), %xmm1, %xmm1
vmovdqa %xmm5, 0xe0(%rsp)
vaesenc %xmm5, %xmm0, %xmm0
vaesenc 0x20(%r14), %xmm1, %xmm1
vmovdqa %xmm6, 0xd0(%rsp)
vaesenc %xmm6, %xmm0, %xmm0
vaesenc 0x30(%r14), %xmm1, %xmm1
vmovdqa 0x1a0(%r14), %xmm3
vaesenc 0x40(%r14), %xmm1, %xmm1
vmovdqa 0x1b0(%r14), %xmm4
vmovdqa %xmm3, 0xc0(%rsp)
vaesenc %xmm3, %xmm0, %xmm0
vmovdqa %xmm4, 0xb0(%rsp)
vaesenc %xmm4, %xmm0, %xmm0
vaesenc 0x50(%r14), %xmm1, %xmm1
vmovdqa 0x1c0(%r14), %xmm3
vmovdqa %xmm3, 0xa0(%rsp)
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x60(%r14), %xmm1, %xmm1
vmovdqa 0x1d0(%r14), %xmm3
vmovdqa %xmm3, 0x90(%rsp)
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x70(%r14), %xmm1, %xmm1
vmovdqa 0x1e0(%r14), %xmm3
vaesenc 0x80(%r14), %xmm1, %xmm1
vmovdqa %xmm3, 0x80(%rsp)
vaesenc %xmm3, %xmm0, %xmm0
vmovdqa 0x1f0(%r14), %xmm3
vmovdqa %xmm3, 0x70(%rsp)
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x90(%r14), %xmm1, %xmm1
vmovdqa 0x200(%r14), %xmm3
vmovdqa %xmm3, 0x60(%rsp)
vaesenclast %xmm3, %xmm0, %xmm0
vaesenclast 0xa0(%r14), %xmm1, %xmm1
vmovdqa %xmm1, 0x50(%rsp)
vpternlogq $0x96, %xmm2, %xmm1, %xmm0
vmovdqa %xmm0, 0x30(%rsp)
movq %r15, %rdi
movq %r13, %rsi
callq 0x24e7
vmovdqa 0x30(%rsp), %xmm0
vmovdqa %xmm0, 0x40(%rsp)
leaq (%rsp,%r13), %rax
addq $0x40, %rax
xorl %ecx, %ecx
cmpq %rcx, %r12
je 0x2081
movb (%rax,%rcx), %dl
movb %dl, (%rbx,%rcx)
incq %rcx
jmp 0x2071
vmovdqa (%rsp), %xmm0
vpxor 0x10(%rsp), %xmm0, %xmm0
vaesenc 0x20(%rsp), %xmm0, %xmm0
vaesenc 0xe0(%rsp), %xmm0, %xmm0
vaesenc 0xd0(%rsp), %xmm0, %xmm0
vaesenc 0xc0(%rsp), %xmm0, %xmm0
vaesenc 0xb0(%rsp), %xmm0, %xmm0
vaesenc 0xa0(%rsp), %xmm0, %xmm0
vaesenc 0x90(%rsp), %xmm0, %xmm0
vaesenc 0x80(%rsp), %xmm0, %xmm0
vaesenc 0x70(%rsp), %xmm0, %xmm0
vaesenclast 0x60(%rsp), %xmm0, %xmm0
vmovdqa 0x240(%r14), %xmm1
vmovdqa 0x50(%rsp), %xmm2
vpxor 0x160(%r14), %xmm2, %xmm2
vpternlogq $0x96, (%r14), %xmm1, %xmm0
vaesenc 0x170(%r14), %xmm2, %xmm2
vaesenc 0x10(%r14), %xmm0, %xmm0
vaesenc 0x180(%r14), %xmm2, %xmm2
vaesenc 0x20(%r14), %xmm0, %xmm0
vaesenc 0x190(%r14), %xmm2, %xmm2
vaesenc 0x30(%r14), %xmm0, %xmm0
vaesenc 0x1a0(%r14), %xmm2, %xmm2
vaesenc 0x40(%r14), %xmm0, %xmm0
vaesenc 0x1b0(%r14), %xmm2, %xmm2
vaesenc 0x50(%r14), %xmm0, %xmm0
vaesenc 0x1c0(%r14), %xmm2, %xmm2
vaesenc 0x60(%r14), %xmm0, %xmm0
vaesenc 0x1d0(%r14), %xmm2, %xmm2
vaesenc 0x70(%r14), %xmm0, %xmm0
vaesenc 0x1e0(%r14), %xmm2, %xmm2
vaesenc 0x80(%r14), %xmm0, %xmm0
vaesenc 0x1f0(%r14), %xmm2, %xmm2
vaesenc 0x90(%r14), %xmm0, %xmm0
vaesenclast 0x200(%r14), %xmm2, %xmm2
vaesenclast 0xa0(%r14), %xmm0, %xmm0
vpternlogq $0x96, %xmm2, %xmm1, %xmm0
vmovdqa %xmm0, 0x40(%rsp)
addq %r12, %rbx
xorl %eax, %eax
cmpq %rax, %r13
je 0x2483
movb 0x40(%rsp,%rax), %cl
movb %cl, (%rbx,%rax)
incq %rax
jmp 0x21b7
vmovdqa 0x240(%r14), %xmm0
vmovdqa 0x10(%rsp), %xmm1
vpxor 0x160(%r14), %xmm1, %xmm1
vpxor (%r14), %xmm0, %xmm0
vpternlogq $0x96, (%rsp), %xmm6, %xmm0
vaesenc 0x170(%r14), %xmm1, %xmm1
vaesenc 0x10(%r14), %xmm0, %xmm0
vaesenc 0x180(%r14), %xmm1, %xmm1
vaesenc 0x20(%r14), %xmm0, %xmm0
vaesenc 0x190(%r14), %xmm1, %xmm1
vaesenc 0x30(%r14), %xmm0, %xmm0
vaesenc 0x1a0(%r14), %xmm1, %xmm1
vaesenc 0x40(%r14), %xmm0, %xmm0
vaesenc 0x1b0(%r14), %xmm1, %xmm1
vaesenc 0x50(%r14), %xmm0, %xmm0
vaesenc 0x1c0(%r14), %xmm1, %xmm1
vaesenc 0x60(%r14), %xmm0, %xmm0
vaesenc 0x1d0(%r14), %xmm1, %xmm1
vaesenc 0x70(%r14), %xmm0, %xmm0
vaesenc 0x1e0(%r14), %xmm1, %xmm1
vaesenc 0x80(%r14), %xmm0, %xmm0
vaesenc 0x1f0(%r14), %xmm1, %xmm1
vaesenc 0x90(%r14), %xmm0, %xmm0
vaesenclast 0x200(%r14), %xmm1, %xmm1
vaesenclast 0xa0(%r14), %xmm0, %xmm0
vpternlogq $0x96, %xmm1, %xmm6, %xmm0
vmovdqu %xmm0, (%rbx)
jmp 0x2483
vpxor 0x20(%rsp), %xmm6, %xmm0
vmovdqa 0x170(%r14), %xmm1
vmovdqa 0x180(%r14), %xmm2
vmovdqa 0x190(%r14), %xmm3
vpxor (%rsp), %xmm6, %xmm4
vmovdqa 0x1a0(%r14), %xmm5
vaesenc %xmm1, %xmm4, %xmm4
vaesenc 0x10(%r14), %xmm0, %xmm0
vmovdqa 0x10(%rsp), %xmm7
vaesenc %xmm1, %xmm7, %xmm1
vaesenc %xmm2, %xmm4, %xmm4
vaesenc %xmm2, %xmm1, %xmm1
vaesenc 0x20(%r14), %xmm0, %xmm0
vaesenc %xmm3, %xmm4, %xmm2
vaesenc %xmm3, %xmm1, %xmm1
vaesenc 0x30(%r14), %xmm0, %xmm0
vaesenc %xmm5, %xmm2, %xmm2
vaesenc 0x40(%r14), %xmm0, %xmm0
vaesenc %xmm5, %xmm1, %xmm1
vmovdqa 0x1b0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc 0x50(%r14), %xmm0, %xmm0
vaesenc %xmm3, %xmm1, %xmm1
vmovdqa 0x1c0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc 0x60(%r14), %xmm0, %xmm0
vaesenc %xmm3, %xmm1, %xmm1
vmovdqa 0x1d0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc 0x70(%r14), %xmm0, %xmm0
vaesenc %xmm3, %xmm1, %xmm1
vmovdqa 0x1e0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc 0x80(%r14), %xmm0, %xmm0
vaesenc %xmm3, %xmm1, %xmm1
vmovdqa 0x1f0(%r14), %xmm3
vaesenc %xmm3, %xmm2, %xmm2
vaesenc 0x90(%r14), %xmm0, %xmm0
vaesenc %xmm3, %xmm1, %xmm1
vmovdqa 0x200(%r14), %xmm3
vaesenclast %xmm3, %xmm2, %xmm2
vaesenclast 0xa0(%r14), %xmm0, %xmm0
vaesenclast %xmm3, %xmm1, %xmm1
vpternlogq $0x96, %xmm6, %xmm0, %xmm1
vmovdqu %xmm1, (%r15)
vmovdqa 0x240(%r14), %xmm1
vpxor 0x160(%r14), %xmm0, %xmm0
vpternlogq $0x96, (%r14), %xmm1, %xmm2
vaesenc 0x170(%r14), %xmm0, %xmm0
vaesenc 0x10(%r14), %xmm2, %xmm2
vaesenc 0x180(%r14), %xmm0, %xmm0
vaesenc 0x20(%r14), %xmm2, %xmm2
vaesenc 0x190(%r14), %xmm0, %xmm0
vaesenc 0x30(%r14), %xmm2, %xmm2
vaesenc 0x1a0(%r14), %xmm0, %xmm0
vaesenc 0x40(%r14), %xmm2, %xmm2
vaesenc 0x1b0(%r14), %xmm0, %xmm0
vaesenc 0x50(%r14), %xmm2, %xmm2
vaesenc 0x1c0(%r14), %xmm0, %xmm0
vaesenc 0x60(%r14), %xmm2, %xmm2
vaesenc 0x1d0(%r14), %xmm0, %xmm0
vaesenc 0x70(%r14), %xmm2, %xmm2
vaesenc 0x1e0(%r14), %xmm0, %xmm0
vaesenc 0x80(%r14), %xmm2, %xmm2
vaesenc 0x1f0(%r14), %xmm0, %xmm0
vaesenc 0x90(%r14), %xmm2, %xmm2
vaesenclast 0x200(%r14), %xmm0, %xmm0
vaesenclast 0xa0(%r14), %xmm2, %xmm2
vpternlogq $0x96, %xmm0, %xmm1, %xmm2
vmovdqu %xmm2, (%rbx)
addq $0xf8, %rsp
popq %rbx
popq %r12
popq %r13
popq %r14
popq %r15
popq %rbp
retq