diff --git a/SecretAgentKit/Agent.swift b/SecretAgentKit/Agent.swift index 3d48fbd..f10f882 100644 --- a/SecretAgentKit/Agent.swift +++ b/SecretAgentKit/Agent.swift @@ -27,7 +27,7 @@ extension Agent { guard !data.isEmpty else { return } let requestTypeInt = data[4] guard let requestType = SSHAgent.RequestType(rawValue: requestTypeInt) else { - writer.write(SSHAgent.ResponseType.agentFailure.data) + writer.write(OpenSSHKeyWriter().lengthAndData(of: SSHAgent.ResponseType.agentFailure.data)) os_log(.debug, "Agent returned %@", SSHAgent.ResponseType.agentFailure.debugDescription) return } diff --git a/SecretAgentKitTests/AgentTests.swift b/SecretAgentKitTests/AgentTests.swift index c570018..cc5ecbd 100644 --- a/SecretAgentKitTests/AgentTests.swift +++ b/SecretAgentKitTests/AgentTests.swift @@ -41,6 +41,7 @@ class AgentTests: XCTestCase { agent.handle(reader: stubReader, writer: stubWriter) let reader = OpenSSHReader(data: stubWriter.data) // TODO: VERIFY + XCTAssertFalse(true) print(stubWriter.data.base64EncodedString()) } @@ -87,7 +88,7 @@ class AgentTests: XCTestCase { // MARK: Exception Handling func testSignatureException() { - let stubReader = StubFileHandleReader(availableData: Constants.Requests.requestIdentities) + let stubReader = StubFileHandleReader(availableData: Constants.Requests.requestSignature) let list = storeList(with: [Constants.Secrets.ecdsa256Secret, Constants.Secrets.ecdsa384Secret]) let store = list.stores.first?.base as! Stub.Store store.shouldThrow = true