mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-08 02:18:53 +00:00
44 lines
1.1 KiB
C++
44 lines
1.1 KiB
C++
/*
|
|
* Copyright (c) 2014-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under both the Apache 2.0 license (found in the
|
|
* LICENSE file in the root directory of this source tree) and the GPLv2 (found
|
|
* in the COPYING file in the root directory of this source tree).
|
|
* You may select, at your option, one of the above-listed licenses.
|
|
*/
|
|
|
|
#include <osquery/logger.h>
|
|
|
|
#include "osquery/dispatcher/io_service.h"
|
|
|
|
namespace osquery {
|
|
|
|
void IOServiceRunner::start() {
|
|
boost::asio::io_service::work work(IOService::get());
|
|
for (;;) {
|
|
try {
|
|
IOService::get().run();
|
|
return;
|
|
} catch (const std::exception& e) {
|
|
LOG(WARNING) << "IOServiceRunner: handler exception: " << e.what();
|
|
} catch (...) {
|
|
LOG(WARNING) << "IOServiceRunner: unknown handler exception";
|
|
}
|
|
}
|
|
}
|
|
|
|
void IOServiceRunner::stop() {
|
|
IOService::get().stop();
|
|
}
|
|
|
|
void startIOService() {
|
|
// Windows service needs notifications that threads should die: #4235
|
|
#ifdef WIN32
|
|
boost::asio::detail::win_thread::set_terminate_threads(true);
|
|
#endif
|
|
|
|
Dispatcher::addService(std::make_shared<IOServiceRunner>());
|
|
}
|
|
} // namespace osquery
|