euh

node_modules/mongodb-core/lib/auth/plain.js

@@ -0,0 +1,183 @@
+'use strict';
+
+var f = require('util').format,
+  retrieveBSON = require('../connection/utils').retrieveBSON,
+  Query = require('../connection/commands').Query,
+  MongoError = require('../error').MongoError;
+
+var BSON = retrieveBSON(),
+  Binary = BSON.Binary;
+
+var AuthSession = function(db, username, password) {
+  this.db = db;
+  this.username = username;
+  this.password = password;
+};
+
+AuthSession.prototype.equal = function(session) {
+  return (
+    session.db === this.db &&
+    session.username === this.username &&
+    session.password === this.password
+  );
+};
+
+/**
+ * Creates a new Plain authentication mechanism
+ * @class
+ * @return {Plain} A cursor instance
+ */
+var Plain = function(bson) {
+  this.bson = bson;
+  this.authStore = [];
+};
+
+/**
+ * Authenticate
+ * @method
+ * @param {{Server}|{ReplSet}|{Mongos}} server Topology the authentication method is being called on
+ * @param {[]Connections} connections Connections to authenticate using this authenticator
+ * @param {string} db Name of the database
+ * @param {string} username Username
+ * @param {string} password Password
+ * @param {authResultCallback} callback The callback to return the result from the authentication
+ * @return {object}
+ */
+Plain.prototype.auth = function(server, connections, db, username, password, callback) {
+  var self = this;
+  // Total connections
+  var count = connections.length;
+  if (count === 0) return callback(null, null);
+
+  // Valid connections
+  var numberOfValidConnections = 0;
+  var errorObject = null;
+
+  // For each connection we need to authenticate
+  while (connections.length > 0) {
+    // Execute MongoCR
+    var execute = function(connection) {
+      // Create payload
+      var payload = new Binary(f('\x00%s\x00%s', username, password));
+
+      // Let's start the sasl process
+      var command = {
+        saslStart: 1,
+        mechanism: 'PLAIN',
+        payload: payload,
+        autoAuthorize: 1
+      };
+
+      // Let's start the process
+      server(
+        connection,
+        new Query(self.bson, '$external.$cmd', command, {
+          numberToSkip: 0,
+          numberToReturn: 1
+        }),
+        function(err, r) {
+          // Adjust count
+          count = count - 1;
+
+          // If we have an error
+          if (err) {
+            errorObject = err;
+          } else if (r.result['$err']) {
+            errorObject = r.result;
+          } else if (r.result['errmsg']) {
+            errorObject = r.result;
+          } else {
+            numberOfValidConnections = numberOfValidConnections + 1;
+          }
+
+          // We have authenticated all connections
+          if (count === 0 && numberOfValidConnections > 0) {
+            // Store the auth details
+            addAuthSession(self.authStore, new AuthSession(db, username, password));
+            // Return correct authentication
+            callback(null, true);
+          } else if (count === 0) {
+            if (errorObject == null)
+              errorObject = new MongoError(f('failed to authenticate using mongocr'));
+            callback(errorObject, false);
+          }
+        }
+      );
+    };
+
+    var _execute = function(_connection) {
+      process.nextTick(function() {
+        execute(_connection);
+      });
+    };
+
+    _execute(connections.shift());
+  }
+};
+
+// Add to store only if it does not exist
+var addAuthSession = function(authStore, session) {
+  var found = false;
+
+  for (var i = 0; i < authStore.length; i++) {
+    if (authStore[i].equal(session)) {
+      found = true;
+      break;
+    }
+  }
+
+  if (!found) authStore.push(session);
+};
+
+/**
+ * Remove authStore credentials
+ * @method
+ * @param {string} db Name of database we are removing authStore details about
+ * @return {object}
+ */
+Plain.prototype.logout = function(dbName) {
+  this.authStore = this.authStore.filter(function(x) {
+    return x.db !== dbName;
+  });
+};
+
+/**
+ * Re authenticate pool
+ * @method
+ * @param {{Server}|{ReplSet}|{Mongos}} server Topology the authentication method is being called on
+ * @param {[]Connections} connections Connections to authenticate using this authenticator
+ * @param {authResultCallback} callback The callback to return the result from the authentication
+ * @return {object}
+ */
+Plain.prototype.reauthenticate = function(server, connections, callback) {
+  var authStore = this.authStore.slice(0);
+  var count = authStore.length;
+  if (count === 0) return callback(null, null);
+  // Iterate over all the auth details stored
+  for (var i = 0; i < authStore.length; i++) {
+    this.auth(
+      server,
+      connections,
+      authStore[i].db,
+      authStore[i].username,
+      authStore[i].password,
+      function(err) {
+        count = count - 1;
+        // Done re-authenticating
+        if (count === 0) {
+          callback(err, null);
+        }
+      }
+    );
+  }
+};
+
+/**
+ * This is a result from a authentication strategy
+ *
+ * @callback authResultCallback
+ * @param {error} error An error object. Set to null if no error present
+ * @param {boolean} result The result of the authentication process
+ */
+
+module.exports = Plain;