mirror of
https://github.com/hslam/msg.git
synced 2025-04-25 13:48:50 +08:00
88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
// Copyright (c) 2020 Meng Huang (mhboy@outlook.com)
|
|
// This package is licensed under a MIT license that can be found in the LICENSE file.
|
|
|
|
package msg
|
|
|
|
import (
|
|
"errors"
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
const (
|
|
// IPC_CREAT creates if key is nonexistent
|
|
IPC_CREAT = 01000
|
|
|
|
// IPC_EXCL fails if key exists.
|
|
IPC_EXCL = 02000
|
|
|
|
// IPC_NOWAIT returns error no wait.
|
|
IPC_NOWAIT = 04000
|
|
|
|
// IPC_PRIVATE is private key
|
|
IPC_PRIVATE = 00000
|
|
|
|
// SEM_UNDO sets up adjust on exit entry
|
|
SEM_UNDO = 010000
|
|
|
|
// IPC_RMID removes identifier
|
|
IPC_RMID = 0
|
|
// IPC_SET sets ipc_perm options.
|
|
IPC_SET = 1
|
|
// IPC_STAT gets ipc_perm options.
|
|
IPC_STAT = 2
|
|
|
|
maxText = 8192
|
|
)
|
|
|
|
// ErrTooLong is returned when the Text length is bigger than maxText.
|
|
var ErrTooLong = errors.New("Text length is too long")
|
|
|
|
type message struct {
|
|
Type uint
|
|
Text [maxText]byte
|
|
}
|
|
|
|
// Get calls the msgget system call.
|
|
func Get(key int, msgflg int) (int, error) {
|
|
r1, _, err := syscall.Syscall(syscall.SYS_MSGGET, uintptr(key), uintptr(msgflg), 0)
|
|
msgid := int(r1)
|
|
if msgid < 0 {
|
|
return msgid, syscall.Errno(err)
|
|
}
|
|
return msgid, nil
|
|
}
|
|
|
|
// Snd calls the msgsnd system call.
|
|
func Snd(msgid int, msgType uint, msgText []byte, flags int) error {
|
|
if len(msgText) > maxText {
|
|
return ErrTooLong
|
|
}
|
|
m := message{Type: msgType}
|
|
copy(m.Text[:], msgText)
|
|
_, _, err := syscall.Syscall6(syscall.SYS_MSGSND, uintptr(msgid), uintptr(unsafe.Pointer(&m)), uintptr(len(msgText)), uintptr(flags), 0, 0)
|
|
if err != 0 {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Rcv calls the msgrcv system call.
|
|
func Rcv(msgid int, msgType uint, flags int) ([]byte, error) {
|
|
m := message{Type: msgType}
|
|
length, _, err := syscall.Syscall6(syscall.SYS_MSGRCV, uintptr(msgid), uintptr(unsafe.Pointer(&m)), maxText, uintptr(msgType), uintptr(flags), 0)
|
|
if err != 0 {
|
|
return nil, err
|
|
}
|
|
return m.Text[:length], nil
|
|
}
|
|
|
|
// Remove removes the message queue with the given id.
|
|
func Remove(msgid int) error {
|
|
r1, _, errno := syscall.Syscall(syscall.SYS_MSGCTL, uintptr(msgid), IPC_RMID, 0)
|
|
if int(r1) < 0 {
|
|
return syscall.Errno(errno)
|
|
}
|
|
return nil
|
|
}
|