main.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896
  1. package main
  2. import (
  3. "crypto/md5"
  4. "encoding/csv"
  5. "encoding/json"
  6. "errors"
  7. "fmt"
  8. "io/ioutil"
  9. "math"
  10. "net"
  11. "net/url"
  12. "os"
  13. "os/user"
  14. "path"
  15. "path/filepath"
  16. "regexp"
  17. "runtime"
  18. "strconv"
  19. "strings"
  20. "sync"
  21. "time"
  22. "gopkg.in/src-d/go-git.v4/plumbing"
  23. "gopkg.in/src-d/go-git.v4/plumbing/object"
  24. "gopkg.in/src-d/go-git.v4/plumbing/transport/ssh"
  25. "gopkg.in/src-d/go-git.v4/storage/memory"
  26. "github.com/BurntSushi/toml"
  27. "github.com/google/go-github/github"
  28. "github.com/hako/durafmt"
  29. flags "github.com/jessevdk/go-flags"
  30. log "github.com/sirupsen/logrus"
  31. git "gopkg.in/src-d/go-git.v4"
  32. )
  33. // Leak represents a leaked secret or regex match.
  34. type Leak struct {
  35. Line string `json:"line"`
  36. Commit string `json:"commit"`
  37. Offender string `json:"offender"`
  38. Type string `json:"reason"`
  39. Message string `json:"commitMsg"`
  40. Author string `json:"author"`
  41. File string `json:"file"`
  42. Branch string `json:"branch"`
  43. Repo string `json:"repo"`
  44. }
  45. // RepoDescriptor contains a src-d git repository and other data about the repo
  46. type RepoDescriptor struct {
  47. path string
  48. url string
  49. name string
  50. repository *git.Repository
  51. err error
  52. }
  53. // Options for gitleaks
  54. type Options struct {
  55. // remote target options
  56. Repo string `short:"r" long:"repo" description:"Repo url to audit"`
  57. GithubUser string `long:"github-user" description:"Github user to audit"`
  58. GithubOrg string `long:"github-org" description:"Github organization to audit"`
  59. GithubURL string `long:"github-url" default:"https://api.github.com/" description:"GitHub API Base URL, use for GitHub Enterprise. Example: https://github.example.com/api/v3/"`
  60. GithubPR string `long:"github-pr" description:"Github PR url to audit. This does not clone the repo. GITHUB_TOKEN must be set"`
  61. /*
  62. TODO:
  63. GitLabUser string `long:"gitlab-user" description:"User url to audit"`
  64. GitLabOrg string `long:"gitlab-org" description:"Organization url to audit"`
  65. */
  66. Branch string `short:"b" long:"branch" description:"branch name to audit (defaults to HEAD)"`
  67. Commit string `short:"c" long:"commit" description:"sha of commit to stop at"`
  68. Depth int `long:"depth" description:"maximum commit depth"`
  69. // local target option
  70. RepoPath string `long:"repo-path" description:"Path to repo"`
  71. OwnerPath string `long:"owner-path" description:"Path to owner directory (repos discovered)"`
  72. // Process options
  73. MaxGoRoutines int `long:"max-go" description:"Maximum number of concurrent go-routines gitleaks spawns"`
  74. Disk bool `long:"disk" description:"Clones repo(s) to disk"`
  75. AuditAllRefs bool `long:"all-refs" description:"run audit on all refs"`
  76. SingleSearch string `long:"single-search" description:"single regular expression to search for"`
  77. ConfigPath string `long:"config" description:"path to gitleaks config"`
  78. SSHKey string `long:"ssh-key" description:"path to ssh key"`
  79. ExcludeForks bool `long:"exclude-forks" description:"exclude forks for organization/user audits"`
  80. Entropy float64 `long:"entropy" short:"e" description:"Include entropy checks during audit. Entropy scale: 0.0(no entropy) - 8.0(max entropy)"`
  81. // TODO: IncludeMessages string `long:"messages" description:"include commit messages in audit"`
  82. // Output options
  83. Log string `short:"l" long:"log" description:"log level"`
  84. Verbose bool `short:"v" long:"verbose" description:"Show verbose output from gitleaks audit"`
  85. Report string `long:"report" description:"path to write report file. Needs to be csv or json"`
  86. Redact bool `long:"redact" description:"redact secrets from log messages and report"`
  87. Version bool `long:"version" description:"version number"`
  88. SampleConfig bool `long:"sample-config" description:"prints a sample config file"`
  89. }
  90. // Config struct for regexes matching and whitelisting
  91. type Config struct {
  92. Regexes []struct {
  93. Description string
  94. Regex string
  95. }
  96. Whitelist struct {
  97. Files []string
  98. Regexes []string
  99. Commits []string
  100. Branches []string
  101. Repos []string
  102. }
  103. Misc struct {
  104. Entropy []string
  105. }
  106. }
  107. type gitDiff struct {
  108. content string
  109. commit *object.Commit
  110. filePath string
  111. branchName string
  112. repoName string
  113. githubCommit *github.RepositoryCommit
  114. sha string
  115. message string
  116. author string
  117. }
  118. type entropyRange struct {
  119. v1 float64
  120. v2 float64
  121. }
  122. const defaultGithubURL = "https://api.github.com/"
  123. const version = "1.16.0"
  124. const errExit = 2
  125. const leakExit = 1
  126. const defaultConfig = `
  127. # This is a sample config file for gitleaks. You can configure gitleaks what to search for and what to whitelist.
  128. # The output you are seeing here is the default gitleaks config. If GITLEAKS_CONFIG environment variable
  129. # is set, gitleaks will load configurations from that path. If option --config-path is set, gitleaks will load
  130. # configurations from that path. Gitleaks does not whitelist anything by default.
  131. title = "gitleaks config"
  132. # add regexes to the regex table
  133. [[regexes]]
  134. description = "AWS"
  135. regex = '''AKIA[0-9A-Z]{16}'''
  136. [[regexes]]
  137. description = "RKCS8"
  138. regex = '''-----BEGIN PRIVATE KEY-----'''
  139. [[regexes]]
  140. description = "RSA"
  141. regex = '''-----BEGIN RSA PRIVATE KEY-----'''
  142. [[regexes]]
  143. description = "Github"
  144. regex = '''(?i)github.*['\"][0-9a-zA-Z]{35,40}['\"]'''
  145. [[regexes]]
  146. description = "SSH"
  147. regex = '''-----BEGIN OPENSSH PRIVATE KEY-----'''
  148. [[regexes]]
  149. description = "Facebook"
  150. regex = '''(?i)facebook.*['\"][0-9a-f]{32}['\"]'''
  151. [[regexes]]
  152. description = "Twitter"
  153. regex = '''(?i)twitter.*['\"][0-9a-zA-Z]{35,44}['\"]'''
  154. [whitelist]
  155. #regexes = [
  156. # "AKAIMYFAKEAWKKEY",
  157. #]
  158. #files = [
  159. # "(.*?)(jpg|gif|doc|pdf|bin)$"
  160. #]
  161. #commits = [
  162. # "BADHA5H1",
  163. # "BADHA5H2",
  164. #]
  165. #branches = [
  166. # "dev/goodrepo"
  167. #]
  168. #repos = [
  169. # "mygoodrepo"
  170. #]
  171. [misc]
  172. #entropy = [
  173. # "3.3-4.30"
  174. # "6.0-8.0
  175. #]
  176. `
  177. var (
  178. opts Options
  179. regexes map[string]*regexp.Regexp
  180. singleSearchRegex *regexp.Regexp
  181. whiteListRegexes []*regexp.Regexp
  182. whiteListFiles []*regexp.Regexp
  183. whiteListCommits map[string]bool
  184. whiteListBranches []string
  185. whiteListRepos []*regexp.Regexp
  186. entropyRanges []entropyRange
  187. fileDiffRegex *regexp.Regexp
  188. sshAuth *ssh.PublicKeys
  189. dir string
  190. maxGo int
  191. totalCommits int64
  192. )
  193. func init() {
  194. log.SetOutput(os.Stdout)
  195. maxGo = runtime.GOMAXPROCS(0) / 2
  196. regexes = make(map[string]*regexp.Regexp)
  197. whiteListCommits = make(map[string]bool)
  198. }
  199. func main() {
  200. parser := flags.NewParser(&opts, flags.Default)
  201. _, err := parser.Parse()
  202. if err != nil {
  203. if flagsErr, ok := err.(*flags.Error); ok && flagsErr.Type == flags.ErrHelp {
  204. os.Exit(0)
  205. }
  206. }
  207. if len(os.Args) == 1 {
  208. parser.WriteHelp(os.Stdout)
  209. os.Exit(0)
  210. }
  211. if opts.Version {
  212. fmt.Println(version)
  213. os.Exit(0)
  214. }
  215. if opts.SampleConfig {
  216. fmt.Println(defaultConfig)
  217. os.Exit(0)
  218. }
  219. now := time.Now()
  220. leaks, err := run()
  221. if err != nil {
  222. if strings.Contains(err.Error(), "whitelisted") {
  223. log.Info(err.Error())
  224. os.Exit(0)
  225. }
  226. log.Error(err)
  227. os.Exit(errExit)
  228. }
  229. if opts.Report != "" {
  230. writeReport(leaks)
  231. }
  232. if len(leaks) != 0 {
  233. log.Warnf("%d leaks detected. %d commits inspected in %s", len(leaks), totalCommits, durafmt.Parse(time.Now().Sub(now)).String())
  234. os.Exit(leakExit)
  235. } else {
  236. log.Infof("%d leaks detected. %d commits inspected in %s", len(leaks), totalCommits, durafmt.Parse(time.Now().Sub(now)).String())
  237. }
  238. }
  239. // run parses options and kicks off the audit
  240. func run() ([]Leak, error) {
  241. var leaks []Leak
  242. setLogs()
  243. err := optsGuard()
  244. if err != nil {
  245. return nil, err
  246. }
  247. err = loadToml()
  248. if err != nil {
  249. return nil, err
  250. }
  251. sshAuth, err = getSSHAuth()
  252. if err != nil {
  253. return leaks, err
  254. }
  255. if opts.Disk {
  256. // temporary directory where all the gitleaks plain clones will reside
  257. dir, err = ioutil.TempDir("", "gitleaks")
  258. defer os.RemoveAll(dir)
  259. if err != nil {
  260. return nil, err
  261. }
  262. }
  263. // start audits
  264. if opts.Repo != "" || opts.RepoPath != "" {
  265. // Audit a single remote repo or a local repo.
  266. repo, err := cloneRepo()
  267. if err != nil {
  268. return leaks, err
  269. }
  270. return auditGitRepo(repo)
  271. } else if opts.OwnerPath != "" {
  272. // Audit local repos. Gitleaks will look for all child directories of OwnerPath for
  273. // git repos and perform an audit on said repos.
  274. repos, err := discoverRepos(opts.OwnerPath)
  275. if err != nil {
  276. return leaks, err
  277. }
  278. for _, repo := range repos {
  279. leaksFromRepo, err := auditGitRepo(repo)
  280. if err != nil {
  281. return leaks, err
  282. }
  283. leaks = append(leaksFromRepo, leaks...)
  284. }
  285. } else if opts.GithubOrg != "" || opts.GithubUser != "" {
  286. // Audit a github owner -- a user or organization.
  287. leaks, err = auditGithubRepos()
  288. if err != nil {
  289. return leaks, err
  290. }
  291. } else if opts.GithubPR != "" {
  292. return auditGithubPR()
  293. }
  294. return leaks, nil
  295. }
  296. // writeReport writes a report to a file specified in the --report= option.
  297. // Default format for report is JSON. You can use the --csv option to write the report as a csv
  298. func writeReport(leaks []Leak) error {
  299. var err error
  300. log.Infof("writing report to %s", opts.Report)
  301. if strings.HasSuffix(opts.Report, ".csv") {
  302. f, err := os.Create(opts.Report)
  303. if err != nil {
  304. return err
  305. }
  306. defer f.Close()
  307. w := csv.NewWriter(f)
  308. w.Write([]string{"repo", "line", "commit", "offender", "reason", "commitMsg", "author", "file", "branch"})
  309. for _, leak := range leaks {
  310. w.Write([]string{leak.Repo, leak.Line, leak.Commit, leak.Offender, leak.Type, leak.Message, leak.Author, leak.File, leak.Branch})
  311. }
  312. w.Flush()
  313. } else {
  314. reportJSON, _ := json.MarshalIndent(leaks, "", "\t")
  315. err = ioutil.WriteFile(opts.Report, reportJSON, 0644)
  316. }
  317. return err
  318. }
  319. // cloneRepo clones a repo to memory(default) or to disk if the --disk option is set.
  320. func cloneRepo() (*RepoDescriptor, error) {
  321. var (
  322. err error
  323. repo *git.Repository
  324. )
  325. // check if whitelist
  326. for _, re := range whiteListRepos {
  327. if re.FindString(opts.Repo) != "" {
  328. return nil, fmt.Errorf("skipping %s, whitelisted", opts.Repo)
  329. }
  330. }
  331. if opts.Disk {
  332. log.Infof("cloning %s", opts.Repo)
  333. cloneTarget := fmt.Sprintf("%s/%x", dir, md5.Sum([]byte(fmt.Sprintf("%s%s", opts.GithubUser, opts.Repo))))
  334. if strings.HasPrefix(opts.Repo, "git") {
  335. repo, err = git.PlainClone(cloneTarget, false, &git.CloneOptions{
  336. URL: opts.Repo,
  337. Progress: os.Stdout,
  338. Auth: sshAuth,
  339. })
  340. } else {
  341. repo, err = git.PlainClone(cloneTarget, false, &git.CloneOptions{
  342. URL: opts.Repo,
  343. Progress: os.Stdout,
  344. })
  345. }
  346. } else if opts.RepoPath != "" {
  347. log.Infof("opening %s", opts.RepoPath)
  348. repo, err = git.PlainOpen(opts.RepoPath)
  349. } else {
  350. log.Infof("cloning %s", opts.Repo)
  351. if strings.HasPrefix(opts.Repo, "git") {
  352. repo, err = git.Clone(memory.NewStorage(), nil, &git.CloneOptions{
  353. URL: opts.Repo,
  354. Progress: os.Stdout,
  355. Auth: sshAuth,
  356. })
  357. } else {
  358. repo, err = git.Clone(memory.NewStorage(), nil, &git.CloneOptions{
  359. URL: opts.Repo,
  360. Progress: os.Stdout,
  361. })
  362. }
  363. }
  364. return &RepoDescriptor{
  365. repository: repo,
  366. path: opts.RepoPath,
  367. url: opts.Repo,
  368. name: filepath.Base(opts.Repo),
  369. err: err,
  370. }, nil
  371. }
  372. // auditGitRepo beings an audit on a git repository by checking the default HEAD branch, all branches, or
  373. // a single branch depending on what gitleaks is configured to do. Note when I say branch I really
  374. // mean reference as these branches are read only.
  375. func auditGitRepo(repo *RepoDescriptor) ([]Leak, error) {
  376. var (
  377. err error
  378. leaks []Leak
  379. )
  380. for _, re := range whiteListRepos {
  381. if re.FindString(repo.name) != "" {
  382. return leaks, fmt.Errorf("skipping %s, whitelisted", repo.name)
  383. }
  384. }
  385. ref, err := repo.repository.Head()
  386. if err != nil {
  387. return leaks, err
  388. }
  389. if opts.AuditAllRefs {
  390. skipBranch := false
  391. refs, err := repo.repository.Storer.IterReferences()
  392. if err != nil {
  393. return leaks, err
  394. }
  395. err = refs.ForEach(func(ref *plumbing.Reference) error {
  396. for _, b := range whiteListBranches {
  397. if strings.HasSuffix(string(ref.Name()), b) {
  398. skipBranch = true
  399. }
  400. }
  401. if skipBranch {
  402. skipBranch = false
  403. return nil
  404. }
  405. branchLeaks := auditGitReference(repo, ref)
  406. for _, leak := range branchLeaks {
  407. leaks = append(leaks, leak)
  408. }
  409. return nil
  410. })
  411. } else {
  412. if opts.Branch != "" {
  413. foundBranch := false
  414. refs, _ := repo.repository.Storer.IterReferences()
  415. branch := strings.Split(opts.Branch, "/")[len(strings.Split(opts.Branch, "/"))-1]
  416. err = refs.ForEach(func(refBranch *plumbing.Reference) error {
  417. if strings.Split(refBranch.Name().String(), "/")[len(strings.Split(refBranch.Name().String(), "/"))-1] == branch {
  418. foundBranch = true
  419. ref = refBranch
  420. }
  421. return nil
  422. })
  423. if foundBranch == false {
  424. return nil, nil
  425. }
  426. }
  427. leaks = auditGitReference(repo, ref)
  428. }
  429. return leaks, err
  430. }
  431. // auditGitReference beings the audit for a git reference. This function will
  432. // traverse the git reference and audit each line of each diff.
  433. func auditGitReference(repo *RepoDescriptor, ref *plumbing.Reference) []Leak {
  434. var (
  435. err error
  436. repoName string
  437. leaks []Leak
  438. commitCount int
  439. commitWg sync.WaitGroup
  440. mutex = &sync.Mutex{}
  441. semaphore chan bool
  442. )
  443. repoName = repo.name
  444. if opts.MaxGoRoutines != 0 {
  445. maxGo = opts.MaxGoRoutines
  446. }
  447. if opts.RepoPath != "" {
  448. maxGo = 1
  449. }
  450. semaphore = make(chan bool, maxGo)
  451. cIter, err := repo.repository.Log(&git.LogOptions{From: ref.Hash()})
  452. if err != nil {
  453. return nil
  454. }
  455. err = cIter.ForEach(func(c *object.Commit) error {
  456. if c == nil || c.Hash.String() == opts.Commit || (opts.Depth != 0 && commitCount == opts.Depth) {
  457. cIter.Close()
  458. return errors.New("ErrStop")
  459. }
  460. commitCount = commitCount + 1
  461. totalCommits = totalCommits + 1
  462. if whiteListCommits[c.Hash.String()] {
  463. log.Infof("skipping commit: %s\n", c.Hash.String())
  464. return nil
  465. }
  466. err = c.Parents().ForEach(func(parent *object.Commit) error {
  467. commitWg.Add(1)
  468. semaphore <- true
  469. go func(c *object.Commit, parent *object.Commit) {
  470. var (
  471. filePath string
  472. skipFile bool
  473. )
  474. defer func() {
  475. commitWg.Done()
  476. <-semaphore
  477. if r := recover(); r != nil {
  478. log.Warnf("recoverying from panic on commit %s, likely large diff causing panic", c.Hash.String())
  479. }
  480. }()
  481. patch, err := c.Patch(parent)
  482. if err != nil {
  483. log.Warnf("problem generating patch for commit: %s\n", c.Hash.String())
  484. return
  485. }
  486. for _, f := range patch.FilePatches() {
  487. skipFile = false
  488. from, to := f.Files()
  489. filePath = "???"
  490. if from != nil {
  491. filePath = from.Path()
  492. } else if to != nil {
  493. filePath = to.Path()
  494. }
  495. for _, re := range whiteListFiles {
  496. if re.FindString(filePath) != "" {
  497. skipFile = true
  498. break
  499. }
  500. }
  501. if skipFile {
  502. continue
  503. }
  504. chunks := f.Chunks()
  505. for _, chunk := range chunks {
  506. if chunk.Type() == 1 || chunk.Type() == 2 {
  507. diff := gitDiff{
  508. branchName: string(ref.Name()),
  509. repoName: repoName,
  510. filePath: filePath,
  511. content: chunk.Content(),
  512. sha: c.Hash.String(),
  513. author: c.Author.String(),
  514. message: c.Message,
  515. }
  516. chunkLeaks := inspect(diff)
  517. for _, leak := range chunkLeaks {
  518. mutex.Lock()
  519. leaks = append(leaks, leak)
  520. mutex.Unlock()
  521. }
  522. }
  523. }
  524. }
  525. }(c, parent)
  526. return nil
  527. })
  528. return nil
  529. })
  530. commitWg.Wait()
  531. return leaks
  532. }
  533. // inspect will parse each line of the git diff's content against a set of regexes or
  534. // a set of regexes set by the config (see gitleaks.toml for example). This function
  535. // will skip lines that include a whitelisted regex. A list of leaks is returned.
  536. // If verbose mode (-v/--verbose) is set, then checkDiff will log leaks as they are discovered.
  537. func inspect(diff gitDiff) []Leak {
  538. lines := strings.Split(diff.content, "\n")
  539. var (
  540. leaks []Leak
  541. skipLine bool
  542. )
  543. for _, line := range lines {
  544. skipLine = false
  545. for leakType, re := range regexes {
  546. match := re.FindString(line)
  547. if match == "" {
  548. continue
  549. }
  550. // if offender matches whitelist regex, ignore it
  551. for _, wRe := range whiteListRegexes {
  552. whitelistMatch := wRe.FindString(line)
  553. if whitelistMatch != "" {
  554. skipLine = true
  555. break
  556. }
  557. }
  558. if skipLine {
  559. break
  560. }
  561. leaks = addLeak(leaks, line, match, leakType, diff)
  562. }
  563. if opts.Entropy > 0 || len(entropyRanges) != 0 {
  564. entropyLeak := false
  565. words := strings.Fields(line)
  566. for _, word := range words {
  567. entropy := getShannonEntropy(word)
  568. if entropy >= opts.Entropy && len(entropyRanges) == 0 {
  569. entropyLeak = true
  570. }
  571. if len(entropyRanges) != 0 {
  572. for _, eR := range entropyRanges {
  573. if entropy > eR.v1 && entropy < eR.v2 {
  574. entropyLeak = true
  575. }
  576. }
  577. }
  578. if entropyLeak {
  579. leaks = addLeak(leaks, line, word, fmt.Sprintf("Entropy: %.2f", entropy), diff)
  580. entropyLeak = false
  581. }
  582. }
  583. }
  584. }
  585. return leaks
  586. }
  587. // addLeak is helper for func inspect() to append leaks if found during a diff check.
  588. func addLeak(leaks []Leak, line string, offender string, leakType string, diff gitDiff) []Leak {
  589. leak := Leak{
  590. Line: line,
  591. Commit: diff.sha,
  592. Offender: offender,
  593. Type: leakType,
  594. Author: diff.author,
  595. File: diff.filePath,
  596. Branch: diff.branchName,
  597. Repo: diff.repoName,
  598. Message: diff.message,
  599. }
  600. if opts.Redact {
  601. leak.Offender = "REDACTED"
  602. leak.Line = "REDACTED"
  603. }
  604. if opts.Verbose {
  605. leak.log()
  606. }
  607. leaks = append(leaks, leak)
  608. return leaks
  609. }
  610. // getShannonEntropy https://en.wiktionary.org/wiki/Shannon_entropy
  611. func getShannonEntropy(data string) (entropy float64) {
  612. if data == "" {
  613. return 0
  614. }
  615. charCounts := make(map[rune]int)
  616. for _, char := range data {
  617. charCounts[char]++
  618. }
  619. invLength := 1.0 / float64(len(data))
  620. for _, count := range charCounts {
  621. freq := float64(count) * invLength
  622. entropy -= freq * math.Log2(freq)
  623. }
  624. return entropy
  625. }
  626. // discoverRepos walks all the children of `path`. If a child directory
  627. // contain a .git file then that repo will be added to the list of repos returned
  628. func discoverRepos(ownerPath string) ([]*RepoDescriptor, error) {
  629. var (
  630. err error
  631. repos []*RepoDescriptor
  632. )
  633. files, err := ioutil.ReadDir(ownerPath)
  634. if err != nil {
  635. return repos, err
  636. }
  637. for _, f := range files {
  638. if f.IsDir() {
  639. repoPath := path.Join(ownerPath, f.Name())
  640. r, err := git.PlainOpen(repoPath)
  641. if err != nil {
  642. continue
  643. }
  644. repos = append(repos, &RepoDescriptor{
  645. repository: r,
  646. name: f.Name(),
  647. path: repoPath,
  648. })
  649. }
  650. }
  651. return repos, err
  652. }
  653. // setLogLevel sets log level for gitleaks. Default is Warning
  654. func setLogs() {
  655. switch opts.Log {
  656. case "info":
  657. log.SetLevel(log.InfoLevel)
  658. case "debug":
  659. log.SetLevel(log.DebugLevel)
  660. case "warn":
  661. log.SetLevel(log.WarnLevel)
  662. default:
  663. log.SetLevel(log.InfoLevel)
  664. }
  665. log.SetFormatter(&log.TextFormatter{
  666. FullTimestamp: true,
  667. })
  668. }
  669. // optsGuard prevents invalid options
  670. func optsGuard() error {
  671. var err error
  672. if opts.GithubOrg != "" && opts.GithubUser != "" {
  673. return fmt.Errorf("github user and organization set")
  674. } else if opts.GithubOrg != "" && opts.OwnerPath != "" {
  675. return fmt.Errorf("github organization set and local owner path")
  676. } else if opts.GithubUser != "" && opts.OwnerPath != "" {
  677. return fmt.Errorf("github user set and local owner path")
  678. }
  679. // do the URL Parse and error checking here, so we can skip it later
  680. // empty string is OK, it will default to the public github URL.
  681. if opts.GithubURL != "" && opts.GithubURL != defaultGithubURL {
  682. if !strings.HasSuffix(opts.GithubURL, "/") {
  683. opts.GithubURL += "/"
  684. }
  685. ghURL, err := url.Parse(opts.GithubURL)
  686. if err != nil {
  687. return err
  688. }
  689. tcpPort := "443"
  690. if ghURL.Scheme == "http" {
  691. tcpPort = "80"
  692. }
  693. timeout := time.Duration(1 * time.Second)
  694. _, err = net.DialTimeout("tcp", ghURL.Host+":"+tcpPort, timeout)
  695. if err != nil {
  696. return fmt.Errorf("%s unreachable, error: %s", ghURL.Host, err)
  697. }
  698. }
  699. if opts.SingleSearch != "" {
  700. singleSearchRegex, err = regexp.Compile(opts.SingleSearch)
  701. if err != nil {
  702. return fmt.Errorf("unable to compile regex: %s, %v", opts.SingleSearch, err)
  703. }
  704. }
  705. if opts.Entropy > 8 {
  706. return fmt.Errorf("The maximum level of entropy is 8")
  707. }
  708. if opts.Report != "" {
  709. if !strings.HasSuffix(opts.Report, ".json") && !strings.HasSuffix(opts.Report, ".csv") {
  710. return fmt.Errorf("Report should be a .json or .csv file")
  711. }
  712. dirPath := filepath.Dir(opts.Report)
  713. if _, err := os.Stat(dirPath); os.IsNotExist(err) {
  714. return fmt.Errorf("%s does not exist", dirPath)
  715. }
  716. }
  717. return nil
  718. }
  719. // loadToml loads of the toml config containing regexes and whitelists.
  720. // This function will first look if the configPath is set and load the config
  721. // from that file. Otherwise will then look for the path set by the GITHLEAKS_CONIFG
  722. // env var. If that is not set, then gitleaks will continue with the default configs
  723. // specified by the const var at the top `defaultConfig`
  724. func loadToml() error {
  725. var (
  726. config Config
  727. configPath string
  728. )
  729. if opts.ConfigPath != "" {
  730. configPath = opts.ConfigPath
  731. _, err := os.Stat(configPath)
  732. if err != nil {
  733. return fmt.Errorf("no gitleaks config at %s", configPath)
  734. }
  735. } else {
  736. configPath = os.Getenv("GITLEAKS_CONFIG")
  737. }
  738. if configPath != "" {
  739. if _, err := toml.DecodeFile(configPath, &config); err != nil {
  740. return fmt.Errorf("problem loading config: %v", err)
  741. }
  742. } else {
  743. _, err := toml.Decode(defaultConfig, &config)
  744. if err != nil {
  745. return fmt.Errorf("problem loading default config: %v", err)
  746. }
  747. }
  748. if len(config.Misc.Entropy) != 0 {
  749. err := entropyLimits(config.Misc.Entropy)
  750. if err != nil {
  751. return err
  752. }
  753. }
  754. if singleSearchRegex != nil {
  755. regexes["singleSearch"] = singleSearchRegex
  756. } else {
  757. for _, regex := range config.Regexes {
  758. regexes[regex.Description] = regexp.MustCompile(regex.Regex)
  759. }
  760. }
  761. whiteListBranches = config.Whitelist.Branches
  762. whiteListCommits = make(map[string]bool)
  763. for _, commit := range config.Whitelist.Commits {
  764. whiteListCommits[commit] = true
  765. }
  766. for _, regex := range config.Whitelist.Files {
  767. whiteListFiles = append(whiteListFiles, regexp.MustCompile(regex))
  768. }
  769. for _, regex := range config.Whitelist.Regexes {
  770. whiteListRegexes = append(whiteListRegexes, regexp.MustCompile(regex))
  771. }
  772. for _, regex := range config.Whitelist.Repos {
  773. whiteListRepos = append(whiteListRepos, regexp.MustCompile(regex))
  774. }
  775. return nil
  776. }
  777. // entropyLimits hydrates entropyRanges which allows for fine tuning entropy checking
  778. func entropyLimits(entropyLimitStr []string) error {
  779. for _, span := range entropyLimitStr {
  780. split := strings.Split(span, "-")
  781. v1, err := strconv.ParseFloat(split[0], 64)
  782. if err != nil {
  783. return err
  784. }
  785. v2, err := strconv.ParseFloat(split[1], 64)
  786. if err != nil {
  787. return err
  788. }
  789. if v1 > v2 {
  790. return fmt.Errorf("entropy range must be ascending")
  791. }
  792. r := entropyRange{
  793. v1: v1,
  794. v2: v2,
  795. }
  796. if r.v1 > 8.0 || r.v1 < 0.0 || r.v2 > 8.0 || r.v2 < 0.0 {
  797. return fmt.Errorf("invalid entropy ranges, must be within 0.0-8.0")
  798. }
  799. entropyRanges = append(entropyRanges, r)
  800. }
  801. return nil
  802. }
  803. // getSSHAuth return an ssh auth use by go-git to clone repos behind authentication.
  804. // If --ssh-key is set then it will attempt to load the key from that path. If not,
  805. // gitleaks will use the default $HOME/.ssh/id_rsa key
  806. func getSSHAuth() (*ssh.PublicKeys, error) {
  807. var (
  808. sshKeyPath string
  809. )
  810. if opts.SSHKey != "" {
  811. sshKeyPath = opts.SSHKey
  812. } else {
  813. c, _ := user.Current()
  814. sshKeyPath = fmt.Sprintf("%s/.ssh/id_rsa", c.HomeDir)
  815. }
  816. sshAuth, err := ssh.NewPublicKeysFromFile("git", sshKeyPath, "")
  817. if err != nil {
  818. if strings.HasPrefix(opts.Repo, "git") {
  819. // if you are attempting to clone a git repo via ssh and supply a bad ssh key,
  820. // the clone will fail.
  821. return nil, fmt.Errorf("unable to generate ssh key: %v", err)
  822. }
  823. }
  824. return sshAuth, nil
  825. }
  826. func (leak Leak) log() {
  827. b, _ := json.MarshalIndent(leak, "", " ")
  828. fmt.Println(string(b))
  829. }