? ? ? ? ?Socket.cc和SocketsOps.cc用來調用一個套接字設置選項和底層的socket API，也包括一些類型轉換，記錄一下，以后可以直接拿來用。

Socket.cc

// Copyright 2010, Shuo Chen. All rights reserved. // http://code.google.com/p/muduo/ // // Use of this source code is governed by a BSD-style license // that can be found in the License file.// Author: Shuo Chen (chenshuo at chenshuo dot com)#include "muduo/net/Socket.h"#include "muduo/base/Logging.h" #include "muduo/net/InetAddress.h" #include "muduo/net/SocketsOps.h"#include <netinet/in.h> #include <netinet/tcp.h> #include <stdio.h> // snprintfusing namespace muduo; using namespace muduo::net;Socket::~Socket() {sockets::close(sockfd_); }bool Socket::getTcpInfo(struct tcp_info* tcpi) const {socklen_t len = sizeof(*tcpi);memZero(tcpi, len);return ::getsockopt(sockfd_, SOL_TCP, TCP_INFO, tcpi, &len) == 0; }bool Socket::getTcpInfoString(char* buf, int len) const {struct tcp_info tcpi;bool ok = getTcpInfo(&tcpi);if (ok){snprintf(buf, len, "unrecovered=%u ""rto=%u ato=%u snd_mss=%u rcv_mss=%u ""lost=%u retrans=%u rtt=%u rttvar=%u ""sshthresh=%u cwnd=%u total_retrans=%u",tcpi.tcpi_retransmits, // Number of unrecovered [RTO] timeoutstcpi.tcpi_rto, // Retransmit timeout in usectcpi.tcpi_ato, // Predicted tick of soft clock in usectcpi.tcpi_snd_mss,tcpi.tcpi_rcv_mss,tcpi.tcpi_lost, // Lost packetstcpi.tcpi_retrans, // Retransmitted packets outtcpi.tcpi_rtt, // Smoothed round trip time in usectcpi.tcpi_rttvar, // Medium deviationtcpi.tcpi_snd_ssthresh,tcpi.tcpi_snd_cwnd,tcpi.tcpi_total_retrans); // Total retransmits for entire connection}return ok; }void Socket::bindAddress(const InetAddress& addr) {sockets::bindOrDie(sockfd_, addr.getSockAddr()); }void Socket::listen() {sockets::listenOrDie(sockfd_); }int Socket::accept(InetAddress* peeraddr) {struct sockaddr_in6 addr;memZero(&addr, sizeof addr);int connfd = sockets::accept(sockfd_, &addr);if (connfd >= 0){peeraddr->setSockAddrInet6(addr);}return connfd; }void Socket::shutdownWrite() {sockets::shutdownWrite(sockfd_); }void Socket::setTcpNoDelay(bool on) {int optval = on ? 1 : 0;::setsockopt(sockfd_, IPPROTO_TCP, TCP_NODELAY,&optval, static_cast<socklen_t>(sizeof optval));// FIXME CHECK }void Socket::setReuseAddr(bool on) {int optval = on ? 1 : 0;::setsockopt(sockfd_, SOL_SOCKET, SO_REUSEADDR,&optval, static_cast<socklen_t>(sizeof optval));// FIXME CHECK }void Socket::setReusePort(bool on) { #ifdef SO_REUSEPORTint optval = on ? 1 : 0;int ret = ::setsockopt(sockfd_, SOL_SOCKET, SO_REUSEPORT,&optval, static_cast<socklen_t>(sizeof optval));if (ret < 0 && on){LOG_SYSERR << "SO_REUSEPORT failed.";} #elseif (on){LOG_ERROR << "SO_REUSEPORT is not supported.";} #endif }void Socket::setKeepAlive(bool on) {int optval = on ? 1 : 0;::setsockopt(sockfd_, SOL_SOCKET, SO_KEEPALIVE,&optval, static_cast<socklen_t>(sizeof optval));// FIXME CHECK }

SocketsOps.cc

// Copyright 2010, Shuo Chen. All rights reserved. // http://code.google.com/p/muduo/ // // Use of this source code is governed by a BSD-style license // that can be found in the License file.// Author: Shuo Chen (chenshuo at chenshuo dot com)#include "muduo/net/SocketsOps.h"#include "muduo/base/Logging.h" #include "muduo/base/Types.h" #include "muduo/net/Endian.h"#include <errno.h> #include <fcntl.h> #include <stdio.h> // snprintf #include <sys/socket.h> #include <sys/uio.h> // readv #include <unistd.h>using namespace muduo; using namespace muduo::net;namespace {typedef struct sockaddr SA;#if VALGRIND || defined (NO_ACCEPT4) void setNonBlockAndCloseOnExec(int sockfd) {// non-blockint flags = ::fcntl(sockfd, F_GETFL, 0);flags |= O_NONBLOCK;int ret = ::fcntl(sockfd, F_SETFL, flags);// FIXME check// close-on-execflags = ::fcntl(sockfd, F_GETFD, 0);flags |= FD_CLOEXEC;ret = ::fcntl(sockfd, F_SETFD, flags);// FIXME check(void)ret; } #endif} // namespaceconst struct sockaddr* sockets::sockaddr_cast(const struct sockaddr_in6* addr) {return static_cast<const struct sockaddr*>(implicit_cast<const void*>(addr)); }struct sockaddr* sockets::sockaddr_cast(struct sockaddr_in6* addr) {return static_cast<struct sockaddr*>(implicit_cast<void*>(addr)); }const struct sockaddr* sockets::sockaddr_cast(const struct sockaddr_in* addr) {return static_cast<const struct sockaddr*>(implicit_cast<const void*>(addr)); }const struct sockaddr_in* sockets::sockaddr_in_cast(const struct sockaddr* addr) {return static_cast<const struct sockaddr_in*>(implicit_cast<const void*>(addr)); }const struct sockaddr_in6* sockets::sockaddr_in6_cast(const struct sockaddr* addr) {return static_cast<const struct sockaddr_in6*>(implicit_cast<const void*>(addr)); }int sockets::createNonblockingOrDie(sa_family_t family) { #if VALGRINDint sockfd = ::socket(family, SOCK_STREAM, IPPROTO_TCP);if (sockfd < 0){LOG_SYSFATAL << "sockets::createNonblockingOrDie";}setNonBlockAndCloseOnExec(sockfd); #elseint sockfd = ::socket(family, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, IPPROTO_TCP);if (sockfd < 0){LOG_SYSFATAL << "sockets::createNonblockingOrDie";} #endifreturn sockfd; }void sockets::bindOrDie(int sockfd, const struct sockaddr* addr) {int ret = ::bind(sockfd, addr, static_cast<socklen_t>(sizeof(struct sockaddr_in6)));if (ret < 0){LOG_SYSFATAL << "sockets::bindOrDie";} }void sockets::listenOrDie(int sockfd) {int ret = ::listen(sockfd, SOMAXCONN);if (ret < 0){LOG_SYSFATAL << "sockets::listenOrDie";} }int sockets::accept(int sockfd, struct sockaddr_in6* addr) {socklen_t addrlen = static_cast<socklen_t>(sizeof *addr); #if VALGRIND || defined (NO_ACCEPT4)int connfd = ::accept(sockfd, sockaddr_cast(addr), &addrlen);setNonBlockAndCloseOnExec(connfd); #elseint connfd = ::accept4(sockfd, sockaddr_cast(addr),&addrlen, SOCK_NONBLOCK | SOCK_CLOEXEC); #endifif (connfd < 0){int savedErrno = errno;LOG_SYSERR << "Socket::accept";switch (savedErrno){case EAGAIN:case ECONNABORTED:case EINTR:case EPROTO: // ???case EPERM:case EMFILE: // per-process lmit of open file desctiptor ???// expected errorserrno = savedErrno;break;case EBADF:case EFAULT:case EINVAL:case ENFILE:case ENOBUFS:case ENOMEM:case ENOTSOCK:case EOPNOTSUPP:// unexpected errorsLOG_FATAL << "unexpected error of ::accept " << savedErrno;break;default:LOG_FATAL << "unknown error of ::accept " << savedErrno;break;}}return connfd; }int sockets::connect(int sockfd, const struct sockaddr* addr) {return ::connect(sockfd, addr, static_cast<socklen_t>(sizeof(struct sockaddr_in6))); }ssize_t sockets::read(int sockfd, void *buf, size_t count) {return ::read(sockfd, buf, count); }ssize_t sockets::readv(int sockfd, const struct iovec *iov, int iovcnt) {return ::readv(sockfd, iov, iovcnt); }ssize_t sockets::write(int sockfd, const void *buf, size_t count) {return ::write(sockfd, buf, count); }void sockets::close(int sockfd) {if (::close(sockfd) < 0){LOG_SYSERR << "sockets::close";} }void sockets::shutdownWrite(int sockfd) {if (::shutdown(sockfd, SHUT_WR) < 0){LOG_SYSERR << "sockets::shutdownWrite";} }void sockets::toIpPort(char* buf, size_t size,const struct sockaddr* addr) {toIp(buf,size, addr);size_t end = ::strlen(buf);const struct sockaddr_in* addr4 = sockaddr_in_cast(addr);uint16_t port = sockets::networkToHost16(addr4->sin_port);assert(size > end);snprintf(buf+end, size-end, ":%u", port); }void sockets::toIp(char* buf, size_t size,const struct sockaddr* addr) {if (addr->sa_family == AF_INET){assert(size >= INET_ADDRSTRLEN);const struct sockaddr_in* addr4 = sockaddr_in_cast(addr);::inet_ntop(AF_INET, &addr4->sin_addr, buf, static_cast<socklen_t>(size));}else if (addr->sa_family == AF_INET6){assert(size >= INET6_ADDRSTRLEN);const struct sockaddr_in6* addr6 = sockaddr_in6_cast(addr);::inet_ntop(AF_INET6, &addr6->sin6_addr, buf, static_cast<socklen_t>(size));} }void sockets::fromIpPort(const char* ip, uint16_t port,struct sockaddr_in* addr) {addr->sin_family = AF_INET;addr->sin_port = hostToNetwork16(port);if (::inet_pton(AF_INET, ip, &addr->sin_addr) <= 0){LOG_SYSERR << "sockets::fromIpPort";} }void sockets::fromIpPort(const char* ip, uint16_t port,struct sockaddr_in6* addr) {addr->sin6_family = AF_INET6;addr->sin6_port = hostToNetwork16(port);if (::inet_pton(AF_INET6, ip, &addr->sin6_addr) <= 0){LOG_SYSERR << "sockets::fromIpPort";} }int sockets::getSocketError(int sockfd) {int optval;socklen_t optlen = static_cast<socklen_t>(sizeof optval);if (::getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &optval, &optlen) < 0){return errno;}else{return optval;} }struct sockaddr_in6 sockets::getLocalAddr(int sockfd) {struct sockaddr_in6 localaddr;memZero(&localaddr, sizeof localaddr);socklen_t addrlen = static_cast<socklen_t>(sizeof localaddr);//static_cast強制類型轉換if (::getsockname(sockfd, sockaddr_cast(&localaddr), &addrlen) < 0){LOG_SYSERR << "sockets::getLocalAddr";}return localaddr; }struct sockaddr_in6 sockets::getPeerAddr(int sockfd) {struct sockaddr_in6 peeraddr;memZero(&peeraddr, sizeof peeraddr);socklen_t addrlen = static_cast<socklen_t>(sizeof peeraddr);if (::getpeername(sockfd, sockaddr_cast(&peeraddr), &addrlen) < 0){LOG_SYSERR << "sockets::getPeerAddr";}return peeraddr; }bool sockets::isSelfConnect(int sockfd) {struct sockaddr_in6 localaddr = getLocalAddr(sockfd);struct sockaddr_in6 peeraddr = getPeerAddr(sockfd);if (localaddr.sin6_family == AF_INET){const struct sockaddr_in* laddr4 = reinterpret_cast<struct sockaddr_in*>(&localaddr);const struct sockaddr_in* raddr4 = reinterpret_cast<struct sockaddr_in*>(&peeraddr);return laddr4->sin_port == raddr4->sin_port&& laddr4->sin_addr.s_addr == raddr4->sin_addr.s_addr;}else if (localaddr.sin6_family == AF_INET6){return localaddr.sin6_port == peeraddr.sin6_port&& memcmp(&localaddr.sin6_addr, &peeraddr.sin6_addr, sizeof localaddr.sin6_addr) == 0;}else{return false;} }

?

《新程序員》：云原生和全面數字化實踐50位技術專家共同創作，文字、視頻、音頻交互閱讀

總結

以上是生活随笔為你收集整理的muduo之Socket和SocketsOps的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。