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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
// Copyright (c) 2010 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 <signal.h>
#include <stdlib.h>
#ifndef _WIN32
#include <sys/time.h>
#include <sys/types.h>
#include <sys/wait.h>
#else
#include <time.h>
#endif
#include <iostream>
#include <fstream>
#include "base/at_exit.h"
#include "base/command_line.h"
#include "base/logging.h"
#include "base/scoped_ptr.h"
#include "base/string_util.h"
#include "base/utf_string_conversions.h"
#include "chrome/test/webdriver/dispatch.h"
#include "chrome/test/webdriver/session_manager.h"
#include "chrome/test/webdriver/utility_functions.h"
#include "chrome/test/webdriver/commands/create_session.h"
#include "chrome/test/webdriver/commands/execute_command.h"
#include "chrome/test/webdriver/commands/find_element_commands.h"
#include "chrome/test/webdriver/commands/implicit_wait_command.h"
#include "chrome/test/webdriver/commands/navigate_commands.h"
#include "chrome/test/webdriver/commands/session_with_id.h"
#include "chrome/test/webdriver/commands/source_command.h"
#include "chrome/test/webdriver/commands/speed_command.h"
#include "chrome/test/webdriver/commands/title_command.h"
#include "chrome/test/webdriver/commands/url_command.h"
#include "third_party/mongoose/mongoose.h"
// Make sure we have ho zombies from CGIs.
static void
signal_handler(int sig_num) {
switch (sig_num) {
#ifdef OS_POSIX
case SIGCHLD:
while (waitpid(-1, &sig_num, WNOHANG) > 0) { }
break;
#elif OS_WIN
case 0: // The win compiler demands at least 1 case statement.
#endif
default:
break;
}
}
namespace webdriver {
template <typename CommandType>
void SetCallback(struct mg_context* ctx, const char* pattern) {
mg_set_uri_callback(ctx, pattern, &Dispatch<CommandType>, NULL);
}
void InitCallbacks(struct mg_context* ctx) {
SetCallback<CreateSession>(ctx, "/session");
SetCallback<BackCommand>(ctx, "/session/*/back");
SetCallback<ExecuteCommand>(ctx, "/session/*/execute");
SetCallback<ForwardCommand>(ctx, "/session/*/forward");
SetCallback<RefreshCommand>(ctx, "/session/*/refresh");
SetCallback<SourceCommand>(ctx, "/session/*/source");
SetCallback<TitleCommand>(ctx, "/session/*/title");
SetCallback<URLCommand>(ctx, "/session/*/url");
SetCallback<SpeedCommand>(ctx, "/session/*/speed");
// WebElement commands
SetCallback<ImplicitWaitCommand>(ctx, "/session/*/timeouts/implicit_wait");
SetCallback<FindOneElementCommand>(ctx, "/session/*/element");
SetCallback<FindManyElementsCommand>(ctx, "/session/*/elements");
SetCallback<FindOneElementCommand>(ctx, "/session/*/element/*/element");
SetCallback<FindManyElementsCommand>(ctx, "/session/*/elements/*/elements");
// Since the /session/* is a wild card that would match the above URIs, this
// line MUST be the last registered URI with the server.
SetCallback<SessionWithID>(ctx, "/session/*");
}
} // namespace webdriver
// Sets up and runs the Mongoose HTTP server for the JSON over HTTP
// protcol of webdriver. The spec is located at:
// http://code.google.com/p/selenium/wiki/JsonWireProtocol.
int main(int argc, char *argv[]) {
struct mg_context *ctx;
base::AtExitManager exit;
std::string port = "9515";
#ifdef OS_POSIX
CommandLine cmd_line = CommandLine(argc, argv);
#elif OS_WIN
std::string c;
for (int i = 0; i < argc; ++i) {
c += std::string(argv[i]);
}
CommandLine& cmd_line = CommandLine::FromString(ASCIIToWide(c));
#endif
// Init the commandline singleton from the env.
CommandLine::Init(0, NULL);
#if OS_POSIX
signal(SIGPIPE, SIG_IGN);
signal(SIGCHLD, &signal_handler);
#endif
srand((unsigned int)time(NULL));
if (cmd_line.HasSwitch(std::string("port"))) {
port = cmd_line.GetSwitchValueASCII(std::string("port"));
}
VLOG(1) << "Using port: " << port;
webdriver::SessionManager* session = webdriver::SessionManager::GetInstance();
session->SetIPAddress(port);
// Initialize SHTTPD context.
// Listen on port 9515 or port specified on command line.
// TODO(jmikhail) Maybe add port 9516 as a secure connection.
ctx = mg_start();
mg_set_option(ctx, "ports", port.c_str());
webdriver::InitCallbacks(ctx);
std::cout << "Starting server on port: " << port << std::endl;
// The default behavior is to run this service forever.
while (true)
PlatformThread::Sleep(3600);
// We should not reach here since the service should never quit.
// TODO(jmikhail): register a listener for SIGTERM and break the
// message loop gracefully.
mg_stop(ctx);
return (EXIT_SUCCESS);
}
|
