added a check for buffer overflow

This commit is contained in:
Nikki 2015-12-29 17:08:46 -07:00
parent 20edf21352
commit 773a15d28a

View file

@ -299,7 +299,11 @@ impl WebSocket {
let chan = global.r().networking_task_source(); let chan = global.r().networking_task_source();
let address = Trusted::new(self, chan.clone()); 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 { if return_after_buffer {
return Ok(false); return Ok(false);