1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
package netproviders
import (
"os"
"strconv"
"github.com/moby/buildkit/solver/pb"
"github.com/moby/buildkit/util/network"
"github.com/moby/buildkit/util/network/cniprovider"
"github.com/pkg/errors"
)
type Opt struct {
CNI cniprovider.Opt
Mode string
}
// Providers returns the network provider set.
// When opt.Mode is "auto" or "", resolvedMode is set to either "cni" or "host".
func Providers(opt Opt) (providers map[pb.NetMode]network.Provider, resolvedMode string, err error) {
var defaultProvider network.Provider
switch opt.Mode {
case "cni":
cniProvider, err := cniprovider.New(opt.CNI)
if err != nil {
return nil, resolvedMode, err
}
defaultProvider = cniProvider
resolvedMode = opt.Mode
case "host":
hostProvider, ok := getHostProvider()
if !ok {
return nil, resolvedMode, errors.New("no host network support on this platform")
}
defaultProvider = hostProvider
resolvedMode = opt.Mode
case "bridge":
bridgeProvider, err := getBridgeProvider(opt.CNI)
if err != nil {
return nil, resolvedMode, err
}
defaultProvider = bridgeProvider
resolvedMode = "cni"
case "auto", "":
if v, err := strconv.ParseBool(os.Getenv("BUILDKIT_NETWORK_BRIDGE_AUTO")); v && err == nil {
bridgeProvider, err := getBridgeProvider(opt.CNI)
if err != nil {
return nil, resolvedMode, err
}
defaultProvider = bridgeProvider
resolvedMode = "cni"
} else if _, err := os.Stat(opt.CNI.ConfigPath); err == nil {
cniProvider, err := cniprovider.New(opt.CNI)
if err != nil {
return nil, resolvedMode, err
}
defaultProvider = cniProvider
resolvedMode = "cni"
} else {
defaultProvider, resolvedMode = getFallback()
}
default:
return nil, resolvedMode, errors.Errorf("invalid network mode: %q", opt.Mode)
}
providers = map[pb.NetMode]network.Provider{
pb.NetMode_UNSET: defaultProvider,
pb.NetMode_NONE: network.NewNoneProvider(),
}
if hostProvider, ok := getHostProvider(); ok {
providers[pb.NetMode_HOST] = hostProvider
}
return providers, resolvedMode, nil
}
|
