perf(macOS): avoid copying large protocol bodies (#1719)

* perf(macOS): avoid copying large protocol bodies

* refatcor: mirage `NSData::initWithBytes_length` to `NSData::with_bytes`

* Refine WKURLSchemeTask body handling

* chore: comment on optimize buffer handling in start_task to reduce copying for larger responses

* perf: reduce NO_COPY_DATA_THRESHOLD to 64 KB for improved buffer handling

* perf: increase NO_COPY_DATA_THRESHOLD to 128 KB for improved buffer handling

* update url_scheme_handler.rs

Co-authored-by: Tony <[email protected]>

* update macos-protocol-body-nocopy.md

* cargo fmt

* Remove unused `NO_COPY_DATA_THRESHOLD`

---------

Co-authored-by: Tony <[email protected]>
Co-authored-by: Tony <[email protected]>

Author: Tunglies

.changes/macos-protocol-body-nocopy.md

@@ -0,0 +1,5 @@
+---
+"wry": patch
+---
+
+On macOS, avoid an extra copy for owned custom protocol response bodies by transferring the body buffer into `NSData`.

Cargo.lock

@@ -138,6 +138,7 @@ objc2-core-foundation = { version = "0.3.0", default-features = false, features
 ] }
 objc2-foundation = { version = "0.3.0", default-features = false, features = [
   "std",
+  "block2",
   "objc2-core-foundation",
   "NSURLRequest",
   "NSURL",

Cargo.toml

@@ -4,7 +4,7 @@
 
 use std::{
   borrow::Cow,
-  ffi::{c_char, c_void, CStr},
+  ffi::{c_char, CStr},
   panic::AssertUnwindSafe,
   ptr::NonNull,
 };
@@ -214,7 +214,7 @@ extern "C" fn start_task(
                 check_webview_id_valid(webview_id)?;
                 check_task_is_valid(&webview, task_key, task_uuid.clone())?;
 
-                let content = sent_response.body();
+                let content_len = sent_response.body().len();
                 // default: application/octet-stream, but should be provided by the client
                 let wanted_mime = sent_response.headers().get(CONTENT_TYPE);
                 // default to 200
@@ -231,7 +231,7 @@ extern "C" fn start_task(
                 }
                 headers.insert(
                   &*NSString::from_str(CONTENT_LENGTH.as_str()),
-                  &*NSString::from_str(&content.len().to_string()),
+                  &*NSString::from_str(&content_len.to_string()),
                 );
 
                 // add headers
@@ -264,15 +264,10 @@ extern "C" fn start_task(
                 }))
                 .map_err(|_e| crate::Error::CustomProtocolTaskInvalid)?;
 
-                // Send data
-                let data = NSData::alloc();
-                // MIGRATE NOTE: we copied the content to the NSData because content will be freed
-                // when out of scope but NSData will also free the content when it's done and cause doube free.
-                let data = NSData::initWithBytes_length(
-                  data,
-                  content.as_ptr() as *mut c_void,
-                  content.len(),
-                );
+                let data = match sent_response.into_body() {
+                  Cow::Owned(content) => NSData::from_vec(content),
+                  Cow::Borrowed(content) => NSData::with_bytes(content),
+                };
 
                 // Check validity again
                 check_webview_id_valid(webview_id)?;