Added getter for block_size and digest_size

Author: oohlaf

main.c

@@ -156,14 +156,33 @@ PyMethodDef whirlpool_methods[] = {
 };
 
 
+static PyObject *
+whirlpool_get_block_size(PyObject *self, void *closure)
+{
+    return PyInt_FromLong(WBLOCKBYTES);
+}
+
+static PyObject *
+whirlpool_get_digest_size(PyObject *self, void *closure)
+{
+    return PyInt_FromLong(DIGESTBYTES);
+}
+
 static PyObject *
 whirlpool_get_name(PyObject *self, void *closure)
 {
     return PyString_FromStringAndSize("WHIRLPOOL", 9);
 }
 
-
 static PyGetSetDef whirlpool_getseters[] = {
+    {"digest_size",
+     (getter)whirlpool_get_digest_size, NULL,
+     NULL,
+     NULL},
+    {"block_size",
+     (getter)whirlpool_get_block_size, NULL,
+     NULL,
+     NULL},
     {"name",
      (getter)whirlpool_get_name, NULL,
      NULL,

tests.py

@@ -77,6 +77,20 @@ def test_update_copy(self):
         self.assertEqual(b2a_hex(wp3.digest()), results['tqbfjotle'])
         self.assertEqual(wp3.hexdigest(), results['tqbfjotle'])
 
+    def test_digest_size(self):
+        wp = whirlpool.new()
+        self.assertEqual(wp.digest_size, 64)
+        with self.assertRaisesRegexp(AttributeError,
+                                     'digest_size.*not writable'):
+            wp.digest_size = 32
+    
+    def test_block_size(self):
+        wp = whirlpool.new()
+        self.assertEqual(wp.block_size, 64)
+        with self.assertRaisesRegexp(AttributeError,
+                                     'block_size.*not writable'):
+            wp.block_size = 32
+
 
 if __name__ == '__main__':
     unittest.main()