Add value range assertions to generators.

pgriess · Aug 21, 2010 · 430c7d1 · 430c7d1
1 parent 04b869c
commit 430c7d1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lib/strtok.js b/lib/strtok.js
@@ -29,6 +29,8 @@ var UINT8 = {
     },
     put : function(s, v) {
         assert.equal(typeof v, 'number');
+        assert.ok(v >= 0 && v <= 255);
+
         s.write(BYTES[v & 0xff], 'binary');
     }
 };
@@ -41,6 +43,8 @@ var UINT16_LE = {
     },
     put : function(s, v) {
         assert.equal(typeof v, 'number');
+        assert.ok(v >= 0 && v <= 65535);
+
         s.write(BYTES[v & 0xff] + BYTES[(v >>> 8) & 0xff], 'binary');
     }
 };
@@ -53,6 +57,8 @@ var UINT16_BE = {
     },
     put : function(s, v) {
         assert.equal(typeof v, 'number');
+        assert.ok(v >= 0 && v <= 65535);
+
         s.write(BYTES[(v >>> 8) & 0xff] + BYTES[v & 0xff], 'binary');
     }
 };
@@ -66,6 +72,8 @@ var UINT32_LE = {
     },
     put : function(s, v) {
         assert.equal(typeof v, 'number');
+        assert.ok(v >= 0 && v <= 4294967295);
+
         s.write(
             BYTES[v & 0xff] + BYTES[(v >>> 8) & 0xff] +
                 BYTES[(v >>> 16) & 0xff] + BYTES[(v >>> 24) & 0xff],
@@ -83,6 +91,8 @@ var UINT32_BE = {
     },
     put : function(s, v) {
         assert.equal(typeof v, 'number');
+        assert.ok(v >= 0 && v <= 4294967295);
+
         s.write(
             BYTES[(v >>> 24) & 0xff] + BYTES[(v >>> 16) & 0xff] +
                 BYTES[(v >>> 8) & 0xff] + BYTES[v & 0xff],