summaryrefslogtreecommitdiff
path: root/src/rhctl/serial_port.go
blob: 7a0af2c62cce8b7d132b0a5d4f6d9f8d71d186fc (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
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
//
//  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) 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) 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
}