/
main.go
106 lines (92 loc) · 2.18 KB
/
main.go
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
103
104
105
106
// Copyright 2023 The Embedded Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// Espnet is an ESP-AT based TCP echo server. See also ../espat that uses the
// espat package directly and has much lower memory requirements. See also the
// same example written for FET1061 and STM32 development boards.
package main
import (
"fmt"
"io"
"net"
"strings"
"time"
"github.com/embeddedgo/espat"
"github.com/embeddedgo/espat/espnet"
"github.com/embeddedgo/imxrt/hal/lpuart"
"github.com/embeddedgo/imxrt/hal/lpuart/lpuart2"
"github.com/embeddedgo/imxrt/devboard/teensy4/board/pins"
)
func logErr(err error) bool {
if err != nil {
fmt.Println("error:", err.Error())
return true
}
return false
}
func fatalErr(err error) {
for err != nil {
fmt.Println("error:", err.Error())
time.Sleep(time.Second)
}
}
func main() {
// IO pins
espTx := pins.P14
espRx := pins.P15
// ESP-AT
u := lpuart2.Driver()
u.Setup(lpuart.Word8b, 115200)
u.UsePin(espRx, lpuart.RXD)
u.UsePin(espTx, lpuart.TXD)
u.EnableRx(512)
u.EnableTx()
time.Sleep(5 * time.Second) // to allow to see messages printed below
fmt.Print("Initializing ESP-AT module... ")
dev := espat.NewDevice("esp0", u, u)
fatalErr(dev.Init(true))
fatalErr(espnet.SetPasvRecv(dev, true))
fmt.Println("OK")
fmt.Println("Waiting for an IP address...")
for msg := range dev.Async() {
fatalErr(msg.Err)
fmt.Println(msg.Str)
if msg.Str == "WIFI GOT IP" {
break
}
}
txt, err := dev.CmdStr("+CIPSTA?")
fatalErr(err)
fmt.Println(strings.ReplaceAll(txt, "+CIPSTA:", ""))
ls, err := espnet.ListenDev(dev, "tcp", ":1111")
fatalErr(err)
fmt.Println("Listen on:", ls.Addr().String())
for {
c, err := ls.Accept()
fatalErr(err)
go handle(c)
}
}
func handle(c net.Conn) {
var buf [64]byte
fmt.Println("Connected:", c.RemoteAddr().String())
_, err := io.WriteString(c, "Echo Server\n\n")
if logErr(err) {
return
}
for {
n, err := c.Read(buf[:])
if err == io.EOF {
break
}
if logErr(err) {
return
}
_, err = c.Write(buf[:n])
if logErr(err) {
return
}
}
c.Close()
fmt.Println("Closed: ", c.RemoteAddr().String())
}