Fix failed request for adapter when not available (#31002)

Signed-off-by: Bentaimia Haddadi <haddadi.taym@gmail.com>
This commit is contained in:
Taym Haddadi 2024-01-09 10:12:53 +01:00 committed by GitHub
parent dd0149d953
commit fddc4a430f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 97 additions and 90 deletions

View file

@ -2063,7 +2063,11 @@ where
}; };
match request { match request {
FromScriptMsg::RequestAdapter(response_sender, options, ids) => match webgpu_chan { FromScriptMsg::RequestAdapter(response_sender, options, ids) => match webgpu_chan {
None => warn!("Failed to send request adapter message, missing WebGPU channel"), None => {
if let Err(e) = response_sender.send(None) {
return warn!("Failed to send request adapter message: {}", e);
}
},
Some(webgpu_chan) => { Some(webgpu_chan) => {
let adapter_request = WebGPURequest::RequestAdapter { let adapter_request = WebGPURequest::RequestAdapter {
sender: response_sender, sender: response_sender,

View file

@ -45,7 +45,7 @@ impl GPU {
} }
pub trait AsyncWGPUListener { pub trait AsyncWGPUListener {
fn handle_response(&self, response: WebGPUResponseResult, promise: &Rc<Promise>); fn handle_response(&self, response: Option<WebGPUResponseResult>, promise: &Rc<Promise>);
} }
struct WGPUResponse<T: AsyncWGPUListener + DomObject> { struct WGPUResponse<T: AsyncWGPUListener + DomObject> {
@ -55,7 +55,7 @@ struct WGPUResponse<T: AsyncWGPUListener + DomObject> {
impl<T: AsyncWGPUListener + DomObject> WGPUResponse<T> { impl<T: AsyncWGPUListener + DomObject> WGPUResponse<T> {
#[allow(crown::unrooted_must_root)] #[allow(crown::unrooted_must_root)]
fn response(self, response: WebGPUResponseResult) { fn response(self, response: Option<WebGPUResponseResult>) {
let promise = self.trusted.root(); let promise = self.trusted.root();
self.receiver.root().handle_response(response, &promise); self.receiver.root().handle_response(response, &promise);
} }
@ -64,13 +64,13 @@ impl<T: AsyncWGPUListener + DomObject> WGPUResponse<T> {
pub fn response_async<T: AsyncWGPUListener + DomObject + 'static>( pub fn response_async<T: AsyncWGPUListener + DomObject + 'static>(
promise: &Rc<Promise>, promise: &Rc<Promise>,
receiver: &T, receiver: &T,
) -> IpcSender<WebGPUResponseResult> { ) -> IpcSender<Option<WebGPUResponseResult>> {
let (action_sender, action_receiver) = ipc::channel().unwrap(); let (action_sender, action_receiver) = ipc::channel().unwrap();
let task_source = receiver.global().dom_manipulation_task_source(); let task_source = receiver.global().dom_manipulation_task_source();
let canceller = receiver let canceller = receiver
.global() .global()
.task_canceller(TaskSourceName::DOMManipulation); .task_canceller(TaskSourceName::DOMManipulation);
let mut trusted = Some(TrustedPromise::new(promise.clone())); let mut trusted: Option<TrustedPromise> = Some(TrustedPromise::new(promise.clone()));
let trusted_receiver = Trusted::new(receiver); let trusted_receiver = Trusted::new(receiver);
ROUTER.add_route( ROUTER.add_route(
action_receiver.to_opaque(), action_receiver.to_opaque(),
@ -139,8 +139,9 @@ impl GPUMethods for GPU {
} }
impl AsyncWGPUListener for GPU { impl AsyncWGPUListener for GPU {
fn handle_response(&self, response: WebGPUResponseResult, promise: &Rc<Promise>) { fn handle_response(&self, response: Option<WebGPUResponseResult>, promise: &Rc<Promise>) {
match response { match response {
Some(response) => match response {
Ok(WebGPUResponse::RequestAdapter { Ok(WebGPUResponse::RequestAdapter {
adapter_info, adapter_info,
adapter_id, adapter_id,
@ -168,9 +169,11 @@ impl AsyncWGPUListener for GPU {
warn!("Could not get GPUAdapter ({:?})", e); warn!("Could not get GPUAdapter ({:?})", e);
promise.resolve_native(&None::<GPUAdapter>); promise.resolve_native(&None::<GPUAdapter>);
}, },
_ => { Ok(_) => unreachable!("GPU received wrong WebGPUResponse"),
warn!("GPU received wrong WebGPUResponse"); },
promise.reject_error(Error::Operation); None => {
warn!("Couldn't get a response, because WebGPU is disabled");
promise.resolve_native(&None::<GPUAdapter>);
}, },
} }
} }

View file

@ -237,8 +237,9 @@ impl GPUAdapterMethods for GPUAdapter {
} }
impl AsyncWGPUListener for GPUAdapter { impl AsyncWGPUListener for GPUAdapter {
fn handle_response(&self, response: WebGPUResponseResult, promise: &Rc<Promise>) { fn handle_response(&self, response: Option<WebGPUResponseResult>, promise: &Rc<Promise>) {
match response { match response {
Some(response) => match response {
Ok(WebGPUResponse::RequestDevice { Ok(WebGPUResponse::RequestDevice {
device_id, device_id,
queue_id, queue_id,
@ -262,10 +263,9 @@ impl AsyncWGPUListener for GPUAdapter {
warn!("Could not get GPUDevice({:?})", e); warn!("Could not get GPUDevice({:?})", e);
promise.reject_error(Error::Operation); promise.reject_error(Error::Operation);
}, },
_ => { Ok(_) => unreachable!("GPUAdapter received wrong WebGPUResponse"),
warn!("GPUAdapter received wrong WebGPUResponse");
promise.reject_error(Error::Operation);
}, },
None => unreachable!("Failed to get a response for RequestDevice"),
} }
} }
} }

View file

@ -346,8 +346,9 @@ impl GPUBufferMethods for GPUBuffer {
impl AsyncWGPUListener for GPUBuffer { impl AsyncWGPUListener for GPUBuffer {
#[allow(unsafe_code)] #[allow(unsafe_code)]
fn handle_response(&self, response: WebGPUResponseResult, promise: &Rc<Promise>) { fn handle_response(&self, response: Option<WebGPUResponseResult>, promise: &Rc<Promise>) {
match response { match response {
Some(response) => match response {
Ok(WebGPUResponse::BufferMapAsync(bytes)) => { Ok(WebGPUResponse::BufferMapAsync(bytes)) => {
*self *self
.map_info .map_info
@ -363,10 +364,9 @@ impl AsyncWGPUListener for GPUBuffer {
warn!("Could not map buffer({:?})", e); warn!("Could not map buffer({:?})", e);
promise.reject_error(Error::Abort); promise.reject_error(Error::Abort);
}, },
_ => { Ok(_) => unreachable!("GPUBuffer received wrong WebGPUResponse"),
warn!("GPUBuffer received wrong WebGPUResponse");
promise.reject_error(Error::Operation);
}, },
None => unreachable!("Failed to get a response for BufferMapAsync"),
} }
*self.map_promise.borrow_mut() = None; *self.map_promise.borrow_mut() = None;
if let Err(e) = self if let Err(e) = self

View file

@ -258,7 +258,7 @@ pub enum ScriptMsg {
MediaSessionEvent(PipelineId, MediaSessionEvent), MediaSessionEvent(PipelineId, MediaSessionEvent),
/// Create a WebGPU Adapter instance /// Create a WebGPU Adapter instance
RequestAdapter( RequestAdapter(
IpcSender<WebGPUResponseResult>, IpcSender<Option<WebGPUResponseResult>>,
wgpu::instance::RequestAdapterOptions, wgpu::instance::RequestAdapterOptions,
SmallVec<[wgpu::id::AdapterId; 4]>, SmallVec<[wgpu::id::AdapterId; 4]>,
), ),

View file

@ -75,7 +75,7 @@ pub type WebGPUResponseResult = Result<WebGPUResponse, String>;
#[derive(Debug, Deserialize, Serialize)] #[derive(Debug, Deserialize, Serialize)]
pub enum WebGPURequest { pub enum WebGPURequest {
BufferMapAsync { BufferMapAsync {
sender: IpcSender<WebGPUResponseResult>, sender: IpcSender<Option<WebGPUResponseResult>>,
buffer_id: id::BufferId, buffer_id: id::BufferId,
device_id: id::DeviceId, device_id: id::DeviceId,
host_map: HostMap, host_map: HostMap,
@ -202,12 +202,12 @@ pub enum WebGPURequest {
device_id: id::DeviceId, device_id: id::DeviceId,
}, },
RequestAdapter { RequestAdapter {
sender: IpcSender<WebGPUResponseResult>, sender: IpcSender<Option<WebGPUResponseResult>>,
options: RequestAdapterOptions, options: RequestAdapterOptions,
ids: SmallVec<[id::AdapterId; 4]>, ids: SmallVec<[id::AdapterId; 4]>,
}, },
RequestDevice { RequestDevice {
sender: IpcSender<WebGPUResponseResult>, sender: IpcSender<Option<WebGPUResponseResult>>,
adapter_id: WebGPUAdapter, adapter_id: WebGPUAdapter,
descriptor: wgt::DeviceDescriptor<Option<String>>, descriptor: wgt::DeviceDescriptor<Option<String>>,
device_id: id::DeviceId, device_id: id::DeviceId,
@ -340,7 +340,7 @@ struct WGPU<'a> {
// Track invalid adapters https://gpuweb.github.io/gpuweb/#invalid // Track invalid adapters https://gpuweb.github.io/gpuweb/#invalid
_invalid_adapters: Vec<WebGPUAdapter>, _invalid_adapters: Vec<WebGPUAdapter>,
// Buffers with pending mapping // Buffers with pending mapping
buffer_maps: HashMap<id::BufferId, Rc<BufferMapInfo<'a, WebGPUResponseResult>>>, buffer_maps: HashMap<id::BufferId, Rc<BufferMapInfo<'a, Option<WebGPUResponseResult>>>>,
// Presentation Buffers with pending mapping // Presentation Buffers with pending mapping
present_buffer_maps: present_buffer_maps:
HashMap<id::BufferId, Rc<BufferMapInfo<'a, (Option<ErrorScopeId>, WebGPURequest)>>>, HashMap<id::BufferId, Rc<BufferMapInfo<'a, (Option<ErrorScopeId>, WebGPURequest)>>>,
@ -423,7 +423,7 @@ impl<'a> WGPU<'a> {
userdata: *mut u8, userdata: *mut u8,
) { ) {
let info = Rc::from_raw( let info = Rc::from_raw(
userdata as *const BufferMapInfo<WebGPUResponseResult>, userdata as *const BufferMapInfo<Option<WebGPUResponseResult>>,
); );
let msg = match status { let msg = match status {
BufferMapAsyncStatus::Success => { BufferMapAsyncStatus::Success => {
@ -442,7 +442,7 @@ impl<'a> WGPU<'a> {
Err(String::from("Failed to map Buffer")) Err(String::from("Failed to map Buffer"))
}, },
}; };
if let Err(e) = info.sender.send(msg) { if let Err(e) = info.sender.send(Some(msg)) {
warn!("Could not send BufferMapAsync Response ({})", e); warn!("Could not send BufferMapAsync Response ({})", e);
} }
} }
@ -461,7 +461,7 @@ impl<'a> WGPU<'a> {
let global = &self.global; let global = &self.global;
let result = gfx_select!(buffer_id => global.buffer_map_async(buffer_id, map_range, operation)); let result = gfx_select!(buffer_id => global.buffer_map_async(buffer_id, map_range, operation));
if let Err(ref e) = result { if let Err(ref e) = result {
if let Err(w) = sender.send(Err(format!("{:?}", e))) { if let Err(w) = sender.send(Some(Err(format!("{:?}", e)))) {
warn!("Failed to send BufferMapAsync Response ({:?})", w); warn!("Failed to send BufferMapAsync Response ({:?})", w);
} }
} }
@ -879,7 +879,7 @@ impl<'a> WGPU<'a> {
) { ) {
Ok(id) => id, Ok(id) => id,
Err(w) => { Err(w) => {
if let Err(e) = sender.send(Err(format!("{:?}", w))) { if let Err(e) = sender.send(Some(Err(format!("{:?}", w)))) {
warn!( warn!(
"Failed to send response to WebGPURequest::RequestAdapter ({})", "Failed to send response to WebGPURequest::RequestAdapter ({})",
e e
@ -898,13 +898,13 @@ impl<'a> WGPU<'a> {
gfx_select!(adapter_id => global.adapter_limits(adapter_id)).unwrap(); gfx_select!(adapter_id => global.adapter_limits(adapter_id)).unwrap();
let features = let features =
gfx_select!(adapter_id => global.adapter_features(adapter_id)).unwrap(); gfx_select!(adapter_id => global.adapter_features(adapter_id)).unwrap();
if let Err(e) = sender.send(Ok(WebGPUResponse::RequestAdapter { if let Err(e) = sender.send(Some(Ok(WebGPUResponse::RequestAdapter {
adapter_info: info, adapter_info: info,
adapter_id: adapter, adapter_id: adapter,
features, features,
limits, limits,
channel: WebGPU(self.sender.clone()), channel: WebGPU(self.sender.clone()),
})) { }))) {
warn!( warn!(
"Failed to send response to WebGPURequest::RequestAdapter ({})", "Failed to send response to WebGPURequest::RequestAdapter ({})",
e e
@ -934,7 +934,7 @@ impl<'a> WGPU<'a> {
) { ) {
Ok(id) => id, Ok(id) => id,
Err(e) => { Err(e) => {
if let Err(w) = sender.send(Err(format!("{:?}", e))) { if let Err(w) = sender.send(Some(Err(format!("{:?}", e)))) {
warn!( warn!(
"Failed to send response to WebGPURequest::RequestDevice ({})", "Failed to send response to WebGPURequest::RequestDevice ({})",
w w
@ -948,11 +948,11 @@ impl<'a> WGPU<'a> {
// since wgpu-core uses the same id for the device and the queue // since wgpu-core uses the same id for the device and the queue
let queue = WebGPUQueue(id); let queue = WebGPUQueue(id);
self.devices.insert(device, pipeline_id); self.devices.insert(device, pipeline_id);
if let Err(e) = sender.send(Ok(WebGPUResponse::RequestDevice { if let Err(e) = sender.send(Some(Ok(WebGPUResponse::RequestDevice {
device_id: device, device_id: device,
queue_id: queue, queue_id: queue,
descriptor, descriptor,
})) { }))) {
warn!( warn!(
"Failed to send response to WebGPURequest::RequestDevice ({})", "Failed to send response to WebGPURequest::RequestDevice ({})",
e e