Merge branch 'worktree-agent-a2fe7ff3'
This commit is contained in:
90
.planning/phases/10-osint-code-hosting/10-03-SUMMARY.md
Normal file
90
.planning/phases/10-osint-code-hosting/10-03-SUMMARY.md
Normal file
@@ -0,0 +1,90 @@
|
|||||||
|
---
|
||||||
|
phase: 10-osint-code-hosting
|
||||||
|
plan: 03
|
||||||
|
subsystem: recon/sources
|
||||||
|
tags: [recon, osint, gitlab, wave-2]
|
||||||
|
requires:
|
||||||
|
- pkg/recon/sources.Client (Plan 10-01)
|
||||||
|
- pkg/recon/sources.BuildQueries (Plan 10-01)
|
||||||
|
- pkg/recon.LimiterRegistry (Phase 9)
|
||||||
|
- pkg/providers.Registry
|
||||||
|
provides:
|
||||||
|
- pkg/recon/sources.GitLabSource
|
||||||
|
affects:
|
||||||
|
- pkg/recon/sources
|
||||||
|
tech_stack_added: []
|
||||||
|
patterns:
|
||||||
|
- "Per-keyword BuildQueries loop driving search API calls"
|
||||||
|
- "PRIVATE-TOKEN header auth with shared retry-aware Client"
|
||||||
|
- "Disabled-when-empty-token semantics (Sweep returns nil with no requests)"
|
||||||
|
- "Bare-keyword → provider-name lookup via local keyword index"
|
||||||
|
key_files_created:
|
||||||
|
- pkg/recon/sources/gitlab.go
|
||||||
|
- pkg/recon/sources/gitlab_test.go
|
||||||
|
key_files_modified: []
|
||||||
|
decisions:
|
||||||
|
- "Bare keyword BuildQueries output (gitlab case in formatQuery) — reverse lookup is a direct map[string]string access"
|
||||||
|
- "gitlabKeywordIndex helper named with gitlab prefix to avoid collision with peer github.go keywordIndex during parallel wave"
|
||||||
|
- "Finding.Source uses constructed /projects/<id>/-/blob/<ref>/<path> URL (per plan) rather than extra /api/v4/projects/<id> lookup to keep request budget tight"
|
||||||
|
- "Confidence=low across all recon findings; Phase 5 verify promotes to high"
|
||||||
|
metrics:
|
||||||
|
duration: ~8 minutes
|
||||||
|
completed_date: 2026-04-05
|
||||||
|
tasks_completed: 1
|
||||||
|
tests_added: 6
|
||||||
|
---
|
||||||
|
|
||||||
|
# Phase 10 Plan 03: GitLabSource Summary
|
||||||
|
|
||||||
|
GitLabSource is a thin recon.ReconSource that queries GitLab's `/api/v4/search?scope=blobs` endpoint with a PRIVATE-TOKEN header, iterating one search call per provider keyword from the shared BuildQueries helper and emitting a Finding per returned blob with Source pointing at a constructed `projects/<id>/-/blob/<ref>/<path>` URL.
|
||||||
|
|
||||||
|
## What Was Built
|
||||||
|
|
||||||
|
`pkg/recon/sources/gitlab.go` contains:
|
||||||
|
|
||||||
|
- `GitLabSource` struct exposing Token, BaseURL, Registry, Limiters (lazy Client)
|
||||||
|
- ReconSource interface methods: `Name()="gitlab"`, `RateLimit()=rate.Every(30ms)`, `Burst()=5`, `RespectsRobots()=false`, `Enabled()` (token non-empty), `Sweep()`
|
||||||
|
- `glBlob` response DTO matching GitLab's documented blob search schema
|
||||||
|
- `gitlabKeywordIndex()` local helper (prefixed to avoid colliding with peer plan helpers during parallel wave execution)
|
||||||
|
- Compile-time `var _ recon.ReconSource = (*GitLabSource)(nil)` assertion
|
||||||
|
|
||||||
|
`pkg/recon/sources/gitlab_test.go` covers all behaviors the plan called out:
|
||||||
|
|
||||||
|
| Test | Verifies |
|
||||||
|
| --- | --- |
|
||||||
|
| `TestGitLabSource_EnabledFalseWhenTokenEmpty` | Enabled gating + Name/RespectsRobots accessors |
|
||||||
|
| `TestGitLabSource_EmptyToken_NoCallsNoError` | No HTTP request issued when Token=="" |
|
||||||
|
| `TestGitLabSource_Sweep_EmitsFindings` | PRIVATE-TOKEN header, `scope=blobs`, two queries × two blobs = 4 Findings, Source URLs contain project_id/ref/path |
|
||||||
|
| `TestGitLabSource_Unauthorized` | 401 propagates as `errors.Is(err, ErrUnauthorized)` |
|
||||||
|
| `TestGitLabSource_CtxCancellation` | Sweep returns promptly on ctx timeout against a hanging server |
|
||||||
|
| `TestGitLabSource_InterfaceAssertion` | Static recon.ReconSource conformance |
|
||||||
|
|
||||||
|
## Verification
|
||||||
|
|
||||||
|
```
|
||||||
|
go build ./... # clean
|
||||||
|
go test ./pkg/recon/sources/ -run TestGitLab -v # 6/6 PASS
|
||||||
|
go test ./pkg/recon/sources/ # full package PASS (3.164s)
|
||||||
|
```
|
||||||
|
|
||||||
|
## Deviations from Plan
|
||||||
|
|
||||||
|
None for must-have behavior. Two minor adjustments:
|
||||||
|
|
||||||
|
1. `keywordIndex` helper renamed to `gitlabKeywordIndex` because `pkg/recon/sources/github.go` (Plan 10-02, wave-2 sibling) introduces an identically-named package-level symbol. Prefixing prevents a redeclared-identifier build failure when the parallel wave merges.
|
||||||
|
2. Provider name lookup simplified to direct `map[string]string` access on the bare keyword because `formatQuery("gitlab", k)` returns the keyword verbatim (no wrapping syntax), avoiding a second `extractKeyword`-style helper.
|
||||||
|
|
||||||
|
## Deferred Issues
|
||||||
|
|
||||||
|
None.
|
||||||
|
|
||||||
|
## Known Stubs
|
||||||
|
|
||||||
|
None.
|
||||||
|
|
||||||
|
## Self-Check: PASSED
|
||||||
|
|
||||||
|
- pkg/recon/sources/gitlab.go — FOUND
|
||||||
|
- pkg/recon/sources/gitlab_test.go — FOUND
|
||||||
|
- .planning/phases/10-osint-code-hosting/10-03-SUMMARY.md — FOUND
|
||||||
|
- commit 0137dc5 — FOUND
|
||||||
175
pkg/recon/sources/gitlab.go
Normal file
175
pkg/recon/sources/gitlab.go
Normal file
@@ -0,0 +1,175 @@
|
|||||||
|
package sources
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"golang.org/x/time/rate"
|
||||||
|
|
||||||
|
"github.com/salvacybersec/keyhunter/pkg/providers"
|
||||||
|
"github.com/salvacybersec/keyhunter/pkg/recon"
|
||||||
|
)
|
||||||
|
|
||||||
|
// GitLabSource implements recon.ReconSource against GitLab's Search API
|
||||||
|
// (/api/v4/search?scope=blobs). It honors PRIVATE-TOKEN header auth and the
|
||||||
|
// published 2000 req/min rate limit. Sweep iterates BuildQueries(reg, "gitlab")
|
||||||
|
// — one request per keyword-derived query — and emits one Finding per returned
|
||||||
|
// blob, with Source pointing at the blob's project/ref/path URL.
|
||||||
|
//
|
||||||
|
// RECON-CODE-02.
|
||||||
|
type GitLabSource struct {
|
||||||
|
// Token is the GitLab personal access token. When empty the source is
|
||||||
|
// disabled (Enabled returns false, Sweep is a no-op).
|
||||||
|
Token string
|
||||||
|
// BaseURL is the GitLab instance root. Defaults to https://gitlab.com.
|
||||||
|
BaseURL string
|
||||||
|
// Registry drives query generation via BuildQueries and provider name
|
||||||
|
// mapping for emitted findings.
|
||||||
|
Registry *providers.Registry
|
||||||
|
// Limiters is the shared per-source rate-limiter registry.
|
||||||
|
Limiters *recon.LimiterRegistry
|
||||||
|
|
||||||
|
// client is the retry-aware HTTP wrapper. A nil client is replaced with
|
||||||
|
// NewClient() lazily inside Sweep so zero-value construction works.
|
||||||
|
client *Client
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compile-time interface assertion.
|
||||||
|
var _ recon.ReconSource = (*GitLabSource)(nil)
|
||||||
|
|
||||||
|
// Name returns the stable source identifier.
|
||||||
|
func (s *GitLabSource) Name() string { return "gitlab" }
|
||||||
|
|
||||||
|
// RateLimit returns ~2000 req/min (one token every 30ms).
|
||||||
|
func (s *GitLabSource) RateLimit() rate.Limit { return rate.Every(30 * time.Millisecond) }
|
||||||
|
|
||||||
|
// Burst allows short bursts of 5 requests.
|
||||||
|
func (s *GitLabSource) Burst() int { return 5 }
|
||||||
|
|
||||||
|
// RespectsRobots returns false: this source uses an authenticated REST API,
|
||||||
|
// not HTML scraping.
|
||||||
|
func (s *GitLabSource) RespectsRobots() bool { return false }
|
||||||
|
|
||||||
|
// Enabled reports whether a token is configured.
|
||||||
|
func (s *GitLabSource) Enabled(_ recon.Config) bool { return strings.TrimSpace(s.Token) != "" }
|
||||||
|
|
||||||
|
// glBlob is the subset of the GitLab Search API blob response we consume.
|
||||||
|
type glBlob struct {
|
||||||
|
Basename string `json:"basename"`
|
||||||
|
Data string `json:"data"`
|
||||||
|
Path string `json:"path"`
|
||||||
|
ProjectID int `json:"project_id"`
|
||||||
|
Ref string `json:"ref"`
|
||||||
|
Startline int `json:"startline"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sweep runs the GitLab blob search for every keyword-derived query and emits
|
||||||
|
// one Finding per blob. It returns nil when the source is disabled (empty
|
||||||
|
// token) so callers can safely skip without special-casing.
|
||||||
|
func (s *GitLabSource) Sweep(ctx context.Context, _ string, out chan<- recon.Finding) error {
|
||||||
|
if !s.Enabled(recon.Config{}) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if s.client == nil {
|
||||||
|
s.client = NewClient()
|
||||||
|
}
|
||||||
|
base := strings.TrimRight(s.BaseURL, "/")
|
||||||
|
if base == "" {
|
||||||
|
base = "https://gitlab.com"
|
||||||
|
}
|
||||||
|
limiters := s.Limiters
|
||||||
|
if limiters == nil {
|
||||||
|
limiters = recon.NewLimiterRegistry()
|
||||||
|
}
|
||||||
|
|
||||||
|
queries := BuildQueries(s.Registry, "gitlab")
|
||||||
|
if len(queries) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
kwIndex := gitlabKeywordIndex(s.Registry)
|
||||||
|
|
||||||
|
for _, q := range queries {
|
||||||
|
if err := ctx.Err(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := limiters.Wait(ctx, s.Name(), s.RateLimit(), s.Burst(), false); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
endpoint := fmt.Sprintf("%s/api/v4/search?scope=blobs&search=%s&per_page=20",
|
||||||
|
base, url.QueryEscape(q))
|
||||||
|
req, err := http.NewRequest(http.MethodGet, endpoint, nil)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("gitlab: build request: %w", err)
|
||||||
|
}
|
||||||
|
req.Header.Set("PRIVATE-TOKEN", s.Token)
|
||||||
|
req.Header.Set("Accept", "application/json")
|
||||||
|
|
||||||
|
resp, err := s.client.Do(ctx, req)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("gitlab: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var blobs []glBlob
|
||||||
|
decErr := json.NewDecoder(resp.Body).Decode(&blobs)
|
||||||
|
_ = resp.Body.Close()
|
||||||
|
if decErr != nil {
|
||||||
|
return fmt.Errorf("gitlab: decode: %w", decErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// For "gitlab", BuildQueries emits bare keywords, so a direct map
|
||||||
|
// lookup recovers the provider name for each query.
|
||||||
|
provName := kwIndex[q]
|
||||||
|
if provName == "" {
|
||||||
|
provName = "unknown"
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, b := range blobs {
|
||||||
|
sourceURL := fmt.Sprintf("%s/projects/%d/-/blob/%s/%s",
|
||||||
|
base, b.ProjectID, b.Ref, b.Path)
|
||||||
|
finding := recon.Finding{
|
||||||
|
ProviderName: provName,
|
||||||
|
Confidence: "low",
|
||||||
|
Source: sourceURL,
|
||||||
|
SourceType: "recon:gitlab",
|
||||||
|
LineNumber: b.Startline,
|
||||||
|
DetectedAt: time.Now().UTC(),
|
||||||
|
}
|
||||||
|
select {
|
||||||
|
case out <- finding:
|
||||||
|
case <-ctx.Done():
|
||||||
|
return ctx.Err()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// gitlabKeywordIndex maps each provider keyword back to its provider name for
|
||||||
|
// Finding.ProviderName population. A name prefixed with "gitlab" avoids
|
||||||
|
// colliding with the shared keywordIndex helper introduced by peer sources
|
||||||
|
// (github.go) in the same package.
|
||||||
|
func gitlabKeywordIndex(reg *providers.Registry) map[string]string {
|
||||||
|
idx := make(map[string]string)
|
||||||
|
if reg == nil {
|
||||||
|
return idx
|
||||||
|
}
|
||||||
|
for _, p := range reg.List() {
|
||||||
|
for _, kw := range p.Keywords {
|
||||||
|
if kw == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if _, exists := idx[kw]; !exists {
|
||||||
|
idx[kw] = p.Name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return idx
|
||||||
|
}
|
||||||
229
pkg/recon/sources/gitlab_test.go
Normal file
229
pkg/recon/sources/gitlab_test.go
Normal file
@@ -0,0 +1,229 @@
|
|||||||
|
package sources
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"strings"
|
||||||
|
"sync/atomic"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/salvacybersec/keyhunter/pkg/providers"
|
||||||
|
"github.com/salvacybersec/keyhunter/pkg/recon"
|
||||||
|
)
|
||||||
|
|
||||||
|
// gitlabTestRegistry returns a synthetic registry with two providers whose
|
||||||
|
// keywords drive the query loop. Keywords are chosen so BuildQueries output is
|
||||||
|
// deterministic and map lookups are unambiguous.
|
||||||
|
func gitlabTestRegistry() *providers.Registry {
|
||||||
|
return providers.NewRegistryFromProviders([]providers.Provider{
|
||||||
|
{
|
||||||
|
Name: "openai",
|
||||||
|
Keywords: []string{"sk-test"},
|
||||||
|
Patterns: []providers.Pattern{{Regex: "sk-test[A-Za-z0-9]+", Confidence: "high"}},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "demo",
|
||||||
|
Keywords: []string{"ghkey"},
|
||||||
|
Patterns: []providers.Pattern{{Regex: "ghkey[A-Za-z0-9]+", Confidence: "low"}},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
type gitlabBlobFixture struct {
|
||||||
|
Basename string `json:"basename"`
|
||||||
|
Data string `json:"data"`
|
||||||
|
Path string `json:"path"`
|
||||||
|
ProjectID int `json:"project_id"`
|
||||||
|
Ref string `json:"ref"`
|
||||||
|
Startline int `json:"startline"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGitLabSource_EnabledFalseWhenTokenEmpty(t *testing.T) {
|
||||||
|
s := &GitLabSource{Token: "", Registry: gitlabTestRegistry(), Limiters: recon.NewLimiterRegistry()}
|
||||||
|
if s.Enabled(recon.Config{}) {
|
||||||
|
t.Fatalf("expected Enabled=false when token empty")
|
||||||
|
}
|
||||||
|
s2 := &GitLabSource{Token: "glpat-xxx", Registry: gitlabTestRegistry(), Limiters: recon.NewLimiterRegistry()}
|
||||||
|
if !s2.Enabled(recon.Config{}) {
|
||||||
|
t.Fatalf("expected Enabled=true when token set")
|
||||||
|
}
|
||||||
|
if s.Name() != "gitlab" {
|
||||||
|
t.Fatalf("expected Name=gitlab, got %q", s.Name())
|
||||||
|
}
|
||||||
|
if s.RespectsRobots() {
|
||||||
|
t.Fatalf("expected RespectsRobots=false for REST API source")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGitLabSource_EmptyToken_NoCallsNoError(t *testing.T) {
|
||||||
|
var calls int32
|
||||||
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
atomic.AddInt32(&calls, 1)
|
||||||
|
w.WriteHeader(200)
|
||||||
|
_, _ = w.Write([]byte("[]"))
|
||||||
|
}))
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
s := &GitLabSource{
|
||||||
|
Token: "",
|
||||||
|
BaseURL: srv.URL,
|
||||||
|
Registry: gitlabTestRegistry(),
|
||||||
|
Limiters: recon.NewLimiterRegistry(),
|
||||||
|
}
|
||||||
|
out := make(chan recon.Finding, 4)
|
||||||
|
if err := s.Sweep(context.Background(), "", out); err != nil {
|
||||||
|
t.Fatalf("expected nil err on empty token, got %v", err)
|
||||||
|
}
|
||||||
|
close(out)
|
||||||
|
if atomic.LoadInt32(&calls) != 0 {
|
||||||
|
t.Fatalf("expected zero HTTP calls, got %d", calls)
|
||||||
|
}
|
||||||
|
if len(out) != 0 {
|
||||||
|
t.Fatalf("expected zero findings, got %d", len(out))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGitLabSource_Sweep_EmitsFindings(t *testing.T) {
|
||||||
|
var gotToken string
|
||||||
|
var gotScopes []string
|
||||||
|
var gotSearches []string
|
||||||
|
|
||||||
|
blobs := []gitlabBlobFixture{
|
||||||
|
{Basename: "config.env", Data: "API_KEY=sk-testABCDEF", Path: "app/config.env", ProjectID: 42, Ref: "main", Startline: 3},
|
||||||
|
{Basename: "README.md", Data: "use ghkeyXYZ", Path: "docs/README.md", ProjectID: 99, Ref: "master", Startline: 10},
|
||||||
|
}
|
||||||
|
|
||||||
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.URL.Path != "/api/v4/search" {
|
||||||
|
http.Error(w, "not found", 404)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
gotToken = r.Header.Get("PRIVATE-TOKEN")
|
||||||
|
gotScopes = append(gotScopes, r.URL.Query().Get("scope"))
|
||||||
|
gotSearches = append(gotSearches, r.URL.Query().Get("search"))
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
_ = json.NewEncoder(w).Encode(blobs)
|
||||||
|
}))
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
s := &GitLabSource{
|
||||||
|
Token: "glpat-secret",
|
||||||
|
BaseURL: srv.URL,
|
||||||
|
Registry: gitlabTestRegistry(),
|
||||||
|
Limiters: recon.NewLimiterRegistry(),
|
||||||
|
}
|
||||||
|
|
||||||
|
out := make(chan recon.Finding, 32)
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
if err := s.Sweep(ctx, "", out); err != nil {
|
||||||
|
t.Fatalf("sweep err: %v", err)
|
||||||
|
}
|
||||||
|
close(out)
|
||||||
|
|
||||||
|
if gotToken != "glpat-secret" {
|
||||||
|
t.Fatalf("expected PRIVATE-TOKEN header, got %q", gotToken)
|
||||||
|
}
|
||||||
|
for _, sc := range gotScopes {
|
||||||
|
if sc != "blobs" {
|
||||||
|
t.Fatalf("expected scope=blobs, got %q", sc)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Two providers → two queries → two requests → 4 findings (2 blobs each).
|
||||||
|
if len(gotSearches) != 2 {
|
||||||
|
t.Fatalf("expected 2 search calls, got %d: %v", len(gotSearches), gotSearches)
|
||||||
|
}
|
||||||
|
|
||||||
|
findings := gitlabDrain(out)
|
||||||
|
if len(findings) != 4 {
|
||||||
|
t.Fatalf("expected 4 findings (2 blobs × 2 queries), got %d", len(findings))
|
||||||
|
}
|
||||||
|
|
||||||
|
var sawP42, sawP99 bool
|
||||||
|
for _, f := range findings {
|
||||||
|
if f.SourceType != "recon:gitlab" {
|
||||||
|
t.Errorf("bad SourceType: %q", f.SourceType)
|
||||||
|
}
|
||||||
|
if f.Confidence != "low" {
|
||||||
|
t.Errorf("bad confidence: %q", f.Confidence)
|
||||||
|
}
|
||||||
|
if strings.Contains(f.Source, "/projects/42/-/blob/main/app/config.env") {
|
||||||
|
sawP42 = true
|
||||||
|
}
|
||||||
|
if strings.Contains(f.Source, "/projects/99/-/blob/master/docs/README.md") {
|
||||||
|
sawP99 = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !sawP42 || !sawP99 {
|
||||||
|
t.Fatalf("expected both project URLs in Source fields: p42=%v p99=%v", sawP42, sawP99)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGitLabSource_Unauthorized(t *testing.T) {
|
||||||
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
w.WriteHeader(401)
|
||||||
|
_, _ = w.Write([]byte(`{"message":"401 Unauthorized"}`))
|
||||||
|
}))
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
s := &GitLabSource{
|
||||||
|
Token: "bad",
|
||||||
|
BaseURL: srv.URL,
|
||||||
|
Registry: gitlabTestRegistry(),
|
||||||
|
Limiters: recon.NewLimiterRegistry(),
|
||||||
|
}
|
||||||
|
out := make(chan recon.Finding, 4)
|
||||||
|
err := s.Sweep(context.Background(), "", out)
|
||||||
|
close(out)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatalf("expected error, got nil")
|
||||||
|
}
|
||||||
|
if !errors.Is(err, ErrUnauthorized) {
|
||||||
|
t.Fatalf("expected ErrUnauthorized, got %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGitLabSource_CtxCancellation(t *testing.T) {
|
||||||
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
select {
|
||||||
|
case <-r.Context().Done():
|
||||||
|
return
|
||||||
|
case <-time.After(2 * time.Second):
|
||||||
|
w.WriteHeader(200)
|
||||||
|
_, _ = w.Write([]byte("[]"))
|
||||||
|
}
|
||||||
|
}))
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
s := &GitLabSource{
|
||||||
|
Token: "glpat-x",
|
||||||
|
BaseURL: srv.URL,
|
||||||
|
Registry: gitlabTestRegistry(),
|
||||||
|
Limiters: recon.NewLimiterRegistry(),
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
|
||||||
|
defer cancel()
|
||||||
|
out := make(chan recon.Finding, 4)
|
||||||
|
err := s.Sweep(ctx, "", out)
|
||||||
|
close(out)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatalf("expected ctx error, got nil")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGitLabSource_InterfaceAssertion(t *testing.T) {
|
||||||
|
var _ recon.ReconSource = (*GitLabSource)(nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
func gitlabDrain(ch <-chan recon.Finding) []recon.Finding {
|
||||||
|
var out []recon.Finding
|
||||||
|
for f := range ch {
|
||||||
|
out = append(out, f)
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user