From 1e21e5d13b21472e2d23ef08feb0ed4372295ef8 Mon Sep 17 00:00:00 2001
From: Andrew Cann <shum@canndrew.org>
Date: Thu, 10 Mar 2016 17:08:11 +0800
Subject: [PATCH] Respect size limits in read_seq and read_map.

---
 src/rustc_serialize/reader.rs | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/rustc_serialize/reader.rs b/src/rustc_serialize/reader.rs
index d3dbfb3..7b2221d 100644
--- a/src/rustc_serialize/reader.rs
+++ b/src/rustc_serialize/reader.rs
@@ -129,7 +129,10 @@ impl<'a, R: Read> DecoderReader<'a, R> {
 
 impl <'a, A> DecoderReader<'a, A> {
     fn read_bytes(&mut self, count: u64) -> Result<(), DecodingError> {
-        self.read += count;
+        self.read = match self.read.checked_add(count) {
+            Some(read) => read,
+            None => return Err(DecodingError::SizeLimit),
+        };
         match self.size_limit {
             SizeLimit::Infinite => Ok(()),
             SizeLimit::Bounded(x) if self.read <= x => Ok(()),
@@ -342,6 +345,11 @@ impl<'a, R: Read> Decoder for DecoderReader<'a, R> {
         where F: FnOnce(&mut DecoderReader<'a, R>, usize) -> DecodingResult<T>
     {
         let len = try!(self.read_usize());
+        match self.size_limit {
+            SizeLimit::Infinite => (),
+            SizeLimit::Bounded(x) if self.read.saturating_add(len as u64) <= x => (),
+            SizeLimit::Bounded(_) => return Err(DecodingError::SizeLimit),
+        };
         f(self, len)
     }
     fn read_seq_elt<T, F>(&mut self, _: usize, f: F) -> DecodingResult<T>
@@ -353,6 +361,11 @@ impl<'a, R: Read> Decoder for DecoderReader<'a, R> {
         where F: FnOnce(&mut DecoderReader<'a, R>, usize) -> DecodingResult<T>
     {
         let len = try!(self.read_usize());
+        match self.size_limit {
+            SizeLimit::Infinite => (),
+            SizeLimit::Bounded(x) if self.read.saturating_add(len as u64) <= x => (),
+            SizeLimit::Bounded(_) => return Err(DecodingError::SizeLimit),
+        };
         f(self, len)
     }
     fn read_map_elt_key<T, F>(&mut self, _: usize, f: F) -> DecodingResult<T>