mongodb/db/curop.h

// curop.h

#pragma once

#include "namespace.h"
#include "security.h"
#include "client.h"

namespace mongo { 

    /* Current operation (for the current Client).
       an embedded member of Client class, and typically used from within the mutex there. */
    class CurOp {
        static BSONObj _tooBig; // { $msg : "query not recording (too large)" }

        bool _active;
        //        unsigned opNum;
        time_t startTime;
        int _op;
        char _ns[Namespace::MaxNsLen+2];
        struct sockaddr_in client;

        char _queryBuf[256];
        bool haveQuery() const { return *((int *) _queryBuf) != 0; }
        void resetQuery(int x=0) { *((int *)_queryBuf) = x; }
        BSONObj query() {
            if( *((int *) _queryBuf) == 1 ) { 
                return _tooBig;
            }
            BSONObj o(_queryBuf);
            return o;
        }

    public:
        void reset(time_t now, const sockaddr_in &_client) { 
            _active = true;
//            opNum++;
            startTime = now;
            _ns[0] = '?'; // just in case not set later
            resetQuery();
            killCurrentOp = 0;
            client = _client;
        }

        bool active() const { return _active; }

        void setActive(bool active) { _active = active; }
        void setNS(const char *ns) {
            strncpy(_ns, ns, Namespace::MaxNsLen);
        }
        void setOp(int op) { _op = op; }
        void setQuery(const BSONObj& query) { 
            if( query.objsize() > (int) sizeof(_queryBuf) ) { 
                resetQuery(1); // flag as too big and return
                return;
            }
            memcpy(_queryBuf, query.objdata(), query.objsize());
        }

        CurOp() { 
            _active = false;
//            opNum = 0; 
            startTime = 0;
            _op = 0;
            // These addresses should never be written to again.  The zeroes are
            // placed here as a precaution because currentOp may be accessed
            // without the db mutex.
            memset(_ns, 0, sizeof(_ns));
            memset(_queryBuf, 0, sizeof(_queryBuf));
        }

        BSONObj info() { 
            AuthenticationInfo *ai = currentClient.get()->ai;
            if( !ai->isAuthorized("admin") ) { 
                BSONObjBuilder b;
                b.append("err", "unauthorized");
                return b.obj();
            }
            return infoNoauth();
        }
        
        BSONObj infoNoauth() {
            BSONObjBuilder b;
//            b.append("opid", opNum);
            b.append("active", _active);
            if( _active ) 
                b.append("secs_running", (int) (time(0)-startTime));
            if( _op == 2004 ) 
                b.append("op", "query");
            else if( _op == 2005 )
                b.append("op", "getMore");
            else if( _op == 2001 )
                b.append("op", "update");
            else if( _op == 2002 )
                b.append("op", "insert");
            else if( _op == 2006 )
                b.append("op", "delete");
            else
                b.append("op", _op);
            b.append("ns", _ns);

            if( haveQuery() ) {
                b.append("query", query());
            }
            // b.append("inLock",  ??
            stringstream clientStr;
            clientStr << inet_ntoa( client.sin_addr ) << ":" << ntohs( client.sin_port );
            b.append("client", clientStr.str());
            return b.obj();
        }
    };

}