| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 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 "components/cronet/android/test/test_upload_data_stream_handler.h" | 5 #include "components/cronet/android/test/test_upload_data_stream_handler.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 #include <string> | 8 #include <string> |
| 9 #include <utility> | 9 #include <utility> |
| 10 | 10 |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 49 network_thread->task_runner()->DeleteSoon(FROM_HERE, this); | 49 network_thread->task_runner()->DeleteSoon(FROM_HERE, this); |
| 50 // Deleting thread stops it after all tasks are completed. | 50 // Deleting thread stops it after all tasks are completed. |
| 51 network_thread.reset(); | 51 network_thread.reset(); |
| 52 } | 52 } |
| 53 | 53 |
| 54 void TestUploadDataStreamHandler::OnInitCompleted(int res) { | 54 void TestUploadDataStreamHandler::OnInitCompleted(int res) { |
| 55 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); | 55 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); |
| 56 init_callback_invoked_ = true; | 56 init_callback_invoked_ = true; |
| 57 JNIEnv* env = base::android::AttachCurrentThread(); | 57 JNIEnv* env = base::android::AttachCurrentThread(); |
| 58 cronet::Java_TestUploadDataStreamHandler_onInitCompleted( | 58 cronet::Java_TestUploadDataStreamHandler_onInitCompleted( |
| 59 env, jtest_upload_data_stream_handler_.obj(), res); | 59 env, jtest_upload_data_stream_handler_, res); |
| 60 } | 60 } |
| 61 | 61 |
| 62 void TestUploadDataStreamHandler::OnReadCompleted(int res) { | 62 void TestUploadDataStreamHandler::OnReadCompleted(int res) { |
| 63 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); | 63 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); |
| 64 read_callback_invoked_ = true; | 64 read_callback_invoked_ = true; |
| 65 bytes_read_ = res; | 65 bytes_read_ = res; |
| 66 NotifyJavaReadCompleted(); | 66 NotifyJavaReadCompleted(); |
| 67 } | 67 } |
| 68 | 68 |
| 69 void TestUploadDataStreamHandler::Init(JNIEnv* env, | 69 void TestUploadDataStreamHandler::Init(JNIEnv* env, |
| (...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 112 | 112 |
| 113 void TestUploadDataStreamHandler::InitOnNetworkThread() { | 113 void TestUploadDataStreamHandler::InitOnNetworkThread() { |
| 114 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); | 114 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); |
| 115 init_callback_invoked_ = false; | 115 init_callback_invoked_ = false; |
| 116 read_buffer_ = nullptr; | 116 read_buffer_ = nullptr; |
| 117 bytes_read_ = 0; | 117 bytes_read_ = 0; |
| 118 int res = upload_data_stream_->Init(base::Bind( | 118 int res = upload_data_stream_->Init(base::Bind( |
| 119 &TestUploadDataStreamHandler::OnInitCompleted, base::Unretained(this))); | 119 &TestUploadDataStreamHandler::OnInitCompleted, base::Unretained(this))); |
| 120 JNIEnv* env = base::android::AttachCurrentThread(); | 120 JNIEnv* env = base::android::AttachCurrentThread(); |
| 121 cronet::Java_TestUploadDataStreamHandler_onInitCalled( | 121 cronet::Java_TestUploadDataStreamHandler_onInitCalled( |
| 122 env, jtest_upload_data_stream_handler_.obj(), res); | 122 env, jtest_upload_data_stream_handler_, res); |
| 123 | 123 |
| 124 if (res == net::OK) { | 124 if (res == net::OK) { |
| 125 cronet::Java_TestUploadDataStreamHandler_onInitCompleted( | 125 cronet::Java_TestUploadDataStreamHandler_onInitCompleted( |
| 126 env, jtest_upload_data_stream_handler_.obj(), res); | 126 env, jtest_upload_data_stream_handler_, res); |
| 127 } | 127 } |
| 128 } | 128 } |
| 129 | 129 |
| 130 void TestUploadDataStreamHandler::ReadOnNetworkThread() { | 130 void TestUploadDataStreamHandler::ReadOnNetworkThread() { |
| 131 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); | 131 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); |
| 132 read_callback_invoked_ = false; | 132 read_callback_invoked_ = false; |
| 133 if (!read_buffer_.get()) | 133 if (!read_buffer_.get()) |
| 134 read_buffer_ = new net::IOBufferWithSize(kReadBufferSize); | 134 read_buffer_ = new net::IOBufferWithSize(kReadBufferSize); |
| 135 | 135 |
| 136 int bytes_read = upload_data_stream_->Read( | 136 int bytes_read = upload_data_stream_->Read( |
| 137 read_buffer_.get(), kReadBufferSize, | 137 read_buffer_.get(), kReadBufferSize, |
| 138 base::Bind(&TestUploadDataStreamHandler::OnReadCompleted, | 138 base::Bind(&TestUploadDataStreamHandler::OnReadCompleted, |
| 139 base::Unretained(this))); | 139 base::Unretained(this))); |
| 140 if (bytes_read == net::OK) { | 140 if (bytes_read == net::OK) { |
| 141 bytes_read_ = bytes_read; | 141 bytes_read_ = bytes_read; |
| 142 NotifyJavaReadCompleted(); | 142 NotifyJavaReadCompleted(); |
| 143 } | 143 } |
| 144 } | 144 } |
| 145 | 145 |
| 146 void TestUploadDataStreamHandler::ResetOnNetworkThread() { | 146 void TestUploadDataStreamHandler::ResetOnNetworkThread() { |
| 147 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); | 147 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); |
| 148 read_buffer_ = nullptr; | 148 read_buffer_ = nullptr; |
| 149 bytes_read_ = 0; | 149 bytes_read_ = 0; |
| 150 upload_data_stream_->Reset(); | 150 upload_data_stream_->Reset(); |
| 151 JNIEnv* env = base::android::AttachCurrentThread(); | 151 JNIEnv* env = base::android::AttachCurrentThread(); |
| 152 cronet::Java_TestUploadDataStreamHandler_onResetCompleted( | 152 cronet::Java_TestUploadDataStreamHandler_onResetCompleted( |
| 153 env, jtest_upload_data_stream_handler_.obj()); | 153 env, jtest_upload_data_stream_handler_); |
| 154 } | 154 } |
| 155 | 155 |
| 156 void TestUploadDataStreamHandler::CheckInitCallbackNotInvokedOnNetworkThread() { | 156 void TestUploadDataStreamHandler::CheckInitCallbackNotInvokedOnNetworkThread() { |
| 157 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); | 157 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); |
| 158 JNIEnv* env = base::android::AttachCurrentThread(); | 158 JNIEnv* env = base::android::AttachCurrentThread(); |
| 159 cronet::Java_TestUploadDataStreamHandler_onCheckInitCallbackNotInvoked( | 159 cronet::Java_TestUploadDataStreamHandler_onCheckInitCallbackNotInvoked( |
| 160 env, jtest_upload_data_stream_handler_.obj(), !init_callback_invoked_); | 160 env, jtest_upload_data_stream_handler_, !init_callback_invoked_); |
| 161 } | 161 } |
| 162 | 162 |
| 163 void TestUploadDataStreamHandler::CheckReadCallbackNotInvokedOnNetworkThread() { | 163 void TestUploadDataStreamHandler::CheckReadCallbackNotInvokedOnNetworkThread() { |
| 164 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); | 164 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); |
| 165 JNIEnv* env = base::android::AttachCurrentThread(); | 165 JNIEnv* env = base::android::AttachCurrentThread(); |
| 166 cronet::Java_TestUploadDataStreamHandler_onCheckReadCallbackNotInvoked( | 166 cronet::Java_TestUploadDataStreamHandler_onCheckReadCallbackNotInvoked( |
| 167 env, jtest_upload_data_stream_handler_.obj(), !read_callback_invoked_); | 167 env, jtest_upload_data_stream_handler_, !read_callback_invoked_); |
| 168 } | 168 } |
| 169 | 169 |
| 170 void TestUploadDataStreamHandler::NotifyJavaReadCompleted() { | 170 void TestUploadDataStreamHandler::NotifyJavaReadCompleted() { |
| 171 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); | 171 DCHECK(network_thread_->task_runner()->BelongsToCurrentThread()); |
| 172 JNIEnv* env = base::android::AttachCurrentThread(); | 172 JNIEnv* env = base::android::AttachCurrentThread(); |
| 173 std::string data_read = ""; | 173 std::string data_read = ""; |
| 174 if (read_buffer_.get() && bytes_read_ > 0) | 174 if (read_buffer_.get() && bytes_read_ > 0) |
| 175 data_read = std::string(read_buffer_->data(), bytes_read_); | 175 data_read = std::string(read_buffer_->data(), bytes_read_); |
| 176 cronet::Java_TestUploadDataStreamHandler_onReadCompleted( | 176 cronet::Java_TestUploadDataStreamHandler_onReadCompleted( |
| 177 env, jtest_upload_data_stream_handler_.obj(), bytes_read_, | 177 env, jtest_upload_data_stream_handler_, bytes_read_, |
| 178 base::android::ConvertUTF8ToJavaString(env, data_read).obj()); | 178 base::android::ConvertUTF8ToJavaString(env, data_read).obj()); |
| 179 } | 179 } |
| 180 | 180 |
| 181 static jlong CreateTestUploadDataStreamHandler( | 181 static jlong CreateTestUploadDataStreamHandler( |
| 182 JNIEnv* env, | 182 JNIEnv* env, |
| 183 const JavaParamRef<jobject>& jtest_upload_data_stream_handler, | 183 const JavaParamRef<jobject>& jtest_upload_data_stream_handler, |
| 184 jlong jupload_data_stream) { | 184 jlong jupload_data_stream) { |
| 185 std::unique_ptr<net::UploadDataStream> upload_data_stream( | 185 std::unique_ptr<net::UploadDataStream> upload_data_stream( |
| 186 reinterpret_cast<net::UploadDataStream*>(jupload_data_stream)); | 186 reinterpret_cast<net::UploadDataStream*>(jupload_data_stream)); |
| 187 TestUploadDataStreamHandler* handler = new TestUploadDataStreamHandler( | 187 TestUploadDataStreamHandler* handler = new TestUploadDataStreamHandler( |
| 188 std::move(upload_data_stream), env, jtest_upload_data_stream_handler); | 188 std::move(upload_data_stream), env, jtest_upload_data_stream_handler); |
| 189 return reinterpret_cast<jlong>(handler); | 189 return reinterpret_cast<jlong>(handler); |
| 190 } | 190 } |
| 191 | 191 |
| 192 bool TestUploadDataStreamHandlerRegisterJni(JNIEnv* env) { | 192 bool TestUploadDataStreamHandlerRegisterJni(JNIEnv* env) { |
| 193 return RegisterNativesImpl(env); | 193 return RegisterNativesImpl(env); |
| 194 } | 194 } |
| 195 | 195 |
| 196 } // namespace cronet | 196 } // namespace cronet |
| OLD | NEW |