1 files changed, 19 insertions, 2 deletions

diff --git a/sys/amd64/include/mptable.h b/sys/amd64/include/mptable.h
index 4788ee47e24f..0c1fc03c3f2d 100644
--- a/sys/amd64/include/mptable.h
+++ b/sys/amd64/include/mptable.h
@@ -2275,15 +2275,23 @@ invltlb(void)
 int
 stop_cpus(u_int map)
 {
+	int count = 0;
+
 	if (!smp_started)
 		return 0;
 
 	/* send the Xcpustop IPI to all CPUs in map */
 	selected_apic_ipi(map, XCPUSTOP_OFFSET, APIC_DELMODE_FIXED);
 	
-	while ((stopped_cpus & map) != map)
+	while (count++ < 100000 && (stopped_cpus & map) != map)
 		/* spin */ ;
 
+#ifdef DIAGNOSTIC
+	if ((stopped_cpus & map) != map)
+		printf("Warning: CPUs 0x%x did not stop!\n",
+		    (~(stopped_cpus & map)) & map);
+#endif
+
 	return 1;
 }
 
@@ -2304,14 +2312,23 @@ stop_cpus(u_int map)
 int
 restart_cpus(u_int map)
 {
+	int count = 0;
+
 	if (!smp_started)
 		return 0;
 
 	started_cpus = map;		/* signal other cpus to restart */
 
-	while ((stopped_cpus & map) != 0) /* wait for each to clear its bit */
+	/* wait for each to clear its bit */
+	while (count++ < 100000 && (stopped_cpus & map) != 0)
 		/* spin */ ;
 
+#ifdef DIAGNOSTIC
+	if ((stopped_cpus & map) != 0)
+		printf("Warning: CPUs 0x%x did not restart!\n",
+		    (~(stopped_cpus & map)) & map);
+#endif
+
 	return 1;
 }