From 616302c13962e6b9be9b0ca552ded9cd37e85597 Mon Sep 17 00:00:00 2001
From: erdgeist <>
Date: Fri, 23 Nov 2007 18:12:50 +0000
Subject: make ot_iovecs fix last interface more sane, also add a convenience
 function that handles the task of fixing, allocating and - if necessary -
 freeing

---
 ot_iovec.c | 21 +++++++++++++++++----
 ot_iovec.h |  5 ++++-
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/ot_iovec.c b/ot_iovec.c
index a098733..f71d4d7 100644
--- a/ot_iovec.c
+++ b/ot_iovec.c
@@ -34,23 +34,36 @@ void iovec_free( int *iovec_entries, struct iovec **iovector ) {
   *iovec_entries = 0;
 }
 
-void  iovec_fixlast( int *iovec_entries, struct iovec **iovector, size_t new_alloc ) {
+void  iovec_fixlast( int *iovec_entries, struct iovec **iovector, void *last_ptr ) {
   int page_size = getpagesize();
-  size_t old_alloc, old_pages, new_pages;
+  size_t old_alloc, new_alloc, old_pages, new_pages;
+  char * base = (char*)((*iovector)[ *iovec_entries - 1 ]).iov_base;
 
   if( !*iovec_entries ) return;
 
   old_alloc = ((*iovector)[ *iovec_entries - 1 ]).iov_len;
+  new_alloc = ((char*)last_ptr) - base;
   old_pages = 1 + old_alloc / page_size;
   new_pages = 1 + new_alloc / page_size;
 
   if( old_pages != new_pages ) {
-    munmap( ((char*)((*iovector)[ *iovec_entries - 1 ]).iov_base ) +  new_pages * page_size,
-            old_alloc - new_pages * page_size );
+    munmap( base + new_pages * page_size, old_alloc - new_pages * page_size );
   }
   ((*iovector)[*iovec_entries - 1 ]).iov_len = new_alloc;
 }
 
+void  *iovec_fix_increase_or_free( int *iovec_entries, struct iovec **iovector, void *last_ptr, size_t new_alloc ) {
+  void *new_ptr;
+
+  iovec_fixlast( iovec_entries, iovector, last_ptr );
+
+  if( !( new_ptr = iovec_increase( iovec_entries, iovector, new_alloc ) ) )
+    iovec_free( iovec_entries, iovector );
+
+  return new_ptr;
+}
+
+
 size_t iovec_length( int *iovec_entries, struct iovec **iovector ) {
   size_t length = 0;
   int i;
diff --git a/ot_iovec.h b/ot_iovec.h
index a2d329d..d52a167 100644
--- a/ot_iovec.h
+++ b/ot_iovec.h
@@ -7,8 +7,11 @@
 #include <sys/uio.h>
 
 void  *iovec_increase( int *iovec_entries, struct iovec **iovector, size_t new_alloc );
-void   iovec_fixlast( int *iovec_entries, struct iovec **iovector, size_t new_alloc );
+void   iovec_fixlast( int *iovec_entries, struct iovec **iovector, void *last_ptr );
 void   iovec_free( int *iovec_entries, struct iovec **iovector );
+
 size_t iovec_length( int *iovec_entries, struct iovec **iovector );
 
+void  *iovec_fix_increase_or_free( int *iovec_entries, struct iovec **iovector, void *last_ptr, size_t new_alloc );
+
 #endif
-- 
cgit v1.2.3