mirror of https://github.com/k3s-io/k3s
68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package systemd
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/bits-and-blooms/bitset"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// RangeToBits converts a text representation of a CPU mask (as written to
|
|
// or read from cgroups' cpuset.* files, e.g. "1,3-5") to a slice of bytes
|
|
// with the corresponding bits set (as consumed by systemd over dbus as
|
|
// AllowedCPUs/AllowedMemoryNodes unit property value).
|
|
func RangeToBits(str string) ([]byte, error) {
|
|
bits := &bitset.BitSet{}
|
|
|
|
for _, r := range strings.Split(str, ",") {
|
|
// allow extra spaces around
|
|
r = strings.TrimSpace(r)
|
|
// allow empty elements (extra commas)
|
|
if r == "" {
|
|
continue
|
|
}
|
|
ranges := strings.SplitN(r, "-", 2)
|
|
if len(ranges) > 1 {
|
|
start, err := strconv.ParseUint(ranges[0], 10, 32)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
end, err := strconv.ParseUint(ranges[1], 10, 32)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if start > end {
|
|
return nil, errors.New("invalid range: " + r)
|
|
}
|
|
for i := uint(start); i <= uint(end); i++ {
|
|
bits.Set(i)
|
|
}
|
|
} else {
|
|
val, err := strconv.ParseUint(ranges[0], 10, 32)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bits.Set(uint(val))
|
|
}
|
|
}
|
|
|
|
val := bits.Bytes()
|
|
if len(val) == 0 {
|
|
// do not allow empty values
|
|
return nil, errors.New("empty value")
|
|
}
|
|
ret := make([]byte, len(val)*8)
|
|
for i := range val {
|
|
// bitset uses BigEndian internally
|
|
binary.BigEndian.PutUint64(ret[i*8:], val[len(val)-1-i])
|
|
}
|
|
// remove upper all-zero bytes
|
|
for ret[0] == 0 {
|
|
ret = ret[1:]
|
|
}
|
|
|
|
return ret, nil
|
|
}
|