Chromium Code Reviews| 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/udp/udp_socket_libevent.h" | 5 #include "net/udp/udp_socket_libevent.h" |
| 6 | 6 |
| 7 #include <errno.h> | 7 #include <errno.h> |
| 8 #include <fcntl.h> | 8 #include <fcntl.h> |
| 9 #include <netdb.h> | 9 #include <netdb.h> |
| 10 #include <net/if.h> | 10 #include <net/if.h> |
| (...skipping 299 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 310 if (rv < 0) { | 310 if (rv < 0) { |
| 311 Close(); | 311 Close(); |
| 312 return rv; | 312 return rv; |
| 313 } | 313 } |
| 314 local_address_.reset(); | 314 local_address_.reset(); |
| 315 return rv; | 315 return rv; |
| 316 } | 316 } |
| 317 | 317 |
| 318 int UDPSocketLibevent::SetReceiveBufferSize(int32 size) { | 318 int UDPSocketLibevent::SetReceiveBufferSize(int32 size) { |
| 319 DCHECK(CalledOnValidThread()); | 319 DCHECK(CalledOnValidThread()); |
| 320 int rv = setsockopt(socket_, SOL_SOCKET, SO_RCVBUF, | 320 if (!is_connected()) { |
| 321 reinterpret_cast<const char*>(&size), sizeof(size)); | 321 socket_options_ |= SOCKET_OPTION_RCVBUF_SIZE; |
| 322 int last_error = errno; | 322 rcvbuf_size_ = size; |
| 323 DCHECK(!rv) << "Could not set socket receive buffer size: " << last_error; | 323 return OK; |
|
mmenke
2014/10/30 14:40:38
Sockets can be disconnected and then reconnected,
hidehiko
2014/11/03 13:34:37
oops, sorry, I forgot to reply this comment. I mad
| |
| 324 return rv == 0 ? OK : MapSystemError(last_error); | 324 } |
| 325 return SetReceiveBufferSizeInternal(size); | |
| 325 } | 326 } |
| 326 | 327 |
| 327 int UDPSocketLibevent::SetSendBufferSize(int32 size) { | 328 int UDPSocketLibevent::SetSendBufferSize(int32 size) { |
| 328 DCHECK(CalledOnValidThread()); | 329 DCHECK(CalledOnValidThread()); |
| 329 int rv = setsockopt(socket_, SOL_SOCKET, SO_SNDBUF, | 330 if (!is_connected()) { |
| 330 reinterpret_cast<const char*>(&size), sizeof(size)); | 331 socket_options_ |= SOCKET_OPTION_SNDBUF_SIZE; |
| 331 int last_error = errno; | 332 sndbuf_size_ = size; |
| 332 DCHECK(!rv) << "Could not set socket send buffer size: " << last_error; | 333 return OK; |
| 333 return rv == 0 ? OK : MapSystemError(last_error); | 334 } |
| 335 return SetSendBufferSizeInternal(size); | |
| 334 } | 336 } |
| 335 | 337 |
| 336 void UDPSocketLibevent::AllowAddressReuse() { | 338 int UDPSocketLibevent::AllowAddressReuse() { |
| 337 DCHECK(CalledOnValidThread()); | 339 DCHECK(CalledOnValidThread()); |
| 338 DCHECK(!is_connected()); | 340 if (!is_connected()) { |
| 339 | 341 socket_options_ |= SOCKET_OPTION_REUSE_ADDRESS; |
| 340 socket_options_ |= SOCKET_OPTION_REUSE_ADDRESS; | 342 return OK; |
| 343 } | |
| 344 return SetAddressReuseInternal(true); | |
| 341 } | 345 } |
| 342 | 346 |
| 343 void UDPSocketLibevent::AllowBroadcast() { | 347 int UDPSocketLibevent::DisallowAddressReuse() { |
| 344 DCHECK(CalledOnValidThread()); | 348 DCHECK(CalledOnValidThread()); |
| 345 DCHECK(!is_connected()); | 349 if (!is_connected()) { |
| 350 socket_options_ &= ~SOCKET_OPTION_REUSE_ADDRESS; | |
| 351 return OK; | |
| 352 } | |
| 353 return SetAddressReuseInternal(false); | |
| 354 } | |
| 346 | 355 |
| 347 socket_options_ |= SOCKET_OPTION_BROADCAST; | 356 int UDPSocketLibevent::AllowBroadcast() { |
| 357 DCHECK(CalledOnValidThread()); | |
| 358 if (!is_connected()) { | |
| 359 socket_options_ |= SOCKET_OPTION_BROADCAST; | |
| 360 return OK; | |
| 361 } | |
| 362 return SetBroadcastInternal(true); | |
| 363 } | |
| 364 | |
| 365 int UDPSocketLibevent::DisallowBroadcast() { | |
| 366 DCHECK(CalledOnValidThread()); | |
| 367 if (!is_connected()) { | |
| 368 socket_options_ &= ~SOCKET_OPTION_BROADCAST; | |
| 369 return OK; | |
| 370 } | |
| 371 return SetBroadcastInternal(false); | |
| 348 } | 372 } |
| 349 | 373 |
| 350 void UDPSocketLibevent::ReadWatcher::OnFileCanReadWithoutBlocking(int) { | 374 void UDPSocketLibevent::ReadWatcher::OnFileCanReadWithoutBlocking(int) { |
| 351 if (!socket_->read_callback_.is_null()) | 375 if (!socket_->read_callback_.is_null()) |
| 352 socket_->DidCompleteRead(); | 376 socket_->DidCompleteRead(); |
| 353 } | 377 } |
| 354 | 378 |
| 355 void UDPSocketLibevent::WriteWatcher::OnFileCanWriteWithoutBlocking(int) { | 379 void UDPSocketLibevent::WriteWatcher::OnFileCanWriteWithoutBlocking(int) { |
| 356 if (!socket_->write_callback_.is_null()) | 380 if (!socket_->write_callback_.is_null()) |
| 357 socket_->DidCompleteWrite(); | 381 socket_->DidCompleteWrite(); |
| (...skipping 151 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 509 addr, | 533 addr, |
| 510 storage.addr_len)); | 534 storage.addr_len)); |
| 511 if (result < 0) | 535 if (result < 0) |
| 512 result = MapSystemError(errno); | 536 result = MapSystemError(errno); |
| 513 if (result != ERR_IO_PENDING) | 537 if (result != ERR_IO_PENDING) |
| 514 LogWrite(result, buf->data(), address); | 538 LogWrite(result, buf->data(), address); |
| 515 return result; | 539 return result; |
| 516 } | 540 } |
| 517 | 541 |
| 518 int UDPSocketLibevent::SetSocketOptions() { | 542 int UDPSocketLibevent::SetSocketOptions() { |
| 519 int true_value = 1; | 543 DCHECK(CalledOnValidThread()); |
| 520 if (socket_options_ & SOCKET_OPTION_REUSE_ADDRESS) { | 544 if (socket_options_ & SOCKET_OPTION_REUSE_ADDRESS) { |
| 521 int rv = setsockopt(socket_, SOL_SOCKET, SO_REUSEADDR, &true_value, | 545 int rv = SetAddressReuseInternal(true); |
| 522 sizeof(true_value)); | 546 if (rv != OK) |
| 523 if (rv < 0) | 547 return rv; |
| 524 return MapSystemError(errno); | |
| 525 } | 548 } |
| 526 if (socket_options_ & SOCKET_OPTION_BROADCAST) { | 549 if (socket_options_ & SOCKET_OPTION_BROADCAST) { |
| 527 int rv; | 550 int rv = SetBroadcastInternal(true); |
| 528 #if defined(OS_MACOSX) | 551 if (rv != OK) |
| 529 // SO_REUSEPORT on OSX permits multiple processes to each receive | 552 return rv; |
| 530 // UDP multicast or broadcast datagrams destined for the bound | |
| 531 // port. | |
| 532 rv = setsockopt(socket_, SOL_SOCKET, SO_REUSEPORT, &true_value, | |
| 533 sizeof(true_value)); | |
| 534 #else | |
| 535 rv = setsockopt(socket_, SOL_SOCKET, SO_BROADCAST, &true_value, | |
| 536 sizeof(true_value)); | |
| 537 #endif // defined(OS_MACOSX) | |
| 538 if (rv < 0) | |
| 539 return MapSystemError(errno); | |
| 540 } | 553 } |
| 541 | |
| 542 if (!(socket_options_ & SOCKET_OPTION_MULTICAST_LOOP)) { | 554 if (!(socket_options_ & SOCKET_OPTION_MULTICAST_LOOP)) { |
| 543 int rv; | 555 int rv; |
| 544 if (addr_family_ == AF_INET) { | 556 if (addr_family_ == AF_INET) { |
| 545 u_char loop = 0; | 557 u_char loop = 0; |
| 546 rv = setsockopt(socket_, IPPROTO_IP, IP_MULTICAST_LOOP, | 558 rv = setsockopt(socket_, IPPROTO_IP, IP_MULTICAST_LOOP, |
| 547 &loop, sizeof(loop)); | 559 &loop, sizeof(loop)); |
| 548 } else { | 560 } else { |
| 549 u_int loop = 0; | 561 u_int loop = 0; |
| 550 rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, | 562 rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, |
| 551 &loop, sizeof(loop)); | 563 &loop, sizeof(loop)); |
| 552 } | 564 } |
| 553 if (rv < 0) | 565 if (rv < 0) |
| 554 return MapSystemError(errno); | 566 return MapSystemError(errno); |
| 555 } | 567 } |
| 568 if (socket_options_ & SOCKET_OPTION_RCVBUF_SIZE) { | |
| 569 int rv = SetReceiveBufferSizeInternal(rcvbuf_size_); | |
| 570 if (rv != OK) | |
| 571 return rv; | |
| 572 } | |
| 573 if (socket_options_ & SOCKET_OPTION_SNDBUF_SIZE) { | |
| 574 int rv = SetSendBufferSizeInternal(sndbuf_size_); | |
| 575 if (rv != OK) | |
| 576 return rv; | |
| 577 } | |
| 578 | |
| 556 if (multicast_time_to_live_ != IP_DEFAULT_MULTICAST_TTL) { | 579 if (multicast_time_to_live_ != IP_DEFAULT_MULTICAST_TTL) { |
| 557 int rv; | 580 int rv; |
| 558 if (addr_family_ == AF_INET) { | 581 if (addr_family_ == AF_INET) { |
| 559 u_char ttl = multicast_time_to_live_; | 582 u_char ttl = multicast_time_to_live_; |
| 560 rv = setsockopt(socket_, IPPROTO_IP, IP_MULTICAST_TTL, | 583 rv = setsockopt(socket_, IPPROTO_IP, IP_MULTICAST_TTL, |
| 561 &ttl, sizeof(ttl)); | 584 &ttl, sizeof(ttl)); |
| 562 } else { | 585 } else { |
| 563 // Signed integer. -1 to use route default. | 586 // Signed integer. -1 to use route default. |
| 564 int ttl = multicast_time_to_live_; | 587 int ttl = multicast_time_to_live_; |
| 565 rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, | 588 rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 598 break; | 621 break; |
| 599 } | 622 } |
| 600 default: | 623 default: |
| 601 NOTREACHED() << "Invalid address family"; | 624 NOTREACHED() << "Invalid address family"; |
| 602 return ERR_ADDRESS_INVALID; | 625 return ERR_ADDRESS_INVALID; |
| 603 } | 626 } |
| 604 } | 627 } |
| 605 return OK; | 628 return OK; |
| 606 } | 629 } |
| 607 | 630 |
| 631 int UDPSocketLibevent::SetAddressReuseInternal(bool value) { | |
| 632 DCHECK(CalledOnValidThread()); | |
| 633 DCHECK(is_connected()); | |
| 634 int opt_value = value ? 1 : 0; | |
| 635 int rv = setsockopt( | |
| 636 socket_, SOL_SOCKET, SO_REUSEADDR, &opt_value, sizeof(opt_value)); | |
| 637 return rv == 0 ? OK : MapSystemError(errno); | |
| 638 } | |
| 639 | |
| 640 int UDPSocketLibevent::SetBroadcastInternal(bool value) { | |
| 641 DCHECK(CalledOnValidThread()); | |
| 642 DCHECK(is_connected()); | |
| 643 int opt_value = value ? 1 : 0; | |
| 644 #if defined(OS_MACOSX) | |
| 645 // SO_REUSEPORT on OSX permits multiple processes to each receive | |
| 646 // UDP multicast or broadcast datagrams destined for the bound | |
| 647 // port. | |
| 648 const int kOptName = SO_REUSEPORT; | |
| 649 #else | |
| 650 const int kOptName = SO_BROADCAST; | |
| 651 #endif | |
| 652 int rv = setsockopt( | |
| 653 socket_, SOL_SOCKET, kOptName, &opt_value, sizeof(opt_value)); | |
| 654 return rv == 0 ? OK : MapSystemError(errno); | |
| 655 } | |
| 656 | |
| 657 int UDPSocketLibevent::SetReceiveBufferSizeInternal(int32 size) { | |
| 658 DCHECK(CalledOnValidThread()); | |
| 659 DCHECK(is_connected()); | |
| 660 int rv = setsockopt(socket_, SOL_SOCKET, SO_RCVBUF, | |
| 661 reinterpret_cast<const char*>(&size), sizeof(size)); | |
| 662 int last_error = errno; | |
| 663 DCHECK(!rv) << "Could not set socket receive buffer size: " << last_error; | |
| 664 return rv == 0 ? OK : MapSystemError(last_error); | |
| 665 } | |
| 666 | |
| 667 int UDPSocketLibevent::SetSendBufferSizeInternal(int32 size) { | |
| 668 DCHECK(CalledOnValidThread()); | |
| 669 DCHECK(is_connected()); | |
| 670 int rv = setsockopt(socket_, SOL_SOCKET, SO_SNDBUF, | |
| 671 reinterpret_cast<const char*>(&size), sizeof(size)); | |
| 672 int last_error = errno; | |
| 673 DCHECK(!rv) << "Could not set socket send buffer size: " << last_error; | |
| 674 return rv == 0 ? OK : MapSystemError(last_error); | |
| 675 } | |
| 676 | |
| 608 int UDPSocketLibevent::DoBind(const IPEndPoint& address) { | 677 int UDPSocketLibevent::DoBind(const IPEndPoint& address) { |
| 609 SockaddrStorage storage; | 678 SockaddrStorage storage; |
| 610 if (!address.ToSockAddr(storage.addr, &storage.addr_len)) | 679 if (!address.ToSockAddr(storage.addr, &storage.addr_len)) |
| 611 return ERR_ADDRESS_INVALID; | 680 return ERR_ADDRESS_INVALID; |
| 612 int rv = bind(socket_, storage.addr, storage.addr_len); | 681 int rv = bind(socket_, storage.addr, storage.addr_len); |
| 613 if (rv == 0) | 682 if (rv == 0) |
| 614 return OK; | 683 return OK; |
| 615 int last_error = errno; | 684 int last_error = errno; |
| 616 UMA_HISTOGRAM_SPARSE_SLOWLY("Net.UdpSocketBindErrorFromPosix", last_error); | 685 UMA_HISTOGRAM_SPARSE_SLOWLY("Net.UdpSocketBindErrorFromPosix", last_error); |
| 617 #if defined(OS_CHROMEOS) | 686 #if defined(OS_CHROMEOS) |
| (...skipping 149 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 767 return MapSystemError(errno); | 836 return MapSystemError(errno); |
| 768 | 837 |
| 769 return OK; | 838 return OK; |
| 770 } | 839 } |
| 771 | 840 |
| 772 void UDPSocketLibevent::DetachFromThread() { | 841 void UDPSocketLibevent::DetachFromThread() { |
| 773 base::NonThreadSafe::DetachFromThread(); | 842 base::NonThreadSafe::DetachFromThread(); |
| 774 } | 843 } |
| 775 | 844 |
| 776 } // namespace net | 845 } // namespace net |
| OLD | NEW |