s4: Fix "Small memory leak after calling several times lp_update"

Should fix bug #6660.

diff --git a/lib/util/charset/iconv.c b/lib/util/charset/iconv.c
index 9825e4be016019360d013217a59a617658b4c218..8256dc665c700796366c21504086b3f5393ed4af 100644 (file)
--- a/lib/util/charset/iconv.c
+++ b/lib/util/charset/iconv.c
@@ -22,7 +22,6 @@
 #include "../lib/util/dlinklist.h"
 #include "system/iconv.h"
 #include "system/filesys.h"
-#undef strcasecmp
 
 
 /**
@@ -50,7 +49,6 @@
 
 static size_t ascii_pull  (void *,const char **, size_t *, char **, size_t *);
 static size_t ascii_push  (void *,const char **, size_t *, char **, size_t *);
-static size_t latin1_push (void *,const char **, size_t *, char **, size_t *);
 static size_t utf8_pull   (void *,const char **, size_t *, char **, size_t *);
 static size_t utf8_push   (void *,const char **, size_t *, char **, size_t *);
 static size_t utf16_munged_pull(void *,const char **, size_t *, char **, size_t *);
@@ -74,8 +72,6 @@ static const struct charset_functions builtin_functions[] = {
        {"UTF16_MUNGED",   utf16_munged_pull,  iconv_copy},
 
        {"ASCII", ascii_pull, ascii_push},
-       {"646", ascii_pull, ascii_push},
-       {"ISO-8859-1", ascii_pull, latin1_push},
        {"UCS2-HEX", ucs2hex_pull, ucs2hex_push}
 };
 
@@ -162,7 +158,19 @@ static bool is_utf16(const char *name)
                strcasecmp(name, "UTF-16LE") == 0;
 }
 
+int smb_iconv_t_destructor(smb_iconv_t hwd)
+{ 
+#ifdef HAVE_NATIVE_ICONV
+       if (hwd->cd_pull != NULL && hwd->cd_pull != (iconv_t)-1)
+               iconv_close(hwd->cd_pull);
+       if (hwd->cd_push != NULL && hwd->cd_push != (iconv_t)-1)
+               iconv_close(hwd->cd_push);
+       if (hwd->cd_direct != NULL && hwd->cd_direct != (iconv_t)-1)
+               iconv_close(hwd->cd_direct);
+#endif
 
+       return 0;
+}
 
 _PUBLIC_ smb_iconv_t smb_iconv_open_ex(TALLOC_CTX *mem_ctx, const char *tocode, 
                              const char *fromcode, bool native_iconv)
@@ -179,6 +187,7 @@ _PUBLIC_ smb_iconv_t smb_iconv_open_ex(TALLOC_CTX *mem_ctx, const char *tocode,
                return (smb_iconv_t)-1;
        }
        memset(ret, 0, sizeof(*ret));
+       talloc_set_destructor(ret, smb_iconv_t_destructor);
 
        /* check for the simplest null conversion */
        if (strcmp(fromcode, tocode) == 0) {
@@ -251,6 +260,9 @@ _PUBLIC_ smb_iconv_t smb_iconv_open_ex(TALLOC_CTX *mem_ctx, const char *tocode,
        }
        if (is_utf16(tocode)) {
                ret->direct = sys_iconv;
+               /* could be set just above - so we need to close iconv */
+               if (ret->cd_direct != NULL && ret->cd_direct != (iconv_t)-1)
+                       iconv_close(ret->cd_direct);
                ret->cd_direct = ret->cd_pull;
                ret->cd_pull = NULL;
                return ret;
@@ -273,7 +285,7 @@ failed:
  */
 _PUBLIC_ smb_iconv_t smb_iconv_open(const char *tocode, const char *fromcode)
 {
-       return smb_iconv_open_ex(NULL, tocode, fromcode, true);
+       return smb_iconv_open_ex(talloc_autofree_context(), tocode, fromcode, true);
 }
 
 /*
@@ -281,12 +293,6 @@ _PUBLIC_ smb_iconv_t smb_iconv_open(const char *tocode, const char *fromcode)
 */
 _PUBLIC_ int smb_iconv_close(smb_iconv_t cd)
 {
-#ifdef HAVE_NATIVE_ICONV
-       if (cd->cd_direct) iconv_close((iconv_t)cd->cd_direct);
-       if (cd->cd_pull) iconv_close((iconv_t)cd->cd_pull);
-       if (cd->cd_push) iconv_close((iconv_t)cd->cd_push);
-#endif
-
        talloc_free(cd);
        return 0;
 }
@@ -344,32 +350,6 @@ static size_t ascii_push(void *cd, const char **inbuf, size_t *inbytesleft,
        return ir_count;
 }
 
-static size_t latin1_push(void *cd, const char **inbuf, size_t *inbytesleft,
-                        char **outbuf, size_t *outbytesleft)
-{
-       int ir_count=0;
-
-       while (*inbytesleft >= 2 && *outbytesleft >= 1) {
-               (*outbuf)[0] = (*inbuf)[0];
-               if ((*inbuf)[1]) ir_count++;
-               (*inbytesleft)  -= 2;
-               (*outbytesleft) -= 1;
-               (*inbuf)  += 2;
-               (*outbuf) += 1;
-       }
-
-       if (*inbytesleft == 1) {
-               errno = EINVAL;
-               return -1;
-       }
-
-       if (*inbytesleft > 1) {
-               errno = E2BIG;
-               return -1;
-       }
-       
-       return ir_count;
-}
 
 static size_t ucs2hex_pull(void *cd, const char **inbuf, size_t *inbytesleft,
                         char **outbuf, size_t *outbytesleft)

diff --git a/lib/util/dprintf.c b/lib/util/dprintf.c
index 3e6d0e8bca99dfe5bf9705818b2f2f96f7444c73..e9a15dcbe625c5e41238f208b6b04de10ae00c3f 100644 (file)
--- a/lib/util/dprintf.c
+++ b/lib/util/dprintf.c
@@ -39,6 +39,9 @@ static smb_iconv_t display_cd = (smb_iconv_t)-1;
 
 void d_set_iconv(smb_iconv_t cd)
 {
+       if (display_cd != (smb_iconv_t)-1)
+               talloc_free(display_cd);
+
        display_cd = cd;
 }