mirror of
https://github.com/netbirdio/netbird.git
synced 2026-05-19 23:29:56 +00:00
97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
//go:build linux && !android
|
|
|
|
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"image"
|
|
|
|
"github.com/jezek/xgb/shm"
|
|
"github.com/jezek/xgb/xproto"
|
|
log "github.com/sirupsen/logrus"
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func (c *X11Capturer) initSHM() error {
|
|
if err := shm.Init(c.conn); err != nil {
|
|
return fmt.Errorf("init SHM extension: %w", err)
|
|
}
|
|
|
|
size := c.w * c.h * 4
|
|
id, err := unix.SysvShmGet(unix.IPC_PRIVATE, size, unix.IPC_CREAT|0600)
|
|
if err != nil {
|
|
return fmt.Errorf("shmget: %w", err)
|
|
}
|
|
|
|
addr, err := unix.SysvShmAttach(id, 0, 0)
|
|
if err != nil {
|
|
if _, ctlErr := unix.SysvShmCtl(id, unix.IPC_RMID, nil); ctlErr != nil {
|
|
log.Debugf("shmctl IPC_RMID on attach failure: %v", ctlErr)
|
|
}
|
|
return fmt.Errorf("shmat: %w", err)
|
|
}
|
|
|
|
if _, err := unix.SysvShmCtl(id, unix.IPC_RMID, nil); err != nil {
|
|
log.Debugf("shmctl IPC_RMID: %v", err)
|
|
}
|
|
|
|
seg, err := shm.NewSegId(c.conn)
|
|
if err != nil {
|
|
if detachErr := unix.SysvShmDetach(addr); detachErr != nil {
|
|
log.Debugf("shmdt on new-seg failure: %v", detachErr)
|
|
}
|
|
return fmt.Errorf("new SHM seg: %w", err)
|
|
}
|
|
|
|
if err := shm.AttachChecked(c.conn, seg, uint32(id), false).Check(); err != nil {
|
|
if detachErr := unix.SysvShmDetach(addr); detachErr != nil {
|
|
log.Debugf("shmdt on attach-checked failure: %v", detachErr)
|
|
}
|
|
return fmt.Errorf("SHM attach to X: %w", err)
|
|
}
|
|
|
|
c.shmID = id
|
|
c.shmAddr = addr
|
|
c.shmSeg = uint32(seg)
|
|
c.useSHM = true
|
|
return nil
|
|
}
|
|
|
|
func (c *X11Capturer) captureSHM() (*image.RGBA, error) {
|
|
if err := c.fillSHM(); err != nil {
|
|
return nil, err
|
|
}
|
|
img := c.nextBuffer()
|
|
swizzleBGRAtoRGBA(img.Pix, c.shmAddr[:c.w*c.h*4])
|
|
return img, nil
|
|
}
|
|
|
|
// captureSHMInto runs a single SHM GetImage and swizzles directly into the
|
|
// caller-provided destination, skipping the internal double-buffer.
|
|
func (c *X11Capturer) captureSHMInto(dst *image.RGBA) error {
|
|
if err := c.fillSHM(); err != nil {
|
|
return err
|
|
}
|
|
swizzleBGRAtoRGBA(dst.Pix, c.shmAddr[:c.w*c.h*4])
|
|
return nil
|
|
}
|
|
|
|
func (c *X11Capturer) fillSHM() error {
|
|
cookie := shm.GetImage(c.conn, xproto.Drawable(c.screen.Root),
|
|
0, 0, uint16(c.w), uint16(c.h), 0xFFFFFFFF,
|
|
xproto.ImageFormatZPixmap, shm.Seg(c.shmSeg), 0)
|
|
if _, err := cookie.Reply(); err != nil {
|
|
return fmt.Errorf("SHM GetImage: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *X11Capturer) closeSHM() {
|
|
if c.useSHM {
|
|
shm.Detach(c.conn, shm.Seg(c.shmSeg))
|
|
if err := unix.SysvShmDetach(c.shmAddr); err != nil {
|
|
log.Debugf("shmdt on close: %v", err)
|
|
}
|
|
}
|
|
}
|