use std::path::PathBuf;
use base::id::PipelineId;
use constellation_traits::EmbedderToConstellationMessage;
use embedder_traits::{
AllowOrDeny, AuthenticationResponse, ContextMenuResult, Cursor, FilterPattern,
GamepadHapticEffectType, InputMethodType, LoadStatus, MediaSessionEvent, Notification,
PermissionFeature, ScreenGeometry, SelectElementOptionOrOptgroup, SimpleDialog,
WebResourceRequest, WebResourceResponse, WebResourceResponseMsg,
};
use ipc_channel::ipc::IpcSender;
use keyboard_types::KeyboardEvent;
use serde::Serialize;
use url::Url;
use webrender_api::units::{DeviceIntPoint, DeviceIntRect, DeviceIntSize};
use crate::responders::ServoErrorSender;
use crate::{ConstellationProxy, WebView};
pub struct NavigationRequest {
pub url: Url,
pub(crate) pipeline_id: PipelineId,
pub(crate) constellation_proxy: ConstellationProxy,
pub(crate) response_sent: bool,
}
impl NavigationRequest {
pub fn allow(mut self) {
self.constellation_proxy
.send(EmbedderToConstellationMessage::AllowNavigationResponse(
self.pipeline_id,
true,
));
self.response_sent = true;
}
pub fn deny(mut self) {
self.constellation_proxy
.send(EmbedderToConstellationMessage::AllowNavigationResponse(
self.pipeline_id,
false,
));
self.response_sent = true;
}
}
impl Drop for NavigationRequest {
fn drop(&mut self) {
if !self.response_sent {
self.constellation_proxy
.send(EmbedderToConstellationMessage::AllowNavigationResponse(
self.pipeline_id,
true,
));
}
}
}
pub(crate) struct IpcResponder<T: Serialize> {
response_sender: IpcSender<T>,
response_sent: bool,
default_response: Option<T>,
}
impl<T: Serialize> IpcResponder<T> {
pub(crate) fn new(response_sender: IpcSender<T>, default_response: T) -> Self {
Self {
response_sender,
response_sent: false,
default_response: Some(default_response),
}
}
pub(crate) fn send(&mut self, response: T) -> bincode::Result<()> {
let result = self.response_sender.send(response);
self.response_sent = true;
result
}
pub(crate) fn into_inner(self) -> IpcSender<T> {
self.response_sender.clone()
}
}
impl<T: Serialize> Drop for IpcResponder<T> {
fn drop(&mut self) {
if !self.response_sent {
let response = self
.default_response
.take()
.expect("Guaranteed by inherent impl");
let _ = self.response_sender.send(response);
}
}
}
pub struct PermissionRequest {
pub(crate) requested_feature: PermissionFeature,
pub(crate) allow_deny_request: AllowOrDenyRequest,
}
impl PermissionRequest {
pub fn feature(&self) -> PermissionFeature {
self.requested_feature
}
pub fn allow(self) {
self.allow_deny_request.allow();
}
pub fn deny(self) {
self.allow_deny_request.deny();
}
}
pub struct AllowOrDenyRequest(IpcResponder<AllowOrDeny>, ServoErrorSender);
impl AllowOrDenyRequest {
pub(crate) fn new(
response_sender: IpcSender<AllowOrDeny>,
default_response: AllowOrDeny,
error_sender: ServoErrorSender,
) -> Self {
Self(
IpcResponder::new(response_sender, default_response),
error_sender,
)
}
pub fn allow(mut self) {
if let Err(error) = self.0.send(AllowOrDeny::Allow) {
self.1.raise_response_send_error(error);
}
}
pub fn deny(mut self) {
if let Err(error) = self.0.send(AllowOrDeny::Deny) {
self.1.raise_response_send_error(error);
}
}
}
pub struct AuthenticationRequest {
pub(crate) url: Url,
pub(crate) for_proxy: bool,
pub(crate) responder: IpcResponder<Option<AuthenticationResponse>>,
pub(crate) error_sender: ServoErrorSender,
}
impl AuthenticationRequest {
pub(crate) fn new(
url: Url,
for_proxy: bool,
response_sender: IpcSender<Option<AuthenticationResponse>>,
error_sender: ServoErrorSender,
) -> Self {
Self {
url,
for_proxy,
responder: IpcResponder::new(response_sender, None),
error_sender,
}
}
pub fn url(&self) -> &Url {
&self.url
}
pub fn for_proxy(&self) -> bool {
self.for_proxy
}
pub fn authenticate(mut self, username: String, password: String) {
if let Err(error) = self
.responder
.send(Some(AuthenticationResponse { username, password }))
{
self.error_sender.raise_response_send_error(error);
}
}
}
pub struct WebResourceLoad {
pub request: WebResourceRequest,
pub(crate) responder: IpcResponder<WebResourceResponseMsg>,
pub(crate) error_sender: ServoErrorSender,
}
impl WebResourceLoad {
pub(crate) fn new(
web_resource_request: WebResourceRequest,
response_sender: IpcSender<WebResourceResponseMsg>,
error_sender: ServoErrorSender,
) -> Self {
Self {
request: web_resource_request,
responder: IpcResponder::new(response_sender, WebResourceResponseMsg::DoNotIntercept),
error_sender,
}
}
pub fn request(&self) -> &WebResourceRequest {
&self.request
}
pub fn intercept(mut self, response: WebResourceResponse) -> InterceptedWebResourceLoad {
if let Err(error) = self.responder.send(WebResourceResponseMsg::Start(response)) {
self.error_sender.raise_response_send_error(error);
}
InterceptedWebResourceLoad {
request: self.request.clone(),
response_sender: self.responder.into_inner(),
finished: false,
error_sender: self.error_sender,
}
}
}
pub struct InterceptedWebResourceLoad {
pub request: WebResourceRequest,
pub(crate) response_sender: IpcSender<WebResourceResponseMsg>,
pub(crate) finished: bool,
pub(crate) error_sender: ServoErrorSender,
}
impl InterceptedWebResourceLoad {
pub fn send_body_data(&self, data: Vec<u8>) {
if let Err(error) = self
.response_sender
.send(WebResourceResponseMsg::SendBodyData(data))
{
self.error_sender.raise_response_send_error(error);
}
}
pub fn finish(mut self) {
if let Err(error) = self
.response_sender
.send(WebResourceResponseMsg::FinishLoad)
{
self.error_sender.raise_response_send_error(error);
}
self.finished = true;
}
pub fn cancel(mut self) {
if let Err(error) = self
.response_sender
.send(WebResourceResponseMsg::CancelLoad)
{
self.error_sender.raise_response_send_error(error);
}
self.finished = true;
}
}
impl Drop for InterceptedWebResourceLoad {
fn drop(&mut self) {
if !self.finished {
if let Err(error) = self
.response_sender
.send(WebResourceResponseMsg::FinishLoad)
{
self.error_sender.raise_response_send_error(error);
}
}
}
}
pub enum FormControl {
SelectElement(SelectElement),
}
pub struct SelectElement {
pub(crate) options: Vec<SelectElementOptionOrOptgroup>,
pub(crate) selected_option: Option<usize>,
pub(crate) position: DeviceIntRect,
pub(crate) responder: IpcResponder<Option<usize>>,
}
impl SelectElement {
pub(crate) fn new(
options: Vec<SelectElementOptionOrOptgroup>,
selected_option: Option<usize>,
position: DeviceIntRect,
ipc_sender: IpcSender<Option<usize>>,
) -> Self {
Self {
options,
selected_option,
position,
responder: IpcResponder::new(ipc_sender, None),
}
}
pub fn position(&self) -> DeviceIntRect {
self.position
}
pub fn options(&self) -> &[SelectElementOptionOrOptgroup] {
&self.options
}
pub fn select(&mut self, id: Option<usize>) {
self.selected_option = id;
}
pub fn selected_option(&self) -> Option<usize> {
self.selected_option
}
pub fn submit(mut self) {
let _ = self.responder.send(self.selected_option);
}
}
pub trait WebViewDelegate {
fn screen_geometry(&self, _webview: WebView) -> Option<ScreenGeometry> {
None
}
fn notify_url_changed(&self, _webview: WebView, _url: Url) {}
fn notify_page_title_changed(&self, _webview: WebView, _title: Option<String>) {}
fn notify_status_text_changed(&self, _webview: WebView, _status: Option<String>) {}
fn notify_focus_changed(&self, _webview: WebView, _focused: bool) {}
fn notify_animating_changed(&self, _webview: WebView, _animating: bool) {}
fn notify_load_status_changed(&self, _webview: WebView, _status: LoadStatus) {}
fn notify_cursor_changed(&self, _webview: WebView, _: Cursor) {}
fn notify_favicon_url_changed(&self, _webview: WebView, _: Url) {}
fn notify_new_frame_ready(&self, _webview: WebView) {}
fn notify_history_changed(&self, _webview: WebView, _: Vec<Url>, _: usize) {}
fn notify_closed(&self, _webview: WebView) {}
fn notify_keyboard_event(&self, _webview: WebView, _: KeyboardEvent) {}
fn notify_crashed(&self, _webview: WebView, _reason: String, _backtrace: Option<String>) {}
fn notify_media_session_event(&self, _webview: WebView, _event: MediaSessionEvent) {}
fn notify_fullscreen_state_changed(&self, _webview: WebView, _: bool) {}
fn request_navigation(&self, _webview: WebView, _navigation_request: NavigationRequest) {}
fn request_unload(&self, _webview: WebView, _unload_request: AllowOrDenyRequest) {}
fn request_move_to(&self, _webview: WebView, _: DeviceIntPoint) {}
fn request_resize_to(&self, _webview: WebView, _: DeviceIntSize) {}
fn request_open_auxiliary_webview(&self, _parent_webview: WebView) -> Option<WebView> {
None
}
fn request_permission(&self, _webview: WebView, _: PermissionRequest) {}
fn request_authentication(
&self,
_webview: WebView,
_authentication_request: AuthenticationRequest,
) {
}
fn show_simple_dialog(&self, _webview: WebView, dialog: SimpleDialog) {
let _ = match dialog {
SimpleDialog::Alert {
response_sender, ..
} => response_sender.send(Default::default()),
SimpleDialog::Confirm {
response_sender, ..
} => response_sender.send(Default::default()),
SimpleDialog::Prompt {
response_sender, ..
} => response_sender.send(Default::default()),
};
}
fn show_context_menu(
&self,
_webview: WebView,
result_sender: IpcSender<ContextMenuResult>,
_: Option<String>,
_: Vec<String>,
) {
let _ = result_sender.send(ContextMenuResult::Ignored);
}
fn show_bluetooth_device_dialog(
&self,
_webview: WebView,
_: Vec<String>,
response_sender: IpcSender<Option<String>>,
) {
let _ = response_sender.send(None);
}
fn show_file_selection_dialog(
&self,
_webview: WebView,
_filter_pattern: Vec<FilterPattern>,
_allow_select_mutiple: bool,
response_sender: IpcSender<Option<Vec<PathBuf>>>,
) {
let _ = response_sender.send(None);
}
fn show_ime(
&self,
_webview: WebView,
_type: InputMethodType,
_text: Option<(String, i32)>,
_multiline: bool,
_position: DeviceIntRect,
) {
}
fn hide_ime(&self, _webview: WebView) {}
fn show_form_control(&self, _webview: WebView, _form_control: FormControl) {}
fn play_gamepad_haptic_effect(
&self,
_webview: WebView,
_: usize,
_: GamepadHapticEffectType,
_: IpcSender<bool>,
) {
}
fn stop_gamepad_haptic_effect(&self, _webview: WebView, _: usize, _: IpcSender<bool>) {}
fn load_web_resource(&self, _webview: WebView, _load: WebResourceLoad) {}
fn show_notification(&self, _webview: WebView, _notification: Notification) {}
}
pub(crate) struct DefaultWebViewDelegate;
impl WebViewDelegate for DefaultWebViewDelegate {}
#[test]
fn test_allow_deny_request() {
use ipc_channel::ipc;
use crate::ServoErrorChannel;
for default_response in [AllowOrDeny::Allow, AllowOrDeny::Deny] {
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AllowOrDenyRequest::new(sender, default_response, errors.sender());
request.allow();
assert_eq!(receiver.try_recv().ok(), Some(AllowOrDeny::Allow));
assert_eq!(receiver.try_recv().ok(), None);
assert!(errors.try_recv().is_none());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AllowOrDenyRequest::new(sender, default_response, errors.sender());
request.deny();
assert_eq!(receiver.try_recv().ok(), Some(AllowOrDeny::Deny));
assert_eq!(receiver.try_recv().ok(), None);
assert!(errors.try_recv().is_none());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AllowOrDenyRequest::new(sender, default_response, errors.sender());
drop(request);
assert_eq!(receiver.try_recv().ok(), Some(default_response));
assert_eq!(receiver.try_recv().ok(), None);
assert!(errors.try_recv().is_none());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AllowOrDenyRequest::new(sender, default_response, errors.sender());
drop(receiver);
request.allow();
assert!(errors.try_recv().is_some());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AllowOrDenyRequest::new(sender, default_response, errors.sender());
drop(receiver);
request.deny();
assert!(errors.try_recv().is_some());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AllowOrDenyRequest::new(sender, default_response, errors.sender());
drop(receiver);
drop(request);
assert!(errors.try_recv().is_none());
}
}
#[test]
fn test_authentication_request() {
use ipc_channel::ipc;
use crate::ServoErrorChannel;
let url = Url::parse("https://example.com").expect("Guaranteed by argument");
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AuthenticationRequest::new(url.clone(), false, sender, errors.sender());
request.authenticate("diffie".to_owned(), "hunter2".to_owned());
assert_eq!(
receiver.try_recv().ok(),
Some(Some(AuthenticationResponse {
username: "diffie".to_owned(),
password: "hunter2".to_owned(),
}))
);
assert_eq!(receiver.try_recv().ok(), None);
assert!(errors.try_recv().is_none());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AuthenticationRequest::new(url.clone(), false, sender, errors.sender());
drop(request);
assert_eq!(receiver.try_recv().ok(), Some(None));
assert_eq!(receiver.try_recv().ok(), None);
assert!(errors.try_recv().is_none());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AuthenticationRequest::new(url.clone(), false, sender, errors.sender());
drop(receiver);
request.authenticate("diffie".to_owned(), "hunter2".to_owned());
assert!(errors.try_recv().is_some());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = AuthenticationRequest::new(url.clone(), false, sender, errors.sender());
drop(receiver);
drop(request);
assert!(errors.try_recv().is_none());
}
#[test]
fn test_web_resource_load() {
use http::{HeaderMap, Method, StatusCode};
use ipc_channel::ipc;
use crate::ServoErrorChannel;
let web_resource_request = || WebResourceRequest {
method: Method::GET,
headers: HeaderMap::default(),
url: Url::parse("https://example.com").expect("Guaranteed by argument"),
is_for_main_frame: false,
is_redirect: false,
};
let web_resource_response = || {
WebResourceResponse::new(Url::parse("https://diffie.test").expect("Guaranteed by argument"))
.status_code(StatusCode::IM_A_TEAPOT)
};
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = WebResourceLoad::new(web_resource_request(), sender, errors.sender());
request.intercept(web_resource_response()).cancel();
assert!(matches!(
receiver.try_recv(),
Ok(WebResourceResponseMsg::Start(_))
));
assert!(matches!(
receiver.try_recv(),
Ok(WebResourceResponseMsg::CancelLoad)
));
assert!(matches!(receiver.try_recv(), Err(_)));
assert!(errors.try_recv().is_none());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = WebResourceLoad::new(web_resource_request(), sender, errors.sender());
drop(request.intercept(web_resource_response()));
assert!(matches!(
receiver.try_recv(),
Ok(WebResourceResponseMsg::Start(_))
));
assert!(matches!(
receiver.try_recv(),
Ok(WebResourceResponseMsg::FinishLoad)
));
assert!(matches!(receiver.try_recv(), Err(_)));
assert!(errors.try_recv().is_none());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = WebResourceLoad::new(web_resource_request(), sender, errors.sender());
drop(request);
assert!(matches!(
receiver.try_recv(),
Ok(WebResourceResponseMsg::DoNotIntercept)
));
assert!(matches!(receiver.try_recv(), Err(_)));
assert!(errors.try_recv().is_none());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = WebResourceLoad::new(web_resource_request(), sender, errors.sender());
drop(receiver);
request.intercept(web_resource_response()).cancel();
assert!(errors.try_recv().is_some());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = WebResourceLoad::new(web_resource_request(), sender, errors.sender());
drop(receiver);
drop(request.intercept(web_resource_response()));
assert!(errors.try_recv().is_some());
let errors = ServoErrorChannel::default();
let (sender, receiver) = ipc::channel().expect("Failed to create IPC channel");
let request = WebResourceLoad::new(web_resource_request(), sender, errors.sender());
drop(receiver);
drop(request);
assert!(errors.try_recv().is_none());
}