3eef755b3b
This tag of hcsshim brings in a couple welcome features/improvements. One being exposing a way to query for hns endpoint statistics (Packets received/sent etc.). This tag also contains some optimizations for querying whether a certain HCN feature is supported, which is a common workflow in kube-proxy on Windows. The first result from querying HCN is now cached so further calls can skip the hcn query as well as the version range parsing that was performed. This also gets rid of some redundant logs that used to hit everytime the version range parsing occurred. The Go-winio dep bump, and all of the ctrd deps are transitive only. Nothing new is needed/intended to be used. Signed-off-by: Daniel Canter <dcanter@microsoft.com>
76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
package winapi
|
|
|
|
import (
|
|
"errors"
|
|
"reflect"
|
|
"syscall"
|
|
"unsafe"
|
|
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
// Uint16BufferToSlice wraps a uint16 pointer-and-length into a slice
|
|
// for easier interop with Go APIs
|
|
func Uint16BufferToSlice(buffer *uint16, bufferLength int) (result []uint16) {
|
|
hdr := (*reflect.SliceHeader)(unsafe.Pointer(&result))
|
|
hdr.Data = uintptr(unsafe.Pointer(buffer))
|
|
hdr.Cap = bufferLength
|
|
hdr.Len = bufferLength
|
|
|
|
return
|
|
}
|
|
|
|
type UnicodeString struct {
|
|
Length uint16
|
|
MaximumLength uint16
|
|
Buffer *uint16
|
|
}
|
|
|
|
//String converts a UnicodeString to a golang string
|
|
func (uni UnicodeString) String() string {
|
|
// UnicodeString is not guaranteed to be null terminated, therefore
|
|
// use the UnicodeString's Length field
|
|
return syscall.UTF16ToString(Uint16BufferToSlice(uni.Buffer, int(uni.Length/2)))
|
|
}
|
|
|
|
// NewUnicodeString allocates a new UnicodeString and copies `s` into
|
|
// the buffer of the new UnicodeString.
|
|
func NewUnicodeString(s string) (*UnicodeString, error) {
|
|
// Get length of original `s` to use in the UnicodeString since the `buf`
|
|
// created later will have an additional trailing null character
|
|
length := len(s)
|
|
if length > 32767 {
|
|
return nil, syscall.ENAMETOOLONG
|
|
}
|
|
|
|
buf, err := windows.UTF16FromString(s)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
uni := &UnicodeString{
|
|
Length: uint16(length * 2),
|
|
MaximumLength: uint16(length * 2),
|
|
Buffer: &buf[0],
|
|
}
|
|
return uni, nil
|
|
}
|
|
|
|
// ConvertStringSetToSlice is a helper function used to convert the contents of
|
|
// `buf` into a string slice. `buf` contains a set of null terminated strings
|
|
// with an additional null at the end to indicate the end of the set.
|
|
func ConvertStringSetToSlice(buf []byte) ([]string, error) {
|
|
var results []string
|
|
prev := 0
|
|
for i := range buf {
|
|
if buf[i] == 0 {
|
|
if prev == i {
|
|
// found two null characters in a row, return result
|
|
return results, nil
|
|
}
|
|
results = append(results, string(buf[prev:i]))
|
|
prev = i + 1
|
|
}
|
|
}
|
|
return nil, errors.New("string set malformed: missing null terminator at end of buffer")
|
|
}
|