5c58ad2d |
1 | /* |
2 | * Pageant client code. |
3 | */ |
4 | |
5 | #include <windows.h> |
6 | #include <stdio.h> |
7 | #include <stdlib.h> |
8 | |
dcbde236 |
9 | #include "puttymem.h" |
10 | |
d70f60ae |
11 | #define AGENT_COPYDATA_ID 0x804e50ba /* random goop */ |
12 | #define AGENT_MAX_MSGLEN 8192 |
13 | |
5c58ad2d |
14 | #ifdef TESTMODE |
15 | #define debug(x) (printf x) |
16 | #else |
17 | #define debug(x) |
18 | #endif |
19 | |
d70f60ae |
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 | |
32874aea |
26 | int agent_exists(void) |
27 | { |
5c58ad2d |
28 | HWND hwnd; |
29 | hwnd = FindWindow("Pageant", "Pageant"); |
30 | if (!hwnd) |
32874aea |
31 | return FALSE; |
5c58ad2d |
32 | else |
32874aea |
33 | return TRUE; |
5c58ad2d |
34 | } |
35 | |
839f10db |
36 | int agent_query(void *in, int inlen, void **out, int *outlen, |
37 | void (*callback)(void *, void *, int), void *callback_ctx) |
32874aea |
38 | { |
5c58ad2d |
39 | HWND hwnd; |
40 | char mapname[64]; |
41 | HANDLE filemap; |
d70f60ae |
42 | unsigned char *p, *ret; |
5c58ad2d |
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) |
839f10db |
52 | return 1; /* *out == NULL, so failure */ |
2d466ffd |
53 | sprintf(mapname, "PageantRequest%08x", (unsigned)GetCurrentThreadId()); |
d70f60ae |
54 | filemap = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, |
32874aea |
55 | 0, AGENT_MAX_MSGLEN, mapname); |
d70f60ae |
56 | if (!filemap) |
839f10db |
57 | return 1; /* *out == NULL, so failure */ |
d70f60ae |
58 | p = MapViewOfFile(filemap, FILE_MAP_WRITE, 0, 0, 0); |
59 | memcpy(p, in, inlen); |
60 | cds.dwData = AGENT_COPYDATA_ID; |
32874aea |
61 | cds.cbData = 1 + strlen(mapname); |
d70f60ae |
62 | cds.lpData = mapname; |
32874aea |
63 | id = SendMessage(hwnd, WM_COPYDATA, (WPARAM) NULL, (LPARAM) & cds); |
5c58ad2d |
64 | debug(("return is %d\n", id)); |
65 | if (id > 0) { |
32874aea |
66 | retlen = 4 + GET_32BIT(p); |
67 | debug(("len is %d\n", retlen)); |
3d88e64d |
68 | ret = snewn(retlen, unsigned char); |
32874aea |
69 | if (ret) { |
70 | memcpy(ret, p, retlen); |
71 | *out = ret; |
72 | *outlen = retlen; |
73 | } |
5c58ad2d |
74 | } |
d70f60ae |
75 | UnmapViewOfFile(p); |
76 | CloseHandle(filemap); |
839f10db |
77 | |
78 | return 1; |
5c58ad2d |
79 | } |
80 | |
81 | #ifdef TESTMODE |
82 | |
32874aea |
83 | int main(void) |
84 | { |
5c58ad2d |
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++) |
32874aea |
92 | debug((" %02x", ((unsigned char *) msg)[i])); |
5c58ad2d |
93 | debug(("\n")); |
94 | return 0; |
95 | } |
96 | |
97 | #endif |