1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "quic_test_server.h"
#include "base/android/jni_android.h"
#include "base/android/jni_string.h"
#include "base/android/path_utils.h"
#include "base/bind.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/test/test_support_android.h"
#include "base/threading/thread.h"
#include "components/cronet/android/test/cronet_test_util.h"
#include "jni/QuicTestServer_jni.h"
#include "net/base/ip_address.h"
#include "net/base/ip_endpoint.h"
#include "net/quic/chromium/crypto/proof_source_chromium.h"
#include "net/test/test_data_directory.h"
#include "net/tools/quic/quic_http_response_cache.h"
#include "net/tools/quic/quic_simple_server.h"
using base::android::JavaParamRef;
using base::android::ScopedJavaLocalRef;
namespace cronet {
namespace {
static const int kServerPort = 6121;
base::Thread* g_quic_server_thread = nullptr;
net::QuicHttpResponseCache* g_quic_response_cache = nullptr;
net::QuicSimpleServer* g_quic_server = nullptr;
void StartOnServerThread(const base::FilePath& test_files_root,
const base::FilePath& test_data_dir) {
DCHECK(g_quic_server_thread->task_runner()->BelongsToCurrentThread());
DCHECK(!g_quic_server);
// Set up in-memory cache.
base::FilePath file_dir = test_files_root.Append("quic_data");
CHECK(base::PathExists(file_dir)) << "Quic data does not exist";
g_quic_response_cache = new net::QuicHttpResponseCache();
g_quic_response_cache->InitializeFromDirectory(file_dir.value());
net::QuicConfig config;
// Set up server certs.
base::FilePath directory = test_data_dir.Append("net/data/ssl/certificates");
std::unique_ptr<net::ProofSourceChromium> proof_source(
new net::ProofSourceChromium());
CHECK(proof_source->Initialize(
directory.Append("quic_test.example.com.crt"),
directory.Append("quic_test.example.com.key.pkcs8"),
directory.Append("quic_test.example.com.key.sct")));
g_quic_server = new net::QuicSimpleServer(
std::move(proof_source), config,
net::QuicCryptoServerConfig::ConfigOptions(), net::AllSupportedVersions(),
g_quic_response_cache);
// Start listening.
int rv = g_quic_server->Listen(
net::IPEndPoint(net::IPAddress::IPv4AllZeros(), kServerPort));
CHECK_GE(rv, 0) << "Quic server fails to start";
JNIEnv* env = base::android::AttachCurrentThread();
Java_QuicTestServer_onServerStarted(env);
}
void ShutdownOnServerThread() {
DCHECK(g_quic_server_thread->task_runner()->BelongsToCurrentThread());
g_quic_server->Shutdown();
delete g_quic_server;
delete g_quic_response_cache;
}
} // namespace
// Quic server is currently hardcoded to run on port 6121 of the localhost on
// the device.
void StartQuicTestServer(JNIEnv* env,
const JavaParamRef<jclass>& /*jcaller*/,
const JavaParamRef<jstring>& jtest_files_root,
const JavaParamRef<jstring>& jtest_data_dir) {
DCHECK(!g_quic_server_thread);
base::FilePath test_data_dir(
base::android::ConvertJavaStringToUTF8(env, jtest_data_dir));
base::InitAndroidTestPaths(test_data_dir);
g_quic_server_thread = new base::Thread("quic server thread");
base::Thread::Options thread_options;
thread_options.message_loop_type = base::MessageLoop::TYPE_IO;
bool started = g_quic_server_thread->StartWithOptions(thread_options);
DCHECK(started);
base::FilePath test_files_root(
base::android::ConvertJavaStringToUTF8(env, jtest_files_root));
g_quic_server_thread->task_runner()->PostTask(
FROM_HERE,
base::Bind(&StartOnServerThread, test_files_root, test_data_dir));
}
void ShutdownQuicTestServer(JNIEnv* env,
const JavaParamRef<jclass>& /*jcaller*/) {
DCHECK(!g_quic_server_thread->task_runner()->BelongsToCurrentThread());
g_quic_server_thread->task_runner()->PostTask(
FROM_HERE, base::Bind(&ShutdownOnServerThread));
delete g_quic_server_thread;
}
int GetServerPort(JNIEnv* env, const JavaParamRef<jclass>& /*jcaller*/) {
return kServerPort;
}
bool RegisterQuicTestServer(JNIEnv* env) {
return RegisterNativesImpl(env);
}
} // namespace cronet
|
