Implement new opool function

* include/mailutils/opool.h (mu_opool_less): New function.
* libmailutils/base/opool.c: Likewise.

2 files changed, 32 insertions, 0 deletions

diff --git a/include/mailutils/opool.h b/include/mailutils/opool.h
index 5c50e8754..9aebd900b 100644
--- a/include/mailutils/opool.h
+++ b/include/mailutils/opool.h
@@ -63,6 +63,9 @@ int mu_opool_union (mu_opool_t *dst, mu_opool_t *src);
    begin a new object. */
 void mu_opool_clear (mu_opool_t opool);
 
+/* Reset the length of the current object to SIZE bytes */
+void mu_opool_less (mu_opool_t opool, size_t size);
+
 /* Free object OBJ from the pool.  If OBJ is NULL, free all created objects,
    including the one being built */
 void mu_opool_free (mu_opool_t pool, void *obj);
diff --git a/libmailutils/base/opool.c b/libmailutils/base/opool.c
index d8a4d155a..30c019bab 100644
--- a/libmailutils/base/opool.c
+++ b/libmailutils/base/opool.c
@@ -182,6 +182,35 @@ mu_opool_clear (mu_opool_t opool)
 }	
 
 void
+mu_opool_less (mu_opool_t opool, size_t sz)
+{
+  union mu_opool_bucket *p;
+  size_t total = 0;
+  
+  if (!opool)
+    return;
+  for (p = opool->bkt_head; p; p = p->hdr.next)
+    {
+      if (total + p->hdr.level >= sz)
+	{
+	  union mu_opool_bucket *t;
+	  p->hdr.level = sz - total;
+	  t = p->hdr.next;
+	  p->hdr.next = NULL;
+
+	  while (t)
+	    {
+	      union mu_opool_bucket *next = t->hdr.next;
+	      free (t);
+	      t = next;
+	    }
+	  break;
+	}
+      total += p->hdr.level;
+    }
+}	    
+
+void
 mu_opool_destroy (mu_opool_t *popool)
 {
   union mu_opool_bucket *p;