Работа с реестром, взрыв мозга

@kpa6uu · Регистрация: 26.11.2014

Author24 — интернет-сервис помощи студентам

Хех, доброго времени суток тебе, форумчанин. В муках уже второй день, никак не могу решить проблему обнаружения данных в реестре. Что я собственно делаю в коде - ищу папку, в которую установлен Steam, далее проверяю, папка "рабочая", или "мёртвая". Рабочая - та папка, внутри которой располагаются два файла:
Первый - сам Steam.exe
Второй - loginusers.vdf(файл создаётся после каждого захода в Steam, т.е. папка живая)

Мёртвая же папка та, которую удалили по-средствам DEL'ита. Т.е. сама папка осталась, а внутри неё пустота.
Но не суть важно.

В общем - достаю путь до папки из реестра -> проверяю наличие файлов -> если файлов нет, то делаю тоже самое только с другим ключом реестра -> если файлов нет даже во второй найденной папке, то проделываю махинацию третий раз.

Надеюсь понятно всё расписал, голова уже не варит, нужно передохнуть немного.
Напоследок - да, я знаю что быдлокод без использования функций, но что поделать, таков мой уровень пока что.

C++

    //#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//
    //#                                                                                                                                #//
    //#                     Начало поиска папки Steam                                                                                 #//
    //#                     Переменная steampath                                                                                     #//
    //#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//
    DWORD BufSize = _MAX_PATH; // Как я понял - буффер
    TCHAR steamfile1[MAX_PATH], steamfile2[MAX_PATH]; // В них храним пути к файлам
    HKEY hKey; // ...
    LSTATUS st; // ...
    TCHAR steampath[MAX_PATH]; // В неё записываем ключ, полученный с реестра(путь к папке)
    TCHAR file1[] = _T("\\Steam.exe"); // Путь к первому файлу относительно "родной" папки Steam
    TCHAR file2[] = _T("\\Config\\loginusers.vdf"); // Путь ко второму файлу. Тоже относительно "родной" папки.
 
    // Начинаю работать с реестром.
    //# Ищу в первом разделе реестра
    st = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
        "SOFTWARE\\Wow6432Node\\Valve\\Steam",
        0,
        KEY_QUERY_VALUE,
        &hKey);
 
    //# Если такой раздел есть
    if (st == ERROR_SUCCESS)
    {
        RegQueryValueEx(hKey, _T("InstallPath"), NULL, NULL, (LPBYTE)steampath, &BufSize); // Записываю полученный ключ в steampath
        RegCloseKey(hKey); // Закрываю ключ
 
        lstrcpy(steamfile1, steampath); // Копирую ключ в заготовленную переменную
        lstrcpy(steamfile2, steampath); // чтобы получить в дальнейшем чистый путь к файлу
        strcat(steamfile1, file1); // steamfile1 - путь к Steam.exe
        strcat(steamfile2, file2); // steamfile2 - путь к Config/loginusers.vdf
 
        std::cout << "steampath 1: " << steampath << "\n"; // Вывожу сообщения для отладки
        std::cout << "Steam.exe: " << steamfile1 << "\n";  // к сожалению не радуют :(
        std::cout << "loginuse.vdf: " << steamfile2 << "\n\n";
    }
 
    // Если файлы не были найдены в папке steampath, то
    // ищем в другом значении реестра другой путь
    if (!FileExists(steamfile1) || !FileExists(steamfile2))
    {
        printf("nety v 1"); // Вывожу сообщение, что в первом ключе реестра файлы найдены не были.
        st = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
            "SOFTWARE\\Valve\\Steam",
            0,
            KEY_QUERY_VALUE,
            &hKey);
        if (st == ERROR_SUCCESS)
        {
            RegQueryValueEx(hKey, _T("InstallPath"), NULL, NULL, (LPBYTE)steampath, &BufSize); // Записываю полученный ключ в steampath
            RegCloseKey(hKey); // Закрываю ключ
 
            lstrcpy(steamfile1, steampath); // Копирую ключ в заготовленную переменную
            lstrcpy(steamfile2, steampath); // чтобы получить в дальнейшем чистый путь к файлу
            strcat(steamfile1, file1); // steamfile1 - путь к Steam.exe
            strcat(steamfile2, file2); // steamfile2 - путь к Config/loginusers.vdf
 
            std::cout << "steampath 2: " << steampath << "\n";
            std::cout << "Steam.exe: " << steamfile1 << "\n";
            std::cout << "loginuse.vdf: " << steamfile2 << "\n\n";
        }
    }
 
    // Если же файлы не были найдены и во втором пути, то
    // начинаю поиск последнего пути по значению реестра
    if (!FileExists(steamfile1) || !FileExists(steamfile2))
    {
        printf("nety v 2");
        st = RegOpenKeyExA(HKEY_CURRENT_USER,
            "Software\\Classes\\steam\\Shell\\Open\\Command",
            0,
            KEY_QUERY_VALUE,
            &hKey);
        if (st == ERROR_SUCCESS)
        {
            RegQueryValueEx(hKey, "assa2", NULL, NULL, (LPBYTE)steampath, &BufSize);
            RegCloseKey(hKey);
 
            lstrcpy(steamfile1, steampath);
            lstrcpy(steamfile2, steampath);
            strcat(steamfile1, file1); // steamfile1 - путь к Steam.exe
            strcat(steamfile2, file2); // steamfile2 - путь к Config/loginusers.vdf
 
            std::cout << "steampath 3: " << steampath << "\n";
            std::cout << "Steam.exe: " << steamfile1 << "\n";
            std::cout << "loginuse.vdf: " << steamfile2 << "\n\n";
        }
    }
 
    //#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//
    //#                                                                                                                                #//
    //#                     Конец поиска папки Steam                                                                                   #//
    //#                     Переменная steampath                                                                                     #//
    //#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//#//

@kpa6uu · 26.11.2014, 17:23 **[ТС]**

Вот скрин, описывающий мою проблему...

@kpa6uu · 26.11.2014, 19:11 **[ТС]**

Ребят, ну хоть намёк дайте куда копать. По логике вещей всё должно работать. Почему выводится одно и тоже?

@demmax2004 · 26.11.2014, 19:16

А почему не проверяешь правильно ли отработала RegQueryValueEx() ???

@kpa6uu · 26.11.2014, 19:30 **[ТС]**

Проверяю. Создаю новый проект VS, пишу в нём код, удостоверяюсь что он правильно работает и копирую в основной проект.

Убежденный · 26.11.2014, 19:41

Сообщение от kpa6uu

Вот скрин, описывающий мою проблему...

А можно на словах описать проблему, а то непонятно как-то ?..

@kpa6uu · 26.11.2014, 19:53 **[ТС]**

Конечно, постараюсь понятно сформулировать...
У меня есть три пути реестра, из которых нужно получить данные:
1. HKEY_LOCAL_MACHINE, "SOFTWARE\Wow6432Node\Valve\Steam\InstallPath"
2. HKEY_LOCAL_MACHINE, "SOFTWARE\Valve\Steam\InstallPath"
3. HKEY_CURRENT_USER, "Software\Classes\Steam\Shell\Open\Command" (Тут нужно вытянуть значение "По умолчанию")

Они хранят в себе путь к папке, в которую установлен Steam. Мне нужно поочерёдно вытаскивать эти значения и проверять, если ли в указанных папках два нужных файла.
Т.е.:
1. Считал путь к папке
2. Проверил, есть ли в этой папке два нужных файла
3.1. Если есть - сохранить путь к этой папке
3.2. Если нет - проделать с новым путём(вытянутым из следующего значения реестра) операции сначала

Я попытался это сделать (код в головном посте), но у меня почему-то везде значения из первого запроса к реестру, хотя код я несколько десятков раз перепроверил, три раза переписывал.

@demmax2004 · 26.11.2014, 19:56

На вид по коду все нормально, если не решишь до завтра проблему, запущу твой код

Убежденный · 26.11.2014, 19:58

Ну так demmax2004 выше подсказал верный путь решения:
добавьте недостающую проверку ошибок. Для RegQueryValueEx.
Почему-то я уверен, что она возвращает ERROR_MORE_DATA вместо
кода успеха - BufSize-то Вы не обновляете !..

@demmax2004 · 26.11.2014, 19:59

Вообще перед каждым вызовом реестра зануляй steampath

Убежденный · 26.11.2014, 20:00

И еще: "Wow6432Node" не нужно указывать в пути к ветке реестра.
Для этого есть флаги KEY_WOW64_32KEY и KEY_WOW64_64KEY.

@kpa6uu · 26.11.2014, 20:19 **[ТС]**

с Wow6432Node как раз таки и получаю, остальные почему-то молчат...
По вашим наказам изменил

C++

1	RegQueryValueEx(hKey, _T("InstallPath"), NULL, NULL, (LPBYTE)steampath, &BufSize); // Записываю полученный ключ в steampath

на

C++

1
2
3

BufSize = NULL;
RegQueryValueEx(hKey, _T("InstallPath"), NULL, NULL, (LPBYTE)steampath, &BufSize); // Записываю полученный ключ в steampath
std::cout << "buffsize2= " << BufSize << " |||\n";

Эффекта, к сожалению, нет. Вот вывод ком. строки:

Убежденный · 26.11.2014, 20:38

kpa6uu, почитайте на MSDN описание функции RegQueryValueEx.
Последний параметр - это [in-out]: на входе размер буфера, в байтах, на
выходе - сколько байт было в него записано. У вас на входе 0.
Да, и без проверки того, что именно возвращает функция RegQueryValueEx
дальнейшие разбирательства не имеют смысла.

@kpa6uu · 26.11.2014, 21:44 **[ТС]**

Две 234-тые ошибки.
Тааакс...
234 ошибка это ERROR_MORE_DATA (согласно winerror.pas)

Убежденный · 26.11.2014, 21:53

О чем и речь.
Схема работы с RegQueryValueEx примерно такая:

1) Выделили буфер (на стеке, через new/malloc или в виде vector<byte> - не суть)
определенного размера, например 100 байт.

2) Установили BufSize = 100.

3) Вызвали функцию RegQueryValueEx. Если успех - она вернет ERROR_SUCCESS и в
BufSize к-во прочитанных байт. Если нет (например, буфер слишком мал) - расширяем
буфер, увеличиваем BufSize и зовем функцию снова. И так далее, пока она не вернет
ERROR_SUCCESS, либо отличный от ERROR_MORE_DATA код.

Да, и еще. Строковые данные, прочитанные из реестра, всегда нужно принудительно
завершать нулем (т.к. строка может храниться в реестре без завершающего нуля).

@kpa6uu · 26.11.2014, 22:09 **[ТС]**

Охо-хо. Спасибо, не можешь помочь с кодом на выделение буфера?

Убежденный · 26.11.2014, 22:11

1.

C++

1	byte Buffer[Size];

2.

C++

1
2
3

byte * pBuffer = new pBuffer[Size];
// ...
delete [] pBuffer;

3.

C++

1	boost::scoped_array<byte> Buffer(new byte[Size]);

4.

C++

1	std::vector<byte> Buffer(Size);

Сам предпочитаю №4.

@kpa6uu · 26.11.2014, 22:35 **[ТС]**

Взялся за гуж, не говори, что не дюж....
Понимаю, что глупый вопрос, но... как мне использовать это?
Просто поставить перед всем кодом, или использовать отдельно для каждого запроса к реестру?
Если же так, то куда вставлять?
Пойму, если не ответишь

@kpa6uu · 26.11.2014, 22:39 **[ТС]**

Если правильно подставил, то всё равно не выходит, видно нужно на второй заход в гугл.

@kpa6uu · 27.11.2014, 00:22 **[ТС]**

До сих пор тоже самое...
Ээх... Кто заглянет в темку - если уж не готовый ответ на подстановку буфера, так киньте статей обучающих работе с ним, буду весьма благодарен

Убежденный Ушел с форума 16478 / 7441 / 1187 Регистрация: 02.05.2013 Сообщений: 11,617 Записей в блоге: 1
	26.11.2014, 20:38	13
	kpa6uu, почитайте на MSDN описание функции RegQueryValueEx. Последний параметр - это [in-out]: на входе размер буфера, в байтах, на выходе - сколько байт было в него записано. У вас на входе 0. Да, и без проверки того, что именно возвращает функция RegQueryValueEx дальнейшие разбирательства не имеют смысла. 1

Убежденный Ушел с форума 16478 / 7441 / 1187 Регистрация: 02.05.2013 Сообщений: 11,617 Записей в блоге: 1
	26.11.2014, 21:53	15
	О чем и речь. Схема работы с RegQueryValueEx примерно такая: 1) Выделили буфер (на стеке, через new/malloc или в виде vector<byte> - не суть) определенного размера, например 100 байт. 2) Установили BufSize = 100. 3) Вызвали функцию RegQueryValueEx. Если успех - она вернет ERROR_SUCCESS и в BufSize к-во прочитанных байт. Если нет (например, буфер слишком мал) - расширяем буфер, увеличиваем BufSize и зовем функцию снова. И так далее, пока она не вернет ERROR_SUCCESS, либо отличный от ERROR_MORE_DATA код. Да, и еще. Строковые данные, прочитанные из реестра, всегда нужно принудительно завершать нулем (т.к. строка может храниться в реестре без завершающего нуля). 1

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	26.11.2014, 22:35 [ТС]	18
	Взялся за гуж, не говори, что не дюж.... Понимаю, что глупый вопрос, но... как мне использовать это? Просто поставить перед всем кодом, или использовать отдельно для каждого запроса к реестру? Если же так, то куда вставлять? Пойму, если не ответишь 0

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	26.11.2014, 17:23 [ТС]	2
	Вот скрин, описывающий мою проблему... Миниатюры 0

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	26.11.2014, 19:11 [ТС]	3
	Ребят, ну хоть намёк дайте куда копать. По логике вещей всё должно работать. Почему выводится одно и тоже? 0

@demmax2004 120 / 142 / 46 Регистрация: 31.10.2014 Сообщений: 721 Записей в блоге: 1
	26.11.2014, 19:16	4
	А почему не проверяешь правильно ли отработала RegQueryValueEx() ??? 0

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	26.11.2014, 19:30 [ТС]	5
	Проверяю. Создаю новый проект VS, пишу в нём код, удостоверяюсь что он правильно работает и копирую в основной проект. 0

Убежденный Ушел с форума 16478 / 7441 / 1187 Регистрация: 02.05.2013 Сообщений: 11,617 Записей в блоге: 1
	26.11.2014, 19:41	6
	Сообщение от kpa6uu Вот скрин, описывающий мою проблему... А можно на словах описать проблему, а то непонятно как-то ?.. 0

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	26.11.2014, 19:53 [ТС]	7
	Конечно, постараюсь понятно сформулировать... У меня есть три пути реестра, из которых нужно получить данные: 1. HKEY_LOCAL_MACHINE, "SOFTWARE\Wow6432Node\Valve\Steam\InstallPath" 2. HKEY_LOCAL_MACHINE, "SOFTWARE\Valve\Steam\InstallPath" 3. HKEY_CURRENT_USER, "Software\Classes\Steam\Shell\Open\Command" (Тут нужно вытянуть значение "По умолчанию") Они хранят в себе путь к папке, в которую установлен Steam. Мне нужно поочерёдно вытаскивать эти значения и проверять, если ли в указанных папках два нужных файла. Т.е.: 1. Считал путь к папке 2. Проверил, есть ли в этой папке два нужных файла 3.1. Если есть - сохранить путь к этой папке 3.2. Если нет - проделать с новым путём(вытянутым из следующего значения реестра) операции сначала Я попытался это сделать (код в головном посте), но у меня почему-то везде значения из первого запроса к реестру, хотя код я несколько десятков раз перепроверил, три раза переписывал. 0

@demmax2004 120 / 142 / 46 Регистрация: 31.10.2014 Сообщений: 721 Записей в блоге: 1
	26.11.2014, 19:56	8
	На вид по коду все нормально, если не решишь до завтра проблему, запущу твой код 1

Убежденный Ушел с форума 16478 / 7441 / 1187 Регистрация: 02.05.2013 Сообщений: 11,617 Записей в блоге: 1
	26.11.2014, 19:58	9
	Ну так demmax2004 выше подсказал верный путь решения: добавьте недостающую проверку ошибок. Для RegQueryValueEx. Почему-то я уверен, что она возвращает ERROR_MORE_DATA вместо кода успеха - BufSize-то Вы не обновляете !.. 1

@demmax2004 120 / 142 / 46 Регистрация: 31.10.2014 Сообщений: 721 Записей в блоге: 1
	26.11.2014, 19:59	10
	Вообще перед каждым вызовом реестра зануляй steampath 0

Убежденный Ушел с форума 16478 / 7441 / 1187 Регистрация: 02.05.2013 Сообщений: 11,617 Записей в блоге: 1
	26.11.2014, 20:00	11
	И еще: "Wow6432Node" не нужно указывать в пути к ветке реестра. Для этого есть флаги KEY_WOW64_32KEY и KEY_WOW64_64KEY. 0

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	26.11.2014, 21:44 [ТС]	14
	Две 234-тые ошибки. Тааакс... 234 ошибка это ERROR_MORE_DATA (согласно winerror.pas) Миниатюры 0

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	26.11.2014, 22:09 [ТС]	16
	Охо-хо. Спасибо, не можешь помочь с кодом на выделение буфера? 0

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	26.11.2014, 22:39 [ТС]	19
	Если правильно подставил, то всё равно не выходит, видно нужно на второй заход в гугл. Миниатюры 0

@kpa6uu 0 / 0 / 0 Регистрация: 26.11.2014 Сообщений: 20
	27.11.2014, 00:22 [ТС]	20
	До сих пор тоже самое... Ээх... Кто заглянет в темку - если уж не готовый ответ на подстановку буфера, так киньте статей обучающих работе с ним, буду весьма благодарен 0