coredump: introduce cn_vprintf()

[deliverable/linux.git] / fs / coredump.c

diff --git a/fs/coredump.c b/fs/coredump.c
index dafafbafa7313e2e7ef46d05619482b2edf5645d..c10a43aae2206ed361eef8b2ab3dc6f4020e42c7 100644 (file)
--- a/fs/coredump.c
+++ b/fs/coredump.c
@@ -58,30 +58,24 @@ static atomic_t call_count = ATOMIC_INIT(1);
 
 static int expand_corename(struct core_name *cn)
 {
-       char *old_corename = cn->corename;
+       int size = CORENAME_MAX_SIZE * atomic_inc_return(&call_count);
+       char *corename = krealloc(cn->corename, size, GFP_KERNEL);
 
-       cn->size = CORENAME_MAX_SIZE * atomic_inc_return(&call_count);
-       cn->corename = krealloc(old_corename, cn->size, GFP_KERNEL);
-
-       if (!cn->corename) {
-               kfree(old_corename);
+       if (!corename)
                return -ENOMEM;
-       }
 
+       cn->size = size;
+       cn->corename = corename;
        return 0;
 }
 
-static int cn_printf(struct core_name *cn, const char *fmt, ...)
+static int cn_vprintf(struct core_name *cn, const char *fmt, va_list arg)
 {
        char *cur;
        int need;
        int ret;
-       va_list arg;
 
-       va_start(arg, fmt);
        need = vsnprintf(NULL, 0, fmt, arg);
-       va_end(arg);
-
        if (likely(need < cn->size - cn->used - 1))
                goto out_printf;
 
@@ -91,9 +85,7 @@ static int cn_printf(struct core_name *cn, const char *fmt, ...)
 
 out_printf:
        cur = cn->corename + cn->used;
-       va_start(arg, fmt);
        vsnprintf(cur, need + 1, fmt, arg);
-       va_end(arg);
        cn->used += need;
        return 0;
 
@@ -101,6 +93,18 @@ expand_fail:
        return ret;
 }
 
+static int cn_printf(struct core_name *cn, const char *fmt, ...)
+{
+       va_list arg;
+       int ret;
+
+       va_start(arg, fmt);
+       ret = cn_vprintf(cn, fmt, arg);
+       va_end(arg);
+
+       return ret;
+}
+
 static void cn_escape(char *str)
 {
        for (; *str; str++)
@@ -157,10 +161,9 @@ static int format_corename(struct core_name *cn, struct coredump_params *cprm)
        int pid_in_pattern = 0;
        int err = 0;
 
+       cn->used = 0;
        cn->size = CORENAME_MAX_SIZE * atomic_read(&call_count);
        cn->corename = kmalloc(cn->size, GFP_KERNEL);
-       cn->used = 0;
-
        if (!cn->corename)
                return -ENOMEM;
 
@@ -549,7 +552,7 @@ void do_coredump(siginfo_t *siginfo)
                if (ispipe < 0) {
                        printk(KERN_WARNING "format_corename failed\n");
                        printk(KERN_WARNING "Aborting core\n");
-                       goto fail_corename;
+                       goto fail_unlock;
                }
 
                if (cprm.limit == 1) {
@@ -669,7 +672,6 @@ fail_dropcount:
                atomic_dec(&core_dump_count);
 fail_unlock:
        kfree(cn.corename);
-fail_corename:
        coredump_finish(mm, core_dumped);
        revert_creds(old_cred);
 fail_creds: