ACE/ace/Process.inl

   1 // -*- C++ -*-
   2 #include "ace/ACE.h"
   3 #include "ace/OS_NS_sys_wait.h"
   4 #include "ace/OS_NS_signal.h"
   5 #include "ace/OS_NS_pwd.h"
   6 #include "ace/OS_NS_string.h"
   7
   8 ACE_BEGIN_VERSIONED_NAMESPACE_DECL
   9
  10 ACE_INLINE void
  11 ACE_Process_Options::enable_unicode_environment ()
  12 {
  13   this->use_unicode_environment_ = true;
  14 }
  15
  16 ACE_INLINE void
  17 ACE_Process_Options::disable_unicode_environment ()
  18 {
  19   this->use_unicode_environment_ = false;
  20 }
  21
  22 ACE_INLINE bool
  23 ACE_Process_Options::use_unicode_environment () const
  24 {
  25   return this->use_unicode_environment_;
  26 }
  27
  28 #if defined (ACE_WIN32)
  29
  30 ACE_INLINE PROCESS_INFORMATION
  31 ACE_Process::process_info ()
  32 {
  33   return process_info_;
  34 }
  35 #endif /* ACE_WIN32 */
  36
  37 ACE_INLINE ACE_HANDLE
  38 ACE_Process::gethandle () const
  39 {
  40 #if defined (ACE_WIN32)
  41   return process_info_.hProcess;
  42 #else
  43   return ACE_HANDLE (child_id_);
  44 #endif /* ACE_WIN32 */
  45 }
  46
  47 ACE_INLINE pid_t
  48 ACE_Process::getpid () const
  49 {
  50 #if defined (ACE_WIN32)
  51   return process_info_.dwProcessId;
  52 #else /* ACE_WIN32 */
  53   return child_id_;
  54 #endif /* ACE_WIN32 */
  55 }
  56
  57 ACE_INLINE pid_t
  58 ACE_Process::wait (ACE_exitcode *status,
  59                    int wait_options)
  60 {
  61   pid_t retv =
  62     ACE_OS::wait (this->getpid (),
  63                   &this->exit_code_,
  64                   wait_options
  65 #if defined (ACE_WIN32)
  66                   , process_info_.hProcess
  67 #endif /* ACE_WIN32 */
  68                   );
  69   if (status != 0)
  70     *status = this->exit_code_;
  71
  72   return retv;
  73 }
  74
  75 ACE_INLINE int
  76 ACE_Process::kill (int signum)
  77 {
  78   if (this->getpid () != -1)
  79     return ACE_OS::kill (this->getpid (), signum);
  80   else
  81     return -1;
  82 }
  83
  84 ACE_INLINE int
  85 ACE_Process::terminate ()
  86 {
  87   if (this->getpid () != -1)
  88     return ACE::terminate_process (this->getpid ());
  89   else
  90     return -1;
  91 }
  92
  93 ACE_INLINE int
  94 ACE_Process::return_value () const
  95 {
  96 #if defined (ACE_WIN32)
  97   return this->exit_code_;
  98 #else
  99   return WEXITSTATUS (this->exit_code_);
 100 #endif /* ACE_WIN32 */
 101 }
 102
 103 ACE_INLINE ACE_exitcode
 104 ACE_Process::exit_code () const
 105 {
 106   return this->exit_code_;
 107 }
 108
 109 ACE_INLINE void
 110 ACE_Process::exit_code (ACE_exitcode code)
 111 {
 112   this->exit_code_ = code;
 113 }
 114
 115 ACE_INLINE u_long
 116 ACE_Process_Options::creation_flags () const
 117 {
 118 #if defined (ACE_USES_WCHAR) && defined (ACE_WIN32)
 119   return creation_flags_ | CREATE_UNICODE_ENVIRONMENT;
 120 #else
 121   return creation_flags_;
 122 #endif /* ACE_USES_WCHAR */
 123 }
 124
 125 ACE_INLINE void
 126 ACE_Process_Options::creation_flags (u_long cf)
 127 {
 128   creation_flags_ = cf;
 129 }
 130
 131 ACE_INLINE pid_t
 132 ACE_Process_Options::getgroup () const
 133 {
 134   return process_group_;
 135 }
 136
 137 ACE_INLINE pid_t
 138 ACE_Process_Options::setgroup (pid_t pgrp)
 139 {
 140   pid_t old = process_group_;
 141   process_group_ = pgrp;
 142   return old;
 143 }
 144
 145 ACE_INLINE int
 146 ACE_Process_Options::handle_inheritance ()
 147 {
 148   return handle_inheritance_;
 149 }
 150
 151 ACE_INLINE void
 152 ACE_Process_Options::handle_inheritance (int hi)
 153 {
 154   handle_inheritance_ = hi;
 155 }
 156
 157 ACE_INLINE int
 158 ACE_Process_Options::avoid_zombies ()
 159 {
 160   return avoid_zombies_;
 161 }
 162 ACE_INLINE void
 163 ACE_Process_Options::avoid_zombies (int avoid_zombies)
 164 {
 165   avoid_zombies_ = avoid_zombies;
 166 }
 167
 168 #if defined (ACE_WIN32)
 169
 170 ACE_INLINE ACE_TEXT_STARTUPINFO *
 171 ACE_Process_Options::startup_info ()
 172 {
 173   return &startup_info_;
 174 }
 175
 176 ACE_INLINE LPSECURITY_ATTRIBUTES
 177 ACE_Process_Options::get_process_attributes () const
 178 {
 179   return process_attributes_;
 180 }
 181
 182 ACE_INLINE LPSECURITY_ATTRIBUTES
 183 ACE_Process_Options::set_process_attributes ()
 184 {
 185   process_attributes_ = &security_buf1_;
 186   return process_attributes_;
 187 }
 188
 189 ACE_INLINE LPSECURITY_ATTRIBUTES
 190 ACE_Process_Options::get_thread_attributes () const
 191 {
 192   return thread_attributes_;
 193 }
 194
 195 ACE_INLINE LPSECURITY_ATTRIBUTES
 196 ACE_Process_Options::set_thread_attributes ()
 197 {
 198   thread_attributes_ = &security_buf2_;
 199   return thread_attributes_;
 200 }
 201
 202 ACE_INLINE HANDLE ACE_Process_Options::get_user_token () const
 203 {
 204   return user_token_;
 205 }
 206
 207 #else /* !defined (ACE_WIN32) */
 208
 209 ACE_INLINE ACE_HANDLE
 210 ACE_Process_Options::get_stdin () const
 211 {
 212   return stdin_;
 213 }
 214
 215 ACE_INLINE ACE_HANDLE
 216 ACE_Process_Options::get_stdout () const
 217 {
 218   return stdout_;
 219 }
 220
 221 ACE_INLINE ACE_HANDLE
 222 ACE_Process_Options::get_stderr () const
 223 {
 224   return stderr_;
 225 }
 226
 227 ACE_INLINE bool
 228 ACE_Process_Options::inherit_environment () const
 229 {
 230   return inherit_environment_;
 231 }
 232
 233 ACE_INLINE void
 234 ACE_Process_Options::inherit_environment (bool nv)
 235 {
 236   inherit_environment_ = nv;
 237 }
 238
 239 ACE_INLINE int
 240 ACE_Process_Options::setreugid (const ACE_TCHAR* user)
 241 {
 242 #if !defined (ACE_LACKS_PWD_FUNCTIONS)
 243   struct passwd *ent = ACE_OS::getpwnam (ACE_TEXT_ALWAYS_CHAR (user));
 244
 245   if (ent != 0)
 246     {
 247       this->euid_ = ent->pw_uid;
 248       this->ruid_ = ent->pw_uid;
 249       this->egid_ = ent->pw_gid;
 250       this->rgid_ = ent->pw_gid;
 251       return 0;
 252     }
 253   else
 254     return -1;
 255 #else
 256   ACE_UNUSED_ARG (user);
 257   ACE_NOTSUP_RETURN (-1);
 258 #endif /* ACE_LACKS_PWD_FUNCTIONS */
 259 }
 260
 261 ACE_INLINE void
 262 ACE_Process_Options::setruid (uid_t id)
 263 {
 264   this->ruid_ = id;
 265 }
 266
 267 ACE_INLINE void
 268 ACE_Process_Options::seteuid (uid_t id)
 269 {
 270   this->euid_ = id;
 271 }
 272
 273 ACE_INLINE void
 274 ACE_Process_Options::setrgid (gid_t id)
 275 {
 276   this->rgid_ = id;
 277 }
 278
 279 ACE_INLINE void
 280 ACE_Process_Options::setegid (gid_t id)
 281 {
 282   this->egid_ = id;
 283 }
 284
 285 ACE_INLINE uid_t
 286 ACE_Process_Options::getruid () const
 287 {
 288   return this->ruid_;
 289 }
 290
 291 ACE_INLINE uid_t
 292 ACE_Process_Options::geteuid () const
 293 {
 294   return this->euid_;
 295 }
 296
 297 ACE_INLINE gid_t
 298 ACE_Process_Options::getrgid () const
 299 {
 300   return this->rgid_;
 301 }
 302
 303 ACE_INLINE gid_t
 304 ACE_Process_Options::getegid () const
 305 {
 306   return this->egid_;
 307 }
 308 #endif /* ACE_WIN32 */
 309
 310 ACE_INLINE ACE_TCHAR *
 311 ACE_Process_Options::command_line_buf (size_t *max_lenp)
 312 {
 313   if (max_lenp != 0)
 314     *max_lenp = this->command_line_buf_len_;
 315   return this->command_line_buf_;
 316 }
 317
 318 ACE_INLINE ACE_TCHAR *
 319 ACE_Process_Options::working_directory ()
 320 {
 321   if (working_directory_[0] == '\0')
 322     return 0;
 323   else
 324     return working_directory_;
 325 }
 326
 327 ACE_INLINE void
 328 ACE_Process_Options::working_directory (const char *wd)
 329 {
 330   ACE_OS::strcpy (working_directory_, ACE_TEXT_CHAR_TO_TCHAR (wd));
 331 }
 332
 333 #if defined (ACE_HAS_WCHAR)
 334 ACE_INLINE void
 335 ACE_Process_Options::working_directory (const wchar_t *wd)
 336 {
 337   ACE_OS::strcpy (working_directory_, ACE_TEXT_WCHAR_TO_TCHAR (wd));
 338 }
 339 #endif /* ACE_HAS_WCHAR */
 340
 341 ACE_INLINE void
 342 ACE_Process_Options::process_name (const ACE_TCHAR *p)
 343 {
 344   ACE_OS::strcpy (this->process_name_, p);
 345 }
 346
 347 ACE_INLINE const ACE_TCHAR *
 348 ACE_Process_Options::process_name ()
 349 {
 350   if (process_name_[0] == '\0')
 351     this->process_name (this->command_line_argv ()[0]);
 352
 353   return this->process_name_;
 354 }
 355
 356 ACE_END_VERSIONED_NAMESPACE_DECL