@@ -12,10 +12,16 @@
  * LICENSE.GPLv2 or LICENSE.GPLv3 respectively included in the packaging of this
  * file.
  */
 
 #include "config.h"
+
+#ifdef HAVE_CET_H
+# include <cet.h>
+#else
+# define _CET_ENDBR
+#endif
 
 .globl _OFForward
 .globl _OFForward_stret
 
 .section __TEXT, __objc_methname, cstring_literals
@@ -29,13 +35,11 @@
 .section __DATA, __objc_imageinfo, regular, no_dead_strip
 	.long 0, 0
 
 .section __TEXT, __text, regular, pure_instructions
 _OFForward:
-#ifdef HAVE_ENDBR64
-	endbr64
-#endif
+	_CET_ENDBR
 
 	pushq	%rbp
 	movq	%rsp, %rbp
 
 	/* Save all arguments */
@@ -106,13 +110,11 @@
 	popq	%rbp
 
 	jmp	_OFMethodNotFound
 
 _OFForward_stret:
-#ifdef HAVE_ENDBR64
-	endbr64
-#endif
+	_CET_ENDBR
 
 	pushq	%rbp
 	movq	%rsp, %rbp
 
 	/* Save all arguments */