Path: blob/trunk/dotnet/src/webdriver/Firefox/FirefoxDriverService.cs
4500 views
// <copyright file="FirefoxDriverService.cs" company="Selenium Committers">
// Licensed to the Software Freedom Conservancy (SFC) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The SFC licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
// </copyright>
using System;
using System.Diagnostics;
using System.Globalization;
using System.IO;
using System.Text;
using OpenQA.Selenium.Internal;
namespace OpenQA.Selenium.Firefox;
/// <summary>
/// Exposes the service provided by the native FirefoxDriver executable.
/// </summary>
public sealed class FirefoxDriverService : DriverService
{
private const string DefaultFirefoxDriverServiceFileName = "geckodriver";
/// <summary>
/// Process management fields for the log writer.
/// </summary>
private StreamWriter? logWriter;
/// <summary>
/// Initializes a new instance of the <see cref="FirefoxDriverService"/> class.
/// </summary>
/// <param name="executablePath">The full path to the Firefox driver executable.</param>
/// <param name="executableFileName">The file name of the Firefox driver executable.</param>
/// <param name="port">The port on which the Firefox driver executable should listen.</param>
private FirefoxDriverService(string? executablePath, string? executableFileName, int port)
: base(executablePath, port, executableFileName)
{
}
/// <inheritdoc />
protected override DriverOptions GetDefaultDriverOptions()
{
return new FirefoxOptions();
}
/// <summary>
/// Gets or sets the location of the Firefox binary executable.
/// </summary>
/// <remarks> A <see langword="null"/> or <see cref="string.Empty"/> value indicates no binary executable path to specify.</remarks>
public string? FirefoxBinaryPath { get; set; }
/// <summary>
/// Gets or sets the port used by the driver executable to communicate with the browser.
/// </summary>
/// <remarks>A negative or zero value indicates no port value to specify.</remarks>
public int BrowserCommunicationPort { get; set; } = -1;
/// <summary>
/// Gets or sets the value of the IP address of the host adapter used by the driver
/// executable to communicate with the browser.
/// </summary>
/// <remarks> A <see langword="null"/> or <see cref="string.Empty"/> value indicates no marionette host adapter to specify.</remarks>
public string? BrowserCommunicationHost { get; set; }
/// <summary>
/// Gets or sets the value of the IP address of the host adapter on which the
/// service should listen for connections.
/// </summary>
/// <remarks> A <see langword="null"/> or <see cref="string.Empty"/> value indicates no host to specify.</remarks>
public string? Host { get; set; }
/// <summary>
/// Gets or sets a value indicating whether to connect to an already-running
/// instance of Firefox.
/// </summary>
public bool ConnectToRunningBrowser { get; set; }
/// <summary>
/// Gets or sets a value indicating whether to open the Firefox Browser Toolbox
/// when Firefox is launched.
/// </summary>
public bool OpenBrowserToolbox { get; set; }
/// <summary>
/// Gets or sets the file path where log output should be written.
/// </summary>
/// <remarks>
/// A <see langword="null"/> or <see cref="string.Empty"/> value indicates no log file to specify.
/// This approach takes the process output and redirects it to a file because GeckoDriver does not
/// offer a way to specify a log file path directly.
/// </remarks>
public string? LogPath { get; set; }
/// <summary>
/// Gets or sets a value indicating whether to disable truncation of long log lines in GeckoDriver.
/// </summary>
/// <remarks>
/// A <see langword="null"/> value indicates no log truncation setting to specify.
/// Set to <see langword="true"/> to disable truncation, or <see langword="false"/> to enable truncation.
/// </remarks>
public bool? LogTruncate { get; set; }
/// <summary>
/// Directory in which GeckoDriver creates profiles.
/// </summary>
public string? ProfileRoot { get; set; }
/// <summary>
/// Gets or sets the level at which log output is displayed.
/// </summary>
/// <remarks>
/// This is largely equivalent to setting the <see cref="FirefoxOptions.LogLevel"/>
/// property, except the log level is set when the driver launches, instead of
/// when the browser is launched, meaning that initial driver logging before
/// initiation of a session can be controlled.
/// </remarks>
public FirefoxDriverLogLevel LogLevel { get; set; } = FirefoxDriverLogLevel.Default;
/// <summary>
/// Gets a value indicating the time to wait for the service to terminate before forcing it to terminate.
/// </summary>
protected override TimeSpan TerminationTimeout
{
// Use a very small timeout for terminating the Firefox driver,
// because the executable does not have a clean shutdown command,
// which means we have to kill the process. Using a short timeout
// gets us to the termination point much faster.
get => TimeSpan.FromMilliseconds(100);
}
/// <summary>
/// Gets a value indicating whether the service has a shutdown API that can be called to terminate
/// it gracefully before forcing a termination.
/// </summary>
protected override bool HasShutdown
{
// The Firefox driver executable does not have a clean shutdown command,
// which means we have to kill the process.
get => false;
}
/// <summary>
/// Gets the command-line arguments for the driver service.
/// </summary>
protected override string CommandLineArguments
{
get
{
StringBuilder argsBuilder = new StringBuilder();
if (this.ConnectToRunningBrowser)
{
argsBuilder.Append(" --connect-existing");
}
else
{
argsBuilder.Append(string.Format(CultureInfo.InvariantCulture, " --websocket-port {0}", PortUtilities.FindFreePort()));
}
if (this.BrowserCommunicationPort > 0)
{
argsBuilder.AppendFormat(CultureInfo.InvariantCulture, " --marionette-port {0}", this.BrowserCommunicationPort);
}
if (!string.IsNullOrEmpty(this.BrowserCommunicationHost))
{
argsBuilder.AppendFormat(CultureInfo.InvariantCulture, " --marionette-host \"{0}\"", this.BrowserCommunicationHost);
}
if (this.Port > 0)
{
argsBuilder.AppendFormat(CultureInfo.InvariantCulture, " --port {0}", this.Port);
}
if (!string.IsNullOrEmpty(this.FirefoxBinaryPath))
{
argsBuilder.AppendFormat(CultureInfo.InvariantCulture, " --binary \"{0}\"", this.FirefoxBinaryPath);
}
if (!string.IsNullOrEmpty(this.Host))
{
argsBuilder.AppendFormat(CultureInfo.InvariantCulture, " --host \"{0}\"", this.Host);
}
if (Environment.GetEnvironmentVariable("SE_DEBUG") is not null)
{
if (this.SuppressInitialDiagnosticInformation ||
(this.LogLevel != FirefoxDriverLogLevel.Default && this.LogLevel != FirefoxDriverLogLevel.Debug))
{
Console.Error.WriteLine("WARNING: Environment Variable `SE_DEBUG` is set; forcing GeckoDriver log level to DEBUG and overriding configured log level.");
}
argsBuilder.Append(" --log debug");
}
else if (this.LogLevel == FirefoxDriverLogLevel.Debug)
{
argsBuilder.Append(" --log debug");
}
else if (this.SuppressInitialDiagnosticInformation)
{
argsBuilder.Append(" --log fatal");
}
else if (this.LogLevel != FirefoxDriverLogLevel.Default)
{
argsBuilder.Append($" --log {this.LogLevel.ToString().ToLowerInvariant()}");
}
if (this.OpenBrowserToolbox)
{
argsBuilder.Append(" --jsdebugger");
}
if (this.LogTruncate is true)
{
argsBuilder.Append(" --log-no-truncate");
}
if (!string.IsNullOrEmpty(this.ProfileRoot))
{
if (!Directory.Exists(this.ProfileRoot))
{
throw new ArgumentException($"Profile root directory does not exist: {this.ProfileRoot}", nameof(ProfileRoot));
}
argsBuilder.AppendFormat(CultureInfo.InvariantCulture, " --profile-root \"{0}\"", this.ProfileRoot);
}
return argsBuilder.ToString().Trim();
}
}
/// <summary>
/// Gets a value indicating whether process output redirection is required.
/// </summary>
protected internal override bool EnableProcessRedirection => base.EnableProcessRedirection || LogPath is not null;
/// <summary>
/// Called when the driver process is starting. This method sets up log file writing if a log path is specified.
/// </summary>
/// <param name="eventArgs">The event arguments containing information about the driver service process.</param>
/// <remarks>
/// This method initializes a log writer if a log path is specified.
/// </remarks>
protected override void OnDriverProcessStarting(DriverProcessStartingEventArgs eventArgs)
{
if (!string.IsNullOrEmpty(this.LogPath))
{
string? directory = Path.GetDirectoryName(this.LogPath);
if (!string.IsNullOrEmpty(directory) && !Directory.Exists(directory))
{
Directory.CreateDirectory(directory);
}
logWriter = new StreamWriter(this.LogPath, append: true) { AutoFlush = true };
}
base.OnDriverProcessStarting(eventArgs);
}
/// <summary>
/// Handles the output and error data received from the driver process and sends it to the log writer if available.
/// </summary>
/// <param name="sender">The sender of the event.</param>
/// <param name="args">The data received event arguments.</param>
protected override void OnDriverProcessDataReceived(object sender, DataReceivedEventArgs args)
{
if (string.IsNullOrEmpty(args.Data))
return;
if (!string.IsNullOrEmpty(this.LogPath))
{
if (logWriter != null)
{
logWriter.WriteLine(args.Data);
}
}
else
{
base.OnDriverProcessDataReceived(sender, args);
}
}
/// <summary>
/// Disposes of the resources used by the <see cref="FirefoxDriverService"/> instance.
/// </summary>
/// <param name="disposing">A value indicating whether the method is being called from Dispose.</param>
/// <remarks>
/// If disposing is true, it disposes of the log writer if it exists.
/// </remarks>
protected override void Dispose(bool disposing)
{
base.Dispose(disposing);
if (logWriter != null && disposing)
{
logWriter.Dispose();
logWriter = null;
}
}
/// <summary>
/// Creates a default instance of the FirefoxDriverService.
/// </summary>
/// <returns>A FirefoxDriverService that implements default settings.</returns>
public static FirefoxDriverService CreateDefaultService()
{
return new FirefoxDriverService(null, null, PortUtilities.FindFreePort());
}
/// <summary>
/// Creates a default instance of the FirefoxDriverService using a specified path to the Firefox driver executable.
/// </summary>
/// <param name="driverPath">The path to the executable or the directory containing the Firefox driver executable.</param>
/// <returns>A FirefoxDriverService using a random port.</returns>
public static FirefoxDriverService CreateDefaultService(string? driverPath)
{
if (File.Exists(driverPath))
{
string fileName = Path.GetFileName(driverPath);
string driverFolder = Path.GetDirectoryName(driverPath)!;
return CreateDefaultService(driverFolder, fileName);
}
else
{
string fileName = FirefoxDriverServiceFileName();
string? driverFolder = driverPath;
return CreateDefaultService(driverFolder, fileName);
}
}
/// <summary>
/// Creates a default instance of the FirefoxDriverService using a specified path to the Firefox driver executable with the given name.
/// </summary>
/// <param name="driverPath">The directory containing the Firefox driver executable.</param>
/// <param name="driverExecutableFileName">The name of the Firefox driver executable file.</param>
/// <returns>A FirefoxDriverService using a random port.</returns>
public static FirefoxDriverService CreateDefaultService(string? driverPath, string? driverExecutableFileName)
{
return new FirefoxDriverService(driverPath, driverExecutableFileName, PortUtilities.FindFreePort());
}
/// <summary>
/// Returns the Firefox driver filename for the currently running platform
/// </summary>
/// <returns>The file name of the Firefox driver service executable.</returns>
private static string FirefoxDriverServiceFileName()
{
string fileName = DefaultFirefoxDriverServiceFileName;
// Unfortunately, detecting the currently running platform isn't as
// straightforward as you might hope.
// See: http://mono.wikia.com/wiki/Detecting_the_execution_platform
// and https://msdn.microsoft.com/en-us/library/3a8hyw88(v=vs.110).aspx
const PlatformID PlatformIDMonoUnix = (PlatformID)128;
switch (Environment.OSVersion.Platform)
{
case PlatformID.Win32NT:
case PlatformID.Win32S:
case PlatformID.Win32Windows:
case PlatformID.WinCE:
fileName += ".exe";
break;
case PlatformID.MacOSX:
case PlatformID.Unix:
case PlatformIDMonoUnix:
break;
// Don't handle the Xbox case. Let default handle it.
// case PlatformID.Xbox:
// break;
default:
throw new WebDriverException("Unsupported platform: " + Environment.OSVersion.Platform);
}
return fileName;
}
}