Faketls works now

This commit is contained in:
9seconds
2021-03-26 12:15:26 +03:00
parent 6d92d5fe21
commit f0efa4697e
4 changed files with 82 additions and 9 deletions
+70
View File
@@ -0,0 +1,70 @@
package faketls
import (
"bytes"
"fmt"
"net"
"github.com/9seconds/mtg/v2/mtglib/internal/faketls/record"
)
type Conn struct {
net.Conn
readBuffer bytes.Buffer
}
func (c *Conn) Read(p []byte) (int, error) {
if n, _ := c.readBuffer.Read(p); n > 0 {
return n, nil
}
rec := record.AcquireRecord()
defer record.ReleaseRecord(rec)
for {
if err := rec.Read(c.Conn); err != nil {
return 0, err // nolint: wrapcheck
}
switch rec.Type { // nolint: exhaustive
case record.TypeChangeCipherSpec:
case record.TypeApplicationData:
rec.Payload.WriteTo(&c.readBuffer)
n, err := c.readBuffer.Read(p)
return n, err
default:
return 0, fmt.Errorf("unsupported record type %v", rec.Type)
}
}
}
func (c *Conn) Write(p []byte) (int, error) {
rec := record.AcquireRecord()
defer record.ReleaseRecord(rec)
rec.Type = record.TypeApplicationData
rec.Version = record.Version12
written := 0
for len(p) > 0 {
chunkSize := record.TLSMaxRecordSize
if chunkSize > len(p) {
chunkSize = len(p)
}
rec.Payload.Reset()
rec.Payload.Write(p[:chunkSize])
if err := rec.Dump(c.Conn); err != nil {
return 0, err
}
written += chunkSize
p = p[chunkSize:]
}
return written, nil
}
+2
View File
@@ -2,6 +2,8 @@ package record
import "fmt"
const TLSMaxRecordSize = 65535 // max uint16
type Type uint8
const (