Index: base/sync_socket_win.cc |
diff --git a/base/sync_socket_win.cc b/base/sync_socket_win.cc |
deleted file mode 100644 |
index e5088162433f2f402a8809c1342142c028e07341..0000000000000000000000000000000000000000 |
--- a/base/sync_socket_win.cc |
+++ /dev/null |
@@ -1,342 +0,0 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "base/sync_socket.h" |
- |
-#include "base/logging.h" |
-#include "base/threading/thread_restrictions.h" |
-#include "base/win/scoped_handle.h" |
- |
-namespace base { |
- |
-using win::ScopedHandle; |
- |
-namespace { |
-// IMPORTANT: do not change how this name is generated because it will break |
-// in sandboxed scenarios as we might have by-name policies that allow pipe |
-// creation. Also keep the secure random number generation. |
-const wchar_t kPipeNameFormat[] = L"\\\\.\\pipe\\chrome.sync.%u.%u.%lu"; |
-const size_t kPipePathMax = arraysize(kPipeNameFormat) + (3 * 10) + 1; |
- |
-// To avoid users sending negative message lengths to Send/Receive |
-// we clamp message lengths, which are size_t, to no more than INT_MAX. |
-const size_t kMaxMessageLength = static_cast<size_t>(INT_MAX); |
- |
-const int kOutBufferSize = 4096; |
-const int kInBufferSize = 4096; |
-const int kDefaultTimeoutMilliSeconds = 1000; |
- |
-bool CreatePairImpl(HANDLE* socket_a, HANDLE* socket_b, bool overlapped) { |
- DCHECK_NE(socket_a, socket_b); |
- DCHECK_EQ(*socket_a, SyncSocket::kInvalidHandle); |
- DCHECK_EQ(*socket_b, SyncSocket::kInvalidHandle); |
- |
- wchar_t name[kPipePathMax]; |
- ScopedHandle handle_a; |
- DWORD flags = PIPE_ACCESS_DUPLEX | FILE_FLAG_FIRST_PIPE_INSTANCE; |
- if (overlapped) |
- flags |= FILE_FLAG_OVERLAPPED; |
- |
- do { |
- unsigned int rnd_name; |
- if (rand_s(&rnd_name) != 0) |
- return false; |
- |
- swprintf(name, kPipePathMax, |
- kPipeNameFormat, |
- GetCurrentProcessId(), |
- GetCurrentThreadId(), |
- rnd_name); |
- |
- handle_a.Set(CreateNamedPipeW( |
- name, |
- flags, |
- PIPE_TYPE_BYTE | PIPE_READMODE_BYTE, |
- 1, |
- kOutBufferSize, |
- kInBufferSize, |
- kDefaultTimeoutMilliSeconds, |
- NULL)); |
- } while (!handle_a.IsValid() && |
- (GetLastError() == ERROR_PIPE_BUSY)); |
- |
- if (!handle_a.IsValid()) { |
- NOTREACHED(); |
- return false; |
- } |
- |
- // The SECURITY_ANONYMOUS flag means that the server side (handle_a) cannot |
- // impersonate the client (handle_b). This allows us not to care which side |
- // ends up in which side of a privilege boundary. |
- flags = SECURITY_SQOS_PRESENT | SECURITY_ANONYMOUS; |
- if (overlapped) |
- flags |= FILE_FLAG_OVERLAPPED; |
- |
- ScopedHandle handle_b(CreateFileW(name, |
- GENERIC_READ | GENERIC_WRITE, |
- 0, // no sharing. |
- NULL, // default security attributes. |
- OPEN_EXISTING, // opens existing pipe. |
- flags, |
- NULL)); // no template file. |
- if (!handle_b.IsValid()) { |
- DPLOG(ERROR) << "CreateFileW failed"; |
- return false; |
- } |
- |
- if (!ConnectNamedPipe(handle_a.Get(), NULL)) { |
- DWORD error = GetLastError(); |
- if (error != ERROR_PIPE_CONNECTED) { |
- DPLOG(ERROR) << "ConnectNamedPipe failed"; |
- return false; |
- } |
- } |
- |
- *socket_a = handle_a.Take(); |
- *socket_b = handle_b.Take(); |
- |
- return true; |
-} |
- |
-// Inline helper to avoid having the cast everywhere. |
-DWORD GetNextChunkSize(size_t current_pos, size_t max_size) { |
- // The following statement is for 64 bit portability. |
- return static_cast<DWORD>(((max_size - current_pos) <= UINT_MAX) ? |
- (max_size - current_pos) : UINT_MAX); |
-} |
- |
-// Template function that supports calling ReadFile or WriteFile in an |
-// overlapped fashion and waits for IO completion. The function also waits |
-// on an event that can be used to cancel the operation. If the operation |
-// is cancelled, the function returns and closes the relevant socket object. |
-template <typename BufferType, typename Function> |
-size_t CancelableFileOperation(Function operation, |
- HANDLE file, |
- BufferType* buffer, |
- size_t length, |
- WaitableEvent* io_event, |
- WaitableEvent* cancel_event, |
- CancelableSyncSocket* socket, |
- DWORD timeout_in_ms) { |
- ThreadRestrictions::AssertIOAllowed(); |
- // The buffer must be byte size or the length check won't make much sense. |
- COMPILE_ASSERT(sizeof(buffer[0]) == sizeof(char), incorrect_buffer_type); |
- DCHECK_GT(length, 0u); |
- DCHECK_LE(length, kMaxMessageLength); |
- DCHECK_NE(file, SyncSocket::kInvalidHandle); |
- |
- // Track the finish time so we can calculate the timeout as data is read. |
- TimeTicks current_time, finish_time; |
- if (timeout_in_ms != INFINITE) { |
- current_time = TimeTicks::Now(); |
- finish_time = |
- current_time + base::TimeDelta::FromMilliseconds(timeout_in_ms); |
- } |
- |
- size_t count = 0; |
- do { |
- // The OVERLAPPED structure will be modified by ReadFile or WriteFile. |
- OVERLAPPED ol = { 0 }; |
- ol.hEvent = io_event->handle(); |
- |
- const DWORD chunk = GetNextChunkSize(count, length); |
- // This is either the ReadFile or WriteFile call depending on whether |
- // we're receiving or sending data. |
- DWORD len = 0; |
- const BOOL operation_ok = operation( |
- file, static_cast<BufferType*>(buffer) + count, chunk, &len, &ol); |
- if (!operation_ok) { |
- if (::GetLastError() == ERROR_IO_PENDING) { |
- HANDLE events[] = { io_event->handle(), cancel_event->handle() }; |
- const int wait_result = WaitForMultipleObjects( |
- arraysize(events), events, FALSE, |
- timeout_in_ms == INFINITE ? |
- timeout_in_ms : |
- static_cast<DWORD>( |
- (finish_time - current_time).InMilliseconds())); |
- if (wait_result != WAIT_OBJECT_0 + 0) { |
- // CancelIo() doesn't synchronously cancel outstanding IO, only marks |
- // outstanding IO for cancellation. We must call GetOverlappedResult() |
- // below to ensure in flight writes complete before returning. |
- CancelIo(file); |
- } |
- |
- // We set the |bWait| parameter to TRUE for GetOverlappedResult() to |
- // ensure writes are complete before returning. |
- if (!GetOverlappedResult(file, &ol, &len, TRUE)) |
- len = 0; |
- |
- if (wait_result == WAIT_OBJECT_0 + 1) { |
- DVLOG(1) << "Shutdown was signaled. Closing socket."; |
- socket->Close(); |
- return count; |
- } |
- |
- // Timeouts will be handled by the while() condition below since |
- // GetOverlappedResult() may complete successfully after CancelIo(). |
- DCHECK(wait_result == WAIT_OBJECT_0 + 0 || wait_result == WAIT_TIMEOUT); |
- } else { |
- break; |
- } |
- } |
- |
- count += len; |
- |
- // Quit the operation if we can't write/read anymore. |
- if (len != chunk) |
- break; |
- |
- // Since TimeTicks::Now() is expensive, only bother updating the time if we |
- // have more work to do. |
- if (timeout_in_ms != INFINITE && count < length) |
- current_time = base::TimeTicks::Now(); |
- } while (count < length && |
- (timeout_in_ms == INFINITE || current_time < finish_time)); |
- |
- return count; |
-} |
- |
-} // namespace |
- |
-#if defined(COMPONENT_BUILD) |
-const SyncSocket::Handle SyncSocket::kInvalidHandle = INVALID_HANDLE_VALUE; |
-#endif |
- |
-SyncSocket::SyncSocket() : handle_(kInvalidHandle) {} |
- |
-SyncSocket::~SyncSocket() { |
- Close(); |
-} |
- |
-// static |
-bool SyncSocket::CreatePair(SyncSocket* socket_a, SyncSocket* socket_b) { |
- return CreatePairImpl(&socket_a->handle_, &socket_b->handle_, false); |
-} |
- |
-// static |
-SyncSocket::Handle SyncSocket::UnwrapHandle( |
- const TransitDescriptor& descriptor) { |
- return descriptor; |
-} |
- |
-bool SyncSocket::PrepareTransitDescriptor(ProcessHandle peer_process_handle, |
- TransitDescriptor* descriptor) { |
- DCHECK(descriptor); |
- if (!::DuplicateHandle(GetCurrentProcess(), handle(), peer_process_handle, |
- descriptor, 0, FALSE, DUPLICATE_SAME_ACCESS)) { |
- DPLOG(ERROR) << "Cannot duplicate socket handle for peer process."; |
- return false; |
- } |
- return true; |
-} |
- |
-bool SyncSocket::Close() { |
- if (handle_ == kInvalidHandle) |
- return true; |
- |
- const BOOL result = CloseHandle(handle_); |
- handle_ = kInvalidHandle; |
- return result == TRUE; |
-} |
- |
-size_t SyncSocket::Send(const void* buffer, size_t length) { |
- ThreadRestrictions::AssertIOAllowed(); |
- DCHECK_GT(length, 0u); |
- DCHECK_LE(length, kMaxMessageLength); |
- DCHECK_NE(handle_, kInvalidHandle); |
- size_t count = 0; |
- while (count < length) { |
- DWORD len; |
- DWORD chunk = GetNextChunkSize(count, length); |
- if (WriteFile(handle_, static_cast<const char*>(buffer) + count, |
- chunk, &len, NULL) == FALSE) { |
- return count; |
- } |
- count += len; |
- } |
- return count; |
-} |
- |
-size_t SyncSocket::ReceiveWithTimeout(void* buffer, |
- size_t length, |
- TimeDelta timeout) { |
- NOTIMPLEMENTED(); |
- return 0; |
-} |
- |
-size_t SyncSocket::Receive(void* buffer, size_t length) { |
- ThreadRestrictions::AssertIOAllowed(); |
- DCHECK_GT(length, 0u); |
- DCHECK_LE(length, kMaxMessageLength); |
- DCHECK_NE(handle_, kInvalidHandle); |
- size_t count = 0; |
- while (count < length) { |
- DWORD len; |
- DWORD chunk = GetNextChunkSize(count, length); |
- if (ReadFile(handle_, static_cast<char*>(buffer) + count, |
- chunk, &len, NULL) == FALSE) { |
- return count; |
- } |
- count += len; |
- } |
- return count; |
-} |
- |
-size_t SyncSocket::Peek() { |
- DWORD available = 0; |
- PeekNamedPipe(handle_, NULL, 0, NULL, &available, NULL); |
- return available; |
-} |
- |
-CancelableSyncSocket::CancelableSyncSocket() |
- : shutdown_event_(true, false), file_operation_(true, false) { |
-} |
- |
-CancelableSyncSocket::CancelableSyncSocket(Handle handle) |
- : SyncSocket(handle), shutdown_event_(true, false), |
- file_operation_(true, false) { |
-} |
- |
-bool CancelableSyncSocket::Shutdown() { |
- // This doesn't shut down the pipe immediately, but subsequent Receive or Send |
- // methods will fail straight away. |
- shutdown_event_.Signal(); |
- return true; |
-} |
- |
-bool CancelableSyncSocket::Close() { |
- const bool result = SyncSocket::Close(); |
- shutdown_event_.Reset(); |
- return result; |
-} |
- |
-size_t CancelableSyncSocket::Send(const void* buffer, size_t length) { |
- static const DWORD kWaitTimeOutInMs = 500; |
- return CancelableFileOperation( |
- &WriteFile, handle_, reinterpret_cast<const char*>(buffer), |
- length, &file_operation_, &shutdown_event_, this, kWaitTimeOutInMs); |
-} |
- |
-size_t CancelableSyncSocket::Receive(void* buffer, size_t length) { |
- return CancelableFileOperation( |
- &ReadFile, handle_, reinterpret_cast<char*>(buffer), length, |
- &file_operation_, &shutdown_event_, this, INFINITE); |
-} |
- |
-size_t CancelableSyncSocket::ReceiveWithTimeout(void* buffer, |
- size_t length, |
- TimeDelta timeout) { |
- return CancelableFileOperation( |
- &ReadFile, handle_, reinterpret_cast<char*>(buffer), length, |
- &file_operation_, &shutdown_event_, this, |
- static_cast<DWORD>(timeout.InMilliseconds())); |
-} |
- |
-// static |
-bool CancelableSyncSocket::CreatePair(CancelableSyncSocket* socket_a, |
- CancelableSyncSocket* socket_b) { |
- return CreatePairImpl(&socket_a->handle_, &socket_b->handle_, true); |
-} |
- |
-} // namespace base |