/*	$OpenBSD: crypto_assembly.h,v 1.5 2026/05/07 15:50:47 jsing Exp $ */
/*
 * Copyright (c) 2026 Joel Sing <jsing@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#ifndef HEADER_CRYPTO_ASSEMBLY_H
#define HEADER_CRYPTO_ASSEMBLY_H

/* Ensure _CET_ENDBR is always defined on amd64. */
#ifdef __amd64__
#ifdef __CET__
#include <cet.h>
#else
#define _CET_ENDBR
#endif
#endif

#if defined(__APPLE__) && defined(__aarch64__)
#define CRYPTO_ASSEMBLY_SEPARATOR		%%
#else
#define CRYPTO_ASSEMBLY_SEPARATOR		;
#endif

#if defined(__APPLE__)
#define CRYPTO_ASSEMBLY_SECTION_TEXT		__TEXT,__text
#define CRYPTO_ASSEMBLY_SECTION_RODATA		__DATA,__const

#define CRYPTO_ASSEMBLY_SYMBOL_NAME(name)	_name
#define CRYPTO_ASSEMBLY_TYPE_FUNCTION(name)
#define CRYPTO_ASSEMBLY_TYPE_OBJECT(name)
#define CRYPTO_ASSEMBLY_OBJECT_SIZE(name)

#define CRYPTO_ASSEMBLY_AARCH64_SYM_HI(name)	name@PAGE
#define CRYPTO_ASSEMBLY_AARCH64_SYM_LO(name)	name@PAGEOFF

#else
#define CRYPTO_ASSEMBLY_SECTION_TEXT		.text
#define CRYPTO_ASSEMBLY_SECTION_RODATA		.rodata

#define CRYPTO_ASSEMBLY_SYMBOL_NAME(name)	name
#define CRYPTO_ASSEMBLY_TYPE_FUNCTION(name)	.type	name,@function
#define CRYPTO_ASSEMBLY_TYPE_OBJECT(name)	.type   name,@object
#define CRYPTO_ASSEMBLY_OBJECT_SIZE(name)	.size   name,.-name

#define CRYPTO_ASSEMBLY_AARCH64_SYM_HI(name)	name
#define CRYPTO_ASSEMBLY_AARCH64_SYM_LO(name)	:lo12:name
#endif

#define CRYPTO_ASSEMBLY_GLOBAL_FUNCTION(name) \
	.global CRYPTO_ASSEMBLY_SYMBOL_NAME(name) CRYPTO_ASSEMBLY_SEPARATOR	\
	CRYPTO_ASSEMBLY_TYPE_FUNCTION(name) CRYPTO_ASSEMBLY_SEPARATOR		\
	CRYPTO_ASSEMBLY_SYMBOL_NAME(name)

#define CRYPTO_ASSEMBLY_OBJECT_START(name) \
	CRYPTO_ASSEMBLY_TYPE_OBJECT(name) CRYPTO_ASSEMBLY_SEPARATOR \
	name

#define CRYPTO_ASSEMBLY_OBJECT_END(name) \
	CRYPTO_ASSEMBLY_OBJECT_SIZE(name)

#endif