Esempio n. 1
0
/**
	@param lpCopyFrom Folder to copy the printer driver files from
	@return true if copied successfully, false if failed
*/
bool PrinterInstall::DoCopyFiles(LPCTSTR lpCopyFrom)
{
	TCHAR lpPath[MAX_PATH + 1];
	DWORD uSize=0;

	// Get the Windows printer driver folder
	if (!GetPrinterDriverDirectory(NULL, NULL, 1, (LPBYTE)lpPath, MAX_PATH + 1, &uSize))
	{
		// Failed?!
		SetError(-1201);
#ifdef TRACE
		TRACE(_T("Could not get printer driver directory: %d\n"), GetLastError());
#endif
		return false;
	}

	// Copy the files from the lpCopyFrom dir to the destination dir:
	std::tstring sFrom, sTo;
	for (STRARRAY::const_iterator i = m_arFiles.begin(); i != m_arFiles.end(); i++)
	{
		// Create the full path name for the files
		sFrom = ConcatPaths(lpCopyFrom, (*i));
		sTo = ConcatPaths(lpPath, (*i));

		// Now copy
		if (!::CopyFile(sFrom.c_str(), sTo.c_str(), TRUE))
		{
			DWORD error = GetLastError();
			if (error == 0 && !::CopyFile(sFrom.c_str(), sTo.c_str(), FALSE))
			{
				// May be a permissions error or something,
				// let's ignore
			} else {
			SetError(error/*-1202*/);
			/*TCHAR msg[1024];
			_tcscpy_s (msg, 1024, "Failed to copy from\n");
			_tcscat_s (msg, 1024, sFrom.c_str());
			_tcscat_s (msg, 1024, "\nto\n");
			_tcscat_s (msg, 1024, sTo.c_str());
			MessageBox (NULL, msg, _T("Error!"), 0);*/
#ifdef TRACE
			TRACE(_T("Cannot copy file: %d\n"), GetLastError());
#endif
			return false;
			}
		}
	}

	return true;
}
Esempio n. 2
0
/**
	@return true if the printer driver files were removed successfully, false if failed
*/
bool PrinterInstall::DoRemoveFiles()
{
	TCHAR lpPath[MAX_PATH + 1];
	DWORD uSize=0;

	// Get the Windows printer driver folder
	if (!::GetPrinterDriverDirectory(NULL, NULL, 1, (LPBYTE)lpPath, MAX_PATH + 1, &uSize))
	{
		// Failed!
		SetError(-1301);
#ifdef TRACE
		TRACE(_T("Cannot get printer driver directory: %d\n"), GetLastError());
#endif
		return false;
	}

	// Remove all the files:
	std::tstring sFullPath;
	for (STRARRAY::const_iterator i = m_arFiles.begin(); i != m_arFiles.end(); i++)
	{
		// Create the full path to the file
		sFullPath = ConcatPaths(lpPath, (*i));
		// Delete it
		if (!::DeleteFile(sFullPath.c_str()))
		{
#ifdef TRACE
			TRACE(_T("Warning: cannot delete file %s: %d\n"), (*i).c_str(), GetLastError());
#endif
		}
	}
	return true;
}
Esempio n. 3
0
// Given directory = "dir", base_name = "test", number = 0,
// extension = "xml", returns "dir/test.xml". If number is greater
// than zero (e.g., 12), returns "dir/test_12.xml".
// On Windows platform, uses \ as the separator rather than /.
FilePath FilePath::MakeFileName(const FilePath& directory,
                                const FilePath& base_name,
                                int number,
                                const char* extension) {
  std::string file;
  if (number == 0) {
    file = base_name.string() + "." + extension;
  } else {
    file = base_name.string() + "_" + StreamableToString(number)
        + "." + extension;
  }
  return ConcatPaths(directory, FilePath(file));
}
Esempio n. 4
0
LList <char *> *ListSubDirectoryNames(const char *_dir)
{
    LList<char *> *result = new LList<char *>();

#ifdef WIN32
    _finddata_t thisfile;
    long fileindex;
	char *dir = ConcatPaths( _dir, "*.*", NULL );
	
	fileindex = _findfirst( dir, &thisfile );

    int exitmeplease = 0;

    while( fileindex != -1 && !exitmeplease ) 
    {
        if( strcmp( thisfile.name, "." ) != 0 &&
            strcmp( thisfile.name, ".." ) != 0 &&
            (thisfile.attrib & _A_SUBDIR) )
        {
            char *newname = strdup( thisfile.name );   
            result->PutData( newname );
        }

        exitmeplease = _findnext( fileindex, &thisfile );
    }
	
	delete[] dir;
#else

    DIR *dir = opendir(_dir);
    if (dir == NULL)
        return result;
    for (struct dirent *entry; (entry = readdir(dir)) != NULL; ) {
        if (entry->d_name[0] == '.')
            continue;
            
        char fullname[strlen(_dir) + strlen(entry->d_name) + 2];
        sprintf(fullname, "%s%s%s", 
            _dir, 
            _dir[0] ? "/" : "",
            entry->d_name);
            
        if (IsDirectory(fullname))
            result->PutData( strdup(entry->d_name) );
    }
    closedir(dir);
    
#endif
    return result;
}
Esempio n. 5
0
/**
	@param lpMonitorName Name of port monitor
	@param lpMonitorFile Port monitor DLL filename
	@param lpFilePath Path where the port monitor file currently resides
	@return true if installed successfully, false if failed
*/
bool PrinterInstall::AddPortMonitor(LPCTSTR lpMonitorName, LPCTSTR lpMonitorFile, LPCTSTR lpFilePath)
{
	// Due to certain stupid Windows 7 mechanisms, we need to disable this thing:
	PVOID oldval = Disable64Redirection();

	// Only add the port monitor if it's not already installed:
	DWORD dwNeeded, dwReturned;
	::EnumMonitors(NULL, 1, NULL, 0, &dwNeeded, &dwReturned);
	if (dwNeeded > 0)
	{
		MONITOR_INFO_1* pMonitors = (MONITOR_INFO_1*)new char[dwNeeded];
		bool bFound = false;
		if (::EnumMonitors(NULL, 1, (LPBYTE)pMonitors, dwNeeded, &dwNeeded, &dwReturned))
		{
			for (DWORD i=0;!bFound && (i<dwReturned);i++)
				bFound = _tcscmp(pMonitors[i].pName, lpMonitorName) == 0;
		}
		delete [] pMonitors;
		if (bFound)
			return true;
	}

	TCHAR lpPath[MAX_PATH + 1];

	// Get the Windows system path (where we will install the port monitor)
	if (!GetSystemDirectory(lpPath, MAX_PATH + 1))
	{
		// Failed
		SetError(-2001);
#ifdef TRACE
		TRACE(_T("Could not get system directory: %d\n"), GetLastError());
#endif
		return false;
	}

	// Copy the file into the system directory
	std::tstring sFrom, sTo;
	sFrom = ConcatPaths(lpFilePath, lpMonitorFile);
	sTo = ConcatPaths(lpPath, lpMonitorFile);

	if (!ExistsAsFile(sTo.c_str()))
	{
		if (!::CopyFile(sFrom.c_str(), sTo.c_str(), FALSE))
		{
			// Couldn't!
			SetError(-2002);
			/*TCHAR msg[1024];
			_tcscpy_s (msg, 1024, "Failed to copy from\n");
			_tcscat_s (msg, 1024, sFrom.c_str());
			_tcscat_s (msg, 1024, "\nto\n");
			_tcscat_s (msg, 1024, sTo.c_str());
			MessageBox (NULL, msg, _T("Error!"), 0);*/

#ifdef TRACE
			TRACE(_T("Cannot copy file: %d\n"), GetLastError());
#endif
			return false;
		}
	}

	// Initialize the data structure
	MONITOR_INFO_2 info;
	info.pName = (TCHAR*)lpMonitorName;
	info.pEnvironment = m_bX64 ? ENVIRONMENT_64 : ENVIRONMENT;
	info.pDLLName = (TCHAR*)lpMonitorFile;

// Add the new monitor
	if (!::AddMonitor(NULL, 2, (LPBYTE)&info))
	{
		int nError = GetLastError();
		// Failed...
		SetError(-2003);
#ifdef TRACE
		TRACE(_T("Cannot add a port monitor: %d\n"), GetLastError());
#endif
		return false;
	}


	Revert64Redirection(oldval);

	// Remember the name!
	m_sPortMonitor = lpMonitorName;
	return true;
}
Esempio n. 6
0
static BOOLEAN
InstallDriver(
    IN HINF hInf,
    IN HANDLE hServices,
    IN HANDLE hDeviceKey,
    IN LPCWSTR DeviceId,
    IN LPCWSTR HardwareId)
{
    UNICODE_STRING PathPrefix = RTL_CONSTANT_STRING(L"System32\\DRIVERS\\");
    UNICODE_STRING ServiceU = RTL_CONSTANT_STRING(L"Service");
    UNICODE_STRING ErrorControlU = RTL_CONSTANT_STRING(L"ErrorControl");
    UNICODE_STRING ImagePathU = RTL_CONSTANT_STRING(L"ImagePath");
    UNICODE_STRING StartU = RTL_CONSTANT_STRING(L"Start");
    UNICODE_STRING TypeU = RTL_CONSTANT_STRING(L"Type");
    UNICODE_STRING UpperFiltersU = RTL_CONSTANT_STRING(L"UpperFilters");
    PWSTR keyboardClass = L"kbdclass\0";

    UNICODE_STRING StringU;
    OBJECT_ATTRIBUTES ObjectAttributes;
    HANDLE hService;
    INFCONTEXT Context;
    PCWSTR Driver, ClassGuid, ImagePath;
    PWSTR FullImagePath;
    ULONG dwValue;
    ULONG Disposition;
    NTSTATUS Status;
    BOOLEAN deviceInstalled = FALSE;

    /* Check if we know the hardware */
    if (!SpInfFindFirstLine(hInf, L"HardwareIdsDatabase", HardwareId, &Context))
        return FALSE;
    if (!INF_GetDataField(&Context, 1, &Driver))
        return FALSE;

    /* Get associated class GUID (if any) */
    if (!INF_GetDataField(&Context, 2, &ClassGuid))
        ClassGuid = NULL;

    /* Find associated driver name */
    /* FIXME: check in other sections too! */
    if (!SpInfFindFirstLine(hInf, L"BootBusExtenders.Load", Driver, &Context)
     && !SpInfFindFirstLine(hInf, L"BusExtenders.Load", Driver, &Context)
     && !SpInfFindFirstLine(hInf, L"SCSI.Load", Driver, &Context)
     && !SpInfFindFirstLine(hInf, L"InputDevicesSupport.Load", Driver, &Context)
     && !SpInfFindFirstLine(hInf, L"Keyboard.Load", Driver, &Context))
    {
        INF_FreeData(ClassGuid);
        INF_FreeData(Driver);
        return FALSE;
    }

    if (!INF_GetDataField(&Context, 1, &ImagePath))
    {
        INF_FreeData(ClassGuid);
        INF_FreeData(Driver);
        return FALSE;
    }

    /* Prepare full driver path */
    dwValue = PathPrefix.MaximumLength + wcslen(ImagePath) * sizeof(WCHAR);
    FullImagePath = (PWSTR)RtlAllocateHeap(ProcessHeap, 0, dwValue);
    if (!FullImagePath)
    {
        DPRINT1("RtlAllocateHeap() failed\n");
        INF_FreeData(ImagePath);
        INF_FreeData(ClassGuid);
        INF_FreeData(Driver);
        return FALSE;
    }
    RtlCopyMemory(FullImagePath, PathPrefix.Buffer, PathPrefix.MaximumLength);
    ConcatPaths(FullImagePath, dwValue / sizeof(WCHAR), 1, ImagePath);

    DPRINT1("Using driver '%S' for device '%S'\n", ImagePath, DeviceId);

    /* Create service key */
    RtlInitUnicodeString(&StringU, Driver);
    InitializeObjectAttributes(&ObjectAttributes, &StringU, OBJ_CASE_INSENSITIVE, hServices, NULL);
    Status = NtCreateKey(&hService, KEY_SET_VALUE, &ObjectAttributes, 0, NULL, REG_OPTION_NON_VOLATILE, &Disposition);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("NtCreateKey('%wZ') failed with status 0x%08x\n", &StringU, Status);
        RtlFreeHeap(ProcessHeap, 0, FullImagePath);
        INF_FreeData(ImagePath);
        INF_FreeData(ClassGuid);
        INF_FreeData(Driver);
        return FALSE;
    }

    /* Fill service key */
    if (Disposition == REG_CREATED_NEW_KEY)
    {
        dwValue = 0;
        NtSetValueKey(hService,
                      &ErrorControlU,
                      0,
                      REG_DWORD,
                      &dwValue,
                      sizeof(dwValue));

        dwValue = 0;
        NtSetValueKey(hService,
                      &StartU,
                      0,
                      REG_DWORD,
                      &dwValue,
                      sizeof(dwValue));

        dwValue = SERVICE_KERNEL_DRIVER;
        NtSetValueKey(hService,
                      &TypeU,
                      0,
                      REG_DWORD,
                      &dwValue,
                      sizeof(dwValue));
    }
    /* HACK: don't put any path in registry */
    NtSetValueKey(hService,
                  &ImagePathU,
                  0,
                  REG_SZ,
                  (PVOID)ImagePath,
                  (wcslen(ImagePath) + 1) * sizeof(WCHAR));

    INF_FreeData(ImagePath);

    if (ClassGuid &&_wcsicmp(ClassGuid, L"{4D36E96B-E325-11CE-BFC1-08002BE10318}") == 0)
    {
        DPRINT1("Installing keyboard class driver for '%S'\n", DeviceId);
        NtSetValueKey(hDeviceKey,
                      &UpperFiltersU,
                      0,
                      REG_MULTI_SZ,
                      keyboardClass,
                      (wcslen(keyboardClass) + 2) * sizeof(WCHAR));
    }

    INF_FreeData(ClassGuid);

    /* Associate device with the service we just filled */
    Status = NtSetValueKey(hDeviceKey,
                           &ServiceU,
                           0,
                           REG_SZ,
                           (PVOID)Driver,
                           (wcslen(Driver) + 1) * sizeof(WCHAR));
    if (NT_SUCCESS(Status))
    {
        /* Restart the device, so it will use the driver we registered */
        deviceInstalled = ResetDevice(DeviceId);
    }

    INF_FreeData(Driver);

    /* HACK: Update driver path */
    NtSetValueKey(hService,
                  &ImagePathU,
                  0,
                  REG_SZ,
                  FullImagePath,
                  (wcslen(FullImagePath) + 1) * sizeof(WCHAR));
    RtlFreeHeap(ProcessHeap, 0, FullImagePath);

    NtClose(hService);

    return deviceInstalled;
}