Send screenshots back to the client for debugging

chrome/test/webdriver/server.cc

 // 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>
 
 #if defined(OS_WIN)
 #include <windows.h>
 #endif
@@ skipping 176 matching lines @@
   // By default, mongoose serves files from the current working directory. The
   // 'root' flag allows the user to specify a different location to serve from.
   if (cmd_line->HasSwitch("root"))
     root = cmd_line->GetSwitchValueASCII("root");
   if (cmd_line->HasSwitch("chrome-dir"))
     chrome_dir = cmd_line->GetSwitchValuePath("chrome-dir");
   if (cmd_line->HasSwitch("url-base"))
     url_base = cmd_line->GetSwitchValueASCII("url-base");
 
   webdriver::SessionManager* manager = webdriver::SessionManager::GetInstance();
-  manager->set_port(port);
-  manager->set_url_base(url_base);
+  manager->SetPort(port);
+  manager->SetUrlBase(url_base);
+  manager->SetScreenshotOnError(cmd_line->HasSwitch("screenshot-on-error"));
   if (!chrome_dir.empty()) {
     if (!file_util::DirectoryExists(chrome_dir)) {
       std::cout << "Given Chrome directory is inaccessible or does not exist: "
                 << chrome_dir.value() << std::endl;
 #if defined(OS_WIN)
       return ERROR_PATH_NOT_FOUND;
 #else
       return ENOENT;
 #endif
     }
-    manager->set_chrome_dir(chrome_dir);
+    manager->SetChromeDir(chrome_dir);
   }
 
   // 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();
   if (!SetMongooseOptions(ctx, port, root)) {
     mg_stop(ctx);
 #if defined(OS_WIN)
     return WSAEADDRINUSE;
@@ skipping 19 matching lines @@
 
   // Run until we receive command to shutdown.
   shutdown_event.Wait();
 
   // 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);
 }