int main(int argc, char *argv[])
{
/*
* Initialize IPRT and convert argv to UTF-8.
*/
int rc = RTR3InitExe(argc, &argv, 0);
if (RT_FAILURE(rc))
return RTMsgInitFailure(rc);
/*
* Parse arguments and read input files.
*/
if (argc < 4)
{
usage(stderr, argv[0]);
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Insufficient arguments.");
}
g_products.reserve(20000);
g_vendors.reserve(3500);
const char *pszOutFile = NULL;
for (int i = 1; i < argc; i++)
{
if (strcmp(argv[i], "-o") == 0)
{
pszOutFile = argv[++i];
continue;
}
if ( strcmp(argv[i], "-h") == 0
|| strcmp(argv[i], "-?") == 0
|| strcmp(argv[i], "--help") == 0)
{
usage(stdout, argv[0]);
return RTEXITCODE_SUCCESS;
}
PRTSTREAM pInStrm;
rc = RTStrmOpen(argv[i], "r", &pInStrm);
if (RT_FAILURE(rc))
return RTMsgErrorExit((RTEXITCODE)ERROR_OPEN_FILE,
"Failed to open file '%s' for reading: %Rrc", argv[i], rc);
rc = ParseUsbIds(pInStrm, argv[i]);
RTStrmClose(pInStrm);
if (rc != 0)
{
RTMsgError("Failed parsing USB devices file '%s'", argv[i]);
return rc;
}
}
/*
* Due to USBIDDBVENDOR::iProduct, there is currently a max of 64KB products.
* (Not a problem as we've only have less that 54K products currently.)
*/
if (g_products.size() > _64K)
return RTMsgErrorExit((RTEXITCODE)ERROR_TOO_MANY_PRODUCTS,
"More than 64K products is not supported: %u products", g_products.size());
/*
* Sort the IDs and fill in the iProduct and cProduct members.
*/
sort(g_products.begin(), g_products.end());
sort(g_vendors.begin(), g_vendors.end());
size_t iProduct = 0;
for (size_t iVendor = 0; iVendor < g_vendors.size(); iVendor++)
{
size_t const idVendor = g_vendors[iVendor].vendorID;
g_vendors[iVendor].iProduct = iProduct;
if ( iProduct < g_products.size()
&& g_products[iProduct].vendorID <= idVendor)
{
if (g_products[iProduct].vendorID == idVendor)
do
iProduct++;
while ( iProduct < g_products.size()
&& g_products[iProduct].vendorID == idVendor);
else
return RTMsgErrorExit((RTEXITCODE)ERROR_IN_PARSE_LINE, "product without vendor after sorting. impossible!");
}
g_vendors[iVendor].cProducts = iProduct - g_vendors[iVendor].iProduct;
}
/*
* Verify that all IDs are unique.
*/
ProductsSet::iterator ita = adjacent_find(g_products.begin(), g_products.end());
if (ita != g_products.end())
return RTMsgErrorExit((RTEXITCODE)ERROR_DUPLICATE_ENTRY, "Duplicate alias detected: idProduct=%#06x", ita->productID);
/*
* Build the string table.
* Do string compression and create the string table.
*/
BLDPROGSTRTAB StrTab;
if (!BldProgStrTab_Init(&StrTab, g_products.size() + g_vendors.size()))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "Out of memory!");
for (ProductsSet::iterator it = g_products.begin(); it != g_products.end(); ++it)
{
it->StrRef.pszString = (char *)it->str.c_str();
BldProgStrTab_AddString(&StrTab, &it->StrRef);
}
for (VendorsSet::iterator it = g_vendors.begin(); it != g_vendors.end(); ++it)
{
it->StrRef.pszString = (char *)it->str.c_str();
BldProgStrTab_AddString(&StrTab, &it->StrRef);
}
if (!BldProgStrTab_CompileIt(&StrTab, g_fVerbose))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "BldProgStrTab_CompileIt failed!\n");
/*
* Print stats. Making a little extra effort to get it all on one line.
*/
size_t const cbVendorEntry = sizeof(USBIdDatabase::s_aVendors[0]) + sizeof(USBIdDatabase::s_aVendorNames[0]);
size_t const cbProductEntry = sizeof(USBIdDatabase::s_aProducts[0]) + sizeof(USBIdDatabase::s_aProductNames[0]);
size_t cbOldRaw = (g_products.size() + g_vendors.size()) * sizeof(const char *) * 2 + g_cbRawStrings;
size_t cbRaw = g_vendors.size() * cbVendorEntry + g_products.size() * cbProductEntry + g_cbRawStrings;
size_t cbActual = g_vendors.size() * cbVendorEntry + g_products.size() * cbProductEntry + StrTab.cchStrTab;
#ifdef USB_ID_DATABASE_WITH_COMPRESSION
cbActual += sizeof(StrTab.aCompDict);
#endif
char szMsg1[32];
RTStrPrintf(szMsg1, sizeof(szMsg1),"Total %zu bytes", cbActual);
char szMsg2[64];
RTStrPrintf(szMsg2, sizeof(szMsg2)," old version %zu bytes + relocs (%zu%% save)",
cbOldRaw, (cbOldRaw - cbActual) * 100 / cbOldRaw);
if (cbActual < cbRaw)
RTMsgInfo("%s - saving %zu%% (%zu bytes);%s", szMsg1, (cbRaw - cbActual) * 100 / cbRaw, cbRaw - cbActual, szMsg2);
else
RTMsgInfo("%s - wasting %zu bytes;%s", szMsg1, cbActual - cbRaw, szMsg2);
/*
* Produce the source file.
*/
if (!pszOutFile)
return RTMsgErrorExit((RTEXITCODE)ERROR_OPEN_FILE, "Output file is not specified.");
FILE *pOut = fopen(pszOutFile, "w");
if (!pOut)
return RTMsgErrorExit((RTEXITCODE)ERROR_OPEN_FILE, "Error opening '%s' for writing", pszOutFile);
WriteSourceFile(pOut, argv[0], &StrTab);
if (ferror(pOut))
return RTMsgErrorExit((RTEXITCODE)ERROR_OPEN_FILE, "Error writing '%s'!", pszOutFile);
if (fclose(pOut) != 0)
return RTMsgErrorExit((RTEXITCODE)ERROR_OPEN_FILE, "Error closing '%s'!", pszOutFile);
return RTEXITCODE_SUCCESS;
}
void DatabaseCommonCustom::WriteFile(const char* out_path, const dios::CDatabaseTableInfo& table_info)
{
WriteHeaderFile(out_path, table_info);
WriteSourceFile(out_path, table_info);
}