1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

// This file is part of network. It is subject to the license terms in the COPYRIGHT file found in the top-level directory of this distribution and at https://raw.githubusercontent.com/lemonrock/network/master/COPYRIGHT. No part of network, including this file, may be copied, modified, propagated, or distributed except according to the terms contained in the COPYRIGHT file.
// Copyright © 2016-2018 The developers of network. See the COPYRIGHT file in the top-level directory of this distribution and at https://raw.githubusercontent.com/lemonrock/network/master/COPYRIGHT.


/// Modelled as a packed 2-byte array rather than an u16 because (a) it is not native endian and (b) its alignment is not necessary 2 bytes (it's actually 1).
#[derive(Default, Debug, Copy, Clone, Eq, PartialEq, Hash)]
#[derive(Deserialize, Serialize)]
#[repr(C, packed)]
pub struct NetworkEndianU16([u8; NetworkEndianU16::Length]);

impl PartialOrd for NetworkEndianU16
{
	#[inline(always)]
	fn partial_cmp(&self, other: &Self) -> Option<Ordering>
	{
		u16::from_be(u16::from_bytes(self.0)).partial_cmp(&u16::from_be(u16::from_bytes(other.0)))
	}
}

impl Ord for NetworkEndianU16
{
	#[inline(always)]
	fn cmp(&self, other: &Self) -> Ordering
	{
		u16::from_be(u16::from_bytes(self.0)).cmp(&u16::from_be(u16::from_bytes(other.0)))
	}
}

impl Into<[u8; NetworkEndianU16::Length]> for NetworkEndianU16
{
	#[inline(always)]
	fn into(self) -> [u8; NetworkEndianU16::Length]
	{
		self.to_bytes()
	}
}

impl From<[u8; NetworkEndianU16::Length]> for NetworkEndianU16
{
	#[inline(always)]
	fn from(bytes: [u8; NetworkEndianU16::Length]) -> Self
	{
		NetworkEndianU16(bytes)
	}
}

impl NetworkEndian for NetworkEndianU16
{
	const Length: usize = 2;
	
	type Bytes = [u8; 2];
	
	#[inline(always)]
	fn to_bytes(self) -> Self::Bytes
	{
		self.0
	}
	
	#[inline(always)]
	fn bytes(&self) -> &[u8]
	{
		&self.0[..]
	}
	
	#[inline(always)]
	fn write_to_hash<H: Hasher>(&self, hasher: &mut H)
	{
		hasher.write_u16(unsafe { transmute_copy(&self.0) })
	}
}

impl Display for NetworkEndianU16
{
	fn fmt(&self, f: &mut Formatter) -> fmt::Result
	{
		write!(f, "{}", self.to_native_endian())
	}
}

impl NetworkEndianU16
{
	/// Zero.
	pub const Zero: Self = NetworkEndianU16([0, 0]);
	
	/// Maximum.
	pub const Maximum: Self = NetworkEndianU16([0xFF, 0xFF]);
	
	/// From network endian.
	#[inline(always)]
	pub const fn from_network_endian(network_endian: [u8; 2]) -> Self
	{
		NetworkEndianU16(network_endian)
	}
	
	/// To network endian.
	#[inline(always)]
	pub fn to_network_endian(self) -> u16
	{
		self.big_endian_from_bytes()
	}
	
	/// To native endian.
	#[inline(always)]
	pub fn to_native_endian(self) -> u16
	{
		u16::from_be(self.big_endian_from_bytes())
	}
	
	/// From native endian.
	#[inline(always)]
	pub fn from_native_endian(native_endian: u16) -> Self
	{
		NetworkEndianU16(native_endian.to_be().to_bytes())
	}
	
	/// Is not zero.
	#[inline(always)]
	pub fn is_not_zero(self) -> bool
	{
		self != Self::Zero
	}
	
	#[inline(always)]
	fn big_endian_from_bytes(self) -> u16
	{
		u16::from_bytes(self.0)
	}
}