Exemple #1
0
    def __init__(self, config):
        Thread.__init__(self)

        self._max_size = int(config['Settings']['max_file_size']) * 1024

        self._path = config['Settings']['dump_dir']
        self._buffer = win32file.AllocateReadBuffer(1024)
        self._overlapped = pywintypes.OVERLAPPED()
        self._overlapped.hEvent = win32event.CreateEvent(None, True, 0, None)
        self._overlapped.object = self._path
        self._stop_event = win32event.CreateEvent(None, True, 0, None)

        self._uploader = FileUploader(config)
Exemple #2
0
class DirMonitor(Thread):
    
    def __init__(self, config):
        Thread.__init__(self)

        self._max_size = int(config['Settings']['max_file_size']) * 1024

        self._path = config['Settings']['dump_dir']
        self._buffer = win32file.AllocateReadBuffer(1024)
        self._overlapped = pywintypes.OVERLAPPED()
        self._overlapped.hEvent = win32event.CreateEvent(None, True, 0, None)
        self._overlapped.object = self._path
        self._stop_event = win32event.CreateEvent(None, True, 0, None)

        self._uploader = FileUploader(config)

    def __del__(self):
        self._stop()

    def stop(self):
        win32event.SetEvent(self._stop_event)

    def _async_watch(self):
        win32file.ReadDirectoryChangesW(
            self._hDir,
            self._buffer,
            False,
            win32con.FILE_NOTIFY_CHANGE_FILE_NAME,
            self._overlapped)

    def run(self):
        logging.info('Monitoring \'{}\''.format(self._path))

        self._hDir = win32file.CreateFile(
            self._path,
            winnt.FILE_LIST_DIRECTORY,
            win32con.FILE_SHARE_READ |
                win32con.FILE_SHARE_WRITE |
                win32con.FILE_SHARE_DELETE,
            None,
            win32con.OPEN_EXISTING,
            win32con.FILE_FLAG_BACKUP_SEMANTICS |
                win32file.FILE_FLAG_OVERLAPPED,
            None)

        self._async_watch()

        while True:
            rc = win32event.WaitForMultipleObjects(
                [self._stop_event, self._overlapped.hEvent],
                False,
                1000)
            if rc == win32event.WAIT_TIMEOUT:
                continue
            if rc == win32event.WAIT_OBJECT_0:
                break

            bytes = win32file.GetOverlappedResult(
                self._hDir, 
                self._overlapped, 
                True)

            result = win32file.FILE_NOTIFY_INFORMATION(self._buffer, bytes)

            for action, name in result:
                filepath = os.path.join(self._path, name)
                if (action == winnt.FILE_ACTION_ADDED and
                        os.path.getsize(filepath) <= self._max_size):
                    self._uploader.upload(filepath)
            
            win32event.ResetEvent(self._overlapped.hEvent)
            self._async_watch()