/**
* Allow install using regsvr32
*/
HRESULT WINAPI
DllRegisterServer(void)
{
TCHAR fn[MAX_PATH], full_fn[MAX_PATH];
LPTSTR name;
WORD len_out;
DWORD cnt;
char *desc = NULL;
BOOL b_res;
if (!GetModuleFileName(hinstFreeTDS, fn, TDS_VECTOR_SIZE(fn)))
return SELFREG_E_CLASS;
if (!GetFullPathName(fn, TDS_VECTOR_SIZE(full_fn), full_fn, &name) || !name || full_fn == name)
return SELFREG_E_CLASS;
if (asprintf(&desc, "FreeTDS%c"
"APILevel=2%c"
"ConnectFunctions=YYN%c"
"DriverODBCVer=03.00%c"
"FileUsage=0%c"
"SQLLevel=2%c"
"Setup=%s%c"
"Driver=%s%c",
0, 0, 0, 0, 0, 0,
name, 0, name, 0
) < 0)
return SELFREG_E_CLASS;
name[-1] = 0;
b_res = SQLInstallDriverEx(desc, full_fn, fn, TDS_VECTOR_SIZE(fn), &len_out, ODBC_INSTALL_COMPLETE, &cnt);
free(desc);
if (!b_res)
return SELFREG_E_CLASS;
return S_OK;
}
static BOOL
InstallMyDriver(const char *driverpath)
{
char driver[300];
char outpath[301];
WORD outpathlen;
DWORD usagecount;
char *p;
/* the correct format of driver keywords are
* "DriverName\0Driver=...\xxxxxx.DLL\0Setup=...\xxxxxx.DLL\0\0" */
snprintf(driver, sizeof(driver),
"%s;Driver=%s\\%s;Setup=%s\\%s;APILevel=1;"
"ConnectFunctions=YYY;DriverODBCVer=%s;SQLLevel=3;",
DriverName, driverpath, DriverDLL, driverpath, DriverDLLs,
MONETDB_ODBC_VER);
for (p = driver; *p; p++)
if (*p == ';')
*p = '\0';
/* call SQLInstallDriverEx to install the driver in the
* registry */
if (!SQLInstallDriverEx(driver, driverpath,
outpath, sizeof(outpath), &outpathlen,
ODBC_INSTALL_COMPLETE, &usagecount) &&
ProcessSQLErrorMessages("SQLInstallDriverEx"))
return FALSE;
return TRUE;
}
BOOL WINAPI SQLInstallDriver(LPCSTR lpszInfFile, LPCSTR lpszDriver,
LPSTR lpszPath, WORD cbPathMax, WORD * pcbPathOut)
{
DWORD usage;
clear_errors();
TRACE("%s %s %p %d %p\n", debugstr_a(lpszInfFile),
debugstr_a(lpszDriver), lpszPath, cbPathMax, pcbPathOut);
if (lpszInfFile)
return FALSE;
return SQLInstallDriverEx(lpszDriver, NULL, lpszPath, cbPathMax,
pcbPathOut, ODBC_INSTALL_COMPLETE, &usage);
}
static BOOL
InUn(int remove, char *drivername, char *dllname, char *dll2name, char *dsname)
{
char path[301], driver[300], attr[300], inst[400], inst2[400];
WORD pathmax = sizeof (path) - 1, pathlen;
DWORD usecnt, mincnt;
if (SQLInstallDriverManager(path, pathmax, &pathlen)) {
char *p;
sprintf(driver, "%s;Driver=%s;Setup=%s;",
drivername, dllname, dllname);
p = driver;
while (*p) {
if (*p == ';') {
*p = '\0';
}
++p;
}
usecnt = 0;
SQLInstallDriverEx(driver, NULL, path, pathmax, &pathlen,
ODBC_INSTALL_INQUIRY, &usecnt);
sprintf(driver, "%s;Driver=%s\\%s;Setup=%s\\%s;",
drivername, path, dllname, path, dllname);
p = driver;
while (*p) {
if (*p == ';') {
*p = '\0';
}
++p;
}
sprintf(inst, "%s\\%s", path, dllname);
if (dll2name) {
sprintf(inst2, "%s\\%s", path, dll2name);
}
if (!remove && usecnt > 0) {
/* first install try: copy over driver dll, keeping DSNs */
if (GetFileAttributes(dllname) != INVALID_FILE_ATTRIBUTES &&
CopyFile(dllname, inst, 0) &&
CopyOrDelModules(dllname, path, 0)) {
if (dll2name != NULL) {
CopyFile(dll2name, inst2, 0);
}
return TRUE;
}
}
mincnt = remove ? 1 : 0;
while (usecnt != mincnt) {
if (!SQLRemoveDriver(driver, TRUE, &usecnt)) {
break;
}
}
if (remove) {
if (!SQLRemoveDriver(driver, TRUE, &usecnt)) {
ProcessErrorMessages("SQLRemoveDriver");
return FALSE;
}
if (!usecnt) {
char buf[512];
DeleteFile(inst);
/* but keep inst2 */
CopyOrDelModules(dllname, path, 1);
if (!quiet) {
sprintf(buf, "%s uninstalled.", drivername);
MessageBox(NULL, buf, "Info",
MB_ICONINFORMATION|MB_OK|MB_TASKMODAL|
MB_SETFOREGROUND);
}
}
if (nosys) {
goto done;
}
sprintf(attr, "DSN=%s;Database=sqlite.db;", dsname);
p = attr;
while (*p) {
if (*p == ';') {
*p = '\0';
}
++p;
}
SQLConfigDataSource(NULL, ODBC_REMOVE_SYS_DSN, drivername, attr);
goto done;
}
if (GetFileAttributes(dllname) == INVALID_FILE_ATTRIBUTES) {
return FALSE;
}
if (!CopyFile(dllname, inst, 0)) {
char buf[512];
sprintf(buf, "Copy %s to %s failed", dllname, inst);
MessageBox(NULL, buf, "CopyFile",
MB_ICONSTOP|MB_OK|MB_TASKMODAL|MB_SETFOREGROUND);
return FALSE;
}
if (dll2name != NULL && !CopyFile(dll2name, inst2, 0)) {
char buf[512];
sprintf(buf, "Copy %s to %s failed", dll2name, inst2);
MessageBox(NULL, buf, "CopyFile",
MB_ICONSTOP|MB_OK|MB_TASKMODAL|MB_SETFOREGROUND);
/* but go on hoping that an SQLite engine is in place */
}
if (!CopyOrDelModules(dllname, path, 0)) {
return FALSE;
}
if (!SQLInstallDriverEx(driver, path, path, pathmax, &pathlen,
ODBC_INSTALL_COMPLETE, &usecnt)) {
ProcessErrorMessages("SQLInstallDriverEx");
return FALSE;
}
if (nosys) {
goto done;
}
sprintf(attr, "DSN=%s;Database=sqlite.db;", dsname);
p = attr;
while (*p) {
if (*p == ';') {
*p = '\0';
}
++p;
}
SQLConfigDataSource(NULL, ODBC_REMOVE_SYS_DSN, drivername, attr);
if (!SQLConfigDataSource(NULL, ODBC_ADD_SYS_DSN, drivername, attr)) {
ProcessErrorMessages("SQLConfigDataSource");
return FALSE;
}
} else {
ProcessErrorMessages("SQLInstallDriverManager");
return FALSE;
}
done:
return TRUE;
}
BOOL INSTAPI
SQLInstallDriverExW (LPCWSTR lpszDriver, LPCWSTR lpszPathIn,
LPWSTR lpszPathOut, WORD cbPathOutMax, WORD * pcbPathOut, WORD fRequest,
LPDWORD lpdwUsageCount)
{
char *_driver_u8 = NULL;
char *_pathin_u8 = NULL;
char *_pathout_u8 = NULL;
BOOL retcode = FALSE;
int length;
SQLWCHAR *ptr;
char *ptr_u8;
for (length = 0, ptr = (SQLWCHAR *) lpszDriver; *ptr;
length += WCSLEN (ptr) + 1, ptr += WCSLEN (ptr) + 1);
if (length > 0)
{
if ((_driver_u8 = malloc (length * UTF8_MAX_CHAR_LEN + 1)) != NULL)
{
for (ptr = (SQLWCHAR *) lpszDriver, ptr_u8 = _driver_u8; *ptr;
ptr += WCSLEN (ptr) + 1, ptr_u8 += STRLEN (ptr_u8) + 1)
dm_StrCopyOut2_W2A (ptr, ptr_u8, WCSLEN (ptr) * UTF8_MAX_CHAR_LEN,
NULL);
*ptr_u8 = '\0';
}
}
else
_driver_u8 = (char *) dm_SQL_WtoU8 ((SQLWCHAR *) lpszDriver, SQL_NTS);
if (_driver_u8 == NULL && lpszDriver)
{
PUSH_ERROR (ODBC_ERROR_OUT_OF_MEM);
goto done;
}
_pathin_u8 = (char *) dm_SQL_WtoU8 ((SQLWCHAR *) lpszPathIn, SQL_NTS);
if (_pathin_u8 == NULL && lpszPathIn)
{
PUSH_ERROR (ODBC_ERROR_OUT_OF_MEM);
goto done;
}
if (cbPathOutMax > 0)
{
if ((_pathout_u8 =
malloc (cbPathOutMax * UTF8_MAX_CHAR_LEN + 1)) == NULL)
{
PUSH_ERROR (ODBC_ERROR_OUT_OF_MEM);
goto done;
}
}
retcode =
SQLInstallDriverEx (_driver_u8, _pathin_u8, _pathout_u8,
cbPathOutMax * UTF8_MAX_CHAR_LEN, pcbPathOut, fRequest, lpdwUsageCount);
if (retcode == TRUE)
{
dm_StrCopyOut2_U8toW (_pathout_u8, lpszPathOut, cbPathOutMax,
pcbPathOut);
}
done:
MEM_FREE (_driver_u8);
MEM_FREE (_pathin_u8);
MEM_FREE (_pathout_u8);
return retcode;
}
BOOL INSTAPI
ConfigDriver (
HWND hwndParent,
WORD fRequest,
LPCSTR lpszDriver,
LPCSTR lpszArgs,
LPSTR lpszMsg,
WORD cbMsgMax,
WORD * pcbMsgOut)
{
char *curr, *cour;
char driverread[4096] = { 0 };
BOOL retcode = FALSE;
UWORD confMode = ODBC_USER_DSN;
/* Map the request User/System */
if (fRequest < ODBC_INSTALL_DRIVER || fRequest > ODBC_CONFIG_DRIVER_MAX)
{
SQLPostInstallerError (ODBC_ERROR_INVALID_REQUEST_TYPE, NULL);
goto done;
}
if (!lpszDriver || !STRLEN (lpszDriver))
{
SQLPostInstallerError (ODBC_ERROR_INVALID_NAME, NULL);
goto done;
}
/* Retrieve the config mode */
SQLGetConfigMode (&confMode);
/* Treat corresponding to the request */
switch (fRequest)
{
case ODBC_INSTALL_DRIVER:
/* Check if the DRIVER with this name already exists */
SQLSetConfigMode (confMode);
#ifdef WIN32
if (hwndParent
&& SQLGetPrivateProfileString ("ODBC 32 bit Drivers", lpszDriver,
"", driverread, sizeof (driverread), "odbcinst.ini")
&& !create_confirm (hwndParent, NULL,
"Are you sure you want to overwrite this driver ?"))
#else
# ifdef _MACX
if (hwndParent
&& SQLGetPrivateProfileString ("ODBC Drivers", lpszDriver, "",
driverread, sizeof (driverread), "odbcinst.ini")
&& !create_confirm (hwndParent, NULL,
"Are you sure you want to overwrite this driver ?"))
# else
if (hwndParent
&& SQLGetPrivateProfileString ("ODBC Drivers", lpszDriver, "",
driverread, sizeof (driverread), "odbcinst.ini")
&& !create_confirm (hwndParent, NULL,
"Are you sure you want to overwrite this driver ?"))
# endif
#endif
{
SQLPostInstallerError (ODBC_ERROR_DRIVER_SPECIFIC,
"Driver already installed previously.");
goto done;
}
/* Add the Driver to the ODBC Drivers */
SQLSetConfigMode (confMode);
if (!SQLInstallDriverEx (lpszArgs, NULL, driverread,
sizeof (driverread), NULL, ODBC_INSTALL_COMPLETE, NULL))
{
SQLPostInstallerError (ODBC_ERROR_DRIVER_SPECIFIC,
"Could not add the driver information.");
goto done;
}
break;
case ODBC_CONFIG_DRIVER:
if (!lpszArgs || !STRLEN (lpszArgs))
{
SQLPostInstallerError (ODBC_ERROR_DRIVER_SPECIFIC,
"No enough parameters for configururation.");
goto done;
}
/* Add each keyword and values */
for (curr = (LPSTR) lpszArgs; *curr; curr += (STRLEN (curr) + 1))
{
STRCPY (driverread, curr);
cour = strchr (driverread, '=');
if (cour)
*cour = 0;
SQLSetConfigMode (confMode);
if (!SQLWritePrivateProfileString (lpszDriver, driverread, (cour
&& STRLEN (cour + 1)) ? cour + 1 : NULL,
"odbcinst.ini"))
goto done;
}
break;
case ODBC_REMOVE_DRIVER:
/* Remove the Driver to the ODBC Drivers */
SQLSetConfigMode (confMode);
if (!SQLRemoveDriver (lpszDriver, TRUE, NULL))
{
SQLPostInstallerError (ODBC_ERROR_DRIVER_SPECIFIC,
"Could not remove driver information.");
goto done;
}
break;
default:
SQLPostInstallerError (ODBC_ERROR_REQUEST_FAILED, NULL);
goto done;
};
quit:
retcode = TRUE;
done:
if (pcbMsgOut)
*pcbMsgOut = 0;
return retcode;
}
