mirror of
https://github.com/jfjallid/go-smb
synced 2026-06-08 15:02:50 +00:00
Jimmy Fjällid
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
// MIT License
|
|
//
|
|
// # Copyright (c) 2017 stacktitan
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in all
|
|
// copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
// SOFTWARE.
|
|
package encoder
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"errors"
|
|
"unicode/utf16"
|
|
)
|
|
|
|
func Utf16ToUtf8(in []uint16) []byte {
|
|
t := make([]byte, 0, len(in)*2)
|
|
for i := 0; i < len(in); i++ {
|
|
t = binary.LittleEndian.AppendUint16(t, in[i])
|
|
}
|
|
return t
|
|
}
|
|
|
|
func Utf8ToUtf16(in []byte) ([]uint16, error) {
|
|
if len(in)%2 != 0 {
|
|
return nil, errors.New("Uneven length of UTF8 array")
|
|
}
|
|
t := make([]uint16, len(in)/2)
|
|
for i := 0; i < len(t); i++ {
|
|
t[i] = binary.LittleEndian.Uint16(in[i*2:])
|
|
}
|
|
return t, nil
|
|
}
|
|
|
|
func FromUnicodeString(d []byte) (string, error) {
|
|
// Credit to https://github.com/Azure/go-ntlmssp/blob/master/unicode.go for logic
|
|
if len(d)%2 > 0 {
|
|
return "", errors.New("Unicode (UTF 16 LE) specified, but uneven data length")
|
|
}
|
|
s := make([]uint16, len(d)/2)
|
|
err := binary.Read(bytes.NewReader(d), binary.LittleEndian, &s)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(utf16.Decode(s)), nil
|
|
}
|
|
|
|
func FromUnicode(d []byte) ([]byte, error) {
|
|
s, err := FromUnicodeString(d)
|
|
return []byte(s), err
|
|
}
|
|
|
|
func ToUnicode(s string) []byte {
|
|
// Credit to https://github.com/Azure/go-ntlmssp/blob/master/unicode.go for logic
|
|
uints := utf16.Encode([]rune(s))
|
|
b := bytes.Buffer{}
|
|
binary.Write(&b, binary.LittleEndian, &uints)
|
|
return b.Bytes()
|
|
}
|