Switch skydb to services/http_server.

This patch makes the skydb debugger use the Mojo http_server app instead
of the /net http_server.

BUG=456128
R=eseidel@chromium.org, qsr@chromium.org

Review URL: https://codereview.chromium.org/930903002
This commit is contained in:
Przemyslaw Pietrzkiewicz 2015-02-17 17:10:24 +01:00
parent 374a768d5b
commit 979663ebc5
2 changed files with 71 additions and 90 deletions

View File

@ -30,8 +30,8 @@ mojo_native_application("sky_debugger") {
"//mojo/public/cpp/system",
"//mojo/public/cpp/utility",
"//mojo/services/window_manager/public/interfaces",
"//net",
"//net:http_server",
"//services/http_server/public",
"//services/http_server/public:util",
"//services/tracing:bindings",
"//sky/viewer:bindings",
]

View File

@ -9,112 +9,91 @@
#include "base/strings/string_number_conversions.h"
#include "base/strings/stringprintf.h"
#include "mojo/application/application_runner_chromium.h"
#include "mojo/common/data_pipe_utils.h"
#include "mojo/public/c/system/main.h"
#include "mojo/public/cpp/application/application_delegate.h"
#include "mojo/public/cpp/application/application_impl.h"
#include "mojo/public/cpp/bindings/binding.h"
#include "mojo/services/window_manager/public/interfaces/window_manager.mojom.h"
#include "net/base/net_errors.h"
#include "net/server/http_server.h"
#include "net/server/http_server_request_info.h"
#include "net/socket/tcp_server_socket.h"
#include "services/http_server/public/http_server.mojom.h"
#include "services/http_server/public/http_server_factory.mojom.h"
#include "services/http_server/public/http_server_util.h"
#include "services/tracing/tracing.mojom.h"
#include "sky/tools/debugger/trace_collector.h"
namespace sky {
namespace debugger {
namespace {
const size_t kMinSendBufferSize = 1024 * 1024;
}
class SkyDebugger : public mojo::ApplicationDelegate,
public net::HttpServer::Delegate {
public http_server::HttpHandler {
public:
SkyDebugger() : is_tracing_(false) {}
SkyDebugger() : is_tracing_(false), handler_binding_(this) {}
virtual ~SkyDebugger() {}
private:
// Overridden from mojo::ApplicationDelegate:
virtual void Initialize(mojo::ApplicationImpl* app) override {
// mojo::ApplicationDelegate:
void Initialize(mojo::ApplicationImpl* app) override {
app->ConnectToService("mojo:tracing", &tracing_);
app->ConnectToService("mojo:window_manager", &window_manager_);
// Format: --args-for="app_url command_port"
if (app->args().size() < 2) {
LOG(ERROR) << "--args-for required to specify command_port";
mojo::ApplicationImpl::Terminate();
return;
}
base::StringToUint(app->args()[1], &command_port_);
http_server::HttpServerFactoryPtr http_server_factory;
app->ConnectToService("mojo:http_server", &http_server_factory);
http_server_factory->CreateHttpServer(GetProxy(&http_server_).Pass(),
command_port_);
scoped_ptr<net::ServerSocket> server_socket(
new net::TCPServerSocket(NULL, net::NetLog::Source()));
int result =
server_socket->ListenWithAddressAndPort("0.0.0.0", command_port_, 1);
if (result != net::OK) {
LOG(ERROR) << "Failed to bind to port " << command_port_
<< " skydb commands will not work.";
mojo::ApplicationImpl::Terminate();
return;
}
web_server_.reset(new net::HttpServer(server_socket.Pass(), this));
app->ConnectToService("mojo:window_manager", &window_manager_);
http_server::HttpHandlerPtr handler_ptr;
handler_binding_.Bind(GetProxy(&handler_ptr).Pass());
http_server_->SetHandler(".*", handler_ptr.Pass(),
[](bool result) { DCHECK(result); });
}
virtual bool ConfigureIncomingConnection(
bool ConfigureIncomingConnection(
mojo::ApplicationConnection* connection) override {
return true;
}
// net::HttpServer::Delegate
void OnConnect(int connection_id) override {}
void OnClose(int connection_id) override {}
void OnHttpRequest(int connection_id,
const net::HttpServerRequestInfo& info) override {
// http_server::HttpHandler:
void HandleRequest(http_server::HttpRequestPtr request,
const HandleRequestCallback& callback) override {
// FIXME: We should use use a fancier lookup system more like what
// services/http_server/http_server.cc does with AddHandler.
if (info.path == "/reload")
Load(connection_id, url_);
else if (info.path == "/quit")
Quit(connection_id);
else if (info.path == "/load")
Load(connection_id, info.data);
else if (info.path == "/start_profiling")
StartProfiling(connection_id);
else if (info.path == "/stop_profiling")
StopProfiling(connection_id);
else if (info.path == "/start_tracing")
StartTracing(connection_id);
else if (info.path == "/stop_tracing")
StopTracing(connection_id);
else
Help(info.path, connection_id);
if (request->relative_url == "/reload") {
Load(callback, url_);
} else if (request->relative_url == "/quit") {
Quit();
} else if (request->relative_url == "/load") {
std::string url;
mojo::common::BlockingCopyToString(request->body.Pass(), &url);
Load(callback, url);
} else if (request->relative_url == "/start_profiling") {
StartProfiling(callback);
} else if (request->relative_url == "/stop_profiling") {
StopProfiling(callback);
} else if (request->relative_url == "/start_tracing") {
StartTracing(callback);
} else if (request->relative_url == "/stop_tracing") {
StopTracing(callback);
} else {
Help(callback, request->relative_url);
}
}
void OnWebSocketRequest(int connection_id,
const net::HttpServerRequestInfo& info) override {
Error(connection_id, "OnWebSocketRequest not implemented");
void Error(const HandleRequestCallback& callback, std::string message) {
callback.Run(http_server::CreateHttpResponse(500, message));
}
void OnWebSocketMessage(int connection_id, const std::string& data) override {
Error(connection_id, "OnWebSocketMessage not implemented");
void Respond(const HandleRequestCallback& callback, std::string response) {
callback.Run(http_server::CreateHttpResponse(200, response));
}
void Error(int connection_id, std::string message) {
web_server_->Send500(connection_id, message);
}
void Respond(int connection_id, std::string response) {
// When sending tracing data back over the wire to the client, we can blow
// through the default send buffer size.
web_server_->SetSendBufferSize(
connection_id, std::max(kMinSendBufferSize, response.length()));
web_server_->Send200(connection_id, response, "text/plain");
}
void Help(std::string path, int connection_id) {
void Help(const HandleRequestCallback& callback, std::string path) {
std::string help = base::StringPrintf(
"Sky Debugger running on port %d\n"
"Supported URLs:\n"
@ -124,14 +103,14 @@ class SkyDebugger : public mojo::ApplicationDelegate,
command_port_);
if (path != "/")
help = "Unknown path: " + path + "\n\n" + help;
Respond(connection_id, help);
Respond(callback, help);
}
void Load(int connection_id, std::string url) {
void Load(const HandleRequestCallback& callback, std::string url) {
url_ = url;
Reload();
std::string response = std::string("Loaded ") + url + "\n";
Respond(connection_id, response);
Respond(callback, response);
}
void Reload() {
@ -139,14 +118,14 @@ class SkyDebugger : public mojo::ApplicationDelegate,
window_manager_->Embed(url_, nullptr, nullptr);
}
void Quit(int connection_id) {
void Quit() {
// TODO(eseidel): We should orderly shutdown once mojo can.
exit(0);
}
void StartTracing(int connection_id) {
void StartTracing(const HandleRequestCallback& callback) {
if (is_tracing_) {
Error(connection_id, "Already tracing. Use stop_tracing to stop.\n");
Error(callback, "Already tracing. Use stop_tracing to stop.\n");
return;
}
@ -154,53 +133,55 @@ class SkyDebugger : public mojo::ApplicationDelegate,
mojo::DataPipe pipe;
tracing_->Start(pipe.producer_handle.Pass(), mojo::String("*"));
trace_collector_.reset(new TraceCollector(pipe.consumer_handle.Pass()));
Respond(connection_id, "Starting trace (type 'stop_tracing' to stop)\n");
Respond(callback, "Starting trace (type 'stop_tracing' to stop)\n");
}
void StopTracing(int connection_id) {
void StopTracing(const HandleRequestCallback& callback) {
if (!is_tracing_) {
Error(connection_id, "Not tracing yet. Use start_tracing to start.\n");
Error(callback, "Not tracing yet. Use start_tracing to start.\n");
return;
}
is_tracing_ = false;
tracing_->StopAndFlush();
trace_collector_->GetTrace(base::Bind(
&SkyDebugger::OnTraceAvailable, base::Unretained(this), connection_id));
trace_collector_->GetTrace(base::Bind(&SkyDebugger::OnTraceAvailable,
base::Unretained(this), callback));
}
void OnTraceAvailable(int connection_id, std::string trace) {
void OnTraceAvailable(HandleRequestCallback callback, std::string trace) {
trace_collector_.reset();
Respond(connection_id, trace);
Respond(callback, trace);
}
void StartProfiling(int connection_id) {
void StartProfiling(const HandleRequestCallback& callback) {
#if !defined(NDEBUG) || !defined(ENABLE_PROFILING)
Error(connection_id,
Error(callback,
"Profiling requires is_debug=false and enable_profiling=true");
return;
#else
base::debug::StartProfiling("sky_viewer.pprof");
Respond(connection_id, "Starting profiling (stop with 'stop_profiling')");
Respond(callback, "Starting profiling (stop with 'stop_profiling')");
#endif
}
void StopProfiling(int connection_id) {
void StopProfiling(const HandleRequestCallback& callback) {
if (!base::debug::BeingProfiled()) {
Error(connection_id, "Profiling not started");
Error(callback, "Profiling not started");
return;
}
base::debug::StopProfiling();
Respond(connection_id, "Stopped profiling");
Respond(callback, "Stopped profiling");
}
bool is_tracing_;
mojo::WindowManagerPtr window_manager_;
tracing::TraceCoordinatorPtr tracing_;
std::string url_;
scoped_ptr<net::HttpServer> web_server_;
uint32_t command_port_;
http_server::HttpServerPtr http_server_;
mojo::Binding<http_server::HttpHandler> handler_binding_;
scoped_ptr<TraceCollector> trace_collector_;
DISALLOW_COPY_AND_ASSIGN(SkyDebugger);