mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-11-23 14:50:24 +00:00
50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
|
package cutil
|
||
|
|
|
||
|
|
import "C"
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
)
|
||
|
|
|
||
|
|
// SplitBuffer splits a byte-slice buffer, typically returned from C code,
|
||
|
|
// into a slice of strings.
|
||
|
|
// The contents of the buffer are assumed to be null-byte separated.
|
||
|
|
// If the buffer contains a sequence of null-bytes it will assume that the
|
||
|
|
// "space" between the bytes are meant to be empty strings.
|
||
|
|
func SplitBuffer(b []byte) []string {
|
||
|
|
return splitBufStrings(b, true)
|
||
|
|
}
|
||
|
|
|
||
|
|
// SplitSparseBuffer splits a byte-slice buffer, typically returned from C code,
|
||
|
|
// into a slice of strings.
|
||
|
|
// The contents of the buffer are assumed to be null-byte separated.
|
||
|
|
// This function assumes that buffer to be "sparse" such that only non-null-byte
|
||
|
|
// strings will be returned, and no "empty" strings exist if null-bytes
|
||
|
|
// are found adjacent to each other.
|
||
|
|
func SplitSparseBuffer(b []byte) []string {
|
||
|
|
return splitBufStrings(b, false)
|
||
|
|
}
|
||
|
|
|
||
|
|
// If keepEmpty is true, empty substrings will be returned, by default they are
|
||
|
|
// excluded from the results.
|
||
|
|
// This is almost certainly a suboptimal implementation, especially for
|
||
|
|
// keepEmpty=true case. Optimizing the functions is a job for another day.
|
||
|
|
func splitBufStrings(b []byte, keepEmpty bool) []string {
|
||
|
|
values := make([]string, 0)
|
||
|
|
// the final null byte should be the terminating null in C
|
||
|
|
// we never want to preserve the empty string after it
|
||
|
|
if len(b) > 0 && b[len(b)-1] == 0 {
|
||
|
|
b = b[:len(b)-1]
|
||
|
|
}
|
||
|
|
if len(b) == 0 {
|
||
|
|
return values
|
||
|
|
}
|
||
|
|
for _, s := range bytes.Split(b, []byte{0}) {
|
||
|
|
if !keepEmpty && len(s) == 0 {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
values = append(values, string(s))
|
||
|
|
}
|
||
|
|
return values
|
||
|
|
}
|