mirror of
https://github.com/tinygo-org/drivers.git
synced 2026-07-26 10:38:41 +00:00
46c9ba9595
Signed-off-by: deadprogram <ron@hybridgroup.com>
141 lines
3.1 KiB
Go
141 lines
3.1 KiB
Go
// This example opens a TCP connection using a device with WiFiNINA firmware
|
|
// and sends some data, for the purpose of testing speed and connectivity.
|
|
//
|
|
// You can open a server to accept connections from this program using:
|
|
//
|
|
// nc -w 5 -lk 8080
|
|
//
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"machine"
|
|
"time"
|
|
|
|
"tinygo.org/x/drivers/net"
|
|
"tinygo.org/x/drivers/wifinina"
|
|
)
|
|
|
|
// access point info
|
|
const ssid = ""
|
|
const pass = ""
|
|
|
|
// IP address of the server aka "hub". Replace with your own info.
|
|
const serverIP = ""
|
|
|
|
// these are the default pins for the Arduino Nano33 IoT.
|
|
// change these to connect to a different UART or pins for the ESP8266/ESP32
|
|
var (
|
|
|
|
// these are the default pins for the Arduino Nano33 IoT.
|
|
uart = machine.UART2
|
|
tx = machine.NINA_TX
|
|
rx = machine.NINA_RX
|
|
spi = machine.NINA_SPI
|
|
|
|
// this is the ESP chip that has the WIFININA firmware flashed on it
|
|
adaptor *wifinina.Device
|
|
)
|
|
|
|
var buf = &bytes.Buffer{}
|
|
|
|
func main() {
|
|
|
|
uart.Configure(machine.UARTConfig{TX: tx, RX: rx})
|
|
|
|
// Configure SPI for 8Mhz, Mode 0, MSB First
|
|
spi.Configure(machine.SPIConfig{
|
|
Frequency: 8 * 1e6,
|
|
SDO: machine.NINA_SDO,
|
|
SDI: machine.NINA_SDI,
|
|
SCK: machine.NINA_SCK,
|
|
})
|
|
|
|
adaptor = wifinina.New(spi,
|
|
machine.NINA_CS,
|
|
machine.NINA_ACK,
|
|
machine.NINA_GPIO0,
|
|
machine.NINA_RESETN)
|
|
adaptor.Configure()
|
|
|
|
connectToAP()
|
|
|
|
for {
|
|
sendBatch()
|
|
time.Sleep(500 * time.Millisecond)
|
|
}
|
|
println("Done.")
|
|
}
|
|
|
|
func sendBatch() {
|
|
|
|
// make TCP connection
|
|
ip := net.ParseIP(serverIP)
|
|
raddr := &net.TCPAddr{IP: ip, Port: 8080}
|
|
laddr := &net.TCPAddr{Port: 8080}
|
|
|
|
message("---------------\r\nDialing TCP connection")
|
|
conn, err := net.DialTCP("tcp", laddr, raddr)
|
|
for ; err != nil; conn, err = net.DialTCP("tcp", laddr, raddr) {
|
|
message(err.Error())
|
|
time.Sleep(5 * time.Second)
|
|
}
|
|
|
|
n := 0
|
|
w := 0
|
|
start := time.Now()
|
|
|
|
// send data
|
|
message("Sending data")
|
|
|
|
for i := 0; i < 1000; i++ {
|
|
buf.Reset()
|
|
fmt.Fprint(buf,
|
|
"\r---------------------------- i == ", i, " ----------------------------"+
|
|
"\r---------------------------- i == ", i, " ----------------------------")
|
|
if w, err = conn.Write(buf.Bytes()); err != nil {
|
|
println("error:", err.Error(), "\r")
|
|
continue
|
|
}
|
|
n += w
|
|
}
|
|
|
|
buf.Reset()
|
|
ms := time.Now().Sub(start).Milliseconds()
|
|
fmt.Fprint(buf, "\nWrote ", n, " bytes in ", ms, " ms\r\n")
|
|
message(buf.String())
|
|
|
|
if _, err := conn.Write(buf.Bytes()); err != nil {
|
|
println("error:", err.Error(), "\r")
|
|
}
|
|
|
|
// Right now this code is never reached. Need a way to trigger it...
|
|
println("Disconnecting TCP...")
|
|
conn.Close()
|
|
}
|
|
|
|
// connect to access point
|
|
func connectToAP() {
|
|
time.Sleep(2 * time.Second)
|
|
message("Connecting to " + ssid)
|
|
adaptor.SetPassphrase(ssid, pass)
|
|
for st, _ := adaptor.GetConnectionStatus(); st != wifinina.StatusConnected; {
|
|
message("Connection status: " + st.String())
|
|
time.Sleep(1 * time.Second)
|
|
st, _ = adaptor.GetConnectionStatus()
|
|
}
|
|
message("Connected.")
|
|
time.Sleep(2 * time.Second)
|
|
ip, _, _, err := adaptor.GetIP()
|
|
for ; err != nil; ip, _, _, err = adaptor.GetIP() {
|
|
message(err.Error())
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
message(ip.String())
|
|
}
|
|
|
|
func message(msg string) {
|
|
println(msg, "\r")
|
|
}
|