76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"net"
|
|
"os"
|
|
|
|
"github.com/AlexSSD7/linsk/vm"
|
|
"github.com/sethvargo/go-password/password"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var runCmd = &cobra.Command{
|
|
Use: "run",
|
|
// TODO: Fill this
|
|
// Short: "",
|
|
Args: cobra.ExactArgs(3),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
vmMountDevName := args[1]
|
|
fsType := args[2]
|
|
|
|
networkSharePort, err := getClosestAvailPort(9000)
|
|
if err != nil {
|
|
slog.Error("Failed to get closest available host port for network file share", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// TODO: `slog` library prints entire stack traces for errors which makes reading errors challenging.
|
|
|
|
os.Exit(runVM(args[0], func(ctx context.Context, i *vm.Instance, fm *vm.FileManager) int {
|
|
err := fm.Mount(vmMountDevName, vm.MountOptions{
|
|
FSType: fsType,
|
|
LUKS: luksFlag,
|
|
})
|
|
if err != nil {
|
|
slog.Error("Failed to mount the disk inside the VM", "error", err)
|
|
return 1
|
|
}
|
|
|
|
sharePWD, err := password.Generate(16, 10, 0, false, false)
|
|
if err != nil {
|
|
slog.Error("Failed to generate ephemeral password for network file share", "error", err)
|
|
return 1
|
|
}
|
|
|
|
shareURI := "smb://linsk:" + sharePWD + "@127.0.0.1:" + fmt.Sprint(networkSharePort)
|
|
|
|
fmt.Fprintf(os.Stderr, "================\n[Network File Share Config]\nThe network file share was started. Please use the credentials below to connect to the file server.\n\nType: SMB\nServer Address: smb://127.0.0.1:%v\nUsername: linsk\nPassword: %v\n\nShare URI: %v\n================\n", networkSharePort, sharePWD, shareURI)
|
|
|
|
err = fm.StartSMB([]byte(sharePWD))
|
|
if err != nil {
|
|
slog.Error("Failed to start SMB server", "error", err)
|
|
return 1
|
|
}
|
|
|
|
slog.Info("Started the network share successfully", "type", "smb")
|
|
|
|
<-ctx.Done()
|
|
return 0
|
|
}, []vm.PortForwardingConfig{{
|
|
HostIP: net.ParseIP("127.0.0.1"), // TODO: Make this changeable.
|
|
HostPort: networkSharePort,
|
|
VMPort: 445,
|
|
}}, false))
|
|
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var luksFlag bool
|
|
|
|
func init() {
|
|
runCmd.Flags().BoolVarP(&luksFlag, "luks", "l", false, "Use cryptsetup to open a LUKS volume (password will be prompted)")
|
|
}
|