Mirrors/trufflehog
2
0
Fork 0
mirror of https://github.com/trufflesecurity/trufflehog.git synced 2024-11-10 15:14:38 +00:00
Code Issues Projects Releases Packages Wiki Activity

Prevent concurrent read and writes to visibility map. (#892)

Browse source
This commit is contained in:
ahrav 2022-11-01 16:20:59 -07:00 committed by GitHub
parent 2884864cd3
commit fe1e475a04
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pkg/sources/github/github.go
View file

@ -62,6 +62,7 @@ type Source struct {
resumeInfoMutex sync.Mutex
resumeInfoSlice []string
apiClient *github.Client
mu sync.Mutex
publicMap map[string]source_metadatapb.Visibility
sources.Progress
}
@ -177,12 +178,18 @@ func (s *Source) Init(aCtx context.Context, name string, jobID, sourceID int64,
}
func (s *Source) visibilityOf(repoURL string) (visibility source_metadatapb.Visibility) {
if visibility, exists := s.publicMap[repoURL]; exists {
s.mu.Lock()
visibility, ok := s.publicMap[repoURL]
s.mu.Unlock()
if ok {
return visibility
}
visibility = source_metadatapb.Visibility_public
defer func() {
s.mu.Lock()
s.publicMap[repoURL] = visibility
s.mu.Unlock()
}()
log.Debugf("Checking public status for %s", repoURL)
u, err := url.Parse(repoURL)