#include <linux/init.h>
 #include <linux/nmi.h>
 #include <linux/console.h>
+#include <linux/bug.h>
 
 #define PANIC_TIMER_STEP 100
 #define PANIC_BLINK_SPD 18
        kmsg_dump(KMSG_DUMP_OOPS);
 }
 
-#ifdef WANT_WARN_ON_SLOWPATH
-struct slowpath_args {
+struct warn_args {
        const char *fmt;
        va_list args;
 };
 
-static void warn_slowpath_common(const char *file, int line, void *caller,
-                                unsigned taint, struct slowpath_args *args)
+void __warn(const char *file, int line, void *caller, unsigned taint,
+           struct pt_regs *regs, struct warn_args *args)
 {
        disable_trace_on_warning();
 
        pr_warn("------------[ cut here ]------------\n");
-       pr_warn("WARNING: CPU: %d PID: %d at %s:%d %pS()\n",
-               raw_smp_processor_id(), current->pid, file, line, caller);
+
+       if (file)
+               pr_warn("WARNING: CPU: %d PID: %d at %s:%d %pS\n",
+                       raw_smp_processor_id(), current->pid, file, line,
+                       caller);
+       else
+               pr_warn("WARNING: CPU: %d PID: %d at %pS\n",
+                       raw_smp_processor_id(), current->pid, caller);
 
        if (args)
                vprintk(args->fmt, args->args);
        }
 
        print_modules();
-       dump_stack();
+
+       if (regs)
+               show_regs(regs);
+       else
+               dump_stack();
+
        print_oops_end_marker();
+
        /* Just a warning, don't kill lockdep. */
        add_taint(taint, LOCKDEP_STILL_OK);
 }
 
+#ifdef WANT_WARN_ON_SLOWPATH
 void warn_slowpath_fmt(const char *file, int line, const char *fmt, ...)
 {
-       struct slowpath_args args;
+       struct warn_args args;
 
        args.fmt = fmt;
        va_start(args.args, fmt);
-       warn_slowpath_common(file, line, __builtin_return_address(0),
-                            TAINT_WARN, &args);
+       __warn(file, line, __builtin_return_address(0), TAINT_WARN, NULL,
+              &args);
        va_end(args.args);
 }
 EXPORT_SYMBOL(warn_slowpath_fmt);
 void warn_slowpath_fmt_taint(const char *file, int line,
                             unsigned taint, const char *fmt, ...)
 {
-       struct slowpath_args args;
+       struct warn_args args;
 
        args.fmt = fmt;
        va_start(args.args, fmt);
-       warn_slowpath_common(file, line, __builtin_return_address(0),
-                            taint, &args);
+       __warn(file, line, __builtin_return_address(0), taint, NULL, &args);
        va_end(args.args);
 }
 EXPORT_SYMBOL(warn_slowpath_fmt_taint);
 
 void warn_slowpath_null(const char *file, int line)
 {
-       warn_slowpath_common(file, line, __builtin_return_address(0),
-                            TAINT_WARN, NULL);
+       __warn(file, line, __builtin_return_address(0), TAINT_WARN, NULL, NULL);
 }
 EXPORT_SYMBOL(warn_slowpath_null);
 #endif
 
 
        if (warning) {
                /* this is a WARN_ON rather than BUG/BUG_ON */
-               pr_warn("------------[ cut here ]------------\n");
-
-               if (file)
-                       pr_warn("WARNING: at %s:%u\n", file, line);
-               else
-                       pr_warn("WARNING: at %p [verbose debug info unavailable]\n",
-                               (void *)bugaddr);
-
-               if (panic_on_warn) {
-                       /*
-                        * This thread may hit another WARN() in the panic path.
-                        * Resetting this prevents additional WARN() from
-                        * panicking the system on this thread.  Other threads
-                        * are blocked by the panic_mutex in panic().
-                        */
-                       panic_on_warn = 0;
-                       panic("panic_on_warn set ...\n");
-               }
-
-               print_modules();
-               show_regs(regs);
-               print_oops_end_marker();
-               /* Just a warning, don't kill lockdep. */
-               add_taint(BUG_GET_TAINT(bug), LOCKDEP_STILL_OK);
+               __warn(file, line, (void *)bugaddr, BUG_GET_TAINT(bug), regs,
+                      NULL);
                return BUG_TRAP_TYPE_WARN;
        }