Add mtrace hooks to spinlock and kalloc.

471c0c15 · Silas Boyd-Wickizer · f70ef994 · 471c0c15 · 471c0c15 · 471c0c15
--- a/Makefile
+++ b/Makefile
@@ -49,6 +49,12 @@ TOOLPREFIX := $(shell if i386-jos-elf-objdump -i 2>&1 | grep '^elf32-i386$$' >/d
 	echo "***" 1>&2; exit 1; fi)
 endif

+QEMUSRC ?= 
+
+ifeq ($(QEMUSRC),)
+$(error You need to set QEMUSRC (e.g. make QEMUSRC=~/qemu))
+endif
+
 # If the makefile can't find QEMU, specify its path here
 #QEMU = 

@@ -71,7 +77,7 @@ AS = $(TOOLPREFIX)gas
 LD = $(TOOLPREFIX)ld
 OBJCOPY = $(TOOLPREFIX)objcopy
 OBJDUMP = $(TOOLPREFIX)objdump
-CFLAGS = -fno-pic -static -fno-builtin -fno-strict-aliasing -O2 -Wall -MD -ggdb -m32 -Werror
+CFLAGS = -fno-pic -static -fno-builtin -fno-strict-aliasing -O2 -Wall -MD -ggdb -m32 -Werror -I$(QEMUSRC)
 CFLAGS += $(shell $(CC) -fno-stack-protector -E -x c /dev/null >/dev/null 2>&1 && echo -fno-stack-protector)
 ASFLAGS = -m32 -gdwarf-2
 # FreeBSD ld wants ``elf_i386_fbsd''

--- a/kalloc.c
+++ b/kalloc.c
@@ -7,6 +7,7 @@
 #include "param.h"
 #include "mmu.h"
 #include "spinlock.h"
+#include "xv6-mtrace.h"

 struct run {
  struct run *next;
@@ -51,6 +52,14 @@ kfree(char *v)
  r = (struct run*)v;
  r->next = kmem.freelist;
  kmem.freelist = r;
+
+  mtrace_label_register(mtrace_label_block,
+			r,
+			0,
+			0,
+			0,
+			RET_EIP());
+
  release(&kmem.lock);
 }

@@ -66,6 +75,14 @@ kalloc(void)
  r = kmem.freelist;
  if(r)
    kmem.freelist = r->next;
+
+  mtrace_label_register(mtrace_label_block,
+			r,
+			4096,
+			"kalloc",
+			sizeof("kalloc"),
+			RET_EIP());
+
  release(&kmem.lock);
  return (char*)r;
 }

--- a/spinlock.c
+++ b/spinlock.c
@@ -7,6 +7,7 @@
 #include "mmu.h"
 #include "proc.h"
 #include "spinlock.h"
+#include "xv6-mtrace.h"

 void
 initlock(struct spinlock *lk, char *name)
@@ -27,12 +28,24 @@ acquire(struct spinlock *lk)
  if(holding(lk))
    panic("acquire");

+  mtrace_lock_register(RET_EIP(),
+		       lk,
+		       lk->name,
+		       mtrace_lockop_acquire,
+		       0);
+
  // The xchg is atomic.
  // It also serializes, so that reads after acquire are not
  // reordered before it.
  while(xchg(&lk->locked, 1) != 0)
    ;

+  mtrace_lock_register(RET_EIP(),
+		       lk,
+		       lk->name,
+		       mtrace_lockop_acquired,
+		       0);
+
  // Record info about lock acquisition for debugging.
  lk->cpu = cpu;
  getcallerpcs(&lk, lk->pcs);
@@ -45,6 +58,12 @@ release(struct spinlock *lk)
  if(!holding(lk))
    panic("release");

+  mtrace_lock_register(RET_EIP(),
+		       lk,
+		       lk->name,
+		       mtrace_lockop_release,
+		       0);
+
  lk->pcs[0] = 0;
  lk->cpu = 0;


--- a/xv6-mtrace.h
+++ b/xv6-mtrace.h
+#include <stdint.h>
+
+void* memcpy(void *dst, const void *src, uint n);
+
+#define RET_EIP() ((unsigned long)__builtin_return_address(0))
+
+#include "mtrace-magic.h"