| // Copyright 2010 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| // +build linux |
| |
| package fsnotify |
| |
| import ( |
| "errors" |
| "fmt" |
| "os" |
| "strings" |
| "sync" |
| "syscall" |
| "unsafe" |
| ) |
| |
| type FileEvent struct { |
| mask uint32 // Mask of events |
| cookie uint32 // Unique cookie associating related events (for rename(2)) |
| Name string // File name (optional) |
| } |
| |
| // IsCreate reports whether the FileEvent was triggerd by a creation |
| func (e *FileEvent) IsCreate() bool { |
| return (e.mask&IN_CREATE) == IN_CREATE || (e.mask&IN_MOVED_TO) == IN_MOVED_TO |
| } |
| |
| // IsDelete reports whether the FileEvent was triggerd by a delete |
| func (e *FileEvent) IsDelete() bool { |
| return (e.mask&IN_DELETE_SELF) == IN_DELETE_SELF || (e.mask&IN_DELETE) == IN_DELETE |
| } |
| |
| // IsModify reports whether the FileEvent was triggerd by a file modification or attribute change |
| func (e *FileEvent) IsModify() bool { |
| return ((e.mask&IN_MODIFY) == IN_MODIFY || (e.mask&IN_ATTRIB) == IN_ATTRIB) |
| } |
| |
| // IsRename reports whether the FileEvent was triggerd by a change name |
| func (e *FileEvent) IsRename() bool { |
| return ((e.mask&IN_MOVE_SELF) == IN_MOVE_SELF || (e.mask&IN_MOVED_FROM) == IN_MOVED_FROM) |
| } |
| |
| type watch struct { |
| wd uint32 // Watch descriptor (as returned by the inotify_add_watch() syscall) |
| flags uint32 // inotify flags of this watch (see inotify(7) for the list of valid flags) |
| } |
| |
| type Watcher struct { |
| mu sync.Mutex // Map access |
| fd int // File descriptor (as returned by the inotify_init() syscall) |
| watches map[string]*watch // Map of inotify watches (key: path) |
| fsnFlags map[string]uint32 // Map of watched files to flags used for filter |
| fsnmut sync.Mutex // Protects access to fsnFlags. |
| paths map[int]string // Map of watched paths (key: watch descriptor) |
| Error chan error // Errors are sent on this channel |
| internalEvent chan *FileEvent // Events are queued on this channel |
| Event chan *FileEvent // Events are returned on this channel |
| done chan bool // Channel for sending a "quit message" to the reader goroutine |
| isClosed bool // Set to true when Close() is first called |
| } |
| |
| // NewWatcher creates and returns a new inotify instance using inotify_init(2) |
| func NewWatcher() (*Watcher, error) { |
| fd, errno := syscall.InotifyInit() |
| if fd == -1 { |
| return nil, os.NewSyscallError("inotify_init", errno) |
| } |
| w := &Watcher{ |
| fd: fd, |
| watches: make(map[string]*watch), |
| fsnFlags: make(map[string]uint32), |
| paths: make(map[int]string), |
| internalEvent: make(chan *FileEvent), |
| Event: make(chan *FileEvent), |
| Error: make(chan error), |
| done: make(chan bool, 1), |
| } |
| |
| go w.readEvents() |
| go w.purgeEvents() |
| return w, nil |
| } |
| |
| // Close closes an inotify watcher instance |
| // It sends a message to the reader goroutine to quit and removes all watches |
| // associated with the inotify instance |
| func (w *Watcher) Close() error { |
| if w.isClosed { |
| return nil |
| } |
| w.isClosed = true |
| |
| // Remove all watches |
| for path := range w.watches { |
| w.RemoveWatch(path) |
| } |
| |
| // Send "quit" message to the reader goroutine |
| w.done <- true |
| |
| return nil |
| } |
| |
| // AddWatch adds path to the watched file set. |
| // The flags are interpreted as described in inotify_add_watch(2). |
| func (w *Watcher) addWatch(path string, flags uint32) error { |
| if w.isClosed { |
| return errors.New("inotify instance already closed") |
| } |
| |
| w.mu.Lock() |
| watchEntry, found := w.watches[path] |
| w.mu.Unlock() |
| if found { |
| watchEntry.flags |= flags |
| flags |= syscall.IN_MASK_ADD |
| } |
| wd, errno := syscall.InotifyAddWatch(w.fd, path, flags) |
| if wd == -1 { |
| return errno |
| } |
| |
| w.mu.Lock() |
| w.watches[path] = &watch{wd: uint32(wd), flags: flags} |
| w.paths[wd] = path |
| w.mu.Unlock() |
| |
| return nil |
| } |
| |
| // Watch adds path to the watched file set, watching all events. |
| func (w *Watcher) watch(path string) error { |
| return w.addWatch(path, OS_AGNOSTIC_EVENTS) |
| } |
| |
| // RemoveWatch removes path from the watched file set. |
| func (w *Watcher) removeWatch(path string) error { |
| w.mu.Lock() |
| defer w.mu.Unlock() |
| watch, ok := w.watches[path] |
| if !ok { |
| return errors.New(fmt.Sprintf("can't remove non-existent inotify watch for: %s", path)) |
| } |
| success, errno := syscall.InotifyRmWatch(w.fd, watch.wd) |
| if success == -1 { |
| return os.NewSyscallError("inotify_rm_watch", errno) |
| } |
| delete(w.watches, path) |
| return nil |
| } |
| |
| // readEvents reads from the inotify file descriptor, converts the |
| // received events into Event objects and sends them via the Event channel |
| func (w *Watcher) readEvents() { |
| var ( |
| buf [syscall.SizeofInotifyEvent * 4096]byte // Buffer for a maximum of 4096 raw events |
| n int // Number of bytes read with read() |
| errno error // Syscall errno |
| ) |
| |
| for { |
| n, errno = syscall.Read(w.fd, buf[0:]) |
| // See if there is a message on the "done" channel |
| var done bool |
| select { |
| case done = <-w.done: |
| default: |
| } |
| |
| // If EOF or a "done" message is received |
| if n == 0 || done { |
| syscall.Close(w.fd) |
| close(w.internalEvent) |
| close(w.Error) |
| return |
| } |
| |
| if n < 0 { |
| w.Error <- os.NewSyscallError("read", errno) |
| continue |
| } |
| if n < syscall.SizeofInotifyEvent { |
| w.Error <- errors.New("inotify: short read in readEvents()") |
| continue |
| } |
| |
| var offset uint32 = 0 |
| // We don't know how many events we just read into the buffer |
| // While the offset points to at least one whole event... |
| for offset <= uint32(n-syscall.SizeofInotifyEvent) { |
| // Point "raw" to the event in the buffer |
| raw := (*syscall.InotifyEvent)(unsafe.Pointer(&buf[offset])) |
| event := new(FileEvent) |
| event.mask = uint32(raw.Mask) |
| event.cookie = uint32(raw.Cookie) |
| nameLen := uint32(raw.Len) |
| // If the event happened to the watched directory or the watched file, the kernel |
| // doesn't append the filename to the event, but we would like to always fill the |
| // the "Name" field with a valid filename. We retrieve the path of the watch from |
| // the "paths" map. |
| w.mu.Lock() |
| event.Name = w.paths[int(raw.Wd)] |
| w.mu.Unlock() |
| watchedName := event.Name |
| if nameLen > 0 { |
| // Point "bytes" at the first byte of the filename |
| bytes := (*[syscall.PathMax]byte)(unsafe.Pointer(&buf[offset+syscall.SizeofInotifyEvent])) |
| // The filename is padded with NUL bytes. TrimRight() gets rid of those. |
| event.Name += "/" + strings.TrimRight(string(bytes[0:nameLen]), "\000") |
| } |
| |
| // Setup FSNotify flags (inherit from directory watch) |
| w.fsnmut.Lock() |
| fsnFlags := w.fsnFlags[watchedName] |
| _, fsnFound := w.fsnFlags[event.Name] |
| if !fsnFound { |
| w.fsnFlags[event.Name] = fsnFlags |
| } |
| w.fsnmut.Unlock() |
| |
| // Send the events that are not ignored on the events channel |
| if (event.mask & IN_IGNORED) == 0 { |
| w.internalEvent <- event |
| } |
| |
| // Move to the next event in the buffer |
| offset += syscall.SizeofInotifyEvent + nameLen |
| } |
| } |
| } |
| |
| const ( |
| // Options for inotify_init() are not exported |
| // IN_CLOEXEC uint32 = syscall.IN_CLOEXEC |
| // IN_NONBLOCK uint32 = syscall.IN_NONBLOCK |
| |
| // Options for AddWatch |
| IN_DONT_FOLLOW uint32 = syscall.IN_DONT_FOLLOW |
| IN_ONESHOT uint32 = syscall.IN_ONESHOT |
| IN_ONLYDIR uint32 = syscall.IN_ONLYDIR |
| |
| // The "IN_MASK_ADD" option is not exported, as AddWatch |
| // adds it automatically, if there is already a watch for the given path |
| // IN_MASK_ADD uint32 = syscall.IN_MASK_ADD |
| |
| // Events |
| IN_ACCESS uint32 = syscall.IN_ACCESS |
| IN_ALL_EVENTS uint32 = syscall.IN_ALL_EVENTS |
| IN_ATTRIB uint32 = syscall.IN_ATTRIB |
| IN_CLOSE uint32 = syscall.IN_CLOSE |
| IN_CLOSE_NOWRITE uint32 = syscall.IN_CLOSE_NOWRITE |
| IN_CLOSE_WRITE uint32 = syscall.IN_CLOSE_WRITE |
| IN_CREATE uint32 = syscall.IN_CREATE |
| IN_DELETE uint32 = syscall.IN_DELETE |
| IN_DELETE_SELF uint32 = syscall.IN_DELETE_SELF |
| IN_MODIFY uint32 = syscall.IN_MODIFY |
| IN_MOVE uint32 = syscall.IN_MOVE |
| IN_MOVED_FROM uint32 = syscall.IN_MOVED_FROM |
| IN_MOVED_TO uint32 = syscall.IN_MOVED_TO |
| IN_MOVE_SELF uint32 = syscall.IN_MOVE_SELF |
| IN_OPEN uint32 = syscall.IN_OPEN |
| |
| OS_AGNOSTIC_EVENTS = IN_MOVED_TO | IN_MOVED_FROM | IN_CREATE | IN_ATTRIB | IN_MODIFY | IN_MOVE_SELF | IN_DELETE | IN_DELETE_SELF |
| |
| // Special events |
| IN_ISDIR uint32 = syscall.IN_ISDIR |
| IN_IGNORED uint32 = syscall.IN_IGNORED |
| IN_Q_OVERFLOW uint32 = syscall.IN_Q_OVERFLOW |
| IN_UNMOUNT uint32 = syscall.IN_UNMOUNT |
| ) |