C/C++ *Example* | "Продвинутое" отключение XP - Компьютерный форум OSzone.net

Несколько вопросов обсуждались в разное время на форуме.
1. Как заставить ХР предупредить о подключенных соединениях при выключении машины;
2. Как прервать отключение машины, если что-то забыл сделать.

Поскольку я сейчас чем-то подобным занимаюсь, набросал небольшую программку, предназначенную для отключения XP машин. Её можно использовать вместо стандарного Shutdown в меню Пуск.

Код:

#ifndef UNICODE 

#define UNICODE 

#endif 

 

#include &lt;stdio.h&gt; 

#include &lt;string.h&gt; 

#include &lt;stdarg.h&gt; 

#include &lt;windows.h&gt; 

#include &lt;winuser.h&gt; 

#include &lt;lm.h&gt; 

 

// собранная информация о сессиях 

static wchar_t * pszSessions = NULL; 

 

// добавление очередной записи к хранилищу 

void save_session(const wchar_t * fmt, ...) 

{ &nbsp; &nbsp; 

 &nbsp; &nbsp;// размер очередной строки, ориентировочно ~100 байт 

 &nbsp; &nbsp;int size = 100; 

 &nbsp; &nbsp;// память под очередную строку 

 &nbsp; &nbsp;wchar_t *p = (wchar_t *) malloc(size * sizeof(wchar_t)); 

 &nbsp; &nbsp;// безопасность 

 &nbsp; &nbsp;if (!p) return; p[0] = '\0'; 

 &nbsp; &nbsp;// формируем строку 

 &nbsp; &nbsp;va_list ap; 

 &nbsp; &nbsp;while (1) 

 &nbsp; &nbsp;{ 

        // попытаемся вместиться в уже выделенную память 

        va_start(ap, fmt); 

        // реальная длина новой строки 

        int n = _vsnwprintf(p, size, fmt, ap); 

        va_end(ap); 

        // если влезли, ок 

        if (n &gt; -1 &amp;&amp; n &lt; size) break; 

        // иначе - увеличиваем буфер под строку 

        size *= 2; 

        if ((p = (wchar_t *) realloc(p, size * sizeof(wchar_t))) == NULL) return; //oops 

 &nbsp; &nbsp;} 

 

 &nbsp; &nbsp;// строку сформирована, надо её добавить к имеющимся 

 &nbsp; &nbsp;size = lstrlen(p); 

 &nbsp; &nbsp;if (pszSessions) size += lstrlen(pszSessions); 

 &nbsp; &nbsp;wchar_t *pn = (wchar_t *)malloc((size + 1)*sizeof(wchar_t)); //eos 

 &nbsp; &nbsp;if (!pn) return; pn[0] = '\0'; 

 &nbsp; &nbsp;//складываем строки 

 &nbsp; &nbsp;if (pszSessions) lstrcpy(pn, pszSessions); 

 &nbsp; &nbsp;lstrcat(pn, p); 

 &nbsp; &nbsp;free(p); 

 &nbsp; &nbsp;free(pszSessions); 

 &nbsp; &nbsp;pszSessions = pn; 

 &nbsp; &nbsp;return; 

} 

 

//int main(int argc, wchar_t *argv[]) 

int WINAPI WinMain(HINSTANCE hInst,HINSTANCE hPrev,LPSTR lpszCmd,int nCmd) 

{ 

 &nbsp; &nbsp;DWORD dwTotalCount = 0; 

 &nbsp; &nbsp;NET_API_STATUS nStatus; 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;// начальная фраза сообщения 

 &nbsp; &nbsp;save_session(L&quot;%s&quot;, L&quot;Warning! Some sessions are not closed:\n\n&quot;); 

 &nbsp; 

 &nbsp; &nbsp;// для всех имеющихся соединений 

 &nbsp; &nbsp;do 

 &nbsp; &nbsp;{ 

        LPSESSION_INFO_10 pBuf = NULL; 

        DWORD dwEntriesRead = 0; 

        DWORD dwTotalEntries = 0; 

        DWORD dwResumeHandle = 0; 

         

        // вызываем очередной NetSessionEnum 

        nStatus = NetSessionEnum(NULL,        //всё локально 

                                NULL,        //для всех сессий 

                                NULL,         //для всех пользователей 

                                10,        //тип запроса хост-юзер-время-простой 

                                (LPBYTE*)&amp;pBuf,        //данные на выходе 

                                MAX_PREFERRED_LENGTH, //длина записи 

                                &amp;dwEntriesRead,        //сколько считано 

                                &amp;dwTotalEntries,                //сколько всего 

                                &amp;dwResumeHandle);        //номер запроса 

         

        // проверяем результат 

        if ((nStatus == NERR_Success) || (nStatus == ERROR_MORE_DATA)) 

        { 

         &nbsp; &nbsp;// запоминаем сессии клиентов 

         &nbsp; &nbsp;if (LPSESSION_INFO_10 pTmpBuf = pBuf) 

                for (int i = 0; i &lt; dwEntriesRead; i++) 

                 &nbsp; &nbsp;if (pTmpBuf) 

                 &nbsp; &nbsp;{ 

                        save_session(L&quot;Client: %s\tUser: %s\tActive: %d\tIdle: %d\n&quot;, 

                                 &nbsp; &nbsp; pTmpBuf-&gt;sesi10_cname, pTmpBuf-&gt;sesi10_username, 

                                 &nbsp; &nbsp; pTmpBuf-&gt;sesi10_time, pTmpBuf-&gt;sesi10_idle_time); 

                        pTmpBuf++; 

                        dwTotalCount++; 

                 &nbsp; &nbsp;} 

        } 

 

        // идём на следующий запрос 

        if (pBuf) 

        { 

         &nbsp; &nbsp;NetApiBufferFree(pBuf); 

         &nbsp; &nbsp;pBuf = NULL; 

        } 

 &nbsp; &nbsp;} while (nStatus == ERROR_MORE_DATA); 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;// если есть сессии, сообщаем об этом выключателю 

 &nbsp; &nbsp;if (dwTotalCount) 

 &nbsp; &nbsp;{ 

        // финальная часть сообщения 

        save_session(L&quot;%s&quot;, L&quot;\nDo you wish to cancel shutdown?\n&quot;); 

        // вывод сообщения 

        if (IDYES == MessageBox(NULL, pszSessions, L&quot;Warning!&quot;, MB_YESNO)) return 0; 

 &nbsp; &nbsp;} 

 

 &nbsp; &nbsp;//чистим память от сообщения - оно нам больше не нужно 

 &nbsp; &nbsp;free(pszSessions); 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;HANDLE hToken; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp;// handle to process token 

 &nbsp; &nbsp;TOKEN_PRIVILEGES tkp; &nbsp; &nbsp; &nbsp; // pointer to token structure 

 

 &nbsp; &nbsp;// Get the current process token handle so we can get shutdown privilege. 

 &nbsp; &nbsp;if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, 

                         &nbsp;&amp;hToken)) return 1; 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;// Get the LUID for shutdown privilege. 

 &nbsp; &nbsp;LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &amp;tkp.Privileges[0].Luid); 

 &nbsp; &nbsp;tkp.PrivilegeCount = 1; &nbsp;// one privilege to set 

 &nbsp; &nbsp;tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;// Get shutdown privilege for this process. 

 &nbsp; &nbsp;AdjustTokenPrivileges(hToken, FALSE, &amp;tkp, 0, (PTOKEN_PRIVILEGES) NULL, 0); 

 &nbsp; &nbsp;if (GetLastError() != ERROR_SUCCESS) return 1; 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;// Вывести окошечко обратного отсчёта 

 &nbsp; &nbsp;if (!InitiateSystemShutdown( 

         &nbsp; &nbsp;NULL, &nbsp; &nbsp;// shut down local computer 

         &nbsp; &nbsp;L&quot;We starts system shutdown.\nIf You have forget to do something, press &quot; 

         &nbsp; &nbsp;L&quot;Ok in message box for stop shutdown.&quot;, &nbsp; // message for user 

         &nbsp; &nbsp;20, &nbsp; &nbsp; &nbsp;// time-out period 

         &nbsp; &nbsp;FALSE, &nbsp; // ask user to close apps 

         &nbsp; &nbsp;FALSE)) //shutdown, no reboot 

        return 1; 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;// Disable shutdown privilege. 

 &nbsp; &nbsp;tkp.Privileges[0].Attributes = 0; 

 &nbsp; &nbsp;AdjustTokenPrivileges(hToken, FALSE, &amp;tkp, 0, (PTOKEN_PRIVILEGES) NULL, 0); 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;MessageBox(NULL, L&quot;Press OK to stop shutdown&quot;, L&quot;Haves forget somehing?&quot;, 

         &nbsp; &nbsp; &nbsp; MB_OK | MB_SYSTEMMODAL); 

 &nbsp; &nbsp;// если мы сюда попали, надо отключать шатдаун 

 &nbsp; &nbsp;if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, 

                         &nbsp;&amp;hToken)) return 1; 

 &nbsp; &nbsp;LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &amp;tkp.Privileges[0].Luid); 

 &nbsp; &nbsp;tkp.PrivilegeCount = 1; 

 &nbsp; &nbsp;tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; 

 &nbsp; &nbsp;AdjustTokenPrivileges(hToken, FALSE, &amp;tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0); 

 &nbsp; &nbsp;if (GetLastError() != ERROR_SUCCESS) return 1; 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;// Prevent the system from shutting down. 

 &nbsp; &nbsp;if ( !AbortSystemShutdown(NULL) ) return 1; 

 &nbsp; &nbsp;// Disable shutdown privilege. 

 &nbsp; &nbsp;tkp.Privileges[0].Attributes = 0; 

 &nbsp; &nbsp;AdjustTokenPrivileges(hToken, FALSE, &amp;tkp, 0, (PTOKEN_PRIVILEGES) NULL, 0); 

 &nbsp; &nbsp; 

 &nbsp; &nbsp;return 0; 

}

FAQ

Q. И как использовать?
A. Компилите. Кладёте ярлык на рабочий стол. При необходимости выключить машину жмёте не Пуск->Выключить, а щёлкаете по этому ярлыку. Он Вас спросит про активные подключения (если они есть). Потом даст 20 секунд на раздумья. Потом выключится.

Q. Почему C++, а не, скажем, VBS?
A. Потому что C++ мне ближе.

Q. Как компилить?
A. Я использую Visual C++ Toolkit. Компиляция:
cl enum_shares.cpp /GA /O1 /link /defaultlib:netapi32.lib /defaultlib:advapi32.lib /defaultlib:user32.lib

Q. А в 9х будет работать?
A. Нет. В 9х сообщение об активных подключениях и так выдается.

Q. Почему UNICODE?
A.Потому что NetSessionEnum

Q. Что так странно (в смысле стиля написания) сформирован текст и комментарии?
A. Потому что при написании использовались фрагменты из PSDK.

Q. А есть ли бинарник?
A. Есть. Тут пока лежит.

Q. Можно использовать в моих программах куски текста?
A. Да, сделано специально для посетителей www.oszone.net

Q. У меня ещё вопросы...
A. Задавайте здесь, я отвечу.