@@ -58,7 +58,8 @@ pub fn run(sector_size: usize) -> Result<(), failure::Error> {
58
58
UnpaddedBytesAmount :: from ( PaddedBytesAmount ( sector_size as u64 ) ) ;
59
59
60
60
// Create files for the staged and sealed sectors.
61
- let staged_file = NamedTempFile :: new ( ) . expect ( "could not create temp file for staged sector" ) ;
61
+ let mut staged_file =
62
+ NamedTempFile :: new ( ) . expect ( "could not create temp file for staged sector" ) ;
62
63
63
64
let sealed_file = NamedTempFile :: new ( ) . expect ( "could not create temp file for sealed sector" ) ;
64
65
@@ -70,7 +71,9 @@ pub fn run(sector_size: usize) -> Result<(), failure::Error> {
70
71
71
72
// Generate the data from which we will create a replica, we will then prove the continued
72
73
// storage of that replica using the PoSt.
73
- let piece_bytes: Vec < u8 > = ( 0 ..sector_size) . map ( |_| rand:: random :: < u8 > ( ) ) . collect ( ) ;
74
+ let piece_bytes: Vec < u8 > = ( 0 ..usize:: from ( sector_size_unpadded_bytes_ammount) )
75
+ . map ( |_| rand:: random :: < u8 > ( ) )
76
+ . collect ( ) ;
74
77
75
78
let mut piece_file = NamedTempFile :: new ( ) ?;
76
79
piece_file. write_all ( & piece_bytes) ?;
@@ -81,10 +84,9 @@ pub fn run(sector_size: usize) -> Result<(), failure::Error> {
81
84
generate_piece_commitment ( piece_file. as_file_mut ( ) , sector_size_unpadded_bytes_ammount) ?;
82
85
piece_file. as_file_mut ( ) . seek ( SeekFrom :: Start ( 0 ) ) ?;
83
86
84
- let mut staged_sector_file = NamedTempFile :: new ( ) ?;
85
87
add_piece (
86
88
& mut piece_file,
87
- & mut staged_sector_file ,
89
+ & mut staged_file ,
88
90
sector_size_unpadded_bytes_ammount,
89
91
& [ ] ,
90
92
) ?;
0 commit comments