From fce9ce75507d9ac32363ffb685caaee027f0bc02 Mon Sep 17 00:00:00 2001
From: Aviv Palivoda <palaviv@gmail.com>
Date: Tue, 14 Apr 2020 12:56:16 +0300
Subject: [PATCH] Fix PyByteArray locking issue

---
 vm/src/obj/objbytearray.rs | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/vm/src/obj/objbytearray.rs b/vm/src/obj/objbytearray.rs
index 4cf49e746b..1e87a92a35 100644
--- a/vm/src/obj/objbytearray.rs
+++ b/vm/src/obj/objbytearray.rs
@@ -486,13 +486,11 @@ impl PyByteArray {
 
     #[pymethod(name = "extend")]
     fn extend(&self, iterable_of_ints: PyIterable, vm: &VirtualMachine) -> PyResult<()> {
-        let mut inner = self.borrow_value_mut();
-
         for x in iterable_of_ints.iter(vm)? {
             let x = x?;
             let x = PyIntRef::try_from_object(vm, x)?;
             let x = x.as_bigint().byte_or(vm)?;
-            inner.elements.push(x);
+            self.borrow_value_mut().elements.push(x);
         }
 
         Ok(())
@@ -526,8 +524,8 @@ impl PyByteArray {
 
     #[pymethod(name = "pop")]
     fn pop(&self, vm: &VirtualMachine) -> PyResult<u8> {
-        let bytes = &mut self.borrow_value_mut().elements;
-        bytes
+        self.borrow_value_mut()
+            .elements
             .pop()
             .ok_or_else(|| vm.new_index_error("pop from empty bytearray".to_owned()))
     }
@@ -623,9 +621,8 @@ impl PyValue for PyByteArrayIterator {
 impl PyByteArrayIterator {
     #[pymethod(name = "__next__")]
     fn next(&self, vm: &VirtualMachine) -> PyResult<u8> {
-        let bytearr = self.bytearray.borrow_value();
         let pos = self.position.fetch_add(1);
-        if let Some(&ret) = bytearr.elements.get(pos) {
+        if let Some(&ret) = self.bytearray.borrow_value().elements.get(pos) {
             Ok(ret)
         } else {
             Err(objiter::new_stop_iteration(vm))