58 lines
1.0 KiB
Go
58 lines
1.0 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
|
||
|
|
}
|