* ---------------------------------------------------------------------------*/
#include "PosixSource.h"
-
#include "Rts.h"
#include "RtsAPI.h"
-#include "RtsFlags.h"
+
#include "RtsUtils.h"
#include "Ticky.h"
+#include "Schedule.h"
#ifdef HAVE_TIME_H
#include <time.h>
#endif
+/* HACK: On Mac OS X 10.4 (at least), time.h doesn't declare ctime_r with
+ * _POSIX_C_SOURCE. If this is the case, we declare it ourselves.
+ */
+#if HAVE_CTIME_R && !HAVE_DECL_CTIME_R
+extern char *ctime_r(const time_t *, char *);
+#endif
+
#ifdef HAVE_FCNTL_H
#include <fcntl.h>
#endif
void
shutdownAllocator(void)
{
+ Allocated *prev, *a;
+
+ if (allocs == NULL) {
+ barf("Allocator shutdown requested, but not initialised!");
+ }
+
#ifdef THREADED_RTS
closeMutex(&allocator_mutex);
#endif
+
+ prev = allocs;
+ while (1) {
+ a = prev->next;
+ free(prev);
+ if (a == NULL) return;
+ IF_DEBUG(sanity,
+ debugBelch("Warning: %ld bytes at %p still allocated at shutdown\n",
+ (long)a->len, a->addr);)
+ prev = a;
+ }
}
-static void allocate(void *addr, size_t len) {
+static void addAllocation(void *addr, size_t len) {
Allocated *a;
size_t alloc_size;
- alloc_size = sizeof(Allocated);
- if ((a = (Allocated *) malloc(alloc_size)) == NULL) {
- /* don't fflush(stdout); WORKAROUND bug in Linux glibc */
- MallocFailHook((W_) alloc_size, "creating info for debugging allocator");
- stg_exit(EXIT_INTERNAL_ERROR);
+ if (allocs != NULL) {
+ alloc_size = sizeof(Allocated);
+ if ((a = (Allocated *) malloc(alloc_size)) == NULL) {
+ /* don't fflush(stdout); WORKAROUND bug in Linux glibc */
+ MallocFailHook((W_) alloc_size,
+ "creating info for debugging allocator");
+ stg_exit(EXIT_INTERNAL_ERROR);
+ }
+ a->addr = addr;
+ a->len = len;
+ ACQUIRE_LOCK(&allocator_mutex);
+ a->next = allocs->next;
+ allocs->next = a;
+ RELEASE_LOCK(&allocator_mutex);
+ }
+ else {
+ /* This doesn't actually help as we haven't looked at the flags
+ * at the time that it matters (while running constructors) */
+ IF_DEBUG(sanity,
+ debugBelch("Ignoring allocation %p %zd as allocs is NULL\n",
+ addr, len);)
}
- a->addr = addr;
- a->len = len;
- ACQUIRE_LOCK(&allocator_mutex);
- a->next = allocs->next;
- allocs->next = a;
- RELEASE_LOCK(&allocator_mutex);
}
-static void deallocate(void *addr) {
+static void removeAllocation(void *addr, int overwrite_with_aa) {
Allocated *prev, *a;
if (addr == NULL) {
barf("Freeing NULL!");
}
- ACQUIRE_LOCK(&allocator_mutex);
- prev = allocs;
- a = prev->next;
- while (a != NULL) {
- if (a->addr == addr) {
- prev->next = a->next;
- memset(addr, 0xaa, a->len);
- free(a);
- RELEASE_LOCK(&allocator_mutex);
- return;
+ if (allocs != NULL) {
+ ACQUIRE_LOCK(&allocator_mutex);
+ prev = allocs;
+ a = prev->next;
+ while (a != NULL) {
+ if (a->addr == addr) {
+ prev->next = a->next;
+ if (overwrite_with_aa) {
+ memset(addr, 0xaa, a->len);
+ }
+ free(a);
+ RELEASE_LOCK(&allocator_mutex);
+ return;
+ }
+ prev = a;
+ a = a->next;
}
- prev = a;
- a = a->next;
+ /* We would like to barf here, but we can't as conc021
+ * allocates some stuff in a constructor which then gets freed
+ * during hs_exit */
+ /* barf("Freeing non-allocated memory at %p", addr); */
+ IF_DEBUG(sanity,
+ debugBelch("Warning: Freeing non-allocated memory at %p\n",
+ addr);)
+ RELEASE_LOCK(&allocator_mutex);
+ }
+ else {
+ IF_DEBUG(sanity,
+ debugBelch("Ignoring free of %p as allocs is NULL\n",
+ addr);)
}
- barf("Freeing non-allocated memory at %p", addr);
}
#endif
stg_exit(EXIT_INTERNAL_ERROR);
}
#if defined(DEBUG)
- allocate(space, n2);
+ addAllocation(space, n2);
#endif
return space;
}
stg_exit(EXIT_INTERNAL_ERROR);
}
#if defined(DEBUG)
- deallocate(p);
- allocate(space, n2);
+ removeAllocation(p, 0);
+ addAllocation(space, n2);
#endif
return space;
}
stg_exit(EXIT_INTERNAL_ERROR);
}
#if defined(DEBUG)
- allocate(space, (size_t) n * (size_t) m);
+ addAllocation(space, (size_t) n * (size_t) m);
#endif
return space;
}
stgFree(void* p)
{
#if defined(DEBUG)
- deallocate(p);
+ removeAllocation(p, 1);
#endif
free(p);
}
void
heapOverflow(void)
{
- /* 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();
-#endif
+ if (!heap_overflow)
+ {
+ /* don't fflush(stdout); WORKAROUND bug in Linux glibc */
+ OutOfHeapHook(0/*unknown request size*/,
+ RtsFlags.GcFlags.maxHeapSize * BLOCK_SIZE);
- stg_exit(EXIT_HEAPOVERFLOW);
-}
-
-/* -----------------------------------------------------------------------------
- Out-of-line strlen.
-
- Used in addr2Integer because the C compiler on x86 chokes on
- strlen, trying to inline it with not enough registers available.
- -------------------------------------------------------------------------- */
-
-nat stg_strlen(char *s)
-{
- char *p = s;
-
- while (*p) p++;
- return p-s;
+ heap_overflow = rtsTrue;
+ }
}
-
/* -----------------------------------------------------------------------------
genSym stuff, used by GHC itself for its splitting unique supply.
ToDo: put this somewhere sensible.
------------------------------------------------------------------------- */
-static I_ __GenSymCounter = 0;
+static HsInt __GenSymCounter = 0;
-I_
+HsInt
genSymZh(void)
{
return(__GenSymCounter++);
}
-I_
+HsInt
resetGenSymZh(void) /* it's your funeral */
{
__GenSymCounter=0;
Get the current time as a string. Used in profiling reports.
-------------------------------------------------------------------------- */
-#if defined(PROFILING) || defined(DEBUG) || defined(PAR) || defined(GRAN)
char *
time_str(void)
{
}
return nowstr;
}
-#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_HOST_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_HOST_ARCH) && !defined(hppa1_1_HOST_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_HOST_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.
* genericRaise(), rather than raise(3).
*/
int genericRaise(int sig) {
-#if defined(THREADED_RTS) && (defined(openbsd_HOST_OS) || defined(freebsd_HOST_OS))
+#if defined(THREADED_RTS) && (defined(openbsd_HOST_OS) || defined(freebsd_HOST_OS) || defined(dragonfly_HOST_OS))
return pthread_kill(pthread_self(), sig);
#else
return raise(sig);
#endif
}
+
+static void mkRtsInfoPair(char *key, char *val) {
+ /* XXX should check for "s, \s etc in key and val */
+ printf(" ,(\"%s\", \"%s\")\n", key, val);
+}
+
+/* This little bit of magic allows us to say TOSTRING(SYM) and get
+ * "5" if SYM is 5 */
+#define TOSTRING2(x) #x
+#define TOSTRING(x) TOSTRING2(x)
+
+void printRtsInfo(void) {
+ /* The first entry is just a hack to make it easy to get the
+ * commas right */
+ printf(" [(\"GHC RTS\", \"YES\")\n");
+ mkRtsInfoPair("GHC version", ProjectVersion);
+ mkRtsInfoPair("RTS way", RtsWay);
+ mkRtsInfoPair("Host platform", HostPlatform);
+ mkRtsInfoPair("Host architecture", HostArch);
+ mkRtsInfoPair("Host OS", HostOS);
+ mkRtsInfoPair("Host vendor", HostVendor);
+ mkRtsInfoPair("Build platform", BuildPlatform);
+ mkRtsInfoPair("Build architecture", BuildArch);
+ mkRtsInfoPair("Build OS", BuildOS);
+ mkRtsInfoPair("Build vendor", BuildVendor);
+ mkRtsInfoPair("Target platform", TargetPlatform);
+ mkRtsInfoPair("Target architecture", TargetArch);
+ mkRtsInfoPair("Target OS", TargetOS);
+ mkRtsInfoPair("Target vendor", TargetVendor);
+ mkRtsInfoPair("Word size", TOSTRING(WORD_SIZE_IN_BITS));
+ mkRtsInfoPair("Compiler unregisterised", GhcUnregisterised);
+ mkRtsInfoPair("Tables next to code", GhcEnableTablesNextToCode);
+ printf(" ]\n");
+}
+
+// Provides a way for Haskell programs to tell whether they're being
+// profiled or not. GHCi uses it (see #2197).
+int rts_isProfiled(void)
+{
+#ifdef PROFILING
+ return 1;
+#else
+ return 0;
+#endif
+}