123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773 |
- package libnetwork
- import (
- "fmt"
- "io"
- "io/ioutil"
- "net"
- "os"
- "os/exec"
- "path/filepath"
- "runtime"
- "strconv"
- "strings"
- "sync"
- "syscall"
- "github.com/Sirupsen/logrus"
- "github.com/docker/docker/pkg/reexec"
- "github.com/docker/libnetwork/iptables"
- "github.com/docker/libnetwork/ipvs"
- "github.com/docker/libnetwork/ns"
- "github.com/gogo/protobuf/proto"
- "github.com/vishvananda/netlink/nl"
- "github.com/vishvananda/netns"
- )
- func init() {
- reexec.Register("fwmarker", fwMarker)
- reexec.Register("redirecter", redirecter)
- }
- // Get all loadbalancers on this network that is currently discovered
- // on this node.
- func (n *network) connectedLoadbalancers() []*loadBalancer {
- c := n.getController()
- c.Lock()
- serviceBindings := make([]*service, 0, len(c.serviceBindings))
- for _, s := range c.serviceBindings {
- serviceBindings = append(serviceBindings, s)
- }
- c.Unlock()
- var lbs []*loadBalancer
- for _, s := range serviceBindings {
- s.Lock()
- if lb, ok := s.loadBalancers[n.ID()]; ok {
- lbs = append(lbs, lb)
- }
- s.Unlock()
- }
- return lbs
- }
- // Populate all loadbalancers on the network that the passed endpoint
- // belongs to, into this sandbox.
- func (sb *sandbox) populateLoadbalancers(ep *endpoint) {
- var gwIP net.IP
- // This is an interface less endpoint. Nothing to do.
- if ep.Iface() == nil {
- return
- }
- n := ep.getNetwork()
- eIP := ep.Iface().Address()
- if n.ingress {
- if err := addRedirectRules(sb.Key(), eIP, ep.ingressPorts); err != nil {
- logrus.Errorf("Failed to add redirect rules for ep %s (%s): %v", ep.Name(), ep.ID()[0:7], err)
- }
- }
- if sb.ingress {
- // For the ingress sandbox if this is not gateway
- // endpoint do nothing.
- if ep != sb.getGatewayEndpoint() {
- return
- }
- // This is the gateway endpoint. Now get the ingress
- // network and plumb the loadbalancers.
- gwIP = ep.Iface().Address().IP
- for _, ep := range sb.getConnectedEndpoints() {
- if !ep.endpointInGWNetwork() {
- n = ep.getNetwork()
- eIP = ep.Iface().Address()
- }
- }
- }
- for _, lb := range n.connectedLoadbalancers() {
- // Skip if vip is not valid.
- if len(lb.vip) == 0 {
- continue
- }
- lb.service.Lock()
- for _, ip := range lb.backEnds {
- sb.addLBBackend(ip, lb.vip, lb.fwMark, lb.service.ingressPorts, eIP, gwIP, n.ingress)
- }
- lb.service.Unlock()
- }
- }
- // Add loadbalancer backend to all sandboxes which has a connection to
- // this network. If needed add the service as well.
- func (n *network) addLBBackend(ip, vip net.IP, fwMark uint32, ingressPorts []*PortConfig) {
- n.WalkEndpoints(func(e Endpoint) bool {
- ep := e.(*endpoint)
- if sb, ok := ep.getSandbox(); ok {
- if !sb.isEndpointPopulated(ep) {
- return false
- }
- var gwIP net.IP
- if ep := sb.getGatewayEndpoint(); ep != nil {
- gwIP = ep.Iface().Address().IP
- }
- sb.addLBBackend(ip, vip, fwMark, ingressPorts, ep.Iface().Address(), gwIP, n.ingress)
- }
- return false
- })
- }
- // Remove loadbalancer backend from all sandboxes which has a
- // connection to this network. If needed remove the service entry as
- // well, as specified by the rmService bool.
- func (n *network) rmLBBackend(ip, vip net.IP, fwMark uint32, ingressPorts []*PortConfig, rmService bool) {
- n.WalkEndpoints(func(e Endpoint) bool {
- ep := e.(*endpoint)
- if sb, ok := ep.getSandbox(); ok {
- if !sb.isEndpointPopulated(ep) {
- return false
- }
- var gwIP net.IP
- if ep := sb.getGatewayEndpoint(); ep != nil {
- gwIP = ep.Iface().Address().IP
- }
- sb.rmLBBackend(ip, vip, fwMark, ingressPorts, ep.Iface().Address(), gwIP, rmService, n.ingress)
- }
- return false
- })
- }
- // Add loadbalancer backend into one connected sandbox.
- func (sb *sandbox) addLBBackend(ip, vip net.IP, fwMark uint32, ingressPorts []*PortConfig, eIP *net.IPNet, gwIP net.IP, isIngressNetwork bool) {
- if sb.osSbox == nil {
- return
- }
- if isIngressNetwork && !sb.ingress {
- return
- }
- i, err := ipvs.New(sb.Key())
- if err != nil {
- logrus.Errorf("Failed to create an ipvs handle for sbox %s (%s,%s) for lb addition: %v", sb.ID()[0:7], sb.ContainerID()[0:7], sb.Key(), err)
- return
- }
- defer i.Close()
- s := &ipvs.Service{
- AddressFamily: nl.FAMILY_V4,
- FWMark: fwMark,
- SchedName: ipvs.RoundRobin,
- }
- if !i.IsServicePresent(s) {
- var filteredPorts []*PortConfig
- if sb.ingress {
- filteredPorts = filterPortConfigs(ingressPorts, false)
- if err := programIngress(gwIP, filteredPorts, false); err != nil {
- logrus.Errorf("Failed to add ingress: %v", err)
- return
- }
- }
- logrus.Debugf("Creating service for vip %s fwMark %d ingressPorts %#v in sbox %s (%s)", vip, fwMark, ingressPorts, sb.ID()[0:7], sb.ContainerID()[0:7])
- if err := invokeFWMarker(sb.Key(), vip, fwMark, ingressPorts, eIP, false); err != nil {
- logrus.Errorf("Failed to add firewall mark rule in sbox %s (%s): %v", sb.ID()[0:7], sb.ContainerID()[0:7], err)
- return
- }
- if err := i.NewService(s); err != nil && err != syscall.EEXIST {
- logrus.Errorf("Failed to create a new service for vip %s fwmark %d in sbox %s (%s): %v", vip, fwMark, sb.ID()[0:7], sb.ContainerID()[0:7], err)
- return
- }
- }
- d := &ipvs.Destination{
- AddressFamily: nl.FAMILY_V4,
- Address: ip,
- Weight: 1,
- }
- // Remove the sched name before using the service to add
- // destination.
- s.SchedName = ""
- if err := i.NewDestination(s, d); err != nil && err != syscall.EEXIST {
- logrus.Errorf("Failed to create real server %s for vip %s fwmark %d in sbox %s (%s): %v", ip, vip, fwMark, sb.ID()[0:7], sb.ContainerID()[0:7], err)
- }
- }
- // Remove loadbalancer backend from one connected sandbox.
- func (sb *sandbox) rmLBBackend(ip, vip net.IP, fwMark uint32, ingressPorts []*PortConfig, eIP *net.IPNet, gwIP net.IP, rmService bool, isIngressNetwork bool) {
- if sb.osSbox == nil {
- return
- }
- if isIngressNetwork && !sb.ingress {
- return
- }
- i, err := ipvs.New(sb.Key())
- if err != nil {
- logrus.Errorf("Failed to create an ipvs handle for sbox %s (%s,%s) for lb removal: %v", sb.ID()[0:7], sb.ContainerID()[0:7], sb.Key(), err)
- return
- }
- defer i.Close()
- s := &ipvs.Service{
- AddressFamily: nl.FAMILY_V4,
- FWMark: fwMark,
- }
- d := &ipvs.Destination{
- AddressFamily: nl.FAMILY_V4,
- Address: ip,
- Weight: 1,
- }
- if err := i.DelDestination(s, d); err != nil && err != syscall.ENOENT {
- logrus.Errorf("Failed to delete real server %s for vip %s fwmark %d in sbox %s (%s): %v", ip, vip, fwMark, sb.ID()[0:7], sb.ContainerID()[0:7], err)
- }
- if rmService {
- s.SchedName = ipvs.RoundRobin
- if err := i.DelService(s); err != nil && err != syscall.ENOENT {
- logrus.Errorf("Failed to delete service for vip %s fwmark %d in sbox %s (%s): %v", vip, fwMark, sb.ID()[0:7], sb.ContainerID()[0:7], err)
- }
- var filteredPorts []*PortConfig
- if sb.ingress {
- filteredPorts = filterPortConfigs(ingressPorts, true)
- if err := programIngress(gwIP, filteredPorts, true); err != nil {
- logrus.Errorf("Failed to delete ingress: %v", err)
- }
- }
- if err := invokeFWMarker(sb.Key(), vip, fwMark, ingressPorts, eIP, true); err != nil {
- logrus.Errorf("Failed to delete firewall mark rule in sbox %s (%s): %v", sb.ID()[0:7], sb.ContainerID()[0:7], err)
- }
- }
- }
- const ingressChain = "DOCKER-INGRESS"
- var (
- ingressOnce sync.Once
- ingressProxyMu sync.Mutex
- ingressProxyTbl = make(map[string]io.Closer)
- portConfigMu sync.Mutex
- portConfigTbl = make(map[PortConfig]int)
- )
- func filterPortConfigs(ingressPorts []*PortConfig, isDelete bool) []*PortConfig {
- portConfigMu.Lock()
- iPorts := make([]*PortConfig, 0, len(ingressPorts))
- for _, pc := range ingressPorts {
- if isDelete {
- if cnt, ok := portConfigTbl[*pc]; ok {
- // This is the last reference to this
- // port config. Delete the port config
- // and add it to filtered list to be
- // plumbed.
- if cnt == 1 {
- delete(portConfigTbl, *pc)
- iPorts = append(iPorts, pc)
- continue
- }
- portConfigTbl[*pc] = cnt - 1
- }
- continue
- }
- if cnt, ok := portConfigTbl[*pc]; ok {
- portConfigTbl[*pc] = cnt + 1
- continue
- }
- // We are adding it for the first time. Add it to the
- // filter list to be plumbed.
- portConfigTbl[*pc] = 1
- iPorts = append(iPorts, pc)
- }
- portConfigMu.Unlock()
- return iPorts
- }
- func programIngress(gwIP net.IP, ingressPorts []*PortConfig, isDelete bool) error {
- addDelOpt := "-I"
- if isDelete {
- addDelOpt = "-D"
- }
- chainExists := iptables.ExistChain(ingressChain, iptables.Nat)
- filterChainExists := iptables.ExistChain(ingressChain, iptables.Filter)
- ingressOnce.Do(func() {
- // Flush nat table and filter table ingress chain rules during init if it
- // exists. It might contain stale rules from previous life.
- if chainExists {
- if err := iptables.RawCombinedOutput("-t", "nat", "-F", ingressChain); err != nil {
- logrus.Errorf("Could not flush nat table ingress chain rules during init: %v", err)
- }
- }
- if filterChainExists {
- if err := iptables.RawCombinedOutput("-F", ingressChain); err != nil {
- logrus.Errorf("Could not flush filter table ingress chain rules during init: %v", err)
- }
- }
- })
- if !isDelete {
- if !chainExists {
- if err := iptables.RawCombinedOutput("-t", "nat", "-N", ingressChain); err != nil {
- return fmt.Errorf("failed to create ingress chain: %v", err)
- }
- }
- if !filterChainExists {
- if err := iptables.RawCombinedOutput("-N", ingressChain); err != nil {
- return fmt.Errorf("failed to create filter table ingress chain: %v", err)
- }
- }
- if !iptables.Exists(iptables.Nat, ingressChain, "-j", "RETURN") {
- if err := iptables.RawCombinedOutput("-t", "nat", "-A", ingressChain, "-j", "RETURN"); err != nil {
- return fmt.Errorf("failed to add return rule in nat table ingress chain: %v", err)
- }
- }
- if !iptables.Exists(iptables.Filter, ingressChain, "-j", "RETURN") {
- if err := iptables.RawCombinedOutput("-A", ingressChain, "-j", "RETURN"); err != nil {
- return fmt.Errorf("failed to add return rule to filter table ingress chain: %v", err)
- }
- }
- for _, chain := range []string{"OUTPUT", "PREROUTING"} {
- if !iptables.Exists(iptables.Nat, chain, "-m", "addrtype", "--dst-type", "LOCAL", "-j", ingressChain) {
- if err := iptables.RawCombinedOutput("-t", "nat", "-I", chain, "-m", "addrtype", "--dst-type", "LOCAL", "-j", ingressChain); err != nil {
- return fmt.Errorf("failed to add jump rule in %s to ingress chain: %v", chain, err)
- }
- }
- }
- if !iptables.Exists(iptables.Filter, "FORWARD", "-j", ingressChain) {
- if err := iptables.RawCombinedOutput("-I", "FORWARD", "-j", ingressChain); err != nil {
- return fmt.Errorf("failed to add jump rule to %s in filter table forward chain: %v", ingressChain, err)
- }
- }
- oifName, err := findOIFName(gwIP)
- if err != nil {
- return fmt.Errorf("failed to find gateway bridge interface name for %s: %v", gwIP, err)
- }
- path := filepath.Join("/proc/sys/net/ipv4/conf", oifName, "route_localnet")
- if err := ioutil.WriteFile(path, []byte{'1', '\n'}, 0644); err != nil {
- return fmt.Errorf("could not write to %s: %v", path, err)
- }
- ruleArgs := strings.Fields(fmt.Sprintf("-m addrtype --src-type LOCAL -o %s -j MASQUERADE", oifName))
- if !iptables.Exists(iptables.Nat, "POSTROUTING", ruleArgs...) {
- if err := iptables.RawCombinedOutput(append([]string{"-t", "nat", "-I", "POSTROUTING"}, ruleArgs...)...); err != nil {
- return fmt.Errorf("failed to add ingress localhost POSTROUTING rule for %s: %v", oifName, err)
- }
- }
- }
- for _, iPort := range ingressPorts {
- if iptables.ExistChain(ingressChain, iptables.Nat) {
- rule := strings.Fields(fmt.Sprintf("-t nat %s %s -p %s --dport %d -j DNAT --to-destination %s:%d",
- addDelOpt, ingressChain, strings.ToLower(PortConfig_Protocol_name[int32(iPort.Protocol)]), iPort.PublishedPort, gwIP, iPort.PublishedPort))
- if err := iptables.RawCombinedOutput(rule...); err != nil {
- errStr := fmt.Sprintf("setting up rule failed, %v: %v", rule, err)
- if !isDelete {
- return fmt.Errorf("%s", errStr)
- }
- logrus.Infof("%s", errStr)
- }
- }
- // Filter table rules to allow a published service to be accessible in the local node from..
- // 1) service tasks attached to other networks
- // 2) unmanaged containers on bridge networks
- rule := strings.Fields(fmt.Sprintf("%s %s -m state -p %s --sport %d --state ESTABLISHED,RELATED -j ACCEPT",
- addDelOpt, ingressChain, strings.ToLower(PortConfig_Protocol_name[int32(iPort.Protocol)]), iPort.PublishedPort))
- if err := iptables.RawCombinedOutput(rule...); err != nil {
- errStr := fmt.Sprintf("setting up rule failed, %v: %v", rule, err)
- if !isDelete {
- return fmt.Errorf("%s", errStr)
- }
- logrus.Warnf("%s", errStr)
- }
- rule = strings.Fields(fmt.Sprintf("%s %s -p %s --dport %d -j ACCEPT",
- addDelOpt, ingressChain, strings.ToLower(PortConfig_Protocol_name[int32(iPort.Protocol)]), iPort.PublishedPort))
- if err := iptables.RawCombinedOutput(rule...); err != nil {
- errStr := fmt.Sprintf("setting up rule failed, %v: %v", rule, err)
- if !isDelete {
- return fmt.Errorf("%s", errStr)
- }
- logrus.Warnf("%s", errStr)
- }
- if err := plumbProxy(iPort, isDelete); err != nil {
- logrus.Warnf("failed to create proxy for port %d: %v", iPort.PublishedPort, err)
- }
- }
- return nil
- }
- // In the filter table FORWARD chain first rule should be to jump to INGRESS-CHAIN
- // This chain has the rules to allow access to the published ports for swarm tasks
- // from local bridge networks and docker_gwbridge (ie:taks on other swarm netwroks)
- func arrangeIngressFilterRule() {
- if iptables.ExistChain(ingressChain, iptables.Filter) {
- if iptables.Exists(iptables.Filter, "FORWARD", "-j", ingressChain) {
- if err := iptables.RawCombinedOutput("-D", "FORWARD", "-j", ingressChain); err != nil {
- logrus.Warnf("failed to delete jump rule to ingressChain in filter table: %v", err)
- }
- }
- if err := iptables.RawCombinedOutput("-I", "FORWARD", "-j", ingressChain); err != nil {
- logrus.Warnf("failed to add jump rule to ingressChain in filter table: %v", err)
- }
- }
- }
- func findOIFName(ip net.IP) (string, error) {
- nlh := ns.NlHandle()
- routes, err := nlh.RouteGet(ip)
- if err != nil {
- return "", err
- }
- if len(routes) == 0 {
- return "", fmt.Errorf("no route to %s", ip)
- }
- // Pick the first route(typically there is only one route). We
- // don't support multipath.
- link, err := nlh.LinkByIndex(routes[0].LinkIndex)
- if err != nil {
- return "", err
- }
- return link.Attrs().Name, nil
- }
- func plumbProxy(iPort *PortConfig, isDelete bool) error {
- var (
- err error
- l io.Closer
- )
- portSpec := fmt.Sprintf("%d/%s", iPort.PublishedPort, strings.ToLower(PortConfig_Protocol_name[int32(iPort.Protocol)]))
- if isDelete {
- ingressProxyMu.Lock()
- if listener, ok := ingressProxyTbl[portSpec]; ok {
- if listener != nil {
- listener.Close()
- }
- }
- ingressProxyMu.Unlock()
- return nil
- }
- switch iPort.Protocol {
- case ProtocolTCP:
- l, err = net.ListenTCP("tcp", &net.TCPAddr{Port: int(iPort.PublishedPort)})
- case ProtocolUDP:
- l, err = net.ListenUDP("udp", &net.UDPAddr{Port: int(iPort.PublishedPort)})
- }
- if err != nil {
- return err
- }
- ingressProxyMu.Lock()
- ingressProxyTbl[portSpec] = l
- ingressProxyMu.Unlock()
- return nil
- }
- func writePortsToFile(ports []*PortConfig) (string, error) {
- f, err := ioutil.TempFile("", "port_configs")
- if err != nil {
- return "", err
- }
- defer f.Close()
- buf, err := proto.Marshal(&EndpointRecord{
- IngressPorts: ports,
- })
- n, err := f.Write(buf)
- if err != nil {
- return "", err
- }
- if n < len(buf) {
- return "", io.ErrShortWrite
- }
- return f.Name(), nil
- }
- func readPortsFromFile(fileName string) ([]*PortConfig, error) {
- buf, err := ioutil.ReadFile(fileName)
- if err != nil {
- return nil, err
- }
- var epRec EndpointRecord
- err = proto.Unmarshal(buf, &epRec)
- if err != nil {
- return nil, err
- }
- return epRec.IngressPorts, nil
- }
- // Invoke fwmarker reexec routine to mark vip destined packets with
- // the passed firewall mark.
- func invokeFWMarker(path string, vip net.IP, fwMark uint32, ingressPorts []*PortConfig, eIP *net.IPNet, isDelete bool) error {
- var ingressPortsFile string
- if len(ingressPorts) != 0 {
- var err error
- ingressPortsFile, err = writePortsToFile(ingressPorts)
- if err != nil {
- return err
- }
- defer os.Remove(ingressPortsFile)
- }
- addDelOpt := "-A"
- if isDelete {
- addDelOpt = "-D"
- }
- cmd := &exec.Cmd{
- Path: reexec.Self(),
- Args: append([]string{"fwmarker"}, path, vip.String(), fmt.Sprintf("%d", fwMark), addDelOpt, ingressPortsFile, eIP.String()),
- Stdout: os.Stdout,
- Stderr: os.Stderr,
- }
- if err := cmd.Run(); err != nil {
- return fmt.Errorf("reexec failed: %v", err)
- }
- return nil
- }
- // Firewall marker reexec function.
- func fwMarker() {
- runtime.LockOSThread()
- defer runtime.UnlockOSThread()
- if len(os.Args) < 7 {
- logrus.Error("invalid number of arguments..")
- os.Exit(1)
- }
- var ingressPorts []*PortConfig
- if os.Args[5] != "" {
- var err error
- ingressPorts, err = readPortsFromFile(os.Args[5])
- if err != nil {
- logrus.Errorf("Failed reading ingress ports file: %v", err)
- os.Exit(6)
- }
- }
- vip := os.Args[2]
- fwMark, err := strconv.ParseUint(os.Args[3], 10, 32)
- if err != nil {
- logrus.Errorf("bad fwmark value(%s) passed: %v", os.Args[3], err)
- os.Exit(2)
- }
- addDelOpt := os.Args[4]
- rules := [][]string{}
- for _, iPort := range ingressPorts {
- rule := strings.Fields(fmt.Sprintf("-t mangle %s PREROUTING -p %s --dport %d -j MARK --set-mark %d",
- addDelOpt, strings.ToLower(PortConfig_Protocol_name[int32(iPort.Protocol)]), iPort.PublishedPort, fwMark))
- rules = append(rules, rule)
- }
- ns, err := netns.GetFromPath(os.Args[1])
- if err != nil {
- logrus.Errorf("failed get network namespace %q: %v", os.Args[1], err)
- os.Exit(3)
- }
- defer ns.Close()
- if err := netns.Set(ns); err != nil {
- logrus.Errorf("setting into container net ns %v failed, %v", os.Args[1], err)
- os.Exit(4)
- }
- if addDelOpt == "-A" {
- eIP, subnet, err := net.ParseCIDR(os.Args[6])
- if err != nil {
- logrus.Errorf("Failed to parse endpoint IP %s: %v", os.Args[6], err)
- os.Exit(9)
- }
- ruleParams := strings.Fields(fmt.Sprintf("-m ipvs --ipvs -d %s -j SNAT --to-source %s", subnet, eIP))
- if !iptables.Exists("nat", "POSTROUTING", ruleParams...) {
- rule := append(strings.Fields("-t nat -A POSTROUTING"), ruleParams...)
- rules = append(rules, rule)
- err := ioutil.WriteFile("/proc/sys/net/ipv4/vs/conntrack", []byte{'1', '\n'}, 0644)
- if err != nil {
- logrus.Errorf("Failed to write to /proc/sys/net/ipv4/vs/conntrack: %v", err)
- os.Exit(8)
- }
- }
- }
- rule := strings.Fields(fmt.Sprintf("-t mangle %s OUTPUT -d %s/32 -j MARK --set-mark %d", addDelOpt, vip, fwMark))
- rules = append(rules, rule)
- rule = strings.Fields(fmt.Sprintf("-t nat %s OUTPUT -p icmp --icmp echo-request -d %s -j DNAT --to 127.0.0.1", addDelOpt, vip))
- rules = append(rules, rule)
- for _, rule := range rules {
- if err := iptables.RawCombinedOutputNative(rule...); err != nil {
- logrus.Errorf("setting up rule failed, %v: %v", rule, err)
- os.Exit(5)
- }
- }
- }
- func addRedirectRules(path string, eIP *net.IPNet, ingressPorts []*PortConfig) error {
- var ingressPortsFile string
- if len(ingressPorts) != 0 {
- var err error
- ingressPortsFile, err = writePortsToFile(ingressPorts)
- if err != nil {
- return err
- }
- defer os.Remove(ingressPortsFile)
- }
- cmd := &exec.Cmd{
- Path: reexec.Self(),
- Args: append([]string{"redirecter"}, path, eIP.String(), ingressPortsFile),
- Stdout: os.Stdout,
- Stderr: os.Stderr,
- }
- if err := cmd.Run(); err != nil {
- return fmt.Errorf("reexec failed: %v", err)
- }
- return nil
- }
- // Redirecter reexec function.
- func redirecter() {
- runtime.LockOSThread()
- defer runtime.UnlockOSThread()
- if len(os.Args) < 4 {
- logrus.Error("invalid number of arguments..")
- os.Exit(1)
- }
- var ingressPorts []*PortConfig
- if os.Args[3] != "" {
- var err error
- ingressPorts, err = readPortsFromFile(os.Args[3])
- if err != nil {
- logrus.Errorf("Failed reading ingress ports file: %v", err)
- os.Exit(2)
- }
- }
- eIP, _, err := net.ParseCIDR(os.Args[2])
- if err != nil {
- logrus.Errorf("Failed to parse endpoint IP %s: %v", os.Args[2], err)
- os.Exit(3)
- }
- rules := [][]string{}
- for _, iPort := range ingressPorts {
- rule := strings.Fields(fmt.Sprintf("-t nat -A PREROUTING -d %s -p %s --dport %d -j REDIRECT --to-port %d",
- eIP.String(), strings.ToLower(PortConfig_Protocol_name[int32(iPort.Protocol)]), iPort.PublishedPort, iPort.TargetPort))
- rules = append(rules, rule)
- // Allow only incoming connections to exposed ports
- iRule := strings.Fields(fmt.Sprintf("-I INPUT -d %s -p %s --dport %d -m conntrack --ctstate NEW,ESTABLISHED -j ACCEPT",
- eIP.String(), strings.ToLower(PortConfig_Protocol_name[int32(iPort.Protocol)]), iPort.TargetPort))
- rules = append(rules, iRule)
- // Allow only outgoing connections from exposed ports
- oRule := strings.Fields(fmt.Sprintf("-I OUTPUT -s %s -p %s --sport %d -m conntrack --ctstate ESTABLISHED -j ACCEPT",
- eIP.String(), strings.ToLower(PortConfig_Protocol_name[int32(iPort.Protocol)]), iPort.TargetPort))
- rules = append(rules, oRule)
- }
- ns, err := netns.GetFromPath(os.Args[1])
- if err != nil {
- logrus.Errorf("failed get network namespace %q: %v", os.Args[1], err)
- os.Exit(4)
- }
- defer ns.Close()
- if err := netns.Set(ns); err != nil {
- logrus.Errorf("setting into container net ns %v failed, %v", os.Args[1], err)
- os.Exit(5)
- }
- for _, rule := range rules {
- if err := iptables.RawCombinedOutputNative(rule...); err != nil {
- logrus.Errorf("setting up rule failed, %v: %v", rule, err)
- os.Exit(6)
- }
- }
- if len(ingressPorts) == 0 {
- return
- }
- // Ensure blocking rules for anything else in/to ingress network
- for _, rule := range [][]string{
- {"-d", eIP.String(), "-p", "udp", "-j", "DROP"},
- {"-d", eIP.String(), "-p", "tcp", "-j", "DROP"},
- } {
- if !iptables.ExistsNative(iptables.Filter, "INPUT", rule...) {
- if err := iptables.RawCombinedOutputNative(append([]string{"-A", "INPUT"}, rule...)...); err != nil {
- logrus.Errorf("setting up rule failed, %v: %v", rule, err)
- os.Exit(7)
- }
- }
- rule[0] = "-s"
- if !iptables.ExistsNative(iptables.Filter, "OUTPUT", rule...) {
- if err := iptables.RawCombinedOutputNative(append([]string{"-A", "OUTPUT"}, rule...)...); err != nil {
- logrus.Errorf("setting up rule failed, %v: %v", rule, err)
- os.Exit(8)
- }
- }
- }
- }
|