src/node_watchdog.cc

   1 // Copyright Joyent, Inc. and other Node contributors.
   2 //
   3 // Permission is hereby granted, free of charge, to any person obtaining a
   4 // copy of this software and associated documentation files (the
   5 // "Software"), to deal in the Software without restriction, including
   6 // without limitation the rights to use, copy, modify, merge, publish,
   7 // distribute, sublicense, and/or sell copies of the Software, and to permit
   8 // persons to whom the Software is furnished to do so, subject to the
   9 // following conditions:
  10 //
  11 // The above copyright notice and this permission notice shall be included
  12 // in all copies or substantial portions of the Software.
  13 //
  14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  15 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  16 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
  17 // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
  18 // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  19 // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  20 // USE OR OTHER DEALINGS IN THE SOFTWARE.
  21
  22 #include "node_watchdog.h"
  23 #include "util.h"
  24 #include <assert.h>
  25
  26 namespace node {
  27
  28 using v8::V8;
  29
  30
  31 Watchdog::Watchdog(uint64_t ms) : destroyed_(false) {
  32   int rc;
  33   loop_ = new uv_loop_t;
  34   CHECK(loop_);
  35   rc = uv_loop_init(loop_);
  36   CHECK_EQ(0, rc);
  37
  38   rc = uv_async_init(loop_, &async_, &Watchdog::Async);
  39   CHECK_EQ(0, rc);
  40
  41   rc = uv_timer_init(loop_, &timer_);
  42   CHECK_EQ(0, rc);
  43
  44   rc = uv_timer_start(&timer_, &Watchdog::Timer, ms, 0);
  45   CHECK_EQ(0, rc);
  46
  47   rc = uv_thread_create(&thread_, &Watchdog::Run, this);
  48   CHECK_EQ(0, rc);
  49 }
  50
  51
  52 Watchdog::~Watchdog() {
  53   Destroy();
  54 }
  55
  56
  57 void Watchdog::Dispose() {
  58   Destroy();
  59 }
  60
  61
  62 void Watchdog::Destroy() {
  63   if (destroyed_) {
  64     return;
  65   }
  66
  67   uv_async_send(&async_);
  68   uv_thread_join(&thread_);
  69
  70   uv_close(reinterpret_cast<uv_handle_t*>(&async_), NULL);
  71
  72   // UV_RUN_DEFAULT so that libuv has a chance to clean up.
  73   uv_run(loop_, UV_RUN_DEFAULT);
  74
  75   int rc = uv_loop_close(loop_);
  76   CHECK_EQ(0, rc);
  77   delete loop_;
  78   loop_ = NULL;
  79
  80   destroyed_ = true;
  81 }
  82
  83
  84 void Watchdog::Run(void* arg) {
  85   Watchdog* wd = static_cast<Watchdog*>(arg);
  86
  87   // UV_RUN_ONCE so async_ or timer_ wakeup exits uv_run() call.
  88   uv_run(wd->loop_, UV_RUN_ONCE);
  89
  90   // Loop ref count reaches zero when both handles are closed.
  91   // Close the timer handle on this side and let Destroy() close async_
  92   uv_close(reinterpret_cast<uv_handle_t*>(&wd->timer_), NULL);
  93 }
  94
  95
  96 void Watchdog::Async(uv_async_t* async, int status) {
  97 }
  98
  99
 100 void Watchdog::Timer(uv_timer_t* timer, int status) {
 101   V8::TerminateExecution();
 102 }
 103
 104
 105 }  // namespace node