2017-12-07 Vladimir Makarov <vmakarov@redhat.com>
[official-gcc.git] / libvtv / vtv_utils.cc
blob4fe7581f5aaaa3e4822ee6c9949946a24ee14c6b
1 /* Copyright (C) 2012-2017 Free Software Foundation, Inc.
3 This file is part of GCC.
5 GCC is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 3, or (at your option)
8 any later version.
10 GCC is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 Under Section 7 of GPL version 3, you are granted additional
16 permissions described in the GCC Runtime Library Exception, version
17 3.1, as published by the Free Software Foundation.
19 You should have received a copy of the GNU General Public License and
20 a copy of the GCC Runtime Library Exception along with this program;
21 see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
22 <http://www.gnu.org/licenses/>. */
24 /* This file is part of the vtable verication runtime library (see
25 comments in vtv_rts.cc for more information about vtable
26 verification). This file contains log file utilities. */
28 #include <sys/types.h>
29 #include <sys/stat.h>
30 #include <fcntl.h>
31 #include <stdarg.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #if defined (__CYGWIN__) || defined (__MINGW32__)
36 #include <windows.h>
37 #else
38 #include <execinfo.h>
39 #endif
41 #include <unistd.h>
42 #include <errno.h>
44 #include "vtv_utils.h"
46 #ifndef HAVE_SECURE_GETENV
47 # ifdef HAVE___SECURE_GETENV
48 # define secure_getenv __secure_getenv
49 # else
50 # define secure_getenv getenv
51 # endif
52 #endif
54 static int vtv_failures_log_fd = -1;
56 /* This function takes the NAME of a log file to open, attempts to
57 open it in the logs_dir directory, and returns the resulting file
58 descriptor.
60 This function first checks to see if the user has specifed (via
61 the environment variable VTV_LOGS_DIR) a directory to use for the
62 vtable verification logs. If that fails, the function will open
63 the logs in the current directory.
66 int
67 __vtv_open_log (const char *name)
69 char log_name[1024];
70 char log_dir[512];
71 #if defined (__CYGWIN__) || defined (__MINGW32__)
72 pid_t process_id = GetCurrentProcessId ();
73 #else
74 uid_t user_id = getuid ();
75 pid_t process_id = getpid ();
76 #endif
77 char *logs_prefix;
78 bool logs_dir_specified = false;
79 int fd = -1;
81 logs_prefix = secure_getenv ("VTV_LOGS_DIR");
82 if (logs_prefix && strlen (logs_prefix) > 0)
84 logs_dir_specified = true;
85 #ifdef __MINGW32__
86 mkdir (logs_prefix);
87 #else
88 mkdir (logs_prefix, S_IRWXU);
89 #endif
91 snprintf (log_dir, sizeof (log_dir), "%s/vtv_logs", logs_prefix);
93 #ifdef __MINGW32__
94 mkdir (log_dir);
95 #else
96 mkdir (log_dir, S_IRWXU);
97 #endif
98 #if defined (__CYGWIN__) || defined (__MINGW32__)
99 snprintf (log_name, sizeof (log_name), "%s_%d_%s", log_dir,
100 (unsigned) process_id, name);
101 fd = open (log_name, O_WRONLY | O_APPEND | O_CREAT, S_IRWXU);
102 #else
103 snprintf (log_name, sizeof (log_name), "%s/%d_%d_%s", log_dir,
104 (unsigned) user_id, (unsigned) process_id, name);
105 fd = open (log_name, O_WRONLY | O_APPEND | O_CREAT | O_NOFOLLOW,
106 S_IRWXU);
107 #endif
109 else
110 fd = dup (2);
112 if (fd == -1)
113 __vtv_add_to_log (2, "Cannot open log file %s %s\n", name,
114 strerror (errno));
115 return fd;
118 /* This function takes a file descriptor (FD) and a string (STR) and
119 tries to write the string to the file. */
121 static int
122 vtv_log_write (int fd, const char *str)
124 if (write (fd, str, strlen (str)) != -1)
125 return 0;
127 if (fd != 2) /* Make sure we dont get in a loop. */
128 __vtv_add_to_log (2, "Error writing to log: %s\n", strerror (errno));
129 return -1;
133 /* This function takes a file decriptor (LOG_FILE) and an output
134 format string (FORMAT), followed by zero or more print format
135 arguments (the same as fprintf, for example). It gets the current
136 process ID and PPID, pre-pends them to the formatted message, and
137 writes write it out to the log file referenced by LOG_FILE via calles
138 to vtv_log_write. */
141 __vtv_add_to_log (int log_file, const char * format, ...)
143 /* We dont want to dynamically allocate this buffer. This should be
144 more than enough in most cases. It if isn't we are careful not to
145 do a buffer overflow. */
146 char output[1024];
148 va_list ap;
149 va_start (ap, format);
151 #if defined (__CYGWIN__) || defined (__MINGW32__)
152 snprintf (output, sizeof (output), "VTV: PID=%ld ", GetCurrentProcessId ());
153 #else
154 snprintf (output, sizeof (output), "VTV: PID=%d PPID=%d ", getpid (),
155 getppid ());
156 #endif
157 vtv_log_write (log_file, output);
158 vsnprintf (output, sizeof (output), format, ap);
159 vtv_log_write (log_file, output);
160 va_end (ap);
162 return 0;
165 /* Open error logging file, if not already open, and write vtable
166 verification failure messages (LOG_MSG) to the log file. Also
167 generate a backtrace in the log file, if GENERATE_BACKTRACE is
168 set. */
170 void
171 __vtv_log_verification_failure (const char *log_msg, bool generate_backtrace)
173 if (vtv_failures_log_fd == -1)
174 vtv_failures_log_fd = __vtv_open_log ("vtable_verification_failures.log");
176 if (vtv_failures_log_fd == -1)
177 return;
179 __vtv_add_to_log (vtv_failures_log_fd, "%s", log_msg);
181 #if !defined (__CYGWIN__) && !defined (__MINGW32__)
182 if (generate_backtrace)
184 #define STACK_DEPTH 20
185 void *callers[STACK_DEPTH];
186 int actual_depth = backtrace (callers, STACK_DEPTH);
187 backtrace_symbols_fd (callers, actual_depth, vtv_failures_log_fd);
189 #endif