You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.9 KiB
72 lines
1.9 KiB
/* 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();
|
|
}
|