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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
|
//
// rhctl
//
// Copyright (C) 2009-2016 Christian Pointner <equinox@helsinki.at>
//
// This file is part of rhctl.
//
// rhctl is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// any later version.
//
// rhctl is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with rhctl. If not, see <http://www.gnu.org/licenses/>.
//
package main
import (
"bufio"
"errors"
"strconv"
"syscall"
"github.com/schleibinger/sio"
)
type Baudrate struct {
value uint32
code uint32
}
func NewBaudrate(value uint32) (b *Baudrate, err error) {
b = &Baudrate{}
err = b.FromUint(value)
return
}
func (b Baudrate) String() string {
return strconv.FormatUint(uint64(b.value), 10)
}
func (b Baudrate) MarshalText() (data []byte, err error) {
data = []byte(b.String())
return
}
func (b Baudrate) MarshalJSON() (data []byte, err error) {
data = []byte(b.String())
return
}
func (b Baudrate) MarshalTOML() (data []byte, err error) {
data = []byte(b.String())
return
}
func (b *Baudrate) FromUint(value uint32) error {
b.value = value
switch b.value {
case 50:
b.code = syscall.B50
case 75:
b.code = syscall.B75
case 110:
b.code = syscall.B110
case 134:
b.code = syscall.B134
case 150:
b.code = syscall.B150
case 200:
b.code = syscall.B200
case 300:
b.code = syscall.B300
case 600:
b.code = syscall.B600
case 1200:
b.code = syscall.B1200
case 1800:
b.code = syscall.B1800
case 2400:
b.code = syscall.B2400
case 4800:
b.code = syscall.B4800
case 9600:
b.code = syscall.B9600
case 19200:
b.code = syscall.B19200
case 38400:
b.code = syscall.B38400
case 57600:
b.code = syscall.B57600
case 115200:
b.code = syscall.B115200
case 230400:
b.code = syscall.B230400
case 460800:
b.code = syscall.B460800
case 500000:
b.code = syscall.B500000
case 576000:
b.code = syscall.B576000
case 921600:
b.code = syscall.B921600
case 1000000:
b.code = syscall.B1000000
case 1152000:
b.code = syscall.B1152000
case 1500000:
b.code = syscall.B1500000
case 2000000:
b.code = syscall.B2000000
case 2500000:
b.code = syscall.B2500000
case 3000000:
b.code = syscall.B3000000
case 3500000:
b.code = syscall.B3500000
case 4000000:
b.code = syscall.B4000000
default:
return errors.New("invalid baudrate")
}
return nil
}
func (b *Baudrate) fromString(str string) error {
vuint, err := strconv.ParseUint(str, 10, 32)
if err != nil {
return err
}
return b.FromUint(uint32(vuint))
}
func (b *Baudrate) UnmarshalText(data []byte) error {
return b.fromString(string(data))
}
func (b *Baudrate) UnmarshalJSON(data []byte) error {
return b.fromString(string(data))
}
func (b *Baudrate) UnmarshalTOML(data []byte) error {
return b.fromString(string(data))
}
type SerialPort struct {
port *sio.Port
rx <-chan string
tx chan<- string
newline string
}
func serialReader(c chan<- string, port *sio.Port, stop chan<- bool) {
defer func() {
port.Close()
stop <- true
}()
scanner := bufio.NewScanner(port)
scanner.Split(bufio.ScanLines)
for scanner.Scan() {
data := scanner.Text()
if len(data) == 0 {
continue
}
c <- data
}
if err := scanner.Err(); err != nil {
rhl.Printf("device(%s): read error: %s", port.LocalAddr(), err)
} else {
rhl.Printf("device(%s): closed", port.LocalAddr())
}
}
func serialWriter(c <-chan string, port *sio.Port, newline string, stop chan<- bool) {
defer func() {
port.Close()
stop <- true
}()
for data := range c {
port.Write([]byte(data + newline))
}
}
func (port *SerialPort) Run(stop chan<- bool) (err error) {
tx := make(chan string, 10)
rx := make(chan string, 20)
go serialReader(rx, port.port, stop)
go serialWriter(tx, port.port, port.newline, stop)
port.rx = rx
port.tx = tx
return
}
func SerialOpen(device string, rate Baudrate, newline string) (port *SerialPort, err error) {
port = &SerialPort{newline: newline}
if port.port, err = sio.Open(device, rate.code); err != nil {
return
}
return
}
|
