75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
|
|
"deadbeef.codes/steven/hyp/otphyp"
|
|
)
|
|
|
|
func main() {
|
|
|
|
if len(os.Args) < 2 {
|
|
usage()
|
|
}
|
|
|
|
switch os.Args[1] {
|
|
case "generatesecret":
|
|
sharedSecret, err := otphyp.GenerateSecret()
|
|
if err != nil {
|
|
log.Fatalf("failed to generate shared secret: %v", err)
|
|
}
|
|
f, err := os.Create("hyp.secret")
|
|
if err != nil {
|
|
log.Fatalf("failed to create file 'hyp.secret': %v", err)
|
|
}
|
|
_, err = f.WriteString(sharedSecret)
|
|
if err != nil {
|
|
log.Fatalf("failed to write to file 'hyp.secret': %v", err)
|
|
}
|
|
f.Close()
|
|
fmt.Println("Created file hyp.secret")
|
|
case "server":
|
|
secretBytes, err := os.ReadFile("hyp.secret")
|
|
if err != nil {
|
|
log.Fatalf("failed to read file 'hyp.secret': %v", err)
|
|
}
|
|
sharedSecret = string(secretBytes)
|
|
if len(os.Args) < 3 {
|
|
usage()
|
|
}
|
|
packetServer(os.Args[2])
|
|
}
|
|
}
|
|
|
|
func usage() {
|
|
fmt.Print(`hyp <command>
|
|
Commands:
|
|
generatesecret - creates a pre shared key file named hyp.secret which can be distributed to a trusted client
|
|
server <device> - TBD
|
|
|
|
Example Usage:
|
|
|
|
# Linux
|
|
hyp server "/dev/eth0"
|
|
|
|
# Windows - get-netadapter | where {$_.Name -eq “Ethernet”} | Select-Object -Property DeviceName
|
|
hyp server "\\Device\\NPF_{A066F7DE-CC2D-4E4B-97C4-BF0EC4C03649}"
|
|
|
|
`)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// TBD: Implement
|
|
func handleSuccess(srcip string) {
|
|
fmt.Println("Success for ", srcip)
|
|
|
|
cmd := exec.Command("iptables", "-A", "INPUT", "-p", "tcp", "-s", srcip, "--dport", "22", "-j", "ACCEPT")
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
log.Printf("failed to execute iptables command for '%s': %v", srcip, err)
|
|
}
|
|
}
|