vec: Fix size_hint() of reverse iterators

Fixes #7558

1 files changed, 16 insertions, 1 deletions

diff --git a/src/libstd/vec.rs b/src/libstd/vec.rs
index 02eadaed5e8..f12e00bfa0a 100644
--- a/src/libstd/vec.rs
+++ b/src/libstd/vec.rs
@@ -2125,7 +2125,12 @@ macro_rules! iterator {
 
             #[inline]
             fn size_hint(&self) -> (Option<uint>, Option<uint>) {
-                let exact = Some(((self.end as uint) - (self.ptr as uint)) / size_of::<$elem>());
+                let diff = if $step > 0 {
+                    (self.end as uint) - (self.ptr as uint)
+                } else {
+                    (self.ptr as uint) - (self.end as uint)
+                };
+                let exact = Some(diff / size_of::<$elem>());
                 (exact, exact)
             }
         }
@@ -3405,6 +3410,16 @@ mod tests {
     }
 
     #[test]
+    fn test_iter_size_hints() {
+        use iterator::*;
+        let mut xs = [1, 2, 5, 10, 11];
+        assert_eq!(xs.iter().size_hint(), (Some(5), Some(5)));
+        assert_eq!(xs.rev_iter().size_hint(), (Some(5), Some(5)));
+        assert_eq!(xs.mut_iter().size_hint(), (Some(5), Some(5)));
+        assert_eq!(xs.mut_rev_iter().size_hint(), (Some(5), Some(5)));
+    }
+
+    #[test]
     fn test_mut_iterator() {
         use iterator::*;
         let mut xs = [1, 2, 3, 4, 5];