Fix race condition in registry/session

Signed-off-by: Alexander Morozov <[email protected]>
ncdc · Jun 1, 2015 · 9d98c28 · 9d98c28
1 parent 64a7431
commit 9d98c28
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/registry/session.go b/registry/session.go
@@ -98,7 +98,11 @@ func (tr *authTransport) RoundTrip(orig *http.Request) (*http.Response, error) {
 	}
 	resp.Body = &transport.OnEOFReader{
 		Rc: resp.Body,
-		Fn: func() { delete(tr.modReq, orig) },
+		Fn: func() {
+			tr.mu.Lock()
+			delete(tr.modReq, orig)
+			tr.mu.Unlock()
+		},
 	}
 	return resp, nil
 }