forked from openshift/origin
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ovs.go
126 lines (108 loc) · 3.8 KB
/
ovs.go
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// Package ovs provides a wrapper around ovs-vsctl and ovs-ofctl
package ovs
import (
"fmt"
"strings"
"github.com/openshift/openshift-sdn/pkg/exec"
)
type Transaction struct {
bridge string
err error
}
// NewTransaction begins a new OVS transaction for a given bridge. If an error
// occurs at any step in the transaction, it will be recorded until
// EndTransaction(), and any further calls on the transaction will be ignored.
func NewTransaction(bridge string) *Transaction {
return &Transaction{bridge: bridge}
}
func (tx *Transaction) exec(cmd string, args ...string) (string, error) {
if tx.err != nil {
return "", tx.err
}
cmdpath, err := exec.LookPath(cmd)
if err != nil {
tx.err = fmt.Errorf("OVS is not installed")
return "", tx.err
}
var output string
output, tx.err = exec.Exec(cmdpath, args...)
return output, tx.err
}
func (tx *Transaction) vsctlExec(args ...string) (string, error) {
return tx.exec("ovs-vsctl", args...)
}
func (tx *Transaction) ofctlExec(args ...string) (string, error) {
args = append([]string{"-O", "OpenFlow13"}, args...)
return tx.exec("ovs-ofctl", args...)
}
// AddBridge creates the bridge associated with the transaction, optionally setting
// properties on it (as with "ovs-vsctl set Bridge ..."). If the bridge already
// existed, it will be destroyed and recreated.
func (tx *Transaction) AddBridge(properties ...string) {
args := []string{"--if-exists", "del-br", tx.bridge, "--", "add-br", tx.bridge}
if len(properties) > 0 {
args = append(args, "--", "set", "Bridge", tx.bridge)
args = append(args, properties...)
}
tx.vsctlExec(args...)
}
// DeleteBridge deletes the bridge associated with the transaction. (It is an
// error if the bridge does not exist.)
func (tx *Transaction) DeleteBridge() {
tx.vsctlExec("del-br", tx.bridge)
}
// AddPort adds an interface to the bridge, requesting the indicated port
// number, and optionally setting properties on it (as with "ovs-vsctl set
// Interface ...").
func (tx *Transaction) AddPort(port string, ofport uint, properties ...string) {
args := []string{"--if-exists", "del-port", port, "--", "add-port", tx.bridge, port, "--", "set", "Interface", port, fmt.Sprintf("ofport_request=%d", ofport)}
if len(properties) > 0 {
args = append(args, properties...)
}
tx.vsctlExec(args...)
}
// DeletePort removes an interface from the bridge. (It is an error if the
// interface is not currently a bridge port.)
func (tx *Transaction) DeletePort(port string) {
tx.vsctlExec("del-port", port)
}
// AddFlow adds a flow to the bridge. The arguments are passed to fmt.Sprintf().
func (tx *Transaction) AddFlow(flow string, args ...interface{}) {
if len(args) > 0 {
flow = fmt.Sprintf(flow, args...)
}
tx.ofctlExec("add-flow", tx.bridge, flow)
}
// DeleteFlows deletes all matching flows from the bridge. The arguments are
// passed to fmt.Sprintf().
func (tx *Transaction) DeleteFlows(flow string, args ...interface{}) {
if len(args) > 0 {
flow = fmt.Sprintf(flow, args...)
}
tx.ofctlExec("del-flows", tx.bridge, flow)
}
// DumpFlows dumps the flow table for the bridge and returns it as an array of
// strings, one per flow. Since this function has a return value, it also
// returns an error immediately if an error occurs.
func (tx *Transaction) DumpFlows() ([]string, error) {
out, err := tx.ofctlExec("dump-flows", tx.bridge)
if err != nil {
return nil, err
}
lines := strings.Split(out, "\n")
flows := make([]string, 0, len(lines))
for _, line := range lines {
if strings.Contains(line, "cookie=") {
flows = append(flows, line)
}
}
return flows, nil
}
// EndTransaction ends an OVS transaction and returns any error that occurred
// during the transaction. You should not use the transaction again after
// calling this function.
func (tx *Transaction) EndTransaction() error {
err := tx.err
tx.err = nil
return err
}