Browse code

Add support for FreeBSD in portallocator

Signed-off-by: Tomoya Tabuchi <t@tomoyat1.com>

Tomoya Tabuchi authored on 2017/04/29 14:55:10
Showing 1 changed files
1 1
new file mode 100644
... ...
@@ -0,0 +1,42 @@
0
+package portallocator
1
+
2
+import (
3
+	"bytes"
4
+	"fmt"
5
+	"os/exec"
6
+)
7
+
8
+func getDynamicPortRange() (start int, end int, err error) {
9
+	portRangeKernelSysctl := []string{"net.inet.ip.portrange.hifirst", "net.ip.portrange.hilast"}
10
+	portRangeFallback := fmt.Sprintf("using fallback port range %d-%d", DefaultPortRangeStart, DefaultPortRangeEnd)
11
+	portRangeLowCmd := exec.Command("/sbin/sysctl", portRangeKernelSysctl[0])
12
+	var portRangeLowOut bytes.Buffer
13
+	portRangeLowCmd.Stdout = &portRangeLowOut
14
+	cmdErr := portRangeLowCmd.Run()
15
+	if cmdErr != nil {
16
+		return 0, 0, fmt.Errorf("port allocator - sysctl net.inet.ip.portrange.hifirst failed - %s: %v", portRangeFallback, err)
17
+	}
18
+	n, err := fmt.Sscanf(portRangeLowOut.String(), "%d", &start)
19
+	if n != 1 || err != nil {
20
+		if err == nil {
21
+			err = fmt.Errorf("unexpected count of parsed numbers (%d)", n)
22
+		}
23
+		return 0, 0, fmt.Errorf("port allocator - failed to parse system ephemeral port range start from %s - %s: %v", portRangeLowOut.String(), portRangeFallback, err)
24
+	}
25
+
26
+	portRangeHighCmd := exec.Command("/sbin/sysctl", portRangeKernelSysctl[1])
27
+	var portRangeHighOut bytes.Buffer
28
+	portRangeHighCmd.Stdout = &portRangeHighOut
29
+	cmdErr = portRangeHighCmd.Run()
30
+	if cmdErr != nil {
31
+		return 0, 0, fmt.Errorf("port allocator - sysctl net.inet.ip.portrange.hilast failed - %s: %v", portRangeFallback, err)
32
+	}
33
+	n, err = fmt.Sscanf(portRangeHighOut.String(), "%d", &end)
34
+	if n != 1 || err != nil {
35
+		if err == nil {
36
+			err = fmt.Errorf("unexpected count of parsed numbers (%d)", n)
37
+		}
38
+		return 0, 0, fmt.Errorf("port allocator - failed to parse system ephemeral port range end from %s - %s: %v", portRangeHighOut.String(), portRangeFallback, err)
39
+	}
40
+	return start, end, nil
41
+}