Skip to content

Commit 58985e1

Browse files
committed
auto merge of #12106 : qpliu/rust/memreader-fill-eof, r=sfackler
I don't know if anything depends on MemReader::fill returning an empty slice instead of EndOfFile, but I'm pretty sure that MemReader::read_until should not go into an infinite loop.
2 parents b66ec34 + 71c88e7 commit 58985e1

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

src/libstd/io/mem.rs

+18-2
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,11 @@ impl Seek for MemReader {
177177

178178
impl Buffer for MemReader {
179179
fn fill<'a>(&'a mut self) -> IoResult<&'a [u8]> {
180-
Ok(self.buf.slice_from(self.pos))
180+
if self.pos < self.buf.len() {
181+
Ok(self.buf.slice_from(self.pos))
182+
} else {
183+
Err(io::standard_error(io::EndOfFile))
184+
}
181185
}
182186
fn consume(&mut self, amt: uint) { self.pos += amt; }
183187
}
@@ -308,7 +312,11 @@ impl<'a> Seek for BufReader<'a> {
308312

309313
impl<'a> Buffer for BufReader<'a> {
310314
fn fill<'a>(&'a mut self) -> IoResult<&'a [u8]> {
311-
Ok(self.buf.slice_from(self.pos))
315+
if self.pos < self.buf.len() {
316+
Ok(self.buf.slice_from(self.pos))
317+
} else {
318+
Err(io::standard_error(io::EndOfFile))
319+
}
312320
}
313321
fn consume(&mut self, amt: uint) { self.pos += amt; }
314322
}
@@ -421,6 +429,10 @@ mod test {
421429
assert_eq!(reader.read(buf), Ok(3));
422430
assert_eq!(buf.slice(0, 3), [5, 6, 7]);
423431
assert!(reader.read(buf).is_err());
432+
let mut reader = MemReader::new(~[0, 1, 2, 3, 4, 5, 6, 7]);
433+
assert_eq!(reader.read_until(3).unwrap(), ~[0, 1, 2, 3]);
434+
assert_eq!(reader.read_until(3).unwrap(), ~[4, 5, 6, 7]);
435+
assert!(reader.read(buf).is_err());
424436
}
425437

426438
#[test]
@@ -441,6 +453,10 @@ mod test {
441453
assert_eq!(reader.read(buf), Ok(3));
442454
assert_eq!(buf.slice(0, 3), [5, 6, 7]);
443455
assert!(reader.read(buf).is_err());
456+
let mut reader = BufReader::new(in_buf);
457+
assert_eq!(reader.read_until(3).unwrap(), ~[0, 1, 2, 3]);
458+
assert_eq!(reader.read_until(3).unwrap(), ~[4, 5, 6, 7]);
459+
assert!(reader.read(buf).is_err());
444460
}
445461

446462
#[test]

0 commit comments

Comments
 (0)