pspy/internal/inotify/event.go

package inotify

import (
	"bytes"
	"fmt"
	"log"
	"strconv"
	"time"
	"unsafe"

	"golang.org/x/sys/unix"
)

var InotifyEvents = map[uint32]string{
	unix.IN_ACCESS:                          "ACCESS",
	unix.IN_ATTRIB:                          "ATTRIB",
	unix.IN_CLOSE_NOWRITE:                   "CLOSE_NOWRITE",
	unix.IN_CLOSE_WRITE:                     "CLOSE_WRITE",
	unix.IN_CREATE:                          "CREATE",
	unix.IN_DELETE:                          "DELETE",
	unix.IN_DELETE_SELF:                     "DELETE_SELF",
	unix.IN_MODIFY:                          "MODIFY",
	unix.IN_MOVED_FROM:                      "MOVED_FROM",
	unix.IN_MOVED_TO:                        "MOVED_TO",
	unix.IN_MOVE_SELF:                       "MOVE_SELF",
	unix.IN_OPEN:                            "OPEN",
	(unix.IN_ACCESS | unix.IN_ISDIR):        "ACCESS DIR",
	(unix.IN_ATTRIB | unix.IN_ISDIR):        "ATTRIB DIR",
	(unix.IN_CLOSE_NOWRITE | unix.IN_ISDIR): "CLOSE_NOWRITE DIR",
	(unix.IN_CLOSE_WRITE | unix.IN_ISDIR):   "CLOSE_WRITE DIR",
	(unix.IN_CREATE | unix.IN_ISDIR):        "CREATE DIR",
	(unix.IN_DELETE | unix.IN_ISDIR):        "DELETE DIR",
	(unix.IN_DELETE_SELF | unix.IN_ISDIR):   "DELETE_SELF DIR",
	(unix.IN_MODIFY | unix.IN_ISDIR):        "MODIFY DIR",
	(unix.IN_MOVED_FROM | unix.IN_ISDIR):    "MOVED_FROM DIR",
	(unix.IN_MOVE_SELF | unix.IN_ISDIR):     "MODE_SELF DIR",
	(unix.IN_OPEN | unix.IN_ISDIR):          "OPEN DIR",
}

type Event struct {
	name string
	op   string
}

func (e Event) String() string {
	return fmt.Sprintf("%20s | %s", e.op, e.name)
}

func newEvent(name string, mask uint32) Event {
	e := Event{name: name}
	op, ok := InotifyEvents[mask]
	if !ok {
		op = strconv.FormatInt(int64(mask), 2)
	}
	e.op = op
	return e
}

func eventLogger(i *Inotify, buffers chan bufRead, print bool) {
	// enable printing only after delay since setting up watchers causes flood of events
	printEnabled := false
	go func() {
		<-time.After(1 * time.Second)
		printEnabled = print
	}()
	for bf := range buffers {
		n := bf.n
		buf := bf.buf

		if n < unix.SizeofInotifyEvent {
			// incomplete or erroneous read
			continue
		}

		var ptr uint32
		var name string
		for ptr <= uint32(n-unix.SizeofInotifyEvent) {
			sys := (*unix.InotifyEvent)(unsafe.Pointer(&buf[ptr]))
			ptr += unix.SizeofInotifyEvent

			watcher, ok := i.watchers[int(sys.Wd)]
			if !ok {
				continue
			}
			name = watcher.dir + "/"
			if sys.Len > 0 && len(buf) >= int(ptr+sys.Len) {
				name += string(bytes.TrimRight(buf[ptr:ptr+sys.Len], "\x00"))
				ptr += sys.Len
			}

			ev := newEvent(name, sys.Mask)
			if printEnabled {
				log.Printf("\x1b[32;1mFS: %+v\x1b[0m", ev)
			}
		}
	}
}