fix #2

sagebind · Feb 2, 2018 · 9e9f1fb · 9e9f1fb
1 parent 298bf06
commit 9e9f1fb
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/buffer.rs b/src/buffer.rs
@@ -192,7 +192,9 @@ impl From<Buffer> for Vec<u8> {
         let len = buffer.copy_to(&mut slice);
 
         unsafe {
-            Vec::from_raw_parts(slice.as_mut_ptr(), len, slice.len())
+            let vec = Vec::from_raw_parts(slice.as_mut_ptr(), len, slice.len());
+            mem::forget(slice);
+            vec
         }
     }
 }
@@ -291,4 +293,17 @@ mod tests {
         buffer.copy_to(&mut out);
         assert!(&out == b"hello world");
     }
+
+    #[test]
+    fn vec_from_buffer() {
+        let mut buffer = Buffer::new();
+        let bytes = b"hello world";
+        buffer.push(bytes);
+
+        assert!(buffer.len() == bytes.len());
+
+        let vec = Vec::from(buffer);
+
+        assert!(&vec == bytes);
+    }
 }