rabies 0.0.1

data/src/cgi.c

@@ -0,0 +1,117 @@
+#include "ruby.h"
+#include "intern.h"
+#include "ctype.h"
+#include "cgi.h"
+
+void Init_CGI() {
+  VALUE cgi = rb_define_class("CGI", rb_cObject);
+  
+  rb_define_module_function(cgi, "escape", CGI_escape, 1);
+  rb_define_module_function(cgi, "escape_without_plus", CGI_escape_without_plus, 1);
+  
+  rb_define_module_function(cgi, "unescape", CGI_unescape, 1);
+  
+  rb_define_module_function(cgi, "escapeHTML", CGI_escapeHTML, 1);
+}
+
+inline char digit2hex(char ch) {
+  if (ch < 9) return '0' + ch;
+  else return 'A' + ch - 10;
+}
+inline char hex2digit(char ch) {
+  if (isdigit(ch)) return ch - '0';
+  else if (ch >= 'A' && ch <= 'F') return ch - 'A';
+  else if (ch >= 'a' && ch <= 'f') return ch - 'a';
+  else return 0;
+}
+
+VALUE CGI_escape(VALUE cgi, VALUE str) {
+  return CGI_internal_escape(cgi, str, 1);
+}
+VALUE CGI_escape_without_plus(VALUE cgi, VALUE str) {
+  return CGI_internal_escape(cgi, str, 0);
+}
+
+VALUE CGI_internal_escape(VALUE cgi, VALUE str, int use_pluses) {
+  long output_length = 1;
+  char *input, *output, *r, *w, ch;
+  
+  r = input = StringValueCStr(str);
+  // First find the length of the resulting string we'll need.
+  while ((ch = *r)) {
+    ++r;
+    if (isalpha(ch) || isdigit(ch) || (ch == ' ' && use_pluses) || ch == '_' || ch == '.' || ch == '-')
+      output_length += 1; // Normal takes 1 character
+    else
+      output_length += 3; // Escaped, needs %XX characters
+  }
+  
+  // Create the string of proper size
+  w = output = ALLOC_N(char, output_length);
+  r = input;
+  
+  // And fill it with the escaped characters
+  while ((ch = *r)) {
+    r++;
+    if (isalpha(ch) || isdigit(ch) || ch == '_' || ch == '.' || ch == '-')
+      *w++ = ch;
+    else if (ch == ' ' && use_pluses)
+      *w++ = '+';
+    else {
+      *w++ = '%';
+      *w++ = digit2hex((ch / 0x10) & 0xF);
+      *w++ = digit2hex(ch & 0xF);
+    }
+  }
+  *w = '\0';
+  return rb_str_new2(output);
+}
+
+VALUE CGI_unescape(VALUE cgi, VALUE str) {
+  char *input, *output, *r, *w, ch;
+  r = input = StringValueCStr(str);
+  w = output = ALLOC_N(char, RSTRING(str)->len + 1);
+  
+  while ((ch = *r)) {
+    ++r;
+    if (ch == '%')
+      *w++ = (hex2digit(*r++) * 0x10) | hex2digit(*r++);
+    else if (ch == '+') *w++ = ' ';
+    else *w++ = ch;
+  }
+  *w = '\0';
+  
+  return rb_str_new2(output);
+}
+
+#define CH(c) (*w++ = c)
+VALUE CGI_escapeHTML(VALUE cgi, VALUE str) {
+  long output_length = 1;
+  char *input, *output, *r, *w, ch;
+
+  r = input = StringValueCStr(str);
+  // First find the length of the resulting string we'll need.
+  while ((ch = *r)) {
+    ++r;
+    if (ch == '&') output_length += 5;
+    else if (ch == '"') output_length += 6;
+    else if (ch == '<' || ch == '>') output_length += 4;
+    else output_length++;
+  }
+  
+  // Create the string of proper size
+  w = output = ALLOC_N(char, output_length);
+  r = input;
+
+  // And fill it with the escaped characters
+  while ((ch = *r)) {
+    r++;
+    if (ch == '&') { CH('&'); CH('a'); CH('m'); CH('p'); CH(';'); }
+    else if (ch == '"') { CH('&'); CH('q'); CH('u'); CH('o'); CH('t'); CH(';'); }
+    else if (ch == '<' || ch == '>') { CH('&'); CH(ch == '>' ? 'g' : 'l'); CH(';'); }
+    else CH(ch);
+  }
+  *w = '\0';
+  
+  return rb_str_new2(output);
+}

data/src/cgi.h

@@ -0,0 +1,11 @@
+#include "ruby.h"
+
+void Init_CGI();
+
+VALUE CGI_escape(VALUE cgi, VALUE str);
+VALUE CGI_escape_without_plus(VALUE cgi, VALUE str);
+VALUE CGI_internal_escape(VALUE cgi, VALUE str, int use_pluses);
+
+VALUE CGI_unescape(VALUE cgi, VALUE str);
+
+VALUE CGI_escapeHTML(VALUE cgi, VALUE str);

data/src/extconf.rb

@@ -0,0 +1,2 @@
+require 'mkmf'
+create_makefile('rabies')

data/src/rabies.c

@@ -0,0 +1,9 @@
+#include "ruby.h"
+
+#include "cgi.h"
+#include "uri.h"
+
+void Init_rabies() {
+  Init_CGI();
+  Init_URI();
+}

data/src/uri.c

@@ -0,0 +1,7 @@
+#include "uri.h"
+
+void Init_URI() {
+  VALUE uri = rb_define_module("URI");
+  
+  rb_define_module_function(uri, "encode", CGI_escape_without_plus, 1);
+}

data/src/uri.h

@@ -0,0 +1,4 @@
+#include "ruby.h"
+#include "cgi.h"
+
+void Init_URI();

data/test/abstract.rb

@@ -0,0 +1,20 @@
+require 'test/unit'
+require File.dirname(__FILE__) + '/preserve_versions.rb'
+
+class << Test::Unit::TestCase
+  
+  def preserve_result_of(target, selector, *arguments)
+    
+    # Find an unused selector for the test case
+    s = "test_preservation_of_#{target}_#{selector}"
+    i = 1
+    i += 1 while instance_methods.include?(s + i.to_s)
+    
+    define_method s + i.to_s do
+      old = target.send "old_#{selector}", *arguments
+      current = target.send selector, *arguments
+      assert_equal old, current, "The result of #{target}.#{selector}(#{arguments.inspect[1..-2]}) changed"
+    end
+  end
+  
+end

data/test/cgi_test.rb

@@ -0,0 +1,15 @@
+
+class CGITest < Test::Unit::TestCase
+  
+  preserve_result_of CGI, :escape, "Hello World!"
+  
+  preserve_result_of CGI, :unescape, "Hello+World%21"
+  preserve_result_of CGI, :unescape, "Hello%20World%21"
+  
+  preserve_result_of CGI, :escapeHTML, "<script>alert('pwned');</script>"
+  preserve_result_of CGI, :escapeHTML, "<b>All your base<b><pwned>are belong to us</pwned>"
+  
+  # This seems broken, but we'll preserve the behavior for now
+  preserve_result_of CGI, :escapeHTML, "<b>I have π for dinner…</b>"
+  
+end

data/test/preserve_versions.rb

@@ -0,0 +1,19 @@
+
+class << Module
+  
+  def preserve_method(*selectors)
+    selectors.flatten.each { |sel| alias_method "old_#{sel}", sel }
+  end
+  
+end
+
+require 'cgi'
+class << CGI; preserve_method %w(escape unescape escapeHTML); end
+
+require 'uri'
+class << URI; preserve_method %w(encode); end
+
+
+# Now require rabies:
+$: << File.expand_path(File.dirname(__FILE__) + '/../src/')
+require 'rabies'

data/test/uri_test.rb

@@ -0,0 +1,6 @@
+
+class URITest < Test::Unit::TestCase
+  
+  preserve_result_of URI, :encode, "Hello World!"
+  
+end

metadata

@@ -0,0 +1,56 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.0
+specification_version: 1
+name: rabies
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+date: 2007-03-12 00:00:00 -04:00
+summary: A library which replaces hotspots in Rails with C implementations to increase performance.
+require_paths: 
+- lib
+email: nseckar@gmail.com
+homepage: http://http://code.google.com/p/rabies/
+rubyforge_project: 
+description: 
+autorequire: rabies
+default_executable: 
+bindir: bin
+has_rdoc: false
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- Nicholas Seckar
+files: 
+- src/cgi.c
+- src/cgi.h
+- src/extconf.rb
+- src/rabies.c
+- src/uri.c
+- src/uri.h
+- test/abstract.rb
+- test/cgi_test.rb
+- test/preserve_versions.rb
+- test/uri_test.rb
+test_files: 
+- test/cgi_test.rb
+- test/uri_test.rb
+rdoc_options: []
+
+extra_rdoc_files: []
+
+executables: []
+
+extensions: 
+- src/extconf.rb
+requirements: []
+
+dependencies: []
+