Opencoded byte-swapping could just use the kernel helper. Signed-off-by: Harvey Harrison <harvey.harrison@xxxxxxxxx> --- net/mac80211/tkip.c | 47 ++++++++++++++++++++++------------------------- 1 files changed, 22 insertions(+), 25 deletions(-) diff --git a/net/mac80211/tkip.c b/net/mac80211/tkip.c index 90c11e7..9a36e83 100644 --- a/net/mac80211/tkip.c +++ b/net/mac80211/tkip.c @@ -60,13 +60,6 @@ static const u16 tkip_sbox[256] = 0x82C3, 0x29B0, 0x5A77, 0x1E11, 0x7BCB, 0xA8FC, 0x6DD6, 0x2C3A, }; - -static inline u16 Mk16(u8 x, u8 y) -{ - return ((u16) x << 8) | (u16) y; -} - - static inline u8 Hi8(u16 v) { return v >> 8; @@ -107,20 +100,22 @@ static void tkip_mixing_phase1(const u8 *ta, const u8 *tk, u32 tsc_IV32, u16 *p1k) { int i, j; + __le16 *leptr = (__le16 *)ta; p1k[0] = Lo16(tsc_IV32); p1k[1] = Hi16(tsc_IV32); - p1k[2] = Mk16(ta[1], ta[0]); - p1k[3] = Mk16(ta[3], ta[2]); - p1k[4] = Mk16(ta[5], ta[4]); + p1k[2] = le16_to_cpup(leptr++); + p1k[3] = le16_to_cpup(leptr++); + p1k[4] = le16_to_cpup(leptr++); + leptr = (__le16 *)tk; for (i = 0; i < PHASE1_LOOP_COUNT; i++) { - j = 2 * (i & 1); - p1k[0] += tkip_S(p1k[4] ^ Mk16(tk[ 1 + j], tk[ 0 + j])); - p1k[1] += tkip_S(p1k[0] ^ Mk16(tk[ 5 + j], tk[ 4 + j])); - p1k[2] += tkip_S(p1k[1] ^ Mk16(tk[ 9 + j], tk[ 8 + j])); - p1k[3] += tkip_S(p1k[2] ^ Mk16(tk[13 + j], tk[12 + j])); - p1k[4] += tkip_S(p1k[3] ^ Mk16(tk[ 1 + j], tk[ 0 + j])) + i; + j = (i & 1); + p1k[0] += tkip_S(p1k[4] ^ le16_to_cpup(leptr + 0 + j)); + p1k[1] += tkip_S(p1k[0] ^ le16_to_cpup(leptr + 2 + j)); + p1k[2] += tkip_S(p1k[1] ^ le16_to_cpup(leptr + 4 + j)); + p1k[3] += tkip_S(p1k[2] ^ le16_to_cpup(leptr + 6 + j)); + p1k[4] += tkip_S(p1k[3] ^ le16_to_cpup(leptr + 0 + j)) + i; } } @@ -130,6 +125,7 @@ static void tkip_mixing_phase2(const u16 *p1k, const u8 *tk, u16 tsc_IV16, { u16 ppk[6]; int i; + __le16 *leptr = (__le16 *)tk; ppk[0] = p1k[0]; ppk[1] = p1k[1]; @@ -138,23 +134,24 @@ static void tkip_mixing_phase2(const u16 *p1k, const u8 *tk, u16 tsc_IV16, ppk[4] = p1k[4]; ppk[5] = p1k[4] + tsc_IV16; - ppk[0] += tkip_S(ppk[5] ^ Mk16(tk[ 1], tk[ 0])); - ppk[1] += tkip_S(ppk[0] ^ Mk16(tk[ 3], tk[ 2])); - ppk[2] += tkip_S(ppk[1] ^ Mk16(tk[ 5], tk[ 4])); - ppk[3] += tkip_S(ppk[2] ^ Mk16(tk[ 7], tk[ 6])); - ppk[4] += tkip_S(ppk[3] ^ Mk16(tk[ 9], tk[ 8])); - ppk[5] += tkip_S(ppk[4] ^ Mk16(tk[11], tk[10])); - ppk[0] += ror16(ppk[5] ^ Mk16(tk[13], tk[12]), 1); - ppk[1] += ror16(ppk[0] ^ Mk16(tk[15], tk[14]), 1); + ppk[0] += tkip_S(ppk[5] ^ le16_to_cpup(leptr++)); + ppk[1] += tkip_S(ppk[0] ^ le16_to_cpup(leptr++)); + ppk[2] += tkip_S(ppk[1] ^ le16_to_cpup(leptr++)); + ppk[3] += tkip_S(ppk[2] ^ le16_to_cpup(leptr++)); + ppk[4] += tkip_S(ppk[3] ^ le16_to_cpup(leptr++)); + ppk[5] += tkip_S(ppk[4] ^ le16_to_cpup(leptr++)); + ppk[0] += ror16(ppk[5] ^ le16_to_cpup(leptr++), 1); + ppk[1] += ror16(ppk[0] ^ le16_to_cpup(leptr++), 1); ppk[2] += ror16(ppk[1], 1); ppk[3] += ror16(ppk[2], 1); ppk[4] += ror16(ppk[3], 1); ppk[5] += ror16(ppk[4], 1); + leptr = (__le16 *)tk; rc4key[0] = Hi8(tsc_IV16); rc4key[1] = (Hi8(tsc_IV16) | 0x20) & 0x7f; rc4key[2] = Lo8(tsc_IV16); - rc4key[3] = Lo8((ppk[5] ^ Mk16(tk[1], tk[0])) >> 1); + rc4key[3] = Lo8((ppk[5] ^ le16_to_cpup(leptr)) >> 1); for (i = 0; i < 6; i++) { rc4key[4 + 2 * i] = Lo8(ppk[i]); -- 1.5.4.GIT -- To unsubscribe from this list: send the line "unsubscribe linux-wireless" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html