Git-Mirrors/constellation
1
0
Fork 0
mirror of https://github.com/edgelesssys/constellation.git synced 2025-06-20 12:14:23 -04:00
Code Issues Projects Releases Packages Wiki Activity

metadata-api: support dnsmasq (#3115)

Browse source
This commit is contained in:
3u13r 2024-05-24 15:14:16 +02:00 • committed by GitHub
parent 37e46b98c2
commit 21c30ebb76
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
18 changed files with 304 additions and 270 deletions
Download patch file Download diff file Expand all files Collapse all files

31
hack/qemu-metadata-api/server/server.go
View file

@ -14,7 +14,7 @@ import (
"net/http"
"strings"
"github.com/edgelesssys/constellation/v2/hack/qemu-metadata-api/virtwrapper"
"github.com/edgelesssys/constellation/v2/hack/qemu-metadata-api/dhcp"
"github.com/edgelesssys/constellation/v2/internal/cloud/metadata"
"github.com/edgelesssys/constellation/v2/internal/role"
)
@ -22,16 +22,16 @@ import (
// Server that provides QEMU metadata.
type Server struct {
log *slog.Logger
virt virConnect
dhcpLeaseGetter LeaseGetter
network string
initSecretHashVal []byte
}
// New creates a new Server.
func New(log *slog.Logger, network, initSecretHash string, conn virConnect) *Server {
func New(log *slog.Logger, network, initSecretHash string, getter LeaseGetter) *Server {
return &Server{
log: log,
virt: conn,
dhcpLeaseGetter: getter,
network: network,
initSecretHashVal: []byte(initSecretHash),
}
@ -139,15 +139,7 @@ func (s *Server) getEndpoint(w http.ResponseWriter, r *http.Request) {
log := s.log.With(slog.String("peer", r.RemoteAddr))
log.Info("Serving GET request for /endpoint")
net, err := s.virt.LookupNetworkByName(s.network)
if err != nil {
log.With(slog.Any("error", err)).Error("Failed to lookup network")
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer net.Free()
leases, err := net.GetDHCPLeases()
leases, err := s.dhcpLeaseGetter.GetDHCPLeases()
if err != nil {
log.With(slog.Any("error", err)).Error("Failed to get DHCP leases")
http.Error(w, err.Error(), http.StatusInternalServerError)
@ -172,13 +164,7 @@ func (s *Server) getEndpoint(w http.ResponseWriter, r *http.Request) {
// listAll returns a list of all active peers.
func (s *Server) listAll() ([]metadata.InstanceMetadata, error) {
net, err := s.virt.LookupNetworkByName(s.network)
if err != nil {
return nil, err
}
defer net.Free()
leases, err := net.GetDHCPLeases()
leases, err := s.dhcpLeaseGetter.GetDHCPLeases()
if err != nil {
return nil, err
}
@ -201,6 +187,7 @@ func (s *Server) listAll() ([]metadata.InstanceMetadata, error) {
return peers, nil
}
type virConnect interface {
LookupNetworkByName(name string) (*virtwrapper.Network, error)
// LeaseGetter is an interface for getting DHCP leases.
type LeaseGetter interface {
GetDHCPLeases() ([]dhcp.NetworkDHCPLease, error)
}