Files
swift-mirror/stdlib/public/core/UTF8.swift
Doug Gregor 1a1f79c0de Introduce safety checkin for ConcurrentValue conformance.
Introduce checking of ConcurrentValue conformances:
- For structs, check that each stored property conforms to ConcurrentValue
- For enums, check that each associated value conforms to ConcurrentValue
- For classes, check that each stored property is immutable and conforms
  to ConcurrentValue

Because all of the stored properties / associated values need to be
visible for this check to work, limit ConcurrentValue conformances to
be in the same source file as the type definition.

This checking can be disabled by conforming to a new marker protocol,
UnsafeConcurrentValue, that refines ConcurrentValue.
UnsafeConcurrentValue otherwise his no specific meaning. This allows
both "I know what I'm doing" for types that manage concurrent access
themselves as well as enabling retroactive conformance, both of which
are fundamentally unsafe but also quite necessary.

The bulk of this change ended up being to the standard library, because
all conformances of standard library types to the ConcurrentValue
protocol needed to be sunk down into the standard library so they
would benefit from the checking above. There were numerous little
mistakes in the initial pass through the stsandard library types that
have now been corrected.
2021-02-04 03:45:09 -08:00

328 lines
12 KiB
Swift

//===--- UTF8.swift -------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
extension Unicode {
@frozen
public enum UTF8: ConcurrentValue {
case _swift3Buffer(Unicode.UTF8.ForwardParser)
}
}
extension Unicode.UTF8 {
/// Returns the number of code units required to encode the given Unicode
/// scalar.
///
/// Because a Unicode scalar value can require up to 21 bits to store its
/// value, some Unicode scalars are represented in UTF-8 by a sequence of up
/// to 4 code units. The first code unit is designated a *lead* byte and the
/// rest are *continuation* bytes.
///
/// let anA: Unicode.Scalar = "A"
/// print(anA.value)
/// // Prints "65"
/// print(UTF8.width(anA))
/// // Prints "1"
///
/// let anApple: Unicode.Scalar = "🍎"
/// print(anApple.value)
/// // Prints "127822"
/// print(UTF8.width(anApple))
/// // Prints "4"
///
/// - Parameter x: A Unicode scalar value.
/// - Returns: The width of `x` when encoded in UTF-8, from `1` to `4`.
@_alwaysEmitIntoClient
public static func width(_ x: Unicode.Scalar) -> Int {
switch x.value {
case 0..<0x80: return 1
case 0x80..<0x0800: return 2
case 0x0800..<0x1_0000: return 3
default: return 4
}
}
}
extension Unicode.UTF8: _UnicodeEncoding {
public typealias CodeUnit = UInt8
public typealias EncodedScalar = _ValidUTF8Buffer
@inlinable
public static var encodedReplacementCharacter: EncodedScalar {
return EncodedScalar.encodedReplacementCharacter
}
@inline(__always)
@inlinable
public static func _isScalar(_ x: CodeUnit) -> Bool {
return isASCII(x)
}
/// Returns whether the given code unit represents an ASCII scalar
@_alwaysEmitIntoClient
@inline(__always)
public static func isASCII(_ x: CodeUnit) -> Bool {
return x & 0b1000_0000 == 0
}
@inline(__always)
@inlinable
public static func decode(_ source: EncodedScalar) -> Unicode.Scalar {
switch source.count {
case 1:
return Unicode.Scalar(_unchecked: source._biasedBits &- 0x01)
case 2:
let bits = source._biasedBits &- 0x0101
var value = (bits & 0b0_______________________11_1111__0000_0000) &>> 8
value |= (bits & 0b0________________________________0001_1111) &<< 6
return Unicode.Scalar(_unchecked: value)
case 3:
let bits = source._biasedBits &- 0x010101
var value = (bits & 0b0____________11_1111__0000_0000__0000_0000) &>> 16
value |= (bits & 0b0_______________________11_1111__0000_0000) &>> 2
value |= (bits & 0b0________________________________0000_1111) &<< 12
return Unicode.Scalar(_unchecked: value)
default:
_internalInvariant(source.count == 4)
let bits = source._biasedBits &- 0x01010101
var value = (bits & 0b0_11_1111__0000_0000__0000_0000__0000_0000) &>> 24
value |= (bits & 0b0____________11_1111__0000_0000__0000_0000) &>> 10
value |= (bits & 0b0_______________________11_1111__0000_0000) &<< 4
value |= (bits & 0b0________________________________0000_0111) &<< 18
return Unicode.Scalar(_unchecked: value)
}
}
@inline(__always)
@inlinable
public static func encode(
_ source: Unicode.Scalar
) -> EncodedScalar? {
var c = source.value
if _fastPath(c < (1&<<7)) {
return EncodedScalar(_containing: UInt8(c))
}
var o = c & 0b0__0011_1111
c &>>= 6
o &<<= 8
if _fastPath(c < (1&<<5)) {
return EncodedScalar(_biasedBits: (o | c) &+ 0b0__1000_0001__1100_0001)
}
o |= c & 0b0__0011_1111
c &>>= 6
o &<<= 8
if _fastPath(c < (1&<<4)) {
return EncodedScalar(
_biasedBits: (o | c) &+ 0b0__1000_0001__1000_0001__1110_0001)
}
o |= c & 0b0__0011_1111
c &>>= 6
o &<<= 8
return EncodedScalar(
_biasedBits: (o | c ) &+ 0b0__1000_0001__1000_0001__1000_0001__1111_0001)
}
@inlinable
@inline(__always)
public static func transcode<FromEncoding: _UnicodeEncoding>(
_ content: FromEncoding.EncodedScalar, from _: FromEncoding.Type
) -> EncodedScalar? {
if _fastPath(FromEncoding.self == UTF16.self) {
let c = _identityCast(content, to: UTF16.EncodedScalar.self)
var u0 = UInt16(truncatingIfNeeded: c._storage)
if _fastPath(u0 < 0x80) {
return EncodedScalar(_containing: UInt8(truncatingIfNeeded: u0))
}
var r = UInt32(u0 & 0b0__11_1111)
r &<<= 8
u0 &>>= 6
if _fastPath(u0 < (1&<<5)) {
return EncodedScalar(
_biasedBits: (UInt32(u0) | r) &+ 0b0__1000_0001__1100_0001)
}
r |= UInt32(u0 & 0b0__11_1111)
r &<<= 8
if _fastPath(u0 & (0xF800 &>> 6) != (0xD800 &>> 6)) {
u0 &>>= 6
return EncodedScalar(
_biasedBits: (UInt32(u0) | r) &+ 0b0__1000_0001__1000_0001__1110_0001)
}
}
else if _fastPath(FromEncoding.self == UTF8.self) {
return _identityCast(content, to: UTF8.EncodedScalar.self)
}
return encode(FromEncoding.decode(content))
}
@frozen
public struct ForwardParser: ConcurrentValue {
public typealias _Buffer = _UIntBuffer<UInt8>
@inline(__always)
@inlinable
public init() { _buffer = _Buffer() }
public var _buffer: _Buffer
}
@frozen
public struct ReverseParser: ConcurrentValue {
public typealias _Buffer = _UIntBuffer<UInt8>
@inline(__always)
@inlinable
public init() { _buffer = _Buffer() }
public var _buffer: _Buffer
}
}
extension UTF8.ReverseParser: Unicode.Parser, _UTFParser {
public typealias Encoding = Unicode.UTF8
@inline(__always)
@inlinable
public func _parseMultipleCodeUnits() -> (isValid: Bool, bitCount: UInt8) {
_internalInvariant(_buffer._storage & 0x80 != 0) // this case handled elsewhere
if _buffer._storage & 0b0__1110_0000__1100_0000
== 0b0__1100_0000__1000_0000 {
// 2-byte sequence. Top 4 bits of decoded result must be nonzero
let top4Bits = _buffer._storage & 0b0__0001_1110__0000_0000
if _fastPath(top4Bits != 0) { return (true, 2*8) }
}
else if _buffer._storage & 0b0__1111_0000__1100_0000__1100_0000
== 0b0__1110_0000__1000_0000__1000_0000 {
// 3-byte sequence. The top 5 bits of the decoded result must be nonzero
// and not a surrogate
let top5Bits = _buffer._storage & 0b0__1111__0010_0000__0000_0000
if _fastPath(
top5Bits != 0 && top5Bits != 0b0__1101__0010_0000__0000_0000) {
return (true, 3*8)
}
}
else if _buffer._storage & 0b0__1111_1000__1100_0000__1100_0000__1100_0000
== 0b0__1111_0000__1000_0000__1000_0000__1000_0000 {
// Make sure the top 5 bits of the decoded result would be in range
let top5bits = _buffer._storage
& 0b0__0111__0011_0000__0000_0000__0000_0000
if _fastPath(
top5bits != 0
&& top5bits <= 0b0__0100__0000_0000__0000_0000__0000_0000
) { return (true, 4*8) }
}
return (false, _invalidLength() &* 8)
}
/// Returns the length of the invalid sequence that ends with the LSB of
/// buffer.
@inline(never)
@usableFromInline
internal func _invalidLength() -> UInt8 {
if _buffer._storage & 0b0__1111_0000__1100_0000
== 0b0__1110_0000__1000_0000 {
// 2-byte prefix of 3-byte sequence. The top 5 bits of the decoded result
// must be nonzero and not a surrogate
let top5Bits = _buffer._storage & 0b0__1111__0010_0000
if top5Bits != 0 && top5Bits != 0b0__1101__0010_0000 { return 2 }
}
else if _buffer._storage & 0b1111_1000__1100_0000
== 0b1111_0000__1000_0000
{
// 2-byte prefix of 4-byte sequence
// Make sure the top 5 bits of the decoded result would be in range
let top5bits = _buffer._storage & 0b0__0111__0011_0000
if top5bits != 0 && top5bits <= 0b0__0100__0000_0000 { return 2 }
}
else if _buffer._storage & 0b0__1111_1000__1100_0000__1100_0000
== 0b0__1111_0000__1000_0000__1000_0000 {
// 3-byte prefix of 4-byte sequence
// Make sure the top 5 bits of the decoded result would be in range
let top5bits = _buffer._storage & 0b0__0111__0011_0000__0000_0000
if top5bits != 0 && top5bits <= 0b0__0100__0000_0000__0000_0000 {
return 3
}
}
return 1
}
@inline(__always)
@inlinable
public func _bufferedScalar(bitCount: UInt8) -> Encoding.EncodedScalar {
let x = UInt32(truncatingIfNeeded: _buffer._storage.byteSwapped)
let shift = 32 &- bitCount
return Encoding.EncodedScalar(_biasedBits: (x &+ 0x01010101) &>> shift)
}
}
extension Unicode.UTF8.ForwardParser: Unicode.Parser, _UTFParser {
public typealias Encoding = Unicode.UTF8
@inline(__always)
@inlinable
public func _parseMultipleCodeUnits() -> (isValid: Bool, bitCount: UInt8) {
_internalInvariant(_buffer._storage & 0x80 != 0) // this case handled elsewhere
if _buffer._storage & 0b0__1100_0000__1110_0000
== 0b0__1000_0000__1100_0000 {
// 2-byte sequence. At least one of the top 4 bits of the decoded result
// must be nonzero.
if _fastPath(_buffer._storage & 0b0_0001_1110 != 0) { return (true, 2*8) }
}
else if _buffer._storage & 0b0__1100_0000__1100_0000__1111_0000
== 0b0__1000_0000__1000_0000__1110_0000 {
// 3-byte sequence. The top 5 bits of the decoded result must be nonzero
// and not a surrogate
let top5Bits = _buffer._storage & 0b0___0010_0000__0000_1111
if _fastPath(top5Bits != 0 && top5Bits != 0b0___0010_0000__0000_1101) {
return (true, 3*8)
}
}
else if _buffer._storage & 0b0__1100_0000__1100_0000__1100_0000__1111_1000
== 0b0__1000_0000__1000_0000__1000_0000__1111_0000 {
// 4-byte sequence. The top 5 bits of the decoded result must be nonzero
// and no greater than 0b0__0100_0000
let top5bits = UInt16(_buffer._storage & 0b0__0011_0000__0000_0111)
if _fastPath(
top5bits != 0
&& top5bits.byteSwapped <= 0b0__0000_0100__0000_0000
) { return (true, 4*8) }
}
return (false, _invalidLength() &* 8)
}
/// Returns the length of the invalid sequence that starts with the LSB of
/// buffer.
@inline(never)
@usableFromInline
internal func _invalidLength() -> UInt8 {
if _buffer._storage & 0b0__1100_0000__1111_0000
== 0b0__1000_0000__1110_0000 {
// 2-byte prefix of 3-byte sequence. The top 5 bits of the decoded result
// must be nonzero and not a surrogate
let top5Bits = _buffer._storage & 0b0__0010_0000__0000_1111
if top5Bits != 0 && top5Bits != 0b0__0010_0000__0000_1101 { return 2 }
}
else if _buffer._storage & 0b0__1100_0000__1111_1000
== 0b0__1000_0000__1111_0000
{
// Prefix of 4-byte sequence. The top 5 bits of the decoded result
// must be nonzero and no greater than 0b0__0100_0000
let top5bits = UInt16(_buffer._storage & 0b0__0011_0000__0000_0111)
if top5bits != 0 && top5bits.byteSwapped <= 0b0__0000_0100__0000_0000 {
return _buffer._storage & 0b0__1100_0000__0000_0000__0000_0000
== 0b0__1000_0000__0000_0000__0000_0000 ? 3 : 2
}
}
return 1
}
@inlinable
public func _bufferedScalar(bitCount: UInt8) -> Encoding.EncodedScalar {
let x = UInt32(_buffer._storage) &+ 0x01010101
return _ValidUTF8Buffer(_biasedBits: x & ._lowBits(bitCount))
}
}