validating arguments and supporting get with undefined key

ipfs · Jul 4, 2017 · 8e8055d · 8e8055d
1 parent 174fff0
commit 8e8055d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/config.js b/src/config.js
@@ -26,6 +26,9 @@ module.exports = (store) => {
         callback = key
         key = undefined
       }
+      if (!key) {
+        key = undefined
+      }
       store.get(configKey, (err, encodedValue) => {
         if (err) { return callback(err) }
 
@@ -55,7 +58,14 @@ module.exports = (store) => {
         callback = value
         value = key
         key = undefined
+      } else if (!key || typeof key !== 'string') {
+        return callback(new Error('Invalid key type'))
       }
+
+      if (value === undefined || Buffer.isBuffer(value)) {
+        return callback(new Error('Invalid value type'))
+      }
+
       setQueue.push({
         key: key,
         value: value