From 3fb2edb00043433fd960de3663a73fb9a327b394 Mon Sep 17 00:00:00 2001
From: dushixiang <helloworld1024@foxmail.com>
Date: Mon, 15 Nov 2021 21:51:37 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BC=9A=E8=AF=9D=E5=9C=A8?=
 =?UTF-8?q?=E6=9C=AA=E5=BD=95=E5=B1=8F=E6=97=B6=E7=9A=84=E5=AE=A1=E8=AE=A1?=
 =?UTF-8?q?=E7=8A=B6=E6=80=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/api/ssh.go    | 4 ++++
 server/api/sshd.go   | 5 +++++
 server/api/tunnel.go | 4 ++++
 3 files changed, 13 insertions(+)

diff --git a/server/api/ssh.go b/server/api/ssh.go
index 7169b3e..e4c375d 100644
--- a/server/api/ssh.go
+++ b/server/api/ssh.go
@@ -166,6 +166,10 @@ func SSHEndpoint(c echo.Context) (err error) {
 		Status:       constant.Connecting,
 		Recording:    recording,
 	}
+	if sess.Recording == "" {
+		// 未录屏时无需审计
+		sess.Reviewed = true
+	}
 	// 创建新会话
 	log.Debugf("创建新会话 %v", sess.ConnectionId)
 	if err := sessionRepository.UpdateById(&sess, sessionId); err != nil {
diff --git a/server/api/sshd.go b/server/api/sshd.go
index 24ca18f..2cc74c9 100644
--- a/server/api/sshd.go
+++ b/server/api/sshd.go
@@ -333,6 +333,11 @@ func handleAccessAsset(sess *ssh.Session, sessionId string) (err error) {
 	sessionForUpdate.Recording = recording
 	sessionForUpdate.ConnectedTime = utils.NowJsonTime()
 
+	if sessionForUpdate.Recording == "" {
+		// 未录屏时无需审计
+		sessionForUpdate.Reviewed = true
+	}
+
 	if err := sessionRepository.UpdateById(&sessionForUpdate, sessionId); err != nil {
 		return err
 	}
diff --git a/server/api/tunnel.go b/server/api/tunnel.go
index 25391f3..88b2351 100644
--- a/server/api/tunnel.go
+++ b/server/api/tunnel.go
@@ -162,6 +162,10 @@ func TunEndpoint(c echo.Context) error {
 			Status:       constant.Connecting,
 			Recording:    configuration.GetParameter(guacd.RecordingPath),
 		}
+		if sess.Recording == "" {
+			// 未录屏时无需审计
+			sess.Reviewed = true
+		}
 		// 创建新会话
 		log.Debugf("[%v:%v] 创建新会话: %v", sessionId, connectionId, sess.ConnectionId)
 		if err := sessionRepository.UpdateById(&sess, sessionId); err != nil {