typedef struct _COMDB
{
HKEY hKey;
- DWORD dwSize;
- PBYTE pBitmap;
- PBYTE pData;
} COMDB, *PCOMDB;
+LONG
+WINAPI
+ComDBClaimNextFreePort(IN HCOMDB hComDB,
+ OUT LPDWORD ComNumber)
+{
+ PCOMDB pComDB;
+ DWORD dwBitIndex;
+ DWORD dwByteIndex;
+ DWORD dwSize;
+ DWORD dwType;
+ PBYTE pBitmap = NULL;
+ BYTE cMask;
+ LONG lError;
+
+ TRACE("ComDBClaimNextFreePort(%p %p)\n", hComDB, ComNumber);
+
+ if (hComDB == INVALID_HANDLE_VALUE ||
+ hComDB == NULL ||
+ ComNumber == NULL)
+ return ERROR_INVALID_PARAMETER;
+
+ pComDB = (PCOMDB)hComDB;
+
+ /* Get the required bitmap size */
+ lError = RegQueryValueExW(pComDB->hKey,
+ L"ComDB",
+ NULL,
+ &dwType,
+ NULL,
+ &dwSize);
+ if (lError != ERROR_SUCCESS)
+ return lError;
+
+ /* Allocate the bitmap */
+ pBitmap = HeapAlloc(GetProcessHeap(),
+ HEAP_ZERO_MEMORY,
+ dwSize);
+ if (pBitmap == NULL)
+ {
+ ERR("Failed to allocate the bitmap!\n");
+ return lError;
+ }
+
+ /* Read the bitmap */
+ lError = RegQueryValueExW(pComDB->hKey,
+ L"ComDB",
+ NULL,
+ &dwType,
+ pBitmap,
+ &dwSize);
+ if (lError != ERROR_SUCCESS)
+ goto done;
+
+ lError = ERROR_INVALID_PARAMETER;
+ for (dwBitIndex = 0; dwBitIndex < (dwSize * BITS_PER_BYTE); dwBitIndex++)
+ {
+ /* Calculate the byte index and a mask for the affected bit */
+ dwByteIndex = dwBitIndex / BITS_PER_BYTE;
+ cMask = 1 << (dwBitIndex % BITS_PER_BYTE);
+
+ if ((pBitmap[dwByteIndex] & cMask) == 0)
+ {
+ pBitmap[dwByteIndex] |= cMask;
+ *ComNumber = dwBitIndex + 1;
+ lError = ERROR_SUCCESS;
+ break;
+ }
+ }
+
+ /* Save the bitmap if it was modified */
+ if (lError == ERROR_SUCCESS)
+ {
+ lError = RegSetValueExW(pComDB->hKey,
+ L"ComDB",
+ 0,
+ REG_BINARY,
+ pBitmap,
+ dwSize);
+ }
+
+done:;
+ if (pBitmap != NULL)
+ HeapFree(GetProcessHeap(), 0, pBitmap);
+
+ return lError;
+}
+
+
LONG
WINAPI
ComDBClaimPort(IN HCOMDB hComDB,
OUT PBOOL Forced)
{
PCOMDB pComDB;
- PBYTE pByte;
- BYTE cMask;
DWORD dwBitIndex;
+ DWORD dwByteIndex;
DWORD dwType;
DWORD dwSize;
+ PBYTE pBitmap = NULL;
+ BYTE cMask;
LONG lError;
+ TRACE("ComDBClaimPort(%p %lu)\n", hComDB, ComNumber);
+
if (hComDB == INVALID_HANDLE_VALUE ||
hComDB == NULL ||
ComNumber == 0 ||
pComDB = (PCOMDB)hComDB;
- /* Update the bitmap */
- dwSize = pComDB->dwSize;
+ /* Get the required bitmap size */
lError = RegQueryValueExW(pComDB->hKey,
L"ComDB",
NULL,
&dwType,
- pComDB->pBitmap,
+ NULL,
&dwSize);
if (lError != ERROR_SUCCESS)
return lError;
+ /* Allocate the bitmap */
+ pBitmap = HeapAlloc(GetProcessHeap(),
+ HEAP_ZERO_MEMORY,
+ dwSize);
+ if (pBitmap == NULL)
+ {
+ ERR("Failed to allocate the bitmap!\n");
+ return lError;
+ }
+
+ /* Read the bitmap */
+ lError = RegQueryValueExW(pComDB->hKey,
+ L"ComDB",
+ NULL,
+ &dwType,
+ pBitmap,
+ &dwSize);
+ if (lError != ERROR_SUCCESS)
+ goto done;
+
/* Get the bit index */
dwBitIndex = ComNumber - 1;
/* Check if the bit to set fits into the bitmap */
- if (dwBitIndex >= (pComDB->dwSize * BITS_PER_BYTE))
+ if (dwBitIndex >= (dwSize * BITS_PER_BYTE))
{
- /* FIXME: Resize the bitmap */
- return ERROR_INVALID_PARAMETER;
+ FIXME("Resize the bitmap\n");
+
+ lError = ERROR_INVALID_PARAMETER;
+ goto done;
}
- /* Get a pointer to the affected byte and calculate a mask for the affected bit */
- pByte = &(pComDB->pBitmap[dwBitIndex / BITS_PER_BYTE]);
+ /* Calculate the byte index and a mask for the affected bit */
+ dwByteIndex = dwBitIndex / BITS_PER_BYTE;
cMask = 1 << (dwBitIndex % BITS_PER_BYTE);
+ lError = ERROR_SHARING_VIOLATION;
+
/* Check if the bit is not set */
- if ((*pByte & cMask) == 0)
+ if ((pBitmap[dwByteIndex] & cMask) == 0)
{
/* Set the bit */
- *pByte |= cMask;
+ pBitmap[dwByteIndex] |= cMask;
lError = ERROR_SUCCESS;
}
- else
- {
- /* The bit is already set */
- lError = ERROR_SHARING_VIOLATION;
- }
/* Save the bitmap if it was modified */
if (lError == ERROR_SUCCESS)
L"ComDB",
0,
REG_BINARY,
- pComDB->pData,
- pComDB->dwSize);
+ pBitmap,
+ dwSize);
}
+done:
+ if (pBitmap != NULL)
+ HeapFree(GetProcessHeap(), 0, pBitmap);
+
return lError;
}
{
PCOMDB pComDB;
+ TRACE("ComDBClose(%p)\n", hComDB);
+
if (hComDB == HCOMDB_INVALID_HANDLE_VALUE || hComDB == NULL)
return ERROR_INVALID_PARAMETER;
if (pComDB->hKey != NULL)
RegCloseKey(pComDB->hKey);
- /* Release the bitmap */
- if (pComDB->pBitmap != NULL)
- HeapFree(GetProcessHeap(), 0, pComDB->pBitmap);
-
/* Release the database */
HeapFree(GetProcessHeap(), 0, pComDB);
}
+LONG
+WINAPI
+ComDBGetCurrentPortUsage(IN HCOMDB hComDB,
+ OUT PBYTE Buffer,
+ IN DWORD BufferSize,
+ IN DWORD ReportType,
+ OUT LPDWORD MaxPortsReported)
+{
+ FIXME("ComDBGetCurrentPortUsage(%p)\n", hComDB);
+ return ERROR_CALL_NOT_IMPLEMENTED;
+}
+
+
LONG
WINAPI
ComDBOpen(OUT HCOMDB *phComDB)
PCOMDB pComDB;
DWORD dwDisposition;
DWORD dwType;
+ DWORD dwSize;
+ PBYTE pBitmap;
LONG lError;
TRACE("ComDBOpen(%p)\n", phComDB);
sizeof(COMDB));
if (pComDB == NULL)
{
- ERR("Failed to allocaete the database!\n");
+ ERR("Failed to allocate the database!\n");
return ERROR_ACCESS_DENIED;
}
NULL,
&dwType,
NULL,
- &pComDB->dwSize);
+ &dwSize);
if (lError == ERROR_FILE_NOT_FOUND)
{
/* Allocate a new bitmap */
- pComDB->dwSize = COMDB_MIN_PORTS_ARBITRATED / BITS_PER_BYTE;
- pComDB->pData = HeapAlloc(GetProcessHeap(),
- HEAP_ZERO_MEMORY,
- pComDB->dwSize);
- if (pComDB->pData == NULL)
+ dwSize = COMDB_MIN_PORTS_ARBITRATED / BITS_PER_BYTE;
+ pBitmap = HeapAlloc(GetProcessHeap(),
+ HEAP_ZERO_MEMORY,
+ dwSize);
+ if (pBitmap == NULL)
{
- ERR("Failed to allocaete the bitmap!\n");
+ ERR("Failed to allocate the bitmap!\n");
lError = ERROR_ACCESS_DENIED;
goto done;
}
- /* Read the bitmap from the registry */
+ /* Write the bitmap to the registry */
lError = RegSetValueExW(pComDB->hKey,
L"ComDB",
0,
REG_BINARY,
- pComDB->pData,
- pComDB->dwSize);
+ pBitmap,
+ dwSize);
+
+ HeapFree(GetProcessHeap(), 0, pBitmap);
}
done:;
if (pComDB->hKey != NULL)
RegCloseKey(pComDB->hKey);
- if (pComDB->pData != NULL)
- HeapFree(GetProcessHeap(), 0, pComDB->pData);
-
HeapFree(GetProcessHeap(), 0, pComDB);
*phComDB = HCOMDB_INVALID_HANDLE_VALUE;
return lError;
}
+
+LONG
+WINAPI
+ComDBReleasePort(IN HCOMDB hComDB,
+ IN DWORD ComNumber)
+{
+ PCOMDB pComDB;
+ DWORD dwByteIndex;
+ DWORD dwBitIndex;
+ DWORD dwType;
+ DWORD dwSize;
+ PBYTE pBitmap = NULL;
+ BYTE cMask;
+ LONG lError;
+
+ TRACE("ComDBReleasePort(%p %lu)\n", hComDB, ComNumber);
+
+ if (hComDB == INVALID_HANDLE_VALUE ||
+ ComNumber == 0 ||
+ ComNumber > COMDB_MAX_PORTS_ARBITRATED)
+ return ERROR_INVALID_PARAMETER;
+
+ pComDB = (PCOMDB)hComDB;
+
+ /* Get the required bitmap size */
+ lError = RegQueryValueExW(pComDB->hKey,
+ L"ComDB",
+ NULL,
+ &dwType,
+ NULL,
+ &dwSize);
+ if (lError != ERROR_SUCCESS)
+ return lError;
+
+ /* Allocate the bitmap */
+ pBitmap = HeapAlloc(GetProcessHeap(),
+ HEAP_ZERO_MEMORY,
+ dwSize);
+ if (pBitmap == NULL)
+ {
+ ERR("Failed to allocate the bitmap!\n");
+ return lError;
+ }
+
+ /* Read the bitmap */
+ lError = RegQueryValueExW(pComDB->hKey,
+ L"ComDB",
+ NULL,
+ &dwType,
+ pBitmap,
+ &dwSize);
+ if (lError != ERROR_SUCCESS)
+ goto done;
+
+ /* Get the bit index */
+ dwBitIndex = ComNumber - 1;
+
+ /* Check if the bit to set fits into the bitmap */
+ if (dwBitIndex >= (dwSize * BITS_PER_BYTE))
+ {
+ lError = ERROR_INVALID_PARAMETER;
+ goto done;
+ }
+
+ /* Calculate the byte index and a mask for the affected bit */
+ dwByteIndex = dwBitIndex / BITS_PER_BYTE;
+ cMask = 1 << (dwBitIndex % BITS_PER_BYTE);
+
+ /* Release the port */
+ pBitmap[dwByteIndex] &= ~cMask;
+
+ lError = RegSetValueExW(pComDB->hKey,
+ L"ComDB",
+ 0,
+ REG_BINARY,
+ pBitmap,
+ dwSize);
+
+done:;
+ if (pBitmap != NULL)
+ HeapFree(GetProcessHeap(), 0, pBitmap);
+
+ return lError;
+}
+
+
+LONG
+WINAPI
+ComDBResizeDatabase(IN HCOMDB hComDB,
+ IN DWORD NewSize)
+{
+ FIXME("ComDBResizeDatabase(%p %lu)\n", hComDB, NewSize);
+ return ERROR_CALL_NOT_IMPLEMENTED;
+}
+
/* EOF */
projects / reactos.git / blobdiff