---

yaml
---
r: 83510
b: refs/heads/try
c: b1ee87f
h: refs/heads/master
v: v3

Author: erickt

[refs]

@@ -2,7 +2,7 @@
 refs/heads/master: 0e4d1fc8cae42e15e00f71d9f439b01bb25a86ae
 refs/heads/snap-stage1: e33de59e47c5076a89eadeb38f4934f58a3618a6
 refs/heads/snap-stage3: 6c08cc2db4f98e9f07ae7d50338396c4123c2f0a
-refs/heads/try: 2d878033fd77f13a41bb9142ba2a4e9b976d0089
+refs/heads/try: b1ee87f402f929689fc028010c450184f661db3b
 refs/tags/release-0.1: 1f5c5126e96c79d22cb7862f75304136e204f105
 refs/heads/ndm: f3868061cd7988080c30d6d5bf352a5a5fe2460b
 refs/heads/try2: 147ecfdd8221e4a4d4e090486829a06da1e0ca3c

branches/try/src/libstd/c_str.rs

@@ -71,6 +71,7 @@ use ptr;
 use str::StrSlice;
 use str;
 use vec::{CopyableVector, ImmutableVector, MutableVector};
+use vec;
 use unstable::intrinsics;
 
 /// Resolution options for the `null_byte` condition
@@ -283,41 +284,32 @@ impl<'self> ToCStr for &'self [u8] {
     }
 
     fn with_c_str<T>(&self, f: &fn(*libc::c_char) -> T) -> T {
-        if self.len() < BUF_LEN {
-            do self.as_imm_buf |self_buf, self_len| {
-                unsafe {
-                    let mut buf: [u8, .. BUF_LEN] = intrinsics::uninit();
-
-                    do buf.as_mut_buf |buf, _| {
-                        ptr::copy_memory(buf, self_buf, self_len);
-                        *ptr::mut_offset(buf, self_len as int) = 0;
-
-                        check_for_null(*self, buf as *mut libc::c_char);
-
-                        f(buf as *libc::c_char)
-                    }
-                }
-            }
-        } else {
-            self.to_c_str().with_ref(f)
-        }
+        unsafe { with_c_str(*self, true, f) }
     }
 
     unsafe fn with_c_str_unchecked<T>(&self, f: &fn(*libc::c_char) -> T) -> T {
-        if self.len() < BUF_LEN {
-            do self.as_imm_buf |self_buf, self_len| {
-                let mut buf: [u8, .. BUF_LEN] = intrinsics::uninit();
+        with_c_str(*self, false, f)
+    }
+}
 
-                do buf.as_mut_buf |buf, _| {
-                    ptr::copy_memory(buf, self_buf, self_len);
-                    *ptr::mut_offset(buf, self_len as int) = 0;
+// Unsafe function that handles possibly copying the &[u8] into a stack array.
+unsafe fn with_c_str<T>(v: &[u8], checked: bool, f: &fn(*libc::c_char) -> T) -> T {
+    if v.len() < BUF_LEN {
+        let mut buf: [u8, .. BUF_LEN] = intrinsics::uninit();
+        vec::bytes::copy_memory(buf, v, v.len());
+        buf[v.len()] = 0;
 
-                    f(buf as *libc::c_char)
-                }
+        do buf.as_mut_buf |buf, _| {
+            if checked {
+                check_for_null(v, buf as *mut libc::c_char);
             }
-        } else {
-            self.to_c_str().with_ref(f)
+
+            f(buf as *libc::c_char)
         }
+    } else if checked {
+        v.to_c_str().with_ref(f)
+    } else {
+        v.to_c_str_unchecked().with_ref(f)
     }
 }