Get rid of magic constants

This commit is contained in:
9seconds
2019-11-04 21:36:59 +03:00
parent 1e428e13fc
commit fbbf9b1634
+8 -4
View File
@@ -19,7 +19,11 @@ var (
fakeTLSWritePrefix = []byte{0x17, 0x03, 0x03} fakeTLSWritePrefix = []byte{0x17, 0x03, 0x03}
) )
const faketlsMaxChunkSize = 16384 + 24 const (
faketlsMaxChunkSize = 16384 + 24
faketlsRecordTypeApplicationData = 0x17
faketlsRecordTypeCCS = 0x14
)
type wrapperFakeTLS struct { type wrapperFakeTLS struct {
bufferedReader bufferedReader
@@ -106,15 +110,15 @@ func NewFakeTLS(socket conntypes.StreamReadWriteCloser) conntypes.StreamReadWrit
faketls.readFunc = func() ([]byte, error) { faketls.readFunc = func() ([]byte, error) {
data := &bytes.Buffer{} data := &bytes.Buffer{}
buf := [2]byte{} buf := [2]byte{}
recordType := byte(0x14) recordType := byte(faketlsRecordTypeCCS)
for recordType == 0x14 { for recordType == faketlsRecordTypeCCS {
if _, err := io.ReadFull(faketls.parent, buf[:1]); err != nil { if _, err := io.ReadFull(faketls.parent, buf[:1]); err != nil {
return nil, fmt.Errorf("cannot read record type: %w", err) return nil, fmt.Errorf("cannot read record type: %w", err)
} }
switch buf[0] { switch buf[0] {
case 0x14, 0x17: case faketlsRecordTypeCCS, faketlsRecordTypeApplicationData:
recordType = buf[0] recordType = buf[0]
default: default:
return nil, fmt.Errorf("incorrect record type %v", buf[0]) return nil, fmt.Errorf("incorrect record type %v", buf[0])