When looking for a local username on Windows, if we can get hold of the

[u/mdw/putty] / windows / winmisc.c

diff --git a/windows/winmisc.c b/windows/winmisc.c
index c8b92a6..c7ac835 100644 (file)
--- a/windows/winmisc.c
+++ b/windows/winmisc.c
@@ -5,17 +5,10 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include "putty.h"
+#include <security.h>
 
 OSVERSIONINFO osVersion;
 
-void platform_get_x11_auth(char *display, int *proto,
-                           unsigned char *data, int *datalen)
-{
-    /* We don't support this at all under Windows. */
-}
-
-const char platform_x11_best_transport[] = "localhost";
-
 char *platform_get_x_display(void) {
     /* We may as well check for DISPLAY in case it's useful. */
     return dupstr(getenv("DISPLAY"));
@@ -48,72 +41,61 @@ char *get_username(void)
 {
     DWORD namelen;
     char *user;
+    int got_username = FALSE;
+    DECL_WINDOWS_FUNCTION(static, BOOLEAN, GetUserNameExA,
+                         (EXTENDED_NAME_FORMAT, LPSTR, PULONG));
+
+    {
+       static int tried_usernameex = FALSE;
+       if (!tried_usernameex) {
+           /* Not available on Win9x, so load dynamically */
+           HMODULE secur32 = LoadLibrary("SECUR32.DLL");
+           GET_WINDOWS_FUNCTION(secur32, GetUserNameExA);
+           tried_usernameex = TRUE;
+       }
+    }
 
-    namelen = 0;
-    if (GetUserName(NULL, &namelen) == FALSE) {
+    if (p_GetUserNameExA) {
        /*
-        * Apparently this doesn't work at least on Windows XP SP2.
-        * Thus assume a maximum of 256. It will fail again if it
-        * doesn't fit.
+        * If available, use the principal -- this avoids the problem
+        * that the local username is case-insensitive but Kerberos
+        * usernames are case-sensitive.
         */
-       namelen = 256;
-    }
-
-    user = snewn(namelen, char);
-    GetUserName(user, &namelen);
-
-    return user;
-}
 
-int SaneDialogBox(HINSTANCE hinst,
-                 LPCTSTR tmpl,
-                 HWND hwndparent,
-                 DLGPROC lpDialogFunc)
-{
-    WNDCLASS wc;
-    HWND hwnd;
-    MSG msg;
-    int flags;
-    int ret;
-    int gm;
-
-    wc.style = CS_DBLCLKS | CS_SAVEBITS | CS_BYTEALIGNWINDOW;
-    wc.lpfnWndProc = DefDlgProc;
-    wc.cbClsExtra = 0;
-    wc.cbWndExtra = DLGWINDOWEXTRA + 8;
-    wc.hInstance = hinst;
-    wc.hIcon = NULL;
-    wc.hCursor = LoadCursor(NULL, IDC_ARROW);
-    wc.hbrBackground = (HBRUSH) (COLOR_BACKGROUND +1);
-    wc.lpszMenuName = NULL;
-    wc.lpszClassName = "PuTTYConfigBox";
-    RegisterClass(&wc);
-
-    hwnd = CreateDialog(hinst, tmpl, hwndparent, lpDialogFunc);
-
-    SetWindowLong(hwnd, BOXFLAGS, 0); /* flags */
-    SetWindowLong(hwnd, BOXRESULT, 0); /* result from SaneEndDialog */
-
-    while ((gm=GetMessage(&msg, NULL, 0, 0)) > 0) {
-       flags=GetWindowLong(hwnd, BOXFLAGS);
-       if (!(flags & DF_END) && !IsDialogMessage(hwnd, &msg))
-           DispatchMessage(&msg);
-       if (flags & DF_END)
-           break;
+       /* Get the length */
+       namelen = 0;
+       (void) p_GetUserNameExA(NameUserPrincipal, NULL, &namelen);
+
+       user = snewn(namelen, char);
+       got_username = p_GetUserNameExA(NameUserPrincipal, user, &namelen);
+       if (got_username) {
+           char *p = strchr(user, '@');
+           if (p) *p = 0;
+       } else {
+           sfree(user);
+       }
     }
 
-    if (gm == 0)
-        PostQuitMessage(msg.wParam); /* We got a WM_QUIT, pass it on */
+    if (!got_username) {
+       /* Fall back to local user name */
+       namelen = 0;
+       if (GetUserName(NULL, &namelen) == FALSE) {
+           /*
+            * Apparently this doesn't work at least on Windows XP SP2.
+            * Thus assume a maximum of 256. It will fail again if it
+            * doesn't fit.
+            */
+           namelen = 256;
+       }
 
-    ret=GetWindowLong(hwnd, BOXRESULT);
-    DestroyWindow(hwnd);
-    return ret;
-}
+       user = snewn(namelen, char);
+       got_username = GetUserName(user, &namelen);
+       if (!got_username) {
+           sfree(user);
+       }
+    }
 
-void SaneEndDialog(HWND hwnd, int ret)
-{
-    SetWindowLong(hwnd, BOXRESULT, ret);
-    SetWindowLong(hwnd, BOXFLAGS, DF_END);
+    return got_username ? user : NULL;
 }
 
 BOOL init_winver(void)