@@ -35,6 +35,8 @@ func IncidentResponseStream(cnf *config.Config, ctx context.Context) {
3535 if ! connErrMsgWritten {
3636 utils .Logger .ErrorF ("failed to start AgentStream: %v" , err )
3737 connErrMsgWritten = true
38+ } else {
39+ utils .Logger .LogF (100 , "failed to start AgentStream: %v" , err )
3840 }
3941 time .Sleep (timeToSleep )
4042 continue
@@ -46,6 +48,7 @@ func IncidentResponseStream(cnf *config.Config, ctx context.Context) {
4648 in , err := stream .Recv ()
4749 if err != nil {
4850 if strings .Contains (err .Error (), "EOF" ) {
51+ utils .Logger .LogF (100 , "error receiving command from server: %v" , err )
4952 time .Sleep (timeToSleep )
5053 break
5154 }
@@ -54,13 +57,17 @@ func IncidentResponseStream(cnf *config.Config, ctx context.Context) {
5457 if ! errorLogged {
5558 utils .Logger .ErrorF ("error receiving command from server: %v" , err )
5659 errorLogged = true
60+ } else {
61+ utils .Logger .LogF (100 , "error receiving command from server: %v" , err )
5762 }
5863 time .Sleep (timeToSleep )
5964 break
6065 } else {
6166 if ! errorLogged {
6267 utils .Logger .ErrorF ("error receiving command from server: %v" , err )
6368 errorLogged = true
69+ } else {
70+ utils .Logger .LogF (100 , "error receiving command from server: %v" , err )
6471 }
6572 time .Sleep (timeToSleep )
6673 continue
@@ -72,6 +79,7 @@ func IncidentResponseStream(cnf *config.Config, ctx context.Context) {
7279 err = commandProcessor (path , stream , cnf , []string {msg .Command .Command , in .GetCommand ().CmdId })
7380 if err != nil {
7481 if strings .Contains (err .Error (), "EOF" ) {
82+ utils .Logger .LogF (100 , "error sending result to server: %v" , err )
7583 time .Sleep (timeToSleep )
7684 break
7785 }
@@ -80,13 +88,17 @@ func IncidentResponseStream(cnf *config.Config, ctx context.Context) {
8088 if ! errorLogged {
8189 utils .Logger .ErrorF ("error sending result to server: %v" , err )
8290 errorLogged = true
91+ } else {
92+ utils .Logger .LogF (100 , "error sending result to server: %v" , err )
8393 }
8494 time .Sleep (timeToSleep )
8595 break
8696 } else {
8797 if ! errorLogged {
8898 utils .Logger .ErrorF ("error sending result to server: %v" , err )
8999 errorLogged = true
100+ } else {
101+ utils .Logger .LogF (100 , "error sending result to server: %v" , err )
90102 }
91103 time .Sleep (timeToSleep )
92104 continue
@@ -102,7 +114,7 @@ func commandProcessor(path string, stream AgentService_AgentStreamClient, cnf *c
102114 var result string
103115 var errB bool
104116
105- utils .Logger .Info ( "Received command: %s" , commandPair [0 ])
117+ utils .Logger .LogF ( 100 , "Received command: %s" , commandPair [0 ])
106118
107119 switch runtime .GOOS {
108120 case "windows" :
@@ -116,7 +128,7 @@ func commandProcessor(path string, stream AgentService_AgentStreamClient, cnf *c
116128 if errB {
117129 utils .Logger .ErrorF ("error executing command %s: %s" , commandPair [0 ], result )
118130 } else {
119- utils .Logger .Info ( "Result when executing the command %s: %s" , commandPair [0 ], result )
131+ utils .Logger .LogF ( 100 , "Result when executing the command %s: %s" , commandPair [0 ], result )
120132 }
121133
122134 if err := stream .Send (& BidirectionalStream {
@@ -126,7 +138,7 @@ func commandProcessor(path string, stream AgentService_AgentStreamClient, cnf *c
126138 }); err != nil {
127139 return err
128140 } else {
129- utils .Logger .Info ( "Result sent to server successfully!!!" )
141+ utils .Logger .LogF ( 100 , "Result sent to server successfully!!!" )
130142 }
131143 return nil
132144}
0 commit comments