mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
bd63aa3c6b
sed -i '1i/*\nCopyright (c) Edgeless Systems GmbH\n\nSPDX-License-Identifier: AGPL-3.0-only\n*/\n' `grep -rL --include='*.go' 'DO NOT EDIT'` gofumpt -w .
51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package iproute
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
// AddToLocalRoutingTable adds the IP to the local routing table.
|
|
func AddToLocalRoutingTable(ctx context.Context, ip string) error {
|
|
return manipulateLocalRoutingTable(ctx, "add", ip)
|
|
}
|
|
|
|
// RemoveFromLocalRoutingTable removes the IPfrom the local routing table.
|
|
func RemoveFromLocalRoutingTable(ctx context.Context, ip string) error {
|
|
return manipulateLocalRoutingTable(ctx, "del", ip)
|
|
}
|
|
|
|
func manipulateLocalRoutingTable(ctx context.Context, action string, ip string) error {
|
|
// https://github.com/GoogleCloudPlatform/guest-agent/blob/792fce795218633bcbde505fb3457a0b24f26d37/google_guest_agent/addresses.go#L179
|
|
if !strings.Contains(ip, "/") {
|
|
ip = ip + "/32"
|
|
}
|
|
|
|
args := []string{"route", action, "to", "local", ip, "scope", "host", "dev", "ens3", "proto", "66"}
|
|
_, err := exec.CommandContext(ctx, "ip", args...).Output()
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
var exitErr *exec.ExitError
|
|
if !errors.As(err, &exitErr) {
|
|
return fmt.Errorf("ip route %s: %w", action, err)
|
|
}
|
|
if exitErr.ExitCode() == 2 {
|
|
// "RTNETLINK answers: File exists" or "RTNETLINK answers: No such process"
|
|
//
|
|
// Ignore, expected in case of adding an existing route or deleting a route
|
|
// that does not exist.
|
|
return nil
|
|
}
|
|
return fmt.Errorf("ip route %s (code %v) with: %s", action, exitErr.ExitCode(), exitErr.Stderr)
|
|
}
|