summaryrefslogtreecommitdiffstats
path: root/cmd/mock/main.go
blob: 88888f59ecfd900439762b84e5ab2deb1f98067e (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
package main

import (
	"encoding/binary"
	"flag"
	"log"
	"net"
)

var listenAddr string

type Encoder interface {
	Encode() []any
}

type Handshake struct {
	ver uint8
}

func (h Handshake) Encode() []any {
	return []any{h.ver}
}

type Test struct {
	msg string
}

func (h Test) Encode() []any {
	return []any{[]byte(h.msg)}
}

func main() {
	flag.StringVar(&listenAddr, "a", "localhost:9999", "Listening address:port")
	flag.Parse()

	log.Print("starting solec daemon mock")
	log.Fatal(listen())
}

func listen() error {
	ln, err := net.Listen("tcp", listenAddr)
	if err != nil {
		return err
	}

	log.Print("server is listening on: ", listenAddr)

	for {
		conn, err := ln.Accept()
		if err != nil {
			log.Print("cannot accept connection: ", err)
		}
		go handle(conn)
	}
}

func handle(conn net.Conn) {
	log.Print("received connection from: ", conn.RemoteAddr())

	hs := Handshake{
		ver: 0x0,
	}

	if err := send(conn, 0x01, hs); err != nil {
		log.Print("cannot send handshake ", err)
		return
	}

	if err := send(conn, 0xFF, Test{msg: "solec kujawski jest zajebisty"}); err != nil {
		log.Print("cannot send test message ", err)
		return
	}
}

func send(conn net.Conn, eType uint8, data Encoder) error {
	err := binary.Write(conn, binary.BigEndian, eType)
	if err != nil {
		return err
	}

	for _, v := range data.Encode() {
		err := binary.Write(conn, binary.BigEndian, v)
		if err != nil {
			return err
		}
	}

	return nil
}