| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/curvecp/client_packetizer.h" | 5 #include "net/curvecp/client_packetizer.h" |
| 6 | 6 |
| 7 #include "base/bind.h" |
| 7 #include "net/base/io_buffer.h" | 8 #include "net/base/io_buffer.h" |
| 8 #include "net/base/net_errors.h" | 9 #include "net/base/net_errors.h" |
| 9 #include "net/base/sys_addrinfo.h" | 10 #include "net/base/sys_addrinfo.h" |
| 10 #include "net/curvecp/protocol.h" | 11 #include "net/curvecp/protocol.h" |
| 11 #include "net/udp/udp_client_socket.h" | 12 #include "net/udp/udp_client_socket.h" |
| 12 | 13 |
| 13 namespace { | 14 namespace { |
| 14 | 15 |
| 15 const int kMaxHelloAttempts = 8; | 16 const int kMaxHelloAttempts = 8; |
| 16 const int kHelloTimeoutMs[kMaxHelloAttempts] = { | 17 const int kHelloTimeoutMs[kMaxHelloAttempts] = { |
| (...skipping 14 matching lines...) Expand all Loading... |
| 31 ClientPacketizer::ClientPacketizer() | 32 ClientPacketizer::ClientPacketizer() |
| 32 : Packetizer(), | 33 : Packetizer(), |
| 33 next_state_(NONE), | 34 next_state_(NONE), |
| 34 listener_(NULL), | 35 listener_(NULL), |
| 35 user_callback_(NULL), | 36 user_callback_(NULL), |
| 36 current_address_(NULL), | 37 current_address_(NULL), |
| 37 hello_attempts_(0), | 38 hello_attempts_(0), |
| 38 initiate_sent_(false), | 39 initiate_sent_(false), |
| 39 ALLOW_THIS_IN_INITIALIZER_LIST( | 40 ALLOW_THIS_IN_INITIALIZER_LIST( |
| 40 io_callback_(this, &ClientPacketizer::OnIOComplete)), | 41 io_callback_(this, &ClientPacketizer::OnIOComplete)), |
| 41 ALLOW_THIS_IN_INITIALIZER_LIST(timers_factory_(this)) { | 42 ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)) { |
| 42 // TODO(mbelshe): Initialize our keys and such properly. | 43 // TODO(mbelshe): Initialize our keys and such properly. |
| 43 // for now we use random values to keep them unique. | 44 // for now we use random values to keep them unique. |
| 44 for (int i = 0; i < 32; ++i) | 45 for (int i = 0; i < 32; ++i) |
| 45 shortterm_public_key_[i] = rand() % 26 + 'a'; | 46 shortterm_public_key_[i] = rand() % 26 + 'a'; |
| 46 } | 47 } |
| 47 | 48 |
| 48 ClientPacketizer::~ClientPacketizer() { | 49 ClientPacketizer::~ClientPacketizer() { |
| 49 } | 50 } |
| 50 | 51 |
| 51 int ClientPacketizer::Connect(const AddressList& server, | 52 int ClientPacketizer::Connect(const AddressList& server, |
| (...skipping 256 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 308 | 309 |
| 309 int rv = socket_->Connect(endpoint); | 310 int rv = socket_->Connect(endpoint); |
| 310 DCHECK_NE(ERR_IO_PENDING, rv); | 311 DCHECK_NE(ERR_IO_PENDING, rv); |
| 311 | 312 |
| 312 return rv; | 313 return rv; |
| 313 } | 314 } |
| 314 | 315 |
| 315 void ClientPacketizer::StartHelloTimer(int milliseconds) { | 316 void ClientPacketizer::StartHelloTimer(int milliseconds) { |
| 316 MessageLoop::current()->PostDelayedTask( | 317 MessageLoop::current()->PostDelayedTask( |
| 317 FROM_HERE, | 318 FROM_HERE, |
| 318 timers_factory_.NewRunnableMethod(&ClientPacketizer::OnHelloTimeout), | 319 base::Bind(&ClientPacketizer::OnHelloTimeout, weak_factory_.GetWeakPtr()), |
| 319 milliseconds); | 320 milliseconds); |
| 320 } | 321 } |
| 321 | 322 |
| 322 void ClientPacketizer::RevokeHelloTimer() { | 323 void ClientPacketizer::RevokeHelloTimer() { |
| 323 timers_factory_.RevokeAll(); | 324 weak_factory_.InvalidateWeakPtrs(); |
| 324 } | 325 } |
| 325 | 326 |
| 326 void ClientPacketizer::OnHelloTimeout() { | 327 void ClientPacketizer::OnHelloTimeout() { |
| 327 DCHECK_EQ(WAITING_COOKIE_COMPLETE, next_state_); | 328 DCHECK_EQ(WAITING_COOKIE_COMPLETE, next_state_); |
| 328 next_state_ = SENDING_HELLO; | 329 next_state_ = SENDING_HELLO; |
| 329 DLOG(INFO) << "HelloTimeout #" << hello_attempts_; | 330 DLOG(INFO) << "HelloTimeout #" << hello_attempts_; |
| 330 int rv = DoLoop(OK); | 331 int rv = DoLoop(OK); |
| 331 if (rv != ERR_IO_PENDING) | 332 if (rv != ERR_IO_PENDING) |
| 332 DoCallback(rv); | 333 DoCallback(rv); |
| 333 } | 334 } |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 379 return rv; | 380 return rv; |
| 380 } | 381 } |
| 381 | 382 |
| 382 void ClientPacketizer::OnIOComplete(int result) { | 383 void ClientPacketizer::OnIOComplete(int result) { |
| 383 int rv = DoLoop(result); | 384 int rv = DoLoop(result); |
| 384 if (rv != ERR_IO_PENDING) | 385 if (rv != ERR_IO_PENDING) |
| 385 DoCallback(rv); | 386 DoCallback(rv); |
| 386 } | 387 } |
| 387 | 388 |
| 388 } // namespace net | 389 } // namespace net |
| OLD | NEW |