--- IO-AIO/AIO.xs	2006/10/22 13:33:28	1.49
+++ IO-AIO/AIO.xs	2006/10/23 00:17:07	1.53
@@ -103,6 +103,7 @@
 static volatile aio_req reqs, reqe; /* queue start, queue end */
 static volatile aio_req ress, rese; /* queue start, queue end */
 
+static void req_invoke (aio_req req);
 static void req_free (aio_req req);
 
 /* must be called at most once */
@@ -119,10 +120,12 @@
 
 static aio_req SvAIO_REQ (SV *sv)
 {
+  MAGIC *mg;
+
   if (!sv_derived_from (sv, AIO_REQ_KLASS) || !SvROK (sv))
     croak ("object of class " AIO_REQ_KLASS " expected");
 
-  MAGIC *mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
+  mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
 
   return mg ? (aio_req)mg->mg_ptr : 0;
 }
@@ -158,6 +161,21 @@
     }
 }
 
+static void aio_grp_dec (aio_req grp)
+{
+  --grp->length;
+
+  /* call feeder, if applicable */
+  aio_grp_feed (grp);
+
+  /* finish, if done */
+  if (!grp->length && grp->fd)
+    {
+      req_invoke (grp);
+      req_free (grp);
+    }
+}
+
 static void poll_wait ()
 {
   if (nreqs && !ress)
@@ -256,16 +274,16 @@
   call_sv (req->callback, G_VOID | G_EVAL);
   SPAGAIN;
 
+  FREETMPS;
+  LEAVE;
+
+  errno = errorno;
+
   if (SvTRUE (ERRSV))
     {
       req_free (req);
       croak (0);
     }
-
-  FREETMPS;
-  LEAVE;
-
-  errno = errorno;
 }
 
 static void req_free (aio_req req)
@@ -281,17 +299,7 @@
       if (grp->grp_first == req)
         grp->grp_first = req->grp_next;
 
-      --grp->length;
-
-      /* call feeder, if applicable */
-      aio_grp_feed (grp);
-
-      /* finish, if done */
-      if (!grp->length && grp->fd)
-        {
-          req_invoke (grp);
-          req_free (grp);
-        }
+      aio_grp_dec (grp);
     }
 
   if (req->self)
@@ -357,7 +365,7 @@
       if (!req)
         break;
 
-      nreqs--;
+      --nreqs;
 
       if (req->type == REQ_QUIT)
         started--;
@@ -418,7 +426,7 @@
   while (started < wanted && nreqs >= started)
     start_thread ();
 
-  nreqs++;
+  ++nreqs;
 
   pthread_mutex_lock (&reqlock);
 
@@ -780,7 +788,7 @@
       errno = 0; /* strictly unnecessary */
 
       if (!req->cancelled)
-        switch (req->type)
+        switch (type = req->type) /* remember type for QUIT check */
           {
             case REQ_READ:      req->result = pread     (req->fd, req->dataptr, req->length, req->offset); break;
             case REQ_WRITE:     req->result = pwrite    (req->fd, req->dataptr, req->length, req->offset); break;
@@ -1273,6 +1281,7 @@
         PPCODE:
 {
 	int i;
+        aio_req req;
 
         if (grp->fd == 2)
           croak ("cannot add requests to IO::AIO::GRP after the group finished");
@@ -1282,7 +1291,7 @@
             if (GIMME_V != G_VOID)
               XPUSHs (sv_2mortal (newSVsv (ST (i))));
 
-            aio_req req = SvAIO_REQ (ST (i));
+            req = SvAIO_REQ (ST (i));
 
             if (req)
               {
@@ -1302,12 +1311,12 @@
 
 void
 result (aio_req grp, ...)
-	CODE:
+        CODE:
 {
-	int i;
-	AV *av = newAV ();
+        int i;
+        AV *av = newAV ();
 
-	for (i = 1; i < items; ++i )
+        for (i = 1; i < items; ++i )
           av_push (av, newSVsv (ST (i)));
 
         SvREFCNT_dec (grp->data);
@@ -1315,6 +1324,16 @@
 }
 
 void
+lock (aio_req grp)
+	CODE:
+        ++grp->length;
+
+void
+unlock (aio_req grp)
+	CODE:
+        aio_grp_dec (grp);
+
+void
 feeder_limit (aio_req grp, int limit)
 	CODE:
         grp->fd2 = limit;