naiveproxy/net/socket/socket_descriptor.cc
2018-02-02 05:49:39 -05:00

57 lines
1.5 KiB
C++

// Copyright 2013 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 "net/socket/socket_descriptor.h"
#if defined(OS_POSIX)
#include <sys/types.h>
#include <sys/socket.h>
#endif
#if defined(OS_WIN)
#include <ws2tcpip.h>
#include "net/base/winsock_init.h"
#endif
#if defined(OS_MACOSX)
#include <unistd.h>
#endif
namespace net {
SocketDescriptor CreatePlatformSocket(int family, int type, int protocol) {
#if defined(OS_WIN)
EnsureWinsockInit();
SocketDescriptor result = ::WSASocket(family, type, protocol, nullptr, 0,
WSA_FLAG_OVERLAPPED);
if (result != kInvalidSocket && family == AF_INET6) {
DWORD value = 0;
if (setsockopt(result, IPPROTO_IPV6, IPV6_V6ONLY,
reinterpret_cast<const char*>(&value), sizeof(value))) {
closesocket(result);
return kInvalidSocket;
}
}
return result;
#else // OS_WIN
SocketDescriptor result = ::socket(family, type, protocol);
#if defined(OS_MACOSX)
// Disable SIGPIPE on this socket. Although Chromium globally disables
// SIGPIPE, the net stack may be used in other consumers which do not do
// this. SO_NOSIGPIPE is a Mac-only API. On Linux, it is a flag on send.
if (result != kInvalidSocket) {
int value = 1;
if (setsockopt(result, SOL_SOCKET, SO_NOSIGPIPE, &value, sizeof(value))) {
close(result);
return kInvalidSocket;
}
}
#endif
return result;
#endif // OS_WIN
}
} // namespace net