summaryrefslogtreecommitdiff
path: root/main.go
blob: 076564c9757c3289619a93ddc1c7dd371507190a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package main

import (
	"encoding/hex"
	"fmt"
	"io"
	"log"
	"net"
	"sync"
)

const (
	client = `ownme.ipredator.se:10002`
	server = `ownme.ipredator.se:10000`
)

type Direction int

const (
	ServerClient Direction = iota
	ClientServer
)

func (d Direction) String() string {
	switch d {
	case ServerClient:
		return "Server → Client"
	case ClientServer:
		return "Client → Server"
	}
	return "unknown"
}

type Data struct {
	Round int
	Dir   Direction
	Data  []byte
	Size  int
}

func dump(dir Direction, src, dst net.Conn, data chan Data) error {
	buf := make([]byte, 4096)
	for k := 1; ; k++ {
		n, err := src.Read(buf)
		if err != nil {
			if err == io.EOF {
				return nil
			}
			return fmt.Errorf("READ %v", err)
		}

		data <- Data{Round: k, Dir: dir, Data: buf[:n], Size: n}

		_, err = dst.Write(buf[:n])
		if err != nil {
			return fmt.Errorf("WRITE %v", err)
		}
	}
}

func main() {
	var wg sync.WaitGroup

	srv, err := net.Dial("tcp", server)
	if err != nil {
		log.Fatal(err)
	}
	defer srv.Close()

	cnt, err := net.Dial("tcp", client)
	if err != nil {
		log.Fatal(err)
	}
	defer cnt.Close()

	data := make(chan Data)

	wg.Add(2)
	go func() {
		wg.Wait()
		close(data)
	}()

	go func(dir Direction) {
		if err := dump(dir, srv, cnt, data); err != nil {
			fmt.Println("ERROR", dir, err)
		}
		wg.Done()
	}(ServerClient)

	go func(dir Direction) {
		if err := dump(dir, cnt, srv, data); err != nil {
			fmt.Println("ERROR", dir, err)
		}
		wg.Done()
	}(ClientServer)

	for d := range data {
		fmt.Println(d.Round, d.Dir, d.Size, "bytes")
		fmt.Println(hex.Dump(d.Data))
	}
}