IpcListener.hpp 3.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. /*
  2. * ZeroTier One - Global Peer to Peer Ethernet
  3. * Copyright (C) 2011-2014 ZeroTier Networks LLC
  4. *
  5. * This program is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. *
  18. * --
  19. *
  20. * ZeroTier may be used and distributed under the terms of the GPLv3, which
  21. * are available at: http://www.gnu.org/licenses/gpl-3.0.html
  22. *
  23. * If you would like to embed ZeroTier into a commercial application or
  24. * redistribute it in a modified binary form, please contact ZeroTier Networks
  25. * LLC. Start here: http://www.zerotier.com/
  26. */
  27. #ifndef ZT_IPCLISTENER_HPP
  28. #define ZT_IPCLISTENER_HPP
  29. #include "Constants.hpp"
  30. #include "Thread.hpp"
  31. #include "NonCopyable.hpp"
  32. #include "IpcConnection.hpp"
  33. #include <string>
  34. #include <stdexcept>
  35. #ifdef __WINDOWS__
  36. #include <WinSock2.h>
  37. #include <Windows.h>
  38. #define ZT_IPC_ENDPOINT_BASE "\\\\.\\pipe\\ZeroTierOne-"
  39. #else
  40. #define ZT_IPC_ENDPOINT_BASE "/tmp/.ZeroTierOne-"
  41. #endif
  42. namespace ZeroTier {
  43. /**
  44. * IPC incoming connection listener (Unix domain sockets or named pipes on Windows)
  45. */
  46. class IpcListener : NonCopyable
  47. {
  48. public:
  49. /**
  50. * Listen for IPC connections
  51. *
  52. * The supplied handler is passed on to incoming instances of IpcConnection. When
  53. * a connection is first opened, it is called with IPC_EVENT_NEW_CONNECTION. The
  54. * receiver must take ownership of the connection object. When a connection is
  55. * closed, IPC_EVENT_CONNECTION_CLOSING is generated. At this point (or after) the
  56. * receiver must delete the object. IPC_EVENT_COMMAND is generated when lines of
  57. * text are read, and in this cases the last argument is not NULL. No closed event
  58. * is generated in the event of manual delete if the connection is still open.
  59. *
  60. * Yeah, this whole callback model sort of sucks. Might rethink and replace with
  61. * some kind of actor model or something if it gets too unweildy. But for now the
  62. * use cases are simple enough that it's not too bad.
  63. *
  64. * @param commandHandler Function to call for each command
  65. * @param arg First argument to pass to handler
  66. * @throws std::runtime_error Unable to bind to endpoint
  67. */
  68. IpcListener(const char *ep,void (*commandHandler)(void *,IpcConnection *,IpcConnection::EventType,const char *),void *arg);
  69. ~IpcListener();
  70. void threadMain()
  71. throw();
  72. private:
  73. std::string _endpoint;
  74. void (*_handler)(void *,IpcConnection *,IpcConnection::EventType,const char *);
  75. void *_arg;
  76. #ifdef __WINDOWS__
  77. volatile bool _run;
  78. volatile bool _running;
  79. #else
  80. volatile int _sock;
  81. #endif
  82. Thread _thread;
  83. };
  84. } // namespace ZeroTier
  85. #endif