Handle set_flags error

solana-labs · Jun 2, 2018 · 8ac9293 · 8ac9293
1 parent 1b400df
commit 8ac9293
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/erasure.rs b/src/erasure.rs
@@ -12,6 +12,7 @@ const NUM_DATA: usize = NUM_CODED - MAX_MISSING;
 pub enum ErasureError {
     NotEnoughBlocksToDecode,
     DecodeError,
+    EncodeError,
     InvalidBlockSize,
 }
 
@@ -229,7 +230,9 @@ pub fn generate_coding(window: &mut Vec<Option<SharedBlob>>, consumed: usize) ->
         let w_l = window[n].clone().unwrap();
         w_l.write().unwrap().meta.size = max_data_size;
         let flags = w_l.write().unwrap().get_flags().unwrap();
-        w_l.write().unwrap().set_flags(flags | BLOB_FLAG_IS_CODING);
+        if w_l.write().unwrap().set_flags(flags | BLOB_FLAG_IS_CODING).is_err() {
+            return Err(ErasureError::EncodeError);
+        }
         coding_blobs.push(
             window[n]
                 .clone()