10 #define AGENT_COPYDATA_ID 0x804e50ba /* random goop */
11 #define AGENT_MAX_MSGLEN 8192
13 #define GET_32BIT(cp) \
14 (((unsigned long)(unsigned char)(cp)[0] << 24) | \
15 ((unsigned long)(unsigned char)(cp)[1] << 16) | \
16 ((unsigned long)(unsigned char)(cp)[2] << 8) | \
17 ((unsigned long)(unsigned char)(cp)[3]))
19 int agent_exists(void)
22 hwnd = FindWindow("Pageant", "Pageant");
29 struct agent_query_data {
31 unsigned char *mapping;
35 void (*callback)(void *, void *, int);
39 DWORD WINAPI agent_query_thread(LPVOID param)
41 struct agent_query_data *data = (struct agent_query_data *)param;
45 id = SendMessage(data->hwnd, WM_COPYDATA, (WPARAM) NULL,
49 retlen = 4 + GET_32BIT(data->mapping);
50 ret = snewn(retlen, unsigned char);
52 memcpy(ret, data->mapping, retlen);
57 UnmapViewOfFile(data->mapping);
58 CloseHandle(data->handle);
61 agent_schedule_callback(data->callback, data->callback_ctx, ret, retlen);
66 int agent_query(void *in, int inlen, void **out, int *outlen,
67 void (*callback)(void *, void *, int), void *callback_ctx)
72 unsigned char *p, *ret;
79 hwnd = FindWindow("Pageant", "Pageant");
81 return 1; /* *out == NULL, so failure */
82 mapname = dupprintf("PageantRequest%08x", (unsigned)GetCurrentThreadId());
83 filemap = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE,
84 0, AGENT_MAX_MSGLEN, mapname);
86 return 1; /* *out == NULL, so failure */
87 p = MapViewOfFile(filemap, FILE_MAP_WRITE, 0, 0, 0);
89 cds.dwData = AGENT_COPYDATA_ID;
90 cds.cbData = 1 + strlen(mapname);
92 if (callback != NULL && !(flags & FLAG_SYNCAGENT)) {
94 * We need an asynchronous Pageant request. Since I know of
95 * no way to stop SendMessage from blocking the thread it's
96 * called in, I see no option but to start a fresh thread.
97 * When we're done we'll PostMessage the result back to our
98 * main window, so that the callback is done in the primary
99 * thread to avoid concurrency.
101 struct agent_query_data *data = snew(struct agent_query_data);
104 data->handle = filemap;
105 data->mapname = mapname;
106 data->callback = callback;
107 data->callback_ctx = callback_ctx;
108 data->cds = cds; /* structure copy */
110 if (CreateThread(NULL, 0, agent_query_thread, data, 0, &threadid))
116 * The user either passed a null callback (indicating that the
117 * query is required to be synchronous) or CreateThread failed.
118 * Either way, we need a synchronous request.
120 id = SendMessage(hwnd, WM_COPYDATA, (WPARAM) NULL, (LPARAM) & cds);
122 retlen = 4 + GET_32BIT(p);
123 ret = snewn(retlen, unsigned char);
125 memcpy(ret, p, retlen);
131 CloseHandle(filemap);