NamedEvent_UNIX.cpp 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. //
  2. // NamedEvent_UNIX.cpp
  3. //
  4. // $Id: //poco/1.4/Foundation/src/NamedEvent_UNIX.cpp#1 $
  5. //
  6. // Library: Foundation
  7. // Package: Processes
  8. // Module: NamedEvent
  9. //
  10. // Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
  11. // and Contributors.
  12. //
  13. // SPDX-License-Identifier: BSL-1.0
  14. //
  15. #include "Poco/NamedEvent_UNIX.h"
  16. #include "Poco/Format.h"
  17. #include "Poco/Exception.h"
  18. #include <fcntl.h>
  19. #include <sys/stat.h>
  20. #include <errno.h>
  21. #if defined(sun) || defined(__APPLE__) || defined(__osf__) || defined(__QNX__) || defined(_AIX)
  22. #include <semaphore.h>
  23. #else
  24. #include <unistd.h>
  25. #include <sys/types.h>
  26. #include <sys/ipc.h>
  27. #include <sys/sem.h>
  28. #endif
  29. namespace Poco {
  30. #if (POCO_OS == POCO_OS_LINUX) || (POCO_OS == POCO_OS_CYGWIN) || (POCO_OS == POCO_OS_FREE_BSD)
  31. union semun
  32. {
  33. int val;
  34. struct semid_ds* buf;
  35. unsigned short int* array;
  36. struct seminfo* __buf;
  37. };
  38. #elif (POCO_OS == POCO_OS_HPUX)
  39. union semun
  40. {
  41. int val;
  42. struct semid_ds* buf;
  43. ushort* array;
  44. };
  45. #endif
  46. NamedEventImpl::NamedEventImpl(const std::string& name):
  47. _name(name)
  48. {
  49. std::string fileName = getFileName();
  50. #if defined(sun) || defined(__APPLE__) || defined(__osf__) || defined(__QNX__) || defined(_AIX)
  51. _sem = sem_open(fileName.c_str(), O_CREAT, S_IRWXU | S_IRWXG | S_IRWXO, 0);
  52. if ((long) _sem == (long) SEM_FAILED)
  53. throw SystemException(Poco::format("cannot create named mutex %s (sem_open() failed, errno=%d)", fileName, errno), _name);
  54. #else
  55. int fd = open(fileName.c_str(), O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
  56. if (fd != -1)
  57. close(fd);
  58. else
  59. throw SystemException(Poco::format("cannot create named event %s (lockfile)", fileName), _name);
  60. key_t key = ftok(fileName.c_str(), 0);
  61. if (key == -1)
  62. throw SystemException(Poco::format("cannot create named mutex %s (ftok() failed, errno=%d)", fileName, errno), _name);
  63. _semid = semget(key, 1, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH | IPC_CREAT | IPC_EXCL);
  64. if (_semid >= 0)
  65. {
  66. union semun arg;
  67. arg.val = 0;
  68. semctl(_semid, 0, SETVAL, arg);
  69. }
  70. else if (errno == EEXIST)
  71. {
  72. _semid = semget(key, 1, 0);
  73. }
  74. else throw SystemException(Poco::format("cannot create named mutex %s (semget() failed, errno=%d)", fileName, errno), _name);
  75. #endif // defined(sun) || defined(__APPLE__) || defined(__osf__) || defined(__QNX__) || defined(_AIX)
  76. }
  77. NamedEventImpl::~NamedEventImpl()
  78. {
  79. #if defined(sun) || defined(__APPLE__) || defined(__osf__) || defined(__QNX__) || defined(_AIX)
  80. sem_close(_sem);
  81. #endif
  82. }
  83. void NamedEventImpl::setImpl()
  84. {
  85. #if defined(sun) || defined(__APPLE__) || defined(__osf__) || defined(__QNX__) || defined(_AIX)
  86. if (sem_post(_sem) != 0)
  87. throw SystemException("cannot set named event", _name);
  88. #else
  89. struct sembuf op;
  90. op.sem_num = 0;
  91. op.sem_op = 1;
  92. op.sem_flg = 0;
  93. if (semop(_semid, &op, 1) != 0)
  94. throw SystemException("cannot set named event", _name);
  95. #endif
  96. }
  97. void NamedEventImpl::waitImpl()
  98. {
  99. #if defined(sun) || defined(__APPLE__) || defined(__osf__) || defined(__QNX__) || defined(_AIX)
  100. int err;
  101. do
  102. {
  103. err = sem_wait(_sem);
  104. }
  105. while (err && errno == EINTR);
  106. if (err) throw SystemException("cannot wait for named event", _name);
  107. #else
  108. struct sembuf op;
  109. op.sem_num = 0;
  110. op.sem_op = -1;
  111. op.sem_flg = 0;
  112. int err;
  113. do
  114. {
  115. err = semop(_semid, &op, 1);
  116. }
  117. while (err && errno == EINTR);
  118. if (err) throw SystemException("cannot wait for named event", _name);
  119. #endif
  120. }
  121. std::string NamedEventImpl::getFileName()
  122. {
  123. #if defined(sun) || defined(__APPLE__) || defined(__QNX__)
  124. std::string fn = "/";
  125. #else
  126. std::string fn = "/tmp/";
  127. #endif
  128. fn.append(_name);
  129. fn.append(".event");
  130. return fn;
  131. }
  132. } // namespace Poco