1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
|
/* $Id: USBIdDatabase.h $ */
/** @file
* USB device vendor and product ID database.
*/
/*
* Copyright (C) 2015-2023 Oracle and/or its affiliates.
*
* This file is part of VirtualBox base platform packages, as
* available from https://www.virtualbox.org.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation, in version 3 of the
* License.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, see <https://www.gnu.org/licenses>.
*
* SPDX-License-Identifier: GPL-3.0-only
*/
#ifndef MAIN_INCLUDED_USBIdDatabase_h
#define MAIN_INCLUDED_USBIdDatabase_h
#ifndef RT_WITHOUT_PRAGMA_ONCE
# pragma once
#endif
#include <iprt/assert.h>
#include <iprt/stdint.h>
#include <iprt/cpp/ministring.h>
#include <iprt/bldprog-strtab.h>
/** Saves a few bytes (~25%) on strings. */
#define USB_ID_DATABASE_WITH_COMPRESSION
/** Max string length. */
#define USB_ID_DATABASE_MAX_STRING _1K
AssertCompileSize(RTBLDPROGSTRREF, sizeof(uint32_t));
/**
* Elements of product table.
*/
typedef struct USBIDDBPROD
{
/** Product ID. */
uint16_t idProduct;
} USBIDDBPROD;
AssertCompileSize(USBIDDBPROD, sizeof(uint16_t));
/**
* Element of vendor table.
*/
typedef struct USBIDDBVENDOR
{
/** Vendor ID. */
uint16_t idVendor;
/** Index of the first product. */
uint16_t iProduct;
/** Number of products. */
uint16_t cProducts;
} USBIDDBVENDOR;
AssertCompileSize(USBIDDBVENDOR, sizeof(uint16_t) * 3);
/**
* Wrapper for static array of Aliases.
*/
class USBIdDatabase
{
public: // For assertions and statis in the generator.
/** The compressed string table. */
static RTBLDPROGSTRTAB const s_StrTab;
/** Number of vendors in the two parallel arrays. */
static const size_t s_cVendors;
/** Vendor IDs lookup table. */
static const USBIDDBVENDOR s_aVendors[];
/** Vendor names table running parallel to s_aVendors. */
static const RTBLDPROGSTRREF s_aVendorNames[];
/** Number of products in the two parallel arrays. */
static const size_t s_cProducts;
/** Vendor+Product keys for lookup purposes. */
static const USBIDDBPROD s_aProducts[];
/** Product names table running parallel to s_aProducts. */
static const RTBLDPROGSTRREF s_aProductNames[];
public:
static RTCString returnString(PCRTBLDPROGSTRREF pStr)
{
char szTmp[USB_ID_DATABASE_MAX_STRING * 2];
ssize_t cchTmp = RTBldProgStrTabQueryString(&s_StrTab, pStr->off, pStr->cch, szTmp, sizeof(szTmp));
return RTCString(szTmp, (size_t)RT_MAX(cchTmp, 0));
}
private:
/**
* Performs a binary lookup of @a idVendor.
*
* @returns The index in the vendor tables, UINT32_MAX if not found.
* @param idVendor The vendor ID.
*/
static uint32_t lookupVendor(uint16_t idVendor)
{
size_t iEnd = s_cVendors;
if (iEnd)
{
size_t iStart = 0;
for (;;)
{
size_t idx = iStart + (iEnd - iStart) / 2;
if (s_aVendors[idx].idVendor < idVendor)
{
idx++;
if (idx < iEnd)
iStart = idx;
else
break;
}
else if (s_aVendors[idx].idVendor > idVendor)
{
if (idx != iStart)
iEnd = idx;
else
break;
}
else
return (uint32_t)idx;
}
}
return UINT32_MAX;
}
/**
* Performs a binary lookup of @a idProduct.
*
* @returns The index in the product tables, UINT32_MAX if not found.
* @param idProduct The product ID.
* @param iStart The index of the first entry for the vendor.
* @param iEnd The index of after the last entry.
*/
static uint32_t lookupProduct(uint16_t idProduct, size_t iStart, size_t iEnd)
{
if (iStart < iEnd)
{
for (;;)
{
size_t idx = iStart + (iEnd - iStart) / 2;
if (s_aProducts[idx].idProduct < idProduct)
{
idx++;
if (idx < iEnd)
iStart = idx;
else
break;
}
else if (s_aProducts[idx].idProduct > idProduct)
{
if (idx != iStart)
iEnd = idx;
else
break;
}
else
return (uint32_t)idx;
}
}
return UINT32_MAX;
}
public:
static RTCString findProduct(uint16_t idVendor, uint16_t idProduct)
{
uint32_t idxVendor = lookupVendor(idVendor);
if (idxVendor != UINT32_MAX)
{
uint32_t idxProduct = lookupProduct(idProduct, s_aVendors[idxVendor].iProduct,
s_aVendors[idxVendor].iProduct + s_aVendors[idxVendor].cProducts);
if (idxProduct != UINT32_MAX)
return returnString(&s_aProductNames[idxProduct]);
}
return RTCString();
}
static RTCString findVendor(uint16_t idVendor)
{
uint32_t idxVendor = lookupVendor(idVendor);
if (idxVendor != UINT32_MAX)
return returnString(&s_aVendorNames[idxVendor]);
return RTCString();
}
static RTCString findVendorAndProduct(uint16_t idVendor, uint16_t idProduct, RTCString *pstrProduct)
{
uint32_t idxVendor = lookupVendor(idVendor);
if (idxVendor != UINT32_MAX)
{
uint32_t idxProduct = lookupProduct(idProduct, s_aVendors[idxVendor].iProduct,
s_aVendors[idxVendor].iProduct + s_aVendors[idxVendor].cProducts);
if (idxProduct != UINT32_MAX)
*pstrProduct = returnString(&s_aProductNames[idxProduct]);
else
pstrProduct->setNull();
return returnString(&s_aVendorNames[idxVendor]);
}
pstrProduct->setNull();
return RTCString();
}
};
#endif /* !MAIN_INCLUDED_USBIdDatabase_h */
|
