mono/profiler/log.c

   1 /*
   2  * mono-profiler-log.c: mono log profiler
   3  *
   4  * Authors:
   5  *   Paolo Molaro (lupus@ximian.com)
   6  *   Alex Rønne Petersen (alexrp@xamarin.com)
   7  *
   8  * Copyright 2010 Novell, Inc (http://www.novell.com)
   9  * Copyright 2011 Xamarin Inc (http://www.xamarin.com)
  10  * Licensed under the MIT license. See LICENSE file in the project root for full license information.
  11  */
  12
  13 #include <config.h>
  14 #include <mono/metadata/assembly.h>
  15 #include <mono/metadata/debug-helpers.h>
  16 #include "../metadata/metadata-internals.h"
  17 #include <mono/metadata/mono-config.h>
  18 #include <mono/metadata/mono-gc.h>
  19 #include <mono/metadata/mono-perfcounters.h>
  20 #include <mono/utils/atomic.h>
  21 #include <mono/utils/hazard-pointer.h>
  22 #include <mono/utils/lock-free-alloc.h>
  23 #include <mono/utils/lock-free-queue.h>
  24 #include <mono/utils/mono-conc-hashtable.h>
  25 #include <mono/utils/mono-counters.h>
  26 #include <mono/utils/mono-linked-list-set.h>
  27 #include <mono/utils/mono-membar.h>
  28 #include <mono/utils/mono-mmap.h>
  29 #include <mono/utils/mono-os-mutex.h>
  30 #include <mono/utils/mono-os-semaphore.h>
  31 #include <mono/utils/mono-threads.h>
  32 #include <mono/utils/mono-threads-api.h>
  33 #include "log.h"
  34
  35 #ifdef HAVE_DLFCN_H
  36 #include <dlfcn.h>
  37 #endif
  38 #include <fcntl.h>
  39 #ifdef HAVE_LINK_H
  40 #include <link.h>
  41 #endif
  42 #ifdef HAVE_UNISTD_H
  43 #include <unistd.h>
  44 #endif
  45 #if defined(__APPLE__)
  46 #include <mach/mach_time.h>
  47 #endif
  48 #include <netinet/in.h>
  49 #ifdef HAVE_SYS_MMAN_H
  50 #include <sys/mman.h>
  51 #endif
  52 #include <sys/socket.h>
  53 #if defined (HAVE_SYS_ZLIB)
  54 #include <zlib.h>
  55 #endif
  56
  57 #define BUFFER_SIZE (4096 * 16)
  58
  59 /* Worst-case size in bytes of a 64-bit value encoded with LEB128. */
  60 #define LEB128_SIZE 10
  61
  62 /* Size of a value encoded as a single byte. */
  63 #undef BYTE_SIZE // mach/i386/vm_param.h on OS X defines this to 8, but it isn't used for anything.
  64 #define BYTE_SIZE 1
  65
  66 /* Size in bytes of the event prefix (ID + time). */
  67 #define EVENT_SIZE (BYTE_SIZE + LEB128_SIZE)
  68
  69 static volatile gint32 runtime_inited;
  70 static volatile gint32 in_shutdown;
  71
  72 static ProfilerConfig config;
  73 static int nocalls = 0;
  74 static int notraces = 0;
  75 static int use_zip = 0;
  76 static int do_report = 0;
  77 static int do_heap_shot = 0;
  78 static int max_call_depth = 0;
  79 static int command_port = 0;
  80 static int heapshot_requested = 0;
  81 static int do_mono_sample = 0;
  82 static int do_debug = 0;
  83 static int do_coverage = 0;
  84 static gboolean no_counters = FALSE;
  85 static gboolean only_coverage = FALSE;
  86 static gboolean debug_coverage = FALSE;
  87 static int max_allocated_sample_hits;
  88
  89 #define ENABLED(EVT) (config.effective_mask & (EVT))
  90
  91 // Statistics for internal profiler data structures.
  92 static gint32 sample_allocations_ctr,
  93               buffer_allocations_ctr;
  94
  95 // Statistics for profiler events.
  96 static gint32 sync_points_ctr,
  97               heap_objects_ctr,
  98               heap_starts_ctr,
  99               heap_ends_ctr,
 100               heap_roots_ctr,
 101               gc_events_ctr,
 102               gc_resizes_ctr,
 103               gc_allocs_ctr,
 104               gc_moves_ctr,
 105               gc_handle_creations_ctr,
 106               gc_handle_deletions_ctr,
 107               finalize_begins_ctr,
 108               finalize_ends_ctr,
 109               finalize_object_begins_ctr,
 110               finalize_object_ends_ctr,
 111               image_loads_ctr,
 112               image_unloads_ctr,
 113               assembly_loads_ctr,
 114               assembly_unloads_ctr,
 115               class_loads_ctr,
 116               class_unloads_ctr,
 117               method_entries_ctr,
 118               method_exits_ctr,
 119               method_exception_exits_ctr,
 120               method_jits_ctr,
 121               code_buffers_ctr,
 122               exception_throws_ctr,
 123               exception_clauses_ctr,
 124               monitor_events_ctr,
 125               thread_starts_ctr,
 126               thread_ends_ctr,
 127               thread_names_ctr,
 128               domain_loads_ctr,
 129               domain_unloads_ctr,
 130               domain_names_ctr,
 131               context_loads_ctr,
 132               context_unloads_ctr,
 133               sample_ubins_ctr,
 134               sample_usyms_ctr,
 135               sample_hits_ctr,
 136               counter_descriptors_ctr,
 137               counter_samples_ctr,
 138               perfcounter_descriptors_ctr,
 139               perfcounter_samples_ctr,
 140               coverage_methods_ctr,
 141               coverage_statements_ctr,
 142               coverage_classes_ctr,
 143               coverage_assemblies_ctr;
 144
 145 static MonoLinkedListSet profiler_thread_list;
 146
 147 /*
 148  * file format:
 149  * [header] [buffer]*
 150  *
 151  * The file is composed by a header followed by 0 or more buffers.
 152  * Each buffer contains events that happened on a thread: for a given thread
 153  * buffers that appear later in the file are guaranteed to contain events
 154  * that happened later in time. Buffers from separate threads could be interleaved,
 155  * though.
 156  * Buffers are not required to be aligned.
 157  *
 158  * header format:
 159  * [id: 4 bytes] constant value: LOG_HEADER_ID
 160  * [major: 1 byte] [minor: 1 byte] major and minor version of the log profiler
 161  * [format: 1 byte] version of the data format for the rest of the file
 162  * [ptrsize: 1 byte] size in bytes of a pointer in the profiled program
 163  * [startup time: 8 bytes] time in milliseconds since the unix epoch when the program started
 164  * [timer overhead: 4 bytes] approximate overhead in nanoseconds of the timer
 165  * [flags: 4 bytes] file format flags, should be 0 for now
 166  * [pid: 4 bytes] pid of the profiled process
 167  * [port: 2 bytes] tcp port for server if != 0
 168  * [args size: 4 bytes] size of args
 169  * [args: string] arguments passed to the profiler
 170  * [arch size: 4 bytes] size of arch
 171  * [arch: string] architecture the profiler is running on
 172  * [os size: 4 bytes] size of os
 173  * [os: string] operating system the profiler is running on
 174  *
 175  * The multiple byte integers are in little-endian format.
 176  *
 177  * buffer format:
 178  * [buffer header] [event]*
 179  * Buffers have a fixed-size header followed by 0 or more bytes of event data.
 180  * Timing information and other values in the event data are usually stored
 181  * as uleb128 or sleb128 integers. To save space, as noted for each item below,
 182  * some data is represented as a difference between the actual value and
 183  * either the last value of the same type (like for timing information) or
 184  * as the difference from a value stored in a buffer header.
 185  *
 186  * For timing information the data is stored as uleb128, since timing
 187  * increases in a monotonic way in each thread: the value is the number of
 188  * nanoseconds to add to the last seen timing data in a buffer. The first value
 189  * in a buffer will be calculated from the time_base field in the buffer head.
 190  *
 191  * Object or heap sizes are stored as uleb128.
 192  * Pointer differences are stored as sleb128, instead.
 193  *
 194  * If an unexpected value is found, the rest of the buffer should be ignored,
 195  * as generally the later values need the former to be interpreted correctly.
 196  *
 197  * buffer header format:
 198  * [bufid: 4 bytes] constant value: BUF_ID
 199  * [len: 4 bytes] size of the data following the buffer header
 200  * [time_base: 8 bytes] time base in nanoseconds since an unspecified epoch
 201  * [ptr_base: 8 bytes] base value for pointers
 202  * [obj_base: 8 bytes] base value for object addresses
 203  * [thread id: 8 bytes] system-specific thread ID (pthread_t for example)
 204  * [method_base: 8 bytes] base value for MonoMethod pointers
 205  *
 206  * event format:
 207  * [extended info: upper 4 bits] [type: lower 4 bits]
 208  * [time diff: uleb128] nanoseconds since last timing
 209  * [data]*
 210  * The data that follows depends on type and the extended info.
 211  * Type is one of the enum values in mono-profiler-log.h: TYPE_ALLOC, TYPE_GC,
 212  * TYPE_METADATA, TYPE_METHOD, TYPE_EXCEPTION, TYPE_MONITOR, TYPE_HEAP.
 213  * The extended info bits are interpreted based on type, see
 214  * each individual event description below.
 215  * strings are represented as a 0-terminated utf8 sequence.
 216  *
 217  * backtrace format:
 218  * [num: uleb128] number of frames following
 219  * [frame: sleb128]* mum MonoMethod* as a pointer difference from the last such
 220  * pointer or the buffer method_base
 221  *
 222  * type alloc format:
 223  * type: TYPE_ALLOC
 224  * exinfo: zero or TYPE_ALLOC_BT
 225  * [ptr: sleb128] class as a byte difference from ptr_base
 226  * [obj: sleb128] object address as a byte difference from obj_base
 227  * [size: uleb128] size of the object in the heap
 228  * If exinfo == TYPE_ALLOC_BT, a backtrace follows.
 229  *
 230  * type GC format:
 231  * type: TYPE_GC
 232  * exinfo: one of TYPE_GC_EVENT, TYPE_GC_RESIZE, TYPE_GC_MOVE, TYPE_GC_HANDLE_CREATED[_BT],
 233  * TYPE_GC_HANDLE_DESTROYED[_BT], TYPE_GC_FINALIZE_START, TYPE_GC_FINALIZE_END,
 234  * TYPE_GC_FINALIZE_OBJECT_START, TYPE_GC_FINALIZE_OBJECT_END
 235  * if exinfo == TYPE_GC_RESIZE
 236  *      [heap_size: uleb128] new heap size
 237  * if exinfo == TYPE_GC_EVENT
 238  *      [event type: byte] GC event (MONO_GC_EVENT_* from profiler.h)
 239  *      [generation: byte] GC generation event refers to
 240  * if exinfo == TYPE_GC_MOVE
 241  *      [num_objects: uleb128] number of object moves that follow
 242  *      [objaddr: sleb128]+ num_objects object pointer differences from obj_base
 243  *      num is always an even number: the even items are the old
 244  *      addresses, the odd numbers are the respective new object addresses
 245  * if exinfo == TYPE_GC_HANDLE_CREATED[_BT]
 246  *      [handle_type: uleb128] GC handle type (System.Runtime.InteropServices.GCHandleType)
 247  *      upper bits reserved as flags
 248  *      [handle: uleb128] GC handle value
 249  *      [objaddr: sleb128] object pointer differences from obj_base
 250  *      If exinfo == TYPE_GC_HANDLE_CREATED_BT, a backtrace follows.
 251  * if exinfo == TYPE_GC_HANDLE_DESTROYED[_BT]
 252  *      [handle_type: uleb128] GC handle type (System.Runtime.InteropServices.GCHandleType)
 253  *      upper bits reserved as flags
 254  *      [handle: uleb128] GC handle value
 255  *      If exinfo == TYPE_GC_HANDLE_DESTROYED_BT, a backtrace follows.
 256  * if exinfo == TYPE_GC_FINALIZE_OBJECT_{START,END}
 257  *      [object: sleb128] the object as a difference from obj_base
 258  *
 259  * type metadata format:
 260  * type: TYPE_METADATA
 261  * exinfo: one of: TYPE_END_LOAD, TYPE_END_UNLOAD (optional for TYPE_THREAD and TYPE_DOMAIN)
 262  * [mtype: byte] metadata type, one of: TYPE_CLASS, TYPE_IMAGE, TYPE_ASSEMBLY, TYPE_DOMAIN,
 263  * TYPE_THREAD, TYPE_CONTEXT
 264  * [pointer: sleb128] pointer of the metadata type depending on mtype
 265  * if mtype == TYPE_CLASS
 266  *      [image: sleb128] MonoImage* as a pointer difference from ptr_base
 267  *      [name: string] full class name
 268  * if mtype == TYPE_IMAGE
 269  *      [name: string] image file name
 270  * if mtype == TYPE_ASSEMBLY
 271  *      [image: sleb128] MonoImage* as a pointer difference from ptr_base
 272  *      [name: string] assembly name
 273  * if mtype == TYPE_DOMAIN && exinfo == 0
 274  *      [name: string] domain friendly name
 275  * if mtype == TYPE_CONTEXT
 276  *      [domain: sleb128] domain id as pointer
 277  * if mtype == TYPE_THREAD && exinfo == 0
 278  *      [name: string] thread name
 279  *
 280  * type method format:
 281  * type: TYPE_METHOD
 282  * exinfo: one of: TYPE_LEAVE, TYPE_ENTER, TYPE_EXC_LEAVE, TYPE_JIT
 283  * [method: sleb128] MonoMethod* as a pointer difference from the last such
 284  * pointer or the buffer method_base
 285  * if exinfo == TYPE_JIT
 286  *      [code address: sleb128] pointer to the native code as a diff from ptr_base
 287  *      [code size: uleb128] size of the generated code
 288  *      [name: string] full method name
 289  *
 290  * type exception format:
 291  * type: TYPE_EXCEPTION
 292  * exinfo: zero, TYPE_CLAUSE, or TYPE_THROW_BT
 293  * if exinfo == TYPE_CLAUSE
 294  *      [clause type: byte] MonoExceptionEnum enum value
 295  *      [clause index: uleb128] index of the current clause
 296  *      [method: sleb128] MonoMethod* as a pointer difference from the last such
 297  *      pointer or the buffer method_base
 298  *      [object: sleb128] the exception object as a difference from obj_base
 299  * else
 300  *      [object: sleb128] the exception object as a difference from obj_base
 301  *      If exinfo == TYPE_THROW_BT, a backtrace follows.
 302  *
 303  * type runtime format:
 304  * type: TYPE_RUNTIME
 305  * exinfo: one of: TYPE_JITHELPER
 306  * if exinfo == TYPE_JITHELPER
 307  *      [type: byte] MonoProfilerCodeBufferType enum value
 308  *      [buffer address: sleb128] pointer to the native code as a diff from ptr_base
 309  *      [buffer size: uleb128] size of the generated code
 310  *      if type == MONO_PROFILER_CODE_BUFFER_SPECIFIC_TRAMPOLINE
 311  *              [name: string] buffer description name
 312  *
 313  * type monitor format:
 314  * type: TYPE_MONITOR
 315  * exinfo: zero or TYPE_MONITOR_BT
 316  * [type: byte] MONO_PROFILER_MONITOR_{CONTENTION,FAIL,DONE}
 317  * [object: sleb128] the lock object as a difference from obj_base
 318  * If exinfo == TYPE_MONITOR_BT, a backtrace follows.
 319  *
 320  * type heap format
 321  * type: TYPE_HEAP
 322  * exinfo: one of TYPE_HEAP_START, TYPE_HEAP_END, TYPE_HEAP_OBJECT, TYPE_HEAP_ROOT
 323  * if exinfo == TYPE_HEAP_OBJECT
 324  *      [object: sleb128] the object as a difference from obj_base
 325  *      [class: sleb128] the object MonoClass* as a difference from ptr_base
 326  *      [size: uleb128] size of the object on the heap
 327  *      [num_refs: uleb128] number of object references
 328  *      each referenced objref is preceded by a uleb128 encoded offset: the
 329  *      first offset is from the object address and each next offset is relative
 330  *      to the previous one
 331  *      [objrefs: sleb128]+ object referenced as a difference from obj_base
 332  *      The same object can appear multiple times, but only the first time
 333  *      with size != 0: in the other cases this data will only be used to
 334  *      provide additional referenced objects.
 335  * if exinfo == TYPE_HEAP_ROOT
 336  *      [num_roots: uleb128] number of root references
 337  *      [num_gc: uleb128] number of major gcs
 338  *      [object: sleb128] the object as a difference from obj_base
 339  *      [root_type: byte] the root_type: MonoProfileGCRootType (profiler.h)
 340  *      [extra_info: uleb128] the extra_info value
 341  *      object, root_type and extra_info are repeated num_roots times
 342  *
 343  * type sample format
 344  * type: TYPE_SAMPLE
 345  * exinfo: one of TYPE_SAMPLE_HIT, TYPE_SAMPLE_USYM, TYPE_SAMPLE_UBIN, TYPE_SAMPLE_COUNTERS_DESC, TYPE_SAMPLE_COUNTERS
 346  * if exinfo == TYPE_SAMPLE_HIT
 347  *      [thread: sleb128] thread id as difference from ptr_base
 348  *      [count: uleb128] number of following instruction addresses
 349  *      [ip: sleb128]* instruction pointer as difference from ptr_base
 350  *      [mbt_count: uleb128] number of managed backtrace frames
 351  *      [method: sleb128]* MonoMethod* as a pointer difference from the last such
 352  *      pointer or the buffer method_base (the first such method can be also indentified by ip, but this is not neccessarily true)
 353  * if exinfo == TYPE_SAMPLE_USYM
 354  *      [address: sleb128] symbol address as a difference from ptr_base
 355  *      [size: uleb128] symbol size (may be 0 if unknown)
 356  *      [name: string] symbol name
 357  * if exinfo == TYPE_SAMPLE_UBIN
 358  *      [address: sleb128] address where binary has been loaded as a difference from ptr_base
 359  *      [offset: uleb128] file offset of mapping (the same file can be mapped multiple times)
 360  *      [size: uleb128] memory size
 361  *      [name: string] binary name
 362  * if exinfo == TYPE_SAMPLE_COUNTERS_DESC
 363  *      [len: uleb128] number of counters
 364  *      for i = 0 to len
 365  *              [section: uleb128] section of counter
 366  *              if section == MONO_COUNTER_PERFCOUNTERS:
 367  *                      [section_name: string] section name of counter
 368  *              [name: string] name of counter
 369  *              [type: byte] type of counter
 370  *              [unit: byte] unit of counter
 371  *              [variance: byte] variance of counter
 372  *              [index: uleb128] unique index of counter
 373  * if exinfo == TYPE_SAMPLE_COUNTERS
 374  *      while true:
 375  *              [index: uleb128] unique index of counter
 376  *              if index == 0:
 377  *                      break
 378  *              [type: byte] type of counter value
 379  *              if type == string:
 380  *                      if value == null:
 381  *                              [0: byte] 0 -> value is null
 382  *                      else:
 383  *                              [1: byte] 1 -> value is not null
 384  *                              [value: string] counter value
 385  *              else:
 386  *                      [value: uleb128/sleb128/double] counter value, can be sleb128, uleb128 or double (determined by using type)
 387  *
 388  * type coverage format
 389  * type: TYPE_COVERAGE
 390  * exinfo: one of TYPE_COVERAGE_METHOD, TYPE_COVERAGE_STATEMENT, TYPE_COVERAGE_ASSEMBLY, TYPE_COVERAGE_CLASS
 391  * if exinfo == TYPE_COVERAGE_METHOD
 392  *  [assembly: string] name of assembly
 393  *  [class: string] name of the class
 394  *  [name: string] name of the method
 395  *  [signature: string] the signature of the method
 396  *  [filename: string] the file path of the file that contains this method
 397  *  [token: uleb128] the method token
 398  *  [method_id: uleb128] an ID for this data to associate with the buffers of TYPE_COVERAGE_STATEMENTS
 399  *  [len: uleb128] the number of TYPE_COVERAGE_BUFFERS associated with this method
 400  * if exinfo == TYPE_COVERAGE_STATEMENTS
 401  *  [method_id: uleb128] an the TYPE_COVERAGE_METHOD buffer to associate this with
 402  *  [offset: uleb128] the il offset relative to the previous offset
 403  *  [counter: uleb128] the counter for this instruction
 404  *  [line: uleb128] the line of filename containing this instruction
 405  *  [column: uleb128] the column containing this instruction
 406  * if exinfo == TYPE_COVERAGE_ASSEMBLY
 407  *  [name: string] assembly name
 408  *  [guid: string] assembly GUID
 409  *  [filename: string] assembly filename
 410  *  [number_of_methods: uleb128] the number of methods in this assembly
 411  *  [fully_covered: uleb128] the number of fully covered methods
 412  *  [partially_covered: uleb128] the number of partially covered methods
 413  *    currently partially_covered will always be 0, and fully_covered is the
 414  *    number of methods that are fully and partially covered.
 415  * if exinfo == TYPE_COVERAGE_CLASS
 416  *  [name: string] assembly name
 417  *  [class: string] class name
 418  *  [number_of_methods: uleb128] the number of methods in this class
 419  *  [fully_covered: uleb128] the number of fully covered methods
 420  *  [partially_covered: uleb128] the number of partially covered methods
 421  *    currently partially_covered will always be 0, and fully_covered is the
 422  *    number of methods that are fully and partially covered.
 423  *
 424  * type meta format:
 425  * type: TYPE_META
 426  * exinfo: one of: TYPE_SYNC_POINT
 427  * if exinfo == TYPE_SYNC_POINT
 428  *      [type: byte] MonoProfilerSyncPointType enum value
 429  */
 430
 431 // Pending data to be written to the log, for a single thread.
 432 // Threads periodically flush their own LogBuffers by calling safe_send
 433 typedef struct _LogBuffer LogBuffer;
 434 struct _LogBuffer {
 435         // Next (older) LogBuffer in processing queue
 436         LogBuffer *next;
 437
 438         uint64_t time_base;
 439         uint64_t last_time;
 440         uintptr_t ptr_base;
 441         uintptr_t method_base;
 442         uintptr_t last_method;
 443         uintptr_t obj_base;
 444         uintptr_t thread_id;
 445
 446         // Bytes allocated for this LogBuffer
 447         int size;
 448
 449         // Start of currently unused space in buffer
 450         unsigned char* cursor;
 451
 452         // Pointer to start-of-structure-plus-size (for convenience)
 453         unsigned char* buf_end;
 454
 455         // Start of data in buffer. Contents follow "buffer format" described above.
 456         unsigned char buf [1];
 457 };
 458
 459 typedef struct {
 460         MonoLinkedListSetNode node;
 461
 462         // Convenience pointer to the profiler structure.
 463         MonoProfiler *profiler;
 464
 465         // Was this thread added to the LLS?
 466         gboolean attached;
 467
 468         // The current log buffer for this thread.
 469         LogBuffer *buffer;
 470
 471         // Methods referenced by events in `buffer`, see `MethodInfo`.
 472         GPtrArray *methods;
 473
 474         // Current call depth for enter/leave events.
 475         int call_depth;
 476
 477         // Indicates whether this thread is currently writing to its `buffer`.
 478         gboolean busy;
 479
 480         // Has this thread written a thread end event to `buffer`?
 481         gboolean ended;
 482 } MonoProfilerThread;
 483
 484 static uintptr_t
 485 thread_id (void)
 486 {
 487         return (uintptr_t) mono_native_thread_id_get ();
 488 }
 489
 490 static uintptr_t
 491 process_id (void)
 492 {
 493 #ifdef HOST_WIN32
 494         return (uintptr_t) GetCurrentProcessId ();
 495 #else
 496         return (uintptr_t) getpid ();
 497 #endif
 498 }
 499
 500 #ifdef __APPLE__
 501 static mach_timebase_info_data_t timebase_info;
 502 #elif defined (HOST_WIN32)
 503 static LARGE_INTEGER pcounter_freq;
 504 #endif
 505
 506 #define TICKS_PER_SEC 1000000000LL
 507
 508 static uint64_t
 509 current_time (void)
 510 {
 511 #ifdef __APPLE__
 512         uint64_t time = mach_absolute_time ();
 513
 514         time *= timebase_info.numer;
 515         time /= timebase_info.denom;
 516
 517         return time;
 518 #elif defined (HOST_WIN32)
 519         LARGE_INTEGER value;
 520
 521         QueryPerformanceCounter (&value);
 522
 523         return value.QuadPart * TICKS_PER_SEC / pcounter_freq.QuadPart;
 524 #elif defined (CLOCK_MONOTONIC)
 525         struct timespec tspec;
 526
 527         clock_gettime (CLOCK_MONOTONIC, &tspec);
 528
 529         return ((uint64_t) tspec.tv_sec * TICKS_PER_SEC + tspec.tv_nsec);
 530 #else
 531         struct timeval tv;
 532
 533         gettimeofday (&tv, NULL);
 534
 535         return ((uint64_t) tv.tv_sec * TICKS_PER_SEC + tv.tv_usec * 1000);
 536 #endif
 537 }
 538
 539 static int timer_overhead;
 540
 541 static void
 542 init_time (void)
 543 {
 544 #ifdef __APPLE__
 545         mach_timebase_info (&timebase_info);
 546 #elif defined (HOST_WIN32)
 547         QueryPerformanceFrequency (&pcounter_freq);
 548 #endif
 549
 550         uint64_t time_start = current_time ();
 551
 552         for (int i = 0; i < 256; ++i)
 553                 current_time ();
 554
 555         uint64_t time_end = current_time ();
 556
 557         timer_overhead = (time_end - time_start) / 256;
 558 }
 559
 560 /*
 561  * These macros should be used when writing an event to a log buffer. They take
 562  * care of a bunch of stuff that can be repetitive and error-prone, such as
 563  * acquiring/releasing the buffer lock, incrementing the event counter,
 564  * expanding the log buffer, processing requests, etc. They also create a scope
 565  * so that it's harder to leak the LogBuffer pointer, which can be problematic
 566  * as the pointer is unstable when the buffer lock isn't acquired.
 567  */
 568
 569 #define ENTER_LOG(COUNTER, BUFFER, SIZE) \
 570         do { \
 571                 MonoProfilerThread *thread__ = PROF_TLS_GET (); \
 572                 if (thread__->attached) \
 573                         buffer_lock (); \
 574                 g_assert (!thread__->busy && "Why are we trying to write a new event while already writing one?"); \
 575                 thread__->busy = TRUE; \
 576                 InterlockedIncrement ((COUNTER)); \
 577                 LogBuffer *BUFFER = ensure_logbuf_unsafe (thread__, (SIZE))
 578
 579 #define EXIT_LOG_EXPLICIT(SEND) \
 580                 thread__->busy = FALSE; \
 581                 if ((SEND)) \
 582                         send_log_unsafe (TRUE); \
 583                 if (thread__->attached) \
 584                         buffer_unlock (); \
 585         } while (0)
 586
 587 // Pass these to EXIT_LOG_EXPLICIT () for easier reading.
 588 #define DO_SEND TRUE
 589 #define NO_SEND FALSE
 590
 591 #define EXIT_LOG EXIT_LOG_EXPLICIT (DO_SEND)
 592
 593 static volatile gint32 buffer_rwlock_count;
 594 static volatile gpointer buffer_rwlock_exclusive;
 595
 596 // Can be used recursively.
 597 static void
 598 buffer_lock (void)
 599 {
 600         /*
 601          * If the thread holding the exclusive lock tries to modify the
 602          * reader count, just make it a no-op. This way, we also avoid
 603          * invoking the GC safe point macros below, which could break if
 604          * done from a thread that is currently the initiator of STW.
 605          *
 606          * In other words, we rely on the fact that the GC thread takes
 607          * the exclusive lock in the gc_event () callback when the world
 608          * is about to stop.
 609          */
 610         if (InterlockedReadPointer (&buffer_rwlock_exclusive) != (gpointer) thread_id ()) {
 611                 MONO_ENTER_GC_SAFE;
 612
 613                 while (InterlockedReadPointer (&buffer_rwlock_exclusive))
 614                         mono_thread_info_yield ();
 615
 616                 InterlockedIncrement (&buffer_rwlock_count);
 617
 618                 MONO_EXIT_GC_SAFE;
 619         }
 620
 621         mono_memory_barrier ();
 622 }
 623
 624 static void
 625 buffer_unlock (void)
 626 {
 627         mono_memory_barrier ();
 628
 629         // See the comment in buffer_lock ().
 630         if (InterlockedReadPointer (&buffer_rwlock_exclusive) == (gpointer) thread_id ())
 631                 return;
 632
 633         g_assert (InterlockedRead (&buffer_rwlock_count) && "Why are we trying to decrement a zero reader count?");
 634
 635         InterlockedDecrement (&buffer_rwlock_count);
 636 }
 637
 638 // Cannot be used recursively.
 639 static void
 640 buffer_lock_excl (void)
 641 {
 642         gpointer tid = (gpointer) thread_id ();
 643
 644         g_assert (InterlockedReadPointer (&buffer_rwlock_exclusive) != tid && "Why are we taking the exclusive lock twice?");
 645
 646         MONO_ENTER_GC_SAFE;
 647
 648         while (InterlockedCompareExchangePointer (&buffer_rwlock_exclusive, tid, 0))
 649                 mono_thread_info_yield ();
 650
 651         while (InterlockedRead (&buffer_rwlock_count))
 652                 mono_thread_info_yield ();
 653
 654         MONO_EXIT_GC_SAFE;
 655
 656         mono_memory_barrier ();
 657 }
 658
 659 static void
 660 buffer_unlock_excl (void)
 661 {
 662         mono_memory_barrier ();
 663
 664         g_assert (InterlockedReadPointer (&buffer_rwlock_exclusive) && "Why is the exclusive lock not held?");
 665         g_assert (InterlockedReadPointer (&buffer_rwlock_exclusive) == (gpointer) thread_id () && "Why does another thread hold the exclusive lock?");
 666         g_assert (!InterlockedRead (&buffer_rwlock_count) && "Why are there readers when the exclusive lock is held?");
 667
 668         InterlockedWritePointer (&buffer_rwlock_exclusive, NULL);
 669 }
 670
 671 typedef struct _BinaryObject BinaryObject;
 672 struct _BinaryObject {
 673         BinaryObject *next;
 674         void *addr;
 675         char *name;
 676 };
 677
 678 struct _MonoProfiler {
 679         FILE* file;
 680 #if defined (HAVE_SYS_ZLIB)
 681         gzFile gzfile;
 682 #endif
 683         char *args;
 684         uint64_t startup_time;
 685         int pipe_output;
 686         int command_port;
 687         int server_socket;
 688         int pipes [2];
 689         MonoNativeThreadId helper_thread;
 690         MonoNativeThreadId writer_thread;
 691         MonoNativeThreadId dumper_thread;
 692         volatile gint32 run_writer_thread;
 693         MonoLockFreeAllocSizeClass writer_entry_size_class;
 694         MonoLockFreeAllocator writer_entry_allocator;
 695         MonoLockFreeQueue writer_queue;
 696         MonoSemType writer_queue_sem;
 697         MonoConcurrentHashTable *method_table;
 698         mono_mutex_t method_table_mutex;
 699         volatile gint32 run_dumper_thread;
 700         MonoLockFreeQueue dumper_queue;
 701         MonoSemType dumper_queue_sem;
 702         MonoLockFreeAllocSizeClass sample_size_class;
 703         MonoLockFreeAllocator sample_allocator;
 704         MonoLockFreeQueue sample_reuse_queue;
 705         BinaryObject *binary_objects;
 706         GPtrArray *coverage_filters;
 707 };
 708
 709 typedef struct {
 710         MonoLockFreeQueueNode node;
 711         GPtrArray *methods;
 712         LogBuffer *buffer;
 713 } WriterQueueEntry;
 714
 715 #define WRITER_ENTRY_BLOCK_SIZE (mono_pagesize ())
 716
 717 typedef struct {
 718         MonoMethod *method;
 719         MonoJitInfo *ji;
 720         uint64_t time;
 721 } MethodInfo;
 722
 723 #ifdef HOST_WIN32
 724
 725 #define PROF_TLS_SET(VAL) (TlsSetValue (profiler_tls, (VAL)))
 726 #define PROF_TLS_GET() ((MonoProfilerThread *) TlsGetValue (profiler_tls))
 727 #define PROF_TLS_INIT() (profiler_tls = TlsAlloc ())
 728 #define PROF_TLS_FREE() (TlsFree (profiler_tls))
 729
 730 static DWORD profiler_tls;
 731
 732 #elif HAVE_KW_THREAD
 733
 734 #define PROF_TLS_SET(VAL) (profiler_tls = (VAL))
 735 #define PROF_TLS_GET() (profiler_tls)
 736 #define PROF_TLS_INIT()
 737 #define PROF_TLS_FREE()
 738
 739 static __thread MonoProfilerThread *profiler_tls;
 740
 741 #else
 742
 743 #define PROF_TLS_SET(VAL) (pthread_setspecific (profiler_tls, (VAL)))
 744 #define PROF_TLS_GET() ((MonoProfilerThread *) pthread_getspecific (profiler_tls))
 745 #define PROF_TLS_INIT() (pthread_key_create (&profiler_tls, NULL))
 746 #define PROF_TLS_FREE() (pthread_key_delete (profiler_tls))
 747
 748 static pthread_key_t profiler_tls;
 749
 750 #endif
 751
 752 static char*
 753 pstrdup (const char *s)
 754 {
 755         int len = strlen (s) + 1;
 756         char *p = (char *) g_malloc (len);
 757         memcpy (p, s, len);
 758         return p;
 759 }
 760
 761 static void *
 762 alloc_buffer (int size)
 763 {
 764         return mono_valloc (NULL, size, MONO_MMAP_READ | MONO_MMAP_WRITE | MONO_MMAP_ANON | MONO_MMAP_PRIVATE, MONO_MEM_ACCOUNT_PROFILER);
 765 }
 766
 767 static void
 768 free_buffer (void *buf, int size)
 769 {
 770         mono_vfree (buf, size, MONO_MEM_ACCOUNT_PROFILER);
 771 }
 772
 773 static LogBuffer*
 774 create_buffer (uintptr_t tid)
 775 {
 776         LogBuffer* buf = (LogBuffer *) alloc_buffer (BUFFER_SIZE);
 777
 778         InterlockedIncrement (&buffer_allocations_ctr);
 779
 780         buf->size = BUFFER_SIZE;
 781         buf->time_base = current_time ();
 782         buf->last_time = buf->time_base;
 783         buf->buf_end = (unsigned char *) buf + buf->size;
 784         buf->cursor = buf->buf;
 785         buf->thread_id = tid;
 786
 787         return buf;
 788 }
 789
 790 /*
 791  * Must be called with the reader lock held if thread is the current thread, or
 792  * the exclusive lock if thread is a different thread. However, if thread is
 793  * the current thread, and init_thread () was called with add_to_lls = FALSE,
 794  * then no locking is necessary.
 795  */
 796 static void
 797 init_buffer_state (MonoProfilerThread *thread)
 798 {
 799         thread->buffer = create_buffer (thread->node.key);
 800         thread->methods = NULL;
 801 }
 802
 803 static void
 804 clear_hazard_pointers (MonoThreadHazardPointers *hp)
 805 {
 806         mono_hazard_pointer_clear (hp, 0);
 807         mono_hazard_pointer_clear (hp, 1);
 808         mono_hazard_pointer_clear (hp, 2);
 809 }
 810
 811 static MonoProfilerThread *
 812 init_thread (MonoProfiler *prof, gboolean add_to_lls)
 813 {
 814         MonoProfilerThread *thread = PROF_TLS_GET ();
 815
 816         /*
 817          * Sometimes we may try to initialize a thread twice. One example is the
 818          * main thread: We initialize it when setting up the profiler, but we will
 819          * also get a thread_start () callback for it. Another example is when
 820          * attaching new threads to the runtime: We may get a gc_alloc () callback
 821          * for that thread's thread object (where we initialize it), soon followed
 822          * by a thread_start () callback.
 823          *
 824          * These cases are harmless anyhow. Just return if we've already done the
 825          * initialization work.
 826          */
 827         if (thread)
 828                 return thread;
 829
 830         thread = g_malloc (sizeof (MonoProfilerThread));
 831         thread->node.key = thread_id ();
 832         thread->profiler = prof;
 833         thread->attached = add_to_lls;
 834         thread->call_depth = 0;
 835         thread->busy = 0;
 836         thread->ended = FALSE;
 837
 838         init_buffer_state (thread);
 839
 840         /*
 841          * Some internal profiler threads don't need to be cleaned up
 842          * by the main thread on shutdown.
 843          */
 844         if (add_to_lls) {
 845                 MonoThreadHazardPointers *hp = mono_hazard_pointer_get ();
 846                 g_assert (mono_lls_insert (&profiler_thread_list, hp, &thread->node) && "Why can't we insert the thread in the LLS?");
 847                 clear_hazard_pointers (hp);
 848         }
 849
 850         PROF_TLS_SET (thread);
 851
 852         return thread;
 853 }
 854
 855 // Only valid if init_thread () was called with add_to_lls = FALSE.
 856 static void
 857 deinit_thread (MonoProfilerThread *thread)
 858 {
 859         g_assert (!thread->attached && "Why are we manually freeing an attached thread?");
 860
 861         g_free (thread);
 862         PROF_TLS_SET (NULL);
 863 }
 864
 865 // Only valid if init_thread () was called with add_to_lls = FALSE.
 866 static LogBuffer *
 867 ensure_logbuf_unsafe (MonoProfilerThread *thread, int bytes)
 868 {
 869         LogBuffer *old = thread->buffer;
 870
 871         if (old && old->cursor + bytes + 100 < old->buf_end)
 872                 return old;
 873
 874         LogBuffer *new_ = create_buffer (thread->node.key);
 875         new_->next = old;
 876         thread->buffer = new_;
 877
 878         return new_;
 879 }
 880
 881 static void
 882 encode_uleb128 (uint64_t value, uint8_t *buf, uint8_t **endbuf)
 883 {
 884         uint8_t *p = buf;
 885
 886         do {
 887                 uint8_t b = value & 0x7f;
 888                 value >>= 7;
 889
 890                 if (value != 0) /* more bytes to come */
 891                         b |= 0x80;
 892
 893                 *p ++ = b;
 894         } while (value);
 895
 896         *endbuf = p;
 897 }
 898
 899 static void
 900 encode_sleb128 (intptr_t value, uint8_t *buf, uint8_t **endbuf)
 901 {
 902         int more = 1;
 903         int negative = (value < 0);
 904         unsigned int size = sizeof (intptr_t) * 8;
 905         uint8_t byte;
 906         uint8_t *p = buf;
 907
 908         while (more) {
 909                 byte = value & 0x7f;
 910                 value >>= 7;
 911
 912                 /* the following is unnecessary if the
 913                  * implementation of >>= uses an arithmetic rather
 914                  * than logical shift for a signed left operand
 915                  */
 916                 if (negative)
 917                         /* sign extend */
 918                         value |= - ((intptr_t) 1 <<(size - 7));
 919
 920                 /* sign bit of byte is second high order bit (0x40) */
 921                 if ((value == 0 && !(byte & 0x40)) ||
 922                     (value == -1 && (byte & 0x40)))
 923                         more = 0;
 924                 else
 925                         byte |= 0x80;
 926
 927                 *p ++= byte;
 928         }
 929
 930         *endbuf = p;
 931 }
 932
 933 static void
 934 emit_byte (LogBuffer *logbuffer, int value)
 935 {
 936         logbuffer->cursor [0] = value;
 937         logbuffer->cursor++;
 938
 939         g_assert (logbuffer->cursor <= logbuffer->buf_end && "Why are we writing past the buffer end?");
 940 }
 941
 942 static void
 943 emit_value (LogBuffer *logbuffer, int value)
 944 {
 945         encode_uleb128 (value, logbuffer->cursor, &logbuffer->cursor);
 946
 947         g_assert (logbuffer->cursor <= logbuffer->buf_end && "Why are we writing past the buffer end?");
 948 }
 949
 950 static void
 951 emit_time (LogBuffer *logbuffer, uint64_t value)
 952 {
 953         uint64_t tdiff = value - logbuffer->last_time;
 954         encode_uleb128 (tdiff, logbuffer->cursor, &logbuffer->cursor);
 955         logbuffer->last_time = value;
 956
 957         g_assert (logbuffer->cursor <= logbuffer->buf_end && "Why are we writing past the buffer end?");
 958 }
 959
 960 static void
 961 emit_event_time (LogBuffer *logbuffer, int event, uint64_t time)
 962 {
 963         emit_byte (logbuffer, event);
 964         emit_time (logbuffer, time);
 965 }
 966
 967 static void
 968 emit_event (LogBuffer *logbuffer, int event)
 969 {
 970         emit_event_time (logbuffer, event, current_time ());
 971 }
 972
 973 static void
 974 emit_svalue (LogBuffer *logbuffer, int64_t value)
 975 {
 976         encode_sleb128 (value, logbuffer->cursor, &logbuffer->cursor);
 977
 978         g_assert (logbuffer->cursor <= logbuffer->buf_end && "Why are we writing past the buffer end?");
 979 }
 980
 981 static void
 982 emit_uvalue (LogBuffer *logbuffer, uint64_t value)
 983 {
 984         encode_uleb128 (value, logbuffer->cursor, &logbuffer->cursor);
 985
 986         g_assert (logbuffer->cursor <= logbuffer->buf_end && "Why are we writing past the buffer end?");
 987 }
 988
 989 static void
 990 emit_ptr (LogBuffer *logbuffer, void *ptr)
 991 {
 992         if (!logbuffer->ptr_base)
 993                 logbuffer->ptr_base = (uintptr_t) ptr;
 994
 995         emit_svalue (logbuffer, (intptr_t) ptr - logbuffer->ptr_base);
 996
 997         g_assert (logbuffer->cursor <= logbuffer->buf_end && "Why are we writing past the buffer end?");
 998 }
 999
1000 static void
1001 emit_method_inner (LogBuffer *logbuffer, void *method)
1002 {
1003         if (!logbuffer->method_base) {
1004                 logbuffer->method_base = (intptr_t) method;
1005                 logbuffer->last_method = (intptr_t) method;
1006         }
1007
1008         encode_sleb128 ((intptr_t) ((char *) method - (char *) logbuffer->last_method), logbuffer->cursor, &logbuffer->cursor);
1009         logbuffer->last_method = (intptr_t) method;
1010
1011         g_assert (logbuffer->cursor <= logbuffer->buf_end && "Why are we writing past the buffer end?");
1012 }
1013
1014 static void
1015 register_method_local (MonoMethod *method, MonoJitInfo *ji)
1016 {
1017         MonoProfilerThread *thread = PROF_TLS_GET ();
1018
1019         if (!mono_conc_hashtable_lookup (thread->profiler->method_table, method)) {
1020                 MethodInfo *info = (MethodInfo *) g_malloc (sizeof (MethodInfo));
1021
1022                 info->method = method;
1023                 info->ji = ji;
1024                 info->time = current_time ();
1025
1026                 GPtrArray *arr = thread->methods ? thread->methods : (thread->methods = g_ptr_array_new ());
1027                 g_ptr_array_add (arr, info);
1028         }
1029 }
1030
1031 static void
1032 emit_method (LogBuffer *logbuffer, MonoMethod *method)
1033 {
1034         register_method_local (method, NULL);
1035         emit_method_inner (logbuffer, method);
1036 }
1037
1038 static void
1039 emit_obj (LogBuffer *logbuffer, void *ptr)
1040 {
1041         if (!logbuffer->obj_base)
1042                 logbuffer->obj_base = (uintptr_t) ptr >> 3;
1043
1044         emit_svalue (logbuffer, ((uintptr_t) ptr >> 3) - logbuffer->obj_base);
1045
1046         g_assert (logbuffer->cursor <= logbuffer->buf_end && "Why are we writing past the buffer end?");
1047 }
1048
1049 static void
1050 emit_string (LogBuffer *logbuffer, const char *str, size_t size)
1051 {
1052         size_t i = 0;
1053         if (str) {
1054                 for (; i < size; i++) {
1055                         if (str[i] == '\0')
1056                                 break;
1057                         emit_byte (logbuffer, str [i]);
1058                 }
1059         }
1060         emit_byte (logbuffer, '\0');
1061 }
1062
1063 static void
1064 emit_double (LogBuffer *logbuffer, double value)
1065 {
1066         int i;
1067         unsigned char buffer[8];
1068         memcpy (buffer, &value, 8);
1069 #if G_BYTE_ORDER == G_BIG_ENDIAN
1070         for (i = 7; i >= 0; i--)
1071 #else
1072         for (i = 0; i < 8; i++)
1073 #endif
1074                 emit_byte (logbuffer, buffer[i]);
1075 }
1076
1077 static char*
1078 write_int16 (char *buf, int32_t value)
1079 {
1080         int i;
1081         for (i = 0; i < 2; ++i) {
1082                 buf [i] = value;
1083                 value >>= 8;
1084         }
1085         return buf + 2;
1086 }
1087
1088 static char*
1089 write_int32 (char *buf, int32_t value)
1090 {
1091         int i;
1092         for (i = 0; i < 4; ++i) {
1093                 buf [i] = value;
1094                 value >>= 8;
1095         }
1096         return buf + 4;
1097 }
1098
1099 static char*
1100 write_int64 (char *buf, int64_t value)
1101 {
1102         int i;
1103         for (i = 0; i < 8; ++i) {
1104                 buf [i] = value;
1105                 value >>= 8;
1106         }
1107         return buf + 8;
1108 }
1109
1110 static char *
1111 write_header_string (char *p, const char *str)
1112 {
1113         size_t len = strlen (str) + 1;
1114
1115         p = write_int32 (p, len);
1116         strcpy (p, str);
1117
1118         return p + len;
1119 }
1120
1121 static void
1122 dump_header (MonoProfiler *profiler)
1123 {
1124         const char *args = profiler->args;
1125         const char *arch = mono_config_get_cpu ();
1126         const char *os = mono_config_get_os ();
1127
1128         char *hbuf = g_malloc (
1129                 sizeof (gint32) /* header id */ +
1130                 sizeof (gint8) /* major version */ +
1131                 sizeof (gint8) /* minor version */ +
1132                 sizeof (gint8) /* data version */ +
1133                 sizeof (gint8) /* word size */ +
1134                 sizeof (gint64) /* startup time */ +
1135                 sizeof (gint32) /* timer overhead */ +
1136                 sizeof (gint32) /* flags */ +
1137                 sizeof (gint32) /* process id */ +
1138                 sizeof (gint16) /* command port */ +
1139                 sizeof (gint32) + strlen (args) + 1 /* arguments */ +
1140                 sizeof (gint32) + strlen (arch) + 1 /* architecture */ +
1141                 sizeof (gint32) + strlen (os) + 1 /* operating system */
1142         );
1143         char *p = hbuf;
1144
1145         p = write_int32 (p, LOG_HEADER_ID);
1146         *p++ = LOG_VERSION_MAJOR;
1147         *p++ = LOG_VERSION_MINOR;
1148         *p++ = LOG_DATA_VERSION;
1149         *p++ = sizeof (void *);
1150         p = write_int64 (p, ((uint64_t) time (NULL)) * 1000);
1151         p = write_int32 (p, timer_overhead);
1152         p = write_int32 (p, 0); /* flags */
1153         p = write_int32 (p, process_id ());
1154         p = write_int16 (p, profiler->command_port);
1155         p = write_header_string (p, args);
1156         p = write_header_string (p, arch);
1157         p = write_header_string (p, os);
1158
1159 #if defined (HAVE_SYS_ZLIB)
1160         if (profiler->gzfile) {
1161                 gzwrite (profiler->gzfile, hbuf, p - hbuf);
1162         } else
1163 #endif
1164         {
1165                 fwrite (hbuf, p - hbuf, 1, profiler->file);
1166                 fflush (profiler->file);
1167         }
1168
1169         g_free (hbuf);
1170 }
1171
1172 /*
1173  * Must be called with the reader lock held if thread is the current thread, or
1174  * the exclusive lock if thread is a different thread. However, if thread is
1175  * the current thread, and init_thread () was called with add_to_lls = FALSE,
1176  * then no locking is necessary.
1177  */
1178 static void
1179 send_buffer (MonoProfilerThread *thread)
1180 {
1181         WriterQueueEntry *entry = mono_lock_free_alloc (&thread->profiler->writer_entry_allocator);
1182         entry->methods = thread->methods;
1183         entry->buffer = thread->buffer;
1184
1185         mono_lock_free_queue_node_init (&entry->node, FALSE);
1186
1187         mono_lock_free_queue_enqueue (&thread->profiler->writer_queue, &entry->node);
1188         mono_os_sem_post (&thread->profiler->writer_queue_sem);
1189 }
1190
1191 static void
1192 free_thread (gpointer p)
1193 {
1194         MonoProfilerThread *thread = p;
1195
1196         if (!thread->ended) {
1197                 /*
1198                  * The thread is being cleaned up by the main thread during
1199                  * shutdown. This typically happens for internal runtime
1200                  * threads. We need to synthesize a thread end event.
1201                  */
1202
1203                 InterlockedIncrement (&thread_ends_ctr);
1204
1205                 if (ENABLED (PROFLOG_THREAD_EVENTS)) {
1206                         LogBuffer *buf = ensure_logbuf_unsafe (thread,
1207                                 EVENT_SIZE /* event */ +
1208                                 BYTE_SIZE /* type */ +
1209                                 LEB128_SIZE /* tid */
1210                         );
1211
1212                         emit_event (buf, TYPE_END_UNLOAD | TYPE_METADATA);
1213                         emit_byte (buf, TYPE_THREAD);
1214                         emit_ptr (buf, (void *) thread->node.key);
1215                 }
1216         }
1217
1218         send_buffer (thread);
1219
1220         g_free (thread);
1221 }
1222
1223 static void
1224 remove_thread (MonoProfilerThread *thread)
1225 {
1226         MonoThreadHazardPointers *hp = mono_hazard_pointer_get ();
1227
1228         if (mono_lls_remove (&profiler_thread_list, hp, &thread->node))
1229                 mono_thread_hazardous_try_free (thread, free_thread);
1230
1231         clear_hazard_pointers (hp);
1232 }
1233
1234 static void
1235 dump_buffer (MonoProfiler *profiler, LogBuffer *buf)
1236 {
1237         char hbuf [128];
1238         char *p = hbuf;
1239
1240         if (buf->next)
1241                 dump_buffer (profiler, buf->next);
1242
1243         if (buf->cursor - buf->buf) {
1244                 p = write_int32 (p, BUF_ID);
1245                 p = write_int32 (p, buf->cursor - buf->buf);
1246                 p = write_int64 (p, buf->time_base);
1247                 p = write_int64 (p, buf->ptr_base);
1248                 p = write_int64 (p, buf->obj_base);
1249                 p = write_int64 (p, buf->thread_id);
1250                 p = write_int64 (p, buf->method_base);
1251
1252 #if defined (HAVE_SYS_ZLIB)
1253                 if (profiler->gzfile) {
1254                         gzwrite (profiler->gzfile, hbuf, p - hbuf);
1255                         gzwrite (profiler->gzfile, buf->buf, buf->cursor - buf->buf);
1256                 } else
1257 #endif
1258                 {
1259                         fwrite (hbuf, p - hbuf, 1, profiler->file);
1260                         fwrite (buf->buf, buf->cursor - buf->buf, 1, profiler->file);
1261                         fflush (profiler->file);
1262                 }
1263         }
1264
1265         free_buffer (buf, buf->size);
1266 }
1267
1268 static void
1269 dump_buffer_threadless (MonoProfiler *profiler, LogBuffer *buf)
1270 {
1271         for (LogBuffer *iter = buf; iter; iter = iter->next)
1272                 iter->thread_id = 0;
1273
1274         dump_buffer (profiler, buf);
1275 }
1276
1277 // Only valid if init_thread () was called with add_to_lls = FALSE.
1278 static void
1279 send_log_unsafe (gboolean if_needed)
1280 {
1281         MonoProfilerThread *thread = PROF_TLS_GET ();
1282
1283         if (!if_needed || (if_needed && thread->buffer->next)) {
1284                 if (!thread->attached)
1285                         for (LogBuffer *iter = thread->buffer; iter; iter = iter->next)
1286                                 iter->thread_id = 0;
1287
1288                 send_buffer (thread);
1289                 init_buffer_state (thread);
1290         }
1291 }
1292
1293 // Assumes that the exclusive lock is held.
1294 static void
1295 sync_point_flush (void)
1296 {
1297         g_assert (InterlockedReadPointer (&buffer_rwlock_exclusive) == (gpointer) thread_id () && "Why don't we hold the exclusive lock?");
1298
1299         MONO_LLS_FOREACH_SAFE (&profiler_thread_list, MonoProfilerThread, thread) {
1300                 g_assert (thread->attached && "Why is a thread in the LLS not attached?");
1301
1302                 send_buffer (thread);
1303                 init_buffer_state (thread);
1304         } MONO_LLS_FOREACH_SAFE_END
1305 }
1306
1307 // Assumes that the exclusive lock is held.
1308 static void
1309 sync_point_mark (MonoProfilerSyncPointType type)
1310 {
1311         g_assert (InterlockedReadPointer (&buffer_rwlock_exclusive) == (gpointer) thread_id () && "Why don't we hold the exclusive lock?");
1312
1313         ENTER_LOG (&sync_points_ctr, logbuffer,
1314                 EVENT_SIZE /* event */ +
1315                 LEB128_SIZE /* type */
1316         );
1317
1318         emit_event (logbuffer, TYPE_META | TYPE_SYNC_POINT);
1319         emit_byte (logbuffer, type);
1320
1321         EXIT_LOG_EXPLICIT (NO_SEND);
1322
1323         send_log_unsafe (FALSE);
1324 }
1325
1326 // Assumes that the exclusive lock is held.
1327 static void
1328 sync_point (MonoProfilerSyncPointType type)
1329 {
1330         sync_point_flush ();
1331         sync_point_mark (type);
1332 }
1333
1334 static int
1335 gc_reference (MonoObject *obj, MonoClass *klass, uintptr_t size, uintptr_t num, MonoObject **refs, uintptr_t *offsets, void *data)
1336 {
1337         /* account for object alignment in the heap */
1338         size += 7;
1339         size &= ~7;
1340
1341         ENTER_LOG (&heap_objects_ctr, logbuffer,
1342                 EVENT_SIZE /* event */ +
1343                 LEB128_SIZE /* obj */ +
1344                 LEB128_SIZE /* klass */ +
1345                 LEB128_SIZE /* size */ +
1346                 LEB128_SIZE /* num */ +
1347                 num * (
1348                         LEB128_SIZE /* offset */ +
1349                         LEB128_SIZE /* ref */
1350                 )
1351         );
1352
1353         emit_event (logbuffer, TYPE_HEAP_OBJECT | TYPE_HEAP);
1354         emit_obj (logbuffer, obj);
1355         emit_ptr (logbuffer, klass);
1356         emit_value (logbuffer, size);
1357         emit_value (logbuffer, num);
1358
1359         uintptr_t last_offset = 0;
1360
1361         for (int i = 0; i < num; ++i) {
1362                 emit_value (logbuffer, offsets [i] - last_offset);
1363                 last_offset = offsets [i];
1364                 emit_obj (logbuffer, refs [i]);
1365         }
1366
1367         EXIT_LOG_EXPLICIT (DO_SEND);
1368
1369         return 0;
1370 }
1371
1372 static unsigned int hs_mode_ms = 0;
1373 static unsigned int hs_mode_gc = 0;
1374 static unsigned int hs_mode_ondemand = 0;
1375 static unsigned int gc_count = 0;
1376 static uint64_t last_hs_time = 0;
1377 static gboolean do_heap_walk = FALSE;
1378 static gboolean ignore_heap_events;
1379
1380 static void
1381 gc_roots (MonoProfiler *prof, int num, void **objects, int *root_types, uintptr_t *extra_info)
1382 {
1383         if (ignore_heap_events)
1384                 return;
1385
1386         ENTER_LOG (&heap_roots_ctr, logbuffer,
1387                 EVENT_SIZE /* event */ +
1388                 LEB128_SIZE /* num */ +
1389                 LEB128_SIZE /* collections */ +
1390                 num * (
1391                         LEB128_SIZE /* object */ +
1392                         LEB128_SIZE /* root type */ +
1393                         LEB128_SIZE /* extra info */
1394                 )
1395         );
1396
1397         emit_event (logbuffer, TYPE_HEAP_ROOT | TYPE_HEAP);
1398         emit_value (logbuffer, num);
1399         emit_value (logbuffer, mono_gc_collection_count (mono_gc_max_generation ()));
1400
1401         for (int i = 0; i < num; ++i) {
1402                 emit_obj (logbuffer, objects [i]);
1403                 emit_byte (logbuffer, root_types [i]);
1404                 emit_value (logbuffer, extra_info [i]);
1405         }
1406
1407         EXIT_LOG_EXPLICIT (DO_SEND);
1408 }
1409
1410
1411 static void
1412 trigger_on_demand_heapshot (void)
1413 {
1414         if (heapshot_requested)
1415                 mono_gc_collect (mono_gc_max_generation ());
1416 }
1417
1418 #define ALL_GC_EVENTS_MASK (PROFLOG_GC_MOVES_EVENTS | PROFLOG_GC_ROOT_EVENTS | PROFLOG_GC_EVENTS | PROFLOG_HEAPSHOT_FEATURE)
1419
1420 static void
1421 gc_event (MonoProfiler *profiler, MonoGCEvent ev, int generation)
1422 {
1423         if (ev == MONO_GC_EVENT_START) {
1424                 uint64_t now = current_time ();
1425
1426                 if (hs_mode_ms && (now - last_hs_time) / 1000 * 1000 >= hs_mode_ms)
1427                         do_heap_walk = TRUE;
1428                 else if (hs_mode_gc && !(gc_count % hs_mode_gc))
1429                         do_heap_walk = TRUE;
1430                 else if (hs_mode_ondemand)
1431                         do_heap_walk = heapshot_requested;
1432                 else if (!hs_mode_ms && !hs_mode_gc && generation == mono_gc_max_generation ())
1433                         do_heap_walk = TRUE;
1434
1435                 //If using heapshot, ignore events for collections we don't care
1436                 if (ENABLED (PROFLOG_HEAPSHOT_FEATURE)) {
1437                         // Ignore events generated during the collection itself (IE GC ROOTS)
1438                         ignore_heap_events = !do_heap_walk;
1439                 }
1440         }
1441
1442
1443         if (ENABLED (PROFLOG_GC_EVENTS)) {
1444                 ENTER_LOG (&gc_events_ctr, logbuffer,
1445                         EVENT_SIZE /* event */ +
1446                         BYTE_SIZE /* gc event */ +
1447                         BYTE_SIZE /* generation */
1448                 );
1449
1450                 emit_event (logbuffer, TYPE_GC_EVENT | TYPE_GC);
1451                 emit_byte (logbuffer, ev);
1452                 emit_byte (logbuffer, generation);
1453
1454                 EXIT_LOG_EXPLICIT (NO_SEND);
1455         }
1456
1457         switch (ev) {
1458         case MONO_GC_EVENT_START:
1459                 if (generation == mono_gc_max_generation ())
1460                         gc_count++;
1461
1462                 break;
1463         case MONO_GC_EVENT_PRE_STOP_WORLD_LOCKED:
1464                 /*
1465                  * Ensure that no thread can be in the middle of writing to
1466                  * a buffer when the world stops...
1467                  */
1468                 buffer_lock_excl ();
1469                 break;
1470         case MONO_GC_EVENT_POST_STOP_WORLD:
1471                 /*
1472                  * ... So that we now have a consistent view of all buffers.
1473                  * This allows us to flush them. We need to do this because
1474                  * they may contain object allocation events that need to be
1475                  * committed to the log file before any object move events
1476                  * that will be produced during this GC.
1477                  */
1478                 if (ENABLED (ALL_GC_EVENTS_MASK))
1479                         sync_point (SYNC_POINT_WORLD_STOP);
1480
1481                 /*
1482                  * All heap events are surrounded by a HEAP_START and a HEAP_ENV event.
1483                  * Right now, that's the case for GC Moves, GC Roots or heapshots.
1484                  */
1485                 if (ENABLED (PROFLOG_GC_MOVES_EVENTS | PROFLOG_GC_ROOT_EVENTS) || do_heap_walk) {
1486                         ENTER_LOG (&heap_starts_ctr, logbuffer,
1487                                 EVENT_SIZE /* event */
1488                         );
1489
1490                         emit_event (logbuffer, TYPE_HEAP_START | TYPE_HEAP);
1491
1492                         EXIT_LOG_EXPLICIT (DO_SEND);
1493                 }
1494
1495                 break;
1496         case MONO_GC_EVENT_PRE_START_WORLD:
1497                 if (do_heap_shot && do_heap_walk)
1498                         mono_gc_walk_heap (0, gc_reference, NULL);
1499
1500                 /* Matching HEAP_END to the HEAP_START from above */
1501                 if (ENABLED (PROFLOG_GC_MOVES_EVENTS | PROFLOG_GC_ROOT_EVENTS) || do_heap_walk) {
1502                         ENTER_LOG (&heap_ends_ctr, logbuffer,
1503                                 EVENT_SIZE /* event */
1504                         );
1505
1506                         emit_event (logbuffer, TYPE_HEAP_END | TYPE_HEAP);
1507
1508                         EXIT_LOG_EXPLICIT (DO_SEND);
1509                 }
1510
1511                 if (do_heap_shot && do_heap_walk) {
1512                         do_heap_walk = FALSE;
1513                         heapshot_requested = 0;
1514                         last_hs_time = current_time ();
1515                 }
1516
1517                 /*
1518                  * Similarly, we must now make sure that any object moves
1519                  * written to the GC thread's buffer are flushed. Otherwise,
1520                  * object allocation events for certain addresses could come
1521                  * after the move events that made those addresses available.
1522                  */
1523                 if (ENABLED (ALL_GC_EVENTS_MASK))
1524                         sync_point_mark (SYNC_POINT_WORLD_START);
1525                 break;
1526         case MONO_GC_EVENT_POST_START_WORLD_UNLOCKED:
1527                 /*
1528                  * Finally, it is safe to allow other threads to write to
1529                  * their buffers again.
1530                  */
1531                 buffer_unlock_excl ();
1532                 break;
1533         default:
1534                 break;
1535         }
1536 }
1537
1538 static void
1539 gc_resize (MonoProfiler *profiler, int64_t new_size)
1540 {
1541         ENTER_LOG (&gc_resizes_ctr, logbuffer,
1542                 EVENT_SIZE /* event */ +
1543                 LEB128_SIZE /* new size */
1544         );
1545
1546         emit_event (logbuffer, TYPE_GC_RESIZE | TYPE_GC);
1547         emit_value (logbuffer, new_size);
1548
1549         EXIT_LOG_EXPLICIT (DO_SEND);
1550 }
1551
1552 typedef struct {
1553         int count;
1554         MonoMethod* methods [MAX_FRAMES];
1555         int32_t il_offsets [MAX_FRAMES];
1556         int32_t native_offsets [MAX_FRAMES];
1557 } FrameData;
1558
1559 static int num_frames = MAX_FRAMES;
1560
1561 static mono_bool
1562 walk_stack (MonoMethod *method, int32_t native_offset, int32_t il_offset, mono_bool managed, void* data)
1563 {
1564         FrameData *frame = (FrameData *)data;
1565         if (method && frame->count < num_frames) {
1566                 frame->il_offsets [frame->count] = il_offset;
1567                 frame->native_offsets [frame->count] = native_offset;
1568                 frame->methods [frame->count++] = method;
1569                 //printf ("In %d %s at %d (native: %d)\n", frame->count, mono_method_get_name (method), il_offset, native_offset);
1570         }
1571         return frame->count == num_frames;
1572 }
1573
1574 /*
1575  * a note about stack walks: they can cause more profiler events to fire,
1576  * so we need to make sure they don't happen after we started emitting an
1577  * event, hence the collect_bt/emit_bt split.
1578  */
1579 static void
1580 collect_bt (FrameData *data)
1581 {
1582         data->count = 0;
1583         mono_stack_walk_no_il (walk_stack, data);
1584 }
1585
1586 static void
1587 emit_bt (MonoProfiler *prof, LogBuffer *logbuffer, FrameData *data)
1588 {
1589         /* FIXME: this is actually tons of data and we should
1590          * just output it the first time and use an id the next
1591          */
1592         if (data->count > num_frames)
1593                 printf ("bad num frames: %d\n", data->count);
1594         emit_value (logbuffer, data->count);
1595         //if (*p != data.count) {
1596         //      printf ("bad num frames enc at %d: %d -> %d\n", count, data.count, *p); printf ("frames end: %p->%p\n", p, logbuffer->cursor); exit(0);}
1597         while (data->count) {
1598                 emit_method (logbuffer, data->methods [--data->count]);
1599         }
1600 }
1601
1602 static void
1603 gc_alloc (MonoProfiler *prof, MonoObject *obj, MonoClass *klass)
1604 {
1605         init_thread (prof, TRUE);
1606
1607         int do_bt = (nocalls && InterlockedRead (&runtime_inited) && !notraces) ? TYPE_ALLOC_BT : 0;
1608         FrameData data;
1609         uintptr_t len = mono_object_get_size (obj);
1610         /* account for object alignment in the heap */
1611         len += 7;
1612         len &= ~7;
1613
1614         if (do_bt)
1615                 collect_bt (&data);
1616
1617         ENTER_LOG (&gc_allocs_ctr, logbuffer,
1618                 EVENT_SIZE /* event */ +
1619                 LEB128_SIZE /* klass */ +
1620                 LEB128_SIZE /* obj */ +
1621                 LEB128_SIZE /* size */ +
1622                 (do_bt ? (
1623                         LEB128_SIZE /* count */ +
1624                         data.count * (
1625                                 LEB128_SIZE /* method */
1626                         )
1627                 ) : 0)
1628         );
1629
1630         emit_event (logbuffer, do_bt | TYPE_ALLOC);
1631         emit_ptr (logbuffer, klass);
1632         emit_obj (logbuffer, obj);
1633         emit_value (logbuffer, len);
1634
1635         if (do_bt)
1636                 emit_bt (prof, logbuffer, &data);
1637
1638         EXIT_LOG;
1639 }
1640
1641 static void
1642 gc_moves (MonoProfiler *prof, void **objects, int num)
1643 {
1644         ENTER_LOG (&gc_moves_ctr, logbuffer,
1645                 EVENT_SIZE /* event */ +
1646                 LEB128_SIZE /* num */ +
1647                 num * (
1648                         LEB128_SIZE /* object */
1649                 )
1650         );
1651
1652         emit_event (logbuffer, TYPE_GC_MOVE | TYPE_GC);
1653         emit_value (logbuffer, num);
1654
1655         for (int i = 0; i < num; ++i)
1656                 emit_obj (logbuffer, objects [i]);
1657
1658         EXIT_LOG_EXPLICIT (DO_SEND);
1659 }
1660
1661 static void
1662 gc_handle (MonoProfiler *prof, int op, int type, uintptr_t handle, MonoObject *obj)
1663 {
1664         int do_bt = nocalls && InterlockedRead (&runtime_inited) && !notraces;
1665         FrameData data;
1666
1667         if (do_bt)
1668                 collect_bt (&data);
1669
1670         gint32 *ctr = op == MONO_PROFILER_GC_HANDLE_CREATED ? &gc_handle_creations_ctr : &gc_handle_deletions_ctr;
1671
1672         ENTER_LOG (ctr, logbuffer,
1673                 EVENT_SIZE /* event */ +
1674                 LEB128_SIZE /* type */ +
1675                 LEB128_SIZE /* handle */ +
1676                 (op == MONO_PROFILER_GC_HANDLE_CREATED ? (
1677                         LEB128_SIZE /* obj */
1678                 ) : 0) +
1679                 (do_bt ? (
1680                         LEB128_SIZE /* count */ +
1681                         data.count * (
1682                                 LEB128_SIZE /* method */
1683                         )
1684                 ) : 0)
1685         );
1686
1687         if (op == MONO_PROFILER_GC_HANDLE_CREATED)
1688                 emit_event (logbuffer, (do_bt ? TYPE_GC_HANDLE_CREATED_BT : TYPE_GC_HANDLE_CREATED) | TYPE_GC);
1689         else if (op == MONO_PROFILER_GC_HANDLE_DESTROYED)
1690                 emit_event (logbuffer, (do_bt ? TYPE_GC_HANDLE_DESTROYED_BT : TYPE_GC_HANDLE_DESTROYED) | TYPE_GC);
1691         else
1692                 g_assert_not_reached ();
1693
1694         emit_value (logbuffer, type);
1695         emit_value (logbuffer, handle);
1696
1697         if (op == MONO_PROFILER_GC_HANDLE_CREATED)
1698                 emit_obj (logbuffer, obj);
1699
1700         if (do_bt)
1701                 emit_bt (prof, logbuffer, &data);
1702
1703         EXIT_LOG;
1704 }
1705
1706 static void
1707 finalize_begin (MonoProfiler *prof)
1708 {
1709         ENTER_LOG (&finalize_begins_ctr, buf,
1710                 EVENT_SIZE /* event */
1711         );
1712
1713         emit_event (buf, TYPE_GC_FINALIZE_START | TYPE_GC);
1714
1715         EXIT_LOG;
1716 }
1717
1718 static void
1719 finalize_end (MonoProfiler *prof)
1720 {
1721         trigger_on_demand_heapshot ();
1722         if (ENABLED (PROFLOG_FINALIZATION_EVENTS)) {
1723                 ENTER_LOG (&finalize_ends_ctr, buf,
1724                         EVENT_SIZE /* event */
1725                 );
1726
1727                 emit_event (buf, TYPE_GC_FINALIZE_END | TYPE_GC);
1728
1729                 EXIT_LOG;
1730         }
1731 }
1732
1733 static void
1734 finalize_object_begin (MonoProfiler *prof, MonoObject *obj)
1735 {
1736         ENTER_LOG (&finalize_object_begins_ctr, buf,
1737                 EVENT_SIZE /* event */ +
1738                 LEB128_SIZE /* obj */
1739         );
1740
1741         emit_event (buf, TYPE_GC_FINALIZE_OBJECT_START | TYPE_GC);
1742         emit_obj (buf, obj);
1743
1744         EXIT_LOG;
1745 }
1746
1747 static void
1748 finalize_object_end (MonoProfiler *prof, MonoObject *obj)
1749 {
1750         ENTER_LOG (&finalize_object_ends_ctr, buf,
1751                 EVENT_SIZE /* event */ +
1752                 LEB128_SIZE /* obj */
1753         );
1754
1755         emit_event (buf, TYPE_GC_FINALIZE_OBJECT_END | TYPE_GC);
1756         emit_obj (buf, obj);
1757
1758         EXIT_LOG;
1759 }
1760
1761 static char*
1762 push_nesting (char *p, MonoClass *klass)
1763 {
1764         MonoClass *nesting;
1765         const char *name;
1766         const char *nspace;
1767         nesting = mono_class_get_nesting_type (klass);
1768         if (nesting) {
1769                 p = push_nesting (p, nesting);
1770                 *p++ = '/';
1771                 *p = 0;
1772         }
1773         name = mono_class_get_name (klass);
1774         nspace = mono_class_get_namespace (klass);
1775         if (*nspace) {
1776                 strcpy (p, nspace);
1777                 p += strlen (nspace);
1778                 *p++ = '.';
1779                 *p = 0;
1780         }
1781         strcpy (p, name);
1782         p += strlen (name);
1783         return p;
1784 }
1785
1786 static char*
1787 type_name (MonoClass *klass)
1788 {
1789         char buf [1024];
1790         char *p;
1791         push_nesting (buf, klass);
1792         p = (char *) g_malloc (strlen (buf) + 1);
1793         strcpy (p, buf);
1794         return p;
1795 }
1796
1797 static void
1798 image_loaded (MonoProfiler *prof, MonoImage *image, int result)
1799 {
1800         if (result != MONO_PROFILE_OK)
1801                 return;
1802
1803         const char *name = mono_image_get_filename (image);
1804         int nlen = strlen (name) + 1;
1805
1806         ENTER_LOG (&image_loads_ctr, logbuffer,
1807                 EVENT_SIZE /* event */ +
1808                 BYTE_SIZE /* type */ +
1809                 LEB128_SIZE /* image */ +
1810                 nlen /* name */
1811         );
1812
1813         emit_event (logbuffer, TYPE_END_LOAD | TYPE_METADATA);
1814         emit_byte (logbuffer, TYPE_IMAGE);
1815         emit_ptr (logbuffer, image);
1816         memcpy (logbuffer->cursor, name, nlen);
1817         logbuffer->cursor += nlen;
1818
1819         EXIT_LOG;
1820 }
1821
1822 static void
1823 image_unloaded (MonoProfiler *prof, MonoImage *image)
1824 {
1825         const char *name = mono_image_get_filename (image);
1826         int nlen = strlen (name) + 1;
1827
1828         ENTER_LOG (&image_unloads_ctr, logbuffer,
1829                 EVENT_SIZE /* event */ +
1830                 BYTE_SIZE /* type */ +
1831                 LEB128_SIZE /* image */ +
1832                 nlen /* name */
1833         );
1834
1835         emit_event (logbuffer, TYPE_END_UNLOAD | TYPE_METADATA);
1836         emit_byte (logbuffer, TYPE_IMAGE);
1837         emit_ptr (logbuffer, image);
1838         memcpy (logbuffer->cursor, name, nlen);
1839         logbuffer->cursor += nlen;
1840
1841         EXIT_LOG;
1842 }
1843
1844 static void
1845 assembly_loaded (MonoProfiler *prof, MonoAssembly *assembly, int result)
1846 {
1847         if (result != MONO_PROFILE_OK)
1848                 return;
1849
1850         char *name = mono_stringify_assembly_name (mono_assembly_get_name (assembly));
1851         int nlen = strlen (name) + 1;
1852         MonoImage *image = mono_assembly_get_image (assembly);
1853
1854         ENTER_LOG (&assembly_loads_ctr, logbuffer,
1855                 EVENT_SIZE /* event */ +
1856                 BYTE_SIZE /* type */ +
1857                 LEB128_SIZE /* assembly */ +
1858                 LEB128_SIZE /* image */ +
1859                 nlen /* name */
1860         );
1861
1862         emit_event (logbuffer, TYPE_END_LOAD | TYPE_METADATA);
1863         emit_byte (logbuffer, TYPE_ASSEMBLY);
1864         emit_ptr (logbuffer, assembly);
1865         emit_ptr (logbuffer, image);
1866         memcpy (logbuffer->cursor, name, nlen);
1867         logbuffer->cursor += nlen;
1868
1869         EXIT_LOG;
1870
1871         mono_free (name);
1872 }
1873
1874 static void
1875 assembly_unloaded (MonoProfiler *prof, MonoAssembly *assembly)
1876 {
1877         char *name = mono_stringify_assembly_name (mono_assembly_get_name (assembly));
1878         int nlen = strlen (name) + 1;
1879         MonoImage *image = mono_assembly_get_image (assembly);
1880
1881         ENTER_LOG (&assembly_unloads_ctr, logbuffer,
1882                 EVENT_SIZE /* event */ +
1883                 BYTE_SIZE /* type */ +
1884                 LEB128_SIZE /* assembly */ +
1885                 LEB128_SIZE /* image */ +
1886                 nlen /* name */
1887         );
1888
1889         emit_event (logbuffer, TYPE_END_UNLOAD | TYPE_METADATA);
1890         emit_byte (logbuffer, TYPE_ASSEMBLY);
1891         emit_ptr (logbuffer, assembly);
1892         emit_ptr (logbuffer, image);
1893         memcpy (logbuffer->cursor, name, nlen);
1894         logbuffer->cursor += nlen;
1895
1896         EXIT_LOG;
1897
1898         mono_free (name);
1899 }
1900
1901 static void
1902 class_loaded (MonoProfiler *prof, MonoClass *klass, int result)
1903 {
1904         if (result != MONO_PROFILE_OK)
1905                 return;
1906
1907         char *name;
1908
1909         if (InterlockedRead (&runtime_inited))
1910                 name = mono_type_get_name (mono_class_get_type (klass));
1911         else
1912                 name = type_name (klass);
1913
1914         int nlen = strlen (name) + 1;
1915         MonoImage *image = mono_class_get_image (klass);
1916
1917         ENTER_LOG (&class_loads_ctr, logbuffer,
1918                 EVENT_SIZE /* event */ +
1919                 BYTE_SIZE /* type */ +
1920                 LEB128_SIZE /* klass */ +
1921                 LEB128_SIZE /* image */ +
1922                 nlen /* name */
1923         );
1924
1925         emit_event (logbuffer, TYPE_END_LOAD | TYPE_METADATA);
1926         emit_byte (logbuffer, TYPE_CLASS);
1927         emit_ptr (logbuffer, klass);
1928         emit_ptr (logbuffer, image);
1929         memcpy (logbuffer->cursor, name, nlen);
1930         logbuffer->cursor += nlen;
1931
1932         EXIT_LOG;
1933
1934         if (runtime_inited)
1935                 mono_free (name);
1936         else
1937                 g_free (name);
1938 }
1939
1940 static void
1941 class_unloaded (MonoProfiler *prof, MonoClass *klass)
1942 {
1943         char *name;
1944
1945         if (InterlockedRead (&runtime_inited))
1946                 name = mono_type_get_name (mono_class_get_type (klass));
1947         else
1948                 name = type_name (klass);
1949
1950         int nlen = strlen (name) + 1;
1951         MonoImage *image = mono_class_get_image (klass);
1952
1953         ENTER_LOG (&class_unloads_ctr, logbuffer,
1954                 EVENT_SIZE /* event */ +
1955                 BYTE_SIZE /* type */ +
1956                 LEB128_SIZE /* klass */ +
1957                 LEB128_SIZE /* image */ +
1958                 nlen /* name */
1959         );
1960
1961         emit_event (logbuffer, TYPE_END_UNLOAD | TYPE_METADATA);
1962         emit_byte (logbuffer, TYPE_CLASS);
1963         emit_ptr (logbuffer, klass);
1964         emit_ptr (logbuffer, image);
1965         memcpy (logbuffer->cursor, name, nlen);
1966         logbuffer->cursor += nlen;
1967
1968         EXIT_LOG;
1969
1970         if (runtime_inited)
1971                 mono_free (name);
1972         else
1973                 g_free (name);
1974 }
1975
1976 static void process_method_enter_coverage (MonoProfiler *prof, MonoMethod *method);
1977
1978 static void
1979 method_enter (MonoProfiler *prof, MonoMethod *method)
1980 {
1981         process_method_enter_coverage (prof, method);
1982
1983         if (!only_coverage && PROF_TLS_GET ()->call_depth++ <= max_call_depth) {
1984                 ENTER_LOG (&method_entries_ctr, logbuffer,
1985                         EVENT_SIZE /* event */ +
1986                         LEB128_SIZE /* method */
1987                 );
1988
1989                 emit_event (logbuffer, TYPE_ENTER | TYPE_METHOD);
1990                 emit_method (logbuffer, method);
1991
1992                 EXIT_LOG;
1993         }
1994 }
1995
1996 static void
1997 method_leave (MonoProfiler *prof, MonoMethod *method)
1998 {
1999         if (!only_coverage && --PROF_TLS_GET ()->call_depth <= max_call_depth) {
2000                 ENTER_LOG (&method_exits_ctr, logbuffer,
2001                         EVENT_SIZE /* event */ +
2002                         LEB128_SIZE /* method */
2003                 );
2004
2005                 emit_event (logbuffer, TYPE_LEAVE | TYPE_METHOD);
2006                 emit_method (logbuffer, method);
2007
2008                 EXIT_LOG;
2009         }
2010 }
2011
2012 static void
2013 method_exc_leave (MonoProfiler *prof, MonoMethod *method)
2014 {
2015         if (!only_coverage && !nocalls && --PROF_TLS_GET ()->call_depth <= max_call_depth) {
2016                 ENTER_LOG (&method_exception_exits_ctr, logbuffer,
2017                         EVENT_SIZE /* event */ +
2018                         LEB128_SIZE /* method */
2019                 );
2020
2021                 emit_event (logbuffer, TYPE_EXC_LEAVE | TYPE_METHOD);
2022                 emit_method (logbuffer, method);
2023
2024                 EXIT_LOG;
2025         }
2026 }
2027
2028 static void
2029 method_jitted (MonoProfiler *prof, MonoMethod *method, MonoJitInfo *ji, int result)
2030 {
2031         if (result != MONO_PROFILE_OK)
2032                 return;
2033
2034         register_method_local (method, ji);
2035 }
2036
2037 static void
2038 code_buffer_new (MonoProfiler *prof, void *buffer, int size, MonoProfilerCodeBufferType type, void *data)
2039 {
2040         char *name;
2041         int nlen;
2042
2043         if (type == MONO_PROFILER_CODE_BUFFER_SPECIFIC_TRAMPOLINE) {
2044                 name = (char *) data;
2045                 nlen = strlen (name) + 1;
2046         } else {
2047                 name = NULL;
2048                 nlen = 0;
2049         }
2050
2051         ENTER_LOG (&code_buffers_ctr, logbuffer,
2052                 EVENT_SIZE /* event */ +
2053                 BYTE_SIZE /* type */ +
2054                 LEB128_SIZE /* buffer */ +
2055                 LEB128_SIZE /* size */ +
2056                 (name ? (
2057                         nlen /* name */
2058                 ) : 0)
2059         );
2060
2061         emit_event (logbuffer, TYPE_JITHELPER | TYPE_RUNTIME);
2062         emit_byte (logbuffer, type);
2063         emit_ptr (logbuffer, buffer);
2064         emit_value (logbuffer, size);
2065
2066         if (name) {
2067                 memcpy (logbuffer->cursor, name, nlen);
2068                 logbuffer->cursor += nlen;
2069         }
2070
2071         EXIT_LOG;
2072 }
2073
2074 static void
2075 throw_exc (MonoProfiler *prof, MonoObject *object)
2076 {
2077         int do_bt = (nocalls && InterlockedRead (&runtime_inited) && !notraces) ? TYPE_THROW_BT : 0;
2078         FrameData data;
2079
2080         if (do_bt)
2081                 collect_bt (&data);
2082
2083         ENTER_LOG (&exception_throws_ctr, logbuffer,
2084                 EVENT_SIZE /* event */ +
2085                 LEB128_SIZE /* object */ +
2086                 (do_bt ? (
2087                         LEB128_SIZE /* count */ +
2088                         data.count * (
2089                                 LEB128_SIZE /* method */
2090                         )
2091                 ) : 0)
2092         );
2093
2094         emit_event (logbuffer, do_bt | TYPE_EXCEPTION);
2095         emit_obj (logbuffer, object);
2096
2097         if (do_bt)
2098                 emit_bt (prof, logbuffer, &data);
2099
2100         EXIT_LOG;
2101 }
2102
2103 static void
2104 clause_exc (MonoProfiler *prof, MonoMethod *method, int clause_type, int clause_num, MonoObject *exc)
2105 {
2106         ENTER_LOG (&exception_clauses_ctr, logbuffer,
2107                 EVENT_SIZE /* event */ +
2108                 BYTE_SIZE /* clause type */ +
2109                 LEB128_SIZE /* clause num */ +
2110                 LEB128_SIZE /* method */
2111         );
2112
2113         emit_event (logbuffer, TYPE_EXCEPTION | TYPE_CLAUSE);
2114         emit_byte (logbuffer, clause_type);
2115         emit_value (logbuffer, clause_num);
2116         emit_method (logbuffer, method);
2117         emit_obj (logbuffer, exc);
2118
2119         EXIT_LOG;
2120 }
2121
2122 static void
2123 monitor_event (MonoProfiler *profiler, MonoObject *object, MonoProfilerMonitorEvent ev)
2124 {
2125         int do_bt = (nocalls && InterlockedRead (&runtime_inited) && !notraces) ? TYPE_MONITOR_BT : 0;
2126         FrameData data;
2127
2128         if (do_bt)
2129                 collect_bt (&data);
2130
2131         ENTER_LOG (&monitor_events_ctr, logbuffer,
2132                 EVENT_SIZE /* event */ +
2133                 BYTE_SIZE /* ev */ +
2134                 LEB128_SIZE /* object */ +
2135                 (do_bt ? (
2136                         LEB128_SIZE /* count */ +
2137                         data.count * (
2138                                 LEB128_SIZE /* method */
2139                         )
2140                 ) : 0)
2141         );
2142
2143         emit_event (logbuffer, do_bt | TYPE_MONITOR);
2144         emit_byte (logbuffer, ev);
2145         emit_obj (logbuffer, object);
2146
2147         if (do_bt)
2148                 emit_bt (profiler, logbuffer, &data);
2149
2150         EXIT_LOG;
2151 }
2152
2153 static void
2154 thread_start (MonoProfiler *prof, uintptr_t tid)
2155 {
2156         init_thread (prof, TRUE);
2157
2158         if (ENABLED (PROFLOG_THREAD_EVENTS)) {
2159                 ENTER_LOG (&thread_starts_ctr, logbuffer,
2160                         EVENT_SIZE /* event */ +
2161                         BYTE_SIZE /* type */ +
2162                         LEB128_SIZE /* tid */
2163                 );
2164
2165                 emit_event (logbuffer, TYPE_END_LOAD | TYPE_METADATA);
2166                 emit_byte (logbuffer, TYPE_THREAD);
2167                 emit_ptr (logbuffer, (void*) tid);
2168
2169                 EXIT_LOG;
2170         }
2171 }
2172
2173 static void
2174 thread_end (MonoProfiler *prof, uintptr_t tid)
2175 {
2176         if (ENABLED (PROFLOG_THREAD_EVENTS)) {
2177                 ENTER_LOG (&thread_ends_ctr, logbuffer,
2178                         EVENT_SIZE /* event */ +
2179                         BYTE_SIZE /* type */ +
2180                         LEB128_SIZE /* tid */
2181                 );
2182
2183                 emit_event (logbuffer, TYPE_END_UNLOAD | TYPE_METADATA);
2184                 emit_byte (logbuffer, TYPE_THREAD);
2185                 emit_ptr (logbuffer, (void*) tid);
2186
2187                 EXIT_LOG_EXPLICIT (NO_SEND);
2188         }
2189
2190         MonoProfilerThread *thread = PROF_TLS_GET ();
2191
2192         thread->ended = TRUE;
2193         remove_thread (thread);
2194
2195         PROF_TLS_SET (NULL);
2196 }
2197
2198 static void
2199 thread_name (MonoProfiler *prof, uintptr_t tid, const char *name)
2200 {
2201         int len = strlen (name) + 1;
2202
2203         if (ENABLED (PROFLOG_THREAD_EVENTS)) {
2204                 ENTER_LOG (&thread_names_ctr, logbuffer,
2205                         EVENT_SIZE /* event */ +
2206                         BYTE_SIZE /* type */ +
2207                         LEB128_SIZE /* tid */ +
2208                         len /* name */
2209                 );
2210
2211                 emit_event (logbuffer, TYPE_METADATA);
2212                 emit_byte (logbuffer, TYPE_THREAD);
2213                 emit_ptr (logbuffer, (void*)tid);
2214                 memcpy (logbuffer->cursor, name, len);
2215                 logbuffer->cursor += len;
2216
2217                 EXIT_LOG;
2218         }
2219 }
2220
2221 static void
2222 domain_loaded (MonoProfiler *prof, MonoDomain *domain, int result)
2223 {
2224         if (result != MONO_PROFILE_OK)
2225                 return;
2226
2227         ENTER_LOG (&domain_loads_ctr, logbuffer,
2228                 EVENT_SIZE /* event */ +
2229                 BYTE_SIZE /* type */ +
2230                 LEB128_SIZE /* domain id */
2231         );
2232
2233         emit_event (logbuffer, TYPE_END_LOAD | TYPE_METADATA);
2234         emit_byte (logbuffer, TYPE_DOMAIN);
2235         emit_ptr (logbuffer, (void*)(uintptr_t) mono_domain_get_id (domain));
2236
2237         EXIT_LOG;
2238 }
2239
2240 static void
2241 domain_unloaded (MonoProfiler *prof, MonoDomain *domain)
2242 {
2243         ENTER_LOG (&domain_unloads_ctr, logbuffer,
2244                 EVENT_SIZE /* event */ +
2245                 BYTE_SIZE /* type */ +
2246                 LEB128_SIZE /* domain id */
2247         );
2248
2249         emit_event (logbuffer, TYPE_END_UNLOAD | TYPE_METADATA);
2250         emit_byte (logbuffer, TYPE_DOMAIN);
2251         emit_ptr (logbuffer, (void*)(uintptr_t) mono_domain_get_id (domain));
2252
2253         EXIT_LOG;
2254 }
2255
2256 static void
2257 domain_name (MonoProfiler *prof, MonoDomain *domain, const char *name)
2258 {
2259         int nlen = strlen (name) + 1;
2260
2261         ENTER_LOG (&domain_names_ctr, logbuffer,
2262                 EVENT_SIZE /* event */ +
2263                 BYTE_SIZE /* type */ +
2264                 LEB128_SIZE /* domain id */ +
2265                 nlen /* name */
2266         );
2267
2268         emit_event (logbuffer, TYPE_METADATA);
2269         emit_byte (logbuffer, TYPE_DOMAIN);
2270         emit_ptr (logbuffer, (void*)(uintptr_t) mono_domain_get_id (domain));
2271         memcpy (logbuffer->cursor, name, nlen);
2272         logbuffer->cursor += nlen;
2273
2274         EXIT_LOG;
2275 }
2276
2277 static void
2278 context_loaded (MonoProfiler *prof, MonoAppContext *context)
2279 {
2280         ENTER_LOG (&context_loads_ctr, logbuffer,
2281                 EVENT_SIZE /* event */ +
2282                 BYTE_SIZE /* type */ +
2283                 LEB128_SIZE /* context id */ +
2284                 LEB128_SIZE /* domain id */
2285         );
2286
2287         emit_event (logbuffer, TYPE_END_LOAD | TYPE_METADATA);
2288         emit_byte (logbuffer, TYPE_CONTEXT);
2289         emit_ptr (logbuffer, (void*)(uintptr_t) mono_context_get_id (context));
2290         emit_ptr (logbuffer, (void*)(uintptr_t) mono_context_get_domain_id (context));
2291
2292         EXIT_LOG;
2293 }
2294
2295 static void
2296 context_unloaded (MonoProfiler *prof, MonoAppContext *context)
2297 {
2298         ENTER_LOG (&context_unloads_ctr, logbuffer,
2299                 EVENT_SIZE /* event */ +
2300                 BYTE_SIZE /* type */ +
2301                 LEB128_SIZE /* context id */ +
2302                 LEB128_SIZE /* domain id */
2303         );
2304
2305         emit_event (logbuffer, TYPE_END_UNLOAD | TYPE_METADATA);
2306         emit_byte (logbuffer, TYPE_CONTEXT);
2307         emit_ptr (logbuffer, (void*)(uintptr_t) mono_context_get_id (context));
2308         emit_ptr (logbuffer, (void*)(uintptr_t) mono_context_get_domain_id (context));
2309
2310         EXIT_LOG;
2311 }
2312
2313 typedef struct {
2314         MonoMethod *method;
2315         MonoDomain *domain;
2316         void *base_address;
2317         int offset;
2318 } AsyncFrameInfo;
2319
2320 typedef struct {
2321         MonoLockFreeQueueNode node;
2322         MonoProfiler *prof;
2323         uint64_t time;
2324         uintptr_t tid;
2325         void *ip;
2326         int count;
2327         AsyncFrameInfo frames [MONO_ZERO_LEN_ARRAY];
2328 } SampleHit;
2329
2330 static mono_bool
2331 async_walk_stack (MonoMethod *method, MonoDomain *domain, void *base_address, int offset, void *data)
2332 {
2333         SampleHit *sample = (SampleHit *) data;
2334
2335         if (sample->count < num_frames) {
2336                 int i = sample->count;
2337
2338                 sample->frames [i].method = method;
2339                 sample->frames [i].domain = domain;
2340                 sample->frames [i].base_address = base_address;
2341                 sample->frames [i].offset = offset;
2342
2343                 sample->count++;
2344         }
2345
2346         return sample->count == num_frames;
2347 }
2348
2349 #define SAMPLE_SLOT_SIZE(FRAMES) (sizeof (SampleHit) + sizeof (AsyncFrameInfo) * (FRAMES - MONO_ZERO_LEN_ARRAY))
2350 #define SAMPLE_BLOCK_SIZE (mono_pagesize ())
2351
2352 static void
2353 enqueue_sample_hit (gpointer p)
2354 {
2355         SampleHit *sample = p;
2356
2357         mono_lock_free_queue_node_unpoison (&sample->node);
2358         mono_lock_free_queue_enqueue (&sample->prof->dumper_queue, &sample->node);
2359         mono_os_sem_post (&sample->prof->dumper_queue_sem);
2360 }
2361
2362 static void
2363 mono_sample_hit (MonoProfiler *profiler, unsigned char *ip, void *context)
2364 {
2365         /*
2366          * Please note: We rely on the runtime loading the profiler with
2367          * MONO_DL_EAGER (RTLD_NOW) so that references to runtime functions within
2368          * this function (and its siblings) are resolved when the profiler is
2369          * loaded. Otherwise, we would potentially invoke the dynamic linker when
2370          * invoking runtime functions, which is not async-signal-safe.
2371          */
2372
2373         if (InterlockedRead (&in_shutdown))
2374                 return;
2375
2376         SampleHit *sample = (SampleHit *) mono_lock_free_queue_dequeue (&profiler->sample_reuse_queue);
2377
2378         if (!sample) {
2379                 /*
2380                  * If we're out of reusable sample events and we're not allowed to
2381                  * allocate more, we have no choice but to drop the event.
2382                  */
2383                 if (InterlockedRead (&sample_allocations_ctr) >= max_allocated_sample_hits)
2384                         return;
2385
2386                 sample = mono_lock_free_alloc (&profiler->sample_allocator);
2387                 sample->prof = profiler;
2388                 mono_lock_free_queue_node_init (&sample->node, TRUE);
2389
2390                 InterlockedIncrement (&sample_allocations_ctr);
2391         }
2392
2393         sample->count = 0;
2394         mono_stack_walk_async_safe (&async_walk_stack, context, sample);
2395
2396         sample->time = current_time ();
2397         sample->tid = thread_id ();
2398         sample->ip = ip;
2399
2400         mono_thread_hazardous_try_free (sample, enqueue_sample_hit);
2401 }
2402
2403 static uintptr_t *code_pages = 0;
2404 static int num_code_pages = 0;
2405 static int size_code_pages = 0;
2406 #define CPAGE_SHIFT (9)
2407 #define CPAGE_SIZE (1 << CPAGE_SHIFT)
2408 #define CPAGE_MASK (~(CPAGE_SIZE - 1))
2409 #define CPAGE_ADDR(p) ((p) & CPAGE_MASK)
2410
2411 static uintptr_t
2412 add_code_page (uintptr_t *hash, uintptr_t hsize, uintptr_t page)
2413 {
2414         uintptr_t i;
2415         uintptr_t start_pos;
2416         start_pos = (page >> CPAGE_SHIFT) % hsize;
2417         i = start_pos;
2418         do {
2419                 if (hash [i] && CPAGE_ADDR (hash [i]) == CPAGE_ADDR (page)) {
2420                         return 0;
2421                 } else if (!hash [i]) {
2422                         hash [i] = page;
2423                         return 1;
2424                 }
2425                 /* wrap around */
2426                 if (++i == hsize)
2427                         i = 0;
2428         } while (i != start_pos);
2429         /* should not happen */
2430         printf ("failed code page store\n");
2431         return 0;
2432 }
2433
2434 static void
2435 add_code_pointer (uintptr_t ip)
2436 {
2437         uintptr_t i;
2438         if (num_code_pages * 2 >= size_code_pages) {
2439                 uintptr_t *n;
2440                 uintptr_t old_size = size_code_pages;
2441                 size_code_pages *= 2;
2442                 if (size_code_pages == 0)
2443                         size_code_pages = 16;
2444                 n = (uintptr_t *) g_calloc (sizeof (uintptr_t) * size_code_pages, 1);
2445                 for (i = 0; i < old_size; ++i) {
2446                         if (code_pages [i])
2447                                 add_code_page (n, size_code_pages, code_pages [i]);
2448                 }
2449                 if (code_pages)
2450                         g_free (code_pages);
2451                 code_pages = n;
2452         }
2453         num_code_pages += add_code_page (code_pages, size_code_pages, ip & CPAGE_MASK);
2454 }
2455
2456 /* ELF code crashes on some systems. */
2457 //#if defined(HAVE_DL_ITERATE_PHDR) && defined(ELFMAG0)
2458 #if 0
2459 static void
2460 dump_ubin (MonoProfiler *prof, const char *filename, uintptr_t load_addr, uint64_t offset, uintptr_t size)
2461 {
2462         int len = strlen (filename) + 1;
2463
2464         ENTER_LOG (&sample_ubins_ctr, logbuffer,
2465                 EVENT_SIZE /* event */ +
2466                 LEB128_SIZE /* load address */ +
2467                 LEB128_SIZE /* offset */ +
2468                 LEB128_SIZE /* size */ +
2469                 nlen /* file name */
2470         );
2471
2472         emit_event (logbuffer, TYPE_SAMPLE | TYPE_SAMPLE_UBIN);
2473         emit_ptr (logbuffer, load_addr);
2474         emit_uvalue (logbuffer, offset);
2475         emit_uvalue (logbuffer, size);
2476         memcpy (logbuffer->cursor, filename, len);
2477         logbuffer->cursor += len;
2478
2479         EXIT_LOG_EXPLICIT (DO_SEND);
2480 }
2481 #endif
2482
2483 static void
2484 dump_usym (MonoProfiler *prof, const char *name, uintptr_t value, uintptr_t size)
2485 {
2486         int len = strlen (name) + 1;
2487
2488         ENTER_LOG (&sample_usyms_ctr, logbuffer,
2489                 EVENT_SIZE /* event */ +
2490                 LEB128_SIZE /* value */ +
2491                 LEB128_SIZE /* size */ +
2492                 len /* name */
2493         );
2494
2495         emit_event (logbuffer, TYPE_SAMPLE | TYPE_SAMPLE_USYM);
2496         emit_ptr (logbuffer, (void*)value);
2497         emit_value (logbuffer, size);
2498         memcpy (logbuffer->cursor, name, len);
2499         logbuffer->cursor += len;
2500
2501         EXIT_LOG_EXPLICIT (DO_SEND);
2502 }
2503
2504 /* ELF code crashes on some systems. */
2505 //#if defined(ELFMAG0)
2506 #if 0
2507
2508 #if SIZEOF_VOID_P == 4
2509 #define ELF_WSIZE 32
2510 #else
2511 #define ELF_WSIZE 64
2512 #endif
2513 #ifndef ElfW
2514 #define ElfW(type)      _ElfW (Elf, ELF_WSIZE, type)
2515 #define _ElfW(e,w,t)    _ElfW_1 (e, w, _##t)
2516 #define _ElfW_1(e,w,t)  e##w##t
2517 #endif
2518
2519 static void
2520 dump_elf_symbols (MonoProfiler *prof, ElfW(Sym) *symbols, int num_symbols, const char *strtab, void *load_addr)
2521 {
2522         int i;
2523         for (i = 0; i < num_symbols; ++i) {
2524                 const char* sym;
2525                 sym =  strtab + symbols [i].st_name;
2526                 if (!symbols [i].st_name || !symbols [i].st_size || (symbols [i].st_info & 0xf) != STT_FUNC)
2527                         continue;
2528                 //printf ("symbol %s at %d\n", sym, symbols [i].st_value);
2529                 dump_usym (sym, (uintptr_t)load_addr + symbols [i].st_value, symbols [i].st_size);
2530         }
2531 }
2532
2533 static int
2534 read_elf_symbols (MonoProfiler *prof, const char *filename, void *load_addr)
2535 {
2536         int fd, i;
2537         void *data;
2538         struct stat statb;
2539         uint64_t file_size;
2540         ElfW(Ehdr) *header;
2541         ElfW(Shdr) *sheader;
2542         ElfW(Shdr) *shstrtabh;
2543         ElfW(Shdr) *symtabh = NULL;
2544         ElfW(Shdr) *strtabh = NULL;
2545         ElfW(Sym) *symbols = NULL;
2546         const char *strtab;
2547         int num_symbols;
2548
2549         fd = open (filename, O_RDONLY);
2550         if (fd < 0)
2551                 return 0;
2552         if (fstat (fd, &statb) != 0) {
2553                 close (fd);
2554                 return 0;
2555         }
2556         file_size = statb.st_size;
2557         data = mmap (NULL, file_size, PROT_READ, MAP_PRIVATE, fd, 0);
2558         close (fd);
2559         if (data == MAP_FAILED)
2560                 return 0;
2561         header = data;
2562         if (header->e_ident [EI_MAG0] != ELFMAG0 ||
2563                         header->e_ident [EI_MAG1] != ELFMAG1 ||
2564                         header->e_ident [EI_MAG2] != ELFMAG2 ||
2565                         header->e_ident [EI_MAG3] != ELFMAG3 ) {
2566                 munmap (data, file_size);
2567                 return 0;
2568         }
2569         sheader = (void*)((char*)data + header->e_shoff);
2570         shstrtabh = (void*)((char*)sheader + (header->e_shentsize * header->e_shstrndx));
2571         strtab = (const char*)data + shstrtabh->sh_offset;
2572         for (i = 0; i < header->e_shnum; ++i) {
2573                 //printf ("section header: %d\n", sheader->sh_type);
2574                 if (sheader->sh_type == SHT_SYMTAB) {
2575                         symtabh = sheader;
2576                         strtabh = (void*)((char*)data + header->e_shoff + sheader->sh_link * header->e_shentsize);
2577                         /*printf ("symtab section header: %d, .strstr: %d\n", i, sheader->sh_link);*/
2578                         break;
2579                 }
2580                 sheader = (void*)((char*)sheader + header->e_shentsize);
2581         }
2582         if (!symtabh || !strtabh) {
2583                 munmap (data, file_size);
2584                 return 0;
2585         }
2586         strtab = (const char*)data + strtabh->sh_offset;
2587         num_symbols = symtabh->sh_size / symtabh->sh_entsize;
2588         symbols = (void*)((char*)data + symtabh->sh_offset);
2589         dump_elf_symbols (symbols, num_symbols, strtab, load_addr);
2590         munmap (data, file_size);
2591         return 1;
2592 }
2593 #endif
2594
2595 /* ELF code crashes on some systems. */
2596 //#if defined(HAVE_DL_ITERATE_PHDR) && defined(ELFMAG0)
2597 #if 0
2598 static int
2599 elf_dl_callback (struct dl_phdr_info *info, size_t size, void *data)
2600 {
2601         MonoProfiler *prof = data;
2602         char buf [256];
2603         const char *filename;
2604         BinaryObject *obj;
2605         char *a = (void*)info->dlpi_addr;
2606         int i, num_sym;
2607         ElfW(Dyn) *dyn = NULL;
2608         ElfW(Sym) *symtab = NULL;
2609         ElfW(Word) *hash_table = NULL;
2610         ElfW(Ehdr) *header = NULL;
2611         const char* strtab = NULL;
2612         for (obj = prof->binary_objects; obj; obj = obj->next) {
2613                 if (obj->addr == a)
2614                         return 0;
2615         }
2616         filename = info->dlpi_name;
2617         if (!filename)
2618                 return 0;
2619         if (!info->dlpi_addr && !filename [0]) {
2620                 int l = readlink ("/proc/self/exe", buf, sizeof (buf) - 1);
2621                 if (l > 0) {
2622                         buf [l] = 0;
2623                         filename = buf;
2624                 }
2625         }
2626         obj = g_calloc (sizeof (BinaryObject), 1);
2627         obj->addr = (void*)info->dlpi_addr;
2628         obj->name = pstrdup (filename);
2629         obj->next = prof->binary_objects;
2630         prof->binary_objects = obj;
2631         //printf ("loaded file: %s at %p, segments: %d\n", filename, (void*)info->dlpi_addr, info->dlpi_phnum);
2632         a = NULL;
2633         for (i = 0; i < info->dlpi_phnum; ++i) {
2634                 //printf ("segment type %d file offset: %d, size: %d\n", info->dlpi_phdr[i].p_type, info->dlpi_phdr[i].p_offset, info->dlpi_phdr[i].p_memsz);
2635                 if (info->dlpi_phdr[i].p_type == PT_LOAD && !header) {
2636                         header = (ElfW(Ehdr)*)(info->dlpi_addr + info->dlpi_phdr[i].p_vaddr);
2637                         if (header->e_ident [EI_MAG0] != ELFMAG0 ||
2638                                         header->e_ident [EI_MAG1] != ELFMAG1 ||
2639                                         header->e_ident [EI_MAG2] != ELFMAG2 ||
2640                                         header->e_ident [EI_MAG3] != ELFMAG3 ) {
2641                                 header = NULL;
2642                         }
2643                         dump_ubin (prof, filename, info->dlpi_addr + info->dlpi_phdr[i].p_vaddr, info->dlpi_phdr[i].p_offset, info->dlpi_phdr[i].p_memsz);
2644                 } else if (info->dlpi_phdr[i].p_type == PT_DYNAMIC) {
2645                         dyn = (ElfW(Dyn) *)(info->dlpi_addr + info->dlpi_phdr[i].p_vaddr);
2646                 }
2647         }
2648         if (read_elf_symbols (prof, filename, (void*)info->dlpi_addr))
2649                 return 0;
2650         if (!info->dlpi_name || !info->dlpi_name[0])
2651                 return 0;
2652         if (!dyn)
2653                 return 0;
2654         for (i = 0; dyn [i].d_tag != DT_NULL; ++i) {
2655                 if (dyn [i].d_tag == DT_SYMTAB) {
2656                         if (symtab && do_debug)
2657                                 printf ("multiple symtabs: %d\n", i);
2658                         symtab = (ElfW(Sym) *)(a + dyn [i].d_un.d_ptr);
2659                 } else if (dyn [i].d_tag == DT_HASH) {
2660                         hash_table = (ElfW(Word) *)(a + dyn [i].d_un.d_ptr);
2661                 } else if (dyn [i].d_tag == DT_STRTAB) {
2662                         strtab = (const char*)(a + dyn [i].d_un.d_ptr);
2663                 }
2664         }
2665         if (!hash_table)
2666                 return 0;
2667         num_sym = hash_table [1];
2668         dump_elf_symbols (prof, symtab, num_sym, strtab, (void*)info->dlpi_addr);
2669         return 0;
2670 }
2671
2672 static int
2673 load_binaries (MonoProfiler *prof)
2674 {
2675         dl_iterate_phdr (elf_dl_callback, prof);
2676         return 1;
2677 }
2678 #else
2679 static int
2680 load_binaries (MonoProfiler *prof)
2681 {
2682         return 0;
2683 }
2684 #endif
2685
2686 static const char*
2687 symbol_for (uintptr_t code)
2688 {
2689 #ifdef HAVE_DLADDR
2690         void *ip = (void*)code;
2691         Dl_info di;
2692         if (dladdr (ip, &di)) {
2693                 if (di.dli_sname)
2694                         return di.dli_sname;
2695         } else {
2696         /*      char **names;
2697                 names = backtrace_symbols (&ip, 1);
2698                 if (names) {
2699                         const char* p = names [0];
2700                         g_free (names);
2701                         return p;
2702                 }
2703                 */
2704         }
2705 #endif
2706         return NULL;
2707 }
2708
2709 static void
2710 dump_unmanaged_coderefs (MonoProfiler *prof)
2711 {
2712         int i;
2713         const char* last_symbol;
2714         uintptr_t addr, page_end;
2715
2716         if (load_binaries (prof))
2717                 return;
2718         for (i = 0; i < size_code_pages; ++i) {
2719                 const char* sym;
2720                 if (!code_pages [i] || code_pages [i] & 1)
2721                         continue;
2722                 last_symbol = NULL;
2723                 addr = CPAGE_ADDR (code_pages [i]);
2724                 page_end = addr + CPAGE_SIZE;
2725                 code_pages [i] |= 1;
2726                 /* we dump the symbols for the whole page */
2727                 for (; addr < page_end; addr += 16) {
2728                         sym = symbol_for (addr);
2729                         if (sym && sym == last_symbol)
2730                                 continue;
2731                         last_symbol = sym;
2732                         if (!sym)
2733                                 continue;
2734                         dump_usym (prof, sym, addr, 0); /* let's not guess the size */
2735                         //printf ("found symbol at %p: %s\n", (void*)addr, sym);
2736                 }
2737         }
2738 }
2739
2740 typedef struct MonoCounterAgent {
2741         MonoCounter *counter;
2742         // MonoCounterAgent specific data :
2743         void *value;
2744         size_t value_size;
2745         short index;
2746         short emitted;
2747         struct MonoCounterAgent *next;
2748 } MonoCounterAgent;
2749
2750 static MonoCounterAgent* counters;
2751 static int counters_index = 1;
2752 static mono_mutex_t counters_mutex;
2753
2754 static void
2755 counters_add_agent (MonoCounter *counter)
2756 {
2757         if (InterlockedRead (&in_shutdown))
2758                 return;
2759
2760         MonoCounterAgent *agent, *item;
2761
2762         mono_os_mutex_lock (&counters_mutex);
2763
2764         for (agent = counters; agent; agent = agent->next) {
2765                 if (agent->counter == counter) {
2766                         agent->value_size = 0;
2767                         if (agent->value) {
2768                                 g_free (agent->value);
2769                                 agent->value = NULL;
2770                         }
2771                         goto done;
2772                 }
2773         }
2774
2775         agent = (MonoCounterAgent *) g_malloc (sizeof (MonoCounterAgent));
2776         agent->counter = counter;
2777         agent->value = NULL;
2778         agent->value_size = 0;
2779         agent->index = counters_index++;
2780         agent->emitted = 0;
2781         agent->next = NULL;
2782
2783         if (!counters) {
2784                 counters = agent;
2785         } else {
2786                 item = counters;
2787                 while (item->next)
2788                         item = item->next;
2789                 item->next = agent;
2790         }
2791
2792 done:
2793         mono_os_mutex_unlock (&counters_mutex);
2794 }
2795
2796 static mono_bool
2797 counters_init_foreach_callback (MonoCounter *counter, gpointer data)
2798 {
2799         counters_add_agent (counter);
2800         return TRUE;
2801 }
2802
2803 static void
2804 counters_init (MonoProfiler *profiler)
2805 {
2806         mono_os_mutex_init (&counters_mutex);
2807
2808         mono_counters_on_register (&counters_add_agent);
2809         mono_counters_foreach (counters_init_foreach_callback, NULL);
2810 }
2811
2812 static void
2813 counters_emit (MonoProfiler *profiler)
2814 {
2815         MonoCounterAgent *agent;
2816         int len = 0;
2817         int size =
2818                 EVENT_SIZE /* event */ +
2819                 LEB128_SIZE /* len */
2820         ;
2821
2822         mono_os_mutex_lock (&counters_mutex);
2823
2824         for (agent = counters; agent; agent = agent->next) {
2825                 if (agent->emitted)
2826                         continue;
2827
2828                 size +=
2829                         LEB128_SIZE /* section */ +
2830                         strlen (mono_counter_get_name (agent->counter)) + 1 /* name */ +
2831                         BYTE_SIZE /* type */ +
2832                         BYTE_SIZE /* unit */ +
2833                         BYTE_SIZE /* variance */ +
2834                         LEB128_SIZE /* index */
2835                 ;
2836
2837                 len++;
2838         }
2839
2840         if (!len)
2841                 goto done;
2842
2843         ENTER_LOG (&counter_descriptors_ctr, logbuffer, size);
2844
2845         emit_event (logbuffer, TYPE_SAMPLE_COUNTERS_DESC | TYPE_SAMPLE);
2846         emit_value (logbuffer, len);
2847
2848         for (agent = counters; agent; agent = agent->next) {
2849                 const char *name;
2850
2851                 if (agent->emitted)
2852                         continue;
2853
2854                 name = mono_counter_get_name (agent->counter);
2855                 emit_value (logbuffer, mono_counter_get_section (agent->counter));
2856                 emit_string (logbuffer, name, strlen (name) + 1);
2857                 emit_byte (logbuffer, mono_counter_get_type (agent->counter));
2858                 emit_byte (logbuffer, mono_counter_get_unit (agent->counter));
2859                 emit_byte (logbuffer, mono_counter_get_variance (agent->counter));
2860                 emit_value (logbuffer, agent->index);
2861
2862                 agent->emitted = 1;
2863         }
2864
2865         EXIT_LOG_EXPLICIT (DO_SEND);
2866
2867 done:
2868         mono_os_mutex_unlock (&counters_mutex);
2869 }
2870
2871 static void
2872 counters_sample (MonoProfiler *profiler, uint64_t timestamp)
2873 {
2874         MonoCounterAgent *agent;
2875         MonoCounter *counter;
2876         int type;
2877         int buffer_size;
2878         void *buffer;
2879         int size;
2880
2881         counters_emit (profiler);
2882
2883         buffer_size = 8;
2884         buffer = g_calloc (1, buffer_size);
2885
2886         mono_os_mutex_lock (&counters_mutex);
2887
2888         size =
2889                 EVENT_SIZE /* event */
2890         ;
2891
2892         for (agent = counters; agent; agent = agent->next) {
2893                 size +=
2894                         LEB128_SIZE /* index */ +
2895                         BYTE_SIZE /* type */ +
2896                         mono_counter_get_size (agent->counter) /* value */
2897                 ;
2898         }
2899
2900         size +=
2901                 LEB128_SIZE /* stop marker */
2902         ;
2903
2904         ENTER_LOG (&counter_samples_ctr, logbuffer, size);
2905
2906         emit_event_time (logbuffer, TYPE_SAMPLE_COUNTERS | TYPE_SAMPLE, timestamp);
2907
2908         for (agent = counters; agent; agent = agent->next) {
2909                 size_t size;
2910
2911                 counter = agent->counter;
2912
2913                 size = mono_counter_get_size (counter);
2914
2915                 if (size > buffer_size) {
2916                         buffer_size = size;
2917                         buffer = g_realloc (buffer, buffer_size);
2918                 }
2919
2920                 memset (buffer, 0, buffer_size);
2921
2922                 g_assert (mono_counters_sample (counter, buffer, size));
2923
2924                 type = mono_counter_get_type (counter);
2925
2926                 if (!agent->value) {
2927                         agent->value = g_calloc (1, size);
2928                         agent->value_size = size;
2929                 } else {
2930                         if (type == MONO_COUNTER_STRING) {
2931                                 if (strcmp (agent->value, buffer) == 0)
2932                                         continue;
2933                         } else {
2934                                 if (agent->value_size == size && memcmp (agent->value, buffer, size) == 0)
2935                                         continue;
2936                         }
2937                 }
2938
2939                 emit_uvalue (logbuffer, agent->index);
2940                 emit_byte (logbuffer, type);
2941                 switch (type) {
2942                 case MONO_COUNTER_INT:
2943 #if SIZEOF_VOID_P == 4
2944                 case MONO_COUNTER_WORD:
2945 #endif
2946                         emit_svalue (logbuffer, *(int*)buffer - *(int*)agent->value);
2947                         break;
2948                 case MONO_COUNTER_UINT:
2949                         emit_uvalue (logbuffer, *(guint*)buffer - *(guint*)agent->value);
2950                         break;
2951                 case MONO_COUNTER_TIME_INTERVAL:
2952                 case MONO_COUNTER_LONG:
2953 #if SIZEOF_VOID_P == 8
2954                 case MONO_COUNTER_WORD:
2955 #endif
2956                         emit_svalue (logbuffer, *(gint64*)buffer - *(gint64*)agent->value);
2957                         break;
2958                 case MONO_COUNTER_ULONG:
2959                         emit_uvalue (logbuffer, *(guint64*)buffer - *(guint64*)agent->value);
2960                         break;
2961                 case MONO_COUNTER_DOUBLE:
2962                         emit_double (logbuffer, *(double*)buffer);
2963                         break;
2964                 case MONO_COUNTER_STRING:
2965                         if (size == 0) {
2966                                 emit_byte (logbuffer, 0);
2967                         } else {
2968                                 emit_byte (logbuffer, 1);
2969                                 emit_string (logbuffer, (char*)buffer, size);
2970                         }
2971                         break;
2972                 default:
2973                         g_assert_not_reached ();
2974                 }
2975
2976                 if (type == MONO_COUNTER_STRING && size > agent->value_size) {
2977                         agent->value = g_realloc (agent->value, size);
2978                         agent->value_size = size;
2979                 }
2980
2981                 if (size > 0)
2982                         memcpy (agent->value, buffer, size);
2983         }
2984         g_free (buffer);
2985
2986         emit_value (logbuffer, 0);
2987
2988         EXIT_LOG_EXPLICIT (DO_SEND);
2989
2990         mono_os_mutex_unlock (&counters_mutex);
2991 }
2992
2993 typedef struct _PerfCounterAgent PerfCounterAgent;
2994 struct _PerfCounterAgent {
2995         PerfCounterAgent *next;
2996         int index;
2997         char *category_name;
2998         char *name;
2999         int type;
3000         gint64 value;
3001         guint8 emitted;
3002         guint8 updated;
3003         guint8 deleted;
3004 };
3005
3006 static PerfCounterAgent *perfcounters = NULL;
3007
3008 static void
3009 perfcounters_emit (MonoProfiler *profiler)
3010 {
3011         PerfCounterAgent *pcagent;
3012         int len = 0;
3013         int size =
3014                 EVENT_SIZE /* event */ +
3015                 LEB128_SIZE /* len */
3016         ;
3017
3018         for (pcagent = perfcounters; pcagent; pcagent = pcagent->next) {
3019                 if (pcagent->emitted)
3020                         continue;
3021
3022                 size +=
3023                         LEB128_SIZE /* section */ +
3024                         strlen (pcagent->category_name) + 1 /* category name */ +
3025                         strlen (pcagent->name) + 1 /* name */ +
3026                         BYTE_SIZE /* type */ +
3027                         BYTE_SIZE /* unit */ +
3028                         BYTE_SIZE /* variance */ +
3029                         LEB128_SIZE /* index */
3030                 ;
3031
3032                 len++;
3033         }
3034
3035         if (!len)
3036                 return;
3037
3038         ENTER_LOG (&perfcounter_descriptors_ctr, logbuffer, size);
3039
3040         emit_event (logbuffer, TYPE_SAMPLE_COUNTERS_DESC | TYPE_SAMPLE);
3041         emit_value (logbuffer, len);
3042
3043         for (pcagent = perfcounters; pcagent; pcagent = pcagent->next) {
3044                 if (pcagent->emitted)
3045                         continue;
3046
3047                 emit_value (logbuffer, MONO_COUNTER_PERFCOUNTERS);
3048                 emit_string (logbuffer, pcagent->category_name, strlen (pcagent->category_name) + 1);
3049                 emit_string (logbuffer, pcagent->name, strlen (pcagent->name) + 1);
3050                 emit_byte (logbuffer, MONO_COUNTER_LONG);
3051                 emit_byte (logbuffer, MONO_COUNTER_RAW);
3052                 emit_byte (logbuffer, MONO_COUNTER_VARIABLE);
3053                 emit_value (logbuffer, pcagent->index);
3054
3055                 pcagent->emitted = 1;
3056         }
3057
3058         EXIT_LOG_EXPLICIT (DO_SEND);
3059 }
3060
3061 static gboolean
3062 perfcounters_foreach (char *category_name, char *name, unsigned char type, gint64 value, gpointer user_data)
3063 {
3064         PerfCounterAgent *pcagent;
3065
3066         for (pcagent = perfcounters; pcagent; pcagent = pcagent->next) {
3067                 if (strcmp (pcagent->category_name, category_name) != 0 || strcmp (pcagent->name, name) != 0)
3068                         continue;
3069                 if (pcagent->value == value)
3070                         return TRUE;
3071
3072                 pcagent->value = value;
3073                 pcagent->updated = 1;
3074                 pcagent->deleted = 0;
3075                 return TRUE;
3076         }
3077
3078         pcagent = g_new0 (PerfCounterAgent, 1);
3079         pcagent->next = perfcounters;
3080         pcagent->index = counters_index++;
3081         pcagent->category_name = g_strdup (category_name);
3082         pcagent->name = g_strdup (name);
3083         pcagent->type = (int) type;
3084         pcagent->value = value;
3085         pcagent->emitted = 0;
3086         pcagent->updated = 1;
3087         pcagent->deleted = 0;
3088
3089         perfcounters = pcagent;
3090
3091         return TRUE;
3092 }
3093
3094 static void
3095 perfcounters_sample (MonoProfiler *profiler, uint64_t timestamp)
3096 {
3097         PerfCounterAgent *pcagent;
3098         int len = 0;
3099         int size;
3100
3101         mono_os_mutex_lock (&counters_mutex);
3102
3103         /* mark all perfcounters as deleted, foreach will unmark them as necessary */
3104         for (pcagent = perfcounters; pcagent; pcagent = pcagent->next)
3105                 pcagent->deleted = 1;
3106
3107         mono_perfcounter_foreach (perfcounters_foreach, perfcounters);
3108
3109         perfcounters_emit (profiler);
3110
3111         size =
3112                 EVENT_SIZE /* event */
3113         ;
3114
3115         for (pcagent = perfcounters; pcagent; pcagent = pcagent->next) {
3116                 if (pcagent->deleted || !pcagent->updated)
3117                         continue;
3118
3119                 size +=
3120                         LEB128_SIZE /* index */ +
3121                         BYTE_SIZE /* type */ +
3122                         LEB128_SIZE /* value */
3123                 ;
3124
3125                 len++;
3126         }
3127
3128         if (!len)
3129                 goto done;
3130
3131         size +=
3132                 LEB128_SIZE /* stop marker */
3133         ;
3134
3135         ENTER_LOG (&perfcounter_samples_ctr, logbuffer, size);
3136
3137         emit_event_time (logbuffer, TYPE_SAMPLE_COUNTERS | TYPE_SAMPLE, timestamp);
3138
3139         for (pcagent = perfcounters; pcagent; pcagent = pcagent->next) {
3140                 if (pcagent->deleted || !pcagent->updated)
3141                         continue;
3142                 emit_uvalue (logbuffer, pcagent->index);
3143                 emit_byte (logbuffer, MONO_COUNTER_LONG);
3144                 emit_svalue (logbuffer, pcagent->value);
3145
3146                 pcagent->updated = 0;
3147         }
3148
3149         emit_value (logbuffer, 0);
3150
3151         EXIT_LOG_EXPLICIT (DO_SEND);
3152
3153 done:
3154         mono_os_mutex_unlock (&counters_mutex);
3155 }
3156
3157 static void
3158 counters_and_perfcounters_sample (MonoProfiler *prof)
3159 {
3160         uint64_t now = current_time ();
3161
3162         counters_sample (prof, now);
3163         perfcounters_sample (prof, now);
3164 }
3165
3166 #define COVERAGE_DEBUG(x) if (debug_coverage) {x}
3167 static mono_mutex_t coverage_mutex;
3168 static MonoConcurrentHashTable *coverage_methods = NULL;
3169 static MonoConcurrentHashTable *coverage_assemblies = NULL;
3170 static MonoConcurrentHashTable *coverage_classes = NULL;
3171
3172 static MonoConcurrentHashTable *filtered_classes = NULL;
3173 static MonoConcurrentHashTable *entered_methods = NULL;
3174 static MonoConcurrentHashTable *image_to_methods = NULL;
3175 static MonoConcurrentHashTable *suppressed_assemblies = NULL;
3176 static gboolean coverage_initialized = FALSE;
3177
3178 static GPtrArray *coverage_data = NULL;
3179 static int previous_offset = 0;
3180
3181 typedef struct {
3182         MonoLockFreeQueueNode node;
3183         MonoMethod *method;
3184 } MethodNode;
3185
3186 typedef struct {
3187         int offset;
3188         int counter;
3189         char *filename;
3190         int line;
3191         int column;
3192 } CoverageEntry;
3193
3194 static void
3195 free_coverage_entry (gpointer data, gpointer userdata)
3196 {
3197         CoverageEntry *entry = (CoverageEntry *)data;
3198         g_free (entry->filename);
3199         g_free (entry);
3200 }
3201
3202 static void
3203 obtain_coverage_for_method (MonoProfiler *prof, const MonoProfileCoverageEntry *entry)
3204 {
3205         int offset = entry->iloffset - previous_offset;
3206         CoverageEntry *e = g_new (CoverageEntry, 1);
3207
3208         previous_offset = entry->iloffset;
3209
3210         e->offset = offset;
3211         e->counter = entry->counter;
3212         e->filename = g_strdup(entry->filename ? entry->filename : "");
3213         e->line = entry->line;
3214         e->column = entry->col;
3215
3216         g_ptr_array_add (coverage_data, e);
3217 }
3218
3219 static char *
3220 parse_generic_type_names(char *name)
3221 {
3222         char *new_name, *ret;
3223         int within_generic_declaration = 0, generic_members = 1;
3224
3225         if (name == NULL || *name == '\0')
3226                 return g_strdup ("");
3227
3228         if (!(ret = new_name = (char *) g_calloc (strlen (name) * 4 + 1, sizeof (char))))
3229                 return NULL;
3230
3231         do {
3232                 switch (*name) {
3233                         case '<':
3234                                 within_generic_declaration = 1;
3235                                 break;
3236
3237                         case '>':
3238                                 within_generic_declaration = 0;
3239
3240                                 if (*(name - 1) != '<') {
3241                                         *new_name++ = '`';
3242                                         *new_name++ = '0' + generic_members;
3243                                 } else {
3244                                         memcpy (new_name, "&lt;&gt;", 8);
3245                                         new_name += 8;
3246                                 }
3247
3248                                 generic_members = 0;
3249                                 break;
3250
3251                         case ',':
3252                                 generic_members++;
3253                                 break;
3254
3255                         default:
3256                                 if (!within_generic_declaration)
3257                                         *new_name++ = *name;
3258
3259                                 break;
3260                 }
3261         } while (*name++);
3262
3263         return ret;
3264 }
3265
3266 static int method_id;
3267 static void
3268 build_method_buffer (gpointer key, gpointer value, gpointer userdata)
3269 {
3270         MonoMethod *method = (MonoMethod *)value;
3271         MonoProfiler *prof = (MonoProfiler *)userdata;
3272         MonoClass *klass;
3273         MonoImage *image;
3274         char *class_name;
3275         const char *image_name, *method_name, *sig, *first_filename;
3276         guint i;
3277
3278         previous_offset = 0;
3279         coverage_data = g_ptr_array_new ();
3280
3281         mono_profiler_coverage_get (prof, method, obtain_coverage_for_method);
3282
3283         klass = mono_method_get_class (method);
3284         image = mono_class_get_image (klass);
3285         image_name = mono_image_get_name (image);
3286
3287         sig = mono_signature_get_desc (mono_method_signature (method), TRUE);
3288         class_name = parse_generic_type_names (mono_type_get_name (mono_class_get_type (klass)));
3289         method_name = mono_method_get_name (method);
3290
3291         if (coverage_data->len != 0) {
3292                 CoverageEntry *entry = (CoverageEntry *)coverage_data->pdata[0];
3293                 first_filename = entry->filename ? entry->filename : "";
3294         } else
3295                 first_filename = "";
3296
3297         image_name = image_name ? image_name : "";
3298         sig = sig ? sig : "";
3299         method_name = method_name ? method_name : "";
3300
3301         ENTER_LOG (&coverage_methods_ctr, logbuffer,
3302                 EVENT_SIZE /* event */ +
3303                 strlen (image_name) + 1 /* image name */ +
3304                 strlen (class_name) + 1 /* class name */ +
3305                 strlen (method_name) + 1 /* method name */ +
3306                 strlen (sig) + 1 /* signature */ +
3307                 strlen (first_filename) + 1 /* first file name */ +
3308                 LEB128_SIZE /* token */ +
3309                 LEB128_SIZE /* method id */ +
3310                 LEB128_SIZE /* entries */
3311         );
3312
3313         emit_event (logbuffer, TYPE_COVERAGE_METHOD | TYPE_COVERAGE);
3314         emit_string (logbuffer, image_name, strlen (image_name) + 1);
3315         emit_string (logbuffer, class_name, strlen (class_name) + 1);
3316         emit_string (logbuffer, method_name, strlen (method_name) + 1);
3317         emit_string (logbuffer, sig, strlen (sig) + 1);
3318         emit_string (logbuffer, first_filename, strlen (first_filename) + 1);
3319
3320         emit_uvalue (logbuffer, mono_method_get_token (method));
3321         emit_uvalue (logbuffer, method_id);
3322         emit_value (logbuffer, coverage_data->len);
3323
3324         EXIT_LOG_EXPLICIT (DO_SEND);
3325
3326         for (i = 0; i < coverage_data->len; i++) {
3327                 CoverageEntry *entry = (CoverageEntry *)coverage_data->pdata[i];
3328
3329                 ENTER_LOG (&coverage_statements_ctr, logbuffer,
3330                         EVENT_SIZE /* event */ +
3331                         LEB128_SIZE /* method id */ +
3332                         LEB128_SIZE /* offset */ +
3333                         LEB128_SIZE /* counter */ +
3334                         LEB128_SIZE /* line */ +
3335                         LEB128_SIZE /* column */
3336                 );
3337
3338                 emit_event (logbuffer, TYPE_COVERAGE_STATEMENT | TYPE_COVERAGE);
3339                 emit_uvalue (logbuffer, method_id);
3340                 emit_uvalue (logbuffer, entry->offset);
3341                 emit_uvalue (logbuffer, entry->counter);
3342                 emit_uvalue (logbuffer, entry->line);
3343                 emit_uvalue (logbuffer, entry->column);
3344
3345                 EXIT_LOG_EXPLICIT (DO_SEND);
3346         }
3347
3348         method_id++;
3349
3350         g_free (class_name);
3351
3352         g_ptr_array_foreach (coverage_data, free_coverage_entry, NULL);
3353         g_ptr_array_free (coverage_data, TRUE);
3354         coverage_data = NULL;
3355 }
3356
3357 /* This empties the queue */
3358 static guint
3359 count_queue (MonoLockFreeQueue *queue)
3360 {
3361         MonoLockFreeQueueNode *node;
3362         guint count = 0;
3363
3364         while ((node = mono_lock_free_queue_dequeue (queue))) {
3365                 count++;
3366                 mono_thread_hazardous_try_free (node, g_free);
3367         }
3368
3369         return count;
3370 }
3371
3372 static void
3373 build_class_buffer (gpointer key, gpointer value, gpointer userdata)
3374 {
3375         MonoClass *klass = (MonoClass *)key;
3376         MonoLockFreeQueue *class_methods = (MonoLockFreeQueue *)value;
3377         MonoImage *image;
3378         char *class_name;
3379         const char *assembly_name;
3380         int number_of_methods, partially_covered;
3381         guint fully_covered;
3382
3383         image = mono_class_get_image (klass);
3384         assembly_name = mono_image_get_name (image);
3385         class_name = mono_type_get_name (mono_class_get_type (klass));
3386
3387         assembly_name = assembly_name ? assembly_name : "";
3388         number_of_methods = mono_class_num_methods (klass);
3389         fully_covered = count_queue (class_methods);
3390         /* We don't handle partial covered yet */
3391         partially_covered = 0;
3392
3393         ENTER_LOG (&coverage_classes_ctr, logbuffer,
3394                 EVENT_SIZE /* event */ +
3395                 strlen (assembly_name) + 1 /* assembly name */ +
3396                 strlen (class_name) + 1 /* class name */ +
3397                 LEB128_SIZE /* no. methods */ +
3398                 LEB128_SIZE /* fully covered */ +
3399                 LEB128_SIZE /* partially covered */
3400         );
3401
3402         emit_event (logbuffer, TYPE_COVERAGE_CLASS | TYPE_COVERAGE);
3403         emit_string (logbuffer, assembly_name, strlen (assembly_name) + 1);
3404         emit_string (logbuffer, class_name, strlen (class_name) + 1);
3405         emit_uvalue (logbuffer, number_of_methods);
3406         emit_uvalue (logbuffer, fully_covered);
3407         emit_uvalue (logbuffer, partially_covered);
3408
3409         EXIT_LOG_EXPLICIT (DO_SEND);
3410
3411         g_free (class_name);
3412 }
3413
3414 static void
3415 get_coverage_for_image (MonoImage *image, int *number_of_methods, guint *fully_covered, int *partially_covered)
3416 {
3417         MonoLockFreeQueue *image_methods = (MonoLockFreeQueue *)mono_conc_hashtable_lookup (image_to_methods, image);
3418
3419         *number_of_methods = mono_image_get_table_rows (image, MONO_TABLE_METHOD);
3420         if (image_methods)
3421                 *fully_covered = count_queue (image_methods);
3422         else
3423                 *fully_covered = 0;
3424
3425         // FIXME: We don't handle partially covered yet.
3426         *partially_covered = 0;
3427 }
3428
3429 static void
3430 build_assembly_buffer (gpointer key, gpointer value, gpointer userdata)
3431 {
3432         MonoAssembly *assembly = (MonoAssembly *)value;
3433         MonoImage *image = mono_assembly_get_image (assembly);
3434         const char *name, *guid, *filename;
3435         int number_of_methods = 0, partially_covered = 0;
3436         guint fully_covered = 0;
3437
3438         name = mono_image_get_name (image);
3439         guid = mono_image_get_guid (image);
3440         filename = mono_image_get_filename (image);
3441
3442         name = name ? name : "";
3443         guid = guid ? guid : "";
3444         filename = filename ? filename : "";
3445
3446         get_coverage_for_image (image, &number_of_methods, &fully_covered, &partially_covered);
3447
3448         ENTER_LOG (&coverage_assemblies_ctr, logbuffer,
3449                 EVENT_SIZE /* event */ +
3450                 strlen (name) + 1 /* name */ +
3451                 strlen (guid) + 1 /* guid */ +
3452                 strlen (filename) + 1 /* file name */ +
3453                 LEB128_SIZE /* no. methods */ +
3454                 LEB128_SIZE /* fully covered */ +
3455                 LEB128_SIZE /* partially covered */
3456         );
3457
3458         emit_event (logbuffer, TYPE_COVERAGE_ASSEMBLY | TYPE_COVERAGE);
3459         emit_string (logbuffer, name, strlen (name) + 1);
3460         emit_string (logbuffer, guid, strlen (guid) + 1);
3461         emit_string (logbuffer, filename, strlen (filename) + 1);
3462         emit_uvalue (logbuffer, number_of_methods);
3463         emit_uvalue (logbuffer, fully_covered);
3464         emit_uvalue (logbuffer, partially_covered);
3465
3466         EXIT_LOG_EXPLICIT (DO_SEND);
3467 }
3468
3469 static void
3470 dump_coverage (MonoProfiler *prof)
3471 {
3472         if (!coverage_initialized)
3473                 return;
3474
3475         COVERAGE_DEBUG(fprintf (stderr, "Coverage: Started dump\n");)
3476         method_id = 0;
3477
3478         mono_os_mutex_lock (&coverage_mutex);
3479         mono_conc_hashtable_foreach (coverage_assemblies, build_assembly_buffer, NULL);
3480         mono_conc_hashtable_foreach (coverage_classes, build_class_buffer, NULL);
3481         mono_conc_hashtable_foreach (coverage_methods, build_method_buffer, prof);
3482         mono_os_mutex_unlock (&coverage_mutex);
3483
3484         COVERAGE_DEBUG(fprintf (stderr, "Coverage: Finished dump\n");)
3485 }
3486
3487 static void
3488 process_method_enter_coverage (MonoProfiler *prof, MonoMethod *method)
3489 {
3490         MonoClass *klass;
3491         MonoImage *image;
3492
3493         if (!coverage_initialized)
3494                 return;
3495
3496         klass = mono_method_get_class (method);
3497         image = mono_class_get_image (klass);
3498
3499         if (mono_conc_hashtable_lookup (suppressed_assemblies, (gpointer) mono_image_get_name (image)))
3500                 return;
3501
3502         mono_os_mutex_lock (&coverage_mutex);
3503         mono_conc_hashtable_insert (entered_methods, method, method);
3504         mono_os_mutex_unlock (&coverage_mutex);
3505 }
3506
3507 static MonoLockFreeQueueNode *
3508 create_method_node (MonoMethod *method)
3509 {
3510         MethodNode *node = (MethodNode *) g_malloc (sizeof (MethodNode));
3511         mono_lock_free_queue_node_init ((MonoLockFreeQueueNode *) node, FALSE);
3512         node->method = method;
3513
3514         return (MonoLockFreeQueueNode *) node;
3515 }
3516
3517 static gboolean
3518 coverage_filter (MonoProfiler *prof, MonoMethod *method)
3519 {
3520         MonoError error;
3521         MonoClass *klass;
3522         MonoImage *image;
3523         MonoAssembly *assembly;
3524         MonoMethodHeader *header;
3525         guint32 iflags, flags, code_size;
3526         char *fqn, *classname;
3527         gboolean has_positive, found;
3528         MonoLockFreeQueue *image_methods, *class_methods;
3529         MonoLockFreeQueueNode *node;
3530
3531         g_assert (coverage_initialized && "Why are we being asked for coverage filter info when we're not doing coverage?");
3532
3533         COVERAGE_DEBUG(fprintf (stderr, "Coverage filter for %s\n", mono_method_get_name (method));)
3534
3535         flags = mono_method_get_flags (method, &iflags);
3536         if ((iflags & 0x1000 /*METHOD_IMPL_ATTRIBUTE_INTERNAL_CALL*/) ||
3537             (flags & 0x2000 /*METHOD_ATTRIBUTE_PINVOKE_IMPL*/)) {
3538                 COVERAGE_DEBUG(fprintf (stderr, "   Internal call or pinvoke - ignoring\n");)
3539                 return FALSE;
3540         }
3541
3542         // Don't need to do anything else if we're already tracking this method
3543         if (mono_conc_hashtable_lookup (coverage_methods, method)) {
3544                 COVERAGE_DEBUG(fprintf (stderr, "   Already tracking\n");)
3545                 return TRUE;
3546         }
3547
3548         klass = mono_method_get_class (method);
3549         image = mono_class_get_image (klass);
3550
3551         // Don't handle coverage for the core assemblies
3552         if (mono_conc_hashtable_lookup (suppressed_assemblies, (gpointer) mono_image_get_name (image)) != NULL)
3553                 return FALSE;
3554
3555         if (prof->coverage_filters) {
3556                 /* Check already filtered classes first */
3557                 if (mono_conc_hashtable_lookup (filtered_classes, klass)) {
3558                         COVERAGE_DEBUG(fprintf (stderr, "   Already filtered\n");)
3559                         return FALSE;
3560                 }
3561
3562                 classname = mono_type_get_name (mono_class_get_type (klass));
3563
3564                 fqn = g_strdup_printf ("[%s]%s", mono_image_get_name (image), classname);
3565
3566                 COVERAGE_DEBUG(fprintf (stderr, "   Looking for %s in filter\n", fqn);)
3567                 // Check positive filters first
3568                 has_positive = FALSE;
3569                 found = FALSE;
3570                 for (guint i = 0; i < prof->coverage_filters->len; ++i) {
3571                         char *filter = (char *)g_ptr_array_index (prof->coverage_filters, i);
3572
3573                         if (filter [0] == '+') {
3574                                 filter = &filter [1];
3575
3576                                 COVERAGE_DEBUG(fprintf (stderr, "   Checking against +%s ...", filter);)
3577
3578                                 if (strstr (fqn, filter) != NULL) {
3579                                         COVERAGE_DEBUG(fprintf (stderr, "matched\n");)
3580                                         found = TRUE;
3581                                 } else
3582                                         COVERAGE_DEBUG(fprintf (stderr, "no match\n");)
3583
3584                                 has_positive = TRUE;
3585                         }
3586                 }
3587
3588                 if (has_positive && !found) {
3589                         COVERAGE_DEBUG(fprintf (stderr, "   Positive match was not found\n");)
3590
3591                         mono_os_mutex_lock (&coverage_mutex);
3592                         mono_conc_hashtable_insert (filtered_classes, klass, klass);
3593                         mono_os_mutex_unlock (&coverage_mutex);
3594                         g_free (fqn);
3595                         g_free (classname);
3596
3597                         return FALSE;
3598                 }
3599
3600                 for (guint i = 0; i < prof->coverage_filters->len; ++i) {
3601                         // FIXME: Is substring search sufficient?
3602                         char *filter = (char *)g_ptr_array_index (prof->coverage_filters, i);
3603                         if (filter [0] == '+')
3604                                 continue;
3605
3606                         // Skip '-'
3607                         filter = &filter [1];
3608                         COVERAGE_DEBUG(fprintf (stderr, "   Checking against -%s ...", filter);)
3609
3610                         if (strstr (fqn, filter) != NULL) {
3611                                 COVERAGE_DEBUG(fprintf (stderr, "matched\n");)
3612
3613                                 mono_os_mutex_lock (&coverage_mutex);
3614                                 mono_conc_hashtable_insert (filtered_classes, klass, klass);
3615                                 mono_os_mutex_unlock (&coverage_mutex);
3616                                 g_free (fqn);
3617                                 g_free (classname);
3618
3619                                 return FALSE;
3620                         } else
3621                                 COVERAGE_DEBUG(fprintf (stderr, "no match\n");)
3622
3623                 }
3624
3625                 g_free (fqn);
3626                 g_free (classname);
3627         }
3628
3629         COVERAGE_DEBUG(fprintf (stderr, "   Handling coverage for %s\n", mono_method_get_name (method));)
3630         header = mono_method_get_header_checked (method, &error);
3631         mono_error_cleanup (&error);
3632
3633         mono_method_header_get_code (header, &code_size, NULL);
3634
3635         assembly = mono_image_get_assembly (image);
3636
3637         // Need to keep the assemblies around for as long as they are kept in the hashtable
3638         // Nunit, for example, has a habit of unloading them before the coverage statistics are
3639         // generated causing a crash. See https://bugzilla.xamarin.com/show_bug.cgi?id=39325
3640         mono_assembly_addref (assembly);
3641
3642         mono_os_mutex_lock (&coverage_mutex);
3643         mono_conc_hashtable_insert (coverage_methods, method, method);
3644         mono_conc_hashtable_insert (coverage_assemblies, assembly, assembly);
3645         mono_os_mutex_unlock (&coverage_mutex);
3646
3647         image_methods = (MonoLockFreeQueue *)mono_conc_hashtable_lookup (image_to_methods, image);
3648
3649         if (image_methods == NULL) {
3650                 image_methods = (MonoLockFreeQueue *) g_malloc (sizeof (MonoLockFreeQueue));
3651                 mono_lock_free_queue_init (image_methods);
3652                 mono_os_mutex_lock (&coverage_mutex);
3653                 mono_conc_hashtable_insert (image_to_methods, image, image_methods);
3654                 mono_os_mutex_unlock (&coverage_mutex);
3655         }
3656
3657         node = create_method_node (method);
3658         mono_lock_free_queue_enqueue (image_methods, node);
3659
3660         class_methods = (MonoLockFreeQueue *)mono_conc_hashtable_lookup (coverage_classes, klass);
3661
3662         if (class_methods == NULL) {
3663                 class_methods = (MonoLockFreeQueue *) g_malloc (sizeof (MonoLockFreeQueue));
3664                 mono_lock_free_queue_init (class_methods);
3665                 mono_os_mutex_lock (&coverage_mutex);
3666                 mono_conc_hashtable_insert (coverage_classes, klass, class_methods);
3667                 mono_os_mutex_unlock (&coverage_mutex);
3668         }
3669
3670         node = create_method_node (method);
3671         mono_lock_free_queue_enqueue (class_methods, node);
3672
3673         return TRUE;
3674 }
3675
3676 #define LINE_BUFFER_SIZE 4096
3677 /* Max file limit of 128KB */
3678 #define MAX_FILE_SIZE 128 * 1024
3679 static char *
3680 get_file_content (FILE *stream)
3681 {
3682         char *buffer;
3683         ssize_t bytes_read;
3684         long filesize;
3685         int res, offset = 0;
3686
3687         res = fseek (stream, 0, SEEK_END);
3688         if (res < 0)
3689           return NULL;
3690
3691         filesize = ftell (stream);
3692         if (filesize < 0)
3693           return NULL;
3694
3695         res = fseek (stream, 0, SEEK_SET);
3696         if (res < 0)
3697           return NULL;
3698
3699         if (filesize > MAX_FILE_SIZE)
3700           return NULL;
3701
3702         buffer = (char *) g_malloc ((filesize + 1) * sizeof (char));
3703         while ((bytes_read = fread (buffer + offset, 1, LINE_BUFFER_SIZE, stream)) > 0)
3704                 offset += bytes_read;
3705
3706         /* NULL terminate our buffer */
3707         buffer[filesize] = '\0';
3708         return buffer;
3709 }
3710
3711 static char *
3712 get_next_line (char *contents, char **next_start)
3713 {
3714         char *p = contents;
3715
3716         if (p == NULL || *p == '\0') {
3717                 *next_start = NULL;
3718                 return NULL;
3719         }
3720
3721         while (*p != '\n' && *p != '\0')
3722                 p++;
3723
3724         if (*p == '\n') {
3725                 *p = '\0';
3726                 *next_start = p + 1;
3727         } else
3728                 *next_start = NULL;
3729
3730         return contents;
3731 }
3732
3733 static void
3734 init_suppressed_assemblies (void)
3735 {
3736         char *content;
3737         char *line;
3738         FILE *sa_file;
3739
3740         suppressed_assemblies = mono_conc_hashtable_new (g_str_hash, g_str_equal);
3741         sa_file = fopen (SUPPRESSION_DIR "/mono-profiler-log.suppression", "r");
3742         if (sa_file == NULL)
3743                 return;
3744
3745         /* Don't need to free @content as it is referred to by the lines stored in @suppressed_assemblies */
3746         content = get_file_content (sa_file);
3747         if (content == NULL) {
3748                 g_error ("mono-profiler-log.suppression is greater than 128kb - aborting\n");
3749         }
3750
3751         while ((line = get_next_line (content, &content))) {
3752                 line = g_strchomp (g_strchug (line));
3753                 /* No locking needed as we're doing initialization */
3754                 mono_conc_hashtable_insert (suppressed_assemblies, line, line);
3755         }
3756
3757         fclose (sa_file);
3758 }
3759
3760 static void
3761 parse_cov_filter_file (GPtrArray *filters, const char *file)
3762 {
3763         FILE *filter_file;
3764         char *line, *content;
3765
3766         filter_file = fopen (file, "r");
3767         if (filter_file == NULL) {
3768                 fprintf (stderr, "Unable to open %s\n", file);
3769                 return;
3770         }
3771
3772         /* Don't need to free content as it is referred to by the lines stored in @filters */
3773         content = get_file_content (filter_file);
3774         if (content == NULL)
3775                 fprintf (stderr, "WARNING: %s is greater than 128kb - ignoring\n", file);
3776
3777         while ((line = get_next_line (content, &content)))
3778                 g_ptr_array_add (filters, g_strchug (g_strchomp (line)));
3779
3780         fclose (filter_file);
3781 }
3782
3783 static void
3784 coverage_init (MonoProfiler *prof)
3785 {
3786         g_assert (!coverage_initialized && "Why are we initializing coverage twice?");
3787
3788         COVERAGE_DEBUG(fprintf (stderr, "Coverage initialized\n");)
3789
3790         mono_os_mutex_init (&coverage_mutex);
3791         coverage_methods = mono_conc_hashtable_new (NULL, NULL);
3792         coverage_assemblies = mono_conc_hashtable_new (NULL, NULL);
3793         coverage_classes = mono_conc_hashtable_new (NULL, NULL);
3794         filtered_classes = mono_conc_hashtable_new (NULL, NULL);
3795         entered_methods = mono_conc_hashtable_new (NULL, NULL);
3796         image_to_methods = mono_conc_hashtable_new (NULL, NULL);
3797         init_suppressed_assemblies ();
3798
3799         coverage_initialized = TRUE;
3800 }
3801
3802 static void
3803 unref_coverage_assemblies (gpointer key, gpointer value, gpointer userdata)
3804 {
3805         MonoAssembly *assembly = (MonoAssembly *)value;
3806         mono_assembly_close (assembly);
3807 }
3808
3809 static void
3810 free_sample_hit (gpointer p)
3811 {
3812         mono_lock_free_free (p, SAMPLE_BLOCK_SIZE);
3813 }
3814
3815 static void
3816 cleanup_reusable_samples (MonoProfiler *prof)
3817 {
3818         SampleHit *sample;
3819
3820         while ((sample = (SampleHit *) mono_lock_free_queue_dequeue (&prof->sample_reuse_queue)))
3821                 mono_thread_hazardous_try_free (sample, free_sample_hit);
3822 }
3823
3824 static void
3825 log_shutdown (MonoProfiler *prof)
3826 {
3827         InterlockedWrite (&in_shutdown, 1);
3828
3829         if (!no_counters)
3830                 counters_and_perfcounters_sample (prof);
3831
3832         dump_coverage (prof);
3833
3834         char c = 1;
3835
3836         if (write (prof->pipes [1], &c, 1) != 1) {
3837                 fprintf (stderr, "Could not write to pipe: %s\n", strerror (errno));
3838                 exit (1);
3839         }
3840
3841         mono_native_thread_join (prof->helper_thread);
3842
3843         mono_os_mutex_destroy (&counters_mutex);
3844
3845         MonoCounterAgent *mc_next;
3846
3847         for (MonoCounterAgent *cur = counters; cur; cur = mc_next) {
3848                 mc_next = cur->next;
3849                 g_free (cur);
3850         }
3851
3852         PerfCounterAgent *pc_next;
3853
3854         for (PerfCounterAgent *cur = perfcounters; cur; cur = pc_next) {
3855                 pc_next = cur->next;
3856                 g_free (cur);
3857         }
3858
3859         /*
3860          * Ensure that we empty the LLS completely, even if some nodes are
3861          * not immediately removed upon calling mono_lls_remove (), by
3862          * iterating until the head is NULL.
3863          */
3864         while (profiler_thread_list.head) {
3865                 MONO_LLS_FOREACH_SAFE (&profiler_thread_list, MonoProfilerThread, thread) {
3866                         g_assert (thread->attached && "Why is a thread in the LLS not attached?");
3867
3868                         remove_thread (thread);
3869                 } MONO_LLS_FOREACH_SAFE_END
3870         }
3871
3872         /*
3873          * Ensure that all threads have been freed, so that we don't miss any
3874          * buffers when we shut down the writer thread below.
3875          */
3876         mono_thread_hazardous_try_free_all ();
3877
3878         InterlockedWrite (&prof->run_dumper_thread, 0);
3879         mono_os_sem_post (&prof->dumper_queue_sem);
3880         mono_native_thread_join (prof->dumper_thread);
3881         mono_os_sem_destroy (&prof->dumper_queue_sem);
3882
3883         InterlockedWrite (&prof->run_writer_thread, 0);
3884         mono_os_sem_post (&prof->writer_queue_sem);
3885         mono_native_thread_join (prof->writer_thread);
3886         mono_os_sem_destroy (&prof->writer_queue_sem);
3887
3888         /*
3889          * Free all writer queue entries, and ensure that all sample hits will be
3890          * added to the sample reuse queue.
3891          */
3892         mono_thread_hazardous_try_free_all ();
3893
3894         cleanup_reusable_samples (prof);
3895
3896         /*
3897          * Finally, make sure that all sample hits are freed. This should cover all
3898          * hazardous data from the profiler. We can now be sure that the runtime
3899          * won't later invoke free functions in the profiler library after it has
3900          * been unloaded.
3901          */
3902         mono_thread_hazardous_try_free_all ();
3903
3904         g_assert (!InterlockedRead (&buffer_rwlock_count) && "Why is the reader count still non-zero?");
3905         g_assert (!InterlockedReadPointer (&buffer_rwlock_exclusive) && "Why does someone still hold the exclusive lock?");
3906
3907 #if defined (HAVE_SYS_ZLIB)
3908         if (prof->gzfile)
3909                 gzclose (prof->gzfile);
3910 #endif
3911         if (prof->pipe_output)
3912                 pclose (prof->file);
3913         else
3914                 fclose (prof->file);
3915
3916         mono_conc_hashtable_destroy (prof->method_table);
3917         mono_os_mutex_destroy (&prof->method_table_mutex);
3918
3919         if (coverage_initialized) {
3920                 mono_os_mutex_lock (&coverage_mutex);
3921                 mono_conc_hashtable_foreach (coverage_assemblies, unref_coverage_assemblies, prof);
3922                 mono_os_mutex_unlock (&coverage_mutex);
3923
3924                 mono_conc_hashtable_destroy (coverage_methods);
3925                 mono_conc_hashtable_destroy (coverage_assemblies);
3926                 mono_conc_hashtable_destroy (coverage_classes);
3927                 mono_conc_hashtable_destroy (filtered_classes);
3928
3929                 mono_conc_hashtable_destroy (entered_methods);
3930                 mono_conc_hashtable_destroy (image_to_methods);
3931                 mono_conc_hashtable_destroy (suppressed_assemblies);
3932                 mono_os_mutex_destroy (&coverage_mutex);
3933         }
3934
3935         PROF_TLS_FREE ();
3936
3937         g_free (prof->args);
3938         g_free (prof);
3939 }
3940
3941 static char*
3942 new_filename (const char* filename)
3943 {
3944         time_t t = time (NULL);
3945         int pid = process_id ();
3946         char pid_buf [16];
3947         char time_buf [16];
3948         char *res, *d;
3949         const char *p;
3950         int count_dates = 0;
3951         int count_pids = 0;
3952         int s_date, s_pid;
3953         struct tm *ts;
3954         for (p = filename; *p; p++) {
3955                 if (*p != '%')
3956                         continue;
3957                 p++;
3958                 if (*p == 't')
3959                         count_dates++;
3960                 else if (*p == 'p')
3961                         count_pids++;
3962                 else if (*p == 0)
3963                         break;
3964         }
3965         if (!count_dates && !count_pids)
3966                 return pstrdup (filename);
3967         snprintf (pid_buf, sizeof (pid_buf), "%d", pid);
3968         ts = gmtime (&t);
3969         snprintf (time_buf, sizeof (time_buf), "%d%02d%02d%02d%02d%02d",
3970                 1900 + ts->tm_year, 1 + ts->tm_mon, ts->tm_mday, ts->tm_hour, ts->tm_min, ts->tm_sec);
3971         s_date = strlen (time_buf);
3972         s_pid = strlen (pid_buf);
3973         d = res = (char *) g_malloc (strlen (filename) + s_date * count_dates + s_pid * count_pids);
3974         for (p = filename; *p; p++) {
3975                 if (*p != '%') {
3976                         *d++ = *p;
3977                         continue;
3978                 }
3979                 p++;
3980                 if (*p == 't') {
3981                         strcpy (d, time_buf);
3982                         d += s_date;
3983                         continue;
3984                 } else if (*p == 'p') {
3985                         strcpy (d, pid_buf);
3986                         d += s_pid;
3987                         continue;
3988                 } else if (*p == '%') {
3989                         *d++ = '%';
3990                         continue;
3991                 } else if (*p == 0)
3992                         break;
3993                 *d++ = '%';
3994                 *d++ = *p;
3995         }
3996         *d = 0;
3997         return res;
3998 }
3999
4000 static void
4001 add_to_fd_set (fd_set *set, int fd, int *max_fd)
4002 {
4003         /*
4004          * This should only trigger for the basic FDs (server socket, pipes) at
4005          * startup if for some mysterious reason they're too large. In this case,
4006          * the profiler really can't function, and we're better off printing an
4007          * error and exiting.
4008          */
4009         if (fd >= FD_SETSIZE) {
4010                 fprintf (stderr, "File descriptor is out of bounds for fd_set: %d\n", fd);
4011                 exit (1);
4012         }
4013
4014         FD_SET (fd, set);
4015
4016         if (*max_fd < fd)
4017                 *max_fd = fd;
4018 }
4019
4020 static void *
4021 helper_thread (void *arg)
4022 {
4023         MonoProfiler *prof = (MonoProfiler *) arg;
4024
4025         mono_threads_attach_tools_thread ();
4026         mono_native_thread_set_name (mono_native_thread_id_get (), "Profiler helper");
4027
4028         MonoProfilerThread *thread = init_thread (prof, FALSE);
4029
4030         GArray *command_sockets = g_array_new (FALSE, FALSE, sizeof (int));
4031
4032         while (1) {
4033                 fd_set rfds;
4034                 int max_fd = -1;
4035
4036                 FD_ZERO (&rfds);
4037
4038                 add_to_fd_set (&rfds, prof->server_socket, &max_fd);
4039                 add_to_fd_set (&rfds, prof->pipes [0], &max_fd);
4040
4041                 for (gint i = 0; i < command_sockets->len; i++)
4042                         add_to_fd_set (&rfds, g_array_index (command_sockets, int, i), &max_fd);
4043
4044                 struct timeval tv = { .tv_sec = 1, .tv_usec = 0 };
4045
4046                 // Sleep for 1sec or until a file descriptor has data.
4047                 if (select (max_fd + 1, &rfds, NULL, NULL, &tv) == -1) {
4048                         if (errno == EINTR)
4049                                 continue;
4050
4051                         fprintf (stderr, "Error in mono-profiler-log server: %s", strerror (errno));
4052                         exit (1);
4053                 }
4054
4055                 if (!no_counters)
4056                         counters_and_perfcounters_sample (prof);
4057
4058                 buffer_lock_excl ();
4059
4060                 sync_point (SYNC_POINT_PERIODIC);
4061
4062                 buffer_unlock_excl ();
4063
4064                 // Are we shutting down?
4065                 if (FD_ISSET (prof->pipes [0], &rfds)) {
4066                         char c;
4067                         read (prof->pipes [0], &c, 1);
4068                         break;
4069                 }
4070
4071                 for (gint i = 0; i < command_sockets->len; i++) {
4072                         int fd = g_array_index (command_sockets, int, i);
4073
4074                         if (!FD_ISSET (fd, &rfds))
4075                                 continue;
4076
4077                         char buf [64];
4078                         int len = read (fd, buf, sizeof (buf) - 1);
4079
4080                         if (len == -1)
4081                                 continue;
4082
4083                         if (!len) {
4084                                 // The other end disconnected.
4085                                 g_array_remove_index (command_sockets, i);
4086                                 close (fd);
4087
4088                                 continue;
4089                         }
4090
4091                         buf [len] = 0;
4092
4093                         if (!strcmp (buf, "heapshot\n") && hs_mode_ondemand) {
4094                                 // Rely on the finalization callback triggering a GC.
4095                                 heapshot_requested = 1;
4096                                 mono_gc_finalize_notify ();
4097                         }
4098                 }
4099
4100                 if (FD_ISSET (prof->server_socket, &rfds)) {
4101                         int fd = accept (prof->server_socket, NULL, NULL);
4102
4103                         if (fd != -1) {
4104                                 if (fd >= FD_SETSIZE)
4105                                         close (fd);
4106                                 else
4107                                         g_array_append_val (command_sockets, fd);
4108                         }
4109                 }
4110         }
4111
4112         for (gint i = 0; i < command_sockets->len; i++)
4113                 close (g_array_index (command_sockets, int, i));
4114
4115         g_array_free (command_sockets, TRUE);
4116
4117         send_log_unsafe (FALSE);
4118         deinit_thread (thread);
4119
4120         mono_thread_info_detach ();
4121
4122         return NULL;
4123 }
4124
4125 static void
4126 start_helper_thread (MonoProfiler* prof)
4127 {
4128         if (pipe (prof->pipes) == -1) {
4129                 fprintf (stderr, "Cannot create pipe: %s\n", strerror (errno));
4130                 exit (1);
4131         }
4132
4133         prof->server_socket = socket (PF_INET, SOCK_STREAM, 0);
4134
4135         if (prof->server_socket == -1) {
4136                 fprintf (stderr, "Cannot create server socket: %s\n", strerror (errno));
4137                 exit (1);
4138         }
4139
4140         struct sockaddr_in server_address;
4141
4142         memset (&server_address, 0, sizeof (server_address));
4143         server_address.sin_family = AF_INET;
4144         server_address.sin_addr.s_addr = INADDR_ANY;
4145         server_address.sin_port = htons (prof->command_port);
4146
4147         if (bind (prof->server_socket, (struct sockaddr *) &server_address, sizeof (server_address)) == -1) {
4148                 fprintf (stderr, "Cannot bind server socket on port %d: %s\n", prof->command_port, strerror (errno));
4149                 close (prof->server_socket);
4150                 exit (1);
4151         }
4152
4153         if (listen (prof->server_socket, 1) == -1) {
4154                 fprintf (stderr, "Cannot listen on server socket: %s\n", strerror (errno));
4155                 close (prof->server_socket);
4156                 exit (1);
4157         }
4158
4159         socklen_t slen = sizeof (server_address);
4160
4161         if (getsockname (prof->server_socket, (struct sockaddr *) &server_address, &slen)) {
4162                 fprintf (stderr, "Could not get assigned port: %s\n", strerror (errno));
4163                 close (prof->server_socket);
4164                 exit (1);
4165         }
4166
4167         prof->command_port = ntohs (server_address.sin_port);
4168
4169         if (!mono_native_thread_create (&prof->helper_thread, helper_thread, prof)) {
4170                 fprintf (stderr, "Could not start helper thread\n");
4171                 close (prof->server_socket);
4172                 exit (1);
4173         }
4174 }
4175
4176 static void
4177 free_writer_entry (gpointer p)
4178 {
4179         mono_lock_free_free (p, WRITER_ENTRY_BLOCK_SIZE);
4180 }
4181
4182 static gboolean
4183 handle_writer_queue_entry (MonoProfiler *prof)
4184 {
4185         WriterQueueEntry *entry;
4186
4187         if ((entry = (WriterQueueEntry *) mono_lock_free_queue_dequeue (&prof->writer_queue))) {
4188                 if (!entry->methods)
4189                         goto no_methods;
4190
4191                 gboolean wrote_methods = FALSE;
4192
4193                 /*
4194                  * Encode the method events in a temporary log buffer that we
4195                  * flush to disk before the main buffer, ensuring that all
4196                  * methods have metadata emitted before they're referenced.
4197                  *
4198                  * We use a 'proper' thread-local buffer for this as opposed
4199                  * to allocating and freeing a buffer by hand because the call
4200                  * to mono_method_full_name () below may trigger class load
4201                  * events when it retrieves the signature of the method. So a
4202                  * thread-local buffer needs to exist when such events occur.
4203                  */
4204                 for (guint i = 0; i < entry->methods->len; i++) {
4205                         MethodInfo *info = (MethodInfo *) g_ptr_array_index (entry->methods, i);
4206
4207                         if (mono_conc_hashtable_lookup (prof->method_table, info->method))
4208                                 goto free_info; // This method already has metadata emitted.
4209
4210                         /*
4211                          * Other threads use this hash table to get a general
4212                          * idea of whether a method has already been emitted to
4213                          * the stream. Due to the way we add to this table, it
4214                          * can easily happen that multiple threads queue up the
4215                          * same methods, but that's OK since eventually all
4216                          * methods will be in this table and the thread-local
4217                          * method lists will just be empty for the rest of the
4218                          * app's lifetime.
4219                          */
4220                         mono_os_mutex_lock (&prof->method_table_mutex);
4221                         mono_conc_hashtable_insert (prof->method_table, info->method, info->method);
4222                         mono_os_mutex_unlock (&prof->method_table_mutex);
4223
4224                         char *name = mono_method_full_name (info->method, 1);
4225                         int nlen = strlen (name) + 1;
4226                         void *cstart = info->ji ? mono_jit_info_get_code_start (info->ji) : NULL;
4227                         int csize = info->ji ? mono_jit_info_get_code_size (info->ji) : 0;
4228
4229                         ENTER_LOG (&method_jits_ctr, logbuffer,
4230                                 EVENT_SIZE /* event */ +
4231                                 LEB128_SIZE /* method */ +
4232                                 LEB128_SIZE /* start */ +
4233                                 LEB128_SIZE /* size */ +
4234                                 nlen /* name */
4235                         );
4236
4237                         emit_event_time (logbuffer, TYPE_JIT | TYPE_METHOD, info->time);
4238                         emit_method_inner (logbuffer, info->method);
4239                         emit_ptr (logbuffer, cstart);
4240                         emit_value (logbuffer, csize);
4241
4242                         memcpy (logbuffer->cursor, name, nlen);
4243                         logbuffer->cursor += nlen;
4244
4245                         EXIT_LOG_EXPLICIT (NO_SEND);
4246
4247                         mono_free (name);
4248
4249                         wrote_methods = TRUE;
4250
4251                 free_info:
4252                         g_free (info);
4253                 }
4254
4255                 g_ptr_array_free (entry->methods, TRUE);
4256
4257                 if (wrote_methods) {
4258                         dump_buffer_threadless (prof, PROF_TLS_GET ()->buffer);
4259                         init_buffer_state (PROF_TLS_GET ());
4260                 }
4261
4262         no_methods:
4263                 dump_buffer (prof, entry->buffer);
4264
4265                 mono_thread_hazardous_try_free (entry, free_writer_entry);
4266
4267                 return TRUE;
4268         }
4269
4270         return FALSE;
4271 }
4272
4273 static void *
4274 writer_thread (void *arg)
4275 {
4276         MonoProfiler *prof = (MonoProfiler *)arg;
4277
4278         mono_threads_attach_tools_thread ();
4279         mono_native_thread_set_name (mono_native_thread_id_get (), "Profiler writer");
4280
4281         dump_header (prof);
4282
4283         MonoProfilerThread *thread = init_thread (prof, FALSE);
4284
4285         while (InterlockedRead (&prof->run_writer_thread)) {
4286                 mono_os_sem_wait (&prof->writer_queue_sem, MONO_SEM_FLAGS_NONE);
4287                 handle_writer_queue_entry (prof);
4288         }
4289
4290         /* Drain any remaining entries on shutdown. */
4291         while (handle_writer_queue_entry (prof));
4292
4293         free_buffer (thread->buffer, thread->buffer->size);
4294         deinit_thread (thread);
4295
4296         mono_thread_info_detach ();
4297
4298         return NULL;
4299 }
4300
4301 static void
4302 start_writer_thread (MonoProfiler* prof)
4303 {
4304         InterlockedWrite (&prof->run_writer_thread, 1);
4305
4306         if (!mono_native_thread_create (&prof->writer_thread, writer_thread, prof)) {
4307                 fprintf (stderr, "Could not start writer thread\n");
4308                 exit (1);
4309         }
4310 }
4311
4312 static void
4313 reuse_sample_hit (gpointer p)
4314 {
4315         SampleHit *sample = p;
4316
4317         mono_lock_free_queue_node_unpoison (&sample->node);
4318         mono_lock_free_queue_enqueue (&sample->prof->sample_reuse_queue, &sample->node);
4319 }
4320
4321 static gboolean
4322 handle_dumper_queue_entry (MonoProfiler *prof)
4323 {
4324         SampleHit *sample;
4325
4326         if ((sample = (SampleHit *) mono_lock_free_queue_dequeue (&prof->dumper_queue))) {
4327                 for (int i = 0; i < sample->count; ++i) {
4328                         MonoMethod *method = sample->frames [i].method;
4329                         MonoDomain *domain = sample->frames [i].domain;
4330                         void *address = sample->frames [i].base_address;
4331
4332                         if (!method) {
4333                                 g_assert (domain && "What happened to the domain pointer?");
4334                                 g_assert (address && "What happened to the instruction pointer?");
4335
4336                                 MonoJitInfo *ji = mono_jit_info_table_find (domain, (char *) address);
4337
4338                                 if (ji)
4339                                         sample->frames [i].method = mono_jit_info_get_method (ji);
4340                         }
4341                 }
4342
4343                 ENTER_LOG (&sample_hits_ctr, logbuffer,
4344                         EVENT_SIZE /* event */ +
4345                         BYTE_SIZE /* type */ +
4346                         LEB128_SIZE /* tid */ +
4347                         LEB128_SIZE /* count */ +
4348                         1 * (
4349                                 LEB128_SIZE /* ip */
4350                         ) +
4351                         LEB128_SIZE /* managed count */ +
4352                         sample->count * (
4353                                 LEB128_SIZE /* method */
4354                         )
4355                 );
4356
4357                 emit_event_time (logbuffer, TYPE_SAMPLE | TYPE_SAMPLE_HIT, sample->time);
4358                 emit_byte (logbuffer, SAMPLE_CYCLES);
4359                 emit_ptr (logbuffer, (void *) sample->tid);
4360                 emit_value (logbuffer, 1);
4361
4362                 // TODO: Actual native unwinding.
4363                 for (int i = 0; i < 1; ++i) {
4364                         emit_ptr (logbuffer, sample->ip);
4365                         add_code_pointer ((uintptr_t) sample->ip);
4366                 }
4367
4368                 /* new in data version 6 */
4369                 emit_uvalue (logbuffer, sample->count);
4370
4371                 for (int i = 0; i < sample->count; ++i)
4372                         emit_method (logbuffer, sample->frames [i].method);
4373
4374                 EXIT_LOG_EXPLICIT (DO_SEND);
4375
4376                 mono_thread_hazardous_try_free (sample, reuse_sample_hit);
4377
4378                 dump_unmanaged_coderefs (prof);
4379         }
4380
4381         return FALSE;
4382 }
4383
4384 static void *
4385 dumper_thread (void *arg)
4386 {
4387         MonoProfiler *prof = (MonoProfiler *)arg;
4388
4389         mono_threads_attach_tools_thread ();
4390         mono_native_thread_set_name (mono_native_thread_id_get (), "Profiler dumper");
4391
4392         MonoProfilerThread *thread = init_thread (prof, FALSE);
4393
4394         while (InterlockedRead (&prof->run_dumper_thread)) {
4395                 mono_os_sem_wait (&prof->dumper_queue_sem, MONO_SEM_FLAGS_NONE);
4396                 handle_dumper_queue_entry (prof);
4397         }
4398
4399         /* Drain any remaining entries on shutdown. */
4400         while (handle_dumper_queue_entry (prof));
4401
4402         send_log_unsafe (FALSE);
4403         deinit_thread (thread);
4404
4405         mono_thread_info_detach ();
4406
4407         return NULL;
4408 }
4409
4410 static void
4411 start_dumper_thread (MonoProfiler* prof)
4412 {
4413         InterlockedWrite (&prof->run_dumper_thread, 1);
4414
4415         if (!mono_native_thread_create (&prof->dumper_thread, dumper_thread, prof)) {
4416                 fprintf (stderr, "Could not start dumper thread\n");
4417                 exit (1);
4418         }
4419 }
4420
4421 static void
4422 register_counter (const char *name, gint32 *counter)
4423 {
4424         mono_counters_register (name, MONO_COUNTER_UINT | MONO_COUNTER_PROFILER | MONO_COUNTER_MONOTONIC, counter);
4425 }
4426
4427 static void
4428 runtime_initialized (MonoProfiler *profiler)
4429 {
4430         InterlockedWrite (&runtime_inited, 1);
4431
4432         register_counter ("Sample events allocated", &sample_allocations_ctr);
4433         register_counter ("Log buffers allocated", &buffer_allocations_ctr);
4434
4435         register_counter ("Event: Sync points", &sync_points_ctr);
4436         register_counter ("Event: Heap objects", &heap_objects_ctr);
4437         register_counter ("Event: Heap starts", &heap_starts_ctr);
4438         register_counter ("Event: Heap ends", &heap_ends_ctr);
4439         register_counter ("Event: Heap roots", &heap_roots_ctr);
4440         register_counter ("Event: GC events", &gc_events_ctr);
4441         register_counter ("Event: GC resizes", &gc_resizes_ctr);
4442         register_counter ("Event: GC allocations", &gc_allocs_ctr);
4443         register_counter ("Event: GC moves", &gc_moves_ctr);
4444         register_counter ("Event: GC handle creations", &gc_handle_creations_ctr);
4445         register_counter ("Event: GC handle deletions", &gc_handle_deletions_ctr);
4446         register_counter ("Event: GC finalize starts", &finalize_begins_ctr);
4447         register_counter ("Event: GC finalize ends", &finalize_ends_ctr);
4448         register_counter ("Event: GC finalize object starts", &finalize_object_begins_ctr);
4449         register_counter ("Event: GC finalize object ends", &finalize_object_ends_ctr);
4450         register_counter ("Event: Image loads", &image_loads_ctr);
4451         register_counter ("Event: Image unloads", &image_unloads_ctr);
4452         register_counter ("Event: Assembly loads", &assembly_loads_ctr);
4453         register_counter ("Event: Assembly unloads", &assembly_unloads_ctr);
4454         register_counter ("Event: Class loads", &class_loads_ctr);
4455         register_counter ("Event: Class unloads", &class_unloads_ctr);
4456         register_counter ("Event: Method entries", &method_entries_ctr);
4457         register_counter ("Event: Method exits", &method_exits_ctr);
4458         register_counter ("Event: Method exception leaves", &method_exception_exits_ctr);
4459         register_counter ("Event: Method JITs", &method_jits_ctr);
4460         register_counter ("Event: Code buffers", &code_buffers_ctr);
4461         register_counter ("Event: Exception throws", &exception_throws_ctr);
4462         register_counter ("Event: Exception clauses", &exception_clauses_ctr);
4463         register_counter ("Event: Monitor events", &monitor_events_ctr);
4464         register_counter ("Event: Thread starts", &thread_starts_ctr);
4465         register_counter ("Event: Thread ends", &thread_ends_ctr);
4466         register_counter ("Event: Thread names", &thread_names_ctr);
4467         register_counter ("Event: Domain loads", &domain_loads_ctr);
4468         register_counter ("Event: Domain unloads", &domain_unloads_ctr);
4469         register_counter ("Event: Domain names", &domain_names_ctr);
4470         register_counter ("Event: Context loads", &context_loads_ctr);
4471         register_counter ("Event: Context unloads", &context_unloads_ctr);
4472         register_counter ("Event: Sample binaries", &sample_ubins_ctr);
4473         register_counter ("Event: Sample symbols", &sample_usyms_ctr);
4474         register_counter ("Event: Sample hits", &sample_hits_ctr);
4475         register_counter ("Event: Counter descriptors", &counter_descriptors_ctr);
4476         register_counter ("Event: Counter samples", &counter_samples_ctr);
4477         register_counter ("Event: Performance counter descriptors", &perfcounter_descriptors_ctr);
4478         register_counter ("Event: Performance counter samples", &perfcounter_samples_ctr);
4479         register_counter ("Event: Coverage methods", &coverage_methods_ctr);
4480         register_counter ("Event: Coverage statements", &coverage_statements_ctr);
4481         register_counter ("Event: Coverage classes", &coverage_classes_ctr);
4482         register_counter ("Event: Coverage assemblies", &coverage_assemblies_ctr);
4483
4484         counters_init (profiler);
4485
4486         /*
4487          * We must start the helper thread before the writer thread. This is
4488          * because the helper thread sets up the command port which is written to
4489          * the log header by the writer thread.
4490          */
4491         start_helper_thread (profiler);
4492         start_writer_thread (profiler);
4493         start_dumper_thread (profiler);
4494 }
4495
4496 static MonoProfiler*
4497 create_profiler (const char *args, const char *filename, GPtrArray *filters)
4498 {
4499         MonoProfiler *prof;
4500         char *nf;
4501         int force_delete = 0;
4502         prof = (MonoProfiler *) g_calloc (1, sizeof (MonoProfiler));
4503
4504         prof->args = pstrdup (args);
4505         prof->command_port = command_port;
4506         if (filename && *filename == '-') {
4507                 force_delete = 1;
4508                 filename++;
4509                 g_warning ("WARNING: the output:-FILENAME option is deprecated, the profiler now always overrides the output file\n");
4510         }
4511
4512         //If filename begin with +, append the pid at the end
4513         if (filename && *filename == '+')
4514                 filename = g_strdup_printf ("%s.%d", filename + 1, getpid ());
4515
4516
4517         if (!filename) {
4518                 if (do_report)
4519                         filename = "|mprof-report -";
4520                 else
4521                         filename = "output.mlpd";
4522                 nf = (char*)filename;
4523         } else {
4524                 nf = new_filename (filename);
4525                 if (do_report) {
4526                         int s = strlen (nf) + 32;
4527                         char *p = (char *) g_malloc (s);
4528                         snprintf (p, s, "|mprof-report '--out=%s' -", nf);
4529                         g_free (nf);
4530                         nf = p;
4531                 }
4532         }
4533         if (*nf == '|') {
4534                 prof->file = popen (nf + 1, "w");
4535                 prof->pipe_output = 1;
4536         } else if (*nf == '#') {
4537                 int fd = strtol (nf + 1, NULL, 10);
4538                 prof->file = fdopen (fd, "a");
4539         } else {
4540                 if (force_delete)
4541                         unlink (nf);
4542                 prof->file = fopen (nf, "wb");
4543         }
4544         if (!prof->file) {
4545                 fprintf (stderr, "Cannot create profiler output: %s\n", nf);
4546                 exit (1);
4547         }
4548
4549 #if defined (HAVE_SYS_ZLIB)
4550         if (use_zip)
4551                 prof->gzfile = gzdopen (fileno (prof->file), "wb");
4552 #endif
4553
4554         /*
4555          * If you hit this assert while increasing MAX_FRAMES, you need to increase
4556          * SAMPLE_BLOCK_SIZE as well.
4557          */
4558         g_assert (SAMPLE_SLOT_SIZE (MAX_FRAMES) * 2 < LOCK_FREE_ALLOC_SB_USABLE_SIZE (SAMPLE_BLOCK_SIZE));
4559
4560         // FIXME: We should free this stuff too.
4561         mono_lock_free_allocator_init_size_class (&prof->sample_size_class, SAMPLE_SLOT_SIZE (num_frames), SAMPLE_BLOCK_SIZE);
4562         mono_lock_free_allocator_init_allocator (&prof->sample_allocator, &prof->sample_size_class, MONO_MEM_ACCOUNT_PROFILER);
4563
4564         mono_lock_free_queue_init (&prof->sample_reuse_queue);
4565
4566         g_assert (sizeof (WriterQueueEntry) * 2 < LOCK_FREE_ALLOC_SB_USABLE_SIZE (WRITER_ENTRY_BLOCK_SIZE));
4567
4568         // FIXME: We should free this stuff too.
4569         mono_lock_free_allocator_init_size_class (&prof->writer_entry_size_class, sizeof (WriterQueueEntry), WRITER_ENTRY_BLOCK_SIZE);
4570         mono_lock_free_allocator_init_allocator (&prof->writer_entry_allocator, &prof->writer_entry_size_class, MONO_MEM_ACCOUNT_PROFILER);
4571
4572         mono_lock_free_queue_init (&prof->writer_queue);
4573         mono_os_sem_init (&prof->writer_queue_sem, 0);
4574
4575         mono_lock_free_queue_init (&prof->dumper_queue);
4576         mono_os_sem_init (&prof->dumper_queue_sem, 0);
4577
4578         mono_os_mutex_init (&prof->method_table_mutex);
4579         prof->method_table = mono_conc_hashtable_new (NULL, NULL);
4580
4581         if (do_coverage)
4582                 coverage_init (prof);
4583         prof->coverage_filters = filters;
4584
4585         prof->startup_time = current_time ();
4586         return prof;
4587 }
4588
4589 /*
4590  * declaration to silence the compiler: this is the entry point that
4591  * mono will load from the shared library and call.
4592  */
4593 extern void
4594 mono_profiler_startup (const char *desc);
4595
4596 extern void
4597 mono_profiler_startup_log (const char *desc);
4598
4599 /*
4600  * this is the entry point that will be used when the profiler
4601  * is embedded inside the main executable.
4602  */
4603 void
4604 mono_profiler_startup_log (const char *desc)
4605 {
4606         mono_profiler_startup (desc);
4607 }
4608
4609 void
4610 mono_profiler_startup (const char *desc)
4611 {
4612         GPtrArray *filters = NULL;
4613         MonoProfiler *prof;
4614
4615         proflog_parse_args (&config, desc [3] == ':' ? desc + 4 : "");
4616
4617         //XXX maybe later cleanup to use config directly
4618         nocalls = !(config.effective_mask & PROFLOG_CALL_EVENTS);
4619         no_counters = !(config.effective_mask & PROFLOG_COUNTER_EVENTS);
4620         do_report = config.do_report;
4621         do_debug = config.do_debug;
4622         do_heap_shot = (config.effective_mask & PROFLOG_HEAPSHOT_FEATURE);
4623         hs_mode_ondemand = config.hs_mode_ondemand;
4624         hs_mode_ms = config.hs_mode_ms;
4625         hs_mode_gc = config.hs_mode_gc;
4626         do_mono_sample = (config.effective_mask & PROFLOG_SAMPLING_FEATURE);
4627         use_zip = config.use_zip;
4628         command_port = config.command_port;
4629         num_frames = config.num_frames;
4630         notraces = config.notraces;
4631         max_allocated_sample_hits = config.max_allocated_sample_hits;
4632         max_call_depth = config.max_call_depth;
4633         do_coverage = (config.effective_mask & PROFLOG_CODE_COV_FEATURE);
4634         debug_coverage = config.debug_coverage;
4635         only_coverage = config.only_coverage;
4636
4637         if (config.cov_filter_files) {
4638                 filters = g_ptr_array_new ();
4639                 int i;
4640                 for (i = 0; i < config.cov_filter_files->len; ++i) {
4641                         const char *name = config.cov_filter_files->pdata [i];
4642                         parse_cov_filter_file (filters, name);
4643                 }
4644         }
4645
4646         init_time ();
4647
4648         PROF_TLS_INIT ();
4649
4650         prof = create_profiler (desc, config.output_filename, filters);
4651         if (!prof) {
4652                 PROF_TLS_FREE ();
4653                 return;
4654         }
4655
4656         mono_lls_init (&profiler_thread_list, NULL);
4657
4658         init_thread (prof, TRUE);
4659
4660         //This two events are required for the profiler to work
4661         int events = MONO_PROFILE_THREADS | MONO_PROFILE_GC;
4662
4663         //Required callbacks
4664         mono_profiler_install (prof, log_shutdown);
4665         mono_profiler_install_runtime_initialized (runtime_initialized);
4666
4667         mono_profiler_install_gc (gc_event, gc_resize);
4668         mono_profiler_install_thread (thread_start, thread_end);
4669
4670         //It's questionable whether we actually want this to be mandatory, maybe put it behind the actual event?
4671         mono_profiler_install_thread_name (thread_name);
4672
4673
4674         if (config.effective_mask & PROFLOG_DOMAIN_EVENTS) {
4675                 events |= MONO_PROFILE_APPDOMAIN_EVENTS;
4676                 mono_profiler_install_appdomain (NULL, domain_loaded, domain_unloaded, NULL);
4677                 mono_profiler_install_appdomain_name (domain_name);
4678         }
4679
4680         if (config.effective_mask & PROFLOG_ASSEMBLY_EVENTS) {
4681                 events |= MONO_PROFILE_ASSEMBLY_EVENTS;
4682                 mono_profiler_install_assembly (NULL, assembly_loaded, assembly_unloaded, NULL);
4683         }
4684
4685         if (config.effective_mask & PROFLOG_MODULE_EVENTS) {
4686                 events |= MONO_PROFILE_MODULE_EVENTS;
4687                 mono_profiler_install_module (NULL, image_loaded, image_unloaded, NULL);
4688         }
4689
4690         if (config.effective_mask & PROFLOG_CLASS_EVENTS) {
4691                 events |= MONO_PROFILE_CLASS_EVENTS;
4692                 mono_profiler_install_class (NULL, class_loaded, class_unloaded, NULL);
4693         }
4694
4695         if (config.effective_mask & PROFLOG_JIT_COMPILATION_EVENTS) {
4696                 events |= MONO_PROFILE_JIT_COMPILATION;
4697                 mono_profiler_install_jit_end (method_jitted);
4698                 mono_profiler_install_code_buffer_new (code_buffer_new);
4699         }
4700
4701         if (config.effective_mask & PROFLOG_EXCEPTION_EVENTS) {
4702                 events |= MONO_PROFILE_EXCEPTIONS;
4703                 mono_profiler_install_exception (throw_exc, method_exc_leave, NULL);
4704                 mono_profiler_install_exception_clause (clause_exc);
4705         }
4706
4707         if (config.effective_mask & PROFLOG_ALLOCATION_EVENTS) {
4708                 events |= MONO_PROFILE_ALLOCATIONS;
4709                 mono_profiler_install_allocation (gc_alloc);
4710         }
4711
4712         //PROFLOG_GC_EVENTS is mandatory
4713         //PROFLOG_THREAD_EVENTS is mandatory
4714
4715         if (config.effective_mask & PROFLOG_CALL_EVENTS) {
4716                 events |= MONO_PROFILE_ENTER_LEAVE;
4717                 mono_profiler_install_enter_leave (method_enter, method_leave);
4718         }
4719
4720         if (config.effective_mask & PROFLOG_INS_COVERAGE_EVENTS) {
4721                 events |= MONO_PROFILE_INS_COVERAGE;
4722                 mono_profiler_install_coverage_filter (coverage_filter);
4723         }
4724
4725         //XXX should we check for PROFLOG_SAMPLING_FEATURE instead??
4726         if (config.effective_mask & PROFLOG_SAMPLING_EVENTS) {
4727                 events |= MONO_PROFILE_STATISTICAL;
4728                 mono_profiler_set_statistical_mode (config.sampling_mode, config.sample_freq);
4729                 mono_profiler_install_statistical (mono_sample_hit);
4730         }
4731
4732         if (config.effective_mask & PROFLOG_MONITOR_EVENTS) {
4733                 events |= MONO_PROFILE_MONITOR_EVENTS;
4734                 mono_profiler_install_monitor (monitor_event);
4735         }
4736
4737         if (config.effective_mask & PROFLOG_GC_MOVES_EVENTS) {
4738                 events |= MONO_PROFILE_GC_MOVES;
4739                 mono_profiler_install_gc_moves (gc_moves);
4740         }
4741
4742         // TODO split those in two profiler events
4743         if (config.effective_mask & (PROFLOG_GC_ROOT_EVENTS | PROFLOG_GC_HANDLE_EVENTS)) {
4744                 events |= MONO_PROFILE_GC_ROOTS;
4745                 mono_profiler_install_gc_roots (
4746                         config.effective_mask & (PROFLOG_GC_HANDLE_EVENTS) ? gc_handle : NULL,
4747                         (config.effective_mask & PROFLOG_GC_ROOT_EVENTS) ? gc_roots : NULL);
4748         }
4749
4750         if (config.effective_mask & PROFLOG_CONTEXT_EVENTS) {
4751                 events |= MONO_PROFILE_CONTEXT_EVENTS;
4752                 mono_profiler_install_context (context_loaded, context_unloaded);
4753         }
4754
4755         if (config.effective_mask & PROFLOG_FINALIZATION_EVENTS) {
4756                 events |= MONO_PROFILE_GC_FINALIZATION;
4757                 mono_profiler_install_gc_finalize (finalize_begin, finalize_object_begin, finalize_object_end, finalize_end);
4758         } else if (ENABLED (PROFLOG_HEAPSHOT_FEATURE) && config.hs_mode_ondemand) {
4759                 //On Demand heapshot uses the finalizer thread to force a collection and thus a heapshot
4760                 events |= MONO_PROFILE_GC_FINALIZATION;
4761                 mono_profiler_install_gc_finalize (NULL, NULL, NULL, finalize_end);
4762         }
4763
4764         //PROFLOG_COUNTER_EVENTS is a pseudo event controled by the no_counters global var
4765         //PROFLOG_GC_HANDLE_EVENTS is handled together with PROFLOG_GC_ROOT_EVENTS
4766
4767         mono_profiler_set_events ((MonoProfileFlags)events);
4768 }