wined3d: Move an extension check into the state template.
[wine] / 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 "windef.h"
31 #include "winbase.h"
32 #include "winreg.h"
33 #include "winuser.h"
34
35 #include "wine/debug.h"
36 #include "mountmgr.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(mountmgr);
39
40 #ifdef HAVE_DISKARBITRATION_DISKARBITRATION_H
41
42 #include <DiskArbitration/DiskArbitration.h>
43
44 static void appeared_callback( DADiskRef disk, void *context )
45 {
46     CFDictionaryRef dict = DADiskCopyDescription( disk );
47     const void *ref;
48     char device[64];
49     char mount_point[PATH_MAX];
50     const char *type = NULL;
51
52     if (!dict) return;
53
54     /* ignore non-removable devices */
55     if (!(ref = CFDictionaryGetValue( dict, CFSTR("DAMediaRemovable") )) ||
56         !CFBooleanGetValue( ref )) goto done;
57
58     /* get device name */
59     if (!(ref = CFDictionaryGetValue( dict, CFSTR("DAMediaBSDName") ))) goto done;
60     strcpy( device, "/dev/r" );
61     CFStringGetCString( ref, device + 6, sizeof(device) - 6, kCFStringEncodingASCII );
62
63     if ((ref = CFDictionaryGetValue( dict, CFSTR("DAVolumePath") )))
64         CFURLGetFileSystemRepresentation( ref, true, (UInt8 *)mount_point, sizeof(mount_point) );
65     else
66         mount_point[0] = 0;
67
68     if ((ref = CFDictionaryGetValue( dict, CFSTR("DAVolumeKind") )))
69     {
70         if (!CFStringCompare( ref, CFSTR("cd9660"), 0 ) ||
71             !CFStringCompare( ref, CFSTR("udf"), 0 ))
72             type = "cdrom";
73     }
74
75     TRACE( "got mount notification for '%s' on '%s'\n", device, mount_point );
76
77     add_dos_device( device, device, mount_point, type );
78 done:
79     CFRelease( dict );
80 }
81
82 static void changed_callback( DADiskRef disk, CFArrayRef keys, void *context )
83 {
84     appeared_callback( disk, context );
85 }
86
87 static void disappeared_callback( DADiskRef disk, void *context )
88 {
89     CFDictionaryRef dict = DADiskCopyDescription( disk );
90     const void *ref;
91     char device[100];
92
93     if (!dict) return;
94
95     /* ignore non-removable devices */
96     if (!(ref = CFDictionaryGetValue( dict, CFSTR("DAMediaRemovable") )) ||
97         !CFBooleanGetValue( ref )) goto done;
98
99     /* get device name */
100     if (!(ref = CFDictionaryGetValue( dict, CFSTR("DAMediaBSDName") ))) goto done;
101     strcpy( device, "/dev/r" );
102     CFStringGetCString( ref, device + 6, sizeof(device) - 6, kCFStringEncodingASCII );
103
104     TRACE( "got unmount notification for '%s'\n", device );
105
106     remove_dos_device( device );
107 done:
108     CFRelease( dict );
109 }
110
111 static DWORD WINAPI runloop_thread( void *arg )
112 {
113     DASessionRef session = DASessionCreate( NULL );
114
115     if (!session) return 1;
116
117     DASessionScheduleWithRunLoop( session, CFRunLoopGetCurrent(), kCFRunLoopDefaultMode );
118     DARegisterDiskAppearedCallback( session, kDADiskDescriptionMatchVolumeMountable,
119                                     appeared_callback, NULL );
120     DARegisterDiskDisappearedCallback( session, kDADiskDescriptionMatchVolumeMountable,
121                                        disappeared_callback, NULL );
122     DARegisterDiskDescriptionChangedCallback( session, kDADiskDescriptionMatchVolumeMountable,
123                                               kDADiskDescriptionWatchVolumePath, changed_callback, NULL );
124     CFRunLoopRun();
125     DASessionUnscheduleFromRunLoop( session, CFRunLoopGetCurrent(), kCFRunLoopDefaultMode );
126     CFRelease( session );
127     return 0;
128 }
129
130 void initialize_diskarbitration(void)
131 {
132     HANDLE handle;
133
134     if (!(handle = CreateThread( NULL, 0, runloop_thread, NULL, 0, NULL ))) return;
135     CloseHandle( handle );
136 }
137
138 #else  /*  HAVE_DISKARBITRATION_DISKARBITRATION_H */
139
140 void initialize_diskarbitration(void)
141 {
142     TRACE( "Skipping, Disk Arbitration support not compiled in\n" );
143 }
144
145 #endif  /* HAVE_DISKARBITRATION_DISKARBITRATION_H */