Lift chunker up to recent sshbytestream changes.

This commit is contained in:
Christian Schwarz 2017-04-30 16:09:41 +02:00
parent c7927b27c1
commit b9361f275a

View File

@ -2,7 +2,6 @@ package main
import ( import (
"flag" "flag"
"github.com/zrepl/zrepl/model"
"github.com/zrepl/zrepl/sshbytestream" "github.com/zrepl/zrepl/sshbytestream"
"github.com/zrepl/zrepl/util" "github.com/zrepl/zrepl/util"
// "bytes" // "bytes"
@ -20,7 +19,9 @@ func main() {
incomingFile := flag.String("incoming.file", "", "file to deliver to callers") incomingFile := flag.String("incoming.file", "", "file to deliver to callers")
outgoingHost := flag.String("outgoing.sshHost", "", "ssh host") outgoingHost := flag.String("outgoing.sshHost", "", "ssh host")
outgoingUser := flag.String("outgoing.sshUser", "", "ssh user") outgoingUser := flag.String("outgoing.sshUser", "", "ssh user")
outgoingIdentity := flag.String("outgoing.sshIdentity", "", "ssh private key")
outgoingPort := flag.Uint("outgoing.sshPort", 22, "ssh port") outgoingPort := flag.Uint("outgoing.sshPort", 22, "ssh port")
outgoingFile := flag.String("outgoing.File", "", "")
flag.Parse() flag.Parse()
switch { switch {
@ -47,21 +48,24 @@ func main() {
case *mode == "outgoing": case *mode == "outgoing":
conn, err := sshbytestream.Outgoing("client", model.SSHTransport{ conn, err := sshbytestream.Outgoing(sshbytestream.SSHTransport{
Host: *outgoingHost, Host: *outgoingHost,
User: *outgoingUser, User: *outgoingUser,
Port: uint16(*outgoingPort), IdentityFile: *outgoingIdentity,
Options: []string{"Compression=no"}, Port: uint16(*outgoingPort),
TransportOpenCommand: []string{"/tmp/sshwrap", "-mode", "incoming", "-incoming.file", "/random.img"},
}) })
if err != nil {
panic(err)
}
f, err := os.OpenFile(*outgoingFile, os.O_WRONLY, 0600)
if err != nil { if err != nil {
panic(err) panic(err)
} }
unchunker := chunking.NewUnchunker(conn) unchunker := chunking.NewUnchunker(conn)
_, err = io.Copy(os.Stdout, &unchunker) _, err = io.Copy(f, unchunker)
if err != nil { if err != nil {
panic(err) panic(err)
} }