shlwapi: Convert SHPackDispParams to use an MS ABI vararg list for x86_64.
[wine] / dlls / wnaspi32 / aspi.c
1 /**************************************************************************
2  * ASPI routines
3  * Copyright (C) 2000 David Elliott <dfe@infinite-internet.net>
4  * Copyright (C) 2005 Vitaliy Margolen
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 /* These routines are to be called from either WNASPI32 or WINASPI */
22
23 /* FIXME:
24  * - No way to override automatic /proc detection, maybe provide an
25  *   HKEY_LOCAL_MACHINE\Software\Wine\Wine\Scsi regkey
26  * - Somewhat debating an #ifdef linux... technically all this code will
27  *   run on another UNIX.. it will fail nicely.
28  * - Please add support for mapping multiple channels on host adapters to
29  *   aspi controllers, e-mail me if you need help.
30  */
31
32
33 #include "config.h"
34 #include "wine/port.h"
35
36 #include <stdio.h>
37 #include <stdarg.h>
38 #include <sys/types.h>
39 #ifdef HAVE_SYS_IOCTL_H
40 #include <sys/ioctl.h>
41 #endif
42 #include <fcntl.h>
43 #ifdef HAVE_DIRENT_H
44 # include <dirent.h>
45 #endif
46 #ifdef HAVE_UNISTD_H
47 # include <unistd.h>
48 #endif
49 #include <errno.h>
50 #include <string.h>
51
52 #include "windef.h"
53 #include "winbase.h"
54 #include "winreg.h"
55 #include "winerror.h"
56 #include "winescsi.h"
57
58 #include "wine/debug.h"
59 #include "wine/unicode.h"
60
61 WINE_DEFAULT_DEBUG_CHANNEL(aspi);
62
63 #ifdef linux
64 static void set_last_error(void)
65 {
66     int save_errno = errno; /* errno gets overwritten by printf */
67     switch (errno)
68     {
69     case EAGAIN:
70         SetLastError( ERROR_SHARING_VIOLATION );
71         break;
72     case EBADF:
73         SetLastError( ERROR_INVALID_HANDLE );
74         break;
75     case ENOSPC:
76         SetLastError( ERROR_HANDLE_DISK_FULL );
77         break;
78     case EACCES:
79     case EPERM:
80     case EROFS:
81         SetLastError( ERROR_ACCESS_DENIED );
82         break;
83     case EBUSY:
84         SetLastError( ERROR_LOCK_VIOLATION );
85         break;
86     case ENOENT:
87         SetLastError( ERROR_FILE_NOT_FOUND );
88         break;
89     case EISDIR:
90         SetLastError( ERROR_CANNOT_MAKE );
91         break;
92     case ENFILE:
93     case EMFILE:
94         SetLastError( ERROR_NO_MORE_FILES );
95         break;
96     case EEXIST:
97         SetLastError( ERROR_FILE_EXISTS );
98         break;
99     case EINVAL:
100     case ESPIPE:
101         SetLastError( ERROR_SEEK );
102         break;
103     case ENOTEMPTY:
104         SetLastError( ERROR_DIR_NOT_EMPTY );
105         break;
106     case ENOEXEC:
107         SetLastError( ERROR_BAD_FORMAT );
108         break;
109     default:
110         WARN( "unknown file error: %s\n", strerror(save_errno) );
111         SetLastError( ERROR_GEN_FAILURE );
112         break;
113     }
114     errno = save_errno;
115 }
116 #endif
117
118 static const WCHAR wDevicemapScsi[] = {'H','A','R','D','W','A','R','E','\\','D','E','V','I','C','E','M','A','P','\\','S','c','s','i',0};
119
120 /* Exported functions */
121 int ASPI_GetNumControllers(void)
122 {
123     HKEY hkeyScsi, hkeyPort;
124     DWORD i = 0, numPorts, num_ha = 0;
125     WCHAR wPortName[15];
126
127     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, wDevicemapScsi, 0,
128         KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS, &hkeyScsi) != ERROR_SUCCESS )
129     {
130         ERR("Could not open HKLM\\%s\n", debugstr_w(wDevicemapScsi));
131         return 0;
132     }
133     while (RegEnumKeyW(hkeyScsi, i++, wPortName, sizeof(wPortName)/sizeof(wPortName[0])) == ERROR_SUCCESS)
134     {
135         if (RegOpenKeyExW(hkeyScsi, wPortName, 0, KEY_QUERY_VALUE, &hkeyPort) == ERROR_SUCCESS)
136         {
137             if (RegQueryInfoKeyW(hkeyPort, NULL, NULL, NULL, &numPorts, NULL, NULL,
138                              NULL, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
139             {
140                 num_ha += numPorts;
141             }
142             RegCloseKey(hkeyPort);
143         }
144     }
145
146     RegCloseKey(hkeyScsi);
147     TRACE("Returning %d host adapters\n", num_ha );
148     return num_ha;
149 }
150
151 BOOL SCSI_GetDeviceName( int h, int c, int t, int d, LPSTR devstr, LPDWORD lpcbData )
152 {
153     char buffer[200];
154     HKEY hkeyScsi;
155     DWORD type;
156
157     snprintf(buffer, sizeof(buffer), KEYNAME_SCSI, h, c, t, d);
158     if( RegOpenKeyExA(HKEY_LOCAL_MACHINE, buffer, 0, KEY_ALL_ACCESS, &hkeyScsi ) != ERROR_SUCCESS )
159     {
160         TRACE("Could not open HKLM\\%s; device does not exist\n", buffer);
161         return FALSE;
162     }
163
164     if( RegQueryValueExA(hkeyScsi, "UnixDeviceName", NULL, &type, (LPBYTE)devstr, lpcbData) != ERROR_SUCCESS )
165     {
166         WARN("Could not query value HKLM\\%s\\UnixDeviceName\n", buffer);
167         RegCloseKey(hkeyScsi);
168         return FALSE;
169     }
170     RegCloseKey(hkeyScsi);
171
172     TRACE("Device name: %s\n", devstr);
173     return TRUE;
174 }
175
176 /* SCSI_GetHCforController
177  * RETURNS
178  *      HIWORD: Host Adapter
179  *      LOWORD: Channel
180  */
181 DWORD ASPI_GetHCforController( int controller )
182 {
183     HKEY hkeyScsi, hkeyPort;
184     DWORD i = 0, numPorts;
185     int num_ha = controller + 1;
186     WCHAR wPortName[15];
187     WCHAR wBusName[15];
188
189     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, wDevicemapScsi, 0,
190         KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS, &hkeyScsi) != ERROR_SUCCESS )
191     {
192         ERR("Could not open HKLM\\%s\n", debugstr_w(wDevicemapScsi));
193         return 0xFFFFFFFF;
194     }
195     while (RegEnumKeyW(hkeyScsi, i++, wPortName, sizeof(wPortName)/sizeof(wPortName[0])) == ERROR_SUCCESS)
196     {
197         if (RegOpenKeyExW(hkeyScsi, wPortName, 0, KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS,
198                           &hkeyPort) == ERROR_SUCCESS)
199         {
200             if (RegQueryInfoKeyW(hkeyPort, NULL, NULL, NULL, &numPorts, NULL, NULL,
201                              NULL, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
202             {
203                 num_ha -= numPorts;
204                 if (num_ha <= 0) break;
205             }
206             else
207             RegCloseKey(hkeyPort);
208         }
209     }
210     RegCloseKey(hkeyScsi);
211
212     if (num_ha > 0)
213     {
214         ERR("Invalid controller(%d)\n", controller);
215         return 0xFFFFFFFF;
216     }
217
218     if (RegEnumKeyW(hkeyPort, -num_ha, wBusName, sizeof(wBusName)/sizeof(wBusName[0])) != ERROR_SUCCESS)
219     {
220         ERR("Failed to enumerate keys\n");
221         RegCloseKey(hkeyPort);
222         return 0xFFFFFFFF;
223     }
224     RegCloseKey(hkeyPort);
225
226     return (atoiW(&wPortName[9]) << 16) + atoiW(&wBusName[9]);
227 }
228
229 int SCSI_OpenDevice( int h, int c, int t, int d )
230 {
231     char devstr[20];
232     DWORD cbData = 20;
233     int fd = -1;
234
235     if(!SCSI_GetDeviceName( h, c, t, d, devstr, &cbData ))
236     {
237         WARN("Could not get device name for h%02dc%02dt%02dd%02d\n", h, c, t, d);
238         return -1;
239     }
240
241     TRACE("Opening device %s mode O_RDWR\n",devstr);
242     fd = open(devstr, O_RDWR);
243     if (fd == -1) {
244         char *errstring = strerror(errno);
245         ERR("Failed to open device %s: %s\n", devstr, errstring);
246     }
247
248     return fd;
249 }
250
251 #ifdef linux
252 /* SCSI_Fix_CMD_LEN
253  *      Checks to make sure the CMD_LEN is correct
254  */
255 void
256 SCSI_Fix_CMD_LEN(int fd, int cmd, int len)
257 {
258         /* This is what the linux kernel thinks.... */
259         static const unsigned char scsi_command_size[8] =
260         {
261                 6, 10, 10, 12,
262                 12, 12, 10, 10
263         };
264
265         int index=(cmd>>5)&7;
266
267         if (len!=scsi_command_size[index])
268         {
269                 TRACE("CDBLen for command %d claims to be %d, expected %d\n",
270                                 cmd, len, scsi_command_size[index]);
271                 ioctl(fd,SG_NEXT_CMD_LEN,&len);
272         }
273 }
274
275 int
276 SCSI_LinuxSetTimeout( int fd, int timeout )
277 {
278         int retval;
279         TRACE("Setting timeout to %d jiffies\n", timeout);
280         retval=ioctl(fd,SG_SET_TIMEOUT,&timeout);
281         if(retval)
282         {
283                 WARN("Could not set timeout ! (%s)\n", strerror(errno));
284         }
285         return retval;
286
287 }
288
289 /* This function takes care of the write/read to the linux sg device.
290  * It returns TRUE or FALSE and uses set_last_error() to convert
291  * UNIX errno to Windows GetLastError().  The reason for that is that
292  * several programs will check that error and we might as well set
293  * it here.  We also return the value of the read call in
294  * lpcbBytesReturned.
295  */
296 BOOL /* NOTE: This function SHOULD BLOCK */
297 SCSI_LinuxDeviceIo( int fd,
298                 struct sg_header * lpInBuffer, DWORD cbInBuffer,
299                 struct sg_header * lpOutBuffer, DWORD cbOutBuffer,
300                 LPDWORD lpcbBytesReturned )
301 {
302         DWORD dwBytes;
303         DWORD save_error;
304
305         TRACE("Writing to Linux sg device\n");
306         dwBytes = write( fd, lpInBuffer, cbInBuffer );
307         if( dwBytes != cbInBuffer )
308         {
309                 set_last_error();
310                 save_error = GetLastError();
311                 WARN("Not enough bytes written to scsi device. bytes=%d .. %d\n", cbInBuffer, dwBytes );
312                 /* FIXME: set_last_error() never sets error to ERROR_NOT_ENOUGH_MEMORY... */
313                 if( save_error == ERROR_NOT_ENOUGH_MEMORY )
314                         MESSAGE("Your Linux kernel was not able to handle the amount of data sent to the scsi device. Try recompiling with a larger SG_BIG_BUFF value (kernel 2.0.x sg.h)\n");
315                 WARN("error= %d\n", save_error );
316                 *lpcbBytesReturned = 0;
317                 return FALSE;
318         }
319
320         TRACE("Reading reply from Linux sg device\n");
321         *lpcbBytesReturned = read( fd, lpOutBuffer, cbOutBuffer );
322         if( *lpcbBytesReturned != cbOutBuffer )
323         {
324                 set_last_error();
325                 save_error = GetLastError();
326                 WARN("Not enough bytes read from scsi device. bytes=%d .. %d\n", cbOutBuffer, *lpcbBytesReturned);
327                 WARN("error= %d\n", save_error );
328                 return FALSE;
329         }
330         return TRUE;
331 }
332
333 static void SCSI_Linux_CheckDevices(void)
334 {
335     DIR *devdir;
336     struct dirent *dent = NULL;
337
338     devdir = opendir("/dev");
339     for (dent=readdir(devdir);dent;dent=readdir(devdir))
340     {
341         if (!(strncmp(dent->d_name, "sg", 2)))
342             break;
343     }
344     closedir(devdir);
345
346     if (dent == NULL)
347     {
348         TRACE("WARNING: You don't have any /dev/sgX generic scsi devices ! \"man MAKEDEV\" !\n");
349         return;
350     }
351 }
352 #endif
353
354 void SCSI_Init(void)
355 {
356 #ifdef linux
357     SCSI_Linux_CheckDevices();
358 #endif
359 }