Cameradar v4 (#212)

* Refactor of cameradar library * Old unit tests updated & improved. New unit tests inc * Update documentation & issue template * Update dependencies * Update TravisCI build script to reflect argument change * Remove outdated contributing guide * Update README with more examples and remove part on library * Add second camera to Travis build script & improve error detection * Fix typo in travis script & add missing image to readme * Remember that travis uses bash syntax not fish * Use relative paths for images in the README
2019-05-26 08:33:08 +02:00
parent 2e49587cc2
commit 212ac2f0d5
28 changed files with 1535 additions and 1163 deletions
@@ -0,0 +1,92 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/spf13/pflag"
+	"github.com/spf13/viper"
+	"github.com/ullaakut/cameradar"
+	"github.com/ullaakut/disgo"
+	"github.com/ullaakut/disgo/style"
+)
+
+func parseArguments() error {
+	viper.SetEnvPrefix("cameradar")
+	viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
+
+	pflag.StringSliceP("targets", "t", []string{}, "The targets on which to scan for open RTSP streams - required (ex: 172.16.100.0/24)")
+	pflag.StringSliceP("ports", "p", []string{"554", "5554", "8554"}, "The ports on which to search for RTSP streams")
+	pflag.StringP("custom-routes", "r", "<GOPATH>/src/github.com/ullaakut/cameradar/dictionaries/routes", "The path on which to load a custom routes dictionary")
+	pflag.StringP("custom-credentials", "c", "<GOPATH>/src/github.com/ullaakut/cameradar/dictionaries/credentials.json", "The path on which to load a custom credentials JSON dictionary")
+	pflag.IntP("speed", "s", 4, "The nmap speed preset to use for discovery")
+	pflag.DurationP("timeout", "T", 2*time.Second, "The timeout in miliseconds to use for attack attempts")
+	pflag.BoolP("debug", "d", true, "Enable the debug logs")
+	pflag.BoolP("verbose", "v", false, "Enable the verbose logs")
+	pflag.BoolP("help", "h", false, "displays this help message")
+
+	viper.AutomaticEnv()
+
+	pflag.Parse()
+
+	err := viper.BindPFlags(pflag.CommandLine)
+	if err != nil {
+		return err
+	}
+
+	if viper.GetBool("help") {
+		pflag.Usage()
+		fmt.Println("\nExamples of usage:")
+		fmt.Println("\tScanning your home network for RTSP streams:\tcameradar -t 192.168.0.0/24")
+		fmt.Println("\tScanning a remote camera on a specific port:\tcameradar -t 172.178.10.14 -p 18554 -s 2")
+		fmt.Println("\tScanning an unstable remote network: \t\tcameradar -t 172.178.10.14/24 -s 1 --timeout 10000 -l")
+		os.Exit(0)
+	}
+
+	if viper.GetStringSlice("targets") == nil {
+		return errors.New("targets (-t, --targets) argument required\n    examples:\n      - 172.16.100.0/24\n      - localhost\n      - 8.8.8.8")
+	}
+
+	return nil
+}
+
+func main() {
+	err := parseArguments()
+	if err != nil {
+		printErr(err)
+	}
+
+	c, err := cameradar.New(
+		cameradar.WithTargets(viper.GetStringSlice("targets")),
+		cameradar.WithPorts(viper.GetStringSlice("ports")),
+		cameradar.WithDebug(viper.GetBool("debug")),
+		cameradar.WithVerbose(viper.GetBool("verbose")),
+		cameradar.WithCustomCredentials(viper.GetString("custom-credentials")),
+		cameradar.WithCustomRoutes(viper.GetString("custom-routes")),
+		cameradar.WithSpeed(viper.GetInt("speed")),
+		cameradar.WithTimeout(viper.GetDuration("timeout")),
+	)
+	if err != nil {
+		printErr(err)
+	}
+
+	scanResult, err := c.Scan()
+	if err != nil {
+		printErr(err)
+	}
+
+	streams, err := c.Attack(scanResult)
+	if err != nil {
+		printErr(err)
+	}
+
+	c.PrintStreams(streams)
+}
+
+func printErr(err error) {
+	disgo.Errorln(style.Failure(style.SymbolCross), err)
+	os.Exit(1)
+}