scrypt_test.go 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. // Copyright 2012 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 scrypt
  5. import (
  6. "bytes"
  7. "testing"
  8. )
  9. type testVector struct {
  10. password string
  11. salt string
  12. N, r, p int
  13. output []byte
  14. }
  15. var good = []testVector{
  16. {
  17. "password",
  18. "salt",
  19. 2, 10, 10,
  20. []byte{
  21. 0x48, 0x2c, 0x85, 0x8e, 0x22, 0x90, 0x55, 0xe6, 0x2f,
  22. 0x41, 0xe0, 0xec, 0x81, 0x9a, 0x5e, 0xe1, 0x8b, 0xdb,
  23. 0x87, 0x25, 0x1a, 0x53, 0x4f, 0x75, 0xac, 0xd9, 0x5a,
  24. 0xc5, 0xe5, 0xa, 0xa1, 0x5f,
  25. },
  26. },
  27. {
  28. "password",
  29. "salt",
  30. 16, 100, 100,
  31. []byte{
  32. 0x88, 0xbd, 0x5e, 0xdb, 0x52, 0xd1, 0xdd, 0x0, 0x18,
  33. 0x87, 0x72, 0xad, 0x36, 0x17, 0x12, 0x90, 0x22, 0x4e,
  34. 0x74, 0x82, 0x95, 0x25, 0xb1, 0x8d, 0x73, 0x23, 0xa5,
  35. 0x7f, 0x91, 0x96, 0x3c, 0x37,
  36. },
  37. },
  38. {
  39. "this is a long \000 password",
  40. "and this is a long \000 salt",
  41. 16384, 8, 1,
  42. []byte{
  43. 0xc3, 0xf1, 0x82, 0xee, 0x2d, 0xec, 0x84, 0x6e, 0x70,
  44. 0xa6, 0x94, 0x2f, 0xb5, 0x29, 0x98, 0x5a, 0x3a, 0x09,
  45. 0x76, 0x5e, 0xf0, 0x4c, 0x61, 0x29, 0x23, 0xb1, 0x7f,
  46. 0x18, 0x55, 0x5a, 0x37, 0x07, 0x6d, 0xeb, 0x2b, 0x98,
  47. 0x30, 0xd6, 0x9d, 0xe5, 0x49, 0x26, 0x51, 0xe4, 0x50,
  48. 0x6a, 0xe5, 0x77, 0x6d, 0x96, 0xd4, 0x0f, 0x67, 0xaa,
  49. 0xee, 0x37, 0xe1, 0x77, 0x7b, 0x8a, 0xd5, 0xc3, 0x11,
  50. 0x14, 0x32, 0xbb, 0x3b, 0x6f, 0x7e, 0x12, 0x64, 0x40,
  51. 0x18, 0x79, 0xe6, 0x41, 0xae,
  52. },
  53. },
  54. {
  55. "p",
  56. "s",
  57. 2, 1, 1,
  58. []byte{
  59. 0x48, 0xb0, 0xd2, 0xa8, 0xa3, 0x27, 0x26, 0x11, 0x98,
  60. 0x4c, 0x50, 0xeb, 0xd6, 0x30, 0xaf, 0x52,
  61. },
  62. },
  63. {
  64. "",
  65. "",
  66. 16, 1, 1,
  67. []byte{
  68. 0x77, 0xd6, 0x57, 0x62, 0x38, 0x65, 0x7b, 0x20, 0x3b,
  69. 0x19, 0xca, 0x42, 0xc1, 0x8a, 0x04, 0x97, 0xf1, 0x6b,
  70. 0x48, 0x44, 0xe3, 0x07, 0x4a, 0xe8, 0xdf, 0xdf, 0xfa,
  71. 0x3f, 0xed, 0xe2, 0x14, 0x42, 0xfc, 0xd0, 0x06, 0x9d,
  72. 0xed, 0x09, 0x48, 0xf8, 0x32, 0x6a, 0x75, 0x3a, 0x0f,
  73. 0xc8, 0x1f, 0x17, 0xe8, 0xd3, 0xe0, 0xfb, 0x2e, 0x0d,
  74. 0x36, 0x28, 0xcf, 0x35, 0xe2, 0x0c, 0x38, 0xd1, 0x89,
  75. 0x06,
  76. },
  77. },
  78. {
  79. "password",
  80. "NaCl",
  81. 1024, 8, 16,
  82. []byte{
  83. 0xfd, 0xba, 0xbe, 0x1c, 0x9d, 0x34, 0x72, 0x00, 0x78,
  84. 0x56, 0xe7, 0x19, 0x0d, 0x01, 0xe9, 0xfe, 0x7c, 0x6a,
  85. 0xd7, 0xcb, 0xc8, 0x23, 0x78, 0x30, 0xe7, 0x73, 0x76,
  86. 0x63, 0x4b, 0x37, 0x31, 0x62, 0x2e, 0xaf, 0x30, 0xd9,
  87. 0x2e, 0x22, 0xa3, 0x88, 0x6f, 0xf1, 0x09, 0x27, 0x9d,
  88. 0x98, 0x30, 0xda, 0xc7, 0x27, 0xaf, 0xb9, 0x4a, 0x83,
  89. 0xee, 0x6d, 0x83, 0x60, 0xcb, 0xdf, 0xa2, 0xcc, 0x06,
  90. 0x40,
  91. },
  92. },
  93. {
  94. "pleaseletmein", "SodiumChloride",
  95. 16384, 8, 1,
  96. []byte{
  97. 0x70, 0x23, 0xbd, 0xcb, 0x3a, 0xfd, 0x73, 0x48, 0x46,
  98. 0x1c, 0x06, 0xcd, 0x81, 0xfd, 0x38, 0xeb, 0xfd, 0xa8,
  99. 0xfb, 0xba, 0x90, 0x4f, 0x8e, 0x3e, 0xa9, 0xb5, 0x43,
  100. 0xf6, 0x54, 0x5d, 0xa1, 0xf2, 0xd5, 0x43, 0x29, 0x55,
  101. 0x61, 0x3f, 0x0f, 0xcf, 0x62, 0xd4, 0x97, 0x05, 0x24,
  102. 0x2a, 0x9a, 0xf9, 0xe6, 0x1e, 0x85, 0xdc, 0x0d, 0x65,
  103. 0x1e, 0x40, 0xdf, 0xcf, 0x01, 0x7b, 0x45, 0x57, 0x58,
  104. 0x87,
  105. },
  106. },
  107. /*
  108. // Disabled: needs 1 GiB RAM and takes too long for a simple test.
  109. {
  110. "pleaseletmein", "SodiumChloride",
  111. 1048576, 8, 1,
  112. []byte{
  113. 0x21, 0x01, 0xcb, 0x9b, 0x6a, 0x51, 0x1a, 0xae, 0xad,
  114. 0xdb, 0xbe, 0x09, 0xcf, 0x70, 0xf8, 0x81, 0xec, 0x56,
  115. 0x8d, 0x57, 0x4a, 0x2f, 0xfd, 0x4d, 0xab, 0xe5, 0xee,
  116. 0x98, 0x20, 0xad, 0xaa, 0x47, 0x8e, 0x56, 0xfd, 0x8f,
  117. 0x4b, 0xa5, 0xd0, 0x9f, 0xfa, 0x1c, 0x6d, 0x92, 0x7c,
  118. 0x40, 0xf4, 0xc3, 0x37, 0x30, 0x40, 0x49, 0xe8, 0xa9,
  119. 0x52, 0xfb, 0xcb, 0xf4, 0x5c, 0x6f, 0xa7, 0x7a, 0x41,
  120. 0xa4,
  121. },
  122. },
  123. */
  124. }
  125. var bad = []testVector{
  126. {"p", "s", 0, 1, 1, nil}, // N == 0
  127. {"p", "s", 1, 1, 1, nil}, // N == 1
  128. {"p", "s", 7, 8, 1, nil}, // N is not power of 2
  129. {"p", "s", 16, maxInt / 2, maxInt / 2, nil}, // p * r too large
  130. }
  131. func TestKey(t *testing.T) {
  132. for i, v := range good {
  133. k, err := Key([]byte(v.password), []byte(v.salt), v.N, v.r, v.p, len(v.output))
  134. if err != nil {
  135. t.Errorf("%d: got unexpected error: %s", i, err)
  136. }
  137. if !bytes.Equal(k, v.output) {
  138. t.Errorf("%d: expected %x, got %x", i, v.output, k)
  139. }
  140. }
  141. for i, v := range bad {
  142. _, err := Key([]byte(v.password), []byte(v.salt), v.N, v.r, v.p, 32)
  143. if err == nil {
  144. t.Errorf("%d: expected error, got nil", i)
  145. }
  146. }
  147. }
  148. var sink []byte
  149. func BenchmarkKey(b *testing.B) {
  150. for i := 0; i < b.N; i++ {
  151. sink, _ = Key([]byte("password"), []byte("salt"), 1<<15, 8, 1, 64)
  152. }
  153. }