/** * Copyright (C) 2008 10gen Inc. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License, version 3, * as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see . * * As a special exception, the copyright holders give permission to link the * code of portions of this program with the OpenSSL library under certain * conditions as described in each individual source file and distribute * linked combinations including the program with the OpenSSL library. You * must comply with the GNU Affero General Public License in all respects for * all of the code used other than as permitted herein. If you modify file(s) * with this exception, you may extend this exception to your version of the * file(s), but you are not obligated to do so. If you do not wish to do so, * delete this exception statement from your version. If you delete this * exception statement from all source files in the program, then also delete * it in the license file. */ #pragma once #include #include #include "mongo/base/disallow_copying.h" #include "mongo/util/decorable.h" #include "mongo/util/net/hostandport.h" #include "mongo/util/net/message_port.h" namespace mongo { class AuthenticationInfo; class AuthenticationSession; class AuthorizationSession; class ServiceContext; /** * this is the base class for Client and ClientInfo * Client is for mongod * ClientInfo is for mongos * They should converge slowly * The idea is this has the basic api so that not all code has to be duplicated */ class ClientBasic : public Decorable { MONGO_DISALLOW_COPYING(ClientBasic); public: virtual ~ClientBasic(); AuthenticationSession* getAuthenticationSession(); void resetAuthenticationSession(AuthenticationSession* newSession); void swapAuthenticationSession(boost::scoped_ptr& other); bool hasAuthorizationSession() const; AuthorizationSession* getAuthorizationSession() const; void setAuthorizationSession(std::unique_ptr authorizationSession); bool getIsLocalHostConnection() { if (!hasRemote()) { return false; } return getRemote().isLocalHost(); } virtual bool hasRemote() const { return _messagingPort; } virtual HostAndPort getRemote() const { verify( _messagingPort ); return _messagingPort->remote(); } /** * Returns the ServiceContext that owns this client session context. */ ServiceContext* getServiceContext() const { return _serviceContext; } /** * Returns the AbstractMessagePort to which this client session is bound, if any. */ AbstractMessagingPort * port() const { return _messagingPort; } static ClientBasic* getCurrent(); protected: ClientBasic(ServiceContext* serviceContext, AbstractMessagingPort* messagingPort); private: boost::scoped_ptr _authenticationSession; std::unique_ptr _authorizationSession; ServiceContext* const _serviceContext; AbstractMessagingPort* const _messagingPort; }; }