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.
172 lines
3.7 KiB
172 lines
3.7 KiB
2 years ago
|
// TODO:
|
||
|
// * Board UI?
|
||
|
package main
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"crypto/ed25519"
|
||
|
"errors"
|
||
|
"flag"
|
||
|
"fmt"
|
||
|
"io/ioutil"
|
||
|
"net/http"
|
||
|
"os"
|
||
|
"os/user"
|
||
|
"path/filepath"
|
||
|
"time"
|
||
|
|
||
|
"github.com/llimllib/springer/client"
|
||
|
)
|
||
|
|
||
|
func fileExists(name string) bool {
|
||
|
if _, err := os.Stat(name); errors.Is(err, os.ErrNotExist) {
|
||
|
return false
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
func getKeys(folder string) (ed25519.PublicKey, ed25519.PrivateKey) {
|
||
|
// get the expected public key file and private key file paths
|
||
|
var pubfile, privfile string
|
||
|
if len(folder) == 0 {
|
||
|
user, err := user.Current()
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
|
||
|
configPath := os.Getenv("XDG_CONFIG_HOME")
|
||
|
if configPath == "" {
|
||
|
configPath = filepath.Join(user.HomeDir, ".config", "spring83")
|
||
|
}
|
||
|
|
||
|
if err = os.MkdirAll(configPath, os.ModePerm); err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
|
||
|
pubfile = filepath.Join(configPath, "key.pub")
|
||
|
privfile = filepath.Join(configPath, "key.priv")
|
||
|
} else {
|
||
|
pubfile = filepath.Join(folder, "key.pub")
|
||
|
privfile = filepath.Join(folder, "key.priv")
|
||
|
}
|
||
|
|
||
|
// try to load the public and private key files as ed25519 keys
|
||
|
var pubkey ed25519.PublicKey
|
||
|
var privkey ed25519.PrivateKey
|
||
|
var err error
|
||
|
if fileExists(pubfile) && fileExists(privfile) {
|
||
|
pubkey, err = ioutil.ReadFile(pubfile)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
|
||
|
privkey, err = ioutil.ReadFile(privfile)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
} else {
|
||
|
fmt.Println("Generating valid key. This will take a minute")
|
||
|
pubkey, privkey = client.ValidKey()
|
||
|
|
||
|
os.WriteFile(pubfile, pubkey, 0666)
|
||
|
os.WriteFile(privfile, privkey, 0600)
|
||
|
}
|
||
|
|
||
|
return pubkey, privkey
|
||
|
}
|
||
|
|
||
|
func getBody(inputFile string) []byte {
|
||
|
var body []byte
|
||
|
var err error
|
||
|
if inputFile == "-" {
|
||
|
body, err = ioutil.ReadAll(os.Stdin)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
} else {
|
||
|
body, err = ioutil.ReadFile(inputFile)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Prepend a time element. Maybe we should check to see if it's already
|
||
|
// been provided?
|
||
|
timeElt := []byte(fmt.Sprintf("<time datetime=\"%s\">", time.Now().UTC().Format(time.RFC3339)))
|
||
|
body = append(timeElt, body...)
|
||
|
|
||
|
if len(body) == 0 {
|
||
|
panic(fmt.Errorf("input required"))
|
||
|
}
|
||
|
if len(body) > 2217 {
|
||
|
panic(fmt.Errorf("input body too long"))
|
||
|
}
|
||
|
|
||
|
return body
|
||
|
}
|
||
|
|
||
|
func sendBody(server string, body []byte, pubkey ed25519.PublicKey, privkey ed25519.PrivateKey) {
|
||
|
client := &http.Client{}
|
||
|
|
||
|
url := fmt.Sprintf("%s/%x", server, pubkey)
|
||
|
req, err := http.NewRequest(http.MethodPut, url, bytes.NewBuffer(body))
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
|
||
|
sig := ed25519.Sign(privkey, body)
|
||
|
req.Header.Set("Authorization", fmt.Sprintf("Spring-83 Signature=%x", sig))
|
||
|
|
||
|
dt := time.Now().Format(time.RFC1123)
|
||
|
req.Header.Set("If-Unmodified-Since", dt)
|
||
|
|
||
|
resp, err := client.Do(req)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
defer resp.Body.Close()
|
||
|
|
||
|
responseBody, err := ioutil.ReadAll(resp.Body)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
|
||
|
if resp.StatusCode != 200 {
|
||
|
fmt.Printf("%s: %s\n", resp.Status, responseBody)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func usage() {
|
||
|
fmt.Print(`springer [-file=filename] [-key=keyfolder] server
|
||
|
|
||
|
Send a spring83 board to a server
|
||
|
|
||
|
flags:
|
||
|
-file=filename
|
||
|
if present, a file to send to the server instead of accepting bytes on
|
||
|
stdin
|
||
|
-key=keyfolder
|
||
|
a folder the program should use for finding your public and private
|
||
|
keys. It will expect there to be two files, one called "key.pub" and
|
||
|
another called "key.priv" which are your public and private keys,
|
||
|
respectively
|
||
|
`)
|
||
|
os.Exit(1)
|
||
|
}
|
||
|
|
||
|
func main() {
|
||
|
inputFile := flag.String("file", "-", "The file to send to the server")
|
||
|
keyFolder := flag.String("key", "", "A folder to check for key.pub and key.priv")
|
||
|
flag.Usage = usage
|
||
|
flag.Parse()
|
||
|
|
||
|
if len(flag.Args()) < 1 {
|
||
|
usage()
|
||
|
}
|
||
|
server := flag.Args()[0]
|
||
|
|
||
|
pubkey, privkey := getKeys(*keyFolder)
|
||
|
|
||
|
sendBody(server, getBody(*inputFile), pubkey, privkey)
|
||
|
}
|