| // Copyright 2009 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. |
| |
| package runner |
| |
| import ( |
| "bytes" |
| "encoding/binary" |
| "fmt" |
| ) |
| |
| func writeLen(buf []byte, v, size int) { |
| for i := 0; i < size; i++ { |
| buf[size-i-1] = byte(v) |
| v >>= 8 |
| } |
| if v != 0 { |
| panic("length is too long") |
| } |
| } |
| |
| type byteBuilder struct { |
| buf *[]byte |
| start int |
| prefixLen int |
| child *byteBuilder |
| } |
| |
| func newByteBuilder() *byteBuilder { |
| buf := make([]byte, 0, 32) |
| return &byteBuilder{buf: &buf} |
| } |
| |
| func (bb *byteBuilder) len() int { |
| return len(*bb.buf) - bb.start - bb.prefixLen |
| } |
| |
| func (bb *byteBuilder) data() []byte { |
| bb.flush() |
| return (*bb.buf)[bb.start+bb.prefixLen:] |
| } |
| |
| func (bb *byteBuilder) flush() { |
| if bb.child == nil { |
| return |
| } |
| bb.child.flush() |
| writeLen((*bb.buf)[bb.child.start:], bb.child.len(), bb.child.prefixLen) |
| bb.child = nil |
| return |
| } |
| |
| func (bb *byteBuilder) finish() []byte { |
| bb.flush() |
| return *bb.buf |
| } |
| |
| func (bb *byteBuilder) addU8(u uint8) { |
| bb.flush() |
| *bb.buf = append(*bb.buf, u) |
| } |
| |
| func (bb *byteBuilder) addU16(u uint16) { |
| bb.flush() |
| *bb.buf = append(*bb.buf, byte(u>>8), byte(u)) |
| } |
| |
| func (bb *byteBuilder) addU24(u int) { |
| bb.flush() |
| *bb.buf = append(*bb.buf, byte(u>>16), byte(u>>8), byte(u)) |
| } |
| |
| func (bb *byteBuilder) addU32(u uint32) { |
| bb.flush() |
| *bb.buf = append(*bb.buf, byte(u>>24), byte(u>>16), byte(u>>8), byte(u)) |
| } |
| |
| func (bb *byteBuilder) addU64(u uint64) { |
| bb.flush() |
| var b [8]byte |
| binary.BigEndian.PutUint64(b[:], u) |
| *bb.buf = append(*bb.buf, b[:]...) |
| } |
| |
| func (bb *byteBuilder) addU8LengthPrefixed() *byteBuilder { |
| return bb.createChild(1) |
| } |
| |
| func (bb *byteBuilder) addU16LengthPrefixed() *byteBuilder { |
| return bb.createChild(2) |
| } |
| |
| func (bb *byteBuilder) addU24LengthPrefixed() *byteBuilder { |
| return bb.createChild(3) |
| } |
| |
| func (bb *byteBuilder) addU32LengthPrefixed() *byteBuilder { |
| return bb.createChild(4) |
| } |
| |
| func (bb *byteBuilder) addBytes(b []byte) { |
| bb.flush() |
| *bb.buf = append(*bb.buf, b...) |
| } |
| |
| func (bb *byteBuilder) createChild(lengthPrefixSize int) *byteBuilder { |
| bb.flush() |
| bb.child = &byteBuilder{ |
| buf: bb.buf, |
| start: len(*bb.buf), |
| prefixLen: lengthPrefixSize, |
| } |
| for i := 0; i < lengthPrefixSize; i++ { |
| *bb.buf = append(*bb.buf, 0) |
| } |
| return bb.child |
| } |
| |
| func (bb *byteBuilder) discardChild() { |
| if bb.child == nil { |
| return |
| } |
| *bb.buf = (*bb.buf)[:bb.child.start] |
| bb.child = nil |
| } |
| |
| type byteReader []byte |
| |
| func (br *byteReader) readInternal(out *byteReader, n int) bool { |
| if len(*br) < n { |
| return false |
| } |
| *out = (*br)[:n] |
| *br = (*br)[n:] |
| return true |
| } |
| |
| func (br *byteReader) readBytes(out *[]byte, n int) bool { |
| var child byteReader |
| if !br.readInternal(&child, n) { |
| return false |
| } |
| *out = []byte(child) |
| return true |
| } |
| |
| func (br *byteReader) readUint(out *uint64, n int) bool { |
| var b []byte |
| if !br.readBytes(&b, n) { |
| return false |
| } |
| *out = 0 |
| for _, v := range b { |
| *out <<= 8 |
| *out |= uint64(v) |
| } |
| return true |
| } |
| |
| func (br *byteReader) readU8(out *uint8) bool { |
| var b []byte |
| if !br.readBytes(&b, 1) { |
| return false |
| } |
| *out = b[0] |
| return true |
| } |
| |
| func (br *byteReader) readU16(out *uint16) bool { |
| var v uint64 |
| if !br.readUint(&v, 2) { |
| return false |
| } |
| *out = uint16(v) |
| return true |
| } |
| |
| func (br *byteReader) readU24(out *uint32) bool { |
| var v uint64 |
| if !br.readUint(&v, 3) { |
| return false |
| } |
| *out = uint32(v) |
| return true |
| } |
| |
| func (br *byteReader) readU32(out *uint32) bool { |
| var v uint64 |
| if !br.readUint(&v, 4) { |
| return false |
| } |
| *out = uint32(v) |
| return true |
| } |
| |
| func (br *byteReader) readU64(out *uint64) bool { |
| return br.readUint(out, 8) |
| } |
| |
| func (br *byteReader) readLengthPrefixed(out *byteReader, n int) bool { |
| var length uint64 |
| return br.readUint(&length, n) && |
| uint64(len(*br)) >= length && |
| br.readInternal(out, int(length)) |
| } |
| |
| func (br *byteReader) readLengthPrefixedBytes(out *[]byte, n int) bool { |
| var length uint64 |
| return br.readUint(&length, n) && |
| uint64(len(*br)) >= length && |
| br.readBytes(out, int(length)) |
| } |
| |
| func (br *byteReader) readU8LengthPrefixed(out *byteReader) bool { |
| return br.readLengthPrefixed(out, 1) |
| } |
| func (br *byteReader) readU8LengthPrefixedBytes(out *[]byte) bool { |
| return br.readLengthPrefixedBytes(out, 1) |
| } |
| |
| func (br *byteReader) readU16LengthPrefixed(out *byteReader) bool { |
| return br.readLengthPrefixed(out, 2) |
| } |
| func (br *byteReader) readU16LengthPrefixedBytes(out *[]byte) bool { |
| return br.readLengthPrefixedBytes(out, 2) |
| } |
| |
| func (br *byteReader) readU24LengthPrefixed(out *byteReader) bool { |
| return br.readLengthPrefixed(out, 3) |
| } |
| func (br *byteReader) readU24LengthPrefixedBytes(out *[]byte) bool { |
| return br.readLengthPrefixedBytes(out, 3) |
| } |
| |
| func (br *byteReader) readU32LengthPrefixed(out *byteReader) bool { |
| return br.readLengthPrefixed(out, 4) |
| } |
| func (br *byteReader) readU32LengthPrefixedBytes(out *[]byte) bool { |
| return br.readLengthPrefixedBytes(out, 4) |
| } |
| |
| type keyShareEntry struct { |
| group CurveID |
| keyExchange []byte |
| } |
| |
| type pskIdentity struct { |
| ticket []uint8 |
| obfuscatedTicketAge uint32 |
| } |
| |
| type clientHelloMsg struct { |
| raw []byte |
| isDTLS bool |
| vers uint16 |
| random []byte |
| sessionId []byte |
| cookie []byte |
| cipherSuites []uint16 |
| compressionMethods []uint8 |
| nextProtoNeg bool |
| serverName string |
| ocspStapling bool |
| supportedCurves []CurveID |
| supportedPoints []uint8 |
| hasKeyShares bool |
| keyShares []keyShareEntry |
| keySharesRaw []byte |
| trailingKeyShareData bool |
| pskIdentities []pskIdentity |
| pskKEModes []byte |
| pskBinders [][]uint8 |
| hasEarlyData bool |
| tls13Cookie []byte |
| ticketSupported bool |
| sessionTicket []uint8 |
| signatureAlgorithms []signatureAlgorithm |
| signatureAlgorithmsCert []signatureAlgorithm |
| supportedVersions []uint16 |
| secureRenegotiation []byte |
| alpnProtocols []string |
| quicTransportParams []byte |
| duplicateExtension bool |
| channelIDSupported bool |
| tokenBindingParams []byte |
| tokenBindingVersion uint16 |
| npnAfterAlpn bool |
| extendedMasterSecret bool |
| srtpProtectionProfiles []uint16 |
| srtpMasterKeyIdentifier string |
| sctListSupported bool |
| customExtension string |
| hasGREASEExtension bool |
| pskBinderFirst bool |
| omitExtensions bool |
| emptyExtensions bool |
| pad int |
| compressedCertAlgs []uint16 |
| delegatedCredentials bool |
| } |
| |
| func (m *clientHelloMsg) equal(i interface{}) bool { |
| m1, ok := i.(*clientHelloMsg) |
| if !ok { |
| return false |
| } |
| |
| return bytes.Equal(m.raw, m1.raw) && |
| m.isDTLS == m1.isDTLS && |
| m.vers == m1.vers && |
| bytes.Equal(m.random, m1.random) && |
| bytes.Equal(m.sessionId, m1.sessionId) && |
| bytes.Equal(m.cookie, m1.cookie) && |
| eqUint16s(m.cipherSuites, m1.cipherSuites) && |
| bytes.Equal(m.compressionMethods, m1.compressionMethods) && |
| m.nextProtoNeg == m1.nextProtoNeg && |
| m.serverName == m1.serverName && |
| m.ocspStapling == m1.ocspStapling && |
| eqCurveIDs(m.supportedCurves, m1.supportedCurves) && |
| bytes.Equal(m.supportedPoints, m1.supportedPoints) && |
| m.hasKeyShares == m1.hasKeyShares && |
| eqKeyShareEntryLists(m.keyShares, m1.keyShares) && |
| m.trailingKeyShareData == m1.trailingKeyShareData && |
| eqPSKIdentityLists(m.pskIdentities, m1.pskIdentities) && |
| bytes.Equal(m.pskKEModes, m1.pskKEModes) && |
| eqByteSlices(m.pskBinders, m1.pskBinders) && |
| m.hasEarlyData == m1.hasEarlyData && |
| bytes.Equal(m.tls13Cookie, m1.tls13Cookie) && |
| m.ticketSupported == m1.ticketSupported && |
| bytes.Equal(m.sessionTicket, m1.sessionTicket) && |
| eqSignatureAlgorithms(m.signatureAlgorithms, m1.signatureAlgorithms) && |
| eqSignatureAlgorithms(m.signatureAlgorithmsCert, m1.signatureAlgorithmsCert) && |
| eqUint16s(m.supportedVersions, m1.supportedVersions) && |
| bytes.Equal(m.secureRenegotiation, m1.secureRenegotiation) && |
| (m.secureRenegotiation == nil) == (m1.secureRenegotiation == nil) && |
| eqStrings(m.alpnProtocols, m1.alpnProtocols) && |
| bytes.Equal(m.quicTransportParams, m1.quicTransportParams) && |
| m.duplicateExtension == m1.duplicateExtension && |
| m.channelIDSupported == m1.channelIDSupported && |
| bytes.Equal(m.tokenBindingParams, m1.tokenBindingParams) && |
| m.tokenBindingVersion == m1.tokenBindingVersion && |
| m.npnAfterAlpn == m1.npnAfterAlpn && |
| m.extendedMasterSecret == m1.extendedMasterSecret && |
| eqUint16s(m.srtpProtectionProfiles, m1.srtpProtectionProfiles) && |
| m.srtpMasterKeyIdentifier == m1.srtpMasterKeyIdentifier && |
| m.sctListSupported == m1.sctListSupported && |
| m.customExtension == m1.customExtension && |
| m.hasGREASEExtension == m1.hasGREASEExtension && |
| m.pskBinderFirst == m1.pskBinderFirst && |
| m.omitExtensions == m1.omitExtensions && |
| m.emptyExtensions == m1.emptyExtensions && |
| m.pad == m1.pad && |
| eqUint16s(m.compressedCertAlgs, m1.compressedCertAlgs) && |
| m.delegatedCredentials == m1.delegatedCredentials |
| } |
| |
| func (m *clientHelloMsg) marshalKeyShares(bb *byteBuilder) { |
| keyShares := bb.addU16LengthPrefixed() |
| for _, keyShare := range m.keyShares { |
| keyShares.addU16(uint16(keyShare.group)) |
| keyExchange := keyShares.addU16LengthPrefixed() |
| keyExchange.addBytes(keyShare.keyExchange) |
| } |
| if m.trailingKeyShareData { |
| keyShares.addU8(0) |
| } |
| } |
| |
| func (m *clientHelloMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| handshakeMsg := newByteBuilder() |
| handshakeMsg.addU8(typeClientHello) |
| hello := handshakeMsg.addU24LengthPrefixed() |
| hello.addU16(m.vers) |
| hello.addBytes(m.random) |
| sessionId := hello.addU8LengthPrefixed() |
| sessionId.addBytes(m.sessionId) |
| if m.isDTLS { |
| cookie := hello.addU8LengthPrefixed() |
| cookie.addBytes(m.cookie) |
| } |
| cipherSuites := hello.addU16LengthPrefixed() |
| for _, suite := range m.cipherSuites { |
| cipherSuites.addU16(suite) |
| } |
| compressionMethods := hello.addU8LengthPrefixed() |
| compressionMethods.addBytes(m.compressionMethods) |
| |
| extensions := hello.addU16LengthPrefixed() |
| if len(m.pskIdentities) > 0 && m.pskBinderFirst { |
| extensions.addU16(extensionPreSharedKey) |
| pskExtension := extensions.addU16LengthPrefixed() |
| |
| pskIdentities := pskExtension.addU16LengthPrefixed() |
| for _, psk := range m.pskIdentities { |
| pskIdentities.addU16LengthPrefixed().addBytes(psk.ticket) |
| pskIdentities.addU32(psk.obfuscatedTicketAge) |
| } |
| pskBinders := pskExtension.addU16LengthPrefixed() |
| for _, binder := range m.pskBinders { |
| pskBinders.addU8LengthPrefixed().addBytes(binder) |
| } |
| } |
| if m.duplicateExtension { |
| // Add a duplicate bogus extension at the beginning and end. |
| extensions.addU16(0xffff) |
| extensions.addU16(0) // 0-length for empty extension |
| } |
| if m.nextProtoNeg && !m.npnAfterAlpn { |
| extensions.addU16(extensionNextProtoNeg) |
| extensions.addU16(0) // The length is always 0 |
| } |
| if len(m.serverName) > 0 { |
| extensions.addU16(extensionServerName) |
| serverNameList := extensions.addU16LengthPrefixed() |
| |
| // RFC 3546, section 3.1 |
| // |
| // struct { |
| // NameType name_type; |
| // select (name_type) { |
| // case host_name: HostName; |
| // } name; |
| // } ServerName; |
| // |
| // enum { |
| // host_name(0), (255) |
| // } NameType; |
| // |
| // opaque HostName<1..2^16-1>; |
| // |
| // struct { |
| // ServerName server_name_list<1..2^16-1> |
| // } ServerNameList; |
| |
| serverName := serverNameList.addU16LengthPrefixed() |
| serverName.addU8(0) // NameType host_name(0) |
| hostName := serverName.addU16LengthPrefixed() |
| hostName.addBytes([]byte(m.serverName)) |
| } |
| if m.ocspStapling { |
| extensions.addU16(extensionStatusRequest) |
| certificateStatusRequest := extensions.addU16LengthPrefixed() |
| |
| // RFC 4366, section 3.6 |
| certificateStatusRequest.addU8(1) // OCSP type |
| // Two zero valued uint16s for the two lengths. |
| certificateStatusRequest.addU16(0) // ResponderID length |
| certificateStatusRequest.addU16(0) // Extensions length |
| } |
| if len(m.supportedCurves) > 0 { |
| // http://tools.ietf.org/html/rfc4492#section-5.1.1 |
| extensions.addU16(extensionSupportedCurves) |
| supportedCurvesList := extensions.addU16LengthPrefixed() |
| supportedCurves := supportedCurvesList.addU16LengthPrefixed() |
| for _, curve := range m.supportedCurves { |
| supportedCurves.addU16(uint16(curve)) |
| } |
| } |
| if len(m.supportedPoints) > 0 { |
| // http://tools.ietf.org/html/rfc4492#section-5.1.2 |
| extensions.addU16(extensionSupportedPoints) |
| supportedPointsList := extensions.addU16LengthPrefixed() |
| supportedPoints := supportedPointsList.addU8LengthPrefixed() |
| supportedPoints.addBytes(m.supportedPoints) |
| } |
| if m.hasKeyShares { |
| extensions.addU16(extensionKeyShare) |
| keyShareList := extensions.addU16LengthPrefixed() |
| m.marshalKeyShares(keyShareList) |
| } |
| if len(m.pskKEModes) > 0 { |
| extensions.addU16(extensionPSKKeyExchangeModes) |
| pskModesExtension := extensions.addU16LengthPrefixed() |
| pskModesExtension.addU8LengthPrefixed().addBytes(m.pskKEModes) |
| } |
| if m.hasEarlyData { |
| extensions.addU16(extensionEarlyData) |
| extensions.addU16(0) // The length is zero. |
| } |
| if len(m.tls13Cookie) > 0 { |
| extensions.addU16(extensionCookie) |
| body := extensions.addU16LengthPrefixed() |
| body.addU16LengthPrefixed().addBytes(m.tls13Cookie) |
| } |
| if m.ticketSupported { |
| // http://tools.ietf.org/html/rfc5077#section-3.2 |
| extensions.addU16(extensionSessionTicket) |
| sessionTicketExtension := extensions.addU16LengthPrefixed() |
| sessionTicketExtension.addBytes(m.sessionTicket) |
| } |
| if len(m.signatureAlgorithms) > 0 { |
| // https://tools.ietf.org/html/rfc5246#section-7.4.1.4.1 |
| extensions.addU16(extensionSignatureAlgorithms) |
| signatureAlgorithmsExtension := extensions.addU16LengthPrefixed() |
| signatureAlgorithms := signatureAlgorithmsExtension.addU16LengthPrefixed() |
| for _, sigAlg := range m.signatureAlgorithms { |
| signatureAlgorithms.addU16(uint16(sigAlg)) |
| } |
| } |
| if len(m.signatureAlgorithmsCert) > 0 { |
| extensions.addU16(extensionSignatureAlgorithmsCert) |
| signatureAlgorithmsCertExtension := extensions.addU16LengthPrefixed() |
| signatureAlgorithmsCert := signatureAlgorithmsCertExtension.addU16LengthPrefixed() |
| for _, sigAlg := range m.signatureAlgorithmsCert { |
| signatureAlgorithmsCert.addU16(uint16(sigAlg)) |
| } |
| } |
| if len(m.supportedVersions) > 0 { |
| extensions.addU16(extensionSupportedVersions) |
| supportedVersionsExtension := extensions.addU16LengthPrefixed() |
| supportedVersions := supportedVersionsExtension.addU8LengthPrefixed() |
| for _, version := range m.supportedVersions { |
| supportedVersions.addU16(uint16(version)) |
| } |
| } |
| if m.secureRenegotiation != nil { |
| extensions.addU16(extensionRenegotiationInfo) |
| secureRenegoExt := extensions.addU16LengthPrefixed() |
| secureRenego := secureRenegoExt.addU8LengthPrefixed() |
| secureRenego.addBytes(m.secureRenegotiation) |
| } |
| if len(m.alpnProtocols) > 0 { |
| // https://tools.ietf.org/html/rfc7301#section-3.1 |
| extensions.addU16(extensionALPN) |
| alpnExtension := extensions.addU16LengthPrefixed() |
| |
| protocolNameList := alpnExtension.addU16LengthPrefixed() |
| for _, s := range m.alpnProtocols { |
| protocolName := protocolNameList.addU8LengthPrefixed() |
| protocolName.addBytes([]byte(s)) |
| } |
| } |
| if len(m.quicTransportParams) > 0 { |
| extensions.addU16(extensionQUICTransportParams) |
| params := extensions.addU16LengthPrefixed() |
| params.addBytes(m.quicTransportParams) |
| } |
| if m.channelIDSupported { |
| extensions.addU16(extensionChannelID) |
| extensions.addU16(0) // Length is always 0 |
| } |
| if m.tokenBindingParams != nil { |
| extensions.addU16(extensionTokenBinding) |
| tokbindExtension := extensions.addU16LengthPrefixed() |
| tokbindExtension.addU16(m.tokenBindingVersion) |
| tokbindParams := tokbindExtension.addU8LengthPrefixed() |
| tokbindParams.addBytes(m.tokenBindingParams) |
| } |
| if m.nextProtoNeg && m.npnAfterAlpn { |
| extensions.addU16(extensionNextProtoNeg) |
| extensions.addU16(0) // Length is always 0 |
| } |
| if m.duplicateExtension { |
| // Add a duplicate bogus extension at the beginning and end. |
| extensions.addU16(0xffff) |
| extensions.addU16(0) |
| } |
| if m.extendedMasterSecret { |
| // https://tools.ietf.org/html/rfc7627 |
| extensions.addU16(extensionExtendedMasterSecret) |
| extensions.addU16(0) // Length is always 0 |
| } |
| if len(m.srtpProtectionProfiles) > 0 { |
| // https://tools.ietf.org/html/rfc5764#section-4.1.1 |
| extensions.addU16(extensionUseSRTP) |
| useSrtpExt := extensions.addU16LengthPrefixed() |
| |
| srtpProtectionProfiles := useSrtpExt.addU16LengthPrefixed() |
| for _, p := range m.srtpProtectionProfiles { |
| srtpProtectionProfiles.addU16(p) |
| } |
| srtpMki := useSrtpExt.addU8LengthPrefixed() |
| srtpMki.addBytes([]byte(m.srtpMasterKeyIdentifier)) |
| } |
| if m.sctListSupported { |
| extensions.addU16(extensionSignedCertificateTimestamp) |
| extensions.addU16(0) // Length is always 0 |
| } |
| if l := len(m.customExtension); l > 0 { |
| extensions.addU16(extensionCustom) |
| customExt := extensions.addU16LengthPrefixed() |
| customExt.addBytes([]byte(m.customExtension)) |
| } |
| if len(m.compressedCertAlgs) > 0 { |
| extensions.addU16(extensionCompressedCertAlgs) |
| body := extensions.addU16LengthPrefixed() |
| algIDs := body.addU8LengthPrefixed() |
| for _, v := range m.compressedCertAlgs { |
| algIDs.addU16(v) |
| } |
| } |
| if m.delegatedCredentials { |
| extensions.addU16(extensionDelegatedCredentials) |
| body := extensions.addU16LengthPrefixed() |
| signatureSchemeList := body.addU16LengthPrefixed() |
| for _, sigAlg := range m.signatureAlgorithms { |
| signatureSchemeList.addU16(uint16(sigAlg)) |
| } |
| } |
| |
| // The PSK extension must be last. See https://tools.ietf.org/html/rfc8446#section-4.2.11 |
| if len(m.pskIdentities) > 0 && !m.pskBinderFirst { |
| extensions.addU16(extensionPreSharedKey) |
| pskExtension := extensions.addU16LengthPrefixed() |
| |
| pskIdentities := pskExtension.addU16LengthPrefixed() |
| for _, psk := range m.pskIdentities { |
| pskIdentities.addU16LengthPrefixed().addBytes(psk.ticket) |
| pskIdentities.addU32(psk.obfuscatedTicketAge) |
| } |
| pskBinders := pskExtension.addU16LengthPrefixed() |
| for _, binder := range m.pskBinders { |
| pskBinders.addU8LengthPrefixed().addBytes(binder) |
| } |
| } |
| |
| if m.pad != 0 && hello.len()%m.pad != 0 { |
| extensions.addU16(extensionPadding) |
| padding := extensions.addU16LengthPrefixed() |
| // Note hello.len() has changed at this point from the length |
| // prefix. |
| if l := hello.len() % m.pad; l != 0 { |
| padding.addBytes(make([]byte, m.pad-l)) |
| } |
| } |
| |
| if m.omitExtensions || m.emptyExtensions { |
| // Silently erase any extensions which were sent. |
| hello.discardChild() |
| if m.emptyExtensions { |
| hello.addU16(0) |
| } |
| } |
| |
| m.raw = handshakeMsg.finish() |
| // Sanity-check padding. |
| if m.pad != 0 && (len(m.raw)-4)%m.pad != 0 { |
| panic(fmt.Sprintf("%d is not a multiple of %d", len(m.raw)-4, m.pad)) |
| } |
| return m.raw |
| } |
| |
| func parseSignatureAlgorithms(reader *byteReader, out *[]signatureAlgorithm, allowEmpty bool) bool { |
| var sigAlgs byteReader |
| if !reader.readU16LengthPrefixed(&sigAlgs) { |
| return false |
| } |
| if !allowEmpty && len(sigAlgs) == 0 { |
| return false |
| } |
| *out = make([]signatureAlgorithm, 0, len(sigAlgs)/2) |
| for len(sigAlgs) > 0 { |
| var v uint16 |
| if !sigAlgs.readU16(&v) { |
| return false |
| } |
| *out = append(*out, signatureAlgorithm(v)) |
| } |
| return true |
| } |
| |
| func checkDuplicateExtensions(extensions byteReader) bool { |
| seen := make(map[uint16]struct{}) |
| for len(extensions) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !extensions.readU16(&extension) || |
| !extensions.readU16LengthPrefixed(&body) { |
| return false |
| } |
| if _, ok := seen[extension]; ok { |
| return false |
| } |
| seen[extension] = struct{}{} |
| } |
| return true |
| } |
| |
| func (m *clientHelloMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| if !reader.readU16(&m.vers) || |
| !reader.readBytes(&m.random, 32) || |
| !reader.readU8LengthPrefixedBytes(&m.sessionId) || |
| len(m.sessionId) > 32 { |
| return false |
| } |
| if m.isDTLS { |
| if !reader.readU8LengthPrefixedBytes(&m.cookie) || |
| len(m.cookie) > 32 { |
| return false |
| } |
| } |
| var cipherSuites byteReader |
| if !reader.readU16LengthPrefixed(&cipherSuites) || |
| !reader.readU8LengthPrefixedBytes(&m.compressionMethods) { |
| return false |
| } |
| |
| m.cipherSuites = make([]uint16, 0, len(cipherSuites)/2) |
| for len(cipherSuites) > 0 { |
| var v uint16 |
| if !cipherSuites.readU16(&v) { |
| return false |
| } |
| m.cipherSuites = append(m.cipherSuites, v) |
| if v == scsvRenegotiation { |
| m.secureRenegotiation = []byte{} |
| } |
| } |
| |
| m.nextProtoNeg = false |
| m.serverName = "" |
| m.ocspStapling = false |
| m.keyShares = nil |
| m.pskIdentities = nil |
| m.hasEarlyData = false |
| m.ticketSupported = false |
| m.sessionTicket = nil |
| m.signatureAlgorithms = nil |
| m.signatureAlgorithmsCert = nil |
| m.supportedVersions = nil |
| m.alpnProtocols = nil |
| m.extendedMasterSecret = false |
| m.customExtension = "" |
| m.delegatedCredentials = false |
| |
| if len(reader) == 0 { |
| // ClientHello is optionally followed by extension data |
| return true |
| } |
| |
| var extensions byteReader |
| if !reader.readU16LengthPrefixed(&extensions) || len(reader) != 0 || !checkDuplicateExtensions(extensions) { |
| return false |
| } |
| for len(extensions) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !extensions.readU16(&extension) || |
| !extensions.readU16LengthPrefixed(&body) { |
| return false |
| } |
| switch extension { |
| case extensionServerName: |
| var names byteReader |
| if !body.readU16LengthPrefixed(&names) || len(body) != 0 { |
| return false |
| } |
| for len(names) > 0 { |
| var nameType byte |
| var name []byte |
| if !names.readU8(&nameType) || |
| !names.readU16LengthPrefixedBytes(&name) { |
| return false |
| } |
| if nameType == 0 { |
| m.serverName = string(name) |
| } |
| } |
| case extensionNextProtoNeg: |
| if len(body) != 0 { |
| return false |
| } |
| m.nextProtoNeg = true |
| case extensionStatusRequest: |
| m.ocspStapling = len(body) > 0 && body[0] == statusTypeOCSP |
| case extensionSupportedCurves: |
| // http://tools.ietf.org/html/rfc4492#section-5.5.1 |
| var curves byteReader |
| if !body.readU16LengthPrefixed(&curves) || len(body) != 0 { |
| return false |
| } |
| m.supportedCurves = make([]CurveID, 0, len(curves)/2) |
| for len(curves) > 0 { |
| var v uint16 |
| if !curves.readU16(&v) { |
| return false |
| } |
| m.supportedCurves = append(m.supportedCurves, CurveID(v)) |
| } |
| case extensionSupportedPoints: |
| // http://tools.ietf.org/html/rfc4492#section-5.5.2 |
| if !body.readU8LengthPrefixedBytes(&m.supportedPoints) || len(body) != 0 { |
| return false |
| } |
| case extensionSessionTicket: |
| // http://tools.ietf.org/html/rfc5077#section-3.2 |
| m.ticketSupported = true |
| m.sessionTicket = []byte(body) |
| case extensionKeyShare: |
| // https://tools.ietf.org/html/rfc8446#section-4.2.8 |
| m.hasKeyShares = true |
| m.keySharesRaw = body |
| var keyShares byteReader |
| if !body.readU16LengthPrefixed(&keyShares) || len(body) != 0 { |
| return false |
| } |
| for len(keyShares) > 0 { |
| var entry keyShareEntry |
| var group uint16 |
| if !keyShares.readU16(&group) || |
| !keyShares.readU16LengthPrefixedBytes(&entry.keyExchange) { |
| return false |
| } |
| entry.group = CurveID(group) |
| m.keyShares = append(m.keyShares, entry) |
| } |
| case extensionPreSharedKey: |
| // https://tools.ietf.org/html/rfc8446#section-4.2.11 |
| var psks, binders byteReader |
| if !body.readU16LengthPrefixed(&psks) || |
| !body.readU16LengthPrefixed(&binders) || |
| len(body) != 0 { |
| return false |
| } |
| for len(psks) > 0 { |
| var psk pskIdentity |
| if !psks.readU16LengthPrefixedBytes(&psk.ticket) || |
| !psks.readU32(&psk.obfuscatedTicketAge) { |
| return false |
| } |
| m.pskIdentities = append(m.pskIdentities, psk) |
| } |
| for len(binders) > 0 { |
| var binder []byte |
| if !binders.readU8LengthPrefixedBytes(&binder) { |
| return false |
| } |
| m.pskBinders = append(m.pskBinders, binder) |
| } |
| |
| // There must be the same number of identities as binders. |
| if len(m.pskIdentities) != len(m.pskBinders) { |
| return false |
| } |
| case extensionPSKKeyExchangeModes: |
| // https://tools.ietf.org/html/rfc8446#section-4.2.9 |
| if !body.readU8LengthPrefixedBytes(&m.pskKEModes) || len(body) != 0 { |
| return false |
| } |
| case extensionEarlyData: |
| // https://tools.ietf.org/html/rfc8446#section-4.2.10 |
| if len(body) != 0 { |
| return false |
| } |
| m.hasEarlyData = true |
| case extensionCookie: |
| if !body.readU16LengthPrefixedBytes(&m.tls13Cookie) || len(body) != 0 { |
| return false |
| } |
| case extensionSignatureAlgorithms: |
| // https://tools.ietf.org/html/rfc5246#section-7.4.1.4.1 |
| if !parseSignatureAlgorithms(&body, &m.signatureAlgorithms, false) || len(body) != 0 { |
| return false |
| } |
| case extensionSignatureAlgorithmsCert: |
| if !parseSignatureAlgorithms(&body, &m.signatureAlgorithmsCert, false) || len(body) != 0 { |
| return false |
| } |
| case extensionSupportedVersions: |
| var versions byteReader |
| if !body.readU8LengthPrefixed(&versions) || len(body) != 0 { |
| return false |
| } |
| m.supportedVersions = make([]uint16, 0, len(versions)/2) |
| for len(versions) > 0 { |
| var v uint16 |
| if !versions.readU16(&v) { |
| return false |
| } |
| m.supportedVersions = append(m.supportedVersions, v) |
| } |
| case extensionRenegotiationInfo: |
| if !body.readU8LengthPrefixedBytes(&m.secureRenegotiation) || len(body) != 0 { |
| return false |
| } |
| case extensionALPN: |
| var protocols byteReader |
| if !body.readU16LengthPrefixed(&protocols) || len(body) != 0 { |
| return false |
| } |
| for len(protocols) > 0 { |
| var protocol []byte |
| if !protocols.readU8LengthPrefixedBytes(&protocol) { |
| return false |
| } |
| m.alpnProtocols = append(m.alpnProtocols, string(protocol)) |
| } |
| case extensionQUICTransportParams: |
| m.quicTransportParams = body |
| case extensionChannelID: |
| if len(body) != 0 { |
| return false |
| } |
| m.channelIDSupported = true |
| case extensionTokenBinding: |
| if !body.readU16(&m.tokenBindingVersion) || |
| !body.readU8LengthPrefixedBytes(&m.tokenBindingParams) || |
| len(body) != 0 { |
| return false |
| } |
| case extensionExtendedMasterSecret: |
| if len(body) != 0 { |
| return false |
| } |
| m.extendedMasterSecret = true |
| case extensionUseSRTP: |
| var profiles byteReader |
| var mki []byte |
| if !body.readU16LengthPrefixed(&profiles) || |
| !body.readU8LengthPrefixedBytes(&mki) || |
| len(body) != 0 { |
| return false |
| } |
| m.srtpProtectionProfiles = make([]uint16, 0, len(profiles)/2) |
| for len(profiles) > 0 { |
| var v uint16 |
| if !profiles.readU16(&v) { |
| return false |
| } |
| m.srtpProtectionProfiles = append(m.srtpProtectionProfiles, v) |
| } |
| m.srtpMasterKeyIdentifier = string(mki) |
| case extensionSignedCertificateTimestamp: |
| if len(body) != 0 { |
| return false |
| } |
| m.sctListSupported = true |
| case extensionCustom: |
| m.customExtension = string(body) |
| case extensionCompressedCertAlgs: |
| var algIDs byteReader |
| if !body.readU8LengthPrefixed(&algIDs) { |
| return false |
| } |
| |
| seen := make(map[uint16]struct{}) |
| for len(algIDs) > 0 { |
| var algID uint16 |
| if !algIDs.readU16(&algID) { |
| return false |
| } |
| if _, ok := seen[algID]; ok { |
| return false |
| } |
| seen[algID] = struct{}{} |
| m.compressedCertAlgs = append(m.compressedCertAlgs, algID) |
| } |
| case extensionPadding: |
| // Padding bytes must be all zero. |
| for _, b := range body { |
| if b != 0 { |
| return false |
| } |
| } |
| case extensionDelegatedCredentials: |
| if len(body) != 0 { |
| return false |
| } |
| m.delegatedCredentials = true |
| } |
| |
| if isGREASEValue(extension) { |
| m.hasGREASEExtension = true |
| } |
| } |
| |
| return true |
| } |
| |
| type serverHelloMsg struct { |
| raw []byte |
| isDTLS bool |
| vers uint16 |
| versOverride uint16 |
| supportedVersOverride uint16 |
| omitSupportedVers bool |
| random []byte |
| sessionId []byte |
| cipherSuite uint16 |
| hasKeyShare bool |
| keyShare keyShareEntry |
| hasPSKIdentity bool |
| pskIdentity uint16 |
| compressionMethod uint8 |
| customExtension string |
| unencryptedALPN string |
| omitExtensions bool |
| emptyExtensions bool |
| extensions serverExtensions |
| } |
| |
| func (m *serverHelloMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| handshakeMsg := newByteBuilder() |
| handshakeMsg.addU8(typeServerHello) |
| hello := handshakeMsg.addU24LengthPrefixed() |
| |
| // m.vers is used both to determine the format of the rest of the |
| // ServerHello and to override the value, so include a second version |
| // field. |
| vers, ok := wireToVersion(m.vers, m.isDTLS) |
| if !ok { |
| panic("unknown version") |
| } |
| if m.versOverride != 0 { |
| hello.addU16(m.versOverride) |
| } else if vers >= VersionTLS13 { |
| hello.addU16(VersionTLS12) |
| } else { |
| hello.addU16(m.vers) |
| } |
| |
| hello.addBytes(m.random) |
| sessionId := hello.addU8LengthPrefixed() |
| sessionId.addBytes(m.sessionId) |
| hello.addU16(m.cipherSuite) |
| hello.addU8(m.compressionMethod) |
| |
| extensions := hello.addU16LengthPrefixed() |
| |
| if vers >= VersionTLS13 { |
| if m.hasKeyShare { |
| extensions.addU16(extensionKeyShare) |
| keyShare := extensions.addU16LengthPrefixed() |
| keyShare.addU16(uint16(m.keyShare.group)) |
| keyExchange := keyShare.addU16LengthPrefixed() |
| keyExchange.addBytes(m.keyShare.keyExchange) |
| } |
| if m.hasPSKIdentity { |
| extensions.addU16(extensionPreSharedKey) |
| extensions.addU16(2) // Length |
| extensions.addU16(m.pskIdentity) |
| } |
| if !m.omitSupportedVers { |
| extensions.addU16(extensionSupportedVersions) |
| extensions.addU16(2) // Length |
| if m.supportedVersOverride != 0 { |
| extensions.addU16(m.supportedVersOverride) |
| } else { |
| extensions.addU16(m.vers) |
| } |
| } |
| if len(m.customExtension) > 0 { |
| extensions.addU16(extensionCustom) |
| customExt := extensions.addU16LengthPrefixed() |
| customExt.addBytes([]byte(m.customExtension)) |
| } |
| if len(m.unencryptedALPN) > 0 { |
| extensions.addU16(extensionALPN) |
| extension := extensions.addU16LengthPrefixed() |
| |
| protocolNameList := extension.addU16LengthPrefixed() |
| protocolName := protocolNameList.addU8LengthPrefixed() |
| protocolName.addBytes([]byte(m.unencryptedALPN)) |
| } |
| } else { |
| m.extensions.marshal(extensions) |
| if m.omitExtensions || m.emptyExtensions { |
| // Silently erasing server extensions will break the handshake. Instead, |
| // assert that tests which use this field also disable all features which |
| // would write an extension. |
| if extensions.len() != 0 { |
| panic(fmt.Sprintf("ServerHello unexpectedly contained extensions: %x, %+v", extensions.data(), m)) |
| } |
| hello.discardChild() |
| if m.emptyExtensions { |
| hello.addU16(0) |
| } |
| } |
| } |
| |
| m.raw = handshakeMsg.finish() |
| return m.raw |
| } |
| |
| func (m *serverHelloMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| if !reader.readU16(&m.vers) || |
| !reader.readBytes(&m.random, 32) { |
| return false |
| } |
| vers, ok := wireToVersion(m.vers, m.isDTLS) |
| if !ok { |
| return false |
| } |
| if !reader.readU8LengthPrefixedBytes(&m.sessionId) || |
| !reader.readU16(&m.cipherSuite) || |
| !reader.readU8(&m.compressionMethod) { |
| return false |
| } |
| |
| if len(reader) == 0 && m.vers < VersionTLS13 { |
| // Extension data is optional before TLS 1.3. |
| m.extensions = serverExtensions{} |
| m.omitExtensions = true |
| return true |
| } |
| |
| var extensions byteReader |
| if !reader.readU16LengthPrefixed(&extensions) || len(reader) != 0 || !checkDuplicateExtensions(extensions) { |
| return false |
| } |
| |
| // Parse out the version from supported_versions if available. |
| if m.vers == VersionTLS12 { |
| extensionsCopy := extensions |
| for len(extensionsCopy) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !extensionsCopy.readU16(&extension) || |
| !extensionsCopy.readU16LengthPrefixed(&body) { |
| return false |
| } |
| if extension == extensionSupportedVersions { |
| if !body.readU16(&m.vers) || len(body) != 0 { |
| return false |
| } |
| vers, ok = wireToVersion(m.vers, m.isDTLS) |
| if !ok { |
| return false |
| } |
| } |
| } |
| } |
| |
| if vers >= VersionTLS13 { |
| for len(extensions) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !extensions.readU16(&extension) || |
| !extensions.readU16LengthPrefixed(&body) { |
| return false |
| } |
| switch extension { |
| case extensionKeyShare: |
| m.hasKeyShare = true |
| var group uint16 |
| if !body.readU16(&group) || |
| !body.readU16LengthPrefixedBytes(&m.keyShare.keyExchange) || |
| len(body) != 0 { |
| return false |
| } |
| m.keyShare.group = CurveID(group) |
| case extensionPreSharedKey: |
| if !body.readU16(&m.pskIdentity) || len(body) != 0 { |
| return false |
| } |
| m.hasPSKIdentity = true |
| case extensionSupportedVersions: |
| // Parsed above. |
| default: |
| // Only allow the 3 extensions that are sent in |
| // the clear in TLS 1.3. |
| return false |
| } |
| } |
| } else if !m.extensions.unmarshal(extensions, vers) { |
| return false |
| } |
| |
| return true |
| } |
| |
| type encryptedExtensionsMsg struct { |
| raw []byte |
| extensions serverExtensions |
| empty bool |
| } |
| |
| func (m *encryptedExtensionsMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| encryptedExtensionsMsg := newByteBuilder() |
| encryptedExtensionsMsg.addU8(typeEncryptedExtensions) |
| encryptedExtensions := encryptedExtensionsMsg.addU24LengthPrefixed() |
| if !m.empty { |
| extensions := encryptedExtensions.addU16LengthPrefixed() |
| m.extensions.marshal(extensions) |
| } |
| |
| m.raw = encryptedExtensionsMsg.finish() |
| return m.raw |
| } |
| |
| func (m *encryptedExtensionsMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| var extensions byteReader |
| if !reader.readU16LengthPrefixed(&extensions) || len(reader) != 0 { |
| return false |
| } |
| return m.extensions.unmarshal(extensions, VersionTLS13) |
| } |
| |
| type serverExtensions struct { |
| nextProtoNeg bool |
| nextProtos []string |
| ocspStapling bool |
| ticketSupported bool |
| secureRenegotiation []byte |
| alpnProtocol string |
| alpnProtocolEmpty bool |
| duplicateExtension bool |
| channelIDRequested bool |
| tokenBindingParams []byte |
| tokenBindingVersion uint16 |
| extendedMasterSecret bool |
| srtpProtectionProfile uint16 |
| srtpMasterKeyIdentifier string |
| sctList []byte |
| customExtension string |
| npnAfterAlpn bool |
| hasKeyShare bool |
| hasEarlyData bool |
| keyShare keyShareEntry |
| supportedVersion uint16 |
| supportedPoints []uint8 |
| supportedCurves []CurveID |
| quicTransportParams []byte |
| serverNameAck bool |
| } |
| |
| func (m *serverExtensions) marshal(extensions *byteBuilder) { |
| if m.duplicateExtension { |
| // Add a duplicate bogus extension at the beginning and end. |
| extensions.addU16(0xffff) |
| extensions.addU16(0) // length = 0 for empty extension |
| } |
| if m.nextProtoNeg && !m.npnAfterAlpn { |
| extensions.addU16(extensionNextProtoNeg) |
| extension := extensions.addU16LengthPrefixed() |
| |
| for _, v := range m.nextProtos { |
| if len(v) > 255 { |
| v = v[:255] |
| } |
| npn := extension.addU8LengthPrefixed() |
| npn.addBytes([]byte(v)) |
| } |
| } |
| if m.ocspStapling { |
| extensions.addU16(extensionStatusRequest) |
| extensions.addU16(0) |
| } |
| if m.ticketSupported { |
| extensions.addU16(extensionSessionTicket) |
| extensions.addU16(0) |
| } |
| if m.secureRenegotiation != nil { |
| extensions.addU16(extensionRenegotiationInfo) |
| extension := extensions.addU16LengthPrefixed() |
| secureRenego := extension.addU8LengthPrefixed() |
| secureRenego.addBytes(m.secureRenegotiation) |
| } |
| if len(m.alpnProtocol) > 0 || m.alpnProtocolEmpty { |
| extensions.addU16(extensionALPN) |
| extension := extensions.addU16LengthPrefixed() |
| |
| protocolNameList := extension.addU16LengthPrefixed() |
| protocolName := protocolNameList.addU8LengthPrefixed() |
| protocolName.addBytes([]byte(m.alpnProtocol)) |
| } |
| if m.channelIDRequested { |
| extensions.addU16(extensionChannelID) |
| extensions.addU16(0) |
| } |
| if m.tokenBindingParams != nil { |
| extensions.addU16(extensionTokenBinding) |
| tokbindExtension := extensions.addU16LengthPrefixed() |
| tokbindExtension.addU16(m.tokenBindingVersion) |
| tokbindParams := tokbindExtension.addU8LengthPrefixed() |
| tokbindParams.addBytes(m.tokenBindingParams) |
| } |
| if m.duplicateExtension { |
| // Add a duplicate bogus extension at the beginning and end. |
| extensions.addU16(0xffff) |
| extensions.addU16(0) |
| } |
| if m.extendedMasterSecret { |
| extensions.addU16(extensionExtendedMasterSecret) |
| extensions.addU16(0) |
| } |
| if m.srtpProtectionProfile != 0 { |
| extensions.addU16(extensionUseSRTP) |
| extension := extensions.addU16LengthPrefixed() |
| |
| srtpProtectionProfiles := extension.addU16LengthPrefixed() |
| srtpProtectionProfiles.addU16(m.srtpProtectionProfile) |
| srtpMki := extension.addU8LengthPrefixed() |
| srtpMki.addBytes([]byte(m.srtpMasterKeyIdentifier)) |
| } |
| if m.sctList != nil { |
| extensions.addU16(extensionSignedCertificateTimestamp) |
| extension := extensions.addU16LengthPrefixed() |
| extension.addBytes(m.sctList) |
| } |
| if l := len(m.customExtension); l > 0 { |
| extensions.addU16(extensionCustom) |
| customExt := extensions.addU16LengthPrefixed() |
| customExt.addBytes([]byte(m.customExtension)) |
| } |
| if m.nextProtoNeg && m.npnAfterAlpn { |
| extensions.addU16(extensionNextProtoNeg) |
| extension := extensions.addU16LengthPrefixed() |
| |
| for _, v := range m.nextProtos { |
| if len(v) > 255 { |
| v = v[0:255] |
| } |
| npn := extension.addU8LengthPrefixed() |
| npn.addBytes([]byte(v)) |
| } |
| } |
| if m.hasKeyShare { |
| extensions.addU16(extensionKeyShare) |
| keyShare := extensions.addU16LengthPrefixed() |
| keyShare.addU16(uint16(m.keyShare.group)) |
| keyExchange := keyShare.addU16LengthPrefixed() |
| keyExchange.addBytes(m.keyShare.keyExchange) |
| } |
| if m.supportedVersion != 0 { |
| extensions.addU16(extensionSupportedVersions) |
| extensions.addU16(2) // Length |
| extensions.addU16(m.supportedVersion) |
| } |
| if len(m.supportedPoints) > 0 { |
| // http://tools.ietf.org/html/rfc4492#section-5.1.2 |
| extensions.addU16(extensionSupportedPoints) |
| supportedPointsList := extensions.addU16LengthPrefixed() |
| supportedPoints := supportedPointsList.addU8LengthPrefixed() |
| supportedPoints.addBytes(m.supportedPoints) |
| } |
| if len(m.supportedCurves) > 0 { |
| // https://tools.ietf.org/html/rfc8446#section-4.2.7 |
| extensions.addU16(extensionSupportedCurves) |
| supportedCurvesList := extensions.addU16LengthPrefixed() |
| supportedCurves := supportedCurvesList.addU16LengthPrefixed() |
| for _, curve := range m.supportedCurves { |
| supportedCurves.addU16(uint16(curve)) |
| } |
| } |
| if len(m.quicTransportParams) > 0 { |
| extensions.addU16(extensionQUICTransportParams) |
| params := extensions.addU16LengthPrefixed() |
| params.addBytes(m.quicTransportParams) |
| } |
| if m.hasEarlyData { |
| extensions.addU16(extensionEarlyData) |
| extensions.addBytes([]byte{0, 0}) |
| } |
| if m.serverNameAck { |
| extensions.addU16(extensionServerName) |
| extensions.addU16(0) // zero length |
| } |
| } |
| |
| func (m *serverExtensions) unmarshal(data byteReader, version uint16) bool { |
| // Reset all fields. |
| *m = serverExtensions{} |
| |
| if !checkDuplicateExtensions(data) { |
| return false |
| } |
| |
| for len(data) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !data.readU16(&extension) || |
| !data.readU16LengthPrefixed(&body) { |
| return false |
| } |
| switch extension { |
| case extensionNextProtoNeg: |
| m.nextProtoNeg = true |
| for len(body) > 0 { |
| var protocol []byte |
| if !body.readU8LengthPrefixedBytes(&protocol) { |
| return false |
| } |
| m.nextProtos = append(m.nextProtos, string(protocol)) |
| } |
| case extensionStatusRequest: |
| if len(body) != 0 { |
| return false |
| } |
| m.ocspStapling = true |
| case extensionSessionTicket: |
| if len(body) != 0 { |
| return false |
| } |
| m.ticketSupported = true |
| case extensionRenegotiationInfo: |
| if !body.readU8LengthPrefixedBytes(&m.secureRenegotiation) || len(body) != 0 { |
| return false |
| } |
| case extensionALPN: |
| var protocols, protocol byteReader |
| if !body.readU16LengthPrefixed(&protocols) || |
| len(body) != 0 || |
| !protocols.readU8LengthPrefixed(&protocol) || |
| len(protocols) != 0 { |
| return false |
| } |
| m.alpnProtocol = string(protocol) |
| m.alpnProtocolEmpty = len(protocol) == 0 |
| case extensionChannelID: |
| if len(body) != 0 { |
| return false |
| } |
| m.channelIDRequested = true |
| case extensionTokenBinding: |
| if !body.readU16(&m.tokenBindingVersion) || |
| !body.readU8LengthPrefixedBytes(&m.tokenBindingParams) || |
| len(m.tokenBindingParams) != 1 || |
| len(body) != 0 { |
| return false |
| } |
| case extensionExtendedMasterSecret: |
| if len(body) != 0 { |
| return false |
| } |
| m.extendedMasterSecret = true |
| case extensionUseSRTP: |
| var profiles, mki byteReader |
| if !body.readU16LengthPrefixed(&profiles) || |
| !profiles.readU16(&m.srtpProtectionProfile) || |
| len(profiles) != 0 || |
| !body.readU8LengthPrefixed(&mki) || |
| len(body) != 0 { |
| return false |
| } |
| m.srtpMasterKeyIdentifier = string(mki) |
| case extensionSignedCertificateTimestamp: |
| m.sctList = []byte(body) |
| case extensionCustom: |
| m.customExtension = string(body) |
| case extensionServerName: |
| if len(body) != 0 { |
| return false |
| } |
| m.serverNameAck = true |
| case extensionSupportedPoints: |
| // supported_points is illegal in TLS 1.3. |
| if version >= VersionTLS13 { |
| return false |
| } |
| // http://tools.ietf.org/html/rfc4492#section-5.5.2 |
| if !body.readU8LengthPrefixedBytes(&m.supportedPoints) || len(body) != 0 { |
| return false |
| } |
| case extensionSupportedCurves: |
| // The server can only send supported_curves in TLS 1.3. |
| if version < VersionTLS13 { |
| return false |
| } |
| case extensionQUICTransportParams: |
| m.quicTransportParams = body |
| case extensionEarlyData: |
| if version < VersionTLS13 || len(body) != 0 { |
| return false |
| } |
| m.hasEarlyData = true |
| default: |
| // Unknown extensions are illegal from the server. |
| return false |
| } |
| } |
| |
| return true |
| } |
| |
| type helloRetryRequestMsg struct { |
| raw []byte |
| vers uint16 |
| isServerHello bool |
| sessionId []byte |
| cipherSuite uint16 |
| compressionMethod uint8 |
| hasSelectedGroup bool |
| selectedGroup CurveID |
| cookie []byte |
| customExtension string |
| duplicateExtensions bool |
| } |
| |
| func (m *helloRetryRequestMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| retryRequestMsg := newByteBuilder() |
| retryRequestMsg.addU8(typeServerHello) |
| retryRequest := retryRequestMsg.addU24LengthPrefixed() |
| retryRequest.addU16(VersionTLS12) |
| retryRequest.addBytes(tls13HelloRetryRequest) |
| sessionId := retryRequest.addU8LengthPrefixed() |
| sessionId.addBytes(m.sessionId) |
| retryRequest.addU16(m.cipherSuite) |
| retryRequest.addU8(m.compressionMethod) |
| |
| extensions := retryRequest.addU16LengthPrefixed() |
| |
| count := 1 |
| if m.duplicateExtensions { |
| count = 2 |
| } |
| |
| for i := 0; i < count; i++ { |
| extensions.addU16(extensionSupportedVersions) |
| extensions.addU16(2) // Length |
| extensions.addU16(m.vers) |
| if m.hasSelectedGroup { |
| extensions.addU16(extensionKeyShare) |
| extensions.addU16(2) // length |
| extensions.addU16(uint16(m.selectedGroup)) |
| } |
| // m.cookie may be a non-nil empty slice for empty cookie tests. |
| if m.cookie != nil { |
| extensions.addU16(extensionCookie) |
| body := extensions.addU16LengthPrefixed() |
| body.addU16LengthPrefixed().addBytes(m.cookie) |
| } |
| if len(m.customExtension) > 0 { |
| extensions.addU16(extensionCustom) |
| extensions.addU16LengthPrefixed().addBytes([]byte(m.customExtension)) |
| } |
| } |
| |
| m.raw = retryRequestMsg.finish() |
| return m.raw |
| } |
| |
| func (m *helloRetryRequestMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| if !reader.readU16(&m.vers) { |
| return false |
| } |
| if m.isServerHello { |
| var random []byte |
| var compressionMethod byte |
| if !reader.readBytes(&random, 32) || |
| !reader.readU8LengthPrefixedBytes(&m.sessionId) || |
| !reader.readU16(&m.cipherSuite) || |
| !reader.readU8(&compressionMethod) || |
| compressionMethod != 0 { |
| return false |
| } |
| } else if !reader.readU16(&m.cipherSuite) { |
| return false |
| } |
| var extensions byteReader |
| if !reader.readU16LengthPrefixed(&extensions) || len(reader) != 0 { |
| return false |
| } |
| extensionsCopy := extensions |
| for len(extensionsCopy) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !extensionsCopy.readU16(&extension) || |
| !extensionsCopy.readU16LengthPrefixed(&body) { |
| return false |
| } |
| switch extension { |
| case extensionSupportedVersions: |
| if !m.isServerHello || |
| !body.readU16(&m.vers) || |
| len(body) != 0 { |
| return false |
| } |
| default: |
| } |
| } |
| for len(extensions) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !extensions.readU16(&extension) || |
| !extensions.readU16LengthPrefixed(&body) { |
| return false |
| } |
| switch extension { |
| case extensionSupportedVersions: |
| // Parsed above. |
| case extensionKeyShare: |
| var v uint16 |
| if !body.readU16(&v) || len(body) != 0 { |
| return false |
| } |
| m.hasSelectedGroup = true |
| m.selectedGroup = CurveID(v) |
| case extensionCookie: |
| if !body.readU16LengthPrefixedBytes(&m.cookie) || len(body) != 0 { |
| return false |
| } |
| default: |
| // Unknown extensions are illegal from the server. |
| return false |
| } |
| } |
| return true |
| } |
| |
| type certificateEntry struct { |
| data []byte |
| ocspResponse []byte |
| sctList []byte |
| duplicateExtensions bool |
| extraExtension []byte |
| delegatedCredential *delegatedCredential |
| } |
| |
| type delegatedCredential struct { |
| // https://tools.ietf.org/html/draft-ietf-tls-subcerts-03#section-3 |
| signedBytes []byte |
| lifetimeSecs uint32 |
| expectedCertVerifyAlgo signatureAlgorithm |
| pkixPublicKey []byte |
| algorithm signatureAlgorithm |
| signature []byte |
| } |
| |
| type certificateMsg struct { |
| raw []byte |
| hasRequestContext bool |
| requestContext []byte |
| certificates []certificateEntry |
| } |
| |
| func (m *certificateMsg) marshal() (x []byte) { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| certMsg := newByteBuilder() |
| certMsg.addU8(typeCertificate) |
| certificate := certMsg.addU24LengthPrefixed() |
| if m.hasRequestContext { |
| context := certificate.addU8LengthPrefixed() |
| context.addBytes(m.requestContext) |
| } |
| certificateList := certificate.addU24LengthPrefixed() |
| for _, cert := range m.certificates { |
| certEntry := certificateList.addU24LengthPrefixed() |
| certEntry.addBytes(cert.data) |
| if m.hasRequestContext { |
| extensions := certificateList.addU16LengthPrefixed() |
| count := 1 |
| if cert.duplicateExtensions { |
| count = 2 |
| } |
| |
| for i := 0; i < count; i++ { |
| if cert.ocspResponse != nil { |
| extensions.addU16(extensionStatusRequest) |
| body := extensions.addU16LengthPrefixed() |
| body.addU8(statusTypeOCSP) |
| response := body.addU24LengthPrefixed() |
| response.addBytes(cert.ocspResponse) |
| } |
| |
| if cert.sctList != nil { |
| extensions.addU16(extensionSignedCertificateTimestamp) |
| extension := extensions.addU16LengthPrefixed() |
| extension.addBytes(cert.sctList) |
| } |
| } |
| if cert.extraExtension != nil { |
| extensions.addBytes(cert.extraExtension) |
| } |
| } |
| } |
| |
| m.raw = certMsg.finish() |
| return m.raw |
| } |
| |
| func (m *certificateMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| |
| if m.hasRequestContext && !reader.readU8LengthPrefixedBytes(&m.requestContext) { |
| return false |
| } |
| |
| var certs byteReader |
| if !reader.readU24LengthPrefixed(&certs) || len(reader) != 0 { |
| return false |
| } |
| m.certificates = nil |
| for len(certs) > 0 { |
| var cert certificateEntry |
| if !certs.readU24LengthPrefixedBytes(&cert.data) { |
| return false |
| } |
| if m.hasRequestContext { |
| var extensions byteReader |
| if !certs.readU16LengthPrefixed(&extensions) || !checkDuplicateExtensions(extensions) { |
| return false |
| } |
| for len(extensions) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !extensions.readU16(&extension) || |
| !extensions.readU16LengthPrefixed(&body) { |
| return false |
| } |
| switch extension { |
| case extensionStatusRequest: |
| var statusType byte |
| if !body.readU8(&statusType) || |
| statusType != statusTypeOCSP || |
| !body.readU24LengthPrefixedBytes(&cert.ocspResponse) || |
| len(body) != 0 { |
| return false |
| } |
| case extensionSignedCertificateTimestamp: |
| cert.sctList = []byte(body) |
| case extensionDelegatedCredentials: |
| // https://tools.ietf.org/html/draft-ietf-tls-subcerts-03#section-3 |
| if cert.delegatedCredential != nil { |
| return false |
| } |
| |
| dc := new(delegatedCredential) |
| origBody := body |
| var expectedCertVerifyAlgo, algorithm uint16 |
| |
| if !body.readU32(&dc.lifetimeSecs) || |
| !body.readU16(&expectedCertVerifyAlgo) || |
| !body.readU24LengthPrefixedBytes(&dc.pkixPublicKey) || |
| !body.readU16(&algorithm) || |
| !body.readU16LengthPrefixedBytes(&dc.signature) || |
| len(body) != 0 { |
| return false |
| } |
| |
| dc.expectedCertVerifyAlgo = signatureAlgorithm(expectedCertVerifyAlgo) |
| dc.algorithm = signatureAlgorithm(algorithm) |
| dc.signedBytes = []byte(origBody)[:4+2+3+len(dc.pkixPublicKey)] |
| cert.delegatedCredential = dc |
| default: |
| return false |
| } |
| } |
| } |
| m.certificates = append(m.certificates, cert) |
| } |
| |
| return true |
| } |
| |
| type compressedCertificateMsg struct { |
| raw []byte |
| algID uint16 |
| uncompressedLength uint32 |
| compressed []byte |
| } |
| |
| func (m *compressedCertificateMsg) marshal() (x []byte) { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| certMsg := newByteBuilder() |
| certMsg.addU8(typeCompressedCertificate) |
| certificate := certMsg.addU24LengthPrefixed() |
| certificate.addU16(m.algID) |
| certificate.addU24(int(m.uncompressedLength)) |
| compressed := certificate.addU24LengthPrefixed() |
| compressed.addBytes(m.compressed) |
| |
| m.raw = certMsg.finish() |
| return m.raw |
| } |
| |
| func (m *compressedCertificateMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| |
| if !reader.readU16(&m.algID) || |
| !reader.readU24(&m.uncompressedLength) || |
| !reader.readU24LengthPrefixedBytes(&m.compressed) || |
| len(reader) != 0 { |
| return false |
| } |
| |
| if m.uncompressedLength >= 1<<17 { |
| return false |
| } |
| |
| return true |
| } |
| |
| type serverKeyExchangeMsg struct { |
| raw []byte |
| key []byte |
| } |
| |
| func (m *serverKeyExchangeMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| msg := newByteBuilder() |
| msg.addU8(typeServerKeyExchange) |
| msg.addU24LengthPrefixed().addBytes(m.key) |
| m.raw = msg.finish() |
| return m.raw |
| } |
| |
| func (m *serverKeyExchangeMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| if len(data) < 4 { |
| return false |
| } |
| m.key = data[4:] |
| return true |
| } |
| |
| type certificateStatusMsg struct { |
| raw []byte |
| statusType uint8 |
| response []byte |
| } |
| |
| func (m *certificateStatusMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| var x []byte |
| if m.statusType == statusTypeOCSP { |
| msg := newByteBuilder() |
| msg.addU8(typeCertificateStatus) |
| body := msg.addU24LengthPrefixed() |
| body.addU8(statusTypeOCSP) |
| body.addU24LengthPrefixed().addBytes(m.response) |
| x = msg.finish() |
| } else { |
| x = []byte{typeCertificateStatus, 0, 0, 1, m.statusType} |
| } |
| |
| m.raw = x |
| return x |
| } |
| |
| func (m *certificateStatusMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| if !reader.readU8(&m.statusType) || |
| m.statusType != statusTypeOCSP || |
| !reader.readU24LengthPrefixedBytes(&m.response) || |
| len(reader) != 0 { |
| return false |
| } |
| return true |
| } |
| |
| type serverHelloDoneMsg struct{} |
| |
| func (m *serverHelloDoneMsg) marshal() []byte { |
| x := make([]byte, 4) |
| x[0] = typeServerHelloDone |
| return x |
| } |
| |
| func (m *serverHelloDoneMsg) unmarshal(data []byte) bool { |
| return len(data) == 4 |
| } |
| |
| type clientKeyExchangeMsg struct { |
| raw []byte |
| ciphertext []byte |
| } |
| |
| func (m *clientKeyExchangeMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| msg := newByteBuilder() |
| msg.addU8(typeClientKeyExchange) |
| msg.addU24LengthPrefixed().addBytes(m.ciphertext) |
| m.raw = msg.finish() |
| return m.raw |
| } |
| |
| func (m *clientKeyExchangeMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| if len(data) < 4 { |
| return false |
| } |
| l := int(data[1])<<16 | int(data[2])<<8 | int(data[3]) |
| if l != len(data)-4 { |
| return false |
| } |
| m.ciphertext = data[4:] |
| return true |
| } |
| |
| type finishedMsg struct { |
| raw []byte |
| verifyData []byte |
| } |
| |
| func (m *finishedMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| msg := newByteBuilder() |
| msg.addU8(typeFinished) |
| msg.addU24LengthPrefixed().addBytes(m.verifyData) |
| m.raw = msg.finish() |
| return m.raw |
| } |
| |
| func (m *finishedMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| if len(data) < 4 { |
| return false |
| } |
| m.verifyData = data[4:] |
| return true |
| } |
| |
| type nextProtoMsg struct { |
| raw []byte |
| proto string |
| } |
| |
| func (m *nextProtoMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| padding := 32 - (len(m.proto)+2)%32 |
| |
| msg := newByteBuilder() |
| msg.addU8(typeNextProtocol) |
| body := msg.addU24LengthPrefixed() |
| body.addU8LengthPrefixed().addBytes([]byte(m.proto)) |
| body.addU8LengthPrefixed().addBytes(make([]byte, padding)) |
| m.raw = msg.finish() |
| return m.raw |
| } |
| |
| func (m *nextProtoMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| var proto, padding []byte |
| if !reader.readU8LengthPrefixedBytes(&proto) || |
| !reader.readU8LengthPrefixedBytes(&padding) || |
| len(reader) != 0 { |
| return false |
| } |
| m.proto = string(proto) |
| |
| // Padding is not meant to be checked normally, but as this is a testing |
| // implementation, we check the padding is as expected. |
| if len(padding) != 32-(len(m.proto)+2)%32 { |
| return false |
| } |
| for _, v := range padding { |
| if v != 0 { |
| return false |
| } |
| } |
| |
| return true |
| } |
| |
| type certificateRequestMsg struct { |
| raw []byte |
| vers uint16 |
| // hasSignatureAlgorithm indicates whether this message includes a list |
| // of signature and hash functions. This change was introduced with TLS |
| // 1.2. |
| hasSignatureAlgorithm bool |
| // hasRequestContext indicates whether this message includes a context |
| // field instead of certificateTypes. This change was introduced with |
| // TLS 1.3. |
| hasRequestContext bool |
| |
| certificateTypes []byte |
| requestContext []byte |
| signatureAlgorithms []signatureAlgorithm |
| signatureAlgorithmsCert []signatureAlgorithm |
| certificateAuthorities [][]byte |
| hasCAExtension bool |
| customExtension uint16 |
| } |
| |
| func (m *certificateRequestMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| // See http://tools.ietf.org/html/rfc4346#section-7.4.4 |
| builder := newByteBuilder() |
| builder.addU8(typeCertificateRequest) |
| body := builder.addU24LengthPrefixed() |
| |
| if m.hasRequestContext { |
| requestContext := body.addU8LengthPrefixed() |
| requestContext.addBytes(m.requestContext) |
| extensions := newByteBuilder() |
| extensions = body.addU16LengthPrefixed() |
| if m.hasSignatureAlgorithm { |
| extensions.addU16(extensionSignatureAlgorithms) |
| signatureAlgorithms := extensions.addU16LengthPrefixed().addU16LengthPrefixed() |
| for _, sigAlg := range m.signatureAlgorithms { |
| signatureAlgorithms.addU16(uint16(sigAlg)) |
| } |
| } |
| if len(m.signatureAlgorithmsCert) > 0 { |
| extensions.addU16(extensionSignatureAlgorithmsCert) |
| signatureAlgorithmsCert := extensions.addU16LengthPrefixed().addU16LengthPrefixed() |
| for _, sigAlg := range m.signatureAlgorithmsCert { |
| signatureAlgorithmsCert.addU16(uint16(sigAlg)) |
| } |
| } |
| if len(m.certificateAuthorities) > 0 { |
| extensions.addU16(extensionCertificateAuthorities) |
| certificateAuthorities := extensions.addU16LengthPrefixed().addU16LengthPrefixed() |
| for _, ca := range m.certificateAuthorities { |
| caEntry := certificateAuthorities.addU16LengthPrefixed() |
| caEntry.addBytes(ca) |
| } |
| } |
| |
| if m.customExtension > 0 { |
| extensions.addU16(m.customExtension) |
| extensions.addU16LengthPrefixed() |
| } |
| } else { |
| certificateTypes := body.addU8LengthPrefixed() |
| certificateTypes.addBytes(m.certificateTypes) |
| |
| if m.hasSignatureAlgorithm { |
| signatureAlgorithms := body.addU16LengthPrefixed() |
| for _, sigAlg := range m.signatureAlgorithms { |
| signatureAlgorithms.addU16(uint16(sigAlg)) |
| } |
| } |
| |
| certificateAuthorities := body.addU16LengthPrefixed() |
| for _, ca := range m.certificateAuthorities { |
| caEntry := certificateAuthorities.addU16LengthPrefixed() |
| caEntry.addBytes(ca) |
| } |
| } |
| |
| m.raw = builder.finish() |
| return m.raw |
| } |
| |
| func parseCAs(reader *byteReader, out *[][]byte) bool { |
| var cas byteReader |
| if !reader.readU16LengthPrefixed(&cas) { |
| return false |
| } |
| for len(cas) > 0 { |
| var ca []byte |
| if !cas.readU16LengthPrefixedBytes(&ca) { |
| return false |
| } |
| *out = append(*out, ca) |
| } |
| return true |
| } |
| |
| func (m *certificateRequestMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| reader := byteReader(data[4:]) |
| |
| if m.hasRequestContext { |
| var extensions byteReader |
| if !reader.readU8LengthPrefixedBytes(&m.requestContext) || |
| !reader.readU16LengthPrefixed(&extensions) || |
| len(reader) != 0 || |
| !checkDuplicateExtensions(extensions) { |
| return false |
| } |
| for len(extensions) > 0 { |
| var extension uint16 |
| var body byteReader |
| if !extensions.readU16(&extension) || |
| !extensions.readU16LengthPrefixed(&body) { |
| return false |
| } |
| switch extension { |
| case extensionSignatureAlgorithms: |
| if !parseSignatureAlgorithms(&body, &m.signatureAlgorithms, false) || len(body) != 0 { |
| return false |
| } |
| case extensionSignatureAlgorithmsCert: |
| if !parseSignatureAlgorithms(&body, &m.signatureAlgorithmsCert, false) || len(body) != 0 { |
| return false |
| } |
| case extensionCertificateAuthorities: |
| if !parseCAs(&body, &m.certificateAuthorities) || len(body) != 0 { |
| return false |
| } |
| m.hasCAExtension = true |
| } |
| } |
| } else { |
| if !reader.readU8LengthPrefixedBytes(&m.certificateTypes) { |
| return false |
| } |
| // In TLS 1.2, the supported_signature_algorithms field in |
| // CertificateRequest may be empty. |
| if m.hasSignatureAlgorithm && !parseSignatureAlgorithms(&reader, &m.signatureAlgorithms, true) { |
| return false |
| } |
| if !parseCAs(&reader, &m.certificateAuthorities) || |
| len(reader) != 0 { |
| return false |
| } |
| } |
| |
| return true |
| } |
| |
| type certificateVerifyMsg struct { |
| raw []byte |
| hasSignatureAlgorithm bool |
| signatureAlgorithm signatureAlgorithm |
| signature []byte |
| } |
| |
| func (m *certificateVerifyMsg) marshal() (x []byte) { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| // See http://tools.ietf.org/html/rfc4346#section-7.4.8 |
| siglength := len(m.signature) |
| length := 2 + siglength |
| if m.hasSignatureAlgorithm { |
| length += 2 |
| } |
| x = make([]byte, 4+length) |
| x[0] = typeCertificateVerify |
| x[1] = uint8(length >> 16) |
| x[2] = uint8(length >> 8) |
| x[3] = uint8(length) |
| y := x[4:] |
| if m.hasSignatureAlgorithm { |
| y[0] = byte(m.signatureAlgorithm >> 8) |
| y[1] = byte(m.signatureAlgorithm) |
| y = y[2:] |
| } |
| y[0] = uint8(siglength >> 8) |
| y[1] = uint8(siglength) |
| copy(y[2:], m.signature) |
| |
| m.raw = x |
| |
| return |
| } |
| |
| func (m *certificateVerifyMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| |
| if len(data) < 6 { |
| return false |
| } |
| |
| length := uint32(data[1])<<16 | uint32(data[2])<<8 | uint32(data[3]) |
| if uint32(len(data))-4 != length { |
| return false |
| } |
| |
| data = data[4:] |
| if m.hasSignatureAlgorithm { |
| m.signatureAlgorithm = signatureAlgorithm(data[0])<<8 | signatureAlgorithm(data[1]) |
| data = data[2:] |
| } |
| |
| if len(data) < 2 { |
| return false |
| } |
| siglength := int(data[0])<<8 + int(data[1]) |
| data = data[2:] |
| if len(data) != siglength { |
| return false |
| } |
| |
| m.signature = data |
| |
| return true |
| } |
| |
| type newSessionTicketMsg struct { |
| raw []byte |
| vers uint16 |
| isDTLS bool |
| ticketLifetime uint32 |
| ticketAgeAdd uint32 |
| ticketNonce []byte |
| ticket []byte |
| maxEarlyDataSize uint32 |
| customExtension string |
| duplicateEarlyDataExtension bool |
| hasGREASEExtension bool |
| } |
| |
| func (m *newSessionTicketMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| version, ok := wireToVersion(m.vers, m.isDTLS) |
| if !ok { |
| panic("unknown version") |
| } |
| |
| // See http://tools.ietf.org/html/rfc5077#section-3.3 |
| ticketMsg := newByteBuilder() |
| ticketMsg.addU8(typeNewSessionTicket) |
| body := ticketMsg.addU24LengthPrefixed() |
| body.addU32(m.ticketLifetime) |
| if version >= VersionTLS13 { |
| body.addU32(m.ticketAgeAdd) |
| body.addU8LengthPrefixed().addBytes(m.ticketNonce) |
| } |
| |
| ticket := body.addU16LengthPrefixed() |
| ticket.addBytes(m.ticket) |
| |
| if version >= VersionTLS13 { |
| extensions := body.addU16LengthPrefixed() |
| if m.maxEarlyDataSize > 0 { |
| extensions.addU16(extensionEarlyData) |
| extensions.addU16LengthPrefixed().addU32(m.maxEarlyDataSize) |
| if m.duplicateEarlyDataExtension { |
| extensions.addU16(extensionEarlyData) |
| extensions.addU16LengthPrefixed().addU32(m.maxEarlyDataSize) |
| } |
| } |
| if len(m.customExtension) > 0 { |
| extensions.addU16(extensionCustom) |
| extensions.addU16LengthPrefixed().addBytes([]byte(m.customExtension)) |
| } |
| } |
| |
| m.raw = ticketMsg.finish() |
| return m.raw |
| } |
| |
| func (m *newSessionTicketMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| |
| version, ok := wireToVersion(m.vers, m.isDTLS) |
| if !ok { |
| panic("unknown version") |
| } |
| |
| if len(data) < 8 { |
| return false |
| } |
| m.ticketLifetime = uint32(data[4])<<24 | uint32(data[5])<<16 | uint32(data[6])<<8 | uint32(data[7]) |
| data = data[8:] |
| |
| if version >= VersionTLS13 { |
| if len(data) < 4 { |
| return false |
| } |
| m.ticketAgeAdd = uint32(data[0])<<24 | uint32(data[1])<<16 | uint32(data[2])<<8 | uint32(data[3]) |
| data = data[4:] |
| nonceLen := int(data[0]) |
| data = data[1:] |
| if len(data) < nonceLen { |
| return false |
| } |
| m.ticketNonce = data[:nonceLen] |
| data = data[nonceLen:] |
| } |
| |
| if len(data) < 2 { |
| return false |
| } |
| ticketLen := int(data[0])<<8 + int(data[1]) |
| data = data[2:] |
| if len(data) < ticketLen { |
| return false |
| } |
| |
| if version >= VersionTLS13 && ticketLen == 0 { |
| return false |
| } |
| |
| m.ticket = data[:ticketLen] |
| data = data[ticketLen:] |
| |
| if version >= VersionTLS13 { |
| if len(data) < 2 { |
| return false |
| } |
| |
| extensionsLength := int(data[0])<<8 | int(data[1]) |
| data = data[2:] |
| if extensionsLength != len(data) { |
| return false |
| } |
| |
| for len(data) != 0 { |
| if len(data) < 4 { |
| return false |
| } |
| extension := uint16(data[0])<<8 | uint16(data[1]) |
| length := int(data[2])<<8 | int(data[3]) |
| data = data[4:] |
| if len(data) < length { |
| return false |
| } |
| |
| switch extension { |
| case extensionEarlyData: |
| if length != 4 { |
| return false |
| } |
| m.maxEarlyDataSize = uint32(data[0])<<24 | uint32(data[1])<<16 | uint32(data[2])<<8 | uint32(data[3]) |
| default: |
| if isGREASEValue(extension) { |
| m.hasGREASEExtension = true |
| } |
| } |
| |
| data = data[length:] |
| } |
| } |
| |
| if len(data) > 0 { |
| return false |
| } |
| |
| return true |
| } |
| |
| type v2ClientHelloMsg struct { |
| raw []byte |
| vers uint16 |
| cipherSuites []uint16 |
| sessionId []byte |
| challenge []byte |
| } |
| |
| func (m *v2ClientHelloMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| length := 1 + 2 + 2 + 2 + 2 + len(m.cipherSuites)*3 + len(m.sessionId) + len(m.challenge) |
| |
| x := make([]byte, length) |
| x[0] = 1 |
| x[1] = uint8(m.vers >> 8) |
| x[2] = uint8(m.vers) |
| x[3] = uint8((len(m.cipherSuites) * 3) >> 8) |
| x[4] = uint8(len(m.cipherSuites) * 3) |
| x[5] = uint8(len(m.sessionId) >> 8) |
| x[6] = uint8(len(m.sessionId)) |
| x[7] = uint8(len(m.challenge) >> 8) |
| x[8] = uint8(len(m.challenge)) |
| y := x[9:] |
| for i, spec := range m.cipherSuites { |
| y[i*3] = 0 |
| y[i*3+1] = uint8(spec >> 8) |
| y[i*3+2] = uint8(spec) |
| } |
| y = y[len(m.cipherSuites)*3:] |
| copy(y, m.sessionId) |
| y = y[len(m.sessionId):] |
| copy(y, m.challenge) |
| |
| m.raw = x |
| |
| return x |
| } |
| |
| type helloVerifyRequestMsg struct { |
| raw []byte |
| vers uint16 |
| cookie []byte |
| } |
| |
| func (m *helloVerifyRequestMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| length := 2 + 1 + len(m.cookie) |
| |
| x := make([]byte, 4+length) |
| x[0] = typeHelloVerifyRequest |
| x[1] = uint8(length >> 16) |
| x[2] = uint8(length >> 8) |
| x[3] = uint8(length) |
| vers := m.vers |
| x[4] = uint8(vers >> 8) |
| x[5] = uint8(vers) |
| x[6] = uint8(len(m.cookie)) |
| copy(x[7:7+len(m.cookie)], m.cookie) |
| |
| return x |
| } |
| |
| func (m *helloVerifyRequestMsg) unmarshal(data []byte) bool { |
| if len(data) < 4+2+1 { |
| return false |
| } |
| m.raw = data |
| m.vers = uint16(data[4])<<8 | uint16(data[5]) |
| cookieLen := int(data[6]) |
| if cookieLen > 32 || len(data) != 7+cookieLen { |
| return false |
| } |
| m.cookie = data[7 : 7+cookieLen] |
| |
| return true |
| } |
| |
| type channelIDMsg struct { |
| raw []byte |
| channelID []byte |
| } |
| |
| func (m *channelIDMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| length := 2 + 2 + len(m.channelID) |
| |
| x := make([]byte, 4+length) |
| x[0] = typeChannelID |
| x[1] = uint8(length >> 16) |
| x[2] = uint8(length >> 8) |
| x[3] = uint8(length) |
| x[4] = uint8(extensionChannelID >> 8) |
| x[5] = uint8(extensionChannelID & 0xff) |
| x[6] = uint8(len(m.channelID) >> 8) |
| x[7] = uint8(len(m.channelID) & 0xff) |
| copy(x[8:], m.channelID) |
| |
| return x |
| } |
| |
| func (m *channelIDMsg) unmarshal(data []byte) bool { |
| if len(data) != 4+2+2+128 { |
| return false |
| } |
| m.raw = data |
| if (uint16(data[4])<<8)|uint16(data[5]) != extensionChannelID { |
| return false |
| } |
| if int(data[6])<<8|int(data[7]) != 128 { |
| return false |
| } |
| m.channelID = data[4+2+2:] |
| |
| return true |
| } |
| |
| type helloRequestMsg struct { |
| } |
| |
| func (*helloRequestMsg) marshal() []byte { |
| return []byte{typeHelloRequest, 0, 0, 0} |
| } |
| |
| func (*helloRequestMsg) unmarshal(data []byte) bool { |
| return len(data) == 4 |
| } |
| |
| type keyUpdateMsg struct { |
| raw []byte |
| keyUpdateRequest byte |
| } |
| |
| func (m *keyUpdateMsg) marshal() []byte { |
| if m.raw != nil { |
| return m.raw |
| } |
| |
| return []byte{typeKeyUpdate, 0, 0, 1, m.keyUpdateRequest} |
| } |
| |
| func (m *keyUpdateMsg) unmarshal(data []byte) bool { |
| m.raw = data |
| |
| if len(data) != 5 { |
| return false |
| } |
| |
| length := int(data[1])<<16 | int(data[2])<<8 | int(data[3]) |
| if len(data)-4 != length { |
| return false |
| } |
| |
| m.keyUpdateRequest = data[4] |
| return m.keyUpdateRequest == keyUpdateNotRequested || m.keyUpdateRequest == keyUpdateRequested |
| } |
| |
| type endOfEarlyDataMsg struct { |
| nonEmpty bool |
| } |
| |
| func (m *endOfEarlyDataMsg) marshal() []byte { |
| if m.nonEmpty { |
| return []byte{typeEndOfEarlyData, 0, 0, 1, 42} |
| } |
| return []byte{typeEndOfEarlyData, 0, 0, 0} |
| } |
| |
| func (*endOfEarlyDataMsg) unmarshal(data []byte) bool { |
| return len(data) == 4 |
| } |
| |
| // ssl3NoCertificateMsg is a dummy message to handle SSL 3.0 using a warning |
| // alert in the handshake. |
| type ssl3NoCertificateMsg struct{} |
| |
| func eqUint16s(x, y []uint16) bool { |
| if len(x) != len(y) { |
| return false |
| } |
| for i, v := range x { |
| if y[i] != v { |
| return false |
| } |
| } |
| return true |
| } |
| |
| func eqCurveIDs(x, y []CurveID) bool { |
| if len(x) != len(y) { |
| return false |
| } |
| for i, v := range x { |
| if y[i] != v { |
| return false |
| } |
| } |
| return true |
| } |
| |
| func eqStrings(x, y []string) bool { |
| if len(x) != len(y) { |
| return false |
| } |
| for i, v := range x { |
| if y[i] != v { |
| return false |
| } |
| } |
| return true |
| } |
| |
| func eqByteSlices(x, y [][]byte) bool { |
| if len(x) != len(y) { |
| return false |
| } |
| for i, v := range x { |
| if !bytes.Equal(v, y[i]) { |
| return false |
| } |
| } |
| return true |
| } |
| |
| func eqSignatureAlgorithms(x, y []signatureAlgorithm) bool { |
| if len(x) != len(y) { |
| return false |
| } |
| for i, v := range x { |
| v2 := y[i] |
| if v != v2 { |
| return false |
| } |
| } |
| return true |
| } |
| |
| func eqKeyShareEntryLists(x, y []keyShareEntry) bool { |
| if len(x) != len(y) { |
| return false |
| } |
| for i, v := range x { |
| if y[i].group != v.group || !bytes.Equal(y[i].keyExchange, v.keyExchange) { |
| return false |
| } |
| } |
| return true |
| |
| } |
| |
| func eqPSKIdentityLists(x, y []pskIdentity) bool { |
| if len(x) != len(y) { |
| return false |
| } |
| for i, v := range x { |
| if !bytes.Equal(y[i].ticket, v.ticket) || y[i].obfuscatedTicketAge != v.obfuscatedTicketAge { |
| return false |
| } |
| } |
| return true |
| |
| } |
