82 lines
2 KiB
Go
82 lines
2 KiB
Go
package share
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"log/slog"
|
|
|
|
"github.com/AlexSSD7/linsk/vm"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const smbPort = 445
|
|
|
|
// TODO: Test SMB backend on macOS.
|
|
|
|
type SMBBackend struct {
|
|
listenIP net.IP
|
|
sharePort *uint16
|
|
}
|
|
|
|
func NewSMBBackend(uc *UserConfiguration) (Backend, *VMShareOptions, error) {
|
|
var ports []vm.PortForwardingRule
|
|
var sharePortPtr *uint16
|
|
if !uc.smbExtMode {
|
|
sharePort, err := getNetworkSharePort(0)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "get network share port")
|
|
}
|
|
|
|
sharePortPtr = &sharePort
|
|
|
|
ports = append(ports, vm.PortForwardingRule{
|
|
HostIP: uc.listenIP,
|
|
HostPort: sharePort,
|
|
VMPort: smbPort,
|
|
})
|
|
}
|
|
|
|
return &SMBBackend{
|
|
listenIP: uc.listenIP,
|
|
sharePort: sharePortPtr,
|
|
}, &VMShareOptions{
|
|
Ports: ports,
|
|
EnableTap: uc.smbExtMode,
|
|
}, nil
|
|
}
|
|
|
|
func (b *SMBBackend) Apply(ctx context.Context, sharePWD string, vc *VMShareContext) (string, error) {
|
|
if b.sharePort != nil && vc.NetTapCtx != nil {
|
|
return "", fmt.Errorf("conflict: configured to use a forwarded port but a net tap configuration was detected")
|
|
}
|
|
|
|
if b.sharePort == nil && vc.NetTapCtx == nil {
|
|
return "", fmt.Errorf("no net tap configuration found")
|
|
}
|
|
|
|
err := vc.FileManager.StartSMB(sharePWD)
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "start smb server")
|
|
}
|
|
|
|
slog.Info("Started the network share successfully", "type", "smb", "ext", vc.NetTapCtx != nil)
|
|
|
|
var shareURL string
|
|
if b.sharePort != nil {
|
|
shareURL = "smb://" + net.JoinHostPort(b.listenIP.String(), fmt.Sprint(*b.sharePort)) + "/linsk"
|
|
} else if vc.NetTapCtx != nil {
|
|
if runtime.GOOS == "windows" {
|
|
shareURL = `\\` + strings.ReplaceAll(vc.NetTapCtx.Net.GuestIP.String(), ":", "-") + ".ipv6-literal.net" + `\linsk`
|
|
} else {
|
|
shareURL = "smb://" + net.JoinHostPort(vc.NetTapCtx.Net.GuestIP.String(), fmt.Sprint(smbPort)) + "/linsk"
|
|
}
|
|
} else {
|
|
return "", fmt.Errorf("no port forwarding and net tap configured")
|
|
}
|
|
|
|
return shareURL, nil
|
|
}
|