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