Method handleNewConnection

agentprotocol/server.go:316–362

(packet *Packet)

}

func (c *ForwardCtx) handleNewConnection(packet *Packet) {
	newConnectionPacket, err := c.unmarshalNewConnection(packet.Payload)
	if err != nil {
		c.logger.Error("Error unmarshalling new connection payload", err)
		return
	}
	pipeReader, pipeWriter := io.Pipe()
	connection := Connection{
		state:        CONNECTION_STATE_WAITINIT,
		id:           packet.ConnectionId,
		details:      newConnectionPacket,
		bufferReader: pipeReader,
		bufferWriter: pipeWriter,
		ctx:          c,
		logger:       c.logger,
	}
	connection.stateCond = sync.NewCond(&connection.lock)
	c.connMapMu.Lock()
	if _, ok := c.connMap[packet.ConnectionId]; ok {
		c.logger.Warning("Remote tried to open connection with re-used connectionId")
		// Cannot send reject here, might interfere with other connection ?
		c.connMapMu.Unlock()
		return
	}
	if packet.ConnectionId <= c.connectionId {
		c.logger.Warning("Suspicious connection, id <= prev")
		// Can't send reject here either
		c.connMapMu.Unlock()
		return
	}
	if packet.ConnectionId != c.connectionId+1 {
		c.logger.Warning("Suspicious connection, id not prev + 1")
	}

	c.connectionId = packet.ConnectionId
	c.connMap[packet.ConnectionId] = &connection
	c.waitGroup.Add(1)
	c.connMapMu.Unlock()

	if c.stopped {
		c.logger.Warning("Client tried opening a connection after stopping")
		_ = connection.Reject()
		return
	}

	c.connectionChannel <- &connection
}

func (c *ForwardCtx) handleBackend() {
	for {