derivepassphrase.git

1) # SPDX-FileCopyrightText: 2024 Marco Ricci <m@the13thletter.info>
2) #
3) # SPDX-License-Identifier: MIT
4) 
5) """Test OpenSSH key loading and signing."""
6) 

7) from __future__ import annotations
8) 
9) import click

10) import pytest
11) 
12) import derivepassphrase

13) import derivepassphrase.cli

14) import ssh_agent_client
15) 
16) import base64

17) import errno
18) import io

19) import os
20) import socket
21) import subprocess
22) 

23) if not os.environ.get('SSH_AUTH_SOCK'):  # pragma: no cover
24)     pytest.skip('no running SSH agent detected', allow_module_level=True)
25) 

26) SUPPORTED = {
27)     'ed25519': {
28)         'private_key': rb'''-----BEGIN OPENSSH PRIVATE KEY-----
29) b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW
30) QyNTUxOQAAACCBeIFoJtYCSF8P/zJIb+TBMIncHGpFBgnpCQ/7whJpdgAAAKDweO7H8Hju
31) xwAAAAtzc2gtZWQyNTUxOQAAACCBeIFoJtYCSF8P/zJIb+TBMIncHGpFBgnpCQ/7whJpdg
32) AAAEAbM/A869nkWZbe2tp3Dm/L6gitvmpH/aRZt8sBII3ExYF4gWgm1gJIXw//Mkhv5MEw
33) idwcakUGCekJD/vCEml2AAAAG3Rlc3Qga2V5IHdpdGhvdXQgcGFzc3BocmFzZQEC
34) -----END OPENSSH PRIVATE KEY-----
35) ''',
36)         'public_key': rb'''ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIIF4gWgm1gJIXw//Mkhv5MEwidwcakUGCekJD/vCEml2 test key without passphrase
37) ''',
38)         'public_key_data': bytes.fromhex('''
39)             00 00 00 0b 73 73 68 2d 65 64 32 35 35 31 39

40)             00 00 00 20
41)             81 78 81 68 26 d6 02 48 5f 0f ff 32 48 6f e4 c1
42)             30 89 dc 1c 6a 45 06 09 e9 09 0f fb c2 12 69 76

43) '''),
44)         'expected_signature': bytes.fromhex('''
45)             00 00 00 0b 73 73 68 2d 65 64 32 35 35 31 39

46)             00 00 00 40
47)             f0 98 19 80 6c 1a 97 d5 26 03 6e cc e3 65 8f 86
48)             66 07 13 19 13 09 21 33 33 f9 e4 36 53 1d af fd
49)             0d 08 1f ec f8 73 9b 8c 5f 55 39 16 7c 53 54 2c
50)             1e 52 bb 30 ed 7f 89 e2 2f 69 51 55 d8 9e a6 02

51)         '''),

52)         'derived_passphrase': rb'8JgZgGwal9UmA27M42WPhmYHExkTCSEzM/nkNlMdr/0NCB/s+HObjF9VORZ8U1QsHlK7MO1/ieIvaVFV2J6mAg==',

53)     },
54)     # Currently only supported by PuTTY (which is deficient in other
55)     # niceties of the SSH agent and the agent's client).
56)     'ed448': {
57)         'private_key': rb'''-----BEGIN OPENSSH PRIVATE KEY-----
58) b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAASgAAAAlz
59) c2gtZWQ0NDgAAAA54vZy009Wu8wExjvEb3hqtLz1GO/+d5vmGUbErWQ4AUO9mYLT
60) zHJHc2m4s+yWzP29Cc3EcxizLG8AAAAA8BdhfCcXYXwnAAAACXNzaC1lZDQ0OAAA
61) ADni9nLTT1a7zATGO8RveGq0vPUY7/53m+YZRsStZDgBQ72ZgtPMckdzabiz7JbM
62) /b0JzcRzGLMsbwAAAAByM7GIMRvWJB3YD6SIpAF2uudX4ozZe0X917wPwiBrs373
63) 9TM1n94Nib6hrxGNmCk2iBQDe2KALPgA4vZy009Wu8wExjvEb3hqtLz1GO/+d5vm
64) GUbErWQ4AUO9mYLTzHJHc2m4s+yWzP29Cc3EcxizLG8AAAAAG3Rlc3Qga2V5IHdp
65) dGhvdXQgcGFzc3BocmFzZQECAwQFBgcICQ==
66) -----END OPENSSH PRIVATE KEY-----
67) ''',
68)         'public_key': rb'''ssh-ed448 AAAACXNzaC1lZDQ0OAAAADni9nLTT1a7zATGO8RveGq0vPUY7/53m+YZRsStZDgBQ72ZgtPMckdzabiz7JbM/b0JzcRzGLMsbwA= test key without passphrase
69) ''',
70)         'public_key_data': bytes.fromhex('''
71)             00 00 00 09 73 73 68 2d 65 64 34 34 38

72)             00 00 00 39
73)             e2 f6 72 d3 4f 56 bb cc 04 c6 3b c4 6f 78 6a b4
74)             bc f5 18 ef fe 77 9b e6 19 46 c4 ad 64 38 01 43
75)             bd 99 82 d3 cc 72 47 73 69 b8 b3 ec 96 cc fd bd
76)             09 cd c4 73 18 b3 2c 6f 00

77)         '''),
78) 

79)         'expected_signature': bytes.fromhex('''
80)             00 00 00 09 73 73 68 2d 65 64 34 34 38
81)             00 00 00 72 06 86
82)             f4 64 a4 a6 ba d9 c3 22 c4 93 49 99 fc 11 de 67
83)             97 08 f2 d8 b7 3c 2c 13 e7 c5 1c 1e 92 a6 0e d8
84)             2f 6d 81 03 82 00 e3 72 e4 32 6d 72 d2 6d 32 84
85)             3f cc a9 1e 57 2c 00 9a b3 99 de 45 da ce 2e d1
86)             db e5 89 f3 35 be 24 58 90 c6 ca 04 f0 db 88 80
87)             db bd 77 7c 80 20 7f 3a 48 61 f6 1f ae a9 5e 53
88)             7b e0 9d 93 1e ea dc eb b5 cd 56 4c ea 8f 08 00
89)         '''),

90)         'derived_passphrase': rb'Bob0ZKSmutnDIsSTSZn8Ed5nlwjy2Lc8LBPnxRwekqYO2C9tgQOCAONy5DJtctJtMoQ/zKkeVywAmrOZ3kXazi7R2+WJ8zW+JFiQxsoE8NuIgNu9d3yAIH86SGH2H66pXlN74J2THurc67XNVkzqjwgA',

91)     },
92)     'rsa': {
93)         'private_key': rb'''-----BEGIN OPENSSH PRIVATE KEY-----
94) b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAABlwAAAAdzc2gtcn
95) NhAAAAAwEAAQAAAYEAsaHu6Xs4cVsuDSNJlMCqoPVgmDgEviI8TfXmHKqX3JkIqI3LsvV7
96) Ijf8WCdTveEq7CkuZhImtsR52AOEVAoU8mDXDNr+nJ5wUPzf1UIaRjDe0lcXW4SlF01hQs
97) G4wYDuqxshwelraB/L3e0zhD7fjYHF8IbFsqGlFHWEwOtlfhhfbxJsTGguLm4A8/gdEJD5
98) 2rkqDcZpIXCHtJbCzW9aQpWcs/PDw5ylwl/3dB7jfxyfrGz4O3QrzsqhWEsip97mOmwl6q
99) CHbq8V8x9zu89D/H+bG5ijqxhijbjcVUW3lZfw/97gy9J6rG31HNar5H8GycLTFwuCFepD
100) mTEpNgQLKoe8ePIEPq4WHhFUovBdwlrOByUKKqxreyvWt5gkpTARz+9Lt8OjBO3rpqK8sZ
101) VKH3sE3de2RJM3V9PJdmZSs2b8EFK3PsUGdlMPM9pn1uk4uIItKWBmooOynuD8Ll6aPwuW
102) AFn3l8nLLyWdrmmEYzHWXiRjQJxy1Bi5AbHMOWiPAAAFkDPkuBkz5LgZAAAAB3NzaC1yc2
103) EAAAGBALGh7ul7OHFbLg0jSZTAqqD1YJg4BL4iPE315hyql9yZCKiNy7L1eyI3/FgnU73h
104) KuwpLmYSJrbEedgDhFQKFPJg1wza/pyecFD839VCGkYw3tJXF1uEpRdNYULBuMGA7qsbIc
105) Hpa2gfy93tM4Q+342BxfCGxbKhpRR1hMDrZX4YX28SbExoLi5uAPP4HRCQ+dq5Kg3GaSFw
106) h7SWws1vWkKVnLPzw8OcpcJf93Qe438cn6xs+Dt0K87KoVhLIqfe5jpsJeqgh26vFfMfc7
107) vPQ/x/mxuYo6sYYo243FVFt5WX8P/e4MvSeqxt9RzWq+R/BsnC0xcLghXqQ5kxKTYECyqH
108) vHjyBD6uFh4RVKLwXcJazgclCiqsa3sr1reYJKUwEc/vS7fDowTt66aivLGVSh97BN3Xtk
109) STN1fTyXZmUrNm/BBStz7FBnZTDzPaZ9bpOLiCLSlgZqKDsp7g/C5emj8LlgBZ95fJyy8l
110) na5phGMx1l4kY0CcctQYuQGxzDlojwAAAAMBAAEAAAF/cNVYT+Om4x9+SItcz5bOByGIOj
111) yWUH8f9rRjnr5ILuwabIDgvFaVG+xM1O1hWADqzMnSEcknHRkTYEsqYPykAtxFvjOFEh70
112) 6qRUJ+fVZkqRGEaI3oWyWKTOhcCIYImtONvb0LOv/HQ2H2AXCoeqjST1qr/xSuljBtcB8u
113) wxs3EqaO1yU7QoZpDcMX9plH7Rmc9nNfZcgrnktPk2deX2+Y/A5tzdVgG1IeqYp6CBMLNM
114) uhL0OPdDehgBoDujx+rhkZ1gpo1wcULIM94NL7VSHBPX0Lgh9T+3j1HVP+YnMAvhfOvfct
115) LlbJ06+TYGRAMuF2LPCAZM/m0FEyAurRgWxAjLXm+4kp2GAJXlw82deDkQ+P8cHNT6s9ZH
116) R5YSy3lpZ35594ZMOLR8KqVvhgJGF6i9019BiF91SDxjE+sp6dNGfN8W+64tHdDv2a0Mso
117) +8Qjyx7sTpi++EjLU8Iy73/e4B8qbXMyheyA/UUfgMtNKShh6sLlrD9h2Sm9RFTuEAAADA
118) Jh3u7WfnjhhKZYbAW4TsPNXDMrB0/t7xyAQgFmko7JfESyrJSLg1cO+QMOiDgD7zuQ9RSp
119) NIKdPsnIna5peh979mVjb2HgnikjyJECmBpLdwZKhX7MnIvgKw5lnQXHboEtWCa1N58l7f
120) srzwbi9pFUuUp9dShXNffmlUCjDRsVLbK5C6+iaIQyCWFYK8mc6dpNkIoPKf+Xg+EJCIFQ
121) oITqeu30Gc1+M+fdZc2ghq0b6XLthh/uHEry8b68M5KglMAAAAwQDw1i+IdcvPV/3u/q9O
122) /kzLpKO3tbT89sc1zhjZsDNjDAGluNr6n38iq/XYRZu7UTL9BG+EgFVfIUV7XsYT5e+BPf
123) 13VS94rzZ7maCsOlULX+VdMO2zBucHIoec9RUlRZrfB21B2W7YGMhbpoa5lN3lKJQ7afHo
124) dXZUMp0cTFbOmbzJgSzO2/NE7BhVwmvcUzTDJGMMKuxBO6w99YKDKRKm0PNLFDz26rWm9L
125) dNS2MVfVuPMTpzT26HQG4pFageq9cAAADBALzRBXdZF8kbSBa5MTUBVTTzgKQm1C772gJ8
126) T01DJEXZsVtOv7mUC1/m/by6Hk4tPyvDBuGj9hHq4N7dPqGutHb1q5n0ADuoQjRW7BXw5Q
127) vC2EAD91xexdorIA5BgXU+qltBqzzBVzVtF7+jOZOjfzOlaTX9I5I5veyeTaTxZj1XXUzi
128) btBNdMEJJp7ifucYmoYAAwE7K+VlWagDEK2y8Mte9y9E+N0uO2j+h85sQt/UIb2iE/vhcg
129) Bgp6142WnSCQAAABt0ZXN0IGtleSB3aXRob3V0IHBhc3NwaHJhc2UB
130) -----END OPENSSH PRIVATE KEY-----
131) ''',
132)         'public_key': rb'''ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQCxoe7pezhxWy4NI0mUwKqg9WCYOAS+IjxN9eYcqpfcmQiojcuy9XsiN/xYJ1O94SrsKS5mEia2xHnYA4RUChTyYNcM2v6cnnBQ/N/VQhpGMN7SVxdbhKUXTWFCwbjBgO6rGyHB6WtoH8vd7TOEPt+NgcXwhsWyoaUUdYTA62V+GF9vEmxMaC4ubgDz+B0QkPnauSoNxmkhcIe0lsLNb1pClZyz88PDnKXCX/d0HuN/HJ+sbPg7dCvOyqFYSyKn3uY6bCXqoIdurxXzH3O7z0P8f5sbmKOrGGKNuNxVRbeVl/D/3uDL0nqsbfUc1qvkfwbJwtMXC4IV6kOZMSk2BAsqh7x48gQ+rhYeEVSi8F3CWs4HJQoqrGt7K9a3mCSlMBHP70u3w6ME7eumoryxlUofewTd17ZEkzdX08l2ZlKzZvwQUrc+xQZ2Uw8z2mfW6Ti4gi0pYGaig7Ke4PwuXpo/C5YAWfeXycsvJZ2uaYRjMdZeJGNAnHLUGLkBscw5aI8= test key without passphrase
133) ''',
134)         'public_key_data': bytes.fromhex('''
135)             00 00 00 07 73 73 68 2d 72 73 61
136)             00 00 00 03 01 00 01

137)             00 00 01 81 00
138)             b1 a1 ee e9 7b 38 71 5b 2e 0d 23 49 94 c0 aa a0
139)             f5 60 98 38 04 be 22 3c 4d f5 e6 1c aa 97 dc 99
140)             08 a8 8d cb b2 f5 7b 22 37 fc 58 27 53 bd e1 2a
141)             ec 29 2e 66 12 26 b6 c4 79 d8 03 84 54 0a 14 f2
142)             60 d7 0c da fe 9c 9e 70 50 fc df d5 42 1a 46 30
143)             de d2 57 17 5b 84 a5 17 4d 61 42 c1 b8 c1 80 ee
144)             ab 1b 21 c1 e9 6b 68 1f cb dd ed 33 84 3e df 8d
145)             81 c5 f0 86 c5 b2 a1 a5 14 75 84 c0 eb 65 7e 18
146)             5f 6f 12 6c 4c 68 2e 2e 6e 00 f3 f8 1d 10 90 f9
147)             da b9 2a 0d c6 69 21 70 87 b4 96 c2 cd 6f 5a 42
148)             95 9c b3 f3 c3 c3 9c a5 c2 5f f7 74 1e e3 7f 1c
149)             9f ac 6c f8 3b 74 2b ce ca a1 58 4b 22 a7 de e6
150)             3a 6c 25 ea a0 87 6e af 15 f3 1f 73 bb cf 43 fc
151)             7f 9b 1b 98 a3 ab 18 62 8d b8 dc 55 45 b7 95 97
152)             f0 ff de e0 cb d2 7a ac 6d f5 1c d6 ab e4 7f 06
153)             c9 c2 d3 17 0b 82 15 ea 43 99 31 29 36 04 0b 2a
154)             87 bc 78 f2 04 3e ae 16 1e 11 54 a2 f0 5d c2 5a
155)             ce 07 25 0a 2a ac 6b 7b 2b d6 b7 98 24 a5 30 11
156)             cf ef 4b b7 c3 a3 04 ed eb a6 a2 bc b1 95 4a 1f
157)             7b 04 dd d7 b6 44 93 37 57 d3 c9 76 66 52 b3 66
158)             fc 10 52 b7 3e c5 06 76 53 0f 33 da 67 d6 e9 38
159)             b8 82 2d 29 60 66 a2 83 b2 9e e0 fc 2e 5e 9a 3f
160)             0b 96 00 59 f7 97 c9 cb 2f 25 9d ae 69 84 63 31
161)             d6 5e 24 63 40 9c 72 d4 18 b9 01 b1 cc 39 68 8f

162) '''),
163)         'expected_signature': bytes.fromhex('''
164)             00 00 00 07 73 73 68 2d 72 73 61

165)             00 00 01 80
166)             a2 10 7c 2e f6 bb 53 a8 74 2a a1 19 99 ad 81 be
167)             79 9c ed d6 9d 09 4e 6e c5 18 48 33 90 77 99 68
168)             f7 9e 03 5a cd 4e 18 eb 89 7d 85 a2 ee ae 4a 92
169)             f6 6f ce b9 fe 86 7f 2a 6b 31 da 6e 1a fe a2 a5
170)             88 b8 44 7f a1 76 73 b3 ec 75 b5 d0 a6 b9 15 97
171)             65 09 13 7d 94 21 d1 fb 5d 0f 8b 23 04 77 c2 c3
172)             55 22 b1 a0 09 8a f5 38 2a d6 7f 1b 87 29 a0 25
173)             d3 25 6f cb 64 61 07 98 dc 14 c5 84 f8 92 24 5e
174)             50 11 6b 49 e5 f0 cc 29 cb 29 a9 19 d8 a7 71 1f
175)             91 0b 05 b1 01 4b c2 5f 00 a5 b6 21 bf f8 2c 9d
176)             67 9b 47 3b 0a 49 6b 79 2d fc 1d ec 0c b0 e5 27
177)             22 d5 a9 f8 d3 c3 f9 df 48 68 e9 fb ef 3c dc 26
178)             bf cf ea 29 43 01 a6 e3 c5 51 95 f4 66 6d 8a 55
179)             e2 47 ec e8 30 45 4c ae 47 e7 c9 a4 21 8b 64 ba
180)             b6 88 f6 21 f8 73 b9 cb 11 a1 78 75 92 c6 5a e5
181)             64 fe ed 42 d9 95 99 e6 2b 6f 3c 16 3c 28 74 a4
182)             72 2f 0d 3f 2c 33 67 aa 35 19 8e e7 b5 11 2f b3
183)             f7 6a c5 02 e2 6f a3 42 e3 62 19 99 03 ea a5 20
184)             e7 a1 e3 bc c8 06 a3 b5 7c d6 76 5d df 6f 60 46
185)             83 2a 08 00 d6 d3 d9 a4 c1 41 8c f8 60 56 45 81
186)             da 3b a2 16 1f 9e 4e 75 83 17 da c3 53 c3 3e 19
187)             a4 1b bc d2 29 b8 78 61 2b 78 e6 b1 52 b0 d5 ec
188)             de 69 2c 48 62 d9 fd d1 9b 6b b0 49 db d3 ff 38
189)             e7 10 d9 2d ce 9f 0d 5e 09 7b 37 d2 7b c3 bf ce

190) '''),

191)         'derived_passphrase': rb'ohB8Lva7U6h0KqEZma2Bvnmc7dadCU5uxRhIM5B3mWj3ngNazU4Y64l9haLurkqS9m/Ouf6GfyprMdpuGv6ipYi4RH+hdnOz7HW10Ka5FZdlCRN9lCHR+10PiyMEd8LDVSKxoAmK9Tgq1n8bhymgJdMlb8tkYQeY3BTFhPiSJF5QEWtJ5fDMKcspqRnYp3EfkQsFsQFLwl8ApbYhv/gsnWebRzsKSWt5Lfwd7Ayw5Sci1an408P530ho6fvvPNwmv8/qKUMBpuPFUZX0Zm2KVeJH7OgwRUyuR+fJpCGLZLq2iPYh+HO5yxGheHWSxlrlZP7tQtmVmeYrbzwWPCh0pHIvDT8sM2eqNRmO57URL7P3asUC4m+jQuNiGZkD6qUg56HjvMgGo7V81nZd329gRoMqCADW09mkwUGM+GBWRYHaO6IWH55OdYMX2sNTwz4ZpBu80im4eGEreOaxUrDV7N5pLEhi2f3Rm2uwSdvT/zjnENktzp8NXgl7N9J7w7/O',

192)     },
193) }
194) 
195) UNSUITABLE = {
196)     'dsa1024': {
197)         'private_key': rb'''-----BEGIN OPENSSH PRIVATE KEY-----
198) b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAABsQAAAAdzc2gtZH
199) NzAAAAgQC7KAZXqBGNVLBQPrcMYAoNW54BhD8aIhe7BDWYzJcsaMt72VKSkguZ8+XR7nRa
200) 0C/ZsBi+uJp0dpxy9ZMTOWX4u5YPMeQcXEdGExZIfimGqSOAsy6fCld2IfJZJZExcCmhe9
201) Ssjsd3YSAPJRluOXFQc95MZoR5hMwlIDD8QzrE7QAAABUA99nOZOgd7aHMVGoXpUEBcn7H
202) ossAAACALr2Ag3hxM3rKdxzVUw8fX0VVPXO+3+Kr8hGe0Kc/7NwVaBVL1GQ8fenBuWynpA
203) UbH0wo3h1wkB/8hX6p+S8cnu5rIBlUuVNwLw/bIYohK98LfqTYK/V+g6KD+8m34wvEiXZm
204) qywY54n2bksch1Nqvj/tNpLzExSx/XS0kSM1aigAAACAbQNRPcVEuGDrEcf+xg5tgAejPX
205) BPXr/Jss+Chk64km3mirMYjAWyWYtVcgT+7hOYxtYRin8LyMLqKRmqa0Q5UrvDfChgLhvs
206) G9YSb/Mpw5qm8PiHSafwhkaz/te3+8hKogqoe7sd+tCF06IpJr5k70ACiNtRGqssNF8Elr
207) l1efYAAAH4swlfVrMJX1YAAAAHc3NoLWRzcwAAAIEAuygGV6gRjVSwUD63DGAKDVueAYQ/
208) GiIXuwQ1mMyXLGjLe9lSkpILmfPl0e50WtAv2bAYvriadHaccvWTEzll+LuWDzHkHFxHRh
209) MWSH4phqkjgLMunwpXdiHyWSWRMXApoXvUrI7Hd2EgDyUZbjlxUHPeTGaEeYTMJSAw/EM6
210) xO0AAAAVAPfZzmToHe2hzFRqF6VBAXJ+x6LLAAAAgC69gIN4cTN6yncc1VMPH19FVT1zvt
211) /iq/IRntCnP+zcFWgVS9RkPH3pwblsp6QFGx9MKN4dcJAf/IV+qfkvHJ7uayAZVLlTcC8P
212) 2yGKISvfC36k2Cv1foOig/vJt+MLxIl2ZqssGOeJ9m5LHIdTar4/7TaS8xMUsf10tJEjNW
213) ooAAAAgG0DUT3FRLhg6xHH/sYObYAHoz1wT16/ybLPgoZOuJJt5oqzGIwFslmLVXIE/u4T
214) mMbWEYp/C8jC6ikZqmtEOVK7w3woYC4b7BvWEm/zKcOapvD4h0mn8IZGs/7Xt/vISqIKqH
215) u7HfrQhdOiKSa+ZO9AAojbURqrLDRfBJa5dXn2AAAAFQDJHfenj4EJ9WkehpdJatPBlqCW
216) 0gAAABt0ZXN0IGtleSB3aXRob3V0IHBhc3NwaHJhc2UBAgMEBQYH
217) -----END OPENSSH PRIVATE KEY-----
218) ''',
219)         'public_key': rb'''ssh-dss AAAAB3NzaC1kc3MAAACBALsoBleoEY1UsFA+twxgCg1bngGEPxoiF7sENZjMlyxoy3vZUpKSC5nz5dHudFrQL9mwGL64mnR2nHL1kxM5Zfi7lg8x5BxcR0YTFkh+KYapI4CzLp8KV3Yh8lklkTFwKaF71KyOx3dhIA8lGW45cVBz3kxmhHmEzCUgMPxDOsTtAAAAFQD32c5k6B3tocxUahelQQFyfseiywAAAIAuvYCDeHEzesp3HNVTDx9fRVU9c77f4qvyEZ7Qpz/s3BVoFUvUZDx96cG5bKekBRsfTCjeHXCQH/yFfqn5Lxye7msgGVS5U3AvD9shiiEr3wt+pNgr9X6DooP7ybfjC8SJdmarLBjnifZuSxyHU2q+P+02kvMTFLH9dLSRIzVqKAAAAIBtA1E9xUS4YOsRx/7GDm2AB6M9cE9ev8myz4KGTriSbeaKsxiMBbJZi1VyBP7uE5jG1hGKfwvIwuopGaprRDlSu8N8KGAuG+wb1hJv8ynDmqbw+IdJp/CGRrP+17f7yEqiCqh7ux360IXToikmvmTvQAKI21Eaqyw0XwSWuXV59g== test key without passphrase
220) ''',
221)         'public_key_data': bytes.fromhex('''
222)             00 00 00 07 73 73 68 2d 64 73 73
223)             00 00 00 81 00
224)             bb 28 06 57 a8 11 8d 54 b0 50 3e b7 0c 60 0a 0d
225)             5b 9e 01 84 3f 1a 22 17 bb 04 35 98 cc 97 2c 68
226)             cb 7b d9 52 92 92 0b 99 f3 e5 d1 ee 74 5a d0 2f
227)             d9 b0 18 be b8 9a 74 76 9c 72 f5 93 13 39 65 f8
228)             bb 96 0f 31 e4 1c 5c 47 46 13 16 48 7e 29 86 a9
229)             23 80 b3 2e 9f 0a 57 76 21 f2 59 25 91 31 70 29
230)             a1 7b d4 ac 8e c7 77 61 20 0f 25 19 6e 39 71 50
231)             73 de 4c 66 84 79 84 cc 25 20 30 fc 43 3a c4 ed

232)             00 00 00 15 00 f7 d9 ce 64
233)             e8 1d ed a1 cc 54 6a 17 a5 41 01 72 7e c7 a2 cb

234)             00 00 00 80
235)             2e bd 80 83 78 71 33 7a ca 77 1c d5 53 0f 1f 5f
236)             45 55 3d 73 be df e2 ab f2 11 9e d0 a7 3f ec dc
237)             15 68 15 4b d4 64 3c 7d e9 c1 b9 6c a7 a4 05 1b
238)             1f 4c 28 de 1d 70 90 1f fc 85 7e a9 f9 2f 1c 9e
239)             ee 6b 20 19 54 b9 53 70 2f 0f db 21 8a 21 2b df
240)             0b 7e a4 d8 2b f5 7e 83 a2 83 fb c9 b7 e3 0b c4
241)             89 76 66 ab 2c 18 e7 89 f6 6e 4b 1c 87 53 6a be
242)             3f ed 36 92 f3 13 14 b1 fd 74 b4 91 23 35 6a 28
243)             00 00 00 80
244)             6d 03 51 3d c5 44 b8 60 eb 11 c7 fe c6 0e 6d 80
245)             07 a3 3d 70 4f 5e bf c9 b2 cf 82 86 4e b8 92 6d
246)             e6 8a b3 18 8c 05 b2 59 8b 55 72 04 fe ee 13 98
247)             c6 d6 11 8a 7f 0b c8 c2 ea 29 19 aa 6b 44 39 52
248)             bb c3 7c 28 60 2e 1b ec 1b d6 12 6f f3 29 c3 9a
249)             a6 f0 f8 87 49 a7 f0 86 46 b3 fe d7 b7 fb c8 4a
250)             a2 0a a8 7b bb 1d fa d0 85 d3 a2 29 26 be 64 ef
251)             40 02 88 db 51 1a ab 2c 34 5f 04 96 b9 75 79 f6
252) '''),
253)         'expected_signature': None,

254)         'derived_passphrase': None,

255)     },
256)     'ecdsa256': {
257)         'private_key': rb'''-----BEGIN OPENSSH PRIVATE KEY-----
258) b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAaAAAABNlY2RzYS
259) 1zaGEyLW5pc3RwMjU2AAAACG5pc3RwMjU2AAAAQQTLbU0zDwsk2Dvp+VYIrsNVf5gWwz2S
260) 3SZ8TbxiQRkpnGSVqyIoHJOJc+NQItAa7xlJ/8Z6gfz57Z3apUkaMJm6AAAAuKeY+YinmP
261) mIAAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBMttTTMPCyTYO+n5
262) Vgiuw1V/mBbDPZLdJnxNvGJBGSmcZJWrIigck4lz41Ai0BrvGUn/xnqB/PntndqlSRowmb
263) oAAAAhAKIl/3n0pKVIxpZkXTGtii782Qr4yIcvHdpxjO/QsIqKAAAAG3Rlc3Qga2V5IHdp
264) dGhvdXQgcGFzc3BocmFzZQECAwQ=
265) -----END OPENSSH PRIVATE KEY-----
266) ''',
267)         'public_key': rb'''ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBMttTTMPCyTYO+n5Vgiuw1V/mBbDPZLdJnxNvGJBGSmcZJWrIigck4lz41Ai0BrvGUn/xnqB/PntndqlSRowmbo= test key without passphrase
268) ''',
269)         'public_key_data': bytes.fromhex('''
270)             00 00 00 13 65 63 64 73 61 2d 73 68 61 32 2d 6e
271)             69 73 74 70 32 35 36
272)             00 00 00 08 6e 69 73 74 70 32 35 36
273)             00 00 00 41 04
274)             cb 6d 4d 33 0f 0b 24 d8 3b e9 f9 56 08 ae c3 55
275)             7f 98 16 c3 3d 92 dd 26 7c 4d bc 62 41 19 29 9c
276)             64 95 ab 22 28 1c 93 89 73 e3 50 22 d0 1a ef 19
277)             49 ff c6 7a 81 fc f9 ed 9d da a5 49 1a 30 99 ba
278) '''),
279)         'expected_signature': None,

280)         'derived_passphrase': None,

281)     },
282)     'ecdsa384': {
283)         'private_key': rb'''-----BEGIN OPENSSH PRIVATE KEY-----
284) b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAiAAAABNlY2RzYS
285) 1zaGEyLW5pc3RwMzg0AAAACG5pc3RwMzg0AAAAYQSgkOjkAvq7v5vHuj3KBL4/EAWcn5hZ
286) DyKcbyV0eBMGFq7hKXQlZqIahLVqeMR0QqmkxNJ2rly2VHcXneq3vZ+9fIsWCOdYk5WP3N
287) ZPzv911Xn7wbEkC7QndD5zKlm4pBUAAADomhj+IZoY/iEAAAATZWNkc2Etc2hhMi1uaXN0
288) cDM4NAAAAAhuaXN0cDM4NAAAAGEEoJDo5AL6u7+bx7o9ygS+PxAFnJ+YWQ8inG8ldHgTBh
289) au4Sl0JWaiGoS1anjEdEKppMTSdq5ctlR3F53qt72fvXyLFgjnWJOVj9zWT87/ddV5+8Gx
290) JAu0J3Q+cypZuKQVAAAAMQD5sTy8p+B1cn/DhOmXquui1BcxvASqzzevkBlbQoBa73y04B
291) 2OdqVOVRkwZWRROz0AAAAbdGVzdCBrZXkgd2l0aG91dCBwYXNzcGhyYXNlAQIDBA==
292) -----END OPENSSH PRIVATE KEY-----
293) ''',
294)         'public_key': rb'''ecdsa-sha2-nistp384 AAAAE2VjZHNhLXNoYTItbmlzdHAzODQAAAAIbmlzdHAzODQAAABhBKCQ6OQC+ru/m8e6PcoEvj8QBZyfmFkPIpxvJXR4EwYWruEpdCVmohqEtWp4xHRCqaTE0nauXLZUdxed6re9n718ixYI51iTlY/c1k/O/3XVefvBsSQLtCd0PnMqWbikFQ== test key without passphrase
295) ''',
296)         'public_key_data': bytes.fromhex('''
297)             00 00 00 13
298)             65 63 64 73 61 2d 73 68 61 32 2d 6e 69 73 74 70
299)             33 38 34
300)             00 00 00 08 6e 69 73 74 70 33 38 34
301)             00 00 00 61 04
302)             a0 90 e8 e4 02 fa bb bf 9b c7 ba 3d ca 04 be 3f
303)             10 05 9c 9f 98 59 0f 22 9c 6f 25 74 78 13 06 16
304)             ae e1 29 74 25 66 a2 1a 84 b5 6a 78 c4 74 42 a9
305)             a4 c4 d2 76 ae 5c b6 54 77 17 9d ea b7 bd 9f bd
306)             7c 8b 16 08 e7 58 93 95 8f dc d6 4f ce ff 75 d5
307)             79 fb c1 b1 24 0b b4 27 74 3e 73 2a 59 b8 a4 15
308) '''),
309)         'expected_signature': None,

310)         'derived_passphrase': None,

311)     },
312) }
313) 
314) def test_client_uint32():
315)     uint32 = ssh_agent_client.SSHAgentClient.uint32
316)     assert uint32(16777216) == b'\x01\x00\x00\x00'
317) 
318) @pytest.mark.parametrize(['value', 'exc_type', 'exc_pattern'], [
319)     (10000000000000000, OverflowError, 'int too big to convert'),
320)     (-1, OverflowError, "can't convert negative int to unsigned"),
321) ])
322) def test_client_uint32_exceptions(value, exc_type, exc_pattern):
323)     uint32 = ssh_agent_client.SSHAgentClient.uint32
324)     with pytest.raises(exc_type, match=exc_pattern):
325)         uint32(value)
326) 
327) @pytest.mark.parametrize(['input', 'expected'], [
328)     (b'ssh-rsa', b'\x00\x00\x00\x07ssh-rsa'),
329)     (b'ssh-ed25519', b'\x00\x00\x00\x0bssh-ed25519'),
330)     (
331)         ssh_agent_client.SSHAgentClient.string(b'ssh-ed25519'),
332)         b'\x00\x00\x00\x0f\x00\x00\x00\x0bssh-ed25519',
333)     ),
334) ])
335) def test_client_string(input, expected):
336)     string = ssh_agent_client.SSHAgentClient.string
337)     assert bytes(string(input)) == expected
338) 

339) @pytest.mark.parametrize(['input', 'exc_type', 'exc_pattern'], [
340)     ('some string', TypeError, 'invalid payload type'),
341) ])
342) def test_client_string_exceptions(input, exc_type, exc_pattern):
343)     string = ssh_agent_client.SSHAgentClient.string
344)     with pytest.raises(exc_type, match=exc_pattern):
345)         string(input)
346) 

347) @pytest.mark.parametrize(['input', 'expected'], [
348)     (b'\x00\x00\x00\x07ssh-rsa', b'ssh-rsa'),
349)     (
350)         ssh_agent_client.SSHAgentClient.string(b'ssh-ed25519'),
351)         b'ssh-ed25519',
352)     ),
353) ])
354) def test_client_unstring(input, expected):
355)     unstring = ssh_agent_client.SSHAgentClient.unstring

356)     unstring_prefix = ssh_agent_client.SSHAgentClient.unstring_prefix

357)     assert bytes(unstring(input)) == expected

358)     assert tuple(bytes(x) for x in unstring_prefix(input)) == (expected, b'')

359) 

360) @pytest.mark.parametrize(
361)     ['input', 'exc_type', 'exc_pattern', 'has_trailer', 'parts'], [
362)         (b'ssh', ValueError, 'malformed SSH byte string', False, None),
363)         (
364)             b'\x00\x00\x00\x08ssh-rsa',
365)             ValueError, 'malformed SSH byte string',
366)             False, None,
367)         ),
368)         (
369)             b'\x00\x00\x00\x04XXX trailing text',
370)             ValueError, 'malformed SSH byte string',
371)             True, (b'XXX ', b'trailing text'),
372)         ),

373) ])

374) def test_client_unstring_exceptions(input, exc_type, exc_pattern,
375)                                     has_trailer, parts):

376)     unstring = ssh_agent_client.SSHAgentClient.unstring

377)     unstring_prefix = ssh_agent_client.SSHAgentClient.unstring_prefix

378)     with pytest.raises(exc_type, match=exc_pattern):
379)         unstring(input)

380)     if has_trailer:
381)         assert tuple(bytes(x) for x in unstring_prefix(input)) == parts
382)     else:
383)         with pytest.raises(exc_type, match=exc_pattern):
384)             unstring_prefix(input)

385) 
386) def test_key_decoding():
387)     public_key = SUPPORTED['ed25519']['public_key']
388)     public_key_data = SUPPORTED['ed25519']['public_key_data']
389)     keydata = base64.b64decode(public_key.split(None, 2)[1])
390)     assert (
391)         keydata == public_key_data
392)     ), "recorded public key data doesn't match"
393) 
394) @pytest.mark.parametrize(['keytype', 'data_dict'], list(SUPPORTED.items()))
395) def test_sign_data_via_agent(keytype, data_dict):
396)     private_key = data_dict['private_key']
397)     try:
398)         result = subprocess.run(['ssh-add', '-t', '30', '-q', '-'],
399)                                 input=private_key, check=True,
400)                                 capture_output=True)
401)     except subprocess.CalledProcessError as e:
402)         pytest.skip(
403)             f"uploading test key: {e!r}, stdout={e.stdout!r}, "
404)             f"stderr={e.stderr!r}"
405)         )
406)     else:
407)         try:
408)             client = ssh_agent_client.SSHAgentClient()

409)         except OSError:  # pragma: no cover

410)             pytest.skip('communication error with the SSH agent')
411)     with client:
412)         key_comment_pairs = {bytes(k): bytes(c)
413)                              for k, c in client.list_keys()}
414)         public_key_data = data_dict['public_key_data']
415)         expected_signature = data_dict['expected_signature']

416)         derived_passphrase = data_dict['derived_passphrase']

417)         if public_key_data not in key_comment_pairs:  # pragma: no cover

418)             pytest.skip('prerequisite SSH key not loaded')
419)         signature = bytes(client.sign(
420)             payload=derivepassphrase.Vault._UUID, key=public_key_data))
421)         assert signature == expected_signature, 'SSH signature mismatch'
422)         signature2 = bytes(client.sign(
423)             payload=derivepassphrase.Vault._UUID, key=public_key_data))
424)         assert signature2 == expected_signature, 'SSH signature mismatch'
425)         assert (

426)             derivepassphrase.Vault.phrase_from_key(public_key_data) ==
427)             derived_passphrase

428)         ), 'SSH signature mismatch'
429) 
430) @pytest.mark.parametrize(['keytype', 'data_dict'], list(UNSUITABLE.items()))
431) def test_sign_data_via_agent_unsupported(keytype, data_dict):
432)     private_key = data_dict['private_key']
433)     try:
434)         result = subprocess.run(['ssh-add', '-t', '30', '-q', '-'],
435)                                 input=private_key, check=True,
436)                                 capture_output=True)

437)     except subprocess.CalledProcessError as e:  # pragma: no cover
438)         pytest.skip(

439)             f"uploading test key: {e!r}, stdout={e.stdout!r}, "
440)             f"stderr={e.stderr!r}"
441)         )
442)     else:
443)         try:
444)             client = ssh_agent_client.SSHAgentClient()

445)         except OSError:  # pragma: no cover

446)             pytest.skip('communication error with the SSH agent')
447)     with client:
448)         key_comment_pairs = {bytes(k): bytes(c)
449)                              for k, c in client.list_keys()}
450)         public_key_data = data_dict['public_key_data']
451)         expected_signature = data_dict['expected_signature']

452)         if public_key_data not in key_comment_pairs:  # pragma: no cover

453)             pytest.skip('prerequisite SSH key not loaded')
454)         signature = bytes(client.sign(
455)             payload=derivepassphrase.Vault._UUID, key=public_key_data))
456)         signature2 = bytes(client.sign(
457)             payload=derivepassphrase.Vault._UUID, key=public_key_data))
458)         assert signature != signature2, 'SSH signature repeatable?!'
459)         with pytest.raises(ValueError, match='unsuitable SSH key'):