From 04e75d6ae5a0b41142652994cbc8404b714b5fa1 Mon Sep 17 00:00:00 2001
From: wqfish <wqfish@calibra.com>
Date: Mon, 13 Jan 2020 03:33:33 -0800
Subject: [PATCH] Fix potential segfault when calling next on DBIterator that
 is at the end of the range (#374)

---
 src/db.rs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/db.rs b/src/db.rs
index 1500ec2..eb4cb6f 100644
--- a/src/db.rs
+++ b/src/db.rs
@@ -595,6 +595,10 @@ impl<'a> Iterator for DBIterator<'a> {
     type Item = KVBytes;
 
     fn next(&mut self) -> Option<KVBytes> {
+        if !self.raw.valid() {
+            return None;
+        }
+
         // Initial call to next() after seeking should not move the iterator
         // or the first item will not be returned
         if !self.just_seeked {
@@ -2169,6 +2173,21 @@ fn iterator_test() {
     assert!(DB::destroy(&opts, path).is_ok());
 }
 
+#[test]
+fn iterator_test_past_end() {
+    let path = "_rust_rocksdb_iteratortest_past_end";
+    {
+        let db = DB::open_default(path).unwrap();
+        db.put(b"k1", b"v1111").unwrap();
+        let mut iter = db.iterator(IteratorMode::Start);
+        assert!(iter.next().is_some());
+        assert!(iter.next().is_none());
+        assert!(iter.next().is_none());
+    }
+    let opts = Options::default();
+    DB::destroy(&opts, path).unwrap();
+}
+
 #[test]
 fn iterator_test_tailing() {
     let path = "_rust_rocksdb_iteratortest_tailing";