opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

transport: Fix a data race when headers are received while the stream is being closed (#1814)

Browse Source
This commit is contained in:
lyuxuan
2018-01-19 09:37:59 -08:00
committed by dfawley
parent 46bef23bc3
commit b71aced4a2
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
transport/http2_client.go
View File

@ -1116,13 +1116,14 @@ func (t *http2Client) operateHeaders(frame *http2.MetaHeadersFrame) {
}() }()
s.mu.Lock() s.mu.Lock()
if !s.headerDone {
// Headers frame is not actually a trailers-only frame.
if !endStream { if !endStream {
s.recvCompress = state.encoding s.recvCompress = state.encoding
} if len(state.mdata) > 0 {
if !s.headerDone {
if !endStream && len(state.mdata) > 0 {
s.header = state.mdata s.header = state.mdata
} }
}
close(s.headerChan) close(s.headerChan)
s.headerDone = true s.headerDone = true
isHeader = true isHeader = true