12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182 |
- package main
-
- import (
- "cwtch.im/cwtch/event"
- "cwtch.im/cwtch/peer"
- "cwtch.im/cwtch/protocol/connections"
- "errors"
- "fmt"
- "git.openprivacy.ca/openprivacy/libricochet-go/connectivity"
- "os"
- "time"
- )
-
- func waitForPeerGroupConnection(peer peer.CwtchPeer, groupID string) error {
- for {
- _, ok := peer.GetProfile().Groups[groupID]
- if ok {
- state := peer.GetGroupState(groupID)
- if state == connections.FAILED {
- return errors.New("Connection to group " + groupID + " failed!")
- }
- if state != connections.AUTHENTICATED {
- fmt.Printf("peer %v waiting to authenticate with group %v 's server, current state: %v\n", peer.GetProfile().Onion, groupID, connections.ConnectionStateName[state])
- time.Sleep(time.Second * 10)
- continue
- }
- } else {
- return errors.New("peer server connections should have entry for server but do not")
- }
- break
- }
- return nil
- }
-
- func main() {
- if len(os.Args) != 2 {
- fmt.Printf("Usage: ./servermon SERVER_ADDRESS\n")
- os.Exit(1)
- }
-
- serverAddr := os.Args[1]
-
- acn, err := connectivity.StartTor(".", "")
- if err != nil {
- fmt.Printf("Could not start tor: %v\n", err)
- os.Exit(1)
- }
-
- botPeer := peer.NewCwtchPeer("servermon")
- botPeer.Init(acn, event.NewEventManager())
-
- fmt.Printf("Connecting to %v...\n", serverAddr)
- botPeer.JoinServer(serverAddr)
- groupID, _, err := botPeer.StartGroup(serverAddr)
- if err != nil {
- fmt.Printf("Error creating group on server %v: %v\n", serverAddr, err)
- os.Exit(1)
- }
-
- err = waitForPeerGroupConnection(botPeer, groupID)
- if err != nil {
- fmt.Printf("Could not connect to server %v: %v\n", serverAddr, err)
- os.Exit(1)
- }
-
- timeout := 1 * time.Second
- timeElapsed := 0 * time.Second
- for {
- err := botPeer.SendMessageToGroup(groupID, timeout.String())
- if err != nil {
- fmt.Printf("Sent to group on server %v failed at interval %v of total %v with: %v\n", serverAddr, timeout, timeElapsed, err)
- os.Exit(1)
- } else {
- fmt.Printf("Successfully sent message to %v at interval %v of total %v\n", serverAddr, timeout, timeElapsed)
- }
- time.Sleep(timeout)
- timeElapsed += timeout
- if timeout < 2*time.Minute {
- timeout = timeout * 2
- }
- }
- }
|