Befator-Inc-Firmen-Netzwerk/MSYS2-packages
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

msys2-unlocker: Minor reorgs + clang-format (WIP)

Browse Source
This commit is contained in:
Ray Donnelly
2014-11-12 00:36:32 +00:00
parent 2da96dc47b
commit d6bddeddd4
1 changed files with 319 additions and 270 deletions
Download Patch File Download Diff File Expand all files Collapse all files

589
msys2-unlocker/main.c
View File

@@ -9,89 +9,209 @@
.email: <mingw.android@gmail.com>.
*/
#include <stdio.h>
#include <tchar.h>
//#define WIN32_MEAN_AND_LEAN
#define WIN32_MEAN_AND_LEAN
#include <windows.h>
#include <psapi.h>
#include <stdio.h>
#include <stdint.h>
#include "resource.h"
typedef enum {
kUnlockHelpOnly,
kUnlockNone,
kUnlockGUI,
kUnlockForce,
} kUnlockAction;
typedef enum {
/* All is good */
kSuccess = 0,
/* User errors */
kUserNoFiles,
kUserTooManyFiles,
kUserExited,
kUserCancelled,
/* Failures */
kFailEnumProc,
kFailForce,
} kExitCode;
typedef struct {
uint32_t lockedFilesMask;
DWORD processId;
} LockingProcess;
/* Global variables plus things
that are updated dynamically */
typedef struct {
HWND appHWnd;
size_t num_processes;
DWORD* p_processes;
} GlobalState;
/* Values from the commandline */
typedef struct {
kUnlockAction unlock_action;
size_t num_filenames;
char** p_filenames;
} Arguments;
HWND FindConsoleHandle(void);
void PrintHelp() {
printf(
"msys2-unlocker: Informs of and / or terminates\n"
" processes locking modules that\n"
" are passed in from the cmdline\n"
"\n"
"Usage: msys2-unlocker [options] files...\n"
"Options:\n"
" --help Display this message\n"
" --force Force quit locking processes\n"
" --gui Present unlocking GUI\n");
}
kExitCode ProcessArguments(int argc, char* argv[], Arguments* args) {
typedef enum {
kArgParseFirst,
kArgParseSwitches = kArgParseFirst,
kArgParseFilenames,
kArgParseCount,
} kArgParsePass;
args->unlock_action = kUnlockNone;
args->num_filenames = 0;
args->p_filenames = NULL;
int argi;
kArgParsePass pass;
size_t num_filename_char = 0;
char* filename_chars;
for (pass = kArgParseFirst; pass < kArgParseCount; ++pass) {
if (pass == kArgParseFilenames) {
args->p_filenames = (char**)malloc((sizeof(char*) * args->num_filenames) +
(sizeof(char) * num_filename_char));
filename_chars = (char*)&args->p_filenames[args->num_filenames];
args->num_filenames = 0;
}
for (argi = 1; argi < argc; ++argi) {
if (argv[argi] != NULL) {
if (!strcmp("--gui", argv[argi])) {
args->unlock_action = kUnlockGUI;
} else if (!strcmp("--force", argv[argi])) {
args->unlock_action = kUnlockForce;
} else if (!strcmp("--help", argv[argi])) {
PrintHelp();
args->unlock_action = kUnlockHelpOnly;
return (kSuccess);
} else {
if (pass == kArgParseSwitches) {
++args->num_filenames;
if (args->num_filenames > 32) {
return (kUserTooManyFiles);
}
num_filename_char += strlen(argv[argi]) + 1;
} else if (pass == kArgParseFilenames) {
args->p_filenames[args->num_filenames++] = filename_chars;
strcpy(filename_chars, argv[argi]);
filename_chars += strlen(argv[argi]) + 1;
}
}
}
}
}
if (args->num_filenames == 0) {
return (kUserNoFiles);
}
return (kSuccess);
}
/* Returns the count, -GetLastError () or -ENOMEM */
ssize_t EnumerateProcessModules(DWORD processId, HMODULE** pp_modules) {
if (pp_modules == NULL) {
return 0;
if (pp_modules == NULL) {
return 0;
}
HANDLE proc_handle = OpenProcess(
PROCESS_QUERY_LIMITED_INFORMATION | PROCESS_VM_READ, FALSE, processId);
if (proc_handle == NULL) {
return -(ssize_t)GetLastError();
}
ssize_t num_entries = 0;
HMODULE* modules = NULL;
size_t bytes_used = 0;
BOOL result = FALSE;
do {
DWORD bytes_needed;
num_entries += 1024;
bytes_used = num_entries * sizeof(DWORD);
modules = (HMODULE*)alloca(bytes_used);
if (modules == NULL) {
CloseHandle(proc_handle);
return -ENOMEM;
}
HANDLE proc_handle = OpenProcess (PROCESS_QUERY_LIMITED_INFORMATION | PROCESS_VM_READ, FALSE, processId);
if (proc_handle == NULL) {
return -(ssize_t)GetLastError ();
result =
EnumProcessModules(proc_handle, modules, bytes_used, &bytes_needed);
if (result != FALSE) {
num_entries = bytes_needed / sizeof(DWORD);
bytes_used = bytes_needed;
}
} while (result == FALSE);
ssize_t num_entries = 0;
HMODULE* modules = NULL;
size_t bytes_used = 0;
BOOL result = FALSE;
*pp_modules = malloc(bytes_used);
if (*pp_modules == NULL) {
CloseHandle(proc_handle);
return -ENOMEM;
}
memcpy(*pp_modules, modules, bytes_used);
CloseHandle(proc_handle);
do {
DWORD bytes_needed;
num_entries += 1024;
bytes_used = num_entries * sizeof (DWORD);
modules = (HMODULE*)alloca (bytes_used);
if (modules == NULL) {
CloseHandle (proc_handle);
return -ENOMEM;
}
result = EnumProcessModules (proc_handle, modules, bytes_used, &bytes_needed);
if (result != FALSE) {
num_entries = bytes_needed / sizeof (DWORD);
bytes_used = bytes_needed;
}
} while (result == FALSE);
*pp_modules = malloc (bytes_used);
if (*pp_modules == NULL) {
CloseHandle (proc_handle);
return -ENOMEM;
}
memcpy (*pp_modules, modules, bytes_used);
CloseHandle (proc_handle);
return num_entries;
return num_entries;
}
uint32_t CheckForConflictingFiles(DWORD processId, char** filenames, size_t num_files) {
HMODULE* modules;
uint32_t result = 0;
ssize_t num_modules = EnumerateProcessModules (processId, &modules);
if (num_modules <= 0) {
return 0;
uint32_t CheckForConflictingFiles(DWORD processId, char** filenames,
size_t num_files) {
HMODULE* modules;
uint32_t result = 0;
ssize_t num_modules = EnumerateProcessModules(processId, &modules);
if (num_modules <= 0) {
return 0;
}
HANDLE proc_handle = OpenProcess(
PROCESS_QUERY_LIMITED_INFORMATION | PROCESS_VM_READ, FALSE, processId);
if (proc_handle == NULL) {
return -(ssize_t)GetLastError();
}
size_t fid;
ssize_t mid;
char module_name[1024];
for (mid = 0; mid < num_modules; ++mid) {
if (GetModuleFileNameExA(proc_handle, modules[mid], module_name,
sizeof(module_name) / sizeof(module_name[0])))
printf("\t%s\n", module_name);
for (fid = 0; fid < num_files; ++fid) {
if (!strcmp(module_name, filenames[fid])) {
result |= (1 << fid);
}
}
}
HANDLE proc_handle = OpenProcess (PROCESS_QUERY_LIMITED_INFORMATION | PROCESS_VM_READ, FALSE, processId);
if (proc_handle == NULL) {
return -(ssize_t)GetLastError ();
}
CloseHandle(proc_handle);
size_t fid;
ssize_t mid;
char module_name[1024];
for (mid = 0; mid < num_modules; ++mid) {
if (GetModuleFileNameExA (proc_handle, modules[mid], module_name,
sizeof(module_name) / sizeof(module_name[0])))
printf ("\t%s\n", module_name);
for (fid = 0; fid < num_files; ++fid) {
if (!strcmp(module_name, filenames[fid])) {
result |= (1<<fid);
}
}
}
CloseHandle (proc_handle);
free (modules);
return result;
free(modules);
return (result);
}
/* Returns error codes as negative numbers.
@@ -100,43 +220,44 @@ uint32_t CheckForConflictingFiles(DWORD processId, char** filenames, size_t num_
-3 == malloc failed (out of heap space)
*/
ssize_t EnumerateProcesses(DWORD** pp_processes) {
ssize_t num_entries = 0;
DWORD* processes = NULL;
size_t bytes_used = 0;
BOOL result = FALSE;
ssize_t num_entries = 0;
DWORD* processes = NULL;
size_t bytes_used = 0;
BOOL result = FALSE;
if (pp_processes == NULL) {
return -1;
if (pp_processes == NULL) {
return -1;
}
do {
DWORD bytes_needed;
num_entries += 1024;
bytes_used = num_entries * sizeof(DWORD);
processes = (DWORD*)alloca(bytes_used);
if (processes == NULL) {
return -2;
}
do {
DWORD bytes_needed;
num_entries += 1024;
bytes_used = num_entries * sizeof(DWORD);
processes = (DWORD*)alloca (bytes_used);
if (processes == NULL) {
return -2;
}
result = EnumProcesses (processes, bytes_used, &bytes_needed);
if (result != FALSE) {
num_entries = bytes_needed / sizeof(DWORD);
bytes_used = bytes_needed;
}
} while (result == FALSE);
*pp_processes = malloc (bytes_used);
if (*pp_processes == NULL) {
return -3;
result = EnumProcesses(processes, bytes_used, &bytes_needed);
if (result != FALSE) {
num_entries = bytes_needed / sizeof(DWORD);
bytes_used = bytes_needed;
}
memcpy (*pp_processes, processes, bytes_used);
return num_entries;
} while (result == FALSE);
*pp_processes = malloc(bytes_used);
if (*pp_processes == NULL) {
return -3;
}
memcpy(*pp_processes, processes, bytes_used);
return (num_entries);
}
/* NYI: http://processhacker.sourceforge.net/forums/viewtopic.php?f=8&t=1584
http://msdn.microsoft.com/en-us/library/windows/desktop/aa813708.aspx
ssize_t GetHandlesForProcess(DWORD processId) {
HANDLE proc_handle = OpenProcess (PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, processId);
HANDLE proc_handle = OpenProcess (PROCESS_QUERY_INFORMATION |
PROCESS_VM_READ, FALSE, processId);
if (proc_handle == NULL) {
DWORD last_error = GetLastError ();
return -last_error;
@@ -150,196 +271,124 @@ ssize_t GetHandlesForProcess(DWORD processId) {
}
*/
typedef enum {
kArgParseFirst,
kArgParseSwitches = kArgParseFirst,
kArgParseFilenames,
kArgParseCount,
} kArgParsePass;
typedef enum {
kUnlockNone,
kUnlockGUI,
kUnlockForce,
} kUnlockAction;
typedef enum {
/* All is good */
kSuccess = 0,
/* User errors */
kUserNoFiles,
kUserTooManyFiles,
kUserExited,
kUserCancelled,
/* Failures */
kFailEnumProc,
kFailForce,
} kExitCode;
typedef struct {
uint32_t lockedFilesMask;
DWORD processId;
} LockingProcess;
typedef struct {
kUnlockAction unlock_action;
HWND appHWnd;
size_t num_filenames;
char** p_filenames;
} Arguments;
HWND FindConsoleHandle() {
const char alphabet[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
char title[33];
char old_title[512];
const char alphabet[] =
"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
char title[33];
char old_title[512];
size_t size = sizeof (title);
size_t i;
for (i = 0; i < size - 1; ++i) {
title[i] = alphabet[rand() % (int) (sizeof (alphabet) - 1)];
}
title[i] = '\0';
if (GetConsoleTitleA (old_title, sizeof(old_title)/sizeof(old_title[0])) == 0) {
return NULL;
}
SetConsoleTitleA (title);
Sleep(40);
HWND wnd = FindWindowA (NULL, title);
SetConsoleTitleA (old_title);
return wnd;
size_t size = sizeof(title);
size_t i;
for (i = 0; i < size - 1; ++i) {
title[i] = alphabet[rand() % (int)(sizeof(alphabet) - 1)];
}
title[i] = '\0';
if (GetConsoleTitleA(old_title, sizeof(old_title) / sizeof(old_title[0])) ==
0) {
return NULL;
}
SetConsoleTitleA(title);
Sleep(40);
HWND wnd = FindWindowA(NULL, title);
SetConsoleTitleA(old_title);
return wnd;
}
INT_PTR CALLBACK DialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) {
(void)lParam;
switch(uMsg)
{
case WM_INITDIALOG: {
return TRUE;
break;
}
case WM_COMMAND: {
switch(wParam)
{
case IDOK:
{
EndDialog (hwndDlg, 0);
return TRUE;
break;
}
}
break;
}
case WM_SIZE: {
HWND hEdit;
RECT rcClient;
GetClientRect (hwndDlg, &rcClient);
hEdit = GetDlgItem (hwndDlg, IDC_UNLOCKERDIALOG);
SetWindowPos (hEdit, NULL, 0, 0, rcClient.right, rcClient.bottom, SWP_NOZORDER);
break;
}
case WM_CLOSE: {
EndDialog (hwndDlg, kUserExited);
return TRUE;
break;
}
default: {
break;
}
INT_PTR CALLBACK
DialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) {
GlobalState* state = (GlobalState*)lParam;
switch (uMsg) {
case WM_INITDIALOG: {
return TRUE;
break;
}
return FALSE;
case WM_COMMAND: {
switch (wParam) {
case IDOK: {
EndDialog(hwndDlg, 0);
return TRUE;
break;
}
}
break;
}
case WM_SIZE: {
HWND hEdit;
RECT rcClient;
GetClientRect(hwndDlg, &rcClient);
hEdit = GetDlgItem(hwndDlg, IDC_UNLOCKERDIALOG);
SetWindowPos(hEdit, NULL, 0, 0, rcClient.right, rcClient.bottom,
SWP_NOZORDER);
break;
}
case WM_CLOSE: {
EndDialog(hwndDlg, kUserExited);
return TRUE;
break;
}
default: { break; }
}
return FALSE;
}
kExitCode ProcessArguments(int argc, char* argv[], Arguments* args) {
args->unlock_action = kUnlockNone;
args->num_filenames = 0;
args->p_filenames = NULL;
args->appHWnd = NULL;
int argi;
kArgParsePass pass;
size_t num_filename_char = 0;
char* filename_chars;
args->appHWnd = FindConsoleHandle ();
/* Process arguments */
for (pass = kArgParseFirst; pass < kArgParseCount; ++pass) {
if (pass == kArgParseFilenames) {
args->p_filenames = (char**)malloc ((sizeof(char*) * args->num_filenames) +
(sizeof(char) * num_filename_char));
filename_chars = (char*)&args->p_filenames[args->num_filenames];
args->num_filenames = 0;
}
for (argi = 1; argi < argc; ++argi) {
if (argv[argi] != NULL) {
if (!strcmp ("--gui", argv[argi])) {
args->unlock_action = kUnlockGUI;
} else if (!strcmp ("--force", argv[argi])) {
args->unlock_action = kUnlockForce;
} else {
if (pass == kArgParseSwitches) {
++args->num_filenames;
if (args->num_filenames > 32) {
return (kUserTooManyFiles);
}
num_filename_char += strlen (argv[argi]) + 1;
}
else if (pass == kArgParseFilenames) {
args->p_filenames[args->num_filenames++] = filename_chars;
strcpy (filename_chars, argv[argi]);
filename_chars += strlen (argv[argi]) + 1;
}
}
}
kExitCode UpdateState(Arguments* args, GlobalState* state) {
ssize_t i, num_processes = EnumerateProcesses(&state->p_processes);
if (num_processes < 0) {
state->num_processes = 0;
return (kFailEnumProc);
}
state->num_processes = (size_t)num_processes;
for (i = 0; i < num_processes; ++i) {
uint32_t conflicts = CheckForConflictingFiles(
state->p_processes[i], args->p_filenames, args->num_filenames);
if (conflicts) {
if (args->unlock_action == kUnlockGUI) {
INT_PTR result =
DialogBoxParamA(NULL, MAKEINTRESOURCEA(IDC_UNLOCKERDIALOG),
state->appHWnd, DialogProc, (LPARAM)state);
if (result == kUserExited) {
return (kUserExited);
}
}
}
if (args->num_filenames == 0) {
return (kUserNoFiles);
}
return (kSuccess);
}
}
int main(int argc, char* argv[]) {
Arguments args;
kExitCode args_result = ProcessArguments (argc, argv, &args);
if (args_result == kUserNoFiles) {
fprintf (stderr, "No filenames specified.\n");
return (kUserNoFiles);
} else if (args_result == kUserTooManyFiles) {
fprintf(stderr, "Too many filenames specified, max is 32.\n");
return (kUserTooManyFiles);
Arguments args;
GlobalState state;
state.appHWnd = FindConsoleHandle();
kExitCode args_result = ProcessArguments(argc, argv, &args);
if (args_result == kUserNoFiles) {
fprintf(stderr, "No filenames specified.\n");
return (kUserNoFiles);
} else if (args_result == kUserTooManyFiles) {
fprintf(stderr, "Too many filenames specified, max is 32.\n");
return (kUserTooManyFiles);
}
if (args.unlock_action == kUnlockHelpOnly) {
return (kSuccess);
}
do {
kExitCode update_result = UpdateState(&args, &state);
if (update_result == kFailEnumProc) {
fprintf(stderr, "Failed to enumerate processes.\n");
return (kFailEnumProc);
}
do {
DWORD* p_processes = NULL;
ssize_t i, num_entries = EnumerateProcesses (&p_processes);
if (num_entries < 0) {
fprintf (stderr, "Failed to enumerate processes.\n");
return (kFailEnumProc);
}
for (i = 0; i < num_entries; ++i) {
uint32_t conflicts = CheckForConflictingFiles (p_processes[i], args.p_filenames, args.num_filenames);
if (conflicts) {
if (args.unlock_action == kUnlockGUI) {
INT_PTR result = DialogBoxA(NULL, MAKEINTRESOURCEA(IDC_UNLOCKERDIALOG), args.appHWnd, DialogProc);
if (result == kUserExited) {
return (kUserExited);
}
}
}
}
} while (0);
} while (0);
return (kSuccess);
return (kSuccess);
}