mono/tests/spinlock-stress.cs

   1 using System;
   2 using System.Threading;
   3
   4 internal class SpinLockWrapper
   5 {
   6         public SpinLock Lock = new SpinLock (false);
   7 }
   8
   9 public class Tests
  10 {
  11         public static void Main (string[] args)
  12         {
  13                 int iterations = 200;
  14                 if (args.Length > 0)
  15                         iterations = Int32.Parse (args [0]);
  16
  17                 ParallelTestHelper.Repeat (delegate {
  18                         int currentCount = 0;
  19                         bool fail = false;
  20                         SpinLockWrapper wrapper = new SpinLockWrapper ();
  21
  22                         ParallelTestHelper.ParallelStressTest (wrapper, delegate {
  23                                 bool taken = false;
  24                                 wrapper.Lock.Enter (ref taken);
  25                                 int current = currentCount++;
  26                                 if (current != 0)
  27                                         fail = true;
  28
  29                                 SpinWait sw = new SpinWait ();
  30                                 for (int i = 0; i < 200; i++)
  31                                         sw.SpinOnce ();
  32                                 currentCount -= 1;
  33
  34                                 wrapper.Lock.Exit ();
  35                         }, 4);
  36
  37                         if (fail)
  38                                 Environment.Exit (1);
  39                 }, iterations);
  40                 Environment.Exit (0);
  41         }
  42 }
  43
  44 static class ParallelTestHelper
  45 {
  46         public static void Repeat (Action action, int numRun)
  47         {
  48                 for (int i = 0; i < numRun; i++) {
  49                         //Console.WriteLine ("Run " + i.ToString ());
  50                         action ();
  51                 }
  52         }
  53
  54         public static void ParallelStressTest<TSource>(TSource obj, Action<TSource> action, int numThread)
  55         {
  56                 Thread[] threads = new Thread[numThread];
  57                 for (int i = 0; i < numThread; i++) {
  58                         threads[i] = new Thread(new ThreadStart(delegate { action(obj); }));
  59                         threads[i].Start();
  60                 }
  61
  62                 // Wait for the completion
  63                 for (int i = 0; i < numThread; i++)
  64                         threads[i].Join();
  65         }
  66 }