Integrate Mojo JS validation bindings with the Router

mojo/public/js/bindings/router.js

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 define("mojo/public/js/bindings/router", [
   "mojo/public/js/bindings/codec",
   "mojo/public/js/bindings/connector",
   "mojo/public/js/bindings/validator",
 ], function(codec, connector, validator) {
 
   function Router(handle) {
     this.connector_ = new connector.Connector(handle);
     this.incomingReceiver_ = null;
     this.nextRequestID_ = 0;
     this.responders_ = {};
+    this.payloadValidators_ = [];
 
     this.connector_.setIncomingReceiver({
         accept: this.handleIncomingMessage_.bind(this),
     });
     this.connector_.setErrorHandler({
         onError: this.handleConnectionError_.bind(this),
     });
   }
 
   Router.prototype.close = function() {
@@ skipping 23 matching lines @@
     // TODO(mpcomplete): accept should return a Promise too, maybe?
     if (result)
       return Promise.resolve();
     return Promise.reject(Error("Connection error"));
   };
 
   Router.prototype.setIncomingReceiver = function(receiver) {
     this.incomingReceiver_ = receiver;
   };
 
+  Router.prototype.setPayloadValidators = function(payloadValidators) {
+    this.payloadValidators_ = payloadValidators;
+  };
+
   Router.prototype.encounteredError = function() {
     return this.connector_.encounteredError();
   };
 
   Router.prototype.handleIncomingMessage_ = function(message) {
-    var v = new validator.Validator(message);
-    if (v.validateMessageHeader() !== validator.validationError.NONE)
+    var noError = validator.validationError.NONE;
+    var messageValidator = new validator.Validator(message);
+    var err = messageValidator.validateMessageHeader();
+    for (var i = 0; err === noError && i < this.payloadValidators_.length; ++i)
+      err = this.payloadValidators_[i](messageValidator);
+    if (err !== noError)
       this.close();
 
     if (message.expectsResponse()) {
       if (this.incomingReceiver_) {
         this.incomingReceiver_.acceptWithResponder(message, this);
       } else {
         // If we receive a request expecting a response when the client is not
         // listening, then we have no choice but to tear down the pipe.
         this.close();
       }
@@ skipping 12 matching lines @@
   Router.prototype.handleConnectionError_ = function(result) {
     for (var each in this.responders_)
       this.responders_[each].reject(result);
     this.close();
   };
 
   var exports = {};
   exports.Router = Router;
   return exports;
 });