OLD | NEW |
1 // Copyright (c) 2015 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2015 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/quic/chromium/quic_chromium_packet_reader.h" | 5 #include "net/quic/chromium/quic_chromium_packet_reader.h" |
6 | 6 |
7 #include "base/location.h" | 7 #include "base/location.h" |
8 #include "base/metrics/histogram_macros.h" | 8 #include "base/metrics/histogram_macros.h" |
9 #include "base/single_thread_task_runner.h" | 9 #include "base/single_thread_task_runner.h" |
10 #include "base/threading/thread_task_runner_handle.h" | 10 #include "base/threading/thread_task_runner_handle.h" |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
59 // Schedule the work through the message loop to 1) prevent infinite | 59 // Schedule the work through the message loop to 1) prevent infinite |
60 // recursion and 2) avoid blocking the thread for too long. | 60 // recursion and 2) avoid blocking the thread for too long. |
61 base::ThreadTaskRunnerHandle::Get()->PostTask( | 61 base::ThreadTaskRunnerHandle::Get()->PostTask( |
62 FROM_HERE, base::Bind(&QuicChromiumPacketReader::OnReadComplete, | 62 FROM_HERE, base::Bind(&QuicChromiumPacketReader::OnReadComplete, |
63 weak_factory_.GetWeakPtr(), rv)); | 63 weak_factory_.GetWeakPtr(), rv)); |
64 } else { | 64 } else { |
65 OnReadComplete(rv); | 65 OnReadComplete(rv); |
66 } | 66 } |
67 } | 67 } |
68 | 68 |
| 69 size_t QuicChromiumPacketReader::EstimateMemoryUsage() const { |
| 70 // Return the size of |read_buffer_|. |
| 71 return kMaxPacketSize; |
| 72 } |
| 73 |
69 void QuicChromiumPacketReader::OnReadComplete(int result) { | 74 void QuicChromiumPacketReader::OnReadComplete(int result) { |
70 read_pending_ = false; | 75 read_pending_ = false; |
71 if (result == 0) | 76 if (result == 0) |
72 result = ERR_CONNECTION_CLOSED; | 77 result = ERR_CONNECTION_CLOSED; |
73 | 78 |
74 if (result < 0) { | 79 if (result < 0) { |
75 visitor_->OnReadError(result, socket_); | 80 visitor_->OnReadError(result, socket_); |
76 return; | 81 return; |
77 } | 82 } |
78 | 83 |
79 QuicReceivedPacket packet(read_buffer_->data(), result, clock_->Now()); | 84 QuicReceivedPacket packet(read_buffer_->data(), result, clock_->Now()); |
80 IPEndPoint local_address; | 85 IPEndPoint local_address; |
81 IPEndPoint peer_address; | 86 IPEndPoint peer_address; |
82 socket_->GetLocalAddress(&local_address); | 87 socket_->GetLocalAddress(&local_address); |
83 socket_->GetPeerAddress(&peer_address); | 88 socket_->GetPeerAddress(&peer_address); |
84 if (!visitor_->OnPacket(packet, local_address, peer_address)) | 89 if (!visitor_->OnPacket(packet, local_address, peer_address)) |
85 return; | 90 return; |
86 | 91 |
87 StartReading(); | 92 StartReading(); |
88 } | 93 } |
89 | 94 |
90 } // namespace net | 95 } // namespace net |
OLD | NEW |