darcs - ruby-ssdeep

1 #include "ruby.h"

2 #include <stdint.h>

3 #include "fuzzy.h"

5 static VALUE ssdeep_compare(VALUE module, VALUE h1, VALUE h2)

7 int n;

9 Check_Type(h1, T_STRING);

10 Check_Type(h2, T_STRING);

12 n = fuzzy_compare(RSTRING(h1)->ptr, RSTRING(h2)->ptr);

13 if (n == -1)

14 rb_raise(rb_eRuntimeError, "Error while comparing hashes");

15 return (INT2FIX(n));

18 static VALUE ssdeep_hash_buf(VALUE module, VALUE buf)

20 char hash[FUZZY_MAX_RESULT];

22 Check_Type(buf, T_STRING);

24 if (fuzzy_hash_buf(RSTRING(buf)->ptr, RSTRING(buf)->len, hash))

25 rb_raise(rb_eRuntimeError, "Error while fuzzy hashing");

27 return rb_str_new2(hash);

30 static VALUE ssdeep_hash_filename(VALUE module, VALUE path)

32 char hash[FUZZY_MAX_RESULT];

34 Check_Type(path, T_STRING);

36 if (fuzzy_hash_filename(RSTRING(path)->ptr, hash))

37 rb_raise(rb_eRuntimeError, "Error while fuzzy hashing");;

39 return rb_str_new2(hash);

42 void Init_ssdeep()

44 VALUE mSsdeep;

45 mSsdeep = rb_define_module("Ssdeep");

46 rb_define_module_function(mSsdeep, "fuzzy_compare", ssdeep_compare, 2);

47 rb_define_module_function(mSsdeep, "fuzzy_hash_buf", ssdeep_hash_buf, 1);

48 rb_define_module_function(mSsdeep, "fuzzy_hash_filename", ssdeep_hash_filename, 1);