yataf/main.go at master · dsecuredcom/yataf · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package main

import (
	"fmt"
	"github.com/dsecuredcom/yataf/pkg/analyze"
	"github.com/dsecuredcom/yataf/pkg/fetch"
	"github.com/dsecuredcom/yataf/pkg/general"
	"os"
	"strings"
)

func main() {
	Arguments := general.GetArguments()

	var Content string

	if strings.HasPrefix(Arguments.File, "http://") || strings.HasPrefix(Arguments.File, "https://") {
		Content = fetch.GetContentOfRemoteFile(Arguments.File)
	} else {
		Content = fetch.GetContentOfLocalFile(Arguments.File)
	}

	Results := analyze.AnalyzeContent(Content, Arguments.Types)

	if len(Results.Result) == 0 {
		fmt.Printf("\033[31m[i] No results found\033[0m\n")
		os.Exit(1)
	}

	fmt.Printf("\n")

	for _, Result := range Results.Result {
		fmt.Printf("\033[34m[i] Name: %s\033[0m\n", Result.RegEx.Name)
		fmt.Printf("\033[36m[i] RegEx:\033[0m %s\n", Result.RegEx.Matcher)
		for Key, DataExtracted := range Result.DataExtracted {
			AddLine := "\r\n"
			if Result.RegEx.Type == "Urls-Paths" || len(DataExtracted) < 150 {
				AddLine = " "
			}
			if len(DataExtracted) < Arguments.CharLimit {
				fmt.Printf("\033[32m[%d]\033[0m\033[36m[DataExtracted]\033[0m%s%s\n", Key, AddLine, DataExtracted)
				continue
			}

			fmt.Printf("\033[32m[%d]\033[0m\033[36m[DataExtracted (limited to %d chars)]\033[0m%s%s\n", Key, Arguments.CharLimit, AddLine, DataExtracted[:Arguments.CharLimit])

		}
		fmt.Printf("\n")
	}

}