修复dtls链接的panic文件

This commit is contained in:
bjdgyc
2022-11-04 15:40:06 +08:00
parent 46721b1453
commit 87dcc63b6f
2 changed files with 34 additions and 11 deletions

View File

@@ -94,18 +94,24 @@ func checkSession() {
timeout := time.Duration(base.Cfg.SessionTimeout) * time.Second
tick := time.NewTicker(time.Second * 60)
for range tick.C {
sessMux.Lock()
outToken := []string{}
sessMux.RLock()
t := time.Now()
for k, v := range sessions {
v.mux.Lock()
v.mux.RLock()
if !v.IsActive {
if t.Sub(v.LastLogin) > timeout {
delete(sessions, k)
outToken = append(outToken, k)
}
}
v.mux.Unlock()
v.mux.RUnlock()
}
sessMux.RUnlock()
// 删除过期session
for _, v := range outToken {
CloseSess(v)
}
sessMux.Unlock()
}
}()
}
@@ -365,6 +371,20 @@ func Dtls2Sess(did string) *Session {
return sessions[token]
}
func Dtls2CSess(did string) *ConnSession {
sessMux.RLock()
defer sessMux.RUnlock()
token := dtlsIds[did]
sess := sessions[token]
if sess == nil {
return nil
}
sess.mux.RLock()
defer sess.mux.RUnlock()
return sess.CSess
}
func Dtls2MasterSecret(did string) string {
sessMux.RLock()
token := dtlsIds[did]
@@ -396,6 +416,7 @@ func CloseSess(token string) {
}
delete(sessions, token)
delete(dtlsIds, sess.DtlsSid)
sess.CSess.Close()
}
@@ -414,7 +435,5 @@ func DelSessByStoken(stoken string) {
stoken = strings.TrimSpace(stoken)
sarr := strings.Split(stoken, "@")
token := sarr[1]
sessMux.Lock()
delete(sessions, token)
sessMux.Unlock()
CloseSess(token)
}