Skip to content

Commit 034ec3f

Browse files
committed
use NetID instead of string name
1 parent 43f157c commit 034ec3f

File tree

4 files changed

+22
-24
lines changed

4 files changed

+22
-24
lines changed

pkg/netlight/network.go

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ func (n *networker) Delete(name string) error {
131131
return err
132132
}
133133

134-
netNR, err := n.networkOf(name)
134+
netNR, err := n.networkOf(pkg.NetID(name))
135135
if err != nil {
136136
return err
137137
}
@@ -204,17 +204,17 @@ func (n *networker) AttachZDB(id string) (string, error) {
204204

205205
// GetSubnet of a local network resource identified by the network ID, ipv4 and ipv6
206206
// subnet respectively
207-
func (n *networker) GetSubnet(name string) (net.IPNet, error) {
208-
localNR, err := n.networkOf(name)
207+
func (n *networker) GetSubnet(networkID pkg.NetID) (net.IPNet, error) {
208+
localNR, err := n.networkOf(networkID)
209209
if err != nil {
210-
return net.IPNet{}, errors.Wrapf(err, "couldn't load network with name (%s)", name)
210+
return net.IPNet{}, errors.Wrapf(err, "couldn't load network with name (%s)", networkID.String())
211211
}
212212

213213
return localNR.Subnet.IPNet, nil
214214
}
215215

216-
func (n *networker) networkOf(name string) (nr pkg.Network, err error) {
217-
path := filepath.Join(n.networkDir, name)
216+
func (n *networker) networkOf(id pkg.NetID) (nr pkg.Network, err error) {
217+
path := filepath.Join(n.networkDir, id.String())
218218
file, err := os.OpenFile(path, os.O_RDWR, 0660)
219219
if err != nil {
220220
return nr, err
@@ -525,21 +525,21 @@ func (n *networker) WireguardPorts() ([]uint, error) {
525525
}
526526

527527
// GetNet of a network identified by the network ID
528-
func (n *networker) GetNet(name string) (net.IPNet, error) {
529-
localNR, err := n.networkOf(name)
528+
func (n *networker) GetNet(id pkg.NetID) (net.IPNet, error) {
529+
localNR, err := n.networkOf(id)
530530
if err != nil {
531-
return net.IPNet{}, errors.Wrapf(err, "couldn't load network (%s)", name)
531+
return net.IPNet{}, errors.Wrapf(err, "couldn't load network (%s)", id.String())
532532
}
533533

534534
return localNR.NetworkIPRange.IPNet, nil
535535
}
536536

537537
// GetDefaultGwIP returns the IPs of the default gateways inside the network
538538
// resource identified by the network ID on the local node, for IPv4
539-
func (n *networker) GetDefaultGwIP(name string) (net.IP, error) {
540-
localNR, err := n.networkOf(name)
539+
func (n *networker) GetDefaultGwIP(id pkg.NetID) (net.IP, error) {
540+
localNR, err := n.networkOf(id)
541541
if err != nil {
542-
return nil, errors.Wrapf(err, "couldn't load network (%s)", name)
542+
return nil, errors.Wrapf(err, "couldn't load network (%s)", id.String())
543543
}
544544

545545
// only IP4 atm
@@ -623,7 +623,7 @@ func (n *networker) releasePort(port uint16) error {
623623
func (n networker) setupWireguard(name string, net zos.NetworkLight, netr *resource.Resource) error {
624624
log.Debug().Msg("setting up wireguard")
625625

626-
storedNR, err := n.networkOf(name)
626+
storedNR, err := n.networkOf(pkg.NetID(name))
627627
if err != nil && !os.IsNotExist(err) {
628628
return errors.Wrap(err, "failed to load previous network setup")
629629
}

pkg/network_light.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,9 @@ type NetworkerLight interface {
3030
LoadPublicConfig() (PublicConfig, error)
3131

3232
WireguardPorts() ([]uint, error)
33-
GetDefaultGwIP(name string) (net.IP, error)
34-
GetNet(name string) (net.IPNet, error)
35-
GetSubnet(name string) (net.IPNet, error)
33+
GetDefaultGwIP(id NetID) (net.IP, error)
34+
GetNet(id NetID) (net.IPNet, error)
35+
GetSubnet(id NetID) (net.IPNet, error)
3636
}
3737

3838
type TapDevice struct {

pkg/primitives/vm-light/utils.go

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -201,9 +201,7 @@ func (p *Manager) newPrivNetworkInterface(ctx context.Context, dl gridtypes.Depl
201201
network := localStubs.NewNetworkerLightStub(p.zbus)
202202
netID := zos.NetworkID(dl.TwinID, inf.Network)
203203

204-
name := netID.String()
205-
206-
subnet, err := network.GetSubnet(ctx, name)
204+
subnet, err := network.GetSubnet(ctx, netID)
207205
if err != nil {
208206
return pkg.VMIface{}, errors.Wrapf(err, "could not get network resource subnet")
209207
}
@@ -222,7 +220,7 @@ func (p *Manager) newPrivNetworkInterface(ctx context.Context, dl gridtypes.Depl
222220
return pkg.VMIface{}, fmt.Errorf("ip %s is reserved", inf.IP.String())
223221
}
224222

225-
privNet, err := network.GetNet(ctx, name)
223+
privNet, err := network.GetNet(ctx, netID)
226224
if err != nil {
227225
return pkg.VMIface{}, errors.Wrapf(err, "could not get network range")
228226
}
@@ -232,7 +230,7 @@ func (p *Manager) newPrivNetworkInterface(ctx context.Context, dl gridtypes.Depl
232230
Mask: subnet.Mask,
233231
}
234232

235-
gw4, err := network.GetDefaultGwIP(ctx, name)
233+
gw4, err := network.GetDefaultGwIP(ctx, netID)
236234
if err != nil {
237235
return pkg.VMIface{}, errors.Wrap(err, "could not get network resource default gateway")
238236
}

pkg/stubs/network_light_stub.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ func (s *NetworkerLightStub) Detach(ctx context.Context, arg0 string) (ret0 erro
126126
return
127127
}
128128

129-
func (s *NetworkerLightStub) GetDefaultGwIP(ctx context.Context, arg0 string) (ret0 []uint8, ret1 error) {
129+
func (s *NetworkerLightStub) GetDefaultGwIP(ctx context.Context, arg0 zos.NetID) (ret0 []uint8, ret1 error) {
130130
args := []interface{}{arg0}
131131
result, err := s.client.RequestContext(ctx, s.module, s.object, "GetDefaultGwIP", args...)
132132
if err != nil {
@@ -143,7 +143,7 @@ func (s *NetworkerLightStub) GetDefaultGwIP(ctx context.Context, arg0 string) (r
143143
return
144144
}
145145

146-
func (s *NetworkerLightStub) GetNet(ctx context.Context, arg0 string) (ret0 net.IPNet, ret1 error) {
146+
func (s *NetworkerLightStub) GetNet(ctx context.Context, arg0 zos.NetID) (ret0 net.IPNet, ret1 error) {
147147
args := []interface{}{arg0}
148148
result, err := s.client.RequestContext(ctx, s.module, s.object, "GetNet", args...)
149149
if err != nil {
@@ -160,7 +160,7 @@ func (s *NetworkerLightStub) GetNet(ctx context.Context, arg0 string) (ret0 net.
160160
return
161161
}
162162

163-
func (s *NetworkerLightStub) GetSubnet(ctx context.Context, arg0 string) (ret0 net.IPNet, ret1 error) {
163+
func (s *NetworkerLightStub) GetSubnet(ctx context.Context, arg0 zos.NetID) (ret0 net.IPNet, ret1 error) {
164164
args := []interface{}{arg0}
165165
result, err := s.client.RequestContext(ctx, s.module, s.object, "GetSubnet", args...)
166166
if err != nil {

0 commit comments

Comments
 (0)