diff options
author | Kevin Wasserman <kevin.wasserman@painless-security.com> | 2012-07-27 16:41:06 -0400 |
---|---|---|
committer | Tom Yu <tlyu@mit.edu> | 2012-08-29 16:28:20 -0400 |
commit | 9d93d99e5b07055f2f87cf60749da575f03c15ac (patch) | |
tree | a55b46649a10167845f3cb46cc1a570ba0fd6b6e /src/ccapi | |
parent | 74f6ca3f4901f3684e64c1f46c3631678de868f3 (diff) | |
download | krb5-9d93d99e5b07055f2f87cf60749da575f03c15ac.zip krb5-9d93d99e5b07055f2f87cf60749da575f03c15ac.tar.gz krb5-9d93d99e5b07055f2f87cf60749da575f03c15ac.tar.bz2 |
CCAPI client rpc fixes
On Windows XP, cci_os_ipc_thread_init() causes additional threads to be
spawned immediately, which results in a vicious cycle until Windows
resources are exhausted. Instead, defer thread_init() until it is really
needed.
Also, use the MSDN-recommended defaults for RPC calls instead of random
constants.
Signed-off-by: Kevin Wasserman <kevin.wasserman@painless-security.com>
(cherry picked from commit 9d528cd3cad2d6ea78310abe12186eedb1ac9314)
ticket: 7322
status: resolved
Diffstat (limited to 'src/ccapi')
-rw-r--r-- | src/ccapi/common/win/tls.c | 3 | ||||
-rw-r--r-- | src/ccapi/common/win/tls.h | 3 | ||||
-rw-r--r-- | src/ccapi/lib/win/ccapi_os_ipc.cxx | 32 | ||||
-rw-r--r-- | src/ccapi/lib/win/dllmain.cxx | 10 |
4 files changed, 24 insertions, 24 deletions
diff --git a/src/ccapi/common/win/tls.c b/src/ccapi/common/win/tls.c index 888230e..48a5bd5 100644 --- a/src/ccapi/common/win/tls.c +++ b/src/ccapi/common/win/tls.c @@ -47,6 +47,8 @@ void tspdata_setUUID(struct tspdata* p, unsigned char __RPC_FAR* uuidString) { strncpy(p->_uuid, uuidString, UUID_SIZE-1); }; +void tspdata_setListening (struct tspdata* p, BOOL b) {p->_listening = b;} + void tspdata_setConnected (struct tspdata* p, BOOL b) {p->_CCAPI_Connected = b;} void tspdata_setReplyEvent(struct tspdata* p, HANDLE h) {p->_replyEvent = h;} @@ -58,6 +60,7 @@ void tspdata_setSST (struct tspdata* p, time_t t) {p->_sst void tspdata_setStream (struct tspdata* p, k5_ipc_stream s) {p->_stream = s;} +BOOL tspdata_getListening (const struct tspdata* p) {return p->_listening;} BOOL tspdata_getConnected (const struct tspdata* p) {return p->_CCAPI_Connected;} diff --git a/src/ccapi/common/win/tls.h b/src/ccapi/common/win/tls.h index 4a8861e..181537f 100644 --- a/src/ccapi/common/win/tls.h +++ b/src/ccapi/common/win/tls.h @@ -41,6 +41,7 @@ */ struct tspdata { + BOOL _listening; BOOL _CCAPI_Connected; RPC_ASYNC_STATE* _rpcState; HANDLE _replyEvent; @@ -52,6 +53,7 @@ struct tspdata { struct tspdata* new_tspdata (char* uuid, time_t sst); void delete_tspdata (struct tspdata* p); +void tspdata_setListening (struct tspdata* p, BOOL b); void tspdata_setConnected (struct tspdata* p, BOOL b); void tspdata_setReplyEvent(struct tspdata* p, HANDLE h); void tspdata_setRpcAState (struct tspdata* p, RPC_ASYNC_STATE* rpcState); @@ -60,6 +62,7 @@ void tspdata_setStream (struct tspdata* p, k5_ipc_stream s); void tspdata_setUUID (struct tspdata* p, unsigned char __RPC_FAR* uuidString); HANDLE tspdata_getReplyEvent(const struct tspdata* p); +BOOL tspdata_getListening(const struct tspdata* p); BOOL tspdata_getConnected(const struct tspdata* p); RPC_ASYNC_STATE* tspdata_getRpcAState(const struct tspdata* p); time_t tspdata_getSST (const struct tspdata* p); diff --git a/src/ccapi/lib/win/ccapi_os_ipc.cxx b/src/ccapi/lib/win/ccapi_os_ipc.cxx index 352c017..35589a5 100644 --- a/src/ccapi/lib/win/ccapi_os_ipc.cxx +++ b/src/ccapi/lib/win/ccapi_os_ipc.cxx @@ -78,8 +78,6 @@ cc_int32 cci_os_ipc_msg( cc_int32 in_launch_server, extern "C" cc_int32 cci_os_ipc_process_init (void) { RPC_STATUS status; - opts.cMinCalls = 1; - opts.cMaxCalls = 20; if (!isNT()) { status = RpcServerRegisterIf(ccs_reply_ServerIfHandle, // interface NULL, // MgrTypeUuid @@ -90,7 +88,7 @@ extern "C" cc_int32 cci_os_ipc_process_init (void) { NULL, // MgrTypeUuid NULL, // MgrEpv; 0 means default RPC_IF_ALLOW_SECURE_ONLY, - opts.cMaxCalls, + RPC_C_LISTEN_MAX_CALLS_DEFAULT, NULL); // No security callback. } cci_check_error(status); @@ -118,10 +116,6 @@ extern "C" cc_int32 cci_os_ipc_thread_init (void) { if (!GetTspData(GetTlsIndex(), &ptspdata)) return ccErrNoMem; - opts.cMinCalls = 1; - opts.cMaxCalls = 20; - opts.fDontWait = TRUE; - err = cci_check_error(UuidCreate(&uuid)); // Get a UUID if (err == RPC_S_OK) { // Convert to string err = UuidToString(&uuid, &uuidString); @@ -131,7 +125,7 @@ extern "C" cc_int32 cci_os_ipc_thread_init (void) { tspdata_setUUID(ptspdata, uuidString); endpoint = clientEndpoint((const char *)uuidString); err = RpcServerUseProtseqEp((RPC_CSTR)"ncalrpc", - opts.cMaxCalls, + RPC_C_PROTSEQ_MAX_REQS_DEFAULT, (RPC_CSTR)endpoint, sa.lpSecurityDescriptor); // SD free(endpoint); @@ -155,9 +149,7 @@ extern "C" cc_int32 cci_os_ipc_thread_init (void) { if (!err) { static bool bListening = false; if (!bListening) { - err = RpcServerListen(opts.cMinCalls, - opts.cMaxCalls, - TRUE); + err = RpcServerListen(1, RPC_C_LISTEN_MAX_CALLS_DEFAULT, TRUE); cci_check_error(err); } bListening = err == 0; @@ -202,25 +194,29 @@ extern "C" cc_int32 cci_os_ipc_msg( cc_int32 in_launch_server, PROCESS_INFORMATION pi = { 0 }; HANDLE replyEvent = 0; BOOL bCCAPI_Connected= FALSE; + BOOL bListening = FALSE; unsigned char tspdata_handle[8] = { 0 }; if (!in_request_stream) { err = cci_check_error (ccErrBadParam); } if (!out_reply_stream ) { err = cci_check_error (ccErrBadParam); } if (!GetTspData(GetTlsIndex(), &ptspdata)) {return ccErrBadParam;} + bListening = tspdata_getListening(ptspdata); + if (!bListening) { + err = cci_check_error(cci_os_ipc_thread_init()); + bListening = !err; + tspdata_setListening(ptspdata, bListening); + } + bCCAPI_Connected = tspdata_getConnected (ptspdata); replyEvent = tspdata_getReplyEvent (ptspdata); sst = tspdata_getSST (ptspdata); uuid = tspdata_getUUID(ptspdata); - // Initialize old CCAPI if necessary: - if (!err) if (!Init:: Initialized()) err = cci_check_error(Init:: Initialize( )); - if (!err) if (!Client::Initialized()) err = cci_check_error(Client::Initialize(0)); - // The lazy connection to the server has been put off as long as possible! // ccapi_connect starts listening for replies as an RPC server and then // calls ccs_rpc_connect. - if (!bCCAPI_Connected) { + if (!err && !bCCAPI_Connected) { err = cci_check_error(ccapi_connect(ptspdata)); bCCAPI_Connected = !err; tspdata_setConnected(ptspdata, bCCAPI_Connected); @@ -330,10 +326,6 @@ cc_int32 ccapi_connect(const struct tspdata* tsp) { replyEvent = tspdata_getReplyEvent(tsp); uuid = tspdata_getUUID(tsp); - opts.cMinCalls = 1; - opts.cMaxCalls = 20; - opts.fDontWait = TRUE; - cci_debug_printf("%s is listening ...", __FUNCTION__); // Clear replyEvent so we can detect when a reply to our connect request has been received: diff --git a/src/ccapi/lib/win/dllmain.cxx b/src/ccapi/lib/win/dllmain.cxx index 3141e19..f9d1e2a 100644 --- a/src/ccapi/lib/win/dllmain.cxx +++ b/src/ccapi/lib/win/dllmain.cxx @@ -106,10 +106,12 @@ BOOL WINAPI DllMain(HINSTANCE hinstDLL, // DLL module handle memset(ptspdata, 0, sizeof(struct tspdata)); - // Initialize CCAPI thread data: - cci_ipc_thread_init(); - - break; + // Do not call cci_ipc_thread_init() yet; defer until we actually + // need it. On XP, cci_ipc_thread_init() will cause additional + // threads to be immediately spawned, which will bring us right + // back here again ad infinitum, until windows + // resources are exhausted. + break; // The thread of the attached process terminates: case DLL_THREAD_DETACH: |