| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/tools/quic/quic_socket_utils.h" | 5 #include "net/tools/quic/quic_socket_utils.h" |
| 6 | 6 |
| 7 #include <errno.h> | 7 #include <errno.h> |
| 8 #include <netinet/in.h> | 8 #include <netinet/in.h> |
| 9 #include <string.h> | 9 #include <string.h> |
| 10 #include <sys/socket.h> | 10 #include <sys/socket.h> |
| (...skipping 136 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 147 sizeof(struct sockaddr_in))); | 147 sizeof(struct sockaddr_in))); |
| 148 } else if (raw_address.ss_family == AF_INET6) { | 148 } else if (raw_address.ss_family == AF_INET6) { |
| 149 CHECK(peer_address->FromSockAddr( | 149 CHECK(peer_address->FromSockAddr( |
| 150 reinterpret_cast<const sockaddr*>(&raw_address), | 150 reinterpret_cast<const sockaddr*>(&raw_address), |
| 151 sizeof(struct sockaddr_in6))); | 151 sizeof(struct sockaddr_in6))); |
| 152 } | 152 } |
| 153 | 153 |
| 154 return bytes_read; | 154 return bytes_read; |
| 155 } | 155 } |
| 156 | 156 |
| 157 void QuicSocketUtils::SetIpInfoInCmsg(const IPAddressNumber& self_address, | 157 size_t QuicSocketUtils::SetIpInfoInCmsg(const IPAddressNumber& self_address, |
| 158 cmsghdr* cmsg) { | 158 cmsghdr* cmsg) { |
| 159 if (GetAddressFamily(self_address) == ADDRESS_FAMILY_IPV4) { | 159 if (GetAddressFamily(self_address) == ADDRESS_FAMILY_IPV4) { |
| 160 cmsg->cmsg_len = CMSG_LEN(sizeof(in_pktinfo)); | 160 cmsg->cmsg_len = CMSG_LEN(sizeof(in_pktinfo)); |
| 161 cmsg->cmsg_level = IPPROTO_IP; | 161 cmsg->cmsg_level = IPPROTO_IP; |
| 162 cmsg->cmsg_type = IP_PKTINFO; | 162 cmsg->cmsg_type = IP_PKTINFO; |
| 163 in_pktinfo* pktinfo = reinterpret_cast<in_pktinfo*>(CMSG_DATA(cmsg)); | 163 in_pktinfo* pktinfo = reinterpret_cast<in_pktinfo*>(CMSG_DATA(cmsg)); |
| 164 memset(pktinfo, 0, sizeof(in_pktinfo)); | 164 memset(pktinfo, 0, sizeof(in_pktinfo)); |
| 165 pktinfo->ipi_ifindex = 0; | 165 pktinfo->ipi_ifindex = 0; |
| 166 memcpy(&pktinfo->ipi_spec_dst, &self_address[0], self_address.size()); | 166 memcpy(&pktinfo->ipi_spec_dst, &self_address[0], self_address.size()); |
| 167 return sizeof(in_pktinfo); |
| 167 } else { | 168 } else { |
| 168 cmsg->cmsg_len = CMSG_LEN(sizeof(in6_pktinfo)); | 169 cmsg->cmsg_len = CMSG_LEN(sizeof(in6_pktinfo)); |
| 169 cmsg->cmsg_level = IPPROTO_IPV6; | 170 cmsg->cmsg_level = IPPROTO_IPV6; |
| 170 cmsg->cmsg_type = IPV6_PKTINFO; | 171 cmsg->cmsg_type = IPV6_PKTINFO; |
| 171 in6_pktinfo* pktinfo = reinterpret_cast<in6_pktinfo*>(CMSG_DATA(cmsg)); | 172 in6_pktinfo* pktinfo = reinterpret_cast<in6_pktinfo*>(CMSG_DATA(cmsg)); |
| 172 memset(pktinfo, 0, sizeof(in6_pktinfo)); | 173 memset(pktinfo, 0, sizeof(in6_pktinfo)); |
| 173 memcpy(&pktinfo->ipi6_addr, &self_address[0], self_address.size()); | 174 memcpy(&pktinfo->ipi6_addr, &self_address[0], self_address.size()); |
| 175 return sizeof(in6_pktinfo); |
| 174 } | 176 } |
| 175 } | 177 } |
| 176 | 178 |
| 177 // static | 179 // static |
| 178 WriteResult QuicSocketUtils::WritePacket(int fd, | 180 WriteResult QuicSocketUtils::WritePacket(int fd, |
| 179 const char* buffer, | 181 const char* buffer, |
| 180 size_t buf_len, | 182 size_t buf_len, |
| 181 const IPAddressNumber& self_address, | 183 const IPAddressNumber& self_address, |
| 182 const IPEndPoint& peer_address) { | 184 const IPEndPoint& peer_address) { |
| 183 sockaddr_storage raw_address; | 185 sockaddr_storage raw_address; |
| (...skipping 30 matching lines...) Expand all Loading... |
| 214 int rc = sendmsg(fd, &hdr, 0); | 216 int rc = sendmsg(fd, &hdr, 0); |
| 215 if (rc >= 0) { | 217 if (rc >= 0) { |
| 216 return WriteResult(WRITE_STATUS_OK, rc); | 218 return WriteResult(WRITE_STATUS_OK, rc); |
| 217 } | 219 } |
| 218 return WriteResult((errno == EAGAIN || errno == EWOULDBLOCK) ? | 220 return WriteResult((errno == EAGAIN || errno == EWOULDBLOCK) ? |
| 219 WRITE_STATUS_BLOCKED : WRITE_STATUS_ERROR, errno); | 221 WRITE_STATUS_BLOCKED : WRITE_STATUS_ERROR, errno); |
| 220 } | 222 } |
| 221 | 223 |
| 222 } // namespace tools | 224 } // namespace tools |
| 223 } // namespace net | 225 } // namespace net |
| OLD | NEW |