Add missing file

cmd/cni-ipvlan-vpc-k8s-tool/cni-ipvlan-vpc-k8s-tool.go

@@ -0,0 +1,258 @@
+package main
+
+import (
+	"fmt"
+	"net"
+	"os"
+	"text/tabwriter"
+
+	"github.com/urfave/cli"
+
+	"github.com/lyft/cni-ipvlan-vpc-k8s"
+	"github.com/lyft/cni-ipvlan-vpc-k8s/aws"
+	"github.com/lyft/cni-ipvlan-vpc-k8s/nl"
+)
+
+var version string
+
+func actionNewInterface(c *cli.Context) error {
+	return cniipvlanvpck8s.LockfileRun(func() error {
+		secGrps := c.Args()
+
+		if len(secGrps) <= 0 {
+			fmt.Println("please specify security groups")
+			return fmt.Errorf("need security groups")
+		}
+		newIf, err := aws.NewInterface(secGrps, nil)
+		if err != nil {
+			fmt.Println(err)
+			return err
+		}
+		fmt.Println(newIf)
+		return nil
+
+	})
+}
+
+func actionRemoveInterface(c *cli.Context) error {
+	return cniipvlanvpck8s.LockfileRun(func() error {
+		interfaces := c.Args()
+
+		if len(interfaces) <= 0 {
+			fmt.Println("please specify an interface")
+			return fmt.Errorf("Insufficent Arguments")
+		}
+
+		if err := aws.RemoveInterface(interfaces); err != nil {
+			fmt.Println(err)
+			return err
+		}
+
+		return nil
+	})
+}
+
+func actionDeallocate(c *cli.Context) error {
+	return cniipvlanvpck8s.LockfileRun(func() error {
+		releaseIps := c.Args()
+		for _, toRelease := range releaseIps {
+
+			if len(toRelease) < 6 {
+				fmt.Println("please specify an IP")
+				return fmt.Errorf("Invalid IP")
+			}
+
+			ip := net.ParseIP(toRelease)
+			if ip == nil {
+				fmt.Println("please specify a valid IP")
+				return fmt.Errorf("IP parse error")
+			}
+
+			err := aws.DeallocateIP(&ip)
+			if err != nil {
+				fmt.Printf("deallocation failed: %v\n", err)
+				return err
+			}
+		}
+		return nil
+	})
+}
+
+func actionAllocate(c *cli.Context) error {
+	return cniipvlanvpck8s.LockfileRun(func() error {
+		index := c.Int("index")
+		res, err := aws.AllocateIPFirstAvailableAtIndex(index)
+		if err != nil {
+			fmt.Println(err)
+			return err
+		}
+
+		fmt.Printf("allocated %v on %v\n", res.IP, res.Interface.LocalName())
+		return nil
+
+	})
+}
+
+func actionFreeIps(c *cli.Context) error {
+	ips, err := cniipvlanvpck8s.FindFreeIPsAtIndex(0)
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 3, ' ', 0)
+	fmt.Fprintln(w, "adapter\tip\t")
+	for _, ip := range ips {
+		fmt.Fprintf(w, "%v\t%v\t\n",
+			ip.Interface.LocalName(),
+			ip.IP)
+	}
+	w.Flush()
+	return nil
+}
+
+func actionLimits(c *cli.Context) error {
+	limit := aws.ENILimits()
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 3, ' ', 0)
+	fmt.Fprintln(w, "adapters\tipv4\tipv6\t")
+	fmt.Fprintf(w, "%v\t%v\t%v\t\n", limit.Adapters,
+		limit.IPv4,
+		limit.IPv6)
+	w.Flush()
+	return nil
+}
+
+func actionAddr(c *cli.Context) error {
+	ips, err := nl.GetIPs()
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 3, ' ', 0)
+	fmt.Fprintln(w, "iface\tip\t")
+	for _, ip := range ips {
+		fmt.Fprintf(w, "%v\t%v\t\n",
+			ip.Label,
+			ip.IPNet.IP)
+	}
+	w.Flush()
+
+	return nil
+}
+
+func actionEniIf(c *cli.Context) error {
+	interfaces, err := aws.GetInterfaces()
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 3, ' ', 0)
+	fmt.Fprintln(w, "iface\tmac\tid\tsubnet\tsubnet_cidr\tsecgrps\tvpc\tips\t")
+	for _, iface := range interfaces {
+		fmt.Fprintf(w, "%v\t%v\t%v\t%v\t%v\t%v\t%v\t%v\t\n", iface.LocalName(),
+			iface.Mac,
+			iface.ID,
+			iface.SubnetID,
+			iface.SubnetCidr,
+			iface.SecurityGroupIds,
+			iface.VpcID,
+			iface.IPv4s)
+
+	}
+
+	w.Flush()
+	return nil
+}
+
+func actionSubnets(c *cli.Context) error {
+	subnets, err := aws.GetSubnetsForInstance()
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+	w := tabwriter.NewWriter(os.Stdout, 0, 0, 3, ' ', 0)
+	fmt.Fprintln(w, "id\tcidr\tdefault\taddresses_available\ttags\t")
+	for _, subnet := range subnets {
+		fmt.Fprintf(w, "%v\t%v\t%v\t%v\t%v\t\n",
+			subnet.ID,
+			subnet.Cidr,
+			subnet.IsDefault,
+			subnet.AvailableAddressCount,
+			subnet.Tags)
+	}
+
+	w.Flush()
+
+	return nil
+}
+
+func main() {
+	if !aws.Available() {
+		fmt.Fprintln(os.Stderr, "This command must be run from a running ec2 instance")
+		os.Exit(1)
+	}
+
+	if os.Getuid() != 0 {
+		fmt.Fprintln(os.Stderr, "This command must be run as root")
+		os.Exit(1)
+	}
+
+	app := cli.NewApp()
+	app.Commands = []cli.Command{
+		{
+			Name:      "new-interface",
+			Usage:     "Create a new interface",
+			Action:    actionNewInterface,
+			ArgsUsage: "[security_group_ids...]",
+		},
+		{
+			Name:      "remove-interface",
+			Usage:     "Remove an existing interface",
+			Action:    actionRemoveInterface,
+			ArgsUsage: "[interface_id...]",
+		},
+		{
+			Name:      "deallocate",
+			Usage:     "Deallocate a private IP",
+			Action:    actionDeallocate,
+			ArgsUsage: "[ip...]",
+		},
+		{
+			Name:   "allocate-first-available",
+			Usage:  "Allocate a private IP on the first available interface",
+			Action: actionAllocate,
+			Flags: []cli.Flag{
+				cli.IntFlag{Name: "index"},
+			},
+		},
+		{
+			Name:   "free-ips",
+			Usage:  "List all currently unassigned AWS IP addresses",
+			Action: actionFreeIps,
+		},
+		{
+			Name:   "eniif",
+			Usage:  "List all ENI interfaces and their setup with addresses",
+			Action: actionEniIf,
+		},
+		{
+			Name:   "addr",
+			Usage:  "List all bound IP addresses",
+			Action: actionAddr,
+		},
+		{
+			Name:   "subnets",
+			Usage:  "Show available subnets for this host",
+			Action: actionSubnets,
+		},
+		{
+			Name:   "limits",
+			Usage:  "Display limits for ENI for this instance type",
+			Action: actionLimits,
+		},
+	}
+	app.Version = version
+	app.Copyright = "(c) 2017 Lyft Inc."
+	app.Usage = "Interface with ENI adapters and CNI bindings for those"
+	app.Run(os.Args)
+}