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.
127 lines
2.6 KiB
127 lines
2.6 KiB
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
)
|
|
|
|
// manage debugging messages
|
|
|
|
const debug debugging = true
|
|
|
|
type debugging bool
|
|
|
|
func (d debugging) log(format string, args ...interface{}) {
|
|
if d {
|
|
log.Printf(format, args...)
|
|
}
|
|
}
|
|
|
|
///////////
|
|
|
|
var conf_file = flag.String("c", "./scorsh.cfg", "Configuration file for SCORSH")
|
|
|
|
func SCORSHerr(err int) error {
|
|
|
|
var err_str string
|
|
|
|
switch err {
|
|
case SCORSH_ERR_NO_FILE:
|
|
err_str = "Invalid file name"
|
|
case SCORSH_ERR_KEYRING:
|
|
err_str = "Invalid keyring"
|
|
case SCORSH_ERR_NO_REPO:
|
|
err_str = "Invalid repository"
|
|
case SCORSH_ERR_NO_COMMIT:
|
|
err_str = "Invalid commit ID"
|
|
case SCORSH_ERR_SIGNATURE:
|
|
err_str = "Invalid signature"
|
|
default:
|
|
err_str = "Generic Error"
|
|
}
|
|
return fmt.Errorf("%s", err_str)
|
|
|
|
}
|
|
|
|
func FindMatchingWorkers(master *SCORSHmaster, msg *SCORSHmsg) []*SCORSHworker {
|
|
|
|
var ret []*SCORSHworker
|
|
|
|
for idx, w := range master.Workers {
|
|
if w.Matches(msg.Repo, msg.Branch) {
|
|
debug.log("--- Worker: %s matches %s:%s\n", w.Name, msg.Repo, msg.Branch)
|
|
ret = append(ret, &(master.Workers[idx]))
|
|
}
|
|
}
|
|
return ret
|
|
}
|
|
|
|
func Master(master *SCORSHmaster) {
|
|
|
|
// master main loop:
|
|
|
|
var matching_workers []*SCORSHworker
|
|
var push_msg SCORSHmsg
|
|
|
|
matching_workers = make([]*SCORSHworker, len(master.Workers))
|
|
|
|
log.Println("[master] Master started ")
|
|
|
|
for {
|
|
select {
|
|
// - receive stuff from the spooler
|
|
case push_msg = <-master.Spooler:
|
|
|
|
debug.log("[master] received message: %s\n", push_msg)
|
|
|
|
// - lookup the repos map for matching workers
|
|
matching_workers = FindMatchingWorkers(master, &push_msg)
|
|
debug.log("[master] matching workers: %s\n", matching_workers)
|
|
|
|
// add the message to PendingMsg
|
|
//...
|
|
// - dispatch the message to all the matching workers
|
|
for _, w := range matching_workers {
|
|
// increase the counter associated to the message
|
|
w.MsgChan <- push_msg
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func InitMaster() *SCORSHmaster {
|
|
|
|
master := ReadGlobalConfig(*conf_file)
|
|
|
|
master.Repos = make(map[string][]*SCORSHworker)
|
|
master.WorkingMsg = make(map[string]int)
|
|
// This is the mutex-channel on which we receive acks from workers
|
|
master.StatusChan = make(chan SCORSHmsg, 1)
|
|
master.Spooler = make(chan SCORSHmsg, 1)
|
|
|
|
err_workers := StartWorkers(master)
|
|
if err_workers != nil {
|
|
log.Fatal("Error starting workers: ", err_workers)
|
|
} else {
|
|
log.Println("Workers started correctly")
|
|
}
|
|
err_spooler := StartSpooler(master)
|
|
if err_spooler != nil {
|
|
log.Fatal("Error starting spooler: ", err_spooler)
|
|
}
|
|
return master
|
|
|
|
}
|
|
|
|
func main() {
|
|
|
|
flag.Parse()
|
|
|
|
master := InitMaster()
|
|
|
|
go Master(master)
|
|
|
|
<-master.StatusChan
|
|
|
|
}
|
|
|