source/modules/vfs_notify_fam.c

   1 /*
   2  * FAM file notification support.
   3  *
   4  * Copyright (c) James Peach 2005
   5  * Copyright (c) Volker Lendecke 2007
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 3 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include "includes.h"
  22
  23 #include <fam.h>
  24
  25 #if !defined(HAVE_FAM_H_FAMCODES_TYPEDEF)
  26 /* Gamin provides this typedef which means we can't use 'enum FAMCodes' as per
  27  * every other FAM implementation. Phooey.
  28  */
  29 typedef enum FAMCodes FAMCodes;
  30 #endif
  31
  32 /* NOTE: There are multiple versions of FAM floating around the net, each with
  33  * slight differences from the original SGI FAM implementation. In this file,
  34  * we rely only on the SGI features and do not assume any extensions. For
  35  * example, we do not look at FAMErrno, because it is not set by the original
  36  * implementation.
  37  *
  38  * Random FAM links:
  39  *      http://oss.sgi.com/projects/fam/
  40  *      http://savannah.nongnu.org/projects/fam/
  41  *      http://sourceforge.net/projects/bsdfam/
  42  */
  43
  44 /* ------------------------------------------------------------------------- */
  45
  46 struct fam_watch_context {
  47         struct fam_watch_context *prev, *next;
  48         FAMConnection *fam_connection;
  49         struct FAMRequest fr;
  50         struct sys_notify_context *sys_ctx;
  51         void (*callback)(struct sys_notify_context *ctx,
  52                          void *private_data,
  53                          struct notify_event *ev);
  54         void *private_data;
  55         uint32_t mask; /* the inotify mask */
  56         uint32_t filter; /* the windows completion filter */
  57         const char *path;
  58 };
  59
  60
  61 /*
  62  * We want one FAM connection per smbd, not one per tcon.
  63  */
  64 static FAMConnection fam_connection;
  65 static bool fam_connection_initialized = False;
  66
  67 static struct fam_watch_context *fam_notify_list;
  68 static void fam_handler(struct event_context *event_ctx,
  69                         struct fd_event *fd_event,
  70                         uint16 flags,
  71                         void *private_data);
  72
  73 static NTSTATUS fam_open_connection(FAMConnection *fam_conn,
  74                                     struct event_context *event_ctx)
  75 {
  76         int res;
  77         char *name;
  78
  79         ZERO_STRUCTP(fam_conn);
  80         FAMCONNECTION_GETFD(fam_conn) = -1;
  81
  82         if (asprintf(&name, "smbd (%lu)", (unsigned long)sys_getpid()) == -1) {
  83                 DEBUG(0, ("No memory\n"));
  84                 return NT_STATUS_NO_MEMORY;
  85         }
  86
  87         res = FAMOpen2(fam_conn, name);
  88         SAFE_FREE(name);
  89
  90         if (res < 0) {
  91                 DEBUG(10, ("FAM file change notifications not available\n"));
  92                 /*
  93                  * No idea how to get NT_STATUS from a FAM result
  94                  */
  95                 FAMCONNECTION_GETFD(fam_conn) = -1;
  96                 return NT_STATUS_UNEXPECTED_IO_ERROR;
  97         }
  98
  99         if (event_add_fd(event_ctx, event_ctx,
 100                          FAMCONNECTION_GETFD(fam_conn),
 101                          EVENT_FD_READ, fam_handler,
 102                          (void *)fam_conn) == NULL) {
 103                 DEBUG(0, ("event_add_fd failed\n"));
 104                 FAMClose(fam_conn);
 105                 FAMCONNECTION_GETFD(fam_conn) = -1;
 106                 return NT_STATUS_NO_MEMORY;
 107         }
 108
 109         return NT_STATUS_OK;
 110 }
 111
 112 static void fam_reopen(FAMConnection *fam_conn,
 113                        struct event_context *event_ctx,
 114                        struct fam_watch_context *notify_list)
 115 {
 116         struct fam_watch_context *ctx;
 117
 118         DEBUG(5, ("Re-opening FAM connection\n"));
 119
 120         FAMClose(fam_conn);
 121
 122         if (!NT_STATUS_IS_OK(fam_open_connection(fam_conn, event_ctx))) {
 123                 DEBUG(5, ("Re-opening fam connection failed\n"));
 124                 return;
 125         }
 126
 127         for (ctx = notify_list; ctx; ctx = ctx->next) {
 128                 FAMMonitorDirectory(fam_conn, ctx->path, &ctx->fr, NULL);
 129         }
 130 }
 131
 132 static void fam_handler(struct event_context *event_ctx,
 133                         struct fd_event *fd_event,
 134                         uint16 flags,
 135                         void *private_data)
 136 {
 137         FAMConnection *fam_conn = (FAMConnection *)private_data;
 138         FAMEvent fam_event;
 139         struct fam_watch_context *ctx;
 140         struct notify_event ne;
 141
 142         if (FAMPending(fam_conn) == 0) {
 143                 DEBUG(10, ("fam_handler called but nothing pending\n"));
 144                 return;
 145         }
 146
 147         if (FAMNextEvent(fam_conn, &fam_event) != 1) {
 148                 DEBUG(5, ("FAMNextEvent returned an error\n"));
 149                 TALLOC_FREE(fd_event);
 150                 fam_reopen(fam_conn, event_ctx, fam_notify_list);
 151                 return;
 152         }
 153
 154         DEBUG(10, ("Got FAMCode %d for %s\n", fam_event.code,
 155                    fam_event.filename));
 156
 157         switch (fam_event.code) {
 158         case FAMCreated:
 159                 ne.action = NOTIFY_ACTION_ADDED;
 160                 break;
 161         case FAMDeleted:
 162                 ne.action = NOTIFY_ACTION_REMOVED;
 163                 break;
 164         default:
 165                 DEBUG(10, ("Ignoring code FAMCode %d for file %s\n",
 166                            (int)fam_event.code, fam_event.filename));
 167                 return;
 168         }
 169
 170         for (ctx = fam_notify_list; ctx; ctx = ctx->next) {
 171                 if (memcmp(&fam_event.fr, &ctx->fr, sizeof(FAMRequest)) == 0) {
 172                         break;
 173                 }
 174         }
 175
 176         if (ctx == NULL) {
 177                 DEBUG(5, ("Discarding event for file %s\n",
 178                           fam_event.filename));
 179                 return;
 180         }
 181
 182         if ((ne.path = strrchr_m(fam_event.filename, '\\')) == NULL) {
 183                 ne.path = fam_event.filename;
 184         }
 185
 186         ctx->callback(ctx->sys_ctx, ctx->private_data, &ne);
 187 }
 188
 189 static int fam_watch_context_destructor(struct fam_watch_context *ctx)
 190 {
 191         if (FAMCONNECTION_GETFD(ctx->fam_connection) != -1) {
 192                 FAMCancelMonitor(&fam_connection, &ctx->fr);
 193         }
 194         DLIST_REMOVE(fam_notify_list, ctx);
 195         return 0;
 196 }
 197
 198 /*
 199   add a watch. The watch is removed when the caller calls
 200   talloc_free() on *handle
 201 */
 202 static NTSTATUS fam_watch(vfs_handle_struct *vfs_handle,
 203                           struct sys_notify_context *ctx,
 204                           struct notify_entry *e,
 205                           void (*callback)(struct sys_notify_context *ctx,
 206                                            void *private_data,
 207                                            struct notify_event *ev),
 208                           void *private_data,
 209                           void *handle_p)
 210 {
 211         const uint32 fam_mask = (FILE_NOTIFY_CHANGE_FILE_NAME|
 212                                  FILE_NOTIFY_CHANGE_DIR_NAME);
 213         struct fam_watch_context *watch;
 214         void **handle = (void **)handle_p;
 215
 216         if ((e->filter & fam_mask) == 0) {
 217                 DEBUG(10, ("filter = %u, ignoring in FAM\n", e->filter));
 218                 return NT_STATUS_OK;
 219         }
 220
 221         if (!fam_connection_initialized) {
 222                 if (!NT_STATUS_IS_OK(fam_open_connection(&fam_connection,
 223                                                          ctx->ev))) {
 224                         /*
 225                          * Just let smbd do all the work itself
 226                          */
 227                         return NT_STATUS_OK;
 228                 }
 229                 fam_connection_initialized = True;
 230         }
 231
 232         if (!(watch = TALLOC_P(ctx, struct fam_watch_context))) {
 233                 return NT_STATUS_NO_MEMORY;
 234         }
 235
 236         watch->fam_connection = &fam_connection;
 237
 238         watch->callback = callback;
 239         watch->private_data = private_data;
 240         watch->sys_ctx = ctx;
 241
 242         if (!(watch->path = talloc_strdup(watch, e->path))) {
 243                 DEBUG(0, ("talloc_asprintf failed\n"));
 244                 TALLOC_FREE(watch);
 245                 return NT_STATUS_NO_MEMORY;
 246         }
 247
 248         /*
 249          * The FAM module in this early state will only take care of
 250          * FAMCreated and FAMDeleted events, Leave the rest to
 251          * notify_internal.c
 252          */
 253
 254         watch->filter = fam_mask;
 255         e->filter &= ~fam_mask;
 256
 257         DLIST_ADD(fam_notify_list, watch);
 258         talloc_set_destructor(watch, fam_watch_context_destructor);
 259
 260         /*
 261          * Only directories monitored so far
 262          */
 263
 264         if (FAMCONNECTION_GETFD(watch->fam_connection) != -1) {
 265                 FAMMonitorDirectory(watch->fam_connection, watch->path,
 266                                     &watch->fr, NULL);
 267         }
 268         else {
 269                 /*
 270                  * If the re-open is successful, this will establish the
 271                  * FAMMonitor from the list
 272                  */
 273                 fam_reopen(watch->fam_connection, ctx->ev, fam_notify_list);
 274         }
 275
 276         *handle = (void *)watch;
 277
 278         return NT_STATUS_OK;
 279 }
 280
 281 /* VFS operations structure */
 282
 283 static vfs_op_tuple notify_fam_op_tuples[] = {
 284
 285         {SMB_VFS_OP(fam_watch),
 286          SMB_VFS_OP_NOTIFY_WATCH,
 287          SMB_VFS_LAYER_OPAQUE},
 288
 289         {SMB_VFS_OP(NULL),
 290          SMB_VFS_OP_NOOP,
 291          SMB_VFS_LAYER_NOOP}
 292
 293 };
 294
 295
 296 NTSTATUS vfs_notify_fam_init(void);
 297 NTSTATUS vfs_notify_fam_init(void)
 298 {
 299         return smb_register_vfs(SMB_VFS_INTERFACE_VERSION, "notify_fam",
 300                                 notify_fam_op_tuples);
 301 }