/* -----------------------------------------------------------------------------
- * $Id: RtsUtils.c,v 1.4 1999/01/27 14:51:21 simonpj Exp $
+ *
+ * (c) The GHC Team, 1998-2004
*
* General utility functions used in the RTS.
*
* ---------------------------------------------------------------------------*/
+/* gettimeofday isn't POSIX */
+/* #include "PosixSource.h" */
+
#include "Rts.h"
#include "RtsAPI.h"
#include "RtsFlags.h"
-#include "Hooks.h"
-#include "Main.h"
#include "RtsUtils.h"
#include "Ticky.h"
#include <time.h>
#endif
-#include <stdarg.h>
+#ifdef HAVE_FCNTL_H
+#include <fcntl.h>
+#endif
-/* variable-argument error function. */
+#ifdef HAVE_GETTIMEOFDAY
+#include <sys/time.h>
+#endif
-void barf(char *s, ...)
-{
- va_list ap;
- va_start(ap,s);
- fflush(stdout);
- if (prog_argv != NULL && prog_argv[0] != NULL) {
- fprintf(stderr, "%s: fatal error: ", prog_argv[0]);
- } else {
- fprintf(stderr, "fatal error: ");
- }
- vfprintf(stderr, s, ap);
- fprintf(stderr, "\n");
- stg_exit(EXIT_FAILURE);
-}
+#include <stdlib.h>
+#include <string.h>
+#include <stdarg.h>
+#include <stdio.h>
-void belch(char *s, ...)
-{
- va_list ap;
- va_start(ap,s);
- fflush(stdout);
- vfprintf(stderr, s, ap);
- fprintf(stderr, "\n");
-}
+#ifdef HAVE_SIGNAL_H
+#include <signal.h>
+#endif
+
+#if defined(THREADED_RTS) && defined(openbsd_TARGET_OS) && defined(HAVE_PTHREAD_H)
+#include <pthread.h>
+#endif
-/* result-checking malloc wrappers. */
+/* -----------------------------------------------------------------------------
+ Result-checking malloc wrappers.
+ -------------------------------------------------------------------------- */
void *
stgMallocBytes (int n, char *msg)
char *space;
if ((space = (char *) malloc((size_t) n)) == NULL) {
- fflush(stdout);
- MallocFailHook((W_) n, msg); /*msg*/
- stg_exit(EXIT_FAILURE);
+ /* don't fflush(stdout); WORKAROUND bug in Linux glibc */
+ MallocFailHook((W_) n, msg); /*msg*/
+ stg_exit(EXIT_INTERNAL_ERROR);
}
return space;
}
char *space;
if ((space = (char *) realloc(p, (size_t) n)) == NULL) {
- fflush(stdout);
- MallocFailHook((W_) n, msg); /*msg*/
- exit(EXIT_FAILURE);
+ /* don't fflush(stdout); WORKAROUND bug in Linux glibc */
+ MallocFailHook((W_) n, msg); /*msg*/
+ stg_exit(EXIT_INTERNAL_ERROR);
}
return space;
}
void *
-stgMallocWords (int n, char *msg)
+stgCallocBytes (int n, int m, char *msg)
{
- return(stgMallocBytes(n * sizeof(W_), msg));
+ int i;
+ int sz = n * m;
+ char* p = stgMallocBytes(sz, msg);
+ for (i = 0; i < sz; i++) p[i] = 0;
+ return p;
}
-void *
-stgReallocWords (void *p, int n, char *msg)
+/* To simplify changing the underlying allocator used
+ * by stgMallocBytes(), provide stgFree() as well.
+ */
+void
+stgFree(void* p)
{
- return(stgReallocBytes(p, n * sizeof(W_), msg));
+ free(p);
}
void
-_stgAssert (char *filename, nat linenum)
+_stgAssert (char *filename, unsigned int linenum)
{
fflush(stdout);
fprintf(stderr, "ASSERTION FAILED: file %s, line %u\n", filename, linenum);
+ fflush(stderr);
abort();
}
-StgStablePtr errorHandler = -1; /* -1 indicates no handler installed */
-
-void
-raiseError( StgStablePtr handler STG_UNUSED )
-{
- shutdownHaskell();
- stg_exit(EXIT_FAILURE);
-}
-
/* -----------------------------------------------------------------------------
Stack overflow
-------------------------------------------------------------------------- */
void
-stackOverflow(nat max_stack_size)
+stackOverflow(void)
{
- fflush(stdout);
- StackOverflowHook(max_stack_size * sizeof(W_)); /*msg*/
+ StackOverflowHook(RtsFlags.GcFlags.maxStkSize * sizeof(W_));
#if defined(TICKY_TICKY)
- if (RtsFlags.TickyFlags.showTickyStats) PrintTickyInfo();
+ if (RtsFlags.TickyFlags.showTickyStats) PrintTickyInfo();
#endif
-
- stg_exit(EXIT_FAILURE);
}
void
heapOverflow(void)
{
- fflush(stdout);
- OutOfHeapHook(0/*unknown request size*/,
- RtsFlags.GcFlags.maxHeapSize * BLOCK_SIZE);
-
+ /* don't fflush(stdout); WORKAROUND bug in Linux glibc */
+ OutOfHeapHook(0/*unknown request size*/,
+ RtsFlags.GcFlags.maxHeapSize * BLOCK_SIZE);
+
#if defined(TICKY_TICKY)
- if (RtsFlags.TickyFlags.showTickyStats) PrintTickyInfo();
+ if (RtsFlags.TickyFlags.showTickyStats) PrintTickyInfo();
#endif
- stg_exit(EXIT_FAILURE);
+ stg_exit(EXIT_HEAPOVERFLOW);
}
/* -----------------------------------------------------------------------------
ToDo: put this somewhere sensible.
------------------------------------------------------------------------- */
-I_ __GenSymCounter = 0;
+static I_ __GenSymCounter = 0;
I_
-genSymzh(void)
+genSymZh(void)
{
return(__GenSymCounter++);
}
I_
-resetGenSymzh(void) /* it's your funeral */
+resetGenSymZh(void) /* it's your funeral */
{
__GenSymCounter=0;
return(__GenSymCounter);
Get the current time as a string. Used in profiling reports.
-------------------------------------------------------------------------- */
-#if defined(PROFILING) || defined(DEBUG)
+#if defined(PROFILING) || defined(DEBUG) || defined(PAR) || defined(GRAN)
char *
time_str(void)
{
#endif
/* -----------------------------------------------------------------------------
+ * Reset a file handle to blocking mode. We do this for the standard
+ * file descriptors before exiting, because the shell doesn't always
+ * clean up for us.
+ * -------------------------------------------------------------------------- */
+
+#if !defined(mingw32_TARGET_OS)
+void
+resetNonBlockingFd(int fd)
+{
+ long fd_flags;
+
+ /* clear the non-blocking flag on this file descriptor */
+ fd_flags = fcntl(fd, F_GETFL);
+ if (fd_flags & O_NONBLOCK) {
+ fcntl(fd, F_SETFL, fd_flags & ~O_NONBLOCK);
+ }
+}
+
+void
+setNonBlockingFd(int fd)
+{
+ long fd_flags;
+
+ /* clear the non-blocking flag on this file descriptor */
+ fd_flags = fcntl(fd, F_GETFL);
+ if (!(fd_flags & O_NONBLOCK)) {
+ fcntl(fd, F_SETFL, fd_flags | O_NONBLOCK);
+ }
+}
+#else
+/* Stub defns -- async / non-blocking IO is not done
+ * via O_NONBLOCK and select() under Win32.
+ */
+void resetNonBlockingFd(int fd STG_UNUSED) {}
+void setNonBlockingFd(int fd STG_UNUSED) {}
+#endif
+
+#ifdef PAR
+static ullong startTime = 0;
+
+/* used in a parallel setup */
+ullong
+msTime(void)
+{
+# if defined(HAVE_GETCLOCK) && !defined(alpha_TARGET_ARCH) && !defined(hppa1_1_TARGET_ARCH)
+ struct timespec tv;
+
+ if (getclock(TIMEOFDAY, &tv) != 0) {
+ fflush(stdout);
+ fprintf(stderr, "Clock failed\n");
+ stg_exit(EXIT_FAILURE);
+ }
+ return tv.tv_sec * LL(1000) + tv.tv_nsec / LL(1000000) - startTime;
+# elif HAVE_GETTIMEOFDAY && !defined(alpha_TARGET_ARCH)
+ struct timeval tv;
+
+ if (gettimeofday(&tv, NULL) != 0) {
+ fflush(stdout);
+ fprintf(stderr, "Clock failed\n");
+ stg_exit(EXIT_FAILURE);
+ }
+ return tv.tv_sec * LL(1000) + tv.tv_usec / LL(1000) - startTime;
+# else
+ time_t t;
+ if ((t = time(NULL)) == (time_t) -1) {
+ fflush(stdout);
+ fprintf(stderr, "Clock failed\n");
+ stg_exit(EXIT_FAILURE);
+ }
+ return t * LL(1000) - startTime;
+# endif
+}
+#endif /* PAR */
+
+/* -----------------------------------------------------------------------------
Print large numbers, with punctuation.
-------------------------------------------------------------------------- */
ullong_format_string(ullong x, char *s, rtsBool with_commas)
{
if (x < (ullong)1000)
- sprintf(s, "%d", (nat)x);
+ sprintf(s, "%lu", (lnat)x);
else if (x < (ullong)1000000)
- sprintf(s, (with_commas) ? "%ld,%3.3ld" : "%ld%3.3ld",
- (nat)((x)/(ullong)1000),
- (nat)((x)%(ullong)1000));
+ sprintf(s, (with_commas) ? "%lu,%3.3lu" : "%lu%3.3lu",
+ (lnat)((x)/(ullong)1000),
+ (lnat)((x)%(ullong)1000));
else if (x < (ullong)1000000000)
- sprintf(s, (with_commas) ? "%ld,%3.3ld,%3.3ld" : "%ld%3.3ld%3.3ld",
- (nat)((x)/(ullong)1000000),
- (nat)((x)/(ullong)1000%(ullong)1000),
- (nat)((x)%(ullong)1000));
+ sprintf(s, (with_commas) ? "%lu,%3.3lu,%3.3lu" : "%lu%3.3lu%3.3lu",
+ (lnat)((x)/(ullong)1000000),
+ (lnat)((x)/(ullong)1000%(ullong)1000),
+ (lnat)((x)%(ullong)1000));
else
- sprintf(s, (with_commas) ? "%ld,%3.3ld,%3.3ld,%3.3ld" : "%ld%3.3ld%3.3ld%3.3ld",
- (nat)((x)/(ullong)1000000000),
- (nat)((x)/(ullong)1000000%(ullong)1000),
- (nat)((x)/(ullong)1000%(ullong)1000),
- (nat)((x)%(ullong)1000));
+ sprintf(s, (with_commas) ? "%lu,%3.3lu,%3.3lu,%3.3lu" : "%lu%3.3lu%3.3lu%3.3lu",
+ (lnat)((x)/(ullong)1000000000),
+ (lnat)((x)/(ullong)1000000%(ullong)1000),
+ (lnat)((x)/(ullong)1000%(ullong)1000),
+ (lnat)((x)%(ullong)1000));
return s;
}
+
+
+// Can be used as a breakpoint to set on every heap check failure.
+#ifdef DEBUG
+void
+heapCheckFail( void )
+{
+}
+#endif
+
+/*
+ * It seems that pthreads and signals interact oddly in OpenBSD & FreeBSD
+ * pthreads (and possibly others). When linking with -lpthreads, we
+ * have to use pthread_kill to send blockable signals. So use that
+ * when we have a threaded rts. So System.Posix.Signals will call
+ * genericRaise(), rather than raise(3).
+ */
+int genericRaise(int sig) {
+#if defined(THREADED_RTS) && (defined(openbsd_TARGET_OS) || defined(freebsd_TARGET_OS))
+ return pthread_kill(pthread_self(), sig);
+#else
+ return raise(sig);
+#endif
+}