#include "hmac_sha512.cuh"

__device__ void hmac_sha512_init(HmacSha512Context *ctx, const uint8_t password[], size_t password_len) {
  uint8_t key[128];
  if (password_len <= 128) {
    memcpy(key, password, password_len);
    memset(key + password_len, 0, 128 - password_len);
  } else {
    Sha512Context ctx_key{};
    sha512_init(&ctx_key);
    sha512_update(&ctx_key, key, password_len);
    sha512_final(&ctx_key);
    sha512_write_output(&ctx_key, key);
    memset(key + 64, 0, 64);
  }

#pragma unroll 128
  for (uint8_t &byte : key) {
    byte ^= 0x5c;
  }

  sha512_init(&ctx->ctx_outside);
  sha512_update(&ctx->ctx_outside, key, 128);

#pragma unroll 128
  for (uint8_t &byte : key) {
    byte ^= 0x5c ^ 0x36;
  }

  sha512_init(&ctx->ctx_inside);
  sha512_update(&ctx->ctx_inside, key, 128);
}

__device__ void hmac_sha512_update(HmacSha512Context *ctx, const uint8_t message[], size_t message_len) {
  sha512_update(&ctx->ctx_inside, message, message_len);
}

__device__ void hmac_sha512_final(HmacSha512Context *ctx) {
  uint8_t inner[64];
  sha512_final(&ctx->ctx_inside);
  sha512_write_output(&ctx->ctx_inside, inner);

  sha512_update(&ctx->ctx_outside, inner, 64);
  sha512_final(&ctx->ctx_outside);
}

__device__ void hmac_sha512_write_output(HmacSha512Context *ctx, uint8_t hash[]) {
  sha512_write_output(&ctx->ctx_outside, hash);
}

	.file	"tmpxft_002f310d_00000000-6_hmac_sha512.cudafe1.cpp"
	.text
#APP

#NO_APP
	.type	_ZL26__cudaUnregisterBinaryUtilv, @function
_ZL26__cudaUnregisterBinaryUtilv:
.LFB2014:
	.cfi_startproc
	endbr64
	movq	_ZL20__cudaFatCubinHandle(%rip), %rdi
	jmp	__cudaUnregisterFatBinary@PLT
	.cfi_endproc
.LFE2014:
	.size	_ZL26__cudaUnregisterBinaryUtilv, .-_ZL26__cudaUnregisterBinaryUtilv
	.globl	_Z16hmac_sha512_initP17HmacSha512ContextPKhm
	.type	_Z16hmac_sha512_initP17HmacSha512ContextPKhm, @function
_Z16hmac_sha512_initP17HmacSha512ContextPKhm:
.LFB2008:
	.cfi_startproc
	endbr64
	pushq	%rax
	.cfi_def_cfa_offset 16
	popq	%rax
	.cfi_def_cfa_offset 8
	subq	$24, %rsp
	.cfi_def_cfa_offset 32
	movl	$1, 12(%rsp)
	movl	12(%rsp), %edi
	call	exit@PLT
	.cfi_endproc
.LFE2008:
	.size	_Z16hmac_sha512_initP17HmacSha512ContextPKhm, .-_Z16hmac_sha512_initP17HmacSha512ContextPKhm
	.globl	_Z18hmac_sha512_updateP17HmacSha512ContextPKhm
	.type	_Z18hmac_sha512_updateP17HmacSha512ContextPKhm, @function
_Z18hmac_sha512_updateP17HmacSha512ContextPKhm:
.LFB2153:
	.cfi_startproc
	endbr64
	pushq	%rax
	.cfi_def_cfa_offset 16
	popq	%rax
	.cfi_def_cfa_offset 8
	subq	$24, %rsp
	.cfi_def_cfa_offset 32
	movl	$1, 12(%rsp)
	movl	12(%rsp), %edi
	call	exit@PLT
	.cfi_endproc
.LFE2153:
	.size	_Z18hmac_sha512_updateP17HmacSha512ContextPKhm, .-_Z18hmac_sha512_updateP17HmacSha512ContextPKhm
	.globl	_Z17hmac_sha512_finalP17HmacSha512Context
	.type	_Z17hmac_sha512_finalP17HmacSha512Context, @function
_Z17hmac_sha512_finalP17HmacSha512Context:
.LFB2010:
	.cfi_startproc
	endbr64
	pushq	%rax
	.cfi_def_cfa_offset 16
	popq	%rax
	.cfi_def_cfa_offset 8
	subq	$24, %rsp
	.cfi_def_cfa_offset 32
	movl	$1, 12(%rsp)
	movl	12(%rsp), %edi
	call	exit@PLT
	.cfi_endproc
.LFE2010:
	.size	_Z17hmac_sha512_finalP17HmacSha512Context, .-_Z17hmac_sha512_finalP17HmacSha512Context
	.globl	_Z24hmac_sha512_write_outputP17HmacSha512ContextPh
	.type	_Z24hmac_sha512_write_outputP17HmacSha512ContextPh, @function
_Z24hmac_sha512_write_outputP17HmacSha512ContextPh:
.LFB2011:
	.cfi_startproc
	endbr64
	pushq	%rax
	.cfi_def_cfa_offset 16
	popq	%rax
	.cfi_def_cfa_offset 8
	subq	$24, %rsp
	.cfi_def_cfa_offset 32
	movl	$1, 12(%rsp)
	movl	12(%rsp), %edi
	call	exit@PLT
	.cfi_endproc
.LFE2011:
	.size	_Z24hmac_sha512_write_outputP17HmacSha512ContextPh, .-_Z24hmac_sha512_write_outputP17HmacSha512ContextPh
	.section	.text.startup,"ax",@progbits
	.type	_ZL24__sti____cudaRegisterAllv, @function
_ZL24__sti____cudaRegisterAllv:
.LFB2037:
	.cfi_startproc
	endbr64
	pushq	%rax
	.cfi_def_cfa_offset 16
	leaq	_ZL15__fatDeviceText(%rip), %rdi
	call	__cudaRegisterFatBinary@PLT
	movq	%rax, %rdi
	movq	%rax, _ZL20__cudaFatCubinHandle(%rip)
	call	__cudaRegisterFatBinaryEnd@PLT
	leaq	_ZL26__cudaUnregisterBinaryUtilv(%rip), %rdi
	popq	%rdx
	.cfi_def_cfa_offset 8
	jmp	atexit@PLT
	.cfi_endproc
.LFE2037:
	.size	_ZL24__sti____cudaRegisterAllv, .-_ZL24__sti____cudaRegisterAllv
	.section	.init_array,"aw"
	.align 8
	.quad	_ZL24__sti____cudaRegisterAllv
	.section	.nvFatBinSegment,"aw"
	.align 8
	.type	_ZL15__fatDeviceText, @object
	.size	_ZL15__fatDeviceText, 24
_ZL15__fatDeviceText:
	.long	1180844977
	.long	1
	.quad	fatbinData
	.quad	0
	.local	_ZL20__cudaFatCubinHandle
	.comm	_ZL20__cudaFatCubinHandle,8,8
	.ident	"GCC: (Ubuntu 13.3.0-6ubuntu2~24.04) 13.3.0"
	.section	.note.GNU-stack,"",@progbits
	.section	.note.gnu.property,"a"
	.align 8
	.long	1f - 0f
	.long	4f - 1f
	.long	5
0:
	.string	"GNU"
1:
	.align 8
	.long	0xc0000002
	.long	3f - 2f
2:
	.long	0x3
3:
	.align 8
4:

#pragma once
#include <hip/hip_runtime.h>


#include <cstdint>

#include "sha512.cuh"

struct HmacSha512Context {
  Sha512Context ctx_inside;
  Sha512Context ctx_outside;
};

__device__ void hmac_sha512_init(HmacSha512Context *ctx, const uint8_t password[], size_t password_len);
__device__ void hmac_sha512_update(HmacSha512Context *ctx, const uint8_t message[], size_t message_len);
__device__ void hmac_sha512_final(HmacSha512Context *ctx);
__device__ void hmac_sha512_write_output(HmacSha512Context *ctx, uint8_t hash[]);

	.text
	.file	"hmac_sha512.hip"
	.type	__hip_cuid_,@object             # @__hip_cuid_
	.bss
	.globl	__hip_cuid_
__hip_cuid_:
	.byte	0                               # 0x0
	.size	__hip_cuid_, 1

	.ident	"AMD clang version 18.0.0git (https://github.com/RadeonOpenCompute/llvm-project roc-6.3.2 25012 e5bf7e55c91490b07c49d8960fa7983d864936c4)"
	.section	".note.GNU-stack","",@progbits
	.addrsig
	.addrsig_sym __hip_cuid_

	.text
	.p2alignl 7, 3214868480
	.fill 96, 4, 3214868480
	.type	__hip_cuid_,@object             ; @__hip_cuid_
	.section	.bss,"aw",@nobits
	.globl	__hip_cuid_
__hip_cuid_:
	.byte	0                               ; 0x0
	.size	__hip_cuid_, 1

	.ident	"AMD clang version 18.0.0git (https://github.com/RadeonOpenCompute/llvm-project roc-6.3.2 25012 e5bf7e55c91490b07c49d8960fa7983d864936c4)"
	.section	".note.GNU-stack","",@progbits
	.addrsig
	.addrsig_sym __hip_cuid_
	.amdgcn_target "amdgcn-amd-amdhsa--gfx1100"
	.amdgpu_metadata
---
amdhsa.kernels:  []
amdhsa.target:   amdgcn-amd-amdhsa--gfx1100
amdhsa.version:
  - 1
  - 2
...

	.end_amdgpu_metadata
