Refined asprintf/vasprintf detection and inclusion

3 files changed, 15 insertions, 9 deletions

diff --git a/src/asprintf.h b/src/asprintf.h
index 6f1250d..3b0072e 100644
--- a/src/asprintf.h
+++ b/src/asprintf.h
@@ -1,15 +1,24 @@
 #ifndef ASPRINTF_H
 #define ASPRINTF_H
-#ifndef vasprintf
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#ifndef HAVE_VASPRINTF
 static inline int vasprintf(char **PTR, const char *TEMPLATE, va_list AP)
 {
         int res;
-        *PTR = (char*)malloc(512);
-        res = vsnprintf(*PTR, 512, TEMPLATE, AP);
+        res = vsnprintf(NULL, 32768, TEMPLATE, AP);
+        if (res > 0) {
+                *PTR = (char*)malloc(res+1);
+                res = vsnprintf(*PTR, res, TEMPLATE, AP);
+        }
         return res;
 }
 #endif
-#ifndef asprintf
+
+#ifndef HAVE_ASPRINTF
 static inline int asprintf(char **PTR, const char *TEMPLATE, ...)
 {
         int res;
@@ -20,4 +29,5 @@ static inline int asprintf(char **PTR, const char *TEMPLATE, ...)
         return res;
 }
 #endif
-#endif
+
+#endif /* ASPRINTF_H */
diff --git a/src/debug.c b/src/debug.c
index 49ec7c7..b1c528d 100644
--- a/src/debug.c
+++ b/src/debug.c
@@ -35,10 +35,8 @@
 #include "libimobiledevice/libimobiledevice.h"
 
 #ifndef STRIP_DEBUG_CODE
-#ifndef HAVE_ASPRINTF
 #include "asprintf.h"
 #endif
-#endif
 
 int debug_level = 0;
 
diff --git a/src/lockdown.c b/src/lockdown.c
index e3d557c..0f1e1b2 100644
--- a/src/lockdown.c
+++ b/src/lockdown.c
@@ -35,9 +35,7 @@
 #include "idevice.h"
 #include "debug.h"
 #include "userpref.h"
-#ifndef HAVE_ASPRINTF
 #include "asprintf.h"
-#endif
 
 #define RESULT_SUCCESS 0
 #define RESULT_FAILURE 1