whisper_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851
  1. // Copyright 2016 The go-ethereum Authors
  2. // This file is part of the go-ethereum library.
  3. //
  4. // The go-ethereum library is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Lesser General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // The go-ethereum library is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Lesser General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Lesser General Public License
  15. // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
  16. package whisperv5
  17. import (
  18. "bytes"
  19. "crypto/ecdsa"
  20. mrand "math/rand"
  21. "testing"
  22. "time"
  23. "github.com/ethereum/go-ethereum/common"
  24. )
  25. func TestWhisperBasic(t *testing.T) {
  26. w := New(&DefaultConfig)
  27. p := w.Protocols()
  28. shh := p[0]
  29. if shh.Name != ProtocolName {
  30. t.Fatalf("failed Protocol Name: %v.", shh.Name)
  31. }
  32. if uint64(shh.Version) != ProtocolVersion {
  33. t.Fatalf("failed Protocol Version: %v.", shh.Version)
  34. }
  35. if shh.Length != NumberOfMessageCodes {
  36. t.Fatalf("failed Protocol Length: %v.", shh.Length)
  37. }
  38. if shh.Run == nil {
  39. t.Fatalf("failed shh.Run.")
  40. }
  41. if uint64(w.Version()) != ProtocolVersion {
  42. t.Fatalf("failed whisper Version: %v.", shh.Version)
  43. }
  44. if w.GetFilter("non-existent") != nil {
  45. t.Fatalf("failed GetFilter.")
  46. }
  47. peerID := make([]byte, 64)
  48. mrand.Read(peerID)
  49. peer, _ := w.getPeer(peerID)
  50. if peer != nil {
  51. t.Fatal("found peer for random key.")
  52. }
  53. if err := w.AllowP2PMessagesFromPeer(peerID); err == nil {
  54. t.Fatalf("failed MarkPeerTrusted.")
  55. }
  56. exist := w.HasSymKey("non-existing")
  57. if exist {
  58. t.Fatalf("failed HasSymKey.")
  59. }
  60. key, err := w.GetSymKey("non-existing")
  61. if err == nil {
  62. t.Fatalf("failed GetSymKey(non-existing): false positive.")
  63. }
  64. if key != nil {
  65. t.Fatalf("failed GetSymKey: false positive.")
  66. }
  67. mail := w.Envelopes()
  68. if len(mail) != 0 {
  69. t.Fatalf("failed w.Envelopes().")
  70. }
  71. m := w.Messages("non-existent")
  72. if len(m) != 0 {
  73. t.Fatalf("failed w.Messages.")
  74. }
  75. var derived []byte
  76. ver := uint64(0xDEADBEEF)
  77. if _, err := deriveKeyMaterial(peerID, ver); err != unknownVersionError(ver) {
  78. t.Fatalf("failed deriveKeyMaterial with param = %v: %s.", peerID, err)
  79. }
  80. derived, err = deriveKeyMaterial(peerID, 0)
  81. if err != nil {
  82. t.Fatalf("failed second deriveKeyMaterial with param = %v: %s.", peerID, err)
  83. }
  84. if !validateSymmetricKey(derived) {
  85. t.Fatalf("failed validateSymmetricKey with param = %v.", derived)
  86. }
  87. if containsOnlyZeros(derived) {
  88. t.Fatalf("failed containsOnlyZeros with param = %v.", derived)
  89. }
  90. buf := []byte{0xFF, 0xE5, 0x80, 0x2, 0}
  91. le := bytesToUintLittleEndian(buf)
  92. be := BytesToUintBigEndian(buf)
  93. if le != uint64(0x280e5ff) {
  94. t.Fatalf("failed bytesToIntLittleEndian: %d.", le)
  95. }
  96. if be != uint64(0xffe5800200) {
  97. t.Fatalf("failed BytesToIntBigEndian: %d.", be)
  98. }
  99. id, err := w.NewKeyPair()
  100. if err != nil {
  101. t.Fatalf("failed to generate new key pair: %s.", err)
  102. }
  103. pk, err := w.GetPrivateKey(id)
  104. if err != nil {
  105. t.Fatalf("failed to retrieve new key pair: %s.", err)
  106. }
  107. if !validatePrivateKey(pk) {
  108. t.Fatalf("failed validatePrivateKey: %v.", pk)
  109. }
  110. if !ValidatePublicKey(&pk.PublicKey) {
  111. t.Fatalf("failed ValidatePublicKey: %v.", pk)
  112. }
  113. }
  114. func TestWhisperAsymmetricKeyImport(t *testing.T) {
  115. var (
  116. w = New(&DefaultConfig)
  117. privateKeys []*ecdsa.PrivateKey
  118. )
  119. for i := 0; i < 50; i++ {
  120. id, err := w.NewKeyPair()
  121. if err != nil {
  122. t.Fatalf("could not generate key: %v", err)
  123. }
  124. pk, err := w.GetPrivateKey(id)
  125. if err != nil {
  126. t.Fatalf("could not export private key: %v", err)
  127. }
  128. privateKeys = append(privateKeys, pk)
  129. if !w.DeleteKeyPair(id) {
  130. t.Fatalf("could not delete private key")
  131. }
  132. }
  133. for _, pk := range privateKeys {
  134. if _, err := w.AddKeyPair(pk); err != nil {
  135. t.Fatalf("could not import private key: %v", err)
  136. }
  137. }
  138. }
  139. func TestWhisperIdentityManagement(t *testing.T) {
  140. w := New(&DefaultConfig)
  141. id1, err := w.NewKeyPair()
  142. if err != nil {
  143. t.Fatalf("failed to generate new key pair: %s.", err)
  144. }
  145. id2, err := w.NewKeyPair()
  146. if err != nil {
  147. t.Fatalf("failed to generate new key pair: %s.", err)
  148. }
  149. pk1, err := w.GetPrivateKey(id1)
  150. if err != nil {
  151. t.Fatalf("failed to retrieve the key pair: %s.", err)
  152. }
  153. pk2, err := w.GetPrivateKey(id2)
  154. if err != nil {
  155. t.Fatalf("failed to retrieve the key pair: %s.", err)
  156. }
  157. if !w.HasKeyPair(id1) {
  158. t.Fatalf("failed HasIdentity(pk1).")
  159. }
  160. if !w.HasKeyPair(id2) {
  161. t.Fatalf("failed HasIdentity(pk2).")
  162. }
  163. if pk1 == nil {
  164. t.Fatalf("failed GetIdentity(pk1).")
  165. }
  166. if pk2 == nil {
  167. t.Fatalf("failed GetIdentity(pk2).")
  168. }
  169. if !validatePrivateKey(pk1) {
  170. t.Fatalf("pk1 is invalid.")
  171. }
  172. if !validatePrivateKey(pk2) {
  173. t.Fatalf("pk2 is invalid.")
  174. }
  175. // Delete one identity
  176. done := w.DeleteKeyPair(id1)
  177. if !done {
  178. t.Fatalf("failed to delete id1.")
  179. }
  180. pk1, err = w.GetPrivateKey(id1)
  181. if err == nil {
  182. t.Fatalf("retrieve the key pair: false positive.")
  183. }
  184. pk2, err = w.GetPrivateKey(id2)
  185. if err != nil {
  186. t.Fatalf("failed to retrieve the key pair: %s.", err)
  187. }
  188. if w.HasKeyPair(id1) {
  189. t.Fatalf("failed DeleteIdentity(pub1): still exist.")
  190. }
  191. if !w.HasKeyPair(id2) {
  192. t.Fatalf("failed DeleteIdentity(pub1): pub2 does not exist.")
  193. }
  194. if pk1 != nil {
  195. t.Fatalf("failed DeleteIdentity(pub1): first key still exist.")
  196. }
  197. if pk2 == nil {
  198. t.Fatalf("failed DeleteIdentity(pub1): second key does not exist.")
  199. }
  200. // Delete again non-existing identity
  201. done = w.DeleteKeyPair(id1)
  202. if done {
  203. t.Fatalf("delete id1: false positive.")
  204. }
  205. pk1, err = w.GetPrivateKey(id1)
  206. if err == nil {
  207. t.Fatalf("retrieve the key pair: false positive.")
  208. }
  209. pk2, err = w.GetPrivateKey(id2)
  210. if err != nil {
  211. t.Fatalf("failed to retrieve the key pair: %s.", err)
  212. }
  213. if w.HasKeyPair(id1) {
  214. t.Fatalf("failed delete non-existing identity: exist.")
  215. }
  216. if !w.HasKeyPair(id2) {
  217. t.Fatalf("failed delete non-existing identity: pub2 does not exist.")
  218. }
  219. if pk1 != nil {
  220. t.Fatalf("failed delete non-existing identity: first key exist.")
  221. }
  222. if pk2 == nil {
  223. t.Fatalf("failed delete non-existing identity: second key does not exist.")
  224. }
  225. // Delete second identity
  226. done = w.DeleteKeyPair(id2)
  227. if !done {
  228. t.Fatalf("failed to delete id2.")
  229. }
  230. pk1, err = w.GetPrivateKey(id1)
  231. if err == nil {
  232. t.Fatalf("retrieve the key pair: false positive.")
  233. }
  234. pk2, err = w.GetPrivateKey(id2)
  235. if err == nil {
  236. t.Fatalf("retrieve the key pair: false positive.")
  237. }
  238. if w.HasKeyPair(id1) {
  239. t.Fatalf("failed delete second identity: first identity exist.")
  240. }
  241. if w.HasKeyPair(id2) {
  242. t.Fatalf("failed delete second identity: still exist.")
  243. }
  244. if pk1 != nil {
  245. t.Fatalf("failed delete second identity: first key exist.")
  246. }
  247. if pk2 != nil {
  248. t.Fatalf("failed delete second identity: second key exist.")
  249. }
  250. }
  251. func TestWhisperSymKeyManagement(t *testing.T) {
  252. InitSingleTest()
  253. var err error
  254. var k1, k2 []byte
  255. w := New(&DefaultConfig)
  256. id1 := string("arbitrary-string-1")
  257. id2 := string("arbitrary-string-2")
  258. id1, err = w.GenerateSymKey()
  259. if err != nil {
  260. t.Fatalf("failed GenerateSymKey with seed %d: %s.", seed, err)
  261. }
  262. k1, err = w.GetSymKey(id1)
  263. if err != nil {
  264. t.Fatalf("failed GetSymKey(id1).")
  265. }
  266. k2, err = w.GetSymKey(id2)
  267. if err == nil {
  268. t.Fatalf("failed GetSymKey(id2): false positive.")
  269. }
  270. if !w.HasSymKey(id1) {
  271. t.Fatalf("failed HasSymKey(id1).")
  272. }
  273. if w.HasSymKey(id2) {
  274. t.Fatalf("failed HasSymKey(id2): false positive.")
  275. }
  276. if k1 == nil {
  277. t.Fatalf("first key does not exist.")
  278. }
  279. if k2 != nil {
  280. t.Fatalf("second key still exist.")
  281. }
  282. // add existing id, nothing should change
  283. randomKey := make([]byte, aesKeyLength)
  284. mrand.Read(randomKey)
  285. id1, err = w.AddSymKeyDirect(randomKey)
  286. if err != nil {
  287. t.Fatalf("failed AddSymKey with seed %d: %s.", seed, err)
  288. }
  289. k1, err = w.GetSymKey(id1)
  290. if err != nil {
  291. t.Fatalf("failed w.GetSymKey(id1).")
  292. }
  293. k2, err = w.GetSymKey(id2)
  294. if err == nil {
  295. t.Fatalf("failed w.GetSymKey(id2): false positive.")
  296. }
  297. if !w.HasSymKey(id1) {
  298. t.Fatalf("failed w.HasSymKey(id1).")
  299. }
  300. if w.HasSymKey(id2) {
  301. t.Fatalf("failed w.HasSymKey(id2): false positive.")
  302. }
  303. if k1 == nil {
  304. t.Fatalf("first key does not exist.")
  305. }
  306. if !bytes.Equal(k1, randomKey) {
  307. t.Fatalf("k1 != randomKey.")
  308. }
  309. if k2 != nil {
  310. t.Fatalf("second key already exist.")
  311. }
  312. id2, err = w.AddSymKeyDirect(randomKey)
  313. if err != nil {
  314. t.Fatalf("failed AddSymKey(id2) with seed %d: %s.", seed, err)
  315. }
  316. k1, err = w.GetSymKey(id1)
  317. if err != nil {
  318. t.Fatalf("failed w.GetSymKey(id1).")
  319. }
  320. k2, err = w.GetSymKey(id2)
  321. if err != nil {
  322. t.Fatalf("failed w.GetSymKey(id2).")
  323. }
  324. if !w.HasSymKey(id1) {
  325. t.Fatalf("HasSymKey(id1) failed.")
  326. }
  327. if !w.HasSymKey(id2) {
  328. t.Fatalf("HasSymKey(id2) failed.")
  329. }
  330. if k1 == nil {
  331. t.Fatalf("k1 does not exist.")
  332. }
  333. if k2 == nil {
  334. t.Fatalf("k2 does not exist.")
  335. }
  336. if !bytes.Equal(k1, k2) {
  337. t.Fatalf("k1 != k2.")
  338. }
  339. if !bytes.Equal(k1, randomKey) {
  340. t.Fatalf("k1 != randomKey.")
  341. }
  342. if len(k1) != aesKeyLength {
  343. t.Fatalf("wrong length of k1.")
  344. }
  345. if len(k2) != aesKeyLength {
  346. t.Fatalf("wrong length of k2.")
  347. }
  348. w.DeleteSymKey(id1)
  349. k1, err = w.GetSymKey(id1)
  350. if err == nil {
  351. t.Fatalf("failed w.GetSymKey(id1): false positive.")
  352. }
  353. if k1 != nil {
  354. t.Fatalf("failed GetSymKey(id1): false positive.")
  355. }
  356. k2, err = w.GetSymKey(id2)
  357. if err != nil {
  358. t.Fatalf("failed w.GetSymKey(id2).")
  359. }
  360. if w.HasSymKey(id1) {
  361. t.Fatalf("failed to delete first key: still exist.")
  362. }
  363. if !w.HasSymKey(id2) {
  364. t.Fatalf("failed to delete first key: second key does not exist.")
  365. }
  366. if k1 != nil {
  367. t.Fatalf("failed to delete first key.")
  368. }
  369. if k2 == nil {
  370. t.Fatalf("failed to delete first key: second key is nil.")
  371. }
  372. w.DeleteSymKey(id1)
  373. w.DeleteSymKey(id2)
  374. k1, err = w.GetSymKey(id1)
  375. if err == nil {
  376. t.Fatalf("failed w.GetSymKey(id1): false positive.")
  377. }
  378. k2, err = w.GetSymKey(id2)
  379. if err == nil {
  380. t.Fatalf("failed w.GetSymKey(id2): false positive.")
  381. }
  382. if k1 != nil || k2 != nil {
  383. t.Fatalf("k1 or k2 is not nil")
  384. }
  385. if w.HasSymKey(id1) {
  386. t.Fatalf("failed to delete second key: first key exist.")
  387. }
  388. if w.HasSymKey(id2) {
  389. t.Fatalf("failed to delete second key: still exist.")
  390. }
  391. if k1 != nil {
  392. t.Fatalf("failed to delete second key: first key is not nil.")
  393. }
  394. if k2 != nil {
  395. t.Fatalf("failed to delete second key: second key is not nil.")
  396. }
  397. randomKey = make([]byte, aesKeyLength+1)
  398. mrand.Read(randomKey)
  399. _, err = w.AddSymKeyDirect(randomKey)
  400. if err == nil {
  401. t.Fatalf("added the key with wrong size, seed %d.", seed)
  402. }
  403. const password = "arbitrary data here"
  404. id1, err = w.AddSymKeyFromPassword(password)
  405. if err != nil {
  406. t.Fatalf("failed AddSymKeyFromPassword(id1) with seed %d: %s.", seed, err)
  407. }
  408. id2, err = w.AddSymKeyFromPassword(password)
  409. if err != nil {
  410. t.Fatalf("failed AddSymKeyFromPassword(id2) with seed %d: %s.", seed, err)
  411. }
  412. k1, err = w.GetSymKey(id1)
  413. if err != nil {
  414. t.Fatalf("failed w.GetSymKey(id1).")
  415. }
  416. k2, err = w.GetSymKey(id2)
  417. if err != nil {
  418. t.Fatalf("failed w.GetSymKey(id2).")
  419. }
  420. if !w.HasSymKey(id1) {
  421. t.Fatalf("HasSymKey(id1) failed.")
  422. }
  423. if !w.HasSymKey(id2) {
  424. t.Fatalf("HasSymKey(id2) failed.")
  425. }
  426. if k1 == nil {
  427. t.Fatalf("k1 does not exist.")
  428. }
  429. if k2 == nil {
  430. t.Fatalf("k2 does not exist.")
  431. }
  432. if !bytes.Equal(k1, k2) {
  433. t.Fatalf("k1 != k2.")
  434. }
  435. if len(k1) != aesKeyLength {
  436. t.Fatalf("wrong length of k1.")
  437. }
  438. if len(k2) != aesKeyLength {
  439. t.Fatalf("wrong length of k2.")
  440. }
  441. if !validateSymmetricKey(k2) {
  442. t.Fatalf("key validation failed.")
  443. }
  444. }
  445. func TestExpiry(t *testing.T) {
  446. InitSingleTest()
  447. w := New(&DefaultConfig)
  448. w.SetMinimumPoW(0.0000001)
  449. defer w.SetMinimumPoW(DefaultMinimumPoW)
  450. w.Start(nil)
  451. defer w.Stop()
  452. params, err := generateMessageParams()
  453. if err != nil {
  454. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  455. }
  456. params.TTL = 1
  457. msg, err := NewSentMessage(params)
  458. if err != nil {
  459. t.Fatalf("failed to create new message with seed %d: %s.", seed, err)
  460. }
  461. env, err := msg.Wrap(params)
  462. if err != nil {
  463. t.Fatalf("failed Wrap with seed %d: %s.", seed, err)
  464. }
  465. err = w.Send(env)
  466. if err != nil {
  467. t.Fatalf("failed to send envelope with seed %d: %s.", seed, err)
  468. }
  469. // wait till received or timeout
  470. var received, expired bool
  471. for j := 0; j < 20; j++ {
  472. time.Sleep(100 * time.Millisecond)
  473. if len(w.Envelopes()) > 0 {
  474. received = true
  475. break
  476. }
  477. }
  478. if !received {
  479. t.Fatalf("did not receive the sent envelope, seed: %d.", seed)
  480. }
  481. // wait till expired or timeout
  482. for j := 0; j < 20; j++ {
  483. time.Sleep(100 * time.Millisecond)
  484. if len(w.Envelopes()) == 0 {
  485. expired = true
  486. break
  487. }
  488. }
  489. if !expired {
  490. t.Fatalf("expire failed, seed: %d.", seed)
  491. }
  492. }
  493. func TestCustomization(t *testing.T) {
  494. InitSingleTest()
  495. w := New(&DefaultConfig)
  496. defer w.SetMinimumPoW(DefaultMinimumPoW)
  497. defer w.SetMaxMessageSize(DefaultMaxMessageSize)
  498. w.Start(nil)
  499. defer w.Stop()
  500. const smallPoW = 0.00001
  501. f, err := generateFilter(t, true)
  502. if err != nil {
  503. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  504. }
  505. params, err := generateMessageParams()
  506. if err != nil {
  507. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  508. }
  509. params.KeySym = f.KeySym
  510. params.Topic = BytesToTopic(f.Topics[2])
  511. params.PoW = smallPoW
  512. params.TTL = 3600 * 24 // one day
  513. msg, err := NewSentMessage(params)
  514. if err != nil {
  515. t.Fatalf("failed to create new message with seed %d: %s.", seed, err)
  516. }
  517. env, err := msg.Wrap(params)
  518. if err != nil {
  519. t.Fatalf("failed Wrap with seed %d: %s.", seed, err)
  520. }
  521. err = w.Send(env)
  522. if err == nil {
  523. t.Fatalf("successfully sent envelope with PoW %.06f, false positive (seed %d).", env.PoW(), seed)
  524. }
  525. w.SetMinimumPoW(smallPoW / 2)
  526. err = w.Send(env)
  527. if err != nil {
  528. t.Fatalf("failed to send envelope with seed %d: %s.", seed, err)
  529. }
  530. params.TTL++
  531. msg, err = NewSentMessage(params)
  532. if err != nil {
  533. t.Fatalf("failed to create new message with seed %d: %s.", seed, err)
  534. }
  535. env, err = msg.Wrap(params)
  536. if err != nil {
  537. t.Fatalf("failed Wrap with seed %d: %s.", seed, err)
  538. }
  539. w.SetMaxMessageSize(uint32(env.size() - 1))
  540. err = w.Send(env)
  541. if err == nil {
  542. t.Fatalf("successfully sent oversized envelope (seed %d): false positive.", seed)
  543. }
  544. w.SetMaxMessageSize(DefaultMaxMessageSize)
  545. err = w.Send(env)
  546. if err != nil {
  547. t.Fatalf("failed to send second envelope with seed %d: %s.", seed, err)
  548. }
  549. // wait till received or timeout
  550. var received bool
  551. for j := 0; j < 20; j++ {
  552. time.Sleep(100 * time.Millisecond)
  553. if len(w.Envelopes()) > 1 {
  554. received = true
  555. break
  556. }
  557. }
  558. if !received {
  559. t.Fatalf("did not receive the sent envelope, seed: %d.", seed)
  560. }
  561. // check w.messages()
  562. id, err := w.Subscribe(f)
  563. if err != nil {
  564. t.Fatalf("failed subscribe with seed %d: %s.", seed, err)
  565. }
  566. time.Sleep(5 * time.Millisecond)
  567. mail := f.Retrieve()
  568. if len(mail) > 0 {
  569. t.Fatalf("received premature mail")
  570. }
  571. mail = w.Messages(id)
  572. if len(mail) != 2 {
  573. t.Fatalf("failed to get whisper messages")
  574. }
  575. }
  576. func TestSymmetricSendCycle(t *testing.T) {
  577. InitSingleTest()
  578. w := New(&DefaultConfig)
  579. defer w.SetMinimumPoW(DefaultMinimumPoW)
  580. defer w.SetMaxMessageSize(DefaultMaxMessageSize)
  581. w.Start(nil)
  582. defer w.Stop()
  583. filter1, err := generateFilter(t, true)
  584. if err != nil {
  585. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  586. }
  587. filter1.PoW = DefaultMinimumPoW
  588. // Copy the first filter since some of its fields
  589. // are randomly gnerated.
  590. filter2 := &Filter{
  591. KeySym: filter1.KeySym,
  592. Topics: filter1.Topics,
  593. PoW: filter1.PoW,
  594. AllowP2P: filter1.AllowP2P,
  595. Messages: make(map[common.Hash]*ReceivedMessage),
  596. }
  597. params, err := generateMessageParams()
  598. if err != nil {
  599. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  600. }
  601. filter1.Src = &params.Src.PublicKey
  602. filter2.Src = &params.Src.PublicKey
  603. params.KeySym = filter1.KeySym
  604. params.Topic = BytesToTopic(filter1.Topics[2])
  605. params.PoW = filter1.PoW
  606. params.WorkTime = 10
  607. params.TTL = 50
  608. msg, err := NewSentMessage(params)
  609. if err != nil {
  610. t.Fatalf("failed to create new message with seed %d: %s.", seed, err)
  611. }
  612. env, err := msg.Wrap(params)
  613. if err != nil {
  614. t.Fatalf("failed Wrap with seed %d: %s.", seed, err)
  615. }
  616. _, err = w.Subscribe(filter1)
  617. if err != nil {
  618. t.Fatalf("failed subscribe 1 with seed %d: %s.", seed, err)
  619. }
  620. _, err = w.Subscribe(filter2)
  621. if err != nil {
  622. t.Fatalf("failed subscribe 2 with seed %d: %s.", seed, err)
  623. }
  624. err = w.Send(env)
  625. if err != nil {
  626. t.Fatalf("Failed sending envelope with PoW %.06f (seed %d): %s", env.PoW(), seed, err)
  627. }
  628. // wait till received or timeout
  629. var received bool
  630. for j := 0; j < 200; j++ {
  631. time.Sleep(10 * time.Millisecond)
  632. if len(w.Envelopes()) > 0 {
  633. received = true
  634. break
  635. }
  636. }
  637. if !received {
  638. t.Fatalf("did not receive the sent envelope, seed: %d.", seed)
  639. }
  640. // check w.messages()
  641. time.Sleep(5 * time.Millisecond)
  642. mail1 := filter1.Retrieve()
  643. mail2 := filter2.Retrieve()
  644. if len(mail2) == 0 {
  645. t.Fatalf("did not receive any email for filter 2")
  646. }
  647. if len(mail1) == 0 {
  648. t.Fatalf("did not receive any email for filter 1")
  649. }
  650. }
  651. func TestSymmetricSendWithoutAKey(t *testing.T) {
  652. InitSingleTest()
  653. w := New(&DefaultConfig)
  654. defer w.SetMinimumPoW(DefaultMinimumPoW)
  655. defer w.SetMaxMessageSize(DefaultMaxMessageSize)
  656. w.Start(nil)
  657. defer w.Stop()
  658. filter, err := generateFilter(t, true)
  659. if err != nil {
  660. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  661. }
  662. filter.PoW = DefaultMinimumPoW
  663. params, err := generateMessageParams()
  664. if err != nil {
  665. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  666. }
  667. filter.Src = nil
  668. params.KeySym = filter.KeySym
  669. params.Topic = BytesToTopic(filter.Topics[2])
  670. params.PoW = filter.PoW
  671. params.WorkTime = 10
  672. params.TTL = 50
  673. msg, err := NewSentMessage(params)
  674. if err != nil {
  675. t.Fatalf("failed to create new message with seed %d: %s.", seed, err)
  676. }
  677. env, err := msg.Wrap(params)
  678. if err != nil {
  679. t.Fatalf("failed Wrap with seed %d: %s.", seed, err)
  680. }
  681. _, err = w.Subscribe(filter)
  682. if err != nil {
  683. t.Fatalf("failed subscribe 1 with seed %d: %s.", seed, err)
  684. }
  685. err = w.Send(env)
  686. if err != nil {
  687. t.Fatalf("Failed sending envelope with PoW %.06f (seed %d): %s", env.PoW(), seed, err)
  688. }
  689. // wait till received or timeout
  690. var received bool
  691. for j := 0; j < 200; j++ {
  692. time.Sleep(10 * time.Millisecond)
  693. if len(w.Envelopes()) > 0 {
  694. received = true
  695. break
  696. }
  697. }
  698. if !received {
  699. t.Fatalf("did not receive the sent envelope, seed: %d.", seed)
  700. }
  701. // check w.messages()
  702. time.Sleep(5 * time.Millisecond)
  703. mail := filter.Retrieve()
  704. if len(mail) == 0 {
  705. t.Fatalf("did not receive message in spite of not setting a public key")
  706. }
  707. }
  708. func TestSymmetricSendKeyMismatch(t *testing.T) {
  709. InitSingleTest()
  710. w := New(&DefaultConfig)
  711. defer w.SetMinimumPoW(DefaultMinimumPoW)
  712. defer w.SetMaxMessageSize(DefaultMaxMessageSize)
  713. w.Start(nil)
  714. defer w.Stop()
  715. filter, err := generateFilter(t, true)
  716. if err != nil {
  717. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  718. }
  719. filter.PoW = DefaultMinimumPoW
  720. params, err := generateMessageParams()
  721. if err != nil {
  722. t.Fatalf("failed generateMessageParams with seed %d: %s.", seed, err)
  723. }
  724. params.KeySym = filter.KeySym
  725. params.Topic = BytesToTopic(filter.Topics[2])
  726. params.PoW = filter.PoW
  727. params.WorkTime = 10
  728. params.TTL = 50
  729. msg, err := NewSentMessage(params)
  730. if err != nil {
  731. t.Fatalf("failed to create new message with seed %d: %s.", seed, err)
  732. }
  733. env, err := msg.Wrap(params)
  734. if err != nil {
  735. t.Fatalf("failed Wrap with seed %d: %s.", seed, err)
  736. }
  737. _, err = w.Subscribe(filter)
  738. if err != nil {
  739. t.Fatalf("failed subscribe 1 with seed %d: %s.", seed, err)
  740. }
  741. err = w.Send(env)
  742. if err != nil {
  743. t.Fatalf("Failed sending envelope with PoW %.06f (seed %d): %s", env.PoW(), seed, err)
  744. }
  745. // wait till received or timeout
  746. var received bool
  747. for j := 0; j < 200; j++ {
  748. time.Sleep(10 * time.Millisecond)
  749. if len(w.Envelopes()) > 0 {
  750. received = true
  751. break
  752. }
  753. }
  754. if !received {
  755. t.Fatalf("did not receive the sent envelope, seed: %d.", seed)
  756. }
  757. // check w.messages()
  758. time.Sleep(5 * time.Millisecond)
  759. mail := filter.Retrieve()
  760. if len(mail) > 0 {
  761. t.Fatalf("received a message when keys weren't matching")
  762. }
  763. }