symmetric_key_encrypted_test.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. // Copyright 2011 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package packet
  5. import (
  6. "bytes"
  7. "encoding/hex"
  8. "io"
  9. "io/ioutil"
  10. "testing"
  11. )
  12. func TestSymmetricKeyEncrypted(t *testing.T) {
  13. buf := readerFromHex(symmetricallyEncryptedHex)
  14. packet, err := Read(buf)
  15. if err != nil {
  16. t.Errorf("failed to read SymmetricKeyEncrypted: %s", err)
  17. return
  18. }
  19. ske, ok := packet.(*SymmetricKeyEncrypted)
  20. if !ok {
  21. t.Error("didn't find SymmetricKeyEncrypted packet")
  22. return
  23. }
  24. err = ske.Decrypt([]byte("password"))
  25. if err != nil {
  26. t.Error(err)
  27. return
  28. }
  29. packet, err = Read(buf)
  30. if err != nil {
  31. t.Errorf("failed to read SymmetricallyEncrypted: %s", err)
  32. return
  33. }
  34. se, ok := packet.(*SymmetricallyEncrypted)
  35. if !ok {
  36. t.Error("didn't find SymmetricallyEncrypted packet")
  37. return
  38. }
  39. r, err := se.Decrypt(ske.CipherFunc, ske.Key)
  40. if err != nil {
  41. t.Error(err)
  42. return
  43. }
  44. contents, err := ioutil.ReadAll(r)
  45. if err != nil && err != io.EOF {
  46. t.Error(err)
  47. return
  48. }
  49. expectedContents, _ := hex.DecodeString(symmetricallyEncryptedContentsHex)
  50. if !bytes.Equal(expectedContents, contents) {
  51. t.Errorf("bad contents got:%x want:%x", contents, expectedContents)
  52. }
  53. }
  54. const symmetricallyEncryptedHex = "8c0d04030302371a0b38d884f02060c91cf97c9973b8e58e028e9501708ccfe618fb92afef7fa2d80ddadd93cf"
  55. const symmetricallyEncryptedContentsHex = "cb1062004d14c4df636f6e74656e74732e0a"
  56. func TestSerializeSymmetricKeyEncrypted(t *testing.T) {
  57. buf := bytes.NewBuffer(nil)
  58. passphrase := []byte("testing")
  59. config := &Config{
  60. DefaultCipher: CipherAES128,
  61. }
  62. key, err := SerializeSymmetricKeyEncrypted(buf, passphrase, config)
  63. if err != nil {
  64. t.Errorf("failed to serialize: %s", err)
  65. return
  66. }
  67. p, err := Read(buf)
  68. if err != nil {
  69. t.Errorf("failed to reparse: %s", err)
  70. return
  71. }
  72. ske, ok := p.(*SymmetricKeyEncrypted)
  73. if !ok {
  74. t.Errorf("parsed a different packet type: %#v", p)
  75. return
  76. }
  77. if !ske.Encrypted {
  78. t.Errorf("SKE not encrypted but should be")
  79. }
  80. if ske.CipherFunc != config.DefaultCipher {
  81. t.Errorf("SKE cipher function is %d (expected %d)", ske.CipherFunc, config.DefaultCipher)
  82. }
  83. err = ske.Decrypt(passphrase)
  84. if err != nil {
  85. t.Errorf("failed to decrypt reparsed SKE: %s", err)
  86. return
  87. }
  88. if !bytes.Equal(key, ske.Key) {
  89. t.Errorf("keys don't match after Decrpyt: %x (original) vs %x (parsed)", key, ske.Key)
  90. }
  91. }