57 lines
1 KiB
Go
57 lines
1 KiB
Go
// Copyright 2015 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// +build windows
|
|
|
|
package sspi
|
|
|
|
import (
|
|
"io"
|
|
"unsafe"
|
|
)
|
|
|
|
func (b *SecBuffer) Set(buftype uint32, data []byte) {
|
|
b.BufferType = buftype
|
|
if len(data) > 0 {
|
|
b.Buffer = &data[0]
|
|
b.BufferSize = uint32(len(data))
|
|
} else {
|
|
b.Buffer = nil
|
|
b.BufferSize = 0
|
|
}
|
|
}
|
|
|
|
func (b *SecBuffer) Free() error {
|
|
if b.Buffer == nil {
|
|
return nil
|
|
}
|
|
return FreeContextBuffer((*byte)(unsafe.Pointer(b.Buffer)))
|
|
}
|
|
|
|
func (b *SecBuffer) Bytes() []byte {
|
|
if b.Buffer == nil || b.BufferSize <= 0 {
|
|
return nil
|
|
}
|
|
return (*[2 << 20]byte)(unsafe.Pointer(b.Buffer))[:b.BufferSize]
|
|
}
|
|
|
|
func (b *SecBuffer) WriteAll(w io.Writer) (int, error) {
|
|
if b.BufferSize == 0 || b.Buffer == nil {
|
|
return 0, nil
|
|
}
|
|
data := b.Bytes()
|
|
total := 0
|
|
for {
|
|
n, err := w.Write(data)
|
|
total += n
|
|
if err != nil {
|
|
return total, err
|
|
}
|
|
if n >= len(data) {
|
|
break
|
|
}
|
|
data = data[n:]
|
|
}
|
|
return total, nil
|
|
}
|