2018-03-31 19:33:32 +00:00
package testing
2018-03-30 21:16:51 +00:00
import (
2018-06-03 19:36:20 +00:00
"crypto/rsa"
2018-05-28 18:05:06 +00:00
"cwtch.im/cwtch/model"
"cwtch.im/cwtch/peer"
2018-06-03 19:36:20 +00:00
"cwtch.im/cwtch/peer/connections"
2018-05-28 18:05:06 +00:00
cwtchserver "cwtch.im/cwtch/server"
2018-05-20 19:58:16 +00:00
"fmt"
2018-06-23 16:15:36 +00:00
"git.openprivacy.ca/openprivacy/libricochet-go/utils"
2018-06-14 00:42:08 +00:00
"golang.org/x/net/proxy"
2018-06-03 19:36:20 +00:00
"io/ioutil"
"log"
2018-04-27 19:20:33 +00:00
"os"
2018-05-30 18:42:17 +00:00
"runtime"
2018-05-20 19:58:16 +00:00
"testing"
"time"
2018-03-30 21:16:51 +00:00
)
2018-04-27 19:20:33 +00:00
const (
2018-06-05 22:06:38 +00:00
serverKeyfile = "./../server/app/private_key"
2018-06-03 19:36:20 +00:00
localKeyfile = "./private_key"
2018-04-27 19:20:33 +00:00
)
var (
2018-06-05 22:06:38 +00:00
aliceLines = [ ] string { "Hello, I'm Alice" , "bye" }
bobLines = [ ] string { "Hi, my name is Bob." , "toodles" , "welcome" }
carolLines = [ ] string { "Howdy, thanks!" }
2018-04-27 19:20:33 +00:00
)
2018-06-14 00:42:08 +00:00
func loadPrivateKey ( t * testing . T ) * rsa . PrivateKey {
2018-06-05 22:06:38 +00:00
if _ , err := os . Stat ( serverKeyfile ) ; os . IsNotExist ( err ) {
2018-06-14 00:42:08 +00:00
return nil
2018-04-27 19:20:33 +00:00
}
2018-06-14 00:42:08 +00:00
2018-06-05 22:06:38 +00:00
fmt . Println ( "Found server key " + serverKeyfile + ", loading..." )
pk , err := utils . LoadPrivateKeyFromFile ( serverKeyfile )
if err != nil {
t . Fatalf ( "Could not load server's key from %v" , serverKeyfile )
}
2018-06-14 00:42:08 +00:00
return pk
}
func genPrivateKey ( t * testing . T ) * rsa . PrivateKey {
fmt . Println ( "generating new private key..." )
pk , err := utils . GeneratePrivateKey ( )
if err != nil {
t . Fatalf ( "error generating new private key: %v\n" , err )
}
err = ioutil . WriteFile ( localKeyfile , [ ] byte ( utils . PrivateKeyToString ( pk ) ) , 0600 )
if err != nil {
t . Fatalf ( "error writing new private key to file %s: %v\n" , localKeyfile , err )
}
return pk
2018-04-27 19:20:33 +00:00
}
2018-06-05 22:06:38 +00:00
func printAndCountVerifedTimeline ( t * testing . T , timeline [ ] model . Message ) int {
numVerified := 0
2018-05-20 19:58:16 +00:00
for _ , message := range timeline {
2018-04-27 19:20:33 +00:00
fmt . Printf ( "%v %v> %s [%t]\n" , message . Timestamp , message . PeerID , message . Message , message . Verified )
2018-06-05 22:06:38 +00:00
if message . Verified {
numVerified ++
2018-04-27 19:20:33 +00:00
}
}
2018-06-05 22:06:38 +00:00
return numVerified
2018-04-27 19:20:33 +00:00
}
2018-06-14 00:42:08 +00:00
func serverCheck ( t * testing . T , serverAddr string ) bool {
torDialer , err := proxy . SOCKS5 ( "tcp" , "127.0.0.1:9050" , nil , proxy . Direct )
if err != nil {
t . Logf ( "Could not get SOCKS5 proxy: %v" , err )
return false
}
2018-05-30 18:42:17 +00:00
2018-06-14 00:42:08 +00:00
// Doesn't seem to be a way to turn the default timeout of 2 minutes down
conn , err := torDialer . Dial ( "tcp" , serverAddr + ".onion:9878" )
if err != nil {
t . Logf ( "Could not dial %v: %v" , serverAddr , err )
return false
2018-04-27 19:20:33 +00:00
}
2018-06-14 00:42:08 +00:00
conn . Close ( )
return true
2018-04-27 19:20:33 +00:00
}
2018-06-19 22:28:44 +00:00
func waitForPeerConnection ( t * testing . T , peer peer . CwtchPeerInterface , server string ) {
2018-06-03 19:36:20 +00:00
for {
servers := peer . GetServers ( )
2018-06-26 18:24:50 +00:00
fmt . Println ( servers )
2018-06-03 19:36:20 +00:00
state , ok := servers [ server ]
if ok {
2018-06-14 00:42:08 +00:00
if state == connections . FAILED {
2018-06-19 22:28:44 +00:00
t . Fatalf ( "%v could not connect to %v" , peer . GetProfile ( ) . Onion , server )
2018-06-14 00:42:08 +00:00
}
2018-06-03 19:36:20 +00:00
if state != connections . AUTHENTICATED {
time . Sleep ( time . Second * 10 )
continue
}
} else {
t . Fatalf ( "peer server connectiond %v should have entry for server %v" , servers , server )
}
break
}
return
}
2018-03-30 21:16:51 +00:00
func TestCwtchPeerIntegration ( t * testing . T ) {
2018-04-27 19:20:33 +00:00
// Hide logging "noise"
log . SetOutput ( ioutil . Discard )
2018-05-30 18:42:17 +00:00
numGoRoutinesStart := runtime . NumGoroutine ( )
2018-04-27 19:20:33 +00:00
2018-05-28 18:36:04 +00:00
// ***** Cwtch Server managment *****
2018-06-15 16:21:07 +00:00
var server * cwtchserver . Server
2018-06-14 00:42:08 +00:00
serverKey := loadPrivateKey ( t )
2018-06-15 16:21:07 +00:00
2018-04-27 19:20:33 +00:00
serverOnline := false
2018-06-14 00:42:08 +00:00
var serverAddr string
if serverKey != nil {
serverAddr , _ = utils . GetOnionAddress ( serverKey )
2018-04-27 19:20:33 +00:00
fmt . Printf ( "Checking if test server %v is online...\n" , serverAddr )
2018-06-14 00:42:08 +00:00
serverOnline = serverCheck ( t , serverAddr )
2018-04-27 19:20:33 +00:00
}
2018-05-20 19:58:16 +00:00
if ! serverOnline {
2018-06-14 00:42:08 +00:00
// launch app with new key
fmt . Println ( "No server found!" )
serverKey = genPrivateKey ( t )
serverAddr , _ = utils . GetOnionAddress ( serverKey )
2018-05-30 18:42:17 +00:00
server = new ( cwtchserver . Server )
2018-06-14 00:42:08 +00:00
fmt . Println ( "Starting cwtch server..." )
2018-06-19 22:38:22 +00:00
config := cwtchserver . Config { PrivateKeyBytes : utils . PrivateKeyToString ( serverKey ) , MaxBufferLines : 100 , ServerReporting : cwtchserver . Reporting { } }
2018-06-18 22:57:21 +00:00
go server . Run ( & config )
2018-04-27 19:20:33 +00:00
// let tor get established
fmt . Printf ( "Establishing Tor hidden service: %v...\n" , serverAddr )
} else {
fmt . Printf ( "Found existing cwtch server %v, using for tests...\n" , serverAddr )
}
2018-05-30 18:42:17 +00:00
numGoRoutinesPostServer := runtime . NumGoroutine ( )
2018-05-28 18:36:04 +00:00
// ***** Peer setup *****
2018-04-27 19:20:33 +00:00
fmt . Println ( "Creating Alice..." )
2018-06-24 18:37:11 +00:00
alice := peer . NewCwtchPeer ( "Alice" , "alicepass" )
2018-05-28 18:36:04 +00:00
go alice . Listen ( )
2018-06-19 22:28:44 +00:00
fmt . Println ( "Alice created:" , alice . GetProfile ( ) . Onion )
2018-04-27 19:20:33 +00:00
2018-05-28 18:36:04 +00:00
fmt . Println ( "Creating Bob..." )
2018-06-24 18:37:11 +00:00
bob := peer . NewCwtchPeer ( "Bob" , "bobpass" )
2018-05-28 18:36:04 +00:00
go bob . Listen ( )
2018-06-19 22:28:44 +00:00
fmt . Println ( "Bob created:" , bob . GetProfile ( ) . Onion )
2018-05-28 18:36:04 +00:00
2018-06-05 22:06:38 +00:00
fmt . Println ( "Creating Carol..." )
2018-06-24 18:37:11 +00:00
carol := peer . NewCwtchPeer ( "Carol" , "carolpass" )
2018-06-05 22:06:38 +00:00
go carol . Listen ( )
2018-06-19 22:28:44 +00:00
fmt . Println ( "Carol created:" , carol . GetProfile ( ) . Onion )
2018-06-05 22:06:38 +00:00
fmt . Println ( "Waiting for Alice, Bob, and Carol to connection with onion network..." )
time . Sleep ( time . Second * 70 )
2018-05-30 18:42:17 +00:00
numGoRoutinesPostPeerStart := runtime . NumGoroutine ( )
2018-05-28 18:36:04 +00:00
2018-06-05 22:06:38 +00:00
// ***** Peering, server joining, group creation / invite *****
2018-05-28 18:36:04 +00:00
fmt . Println ( "Creating group on " , serverAddr , "..." )
2018-06-19 22:28:44 +00:00
groupID , _ , err := alice . StartGroup ( serverAddr )
2018-06-15 16:21:07 +00:00
fmt . Printf ( "Created group: %v!\n" , groupID )
2018-04-27 19:20:33 +00:00
if err != nil {
2018-06-05 22:06:38 +00:00
t . Errorf ( "Failed to init group: %v" , err )
2018-04-27 19:20:33 +00:00
return
}
2018-06-05 22:06:38 +00:00
fmt . Println ( "Alice peering with Bob..." )
2018-06-19 22:28:44 +00:00
alice . PeerWithOnion ( bob . GetProfile ( ) . Onion )
2018-06-05 22:06:38 +00:00
fmt . Println ( "Alice peering with Carol..." )
2018-06-19 22:28:44 +00:00
alice . PeerWithOnion ( carol . GetProfile ( ) . Onion )
2018-04-27 19:20:33 +00:00
2018-06-05 22:06:38 +00:00
fmt . Println ( "Alice joining server..." )
alice . JoinServer ( serverAddr )
fmt . Println ( "Bob joining server..." )
bob . JoinServer ( serverAddr )
fmt . Println ( "Waiting for peerings and server joins..." )
2018-06-22 18:11:23 +00:00
time . Sleep ( time . Second * 120 )
2018-04-27 19:20:33 +00:00
2018-05-28 18:36:04 +00:00
fmt . Println ( "Alice inviting Bob to group..." )
2018-06-19 22:28:44 +00:00
err = alice . InviteOnionToGroup ( bob . GetProfile ( ) . Onion , groupID )
2018-05-28 18:36:04 +00:00
if err != nil {
t . Fatalf ( "Error for Alice inviting Bob to group: %v" , err )
}
time . Sleep ( time . Second * 10 )
2018-04-27 19:20:33 +00:00
2018-05-28 18:36:04 +00:00
fmt . Println ( "Bob examining groups and accepting invites..." )
2018-06-19 22:28:44 +00:00
for _ , groupID := range bob . GetGroups ( ) {
group := bob . GetGroup ( groupID )
2018-05-28 18:36:04 +00:00
fmt . Printf ( "Bob group: %v (Accepted: %v)\n" , group . GroupID , group . Accepted )
if group . Accepted == false {
fmt . Printf ( "Bob received and accepting group invite: %v\n" , group . GroupID )
bob . AcceptInvite ( group . GroupID )
}
}
2018-06-05 22:06:38 +00:00
time . Sleep ( time . Second * 5 )
2018-04-27 19:20:33 +00:00
2018-06-05 22:06:38 +00:00
numGoRoutinesPostServerConnect := runtime . NumGoroutine ( )
2018-05-30 18:42:17 +00:00
2018-06-05 22:06:38 +00:00
// ***** Fill up message history of server ******
2018-04-27 19:20:33 +00:00
2018-06-05 22:06:38 +00:00
/ *
2018-06-03 19:36:20 +00:00
// filler group will be used to fill up the servers message history a bit to stress test fetch later for carol
fillerGroupId , _ , err := alice . Profile . StartGroup ( serverAddr )
if err != nil {
t . Errorf ( "Failed to init filler group: %v" , err )
return
}
2018-06-05 22:06:38 +00:00
2018-06-03 19:36:20 +00:00
fmt . Println ( "Alice filling message history of server..." )
for i := 0 ; i < 100 ; i ++ {
go func ( x int ) {
time . Sleep ( time . Second * time . Duration ( x ) )
err := alice . SendMessageToGroup ( fillerGroupId , aliceLines [ 0 ] )
if err != nil {
fmt . Println ( "SEND" , x , "ERROR:" , err )
} else {
fmt . Println ( "SEND" , x , " SUCCESS!" )
}
} ( i )
}
2018-04-27 19:20:33 +00:00
2018-06-03 19:36:20 +00:00
time . Sleep ( time . Second * 110 )
2018-06-05 22:06:38 +00:00
* /
2018-06-03 19:36:20 +00:00
// Wait for them to join the server
waitForPeerConnection ( t , alice , serverAddr )
waitForPeerConnection ( t , bob , serverAddr )
//numGouRoutinesPostServerConnect := runtime.NumGoroutine()
2018-05-28 18:36:04 +00:00
// ***** Conversation *****
2018-04-27 19:20:33 +00:00
fmt . Println ( "Starting conversation in group..." )
2018-04-27 19:20:33 +00:00
// Conversation
fmt . Println ( "Alice> " , aliceLines [ 0 ] )
2018-06-15 16:21:07 +00:00
err = alice . SendMessageToGroup ( groupID , aliceLines [ 0 ] )
2018-05-20 19:58:16 +00:00
if err != nil {
2018-05-28 18:36:04 +00:00
t . Fatalf ( "Alice failed to send a message to the group: %v" , err )
2018-05-20 19:58:16 +00:00
}
time . Sleep ( time . Second * 10 )
2018-04-27 19:20:33 +00:00
fmt . Println ( "Bob> " , bobLines [ 0 ] )
2018-06-15 16:21:07 +00:00
err = bob . SendMessageToGroup ( groupID , bobLines [ 0 ] )
2018-05-20 19:58:16 +00:00
if err != nil {
2018-05-28 18:36:04 +00:00
t . Fatalf ( "Bob failed to send a message to the group: %v" , err )
2018-05-20 19:58:16 +00:00
}
time . Sleep ( time . Second * 10 )
2018-04-27 19:20:33 +00:00
fmt . Println ( "Alice> " , aliceLines [ 1 ] )
2018-06-15 16:21:07 +00:00
alice . SendMessageToGroup ( groupID , aliceLines [ 1 ] )
2018-06-05 22:06:38 +00:00
time . Sleep ( time . Second * 10 )
2018-04-27 19:20:33 +00:00
fmt . Println ( "Bob> " , bobLines [ 1 ] )
2018-06-15 16:21:07 +00:00
bob . SendMessageToGroup ( groupID , bobLines [ 1 ] )
2018-05-20 19:58:16 +00:00
time . Sleep ( time . Second * 10 )
2018-04-27 19:20:33 +00:00
2018-06-05 22:06:38 +00:00
fmt . Println ( "Alice inviting Carol to group..." )
2018-06-19 22:28:44 +00:00
err = alice . InviteOnionToGroup ( carol . GetProfile ( ) . Onion , groupID )
2018-06-05 22:06:38 +00:00
if err != nil {
t . Fatalf ( "Error for Alice inviting Carol to group: %v" , err )
}
2018-05-20 19:58:16 +00:00
time . Sleep ( time . Second * 10 )
2018-06-05 22:06:38 +00:00
fmt . Println ( "Carol examining groups and accepting invites..." )
2018-06-19 22:28:44 +00:00
for _ , groupID := range carol . GetGroups ( ) {
group := carol . GetGroup ( groupID )
2018-06-05 22:06:38 +00:00
fmt . Printf ( "Carol group: %v (Accepted: %v)\n" , group . GroupID , group . Accepted )
if group . Accepted == false {
fmt . Printf ( "Carol received and accepting group invite: %v\n" , group . GroupID )
carol . AcceptInvite ( group . GroupID )
}
}
fmt . Println ( "Shutting down Alice..." )
alice . Shutdown ( )
time . Sleep ( time . Second * 5 )
numGoRoutinesPostAlice := runtime . NumGoroutine ( )
fmt . Println ( "Carol joining server..." )
carol . JoinServer ( serverAddr )
2018-06-03 19:36:20 +00:00
waitForPeerConnection ( t , carol , serverAddr )
2018-06-05 22:06:38 +00:00
numGoRotinesPostCarolConnect := runtime . NumGoroutine ( )
2018-04-27 19:20:33 +00:00
fmt . Println ( "Bob> " , bobLines [ 2 ] )
2018-06-15 16:21:07 +00:00
bob . SendMessageToGroup ( groupID , bobLines [ 2 ] )
2018-05-20 19:58:16 +00:00
time . Sleep ( time . Second * 10 )
2018-04-27 19:20:33 +00:00
2018-06-05 22:06:38 +00:00
fmt . Println ( "Carol> " , carolLines [ 0 ] )
2018-06-15 16:21:07 +00:00
carol . SendMessageToGroup ( groupID , carolLines [ 0 ] )
2018-06-05 22:06:38 +00:00
time . Sleep ( time . Second * 10 )
2018-04-27 19:20:33 +00:00
2018-05-28 18:36:04 +00:00
// ***** Verify Test *****
2018-04-27 19:20:33 +00:00
2018-06-26 18:24:50 +00:00
// final syncing time...
time . Sleep ( time . Second * 15 )
2018-04-27 19:20:33 +00:00
2018-06-19 22:28:44 +00:00
alicesGroup := alice . GetGroup ( groupID )
2018-04-27 19:20:33 +00:00
if alicesGroup == nil {
t . Error ( "aliceGroup == nil" )
return
}
2018-04-27 19:20:33 +00:00
2018-06-05 22:06:38 +00:00
fmt . Printf ( "Alice's TimeLine:\n" )
aliceVerified := printAndCountVerifedTimeline ( t , alicesGroup . GetTimeline ( ) )
if aliceVerified != 4 {
t . Errorf ( "Alice did not have 4 verified messages" )
}
2018-04-27 19:20:33 +00:00
2018-06-19 22:28:44 +00:00
bobsGroup := bob . GetGroup ( groupID )
2018-04-27 19:20:33 +00:00
if bobsGroup == nil {
t . Error ( "bobGroup == nil" )
return
}
2018-06-05 22:06:38 +00:00
fmt . Printf ( "Bob's TimeLine:\n" )
bobVerified := printAndCountVerifedTimeline ( t , bobsGroup . GetTimeline ( ) )
if bobVerified != 5 {
t . Errorf ( "Bob did not have 5 verified messages" )
}
2018-06-19 22:28:44 +00:00
carolsGroup := carol . GetGroup ( groupID )
2018-06-05 22:06:38 +00:00
fmt . Printf ( "Carol's TimeLine:\n" )
carolVerified := printAndCountVerifedTimeline ( t , carolsGroup . GetTimeline ( ) )
if carolVerified != 3 {
t . Errorf ( "Carol did not have 3 verified messages" )
}
2018-04-27 19:20:33 +00:00
2018-06-05 22:06:38 +00:00
if len ( alicesGroup . GetTimeline ( ) ) != 4 {
2018-04-27 19:20:33 +00:00
t . Errorf ( "Alice's timeline does not have all messages" )
2018-06-26 18:24:50 +00:00
return
2018-04-27 19:20:33 +00:00
}
2018-06-05 22:06:38 +00:00
if len ( bobsGroup . GetTimeline ( ) ) != 6 {
t . Errorf ( "Bob's timeline does not have all messages" )
2018-06-26 18:24:50 +00:00
}
// check message 0,1,2,3
aliceGroupTimeline := alicesGroup . GetTimeline ( )
if aliceGroupTimeline [ 0 ] . Message != aliceLines [ 0 ] || aliceGroupTimeline [ 1 ] . Message != bobLines [ 0 ] ||
aliceGroupTimeline [ 2 ] . Message != aliceLines [ 1 ] || aliceGroupTimeline [ 3 ] . Message != bobLines [ 1 ] {
t . Errorf ( "Some of Alice's timeline messages did not have the expected content!" )
}
// check message 0,1,2,3,4,5
bobGroupTimeline := bobsGroup . GetTimeline ( )
if bobGroupTimeline [ 0 ] . Message != aliceLines [ 0 ] || bobGroupTimeline [ 1 ] . Message != bobLines [ 0 ] ||
bobGroupTimeline [ 2 ] . Message != aliceLines [ 1 ] || bobGroupTimeline [ 3 ] . Message != bobLines [ 1 ] ||
bobGroupTimeline [ 4 ] . Message != bobLines [ 2 ] || bobGroupTimeline [ 5 ] . Message != carolLines [ 0 ] {
t . Errorf ( "Some of Bob's timeline messages did not have the expected content!" )
2018-06-05 22:06:38 +00:00
}
if len ( carolsGroup . GetTimeline ( ) ) != 6 {
t . Errorf ( "Carol's timeline does not have all messages" )
2018-06-26 18:24:50 +00:00
}
// check message 0,1,2,3,4,5
carolGroupTimeline := carolsGroup . GetTimeline ( )
if carolGroupTimeline [ 0 ] . Message != aliceLines [ 0 ] || carolGroupTimeline [ 1 ] . Message != bobLines [ 0 ] ||
carolGroupTimeline [ 2 ] . Message != aliceLines [ 1 ] || carolGroupTimeline [ 3 ] . Message != bobLines [ 1 ] ||
carolGroupTimeline [ 4 ] . Message != bobLines [ 2 ] || carolGroupTimeline [ 5 ] . Message != carolLines [ 0 ] {
t . Errorf ( "Some of Carol's timeline messages did not have the expected content!" )
2018-04-27 19:20:33 +00:00
}
2018-05-30 18:42:17 +00:00
fmt . Println ( "Shutting down Bob..." )
bob . Shutdown ( )
time . Sleep ( time . Second * 3 )
numGoRoutinesPostBob := runtime . NumGoroutine ( )
if server != nil {
fmt . Println ( "Shutting down server..." )
server . Shutdown ( )
time . Sleep ( time . Second * 3 )
}
numGoRoutinesPostServerShutdown := runtime . NumGoroutine ( )
2018-06-05 22:06:38 +00:00
fmt . Println ( "Shuttind down Carol..." )
carol . Shutdown ( )
2018-05-30 18:42:17 +00:00
time . Sleep ( time . Second * 3 )
2018-06-05 22:06:38 +00:00
numGoRoutinesPostCarol := runtime . NumGoroutine ( )
2018-05-30 18:42:17 +00:00
2018-06-03 19:36:20 +00:00
fmt . Printf ( "numGoRoutinesStart: %v\nnumGoRoutinesPostServer: %v\nnumGoRoutinesPostPeerStart: %v\nnumGoRoutinesPostPeerAndServerConnect: %v\n" +
2018-06-05 22:06:38 +00:00
"numGoRoutinesPostAlice: %v\nnumGoRotinesPostCarolConnect: %v\nnumGoRoutinesPostBob: %v\nnumGoRoutinesPostServerShutdown: %v\nnumGoRoutinesPostCarol: %v\n" ,
numGoRoutinesStart , numGoRoutinesPostServer , numGoRoutinesPostPeerStart , numGoRoutinesPostServerConnect ,
numGoRoutinesPostAlice , numGoRotinesPostCarolConnect , numGoRoutinesPostBob , numGoRoutinesPostServerShutdown , numGoRoutinesPostCarol )
2018-05-30 18:42:17 +00:00
2018-06-14 00:42:08 +00:00
if numGoRoutinesStart != numGoRoutinesPostCarol {
t . Errorf ( "Number of GoRoutines at start (%v) does not match number of goRoutines after cleanup of peers and servers (%v), clean up failed, leak detected!" , numGoRoutinesStart , numGoRoutinesPostCarol )
2018-05-30 18:42:17 +00:00
}
2018-06-14 00:42:08 +00:00
2018-03-30 21:16:51 +00:00
}