183 lines
3.3 KiB
ArmAsm
183 lines
3.3 KiB
ArmAsm
|
// Copyright 2019 The Go Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
//go:build gc && !purego
|
||
|
|
// +build gc,!purego
|
||
|
|
|
||
|
|
#include "textflag.h"
|
||
|
|
|
||
|
|
// This was ported from the amd64 implementation.
|
||
|
|
|
||
|
|
#define POLY1305_ADD(msg, h0, h1, h2, t0, t1, t2) \
|
||
|
|
MOVD (msg), t0; \
|
||
|
|
MOVD 8(msg), t1; \
|
||
|
|
MOVD $1, t2; \
|
||
|
|
ADDC t0, h0, h0; \
|
||
|
|
ADDE t1, h1, h1; \
|
||
|
|
ADDE t2, h2; \
|
||
|
|
ADD $16, msg
|
||
|
|
|
||
|
|
#define POLY1305_MUL(h0, h1, h2, r0, r1, t0, t1, t2, t3, t4, t5) \
|
||
|
|
MULLD r0, h0, t0; \
|
||
|
|
MULLD r0, h1, t4; \
|
||
|
|
MULHDU r0, h0, t1; \
|
||
|
|
MULHDU r0, h1, t5; \
|
||
|
|
ADDC t4, t1, t1; \
|
||
|
|
MULLD r0, h2, t2; \
|
||
|
|
ADDZE t5; \
|
||
|
|
MULHDU r1, h0, t4; \
|
||
|
|
MULLD r1, h0, h0; \
|
||
|
|
ADD t5, t2, t2; \
|
||
|
|
ADDC h0, t1, t1; \
|
||
|
|
MULLD h2, r1, t3; \
|
||
|
|
ADDZE t4, h0; \
|
||
|
|
MULHDU r1, h1, t5; \
|
||
|
|
MULLD r1, h1, t4; \
|
||
|
|
ADDC t4, t2, t2; \
|
||
|
|
ADDE t5, t3, t3; \
|
||
|
|
ADDC h0, t2, t2; \
|
||
|
|
MOVD $-4, t4; \
|
||
|
|
MOVD t0, h0; \
|
||
|
|
MOVD t1, h1; \
|
||
|
|
ADDZE t3; \
|
||
|
|
ANDCC $3, t2, h2; \
|
||
|
|
AND t2, t4, t0; \
|
||
|
|
ADDC t0, h0, h0; \
|
||
|
|
ADDE t3, h1, h1; \
|
||
|
|
SLD $62, t3, t4; \
|
||
|
|
SRD $2, t2; \
|
||
|
|
ADDZE h2; \
|
||
|
|
OR t4, t2, t2; \
|
||
|
|
SRD $2, t3; \
|
||
|
|
ADDC t2, h0, h0; \
|
||
|
|
ADDE t3, h1, h1; \
|
||
|
|
ADDZE h2
|
||
|
|
|
||
|
|
DATA ·poly1305Mask<>+0x00(SB)/8, $0x0FFFFFFC0FFFFFFF
|
||
|
|
DATA ·poly1305Mask<>+0x08(SB)/8, $0x0FFFFFFC0FFFFFFC
|
||
|
|
GLOBL ·poly1305Mask<>(SB), RODATA, $16
|
||
|
|
|
||
|
|
// func update(state *[7]uint64, msg []byte)
|
||
|
|
TEXT ·update(SB), $0-32
|
||
|
|
MOVD state+0(FP), R3
|
||
|
|
MOVD msg_base+8(FP), R4
|
||
|
|
MOVD msg_len+16(FP), R5
|
||
|
|
|
||
|
|
MOVD 0(R3), R8 // h0
|
||
|
|
MOVD 8(R3), R9 // h1
|
||
|
|
MOVD 16(R3), R10 // h2
|
||
|
|
MOVD 24(R3), R11 // r0
|
||
|
|
MOVD 32(R3), R12 // r1
|
||
|
|
|
||
|
|
CMP R5, $16
|
||
|
|
BLT bytes_between_0_and_15
|
||
|
|
|
||
|
|
loop:
|
||
|
|
POLY1305_ADD(R4, R8, R9, R10, R20, R21, R22)
|
||
|
|
|
||
|
|
multiply:
|
||
|
|
POLY1305_MUL(R8, R9, R10, R11, R12, R16, R17, R18, R14, R20, R21)
|
||
|
|
ADD $-16, R5
|
||
|
|
CMP R5, $16
|
||
|
|
BGE loop
|
||
|
|
|
||
|
|
bytes_between_0_and_15:
|
||
|
|
CMP R5, $0
|
||
|
|
BEQ done
|
||
|
|
MOVD $0, R16 // h0
|
||
|
|
MOVD $0, R17 // h1
|
||
|
|
|
||
|
|
flush_buffer:
|
||
|
|
CMP R5, $8
|
||
|
|
BLE just1
|
||
|
|
|
||
|
|
MOVD $8, R21
|
||
|
|
SUB R21, R5, R21
|
||
|
|
|
||
|
|
// Greater than 8 -- load the rightmost remaining bytes in msg
|
||
|
|
// and put into R17 (h1)
|
||
|
|
MOVD (R4)(R21), R17
|
||
|
|
MOVD $16, R22
|
||
|
|
|
||
|
|
// Find the offset to those bytes
|
||
|
|
SUB R5, R22, R22
|
||
|
|
SLD $3, R22
|
||
|
|
|
||
|
|
// Shift to get only the bytes in msg
|
||
|
|
SRD R22, R17, R17
|
||
|
|
|
||
|
|
// Put 1 at high end
|
||
|
|
MOVD $1, R23
|
||
|
|
SLD $3, R21
|
||
|
|
SLD R21, R23, R23
|
||
|
|
OR R23, R17, R17
|
||
|
|
|
||
|
|
// Remainder is 8
|
||
|
|
MOVD $8, R5
|
||
|
|
|
||
|
|
just1:
|
||
|
|
CMP R5, $8
|
||
|
|
BLT less8
|
||
|
|
|
||
|
|
// Exactly 8
|
||
|
|
MOVD (R4), R16
|
||
|
|
|
||
|
|
CMP R17, $0
|
||
|
|
|
||
|
|
// Check if we've already set R17; if not
|
||
|
|
// set 1 to indicate end of msg.
|
||
|
|
BNE carry
|
||
|
|
MOVD $1, R17
|
||
|
|
BR carry
|
||
|
|
|
||
|
|
less8:
|
||
|
|
MOVD $0, R16 // h0
|
||
|
|
MOVD $0, R22 // shift count
|
||
|
|
CMP R5, $4
|
||
|
|
BLT less4
|
||
|
|
MOVWZ (R4), R16
|
||
|
|
ADD $4, R4
|
||
|
|
ADD $-4, R5
|
||
|
|
MOVD $32, R22
|
||
|
|
|
||
|
|
less4:
|
||
|
|
CMP R5, $2
|
||
|
|
BLT less2
|
||
|
|
MOVHZ (R4), R21
|
||
|
|
SLD R22, R21, R21
|
||
|
|
OR R16, R21, R16
|
||
|
|
ADD $16, R22
|
||
|
|
ADD $-2, R5
|
||
|
|
ADD $2, R4
|
||
|
|
|
||
|
|
less2:
|
||
|
|
CMP R5, $0
|
||
|
|
BEQ insert1
|
||
|
|
MOVBZ (R4), R21
|
||
|
|
SLD R22, R21, R21
|
||
|
|
OR R16, R21, R16
|
||
|
|
ADD $8, R22
|
||
|
|
|
||
|
|
insert1:
|
||
|
|
// Insert 1 at end of msg
|
||
|
|
MOVD $1, R21
|
||
|
|
SLD R22, R21, R21
|
||
|
|
OR R16, R21, R16
|
||
|
|
|
||
|
|
carry:
|
||
|
|
// Add new values to h0, h1, h2
|
||
|
|
ADDC R16, R8
|
||
|
|
ADDE R17, R9
|
||
|
|
ADDZE R10, R10
|
||
|
|
MOVD $16, R5
|
||
|
|
ADD R5, R4
|
||
|
|
BR multiply
|
||
|
|
|
||
|
|
done:
|
||
|
|
// Save h0, h1, h2 in state
|
||
|
|
MOVD R8, 0(R3)
|
||
|
|
MOVD R9, 8(R3)
|
||
|
|
MOVD R10, 16(R3)
|
||
|
|
RET
|