103 lines
1.7 KiB
Go
103 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"sync"
|
|
)
|
|
|
|
/* Index=0 is reserved for unset indecies
|
|
*
|
|
* TODO: Rethink map[id] -> peer VS map[id] -> handshake and handshake <ref> peer
|
|
*
|
|
*/
|
|
|
|
type IndexTableEntry struct {
|
|
peer *Peer
|
|
handshake *Handshake
|
|
keyPair *KeyPair
|
|
}
|
|
|
|
type IndexTable struct {
|
|
mutex sync.RWMutex
|
|
table map[uint32]IndexTableEntry
|
|
}
|
|
|
|
func randUint32() (uint32, error) {
|
|
var buff [4]byte
|
|
_, err := rand.Read(buff[:])
|
|
id := uint32(buff[0])
|
|
id <<= 8
|
|
id |= uint32(buff[1])
|
|
id <<= 8
|
|
id |= uint32(buff[2])
|
|
id <<= 8
|
|
id |= uint32(buff[3])
|
|
return id, err
|
|
}
|
|
|
|
func (table *IndexTable) Init() {
|
|
table.mutex.Lock()
|
|
table.table = make(map[uint32]IndexTableEntry)
|
|
table.mutex.Unlock()
|
|
}
|
|
|
|
func (table *IndexTable) Delete(index uint32) {
|
|
if index == 0 {
|
|
return
|
|
}
|
|
table.mutex.Lock()
|
|
delete(table.table, index)
|
|
table.mutex.Unlock()
|
|
}
|
|
|
|
func (table *IndexTable) Insert(key uint32, value IndexTableEntry) {
|
|
table.mutex.Lock()
|
|
table.table[key] = value
|
|
table.mutex.Unlock()
|
|
}
|
|
|
|
func (table *IndexTable) NewIndex(peer *Peer) (uint32, error) {
|
|
for {
|
|
// generate random index
|
|
|
|
index, err := randUint32()
|
|
if err != nil {
|
|
return index, err
|
|
}
|
|
if index == 0 {
|
|
continue
|
|
}
|
|
|
|
// check if index used
|
|
|
|
table.mutex.RLock()
|
|
_, ok := table.table[index]
|
|
if ok {
|
|
continue
|
|
}
|
|
table.mutex.RUnlock()
|
|
|
|
// replace index
|
|
|
|
table.mutex.Lock()
|
|
_, found := table.table[index]
|
|
if found {
|
|
table.mutex.Unlock()
|
|
continue
|
|
}
|
|
table.table[index] = IndexTableEntry{
|
|
peer: peer,
|
|
handshake: &peer.handshake,
|
|
keyPair: nil,
|
|
}
|
|
table.mutex.Unlock()
|
|
return index, nil
|
|
}
|
|
}
|
|
|
|
func (table *IndexTable) Lookup(id uint32) IndexTableEntry {
|
|
table.mutex.RLock()
|
|
defer table.mutex.RUnlock()
|
|
return table.table[id]
|
|
}
|