1
0

sign_test.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533
  1. //go:build !windows
  2. package main
  3. import (
  4. "bytes"
  5. "crypto/rand"
  6. "errors"
  7. "os"
  8. "testing"
  9. "time"
  10. "github.com/slackhq/nebula/cert"
  11. "github.com/stretchr/testify/assert"
  12. "github.com/stretchr/testify/require"
  13. "golang.org/x/crypto/ed25519"
  14. )
  15. func Test_signSummary(t *testing.T) {
  16. assert.Equal(t, "sign <flags>: create and sign a certificate", signSummary())
  17. }
  18. func Test_signHelp(t *testing.T) {
  19. ob := &bytes.Buffer{}
  20. signHelp(ob)
  21. assert.Equal(
  22. t,
  23. "Usage of "+os.Args[0]+" sign <flags>: create and sign a certificate\n"+
  24. " Pass \"-\" to any path flag to read from stdin or write to stdout.\n"+
  25. " -ca-crt string\n"+
  26. " \tOptional: path to the signing CA cert (default \"ca.crt\")\n"+
  27. " -ca-key string\n"+
  28. " \tOptional: path to the signing CA key (default \"ca.key\")\n"+
  29. " -duration duration\n"+
  30. " \tOptional: how long the cert should be valid for. The default is 1 second before the signing cert expires. Valid time units are seconds: \"s\", minutes: \"m\", hours: \"h\"\n"+
  31. " -groups string\n"+
  32. " \tOptional: comma separated list of groups\n"+
  33. " -in-pub string\n"+
  34. " \tOptional (if out-key not set): path to read a previously generated public key\n"+
  35. " -ip string\n"+
  36. " \tDeprecated, see -networks\n"+
  37. " -name string\n"+
  38. " \tRequired: name of the cert, usually a hostname\n"+
  39. " -networks string\n"+
  40. " \tRequired: comma separated list of ip address and network in CIDR notation to assign to this cert\n"+
  41. " -out-crt string\n"+
  42. " \tOptional: path to write the certificate to\n"+
  43. " -out-key string\n"+
  44. " \tOptional (if in-pub not set): path to write the private key to\n"+
  45. " -out-qr string\n"+
  46. " \tOptional: output a qr code image (png) of the certificate\n"+
  47. optionalPkcs11String(" -pkcs11 string\n \tOptional: PKCS#11 URI to an existing private key\n")+
  48. " -subnets string\n"+
  49. " \tDeprecated, see -unsafe-networks\n"+
  50. " -unsafe-networks string\n"+
  51. " \tOptional: comma separated list of ip address and network in CIDR notation. Unsafe networks this cert can route for\n"+
  52. " -version uint\n"+
  53. " \tOptional: version of the certificate format to use. The default is to match the version of the signing CA\n",
  54. ob.String(),
  55. )
  56. }
  57. func Test_signCert(t *testing.T) {
  58. ob := &bytes.Buffer{}
  59. eb := &bytes.Buffer{}
  60. nopw := &StubPasswordReader{
  61. password: []byte(""),
  62. err: nil,
  63. }
  64. errpw := &StubPasswordReader{
  65. password: []byte(""),
  66. err: errors.New("stub error"),
  67. }
  68. passphrase := []byte("DO NOT USE THIS KEY")
  69. testpw := &StubPasswordReader{
  70. password: passphrase,
  71. err: nil,
  72. }
  73. // required args
  74. assertHelpError(t, signCert(
  75. []string{"-version", "1", "-ca-crt", "./nope", "-ca-key", "./nope", "-ip", "1.1.1.1/24", "-out-key", "nope", "-out-crt", "nope"}, ob, eb, nopw,
  76. ), "-name is required")
  77. assert.Empty(t, ob.String())
  78. assert.Empty(t, eb.String())
  79. assertHelpError(t, signCert(
  80. []string{"-version", "1", "-ca-crt", "./nope", "-ca-key", "./nope", "-name", "test", "-out-key", "nope", "-out-crt", "nope"}, ob, eb, nopw,
  81. ), "-networks is required")
  82. assert.Empty(t, ob.String())
  83. assert.Empty(t, eb.String())
  84. // cannot set -in-pub and -out-key
  85. assertHelpError(t, signCert(
  86. []string{"-version", "1", "-ca-crt", "./nope", "-ca-key", "./nope", "-name", "test", "-in-pub", "nope", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope"}, ob, eb, nopw,
  87. ), "cannot set both -in-pub and -out-key")
  88. assert.Empty(t, ob.String())
  89. assert.Empty(t, eb.String())
  90. // failed to read key
  91. ob.Reset()
  92. eb.Reset()
  93. args := []string{"-version", "1", "-ca-crt", "./nope", "-ca-key", "./nope", "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m"}
  94. require.EqualError(t, signCert(args, ob, eb, nopw), "error while reading ca-key: open ./nope: "+NoSuchFileError)
  95. // failed to unmarshal key
  96. ob.Reset()
  97. eb.Reset()
  98. caKeyF, err := os.CreateTemp("", "sign-cert.key")
  99. require.NoError(t, err)
  100. defer os.Remove(caKeyF.Name())
  101. args = []string{"-version", "1", "-ca-crt", "./nope", "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m"}
  102. require.EqualError(t, signCert(args, ob, eb, nopw), "error while parsing ca-key: input did not contain a valid PEM encoded block")
  103. assert.Empty(t, ob.String())
  104. assert.Empty(t, eb.String())
  105. // Write a proper ca key for later
  106. ob.Reset()
  107. eb.Reset()
  108. caPub, caPriv, _ := ed25519.GenerateKey(rand.Reader)
  109. caKeyF.Write(cert.MarshalSigningPrivateKeyToPEM(cert.Curve_CURVE25519, caPriv))
  110. // failed to read cert
  111. args = []string{"-version", "1", "-ca-crt", "./nope", "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m"}
  112. require.EqualError(t, signCert(args, ob, eb, nopw), "error while reading ca-crt: open ./nope: "+NoSuchFileError)
  113. assert.Empty(t, ob.String())
  114. assert.Empty(t, eb.String())
  115. // failed to unmarshal cert
  116. ob.Reset()
  117. eb.Reset()
  118. caCrtF, err := os.CreateTemp("", "sign-cert.crt")
  119. require.NoError(t, err)
  120. defer os.Remove(caCrtF.Name())
  121. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m"}
  122. require.EqualError(t, signCert(args, ob, eb, nopw), "error while parsing ca-crt: input did not contain a valid PEM encoded block")
  123. assert.Empty(t, ob.String())
  124. assert.Empty(t, eb.String())
  125. // write a proper ca cert for later
  126. ca, _ := NewTestCaCert("ca", caPub, caPriv, time.Now(), time.Now().Add(time.Minute*200), nil, nil, nil)
  127. b, _ := ca.MarshalPEM()
  128. caCrtF.Write(b)
  129. // failed to read pub
  130. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-in-pub", "./nope", "-duration", "100m"}
  131. require.EqualError(t, signCert(args, ob, eb, nopw), "error while reading in-pub: open ./nope: "+NoSuchFileError)
  132. assert.Empty(t, ob.String())
  133. assert.Empty(t, eb.String())
  134. // failed to unmarshal pub
  135. ob.Reset()
  136. eb.Reset()
  137. inPubF, err := os.CreateTemp("", "in.pub")
  138. require.NoError(t, err)
  139. defer os.Remove(inPubF.Name())
  140. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-in-pub", inPubF.Name(), "-duration", "100m"}
  141. require.EqualError(t, signCert(args, ob, eb, nopw), "error while parsing in-pub: input did not contain a valid PEM encoded block")
  142. assert.Empty(t, ob.String())
  143. assert.Empty(t, eb.String())
  144. // write a proper pub for later
  145. ob.Reset()
  146. eb.Reset()
  147. inPub, _ := x25519Keypair()
  148. inPubF.Write(cert.MarshalPublicKeyToPEM(cert.Curve_CURVE25519, inPub))
  149. // bad ip cidr
  150. ob.Reset()
  151. eb.Reset()
  152. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "a1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m"}
  153. assertHelpError(t, signCert(args, ob, eb, nopw), "invalid -networks definition: a1.1.1.1/24")
  154. assert.Empty(t, ob.String())
  155. assert.Empty(t, eb.String())
  156. ob.Reset()
  157. eb.Reset()
  158. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "100::100/100", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m"}
  159. assertHelpError(t, signCert(args, ob, eb, nopw), "invalid -networks definition: v1 certificates can only have a single ipv4 address")
  160. assert.Empty(t, ob.String())
  161. assert.Empty(t, eb.String())
  162. ob.Reset()
  163. eb.Reset()
  164. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24,1.1.1.2/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m"}
  165. assertHelpError(t, signCert(args, ob, eb, nopw), "invalid -networks definition: v1 certificates can only have a single ipv4 address")
  166. assert.Empty(t, ob.String())
  167. assert.Empty(t, eb.String())
  168. // bad subnet cidr
  169. ob.Reset()
  170. eb.Reset()
  171. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m", "-subnets", "a"}
  172. assertHelpError(t, signCert(args, ob, eb, nopw), "invalid -unsafe-networks definition: a")
  173. assert.Empty(t, ob.String())
  174. assert.Empty(t, eb.String())
  175. ob.Reset()
  176. eb.Reset()
  177. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m", "-subnets", "100::100/100"}
  178. assertHelpError(t, signCert(args, ob, eb, nopw), "invalid -unsafe-networks definition: v1 certificates can only contain ipv4 addresses")
  179. assert.Empty(t, ob.String())
  180. assert.Empty(t, eb.String())
  181. // mismatched ca key
  182. _, caPriv2, _ := ed25519.GenerateKey(rand.Reader)
  183. caKeyF2, err := os.CreateTemp("", "sign-cert-2.key")
  184. require.NoError(t, err)
  185. defer os.Remove(caKeyF2.Name())
  186. caKeyF2.Write(cert.MarshalSigningPrivateKeyToPEM(cert.Curve_CURVE25519, caPriv2))
  187. ob.Reset()
  188. eb.Reset()
  189. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF2.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m", "-subnets", "a"}
  190. require.EqualError(t, signCert(args, ob, eb, nopw), "refusing to sign, root certificate does not match private key")
  191. assert.Empty(t, ob.String())
  192. assert.Empty(t, eb.String())
  193. // failed key write
  194. ob.Reset()
  195. eb.Reset()
  196. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "/do/not/write/pleasecrt", "-out-key", "/do/not/write/pleasekey", "-duration", "100m", "-subnets", "10.1.1.1/32"}
  197. require.EqualError(t, signCert(args, ob, eb, nopw), "error while writing out-key: open /do/not/write/pleasekey: "+NoSuchDirError)
  198. assert.Empty(t, ob.String())
  199. assert.Empty(t, eb.String())
  200. // create temp key file
  201. keyF, err := os.CreateTemp("", "test.key")
  202. require.NoError(t, err)
  203. os.Remove(keyF.Name())
  204. // failed cert write
  205. ob.Reset()
  206. eb.Reset()
  207. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", "/do/not/write/pleasecrt", "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32"}
  208. require.EqualError(t, signCert(args, ob, eb, nopw), "error while writing out-crt: open /do/not/write/pleasecrt: "+NoSuchDirError)
  209. assert.Empty(t, ob.String())
  210. assert.Empty(t, eb.String())
  211. os.Remove(keyF.Name())
  212. // create temp cert file
  213. crtF, err := os.CreateTemp("", "test.crt")
  214. require.NoError(t, err)
  215. os.Remove(crtF.Name())
  216. // test proper cert with removed empty groups and subnets
  217. ob.Reset()
  218. eb.Reset()
  219. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  220. require.NoError(t, signCert(args, ob, eb, nopw))
  221. assert.Empty(t, ob.String())
  222. assert.Empty(t, eb.String())
  223. // read cert and key files
  224. rb, _ := os.ReadFile(keyF.Name())
  225. lKey, b, curve, err := cert.UnmarshalPrivateKeyFromPEM(rb)
  226. assert.Equal(t, cert.Curve_CURVE25519, curve)
  227. assert.Empty(t, b)
  228. require.NoError(t, err)
  229. assert.Len(t, lKey, 32)
  230. rb, _ = os.ReadFile(crtF.Name())
  231. lCrt, b, err := cert.UnmarshalCertificateFromPEM(rb)
  232. assert.Empty(t, b)
  233. require.NoError(t, err)
  234. assert.Equal(t, "test", lCrt.Name())
  235. assert.Equal(t, "1.1.1.1/24", lCrt.Networks()[0].String())
  236. assert.Len(t, lCrt.Networks(), 1)
  237. assert.False(t, lCrt.IsCA())
  238. assert.Equal(t, []string{"1", "2", "3", "4", "5"}, lCrt.Groups())
  239. assert.Len(t, lCrt.UnsafeNetworks(), 3)
  240. assert.Len(t, lCrt.PublicKey(), 32)
  241. assert.Equal(t, time.Duration(time.Minute*100), lCrt.NotAfter().Sub(lCrt.NotBefore()))
  242. sns := []string{}
  243. for _, sn := range lCrt.UnsafeNetworks() {
  244. sns = append(sns, sn.String())
  245. }
  246. assert.Equal(t, []string{"10.1.1.1/32", "10.2.2.2/32", "10.5.5.5/32"}, sns)
  247. issuer, _ := ca.Fingerprint()
  248. assert.Equal(t, issuer, lCrt.Issuer())
  249. assert.True(t, lCrt.CheckSignature(caPub))
  250. // test proper cert with in-pub
  251. os.Remove(keyF.Name())
  252. os.Remove(crtF.Name())
  253. ob.Reset()
  254. eb.Reset()
  255. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-in-pub", inPubF.Name(), "-duration", "100m", "-groups", "1"}
  256. require.NoError(t, signCert(args, ob, eb, nopw))
  257. assert.Empty(t, ob.String())
  258. assert.Empty(t, eb.String())
  259. // read cert file and check pub key matches in-pub
  260. rb, _ = os.ReadFile(crtF.Name())
  261. lCrt, b, err = cert.UnmarshalCertificateFromPEM(rb)
  262. assert.Empty(t, b)
  263. require.NoError(t, err)
  264. assert.Equal(t, lCrt.PublicKey(), inPub)
  265. // test refuse to sign cert with duration beyond root
  266. ob.Reset()
  267. eb.Reset()
  268. os.Remove(keyF.Name())
  269. os.Remove(crtF.Name())
  270. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "1000m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  271. require.EqualError(t, signCert(args, ob, eb, nopw), "error while signing: certificate expires after signing certificate")
  272. assert.Empty(t, ob.String())
  273. assert.Empty(t, eb.String())
  274. // create valid cert/key for overwrite tests
  275. os.Remove(keyF.Name())
  276. os.Remove(crtF.Name())
  277. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  278. require.NoError(t, signCert(args, ob, eb, nopw))
  279. // test that we won't overwrite existing key file
  280. os.Remove(crtF.Name())
  281. ob.Reset()
  282. eb.Reset()
  283. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  284. require.EqualError(t, signCert(args, ob, eb, nopw), "refusing to overwrite existing key: "+keyF.Name())
  285. assert.Empty(t, ob.String())
  286. assert.Empty(t, eb.String())
  287. // create valid cert/key for overwrite tests
  288. os.Remove(keyF.Name())
  289. os.Remove(crtF.Name())
  290. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  291. require.NoError(t, signCert(args, ob, eb, nopw))
  292. // test that we won't overwrite existing certificate file
  293. os.Remove(keyF.Name())
  294. ob.Reset()
  295. eb.Reset()
  296. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  297. require.EqualError(t, signCert(args, ob, eb, nopw), "refusing to overwrite existing cert: "+crtF.Name())
  298. assert.Empty(t, ob.String())
  299. assert.Empty(t, eb.String())
  300. // create valid cert/key using encrypted CA key
  301. os.Remove(caKeyF.Name())
  302. os.Remove(caCrtF.Name())
  303. os.Remove(keyF.Name())
  304. os.Remove(crtF.Name())
  305. ob.Reset()
  306. eb.Reset()
  307. caKeyF, err = os.CreateTemp("", "sign-cert.key")
  308. require.NoError(t, err)
  309. defer os.Remove(caKeyF.Name())
  310. caCrtF, err = os.CreateTemp("", "sign-cert.crt")
  311. require.NoError(t, err)
  312. defer os.Remove(caCrtF.Name())
  313. // generate the encrypted key
  314. caPub, caPriv, _ = ed25519.GenerateKey(rand.Reader)
  315. kdfParams := cert.NewArgon2Parameters(64*1024, 4, 3)
  316. b, _ = cert.EncryptAndMarshalSigningPrivateKey(cert.Curve_CURVE25519, caPriv, passphrase, kdfParams)
  317. caKeyF.Write(b)
  318. ca, _ = NewTestCaCert("ca", caPub, caPriv, time.Now(), time.Now().Add(time.Minute*200), nil, nil, nil)
  319. b, _ = ca.MarshalPEM()
  320. caCrtF.Write(b)
  321. // test with the proper password
  322. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  323. require.NoError(t, signCert(args, ob, eb, testpw))
  324. assert.Empty(t, ob.String())
  325. assert.Equal(t, "Enter passphrase: ", eb.String())
  326. // test with the proper password in the environment
  327. os.Remove(crtF.Name())
  328. os.Remove(keyF.Name())
  329. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  330. os.Setenv("NEBULA_CA_PASSPHRASE", string(passphrase))
  331. ob.Reset()
  332. eb.Reset()
  333. require.NoError(t, signCert(args, ob, eb, testpw))
  334. assert.Empty(t, ob.String())
  335. assert.Empty(t, eb.String())
  336. os.Setenv("NEBULA_CA_PASSPHRASE", "")
  337. // test with the wrong password
  338. ob.Reset()
  339. eb.Reset()
  340. testpw.password = []byte("invalid password")
  341. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  342. require.Error(t, signCert(args, ob, eb, testpw))
  343. assert.Empty(t, ob.String())
  344. assert.Equal(t, "Enter passphrase: ", eb.String())
  345. // test with the wrong password in environment
  346. ob.Reset()
  347. eb.Reset()
  348. os.Setenv("NEBULA_CA_PASSPHRASE", "invalid password")
  349. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  350. require.EqualError(t, signCert(args, ob, eb, nopw), "error while parsing encrypted ca-key: invalid passphrase or corrupt private key")
  351. assert.Empty(t, ob.String())
  352. assert.Empty(t, eb.String())
  353. os.Setenv("NEBULA_CA_PASSPHRASE", "")
  354. // test with the user not entering a password
  355. ob.Reset()
  356. eb.Reset()
  357. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  358. require.Error(t, signCert(args, ob, eb, nopw))
  359. // normally the user hitting enter on the prompt would add newlines between these
  360. assert.Empty(t, ob.String())
  361. assert.Equal(t, "Enter passphrase: Enter passphrase: Enter passphrase: Enter passphrase: Enter passphrase: ", eb.String())
  362. // test an error condition
  363. ob.Reset()
  364. eb.Reset()
  365. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", keyF.Name(), "-duration", "100m", "-subnets", "10.1.1.1/32, , 10.2.2.2/32 , , ,, 10.5.5.5/32", "-groups", "1,, 2 , ,,,3,4,5"}
  366. require.Error(t, signCert(args, ob, eb, errpw))
  367. assert.Empty(t, ob.String())
  368. assert.Equal(t, "Enter passphrase: ", eb.String())
  369. }
  370. func Test_signCert_stdio(t *testing.T) {
  371. nopw := &StubPasswordReader{
  372. password: []byte(""),
  373. err: nil,
  374. }
  375. caPub, caPriv, _ := ed25519.GenerateKey(rand.Reader)
  376. rawCAKey := cert.MarshalSigningPrivateKeyToPEM(cert.Curve_CURVE25519, caPriv)
  377. ca, _ := NewTestCaCert("ca", caPub, caPriv, time.Now(), time.Now().Add(time.Minute*200), nil, nil, nil)
  378. rawCACrt, _ := ca.MarshalPEM()
  379. caCrtF, err := os.CreateTemp("", "sign-cert.crt")
  380. require.NoError(t, err)
  381. defer os.Remove(caCrtF.Name())
  382. caCrtF.Write(rawCACrt)
  383. caKeyF, err := os.CreateTemp("", "sign-cert.key")
  384. require.NoError(t, err)
  385. defer os.Remove(caKeyF.Name())
  386. caKeyF.Write(rawCAKey)
  387. keyF, err := os.CreateTemp("", "sign.key")
  388. require.NoError(t, err)
  389. os.Remove(keyF.Name())
  390. defer os.Remove(keyF.Name())
  391. // ca-key on stdin, cert to stdout
  392. withStdin(t, bytes.NewReader(rawCAKey))
  393. ob := &bytes.Buffer{}
  394. eb := &bytes.Buffer{}
  395. args := []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", "-", "-name", "stdin-test", "-ip", "1.1.1.1/24", "-out-crt", "-", "-out-key", keyF.Name(), "-duration", "100m"}
  396. require.NoError(t, signCert(args, ob, eb, nopw))
  397. assert.Empty(t, eb.String())
  398. lCrt, _, err := cert.UnmarshalCertificateFromPEM(ob.Bytes())
  399. require.NoError(t, err)
  400. assert.Equal(t, "stdin-test", lCrt.Name())
  401. assert.True(t, lCrt.CheckSignature(caPub))
  402. // two flags reading from stdin should error before any read attempt;
  403. // otherwise an interactive shell would hang on io.ReadAll
  404. stdinIn := bytes.NewReader(rawCAKey)
  405. withStdin(t, stdinIn)
  406. ob.Reset()
  407. eb.Reset()
  408. args = []string{"-version", "1", "-ca-crt", "-", "-ca-key", "-", "-name", "stdin-test", "-ip", "1.1.1.1/24", "-out-crt", "nope", "-out-key", "nope", "-duration", "100m"}
  409. require.EqualError(t, signCert(args, ob, eb, nopw),
  410. `-ca-key and -ca-crt both set to "-", only one input may read from stdin`)
  411. assert.Equal(t, len(rawCAKey), stdinIn.Len(), "stdin should be untouched when conflict is caught up front")
  412. // two flags writing to stdout should error before any output is written
  413. // AND before stdin is consumed
  414. stdinR := bytes.NewReader(rawCAKey)
  415. withStdin(t, stdinR)
  416. ob.Reset()
  417. eb.Reset()
  418. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", "-", "-name", "stdin-test", "-ip", "1.1.1.1/24", "-out-crt", "-", "-out-key", "-", "-duration", "100m"}
  419. require.EqualError(t, signCert(args, ob, eb, nopw),
  420. `-out-key and -out-crt both set to "-", only one output may write to stdout`)
  421. assert.Empty(t, ob.String())
  422. // stdin should be untouched because the conflict was caught up front
  423. assert.Equal(t, len(rawCAKey), stdinR.Len())
  424. // out-key on stdout, cert on disk
  425. keyF2, err := os.CreateTemp("", "sign.key")
  426. require.NoError(t, err)
  427. os.Remove(keyF2.Name())
  428. defer os.Remove(keyF2.Name())
  429. crtF, err := os.CreateTemp("", "sign.crt")
  430. require.NoError(t, err)
  431. os.Remove(crtF.Name())
  432. defer os.Remove(crtF.Name())
  433. ob.Reset()
  434. eb.Reset()
  435. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "stdin-test", "-ip", "1.1.1.1/24", "-out-crt", crtF.Name(), "-out-key", "-", "-duration", "100m"}
  436. require.NoError(t, signCert(args, ob, eb, nopw))
  437. assert.Empty(t, eb.String())
  438. _, _, curve, err := cert.UnmarshalPrivateKeyFromPEM(ob.Bytes())
  439. require.NoError(t, err)
  440. assert.Equal(t, cert.Curve_CURVE25519, curve)
  441. // in-pub on stdin (caller already has a keypair, only the cert is generated)
  442. inPub, _ := x25519Keypair()
  443. rawInPub := cert.MarshalPublicKeyToPEM(cert.Curve_CURVE25519, inPub)
  444. withStdin(t, bytes.NewReader(rawInPub))
  445. os.Remove(crtF.Name())
  446. ob.Reset()
  447. eb.Reset()
  448. args = []string{"-version", "1", "-ca-crt", caCrtF.Name(), "-ca-key", caKeyF.Name(), "-name", "in-pub-test", "-ip", "1.1.1.1/24", "-in-pub", "-", "-out-crt", "-", "-duration", "100m"}
  449. require.NoError(t, signCert(args, ob, eb, nopw))
  450. assert.Empty(t, eb.String())
  451. stdinCrt, _, err := cert.UnmarshalCertificateFromPEM(ob.Bytes())
  452. require.NoError(t, err)
  453. assert.Equal(t, "in-pub-test", stdinCrt.Name())
  454. assert.Equal(t, inPub, stdinCrt.PublicKey())
  455. }