@@ -46,6 +46,7 @@ type Conn struct {
46
46
closeErr error
47
47
closed chan struct {}
48
48
49
+ // messageWriter state.
49
50
// writeMsgLock is acquired to write a data message.
50
51
writeMsgLock chan struct {}
51
52
// writeFrameLock is acquired to write a single frame.
@@ -72,11 +73,11 @@ type Conn struct {
72
73
readHeaderBuf []byte
73
74
controlPayloadBuf []byte
74
75
75
- // messageReader state
76
- readMsgCtx context.Context
77
- readMsgHeader header
78
- readFrameEOF bool
79
- readMaskPos int
76
+ // messageReader state.
77
+ readerMsgCtx context.Context
78
+ readerMsgHeader header
79
+ readerFrameEOF bool
80
+ readerMaskPos int
80
81
81
82
setReadTimeout chan context.Context
82
83
setWriteTimeout chan context.Context
@@ -360,7 +361,7 @@ func (c *Conn) Reader(ctx context.Context) (MessageType, io.Reader, error) {
360
361
}
361
362
362
363
func (c * Conn ) reader (ctx context.Context ) (MessageType , io.Reader , error ) {
363
- if c .activeReader != nil && ! c .readFrameEOF {
364
+ if c .activeReader != nil && ! c .readerFrameEOF {
364
365
// The only way we know for sure the previous reader is not yet complete is
365
366
// if there is an active frame not yet fully read.
366
367
// Otherwise, a user may have read the last byte but not the EOF if the EOF
@@ -396,10 +397,10 @@ func (c *Conn) reader(ctx context.Context) (MessageType, io.Reader, error) {
396
397
return 0 , nil , err
397
398
}
398
399
399
- c .readMsgCtx = ctx
400
- c .readMsgHeader = h
401
- c .readFrameEOF = false
402
- c .readMaskPos = 0
400
+ c .readerMsgCtx = ctx
401
+ c .readerMsgHeader = h
402
+ c .readerFrameEOF = false
403
+ c .readerMaskPos = 0
403
404
c .readMsgLeft = c .msgReadLimit
404
405
405
406
r := & messageReader {
@@ -468,8 +469,8 @@ func (r *messageReader) read(p []byte) (int, error) {
468
469
p = p [:r .c .readMsgLeft ]
469
470
}
470
471
471
- if r .c .readFrameEOF {
472
- h , err := r .c .readTillMsg (r .c .readMsgCtx )
472
+ if r .c .readerFrameEOF {
473
+ h , err := r .c .readTillMsg (r .c .readerMsgCtx )
473
474
if err != nil {
474
475
return 0 , err
475
476
}
@@ -480,31 +481,31 @@ func (r *messageReader) read(p []byte) (int, error) {
480
481
return 0 , err
481
482
}
482
483
483
- r .c .readMsgHeader = h
484
- r .c .readFrameEOF = false
485
- r .c .readMaskPos = 0
484
+ r .c .readerMsgHeader = h
485
+ r .c .readerFrameEOF = false
486
+ r .c .readerMaskPos = 0
486
487
}
487
488
488
- h := r .c .readMsgHeader
489
+ h := r .c .readerMsgHeader
489
490
if int64 (len (p )) > h .payloadLength {
490
491
p = p [:h .payloadLength ]
491
492
}
492
493
493
- n , err := r .c .readFramePayload (r .c .readMsgCtx , p )
494
+ n , err := r .c .readFramePayload (r .c .readerMsgCtx , p )
494
495
495
496
h .payloadLength -= int64 (n )
496
497
r .c .readMsgLeft -= int64 (n )
497
498
if h .masked {
498
- r .c .readMaskPos = fastXOR (h .maskKey , r .c .readMaskPos , p )
499
+ r .c .readerMaskPos = fastXOR (h .maskKey , r .c .readerMaskPos , p )
499
500
}
500
- r .c .readMsgHeader = h
501
+ r .c .readerMsgHeader = h
501
502
502
503
if err != nil {
503
504
return n , err
504
505
}
505
506
506
507
if h .payloadLength == 0 {
507
- r .c .readFrameEOF = true
508
+ r .c .readerFrameEOF = true
508
509
509
510
if h .fin {
510
511
r .c .activeReader = nil
0 commit comments