]> git.bitcoin.ninja Git - rust-lightning/blob - lightning/src/util/byte_utils.rs
Merge pull request #739 from rloomba/rloomba/add-clippy-to-travis
[rust-lightning] / lightning / src / util / byte_utils.rs
1 // This file is Copyright its original authors, visible in version control
2 // history.
3 //
4 // This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
5 // or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
6 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
7 // You may not use this file except in accordance with one or both of these
8 // licenses.
9
10 #[inline]
11 pub fn slice_to_be16(v: &[u8]) -> u16 {
12         ((v[0] as u16) << 8*1) |
13         ((v[1] as u16) << 8*0)
14 }
15 #[inline]
16 pub fn slice_to_be32(v: &[u8]) -> u32 {
17         ((v[0] as u32) << 8*3) |
18         ((v[1] as u32) << 8*2) |
19         ((v[2] as u32) << 8*1) |
20         ((v[3] as u32) << 8*0)
21 }
22 #[cfg(not(feature = "fuzztarget"))] // Used only by poly1305
23 #[inline]
24 pub fn slice_to_le32(v: &[u8]) -> u32 {
25         ((v[0] as u32) << 8*0) |
26         ((v[1] as u32) << 8*1) |
27         ((v[2] as u32) << 8*2) |
28         ((v[3] as u32) << 8*3)
29 }
30 #[inline]
31 pub fn slice_to_be48(v: &[u8]) -> u64 {
32         ((v[0] as u64) << 8*5) |
33         ((v[1] as u64) << 8*4) |
34         ((v[2] as u64) << 8*3) |
35         ((v[3] as u64) << 8*2) |
36         ((v[4] as u64) << 8*1) |
37         ((v[5] as u64) << 8*0)
38 }
39 #[inline]
40 pub fn slice_to_be64(v: &[u8]) -> u64 {
41         ((v[0] as u64) << 8*7) |
42         ((v[1] as u64) << 8*6) |
43         ((v[2] as u64) << 8*5) |
44         ((v[3] as u64) << 8*4) |
45         ((v[4] as u64) << 8*3) |
46         ((v[5] as u64) << 8*2) |
47         ((v[6] as u64) << 8*1) |
48         ((v[7] as u64) << 8*0)
49 }
50
51 #[inline]
52 pub fn be16_to_array(u: u16) -> [u8; 2] {
53         let mut v = [0; 2];
54         v[0] = ((u >> 8*1) & 0xff) as u8;
55         v[1] = ((u >> 8*0) & 0xff) as u8;
56         v
57 }
58 #[inline]
59 pub fn be32_to_array(u: u32) -> [u8; 4] {
60         let mut v = [0; 4];
61         v[0] = ((u >> 8*3) & 0xff) as u8;
62         v[1] = ((u >> 8*2) & 0xff) as u8;
63         v[2] = ((u >> 8*1) & 0xff) as u8;
64         v[3] = ((u >> 8*0) & 0xff) as u8;
65         v
66 }
67 #[cfg(not(feature = "fuzztarget"))] // Used only by poly1305
68 #[inline]
69 pub fn le32_to_array(u: u32) -> [u8; 4] {
70         let mut v = [0; 4];
71         v[0] = ((u >> 8*0) & 0xff) as u8;
72         v[1] = ((u >> 8*1) & 0xff) as u8;
73         v[2] = ((u >> 8*2) & 0xff) as u8;
74         v[3] = ((u >> 8*3) & 0xff) as u8;
75         v
76 }
77 #[inline]
78 pub fn be48_to_array(u: u64) -> [u8; 6] {
79         assert!(u & 0xffff_0000_0000_0000 == 0);
80         let mut v = [0; 6];
81         v[0] = ((u >> 8*5) & 0xff) as u8;
82         v[1] = ((u >> 8*4) & 0xff) as u8;
83         v[2] = ((u >> 8*3) & 0xff) as u8;
84         v[3] = ((u >> 8*2) & 0xff) as u8;
85         v[4] = ((u >> 8*1) & 0xff) as u8;
86         v[5] = ((u >> 8*0) & 0xff) as u8;
87         v
88 }
89 #[inline]
90 pub fn be64_to_array(u: u64) -> [u8; 8] {
91         let mut v = [0; 8];
92         v[0] = ((u >> 8*7) & 0xff) as u8;
93         v[1] = ((u >> 8*6) & 0xff) as u8;
94         v[2] = ((u >> 8*5) & 0xff) as u8;
95         v[3] = ((u >> 8*4) & 0xff) as u8;
96         v[4] = ((u >> 8*3) & 0xff) as u8;
97         v[5] = ((u >> 8*2) & 0xff) as u8;
98         v[6] = ((u >> 8*1) & 0xff) as u8;
99         v[7] = ((u >> 8*0) & 0xff) as u8;
100         v
101 }
102
103 #[inline]
104 pub fn le64_to_array(u: u64) -> [u8; 8] {
105         let mut v = [0; 8];
106         v[0] = ((u >> 8*0) & 0xff) as u8;
107         v[1] = ((u >> 8*1) & 0xff) as u8;
108         v[2] = ((u >> 8*2) & 0xff) as u8;
109         v[3] = ((u >> 8*3) & 0xff) as u8;
110         v[4] = ((u >> 8*4) & 0xff) as u8;
111         v[5] = ((u >> 8*5) & 0xff) as u8;
112         v[6] = ((u >> 8*6) & 0xff) as u8;
113         v[7] = ((u >> 8*7) & 0xff) as u8;
114         v
115 }
116
117 #[cfg(test)]
118 mod tests {
119         use super::*;
120         
121         #[test]
122         fn test_all() {
123                 assert_eq!(slice_to_be16(&[0xde, 0xad]), 0xdead);
124                 assert_eq!(slice_to_be32(&[0xde, 0xad, 0xbe, 0xef]), 0xdeadbeef);
125                 assert_eq!(slice_to_le32(&[0xef, 0xbe, 0xad, 0xde]), 0xdeadbeef);
126                 assert_eq!(slice_to_be48(&[0xde, 0xad, 0xbe, 0xef, 0x1b, 0xad]), 0xdeadbeef1bad);
127                 assert_eq!(slice_to_be64(&[0xde, 0xad, 0xbe, 0xef, 0x1b, 0xad, 0x1d, 0xea]), 0xdeadbeef1bad1dea);
128                 assert_eq!(be16_to_array(0xdead), [0xde, 0xad]);
129                 assert_eq!(be32_to_array(0xdeadbeef), [0xde, 0xad, 0xbe, 0xef]);
130                 assert_eq!(le32_to_array(0xdeadbeef), [0xef, 0xbe, 0xad, 0xde]);
131                 assert_eq!(be48_to_array(0xdeadbeef1bad), [0xde, 0xad, 0xbe, 0xef, 0x1b, 0xad]);
132                 assert_eq!(be64_to_array(0xdeadbeef1bad1dea), [0xde, 0xad, 0xbe, 0xef, 0x1b, 0xad, 0x1d, 0xea]);
133                 assert_eq!(le64_to_array(0xdeadbeef1bad1dea), [0xea, 0x1d, 0xad, 0x1b, 0xef, 0xbe, 0xad, 0xde]);
134         }
135 }