You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package commands
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
type UploadFileArgs struct {
|
|
Command string `json:"command"`
|
|
LocalFilepath string `json:"local_filepath"`
|
|
DestinationFilepath string `json:"destination_filepath"`
|
|
PrivateKeyFile string `json:"private_key_file"`
|
|
Username string `json:"username"`
|
|
Host string `json:"host"`
|
|
}
|
|
|
|
func (ufa *UploadFileArgs) ReadArgs() {
|
|
|
|
if len(os.Args) != 7 {
|
|
fmt.Fprintf(os.Stderr, "Usage: %s upload_file <local-file> <remote-path> <private-key> <username> <host>\n", os.Args[0])
|
|
os.Exit(1)
|
|
}
|
|
|
|
ufa.Command = os.Args[1]
|
|
ufa.LocalFilepath = os.Args[2]
|
|
ufa.DestinationFilepath = os.Args[3]
|
|
ufa.PrivateKeyFile = os.Args[4]
|
|
ufa.Username = os.Args[5]
|
|
ufa.Host = os.Args[6]
|
|
|
|
}
|
|
|
|
func UploadFile(args UploadFileArgs, stdout, stderr io.Writer) error {
|
|
|
|
client, err := GetClient(args.PrivateKeyFile, args.Username, args.Host)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer client.Close()
|
|
|
|
session, err := client.NewSession()
|
|
if err != nil {
|
|
return fmt.Errorf("error creating SSH session: %w", err)
|
|
}
|
|
defer session.Close()
|
|
|
|
source_file, err := os.Open(args.LocalFilepath)
|
|
if err != nil {
|
|
return fmt.Errorf("error opening local file: %w", err)
|
|
}
|
|
defer source_file.Close()
|
|
|
|
info, err := source_file.Stat()
|
|
if err != nil {
|
|
return fmt.Errorf("error stating local file: %w", err)
|
|
}
|
|
|
|
w, err := session.StdinPipe()
|
|
if err != nil {
|
|
return fmt.Errorf("error creating stdin pipe: %w", err)
|
|
}
|
|
|
|
if err := session.Start("scp -t " + args.DestinationFilepath); err != nil {
|
|
return fmt.Errorf("error starting scp: %w", err)
|
|
}
|
|
|
|
fmt.Fprintf(w, "C0644 %d %s\n", info.Size(), filepath.Base(args.LocalFilepath))
|
|
if _, err := io.Copy(w, source_file); err != nil {
|
|
return fmt.Errorf("error sending file: %w", err)
|
|
}
|
|
fmt.Fprint(w, "\x00")
|
|
w.Close()
|
|
|
|
if err := session.Wait(); err != nil {
|
|
return fmt.Errorf("error during scp transfer: %w", err)
|
|
}
|
|
|
|
fmt.Fprintln(stdout, "Success")
|
|
return nil
|
|
|
|
}
|