2 * sgen-protocol.c: Binary protocol of internal activity, to aid
5 * Copyright 2001-2003 Ximian, Inc
6 * Copyright 2003-2010 Novell, Inc.
7 * Copyright (C) 2012 Xamarin Inc
9 * Licensed under the MIT license. See LICENSE file in the project root for full license information.
15 #include "sgen-conf.h"
17 #include "sgen-protocol.h"
18 #include "sgen-memory-governor.h"
19 #include "sgen-thread-pool.h"
20 #include "sgen-client.h"
21 #include "mono/utils/mono-membar.h"
22 #include "mono/utils/mono-proclib.h"
26 #if defined(HAVE_UNISTD_H)
29 #elif defined(HOST_WIN32)
33 #if defined(HOST_WIN32)
34 static const HANDLE invalid_file_value = INVALID_HANDLE_VALUE;
35 /* If valid, dump binary protocol to this file */
36 static HANDLE binary_protocol_file = INVALID_HANDLE_VALUE;
38 static const int invalid_file_value = -1;
39 static int binary_protocol_file = -1;
42 /* We set this to -1 to indicate an exclusive lock */
43 static volatile int binary_protocol_use_count = 0;
45 #define BINARY_PROTOCOL_BUFFER_SIZE (65536 - 2 * 8)
47 typedef struct _BinaryProtocolBuffer BinaryProtocolBuffer;
48 struct _BinaryProtocolBuffer {
49 BinaryProtocolBuffer * volatile next;
51 unsigned char buffer [BINARY_PROTOCOL_BUFFER_SIZE];
54 static BinaryProtocolBuffer * volatile binary_protocol_buffers = NULL;
56 static char* filename_or_prefix = NULL;
57 static int current_file_index = 0;
58 static long long current_file_size = 0;
59 static long long file_size_limit;
62 filename_for_index (int index)
66 SGEN_ASSERT (0, file_size_limit > 0, "Indexed binary protocol filename must only be used with file size limit");
68 filename = (char *)sgen_alloc_internal_dynamic (strlen (filename_or_prefix) + 32, INTERNAL_MEM_BINARY_PROTOCOL, TRUE);
69 sprintf (filename, "%s.%d", filename_or_prefix, index);
75 free_filename (char *filename)
77 SGEN_ASSERT (0, file_size_limit > 0, "Indexed binary protocol filename must only be used with file size limit");
79 sgen_free_internal_dynamic (filename, strlen (filename_or_prefix) + 32, INTERNAL_MEM_BINARY_PROTOCOL);
83 binary_protocol_open_file (gboolean assert_on_failure)
88 lock.l_type = F_WRLCK;
89 lock.l_whence = SEEK_SET;
94 if (file_size_limit > 0)
95 filename = filename_for_index (current_file_index);
97 filename = filename_or_prefix;
99 #if defined(HAVE_UNISTD_H)
101 binary_protocol_file = open (filename, O_CREAT | O_WRONLY, 0644);
102 if (binary_protocol_file == -1) {
106 } else if (fcntl (binary_protocol_file, F_SETLK, &lock) == -1) {
107 /* The lock for the file is already taken. Fail */
108 close (binary_protocol_file);
109 binary_protocol_file = -1;
113 /* We have acquired the lock. Truncate the file */
114 ftruncate (binary_protocol_file, 0);
116 } while (binary_protocol_file == -1);
117 #elif defined(HOST_WIN32)
118 binary_protocol_file = CreateFileA (filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
120 g_error ("sgen binary protocol: not supported");
123 if (binary_protocol_file == invalid_file_value && assert_on_failure)
124 g_error ("sgen binary protocol: failed to open file");
126 if (file_size_limit > 0)
127 free_filename (filename);
131 binary_protocol_init (const char *filename, long long limit)
133 file_size_limit = limit;
135 /* Original name length + . + pid length in hex + null terminator */
136 filename_or_prefix = g_strdup_printf ("%s", filename);
137 binary_protocol_open_file (FALSE);
139 if (binary_protocol_file == invalid_file_value) {
140 /* Another process owns the file, try adding the pid suffix to the filename */
141 gint32 pid = mono_process_current_pid ();
142 g_free (filename_or_prefix);
143 filename_or_prefix = g_strdup_printf ("%s.%x", filename, pid);
144 binary_protocol_open_file (TRUE);
147 /* If we have a file size limit, we might need to open additional files */
148 if (file_size_limit == 0)
149 g_free (filename_or_prefix);
151 binary_protocol_header (PROTOCOL_HEADER_CHECK, PROTOCOL_HEADER_VERSION, SIZEOF_VOID_P, G_BYTE_ORDER == G_LITTLE_ENDIAN);
155 binary_protocol_is_enabled (void)
157 return binary_protocol_file != invalid_file_value;
161 close_binary_protocol_file (void)
163 #if defined(HAVE_UNISTD_H)
164 while (close (binary_protocol_file) == -1 && errno == EINTR)
166 #elif defined(HOST_WIN32)
167 CloseHandle (binary_protocol_file);
169 binary_protocol_file = invalid_file_value;
173 try_lock_exclusive (void)
176 if (binary_protocol_use_count)
178 } while (InterlockedCompareExchange (&binary_protocol_use_count, -1, 0) != 0);
179 mono_memory_barrier ();
184 unlock_exclusive (void)
186 mono_memory_barrier ();
187 SGEN_ASSERT (0, binary_protocol_use_count == -1, "Exclusively locked count must be -1");
188 if (InterlockedCompareExchange (&binary_protocol_use_count, 0, -1) != -1)
189 SGEN_ASSERT (0, FALSE, "Somebody messed with the exclusive lock");
193 lock_recursive (void)
198 old_count = binary_protocol_use_count;
200 /* Exclusively locked - retry */
201 /* FIXME: short back-off */
204 } while (InterlockedCompareExchange (&binary_protocol_use_count, old_count + 1, old_count) != old_count);
205 mono_memory_barrier ();
209 unlock_recursive (void)
212 mono_memory_barrier ();
214 old_count = binary_protocol_use_count;
215 SGEN_ASSERT (0, old_count > 0, "Locked use count must be at least 1");
216 } while (InterlockedCompareExchange (&binary_protocol_use_count, old_count - 1, old_count) != old_count);
220 binary_protocol_flush_buffer (BinaryProtocolBuffer *buffer)
223 size_t to_write = buffer->index;
225 g_assert (buffer->index > 0);
227 while (written < to_write) {
228 #if defined(HAVE_UNISTD_H)
229 ret = write (binary_protocol_file, buffer->buffer + written, to_write - written);
232 else if (errno == EINTR)
235 close_binary_protocol_file ();
236 #elif defined(HOST_WIN32)
238 if (WriteFile (binary_protocol_file, buffer->buffer + written, to_write - written, &tmp_written, NULL))
239 written += tmp_written;
241 close_binary_protocol_file ();
245 current_file_size += buffer->index;
247 sgen_free_os_memory (buffer, sizeof (BinaryProtocolBuffer), SGEN_ALLOC_INTERNAL, MONO_MEM_ACCOUNT_SGEN_BINARY_PROTOCOL);
251 binary_protocol_check_file_overflow (void)
253 if (file_size_limit <= 0 || current_file_size < file_size_limit)
256 close_binary_protocol_file ();
258 if (current_file_index > 0) {
259 char *filename = filename_for_index (current_file_index - 1);
261 free_filename (filename);
264 ++current_file_index;
265 current_file_size = 0;
267 binary_protocol_open_file (TRUE);
271 * Flushing buffers takes an exclusive lock, so it must only be done when the world is
272 * stopped, otherwise we might end up with a deadlock because a stopped thread owns the
275 * The protocol entries that do flush have `FLUSH()` in their definition.
278 binary_protocol_flush_buffers (gboolean force)
280 int num_buffers = 0, i;
281 BinaryProtocolBuffer *header;
282 BinaryProtocolBuffer *buf;
283 BinaryProtocolBuffer **bufs;
285 if (binary_protocol_file == invalid_file_value)
288 if (!force && !try_lock_exclusive ())
291 header = binary_protocol_buffers;
292 for (buf = header; buf != NULL; buf = buf->next)
294 bufs = (BinaryProtocolBuffer **)sgen_alloc_internal_dynamic (num_buffers * sizeof (BinaryProtocolBuffer*), INTERNAL_MEM_BINARY_PROTOCOL, TRUE);
295 for (buf = header, i = 0; buf != NULL; buf = buf->next, i++)
297 SGEN_ASSERT (0, i == num_buffers, "Binary protocol buffer count error");
300 * This might be incorrect when forcing, but all bets are off in that case, anyway,
301 * because we're trying to figure out a bug in the debugger.
303 binary_protocol_buffers = NULL;
305 for (i = num_buffers - 1; i >= 0; --i) {
306 binary_protocol_flush_buffer (bufs [i]);
307 binary_protocol_check_file_overflow ();
310 sgen_free_internal_dynamic (buf, num_buffers * sizeof (BinaryProtocolBuffer*), INTERNAL_MEM_BINARY_PROTOCOL);
318 static BinaryProtocolBuffer*
319 binary_protocol_get_buffer (int length)
321 BinaryProtocolBuffer *buffer, *new_buffer;
323 buffer = binary_protocol_buffers;
324 if (buffer && buffer->index + length <= BINARY_PROTOCOL_BUFFER_SIZE)
327 new_buffer = (BinaryProtocolBuffer *)sgen_alloc_os_memory (sizeof (BinaryProtocolBuffer), (SgenAllocFlags)(SGEN_ALLOC_INTERNAL | SGEN_ALLOC_ACTIVATE), "debugging memory", MONO_MEM_ACCOUNT_SGEN_BINARY_PROTOCOL);
328 new_buffer->next = buffer;
329 new_buffer->index = 0;
331 if (InterlockedCompareExchangePointer ((void**)&binary_protocol_buffers, new_buffer, buffer) != buffer) {
332 sgen_free_os_memory (new_buffer, sizeof (BinaryProtocolBuffer), SGEN_ALLOC_INTERNAL, MONO_MEM_ACCOUNT_SGEN_BINARY_PROTOCOL);
340 protocol_entry (unsigned char type, gpointer data, int size)
343 BinaryProtocolBuffer *buffer;
345 if (binary_protocol_file == invalid_file_value)
348 if (sgen_thread_pool_is_thread_pool_thread (mono_native_thread_id_get ()))
354 buffer = binary_protocol_get_buffer (size + 1);
356 index = buffer->index;
357 if (index + 1 + size > BINARY_PROTOCOL_BUFFER_SIZE)
360 if (InterlockedCompareExchange (&buffer->index, index + 1 + size, index) != index)
361 goto retry_same_buffer;
363 /* FIXME: if we're interrupted at this point, we have a buffer
364 entry that contains random data. */
366 buffer->buffer [index++] = type;
367 memcpy (buffer->buffer + index, data, size);
370 g_assert (index <= BINARY_PROTOCOL_BUFFER_SIZE);
376 #define TYPE_LONGLONG long long
377 #define TYPE_SIZE size_t
378 #define TYPE_POINTER gpointer
379 #define TYPE_BOOL gboolean
381 #define BEGIN_PROTOCOL_ENTRY0(method) \
382 void method (void) { \
383 int __type = PROTOCOL_ID(method); \
384 gpointer __data = NULL; \
386 CLIENT_PROTOCOL_NAME (method) ();
387 #define BEGIN_PROTOCOL_ENTRY1(method,t1,f1) \
388 void method (t1 f1) { \
389 PROTOCOL_STRUCT(method) __entry = { f1 }; \
390 int __type = PROTOCOL_ID(method); \
391 gpointer __data = &__entry; \
392 int __size = sizeof (PROTOCOL_STRUCT(method)); \
393 CLIENT_PROTOCOL_NAME (method) (f1);
394 #define BEGIN_PROTOCOL_ENTRY2(method,t1,f1,t2,f2) \
395 void method (t1 f1, t2 f2) { \
396 PROTOCOL_STRUCT(method) __entry = { f1, f2 }; \
397 int __type = PROTOCOL_ID(method); \
398 gpointer __data = &__entry; \
399 int __size = sizeof (PROTOCOL_STRUCT(method)); \
400 CLIENT_PROTOCOL_NAME (method) (f1, f2);
401 #define BEGIN_PROTOCOL_ENTRY3(method,t1,f1,t2,f2,t3,f3) \
402 void method (t1 f1, t2 f2, t3 f3) { \
403 PROTOCOL_STRUCT(method) __entry = { f1, f2, f3 }; \
404 int __type = PROTOCOL_ID(method); \
405 gpointer __data = &__entry; \
406 int __size = sizeof (PROTOCOL_STRUCT(method)); \
407 CLIENT_PROTOCOL_NAME (method) (f1, f2, f3);
408 #define BEGIN_PROTOCOL_ENTRY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
409 void method (t1 f1, t2 f2, t3 f3, t4 f4) { \
410 PROTOCOL_STRUCT(method) __entry = { f1, f2, f3, f4 }; \
411 int __type = PROTOCOL_ID(method); \
412 gpointer __data = &__entry; \
413 int __size = sizeof (PROTOCOL_STRUCT(method)); \
414 CLIENT_PROTOCOL_NAME (method) (f1, f2, f3, f4);
415 #define BEGIN_PROTOCOL_ENTRY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
416 void method (t1 f1, t2 f2, t3 f3, t4 f4, t5 f5) { \
417 PROTOCOL_STRUCT(method) __entry = { f1, f2, f3, f4, f5 }; \
418 int __type = PROTOCOL_ID(method); \
419 gpointer __data = &__entry; \
420 int __size = sizeof (PROTOCOL_STRUCT(method)); \
421 CLIENT_PROTOCOL_NAME (method) (f1, f2, f3, f4, f5);
422 #define BEGIN_PROTOCOL_ENTRY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
423 void method (t1 f1, t2 f2, t3 f3, t4 f4, t5 f5, t6 f6) { \
424 PROTOCOL_STRUCT(method) __entry = { f1, f2, f3, f4, f5, f6 }; \
425 int __type = PROTOCOL_ID(method); \
426 gpointer __data = &__entry; \
427 int __size = sizeof (PROTOCOL_STRUCT(method)); \
428 CLIENT_PROTOCOL_NAME (method) (f1, f2, f3, f4, f5, f6);
430 #define DEFAULT_PRINT()
431 #define CUSTOM_PRINT(_)
433 #define IS_ALWAYS_MATCH(_)
434 #define MATCH_INDEX(_)
435 #define IS_VTABLE_MATCH(_)
437 #define END_PROTOCOL_ENTRY \
438 protocol_entry (__type, __data, __size); \
441 #define END_PROTOCOL_ENTRY_FLUSH \
442 protocol_entry (__type, __data, __size); \
443 binary_protocol_flush_buffers (FALSE); \
446 #ifdef SGEN_HEAVY_BINARY_PROTOCOL
447 #define BEGIN_PROTOCOL_ENTRY_HEAVY0(method) \
448 BEGIN_PROTOCOL_ENTRY0 (method)
449 #define BEGIN_PROTOCOL_ENTRY_HEAVY1(method,t1,f1) \
450 BEGIN_PROTOCOL_ENTRY1 (method,t1,f1)
451 #define BEGIN_PROTOCOL_ENTRY_HEAVY2(method,t1,f1,t2,f2) \
452 BEGIN_PROTOCOL_ENTRY2 (method,t1,f1,t2,f2)
453 #define BEGIN_PROTOCOL_ENTRY_HEAVY3(method,t1,f1,t2,f2,t3,f3) \
454 BEGIN_PROTOCOL_ENTRY3 (method,t1,f1,t2,f2,t3,f3)
455 #define BEGIN_PROTOCOL_ENTRY_HEAVY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
456 BEGIN_PROTOCOL_ENTRY4 (method,t1,f1,t2,f2,t3,f3,t4,f4)
457 #define BEGIN_PROTOCOL_ENTRY_HEAVY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
458 BEGIN_PROTOCOL_ENTRY5 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5)
459 #define BEGIN_PROTOCOL_ENTRY_HEAVY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
460 BEGIN_PROTOCOL_ENTRY6 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6)
462 #define END_PROTOCOL_ENTRY_HEAVY \
465 #define BEGIN_PROTOCOL_ENTRY_HEAVY0(method)
466 #define BEGIN_PROTOCOL_ENTRY_HEAVY1(method,t1,f1)
467 #define BEGIN_PROTOCOL_ENTRY_HEAVY2(method,t1,f1,t2,f2)
468 #define BEGIN_PROTOCOL_ENTRY_HEAVY3(method,t1,f1,t2,f2,t3,f3)
469 #define BEGIN_PROTOCOL_ENTRY_HEAVY4(method,t1,f1,t2,f2,t3,f3,t4,f4)
470 #define BEGIN_PROTOCOL_ENTRY_HEAVY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5)
471 #define BEGIN_PROTOCOL_ENTRY_HEAVY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6)
473 #define END_PROTOCOL_ENTRY_HEAVY
476 #include "sgen-protocol-def.h"
484 #endif /* HAVE_SGEN_GC */