dlls/mountmgr.sys/diskarb.c

   1 /*
   2  * Devices support using the MacOS Disk Arbitration library.
   3  *
   4  * Copyright 2006 Alexandre Julliard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #include <assert.h>
  25 #include <errno.h>
  26 #include <stdarg.h>
  27 #include <stdio.h>
  28 #include <sys/time.h>
  29
  30 #include "mountmgr.h"
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(mountmgr);
  34
  35 #ifdef HAVE_DISKARBITRATION_DISKARBITRATION_H
  36
  37 #include <DiskArbitration/DiskArbitration.h>
  38
  39 static void appeared_callback( DADiskRef disk, void *context )
  40 {
  41     CFDictionaryRef dict = DADiskCopyDescription( disk );
  42     const void *ref;
  43     char device[64];
  44     char mount_point[PATH_MAX];
  45     DWORD type = DRIVE_UNKNOWN;
  46
  47     if (!dict) return;
  48
  49     /* ignore non-removable devices */
  50     if (!(ref = CFDictionaryGetValue( dict, CFSTR("DAMediaRemovable") )) ||
  51         !CFBooleanGetValue( ref )) goto done;
  52
  53     /* get device name */
  54     if (!(ref = CFDictionaryGetValue( dict, CFSTR("DAMediaBSDName") ))) goto done;
  55     strcpy( device, "/dev/r" );
  56     CFStringGetCString( ref, device + 6, sizeof(device) - 6, kCFStringEncodingASCII );
  57
  58     if ((ref = CFDictionaryGetValue( dict, CFSTR("DAVolumePath") )))
  59         CFURLGetFileSystemRepresentation( ref, true, (UInt8 *)mount_point, sizeof(mount_point) );
  60     else
  61         mount_point[0] = 0;
  62
  63     if ((ref = CFDictionaryGetValue( dict, CFSTR("DAVolumeKind") )))
  64     {
  65         if (!CFStringCompare( ref, CFSTR("cd9660"), 0 ) ||
  66             !CFStringCompare( ref, CFSTR("udf"), 0 ))
  67             type = DRIVE_CDROM;
  68     }
  69
  70     TRACE( "got mount notification for '%s' on '%s'\n", device, mount_point );
  71
  72     add_dos_device( -1, device, device, mount_point, type );
  73 done:
  74     CFRelease( dict );
  75 }
  76
  77 static void changed_callback( DADiskRef disk, CFArrayRef keys, void *context )
  78 {
  79     appeared_callback( disk, context );
  80 }
  81
  82 static void disappeared_callback( DADiskRef disk, void *context )
  83 {
  84     CFDictionaryRef dict = DADiskCopyDescription( disk );
  85     const void *ref;
  86     char device[100];
  87
  88     if (!dict) return;
  89
  90     /* ignore non-removable devices */
  91     if (!(ref = CFDictionaryGetValue( dict, CFSTR("DAMediaRemovable") )) ||
  92         !CFBooleanGetValue( ref )) goto done;
  93
  94     /* get device name */
  95     if (!(ref = CFDictionaryGetValue( dict, CFSTR("DAMediaBSDName") ))) goto done;
  96     strcpy( device, "/dev/r" );
  97     CFStringGetCString( ref, device + 6, sizeof(device) - 6, kCFStringEncodingASCII );
  98
  99     TRACE( "got unmount notification for '%s'\n", device );
 100
 101     remove_dos_device( -1, device );
 102 done:
 103     CFRelease( dict );
 104 }
 105
 106 static DWORD WINAPI runloop_thread( void *arg )
 107 {
 108     DASessionRef session = DASessionCreate( NULL );
 109
 110     if (!session) return 1;
 111
 112     DASessionScheduleWithRunLoop( session, CFRunLoopGetCurrent(), kCFRunLoopDefaultMode );
 113     DARegisterDiskAppearedCallback( session, kDADiskDescriptionMatchVolumeMountable,
 114                                     appeared_callback, NULL );
 115     DARegisterDiskDisappearedCallback( session, kDADiskDescriptionMatchVolumeMountable,
 116                                        disappeared_callback, NULL );
 117     DARegisterDiskDescriptionChangedCallback( session, kDADiskDescriptionMatchVolumeMountable,
 118                                               kDADiskDescriptionWatchVolumePath, changed_callback, NULL );
 119     CFRunLoopRun();
 120     DASessionUnscheduleFromRunLoop( session, CFRunLoopGetCurrent(), kCFRunLoopDefaultMode );
 121     CFRelease( session );
 122     return 0;
 123 }
 124
 125 void initialize_diskarbitration(void)
 126 {
 127     HANDLE handle;
 128
 129     if (!(handle = CreateThread( NULL, 0, runloop_thread, NULL, 0, NULL ))) return;
 130     CloseHandle( handle );
 131 }
 132
 133 #else  /*  HAVE_DISKARBITRATION_DISKARBITRATION_H */
 134
 135 void initialize_diskarbitration(void)
 136 {
 137     TRACE( "Skipping, Disk Arbitration support not compiled in\n" );
 138 }
 139
 140 #endif  /* HAVE_DISKARBITRATION_DISKARBITRATION_H */