server/common/logger.C

/*
 * This file is part of Deliantra, the Roguelike Realtime MMORPG.
 * 
 * Copyright (©) 2005,2006,2007,2008,2009,2010,2011 Marc Alexander Lehmann / Robin Redeker / the Deliantra team
 * 
 * Deliantra is free software: you can redistribute it and/or modify it under
 * the terms of the Affero GNU General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or (at your
 * option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the Affero GNU General Public License
 * and the GNU General Public License along with this program. If not, see
 * <http://www.gnu.org/licenses/>.
 * 
 * The authors can be reached via e-mail to <support@deliantra.net>
 */

#include <cstdarg>
#include <cstring>

#include <vector>

#include <sys/uio.h>
#include <pthread.h>

#include "global.h"
#include "dynbuf.h"
#include "util.h"

struct logline
{
  struct timeval tv;
  char *buf; // includes PREFIX_LEN garbage bytes
  int len;
  int flags;
};

typedef std::vector<logline, slice_allocator<logline> > logvector;

static SMUTEX(mutex);
static SMUTEX(fdlock);
static SCOND(cond);
static int logfd = STDERR_FILENO;
static int logsync = 1;
static logvector queue;

int log_setfd (int fd)
{
  SMUTEX_LOCK (fdlock);
  int old = logfd;
  logfd = fd < 0 ? STDERR_FILENO : fd;
  SMUTEX_UNLOCK (fdlock);

  return old;
}

#define PREFIX_LEN sizeof ("0000-00-00 00:00:00.0000 L+") - 1

static void
log_sync (logline &line)
{
  static const char levelchar [16+1] = "EWIDt???????????";
  struct tm lt;
  char pfx [PREFIX_LEN];

  localtime_r (&line.tv.tv_sec, &lt);

  sprintf (pfx, "%04d-%02d-%02d %02d:%02d:%02d.%04d %c",
    lt.tm_year + 1900,
    lt.tm_mon + 1,
    lt.tm_mday,
    lt.tm_hour,
    lt.tm_min,
    lt.tm_sec,
    (int)(line.tv.tv_usec / 100),
    levelchar [line.flags & 15]
  );

  pfx [PREFIX_LEN - 1] = line.flags & logSync ? '=' : ' ';

  struct iovec iov [2];

  iov [0].iov_base = pfx;
  iov [0].iov_len = PREFIX_LEN;

  char *buf = line.buf;

  if (logsync != 2)
    SMUTEX_LOCK (fdlock);

  while (char *end = strchr (buf, '\n'))
    {
      iov [1].iov_base = buf;
      iov [1].iov_len = end - buf + 1;

      writev (STDERR_FILENO, iov, 2);
      if (logfd != STDERR_FILENO)
        writev (logfd, iov, 2);

      buf = end + 1;

      if (buf == line.buf + line.len)
        break;

      pfx [PREFIX_LEN - 1] = '+';
    }

  if (logsync != 2)
    SMUTEX_UNLOCK (fdlock);

  sfree (line.buf, line.len);
}

static void *
logthread_proc (void *arg)
{
  int idx = 0;

  for (;;)
    {
      logline line;

      SMUTEX_LOCK (mutex);

      while (queue.empty ())
        SCOND_WAIT (cond, mutex);

      line = queue [idx++];

      // this algorithm could result in an ever-increasing vector
      // size if we log faster than we can write, but if that happens
      // we have bigger problems.
      if (idx == queue.size ())
        {
          if (idx < 32)
            queue.clear ();
          else
            logvector ().swap (queue); // free memory, hopefully

          idx = 0;
        }

      SMUTEX_UNLOCK (mutex);

      log_sync (line);
    }
}

void
log_cleanup ()
{
  logsync = 1;
  SMUTEX_UNLOCK (fdlock);

  for (;;)
    {
      int done;

      SMUTEX_LOCK (mutex);
      done = queue.empty ();
      SMUTEX_UNLOCK (mutex);

      if (done)
        break;

      usleep (10000);
      SMUTEX_LOCK (fdlock);
      SMUTEX_UNLOCK (fdlock);
    }
}

static void
af_child ()
{
  logsync = 2;
}

static struct logthread : thread
{
  logthread ()
  {
    pthread_atfork (0, 0, af_child);
    start (logthread_proc);
    logsync = 0;
  }
} logthread;

void
LOG (int flags, const_utf8_string format, ...)
{
  int level = flags & 15;

  if (level > settings.debug)
    return;

  logline line;
  gettimeofday (&line.tv, 0);

  static dynbuf_text buf;

  va_list ap;
  va_start (ap, format);
  buf.vprintf (format, ap);
  va_end (ap);

  buf << '\n';

  line.buf = buf.linearise ();
  line.len = buf.size ();

  if (line.buf [line.len - 2] == '\n')
    --line.len;

  line.buf = salloc<char> (line.len, line.buf);

  buf.clear ();

  if (logsync)
    flags |= logSync;

  if (flags & logBacktrace)
    {
      line.buf [line.len - 1] = 0;
      log_backtrace (line.buf);
      line.buf [line.len - 1] = '\n';
    }

  line.flags = flags;

  if (line.flags & logSync)
    log_sync (line);
  else
    {
      SMUTEX_LOCK (mutex);
      queue.push_back (line);
      SMUTEX_UNLOCK (mutex);
      SCOND_SIGNAL (cond);
    }
}

static int suspended;

void
log_suspend ()
{
  if (!suspended++)
    {
      LOG (llevDebug, "logging suspended.");
      SMUTEX_LOCK (fdlock);
    }
}

void
log_resume ()
{
  if (!--suspended)
    {
      SMUTEX_UNLOCK (fdlock);
      LOG (llevDebug, "logging resumed.");
    }
}

Revision:	1.27
Committed:	Sat Apr 23 04:56:46 2011 UTC (13 years, 1 month ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.26:	+1 -1 lines
Log Message:	update copyright to 2011
#	Content
1	/*
2	* This file is part of Deliantra, the Roguelike Realtime MMORPG.
3	*
4	* Copyright (©) 2005,2006,2007,2008,2009,2010,2011 Marc Alexander Lehmann / Robin Redeker / the Deliantra team
5	*
6	* Deliantra is free software: you can redistribute it and/or modify it under
7	* the terms of the Affero GNU General Public License as published by the
8	* Free Software Foundation, either version 3 of the License, or (at your
9	* option) any later version.
10	*
11	* This program is distributed in the hope that it will be useful,
12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14	* GNU General Public License for more details.
15	*
16	* You should have received a copy of the Affero GNU General Public License
17	* and the GNU General Public License along with this program. If not, see
18	* <http://www.gnu.org/licenses/>.
19	*
20	* The authors can be reached via e-mail to <support@deliantra.net>
21	*/
22
23	#include <cstdarg>
24	#include <cstring>
25
26	#include <vector>
27
28	#include <sys/uio.h>
29	#include <pthread.h>
30
31	#include "global.h"
32	#include "dynbuf.h"
33	#include "util.h"
34
35	struct logline
36	{
37	struct timeval tv;
38	char *buf; // includes PREFIX_LEN garbage bytes
39	int len;
40	int flags;
41	};
42
43	typedef std::vector<logline, slice_allocator<logline> > logvector;
44
45	static SMUTEX(mutex);
46	static SMUTEX(fdlock);
47	static SCOND(cond);
48	static int logfd = STDERR_FILENO;
49	static int logsync = 1;
50	static logvector queue;
51
52	int log_setfd (int fd)
53	{
54	SMUTEX_LOCK (fdlock);
55	int old = logfd;
56	logfd = fd < 0 ? STDERR_FILENO : fd;
57	SMUTEX_UNLOCK (fdlock);
58
59	return old;
60	}
61
62	#define PREFIX_LEN sizeof ("0000-00-00 00:00:00.0000 L+") - 1
63
64	static void
65	log_sync (logline &line)
66	{
67	static const char levelchar [16+1] = "EWIDt???????????";
68	struct tm lt;
69	char pfx [PREFIX_LEN];
70
71	localtime_r (&line.tv.tv_sec, &lt);
72
73	sprintf (pfx, "%04d-%02d-%02d %02d:%02d:%02d.%04d %c",
74	lt.tm_year + 1900,
75	lt.tm_mon + 1,
76	lt.tm_mday,
77	lt.tm_hour,
78	lt.tm_min,
79	lt.tm_sec,
80	(int)(line.tv.tv_usec / 100),
81	levelchar [line.flags & 15]
82	);
83
84	pfx [PREFIX_LEN - 1] = line.flags & logSync ? '=' : ' ';
85
86	struct iovec iov [2];
87
88	iov [0].iov_base = pfx;
89	iov [0].iov_len = PREFIX_LEN;
90
91	char *buf = line.buf;
92
93	if (logsync != 2)
94	SMUTEX_LOCK (fdlock);
95
96	while (char *end = strchr (buf, '\n'))
97	{
98	iov [1].iov_base = buf;
99	iov [1].iov_len = end - buf + 1;
100
101	writev (STDERR_FILENO, iov, 2);
102	if (logfd != STDERR_FILENO)
103	writev (logfd, iov, 2);
104
105	buf = end + 1;
106
107	if (buf == line.buf + line.len)
108	break;
109
110	pfx [PREFIX_LEN - 1] = '+';
111	}
112
113	if (logsync != 2)
114	SMUTEX_UNLOCK (fdlock);
115
116	sfree (line.buf, line.len);
117	}
118
119	static void *
120	logthread_proc (void *arg)
121	{
122	int idx = 0;
123
124	for (;;)
125	{
126	logline line;
127
128	SMUTEX_LOCK (mutex);
129
130	while (queue.empty ())
131	SCOND_WAIT (cond, mutex);
132
133	line = queue [idx++];
134
135	// this algorithm could result in an ever-increasing vector
136	// size if we log faster than we can write, but if that happens
137	// we have bigger problems.
138	if (idx == queue.size ())
139	{
140	if (idx < 32)
141	queue.clear ();
142	else
143	logvector ().swap (queue); // free memory, hopefully
144
145	idx = 0;
146	}
147
148	SMUTEX_UNLOCK (mutex);
149
150	log_sync (line);
151	}
152	}
153
154	void
155	log_cleanup ()
156	{
157	logsync = 1;
158	SMUTEX_UNLOCK (fdlock);
159
160	for (;;)
161	{
162	int done;
163
164	SMUTEX_LOCK (mutex);
165	done = queue.empty ();
166	SMUTEX_UNLOCK (mutex);
167
168	if (done)
169	break;
170
171	usleep (10000);
172	SMUTEX_LOCK (fdlock);
173	SMUTEX_UNLOCK (fdlock);
174	}
175	}
176
177	static void
178	af_child ()
179	{
180	logsync = 2;
181	}
182
183	static struct logthread : thread
184	{
185	logthread ()
186	{
187	pthread_atfork (0, 0, af_child);
188	start (logthread_proc);
189	logsync = 0;
190	}
191	} logthread;
192
193	void
194	LOG (int flags, const_utf8_string format, ...)
195	{
196	int level = flags & 15;
197
198	if (level > settings.debug)
199	return;
200
201	logline line;
202	gettimeofday (&line.tv, 0);
203
204	static dynbuf_text buf;
205
206	va_list ap;
207	va_start (ap, format);
208	buf.vprintf (format, ap);
209	va_end (ap);
210
211	buf << '\n';
212
213	line.buf = buf.linearise ();
214	line.len = buf.size ();
215
216	if (line.buf [line.len - 2] == '\n')
217	--line.len;
218
219	line.buf = salloc<char> (line.len, line.buf);
220
221	buf.clear ();
222
223	if (logsync)
224	flags \|= logSync;
225
226	if (flags & logBacktrace)
227	{
228	line.buf [line.len - 1] = 0;
229	log_backtrace (line.buf);
230	line.buf [line.len - 1] = '\n';
231	}
232
233	line.flags = flags;
234
235	if (line.flags & logSync)
236	log_sync (line);
237	else
238	{
239	SMUTEX_LOCK (mutex);
240	queue.push_back (line);
241	SMUTEX_UNLOCK (mutex);
242	SCOND_SIGNAL (cond);
243	}
244	}
245
246	static int suspended;
247
248	void
249	log_suspend ()
250	{
251	if (!suspended++)
252	{
253	LOG (llevDebug, "logging suspended.");
254	SMUTEX_LOCK (fdlock);
255	}
256	}
257
258	void
259	log_resume ()
260	{
261	if (!--suspended)
262	{
263	SMUTEX_UNLOCK (fdlock);
264	LOG (llevDebug, "logging resumed.");
265	}
266	}
267