Fix regression of r23.
[platform/upstream/glog.git] / src / logging.cc
1 // Copyright (c) 1999, Google Inc.
2 // All rights reserved.
3 //
4 // Redistribution and use in source and binary forms, with or without
5 // modification, are permitted provided that the following conditions are
6 // met:
7 //
8 //     * Redistributions of source code must retain the above copyright
9 // notice, this list of conditions and the following disclaimer.
10 //     * Redistributions in binary form must reproduce the above
11 // copyright notice, this list of conditions and the following disclaimer
12 // in the documentation and/or other materials provided with the
13 // distribution.
14 //     * Neither the name of Google Inc. nor the names of its
15 // contributors may be used to endorse or promote products derived from
16 // this software without specific prior written permission.
17 //
18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29
30 #define _GNU_SOURCE 1 // needed for O_NOFOLLOW and pread()/pwrite()
31
32 #include "utilities.h"
33
34 #include <assert.h>
35 #include <iomanip>
36 #include <string>
37 #ifdef HAVE_UNISTD_H
38 # include <unistd.h>  // For _exit.
39 #endif
40 #include <climits>
41 #include <sys/types.h>
42 #include <sys/stat.h>
43 #ifdef HAVE_SYS_UTSNAME_H
44 # include <sys/utsname.h>  // For uname.
45 #endif
46 #include <fcntl.h>
47 #include <cstdio>
48 #include <iostream>
49 #include <stdarg.h>
50 #include <stdlib.h>
51 #ifdef HAVE_PWD_H
52 # include <pwd.h>
53 #endif
54 #ifdef HAVE_SYSLOG_H
55 # include <syslog.h>
56 #endif
57 #include <vector>
58 #include <errno.h>                   // for errno
59 #include <sstream>
60 #include "base/commandlineflags.h"        // to get the program name
61 #include "glog/logging.h"
62 #include "glog/raw_logging.h"
63 #include "base/googleinit.h"
64
65 #ifdef HAVE_STACKTRACE
66 # include "stacktrace.h"
67 #endif
68
69 using std::string;
70 using std::vector;
71 using std::ostrstream;
72 using std::setw;
73 using std::setfill;
74 using std::hex;
75 using std::dec;
76 using std::min;
77 using std::ostream;
78 using std::ostringstream;
79 using std::strstream;
80
81 // There is no thread annotation support.
82 #define EXCLUSIVE_LOCKS_REQUIRED(mu)
83
84 static bool BoolFromEnv(const char *varname, bool defval) {
85   const char* const valstr = getenv(varname);
86   if (!valstr) {
87     return defval;
88   }
89   return memchr("tTyY1\0", valstr[0], 6) != NULL;
90 }
91
92 DEFINE_bool(logtostderr, BoolFromEnv("GOOGLE_LOGTOSTDERR", false),
93             "log messages go to stderr instead of logfiles");
94 DEFINE_bool(alsologtostderr, BoolFromEnv("GOOGLE_ALSOLOGTOSTDERR", false),
95             "log messages go to stderr in addition to logfiles");
96 #ifdef OS_LINUX
97 DEFINE_bool(drop_log_memory, true, "Drop in-memory buffers of log contents. "
98             "Logs can grow very quickly and they are rarely read before they "
99             "need to be evicted from memory. Instead, drop them from memory "
100             "as soon as they are flushed to disk.");
101 _START_GOOGLE_NAMESPACE_
102 namespace logging {
103 static const int64 kPageSize = getpagesize();
104 }
105 _END_GOOGLE_NAMESPACE_
106 #endif
107
108 // By default, errors (including fatal errors) get logged to stderr as
109 // well as the file.
110 //
111 // The default is ERROR instead of FATAL so that users can see problems
112 // when they run a program without having to look in another file.
113 DEFINE_int32(stderrthreshold,
114              GOOGLE_NAMESPACE::ERROR,
115              "log messages at or above this level are copied to stderr in "
116              "addition to logfiles.  This flag obsoletes --alsologtostderr.");
117
118 DEFINE_string(alsologtoemail, "",
119               "log messages go to these email addresses "
120               "in addition to logfiles");
121 DEFINE_bool(log_prefix, true,
122             "Prepend the log prefix to the start of each log line");
123 DEFINE_int32(minloglevel, 0, "Messages logged at a lower level than this don't "
124              "actually get logged anywhere");
125 DEFINE_int32(logbuflevel, 0,
126              "Buffer log messages logged at this level or lower"
127              " (-1 means don't buffer; 0 means buffer INFO only;"
128              " ...)");
129 DEFINE_int32(logbufsecs, 30,
130              "Buffer log messages for at most this many seconds");
131 DEFINE_int32(logemaillevel, 999,
132              "Email log messages logged at this level or higher"
133              " (0 means email all; 3 means email FATAL only;"
134              " ...)");
135 DEFINE_string(logmailer, "/bin/mail",
136               "Mailer used to send logging email");
137
138 // Compute the default value for --log_dir
139 static const char* DefaultLogDir() {
140   const char* env;
141   env = getenv("GOOGLE_LOG_DIR");
142   if (env != NULL && env[0] != '\0') {
143     return env;
144   }
145   env = getenv("TEST_TMPDIR");
146   if (env != NULL && env[0] != '\0') {
147     return env;
148   }
149   return "";
150 }
151
152 DEFINE_string(log_dir, DefaultLogDir(),
153               "If specified, logfiles are written into this directory instead "
154               "of the default logging directory.");
155 DEFINE_string(log_link, "", "Put additional links to the log "
156               "files in this directory");
157
158 DEFINE_int32(max_log_size, 1800,
159              "approx. maximum log file size (in MB). A value of 0 will "
160              "be silently overridden to 1.");
161
162 DEFINE_bool(stop_logging_if_full_disk, false,
163             "Stop attempting to log to disk if the disk is full.");
164
165 DEFINE_string(log_backtrace_at, "",
166               "Emit a backtrace when logging at file:linenum.");
167
168 // TODO(hamaji): consider windows
169 #define PATH_SEPARATOR '/'
170
171 static void GetHostName(string* hostname) {
172 #if defined(HAVE_SYS_UTSNAME_H)
173   struct utsname buf;
174   if (0 != uname(&buf)) {
175     // ensure null termination on failure
176     *buf.nodename = '\0';
177   }
178   *hostname = buf.nodename;
179 #elif defined(OS_WINDOWS)
180   char buf[256];
181   DWORD len;
182   if (GetComputerNameA(buf, &len)) {
183     *hostname = buf;
184   } else {
185     hostname->clear();
186   }
187 #else
188 # warning There is no way to retrieve the host name.
189   *hostname = "(unknown)";
190 #endif
191 }
192
193 _START_GOOGLE_NAMESPACE_
194
195 // Safely get max_log_size, overriding to 1 if it somehow gets defined as 0
196 static int32 MaxLogSize() {
197   return (FLAGS_max_log_size > 0 ? FLAGS_max_log_size : 1);
198 }
199
200 // A mutex that allows only one thread to log at a time, to keep things from
201 // getting jumbled.  Some other very uncommon logging operations (like
202 // changing the destination file for log messages of a given severity) also
203 // lock this mutex.  Please be sure that anybody who might possibly need to
204 // lock it does so.
205 static Mutex log_mutex;
206
207 // Number of messages sent at each severity.  Under log_mutex.
208 int64 LogMessage::num_messages_[NUM_SEVERITIES] = {0, 0, 0, 0};
209
210 // Globally disable log writing (if disk is full)
211 static bool stop_writing = false;
212
213 const char*const LogSeverityNames[NUM_SEVERITIES] = {
214   "INFO", "WARNING", "ERROR", "FATAL"
215 };
216
217 // Has the user called SetExitOnDFatal(true)?
218 static bool exit_on_dfatal = true;
219
220 const char* GetLogSeverityName(LogSeverity severity) {
221   return LogSeverityNames[severity];
222 }
223
224 static bool SendEmailInternal(const char*dest, const char *subject,
225                               const char*body, bool use_logging);
226
227 base::Logger::~Logger() {
228 }
229
230 namespace {
231
232 // Encapsulates all file-system related state
233 class LogFileObject : public base::Logger {
234  public:
235   LogFileObject(LogSeverity severity, const char* base_filename);
236   ~LogFileObject();
237
238   virtual void Write(bool force_flush, // Should we force a flush here?
239                      time_t timestamp,  // Timestamp for this entry
240                      const char* message,
241                      int message_len);
242
243   // Configuration options
244   void SetBasename(const char* basename);
245   void SetExtension(const char* ext);
246   void SetSymlinkBasename(const char* symlink_basename);
247
248   // Normal flushing routine
249   virtual void Flush();
250
251   // It is the actual file length for the system loggers,
252   // i.e., INFO, ERROR, etc.
253   virtual uint32 LogSize() {
254     MutexLock l(&lock_);
255     return file_length_;
256   }
257
258   // Internal flush routine.  Exposed so that FlushLogFilesUnsafe()
259   // can avoid grabbing a lock.  Usually Flush() calls it after
260   // acquiring lock_.
261   void FlushUnlocked();
262
263  private:
264   static const uint32 kRolloverAttemptFrequency = 0x20;
265
266   Mutex lock_;
267   bool base_filename_selected_;
268   string base_filename_;
269   string symlink_basename_;
270   string filename_extension_;     // option users can specify (eg to add port#)
271   FILE* file_;
272   LogSeverity severity_;
273   uint32 bytes_since_flush_;
274   uint32 file_length_;
275   unsigned int rollover_attempt_;
276   int64 next_flush_time_;         // cycle count at which to flush log
277
278   // Actually create a logfile using the value of base_filename_ and the
279   // supplied argument time_pid_string
280   // REQUIRES: lock_ is held
281   bool CreateLogfile(const char* time_pid_string);
282 };
283
284 }  // namespace
285
286 class LogDestination {
287  public:
288   friend class LogMessage;
289   friend void ReprintFatalMessage();
290   friend base::Logger* base::GetLogger(LogSeverity);
291   friend void base::SetLogger(LogSeverity, base::Logger*);
292
293   // These methods are just forwarded to by their global versions.
294   static void SetLogDestination(LogSeverity severity,
295                                 const char* base_filename);
296   static void SetLogSymlink(LogSeverity severity,
297                             const char* symlink_basename);
298   static void AddLogSink(LogSink *destination);
299   static void RemoveLogSink(LogSink *destination);
300   static void SetLogFilenameExtension(const char* filename_extension);
301   static void SetStderrLogging(LogSeverity min_severity);
302   static void SetEmailLogging(LogSeverity min_severity, const char* addresses);
303   static void LogToStderr();
304   // Flush all log files that are at least at the given severity level
305   static void FlushLogFiles(int min_severity);
306   static void FlushLogFilesUnsafe(int min_severity);
307
308   // we set the maximum size of our packet to be 1400, the logic being
309   // to prevent fragmentation.
310   // Really this number is arbitrary.
311   static const int kNetworkBytes = 1400;
312
313   static const string& hostname();
314  private:
315
316   LogDestination(LogSeverity severity, const char* base_filename);
317   ~LogDestination() { }
318
319   // Take a log message of a particular severity and log it to stderr
320   // iff it's of a high enough severity to deserve it.
321   static void MaybeLogToStderr(LogSeverity severity, const char* message,
322                                size_t len);
323
324   // Take a log message of a particular severity and log it to email
325   // iff it's of a high enough severity to deserve it.
326   static void MaybeLogToEmail(LogSeverity severity, const char* message,
327                               size_t len);
328   // Take a log message of a particular severity and log it to a file
329   // iff the base filename is not "" (which means "don't log to me")
330   static void MaybeLogToLogfile(LogSeverity severity,
331                                 time_t timestamp,
332                                 const char* message, size_t len);
333   // Take a log message of a particular severity and log it to the file
334   // for that severity and also for all files with severity less than
335   // this severity.
336   static void LogToAllLogfiles(LogSeverity severity,
337                                time_t timestamp,
338                                const char* message, size_t len);
339
340   // Send logging info to all registered sinks.
341   static void LogToSinks(LogSeverity severity,
342                          const char *full_filename,
343                          const char *base_filename,
344                          int line,
345                          const struct ::tm* tm_time,
346                          const char* message,
347                          size_t message_len);
348
349   // Wait for all registered sinks via WaitTillSent
350   // including the optional one in "data".
351   static void WaitForSinks(LogMessage::LogMessageData* data);
352
353   static LogDestination* log_destination(LogSeverity severity);
354
355   LogFileObject fileobject_;
356   base::Logger* logger_;      // Either &fileobject_, or wrapper around it
357
358   static LogDestination* log_destinations_[NUM_SEVERITIES];
359   static LogSeverity email_logging_severity_;
360   static string addresses_;
361   static string hostname_;
362
363   // arbitrary global logging destinations.
364   static vector<LogSink*>* sinks_;
365
366   // Protects the vector sinks_,
367   // but not the LogSink objects its elements reference.
368   static Mutex sink_mutex_;
369
370   // Disallow
371   LogDestination(const LogDestination&);
372   LogDestination& operator=(const LogDestination&);
373 };
374
375 // Errors do not get logged to email by default.
376 LogSeverity LogDestination::email_logging_severity_ = 99999;
377
378 string LogDestination::addresses_;
379 string LogDestination::hostname_;
380
381 vector<LogSink*>* LogDestination::sinks_ = NULL;
382 Mutex LogDestination::sink_mutex_;
383
384 /* static */
385 const string& LogDestination::hostname() {
386   if (hostname_.empty()) {
387     GetHostName(&hostname_);
388     if (hostname_.empty()) {
389       hostname_ = "(unknown)";
390     }
391   }
392   return hostname_;
393 }
394
395 LogDestination::LogDestination(LogSeverity severity,
396                                const char* base_filename)
397   : fileobject_(severity, base_filename),
398     logger_(&fileobject_) {
399 }
400
401 inline void LogDestination::FlushLogFilesUnsafe(int min_severity) {
402   // assume we have the log_mutex or we simply don't care
403   // about it
404   for (int i = min_severity; i < NUM_SEVERITIES; i++) {
405     LogDestination* log = log_destination(i);
406     if (log != NULL) {
407       // Flush the base fileobject_ logger directly instead of going
408       // through any wrappers to reduce chance of deadlock.
409       log->fileobject_.FlushUnlocked();
410     }
411   }
412 }
413
414 inline void LogDestination::FlushLogFiles(int min_severity) {
415   // Prevent any subtle race conditions by wrapping a mutex lock around
416   // all this stuff.
417   MutexLock l(&log_mutex);
418   for (int i = min_severity; i < NUM_SEVERITIES; i++) {
419     LogDestination* log = log_destination(i);
420     if (log != NULL) {
421       log->logger_->Flush();
422     }
423   }
424 }
425
426 inline void LogDestination::SetLogDestination(LogSeverity severity,
427                                               const char* base_filename) {
428   assert(severity >= 0 && severity < NUM_SEVERITIES);
429   // Prevent any subtle race conditions by wrapping a mutex lock around
430   // all this stuff.
431   MutexLock l(&log_mutex);
432   log_destination(severity)->fileobject_.SetBasename(base_filename);
433 }
434
435 inline void LogDestination::SetLogSymlink(LogSeverity severity,
436                                           const char* symlink_basename) {
437   CHECK_GE(severity, 0);
438   CHECK_LT(severity, NUM_SEVERITIES);
439   MutexLock l(&log_mutex);
440   log_destination(severity)->fileobject_.SetSymlinkBasename(symlink_basename);
441 }
442
443 inline void LogDestination::AddLogSink(LogSink *destination) {
444   // Prevent any subtle race conditions by wrapping a mutex lock around
445   // all this stuff.
446   MutexLock l(&sink_mutex_);
447   if (!sinks_)  sinks_ = new vector<LogSink*>;
448   sinks_->push_back(destination);
449 }
450
451 inline void LogDestination::RemoveLogSink(LogSink *destination) {
452   // Prevent any subtle race conditions by wrapping a mutex lock around
453   // all this stuff.
454   MutexLock l(&sink_mutex_);
455   // This doesn't keep the sinks in order, but who cares?
456   if (sinks_) {
457     for (int i = sinks_->size() - 1; i >= 0; i--) {
458       if ((*sinks_)[i] == destination) {
459         (*sinks_)[i] = (*sinks_)[sinks_->size() - 1];
460         sinks_->pop_back();
461         break;
462       }
463     }
464   }
465 }
466
467 inline void LogDestination::SetLogFilenameExtension(const char* ext) {
468   // Prevent any subtle race conditions by wrapping a mutex lock around
469   // all this stuff.
470   MutexLock l(&log_mutex);
471   for ( int severity = 0; severity < NUM_SEVERITIES; ++severity ) {
472     log_destination(severity)->fileobject_.SetExtension(ext);
473   }
474 }
475
476 inline void LogDestination::SetStderrLogging(LogSeverity min_severity) {
477   assert(min_severity >= 0 && min_severity < NUM_SEVERITIES);
478   // Prevent any subtle race conditions by wrapping a mutex lock around
479   // all this stuff.
480   MutexLock l(&log_mutex);
481   FLAGS_stderrthreshold = min_severity;
482 }
483
484 inline void LogDestination::LogToStderr() {
485   // *Don't* put this stuff in a mutex lock, since SetStderrLogging &
486   // SetLogDestination already do the locking!
487   SetStderrLogging(0);            // thus everything is "also" logged to stderr
488   for ( int i = 0; i < NUM_SEVERITIES; ++i ) {
489     SetLogDestination(i, "");     // "" turns off logging to a logfile
490   }
491 }
492
493 inline void LogDestination::SetEmailLogging(LogSeverity min_severity,
494                                             const char* addresses) {
495   assert(min_severity >= 0 && min_severity < NUM_SEVERITIES);
496   // Prevent any subtle race conditions by wrapping a mutex lock around
497   // all this stuff.
498   MutexLock l(&log_mutex);
499   LogDestination::email_logging_severity_ = min_severity;
500   LogDestination::addresses_ = addresses;
501 }
502
503 static void WriteToStderr(const char* message, size_t len) {
504   // Avoid using cerr from this module since we may get called during
505   // exit code, and cerr may be partially or fully destroyed by then.
506   write(STDERR_FILENO, message, len);
507 }
508
509 inline void LogDestination::MaybeLogToStderr(LogSeverity severity,
510                                              const char* message, size_t len) {
511   if ((severity >= FLAGS_stderrthreshold) || FLAGS_alsologtostderr) {
512     WriteToStderr(message, len);
513 #ifdef OS_WINDOWS
514     // On Windows, also output to the debugger
515     ::OutputDebugStringA(string(message,len).c_str());
516 #endif
517   }
518 }
519
520
521 inline void LogDestination::MaybeLogToEmail(LogSeverity severity,
522                                             const char* message, size_t len) {
523   if (severity >= email_logging_severity_ ||
524       severity >= FLAGS_logemaillevel) {
525     string to(FLAGS_alsologtoemail);
526     if (!addresses_.empty()) {
527       if (!to.empty()) {
528         to += ",";
529       }
530       to += addresses_;
531     }
532     const string subject(string("[LOG] ") + LogSeverityNames[severity] + ": " +
533                          ProgramInvocationShortName());
534     string body(hostname());
535     body += "\n\n";
536     body.append(message, len);
537
538     // should NOT use SendEmail().  The caller of this function holds the
539     // log_mutex and SendEmail() calls LOG/VLOG which will block trying to
540     // acquire the log_mutex object.  Use SendEmailInternal() and set
541     // use_logging to false.
542     SendEmailInternal(to.c_str(), subject.c_str(), body.c_str(), false);
543   }
544 }
545
546
547 inline void LogDestination::MaybeLogToLogfile(LogSeverity severity,
548                                               time_t timestamp,
549                                               const char* message,
550                                               size_t len) {
551   const bool should_flush = severity > FLAGS_logbuflevel;
552   LogDestination* destination = log_destination(severity);
553   destination->logger_->Write(should_flush, timestamp, message, len);
554 }
555
556 inline void LogDestination::LogToAllLogfiles(LogSeverity severity,
557                                              time_t timestamp,
558                                              const char* message,
559                                              size_t len) {
560
561   if ( FLAGS_logtostderr )            // global flag: never log to file
562     WriteToStderr(message, len);
563   else
564     for (int i = severity; i >= 0; --i)
565       LogDestination::MaybeLogToLogfile(i, timestamp, message, len);
566
567 }
568
569 inline void LogDestination::LogToSinks(LogSeverity severity,
570                                        const char *full_filename,
571                                        const char *base_filename,
572                                        int line,
573                                        const struct ::tm* tm_time,
574                                        const char* message,
575                                        size_t message_len) {
576   ReaderMutexLock l(&sink_mutex_);
577   if (sinks_) {
578     for (int i = sinks_->size() - 1; i >= 0; i--) {
579       (*sinks_)[i]->send(severity, full_filename, base_filename,
580                          line, tm_time, message, message_len);
581     }
582   }
583 }
584
585 inline void LogDestination::WaitForSinks(LogMessage::LogMessageData* data) {
586   ReaderMutexLock l(&sink_mutex_);
587   if (sinks_) {
588     for (int i = sinks_->size() - 1; i >= 0; i--) {
589       (*sinks_)[i]->WaitTillSent();
590     }
591   }
592   const bool send_to_sink =
593       (data->send_method_ == &LogMessage::SendToSink) ||
594       (data->send_method_ == &LogMessage::SendToSinkAndLog);
595   if (send_to_sink && data->sink_ != NULL) {
596     data->sink_->WaitTillSent();
597   }
598 }
599
600 LogDestination* LogDestination::log_destinations_[NUM_SEVERITIES];
601
602 inline LogDestination* LogDestination::log_destination(LogSeverity severity) {
603   assert(severity >=0 && severity < NUM_SEVERITIES);
604   if (!log_destinations_[severity]) {
605     log_destinations_[severity] = new LogDestination(severity, NULL);
606   }
607   return log_destinations_[severity];
608 }
609
610 namespace {
611
612 LogFileObject::LogFileObject(LogSeverity severity,
613                              const char* base_filename)
614   : base_filename_selected_(base_filename != NULL),
615     base_filename_((base_filename != NULL) ? base_filename : ""),
616     symlink_basename_(ProgramInvocationShortName()),
617     filename_extension_(),
618     file_(NULL),
619     severity_(severity),
620     bytes_since_flush_(0),
621     file_length_(0),
622     rollover_attempt_(kRolloverAttemptFrequency-1),
623     next_flush_time_(0) {
624   assert(severity >= 0);
625   assert(severity < NUM_SEVERITIES);
626 }
627
628 LogFileObject::~LogFileObject() {
629   MutexLock l(&lock_);
630   if (file_ != NULL) {
631     fclose(file_);
632     file_ = NULL;
633   }
634 }
635
636 void LogFileObject::SetBasename(const char* basename) {
637   MutexLock l(&lock_);
638   base_filename_selected_ = true;
639   if (base_filename_ != basename) {
640     // Get rid of old log file since we are changing names
641     if (file_ != NULL) {
642       fclose(file_);
643       file_ = NULL;
644       rollover_attempt_ = kRolloverAttemptFrequency-1;
645     }
646     base_filename_ = basename;
647   }
648 }
649
650 void LogFileObject::SetExtension(const char* ext) {
651   MutexLock l(&lock_);
652   if (filename_extension_ != ext) {
653     // Get rid of old log file since we are changing names
654     if (file_ != NULL) {
655       fclose(file_);
656       file_ = NULL;
657       rollover_attempt_ = kRolloverAttemptFrequency-1;
658     }
659     filename_extension_ = ext;
660   }
661 }
662
663 void LogFileObject::SetSymlinkBasename(const char* symlink_basename) {
664   MutexLock l(&lock_);
665   symlink_basename_ = symlink_basename;
666 }
667
668 void LogFileObject::Flush() {
669   MutexLock l(&lock_);
670   FlushUnlocked();
671 }
672
673 void LogFileObject::FlushUnlocked(){
674   if (file_ != NULL) {
675     fflush(file_);
676     bytes_since_flush_ = 0;
677   }
678   // Figure out when we are due for another flush.
679   const int64 next = (FLAGS_logbufsecs
680                       * static_cast<int64>(1000000));  // in usec
681   next_flush_time_ = CycleClock_Now() + UsecToCycles(next);
682 }
683
684 bool LogFileObject::CreateLogfile(const char* time_pid_string) {
685   string string_filename = base_filename_+filename_extension_+
686                            time_pid_string;
687   const char* filename = string_filename.c_str();
688   int fd = open(filename, O_WRONLY | O_CREAT | O_EXCL, 0664);
689   if (fd == -1) return false;
690 #ifdef HAVE_FCNTL
691   // Mark the file close-on-exec. We don't really care if this fails
692   fcntl(fd, F_SETFD, FD_CLOEXEC);
693 #endif
694
695   file_ = fdopen(fd, "a");  // Make a FILE*.
696   if (file_ == NULL) {  // Man, we're screwed!
697     close(fd);
698     unlink(filename);  // Erase the half-baked evidence: an unusable log file
699     return false;
700   }
701
702   // We try to create a symlink called <program_name>.<severity>,
703   // which is easier to use.  (Every time we create a new logfile,
704   // we destroy the old symlink and create a new one, so it always
705   // points to the latest logfile.)  If it fails, we're sad but it's
706   // no error.
707   if (!symlink_basename_.empty()) {
708     // take directory from filename
709     const char* slash = strrchr(filename, PATH_SEPARATOR);
710     const string linkname =
711       symlink_basename_ + '.' + LogSeverityNames[severity_];
712     string linkpath;
713     if ( slash ) linkpath = string(filename, slash-filename+1);  // get dirname
714     linkpath += linkname;
715     unlink(linkpath.c_str());                    // delete old one if it exists
716
717     // We must have unistd.h.
718 #ifdef HAVE_UNISTD_H
719     // Make the symlink be relative (in the same dir) so that if the
720     // entire log directory gets relocated the link is still valid.
721     const char *linkdest = slash ? (slash + 1) : filename;
722     symlink(linkdest, linkpath.c_str());         // silently ignore failures
723
724     // Make an additional link to the log file in a place specified by
725     // FLAGS_log_link, if indicated
726     if (!FLAGS_log_link.empty()) {
727       linkpath = FLAGS_log_link + "/" + linkname;
728       unlink(linkpath.c_str());                  // delete old one if it exists
729       symlink(filename, linkpath.c_str());       // silently ignore failures
730     }
731 #endif
732   }
733
734   return true;  // Everything worked
735 }
736
737 void LogFileObject::Write(bool force_flush,
738                           time_t timestamp,
739                           const char* message,
740                           int message_len) {
741   MutexLock l(&lock_);
742
743   // We don't log if the base_name_ is "" (which means "don't write")
744   if (base_filename_selected_ && base_filename_.empty()) {
745     return;
746   }
747
748   if (static_cast<int>(file_length_ >> 20) >= MaxLogSize()) {
749     if (file_ != NULL) fclose(file_);
750     file_ = NULL;
751     file_length_ = bytes_since_flush_ = 0;
752     rollover_attempt_ = kRolloverAttemptFrequency-1;
753   }
754
755   // If there's no destination file, make one before outputting
756   if (file_ == NULL) {
757     // Try to rollover the log file every 32 log messages.  The only time
758     // this could matter would be when we have trouble creating the log
759     // file.  If that happens, we'll lose lots of log messages, of course!
760     if (++rollover_attempt_ != kRolloverAttemptFrequency) return;
761     rollover_attempt_ = 0;
762
763     struct ::tm tm_time;
764     localtime_r(&timestamp, &tm_time);
765
766     // The logfile's filename will have the date/time & pid in it
767     char time_pid_string[256];  // More than enough chars for time, pid, \0
768     ostrstream time_pid_stream(time_pid_string, sizeof(time_pid_string));
769     time_pid_stream.fill('0');
770     time_pid_stream << 1900+tm_time.tm_year
771                     << setw(2) << 1+tm_time.tm_mon
772                     << setw(2) << tm_time.tm_mday
773                     << '-'
774                     << setw(2) << tm_time.tm_hour
775                     << setw(2) << tm_time.tm_min
776                     << setw(2) << tm_time.tm_sec
777                     << '.'
778                     << GetMainThreadPid()
779                     << '\0';
780
781     if (base_filename_selected_) {
782       if (!CreateLogfile(time_pid_string)) {
783         perror("Could not create log file");
784         fprintf(stderr, "COULD NOT CREATE LOGFILE '%s'!\n", time_pid_string);
785         return;
786       }
787     } else {
788       // If no base filename for logs of this severity has been set, use a
789       // default base filename of
790       // "<program name>.<hostname>.<user name>.log.<severity level>.".  So
791       // logfiles will have names like
792       // webserver.examplehost.root.log.INFO.19990817-150000.4354, where
793       // 19990817 is a date (1999 August 17), 150000 is a time (15:00:00),
794       // and 4354 is the pid of the logging process.  The date & time reflect
795       // when the file was created for output.
796       //
797       // Where does the file get put?  Successively try the directories
798       // "/tmp", and "."
799       string stripped_filename(ProgramInvocationShortName());  // in cmdlineflag
800       string hostname;
801       GetHostName(&hostname);
802
803       string uidname = MyUserName();
804       // We should not call CHECK() here because this function can be
805       // called after holding on to log_mutex. We don't want to
806       // attempt to hold on to the same mutex, and get into a
807       // deadlock. Simply use a name like invalid-user.
808       if (uidname.empty()) uidname = "invalid-user";
809
810       stripped_filename = stripped_filename+'.'+hostname+'.'
811                           +uidname+".log."
812                           +LogSeverityNames[severity_]+'.';
813       // We're going to (potentially) try to put logs in several different dirs
814       const vector<string> & log_dirs = GetLoggingDirectories();
815
816       // Go through the list of dirs, and try to create the log file in each
817       // until we succeed or run out of options
818       bool success = false;
819       for (vector<string>::const_iterator dir = log_dirs.begin();
820            dir != log_dirs.end();
821            ++dir) {
822         base_filename_ = *dir + "/" + stripped_filename;
823         if ( CreateLogfile(time_pid_string) ) {
824           success = true;
825           break;
826         }
827       }
828       // If we never succeeded, we have to give up
829       if ( success == false ) {
830         perror("Could not create logging file");
831         fprintf(stderr, "COULD NOT CREATE A LOGGINGFILE %s!", time_pid_string);
832         return;
833       }
834     }
835
836     // Write a header message into the log file
837     char file_header_string[512];  // Enough chars for time and binary info
838     ostrstream file_header_stream(file_header_string,
839                                   sizeof(file_header_string));
840     file_header_stream.fill('0');
841     file_header_stream << "Log file created at: "
842                        << 1900+tm_time.tm_year << '/'
843                        << setw(2) << 1+tm_time.tm_mon << '/'
844                        << setw(2) << tm_time.tm_mday
845                        << ' '
846                        << setw(2) << tm_time.tm_hour << ':'
847                        << setw(2) << tm_time.tm_min << ':'
848                        << setw(2) << tm_time.tm_sec << '\n'
849                        << "Running on machine: "
850                        << LogDestination::hostname() << '\n'
851                        << "Log line format: [IWEF]mmdd hh:mm:ss.uuuuuu "
852                        << "threadid file:line] msg" << '\n'
853                        << '\0';
854     int header_len = strlen(file_header_string);
855     fwrite(file_header_string, 1, header_len, file_);
856     file_length_ += header_len;
857     bytes_since_flush_ += header_len;
858   }
859
860   // Write to LOG file
861   if ( !stop_writing ) {
862     // fwrite() doesn't return an error when the disk is full, for
863     // messages that are less than 4096 bytes. When the disk is full,
864     // it returns the message length for messages that are less than
865     // 4096 bytes. fwrite() returns 4096 for message lengths that are
866     // greater than 4096, thereby indicating an error.
867     errno = 0;
868     fwrite(message, 1, message_len, file_);
869     if ( FLAGS_stop_logging_if_full_disk &&
870          errno == ENOSPC ) {  // disk full, stop writing to disk
871       stop_writing = true;  // until the disk is
872       return;
873     } else {
874       file_length_ += message_len;
875       bytes_since_flush_ += message_len;
876     }
877   } else {
878     if ( CycleClock_Now() >= next_flush_time_ )
879       stop_writing = false;  // check to see if disk has free space.
880     return;  // no need to flush
881   }
882
883   // See important msgs *now*.  Also, flush logs at least every 10^6 chars,
884   // or every "FLAGS_logbufsecs" seconds.
885   if ( force_flush ||
886        (bytes_since_flush_ >= 1000000) ||
887        (CycleClock_Now() >= next_flush_time_) ) {
888     FlushUnlocked();
889 #ifdef OS_LINUX
890     if (FLAGS_drop_log_memory) {
891       if (file_length_ >= logging::kPageSize) {
892         // don't evict the most recent page
893         uint32 len = file_length_ & ~(logging::kPageSize - 1);
894         posix_fadvise(fileno(file_), 0, len, POSIX_FADV_DONTNEED);
895       }
896     }
897 #endif
898   }
899 }
900
901 }  // namespace
902
903 // An arbitrary limit on the length of a single log message.  This
904 // is so that streaming can be done more efficiently.
905 const size_t LogMessage::kMaxLogMessageLen = 30000;
906
907 // Static log data space to avoid alloc failures in a LOG(FATAL)
908 //
909 // Since multiple threads may call LOG(FATAL), and we want to preserve
910 // the data from the first call, we allocate two sets of space.  One
911 // for exclusive use by the first thread, and one for shared use by
912 // all other threads.
913 static Mutex fatal_msg_lock;
914 static CrashReason crash_reason;
915 static bool fatal_msg_exclusive = true;
916 static char fatal_msg_buf_exclusive[LogMessage::kMaxLogMessageLen+1];
917 static char fatal_msg_buf_shared[LogMessage::kMaxLogMessageLen+1];
918 static LogMessage::LogStream fatal_msg_stream_exclusive(
919     fatal_msg_buf_exclusive, LogMessage::kMaxLogMessageLen, 0);
920 static LogMessage::LogStream fatal_msg_stream_shared(
921     fatal_msg_buf_shared, LogMessage::kMaxLogMessageLen, 0);
922 LogMessage::LogMessageData LogMessage::fatal_msg_data_exclusive_;
923 LogMessage::LogMessageData LogMessage::fatal_msg_data_shared_;
924
925 LogMessage::LogMessageData::~LogMessageData() {
926   delete[] buf_;
927   delete stream_alloc_;
928 }
929
930 LogMessage::LogMessage(const char* file, int line, LogSeverity severity,
931                        int ctr, void (LogMessage::*send_method)()) {
932   Init(file, line, severity, send_method);
933   data_->stream_->set_ctr(ctr);
934 }
935
936 LogMessage::LogMessage(const char* file, int line,
937                        const CheckOpString& result) {
938   Init(file, line, FATAL, &LogMessage::SendToLog);
939   stream() << "Check failed: " << (*result.str_) << " ";
940 }
941
942 LogMessage::LogMessage(const char* file, int line) {
943   Init(file, line, INFO, &LogMessage::SendToLog);
944 }
945
946 LogMessage::LogMessage(const char* file, int line, LogSeverity severity) {
947   Init(file, line, severity, &LogMessage::SendToLog);
948 }
949
950 LogMessage::LogMessage(const char* file, int line, LogSeverity severity,
951                        LogSink* sink, bool also_send_to_log) {
952   Init(file, line, severity, also_send_to_log ? &LogMessage::SendToSinkAndLog :
953                                                 &LogMessage::SendToSink);
954   data_->sink_ = sink;  // override Init()'s setting to NULL
955 }
956
957 LogMessage::LogMessage(const char* file, int line, LogSeverity severity,
958                        vector<string> *outvec) {
959   Init(file, line, severity, &LogMessage::SaveOrSendToLog);
960   data_->outvec_ = outvec; // override Init()'s setting to NULL
961 }
962
963 LogMessage::LogMessage(const char* file, int line, LogSeverity severity,
964                        string *message) {
965   Init(file, line, severity, &LogMessage::WriteToStringAndLog);
966   data_->message_ = message;  // override Init()'s setting to NULL
967 }
968
969 void LogMessage::Init(const char* file,
970                       int line,
971                       LogSeverity severity,
972                       void (LogMessage::*send_method)()) {
973   allocated_ = NULL;
974   if (severity != FATAL || !exit_on_dfatal) {
975     allocated_ = new LogMessageData();
976     data_ = allocated_;
977     data_->buf_ = new char[kMaxLogMessageLen+1];
978     data_->message_text_ = data_->buf_;
979     data_->stream_alloc_ =
980         new LogStream(data_->message_text_, kMaxLogMessageLen, 0);
981     data_->stream_ = data_->stream_alloc_;
982     data_->first_fatal_ = false;
983   } else {
984     MutexLock l(&fatal_msg_lock);
985     if (fatal_msg_exclusive) {
986       fatal_msg_exclusive = false;
987       data_ = &fatal_msg_data_exclusive_;
988       data_->message_text_ = fatal_msg_buf_exclusive;
989       data_->stream_ = &fatal_msg_stream_exclusive;
990       data_->first_fatal_ = true;
991     } else {
992       data_ = &fatal_msg_data_shared_;
993       data_->message_text_ = fatal_msg_buf_shared;
994       data_->stream_ = &fatal_msg_stream_shared;
995       data_->first_fatal_ = false;
996     }
997     data_->stream_alloc_ = NULL;
998   }
999
1000   stream().fill('0');
1001   data_->preserved_errno_ = errno;
1002   data_->severity_ = severity;
1003   data_->line_ = line;
1004   data_->send_method_ = send_method;
1005   data_->sink_ = NULL;
1006   data_->outvec_ = NULL;
1007   WallTime now = WallTime_Now();
1008   data_->timestamp_ = static_cast<time_t>(now);
1009   localtime_r(&data_->timestamp_, &data_->tm_time_);
1010   int usecs = static_cast<int>((now - data_->timestamp_) * 1000000);
1011   RawLog__SetLastTime(data_->tm_time_, usecs);
1012
1013   data_->num_chars_to_log_ = 0;
1014   data_->num_chars_to_syslog_ = 0;
1015   data_->basename_ = const_basename(file);
1016   data_->fullname_ = file;
1017   data_->has_been_flushed_ = false;
1018
1019   // If specified, prepend a prefix to each line.  For example:
1020   //    I1018 160715 f5d4fbb0 logging.cc:1153]
1021   //    (log level, GMT month, date, time, thread_id, file basename, line)
1022   // We exclude the thread_id for the default thread.
1023   if (FLAGS_log_prefix && (line != kNoLogPrefix)) {
1024     stream() << LogSeverityNames[severity][0]
1025              << setw(2) << 1+data_->tm_time_.tm_mon
1026              << setw(2) << data_->tm_time_.tm_mday
1027              << ' '
1028              << setw(2) << data_->tm_time_.tm_hour  << ':'
1029              << setw(2) << data_->tm_time_.tm_min   << ':'
1030              << setw(2) << data_->tm_time_.tm_sec   << "."
1031              << setw(6) << usecs
1032              << ' '
1033              << setfill(' ') << setw(5)
1034              << static_cast<unsigned int>(GetTID()) << setfill('0')
1035              << ' '
1036              << data_->basename_ << ':' << data_->line_ << "] ";
1037   }
1038   data_->num_prefix_chars_ = data_->stream_->pcount();
1039
1040   if (!FLAGS_log_backtrace_at.empty()) {
1041     char fileline[128];
1042     snprintf(fileline, sizeof(fileline), "%s:%d", data_->basename_, line);
1043 #ifdef HAVE_STACKTRACE
1044     if (!strcmp(FLAGS_log_backtrace_at.c_str(), fileline)) {
1045       string stacktrace;
1046       DumpStackTraceToString(&stacktrace);
1047       stream() << " (stacktrace:\n" << stacktrace << ") ";
1048     }
1049 #endif
1050   }
1051 }
1052
1053 LogMessage::~LogMessage() {
1054   Flush();
1055   delete allocated_;
1056 }
1057
1058 // Flush buffered message, called by the destructor, or any other function
1059 // that needs to synchronize the log.
1060 void LogMessage::Flush() {
1061   if (data_->has_been_flushed_ || data_->severity_ < FLAGS_minloglevel)
1062     return;
1063
1064   data_->num_chars_to_log_ = data_->stream_->pcount();
1065   data_->num_chars_to_syslog_ =
1066     data_->num_chars_to_log_ - data_->num_prefix_chars_;
1067
1068   // Do we need to add a \n to the end of this message?
1069   bool append_newline =
1070       (data_->message_text_[data_->num_chars_to_log_-1] != '\n');
1071   char original_final_char = '\0';
1072
1073   // If we do need to add a \n, we'll do it by violating the memory of the
1074   // ostrstream buffer.  This is quick, and we'll make sure to undo our
1075   // modification before anything else is done with the ostrstream.  It
1076   // would be preferable not to do things this way, but it seems to be
1077   // the best way to deal with this.
1078   if (append_newline) {
1079     original_final_char = data_->message_text_[data_->num_chars_to_log_];
1080     data_->message_text_[data_->num_chars_to_log_++] = '\n';
1081   }
1082
1083   // Prevent any subtle race conditions by wrapping a mutex lock around
1084   // the actual logging action per se.
1085   {
1086     MutexLock l(&log_mutex);
1087     (this->*(data_->send_method_))();
1088     ++num_messages_[static_cast<int>(data_->severity_)];
1089   }
1090   LogDestination::WaitForSinks(data_);
1091
1092   if (append_newline) {
1093     // Fix the ostrstream back how it was before we screwed with it.
1094     // It's 99.44% certain that we don't need to worry about doing this.
1095     data_->message_text_[data_->num_chars_to_log_-1] = original_final_char;
1096   }
1097
1098   // If errno was already set before we enter the logging call, we'll
1099   // set it back to that value when we return from the logging call.
1100   // It happens often that we log an error message after a syscall
1101   // failure, which can potentially set the errno to some other
1102   // values.  We would like to preserve the original errno.
1103   if (data_->preserved_errno_ != 0) {
1104     errno = data_->preserved_errno_;
1105   }
1106
1107   // Note that this message is now safely logged.  If we're asked to flush
1108   // again, as a result of destruction, say, we'll do nothing on future calls.
1109   data_->has_been_flushed_ = true;
1110 }
1111
1112 // Copy of first FATAL log message so that we can print it out again
1113 // after all the stack traces.  To preserve legacy behavior, we don't
1114 // use fatal_msg_buf_exclusive.
1115 static time_t fatal_time;
1116 static char fatal_message[256];
1117
1118 void ReprintFatalMessage() {
1119   if (fatal_message[0]) {
1120     const int n = strlen(fatal_message);
1121     if (!FLAGS_logtostderr) {
1122       // Also write to stderr
1123       WriteToStderr(fatal_message, n);
1124     }
1125     LogDestination::LogToAllLogfiles(ERROR, fatal_time, fatal_message, n);
1126   }
1127 }
1128
1129 // L >= log_mutex (callers must hold the log_mutex).
1130 void LogMessage::SendToLog() EXCLUSIVE_LOCKS_REQUIRED(log_mutex) {
1131   static bool already_warned_before_initgoogle = false;
1132
1133   log_mutex.AssertHeld();
1134
1135   RAW_DCHECK(data_->num_chars_to_log_ > 0 &&
1136              data_->message_text_[data_->num_chars_to_log_-1] == '\n', "");
1137
1138   // Messages of a given severity get logged to lower severity logs, too
1139
1140   if (!already_warned_before_initgoogle && !IsGoogleLoggingInitialized()) {
1141     const char w[] = "WARNING: Logging before InitGoogleLogging() is "
1142                      "written to STDERR\n";
1143     WriteToStderr(w, strlen(w));
1144     already_warned_before_initgoogle = true;
1145   }
1146
1147   // global flag: never log to file if set.  Also -- don't log to a
1148   // file if we haven't parsed the command line flags to get the
1149   // program name.
1150   if (FLAGS_logtostderr || !IsGoogleLoggingInitialized()) {
1151     WriteToStderr(data_->message_text_, data_->num_chars_to_log_);
1152
1153     // this could be protected by a flag if necessary.
1154     LogDestination::LogToSinks(data_->severity_,
1155                                data_->fullname_, data_->basename_,
1156                                data_->line_, &data_->tm_time_,
1157                                data_->message_text_ + data_->num_prefix_chars_,
1158                                (data_->num_chars_to_log_ -
1159                                 data_->num_prefix_chars_ - 1));
1160   } else {
1161
1162     // log this message to all log files of severity <= severity_
1163     LogDestination::LogToAllLogfiles(data_->severity_, data_->timestamp_,
1164                                      data_->message_text_,
1165                                      data_->num_chars_to_log_);
1166
1167     LogDestination::MaybeLogToStderr(data_->severity_, data_->message_text_,
1168                                      data_->num_chars_to_log_);
1169     LogDestination::MaybeLogToEmail(data_->severity_, data_->message_text_,
1170                                     data_->num_chars_to_log_);
1171     LogDestination::LogToSinks(data_->severity_,
1172                                data_->fullname_, data_->basename_,
1173                                data_->line_, &data_->tm_time_,
1174                                data_->message_text_ + data_->num_prefix_chars_,
1175                                (data_->num_chars_to_log_
1176                                 - data_->num_prefix_chars_ - 1));
1177     // NOTE: -1 removes trailing \n
1178   }
1179
1180   // If we log a FATAL message, flush all the log destinations, then toss
1181   // a signal for others to catch. We leave the logs in a state that
1182   // someone else can use them (as long as they flush afterwards)
1183   if (data_->severity_ == FATAL && exit_on_dfatal) {
1184     if (data_->first_fatal_) {
1185       // Store crash information so that it is accessible from within signal
1186       // handlers that may be invoked later.
1187       RecordCrashReason(&crash_reason);
1188       SetCrashReason(&crash_reason);
1189
1190       // Store shortened fatal message for other logs and GWQ status
1191       const int copy = min<int>(data_->num_chars_to_log_,
1192                                 sizeof(fatal_message)-1);
1193       memcpy(fatal_message, data_->message_text_, copy);
1194       fatal_message[copy] = '\0';
1195       fatal_time = data_->timestamp_;
1196     }
1197
1198     if (!FLAGS_logtostderr) {
1199       for (int i = 0; i < NUM_SEVERITIES; ++i) {
1200         if ( LogDestination::log_destinations_[i] )
1201           LogDestination::log_destinations_[i]->logger_->Write(true, 0, "", 0);
1202       }
1203     }
1204
1205     // release the lock that our caller (directly or indirectly)
1206     // LogMessage::~LogMessage() grabbed so that signal handlers
1207     // can use the logging facility. Alternately, we could add
1208     // an entire unsafe logging interface to bypass locking
1209     // for signal handlers but this seems simpler.
1210     log_mutex.Unlock();
1211     LogDestination::WaitForSinks(data_);
1212
1213     const char* message = "*** Check failure stack trace: ***\n";
1214     write(STDERR_FILENO, message, strlen(message));
1215     Fail();
1216   }
1217 }
1218
1219 void LogMessage::RecordCrashReason(
1220     glog_internal_namespace_::CrashReason* reason) {
1221   reason->filename = fatal_msg_data_exclusive_.fullname_;
1222   reason->line_number = fatal_msg_data_exclusive_.line_;
1223   reason->message = fatal_msg_buf_exclusive +
1224                     fatal_msg_data_exclusive_.num_prefix_chars_;
1225 #ifdef HAVE_STACKTRACE
1226   // Retrieve the stack trace, omitting the logging frames that got us here.
1227   reason->depth = GetStackTrace(reason->stack, ARRAYSIZE(reason->stack), 4);
1228 #else
1229   reason->depth = 0;
1230 #endif
1231 }
1232
1233 static void logging_fail() {
1234 #if defined(_DEBUG) && defined(_MSC_VER)
1235   // When debugging on windows, avoid the obnoxious dialog and make
1236   // it possible to continue past a LOG(FATAL) in the debugger
1237   _asm int 3
1238 #else
1239   abort();
1240 #endif
1241 }
1242
1243 #ifdef HAVE___ATTRIBUTE__
1244 GOOGLE_GLOG_DLL_DECL
1245 void (*g_logging_fail_func)() __attribute__((noreturn)) = &logging_fail;
1246 #else
1247 GOOGLE_GLOG_DLL_DECL void (*g_logging_fail_func)() = &logging_fail;
1248 #endif
1249
1250 void InstallFailureFunction(void (*fail_func)()) {
1251   g_logging_fail_func = fail_func;
1252 }
1253
1254 void LogMessage::Fail() {
1255   g_logging_fail_func();
1256 }
1257
1258 // L >= log_mutex (callers must hold the log_mutex).
1259 void LogMessage::SendToSink() EXCLUSIVE_LOCKS_REQUIRED(log_mutex) {
1260   if (data_->sink_ != NULL) {
1261     RAW_DCHECK(data_->num_chars_to_log_ > 0 &&
1262                data_->message_text_[data_->num_chars_to_log_-1] == '\n', "");
1263     data_->sink_->send(data_->severity_, data_->fullname_, data_->basename_,
1264                        data_->line_, &data_->tm_time_,
1265                        data_->message_text_ + data_->num_prefix_chars_,
1266                        (data_->num_chars_to_log_ -
1267                         data_->num_prefix_chars_ - 1));
1268   }
1269 }
1270
1271 // L >= log_mutex (callers must hold the log_mutex).
1272 void LogMessage::SendToSinkAndLog() EXCLUSIVE_LOCKS_REQUIRED(log_mutex) {
1273   SendToSink();
1274   SendToLog();
1275 }
1276
1277 // L >= log_mutex (callers must hold the log_mutex).
1278 void LogMessage::SaveOrSendToLog() EXCLUSIVE_LOCKS_REQUIRED(log_mutex) {
1279   if (data_->outvec_ != NULL) {
1280     RAW_DCHECK(data_->num_chars_to_log_ > 0 &&
1281                data_->message_text_[data_->num_chars_to_log_-1] == '\n', "");
1282     // Omit prefix of message and trailing newline when recording in outvec_.
1283     const char *start = data_->message_text_ + data_->num_prefix_chars_;
1284     int len = data_->num_chars_to_log_ - data_->num_prefix_chars_ - 1;
1285     data_->outvec_->push_back(string(start, len));
1286   } else {
1287     SendToLog();
1288   }
1289 }
1290
1291 void LogMessage::WriteToStringAndLog() EXCLUSIVE_LOCKS_REQUIRED(log_mutex) {
1292   if (data_->message_ != NULL) {
1293     RAW_DCHECK(data_->num_chars_to_log_ > 0 &&
1294                data_->message_text_[data_->num_chars_to_log_-1] == '\n', "");
1295     // Omit prefix of message and trailing newline when writing to message_.
1296     const char *start = data_->message_text_ + data_->num_prefix_chars_;
1297     int len = data_->num_chars_to_log_ - data_->num_prefix_chars_ - 1;
1298     data_->message_->assign(start, len);
1299   }
1300   SendToLog();
1301 }
1302
1303 // L >= log_mutex (callers must hold the log_mutex).
1304 void LogMessage::SendToSyslogAndLog() {
1305 #ifdef HAVE_SYSLOG_H
1306   // Before any calls to syslog(), make a single call to openlog()
1307   static bool openlog_already_called = false;
1308   if (!openlog_already_called) {
1309     openlog(ProgramInvocationShortName(), LOG_CONS | LOG_NDELAY | LOG_PID,
1310             LOG_USER);
1311     openlog_already_called = true;
1312   }
1313
1314   // This array maps Google severity levels to syslog levels
1315   const int SEVERITY_TO_LEVEL[] = { LOG_INFO, LOG_WARNING, LOG_ERR, LOG_EMERG };
1316   syslog(LOG_USER | SEVERITY_TO_LEVEL[static_cast<int>(data_->severity_)], "%.*s",
1317          int(data_->num_chars_to_syslog_),
1318          data_->message_text_ + data_->num_prefix_chars_);
1319   SendToLog();
1320 #else
1321   LOG(ERROR) << "No syslog support: message=" << data_->message_text_;
1322 #endif
1323 }
1324
1325 base::Logger* base::GetLogger(LogSeverity severity) {
1326   MutexLock l(&log_mutex);
1327   return LogDestination::log_destination(severity)->logger_;
1328 }
1329
1330 void base::SetLogger(LogSeverity severity, base::Logger* logger) {
1331   MutexLock l(&log_mutex);
1332   LogDestination::log_destination(severity)->logger_ = logger;
1333 }
1334
1335 // L < log_mutex.  Acquires and releases mutex_.
1336 int64 LogMessage::num_messages(int severity) {
1337   MutexLock l(&log_mutex);
1338   return num_messages_[severity];
1339 }
1340
1341 // Output the COUNTER value. This is only valid if ostream is a
1342 // LogStream.
1343 ostream& operator<<(ostream &os, const PRIVATE_Counter&) {
1344   LogMessage::LogStream *log = dynamic_cast<LogMessage::LogStream*>(&os);
1345   CHECK(log == log->self());
1346   os << log->ctr();
1347   return os;
1348 }
1349
1350 ErrnoLogMessage::ErrnoLogMessage(const char* file, int line,
1351                                  LogSeverity severity, int ctr,
1352                                  void (LogMessage::*send_method)())
1353     : LogMessage(file, line, severity, ctr, send_method) {
1354 }
1355
1356 ErrnoLogMessage::~ErrnoLogMessage() {
1357   // Don't access errno directly because it may have been altered
1358   // while streaming the message.
1359   char buf[100];
1360   posix_strerror_r(preserved_errno(), buf, sizeof(buf));
1361   stream() << ": " << buf << " [" << preserved_errno() << "]";
1362 }
1363
1364 void FlushLogFiles(LogSeverity min_severity) {
1365   LogDestination::FlushLogFiles(min_severity);
1366 }
1367
1368 void FlushLogFilesUnsafe(LogSeverity min_severity) {
1369   LogDestination::FlushLogFilesUnsafe(min_severity);
1370 }
1371
1372 void SetLogDestination(LogSeverity severity, const char* base_filename) {
1373   LogDestination::SetLogDestination(severity, base_filename);
1374 }
1375
1376 void SetLogSymlink(LogSeverity severity, const char* symlink_basename) {
1377   LogDestination::SetLogSymlink(severity, symlink_basename);
1378 }
1379
1380 LogSink::~LogSink() {
1381 }
1382
1383 void LogSink::WaitTillSent() {
1384   // noop default
1385 }
1386
1387 string LogSink::ToString(LogSeverity severity, const char* file, int line,
1388                          const struct ::tm* tm_time,
1389                          const char* message, size_t message_len) {
1390   ostringstream stream(string(message, message_len));
1391   stream.fill('0');
1392
1393   // FIXME(jrvb): Updating this to use the correct value for usecs
1394   // requires changing the signature for both this method and
1395   // LogSink::send().  This change needs to be done in a separate CL
1396   // so subclasses of LogSink can be updated at the same time.
1397   int usecs = 0;
1398
1399   stream << LogSeverityNames[severity][0]
1400          << setw(2) << 1+tm_time->tm_mon
1401          << setw(2) << tm_time->tm_mday
1402          << ' '
1403          << setw(2) << tm_time->tm_hour << ':'
1404          << setw(2) << tm_time->tm_min << ':'
1405          << setw(2) << tm_time->tm_sec << '.'
1406          << setw(6) << usecs
1407          << ' '
1408          << setfill(' ') << setw(5) << GetTID() << setfill('0')
1409          << ' '
1410          << file << ':' << line << "] ";
1411
1412   stream << string(message, message_len);
1413   return stream.str();
1414 }
1415
1416 void AddLogSink(LogSink *destination) {
1417   LogDestination::AddLogSink(destination);
1418 }
1419
1420 void RemoveLogSink(LogSink *destination) {
1421   LogDestination::RemoveLogSink(destination);
1422 }
1423
1424 void SetLogFilenameExtension(const char* ext) {
1425   LogDestination::SetLogFilenameExtension(ext);
1426 }
1427
1428 void SetStderrLogging(LogSeverity min_severity) {
1429   LogDestination::SetStderrLogging(min_severity);
1430 }
1431
1432 void SetEmailLogging(LogSeverity min_severity, const char* addresses) {
1433   LogDestination::SetEmailLogging(min_severity, addresses);
1434 }
1435
1436 void LogToStderr() {
1437   LogDestination::LogToStderr();
1438 }
1439
1440 namespace base {
1441 namespace internal {
1442
1443 bool GetExitOnDFatal() {
1444   MutexLock l(&log_mutex);
1445   return exit_on_dfatal;
1446 }
1447
1448 // Determines whether we exit the program for a LOG(DFATAL) message in
1449 // debug mode.  It does this by skipping the call to Fail/FailQuietly.
1450 // This is intended for testing only.
1451 //
1452 // This can have some effects on LOG(FATAL) as well.  Failure messages
1453 // are always allocated (rather than sharing a buffer), the crash
1454 // reason is not recorded, the "gwq" status message is not updated,
1455 // and the stack trace is not recorded.  The LOG(FATAL) *will* still
1456 // exit the program.  Since this function is used only in testing,
1457 // these differences are acceptable.
1458 void SetExitOnDFatal(bool value) {
1459   MutexLock l(&log_mutex);
1460   exit_on_dfatal = value;
1461 }
1462
1463 }  // namespace internal
1464 }  // namespace base
1465
1466 // use_logging controls whether the logging functions LOG/VLOG are used
1467 // to log errors.  It should be set to false when the caller holds the
1468 // log_mutex.
1469 static bool SendEmailInternal(const char*dest, const char *subject,
1470                               const char*body, bool use_logging) {
1471   if (dest && *dest) {
1472     if ( use_logging ) {
1473       VLOG(1) << "Trying to send TITLE:" << subject
1474               << " BODY:" << body << " to " << dest;
1475     } else {
1476       fprintf(stderr, "Trying to send TITLE: %s BODY: %s to %s\n",
1477               subject, body, dest);
1478     }
1479
1480     string cmd =
1481         FLAGS_logmailer + " -s\"" + subject + "\" " + dest;
1482     FILE* pipe = popen(cmd.c_str(), "w");
1483     if (pipe != NULL) {
1484       // Add the body if we have one
1485       if (body)
1486         fwrite(body, sizeof(char), strlen(body), pipe);
1487       bool ok = pclose(pipe) != -1;
1488       if ( !ok ) {
1489         if ( use_logging ) {
1490           char buf[100];
1491           posix_strerror_r(errno, buf, sizeof(buf));
1492           LOG(ERROR) << "Problems sending mail to " << dest << ": " << buf;
1493         } else {
1494           char buf[100];
1495           posix_strerror_r(errno, buf, sizeof(buf));
1496           fprintf(stderr, "Problems sending mail to %s: %s\n", dest, buf);
1497         }
1498       }
1499       return ok;
1500     } else {
1501       if ( use_logging ) {
1502         LOG(ERROR) << "Unable to send mail to " << dest;
1503       } else {
1504         fprintf(stderr, "Unable to send mail to %s\n", dest);
1505       }
1506     }
1507   }
1508   return false;
1509 }
1510
1511 bool SendEmail(const char*dest, const char *subject, const char*body){
1512   return SendEmailInternal(dest, subject, body, true);
1513 }
1514
1515 static void GetTempDirectories(vector<string>* list) {
1516   list->clear();
1517 #ifdef OS_WINDOWS
1518   // On windows we'll try to find a directory in this order:
1519   //   C:/Documents & Settings/whomever/TEMP (or whatever GetTempPath() is)
1520   //   C:/TMP/
1521   //   C:/TEMP/
1522   //   C:/WINDOWS/ or C:/WINNT/
1523   //   .
1524   char tmp[MAX_PATH];
1525   if (GetTempPathA(MAX_PATH, tmp))
1526     list->push_back(tmp);
1527   list->push_back("C:\\tmp\\");
1528   list->push_back("C:\\temp\\");
1529 #else
1530   // Directories, in order of preference. If we find a dir that
1531   // exists, we stop adding other less-preferred dirs
1532   const char * candidates[] = {
1533     // Non-null only during unittest/regtest
1534     getenv("TEST_TMPDIR"),
1535
1536     // Explicitly-supplied temp dirs
1537     getenv("TMPDIR"), getenv("TMP"),
1538
1539     // If all else fails
1540     "/tmp",
1541   };
1542
1543   for (int i = 0; i < ARRAYSIZE(candidates); i++) {
1544     const char *d = candidates[i];
1545     if (!d) continue;  // Empty env var
1546
1547     // Make sure we don't surprise anyone who's expecting a '/'
1548     string dstr = d;
1549     if (dstr[dstr.size() - 1] != '/') {
1550       dstr += "/";
1551     }
1552     list->push_back(dstr);
1553
1554     struct stat statbuf;
1555     if (!stat(d, &statbuf) && S_ISDIR(statbuf.st_mode)) {
1556       // We found a dir that exists - we're done.
1557       return;
1558     }
1559   }
1560
1561 #endif
1562 }
1563
1564 static vector<string>* logging_directories_list;
1565
1566 const vector<string>& GetLoggingDirectories() {
1567   // Not strictly thread-safe but we're called early in InitGoogle().
1568   if (logging_directories_list == NULL) {
1569     logging_directories_list = new vector<string>;
1570
1571     if ( !FLAGS_log_dir.empty() ) {
1572       // A dir was specified, we should use it
1573       logging_directories_list->push_back(FLAGS_log_dir.c_str());
1574     } else {
1575       GetTempDirectories(logging_directories_list);
1576 #ifdef OS_WINDOWS
1577       char tmp[MAX_PATH];
1578       if (GetWindowsDirectoryA(tmp, MAX_PATH))
1579         logging_directories_list->push_back(tmp);
1580       logging_directories_list->push_back(".\\");
1581 #else
1582       logging_directories_list->push_back("./");
1583 #endif
1584     }
1585   }
1586   return *logging_directories_list;
1587 }
1588
1589 void TestOnly_ClearLoggingDirectoriesList() {
1590   fprintf(stderr, "TestOnly_ClearLoggingDirectoriesList should only be "
1591           "called from test code.\n");
1592   delete logging_directories_list;
1593   logging_directories_list = NULL;
1594 }
1595
1596 void GetExistingTempDirectories(vector<string>* list) {
1597   GetTempDirectories(list);
1598   vector<string>::iterator i_dir = list->begin();
1599   while( i_dir != list->end() ) {
1600     // zero arg to access means test for existence; no constant
1601     // defined on windows
1602     if ( access(i_dir->c_str(), 0) ) {
1603       i_dir = list->erase(i_dir);
1604     } else {
1605       ++i_dir;
1606     }
1607   }
1608 }
1609
1610 void TruncateLogFile(const char *path, int64 limit, int64 keep) {
1611 #ifdef HAVE_UNISTD_H
1612   struct stat statbuf;
1613   const int kCopyBlockSize = 8 << 10;
1614   char copybuf[kCopyBlockSize];
1615   int64 read_offset, write_offset;
1616   // Don't follow symlinks unless they're our own fd symlinks in /proc
1617   int flags = O_RDWR;
1618   const char *procfd_prefix = "/proc/self/fd/";
1619   if (strncmp(procfd_prefix, path, strlen(procfd_prefix))) flags |= O_NOFOLLOW;
1620
1621   int fd = open(path, flags);
1622   if (fd == -1) {
1623     if (errno == EFBIG) {
1624       // The log file in question has got too big for us to open. The
1625       // real fix for this would be to compile logging.cc (or probably
1626       // all of base/...) with -D_FILE_OFFSET_BITS=64 but that's
1627       // rather scary.
1628       // Instead just truncate the file to something we can manage
1629       if (truncate(path, 0) == -1) {
1630         PLOG(ERROR) << "Unable to truncate " << path;
1631       } else {
1632         LOG(ERROR) << "Truncated " << path << " due to EFBIG error";
1633       }
1634     } else {
1635       PLOG(ERROR) << "Unable to open " << path;
1636     }
1637     return;
1638   }
1639
1640   if (fstat(fd, &statbuf) == -1) {
1641     PLOG(ERROR) << "Unable to fstat()";
1642     goto out_close_fd;
1643   }
1644
1645   // See if the path refers to a regular file bigger than the
1646   // specified limit
1647   if (!S_ISREG(statbuf.st_mode)) goto out_close_fd;
1648   if (statbuf.st_size <= limit)  goto out_close_fd;
1649   if (statbuf.st_size <= keep) goto out_close_fd;
1650
1651   // This log file is too large - we need to truncate it
1652   LOG(INFO) << "Truncating " << path << " to " << keep << " bytes";
1653
1654   // Copy the last "keep" bytes of the file to the beginning of the file
1655   read_offset = statbuf.st_size - keep;
1656   write_offset = 0;
1657   int bytesin, bytesout;
1658   while ((bytesin = pread(fd, copybuf, sizeof(copybuf), read_offset)) > 0) {
1659     bytesout = pwrite(fd, copybuf, bytesin, write_offset);
1660     if (bytesout == -1) {
1661       PLOG(ERROR) << "Unable to write to " << path;
1662       break;
1663     } else if (bytesout != bytesin) {
1664       LOG(ERROR) << "Expected to write " << bytesin << ", wrote " << bytesout;
1665     }
1666     read_offset += bytesin;
1667     write_offset += bytesout;
1668   }
1669   if (bytesin == -1) PLOG(ERROR) << "Unable to read from " << path;
1670
1671   // Truncate the remainder of the file. If someone else writes to the
1672   // end of the file after our last read() above, we lose their latest
1673   // data. Too bad ...
1674   if (ftruncate(fd, write_offset) == -1) {
1675     PLOG(ERROR) << "Unable to truncate " << path;
1676   }
1677
1678  out_close_fd:
1679   close(fd);
1680 #else
1681   LOG(ERROR) << "No log truncation support.";
1682 #endif
1683 }
1684
1685 void TruncateStdoutStderr() {
1686 #ifdef HAVE_UNISTD_H
1687   int64 limit = MaxLogSize() << 20;
1688   int64 keep = 1 << 20;
1689   TruncateLogFile("/proc/self/fd/1", limit, keep);
1690   TruncateLogFile("/proc/self/fd/2", limit, keep);
1691 #else
1692   LOG(ERROR) << "No log truncation support.";
1693 #endif
1694 }
1695
1696
1697 // Helper functions for string comparisons.
1698 #define DEFINE_CHECK_STROP_IMPL(name, func, expected)                   \
1699   string* Check##func##expected##Impl(const char* s1, const char* s2,   \
1700                                       const char* names) {              \
1701     bool equal = s1 == s2 || (s1 && s2 && !func(s1, s2));               \
1702     if (equal == expected) return NULL;                                 \
1703     else {                                                              \
1704       strstream ss;                                                     \
1705       if (!s1) s1 = "";                                                 \
1706       if (!s2) s2 = "";                                                 \
1707       ss << #name " failed: " << names << " (" << s1 << " vs. " << s2 << ")"; \
1708       return new string(ss.str(), ss.pcount());                         \
1709     }                                                                   \
1710   }
1711 DEFINE_CHECK_STROP_IMPL(CHECK_STREQ, strcmp, true)
1712 DEFINE_CHECK_STROP_IMPL(CHECK_STRNE, strcmp, false)
1713 DEFINE_CHECK_STROP_IMPL(CHECK_STRCASEEQ, strcasecmp, true)
1714 DEFINE_CHECK_STROP_IMPL(CHECK_STRCASENE, strcasecmp, false)
1715 #undef DEFINE_CHECK_STROP_IMPL
1716
1717 int posix_strerror_r(int err, char *buf, size_t len) {
1718   // Sanity check input parameters
1719   if (buf == NULL || len <= 0) {
1720     errno = EINVAL;
1721     return -1;
1722   }
1723
1724   // Reset buf and errno, and try calling whatever version of strerror_r()
1725   // is implemented by glibc
1726   buf[0] = '\000';
1727   int old_errno = errno;
1728   errno = 0;
1729   char *rc = reinterpret_cast<char *>(strerror_r(err, buf, len));
1730
1731   // Both versions set errno on failure
1732   if (errno) {
1733     // Should already be there, but better safe than sorry
1734     buf[0]     = '\000';
1735     return -1;
1736   }
1737   errno = old_errno;
1738
1739   // POSIX is vague about whether the string will be terminated, although
1740   // is indirectly implies that typically ERANGE will be returned, instead
1741   // of truncating the string. This is different from the GNU implementation.
1742   // We play it safe by always terminating the string explicitly.
1743   buf[len-1] = '\000';
1744
1745   // If the function succeeded, we can use its exit code to determine the
1746   // semantics implemented by glibc
1747   if (!rc) {
1748     return 0;
1749   } else {
1750     // GNU semantics detected
1751     if (rc == buf) {
1752       return 0;
1753     } else {
1754       buf[0] = '\000';
1755 #if defined(OS_MACOSX) || defined(OS_FREEBSD)
1756       if (reinterpret_cast<int>(rc) < sys_nerr) {
1757         // This means an error on MacOSX or FreeBSD.
1758         return -1;
1759       }
1760 #endif
1761       strncat(buf, rc, len-1);
1762       return 0;
1763     }
1764   }
1765 }
1766
1767 LogMessageFatal::LogMessageFatal(const char* file, int line) :
1768     LogMessage(file, line, FATAL) {}
1769
1770 LogMessageFatal::LogMessageFatal(const char* file, int line,
1771                                  const CheckOpString& result) :
1772     LogMessage(file, line, result) {}
1773
1774 LogMessageFatal::~LogMessageFatal() {
1775     Flush();
1776     LogMessage::Fail();
1777 }
1778
1779 _END_GOOGLE_NAMESPACE_