git.distorted.org.uk Git - u/mdw/putty/blob - pageantc.c

   1 /*
   2  * Pageant client code.
   3  */
   4
   5 #include <windows.h>
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8
   9 #include "puttymem.h"
  10
  11 #define AGENT_COPYDATA_ID 0x804e50ba   /* random goop */
  12 #define AGENT_MAX_MSGLEN  8192
  13
  14 #ifdef TESTMODE
  15 #define debug(x) (printf x)
  16 #else
  17 #define debug(x)
  18 #endif
  19
  20 #define GET_32BIT(cp) \
  21     (((unsigned long)(unsigned char)(cp)[0] << 24) | \
  22     ((unsigned long)(unsigned char)(cp)[1] << 16) | \
  23     ((unsigned long)(unsigned char)(cp)[2] << 8) | \
  24     ((unsigned long)(unsigned char)(cp)[3]))
  25
  26 int agent_exists(void)
  27 {
  28     HWND hwnd;
  29     hwnd = FindWindow("Pageant", "Pageant");
  30     if (!hwnd)
  31         return FALSE;
  32     else
  33         return TRUE;
  34 }
  35
  36 void agent_query(void *in, int inlen, void **out, int *outlen)
  37 {
  38     HWND hwnd;
  39     char mapname[64];
  40     HANDLE filemap;
  41     unsigned char *p, *ret;
  42     int id, retlen;
  43     COPYDATASTRUCT cds;
  44
  45     *out = NULL;
  46     *outlen = 0;
  47
  48     hwnd = FindWindow("Pageant", "Pageant");
  49     debug(("hwnd is %p\n", hwnd));
  50     if (!hwnd)
  51         return;
  52     sprintf(mapname, "PageantRequest%08x", (unsigned)GetCurrentThreadId());
  53     filemap = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE,
  54                                 0, AGENT_MAX_MSGLEN, mapname);
  55     if (!filemap)
  56         return;
  57     p = MapViewOfFile(filemap, FILE_MAP_WRITE, 0, 0, 0);
  58     memcpy(p, in, inlen);
  59     cds.dwData = AGENT_COPYDATA_ID;
  60     cds.cbData = 1 + strlen(mapname);
  61     cds.lpData = mapname;
  62     id = SendMessage(hwnd, WM_COPYDATA, (WPARAM) NULL, (LPARAM) & cds);
  63     debug(("return is %d\n", id));
  64     if (id > 0) {
  65         retlen = 4 + GET_32BIT(p);
  66         debug(("len is %d\n", retlen));
  67         ret = snewn(retlen, unsigned char);
  68         if (ret) {
  69             memcpy(ret, p, retlen);
  70             *out = ret;
  71             *outlen = retlen;
  72         }
  73     }
  74     UnmapViewOfFile(p);
  75     CloseHandle(filemap);
  76 }
  77
  78 #ifdef TESTMODE
  79
  80 int main(void)
  81 {
  82     void *msg;
  83     int len;
  84     int i;
  85
  86     agent_query("\0\0\0\1\1", 5, &msg, &len);
  87     debug(("%d:", len));
  88     for (i = 0; i < len; i++)
  89         debug((" %02x", ((unsigned char *) msg)[i]));
  90     debug(("\n"));
  91     return 0;
  92 }
  93
  94 #endif