| /* Distributed under the OSI-approved BSD 3-Clause License. See accompanying |
| file Copyright.txt or https://cmake.org/licensing for details. */ |
| #include "cmPipeConnection.h" |
| |
| #include <utility> |
| |
| #include "cmServer.h" |
| |
| cmPipeConnection::cmPipeConnection(std::string name, |
| cmConnectionBufferStrategy* bufferStrategy) |
| : cmEventBasedConnection(bufferStrategy) |
| , PipeName(std::move(name)) |
| { |
| } |
| |
| void cmPipeConnection::Connect(uv_stream_t* server) |
| { |
| if (this->WriteStream.get()) { |
| // Accept and close all pipes but the first: |
| cm::uv_pipe_ptr rejectPipe; |
| |
| rejectPipe.init(*this->Server->GetLoop(), 0); |
| uv_accept(server, rejectPipe); |
| |
| return; |
| } |
| |
| cm::uv_pipe_ptr ClientPipe; |
| ClientPipe.init(*this->Server->GetLoop(), 0, |
| static_cast<cmEventBasedConnection*>(this)); |
| |
| if (uv_accept(server, ClientPipe) != 0) { |
| return; |
| } |
| |
| uv_read_start(ClientPipe, on_alloc_buffer, on_read); |
| WriteStream = std::move(ClientPipe); |
| Server->OnConnected(this); |
| } |
| |
| bool cmPipeConnection::OnServeStart(std::string* errorMessage) |
| { |
| this->ServerPipe.init(*this->Server->GetLoop(), 0, |
| static_cast<cmEventBasedConnection*>(this)); |
| |
| int r; |
| if ((r = uv_pipe_bind(this->ServerPipe, this->PipeName.c_str())) != 0) { |
| *errorMessage = std::string("Internal Error with ") + this->PipeName + |
| ": " + uv_err_name(r); |
| return false; |
| } |
| |
| if ((r = uv_listen(this->ServerPipe, 1, on_new_connection)) != 0) { |
| *errorMessage = std::string("Internal Error listening on ") + |
| this->PipeName + ": " + uv_err_name(r); |
| return false; |
| } |
| |
| return cmConnection::OnServeStart(errorMessage); |
| } |
| |
| bool cmPipeConnection::OnConnectionShuttingDown() |
| { |
| if (this->WriteStream.get()) { |
| this->WriteStream->data = nullptr; |
| } |
| |
| this->ServerPipe.reset(); |
| |
| return cmEventBasedConnection::OnConnectionShuttingDown(); |
| } |