diff options
Diffstat (limited to 'src/detection/packages/packages_windows.c')
| -rw-r--r-- | src/detection/packages/packages_windows.c | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/src/detection/packages/packages_windows.c b/src/detection/packages/packages_windows.c new file mode 100644 index 0000000..a3faee5 --- /dev/null +++ b/src/detection/packages/packages_windows.c @@ -0,0 +1,203 @@ +#include "packages.h" +#include "common/processing.h" +#include "common/strutil.h" +#include "common/path.h" +#include "common/windows/unicode.h" +#include "common/mallocHelper.h" +#include "common/io.h" + +#include <stdalign.h> +#include <windows.h> +#include "common/windows/nt.h" +#include <ntstatus.h> +#include <shlobj.h> + +static uint32_t getNumElements(const char* searchPath, DWORD type, const wchar_t* ignore) { + FF_AUTO_CLOSE_FD HANDLE dfd = CreateFileA(searchPath, FILE_LIST_DIRECTORY | SYNCHRONIZE, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL); + if (dfd == INVALID_HANDLE_VALUE) { + return 0; + } + + bool flag = ignore == NULL; + uint32_t counter = 0; + alignas(8) uint8_t buffer[64 * 1024]; + BOOLEAN firstScan = TRUE; + + size_t ignoreLen = ignore ? wcslen(ignore) : 0; + + while (true) { + IO_STATUS_BLOCK ioStatus = {}; + NTSTATUS status = NtQueryDirectoryFile( + dfd, + NULL, + NULL, + NULL, + &ioStatus, + buffer, + ARRAY_SIZE(buffer), + FileDirectoryInformation, + FALSE, + NULL, + firstScan); + firstScan = FALSE; + + if (!NT_SUCCESS(status) && status != STATUS_BUFFER_OVERFLOW) { + break; + } + + for (FILE_DIRECTORY_INFORMATION* entry = (FILE_DIRECTORY_INFORMATION*) buffer; + ; + entry = (FILE_DIRECTORY_INFORMATION*) ((uint8_t*) entry + entry->NextEntryOffset)) { + if (!(entry->FileAttributes & type)) { + continue; + } + + if (!flag && + ignoreLen == entry->FileNameLength / sizeof(*entry->FileName) && + _wcsnicmp(entry->FileName, ignore, ignoreLen) == 0) { + flag = true; + continue; + } + + counter++; + + if (entry->NextEntryOffset == 0) { + break; + } + } + + if (status == STATUS_SUCCESS) { + break; // No next page + } + } + + if (type == FILE_ATTRIBUTE_DIRECTORY && counter >= 2) { + counter -= 2; // accounting for . and .. + } + + return counter; +} + +static inline void wrapYyjsonFree(yyjson_doc** doc) { + assert(doc); + if (*doc) { + yyjson_doc_free(*doc); + } +} + +static void detectScoop(FFPackagesResult* result) { + FF_STRBUF_AUTO_DESTROY scoopPath = ffStrbufCreateA(MAX_PATH + 3); + ffStrbufAppend(&scoopPath, &instance.state.platform.homeDir); + ffStrbufAppendS(&scoopPath, ".config/scoop/config.json"); + + yyjson_val* root = NULL; + + yyjson_doc* FF_A_CLEANUP(wrapYyjsonFree) doc = yyjson_read_file(scoopPath.chars, 0, NULL, NULL); + if (doc) { + root = yyjson_doc_get_root(doc); + if (!yyjson_is_obj(root)) { + root = NULL; + } + } + + { + ffStrbufClear(&scoopPath); + if (root) { + ffStrbufSetJsonVal(&scoopPath, yyjson_obj_get(root, "root_path")); + } + if (scoopPath.length == 0) { + ffStrbufSet(&scoopPath, &instance.state.platform.homeDir); + ffStrbufAppendS(&scoopPath, "/scoop"); + } + ffStrbufAppendS(&scoopPath, "/apps/"); + result->scoopUser = getNumElements(scoopPath.chars, FILE_ATTRIBUTE_DIRECTORY, L"scoop"); + } + + { + ffStrbufClear(&scoopPath); + if (root) { + ffStrbufSetJsonVal(&scoopPath, yyjson_obj_get(root, "global_path")); + } + if (scoopPath.length == 0) { + PWSTR pPath = NULL; + if (SUCCEEDED(SHGetKnownFolderPath(&FOLDERID_ProgramData, KF_FLAG_DEFAULT, NULL, &pPath))) { + ffStrbufSetWS(&scoopPath, pPath); + CoTaskMemFree(pPath); + } + ffStrbufAppendS(&scoopPath, "/scoop"); + } + ffStrbufAppendS(&scoopPath, "/apps/"); + result->scoopGlobal = getNumElements(scoopPath.chars, FILE_ATTRIBUTE_DIRECTORY, L"scoop"); + } +} + +static void detectChoco(FF_A_UNUSED FFPackagesResult* result) { + const char* chocoInstall = getenv("ChocolateyInstall"); + if (!chocoInstall || chocoInstall[0] == '\0') { + return; + } + + char chocoPath[MAX_PATH + 3]; + char* pend = ffStrCopy(chocoPath, chocoInstall, ARRAY_SIZE(chocoPath)); + ffStrCopy(pend, "/lib/", ARRAY_SIZE(chocoPath) - (size_t) (pend - chocoPath)); + result->choco = getNumElements(chocoPath, FILE_ATTRIBUTE_DIRECTORY, L"choco"); +} + +static void detectPacman(FFPackagesResult* result) { + const char* msystemPrefix = getenv("MSYSTEM_PREFIX"); + if (!msystemPrefix) { + return; + } + + // MSYS2 + char pacmanPath[MAX_PATH + 3]; + char* pend = ffStrCopy(pacmanPath, msystemPrefix, ARRAY_SIZE(pacmanPath)); + ffStrCopy(pend, "/../var/lib/pacman/local/", ARRAY_SIZE(pacmanPath) - (size_t) (pend - pacmanPath)); + result->pacman = getNumElements(pacmanPath, FILE_ATTRIBUTE_DIRECTORY, NULL); +} + +static void detectWinget(FFPackagesResult* result) { + FF_STRBUF_AUTO_DESTROY buffer = ffStrbufCreate(); + if (ffProcessAppendStdOut(&buffer, (char*[]) { + "winget.exe", + "list", + "--disable-interactivity", + NULL, + })) { + return; + } + + uint32_t index = ffStrbufFirstIndexS(&buffer, "--\r\n"); // Ignore garbage and table headers + if (index == buffer.length) { + return; + } + + uint32_t count = 0; + for ( + index += strlen("--\r\n"); + (index = ffStrbufNextIndexC(&buffer, index, '\n')) < buffer.length; + ++index) { + ++count; + } + + if (buffer.chars[buffer.length - 1] != '\n') { // count last line + ++count; + } + + result->winget = count; +} + +void ffDetectPackagesImpl(FFPackagesResult* result, FFPackagesOptions* options) { + if (FF_PACKAGES_IS_ENABLED(options, SCOOP)) { + detectScoop(result); + } + if (FF_PACKAGES_IS_ENABLED(options, CHOCO)) { + detectChoco(result); + } + if (FF_PACKAGES_IS_ENABLED(options, PACMAN)) { + detectPacman(result); + } + if (FF_PACKAGES_IS_ENABLED(options, WINGET)) { + detectWinget(result); + } +} |