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.
171 lines
3.7 KiB
171 lines
3.7 KiB
// 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) |
|
}
|
|
|