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
|
// 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/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/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<ForwardCommand>(ctx, "/session/*/forward");
SetCallback<RefreshCommand>(ctx, "/session/*/refresh");
SetCallback<SourceCommand>(ctx, "/session/*/source");
SetCallback<TitleCommand>(ctx, "/session/*/title");
SetCallback<URLCommand>(ctx, "/session/*/url");
// 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 = "8080";
#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"));
}
LOG(INFO) << "Using port: " << port;
webdriver::SessionManager* session =
Singleton<webdriver::SessionManager>::get();
session->SetIPAddress(port);
// Initialize SHTTPD context.
// Listen on port 8080 or port specified on command line.
// TODO(jmikhail) Maybe add port 8081 as a secure connection.
ctx = mg_start();
mg_set_option(ctx, "ports", port.c_str());
webdriver::InitCallbacks(ctx);
std::cout << "Starting server" << 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);
}
|
