wireguard-go/src/device.go

package main

import (
	"sync"
)

type Device struct {
	mutex        sync.RWMutex
	peers        map[NoisePublicKey]*Peer
	privateKey   NoisePrivateKey
	publicKey    NoisePublicKey
	fwMark       uint32
	listenPort   uint16
	routingTable RoutingTable
}

func (dev *Device) RemovePeer(key NoisePublicKey) {
	dev.mutex.Lock()
	defer dev.mutex.Unlock()
	peer, ok := dev.peers[key]
	if !ok {
		return
	}
	peer.mutex.Lock()
	dev.routingTable.RemovePeer(peer)
	delete(dev.peers, key)
}

func (dev *Device) RemoveAllAllowedIps(peer *Peer) {

}

func (dev *Device) RemoveAllPeers() {
	dev.mutex.Lock()
	defer dev.mutex.Unlock()

	for key, peer := range dev.peers {
		peer.mutex.Lock()
		dev.routingTable.RemovePeer(peer)
		delete(dev.peers, key)
		peer.mutex.Unlock()
	}
}