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
116
117
118
119
120
|
#include "CPU_Affinity.h"
#include "dance/DAnCE_PropertiesC.h"
#include "dance/Deployment/Deployment_StartErrorC.h"
#include "dance/Logger/Log_Macros.h"
#include "ace/os_include/os_sched.h"
#include "ace/Auto_Ptr.h"
#include "ace/Tokenizer_T.h"
#include "ace/OS_NS_unistd.h"
#include <sstream>
namespace DAnCE
{
CPU_Affinity::CPU_Affinity (void)
{
}
// Implementation skeleton destructor
CPU_Affinity::~CPU_Affinity (void)
{
}
char * CPU_Affinity::type (void)
{
return CORBA::string_dup (DAnCE::DANCE_LM_CPUAFFINITY);
}
void CPU_Affinity::configure (const ::Deployment::Property & prop)
{
#if defined (ACE_HAS_SCHED_SETAFFINITY)
const char *extracted_affinity = 0;
if (! (prop.value >>= CORBA::Any::to_string (extracted_affinity, 0)))
{
DANCE_ERROR (DANCE_LOG_TERMINAL_ERROR,
(LM_ERROR, DLINFO
ACE_TEXT ("CPU_Affinity::configure - ")
ACE_TEXT ("Unable to extract CPU affinity string")));
throw ::Deployment::StartError (prop.name.in (),
"Unable to extract CPU affinity string");
}
char *affinity = ACE_OS::strdup (extracted_affinity);
ACE_Tokenizer_T<char> tokenizer(affinity);
tokenizer.delimiter (',');
char *token = 0;
cpu_set_t mask;
CPU_ZERO (&mask);
while ((token = tokenizer.next ()))
{
int i = ACE_OS::atoi (token);
if (i >= 0)
{
DANCE_DEBUG (DANCE_LOG_MINOR_EVENT,
(LM_DEBUG, DLINFO
ACE_TEXT ("CPU_Affinity::configure - ")
ACE_TEXT ("Toggling affinity for CPU %i\n"),
i));
CPU_SET (i, &mask);
}
else
{
DANCE_ERROR (DANCE_LOG_TERMINAL_ERROR,
(LM_ERROR, DLINFO
ACE_TEXT ("CPU_Affinity::configure - ")
ACE_TEXT ("All affinity values should be greater than 0")));
throw ::Deployment::StartError (prop.name.in (),
"All affinity values should be greater than 0");
}
}
ACE_OS::free (affinity);
pid_t const pid = ACE_OS::getpid ();
int retval = ::sched_setaffinity (pid,
sizeof (cpu_set_t),
&mask);
if (retval != 0)
{
std::stringstream str;
ACE_Auto_Basic_Array_Ptr<char> safe_error (ACE_OS::strerror (ACE_OS::last_error ()));
str << "Unable to set CPU affinity to <" << affinity << ">: "
<< safe_error.get ();
std::string message = str.str ();
DANCE_ERROR (DANCE_LOG_TERMINAL_ERROR,
(LM_ERROR, DLINFO
ACE_TEXT ("CPU_Affinity::configure - %C\n"),
message.c_str ()));
throw ::Deployment::StartError (prop.name.in (),
message.c_str ());
}
#else
throw ::Deployment::StartError (prop.name.in (),
"CPU Affinity not supported on this platform");
#endif
}
}
extern "C"
{
DAnCE::LocalityConfiguration_ptr create_CPU_Affinity (void)
{
DAnCE::LocalityConfiguration_ptr retval (0);
ACE_NEW_RETURN (retval,
DAnCE::CPU_Affinity (),
0);
return retval;
}
}
|
