X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=rpc%2Fdial.go;h=c098714cc0759388dd32e0d723819c4b4c598665;hb=HEAD;hp=c0c6cd4088b2859d758aa0d06cf64405a2b70a16;hpb=8a3f6af248e28ea7efc1bf89751a597d28834942;p=safcm%2Fsafcm.git diff --git a/rpc/dial.go b/rpc/dial.go index c0c6cd4..fa1efcd 100644 --- a/rpc/dial.go +++ b/rpc/dial.go @@ -1,19 +1,7 @@ // Simple RPC-like protocol: establish new connection and upload helper -// Copyright (C) 2021 Simon Ruderich -// -// This program is free software: you can redistribute it and/or modify -// it under the terms of the GNU General Public License as published by -// the Free Software Foundation, either version 3 of the License, or -// (at your option) any later version. -// -// This program is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -// GNU General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with this program. If not, see . +// SPDX-License-Identifier: GPL-3.0-or-later +// Copyright (C) 2021-2024 Simon Ruderich package rpc @@ -24,18 +12,20 @@ import ( "encoding/hex" "fmt" "io" + "io/fs" "os/exec" "strconv" "strings" "ruderich.org/simon/safcm" - "ruderich.org/simon/safcm/remote" ) type SSHConfig struct { Host string User string // optional SshConfig string // optional + + RemoteHelpers fs.FS } func (c *Conn) DialSSH(cfg SSHConfig) error { @@ -43,6 +33,11 @@ func (c *Conn) DialSSH(cfg SSHConfig) error { return fmt.Errorf("cannot reuse Conn") } + if cfg.RemoteHelpers == nil { + return fmt.Errorf("SSHConfig.RemoteHelpers not set") + } + c.remoteHelpers = cfg.RemoteHelpers + remote := cfg.Host if cfg.User != "" { remote = cfg.User + "@" + cfg.Host @@ -83,7 +78,7 @@ func (c *Conn) DialSSH(cfg SSHConfig) error { err = c.dialSSH(stdin, stdout) if err != nil { - c.Kill() + c.Kill() //nolint:errcheck return err } c.conn = safcm.NewGobConn(stdout, stdin) @@ -208,9 +203,9 @@ f return err } - // Get embedded helper binary - helper, err := remote.Helpers.ReadFile( - fmt.Sprintf("helpers/%s-%s", goos, goarch)) + // Get remote helper binary + helper, err := fs.ReadFile(c.remoteHelpers, + fmt.Sprintf("%s-%s", goos, goarch)) if err != nil { return fmt.Errorf("remote not built for GOOS/GOARCH %s/%s", goos, goarch)