signed-commit remote shell (see also https://github.com/dyne/scorsh)
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
2.2 KiB
90 lines
2.2 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/fsnotify/fsnotify"
|
|
"github.com/go-yaml/yaml"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
// "time"
|
|
)
|
|
|
|
// parse a request file and return a SCORSHmessage
|
|
func parse_request(fname string, msg *SCORSHmsg) error {
|
|
|
|
debug.log("[parse_request] message at start: %s\n", msg)
|
|
|
|
data, err := ioutil.ReadFile(fname)
|
|
if err != nil {
|
|
log.Printf("Unable to open file: %s\n", fname)
|
|
return SCORSHerr(SCORSH_ERR_NO_FILE)
|
|
}
|
|
|
|
debug.log("[parse_request] file contains: \n%s\n", data)
|
|
|
|
debug.log("[parse_request] reading message from file: %s\n", fname)
|
|
|
|
err = yaml.Unmarshal([]byte(data), msg)
|
|
if err != nil {
|
|
return fmt.Errorf("Error parsing message: %s", err)
|
|
}
|
|
debug.log("[parse_request] got message: %s\n", msg)
|
|
return nil
|
|
}
|
|
|
|
func spooler(watcher *fsnotify.Watcher, master chan SCORSHmsg) {
|
|
|
|
log.Println("Spooler started correctly")
|
|
|
|
for {
|
|
select {
|
|
case event := <-watcher.Events:
|
|
// Here we manage genuine events from fsnotify. We catch the
|
|
// "Write" event, which should happen only when the file is
|
|
// created
|
|
if event.Op == fsnotify.Write {
|
|
var msg SCORSHmsg
|
|
debug.log("[spooler] new file %s detected\n", event.Name)
|
|
err := parse_request(event.Name, &msg)
|
|
if err != nil {
|
|
log.Printf("Invalid packet received. [%s]\n", err)
|
|
}
|
|
debug.log("[spooler] read message: %s\n", msg)
|
|
msg.Path = event.Name
|
|
master <- msg
|
|
}
|
|
case err := <-watcher.Errors:
|
|
// here we manage event errors
|
|
log.Println("[spooler] error: ", err)
|
|
case msg := <-master:
|
|
// Here we receive messages from the master about files to be
|
|
// removed
|
|
log.Printf("[spooler] received deletion request for: %s\n", msg.Path)
|
|
err := os.Remove(msg.Path)
|
|
if err != nil {
|
|
log.Printf("[spooler] error removing file: %s\n", err)
|
|
} else {
|
|
log.Printf("[spooler] file %s successfully removed\n", msg.Path)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func StartSpooler(master *SCORSHmaster) error {
|
|
|
|
watcher, err := fsnotify.NewWatcher()
|
|
|
|
if err != nil {
|
|
return fmt.Errorf("Error creating watcher: %s\n", err)
|
|
}
|
|
|
|
err = watcher.Add(master.Spooldir)
|
|
if err != nil {
|
|
return fmt.Errorf("Error adding folder: %s\n", err)
|
|
}
|
|
|
|
go spooler(watcher, master.Spooler)
|
|
|
|
return nil
|
|
}
|
|
|