Merge pull request RustPython#506 from janczer/add_check_byte_value

Add checking if value more then 255 in bytearraya

Author: windelbouwman

tests/snippets/bytearray.py

@@ -34,6 +34,13 @@
 a.clear()
 assert len(a) == 0
 
+try:
+    bytearray([400])
+except ValueError:
+      pass
+else:
+    assert False
+
 b = bytearray(b'test')
 assert len(b) == 4
 b.pop()

vm/src/obj/objbytearray.rs

@@ -121,7 +121,11 @@ fn bytearray_new(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
         let mut data_bytes = vec![];
         for elem in elements.iter() {
             let v = objint::to_int(vm, elem, 10)?;
-            data_bytes.push(v.to_u8().unwrap());
+            if let Some(i) = v.to_u8() {
+                data_bytes.push(i);
+            } else {
+                return Err(vm.new_value_error("byte must be in range(0, 256)".to_string()));
+            }
         }
         data_bytes
     // return Err(vm.new_type_error("Cannot construct bytes".to_string()));