Return to WbemExecClient.cpp CVS log | Up to [Pegasus] / pegasus / src / Clients / wbemexec |
File: [Pegasus] / pegasus / src / Clients / wbemexec / WbemExecClient.cpp
(download)
Revision: 1.29, Sun Jan 16 01:42:06 2005 UTC (19 years, 5 months ago) by david.dillard Branch: MAIN CVS Tags: SLPPERFINST-root, SLPPERFINST-branch Changes since 1.28: +6 -6 lines BUG#: 2297 TITLE: 'Types' changes USER: David Dillard MAIL: david.dillard@veritas.com DESCRIPTION: Changes for typedef'ing Sint8 to be a 'signed char'. |
//%2004//////////////////////////////////////////////////////////////////////// // // Copyright (c) 2000, 2001, 2002 BMC Software; Hewlett-Packard Development // Company, L.P.; IBM Corp.; The Open Group; Tivoli Systems. // Copyright (c) 2003 BMC Software; Hewlett-Packard Development Company, L.P.; // IBM Corp.; EMC Corporation, The Open Group. // Copyright (c) 2004 BMC Software; Hewlett-Packard Development Company, L.P.; // IBM Corp.; EMC Corporation; VERITAS Software Corporation; The Open Group. // // Permission is hereby granted, free of charge, to any person obtaining a copy // of this software and associated documentation files (the "Software"), to // deal in the Software without restriction, including without limitation the // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or // sell copies of the Software, and to permit persons to whom the Software is // furnished to do so, subject to the following conditions: // // THE ABOVE COPYRIGHT NOTICE AND THIS PERMISSION NOTICE SHALL BE INCLUDED IN // ALL COPIES OR SUBSTANTIAL PORTIONS OF THE SOFTWARE. THE SOFTWARE IS PROVIDED // "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT // LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR // PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT // HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN // ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. // //============================================================================== // // Author: Roger Kumpf, Hewlett-Packard Company (roger_kumpf@hp.com) // // Modified By: Warren Otsuka, Hewlett-Packard Company (warren_otsuka@hp.com) // Carol Ann Krug Graves, Hewlett-Packard Company // (carolann_graves@hp.com) // Dan Gorey, IBM (djgorey@us.ibm.com) // Amit Arora, IBM (amita@in.ibm.com) for Bug#1170, PEP-101 // //%///////////////////////////////////////////////////////////////////////////// #include <Pegasus/Common/Config.h> #include <Pegasus/Common/Constants.h> #include <Pegasus/Common/FileSystem.h> #include <Pegasus/Common/HTTPConnection.h> #include <Pegasus/Common/Destroyer.h> #include <Pegasus/Common/XmlWriter.h> #include <Pegasus/Common/TimeValue.h> #include <Pegasus/Common/PegasusVersion.h> #include <Pegasus/Common/System.h> #include <Pegasus/Common/HTTPMessage.h> #include <Pegasus/Common/AutoPtr.h> #include "HttpConstants.h" #include "WbemExecClient.h" #include <iostream> #ifdef PEGASUS_PLATFORM_WIN32_IX86_MSVC # include <windows.h> #else # include <netinet/in.h> # include <arpa/inet.h> # include <sys/socket.h> #endif PEGASUS_USING_STD; PEGASUS_NAMESPACE_BEGIN static const char PASSWORD_PROMPT [] = "Please enter your password: "; static const char PASSWORD_BLANK [] = "Password cannot be blank. Please re-enter your password."; static const Uint32 MAX_PW_RETRIES = 3; static Boolean verifyServerCertificate(SSLCertificateInfo &certInfo) { // // If server certificate was found in CA trust store and validated, then // return 'true' to accept the certificate, otherwise return 'false'. // if (certInfo.getResponseCode() == 1) { return true; } else { return false; } } WbemExecClient::WbemExecClient(Uint32 timeoutMilliseconds) : MessageQueue(PEGASUS_QUEUENAME_WBEMEXECCLIENT), _timeoutMilliseconds(timeoutMilliseconds), _connected(false), _isRemote( false ), _password( String::EMPTY ) { // // Create Monitor and HTTPConnector // #ifdef PEGASUS_USE_23HTTPMONITOR_CLIENT _monitor.reset(new Monitor()); _httpConnector.reset(new HTTPConnector(_monitor.get())); #else _monitor.reset(new monitor_2()); _httpConnector.reset(new HTTPConnector2(_monitor.get())); #endif } WbemExecClient::~WbemExecClient() { disconnect(); } void WbemExecClient::handleEnqueue() { } void WbemExecClient::_connect( const String& host, const Uint32 portNumber, AutoPtr<SSLContext>& sslContext ) throw(CannotCreateSocketException, CannotConnectException, InvalidLocatorException) { // // Attempt to establish a connection: // //try //{ #ifdef PEGASUS_USE_23HTTPMONITOR_CLIENT _httpConnection.reset(_httpConnector->connect(host, portNumber, sslContext.get(), this)); sslContext.release(); #else _httpConnection.reset(_httpConnector->connect(host, portNumber, sslContext.get(), this)); sslContext.release(); _monitor->set_session_dispatch(_httpConnection->connection_dispatch);//bug#1170 #endif //} // Could catch CannotCreateSocketException, CannotConnectException, // or InvalidLocatorException //catch (Exception& e) //{ // throw e; //} _connected = true; _isRemote = true; } void WbemExecClient::connect( const String& host, const Uint32 portNumber, AutoPtr<SSLContext>& sslContext, const String& userName, const String& password ) throw(AlreadyConnectedException, InvalidLocatorException, CannotCreateSocketException, CannotConnectException) { // // If already connected, bail out! // if (_connected) throw AlreadyConnectedException(); // // If the host is empty, set hostName to "localhost" // String hostName = host; if (host == String::EMPTY) { hostName = "localhost"; } // // Set authentication information // _authenticator.clear(); if (userName.size()) { _authenticator.setUserName(userName); } if (password.size()) { _authenticator.setPassword(password); _password = password; } _connect(hostName, portNumber, sslContext); _isRemote = true; } void WbemExecClient::connectLocal() throw(AlreadyConnectedException, InvalidLocatorException, CannotCreateSocketException, CannotConnectException) { // // If already connected, bail out! // if (_connected) throw AlreadyConnectedException(); String host = String::EMPTY; Uint32 portNumber = 0; // // Set authentication type // _authenticator.clear(); _authenticator.setAuthType(ClientAuthenticator::LOCAL); AutoPtr<SSLContext> sslContext; #ifdef PEGASUS_LOCAL_DOMAIN_SOCKET _connect(host, portNumber, sslContext); #else try { // // Look up the WBEM HTTP port number for the local system // portNumber = System::lookupPort(WBEM_HTTP_SERVICE_NAME, WBEM_DEFAULT_HTTP_PORT); // // Assign host // host.assign(_getLocalHostName()); _connect(host, portNumber, sslContext); } catch(CannotConnectException &e) { // // Look up the WBEM HTTPS port number for the local system // portNumber = System::lookupPort(WBEM_HTTPS_SERVICE_NAME, WBEM_DEFAULT_HTTPS_PORT); // // Assign host // host.assign(_getLocalHostName()); // // Create SSLContext // const char* pegasusHome = getenv("PEGASUS_HOME"); String certpath = FileSystem::getAbsolutePath( pegasusHome, PEGASUS_SSLCLIENT_CERTIFICATEFILE); String randFile = String::EMPTY; #ifdef PEGASUS_SSL_RANDOMFILE randFile = FileSystem::getAbsolutePath( pegasusHome, PEGASUS_SSLCLIENT_RANDOMFILE); #endif AutoPtr<SSLContext> sslContext(new SSLContext(certpath, verifyServerCertificate, randFile));//PEP101 _connect(host, portNumber, sslContext); } #endif _isRemote = false; } void WbemExecClient::disconnect() { if (_connected) { // // Close the connection // _httpConnector->disconnect(_httpConnection.get()); _authenticator.clear(); _connected = false; } } /** Prompt for password. @return String value of the user entered password */ String WbemExecClient::_promptForPassword() { // // Password is not set, prompt for the old password once // String pw = String::EMPTY; Uint32 retries = 1; do { pw = System::getPassword( PASSWORD_PROMPT ); if ( pw == String::EMPTY || pw == "" ) { if( retries < MAX_PW_RETRIES ) { retries++; } else { break; } cerr << PASSWORD_BLANK << endl; pw = String::EMPTY; continue; } } while ( pw == String::EMPTY ); return( pw ); } Array<char> WbemExecClient::issueRequest( const Array<char>& request ) throw(NotConnectedException, ConnectionTimeoutException, UnauthorizedAccess) { if (!_connected) { throw NotConnectedException(); } HTTPMessage* httpRequest = new HTTPMessage(request); _authenticator.setRequestMessage(httpRequest); Boolean finished = false; Boolean challenge = false; HTTPMessage* httpResponse; do { HTTPMessage* httpRequestCopy = new HTTPMessage(*(HTTPMessage*)_authenticator.getRequestMessage()); _addAuthHeader(httpRequestCopy); Message* response = _doRequest(httpRequestCopy); PEGASUS_ASSERT(response->getType() == HTTP_MESSAGE); httpResponse = (HTTPMessage*)response; finished = !_checkNeedToResend(httpResponse); if (!finished) { if (!challenge) { challenge = true; if( ( _password == String::EMPTY ) && _isRemote ) { _password = _promptForPassword(); _authenticator.setPassword( _password ); } } else { break; } delete httpResponse; } } while (!finished); AutoPtr<HTTPMessage> origRequest((HTTPMessage*)_authenticator.getRequestMessage()); _authenticator.setRequestMessage(0); AutoPtr<HTTPMessage> destroyer(httpResponse); return(httpResponse->message); } Message* WbemExecClient::_doRequest(HTTPMessage * request) throw(ConnectionTimeoutException) { // ATTN-RK-P2-20020416: We should probably clear out the queue first. PEGASUS_ASSERT(getCount() == 0); // Shouldn't be any messages in our queue _httpConnection->enqueue(request); Uint64 startMilliseconds = TimeValue::getCurrentTime().toMilliseconds(); Uint64 nowMilliseconds = startMilliseconds; Uint64 stopMilliseconds = nowMilliseconds + _timeoutMilliseconds; while (nowMilliseconds < stopMilliseconds) { // // Wait until the timeout expires or an event occurs: // #ifdef PEGASUS_USE_23HTTPMONITOR_CLIENT _monitor->run(Uint32(stopMilliseconds - nowMilliseconds)); #else _monitor->run(); #endif // // Check to see if incoming queue has a message // Message* response = dequeue(); if (response) { return response; } nowMilliseconds = TimeValue::getCurrentTime().toMilliseconds(); } // // Throw timed out exception: // throw ConnectionTimeoutException(); } String WbemExecClient::_getLocalHostName() { static String hostname; if (!hostname.size()) { hostname.assign(System::getHostName()); } return hostname; } void WbemExecClient::_addAuthHeader(HTTPMessage*& httpMessage) { // // Add authentication headers to the message // String authHeader = _authenticator.buildRequestAuthHeader(); if (authHeader != String::EMPTY) { // // Parse the HTTP message: // String startLine; Array<HTTPHeader> headers; char* content; Uint32 contentLength; httpMessage->parse(startLine, headers, contentLength); // Calculate the beginning of the content from the message size and // the content length content = (char*) httpMessage->message.getData() + httpMessage->message.size() - contentLength; Array<char> newMessageBuffer; newMessageBuffer << startLine << HTTP_CRLF; newMessageBuffer << authHeader << HTTP_CRLF; for (Uint32 i=0; i<headers.size(); i++) { newMessageBuffer << headers[i].first << HEADER_SEPARATOR << HTTP_SP << headers[i].second << HTTP_CRLF; } newMessageBuffer << HTTP_CRLF; newMessageBuffer << content << HTTP_CRLF; HTTPMessage* newMessage = new HTTPMessage(newMessageBuffer); delete httpMessage; httpMessage = newMessage; } } Boolean WbemExecClient::_checkNeedToResend(HTTPMessage* httpMessage) { // // Parse the HTTP message: // String startLine; Array<HTTPHeader> headers; Uint32 contentLength; httpMessage->parse(startLine, headers, contentLength); try { return _authenticator.checkResponseHeaderForChallenge(headers); } catch(InvalidAuthHeader&) { // We're done, send (garbage) response back to the user. return false; } } PEGASUS_NAMESPACE_END
No CVS admin address has been configured |
Powered by ViewCVS 0.9.2 |