1
0
Fork 0

stuff? stuff!

This commit is contained in:
fanir 2014-01-14 01:18:52 +01:00
parent d1e405ba67
commit d564133b72
3 changed files with 110 additions and 39 deletions

18
config.json Normal file
View file

@ -0,0 +1,18 @@
// This is JSON. But with comments! They start with "//" (without the quotes)
{
// The file(s) to watch. If no given file contains the
// signature and key in the keyfile, the screen is locked.
// You can use shell-replacemants like * and ?
"Lockfile": "/dev/sd??",
// The file containing the key.
"Keyfile": "usblocker.key",
// The location of the xlock binary.
"Xlock_path": "/usr/bin/xlock",
// Exclude possible lockfiles, if they contain the wrong
// signature or no signature at all.
"Blacklist_devices": true
}

BIN
usblocker

Binary file not shown.

View file

@ -1,15 +1,39 @@
package main package main
import ( import (
//"code.google.com/p/gcfg"
"encoding/json"
"fmt" "fmt"
"io/ioutil"
"log" "log"
"os" "os"
"os/exec"
"path/filepath" "path/filepath"
"regexp"
"strings" "strings"
"time" "time"
) )
type Configuration struct {
Lockfile string
Keyfile string
Xlock_path string
Blacklist_devices bool
}
var VERSION = "v0.1.0" var VERSION = "v0.1.0"
var CONFIGFILE = "config.json"
// Default configuration
var cfg = Configuration{
Lockfile: "/dev/sd??",
Keyfile: "usblocker.key",
Xlock_path: "/bin/xlock",
Blacklist_devices: true,
}
var FIRSTRUN = true
var cmd = exec.Command("")
/* Shows the help message /* Shows the help message
*/ */
@ -17,12 +41,12 @@ func ShowHelp() {
fmt.Println("Watches a device and locks the screen if the device is missing.") fmt.Println("Watches a device and locks the screen if the device is missing.")
fmt.Println() fmt.Println()
fmt.Println("Usage:") fmt.Println("Usage:")
fmt.Println(" usblocker run: Start USBLocker.") fmt.Println(" usblocker start: Start USBLocker.")
//fmt.Println(" usblocker daemon: Runs in background.") //fmt.Println(" usblocker daemon: Runs in background.")
//fmt.Println(" usblocker create: Prepares a device for usage with USBLocker.") //fmt.Println(" usblocker create: Prepares a device for usage with USBLocker.")
} }
/* /* Shows the message of the day
*/ */
func ShowMOTD() { func ShowMOTD() {
fmt.Println("Welcome to USBLocker ", VERSION) fmt.Println("Welcome to USBLocker ", VERSION)
@ -42,74 +66,81 @@ func WatcherUnlocked(device string) {
time.Sleep(time.Duration(1) * time.Second) time.Sleep(time.Duration(1) * time.Second)
} }
log.Println("Locked") log.Println("Locked")
var args []string cmd = exec.Command("/usr/bin/xlock", "-mode", "blank")
args = append(args, "-nolock") // Fix for... something. First passed argument keeps beeing ignored, for whatever reason. err := cmd.Start()
//args = append(args, "-info")
//args = append(args, "")
args = append(args, "-mode")
args = append(args, "blank")
//log.Println(args)
var attr os.ProcAttr
lpid, err := os.StartProcess("/usr/bin/xlock", args, &attr)
if err != nil { if err != nil {
log.Fatal("Error: ", err) log.Fatal("Error: ", err)
} }
WatcherLocked(lpid)
WatcherLocked()
} }
/* Watcher for locked state / no device /* Watcher for locked state / no device
*/ */
func WatcherLocked(lpid *os.Process) { func WatcherLocked() {
var passphrase string // TEH KEY!!1! var passphrase string // the key
var passtest = make([]byte, 1024) // slice for testing the passphrase
var devices []string // list of devices var devices []string // list of devices
var devblacklist []string // blacklist of devices
var lfile *os.File // device file var lfile *os.File // device file
ldev := "" // device file name ldev := "" // device file name
version := "v0" // version string lfileversion := "v1" // version string
var passtest []byte
for i := 0; i < 1024; i++ {
passtest = append(passtest, 0)
}
// Set the passphrase // Set the passphrase
passphrase = "wictQQT86qi7YhZP3wVIZek+x9X0ELJA5LN2amEdh54xXs0yGlGx2kRrfIXC+C3iIX9iYRMKrv0bFSyR29SVTVhl2Q0LFOvvAmawiRrJ9MLwyiBrTVfhsVLcZcKNI5Up6xjebezqyhTcZBzEuRYk4VueCXbjI0ISK0LQgMdKxiybQOt1+FLAmgtYj+izB4xvEd9A+wQOnb+wKqZ8TpWh8h/HVfQLgobXDND8I7SXYC1Qm5Kv4wUBWUDg3YOWwMLWfYvyCU4ZyKi5FG3GWvLvKKMxk7vHA/YDyz/eXo3x07zmqDqZHP19k9nGex6ubzmZOWLZz1zEwsdweXCqYps9waXwhJ/9zgrx735IRE3gPL/KYP/X5WTMBXMYFkCAzT6LHNO2gWzdHqdIjyBPBoOHDtVFhKp7prN6GCn/h4Bwr5VzxC+NrUQPG3AqdQdnNGf9zmUV7VE4ZtRdC5dA5I0rJkX9xIbps7IiF9TACIHwXWVEQ+VpQFXRT063DcMxzlmP0b55k4Y2YEZsFEIYO0/hNqMl5D/ETNriiKIpr296WfRqPC6RsGgodZ6TIA8/XqRWsmkCjBHHPSv7BfVDNUbH/hiINH+8U/YfUGfFOly301HXZlEpqLjCXdEw8kzNN06anFTd+6YsTOScGLeddi1urqUCQ9v5nM1iAoXj0djOUdbZks8RgXdf583hhwHWG/Ib5HLSr4ISMQxFhjWKcHvo4ffeYLiOP85ulNsL5ZbtOUL2MX/DuLE1mKopoj50Jjb0mV+QiXUkifjZgGuXyGuZYfNYtMSmkRGERkJfrLeQyd69o+CLG7sfywOgelFWHTwkeVQNbpjRPmTLdSsq0MuJeIlcIAEKECQgBkNMOE2WuHOC2jr8Ba018wESW/hXAairG7ft7Mmf+AMCdjamFhb56YFbE9LumQ8ILgK9q6FQ6OU39qP+wCivdtSEnDnip4EVw7yItj1/0GVs1QPr20gDHO0G2t90JPyApmqtfKq4CY/tKgDhAkBwNXyGEVCrsA==" passphrase = "wictQQT86qi7YhZP3wVIZek+x9X0ELJA5LN2amEdh54xXs0yGlGx2kRrfIXC+C3iIX9iYRMKrv0bFSyR29SVTVhl2Q0LFOvvAmawiRrJ9MLwyiBrTVfhsVLcZcKNI5Up6xjebezqyhTcZBzEuRYk4VueCXbjI0ISK0LQgMdKxiybQOt1+FLAmgtYj+izB4xvEd9A+wQOnb+wKqZ8TpWh8h/HVfQLgobXDND8I7SXYC1Qm5Kv4wUBWUDg3YOWwMLWfYvyCU4ZyKi5FG3GWvLvKKMxk7vHA/YDyz/eXo3x07zmqDqZHP19k9nGex6ubzmZOWLZz1zEwsdweXCqYps9waXwhJ/9zgrx735IRE3gPL/KYP/X5WTMBXMYFkCAzT6LHNO2gWzdHqdIjyBPBoOHDtVFhKp7prN6GCn/h4Bwr5VzxC+NrUQPG3AqdQdnNGf9zmUV7VE4ZtRdC5dA5I0rJkX9xIbps7IiF9TACIHwXWVEQ+VpQFXRT063DcMxzlmP0b55k4Y2YEZsFEIYO0/hNqMl5D/ETNriiKIpr296WfRqPC6RsGgodZ6TIA8/XqRWsmkCjBHHPSv7BfVDNUbH/hiINH+8U/YfUGfFOly301HXZlEpqLjCXdEw8kzNN06anFTd+6YsTOScGLeddi1urqUCQ9v5nM1iAoXj0djOUdbZks8RgXdf583hhwHWG/Ib5HLSr4ISMQxFhjWKcHvo4ffeYLiOP85ulNsL5ZbtOUL2MX/DuLE1mKopoj50Jjb0mV+QiXUkifjZgGuXyGuZYfNYtMSmkRGERkJfrLeQyd69o+CLG7sfywOgelFWHTwkeVQNbpjRPmTLdSsq0MuJeIlcIAEKECQgBkNMOE2WuHOC2jr8Ba018wESW/hXAairG7ft7Mmf+AMCdjamFhb56YFbE9LumQ8ILgK9q6FQ6OU39qP+wCivdtSEnDnip4EVw7yItj1/0GVs1QPr20gDHO0G2t90JPyApmqtfKq4CY/tKgDhAkBwNXyGEVCrsA=="
for { for {
var err error var err error
devices, err = filepath.Glob("/dev/?d?1") devices, err = filepath.Glob(cfg.Lockfile)
if err != nil { if err != nil {
log.Fatal("Error while getting list of aviable devices (/dev/?d?1):", err) log.Fatal("Error while getting list of aviable devices ", cfg.Lockfile, ":", err)
}
if len(devices) == 0 {
log.Fatal("Not found: ", cfg.Lockfile)
} }
// Search lock device // Search lock device
for i := range devices { for _, device := range devices {
log.Println("Checking", devices[i], "for usblocker data...") log.Println("Checking", device, "for usblocker data...")
lfile, err = os.Open(devices[i]) lfile, err = os.Open(device)
if err != nil { if err != nil {
log.Fatal("Error while opening the device file:", err) log.Fatal("Error while opening the device file:", err)
} }
// Check for magic string // Check for magic string
magictest := []byte(" ") magictest := make([]byte, 9)
length, err := lfile.Read(magictest) length, err := lfile.Read(magictest)
if err != nil { if err != nil {
log.Fatal("Error while reading the device file:", err) log.Fatal("Error while reading the device file:", err)
_ = lfile.Close() _ = lfile.Close()
} }
if length == 9 && string(magictest) == "usblocker" { if length == 9 && string(magictest) == "usblocker" {
ldev = devices[i] ldev = device
break break
} else {
if cfg.Blacklist_devices && FIRSTRUN {
if len(devblacklist) == 0 {
devblacklist = append(devblacklist, device)
} else {
for _, v := range devblacklist {
if v != device {
devblacklist = append(devblacklist, device)
}
}
}
FIRSTRUN = false
}
} }
} }
// Check lockdev version // Check lockdev version
if ldev != "" { if ldev != "" {
log.Println("Checking", ldev, "for right version...") log.Println("Checking", ldev, "for right version...")
versiontest := []byte(" ") versiontest := make([]byte, 7)
length, err := lfile.Read(versiontest) length, err := lfile.Read(versiontest)
if err != nil { if err != nil {
log.Fatal("Error while reading the device file:", err) log.Fatal("Error while reading the device file:", err)
} }
if length == 7 && strings.HasPrefix(string(versiontest), version) { if length == 7 && strings.HasPrefix(string(versiontest), lfileversion) {
log.Println("Checking passphrase...") log.Println("Checking passphrase...")
length, err := lfile.Read(passtest) length, err := lfile.Read(passtest)
_ = lfile.Close() _ = lfile.Close()
@ -119,9 +150,9 @@ func WatcherLocked(lpid *os.Process) {
if length == 1024 && string(passtest) == passphrase { if length == 1024 && string(passtest) == passphrase {
// And finally: UNLOCK! // And finally: UNLOCK!
log.Println("Unlocked") log.Println("Unlocked")
if lpid != nil { if cmd.Process != nil {
lpid.Kill() cmd.Process.Kill()
_, _ = lpid.Wait() _ = cmd.Wait()
} }
WatcherUnlocked(ldev) WatcherUnlocked(ldev)
} }
@ -136,18 +167,19 @@ func WatcherLocked(lpid *os.Process) {
*/ */
func main() { func main() {
// Determining what to do // Determining what to do
if len(os.Args) == 1 { if len(os.Args) < 2 {
ShowHelp() ShowHelp()
return return
} }
switch os.Args[1] { switch os.Args[1] {
case "run": case "start":
//case "daemon": //case "daemon":
// log.Fatal("Not yet implemented. Probably won't be implemented anyway, go doesn't seem to support forking...") // log.Fatal("Not yet implemented. Probably won't be implemented anyway, go doesn't seem to support forking...")
case "create": case "create":
log.Fatal("Not yet implemented.") log.Fatal("Not yet implemented.")
default: default:
ShowHelp() ShowHelp()
return
} }
// Check for root // Check for root
@ -155,6 +187,27 @@ func main() {
log.Fatal("Y U NO GOT ROOT??") log.Fatal("Y U NO GOT ROOT??")
} }
ShowMOTD() // Get configuration from file config.json
WatcherLocked(nil) b, err := ioutil.ReadFile(CONFIGFILE)
if err != nil {
log.Println("Cannot read configuration file:", err)
log.Println("Using default configuration")
} else {
re := regexp.MustCompile("(?m)^\\s*//.*$")
jdata := re.ReplaceAllString(string(b), "")
err = json.Unmarshal([]byte(jdata), &cfg)
if err != nil {
log.Fatal("Failed to get configuration from json:", err)
}
}
// Check for xlock
_, err = os.Stat(cfg.Xlock_path)
if err != nil {
os.Stdout.Write([]byte("Is xlock installed and the right path set in config.json?\n"))
log.Fatal("Stat for ", cfg.Xlock_path, " failed:", err)
}
ShowMOTD()
WatcherLocked()
} }