From 377ef5136dff3fc340df312f0c422f96d8256802 Mon Sep 17 00:00:00 2001
From: Spencer T Brody <spencer@10gen.com>
Date: Fri, 21 Dec 2012 15:56:12 -0500
Subject: SERVER-7572 Make profiler record user names from AuthorizationManager

---
 src/mongo/db/introspect.cpp | 23 ++++++++++++++++++-----
 1 file changed, 18 insertions(+), 5 deletions(-)

(limited to 'src/mongo/db/introspect.cpp')

diff --git a/src/mongo/db/introspect.cpp b/src/mongo/db/introspect.cpp
index 56b281384a6..6029281d0c0 100644
--- a/src/mongo/db/introspect.cpp
+++ b/src/mongo/db/introspect.cpp
@@ -19,6 +19,8 @@
 #include "mongo/pch.h"
 
 #include "mongo/bson/util/builder.h"
+#include "mongo/db/auth/authorization_manager.h"
+#include "mongo/db/auth/principal_set.h"
 #include "mongo/db/curop.h"
 #include "mongo/db/databaseholder.h"
 #include "mongo/db/introspect.h"
@@ -32,6 +34,19 @@ namespace {
 
 namespace mongo {
 
+namespace {
+    void _appendUserInfo(BSONObjBuilder& builder, AuthorizationManager* authManager) {
+        PrincipalSet::NameIterator nameIter = authManager->getAuthenticatedPrincipalNames();
+
+        builder.append("user", nameIter.more() ? nameIter->getFullName() : "");
+        BSONArrayBuilder allUsers(builder.subarrayStart("allUsers"));
+        for ( ; nameIter.more(); nameIter.next()) {
+            allUsers.append(nameIter->getFullName());
+        }
+        allUsers.doneFast();
+    }
+} // namespace
+
     static void _profile(const Client& c, CurOp& currentOp, BufBuilder& profileBufBuilder) {
         Database *db = c.database();
         DEV verify( db );
@@ -46,9 +61,8 @@ namespace mongo {
         b.appendDate("ts", jsTime());
         b.append("client", c.clientAddress());
 
-        if (c.getAuthenticationInfo()) {
-            b.append("user", c.getAuthenticationInfo()->getUser(nsToDatabase(ns)));
-        }
+        AuthorizationManager* authManager = c.getAuthorizationManager();
+        _appendUserInfo(b, authManager);
 
         BSONObj p = b.done();
 
@@ -61,8 +75,7 @@ namespace mongo {
             BSONObjBuilder b(profileBufBuilder);
             b.appendDate("ts", jsTime());
             b.append("client", c.clientAddress() );
-            if ( c.getAuthenticationInfo() )
-                b.append( "user" , c.getAuthenticationInfo()->getUser( nsToDatabase( ns ) ) );
+            _appendUserInfo(b, authManager);
 
             b.append("err", "profile line too large (max is 100KB)");
 
-- 
cgit v1.2.1