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 int agent_query(void *in, int inlen, void **out, int *outlen,
  37                 void (*callback)(void *, void *, int), void *callback_ctx)
  38 {
  39     HWND hwnd;
  40     char mapname[64];
  41     HANDLE filemap;
  42     unsigned char *p, *ret;
  43     int id, retlen;
  44     COPYDATASTRUCT cds;
  45
  46     *out = NULL;
  47     *outlen = 0;
  48
  49     hwnd = FindWindow("Pageant", "Pageant");
  50     debug(("hwnd is %p\n", hwnd));
  51     if (!hwnd)
  52         return 1;                      /* *out == NULL, so failure */
  53     sprintf(mapname, "PageantRequest%08x", (unsigned)GetCurrentThreadId());
  54     filemap = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE,
  55                                 0, AGENT_MAX_MSGLEN, mapname);
  56     if (!filemap)
  57         return 1;                      /* *out == NULL, so failure */
  58     p = MapViewOfFile(filemap, FILE_MAP_WRITE, 0, 0, 0);
  59     memcpy(p, in, inlen);
  60     cds.dwData = AGENT_COPYDATA_ID;
  61     cds.cbData = 1 + strlen(mapname);
  62     cds.lpData = mapname;
  63     id = SendMessage(hwnd, WM_COPYDATA, (WPARAM) NULL, (LPARAM) & cds);
  64     debug(("return is %d\n", id));
  65     if (id > 0) {
  66         retlen = 4 + GET_32BIT(p);
  67         debug(("len is %d\n", retlen));
  68         ret = snewn(retlen, unsigned char);
  69         if (ret) {
  70             memcpy(ret, p, retlen);
  71             *out = ret;
  72             *outlen = retlen;
  73         }
  74     }
  75     UnmapViewOfFile(p);
  76     CloseHandle(filemap);
  77
  78     return 1;
  79 }
  80
  81 #ifdef TESTMODE
  82
  83 int main(void)
  84 {
  85     void *msg;
  86     int len;
  87     int i;
  88
  89     agent_query("\0\0\0\1\1", 5, &msg, &len);
  90     debug(("%d:", len));
  91     for (i = 0; i < len; i++)
  92         debug((" %02x", ((unsigned char *) msg)[i]));
  93     debug(("\n"));
  94     return 0;
  95 }
  96
  97 #endif