pktgen: use dynamic allocation for debug print buffer

[tomoyo/tomoyo-test1.git] / net / core / pktgen.c

diff --git a/net/core/pktgen.c b/net/core/pktgen.c
index e2d6ae3..fd65761 100644 (file)
--- a/net/core/pktgen.c
+++ b/net/core/pktgen.c
@@ -906,13 +906,14 @@ static ssize_t pktgen_if_write(struct file *file,
        i += len;
 
        if (debug) {
-               size_t copy = min_t(size_t, count, 1023);
-               char tb[1024];
-               if (copy_from_user(tb, user_buffer, copy))
-                       return -EFAULT;
-               tb[copy] = 0;
-               pr_debug("%s,%lu  buffer -:%s:-\n",
-                        name, (unsigned long)count, tb);
+               size_t copy = min_t(size_t, count + 1, 1024);
+               char *tp = strndup_user(user_buffer, copy);
+
+               if (IS_ERR(tp))
+                       return PTR_ERR(tp);
+
+               pr_debug("%s,%zu  buffer -:%s:-\n", name, count, tp);
+               kfree(buf);
        }
 
        if (!strcmp(name, "min_pkt_size")) {