fix: replace other String usage by bavshin-f5\'s impl

Author: JyJyJcr

nginx-sys/src/lib.rs

@@ -116,27 +116,6 @@ impl ngx_str_t {
         bytes_to_uchar(pool, src).map(|data| Self { data, len: src.len() })
     }
 
-    /// Create an `ngx_str_t` instance from a `String`.
-    ///
-    /// # Arguments
-    ///
-    /// * `pool` - A pointer to the nginx memory pool (`ngx_pool_t`).
-    /// * `data` - The `String` from which to create the nginx string.
-    ///
-    /// # Safety
-    /// This function is marked as unsafe because it accepts a raw pointer argument. There is no
-    /// way to know if `pool` is pointing to valid memory. The caller must provide a valid pool to
-    /// avoid indeterminate behavior.
-    ///
-    /// # Returns
-    /// An `ngx_str_t` instance representing the given `String`.
-    pub unsafe fn from_string(pool: *mut ngx_pool_t, data: String) -> Self {
-        ngx_str_t {
-            data: str_to_uchar(pool, data.as_str()),
-            len: data.len(),
-        }
-    }
-
     /// Create an `ngx_str_t` instance from a string slice (`&str`).
     ///
     /// # Arguments
@@ -221,18 +200,46 @@ impl TryFrom<ngx_str_t> for &str {
 pub unsafe fn add_to_ngx_table(
     table: *mut ngx_table_elt_t,
     pool: *mut ngx_pool_t,
-    key: &str,
-    value: &str,
+    key: impl AsRef<[u8]>,
+    value: impl AsRef<[u8]>,
 ) -> Option<()> {
-    if table.is_null() {
-        return None;
+    if let Some(table) = table.as_mut() {
+        let key = key.as_ref();
+        table.key = ngx_str_t::from_bytes(pool, key)?;
+        table.value = ngx_str_t::from_bytes(pool, value.as_ref())?;
+        table.lowcase_key = ngx_pnalloc(pool, table.key.len).cast();
+        if table.lowcase_key.is_null() {
+            return None;
+        }
+        table.hash = ngx_hash_strlow(table.lowcase_key, table.key.data, table.key.len);
+        return Some(());
+    }
+    None
+}
+
+#[cfg(test)]
+mod tests {
+    use std::string::ToString;
+
+    use super::*;
+
+    #[test]
+    fn ngx_str_display() {
+        let pairs: &[(&[u8], &str)] = &[
+            (b"", ""),
+            (b"Ferris the \xf0\x9f\xa6\x80", "Ferris the 🦀"),
+            (b"\xF0\x90\x80", "\\xf0\\x90\\x80"),
+            (b"\xF0\x90\x80Hello World", "\\xf0\\x90\\x80Hello World"),
+            (b"Hello \xF0\x90\x80World", "Hello \\xf0\\x90\\x80World"),
+            (b"Hello World\xF0\x90\x80", "Hello World\\xf0\\x90\\x80"),
+        ];
+
+        for (bytes, expected) in pairs {
+            let str = ngx_str_t {
+                data: bytes.as_ptr().cast_mut(),
+                len: bytes.len(),
+            };
+            assert_eq!(str.to_string(), *expected);
+        }
     }
-    table.as_mut().map(|table| {
-        table.hash = 1;
-        table.key.len = key.len();
-        table.key.data = str_to_uchar(pool, key);
-        table.value.len = value.len();
-        table.value.data = str_to_uchar(pool, value);
-        table.lowcase_key = str_to_uchar(pool, String::from(key).to_ascii_lowercase().as_str());
-    })
 }

src/http/status.rs

@@ -25,7 +25,7 @@ impl InvalidHTTPStatusCode {
 
 impl fmt::Display for InvalidHTTPStatusCode {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
-        f.write_str("invalid status code".to_string().as_str())
+        f.write_str("invalid status code")
     }
 }