Compare commits

..

No commits in common. "604e5bb4addddeeef5db6683147df5dd6feee111" and "5bf90d1b6328c9785c9c92dd84708b29847b8753" have entirely different histories.

5 changed files with 28 additions and 33 deletions

View File

@ -10,10 +10,10 @@
* [x] 双向编码转换 * [x] 双向编码转换
* [x] 活动端口探测 * [x] 活动端口探测
* [x] 数据日志保存 * [x] 数据日志保存
* [x] Hex断帧设置 * [ ] 自动断帧设置
* [x] UDP数据转发
* [x] TCP数据转发
* [ ] 文件接收发送 * [ ] 文件接收发送
* [ ] UDP数据转发
* [ ] TCP数据转发
## 运行示例 ## 运行示例

View File

@ -18,7 +18,7 @@ var commands []Command
func cmdhelp() { func cmdhelp() {
var page = 0 var page = 0
strout(out, config.outputCode, fmt.Sprintf(">-------Help(%v)-------<\n", page)) fmt.Printf(">-------Help(%v)-------<\n", page)
for i := 0; i < len(commands); i++ { for i := 0; i < len(commands); i++ {
strout(out, config.outputCode, fmt.Sprintf(" %-10v --%v\n", commands[i].name, commands[i].description)) strout(out, config.outputCode, fmt.Sprintf(" %-10v --%v\n", commands[i].name, commands[i].description))
} }
@ -27,12 +27,12 @@ func cmdexit() {
os.Exit(0) os.Exit(0)
} }
func cmdargs() { func cmdargs() {
strout(out, config.outputCode, fmt.Sprintf(">-------Args(%v)-------<\n", len(args)-1)) fmt.Printf(">-------Args(%v)-------<\n", len(args)-1)
strout(out, config.outputCode, fmt.Sprintf("%q\n", args[1:])) fmt.Printf("%q\n", args[1:])
} }
func cmdhex() { func cmdhex() {
strout(out, config.outputCode, fmt.Sprintf(">-----Hex Send-----<\n")) fmt.Printf(">-----Hex Send-----<\n")
strout(out, config.outputCode, fmt.Sprintf("%q\n", args[1:])) fmt.Printf("%q\n", args[1:])
s := strings.Join(args[1:], "") s := strings.Join(args[1:], "")
b, err := hex.DecodeString(s) b, err := hex.DecodeString(s)
if err != nil { if err != nil {

View File

@ -17,28 +17,31 @@ type Config struct {
enableLog bool enableLog bool
logFilePath string logFilePath string
forWard int forWard int
frameSize int
address string address string
} }
type FoeWardMode int type FoeWardMode int
const ( const (
NOT FoeWardMode = iota NOT FoeWardMode = iota
TCPS
TCPC TCPC
UDPS
UDPC UDPC
) )
var config Config var config Config
func setForWardClient() (conn net.Conn) { func setForWard() (conn net.Conn) {
switch FoeWardMode(config.forWard) { switch FoeWardMode(config.forWard) {
case NOT: case TCPS:
case TCPC: case TCPC:
conn, err = net.Dial("tcp", config.address) conn, err = net.Dial("tcp", config.address)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
case UDPS:
case UDPC: case UDPC:
conn, err = net.Dial("udp", config.address) conn, err = net.Dial("udp", config.address)
if err != nil { if err != nil {

View File

@ -38,11 +38,10 @@ var (
endStr = Flag{ptrVal{string: &config.endStr}, "e", "end", Val{string: "\n"}, "终端换行符"} endStr = Flag{ptrVal{string: &config.endStr}, "e", "end", Val{string: "\n"}, "终端换行符"}
enableLog = Flag{ptrVal{bool: &config.enableLog}, "l", "log", Val{bool: false}, "是否启用日志保存"} enableLog = Flag{ptrVal{bool: &config.enableLog}, "l", "log", Val{bool: false}, "是否启用日志保存"}
logFilePath = Flag{ptrVal{string: &config.logFilePath}, "P", "Path", Val{string: "./Log.txt"}, "日志保存路径"} logFilePath = Flag{ptrVal{string: &config.logFilePath}, "P", "Path", Val{string: "./Log.txt"}, "日志保存路径"}
forWard = Flag{ptrVal{int: &config.forWard}, "f", "forward", Val{int: 0}, "转发模式(0: 无 1:TCP-C 2:UDP-C)"} forWard = Flag{ptrVal{int: &config.forWard}, "f", "forward", Val{int: 0}, "转发模式(0: 无 1:TCP-S 2:TCP-C 3:UDP-S 4:UDP-C)"}
address = Flag{ptrVal{string: &config.address}, "a", "address", Val{string: "127.0.0.1:12345"}, "转发服务地址"} address = Flag{ptrVal{string: &config.address}, "a", "address", Val{string: "127.0.0.1:12345"}, "转发服务地址"}
frameSize = Flag{ptrVal{int: &config.frameSize}, "F", "Frame", Val{int: 16}, "帧大小"}
parityBit = Flag{ptrVal{int: &config.parityBit}, "v", "verify", Val{int: 0}, "奇偶校验(0:无校验、1:奇校验、2:偶校验、3:1校验、4:0校验)"} parityBit = Flag{ptrVal{int: &config.parityBit}, "v", "verify", Val{int: 0}, "奇偶校验(0:无校验、1:奇校验、2:偶校验、3:1校验、4:0校验)"}
flags = []Flag{portName, baudRate, dataBits, stopBits, outputCode, inputCode, endStr, enableLog, logFilePath, forWard, address, frameSize, parityBit} flags = []Flag{portName, baudRate, dataBits, stopBits, outputCode, inputCode, endStr, enableLog, logFilePath, forWard, address, parityBit}
) )
type ValType int type ValType int

31
main.go
View File

@ -8,7 +8,6 @@ import (
"go.bug.st/serial" "go.bug.st/serial"
"io" "io"
"log" "log"
"net"
"os" "os"
"strings" "strings"
) )
@ -56,8 +55,7 @@ func init() {
func input(in io.Reader) { func input(in io.Reader) {
input := bufio.NewScanner(in) input := bufio.NewScanner(in)
var ok = false var ok = false
for { for input.Scan() {
input.Scan()
ok = false ok = false
args = strings.Split(input.Text(), " ") args = strings.Split(input.Text(), " ")
for _, cmd := range commands { for _, cmd := range commands {
@ -76,7 +74,7 @@ func input(in io.Reader) {
log.Fatal(err) log.Fatal(err)
} }
} }
err = serialPort.Drain() err := serialPort.Drain()
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
@ -90,10 +88,10 @@ func strout(out io.Writer, cs, str string) {
} }
} }
func output() { func output(out io.Writer) {
if strings.Compare(config.inputCode, "hex") == 0 { if strings.Compare(config.inputCode, "hex") == 0 {
b := make([]byte, 16) b := make([]byte, 16)
r, _ := io.LimitReader(serialPort, int64(config.frameSize)).Read(b) r, _ := io.LimitReader(serialPort, 16).Read(b)
if r != 0 { if r != 0 {
strout(out, config.outputCode, fmt.Sprintf("% X %q \n", b, b)) strout(out, config.outputCode, fmt.Sprintf("% X %q \n", b, b))
} }
@ -125,26 +123,21 @@ func main() {
defer func(port serial.Port) { defer func(port serial.Port) {
err := port.Close() err := port.Close()
if err != nil { if err != nil {
log.Fatal(err)
} }
}(serialPort) }(serialPort)
if FoeWardMode(config.forWard) != NOT { if FoeWardMode(config.forWard) != NOT {
conn := setForWardClient() conn := setForWard()
ins = append(ins, conn) ins = append(ins, conn)
outs = append(outs, conn) outs = append(outs, conn)
defer func(conn net.Conn) { defer conn.Close()
err := conn.Close()
if err != nil {
log.Fatal(err)
}
}(conn)
} }
if len(ins) != 0 {
for _, reader := range ins { if len(ins) != 1 {
go input(reader) in = io.MultiReader(ins...)
}
} }
go input(in)
if config.enableLog { if config.enableLog {
f, err := os.OpenFile(config.logFilePath, os.O_RDWR|os.O_APPEND|os.O_CREATE, 0666) f, err := os.OpenFile(config.logFilePath, os.O_RDWR|os.O_APPEND|os.O_CREATE, 0666)
if err != nil { if err != nil {
@ -156,6 +149,6 @@ func main() {
out = io.MultiWriter(outs...) out = io.MultiWriter(outs...)
} }
for { for {
output() output(out)
} }
} }