Initial Windows support.

Author: Chung-Yu Liao

include/nvtop/get_process_info.h

@@ -22,6 +22,11 @@
 #ifndef __GET_PROCESS_INFO_H_
 #define __GET_PROCESS_INFO_H_
 
+#ifdef _WIN32
+#include <windef.h>
+#define pid_t DWORD
+#endif
+
 #include <stdlib.h>
 
 void get_username_from_pid(pid_t pid, size_t size_buffer, char *buffer);

src/CMakeLists.txt

@@ -3,20 +3,33 @@ configure_file(
   "${PROJECT_BINARY_DIR}/include/nvtop/version.h"
   IMMEDIATE @ONLY)
 
-add_executable (nvtop
-  nvtop.c
-  interface.c
-  get_process_info_linux.c
-  extract_gpuinfo.c)
+if(MINGW)
+  add_executable(nvtop
+    nvtop.c
+    interface.c
+    get_process_info_windows.c
+    extract_gpuinfo.c)
+else()
+  add_executable(nvtop
+    nvtop.c
+    interface.c
+    get_process_info_linux.c
+    extract_gpuinfo.c)
+endif()
 
 target_include_directories(nvtop PRIVATE
   $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/include>
   $<BUILD_INTERFACE:${PROJECT_BINARY_DIR}/include>)
 
 set_property(TARGET nvtop PROPERTY C_STANDARD 11)
 
-target_link_libraries(nvtop
-  PRIVATE nvml ncurses m)
+if(MINGW)
+  target_link_libraries(nvtop
+    PRIVATE nvml ncurses m)
+else()
+  target_link_libraries(nvtop
+    PRIVATE nvml ncurses m "-Wl,-Bstatic")
+endif()
 
 install (TARGETS nvtop
   RUNTIME DESTINATION bin)

src/get_process_info_windows.c

@@ -0,0 +1,68 @@
+/*
+ *
+ * Copyright (C) 2019 Chung-Yu Liao <[email protected]>
+ *
+ * This file is part of Nvtop.
+ *
+ * Nvtop is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Nvtop is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with nvtop.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+#include "nvtop/get_process_info.h"
+
+#include <stdio.h>
+#include <string.h>
+#include <winbase.h>
+#include <sddl.h>
+
+void get_username_from_pid(pid_t pid, size_t size_buffer, char *buffer) {
+  HANDLE hProc       = NULL;
+  HANDLE hProcToken  = NULL;
+  DWORD  dwTokenSize = 0u;
+  
+  PTOKEN_USER pTokenUser  = NULL;
+  LPSTR       lpStringSid = NULL;
+  
+  buffer[0] = '\0';
+  
+  hProc = OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION | PROCESS_VM_READ, FALSE, pid);
+  
+  if (!OpenProcessToken(hProc, TOKEN_QUERY, hProcToken)) {
+    return;
+  }
+
+  if (!GetTokenInformation(hProcToken, TokenUser, NULL, dwTokenSize, &dwTokenSize)){
+    CloseHandle(hProcToken);
+    return;
+  }
+  pTokenUser = (TOKEN_USER *)LocalAlloc(LMEM_FIXED, dwTokenSize);
+  if (!GetTokenInformation(hProcToken, TokenUser, pTokenUser, dwTokenSize, &dwTokenSize)){
+    LocalFree(pTokenUser);
+    CloseHandle(hProcToken);
+    return;
+  }
+  
+  if (!ConvertSidToStringSid(pTokenUser->User.Sid, &lpStringSid)) {
+    LocalFree(pTokenUser);
+    CloseHandle(hProcToken);
+    CloseHandle(hProc);
+    return;
+  }
+  strncpy(buffer, lpStringSid, size_buffer);
+  
+  LocalFree(lpStringSid);
+  LocalFree(pTokenUser);
+  CloseHandle(hProcToken);
+  CloseHandle(hProc);
+}

src/interface.c

@@ -19,6 +19,13 @@
  *
  */
 
+#ifdef _WIN32
+#define __USE_MINGW_ANSI_STDIO 1
+#define pid_t DWORD
+#include <windef.h>
+#include <winbase.h>
+#endif
+
 #include "nvtop/interface.h"
 
 #include <stdlib.h>
@@ -988,6 +995,26 @@ static void draw_processes(
   print_processes_on_screen(total_processes, &interface->process);
 }
 
+#ifdef _WIN32
+static const char *signalsName[] = {
+"Cancel",
+"SIGABRT",
+"SIGFPE",
+"SIGILL",
+"SIGINT",
+"SIGSEGV",
+"SIGTERM",
+};
+
+static const int signalsValues[] = {
+SIGABRT,
+SIGFPE ,
+SIGILL ,
+SIGINT ,
+SIGSEGV,
+SIGTERM,
+};
+#else
 static const char *signalsName[] = {
 "Cancel",
 "SIGABRT",
@@ -1050,6 +1077,7 @@ SIGVTALRM,
 SIGXCPU  ,
 SIGXFSZ  ,
 };
+#endif
 
 static const size_t nvtop_num_signals = 28;
 
@@ -1272,7 +1300,15 @@ static void option_do_kill(struct nvtop_interface *inter) {
     return;
   pid_t pid = inter->process.all_process[inter->process.selected_row].pid;
   int sig = signalsValues[inter->option_window.selected_row-1];
-  kill(pid,sig);
+
+#ifdef _WIN32
+  HANDLE hProc;
+  hProc = OpenProcess(PROCESS_TERMINATE, false, pid);
+  TerminateProcess(hProc, sig);
+  CloseHandle(hProc);
+#else
+  kill(pid, sig);
+#endif
 }
 
 static void option_change_sort(struct nvtop_interface *inter) {

src/nvtop.c

@@ -19,6 +19,17 @@
  *
  */
 
+#ifdef _WIN32
+#define __USE_MINGW_ANSI_STDIO 1
+#define setenv(name, value, overwrite) \
+        _putenv_s(name, value)
+#include <windef.h>
+#include <winbase.h>
+#include <wincon.h>
+#include <winnls.h>
+#include <fcntl.h>
+#endif
+
 #include <signal.h>
 #include <stdio.h>
 #include <stdlib.h>
@@ -42,10 +53,12 @@ static void exit_handler(int signum) {
   signal_bits |= STOP_SIGNAL;
 }
 
+#ifndef _WIN32
 static void resize_handler(int signum) {
   (void) signum;
   signal_bits |= RESIZE_SIGNAL;
 }
+#endif
 
 static const char helpstring[] =
 "Available options:\n"
@@ -136,7 +149,12 @@ static size_t update_mask_value(const char *str, size_t entry_mask, bool addTo)
 }
 
 int main (int argc, char **argv) {
+#ifdef _WIN32
+  SetConsoleCP(CP_UTF8);
+  SetConsoleOutputCP(CP_UTF8);
+#else
   (void) setlocale(LC_CTYPE, "");
+#endif
 
   opterr = 0;
   int refresh_interval = 1000;
@@ -195,6 +213,12 @@ int main (int argc, char **argv) {
 
   setenv("ESCDELAY", "10", 1);
 
+#ifdef _WIN32
+  if (signal(SIGINT, exit_handler) == SIG_ERR) {
+    perror("Impossible to set signal handler for SIGINT: ");
+    exit(EXIT_FAILURE);
+  }
+#else
   struct sigaction siga;
   siga.sa_flags = 0;
   sigemptyset(&siga.sa_mask);
@@ -213,6 +237,7 @@ int main (int argc, char **argv) {
     perror("Impossible to set signal handler for SIGQUIT: ");
     exit(EXIT_FAILURE);
   }
+#endif
 
   size_t gpu_mask;
   if (selectedGPU != NULL) {