Method handleData

agentprotocol/server.go:190–233

(packet *Packet)

}

func (c *ForwardCtx) handleData(packet *Packet) {
	c.connMapMu.RLock()
	conn, ok := c.connMap[packet.ConnectionId]
	c.connMapMu.RUnlock()
	if !ok {
		c.logger.Info(
			message.NewMessage(
				message.EAgentUnknownConnection,
				"Received data packet with unknown connection id %d",
				packet.ConnectionId,
			),
		)
		return
	}
	conn.lock.Lock()
	defer conn.lock.Unlock()
	if conn.state != CONNECTION_STATE_STARTED {
		c.logger.Info(
			message.NewMessage(
				message.EAgentConnectionInvalidState,
				"Received data packet for a connection in a non-started state",
			),
		)
		return
	}
	nByte, err := conn.bufferWriter.Write(packet.Payload)
	if err != nil {
		c.logger.Error(message.Wrap(
			err,
			message.MSSHConnected,
			"Error handling data packet",
		))
		return
	}
	if nByte != len(packet.Payload) {
		c.logger.Warning(
			message.NewMessage(
				message.EAgentWriteFailed,
				"Failed to write connection packet to agent",
			),
		)
		return
	}
}

func (c *ForwardCtx) handleClose(packet *Packet) {
	c.connMapMu.Lock()