From 98113964ade47518947a1167d35c53a6a95de15b Mon Sep 17 00:00:00 2001
From: PICKQUE <925537712@qq.com>
Date: Wed, 12 Jun 2024 17:55:32 +0800
Subject: [PATCH] optimize ssh Feed function example

---
 analyzer/tcp/ssh.go | 50 +++++++++++++++++++++++++++------------------
 1 file changed, 30 insertions(+), 20 deletions(-)

diff --git a/analyzer/tcp/ssh.go b/analyzer/tcp/ssh.go
index a636441..abd4046 100644
--- a/analyzer/tcp/ssh.go
+++ b/analyzer/tcp/ssh.go
@@ -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)
 }