兄弟連區塊鏈教程open-ethereum-pool礦池源碼分析policy模塊
阿新 • • 發佈:2018-10-11
lang ids onf lock else unlock urn HERE fun open-ethereum-pooleth礦池-policy模塊
PolicyServer定義
type PolicyServer struct { sync.RWMutex statsMu sync.Mutex config *Config stats map[string]*Stats banChannel chan string startedAt int64 grace int64 timeout int64 blacklist []string whitelist []string storage *storage.RedisClient }
GetBlacklist和GetWhitelist
// Always returns list of addresses. If Redis fails it will return empty list. func (r *RedisClient) GetBlacklist() ([]string, error) { //SMEMBERS eth:blacklist //Smembers 命令返回集合中的所有的成員 cmd := r.client.SMembers(r.formatKey("blacklist")) if cmd.Err() != nil { return []string{}, cmd.Err() } return cmd.Val(), nil } // Always returns list of IPs. If Redis fails it will return empty list. func (r *RedisClient) GetWhitelist() ([]string, error) { //SMEMBERS eth:whitelist //Smembers 命令返回集合中的所有的成員 cmd := r.client.SMembers(r.formatKey("whitelist")) if cmd.Err() != nil { return []string{}, cmd.Err() } return cmd.Val(), nil }
IsBanned
func (s *PolicyServer) IsBanned(ip string) bool { x := s.Get(ip) return atomic.LoadInt32(&x.Banned) > 0 } func (s *PolicyServer) Get(ip string) *Stats { s.statsMu.Lock() defer s.statsMu.Unlock() if x, ok := s.stats[ip]; !ok { x = s.NewStats() s.stats[ip] = x return x } else { x.heartbeat() return x } } func (s *PolicyServer) NewStats() *Stats { x := &Stats{ ConnLimit: s.config.Limits.Limit, } x.heartbeat() return x }
處理ApplyMalformedPolicy
//應用格式錯誤的策略
//malformedLimit為5次
func (s *PolicyServer) ApplyMalformedPolicy(ip string) bool {
x := s.Get(ip)
n := x.incrMalformed()
if n >= s.config.Banning.MalformedLimit {
s.forceBan(x, ip)
return false
}
return true
}
func (x *Stats) incrMalformed() int32 {
return atomic.AddInt32(&x.Malformed, 1)
}
func (s *PolicyServer) forceBan(x *Stats, ip string) {
//沒啟用Banning或在白名單
if !s.config.Banning.Enabled || s.InWhiteList(ip) {
return
}
atomic.StoreInt64(&x.BannedAt, util.MakeTimestamp())
//x.Banned賦值為1
if atomic.CompareAndSwapInt32(&x.Banned, 0, 1) {
//"ipset": "blacklist",
if len(s.config.Banning.IPSet) > 0 {
s.banChannel <- ip
} else {
log.Println("Banned peer", ip)
}
}
}
處理ApplyLoginPolicy
func (s *PolicyServer) ApplyLoginPolicy(addy, ip string) bool {
if s.InBlackList(addy) {
x := s.Get(ip)
s.forceBan(x, ip)
return false
}
return true
}
func (s *PolicyServer) InBlackList(addy string) bool {
s.RLock()
defer s.RUnlock()
return util.StringInSlice(addy, s.blacklist)
}
處理ApplyLoginPolicy
func (s *PolicyServer) ApplySharePolicy(ip string, validShare bool) bool {
x := s.Get(ip)
x.Lock()
if validShare {
x.ValidShares++
if s.config.Limits.Enabled {
//Increase allowed number of connections on each valid share
//每個有效share可以增加的連接數
//"limitJump": 10
x.incrLimit(s.config.Limits.LimitJump)
}
} else {
x.InvalidShares++
}
totalShares := x.ValidShares + x.InvalidShares
//Check after after miner submitted this number of shares
//30個shares後檢查
//"checkThreshold": 30,
if totalShares < s.config.Banning.CheckThreshold {
x.Unlock()
return true
}
validShares := float32(x.ValidShares)
invalidShares := float32(x.InvalidShares)
x.resetShares()
x.Unlock()
//無效share比例
ratio := invalidShares / validShares
// Percent of invalid shares from all shares to ban miner
//"invalidPercent": 30,
if ratio >= s.config.Banning.InvalidPercent/100.0 {
s.forceBan(x, ip)
return false
}
return true
}
//加
func (x *Stats) incrLimit(n int32) {
atomic.AddInt32(&x.ConnLimit, n)
}
//reset
func (x *Stats) resetShares() {
x.ValidShares = 0
x.InvalidShares = 0
}
處理ApplyLimitPolicy
func (s *PolicyServer) ApplyLimitPolicy(ip string) bool {
if !s.config.Limits.Enabled {
return true
}
now := util.MakeTimestamp()
//"grace": "5m",
if now-s.startedAt > s.grace {
//減1
return s.Get(ip).decrLimit() > 0
}
return true
}
func (x *Stats) decrLimit() int32 {
return atomic.AddInt32(&x.ConnLimit, -1)
}
處理BanClient
func (s *PolicyServer) BanClient(ip string) {
x := s.Get(ip)
s.forceBan(x, ip)
}
兄弟連區塊鏈教程open-ethereum-pool礦池源碼分析policy模塊