mirror of
https://github.com/vlang/v.git
synced 2023-08-10 21:13:21 +03:00
sha1 implementation + helper funcs
This commit is contained in:
parent
37aff9b107
commit
a7529b7b05
@ -225,3 +225,11 @@ fn free(a voidptr) {
|
||||
C.free(a)
|
||||
}
|
||||
|
||||
pub fn (b []byte) hex() string {
|
||||
mut hex := malloc(b.len*2+1)
|
||||
mut ptr := &hex[0]
|
||||
for i := 0; i < b.len ; i++ {
|
||||
ptr += C.sprintf(ptr, '%02X', b[i])
|
||||
}
|
||||
return string(hex)
|
||||
}
|
||||
|
@ -812,3 +812,11 @@ pub fn (s string) hash() int {
|
||||
return h
|
||||
}
|
||||
|
||||
pub fn (s string) bytes() []byte {
|
||||
if s.len == 0 {
|
||||
return []byte
|
||||
}
|
||||
mut buf := [byte(0); s.len]
|
||||
C.memcpy(buf.data, s.str, s.len)
|
||||
return buf
|
||||
}
|
||||
|
147
vlib/crypto/sha1/sha1.v
Normal file
147
vlib/crypto/sha1/sha1.v
Normal file
@ -0,0 +1,147 @@
|
||||
// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
|
||||
// Use of this source code is governed by an MIT license
|
||||
// that can be found in the LICENSE file.
|
||||
|
||||
// Package sha1 implements the SHA-1 hash algorithm as defined in RFC 3174.
|
||||
//
|
||||
// SHA-1 is cryptographically broken and should not be used for secure
|
||||
// applications.
|
||||
// Adapted from: https://github.com/golang/go/blob/master/src/crypto/sha1
|
||||
module sha1
|
||||
|
||||
import math
|
||||
import encoding.binary
|
||||
|
||||
|
||||
const(
|
||||
// The size of a SHA-1 checksum in bytes.
|
||||
Size = 20
|
||||
// The blocksize of SHA-1 in bytes.
|
||||
BlockSize = 64
|
||||
)
|
||||
|
||||
const (
|
||||
Chunk = 64
|
||||
Init0 = 0x67452301
|
||||
Init1 = 0xEFCDAB89
|
||||
Init2 = 0x98BADCFE
|
||||
Init3 = 0x10325476
|
||||
Init4 = 0xC3D2E1F0
|
||||
)
|
||||
|
||||
// digest represents the partial evaluation of a checksum.
|
||||
struct Digest {
|
||||
mut:
|
||||
h []u32
|
||||
x []byte
|
||||
nx int
|
||||
len u64
|
||||
}
|
||||
|
||||
fn (d mut Digest) reset() {
|
||||
d.x = [byte(0); Chunk]
|
||||
d.h = [u32(0); 5]
|
||||
d.h[0] = u32(Init0)
|
||||
d.h[1] = u32(Init1)
|
||||
d.h[2] = u32(Init2)
|
||||
d.h[3] = u32(Init3)
|
||||
d.h[4] = u32(Init4)
|
||||
d.nx = 0
|
||||
d.len = u64(0)
|
||||
}
|
||||
|
||||
// New returns a new Digest (implementing hash.Hash) computing the SHA1 checksum.
|
||||
pub fn new() &Digest {
|
||||
mut d := &Digest{}
|
||||
d.reset()
|
||||
return d
|
||||
}
|
||||
|
||||
pub fn (d mut Digest) write(p []byte) ?int {
|
||||
nn := p.len
|
||||
d.len += u64(nn)
|
||||
|
||||
if d.nx > 0 {
|
||||
n := int(math.min(f64(d.x.len), f64(p.len)))
|
||||
for i:=0; i<n; i++ {
|
||||
d.x.set(i+d.nx, p[i])
|
||||
}
|
||||
d.nx += n
|
||||
if d.nx == Chunk {
|
||||
block(d, d.x)
|
||||
d.nx = 0
|
||||
}
|
||||
if n >= p.len {
|
||||
p = []byte
|
||||
} else {
|
||||
p = p.right(n)
|
||||
}
|
||||
}
|
||||
if p.len >= Chunk {
|
||||
n := p.len &~ (Chunk - 1)
|
||||
block(d, p.left(n))
|
||||
if n >= p.len {
|
||||
p = []byte
|
||||
} else {
|
||||
p = p.right(n)
|
||||
}
|
||||
}
|
||||
if p.len > 0 {
|
||||
d.nx = int(math.min(f64(d.x.len), f64(p.len)))
|
||||
for i:=0; i<d.nx; i++ {
|
||||
d.x.set(i, p[i])
|
||||
}
|
||||
}
|
||||
return nn
|
||||
}
|
||||
|
||||
pub fn (d &Digest) sum(b_in mut []byte) []byte {
|
||||
// Make a copy of d so that caller can keep writing and summing.
|
||||
mut d0 := *d
|
||||
hash := d0.check_sum()
|
||||
for b in hash {
|
||||
b_in << b
|
||||
}
|
||||
return *b_in
|
||||
}
|
||||
|
||||
fn (d mut Digest) check_sum() []byte {
|
||||
mut len := d.len
|
||||
// Padding. Add a 1 bit and 0 bits until 56 bytes mod 64.
|
||||
mut tmp := [byte(0); 64]
|
||||
|
||||
tmp[0] = 0x80
|
||||
|
||||
if int(len)%64 < 56 {
|
||||
d.write(tmp.left(56-int(len)%64))
|
||||
} else {
|
||||
d.write(tmp.left(64+56-int(len)%64))
|
||||
}
|
||||
|
||||
// Length in bits.
|
||||
len <<= u64(3)
|
||||
binary.big_endian_put_u64(tmp, len)
|
||||
d.write(tmp.left(8))
|
||||
|
||||
mut digest := [byte(0); Size]
|
||||
|
||||
binary.big_endian_put_u32(digest, d.h[0])
|
||||
binary.big_endian_put_u32(digest.right(4), d.h[1])
|
||||
binary.big_endian_put_u32(digest.right(8), d.h[2])
|
||||
binary.big_endian_put_u32(digest.right(12), d.h[3])
|
||||
binary.big_endian_put_u32(digest.right(16), d.h[4])
|
||||
|
||||
return digest
|
||||
}
|
||||
|
||||
// Sum returns the SHA-1 checksum of the data.
|
||||
pub fn sum(data []byte) []byte {
|
||||
mut d := Digest{}
|
||||
d.reset()
|
||||
d.write(data)
|
||||
return d.check_sum()
|
||||
}
|
||||
|
||||
pub fn (d &Digest) size() int { return Size }
|
||||
|
||||
pub fn (d &Digest) block_size() int { return BlockSize }
|
5
vlib/crypto/sha1/sha1_test.v
Normal file
5
vlib/crypto/sha1/sha1_test.v
Normal file
@ -0,0 +1,5 @@
|
||||
import crypto.sha1
|
||||
|
||||
fn test_crypto_sha1() {
|
||||
assert sha1.sum('This is a sha1 hash.'.bytes()).hex() == '6FF5FA4D5166D5C2576FE56ED1EC2D5AB0FDF936'
|
||||
}
|
117
vlib/crypto/sha1/sha1block.v
Normal file
117
vlib/crypto/sha1/sha1block.v
Normal file
@ -0,0 +1,117 @@
|
||||
module sha1
|
||||
|
||||
import math.bits
|
||||
|
||||
const (
|
||||
_K0 = 0x5A827999
|
||||
_K1 = 0x6ED9EBA1
|
||||
_K2 = 0x8F1BBCDC
|
||||
_K3 = 0xCA62C1D6
|
||||
)
|
||||
|
||||
fn block(dig &Digest, p []byte) {
|
||||
mut w := [u32(0); 16]
|
||||
mut h0 := dig.h[0]
|
||||
mut h1 := dig.h[1]
|
||||
mut h2 := dig.h[2]
|
||||
mut h3 := dig.h[3]
|
||||
mut h4 := dig.h[4]
|
||||
for p.len >= Chunk {
|
||||
// Can interlace the computation of w with the
|
||||
// rounds below if needed for speed.
|
||||
for i := 0; i < 16; i++ {
|
||||
j := i * 4
|
||||
w[i] = u32(u32(p[j])<<u32(24)) | u32(u32(p[j+1])<<u32(16)) | u32(u32(p[j+2])<<u32(8)) | u32(u32(p[j+3]))
|
||||
}
|
||||
|
||||
mut a := h0
|
||||
mut b := h1
|
||||
mut c := h2
|
||||
mut d := h3
|
||||
mut e := h4
|
||||
|
||||
// Each of the four 20-iteration rounds
|
||||
// differs only in the computation of f and
|
||||
// the choice of K (_K0, _K1, etc).
|
||||
mut i := 0
|
||||
for i < 16 {
|
||||
f := u32(b&c | (~b)&d)
|
||||
t := bits.rotate_left_32(a, 5) + f + e + w[i&0xf] + u32(_K0)
|
||||
e = d
|
||||
d = c
|
||||
c = bits.rotate_left_32(b, 30)
|
||||
b = a
|
||||
a = t
|
||||
i++
|
||||
}
|
||||
for i < 20 {
|
||||
tmp := w[(i-3)&0xf] ^ w[(i-8)&0xf] ^ w[(i-14)&0xf] ^ w[(i)&0xf]
|
||||
w[i&0xf] = u32(tmp<<u32(1)) | u32(tmp>>u32(32-1))
|
||||
f := b&c | (~b)&d
|
||||
t := bits.rotate_left_32(a, 5) + f + e + w[i&0xf] + u32(_K0)
|
||||
e = d
|
||||
d = c
|
||||
c = bits.rotate_left_32(b, 30)
|
||||
b = a
|
||||
a = t
|
||||
i++
|
||||
}
|
||||
for i < 40 {
|
||||
tmp := w[(i-3)&0xf] ^ w[(i-8)&0xf] ^ w[(i-14)&0xf] ^ w[(i)&0xf]
|
||||
w[i&0xf] = u32(tmp<<u32(1)) | u32(tmp>>u32(32-1))
|
||||
f := b ^ c ^ d
|
||||
t := bits.rotate_left_32(a, 5) + f + e + w[i&0xf] + u32(_K1)
|
||||
e = d
|
||||
d = c
|
||||
c = bits.rotate_left_32(b, 30)
|
||||
b = a
|
||||
a = t
|
||||
i++
|
||||
}
|
||||
for i < 60 {
|
||||
tmp := w[(i-3)&0xf] ^ w[(i-8)&0xf] ^ w[(i-14)&0xf] ^ w[(i)&0xf]
|
||||
w[i&0xf] = u32(tmp<<u32(1)) | u32(tmp>>u32(32-1))
|
||||
f := ((b | c) & d) | (b & c)
|
||||
t := bits.rotate_left_32(a, 5) + f + e + w[i&0xf] + u32(_K2)
|
||||
|
||||
e = d
|
||||
d = c
|
||||
c = bits.rotate_left_32(b, 30)
|
||||
b = a
|
||||
a = t
|
||||
|
||||
i++
|
||||
}
|
||||
for i < 80 {
|
||||
tmp := w[(i-3)&0xf] ^ w[(i-8)&0xf] ^ w[(i-14)&0xf] ^ w[(i)&0xf]
|
||||
w[i&0xf] = u32(tmp<<u32(1)) | u32(tmp>>u32(32-1))
|
||||
f := b ^ c ^ d
|
||||
t := bits.rotate_left_32(a, 5) + f + e + w[i&0xf] + u32(_K3)
|
||||
e = d
|
||||
d = c
|
||||
c = bits.rotate_left_32(b, 30)
|
||||
b = a
|
||||
a = t
|
||||
i++
|
||||
}
|
||||
|
||||
h0 += a
|
||||
h1 += b
|
||||
h2 += c
|
||||
h3 += d
|
||||
h4 += e
|
||||
|
||||
if Chunk >= p.len {
|
||||
p = []byte
|
||||
} else {
|
||||
p = p.right(Chunk)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
dig.h[0] = h0
|
||||
dig.h[1] = h1
|
||||
dig.h[2] = h2
|
||||
dig.h[3] = h3
|
||||
dig.h[4] = h4
|
||||
}
|
93
vlib/encoding/binary/binary.v
Normal file
93
vlib/encoding/binary/binary.v
Normal file
@ -0,0 +1,93 @@
|
||||
// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
|
||||
// Use of this source code is governed by an MIT license
|
||||
// that can be found in the LICENSE file.
|
||||
|
||||
module binary
|
||||
|
||||
|
||||
// Little Endian
|
||||
pub fn little_endian_endian_u16(b []byte) u16 {
|
||||
_ := b[1] // bounds check
|
||||
return u16(b[0]) | u16(u16(b[1])<<u16(8))
|
||||
}
|
||||
|
||||
|
||||
pub fn little_endian_put_u16(b []byte, v u16) {
|
||||
_ := b[1] // bounds check
|
||||
b[0] = byte(v)
|
||||
b[1] = byte(v >> u16(8))
|
||||
}
|
||||
|
||||
pub fn little_endian_u32(b []byte) u32 {
|
||||
_ := b[3] // bounds check
|
||||
return u32(b[0]) | u32(u32(b[1])<<u32(8)) | u32(u32(b[2])<<u32(16)) | u32(u32(b[3])<<u32(24))
|
||||
}
|
||||
|
||||
pub fn little_endian_put_u32(b []byte, v u32) {
|
||||
_ := b[3] // bounds check
|
||||
b[0] = byte(v)
|
||||
b[1] = byte(v >> u32(8))
|
||||
b[2] = byte(v >> u32(16))
|
||||
b[3] = byte(v >> u32(24))
|
||||
}
|
||||
|
||||
pub fn little_endian_u64(b []byte) u64 {
|
||||
_ := b[7] // bounds check
|
||||
return u64(b[0]) | u64(u64(b[1])<<u64(8)) | u64(u64(b[2])<<u64(16)) | u64(u64(b[3])<<u64(24)) |
|
||||
u64(u64(b[4])<<u64(32)) | u64(u64(b[5])<<u64(40)) | u64(u64(b[6])<<u64(48)) | u64(u64(b[7])<<u64(56))
|
||||
}
|
||||
|
||||
pub fn little_endian_put_u64(b []byte, v u64) {
|
||||
_ := b[7] // bounds check
|
||||
b[0] = byte(v)
|
||||
b[1] = byte(v >> u64(8))
|
||||
b[2] = byte(v >> u64(16))
|
||||
b[3] = byte(v >> u64(24))
|
||||
b[4] = byte(v >> u64(32))
|
||||
b[5] = byte(v >> u64(40))
|
||||
b[6] = byte(v >> u64(48))
|
||||
b[7] = byte(v >> u64(56))
|
||||
}
|
||||
|
||||
// Big Endian
|
||||
pub fn big_endian_u16(b []byte) u16 {
|
||||
_ := b[1] // bounds check
|
||||
return u16(b[1]) | u16(u16(b[0])<<u16(8))
|
||||
}
|
||||
|
||||
pub fn big_endian_put_u16(b []byte, v u16) {
|
||||
_ := b[1] // bounds check
|
||||
b[0] = byte(v >> u16(8))
|
||||
b[1] = byte(v)
|
||||
}
|
||||
|
||||
pub fn big_endian_u32(b []byte) u32 {
|
||||
_ := b[3] // bounds check
|
||||
return u32(b[3]) | u32(u32(b[2])<<u32(8)) | u32(u32(b[1])<<u32(16)) | u32(u32(b[0])<<u32(24))
|
||||
}
|
||||
|
||||
pub fn big_endian_put_u32(b []byte, v u32) {
|
||||
_ := b[3] // bounds check
|
||||
b[0] = byte(v >> u32(24))
|
||||
b[1] = byte(v >> u32(16))
|
||||
b[2] = byte(v >> u32(8))
|
||||
b[3] = byte(v)
|
||||
}
|
||||
|
||||
pub fn big_endian_u64(b []byte) u64 {
|
||||
_ := b[7] // bounds check
|
||||
return u64(b[7]) | u64(u64(b[6])<<u64(8)) | u64(u64(b[5])<<u64(16)) | u64(u64(b[4])<<u64(24)) |
|
||||
u64(u64(b[3])<<u64(32)) | u64(u64(b[2])<<u64(40)) | u64(u64(b[1])<<u64(48)) | u64(u64(b[0])<<u64(56))
|
||||
}
|
||||
|
||||
pub fn big_endian_put_u64(b []byte, v u64) {
|
||||
_ := b[7] // bounds check
|
||||
b[0] = byte(v >> u64(56))
|
||||
b[1] = byte(v >> u64(48))
|
||||
b[2] = byte(v >> u64(40))
|
||||
b[3] = byte(v >> u64(32))
|
||||
b[4] = byte(v >> u64(24))
|
||||
b[5] = byte(v >> u64(16))
|
||||
b[6] = byte(v >> u64(8))
|
||||
b[7] = byte(v)
|
||||
}
|
@ -13,6 +13,11 @@ const (
|
||||
Koopman = 0xeb31d82e
|
||||
)
|
||||
|
||||
// The size of a CRC-32 checksum in bytes.
|
||||
const (
|
||||
Size = 4
|
||||
)
|
||||
|
||||
struct Crc32 {
|
||||
mut:
|
||||
table []u32
|
||||
@ -32,7 +37,7 @@ fn(c mut Crc32) generate_table(poly int) {
|
||||
}
|
||||
}
|
||||
|
||||
fn(c &Crc32) sum_32(s string) u32 {
|
||||
fn(c &Crc32) sum32(s string) u32 {
|
||||
mut crc := ~u32(0)
|
||||
for i := 0; i < s.len; i++ {
|
||||
crc = c.table[byte(crc)^s[i]] ^ u32(crc >> u32(8))
|
||||
@ -41,7 +46,7 @@ fn(c &Crc32) sum_32(s string) u32 {
|
||||
}
|
||||
|
||||
pub fn(c &Crc32) checksum(s string) u32 {
|
||||
return c.sum_32(s)
|
||||
return c.sum32(s)
|
||||
}
|
||||
|
||||
// pass the polinomial to use
|
||||
@ -54,6 +59,6 @@ pub fn new(poly int) *Crc32 {
|
||||
// calculate crc32 using IEEE
|
||||
pub fn sum(s string) u32 {
|
||||
mut c := new(IEEE)
|
||||
return c.sum_32(s)
|
||||
return c.sum32(s)
|
||||
}
|
||||
|
||||
|
21
vlib/hash/hash.v
Normal file
21
vlib/hash/hash.v
Normal file
@ -0,0 +1,21 @@
|
||||
// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
|
||||
// Use of this source code is governed by an MIT license
|
||||
// that can be found in the LICENSE file.
|
||||
|
||||
module hash
|
||||
|
||||
interface Hash {
|
||||
// Sum appends the current hash to b and returns the resulting array.
|
||||
// It does not change the underlying hash state.
|
||||
sum(b []byte) []byte
|
||||
size() int
|
||||
block_size() int
|
||||
}
|
||||
|
||||
interface Hash32 {
|
||||
sum32() uint32
|
||||
}
|
||||
|
||||
interface Hash64 {
|
||||
sum64() uint64
|
||||
}
|
47
vlib/math/bits/bits.v
Normal file
47
vlib/math/bits/bits.v
Normal file
@ -0,0 +1,47 @@
|
||||
// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
|
||||
// Use of this source code is governed by an MIT license
|
||||
// that can be found in the LICENSE file.
|
||||
|
||||
module bits
|
||||
|
||||
// --- RotateLeft ---
|
||||
|
||||
// rotate_left_8 returns the value of x rotated left by (k mod 8) bits.
|
||||
// To rotate x right by k bits, call rotate_left_8(x, -k).
|
||||
//
|
||||
// This function's execution time does not depend on the inputs.
|
||||
pub fn rotate_left_8(x u8, k int) u8 {
|
||||
n := u8(8)
|
||||
s := u8(k) & u8(n - u8(1))
|
||||
return u8((x<<s) | (x>>(n-s)))
|
||||
}
|
||||
|
||||
// rotate_left_16 returns the value of x rotated left by (k mod 16) bits.
|
||||
// To rotate x right by k bits, call rotate_left_16(x, -k).
|
||||
//
|
||||
// This function's execution time does not depend on the inputs.
|
||||
pub fn rotate_left_16(x u16, k int) u16 {
|
||||
n := u16(16)
|
||||
s := u16(k) & (n - u16(1))
|
||||
return u16((x<<s) | (x>>(n-s)))
|
||||
}
|
||||
|
||||
// rotate_left_32 returns the value of x rotated left by (k mod 32) bits.
|
||||
// To rotate x right by k bits, call rotate_left_32(x, -k).
|
||||
//
|
||||
// This function's execution time does not depend on the inputs.
|
||||
pub fn rotate_left_32(x u32, k int) u32 {
|
||||
n := u32(32)
|
||||
s := u32(k) & (n - u32(1))
|
||||
return u32(u32(x<<s) | u32(x>>(n-s)))
|
||||
}
|
||||
|
||||
// rotate_left_64 returns the value of x rotated left by (k mod 64) bits.
|
||||
// To rotate x right by k bits, call rotate_left_64(x, -k).
|
||||
//
|
||||
// This function's execution time does not depend on the inputs.
|
||||
pub fn rotate_left_64(x u64, k int) u64 {
|
||||
n := u64(64)
|
||||
s := u64(k) & (n - u64(1))
|
||||
return u64(u64(x<<s) | u64(x>>(n-s)))
|
||||
}
|
Loading…
Reference in New Issue
Block a user