Skip to content

Commit

Permalink
demo dispatch: 当超过MaxSubSessionPerIp时,踢出超过限制的、时间上偏后的连接
Browse files Browse the repository at this point in the history
  • Loading branch information
q191201771 committed Jun 21, 2024
1 parent 455a9b2 commit 090bbe1
Showing 1 changed file with 21 additions and 10 deletions.
31 changes: 21 additions & 10 deletions app/demo/dispatch/dispatch__security.go
Original file line number Diff line number Diff line change
Expand Up @@ -26,28 +26,39 @@ func securityMaxSubSessionPerIp(info base.UpdateInfo) {
for _, sub := range g.StatSubs {
host, _, err := net.SplitHostPort(sub.RemoteAddr)
if err != nil {
nazalog.Warnf("split host port failed. remote addr=%s", sub.RemoteAddr)
nazalog.Warnf("securityMaxSubSessionPerIp. split host port failed. remote addr=%s", sub.RemoteAddr)
continue
}
ip2SubSessions[host] = append(ip2SubSessions[host], sub)
sessionId2StreamName[sub.SessionId] = g.StreamName
}
}

lastUnix := int64(-1)
lastId := ""
for ip, subs := range ip2SubSessions {
if len(subs) <= config.MaxSubSessionPerIp {
continue
}
nazalog.Debugf("close session. ip=%s, session count=%d", ip, len(subs))
nazalog.Debugf("securityMaxSubSessionPerIp. close session. ip=%s, session count=%d", ip, len(subs))

for _, sub := range subs {
//if sub.Protocol == base.SessionProtocolHlsStr {
// host, _, err := net.SplitHostPort(sub.RemoteAddr)
// if err != nil {
// nazalog.Warnf("split host port failed. remote addr=%s", sub.RemoteAddr)
// continue
// }
// addIpBlacklist(info.ServerId, host, 60)
//}
st, err := base.ParseReadableTime(sub.StartTime)
if err != nil {
nazalog.Warnf("securityMaxSubSessionPerIp. parse readable time failed. start time=%s, err=%+v", sub.StartTime, err)
continue
}
if st.UnixMilli() > lastUnix {
lastUnix = st.UnixMilli()
lastId = sub.SessionId
}
}

for _, sub := range subs {
if sub.SessionId == lastId {
nazalog.Debugf("securityMaxSubSessionPerIp. skip kick. last id=%s", lastId)
continue
}
kickSession(info.ServerId, sessionId2StreamName[sub.SessionId], sub.SessionId)
}
}
Expand Down

0 comments on commit 090bbe1

Please sign in to comment.