1 /* GIO - GLib Input, Output and Streaming Library
2 *
3 * Copyright (C) 2006-2007 Red Hat, Inc.
4 *
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
9 *
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General
16 * Public License along with this library; if not, write to the
17 * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
18 * Boston, MA 02111-1307, USA.
19 *
20 * Author: Vlad Grecescu <b100dian@gmail.com>
21 *
22 */
23
24 #define _WIN32_WINNT 0x0400
25
26 #include "config.h"
27 #include "gwin32directorymonitor.h"
28 #include "giomodule.h"
29 #include <windows.h>
30
31 G_DEFINE_TYPE_WITH_CODE (GWin32DirectoryMonitor,
32 g_win32_directory_monitor,
33 G_TYPE_LOCAL_DIRECTORY_MONITOR,
34 g_io_extension_point_implement (G_LOCAL_DIRECTORY_MONITOR_EXTENSION_POINT_NAME,
35 g_define_type_id,
36 "readdirectorychanges",
37 20))
38
39 struct _GWin32DirectoryMonitorPrivate {
40 OVERLAPPED overlapped;
41 DWORD buffer_allocated_bytes;
42 gchar *file_notify_buffer;
43 DWORD buffer_filled_bytes;
44 HANDLE hDirectory;
45 /* Needed in the APC where we only have this private struct */
46 GFileMonitor *self;
47 };
48
49 static void g_win32_directory_monitor_finalize (GObject *base);
50 static gboolean g_win32_directory_monitor_cancel (GFileMonitor *base);
51
52 static GObject *g_win32_directory_monitor_constructor (GType type,
53 guint n_construct_properties,
54 GObjectConstructParam *construct_properties);
55
56 static gboolean
g_win32_directory_monitor_is_supported(void)57 g_win32_directory_monitor_is_supported (void)
58 {
59 return TRUE;
60 }
61
62 static void
g_win32_directory_monitor_finalize(GObject * base)63 g_win32_directory_monitor_finalize (GObject *base)
64 {
65 GWin32DirectoryMonitor *self;
66 self = G_WIN32_DIRECTORY_MONITOR (base);
67
68 g_free (self->priv->file_notify_buffer);
69 g_free (self->priv);
70
71 if (G_OBJECT_CLASS (g_win32_directory_monitor_parent_class)->finalize)
72 (*G_OBJECT_CLASS (g_win32_directory_monitor_parent_class)->finalize) (base);
73 }
74
75 static gboolean
g_win32_directory_monitor_cancel(GFileMonitor * base)76 g_win32_directory_monitor_cancel (GFileMonitor *base)
77 {
78 GWin32DirectoryMonitor *self;
79 self = G_WIN32_DIRECTORY_MONITOR (base);
80
81 /* This triggers a last callback() with nBytes=0 */
82 if (self->priv->hDirectory != INVALID_HANDLE_VALUE)
83 CloseHandle (self->priv->hDirectory);
84
85 if (G_FILE_MONITOR_CLASS (g_win32_directory_monitor_parent_class)->cancel)
86 (*G_FILE_MONITOR_CLASS (g_win32_directory_monitor_parent_class)->cancel) (base);
87 return TRUE;
88 }
89
90 static void CALLBACK
g_win32_directory_monitor_callback(DWORD error,DWORD nBytes,LPOVERLAPPED lpOverlapped)91 g_win32_directory_monitor_callback (DWORD error,
92 DWORD nBytes,
93 LPOVERLAPPED lpOverlapped)
94 {
95 gulong offset;
96 PFILE_NOTIFY_INFORMATION pfile_notify_walker;
97 gulong file_name_len;
98 gchar *file_name;
99 gchar *path;
100 GFile *file;
101 GWin32DirectoryMonitorPrivate *priv = (GWin32DirectoryMonitorPrivate *) lpOverlapped;
102
103 static GFileMonitorEvent events[] =
104 {
105 0,
106 G_FILE_MONITOR_EVENT_CREATED, /* FILE_ACTION_ADDED */
107 G_FILE_MONITOR_EVENT_DELETED, /* FILE_ACTION_REMOVED */
108 G_FILE_MONITOR_EVENT_CHANGED, /* FILE_ACTION_MODIFIED */
109 G_FILE_MONITOR_EVENT_DELETED, /* FILE_ACTION_RENAMED_OLD_NAME */
110 G_FILE_MONITOR_EVENT_CREATED, /* FILE_ACTION_RENAMED_NEW_NAME */
111 };
112
113 if (!nBytes) /* Monitor was cancelled/finalized */
114 return;
115
116 if (g_file_monitor_is_cancelled (G_FILE_MONITOR (priv->self)))
117 return; /* and ReadDirectoryChangesW doesn't get called this time */
118
119 offset = 0;
120 do {
121 pfile_notify_walker = (PFILE_NOTIFY_INFORMATION)(priv->file_notify_buffer + offset);
122 offset += pfile_notify_walker->NextEntryOffset;
123 file_name = g_utf16_to_utf8 (pfile_notify_walker->FileName, pfile_notify_walker->FileNameLength / sizeof(WCHAR), NULL, &file_name_len, NULL);
124 path = g_build_filename(G_LOCAL_DIRECTORY_MONITOR (priv->self)->dirname, file_name, NULL);
125 file = g_file_new_for_path (path);
126 g_file_monitor_emit_event (priv->self, file, NULL, events [pfile_notify_walker->Action]);
127 g_object_unref (file);
128 g_free (path);
129 g_free (file_name);
130 } while (pfile_notify_walker->NextEntryOffset);
131
132 ReadDirectoryChangesW (priv->hDirectory,
133 (gpointer)priv->file_notify_buffer,
134 priv->buffer_allocated_bytes,
135 FALSE,
136 FILE_NOTIFY_CHANGE_FILE_NAME |
137 FILE_NOTIFY_CHANGE_DIR_NAME |
138 FILE_NOTIFY_CHANGE_ATTRIBUTES |
139 FILE_NOTIFY_CHANGE_SIZE,
140 &priv->buffer_filled_bytes,
141 &priv->overlapped,
142 g_win32_directory_monitor_callback);
143 }
144
145 static GObject *
g_win32_directory_monitor_constructor(GType type,guint n_construct_properties,GObjectConstructParam * construct_properties)146 g_win32_directory_monitor_constructor (GType type,
147 guint n_construct_properties,
148 GObjectConstructParam *construct_properties) {
149 GObject *obj;
150 GWin32DirectoryMonitorClass *klass;
151 GObjectClass *parent_class;
152 GWin32DirectoryMonitor *self;
153 wchar_t *wdirname;
154 gboolean result;
155
156 klass = G_WIN32_DIRECTORY_MONITOR_CLASS (g_type_class_peek (G_TYPE_WIN32_DIRECTORY_MONITOR));
157 parent_class = G_OBJECT_CLASS (g_type_class_peek_parent (klass));
158 obj = parent_class->constructor (type, n_construct_properties, construct_properties);
159 self = G_WIN32_DIRECTORY_MONITOR (obj);
160 wdirname = g_utf8_to_utf16 (G_LOCAL_DIRECTORY_MONITOR (obj)->dirname, -1, NULL, NULL, NULL);
161
162 self->priv->hDirectory = CreateFileW (wdirname,
163 FILE_LIST_DIRECTORY,
164 FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
165
166 NULL,
167 OPEN_EXISTING,
168 FILE_FLAG_BACKUP_SEMANTICS | FILE_FLAG_OVERLAPPED,
169 NULL);
170 g_free (wdirname);
171 if (self->priv->hDirectory == INVALID_HANDLE_VALUE)
172 {
173 /* Ignore errors */
174 return obj;
175 }
176
177 result = ReadDirectoryChangesW (self->priv->hDirectory,
178 (gpointer)self->priv->file_notify_buffer,
179 self->priv->buffer_allocated_bytes,
180 FALSE,
181 FILE_NOTIFY_CHANGE_FILE_NAME |
182 FILE_NOTIFY_CHANGE_DIR_NAME |
183 FILE_NOTIFY_CHANGE_ATTRIBUTES |
184 FILE_NOTIFY_CHANGE_SIZE,
185 &self->priv->buffer_filled_bytes,
186 &self->priv->overlapped,
187 g_win32_directory_monitor_callback);
188 /* Ignore errors */
189
190 return obj;
191 }
192
193 static void
g_win32_directory_monitor_class_init(GWin32DirectoryMonitorClass * klass)194 g_win32_directory_monitor_class_init (GWin32DirectoryMonitorClass *klass)
195 {
196 g_win32_directory_monitor_parent_class = g_type_class_peek_parent (klass);
197
198 G_OBJECT_CLASS (klass)->constructor = g_win32_directory_monitor_constructor;
199 G_OBJECT_CLASS (klass)->finalize = g_win32_directory_monitor_finalize;
200 G_FILE_MONITOR_CLASS (klass)->cancel = g_win32_directory_monitor_cancel;
201
202 G_LOCAL_DIRECTORY_MONITOR_CLASS (klass)->mount_notify = FALSE;
203 G_LOCAL_DIRECTORY_MONITOR_CLASS (klass)->is_supported = g_win32_directory_monitor_is_supported;
204 }
205
206 static void
g_win32_directory_monitor_init(GWin32DirectoryMonitor * self)207 g_win32_directory_monitor_init (GWin32DirectoryMonitor *self)
208 {
209 self->priv = (GWin32DirectoryMonitorPrivate*)g_new0 (GWin32DirectoryMonitorPrivate, 1);
210 g_assert (self->priv != 0);
211
212 self->priv->buffer_allocated_bytes = 32768;
213 self->priv->file_notify_buffer = g_new0 (gchar, self->priv->buffer_allocated_bytes);
214 g_assert (self->priv->file_notify_buffer);
215
216 self->priv->self = G_FILE_MONITOR (self);
217 }
218