diff options
| author | Mason Reed <mason@vector35.com> | 2025-05-04 19:10:56 -0400 |
|---|---|---|
| committer | Mason Reed <35282038+emesare@users.noreply.github.com> | 2025-05-12 17:45:24 -0400 |
| commit | a826c589dfc10c542deba7ca3343a462e02d6bde (patch) | |
| tree | f116254bef39f787268bbecc5eac19da310db9ce /rust/src/websocket/client.rs | |
| parent | 28b3c4044af06fdc32c9c85bf8381b5058306427 (diff) | |
[Rust] Simplify `BnStrCompatible` trait
Followup to https://github.com/Vector35/binaryninja-api/pull/5897/
This simplifies usage of the trait in user code, should just be able to `to_cstr` to get the cstr repr and then call `as_ptr`.
Co-authored-by: Michael Krasnitski <michael.krasnitski@gmail.com>
Diffstat (limited to 'rust/src/websocket/client.rs')
| -rw-r--r-- | rust/src/websocket/client.rs | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/rust/src/websocket/client.rs b/rust/src/websocket/client.rs index 36c7bedd..43e1e1f7 100644 --- a/rust/src/websocket/client.rs +++ b/rust/src/websocket/client.rs @@ -1,5 +1,5 @@ use crate::rc::{Ref, RefCountable}; -use crate::string::{BnStrCompatible, BnString}; +use crate::string::{AsCStr, BnString}; use binaryninjacore_sys::*; use std::ffi::{c_char, c_void, CStr}; use std::ptr::NonNull; @@ -21,8 +21,8 @@ pub trait WebsocketClient: Sync + Send { fn connect<I, K, V>(&self, host: &str, headers: I) -> bool where I: IntoIterator<Item = (K, V)>, - K: BnStrCompatible, - V: BnStrCompatible; + K: AsCStr, + V: AsCStr; fn write(&self, data: &[u8]) -> bool; @@ -77,14 +77,14 @@ impl CoreWebsocketClient { ) -> bool where I: IntoIterator<Item = (K, V)>, - K: BnStrCompatible, - V: BnStrCompatible, + K: AsCStr, + V: AsCStr, C: WebsocketClientCallback, { - let url = host.into_bytes_with_nul(); + let url = host.to_cstr(); let (header_keys, header_values): (Vec<K::Result>, Vec<V::Result>) = headers .into_iter() - .map(|(k, v)| (k.into_bytes_with_nul(), v.into_bytes_with_nul())) + .map(|(k, v)| (k.to_cstr(), v.to_cstr())) .unzip(); let header_keys: Vec<*const c_char> = header_keys .iter() @@ -129,7 +129,7 @@ impl CoreWebsocketClient { /// Call the error callback function pub fn notify_error(&self, msg: &str) { - let error = msg.into_bytes_with_nul(); + let error = msg.to_cstr(); unsafe { BNNotifyWebsocketClientError(self.handle.as_ptr(), error.as_ptr() as *const c_char) } |
