2008-12-19 Rolf Bjarne Kvinge <RKvinge@novell.com>
[mono.git] / mono / io-layer / daemon.c
index 80c6b956ebfb298aa7aa5dfb8c451da82911a5d9..0f5ad38ab882c7136f58c7e02e21ca1e4594c979 100644 (file)
@@ -62,9 +62,24 @@ static int main_sock;
 /* Set to TRUE by the SIGCHLD signal handler */
 static volatile gboolean check_processes=FALSE;
 
+/* The file_share_hash is used to emulate the windows file sharing mode */
+typedef struct _share_key
+{
+       dev_t device;
+       ino_t inode;
+} ShareKey;
+
+typedef struct _share_data
+{
+       guint32 sharemode;
+       guint32 access;
+} ShareData;
+
+static GHashTable *file_share_hash = NULL;
+
 static gboolean fd_activity (GIOChannel *channel, GIOCondition condition,
                             gpointer data);
-
+static void check_sharing (dev_t device, ino_t inode);
 
 /* Deletes the shared memory segment.  If we're exiting on error,
  * clients will get EPIPEs.
@@ -194,10 +209,27 @@ static void sigchld_handler (int unused)
        check_processes=TRUE;
 }
 
+static guint sharedata_hash (gconstpointer key)
+{
+       ShareKey *sharekey = (ShareKey *)key;
+       
+       return(g_int_hash (&(sharekey->inode)));
+}
+
+static gboolean sharedata_equal (gconstpointer a, gconstpointer b)
+{
+       ShareKey *share_a = (ShareKey *)a;
+       ShareKey *share_b = (ShareKey *)b;
+       
+       return(share_a->device == share_b->device &&
+              share_a->inode == share_b->inode);
+}
+
 /*
  * startup:
  *
- * Bind signals and attach to shared memory
+ * Bind signals, attach to shared memory and set up any internal data
+ * structures needed.
  */
 static void startup (void)
 {
@@ -235,6 +267,10 @@ static void startup (void)
                  "mono-handle-daemon-%d-%d-%ld", getuid (), getpid (),
                  time (NULL));
 #endif
+
+       file_share_hash = g_hash_table_new_full (sharedata_hash,
+                                                sharedata_equal, g_free,
+                                                g_free);
 }
 
 
@@ -304,12 +340,14 @@ static gboolean unref_handle (ChannelData *channel_data, guint32 handle)
                   channel_data->open_handles[handle]);
 #endif
 
-       if(channel_data->open_handles[handle]==0) {
+       if (_wapi_shared_data[segment]->handles[idx].ref == 0) {
+               gboolean was_file;
+               dev_t device = 0;
+               ino_t inode = 0;
+               
                /* This client has released the handle */
                destroy=TRUE;
-       }
        
-       if(_wapi_shared_data[segment]->handles[idx].ref==0) {
                if (channel_data->open_handles[handle]!=0) {
                        g_warning (G_GNUC_PRETTY_FUNCTION ": per-process open_handles mismatch, set to %d, should be 0", 
                                        channel_data->open_handles[handle]);
@@ -319,6 +357,30 @@ static gboolean unref_handle (ChannelData *channel_data, guint32 handle)
                g_message (G_GNUC_PRETTY_FUNCTION ": Destroying handle 0x%x",
                           handle);
 #endif
+
+               /* if this was a file handle, save the device and
+                * inode numbers so we can scan the share info data
+                * later to see if the last handle to a file has been
+                * closed, and delete the data if so.
+                */
+               was_file = (_wapi_shared_data[segment]->handles[idx].type == WAPI_HANDLE_FILE);
+               if (was_file) {
+                       struct _WapiHandle_file *file_handle;
+                       gboolean ok;
+                       
+                       ok = _wapi_lookup_handle (GUINT_TO_POINTER (handle),
+                                                 WAPI_HANDLE_FILE,
+                                                 (gpointer *)&file_handle,
+                                                 NULL);
+                       if (ok == FALSE) {
+                               g_warning (G_GNUC_PRETTY_FUNCTION
+                                          ": error looking up file handle %x",
+                                          handle);
+                       } else {
+                               device = file_handle->device;
+                               inode = file_handle->inode;
+                       }
+               }
                
                _wapi_handle_ops_close_shared (GUINT_TO_POINTER (handle));
                
@@ -329,6 +391,10 @@ static gboolean unref_handle (ChannelData *channel_data, guint32 handle)
 
                memset (&_wapi_shared_data[segment]->handles[idx].u, '\0', sizeof(_wapi_shared_data[segment]->handles[idx].u));
                _wapi_shared_data[segment]->handles[idx].type=WAPI_HANDLE_UNUSED;
+
+               if (was_file) {
+                       check_sharing (device, inode);
+               }
        }
 
        if(channel_data == daemon_channel_data) {
@@ -425,6 +491,14 @@ static void rem_fd(GIOChannel *channel, ChannelData *channel_data)
        g_message (G_GNUC_PRETTY_FUNCTION ": Removing client fd %d", fd);
 #endif
 
+       if (channel_data->io_source == 0) {
+#ifdef DEBUG
+               g_message (G_GNUC_PRETTY_FUNCTION ": channel already closed for fd %d", fd);
+#endif
+               return;
+       }
+
+
        g_io_channel_shutdown (channel, TRUE, NULL);
        g_source_remove (channel_data->io_source);
        g_io_channel_unref (channel);
@@ -458,6 +532,104 @@ static void rem_fd(GIOChannel *channel, ChannelData *channel_data)
        }
 }
 
+static void sharemode_set (dev_t device, ino_t inode, guint32 sharemode,
+                          guint32 access)
+{
+       ShareKey *sharekey;
+       ShareData *sharedata;
+       
+       sharekey = g_new (ShareKey, 1);
+       sharekey->device = device;
+       sharekey->inode = inode;
+
+       sharedata = g_new (ShareData, 1);
+       sharedata->sharemode = sharemode;
+       sharedata->access = access;
+       
+       /* Setting share mode to include all access bits is really
+        * removing the share info
+        */
+       if (sharemode == (FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE)) {
+               g_hash_table_remove (file_share_hash, sharekey);
+       } else {
+               g_hash_table_insert (file_share_hash, sharekey, sharedata);
+       }
+}
+
+static gboolean sharemode_get (dev_t device, ino_t inode, guint32 *sharemode,
+                              guint32 *access)
+{
+       ShareKey sharekey;
+       ShareData *sharedata;
+       
+       sharekey.device = device;
+       sharekey.inode = inode;
+       
+       sharedata = (ShareData *)g_hash_table_lookup (file_share_hash,
+                                                      &sharekey);
+       if (sharedata == NULL) {
+               return(FALSE);
+       }
+       
+       *sharemode = sharedata->sharemode;
+       *access = sharedata->access;
+       
+       return(TRUE);
+}
+
+static gboolean share_compare (gpointer handle, gpointer user_data)
+{
+       struct _WapiHandle_file *file_handle;
+       gboolean ok;
+       ShareKey *sharekey = (ShareKey *)user_data;
+       
+       ok = _wapi_lookup_handle (handle, WAPI_HANDLE_FILE,
+                                 (gpointer *)&file_handle, NULL);
+       if (ok == FALSE) {
+               g_warning (G_GNUC_PRETTY_FUNCTION
+                          ": error looking up file handle %p", handle);
+               return(FALSE);
+       }
+       
+       if (file_handle->device == sharekey->device &&
+           file_handle->inode == sharekey->inode) {
+#ifdef DEBUG
+               g_message (G_GNUC_PRETTY_FUNCTION ": found one, handle %p",
+                          handle);
+#endif
+               return(TRUE);
+       } else {
+               return(FALSE);
+       }
+}
+
+static void check_sharing (dev_t device, ino_t inode)
+{
+       ShareKey sharekey;
+       gpointer file_handle;
+       
+#ifdef DEBUG
+       g_message (G_GNUC_PRETTY_FUNCTION ": Checking if anything has (dev 0x%llx, inode %lld) still open", device, inode);
+#endif
+
+       sharekey.device = device;
+       sharekey.inode = inode;
+       
+       file_handle = _wapi_search_handle (WAPI_HANDLE_FILE, share_compare,
+                                          &sharekey, NULL, NULL);
+
+       if (file_handle == NULL) {
+               /* Delete this share info, as the last handle to it
+                * has been closed
+                */
+#ifdef DEBUG
+               g_message (G_GNUC_PRETTY_FUNCTION ": Deleting share data for (dev 0x%llx inode %lld)", device, inode);
+#endif
+               
+               g_hash_table_remove (file_share_hash, &sharekey);
+       }
+}
+
 static gboolean process_compare (gpointer handle, gpointer user_data)
 {
        struct _WapiHandle_process *process_handle;
@@ -571,11 +743,14 @@ static void process_post_mortem (pid_t pid, int status)
                           WEXITSTATUS (status));
 #endif
                
-               /* Technically WEXITSTATUS is only valid if the
-                * process exited normally, but I don't care if the
-                * process caught a signal or not.
+               /* If the child terminated due to the receipt of a signal,
+                * the exit status must be based on WTERMSIG, since WEXITSTATUS
+                * returns 0 in this case.
                 */
-               process_handle_data->exitstatus=WEXITSTATUS (status);
+               if (WIFSIGNALED(status))
+                       process_handle_data->exitstatus=128 + WTERMSIG (status);
+               else
+                       process_handle_data->exitstatus=WEXITSTATUS (status);
 
                /* Ignore errors */
                gettimeofday (&tv, NULL);
@@ -655,23 +830,11 @@ static void send_reply (GIOChannel *channel, WapiHandleResponse *resp)
        _wapi_daemon_response (g_io_channel_unix_get_fd (channel), resp);
 }
 
-/*
- * process_new:
- * @channel: The client making the request
- * @channel_data: Our data for this channel
- * @type: type to init handle to
- *
- * Find a free handle and initialize it to 'type', increase refcnt and
- * send back a reply to the client.
- */
-static void process_new (GIOChannel *channel, ChannelData *channel_data,
-                        WapiHandleType type)
+static guint32 new_handle_with_shared_check (WapiHandleType type)
 {
-       guint32 handle;
-       WapiHandleResponse resp={0};
-       
-       handle=_wapi_handle_new_internal (type);
-       if(handle==0) {
+       guint32 handle = 0;
+
+       while ((handle = _wapi_handle_new_internal (type)) == 0) {
                /* Try and allocate a new shared segment, and have
                 * another go
                 */
@@ -696,15 +859,34 @@ static void process_new (GIOChannel *channel, ChannelData *channel_data,
                                        channels[i].open_handles=_wapi_g_renew0 (channels[i].open_handles, old_len, new_len);
                                }
                        }
-
-                       handle=_wapi_handle_new_internal (type);
                } else {
                        /* Map failed.  Just return 0 meaning "out of
                         * handles"
                         */
+                       break;
                }
        }
        
+       return(handle);
+}
+
+/*
+ * process_new:
+ * @channel: The client making the request
+ * @channel_data: Our data for this channel
+ * @type: type to init handle to
+ *
+ * Find a free handle and initialize it to 'type', increase refcnt and
+ * send back a reply to the client.
+ */
+static void process_new (GIOChannel *channel, ChannelData *channel_data,
+                        WapiHandleType type)
+{
+       guint32 handle;
+       WapiHandleResponse resp={0};
+       
+       handle = new_handle_with_shared_check (type);
+       
        /* handle might still be set to 0.  This is handled at the
         * client end
         */
@@ -832,10 +1014,39 @@ static void process_scratch_free (GIOChannel *channel, guint32 scratch_idx)
        send_reply (channel, &resp);
 }
 
+/*
+ * process_process_kill:
+ * @channel: The client making the request
+ * @process_kill: pid and signal to send to the pid.
+ *
+ * Sends the specified signal to the process.
+ */
+static void
+process_process_kill (GIOChannel *channel,
+                     WapiHandleRequest_ProcessKill process_kill)
+{
+       WapiHandleResponse resp = {0};
+
+       resp.type = WapiHandleResponseType_ProcessKill;
+
+#ifdef DEBUG
+       g_message (G_GNUC_PRETTY_FUNCTION ": kill (%d, %d)",
+                  process_kill.pid, process_kill.signo);
+#endif
+       if (kill (process_kill.pid, process_kill.signo) == -1) {
+               resp.u.process_kill.err = errno;
+#ifdef DEBUG
+       g_message (G_GNUC_PRETTY_FUNCTION ": kill (%d, %d) failed: %d",
+                  process_kill.pid, process_kill.signo, resp.u.process_kill.err);
+#endif
+       }
+
+       send_reply (channel, &resp);
+}
+
 /*
  * process_process_fork:
  * @channel: The client making the request
- * @open_handles: An array of handles referenced by this client
  * @process_fork: Describes the process to fork
  * @fds: stdin, stdout, and stderr for the new process
  *
@@ -850,7 +1061,7 @@ static void process_process_fork (GIOChannel *channel, ChannelData *channel_data
        guint32 process_handle, thread_handle;
        struct _WapiHandle_process *process_handle_data;
        struct _WapiHandle_thread *thread_handle_data;
-       pid_t pid;
+       pid_t pid = 0;
        
        resp.type=WapiHandleResponseType_ProcessFork;
        
@@ -860,11 +1071,11 @@ static void process_process_fork (GIOChannel *channel, ChannelData *channel_data
         * client must check if either handle is 0 and take
         * appropriate error handling action.
         */
-       process_handle=_wapi_handle_new_internal (WAPI_HANDLE_PROCESS);
+       process_handle = new_handle_with_shared_check (WAPI_HANDLE_PROCESS);
        ref_handle (daemon_channel_data, process_handle);
        ref_handle (channel_data, process_handle);
        
-       thread_handle=_wapi_handle_new_internal (WAPI_HANDLE_THREAD);
+       thread_handle = new_handle_with_shared_check (WAPI_HANDLE_THREAD);
        ref_handle (daemon_channel_data, thread_handle);
        ref_handle (channel_data, thread_handle);
        
@@ -893,6 +1104,16 @@ static void process_process_fork (GIOChannel *channel, ChannelData *channel_data
                cmd=_wapi_handle_scratch_lookup (process_fork.cmd);
                dir=_wapi_handle_scratch_lookup (process_fork.dir);
                env=_wapi_handle_scratch_lookup_string_array (process_fork.env);
+
+               _wapi_lookup_handle (GUINT_TO_POINTER (process_handle),
+                                    WAPI_HANDLE_PROCESS,
+                                    (gpointer *)&process_handle_data,
+                                    NULL);
+
+               _wapi_lookup_handle (GUINT_TO_POINTER (thread_handle),
+                                    WAPI_HANDLE_THREAD,
+                                    (gpointer *)&thread_handle_data,
+                                    NULL);
                
                ret=g_shell_parse_argv (cmd, NULL, &argv, &gerr);
                if(ret==FALSE) {
@@ -905,16 +1126,6 @@ static void process_process_fork (GIOChannel *channel, ChannelData *channel_data
                        g_message (G_GNUC_PRETTY_FUNCTION ": forking");
 #endif
 
-                       _wapi_lookup_handle (GUINT_TO_POINTER (process_handle),
-                                            WAPI_HANDLE_PROCESS,
-                                            (gpointer *)&process_handle_data,
-                                            NULL);
-
-                       _wapi_lookup_handle (GUINT_TO_POINTER (thread_handle),
-                                            WAPI_HANDLE_THREAD,
-                                            (gpointer *)&thread_handle_data,
-                                            NULL);
-
                        /* Fork, exec cmd with args and optional env,
                         * and return the handles with pid and blank
                         * thread id
@@ -1003,12 +1214,20 @@ static void process_process_fork (GIOChannel *channel, ChannelData *channel_data
 
                /* store process name, based on the last section of the cmd */
                {
-                       char *slash=strrchr (argv[0], '/');
+                       char *slash;
                        
-                       if(slash!=NULL) {
-                               process_handle_data->proc_name=_wapi_handle_scratch_store (slash+1, strlen (slash+1));
+                       /* This should never fail, but it seems it can...
+                        */
+                       if (argv[0] != NULL) {
+                               slash=strrchr (argv[0], '/');
+                       
+                               if(slash!=NULL) {
+                                       process_handle_data->proc_name=_wapi_handle_scratch_store (slash+1, strlen (slash+1));
+                               } else {
+                                       process_handle_data->proc_name=_wapi_handle_scratch_store (argv[0], strlen (argv[0]));
+                               }
                        } else {
-                               process_handle_data->proc_name=_wapi_handle_scratch_store (argv[0], strlen (argv[0]));
+                               process_handle_data->proc_name = _wapi_handle_scratch_store (cmd, strlen(cmd));
                        }
                }
                
@@ -1042,22 +1261,90 @@ static void process_process_fork (GIOChannel *channel, ChannelData *channel_data
 
                resp.u.process_fork.pid=pid;
        }
-                       
+
        resp.u.process_fork.process_handle=process_handle;
        resp.u.process_fork.thread_handle=thread_handle;
 
        send_reply (channel, &resp);
 }
 
+/*
+ * process_set_share:
+ * @channel: The client making the request
+ * @channel_data: The channel data
+ * @set_share: Set share data passed from the client
+ *
+ * Sets file share info
+ */
+static void process_set_share (GIOChannel *channel, ChannelData *channel_data,
+                              WapiHandleRequest_SetShare set_share)
+{
+       WapiHandleResponse resp = {0};
+
+       resp.type = WapiHandleResponseType_SetShare;
+       
+#ifdef DEBUG
+       g_message (G_GNUC_PRETTY_FUNCTION ": Setting share for file (dev:0x%llx, ino:%lld) mode 0x%x access 0x%x", set_share.device, set_share.inode, set_share.sharemode, set_share.access);
+#endif
+       
+       sharemode_set (set_share.device, set_share.inode, set_share.sharemode,
+                      set_share.access);
+       
+       send_reply (channel, &resp);
+}
+
+/*
+ * process_get_or_set_share:
+ * @channel: The client making the request
+ * @channel_data: The channel data
+ * @get_share: GetOrSetShare data passed from the client
+ *
+ * Gets a file share status, and sets the status if it doesn't already
+ * exist
+ */
+static void process_get_or_set_share (GIOChannel *channel,
+                                     ChannelData *channel_data,
+                                     WapiHandleRequest_GetOrSetShare get_share)
+{
+       WapiHandleResponse resp = {0};
+       
+       resp.type = WapiHandleResponseType_GetOrSetShare;
+       
+#ifdef DEBUG
+       g_message (G_GNUC_PRETTY_FUNCTION
+                  ": Getting share status for file (dev:0x%llx, ino:%lld)",
+                  get_share.device, get_share.inode);
+#endif
+
+       resp.u.get_or_set_share.exists = sharemode_get (get_share.device, get_share.inode, &resp.u.get_or_set_share.sharemode, &resp.u.get_or_set_share.access);
+       
+       if (resp.u.get_or_set_share.exists) {
+#ifdef DEBUG
+               g_message (G_GNUC_PRETTY_FUNCTION ": Share mode: 0x%x",
+                          resp.u.get_or_set_share.sharemode);
+#endif
+       } else {
+#ifdef DEBUG
+               g_message (G_GNUC_PRETTY_FUNCTION
+                          ": file share info not already known, setting");
+#endif
+               sharemode_set (get_share.device, get_share.inode,
+                              get_share.new_sharemode, get_share.new_access);
+       }
+       
+       send_reply (channel, &resp);
+}
+
 /*
  * read_message:
  * @channel: The client to read the request from
  * @open_handles: An array of handles referenced by this client
  *
  * Read a message (A WapiHandleRequest) from a client and dispatch
- * whatever it wants to the process_* calls.
+ * whatever it wants to the process_* calls.  Return TRUE if the message
+ * was read successfully, FALSE otherwise.
  */
-static void read_message (GIOChannel *channel, ChannelData *channel_data)
+static gboolean read_message (GIOChannel *channel, ChannelData *channel_data)
 {
        WapiHandleRequest req;
        int fds[3]={0, 1, 2};
@@ -1074,7 +1361,7 @@ static void read_message (GIOChannel *channel, ChannelData *channel_data)
                           g_io_channel_unix_get_fd (channel));
 #endif
                rem_fd (channel, channel_data);
-               return;
+               return(FALSE);
        }
        
 #ifdef DEBUG
@@ -1106,6 +1393,16 @@ static void read_message (GIOChannel *channel, ChannelData *channel_data)
                process_process_fork (channel, channel_data,
                                      req.u.process_fork, fds);
                break;
+       case WapiHandleRequestType_ProcessKill:
+               process_process_kill (channel, req.u.process_kill);
+               break;
+       case WapiHandleRequestType_SetShare:
+               process_set_share (channel, channel_data, req.u.set_share);
+               break;
+       case WapiHandleRequestType_GetOrSetShare:
+               process_get_or_set_share (channel, channel_data,
+                                         req.u.get_or_set_share);
+               break;
        case WapiHandleRequestType_Error:
                /* fall through */
        default:
@@ -1125,6 +1422,8 @@ static void read_message (GIOChannel *channel, ChannelData *channel_data)
                close (fds[1]);
                close (fds[2]);
        }
+
+       return(TRUE);
 }
 
 /*
@@ -1176,7 +1475,7 @@ static gboolean fd_activity (GIOChannel *channel, GIOCondition condition,
                        g_message ("reading data on fd %d", fd);
 #endif
 
-                       read_message (channel, channel_data);
+                       return(read_message (channel, channel_data));
                }
                return(TRUE);
        }
@@ -1207,6 +1506,9 @@ void _wapi_daemon_main(gpointer data, gpointer scratch)
        /* Note that we've got the starting segment already */
        _wapi_shared_data[0]->num_segments=1;
        _wapi_shm_mapped_segments=1;
+
+       _wapi_fd_offset_table_size=getdtablesize ();
+       _wapi_shared_data[0]->fd_offset_table_size = _wapi_fd_offset_table_size;
        
        startup ();