| /* |
| * |
| * Copyright 2015 gRPC authors. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| |
| /* Windows implementation for gpr threads. */ |
| |
| #include <grpc/support/port_platform.h> |
| |
| #ifdef GPR_WINDOWS |
| |
| #include "src/core/lib/gprpp/thd.h" |
| |
| #include <grpc/support/alloc.h> |
| #include <grpc/support/log.h> |
| #include <grpc/support/thd_id.h> |
| #include <string.h> |
| |
| #include "src/core/lib/gprpp/memory.h" |
| |
| #if defined(_MSC_VER) |
| #define thread_local __declspec(thread) |
| #define WIN_LAMBDA |
| #elif defined(__GNUC__) |
| #define thread_local __thread |
| #define WIN_LAMBDA WINAPI |
| #else |
| #error "Unknown compiler - please file a bug report" |
| #endif |
| |
| namespace { |
| class ThreadInternalsWindows; |
| struct thd_info { |
| ThreadInternalsWindows* thread; |
| void (*body)(void* arg); /* body of a thread */ |
| void* arg; /* argument to a thread */ |
| HANDLE join_event; /* the join event */ |
| }; |
| |
| thread_local struct thd_info* g_thd_info; |
| |
| class ThreadInternalsWindows |
| : public grpc_core::internal::ThreadInternalsInterface { |
| public: |
| ThreadInternalsWindows(void (*thd_body)(void* arg), void* arg, bool* success) |
| : started_(false) { |
| gpr_mu_init(&mu_); |
| gpr_cv_init(&ready_); |
| |
| HANDLE handle; |
| info_ = (struct thd_info*)gpr_malloc(sizeof(*info_)); |
| info_->thread = this; |
| info_->body = thd_body; |
| info_->arg = arg; |
| |
| info_->join_event = CreateEvent(nullptr, FALSE, FALSE, nullptr); |
| if (info_->join_event == nullptr) { |
| gpr_free(info_); |
| *success = false; |
| } else { |
| handle = CreateThread( |
| nullptr, 64 * 1024, |
| [](void* v) WIN_LAMBDA -> DWORD { |
| g_thd_info = static_cast<thd_info*>(v); |
| gpr_mu_lock(&g_thd_info->thread->mu_); |
| while (!g_thd_info->thread->started_) { |
| gpr_cv_wait(&g_thd_info->thread->ready_, &g_thd_info->thread->mu_, |
| gpr_inf_future(GPR_CLOCK_MONOTONIC)); |
| } |
| gpr_mu_unlock(&g_thd_info->thread->mu_); |
| g_thd_info->body(g_thd_info->arg); |
| BOOL ret = SetEvent(g_thd_info->join_event); |
| GPR_ASSERT(ret); |
| return 0; |
| }, |
| info_, 0, nullptr); |
| if (handle == nullptr) { |
| destroy_thread(); |
| *success = false; |
| } else { |
| CloseHandle(handle); |
| *success = true; |
| } |
| } |
| } |
| |
| ~ThreadInternalsWindows() override { |
| gpr_mu_destroy(&mu_); |
| gpr_cv_destroy(&ready_); |
| } |
| |
| void Start() override { |
| gpr_mu_lock(&mu_); |
| started_ = true; |
| gpr_cv_signal(&ready_); |
| gpr_mu_unlock(&mu_); |
| } |
| |
| void Join() override { |
| DWORD ret = WaitForSingleObject(info_->join_event, INFINITE); |
| GPR_ASSERT(ret == WAIT_OBJECT_0); |
| destroy_thread(); |
| } |
| |
| private: |
| void destroy_thread() { |
| CloseHandle(info_->join_event); |
| gpr_free(info_); |
| } |
| |
| gpr_mu mu_; |
| gpr_cv ready_; |
| bool started_; |
| thd_info* info_; |
| }; |
| |
| } // namespace |
| |
| namespace grpc_core { |
| |
| Thread::Thread(const char* thd_name, void (*thd_body)(void* arg), void* arg, |
| bool* success) { |
| bool outcome = false; |
| impl_ = grpc_core::New<ThreadInternalsWindows>(thd_body, arg, &outcome); |
| if (outcome) { |
| state_ = ALIVE; |
| } else { |
| state_ = FAILED; |
| grpc_core::Delete(impl_); |
| impl_ = nullptr; |
| } |
| |
| if (success != nullptr) { |
| *success = outcome; |
| } |
| } |
| |
| } // namespace grpc_core |
| |
| gpr_thd_id gpr_thd_currentid(void) { return (gpr_thd_id)g_thd_info; } |
| |
| #endif /* GPR_WINDOWS */ |