Changeset 98

Timestamp:

01/27/2007 12:51:54 (23 months ago)

Author:

why

Message:

• ext/hpricot_scan/hpricot_scan.rl: added buffer_size to allow increasing of Hpricot's buffer to compensate for ridiculous ASP.NET viewstate sizes!! The attribute values can be LARGER THAN 16K!! See #13 for how hassled I get about this. :)
  

Files:

• trunk/ext/hpricot_scan/hpricot_scan.rl (modified) (6 diffs)

trunk/ext/hpricot_scan/hpricot_scan.rl

@@ -196 +196 @@
 VALUE hpricot_scan(VALUE self, VALUE port)
 {
-  char buf[BUFSIZE];
   int cs, act, have = 0, nread = 0, curline = 1, text = 0;
-  char *tokstart = 0, *tokend = 0;
-
-  VALUE attr = Qnil, tag = Qnil, akey = Qnil, aval = Qnil;
+  char *tokstart = 0, *tokend = 0, *buf = NULL;
+
+  VALUE attr = Qnil, tag = Qnil, akey = Qnil, aval = Qnil, bufsize = Qnil;
   char *mark_tag = 0, *mark_akey = 0, *mark_aval = 0;
-  int done = 0, ele_open = 0;
+  int done = 0, ele_open = 0, buffer_size = 0;
 
   int taint = OBJ_TAINTED( port );
@@ -217 +216 @@
     }
   }
+
+  bufsize = rb_ivar_get(self, rb_intern("@buffer_size"));
+  buffer_size = BUFSIZE;
+  if (!NIL_P(bufsize)) {
+    buffer_size = NUM2INT(bufsize);
+  }
+  buf = ALLOC_N(char, buffer_size);
 
   %% write init;
@@ -223 +229 @@
     VALUE str;
     char *p = buf + have, *pe;
-    int len, space = BUFSIZE - have;
+    int len, space = buffer_size - have;
 
     if ( space == 0 ) {
@@ -256 +262 @@
     
     if ( cs == hpricot_scan_error ) {
+      free(buf);
       if ( !NIL_P(tag) )
       {
@@ -308 +315 @@
     }
   }
+  free(buf);
 }
 
@@ -313 +321 @@
 {
   VALUE mHpricot = rb_define_module("Hpricot");
+  rb_define_attr(rb_singleton_class(mHpricot), "buffer_size", 1, 1);
   rb_define_singleton_method(mHpricot, "scan", hpricot_scan, 1);
   rb_eHpricotParseError = rb_define_class_under(mHpricot, "ParseError", rb_eException);