Start 1.33.0 development cycle
[oweals/busybox.git] / libbb / verror_msg.c
index be206e4771aeefafc9772390d92e36fcb78e22cb..71512aee2c9751520e9aa99fbca77bc2ae49f339 100644 (file)
  *
  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
  *
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
-
-#include <stdio.h>
-#include <errno.h>
-#include <string.h>
-#include <stdlib.h>
-#include <syslog.h>
 #include "libbb.h"
+#if ENABLE_FEATURE_SYSLOG
+# include <syslog.h>
+#endif
 
-int logmode = LOGMODE_STDIO;
+#if ENABLE_FEATURE_SYSLOG
+static smallint syslog_level = LOG_ERR;
+#endif
+smallint logmode = LOGMODE_STDIO;
+const char *msg_eol = "\n";
 
-void bb_verror_msg(const char *s, va_list p, const char* strerr)
+void FAST_FUNC bb_verror_msg(const char *s, va_list p, const char* strerr)
 {
-       /* va_copy is used because it is not portable
-        * to use va_list p twice */
-       va_list p2;
-       va_copy(p2, p);
+       char *msg, *msg1;
+       char stack_msg[80];
+       int applet_len, strerr_len, msgeol_len, used;
 
-       if (logmode & LOGMODE_STDIO) {
-               fflush(stdout);
-               fprintf(stderr, "%s: ", bb_applet_name);
-               vfprintf(stderr, s, p);
-               if (!strerr)
-                       fputc('\n', stderr);
-               else
-                       fprintf(stderr, ": %s\n", strerr);
+       if (!logmode)
+               return;
+
+       if (!s) /* nomsg[_and_die] uses NULL fmt */
+               s = ""; /* some libc don't like printf(NULL) */
+
+       applet_len = strlen(applet_name) + 2; /* "applet: " */
+       strerr_len = strerr ? strlen(strerr) : 0;
+       msgeol_len = strlen(msg_eol);
+
+       /* This costs ~90 bytes of code, but avoids costly
+        * malloc()[in vasprintf]+realloc()+memmove()+free() in 99% of cases.
+        * ~40% speedup.
+        */
+       if ((int)sizeof(stack_msg) - applet_len > 0) {
+               va_list p2;
+
+               /* It is not portable to use va_list twice, need to va_copy it */
+               va_copy(p2, p);
+               used = vsnprintf(stack_msg + applet_len, (int)sizeof(stack_msg) - applet_len, s, p2);
+               va_end(p2);
+               msg = stack_msg;
+               used += applet_len;
+               if (used < (int)sizeof(stack_msg) - 3 - msgeol_len - strerr_len)
+                       goto add_pfx_and_sfx;
        }
-       if (ENABLE_FEATURE_SYSLOG & (logmode & LOGMODE_SYSLOG)) {
-               if (!strerr)
-                       vsyslog(LOG_ERR, s, p2);
-               else  {
-                       char *msg;
-                       if (vasprintf(&msg, s, p2) < 0)
-                               bb_error_msg_and_die(bb_msg_memory_exhausted);
-                       syslog(LOG_ERR, "%s: %s", msg, strerr);
-                       free(msg);
+
+       used = vasprintf(&msg, s, p);
+       if (used < 0)
+               return;
+
+       /* This is ugly and costs +60 bytes compared to multiple
+        * fprintf's, but is guaranteed to do a single write.
+        * This is needed for e.g. httpd logging, when multiple
+        * children can produce log messages simultaneously. */
+
+       /* can't use xrealloc: it calls error_msg on failure,
+        * that may result in a recursion */
+       /* +3 is for ": " before strerr and for terminating NUL */
+       msg1 = realloc(msg, applet_len + used + strerr_len + msgeol_len + 3);
+       if (!msg1) {
+               msg[used++] = '\n'; /* overwrites NUL */
+               applet_len = 0;
+       } else {
+               msg = msg1;
+               /* TODO: maybe use writev instead of memmoving? Need full_writev? */
+               memmove(msg + applet_len, msg, used);
+               used += applet_len;
+ add_pfx_and_sfx:
+               strcpy(msg, applet_name);
+               msg[applet_len - 2] = ':';
+               msg[applet_len - 1] = ' ';
+               if (strerr) {
+                       if (s[0]) { /* not perror_nomsg? */
+                               msg[used++] = ':';
+                               msg[used++] = ' ';
+                       }
+                       strcpy(&msg[used], strerr);
+                       used += strerr_len;
                }
+               strcpy(&msg[used], msg_eol);
+               used += msgeol_len;
+       }
+
+       if (logmode & LOGMODE_STDIO) {
+               fflush_all();
+               full_write(STDERR_FILENO, msg, used);
+       }
+#if ENABLE_FEATURE_SYSLOG
+       if (logmode & LOGMODE_SYSLOG) {
+               syslog(syslog_level, "%s", msg + applet_len);
+       }
+#endif
+       if (msg != stack_msg)
+               free(msg);
+}
+
+#ifdef VERSION_WITH_WRITEV
+/* Code size is approximately the same, but currently it's the only user
+ * of writev in entire bbox. __libc_writev in uclibc is ~50 bytes. */
+void FAST_FUNC bb_verror_msg(const char *s, va_list p, const char* strerr)
+{
+       int strerr_len, msgeol_len;
+       struct iovec iov[3];
+
+#define used   (iov[2].iov_len)
+#define msgv   (iov[2].iov_base)
+#define msgc   ((char*)(iov[2].iov_base))
+#define msgptr (&(iov[2].iov_base))
+
+       if (!logmode)
+               return;
+
+       if (!s) /* nomsg[_and_die] uses NULL fmt */
+               s = ""; /* some libc don't like printf(NULL) */
+
+       /* Prevent "derefing type-punned ptr will break aliasing rules" */
+       used = vasprintf((char**)(void*)msgptr, s, p);
+       if (used < 0)
+               return;
+
+       /* This is ugly and costs +60 bytes compared to multiple
+        * fprintf's, but is guaranteed to do a single write.
+        * This is needed for e.g. httpd logging, when multiple
+        * children can produce log messages simultaneously. */
+
+       strerr_len = strerr ? strlen(strerr) : 0;
+       msgeol_len = strlen(msg_eol);
+       /* +3 is for ": " before strerr and for terminating NUL */
+       msgv = xrealloc(msgv, used + strerr_len + msgeol_len + 3);
+       if (strerr) {
+               msgc[used++] = ':';
+               msgc[used++] = ' ';
+               strcpy(msgc + used, strerr);
+               used += strerr_len;
+       }
+       strcpy(msgc + used, msg_eol);
+       used += msgeol_len;
+
+       if (logmode & LOGMODE_STDIO) {
+               iov[0].iov_base = (char*)applet_name;
+               iov[0].iov_len = strlen(applet_name);
+               iov[1].iov_base = (char*)": ";
+               iov[1].iov_len = 2;
+               /*iov[2].iov_base = msgc;*/
+               /*iov[2].iov_len = used;*/
+               fflush_all();
+               writev(STDERR_FILENO, iov, 3);
        }
-       va_end(p2);
+# if ENABLE_FEATURE_SYSLOG
+       if (logmode & LOGMODE_SYSLOG) {
+               syslog(syslog_level, "%s", msgc);
+       }
+# endif
+       free(msgc);
+}
+#endif
+
+
+void FAST_FUNC bb_error_msg_and_die(const char *s, ...)
+{
+       va_list p;
+
+       va_start(p, s);
+       bb_verror_msg(s, p, NULL);
+       va_end(p);
+       xfunc_die();
+}
+
+void FAST_FUNC bb_error_msg(const char *s, ...)
+{
+       va_list p;
+
+       va_start(p, s);
+       bb_verror_msg(s, p, NULL);
+       va_end(p);
+}
+
+#if ENABLE_FEATURE_SYSLOG_INFO
+void FAST_FUNC bb_vinfo_msg(const char *s, va_list p)
+{
+       syslog_level = LOG_INFO;
+       bb_verror_msg(s, p, NULL);
+       syslog_level = LOG_ERR;
+}
+
+void FAST_FUNC bb_info_msg(const char *s, ...)
+{
+       va_list p;
+
+       va_start(p, s);
+       bb_vinfo_msg(s, p);
+       va_end(p);
+}
+
+void FAST_FUNC bb_simple_info_msg(const char *s)
+{
+       bb_info_msg("%s", s);
+}
+#endif
+
+void FAST_FUNC bb_simple_error_msg(const char *s)
+{
+       bb_error_msg("%s", s);
+}
+
+void FAST_FUNC bb_simple_error_msg_and_die(const char *s)
+{
+       bb_error_msg_and_die("%s", s);
 }