added a check for buffer overflow

servo · Dec 30, 2015 · 773a15d · 773a15d
1 parent 20edf21
commit 773a15d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/components/script/dom/websocket.rs b/components/script/dom/websocket.rs
@@ -299,7 +299,11 @@ impl WebSocket {
         let chan = global.r().networking_task_source();
         let address = Trusted::new(self, chan.clone());
 
-        self.buffered_amount.set(self.buffered_amount.get() + data_byte_len);
+        match data_byte_len.checked_add(self.buffered_amount.get()) {
+            None => return Ok(false),
+            Some(new_amount) => self.buffered_amount.set(new_amount)
+        };
+        // self.buffered_amount.set(self.buffered_amount.get() + data_byte_len);
 
         if return_after_buffer {
             return Ok(false);