summaryrefslogtreecommitdiff
path: root/tests/Mutex_Test.cpp
blob: f7178182e244446503dbdda4f303bbf90c486707 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
// $Id$

// ============================================================================
//
// = LIBRARY
//    tests
// 
// = FILENAME
//    Mutex_Test.cpp
//
// = DESCRIPTION
//      This is a simple test to illustrate the functionality of
//      ACE_Process_Mutex. The test acquires and releases mutexes. No
//      command line arguments are needed to run the test.
//
// = AUTHOR
//    Prashant Jain and Doug Schmidt
// 
// ============================================================================

#include "ace/Service_Config.h"

#include "ace/Thread_Manager.h"
#include "test_config.h"

static void *
test (void *args)
{
  ACE_Process_Mutex *pm = (ACE_Process_Mutex *) args;
#if (defined (ACE_WIN32) || defined (VXWORKS)) && defined (ACE_HAS_THREADS)
  ACE_Thread_Control tc (ACE_Service_Config::thr_mgr ());
  ACE_NEW_THREAD;
#endif /* (defined (ACE_WIN32) || defined (VXWORKS)) && defined (ACE_HAS_THREADS) */

  ACE_OS::srand (ACE_OS::time (0));

  for (int i = 0; i < ACE_MAX_ITERATIONS; i++)
    {
      ACE_DEBUG ((LM_DEBUG, "(%P|%t) = trying to acquire on iteration %d\n", i));
      ACE_ASSERT (pm->acquire () == 0);
      ACE_DEBUG ((LM_DEBUG, "(%P|%t) = acquired on iteration %d\n", i));

      // Sleep for a random amount of time between 0 and 5 seconds.
      // Note that it's ok to use rand() here because we are running
      // within the critical section defined by the Process_Mutex.
      ACE_OS::sleep (ACE_OS::rand () % 5);

      ACE_ASSERT (pm->release () == 0);
      ACE_DEBUG ((LM_DEBUG, "(%P|%t) = released on iteration %d\n", i));
    }

  return 0;
}

static void
spawn (void)
{
  char *name = "hello";

#if !defined (ACE_WIN32) && !defined (VXWORKS)
  switch (ACE_OS::fork ())
    {
    case -1:
      ACE_ERROR ((LM_ERROR, "%p\n%a", "fork failed"));
      exit (-1);
    case 0: // In child
      {
	ACE_LOG_MSG->sync ("child");
	ACE_Process_Mutex pm (ACE_WIDE_STRING (name));
	test (&pm);
	break;
      }
    default: // In parent
      {
	ACE_Process_Mutex pm (name);
	test (&pm);

	// Allow the client to exit, then remove the Process_Mutex.
	ACE_OS::wait ();
	pm.remove ();
	break;
      }
    }
#elif defined (ACE_HAS_THREADS)
  ACE_Process_Mutex pm (ACE_WIDE_STRING (name));

  if (ACE_Service_Config::thr_mgr ()->spawn (ACE_THR_FUNC (test),
					     (void *) &pm,
					     THR_NEW_LWP | THR_DETACHED) == -1)
    ACE_ERROR ((LM_ERROR, "%p\n%a", "thread create failed"));

  if (ACE_Service_Config::thr_mgr ()->spawn (ACE_THR_FUNC (test),
					     (void *) &pm,
					     THR_NEW_LWP | THR_DETACHED) == -1)
    ACE_ERROR ((LM_ERROR, "%p\n%a", "thread create failed"));

  // Wait for the threads to exit.
  ACE_Service_Config::thr_mgr ()->wait ();

#else
  ACE_ERROR ((LM_ERROR, "threads not supported on this platform\n%a", 1));
#endif /* ACE_HAS_THREADS */	
}

int
main (int, char *[])
{
  ACE_START_TEST ("Mutex_Test");

  spawn ();

  ACE_END_TEST;
  return 0;
}