80 lines
2.5 KiB
Go
80 lines
2.5 KiB
Go
|
package openssh
|
||
|
|
|
||
|
|
import (
|
||
|
|
sshfx "github.com/pkg/sftp/internal/encoding/ssh/filexfer"
|
||
|
|
)
|
||
|
|
|
||
|
|
const extensionHardlink = "hardlink@openssh.com"
|
||
|
|
|
||
|
|
// RegisterExtensionHardlink registers the "hardlink@openssh.com" extended packet with the encoding/ssh/filexfer package.
|
||
|
|
func RegisterExtensionHardlink() {
|
||
|
|
sshfx.RegisterExtendedPacketType(extensionHardlink, func() sshfx.ExtendedData {
|
||
|
|
return new(HardlinkExtendedPacket)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// ExtensionHardlink returns an ExtensionPair suitable to append into an sshfx.InitPacket or sshfx.VersionPacket.
|
||
|
|
func ExtensionHardlink() *sshfx.ExtensionPair {
|
||
|
|
return &sshfx.ExtensionPair{
|
||
|
|
Name: extensionHardlink,
|
||
|
|
Data: "1",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// HardlinkExtendedPacket defines the hardlink@openssh.com extend packet.
|
||
|
|
type HardlinkExtendedPacket struct {
|
||
|
|
OldPath string
|
||
|
|
NewPath string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Type returns the SSH_FXP_EXTENDED packet type.
|
||
|
|
func (ep *HardlinkExtendedPacket) Type() sshfx.PacketType {
|
||
|
|
return sshfx.PacketTypeExtended
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalPacket returns ep as a two-part binary encoding of the full extended packet.
|
||
|
|
func (ep *HardlinkExtendedPacket) MarshalPacket(reqid uint32, b []byte) (header, payload []byte, err error) {
|
||
|
|
p := &sshfx.ExtendedPacket{
|
||
|
|
ExtendedRequest: extensionHardlink,
|
||
|
|
|
||
|
|
Data: ep,
|
||
|
|
}
|
||
|
|
return p.MarshalPacket(reqid, b)
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalInto encodes ep into the binary encoding of the hardlink@openssh.com extended packet-specific data.
|
||
|
|
func (ep *HardlinkExtendedPacket) MarshalInto(buf *sshfx.Buffer) {
|
||
|
|
buf.AppendString(ep.OldPath)
|
||
|
|
buf.AppendString(ep.NewPath)
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalBinary encodes ep into the binary encoding of the hardlink@openssh.com extended packet-specific data.
|
||
|
|
//
|
||
|
|
// NOTE: This _only_ encodes the packet-specific data, it does not encode the full extended packet.
|
||
|
|
func (ep *HardlinkExtendedPacket) MarshalBinary() ([]byte, error) {
|
||
|
|
// string(oldpath) + string(newpath)
|
||
|
|
size := 4 + len(ep.OldPath) + 4 + len(ep.NewPath)
|
||
|
|
|
||
|
|
buf := sshfx.NewBuffer(make([]byte, 0, size))
|
||
|
|
ep.MarshalInto(buf)
|
||
|
|
return buf.Bytes(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// UnmarshalFrom decodes the hardlink@openssh.com extended packet-specific data from buf.
|
||
|
|
func (ep *HardlinkExtendedPacket) UnmarshalFrom(buf *sshfx.Buffer) (err error) {
|
||
|
|
if ep.OldPath, err = buf.ConsumeString(); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if ep.NewPath, err = buf.ConsumeString(); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// UnmarshalBinary decodes the hardlink@openssh.com extended packet-specific data into ep.
|
||
|
|
func (ep *HardlinkExtendedPacket) UnmarshalBinary(data []byte) (err error) {
|
||
|
|
return ep.UnmarshalFrom(sshfx.NewBuffer(data))
|
||
|
|
}
|