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  }