ZeroTierOneService.h 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. /*
  2. * ZeroTier One - Network Virtualization Everywhere
  3. * Copyright (C) 2011-2015 ZeroTier, Inc.
  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. #pragma once
  28. #include <stdio.h>
  29. #include "ServiceBase.h"
  30. #include <string>
  31. #include "../../node/Mutex.hpp"
  32. #include "../../osdep/Thread.hpp"
  33. #include "../../service/OneService.hpp"
  34. // Uncomment to make debugging Windows services suck slightly less hard.
  35. //#define ZT_DEBUG_SERVICE "C:\\ZeroTierOneServiceDebugLog.txt"
  36. #ifdef ZT_DEBUG_SERVICE
  37. extern FILE *SVCDBGfile;
  38. extern ZeroTier::Mutex SVCDBGfile_m;
  39. #define ZT_SVCDBG(f,...) { SVCDBGfile_m.lock(); fprintf(SVCDBGfile,f,##__VA_ARGS__); fflush(SVCDBGfile); SVCDBGfile_m.unlock(); }
  40. #else
  41. #define ZT_SVCDBG(f,...) {}
  42. #endif
  43. #define ZT_SERVICE_NAME "ZeroTierOneService"
  44. #define ZT_SERVICE_DISPLAY_NAME "ZeroTier One"
  45. #define ZT_SERVICE_START_TYPE SERVICE_AUTO_START
  46. #define ZT_SERVICE_DEPENDENCIES ""
  47. //#define ZT_SERVICE_ACCOUNT "NT AUTHORITY\\LocalService"
  48. #define ZT_SERVICE_ACCOUNT NULL
  49. #define ZT_SERVICE_PASSWORD NULL
  50. class ZeroTierOneService : public CServiceBase
  51. {
  52. public:
  53. ZeroTierOneService();
  54. virtual ~ZeroTierOneService(void);
  55. /**
  56. * Thread main method; do not call elsewhere
  57. */
  58. void threadMain()
  59. throw();
  60. // Returns false on failure.
  61. static bool doStartUpgrade(const std::string &msiPath);
  62. protected:
  63. virtual void OnStart(DWORD dwArgc, PSTR *pszArgv);
  64. virtual void OnStop();
  65. virtual void OnShutdown();
  66. private:
  67. ZeroTier::OneService *volatile _service;
  68. ZeroTier::Mutex _lock;
  69. ZeroTier::Thread _thread;
  70. };