ext/xxhash/xxhash.cc in xxhash-0.1.1 vs ext/xxhash/xxhash.cc in xxhash-0.2.0

@@ -8,12 +8,49 @@
 extern "C" VALUE xxhash_xxh32(VALUE mod, VALUE input, VALUE seed)
 {
   return ULL2NUM(XXH32(StringValuePtr(input), RSTRING_LEN(input), NUM2ULL(seed)));
 }
 
+extern "C" void xxhash_streaming_hash_free(void* state)
+{
+  // Digest frees the memory.
+  (void) XXH32_digest(state);
+}
+
+extern "C" VALUE xxhash_streaming_hash_new(VALUE klass, VALUE seed)
+{
+  void* state = XXH32_init(NUM2ULL(seed));
+  return Data_Wrap_Struct(klass, 0, xxhash_streaming_hash_free, state);
+}
+
+extern "C" VALUE xxhash_streaming_hash_update(VALUE self, VALUE data)
+{
+  void* state;
+  Data_Get_Struct(self, void, state);
+
+  XXH32_update(state, StringValuePtr(data), RSTRING_LEN(data));
+  return Qnil;
+}
+
+extern "C" VALUE xxhash_streaming_hash_digest(VALUE self)
+{
+  void* state;
+  Data_Get_Struct(self, void, state);
+
+  // Do not free memory now.
+  return ULL2NUM(XXH32_intermediateDigest(state));
+}
+
 extern "C" void Init_xxhash()
 {
   VALUE mXXhash = rb_define_module("XXhash");
   VALUE mInternal = rb_define_module_under(mXXhash, "Internal");
 
   rb_define_singleton_method(mInternal, "xxh32", (ruby_method*) &xxhash_xxh32, 2);
+
+  VALUE cStreamingHash = rb_define_class_under(mInternal, "StreamingHash", rb_cObject);
+
+  rb_define_singleton_method(cStreamingHash, "new", (ruby_method*) &xxhash_streaming_hash_new, 1);
+  rb_define_method(cStreamingHash, "update", (ruby_method*) &xxhash_streaming_hash_update, 1);
+  rb_define_method(cStreamingHash, "digest", (ruby_method*) &xxhash_streaming_hash_digest, 0);
+  rb_define_method(cStreamingHash, "intermediate_digest", (ruby_method*) &xxhash_streaming_hash_digest, 0);
 }