在golang中将控制信号转换为字节ascii十六进制代码

时间:2018-02-21 19:54:24

标签: go websocket ascii gorilla

我试图在golang中捕获ctrl + c,ctrl + z,ctrl + \等信号,并将其等效的十六进制代码写入websocket。我正在做类似下面的事情。

func (c *poc) writePump() {
    var b = make([]byte, 1)
    d := make(chan os.Signal, 1)
    signal.Notify(d, syscall.SIGINT, syscall.SIGTSTP, syscall.SIGQUIT)

    for {
        os.Stdin.Read(b)
        err = c.ws.WriteMessage(websocket.TextMessage, b)

        go func() {
            for {
                s := <-d
                err = cli.sendControlSignalToWebsockets(s)
            }
        }()

        if err != nil {
            log.Printf("Failed to send UTF8 char: %s", err)
        }
    }
}

func (c *poc) sendControlSignalToWebsockets(s os.Signal) error {
    var err error

    switch s {
    case syscall.SIGINT:
        err = c.ws.WriteMessage(websocket.TextMessage, []byte{'\003'})
    case syscall.SIGTSTP:
        err = c.ws.WriteMessage(websocket.TextMessage, []byte{'\x1a'})
    case syscall.SIGQUIT:
        err = c.ws.WriteMessage(websocket.TextMessage, []byte{'\x1c'})
    }

    if err != nil {
        return err
    }
    return nil
}

相反,我想做一些事情,比如捕获所有信号,而不是硬编码每个信号相当于它的十六进制码,将信号转换为十六进制码。

func (c *poc) writePump() {
    var b = make([]byte, 1)
    d := make(chan os.Signal, 1)
    signal.Notify(d)

    for {
        os.Stdin.Read(b)
        err = c.ws.WriteMessage(websocket.TextMessage, b)

        go func() {
            for {
                s := <-d
                //???? do some conversions and write it to websocket
            }
        }()

        if err != nil {
            log.Printf("Failed to send UTF8 char: %s", err)
        }
    }
}

很想知道是否有人有任何想法。谢谢!

1 个答案:

答案 0 :(得分:0)

听起来你正试图通过websocket传递信号。因此,您需要一种编码方式,不会与其他字符混淆,因此使用不可打印的ascii字符是一个合理的举措。

由于没有将syscall信号映射到byte的惯例,因此您无法导入它 - 但这是个好消息! - 你可以创建自己的!

创建map[syscall.Signal]bytemap[os.Signal]byte并从那里进行翻译:

// Create a mapping between syscalls and bytes
var syscallByteMap = map[syscall.Signal]byte{
  syscall.SIGINT: '\003',
  ...more here...
}

...truncated...

        go func() {
        for {
            s := <-d
            // Look up the signal in the map
            b, ok := syscallByteMap[s]
            if ok {
                err = c.ws.WriteMessage(websocket.TextMessage, []byte{b})
            } else {
                err = nil
            }

        }
    }()