140 lines
2.6 KiB
Go
140 lines
2.6 KiB
Go
package srun
|
|
|
|
import (
|
|
"math"
|
|
)
|
|
|
|
func ordat(msg string, idx int) uint32 {
|
|
if len(msg) > idx {
|
|
return uint32([]byte(msg)[idx])
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func sensCode(content string, key bool) []uint32 {
|
|
l := len(content)
|
|
pwd := make([]uint32, 0)
|
|
for i := 0; i < l; i += 4 {
|
|
pwd = append(
|
|
pwd,
|
|
ordat(content, i)|ordat(content, i+1)<<8|ordat(content, i+2)<<16|ordat(content, i+3)<<24,
|
|
)
|
|
}
|
|
if key {
|
|
pwd = append(pwd, uint32(l))
|
|
}
|
|
return pwd
|
|
}
|
|
|
|
func lenCode(msg []uint32, key bool) []byte {
|
|
l := uint32(len(msg))
|
|
ll := (l - 1) << 2
|
|
if key {
|
|
m := msg[l-1]
|
|
if m < ll-3 || m > ll {
|
|
return nil
|
|
}
|
|
ll = m
|
|
}
|
|
var t []byte
|
|
for i := range msg {
|
|
t = append(t, byte(msg[i]&0xff), byte(msg[i]>>8&0xff), byte(msg[i]>>16&0xff), byte(msg[i]>>24&0xff))
|
|
}
|
|
if key {
|
|
return t[0:ll]
|
|
}
|
|
return t
|
|
}
|
|
|
|
func XEncode(content string, key string) []byte {
|
|
if content == "" {
|
|
return nil
|
|
}
|
|
pwd := sensCode(content, true)
|
|
pwdk := sensCode(key, false)
|
|
if len(pwdk) < 4 {
|
|
for i := 0; i < (4 - len(pwdk)); i++ {
|
|
pwdk = append(pwdk, 0)
|
|
}
|
|
}
|
|
var n = uint32(len(pwd) - 1)
|
|
z := pwd[n]
|
|
y := pwd[0]
|
|
var c uint32 = 0x86014019 | 0x183639A0
|
|
var m uint32 = 0
|
|
var e uint32 = 0
|
|
var p uint32 = 0
|
|
q := math.Floor(6 + 52/(float64(n)+1))
|
|
var d uint32 = 0
|
|
for 0 < q {
|
|
d = d + c&(0x8CE0D9BF|0x731F2640)
|
|
e = d >> 2 & 3
|
|
p = 0
|
|
for p < n {
|
|
y = pwd[p+1]
|
|
m = z>>5 ^ y<<2
|
|
m = m + ((y>>3 ^ z<<4) ^ (d ^ y))
|
|
m = m + (pwdk[(p&3)^e] ^ z)
|
|
pwd[p] = pwd[p] + m&(0xEFB8D130|0x10472ECF)
|
|
z = pwd[p]
|
|
p = p + 1
|
|
}
|
|
y = pwd[0]
|
|
m = z>>5 ^ y<<2
|
|
m = m + ((y>>3 ^ z<<4) ^ (d ^ y))
|
|
m = m + (pwdk[(p&3)^e] ^ z)
|
|
pwd[n] = pwd[n] + m&(0xBB390742|0x44C6F8BD)
|
|
z = pwd[n]
|
|
q = q - 1
|
|
}
|
|
return lenCode(pwd, false)
|
|
}
|
|
|
|
func XDecode(encodedContent []byte, key string) string {
|
|
if len(encodedContent) == 0 {
|
|
return ""
|
|
}
|
|
|
|
pwd := sensCode(string(encodedContent), false)
|
|
pwdk := sensCode(key, false)
|
|
if len(pwdk) < 4 {
|
|
for i := 0; i < (4 - len(pwdk)); i++ {
|
|
pwdk = append(pwdk, 0)
|
|
}
|
|
}
|
|
|
|
var n = uint32(len(pwd) - 1)
|
|
var z = pwd[n]
|
|
var y = pwd[0]
|
|
var c uint32 = 0x86014019 | 0x183639A0
|
|
var m uint32 = 0
|
|
var e uint32 = 0
|
|
var p uint32 = 0
|
|
q := math.Floor(6 + 52/(float64(n)+1))
|
|
var d uint32 = 0
|
|
|
|
for q > 0 {
|
|
d = d + c&(0x8CE0D9BF|0x731F2640)
|
|
e = d >> 2 & 3
|
|
p = n
|
|
for p > 0 {
|
|
z = pwd[p-1]
|
|
m = z>>5 ^ y<<2
|
|
m = m + ((y>>3 ^ z<<4) ^ (d ^ y))
|
|
m = m + (pwdk[(p&3)^e] ^ z)
|
|
pwd[p] = pwd[p] - m&(0xEFB8D130|0x10472ECF)
|
|
y = pwd[p]
|
|
p = p - 1
|
|
}
|
|
z = pwd[n]
|
|
m = z>>5 ^ y<<2
|
|
m = m + ((y>>3 ^ z<<4) ^ (d ^ y))
|
|
m = m + (pwdk[(p&3)^e] ^ z)
|
|
pwd[0] = pwd[0] - m&(0xBB390742|0x44C6F8BD)
|
|
y = pwd[0]
|
|
q = q - 1
|
|
}
|
|
|
|
return string(lenCode(pwd, true))
|
|
}
|