MigrateRepos/OpenGFW
1
0
Fork 0
mirror of https://github.com/apernet/OpenGFW.git synced 2024-11-14 14:29:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimize ssh Feed function example

Browse Source
This commit is contained in:
PICKQUE 2024-06-12 17:55:32 +08:00
parent 1de95ed53e
commit 98113964ad
1 changed files with 31 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
analyzer/tcp/ssh.go
View File

@ -50,38 +50,48 @@ func newSSHStream(logger analyzer.Logger) *sshStream {
return s
}
func (s *sshStream) Feed(rev, start, end bool, skip int, data []byte) (u *analyzer.PropUpdate, done bool) {
func (s *sshStream) Feed(rev, start, end bool, skip int, data []byte) (u *analyzer.PropUpdate, _done bool) {
if skip != 0 {
return nil, true
}
if len(data) == 0 {
return nil, false
}
var update *analyzer.PropUpdate
var cancelled bool
var buf *utils.ByteBuffer
var updated, done *bool
var m *analyzer.PropMap
var lsm *utils.LinearStateMachine
var key string
if rev {
s.serverBuf.Append(data)
s.serverUpdated = false
cancelled, s.serverDone = s.serverLSM.Run()
if s.serverUpdated {
update = &analyzer.PropUpdate{
Type: analyzer.PropUpdateMerge,
M: analyzer.PropMap{"server": s.serverMap},
}
s.serverUpdated = false
}
buf = s.serverBuf
updated = &s.serverUpdated
lsm = s.serverLSM
m = &s.serverMap
done = &s.serverDone
key = "server"
} else {
s.clientBuf.Append(data)
s.clientUpdated = false
cancelled, s.clientDone = s.clientLSM.Run()
if s.clientUpdated {
update = &analyzer.PropUpdate{
Type: analyzer.PropUpdateMerge,
M: analyzer.PropMap{"client": s.clientMap},
}
s.clientUpdated = false
}
buf = s.clientBuf
updated = &s.serverUpdated
lsm = s.clientLSM
m = &s.serverMap
done = &s.clientDone
key = "client"
}
buf.Append(data)
cancelled, *done = lsm.Run()
if *updated {
update = &analyzer.PropUpdate{
Type: analyzer.PropUpdateMerge,
M: analyzer.PropMap{key: *m},
}
*updated = false
}
return update, cancelled || (s.clientDone && s.serverDone)
}