// File: /workspace/github_c_makefile_star2_O2/mjosaarinen[P]brutus/crypto_aead/aes128poetv2aes4ni/ni/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* k = ctx->aes_enc;
    __m128i* hk = ctx->aes_axu;
    __m128i x = ctx->x;
    __m128i y = ctx->y;
    __m128i z;
    const uint64_t plaintext_len = plen;

    if (clen) {
        *clen = plen;
    }

    x = aesfour_encrypt(x, hk);

    if (plen >= BLOCKLEN) {
        x = vxor(x, loadu(plaintext));
        z = x;
        xor_3blocks(x, y, z, hk[0], hk[0], k[0]); \
        plaintext += BLOCKLEN;
    }

    __m128i tmp_x;
    __m128i tmp_z;
    __m128i tmp_y;

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

    const __m128i s = aes_encrypt(encode_length(plaintext_len), k);

    if (plen == 0) { // Empty message
        z = vxor3(x, s, ctx->tau);
        aes_encrypt_2blocks(y, z, hk, k);
        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, k, hk);
        storeu(ciphertext, vxor3(z, y, s));

        // Generate tag
        y = z;
        z = vxor(x, ctx->tau);
        aes_encrypt_2blocks(y, z, hk, k);
        z = vxor3(z, y, ctx->tau);
        storeu(tag, z);
    } else { // Tag splitting
        if (plen > BLOCKLEN) {
            aes_encrypt_3blocks_no_whitening(x, y, z, k, hk);
            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, hk[0], hk[0], k[0]);
        aes_encrypt_3blocks_no_whitening(x, y, z, k, hk);

        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, hk[0], hk[0], k[0]);
        aes_encrypt_3blocks_no_whitening(x, y, z, k, hk);

        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 $0x98, %rsp
movq %r9, %rbx
movq %rcx, %r15
movq %rdx, %r13
movq %rsi, %r12
movq %rdi, %r14
addq $0x160, %rdi            # imm = 0x160
vmovdqa 0x1c0(%r14), %xmm0
vmovdqa 0x1d0(%r14), %xmm1
vmovdqa %xmm1, (%rsp)
testq %r8, %r8
je 0x1c7f
movq %r13, (%r8)
callq 0x22a0
cmpq $0x10, %r13
jb 0x1cb1
vpxor (%r12), %xmm0, %xmm2
vmovdqa 0x160(%r14), %xmm1
vpxor %xmm1, %xmm2, %xmm0
vmovdqa (%rsp), %xmm5
vpxor %xmm5, %xmm1, %xmm5
vpxor (%r14), %xmm2, %xmm8
addq $0x10, %r12
jmp 0x1cb6
vmovdqa (%rsp), %xmm5
movq %r13, %rbp
cmpq $0x21, %rbp
jb 0x1d8f
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
vpxor 0x1a0(%r14), %xmm7, %xmm2
vpxor (%r12), %xmm2, %xmm6
vaesenc %xmm3, %xmm4, %xmm4
vaesenc %xmm3, %xmm1, %xmm1
vaesenc 0x30(%r14), %xmm5, %xmm3
vaesenc %xmm2, %xmm1, %xmm1
vaesenc 0x40(%r14), %xmm3, %xmm2
vaesenc 0x50(%r14), %xmm2, %xmm2
vaesenc 0x60(%r14), %xmm2, %xmm2
vaesenc 0x70(%r14), %xmm2, %xmm2
vaesenc 0x80(%r14), %xmm2, %xmm2
vpxor 0xa0(%r14), %xmm7, %xmm0
vaesenc 0x90(%r14), %xmm2, %xmm2
vaesenclast %xmm0, %xmm2, %xmm5
vpxor %xmm1, %xmm5, %xmm0
vmovdqa %xmm0, (%r15)
vmovdqa (%r14), %xmm8
vaesenc %xmm6, %xmm4, %xmm0
vpternlogq $0x96, 0x160(%r14), %xmm0, %xmm8
addq $0x10, %r12
addq $0x10, %r15
addq $-0x10, %rbp
jmp 0x1cb9
vmovdqa %xmm8, 0x20(%rsp)
vmovdqa %xmm0, 0x10(%rsp)
vmovdqa %xmm5, (%rsp)
shlq $0x3, %r13
vmovq %r13, %xmm0
movq %r14, %rdi
callq 0x1b9a
vmovdqa %xmm0, %xmm7
testq %rbp, %rbp
je 0x20b5
movq %rbp, %r13
andq $0xf, %r13
je 0x2153
cmpq $0x11, %rbp
vmovdqa 0x20(%rsp), %xmm6
vmovdqa %xmm7, 0x30(%rsp)
jb 0x1e95
vmovdqa 0x170(%r14), %xmm0
vmovdqa 0x180(%r14), %xmm1
vmovdqa 0x190(%r14), %xmm2
vmovdqa 0x1a0(%r14), %xmm3
vmovdqa 0x10(%rsp), %xmm4
vaesenc %xmm0, %xmm4, %xmm4
vmovdqa (%rsp), %xmm5
vaesenc %xmm0, %xmm5, %xmm0
vaesenc 0x10(%r14), %xmm6, %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
vmovdqa %xmm2, 0x10(%rsp)
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x40(%r14), %xmm1, %xmm1
vaesenc 0x50(%r14), %xmm1, %xmm1
vaesenc 0x60(%r14), %xmm1, %xmm1
vaesenc 0x70(%r14), %xmm1, %xmm1
vaesenc 0x80(%r14), %xmm1, %xmm1
vaesenc 0x90(%r14), %xmm1, %xmm1
vaesenclast 0xa0(%r14), %xmm1, %xmm1
vmovdqa %xmm1, (%rsp)
vpxor %xmm0, %xmm1, %xmm0
vmovdqu %xmm0, (%r15)
addq $0x10, %r15
movq %r12, %rdi
movq %r13, %rsi
callq 0x1c09
vmovdqa %xmm0, 0x20(%rsp)
pushq $0x10
popq %r12
subq %r13, %r12
leaq 0x117c(%rip), %rbp      # 0x3030
vmovdqa 0x1e0(%r14), %xmm0
movq %r13, %rdi
callq 0x22bd
vmovdqa 0x20(%rsp), %xmm1
vpternlogq $0xf8, (%r12,%rbp), %xmm1, %xmm0
vmovdqa 0x30(%rsp), %xmm2
vpternlogq $0x96, 0x10(%rsp), %xmm2, %xmm0
vmovdqa 0x160(%r14), %xmm1
vmovdqa 0x170(%r14), %xmm4
vmovdqa 0x180(%r14), %xmm5
vmovdqa 0x190(%r14), %xmm6
vmovdqa %xmm1, 0x10(%rsp)
vpxor (%rsp), %xmm1, %xmm3
vmovdqa %xmm0, (%rsp)
vpxor (%r14), %xmm0, %xmm1
vmovdqa %xmm4, 0x20(%rsp)
vaesenc %xmm4, %xmm3, %xmm0
vaesenc 0x10(%r14), %xmm1, %xmm1
vmovdqa %xmm5, 0x80(%rsp)
vaesenc %xmm5, %xmm0, %xmm0
vaesenc 0x20(%r14), %xmm1, %xmm1
vmovdqa %xmm6, 0x70(%rsp)
vaesenc %xmm6, %xmm0, %xmm0
vaesenc 0x30(%r14), %xmm1, %xmm1
vmovdqa 0x1a0(%r14), %xmm3
vaesenc 0x40(%r14), %xmm1, %xmm1
vaesenc 0x50(%r14), %xmm1, %xmm1
vaesenc 0x60(%r14), %xmm1, %xmm1
vaesenc 0x70(%r14), %xmm1, %xmm1
vaesenc 0x80(%r14), %xmm1, %xmm1
vmovdqa %xmm3, 0x60(%rsp)
vaesenc %xmm3, %xmm0, %xmm0
vaesenc 0x90(%r14), %xmm1, %xmm1
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 0x230f
vmovdqa 0x30(%rsp), %xmm0
vmovdqa %xmm0, 0x40(%rsp)
leaq (%rsp,%r13), %rax
addq $0x40, %rax
xorl %ecx, %ecx
cmpq %rcx, %r12
je 0x1fdc
movb (%rax,%rcx), %dl
movb %dl, (%rbx,%rcx)
incq %rcx
jmp 0x1fcc
vmovdqa 0x10(%rsp), %xmm0
vpxor (%rsp), %xmm0, %xmm0
vaesenc 0x20(%rsp), %xmm0, %xmm0
vaesenc 0x80(%rsp), %xmm0, %xmm0
vaesenc 0x70(%rsp), %xmm0, %xmm0
vaesenc 0x60(%rsp), %xmm0, %xmm0
vmovdqa 0x1e0(%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 0x50(%r14), %xmm0, %xmm0
vaesenc 0x60(%r14), %xmm0, %xmm0
vaesenc 0x70(%r14), %xmm0, %xmm0
vaesenc 0x80(%r14), %xmm0, %xmm0
vaesenc 0x90(%r14), %xmm0, %xmm0
vaesenclast 0xa0(%r14), %xmm0, %xmm0
vpternlogq $0x96, %xmm2, %xmm1, %xmm0
vmovdqa %xmm0, 0x40(%rsp)
addq %r12, %rbx
xorl %eax, %eax
cmpq %rax, %r13
je 0x228e
movb 0x40(%rsp,%rax), %cl
movb %cl, (%rbx,%rax)
incq %rax
jmp 0x20a0
vmovdqa 0x1e0(%r14), %xmm0
vmovdqa (%rsp), %xmm1
vpxor 0x160(%r14), %xmm1, %xmm1
vpxor (%r14), %xmm0, %xmm0
vpternlogq $0x96, 0x10(%rsp), %xmm7, %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 0x50(%r14), %xmm0, %xmm0
vaesenc 0x60(%r14), %xmm0, %xmm0
vaesenc 0x70(%r14), %xmm0, %xmm0
vaesenc 0x80(%r14), %xmm0, %xmm0
vaesenc 0x90(%r14), %xmm0, %xmm0
vaesenclast 0xa0(%r14), %xmm0, %xmm0
vpternlogq $0x96, %xmm1, %xmm7, %xmm0
vmovdqu %xmm0, (%rbx)
jmp 0x228e
vpxor 0x20(%rsp), %xmm7, %xmm0
vmovdqa 0x170(%r14), %xmm1
vmovdqa 0x180(%r14), %xmm2
vmovdqa 0x190(%r14), %xmm3
vpxor 0x10(%rsp), %xmm7, %xmm4
vmovdqa 0x1a0(%r14), %xmm5
vaesenc %xmm1, %xmm4, %xmm4
vaesenc 0x10(%r14), %xmm0, %xmm0
vmovdqa (%rsp), %xmm6
vaesenc %xmm1, %xmm6, %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 0x50(%r14), %xmm0, %xmm0
vaesenc 0x60(%r14), %xmm0, %xmm0
vaesenc 0x70(%r14), %xmm0, %xmm0
vaesenc 0x80(%r14), %xmm0, %xmm0
vaesenc 0x90(%r14), %xmm0, %xmm0
vaesenclast 0xa0(%r14), %xmm0, %xmm0
vaesenc %xmm5, %xmm1, %xmm1
vpternlogq $0x96, %xmm7, %xmm0, %xmm1
vmovdqu %xmm1, (%r15)
vmovdqa 0x1e0(%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 0x50(%r14), %xmm2, %xmm2
vaesenc 0x60(%r14), %xmm2, %xmm2
vaesenc 0x70(%r14), %xmm2, %xmm2
vaesenc 0x80(%r14), %xmm2, %xmm2
vaesenc 0x90(%r14), %xmm2, %xmm2
vaesenclast 0xa0(%r14), %xmm2, %xmm2
vpternlogq $0x96, %xmm0, %xmm1, %xmm2
vmovdqu %xmm2, (%rbx)
addq $0x98, %rsp
popq %rbx
popq %r12
popq %r13
popq %r14
popq %r15
popq %rbp
retq