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 #include <stdio.h>
  33 #include <stdarg.h>
  34 #include <sys/types.h>
  35 #include <string.h>
  36 #include <stdlib.h>
  37
  38 #include "windef.h"
  39 #include "winbase.h"
  40 #include "winreg.h"
  41 #include "winerror.h"
  42 #include "winescsi.h"
  43
  44 #include "wine/debug.h"
  45
  46 WINE_DEFAULT_DEBUG_CHANNEL(aspi);
  47
  48 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};
  49
  50 /* Exported functions */
  51 int ASPI_GetNumControllers(void)
  52 {
  53     HKEY hkeyScsi, hkeyPort;
  54     DWORD i = 0, numPorts, num_ha = 0;
  55     WCHAR wPortName[15];
  56
  57     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, wDevicemapScsi, 0,
  58         KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS, &hkeyScsi) != ERROR_SUCCESS )
  59     {
  60         ERR("Could not open HKLM\\%s\n", debugstr_w(wDevicemapScsi));
  61         return 0;
  62     }
  63     while (RegEnumKeyW(hkeyScsi, i++, wPortName, ARRAY_SIZE(wPortName)) == ERROR_SUCCESS)
  64     {
  65         if (RegOpenKeyExW(hkeyScsi, wPortName, 0, KEY_QUERY_VALUE, &hkeyPort) == ERROR_SUCCESS)
  66         {
  67             if (RegQueryInfoKeyW(hkeyPort, NULL, NULL, NULL, &numPorts, NULL, NULL,
  68                              NULL, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
  69             {
  70                 num_ha += numPorts;
  71             }
  72             RegCloseKey(hkeyPort);
  73         }
  74     }
  75
  76     RegCloseKey(hkeyScsi);
  77     TRACE("Returning %ld host adapters\n", num_ha );
  78     return num_ha;
  79 }
  80
  81 /* SCSI_GetHCforController
  82  * RETURNS
  83  *      HIWORD: Host Adapter
  84  *      LOWORD: Channel
  85  */
  86 DWORD ASPI_GetHCforController( int controller )
  87 {
  88     HKEY hkeyScsi, hkeyPort;
  89     DWORD i = 0, numPorts;
  90     int num_ha = controller + 1;
  91     WCHAR wPortName[15];
  92     WCHAR wBusName[15];
  93
  94     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, wDevicemapScsi, 0,
  95         KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS, &hkeyScsi) != ERROR_SUCCESS )
  96     {
  97         ERR("Could not open HKLM\\%s\n", debugstr_w(wDevicemapScsi));
  98         return 0xFFFFFFFF;
  99     }
 100     while (RegEnumKeyW(hkeyScsi, i++, wPortName, ARRAY_SIZE(wPortName)) == ERROR_SUCCESS)
 101     {
 102         if (RegOpenKeyExW(hkeyScsi, wPortName, 0, KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS,
 103                           &hkeyPort) == ERROR_SUCCESS)
 104         {
 105             if (RegQueryInfoKeyW(hkeyPort, NULL, NULL, NULL, &numPorts, NULL, NULL,
 106                              NULL, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
 107             {
 108                 num_ha -= numPorts;
 109                 if (num_ha <= 0) break;
 110             }
 111             else
 112             RegCloseKey(hkeyPort);
 113         }
 114     }
 115     RegCloseKey(hkeyScsi);
 116
 117     if (num_ha > 0)
 118     {
 119         ERR("Invalid controller(%d)\n", controller);
 120         return 0xFFFFFFFF;
 121     }
 122
 123     if (RegEnumKeyW(hkeyPort, -num_ha, wBusName, ARRAY_SIZE(wBusName)) != ERROR_SUCCESS)
 124     {
 125         ERR("Failed to enumerate keys\n");
 126         RegCloseKey(hkeyPort);
 127         return 0xFFFFFFFF;
 128     }
 129     RegCloseKey(hkeyPort);
 130
 131     return (wcstol(&wPortName[9], NULL, 10) << 16) + wcstol(&wBusName[9], NULL, 10);
 132 }