2603cdfa8b
* chore: Update dependencies in go.mod and go.sum * chore: Update submodule commit for crypto * refactor: Update hook functions to return ssh.PipePackageHookMethod for better integration * fix: Increase sleep duration in TestFixed to ensure proper execution * fix: Increase sleep duration in TestFixed to ensure proper file flush * fix: Update docker-compose service images and modify test to use host-password-old * refactor: Simplify hook functions to return error instead of PipePackageHookMethod for better error handling * feat: Add reply-ping flag to enable ping response for compatibility with older sshd * fix: Update subproject commit reference in crypto
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
func TestHookChain_Hook(t *testing.T) {
|
|
hc := &hookChain{}
|
|
|
|
// Mock hooks
|
|
hook1 := func(packet []byte) (ssh.PipePacketHookMethod, []byte, error) {
|
|
return ssh.PipePacketHookTransform, append(packet, '1'), nil
|
|
}
|
|
|
|
hook2 := func(packet []byte) (ssh.PipePacketHookMethod, []byte, error) {
|
|
return ssh.PipePacketHookTransform, append(packet, '2'), nil
|
|
}
|
|
|
|
hook3 := func(packet []byte) (ssh.PipePacketHookMethod, []byte, error) {
|
|
return ssh.PipePacketHookTransform, append(packet, '3'), nil
|
|
}
|
|
|
|
hook4 := func(packet []byte) (ssh.PipePacketHookMethod, []byte, error) {
|
|
return ssh.PipePacketHookReply, append(packet, '4'), nil
|
|
}
|
|
|
|
hook5 := func(packet []byte) (ssh.PipePacketHookMethod, []byte, error) {
|
|
return ssh.PipePacketHookTransform, append(packet, '5'), nil
|
|
}
|
|
|
|
hc.append(hook1)
|
|
hc.append(hook2)
|
|
hc.append(hook3)
|
|
hc.append(hook4)
|
|
hc.append(hook5)
|
|
|
|
finalHook := hc.hook()
|
|
if finalHook == nil {
|
|
t.Fatal("expected a non-nil hook")
|
|
}
|
|
|
|
packet := []byte("test")
|
|
method, packetOut, err := finalHook(packet)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
|
|
if method != ssh.PipePacketHookReply {
|
|
t.Errorf("expected method to be PipePacketHookReply, got %v", method)
|
|
}
|
|
|
|
expectedPacket := "test1234"
|
|
if string(packetOut) != expectedPacket {
|
|
t.Errorf("expected packetOut to be %q, got %q", expectedPacket, string(packetOut))
|
|
}
|
|
}
|
|
|
|
func TestHookChain_HookWithError(t *testing.T) {
|
|
hc := &hookChain{}
|
|
|
|
// Mock hooks
|
|
hook1 := func(packet []byte) (ssh.PipePacketHookMethod, []byte, error) {
|
|
return ssh.PipePacketHookTransform, append(packet, '1'), nil
|
|
}
|
|
|
|
hookWithError := func(packet []byte) (ssh.PipePacketHookMethod, []byte, error) {
|
|
return ssh.PipePacketHookTransform, nil, errors.New("mock error")
|
|
}
|
|
|
|
hc.append(hook1)
|
|
hc.append(hookWithError)
|
|
|
|
finalHook := hc.hook()
|
|
if finalHook == nil {
|
|
t.Fatal("expected a non-nil hook")
|
|
}
|
|
|
|
packet := []byte("test")
|
|
_, _, err := finalHook(packet)
|
|
if err == nil {
|
|
t.Fatal("expected an error, got nil")
|
|
}
|
|
|
|
expectedError := "mock error"
|
|
if err.Error() != expectedError {
|
|
t.Errorf("expected error %q, got %q", expectedError, err.Error())
|
|
}
|
|
}
|