github.com/yinchengtsinghua/golang-Eos-dpos-Ethereum@v0.0.0-20190121132951-92cc4225ed8e/whisper/whisperv5/peer_test.go (about) 1 2 //此源码被清华学神尹成大魔王专业翻译分析并修改 3 //尹成QQ77025077 4 //尹成微信18510341407 5 //尹成所在QQ群721929980 6 //尹成邮箱 yinc13@mails.tsinghua.edu.cn 7 //尹成毕业于清华大学,微软区块链领域全球最有价值专家 8 //https://mvp.microsoft.com/zh-cn/PublicProfile/4033620 9 // 10 // 11 // 12 // 13 // 14 // 15 // 16 // 17 // 18 // 19 // 20 // 21 // 22 // 23 // 24 25 package whisperv5 26 27 import ( 28 "bytes" 29 "crypto/ecdsa" 30 "net" 31 "sync" 32 "testing" 33 "time" 34 35 "github.com/ethereum/go-ethereum/common" 36 "github.com/ethereum/go-ethereum/crypto" 37 "github.com/ethereum/go-ethereum/p2p" 38 "github.com/ethereum/go-ethereum/p2p/discover" 39 "github.com/ethereum/go-ethereum/p2p/nat" 40 ) 41 42 var keys = []string{ 43 "d49dcf37238dc8a7aac57dc61b9fee68f0a97f062968978b9fafa7d1033d03a9", 44 "73fd6143c48e80ed3c56ea159fe7494a0b6b393a392227b422f4c3e8f1b54f98", 45 "119dd32adb1daa7a4c7bf77f847fb28730785aa92947edf42fdd997b54de40dc", 46 "deeda8709dea935bb772248a3144dea449ffcc13e8e5a1fd4ef20ce4e9c87837", 47 "5bd208a079633befa349441bdfdc4d85ba9bd56081525008380a63ac38a407cf", 48 "1d27fb4912002d58a2a42a50c97edb05c1b3dffc665dbaa42df1fe8d3d95c9b5", 49 "15def52800c9d6b8ca6f3066b7767a76afc7b611786c1276165fbc61636afb68", 50 "51be6ab4b2dc89f251ff2ace10f3c1cc65d6855f3e083f91f6ff8efdfd28b48c", 51 "ef1ef7441bf3c6419b162f05da6037474664f198b58db7315a6f4de52414b4a0", 52 "09bdf6985aabc696dc1fbeb5381aebd7a6421727343872eb2fadfc6d82486fd9", 53 "15d811bf2e01f99a224cdc91d0cf76cea08e8c67905c16fee9725c9be71185c4", 54 "2f83e45cf1baaea779789f755b7da72d8857aeebff19362dd9af31d3c9d14620", 55 "73f04e34ac6532b19c2aae8f8e52f38df1ac8f5cd10369f92325b9b0494b0590", 56 "1e2e07b69e5025537fb73770f483dc8d64f84ae3403775ef61cd36e3faf162c1", 57 "8963d9bbb3911aac6d30388c786756b1c423c4fbbc95d1f96ddbddf39809e43a", 58 "0422da85abc48249270b45d8de38a4cc3c02032ede1fcf0864a51092d58a2f1f", 59 "8ae5c15b0e8c7cade201fdc149831aa9b11ff626a7ffd27188886cc108ad0fa8", 60 "acd8f5a71d4aecfcb9ad00d32aa4bcf2a602939b6a9dd071bab443154184f805", 61 "a285a922125a7481600782ad69debfbcdb0316c1e97c267aff29ef50001ec045", 62 "28fd4eee78c6cd4bf78f39f8ab30c32c67c24a6223baa40e6f9c9a0e1de7cef5", 63 "c5cca0c9e6f043b288c6f1aef448ab59132dab3e453671af5d0752961f013fc7", 64 "46df99b051838cb6f8d1b73f232af516886bd8c4d0ee07af9a0a033c391380fd", 65 "c6a06a53cbaadbb432884f36155c8f3244e244881b5ee3e92e974cfa166d793f", 66 "783b90c75c63dc72e2f8d11b6f1b4de54d63825330ec76ee8db34f06b38ea211", 67 "9450038f10ca2c097a8013e5121b36b422b95b04892232f930a29292d9935611", 68 "e215e6246ed1cfdcf7310d4d8cdbe370f0d6a8371e4eb1089e2ae05c0e1bc10f", 69 "487110939ed9d64ebbc1f300adeab358bc58875faf4ca64990fbd7fe03b78f2b", 70 "824a70ea76ac81366da1d4f4ac39de851c8ac49dca456bb3f0a186ceefa269a5", 71 "ba8f34fa40945560d1006a328fe70c42e35cc3d1017e72d26864cd0d1b150f15", 72 "30a5dfcfd144997f428901ea88a43c8d176b19c79dde54cc58eea001aa3d246c", 73 "de59f7183aca39aa245ce66a05245fecfc7e2c75884184b52b27734a4a58efa2", 74 "92629e2ff5f0cb4f5f08fffe0f64492024d36f045b901efb271674b801095c5a", 75 "7184c1701569e3a4c4d2ddce691edd983b81e42e09196d332e1ae2f1e062cff4", 76 } 77 78 const NumNodes = 16 // 79 80 type TestData struct { 81 counter [NumNodes]int 82 mutex sync.RWMutex 83 } 84 85 type TestNode struct { 86 shh *Whisper 87 id *ecdsa.PrivateKey 88 server *p2p.Server 89 filerId string 90 } 91 92 var result TestData 93 var nodes [NumNodes]*TestNode 94 var sharedKey = []byte("some arbitrary data here") 95 var sharedTopic TopicType = TopicType{0xF, 0x1, 0x2, 0} 96 var expectedMessage = []byte("per rectum ad astra") 97 98 // 99 // 100 // 101 // 102 // 103 // 104 func TestSimulation(t *testing.T) { 105 initialize(t) 106 107 for i := 0; i < NumNodes; i++ { 108 sendMsg(t, false, i) 109 } 110 111 sendMsg(t, true, 0) 112 checkPropagation(t) 113 stopServers() 114 } 115 116 func initialize(t *testing.T) { 117 var err error 118 119 for i := 0; i < NumNodes; i++ { 120 var node TestNode 121 node.shh = New(&DefaultConfig) 122 node.shh.SetMinimumPoW(0.00000001) 123 node.shh.Start(nil) 124 topics := make([]TopicType, 0) 125 topics = append(topics, sharedTopic) 126 f := Filter{KeySym: sharedKey} 127 f.Topics = [][]byte{topics[0][:]} 128 node.filerId, err = node.shh.Subscribe(&f) 129 if err != nil { 130 t.Fatalf("failed to install the filter: %s.", err) 131 } 132 node.id, err = crypto.HexToECDSA(keys[i]) 133 if err != nil { 134 t.Fatalf("failed convert the key: %s.", keys[i]) 135 } 136 name := common.MakeName("whisper-go", "2.0") 137 node.server = &p2p.Server{ 138 Config: p2p.Config{ 139 PrivateKey: node.id, 140 MaxPeers: NumNodes/2 + 1, 141 Name: name, 142 Protocols: node.shh.Protocols(), 143 ListenAddr: "127.0.0.1:0", 144 NAT: nat.Any(), 145 }, 146 } 147 148 err = node.server.Start() 149 if err != nil { 150 t.Fatalf("failed to start server %d.", i) 151 } 152 153 for j := 0; j < i; j++ { 154 peerNodeId := nodes[j].id 155 address, _ := net.ResolveTCPAddr("tcp", nodes[j].server.ListenAddr) 156 peerPort := uint16(address.Port) 157 peerNode := discover.PubkeyID(&peerNodeId.PublicKey) 158 peer := discover.NewNode(peerNode, address.IP, peerPort, peerPort) 159 node.server.AddPeer(peer) 160 } 161 162 nodes[i] = &node 163 } 164 } 165 166 func stopServers() { 167 for i := 0; i < NumNodes; i++ { 168 n := nodes[i] 169 if n != nil { 170 n.shh.Unsubscribe(n.filerId) 171 n.shh.Stop() 172 n.server.Stop() 173 } 174 } 175 } 176 177 func checkPropagation(t *testing.T) { 178 if t.Failed() { 179 return 180 } 181 182 const cycle = 100 183 const iterations = 100 184 185 for j := 0; j < iterations; j++ { 186 time.Sleep(cycle * time.Millisecond) 187 188 for i := 0; i < NumNodes; i++ { 189 f := nodes[i].shh.GetFilter(nodes[i].filerId) 190 if f == nil { 191 t.Fatalf("failed to get filterId %s from node %d.", nodes[i].filerId, i) 192 } 193 194 mail := f.Retrieve() 195 if !validateMail(t, i, mail) { 196 return 197 } 198 199 if isTestComplete() { 200 return 201 } 202 } 203 } 204 205 t.Fatalf("Test was not complete: timeout %d seconds.", iterations*cycle/1000) 206 } 207 208 func validateMail(t *testing.T, index int, mail []*ReceivedMessage) bool { 209 var cnt int 210 for _, m := range mail { 211 if bytes.Equal(m.Payload, expectedMessage) { 212 cnt++ 213 } 214 } 215 216 if cnt == 0 { 217 // 218 return true 219 } 220 if cnt > 1 { 221 t.Fatalf("node %d received %d.", index, cnt) 222 return false 223 } 224 225 if cnt > 0 { 226 result.mutex.Lock() 227 defer result.mutex.Unlock() 228 result.counter[index] += cnt 229 if result.counter[index] > 1 { 230 t.Fatalf("node %d accumulated %d.", index, result.counter[index]) 231 } 232 } 233 return true 234 } 235 236 func isTestComplete() bool { 237 result.mutex.RLock() 238 defer result.mutex.RUnlock() 239 240 for i := 0; i < NumNodes; i++ { 241 if result.counter[i] < 1 { 242 return false 243 } 244 } 245 246 for i := 0; i < NumNodes; i++ { 247 envelopes := nodes[i].shh.Envelopes() 248 if len(envelopes) < 2 { 249 return false 250 } 251 } 252 253 return true 254 } 255 256 func sendMsg(t *testing.T, expected bool, id int) { 257 if t.Failed() { 258 return 259 } 260 261 opt := MessageParams{KeySym: sharedKey, Topic: sharedTopic, Payload: expectedMessage, PoW: 0.00000001, WorkTime: 1} 262 if !expected { 263 opt.KeySym[0]++ 264 opt.Topic[0]++ 265 opt.Payload = opt.Payload[1:] 266 } 267 268 msg, err := NewSentMessage(&opt) 269 if err != nil { 270 t.Fatalf("failed to create new message with seed %d: %s.", seed, err) 271 } 272 envelope, err := msg.Wrap(&opt) 273 if err != nil { 274 t.Fatalf("failed to seal message: %s", err) 275 } 276 277 err = nodes[id].shh.Send(envelope) 278 if err != nil { 279 t.Fatalf("failed to send message: %s", err) 280 } 281 } 282 283 func TestPeerBasic(t *testing.T) { 284 InitSingleTest() 285 286 params, err := generateMessageParams() 287 if err != nil { 288 t.Fatalf("failed generateMessageParams with seed %d.", seed) 289 } 290 291 params.PoW = 0.001 292 msg, err := NewSentMessage(params) 293 if err != nil { 294 t.Fatalf("failed to create new message with seed %d: %s.", seed, err) 295 } 296 env, err := msg.Wrap(params) 297 if err != nil { 298 t.Fatalf("failed Wrap with seed %d.", seed) 299 } 300 301 p := newPeer(nil, nil, nil) 302 p.mark(env) 303 if !p.marked(env) { 304 t.Fatalf("failed mark with seed %d.", seed) 305 } 306 }