Add signTypedData input validations

1 files changed, 11 insertions, 0 deletions

diff --git a/app/scripts/lib/typed-message-manager.js b/app/scripts/lib/typed-message-manager.js
index e3efdb45d..e041ae9f3 100644
--- a/app/scripts/lib/typed-message-manager.js
+++ b/app/scripts/lib/typed-message-manager.js
@@ -1,6 +1,7 @@
 const EventEmitter = require('events')
 const ObservableStore = require('obs-store')
 const createId = require('./random-id')
+const assert = require('assert')
 
 
 module.exports = class TypedMessageManager extends EventEmitter {
@@ -23,6 +24,8 @@ module.exports = class TypedMessageManager extends EventEmitter {
   }
 
   addUnapprovedMessage (msgParams) {
+    this.validateParams(msgParams)
+
     log.debug(`TypedMessageManager addUnapprovedMessage: ${JSON.stringify(msgParams)}`)
     // create txData obj with parameters and meta data
     var time = (new Date()).getTime()
@@ -41,6 +44,14 @@ module.exports = class TypedMessageManager extends EventEmitter {
     return msgId
   }
 
+  validateParams (params) {
+    assert.equal(typeof params, 'object', 'Params should ben an object.')
+    assert.ok('data' in params, 'Params must include a data field.')
+    assert.ok('from' in params, 'Params must include a from field.')
+    assert.ok(Array.isArray(params.data), 'Data should be an array.')
+    assert.equal(typeof params.from, 'string', 'From field must be a string.')
+  }
+
   addMsg (msg) {
     this.messages.push(msg)
     this._saveMsgList()