OSDN Git Service

2003-03-03 Hans Boehm <Hans.Boehm@hp.com>
[pf3gnuchains/gcc-fork.git] / boehm-gc / dyn_load.c
1 /*
2  * Copyright (c) 1991-1994 by Xerox Corporation.  All rights reserved.
3  * Copyright (c) 1997 by Silicon Graphics.  All rights reserved.
4  *
5  * THIS MATERIAL IS PROVIDED AS IS, WITH ABSOLUTELY NO WARRANTY EXPRESSED
6  * OR IMPLIED.  ANY USE IS AT YOUR OWN RISK.
7  *
8  * Permission is hereby granted to use or copy this program
9  * for any purpose,  provided the above notices are retained on all copies.
10  * Permission to modify the code and to distribute modified code is granted,
11  * provided the above notices are retained, and a notice that the code was
12  * modified is included with the above copyright notice.
13  *
14  * Original author: Bill Janssen
15  * Heavily modified by Hans Boehm and others
16  */
17
18 /*
19  * This is incredibly OS specific code for tracking down data sections in
20  * dynamic libraries.  There appears to be no way of doing this quickly
21  * without groveling through undocumented data structures.  We would argue
22  * that this is a bug in the design of the dlopen interface.  THIS CODE
23  * MAY BREAK IN FUTURE OS RELEASES.  If this matters to you, don't hesitate
24  * to let your vendor know ...
25  *
26  * None of this is safe with dlclose and incremental collection.
27  * But then not much of anything is safe in the presence of dlclose.
28  */
29 #if defined(__linux__) && !defined(_GNU_SOURCE)
30     /* Can't test LINUX, since this must be define before other includes */
31 #   define _GNU_SOURCE
32 #endif
33 #if !defined(MACOS) && !defined(_WIN32_WCE)
34 #  include <sys/types.h>
35 #endif
36 #include "private/gc_priv.h"
37
38 /* BTL: avoid circular redefinition of dlopen if GC_SOLARIS_THREADS defined */
39 # if (defined(GC_PTHREADS) || defined(GC_SOLARIS_THREADS)) \
40       && defined(dlopen) && !defined(GC_USE_LD_WRAP)
41     /* To support threads in Solaris, gc.h interposes on dlopen by       */
42     /* defining "dlopen" to be "GC_dlopen", which is implemented below.  */
43     /* However, both GC_FirstDLOpenedLinkMap() and GC_dlopen() use the   */
44     /* real system dlopen() in their implementation. We first remove     */
45     /* gc.h's dlopen definition and restore it later, after GC_dlopen(). */
46 #   undef dlopen
47 #   define GC_must_restore_redefined_dlopen
48 # else
49 #   undef GC_must_restore_redefined_dlopen
50 # endif
51
52 #if (defined(DYNAMIC_LOADING) || defined(MSWIN32) || defined(MSWINCE)) \
53     && !defined(PCR)
54 #if !defined(SUNOS4) && !defined(SUNOS5DL) && !defined(IRIX5) && \
55     !defined(MSWIN32) && !defined(MSWINCE) && \
56     !(defined(ALPHA) && defined(OSF1)) && \
57     !defined(HPUX) && !(defined(LINUX) && defined(__ELF__)) && \
58     !defined(RS6000) && !defined(SCO_ELF) && \
59     !(defined(FREEBSD) && defined(__ELF__)) && \
60     !(defined(NETBSD) && defined(__ELF__)) && !defined(HURD)
61  --> We only know how to find data segments of dynamic libraries for the
62  --> above.  Additional SVR4 variants might not be too
63  --> hard to add.
64 #endif
65
66 #include <stdio.h>
67 #ifdef SUNOS5DL
68 #   include <sys/elf.h>
69 #   include <dlfcn.h>
70 #   include <link.h>
71 #endif
72 #ifdef SUNOS4
73 #   include <dlfcn.h>
74 #   include <link.h>
75 #   include <a.out.h>
76   /* struct link_map field overrides */
77 #   define l_next       lm_next
78 #   define l_addr       lm_addr
79 #   define l_name       lm_name
80 #endif
81
82 #if defined(LINUX) && defined(__ELF__) || defined(SCO_ELF) || \
83     (defined(FREEBSD) && defined(__ELF__)) || \
84     (defined(NETBSD) && defined(__ELF__)) || defined(HURD)
85 #   include <stddef.h>
86 #   include <elf.h>
87 #   include <link.h>
88 #endif
89
90 /* Newer versions of GNU/Linux define this macro.  We
91  * define it similarly for any ELF systems that don't.  */
92 #  ifndef ElfW
93 #    if !defined(ELF_CLASS) || ELF_CLASS == ELFCLASS32
94 #      define ElfW(type) Elf32_##type
95 #    else
96 #      define ElfW(type) Elf64_##type
97 #    endif
98 #  endif
99
100 #if defined(SUNOS5DL) && !defined(USE_PROC_FOR_LIBRARIES)
101
102 #ifdef LINT
103     Elf32_Dyn _DYNAMIC;
104 #endif
105
106 static struct link_map *
107 GC_FirstDLOpenedLinkMap()
108 {
109     extern ElfW(Dyn) _DYNAMIC;
110     ElfW(Dyn) *dp;
111     struct r_debug *r;
112     static struct link_map * cachedResult = 0;
113     static ElfW(Dyn) *dynStructureAddr = 0;
114                         /* BTL: added to avoid Solaris 5.3 ld.so _DYNAMIC bug */
115
116 #   ifdef SUNOS53_SHARED_LIB
117         /* BTL: Avoid the Solaris 5.3 bug that _DYNAMIC isn't being set */
118         /* up properly in dynamically linked .so's. This means we have  */
119         /* to use its value in the set of original object files loaded  */
120         /* at program startup.                                          */
121         if( dynStructureAddr == 0 ) {
122           void* startupSyms = dlopen(0, RTLD_LAZY);
123           dynStructureAddr = (ElfW(Dyn)*)dlsym(startupSyms, "_DYNAMIC");
124                 }
125 #   else
126         dynStructureAddr = &_DYNAMIC;
127 #   endif
128
129     if( dynStructureAddr == 0) {
130         return(0);
131     }
132     if( cachedResult == 0 ) {
133         int tag;
134         for( dp = ((ElfW(Dyn) *)(&_DYNAMIC)); (tag = dp->d_tag) != 0; dp++ ) {
135             if( tag == DT_DEBUG ) {
136                 struct link_map *lm
137                         = ((struct r_debug *)(dp->d_un.d_ptr))->r_map;
138                 if( lm != 0 ) cachedResult = lm->l_next; /* might be NIL */
139                 break;
140             }
141         }
142     }
143     return cachedResult;
144 }
145
146 #endif /* SUNOS5DL ... */
147
148 /* BTL: added to fix circular dlopen definition if GC_SOLARIS_THREADS defined */
149 # if defined(GC_must_restore_redefined_dlopen)
150 #   define dlopen GC_dlopen
151 # endif
152
153 #if defined(SUNOS4) && !defined(USE_PROC_FOR_LIBRARIES)
154
155 #ifdef LINT
156     struct link_dynamic _DYNAMIC;
157 #endif
158
159 static struct link_map *
160 GC_FirstDLOpenedLinkMap()
161 {
162     extern struct link_dynamic _DYNAMIC;
163
164     if( &_DYNAMIC == 0) {
165         return(0);
166     }
167     return(_DYNAMIC.ld_un.ld_1->ld_loaded);
168 }
169
170 /* Return the address of the ld.so allocated common symbol      */
171 /* with the least address, or 0 if none.                        */
172 static ptr_t GC_first_common()
173 {
174     ptr_t result = 0;
175     extern struct link_dynamic _DYNAMIC;
176     struct rtc_symb * curr_symbol;
177     
178     if( &_DYNAMIC == 0) {
179         return(0);
180     }
181     curr_symbol = _DYNAMIC.ldd -> ldd_cp;
182     for (; curr_symbol != 0; curr_symbol = curr_symbol -> rtc_next) {
183         if (result == 0
184             || (ptr_t)(curr_symbol -> rtc_sp -> n_value) < result) {
185             result = (ptr_t)(curr_symbol -> rtc_sp -> n_value);
186         }
187     }
188     return(result);
189 }
190
191 #endif  /* SUNOS4 ... */
192
193 # if defined(SUNOS4) || defined(SUNOS5DL)
194 /* Add dynamic library data sections to the root set.           */
195 # if !defined(PCR) && !defined(GC_SOLARIS_THREADS) && defined(THREADS)
196 #   ifndef SRC_M3
197         --> fix mutual exclusion with dlopen
198 #   endif  /* We assume M3 programs don't call dlopen for now */
199 # endif
200
201 # ifndef USE_PROC_FOR_LIBRARIES
202 void GC_register_dynamic_libraries()
203 {
204   struct link_map *lm = GC_FirstDLOpenedLinkMap();
205   
206
207   for (lm = GC_FirstDLOpenedLinkMap();
208        lm != (struct link_map *) 0;  lm = lm->l_next)
209     {
210 #     ifdef SUNOS4
211         struct exec *e;
212          
213         e = (struct exec *) lm->lm_addr;
214         GC_add_roots_inner(
215                     ((char *) (N_DATOFF(*e) + lm->lm_addr)),
216                     ((char *) (N_BSSADDR(*e) + e->a_bss + lm->lm_addr)),
217                     TRUE);
218 #     endif
219 #     ifdef SUNOS5DL
220         ElfW(Ehdr) * e;
221         ElfW(Phdr) * p;
222         unsigned long offset;
223         char * start;
224         register int i;
225         
226         e = (ElfW(Ehdr) *) lm->l_addr;
227         p = ((ElfW(Phdr) *)(((char *)(e)) + e->e_phoff));
228         offset = ((unsigned long)(lm->l_addr));
229         for( i = 0; i < (int)(e->e_phnum); ((i++),(p++)) ) {
230           switch( p->p_type ) {
231             case PT_LOAD:
232               {
233                 if( !(p->p_flags & PF_W) ) break;
234                 start = ((char *)(p->p_vaddr)) + offset;
235                 GC_add_roots_inner(
236                   start,
237                   start + p->p_memsz,
238                   TRUE
239                 );
240               }
241               break;
242             default:
243               break;
244           }
245         }
246 #     endif
247     }
248 #   ifdef SUNOS4
249       {
250         static ptr_t common_start = 0;
251         ptr_t common_end;
252         extern ptr_t GC_find_limit();
253         
254         if (common_start == 0) common_start = GC_first_common();
255         if (common_start != 0) {
256             common_end = GC_find_limit(common_start, TRUE);
257             GC_add_roots_inner((char *)common_start, (char *)common_end, TRUE);
258         }
259       }
260 #   endif
261 }
262
263 # endif /* !USE_PROC ... */
264 # endif /* SUNOS */
265
266 #if defined(LINUX) && defined(__ELF__) || defined(SCO_ELF) || \
267     (defined(FREEBSD) && defined(__ELF__)) || \
268     (defined(NETBSD) && defined(__ELF__)) || defined(HURD)
269
270
271 #ifdef USE_PROC_FOR_LIBRARIES
272
273 #include <string.h>
274
275 #include <sys/stat.h>
276 #include <fcntl.h>
277 #include <unistd.h>
278
279 #define MAPS_BUF_SIZE (32*1024)
280
281 extern ssize_t GC_repeat_read(int fd, char *buf, size_t count);
282         /* Repeatedly read until buffer is filled, or EOF is encountered */
283         /* Defined in os_dep.c.                                          */
284
285 static char *parse_map_entry(char *buf_ptr, word *start, word *end,
286                              char *prot_buf, unsigned int *maj_dev);
287
288 void GC_register_dynamic_libraries()
289 {
290     int f;
291     int result;
292     char prot_buf[5];
293     int maps_size;
294     char maps_temp[32768];
295     char *maps_buf;
296     char *buf_ptr;
297     int count;
298     word start, end;
299     unsigned int maj_dev, min_dev;
300     word least_ha, greatest_ha;
301     unsigned i;
302     word datastart = (word)(DATASTART);
303
304     /* Read /proc/self/maps     */
305         /* Note that we may not allocate, and thus can't use stdio.     */
306         f = open("/proc/self/maps", O_RDONLY);
307         if (-1 == f) ABORT("Couldn't open /proc/self/maps");
308         /* stat() doesn't work for /proc/self/maps, so we have to
309            read it to find out how large it is... */
310         maps_size = 0;
311         do {
312             result = GC_repeat_read(f, maps_temp, sizeof(maps_temp));
313             if (result <= 0) ABORT("Couldn't read /proc/self/maps");
314             maps_size += result;
315         } while (result == sizeof(maps_temp));
316
317         if (maps_size > sizeof(maps_temp)) {
318             /* If larger than our buffer, close and re-read it. */
319             close(f);
320             f = open("/proc/self/maps", O_RDONLY);
321             if (-1 == f) ABORT("Couldn't open /proc/self/maps");
322             maps_buf = alloca(maps_size);
323             if (NULL == maps_buf) ABORT("/proc/self/maps alloca failed");
324             result = GC_repeat_read(f, maps_buf, maps_size);
325             if (result <= 0) ABORT("Couldn't read /proc/self/maps");
326         } else {
327             /* Otherwise use the fixed size buffer */
328             maps_buf = maps_temp;
329         }
330
331         close(f);
332         maps_buf[result] = '\0';
333         buf_ptr = maps_buf;
334     /* Compute heap bounds. Should be done by add_to_heap?      */
335         least_ha = (word)(-1);
336         greatest_ha = 0;
337         for (i = 0; i < GC_n_heap_sects; ++i) {
338             word sect_start = (word)GC_heap_sects[i].hs_start;
339             word sect_end = sect_start + GC_heap_sects[i].hs_bytes;
340             if (sect_start < least_ha) least_ha = sect_start;
341             if (sect_end > greatest_ha) greatest_ha = sect_end;
342         }
343         if (greatest_ha < (word)GC_scratch_last_end_ptr)
344             greatest_ha = (word)GC_scratch_last_end_ptr; 
345     for (;;) {
346
347         buf_ptr = parse_map_entry(buf_ptr, &start, &end, prot_buf, &maj_dev);
348         if (buf_ptr == NULL) return;
349
350         if (prot_buf[1] == 'w') {
351             /* This is a writable mapping.  Add it to           */
352             /* the root set unless it is already otherwise      */
353             /* accounted for.                                   */
354             if (start <= (word)GC_stackbottom && end >= (word)GC_stackbottom) {
355                 /* Stack mapping; discard       */
356                 continue;
357             }
358 #           ifdef THREADS
359               if (GC_segment_is_thread_stack(start, end)) continue;
360 #           endif
361             /* The rest of this assumes that there is no mapping        */
362             /* spanning the beginning of the data segment, or extending */
363             /* beyond the entire heap at both ends.                     */
364             /* Empirically these assumptions hold.                      */
365             
366             if (start < (word)DATAEND && end > (word)DATAEND) {
367                 /* Rld may use space at the end of the main data        */
368                 /* segment.  Thus we add that in.                       */
369                 start = (word)DATAEND;
370             }
371             if (start < least_ha && end > least_ha) {
372                 end = least_ha;
373             }
374             if (start < greatest_ha && end > greatest_ha) {
375                 start = greatest_ha;
376             }
377             if (start >= least_ha && end <= greatest_ha) continue;
378             GC_add_roots_inner((char *)start, (char *)end, TRUE);
379         }
380      }
381 }
382
383 /* We now take care of the main data segment ourselves: */
384 GC_bool GC_register_main_static_data()
385 {
386   return FALSE;
387 }
388   
389 # define HAVE_REGISTER_MAIN_STATIC_DATA
390 //
391 //  parse_map_entry parses an entry from /proc/self/maps so we can
392 //  locate all writable data segments that belong to shared libraries.
393 //  The format of one of these entries and the fields we care about
394 //  is as follows:
395 //  XXXXXXXX-XXXXXXXX r-xp 00000000 30:05 260537     name of mapping...\n
396 //  ^^^^^^^^ ^^^^^^^^ ^^^^          ^^
397 //  start    end      prot          maj_dev
398 //  0        9        18            32
399 //
400 //  The parser is called with a pointer to the entry and the return value
401 //  is either NULL or is advanced to the next entry(the byte after the
402 //  trailing '\n'.)
403 //
404 #define OFFSET_MAP_START   0
405 #define OFFSET_MAP_END     9
406 #define OFFSET_MAP_PROT   18
407 #define OFFSET_MAP_MAJDEV 32
408
409 static char *parse_map_entry(char *buf_ptr, word *start, word *end,
410                              char *prot_buf, unsigned int *maj_dev)
411 {
412     int i;
413     unsigned int val;
414     char *tok;
415
416     if (buf_ptr == NULL || *buf_ptr == '\0') {
417         return NULL;
418     }
419
420     memcpy(prot_buf, buf_ptr+OFFSET_MAP_PROT, 4); // do the protections first
421     prot_buf[4] = '\0';
422
423     if (prot_buf[1] == 'w') { // we can skip all of this if it's not writable
424
425         tok = buf_ptr;
426         buf_ptr[OFFSET_MAP_START+8] = '\0';
427         *start = strtoul(tok, NULL, 16);
428
429         tok = buf_ptr+OFFSET_MAP_END;
430         buf_ptr[OFFSET_MAP_END+8] = '\0';
431         *end = strtoul(tok, NULL, 16);
432
433         buf_ptr += OFFSET_MAP_MAJDEV;
434         tok = buf_ptr;
435         while (*buf_ptr != ':') buf_ptr++;
436         *buf_ptr++ = '\0';
437         *maj_dev = strtoul(tok, NULL, 16);
438     }
439
440     while (*buf_ptr && *buf_ptr++ != '\n');
441
442     return buf_ptr;
443 }
444
445 #endif /* USE_PROC_FOR_LIBRARIES */
446
447 #if !defined(USE_PROC_FOR_LIBRARIES)
448 /* The following is the preferred way to walk dynamic libraries */
449 /* For glibc 2.2.4+.  Unfortunately, it doesn't work for older  */
450 /* versions.  Thanks to Jakub Jelinek for most of the code.     */
451
452 # if defined(LINUX) /* Are others OK here, too? */ \
453      && (__GLIBC__ > 2 || (__GLIBC__ == 2 && __GLIBC_MINOR__ > 2) \
454          || (__GLIBC__ == 2 && __GLIBC_MINOR__ == 2 && defined(DT_CONFIG))) 
455
456 /* We have the header files for a glibc that includes dl_iterate_phdr.  */
457 /* It may still not be available in the library on the target system.   */
458 /* Thus we also treat it as a weak symbol.                              */
459 #define HAVE_DL_ITERATE_PHDR
460
461 static int GC_register_dynlib_callback(info, size, ptr)
462      struct dl_phdr_info * info;
463      size_t size;
464      void * ptr;
465 {
466   const ElfW(Phdr) * p;
467   char * start;
468   register int i;
469
470   /* Make sure struct dl_phdr_info is at least as big as we need.  */
471   if (size < offsetof (struct dl_phdr_info, dlpi_phnum)
472       + sizeof (info->dlpi_phnum))
473     return -1;
474
475   p = info->dlpi_phdr;
476   for( i = 0; i < (int)(info->dlpi_phnum); ((i++),(p++)) ) {
477     switch( p->p_type ) {
478       case PT_LOAD:
479         {
480           if( !(p->p_flags & PF_W) ) break;
481           start = ((char *)(p->p_vaddr)) + info->dlpi_addr;
482           GC_add_roots_inner(start, start + p->p_memsz, TRUE);
483         }
484       break;
485       default:
486         break;
487     }
488   }
489
490   return 0;
491 }     
492
493 /* Return TRUE if we succeed, FALSE if dl_iterate_phdr wasn't there. */
494
495 #pragma weak dl_iterate_phdr
496
497 GC_bool GC_register_dynamic_libraries_dl_iterate_phdr()
498 {
499   int tmp = 0;
500
501   if (dl_iterate_phdr) {
502     dl_iterate_phdr(GC_register_dynlib_callback, &tmp);
503     return TRUE;
504   } else {
505     return FALSE;
506   }
507 }
508
509 /* Do we need to separately register the main static data segment? */
510 GC_bool GC_register_main_static_data()
511 {
512   return (dl_iterate_phdr == 0);
513 }
514
515 #define HAVE_REGISTER_MAIN_STATIC_DATA
516
517 # else /* !LINUX || version(glibc) < 2.2.4 */
518
519 /* Dynamic loading code for Linux running ELF. Somewhat tested on
520  * Linux/x86, untested but hopefully should work on Linux/Alpha. 
521  * This code was derived from the Solaris/ELF support. Thanks to
522  * whatever kind soul wrote that.  - Patrick Bridges */
523
524 /* This doesn't necessarily work in all cases, e.g. with preloaded
525  * dynamic libraries.                                           */
526
527 #if defined(NETBSD)
528 #  include <sys/exec_elf.h>
529 #else
530 #  include <elf.h>
531 #endif
532 #include <link.h>
533
534 # endif
535
536 #ifdef __GNUC__
537 # pragma weak _DYNAMIC
538 #endif
539 extern ElfW(Dyn) _DYNAMIC[];
540
541 static struct link_map *
542 GC_FirstDLOpenedLinkMap()
543 {
544     ElfW(Dyn) *dp;
545     struct r_debug *r;
546     static struct link_map *cachedResult = 0;
547
548     if( _DYNAMIC == 0) {
549         return(0);
550     }
551     if( cachedResult == 0 ) {
552         int tag;
553         for( dp = _DYNAMIC; (tag = dp->d_tag) != 0; dp++ ) {
554             if( tag == DT_DEBUG ) {
555                 struct link_map *lm
556                         = ((struct r_debug *)(dp->d_un.d_ptr))->r_map;
557                 if( lm != 0 ) cachedResult = lm->l_next; /* might be NIL */
558                 break;
559             }
560         }
561     }
562     return cachedResult;
563 }
564
565
566 void GC_register_dynamic_libraries()
567 {
568   struct link_map *lm;
569   
570
571 # ifdef HAVE_DL_ITERATE_PHDR
572     if (GC_register_dynamic_libraries_dl_iterate_phdr()) {
573         return;
574     }
575 # endif
576   lm = GC_FirstDLOpenedLinkMap();
577   for (lm = GC_FirstDLOpenedLinkMap();
578        lm != (struct link_map *) 0;  lm = lm->l_next)
579     {
580         ElfW(Ehdr) * e;
581         ElfW(Phdr) * p;
582         unsigned long offset;
583         char * start;
584         register int i;
585         
586         e = (ElfW(Ehdr) *) lm->l_addr;
587         p = ((ElfW(Phdr) *)(((char *)(e)) + e->e_phoff));
588         offset = ((unsigned long)(lm->l_addr));
589         for( i = 0; i < (int)(e->e_phnum); ((i++),(p++)) ) {
590           switch( p->p_type ) {
591             case PT_LOAD:
592               {
593                 if( !(p->p_flags & PF_W) ) break;
594                 start = ((char *)(p->p_vaddr)) + offset;
595                 GC_add_roots_inner(start, start + p->p_memsz, TRUE);
596               }
597               break;
598             default:
599               break;
600           }
601         }
602     }
603 }
604
605 #endif /* !USE_PROC_FOR_LIBRARIES */
606
607 #endif /* LINUX */
608
609 #if defined(IRIX5) || (defined(USE_PROC_FOR_LIBRARIES) && !defined(LINUX))
610
611 #include <sys/procfs.h>
612 #include <sys/stat.h>
613 #include <fcntl.h>
614 #include <elf.h>
615 #include <errno.h>
616 #include <signal.h>  /* Only for the following test. */
617 #ifndef _sigargs
618 # define IRIX6
619 #endif
620
621 extern void * GC_roots_present();
622         /* The type is a lie, since the real type doesn't make sense here, */
623         /* and we only test for NULL.                                      */
624
625 /* We use /proc to track down all parts of the address space that are   */
626 /* mapped by the process, and throw out regions we know we shouldn't    */
627 /* worry about.  This may also work under other SVR4 variants.          */
628 void GC_register_dynamic_libraries()
629 {
630     static int fd = -1;
631     char buf[30];
632     static prmap_t * addr_map = 0;
633     static int current_sz = 0;  /* Number of records currently in addr_map */
634     static int needed_sz;       /* Required size of addr_map            */
635     register int i;
636     register long flags;
637     register ptr_t start;
638     register ptr_t limit;
639     ptr_t heap_start = (ptr_t)HEAP_START;
640     ptr_t heap_end = heap_start;
641
642 #   ifdef SUNOS5DL
643 #     define MA_PHYS 0
644 #   endif /* SUNOS5DL */
645
646     if (fd < 0) {
647       sprintf(buf, "/proc/%d", getpid());
648         /* The above generates a lint complaint, since pid_t varies.    */
649         /* It's unclear how to improve this.                            */
650       fd = open(buf, O_RDONLY);
651       if (fd < 0) {
652         ABORT("/proc open failed");
653       }
654     }
655     if (ioctl(fd, PIOCNMAP, &needed_sz) < 0) {
656         GC_err_printf2("fd = %d, errno = %d\n", fd, errno);
657         ABORT("/proc PIOCNMAP ioctl failed");
658     }
659     if (needed_sz >= current_sz) {
660         current_sz = needed_sz * 2 + 1;
661                         /* Expansion, plus room for 0 record */
662         addr_map = (prmap_t *)GC_scratch_alloc((word)
663                                                 (current_sz * sizeof(prmap_t)));
664     }
665     if (ioctl(fd, PIOCMAP, addr_map) < 0) {
666         GC_err_printf4("fd = %d, errno = %d, needed_sz = %d, addr_map = 0x%X\n",
667                         fd, errno, needed_sz, addr_map);
668         ABORT("/proc PIOCMAP ioctl failed");
669     };
670     if (GC_n_heap_sects > 0) {
671         heap_end = GC_heap_sects[GC_n_heap_sects-1].hs_start
672                         + GC_heap_sects[GC_n_heap_sects-1].hs_bytes;
673         if (heap_end < GC_scratch_last_end_ptr) heap_end = GC_scratch_last_end_ptr; 
674     }
675     for (i = 0; i < needed_sz; i++) {
676         flags = addr_map[i].pr_mflags;
677         if ((flags & (MA_BREAK | MA_STACK | MA_PHYS)) != 0) goto irrelevant;
678         if ((flags & (MA_READ | MA_WRITE)) != (MA_READ | MA_WRITE))
679             goto irrelevant;
680           /* The latter test is empirically useless in very old Irix    */
681           /* versions.  Other than the                                  */
682           /* main data and stack segments, everything appears to be     */
683           /* mapped readable, writable, executable, and shared(!!).     */
684           /* This makes no sense to me. - HB                            */
685         start = (ptr_t)(addr_map[i].pr_vaddr);
686         if (GC_roots_present(start)) goto irrelevant;
687         if (start < heap_end && start >= heap_start)
688                 goto irrelevant;
689 #       ifdef MMAP_STACKS
690           if (GC_is_thread_stack(start)) goto irrelevant;
691 #       endif /* MMAP_STACKS */
692
693         limit = start + addr_map[i].pr_size;
694         /* The following seemed to be necessary for very old versions   */
695         /* of Irix, but it has been reported to discard relevant        */
696         /* segments under Irix 6.5.                                     */
697 #       ifndef IRIX6
698           if (addr_map[i].pr_off == 0 && strncmp(start, ELFMAG, 4) == 0) {
699             /* Discard text segments, i.e. 0-offset mappings against    */
700             /* executable files which appear to have ELF headers.       */
701             caddr_t arg;
702             int obj;
703 #           define MAP_IRR_SZ 10
704             static ptr_t map_irr[MAP_IRR_SZ];
705                                         /* Known irrelevant map entries */
706             static int n_irr = 0;
707             struct stat buf;
708             register int i;
709             
710             for (i = 0; i < n_irr; i++) {
711                 if (map_irr[i] == start) goto irrelevant;
712             }
713             arg = (caddr_t)start;
714             obj = ioctl(fd, PIOCOPENM, &arg);
715             if (obj >= 0) {
716                 fstat(obj, &buf);
717                 close(obj);
718                 if ((buf.st_mode & 0111) != 0) {
719                     if (n_irr < MAP_IRR_SZ) {
720                         map_irr[n_irr++] = start;
721                     }
722                     goto irrelevant;
723                 }
724             }
725           }
726 #       endif /* !IRIX6 */
727         GC_add_roots_inner(start, limit, TRUE);
728       irrelevant: ;
729     }
730     /* Dont keep cached descriptor, for now.  Some kernels don't like us */
731     /* to keep a /proc file descriptor around during kill -9.            */
732         if (close(fd) < 0) ABORT("Couldnt close /proc file");
733         fd = -1;
734 }
735
736 # endif /* USE_PROC || IRIX5 */
737
738 # if defined(MSWIN32) || defined(MSWINCE)
739
740 # define WIN32_LEAN_AND_MEAN
741 # define NOSERVICE
742 # include <windows.h>
743 # include <stdlib.h>
744
745   /* We traverse the entire address space and register all segments     */
746   /* that could possibly have been written to.                          */
747   
748   extern GC_bool GC_is_heap_base (ptr_t p);
749
750 # ifdef GC_WIN32_THREADS
751     extern void GC_get_next_stack(char *start, char **lo, char **hi);
752     void GC_cond_add_roots(char *base, char * limit)
753     {
754       char * curr_base = base;
755       char * next_stack_lo;
756       char * next_stack_hi;
757    
758       if (base == limit) return;
759       for(;;) {
760           GC_get_next_stack(curr_base, &next_stack_lo, &next_stack_hi);
761           if (next_stack_lo >= limit) break;
762           GC_add_roots_inner(curr_base, next_stack_lo, TRUE);
763           curr_base = next_stack_hi;
764       }
765       if (curr_base < limit) GC_add_roots_inner(curr_base, limit, TRUE);
766     }
767 # else
768     void GC_cond_add_roots(char *base, char * limit)
769     {
770       char dummy;
771       char * stack_top
772          = (char *) ((word)(&dummy) & ~(GC_sysinfo.dwAllocationGranularity-1));
773       if (base == limit) return;
774       if (limit > stack_top && base < GC_stackbottom) {
775           /* Part of the stack; ignore it. */
776           return;
777       }
778       GC_add_roots_inner(base, limit, TRUE);
779     }
780 # endif
781
782 # ifdef MSWINCE
783   /* Do we need to separately register the main static data segment? */
784   GC_bool GC_register_main_static_data()
785   {
786     return FALSE;
787   }
788 # else /* win32 */
789   extern GC_bool GC_no_win32_dlls;
790
791   GC_bool GC_register_main_static_data()
792   {
793     return GC_no_win32_dlls;
794   }
795 # endif /* win32 */
796   
797 # define HAVE_REGISTER_MAIN_STATIC_DATA
798
799   void GC_register_dynamic_libraries()
800   {
801     MEMORY_BASIC_INFORMATION buf;
802     DWORD result;
803     DWORD protect;
804     LPVOID p;
805     char * base;
806     char * limit, * new_limit;
807
808 #   ifdef MSWIN32
809       if (GC_no_win32_dlls) return;
810 #   endif
811     base = limit = p = GC_sysinfo.lpMinimumApplicationAddress;
812 #   if defined(MSWINCE) && !defined(_WIN32_WCE_EMULATION)
813     /* Only the first 32 MB of address space belongs to the current process */
814     while (p < (LPVOID)0x02000000) {
815         result = VirtualQuery(p, &buf, sizeof(buf));
816         if (result == 0) {
817             /* Page is free; advance to the next possible allocation base */
818             new_limit = (char *)
819                 (((DWORD) p + GC_sysinfo.dwAllocationGranularity)
820                  & ~(GC_sysinfo.dwAllocationGranularity-1));
821         } else
822 #   else
823     while (p < GC_sysinfo.lpMaximumApplicationAddress) {
824         result = VirtualQuery(p, &buf, sizeof(buf));
825 #   endif
826         {
827             if (result != sizeof(buf)) {
828                 ABORT("Weird VirtualQuery result");
829             }
830             new_limit = (char *)p + buf.RegionSize;
831             protect = buf.Protect;
832             if (buf.State == MEM_COMMIT
833                 && (protect == PAGE_EXECUTE_READWRITE
834                     || protect == PAGE_READWRITE)
835                 && !GC_is_heap_base(buf.AllocationBase)) {
836                 if ((char *)p != limit) {
837                     GC_cond_add_roots(base, limit);
838                     base = p;
839                 }
840                 limit = new_limit;
841             }
842         }
843         if (p > (LPVOID)new_limit /* overflow */) break;
844         p = (LPVOID)new_limit;
845     }
846     GC_cond_add_roots(base, limit);
847   }
848
849 #endif /* MSWIN32 || MSWINCE */
850   
851 #if defined(ALPHA) && defined(OSF1)
852
853 #include <loader.h>
854
855 void GC_register_dynamic_libraries()
856 {
857   int status;
858   ldr_process_t mypid;
859
860   /* module */
861     ldr_module_t moduleid = LDR_NULL_MODULE;
862     ldr_module_info_t moduleinfo;
863     size_t moduleinfosize = sizeof(moduleinfo);
864     size_t modulereturnsize;    
865
866   /* region */
867     ldr_region_t region; 
868     ldr_region_info_t regioninfo;
869     size_t regioninfosize = sizeof(regioninfo);
870     size_t regionreturnsize;
871
872   /* Obtain id of this process */
873     mypid = ldr_my_process();
874   
875   /* For each module */
876     while (TRUE) {
877
878       /* Get the next (first) module */
879         status = ldr_next_module(mypid, &moduleid);
880
881       /* Any more modules? */
882         if (moduleid == LDR_NULL_MODULE)
883             break;    /* No more modules */
884
885       /* Check status AFTER checking moduleid because */
886       /* of a bug in the non-shared ldr_next_module stub */
887         if (status != 0 ) {
888             GC_printf1("dynamic_load: status = %ld\n", (long)status);
889             {
890                 extern char *sys_errlist[];
891                 extern int sys_nerr;
892                 extern int errno;
893                 if (errno <= sys_nerr) {
894                     GC_printf1("dynamic_load: %s\n", (long)sys_errlist[errno]);
895                } else {
896                     GC_printf1("dynamic_load: %d\n", (long)errno);
897                 }
898         }
899             ABORT("ldr_next_module failed");
900          }
901
902       /* Get the module information */
903         status = ldr_inq_module(mypid, moduleid, &moduleinfo,
904                                 moduleinfosize, &modulereturnsize); 
905         if (status != 0 )
906             ABORT("ldr_inq_module failed");
907
908       /* is module for the main program (i.e. nonshared portion)? */
909           if (moduleinfo.lmi_flags & LDR_MAIN)
910               continue;    /* skip the main module */
911
912 #     ifdef VERBOSE
913           GC_printf("---Module---\n");
914           GC_printf("Module ID            = %16ld\n", moduleinfo.lmi_modid);
915           GC_printf("Count of regions     = %16d\n", moduleinfo.lmi_nregion);
916           GC_printf("flags for module     = %16lx\n", moduleinfo.lmi_flags); 
917           GC_printf("pathname of module   = \"%s\"\n", moduleinfo.lmi_name);
918 #     endif
919
920       /* For each region in this module */
921         for (region = 0; region < moduleinfo.lmi_nregion; region++) {
922
923           /* Get the region information */
924             status = ldr_inq_region(mypid, moduleid, region, &regioninfo,
925                                     regioninfosize, &regionreturnsize);
926             if (status != 0 )
927                 ABORT("ldr_inq_region failed");
928
929           /* only process writable (data) regions */
930             if (! (regioninfo.lri_prot & LDR_W))
931                 continue;
932
933 #         ifdef VERBOSE
934               GC_printf("--- Region ---\n");
935               GC_printf("Region number    = %16ld\n",
936                         regioninfo.lri_region_no);
937               GC_printf("Protection flags = %016x\n",  regioninfo.lri_prot);
938               GC_printf("Virtual address  = %16p\n",   regioninfo.lri_vaddr);
939               GC_printf("Mapped address   = %16p\n",   regioninfo.lri_mapaddr);
940               GC_printf("Region size      = %16ld\n",  regioninfo.lri_size);
941               GC_printf("Region name      = \"%s\"\n", regioninfo.lri_name);
942 #         endif
943
944           /* register region as a garbage collection root */
945             GC_add_roots_inner (
946                 (char *)regioninfo.lri_mapaddr,
947                 (char *)regioninfo.lri_mapaddr + regioninfo.lri_size,
948                 TRUE);
949
950         }
951     }
952 }
953 #endif
954
955 #if defined(HPUX)
956
957 #include <errno.h>
958 #include <dl.h>
959
960 extern int errno;
961 extern char *sys_errlist[];
962 extern int sys_nerr;
963
964 void GC_register_dynamic_libraries()
965 {
966   int status;
967   int index = 1; /* Ordinal position in shared library search list */
968   struct shl_descriptor *shl_desc; /* Shared library info, see dl.h */
969
970   /* For each dynamic library loaded */
971     while (TRUE) {
972
973       /* Get info about next shared library */
974         status = shl_get(index, &shl_desc);
975
976       /* Check if this is the end of the list or if some error occured */
977         if (status != 0) {
978 #        ifdef GC_HPUX_THREADS
979            /* I've seen errno values of 0.  The man page is not clear   */
980            /* as to whether errno should get set on a -1 return.        */
981            break;
982 #        else
983           if (errno == EINVAL) {
984               break; /* Moved past end of shared library list --> finished */
985           } else {
986               if (errno <= sys_nerr) {
987                     GC_printf1("dynamic_load: %s\n", (long) sys_errlist[errno]);
988               } else {
989                     GC_printf1("dynamic_load: %d\n", (long) errno);
990               }
991               ABORT("shl_get failed");
992           }
993 #        endif
994         }
995
996 #     ifdef VERBOSE
997           GC_printf0("---Shared library---\n");
998           GC_printf1("\tfilename        = \"%s\"\n", shl_desc->filename);
999           GC_printf1("\tindex           = %d\n", index);
1000           GC_printf1("\thandle          = %08x\n",
1001                                         (unsigned long) shl_desc->handle);
1002           GC_printf1("\ttext seg. start = %08x\n", shl_desc->tstart);
1003           GC_printf1("\ttext seg. end   = %08x\n", shl_desc->tend);
1004           GC_printf1("\tdata seg. start = %08x\n", shl_desc->dstart);
1005           GC_printf1("\tdata seg. end   = %08x\n", shl_desc->dend);
1006           GC_printf1("\tref. count      = %lu\n", shl_desc->ref_count);
1007 #     endif
1008
1009       /* register shared library's data segment as a garbage collection root */
1010         GC_add_roots_inner((char *) shl_desc->dstart,
1011                            (char *) shl_desc->dend, TRUE);
1012
1013         index++;
1014     }
1015 }
1016 #endif /* HPUX */
1017
1018 #ifdef RS6000
1019 #pragma alloca
1020 #include <sys/ldr.h>
1021 #include <sys/errno.h>
1022 void GC_register_dynamic_libraries()
1023 {
1024         int len;
1025         char *ldibuf;
1026         int ldibuflen;
1027         struct ld_info *ldi;
1028
1029         ldibuf = alloca(ldibuflen = 8192);
1030
1031         while ( (len = loadquery(L_GETINFO,ldibuf,ldibuflen)) < 0) {
1032                 if (errno != ENOMEM) {
1033                         ABORT("loadquery failed");
1034                 }
1035                 ldibuf = alloca(ldibuflen *= 2);
1036         }
1037
1038         ldi = (struct ld_info *)ldibuf;
1039         while (ldi) {
1040                 len = ldi->ldinfo_next;
1041                 GC_add_roots_inner(
1042                                 ldi->ldinfo_dataorg,
1043                                 (unsigned long)ldi->ldinfo_dataorg
1044                                 + ldi->ldinfo_datasize,
1045                                 TRUE);
1046                 ldi = len ? (struct ld_info *)((char *)ldi + len) : 0;
1047         }
1048 }
1049 #endif /* RS6000 */
1050
1051
1052
1053 #else /* !DYNAMIC_LOADING */
1054
1055 #ifdef PCR
1056
1057 #   include "il/PCR_IL.h"
1058 #   include "th/PCR_ThCtl.h"
1059 #   include "mm/PCR_MM.h"
1060
1061 void GC_register_dynamic_libraries()
1062 {
1063     /* Add new static data areas of dynamically loaded modules. */
1064         {
1065           PCR_IL_LoadedFile * p = PCR_IL_GetLastLoadedFile();
1066           PCR_IL_LoadedSegment * q;
1067           
1068           /* Skip uncommited files */
1069           while (p != NIL && !(p -> lf_commitPoint)) {
1070               /* The loading of this file has not yet been committed    */
1071               /* Hence its description could be inconsistent.           */
1072               /* Furthermore, it hasn't yet been run.  Hence its data   */
1073               /* segments can't possibly reference heap allocated       */
1074               /* objects.                                               */
1075               p = p -> lf_prev;
1076           }
1077           for (; p != NIL; p = p -> lf_prev) {
1078             for (q = p -> lf_ls; q != NIL; q = q -> ls_next) {
1079               if ((q -> ls_flags & PCR_IL_SegFlags_Traced_MASK)
1080                   == PCR_IL_SegFlags_Traced_on) {
1081                 GC_add_roots_inner
1082                         ((char *)(q -> ls_addr), 
1083                          (char *)(q -> ls_addr) + q -> ls_bytes,
1084                          TRUE);
1085               }
1086             }
1087           }
1088         }
1089 }
1090
1091
1092 #else /* !PCR */
1093
1094 void GC_register_dynamic_libraries(){}
1095
1096 int GC_no_dynamic_loading;
1097
1098 #endif /* !PCR */
1099
1100 #endif /* !DYNAMIC_LOADING */
1101
1102 #ifndef HAVE_REGISTER_MAIN_STATIC_DATA
1103
1104 /* Do we need to separately register the main static data segment? */
1105 GC_bool GC_register_main_static_data()
1106 {
1107   return TRUE;
1108 }
1109 #endif /* HAVE_REGISTER_MAIN_STATIC_DATA */
1110