Archives/Cloak
2
0
Fork 0
mirror of https://github.com/cbeuw/Cloak.git synced 2024-11-05 21:20:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

TLSConn.Write returns correct length

Browse Source
This commit is contained in:
Andy Wang 2020-04-10 16:15:23 +01:00
parent 78b27c0705
commit 8c9bab2a6f
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
internal/common/tls.go
View File

@ -11,6 +11,8 @@ const (
VersionTLS11 = 0x0301 VersionTLS11 = 0x0301
VersionTLS13 = 0x0303 VersionTLS13 = 0x0303
recordLayerLength = 5
Handshake = 22 Handshake = 22
ApplicationData = 23 ApplicationData = 23
) )
@ -55,7 +57,7 @@ func (tls *TLSConn) Read(buffer []byte) (n int, err error) {
// a single message can also be segmented due to MTU of the IP layer. // a single message can also be segmented due to MTU of the IP layer.
// This function guareentees a single TLS message to be read and everything // This function guareentees a single TLS message to be read and everything
// else is left in the buffer. // else is left in the buffer.
_, err = io.ReadFull(tls.Conn, buffer[:5]) _, err = io.ReadFull(tls.Conn, buffer[:recordLayerLength])
if err != nil { if err != nil {
return return
} }
@ -71,7 +73,8 @@ func (tls *TLSConn) Read(buffer []byte) (n int, err error) {
func (tls *TLSConn) Write(in []byte) (n int, err error) { func (tls *TLSConn) Write(in []byte) (n int, err error) {
// TODO: write record layer directly first? // TODO: write record layer directly first?
toWrite := AddRecordLayer(in, ApplicationData, VersionTLS13) toWrite := AddRecordLayer(in, ApplicationData, VersionTLS13)
return tls.Conn.Write(toWrite) n, err = tls.Conn.Write(toWrite)
return n - recordLayerLength, err
} }
func (tls *TLSConn) Close() error { func (tls *TLSConn) Close() error {