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 int result = SetReceiveBufferSizeInternal(size); |
| 322 int last_error = errno; | 322 if (result != OK) |
| 323 DCHECK(!rv) << "Could not set socket receive buffer size: " << last_error; | 323 return result; |
| 324 return rv == 0 ? OK : MapSystemError(last_error); | 324 } |
| 325 socket_options_ |= SOCKET_OPTION_RCVBUF_SIZE; | |
| 326 rcvbuf_size_ = size; | |
| 327 return OK; | |
| 325 } | 328 } |
| 326 | 329 |
| 327 int UDPSocketLibevent::SetSendBufferSize(int32 size) { | 330 int UDPSocketLibevent::SetSendBufferSize(int32 size) { |
| 328 DCHECK(CalledOnValidThread()); | 331 DCHECK(CalledOnValidThread()); |
| 329 int rv = setsockopt(socket_, SOL_SOCKET, SO_SNDBUF, | 332 if (is_connected()) { |
| 330 reinterpret_cast<const char*>(&size), sizeof(size)); | 333 int result = SetSendBufferSizeInternal(size); |
| 331 int last_error = errno; | 334 if (result != OK) |
| 332 DCHECK(!rv) << "Could not set socket send buffer size: " << last_error; | 335 return result; |
| 333 return rv == 0 ? OK : MapSystemError(last_error); | 336 } |
| 337 socket_options_ |= SOCKET_OPTION_SNDBUF_SIZE; | |
| 338 sndbuf_size_ = size; | |
| 339 return OK; | |
| 334 } | 340 } |
| 335 | 341 |
| 336 void UDPSocketLibevent::AllowAddressReuse() { | 342 int UDPSocketLibevent::AllowAddressReuse() { |
| 337 DCHECK(CalledOnValidThread()); | 343 DCHECK(CalledOnValidThread()); |
| 338 DCHECK(!is_connected()); | 344 if (is_connected()) { |
|
jar (doing other things)
2014/11/04 22:51:09
What are the semantics for setting AllowAddressRus
hidehiko
2014/11/05 12:48:30
It affects the behavior of other socket's for the
| |
| 339 | 345 int result = SetAddressReuseInternal(true); |
| 346 if (result != OK) | |
| 347 return result; | |
| 348 } | |
| 340 socket_options_ |= SOCKET_OPTION_REUSE_ADDRESS; | 349 socket_options_ |= SOCKET_OPTION_REUSE_ADDRESS; |
| 350 return OK; | |
| 341 } | 351 } |
| 342 | 352 |
| 343 void UDPSocketLibevent::AllowBroadcast() { | 353 int UDPSocketLibevent::DisallowAddressReuse() { |
| 344 DCHECK(CalledOnValidThread()); | 354 DCHECK(CalledOnValidThread()); |
| 345 DCHECK(!is_connected()); | 355 if (is_connected()) { |
| 356 int result = SetAddressReuseInternal(false); | |
| 357 if (result != OK) | |
| 358 return result; | |
| 359 } | |
| 360 socket_options_ &= ~SOCKET_OPTION_REUSE_ADDRESS; | |
| 361 return OK; | |
| 362 } | |
| 346 | 363 |
| 364 int UDPSocketLibevent::AllowBroadcast() { | |
|
jar (doing other things)
2014/11/04 22:51:09
Why don't you use a single setter method, taking a
hidehiko
2014/11/05 12:48:30
I just kept the current interface, as UDPSocket{Li
| |
| 365 DCHECK(CalledOnValidThread()); | |
| 366 if (is_connected()) { | |
| 367 int result = SetBroadcastInternal(true); | |
| 368 if (result != OK) | |
| 369 return result; | |
| 370 } | |
| 347 socket_options_ |= SOCKET_OPTION_BROADCAST; | 371 socket_options_ |= SOCKET_OPTION_BROADCAST; |
| 372 return OK; | |
| 373 } | |
| 374 | |
| 375 int UDPSocketLibevent::DisallowBroadcast() { | |
| 376 DCHECK(CalledOnValidThread()); | |
| 377 if (is_connected()) { | |
| 378 int result = SetBroadcastInternal(false); | |
| 379 if (result != OK) | |
| 380 return result; | |
| 381 } | |
| 382 socket_options_ &= ~SOCKET_OPTION_BROADCAST; | |
| 383 return OK; | |
| 348 } | 384 } |
| 349 | 385 |
| 350 void UDPSocketLibevent::ReadWatcher::OnFileCanReadWithoutBlocking(int) { | 386 void UDPSocketLibevent::ReadWatcher::OnFileCanReadWithoutBlocking(int) { |
| 351 if (!socket_->read_callback_.is_null()) | 387 if (!socket_->read_callback_.is_null()) |
| 352 socket_->DidCompleteRead(); | 388 socket_->DidCompleteRead(); |
| 353 } | 389 } |
| 354 | 390 |
| 355 void UDPSocketLibevent::WriteWatcher::OnFileCanWriteWithoutBlocking(int) { | 391 void UDPSocketLibevent::WriteWatcher::OnFileCanWriteWithoutBlocking(int) { |
| 356 if (!socket_->write_callback_.is_null()) | 392 if (!socket_->write_callback_.is_null()) |
| 357 socket_->DidCompleteWrite(); | 393 socket_->DidCompleteWrite(); |
| (...skipping 151 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 509 addr, | 545 addr, |
| 510 storage.addr_len)); | 546 storage.addr_len)); |
| 511 if (result < 0) | 547 if (result < 0) |
| 512 result = MapSystemError(errno); | 548 result = MapSystemError(errno); |
| 513 if (result != ERR_IO_PENDING) | 549 if (result != ERR_IO_PENDING) |
| 514 LogWrite(result, buf->data(), address); | 550 LogWrite(result, buf->data(), address); |
| 515 return result; | 551 return result; |
| 516 } | 552 } |
| 517 | 553 |
| 518 int UDPSocketLibevent::SetSocketOptions() { | 554 int UDPSocketLibevent::SetSocketOptions() { |
| 519 int true_value = 1; | 555 DCHECK(CalledOnValidThread()); |
| 520 if (socket_options_ & SOCKET_OPTION_REUSE_ADDRESS) { | 556 if (socket_options_ & SOCKET_OPTION_REUSE_ADDRESS) { |
| 521 int rv = setsockopt(socket_, SOL_SOCKET, SO_REUSEADDR, &true_value, | 557 int rv = SetAddressReuseInternal(true); |
| 522 sizeof(true_value)); | 558 if (rv != OK) |
| 523 if (rv < 0) | 559 return rv; |
| 524 return MapSystemError(errno); | |
| 525 } | 560 } |
| 526 if (socket_options_ & SOCKET_OPTION_BROADCAST) { | 561 if (socket_options_ & SOCKET_OPTION_BROADCAST) { |
| 527 int rv; | 562 int rv = SetBroadcastInternal(true); |
| 528 #if defined(OS_MACOSX) | 563 if (rv != OK) |
| 529 // SO_REUSEPORT on OSX permits multiple processes to each receive | 564 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 } | 565 } |
| 541 | |
| 542 if (!(socket_options_ & SOCKET_OPTION_MULTICAST_LOOP)) { | 566 if (!(socket_options_ & SOCKET_OPTION_MULTICAST_LOOP)) { |
| 543 int rv; | 567 int rv; |
| 544 if (addr_family_ == AF_INET) { | 568 if (addr_family_ == AF_INET) { |
| 545 u_char loop = 0; | 569 u_char loop = 0; |
| 546 rv = setsockopt(socket_, IPPROTO_IP, IP_MULTICAST_LOOP, | 570 rv = setsockopt(socket_, IPPROTO_IP, IP_MULTICAST_LOOP, |
| 547 &loop, sizeof(loop)); | 571 &loop, sizeof(loop)); |
| 548 } else { | 572 } else { |
| 549 u_int loop = 0; | 573 u_int loop = 0; |
| 550 rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, | 574 rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, |
| 551 &loop, sizeof(loop)); | 575 &loop, sizeof(loop)); |
| 552 } | 576 } |
| 553 if (rv < 0) | 577 if (rv < 0) |
| 554 return MapSystemError(errno); | 578 return MapSystemError(errno); |
| 555 } | 579 } |
| 580 if (socket_options_ & SOCKET_OPTION_RCVBUF_SIZE) { | |
| 581 int rv = SetReceiveBufferSizeInternal(rcvbuf_size_); | |
| 582 if (rv != OK) | |
| 583 return rv; | |
| 584 } | |
| 585 if (socket_options_ & SOCKET_OPTION_SNDBUF_SIZE) { | |
| 586 int rv = SetSendBufferSizeInternal(sndbuf_size_); | |
| 587 if (rv != OK) | |
| 588 return rv; | |
| 589 } | |
| 590 | |
| 556 if (multicast_time_to_live_ != IP_DEFAULT_MULTICAST_TTL) { | 591 if (multicast_time_to_live_ != IP_DEFAULT_MULTICAST_TTL) { |
| 557 int rv; | 592 int rv; |
| 558 if (addr_family_ == AF_INET) { | 593 if (addr_family_ == AF_INET) { |
| 559 u_char ttl = multicast_time_to_live_; | 594 u_char ttl = multicast_time_to_live_; |
| 560 rv = setsockopt(socket_, IPPROTO_IP, IP_MULTICAST_TTL, | 595 rv = setsockopt(socket_, IPPROTO_IP, IP_MULTICAST_TTL, |
| 561 &ttl, sizeof(ttl)); | 596 &ttl, sizeof(ttl)); |
| 562 } else { | 597 } else { |
| 563 // Signed integer. -1 to use route default. | 598 // Signed integer. -1 to use route default. |
| 564 int ttl = multicast_time_to_live_; | 599 int ttl = multicast_time_to_live_; |
| 565 rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, | 600 rv = setsockopt(socket_, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 598 break; | 633 break; |
| 599 } | 634 } |
| 600 default: | 635 default: |
| 601 NOTREACHED() << "Invalid address family"; | 636 NOTREACHED() << "Invalid address family"; |
| 602 return ERR_ADDRESS_INVALID; | 637 return ERR_ADDRESS_INVALID; |
| 603 } | 638 } |
| 604 } | 639 } |
| 605 return OK; | 640 return OK; |
| 606 } | 641 } |
| 607 | 642 |
| 643 int UDPSocketLibevent::SetAddressReuseInternal(bool value) { | |
| 644 DCHECK(CalledOnValidThread()); | |
| 645 DCHECK(is_connected()); | |
| 646 int opt_value = value ? 1 : 0; | |
| 647 int rv = setsockopt( | |
| 648 socket_, SOL_SOCKET, SO_REUSEADDR, &opt_value, sizeof(opt_value)); | |
| 649 return rv == 0 ? OK : MapSystemError(errno); | |
| 650 } | |
| 651 | |
| 652 int UDPSocketLibevent::SetBroadcastInternal(bool value) { | |
| 653 DCHECK(CalledOnValidThread()); | |
| 654 DCHECK(is_connected()); | |
| 655 int opt_value = value ? 1 : 0; | |
| 656 #if defined(OS_MACOSX) | |
| 657 // SO_REUSEPORT on OSX permits multiple processes to each receive | |
| 658 // UDP multicast or broadcast datagrams destined for the bound | |
| 659 // port. | |
| 660 const int kOptName = SO_REUSEPORT; | |
| 661 #else | |
| 662 const int kOptName = SO_BROADCAST; | |
| 663 #endif | |
| 664 int rv = setsockopt( | |
| 665 socket_, SOL_SOCKET, kOptName, &opt_value, sizeof(opt_value)); | |
| 666 return rv == 0 ? OK : MapSystemError(errno); | |
| 667 } | |
| 668 | |
| 669 int UDPSocketLibevent::SetReceiveBufferSizeInternal(int32 size) { | |
| 670 DCHECK(CalledOnValidThread()); | |
| 671 DCHECK(is_connected()); | |
| 672 int rv = setsockopt(socket_, SOL_SOCKET, SO_RCVBUF, | |
| 673 reinterpret_cast<const char*>(&size), sizeof(size)); | |
| 674 int last_error = errno; | |
| 675 DCHECK(!rv) << "Could not set socket receive buffer size: " << last_error; | |
| 676 return rv == 0 ? OK : MapSystemError(last_error); | |
| 677 } | |
| 678 | |
| 679 int UDPSocketLibevent::SetSendBufferSizeInternal(int32 size) { | |
| 680 DCHECK(CalledOnValidThread()); | |
| 681 DCHECK(is_connected()); | |
| 682 int rv = setsockopt(socket_, SOL_SOCKET, SO_SNDBUF, | |
| 683 reinterpret_cast<const char*>(&size), sizeof(size)); | |
| 684 int last_error = errno; | |
| 685 DCHECK(!rv) << "Could not set socket send buffer size: " << last_error; | |
| 686 return rv == 0 ? OK : MapSystemError(last_error); | |
| 687 } | |
| 688 | |
| 608 int UDPSocketLibevent::DoBind(const IPEndPoint& address) { | 689 int UDPSocketLibevent::DoBind(const IPEndPoint& address) { |
| 609 SockaddrStorage storage; | 690 SockaddrStorage storage; |
| 610 if (!address.ToSockAddr(storage.addr, &storage.addr_len)) | 691 if (!address.ToSockAddr(storage.addr, &storage.addr_len)) |
| 611 return ERR_ADDRESS_INVALID; | 692 return ERR_ADDRESS_INVALID; |
| 612 int rv = bind(socket_, storage.addr, storage.addr_len); | 693 int rv = bind(socket_, storage.addr, storage.addr_len); |
| 613 if (rv == 0) | 694 if (rv == 0) |
| 614 return OK; | 695 return OK; |
| 615 int last_error = errno; | 696 int last_error = errno; |
| 616 UMA_HISTOGRAM_SPARSE_SLOWLY("Net.UdpSocketBindErrorFromPosix", last_error); | 697 UMA_HISTOGRAM_SPARSE_SLOWLY("Net.UdpSocketBindErrorFromPosix", last_error); |
| 617 #if defined(OS_CHROMEOS) | 698 #if defined(OS_CHROMEOS) |
| (...skipping 149 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 767 return MapSystemError(errno); | 848 return MapSystemError(errno); |
| 768 | 849 |
| 769 return OK; | 850 return OK; |
| 770 } | 851 } |
| 771 | 852 |
| 772 void UDPSocketLibevent::DetachFromThread() { | 853 void UDPSocketLibevent::DetachFromThread() { |
| 773 base::NonThreadSafe::DetachFromThread(); | 854 base::NonThreadSafe::DetachFromThread(); |
| 774 } | 855 } |
| 775 | 856 |
| 776 } // namespace net | 857 } // namespace net |
| OLD | NEW |