Skip to content

Commit

Permalink
Merge pull request #39008 from cuviper/beta-blake-endian
Browse files Browse the repository at this point in the history
Backport #38960 to beta
  • Loading branch information
alexcrichton committed Jan 12, 2017
2 parents a035041 + f3639ac commit dc4f0e8
Showing 1 changed file with 10 additions and 6 deletions.
16 changes: 10 additions & 6 deletions src/librustc_data_structures/blake2b.rs
Original file line number Diff line number Diff line change
Expand Up @@ -113,17 +113,20 @@ fn blake2b_compress(ctx: &mut Blake2bCtx, last: bool) {
}

{
// Re-interpret the input buffer in the state as u64s
// Re-interpret the input buffer in the state as an array
// of little-endian u64s, converting them to machine
// endianness. It's OK to modify the buffer in place
// since this is the last time this data will be accessed
// before it's overwritten.

let m: &mut [u64; 16] = unsafe {
let b: &mut [u8; 128] = &mut ctx.b;
::std::mem::transmute(b)
};

// It's OK to modify the buffer in place since this is the last time
// this data will be accessed before it's overwritten
if cfg!(target_endian = "big") {
for word in &mut m[..] {
*word = word.to_be();
*word = u64::from_le(*word);
}
}

Expand Down Expand Up @@ -209,9 +212,10 @@ fn blake2b_final(ctx: &mut Blake2bCtx)

blake2b_compress(ctx, true);

// Modify our buffer to little-endian format as it will be read
// as a byte array. It's OK to modify the buffer in place since
// this is the last time this data will be accessed.
if cfg!(target_endian = "big") {
// Make sure that the data is in memory in little endian format, as is
// demanded by BLAKE2
for word in &mut ctx.h {
*word = word.to_le();
}
Expand Down

0 comments on commit dc4f0e8

Please sign in to comment.