fargo 0.1.1 → 0.2.0

data/ext/fargo/base32.c

@@ -0,0 +1,61 @@
+#include <stdlib.h>
+
+#include "base32.h"
+
+char base32_alphabet[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567";
+
+char* base32_encode(const unsigned char* buffer, int len)
+{
+    unsigned char* digest = NULL;
+    if (len > 0) {
+        int i, j = 0;
+        int bits_remain = 0;
+        unsigned short value = 0;
+        unsigned long digest_len = len*8;
+        digest_len = 1 + ((digest_len % 5) == 0 ? (digest_len / 5) : (digest_len / 5) + 1);
+
+        digest = (unsigned char*)malloc(digest_len);
+        if (digest != NULL) {
+            for (i = 0; i < len; i++) {
+                value = (value << 8) | buffer[i];
+                bits_remain += 8;
+                while (bits_remain > 5 && j < 1023) {
+                    int idx = (value >> (bits_remain-5)) & 0x1F;
+                    digest[j++] = base32_alphabet[idx];
+                    bits_remain -= 5;
+                }
+            }
+            if (bits_remain > 0) {
+                int idx = (value << (5-bits_remain)) & 0x1F;
+                digest[j++] = base32_alphabet[idx];
+            }
+            digest[j] = '\0';
+        }
+    }
+    return digest;
+}
+
+#if 0
+char* base32_decode(const unsigned char* buffer, int len)
+{
+    int i, j = 0;
+    int bits_remain = 0;
+    unsigned short value = 0;
+
+    for (i = 0; i < len; i++) {
+        value |= (buffer[i] << bits_remain);
+        bits_remain += 8;
+        while (bits_remain > 5 && j < 1023) {
+            base32_digest[j++] = (value & 0x1F) < 26 ? 'A'+(value & 0x1F) : '2' + (value & 0x1F);
+            value >>= 5;
+            bits_remain -= 5;
+        }
+    }
+    if (bits_remain > 0) {
+        base32_digest[j++] = (value & 0x1F) < 26 ? 'A'+(value & 0x1F) : '2' + (value & 0x1F);
+    }
+    base32_digest[j] = '\0';
+    return base32_digest;
+}
+
+#endif

data/ext/fargo/base32.h

@@ -0,0 +1,16 @@
+#ifndef __BASE32_H
+#define __BASE32_H
+
+#if defined(__cplusplus)
+extern "C" {
+#endif
+
+// user must call free() funtion on pointer returned from these functions
+
+char* base32_encode(const unsigned char* in, int inlen);
+
+#if defined(__cplusplus)
+}
+#endif
+
+#endif // ifndef __BASE32_H

data/ext/fargo/extconf.rb

@@ -0,0 +1,8 @@
+# This uses mkmf
+require 'mkmf'
+
+if RUBY_VERSION =~ /1.9/
+  $CFLAGS << ' -DRUBY_19'
+end
+
+create_makefile('tth')

data/ext/fargo/fargo_tth.c

@@ -0,0 +1,34 @@
+#include <ruby.h>
+#include <string.h>
+#include "tth.h"
+
+#ifndef RUBY_19
+# define RSTRING_LEN(s) (RSTRING(s)->len)
+# define RSTRING_PTR(s) (RSTRING(s)->ptr)
+# define RARRAY_LEN(s) (RARRAY(s)->len)
+# define RARRAY_PTR(s) (RARRAY(s)->ptr)
+#endif
+
+VALUE rb_tth_file(VALUE self, VALUE filename) {
+  char *file = RSTRING_PTR(filename);
+
+  char* tthl = NULL;
+  size_t tthl_size;
+  char* hash = tth(file, &tthl, &tthl_size);
+  if (tthl != NULL) {
+    free(tthl);
+  }
+
+  if (hash != NULL) {
+    return rb_str_new(hash, strlen(hash));
+  } else {
+    return Qnil;
+  }
+}
+
+Init_tth() {
+  VALUE cFargo = rb_define_module("Fargo");
+  VALUE cTTH = rb_define_module_under(cFargo, "TTH");
+
+  rb_define_method(cTTH, "file_tth", rb_tth_file, 1);
+}