fix mcount

Author: jwt27

src/libc/crt0/makefile

@@ -12,7 +12,8 @@ SRC += c1args.c
 SRC += c1loadef.c
 SRC += c1pglob.c
 SRC += crt1.c
-SRC += mcount.c
+SRC += mcount_i.c
+SRC += mcount.S
 SRC += memhandl.c
 SRC += rfinfo.c
 SRC += dfinfo.c
@@ -40,5 +41,5 @@ $(LIB)/gcrt0.o : gcrt0.S crt0.S exit16.ah sbrk16.ah
 	$(MISC) rm gcrt0.o
 	sed 's@gcrt0.o@$(LIB)/gcrt0.o@' gcrt0.d > gcrt02.d
 
-mcount.o : mcount.c
+mcount_i.o : mcount_i.c
 	$(XNOPGGCC) -c $<

src/libc/crt0/mcount.S

@@ -0,0 +1,27 @@
+/* Copyright (C) 2024 DJ Delorie, see COPYING.DJ for details */
+
+	.intel_syntax noprefix
+	.text
+	.globl	_mcount
+_mcount:
+	push	ebp
+	mov	ebp, esp
+	push	ecx
+	push	eax
+	cmp	ebp, offset _etext
+	jb	Lstack_overflow
+	sub	esp, 4			# Keep stack aligned
+	mov	ecx, [ebp]
+	mov	eax, [ebp+4]
+	mov	ecx, [ecx+4]
+	push	edx			# Pointer to cached MTAB entry
+	push	eax			# Our return address (callee)
+	push	ecx			# Callee's return address (caller)
+	call	___mcount_internal
+	add	esp, 16
+	pop	eax
+	pop	ecx
+	pop	ebp
+	ret
+Lstack_overflow:
+	ud2

src/libc/crt0/mcount.c renamed to src/libc/crt0/mcount_i.c

@@ -1,3 +1,4 @@
+/* Copyright (C) 2024 DJ Delorie, see COPYING.DJ for details */
 /* Copyright (C) 2001 DJ Delorie, see COPYING.DJ for details */
 /* Copyright (C) 1997 DJ Delorie, see COPYING.DJ for details */
 /* Copyright (C) 1995 DJ Delorie, see COPYING.DJ for details */
@@ -38,7 +39,7 @@ typedef struct MTAB {
 
 static header h;
 static short *histogram;
-static int mcount_skip = 1;
+static volatile unsigned char mcount_skip = 1;
 static int histlen;
 static MTAB *mtab=0;
 
@@ -48,35 +49,24 @@ extern int etext __asm__("etext");
 
 static int profiling_p;
 
-/* called by functions.  Use the pointer it provides to cache
+/* called by mcount.  Use the pointer it provides to cache
 ** the last used MTABE, so that repeated calls to/from the same
 ** pair works quickly - no lookup.
 */
-void mcount(int _to);
-void mcount(int _to)
+void __mcount_internal(unsigned long from, unsigned long to, MTABE **cache);
+void __mcount_internal(unsigned long from, unsigned long to, MTABE **cache)
 {
   MTAB *m;
   int i;
-  unsigned int to;
-  int ebp;
-  unsigned int from;
   int mtabi;
-  MTABE **cache;
-
-  /* obtain the cached pointer */
-  __asm__ __volatile__ ("movl %%edx,%0" : "=g" (cache));
 
   mcount_skip = 1;
   /* Do nothing if profiling is disabled.  */
   if (!profiling_p)
     return;
 
-  if (&_to < &etext)
-    *(int *)(-1) = 0; /* fault! */
+  to -= 12;
 
-  to = *((&_to)-1) - 12;
-  ebp = *((&_to)-2); /* glean the caller's return address from the stack */
-  from = ((int *)ebp)[1];
   /* Do nothing if the FROM address is outside the sampling range.  */
   if (from < h.low || from >= h.high)
     return;