57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
// Copyright (c) 2021 Tailscale Inc & AUTHORS All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package dns
|
|
|
|
// This code is only used in Windows builds, but is in an
|
|
// OS-independent file so tests can run all the time.
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"unicode/utf16"
|
|
)
|
|
|
|
// maybeUnUTF16 tries to detect whether bs contains UTF-16, and if so
|
|
// translates it to regular UTF-8.
|
|
//
|
|
// Some of wsl.exe's output get printed as UTF-16, which breaks a
|
|
// bunch of things. Try to detect this by looking for a zero byte in
|
|
// the first few bytes of output (which will appear if any of those
|
|
// codepoints are basic ASCII - very likely). From that we can infer
|
|
// that UTF-16 is being printed, and the byte order in use, and we
|
|
// decode that back to UTF-8.
|
|
//
|
|
// https://github.com/microsoft/WSL/issues/4607
|
|
func maybeUnUTF16(bs []byte) []byte {
|
|
if len(bs)%2 != 0 {
|
|
// Can't be complete UTF-16.
|
|
return bs
|
|
}
|
|
checkLen := 20
|
|
if len(bs) < checkLen {
|
|
checkLen = len(bs)
|
|
}
|
|
zeroOff := bytes.IndexByte(bs[:checkLen], 0)
|
|
if zeroOff == -1 {
|
|
return bs
|
|
}
|
|
|
|
// We assume wsl.exe is trying to print an ASCII codepoint,
|
|
// meaning the zero byte is in the upper 8 bits of the
|
|
// codepoint. That means we can use the zero's byte offset to
|
|
// work out if we're seeing little-endian or big-endian
|
|
// UTF-16.
|
|
var endian binary.ByteOrder = binary.LittleEndian
|
|
if zeroOff%2 == 0 {
|
|
endian = binary.BigEndian
|
|
}
|
|
|
|
var u16 []uint16
|
|
for i := 0; i < len(bs); i += 2 {
|
|
u16 = append(u16, endian.Uint16(bs[i:]))
|
|
}
|
|
return []byte(string(utf16.Decode(u16)))
|
|
}
|