linux-packaging-mono/tools/sgen/sgen-grep-binprot.c
Jo Shields 3c1f479b9d Imported Upstream version 4.0.0~alpha1
Former-commit-id: 806294f5ded97629b74c85c09952f2a74fe182d9
2015-04-07 09:35:12 +01:00

686 lines
21 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <glib.h>
#include <unistd.h>
#include <fcntl.h>
#define SGEN_BINARY_PROTOCOL
#define MONO_INTERNAL
#include <mono/metadata/sgen-protocol.h>
#define SGEN_PROTOCOL_EOF 255
#define TYPE(t) ((t) & 0x7f)
#define WORKER(t) ((t) & 0x80)
#define MAX_ENTRY_SIZE (1 << 10)
#define BUFFER_SIZE (1 << 20)
typedef struct {
int file;
char *buffer;
const char *end;
const char *pos;
} EntryStream;
static void
init_stream (EntryStream *stream, int file)
{
stream->file = file;
stream->buffer = g_malloc0 (BUFFER_SIZE);
stream->end = stream->buffer + BUFFER_SIZE;
stream->pos = stream->end;
}
static void
close_stream (EntryStream *stream)
{
g_free (stream->buffer);
}
static gboolean
refill_stream (EntryStream *in, size_t size)
{
size_t remainder = in->end - in->pos;
ssize_t refilled;
g_assert (size > 0);
g_assert (in->pos >= in->buffer);
if (in->pos + size <= in->end)
return TRUE;
memmove (in->buffer, in->pos, remainder);
in->pos = in->buffer;
refilled = read (in->file, in->buffer + remainder, BUFFER_SIZE - remainder);
if (refilled < 0)
return FALSE;
g_assert (refilled + remainder <= BUFFER_SIZE);
in->end = in->buffer + refilled + remainder;
return in->end - in->buffer >= size;
}
static ssize_t
read_stream (EntryStream *stream, void *out, size_t size)
{
if (refill_stream (stream, size)) {
memcpy (out, stream->pos, size);
stream->pos += size;
return size;
}
return 0;
}
static int
read_entry (EntryStream *stream, void *data)
{
unsigned char type;
ssize_t size;
if (read_stream (stream, &type, 1) <= 0)
return SGEN_PROTOCOL_EOF;
switch (TYPE (type)) {
#define BEGIN_PROTOCOL_ENTRY0(method) \
case PROTOCOL_ID(method): size = 0; break;
#define BEGIN_PROTOCOL_ENTRY1(method,t1,f1) \
case PROTOCOL_ID(method): size = sizeof (PROTOCOL_STRUCT(method)); break;
#define BEGIN_PROTOCOL_ENTRY2(method,t1,f1,t2,f2) \
case PROTOCOL_ID(method): size = sizeof (PROTOCOL_STRUCT(method)); break;
#define BEGIN_PROTOCOL_ENTRY3(method,t1,f1,t2,f2,t3,f3) \
case PROTOCOL_ID(method): size = sizeof (PROTOCOL_STRUCT(method)); break;
#define BEGIN_PROTOCOL_ENTRY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
case PROTOCOL_ID(method): size = sizeof (PROTOCOL_STRUCT(method)); break;
#define BEGIN_PROTOCOL_ENTRY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
case PROTOCOL_ID(method): size = sizeof (PROTOCOL_STRUCT(method)); break;
#define BEGIN_PROTOCOL_ENTRY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
case PROTOCOL_ID(method): size = sizeof (PROTOCOL_STRUCT(method)); break;
#define BEGIN_PROTOCOL_ENTRY_HEAVY0(method) \
BEGIN_PROTOCOL_ENTRY0 (method)
#define BEGIN_PROTOCOL_ENTRY_HEAVY1(method,t1,f1) \
BEGIN_PROTOCOL_ENTRY1 (method,t1,f1)
#define BEGIN_PROTOCOL_ENTRY_HEAVY2(method,t1,f1,t2,f2) \
BEGIN_PROTOCOL_ENTRY2 (method,t1,f1,t2,f2)
#define BEGIN_PROTOCOL_ENTRY_HEAVY3(method,t1,f1,t2,f2,t3,f3) \
BEGIN_PROTOCOL_ENTRY3 (method,t1,f1,t2,f2,t3,f3)
#define BEGIN_PROTOCOL_ENTRY_HEAVY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
BEGIN_PROTOCOL_ENTRY4 (method,t1,f1,t2,f2,t3,f3,t4,f4)
#define BEGIN_PROTOCOL_ENTRY_HEAVY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
BEGIN_PROTOCOL_ENTRY5 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5)
#define BEGIN_PROTOCOL_ENTRY_HEAVY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
BEGIN_PROTOCOL_ENTRY6 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6)
#define FLUSH()
#define DEFAULT_PRINT()
#define CUSTOM_PRINT(_)
#define IS_ALWAYS_MATCH(_)
#define MATCH_INDEX(_)
#define IS_VTABLE_MATCH(_)
#define END_PROTOCOL_ENTRY
#define END_PROTOCOL_ENTRY_HEAVY
#include <mono/metadata/sgen-protocol-def.h>
default: assert (0);
}
if (size) {
size_t size_read = read_stream (stream, data, size);
g_assert (size_read == size);
}
return (int)type;
}
static gboolean
is_always_match (int type)
{
switch (TYPE (type)) {
#define BEGIN_PROTOCOL_ENTRY0(method) \
case PROTOCOL_ID(method):
#define BEGIN_PROTOCOL_ENTRY1(method,t1,f1) \
case PROTOCOL_ID(method):
#define BEGIN_PROTOCOL_ENTRY2(method,t1,f1,t2,f2) \
case PROTOCOL_ID(method):
#define BEGIN_PROTOCOL_ENTRY3(method,t1,f1,t2,f2,t3,f3) \
case PROTOCOL_ID(method):
#define BEGIN_PROTOCOL_ENTRY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
case PROTOCOL_ID(method):
#define BEGIN_PROTOCOL_ENTRY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
case PROTOCOL_ID(method):
#define BEGIN_PROTOCOL_ENTRY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
case PROTOCOL_ID(method):
#define BEGIN_PROTOCOL_ENTRY_HEAVY0(method) \
BEGIN_PROTOCOL_ENTRY0 (method)
#define BEGIN_PROTOCOL_ENTRY_HEAVY1(method,t1,f1) \
BEGIN_PROTOCOL_ENTRY1 (method,t1,f1)
#define BEGIN_PROTOCOL_ENTRY_HEAVY2(method,t1,f1,t2,f2) \
BEGIN_PROTOCOL_ENTRY2 (method,t1,f1,t2,f2)
#define BEGIN_PROTOCOL_ENTRY_HEAVY3(method,t1,f1,t2,f2,t3,f3) \
BEGIN_PROTOCOL_ENTRY3 (method,t1,f1,t2,f2,t3,f3)
#define BEGIN_PROTOCOL_ENTRY_HEAVY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
BEGIN_PROTOCOL_ENTRY4 (method,t1,f1,t2,f2,t3,f3,t4,f4)
#define BEGIN_PROTOCOL_ENTRY_HEAVY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
BEGIN_PROTOCOL_ENTRY5 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5)
#define BEGIN_PROTOCOL_ENTRY_HEAVY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
BEGIN_PROTOCOL_ENTRY6 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6)
#define FLUSH()
#define DEFAULT_PRINT()
#define CUSTOM_PRINT(_)
#define IS_ALWAYS_MATCH(is_always_match) \
return is_always_match;
#define MATCH_INDEX(_)
#define IS_VTABLE_MATCH(_)
#define END_PROTOCOL_ENTRY
#define END_PROTOCOL_ENTRY_HEAVY
#include <mono/metadata/sgen-protocol-def.h>
default: assert (0);
}
}
#define WORKER_PREFIX(t) (WORKER ((t)) ? "w" : " ")
enum { NO_COLOR = -1 };
typedef struct {
int type;
const char *name;
void *data;
/* The index of the ANSI color with which to highlight
* this entry, or NO_COLOR for no highlighting.
*/
int color;
} PrintEntry;
#define TYPE_INT 0
#define TYPE_LONGLONG 1
#define TYPE_SIZE 2
#define TYPE_POINTER 3
static void
print_entry_content (int entries_size, PrintEntry *entries, gboolean color_output)
{
int i;
for (i = 0; i < entries_size; ++i) {
printf ("%s%s ", i == 0 ? "" : " ", entries [i].name);
if (color_output && entries [i].color != NO_COLOR)
/* Set foreground color, excluding black & white. */
printf ("\x1B[%dm", 31 + (entries [i].color % 6));
switch (entries [i].type) {
case TYPE_INT:
printf ("%d", *(int*) entries [i].data);
break;
case TYPE_LONGLONG:
printf ("%lld", *(long long*) entries [i].data);
break;
case TYPE_SIZE:
printf ("%lu", *(size_t*) entries [i].data);
break;
case TYPE_POINTER:
printf ("%p", *(gpointer*) entries [i].data);
break;
default:
assert (0);
}
if (color_output && entries [i].color != NO_COLOR)
/* Reset foreground color to default. */
printf ("\x1B[0m");
}
}
static int
index_color (int index, int num_nums, int *match_indices)
{
int result;
for (result = 0; result < num_nums + 1; ++result)
if (index == match_indices [result])
return result;
return NO_COLOR;
}
static void
print_entry (int type, void *data, int num_nums, int *match_indices, gboolean color_output)
{
const char *always_prefix = is_always_match (type) ? " " : "";
printf ("%s%s ", WORKER_PREFIX (type), always_prefix);
switch (TYPE (type)) {
#define BEGIN_PROTOCOL_ENTRY0(method) \
case PROTOCOL_ID(method): { \
const int pes_size G_GNUC_UNUSED = 0; \
PrintEntry pes [1] G_GNUC_UNUSED; \
printf ("%s", #method + strlen ("binary_protocol_"));
#define BEGIN_PROTOCOL_ENTRY1(method,t1,f1) \
case PROTOCOL_ID(method): { \
PROTOCOL_STRUCT (method) *entry = data; \
const int pes_size G_GNUC_UNUSED = 1; \
PrintEntry pes [1] G_GNUC_UNUSED; \
pes [0].type = t1; \
pes [0].name = #f1; \
pes [0].data = &entry->f1; \
pes [0].color = index_color(0, num_nums, match_indices); \
printf ("%s ", #method + strlen ("binary_protocol_"));
#define BEGIN_PROTOCOL_ENTRY2(method,t1,f1,t2,f2) \
case PROTOCOL_ID(method): { \
PROTOCOL_STRUCT (method) *entry = data; \
const int pes_size G_GNUC_UNUSED = 2; \
PrintEntry pes [2] G_GNUC_UNUSED; \
pes [0].type = t1; \
pes [0].name = #f1; \
pes [0].data = &entry->f1; \
pes [0].color = index_color(0, num_nums, match_indices); \
pes [1].type = t2; \
pes [1].name = #f2; \
pes [1].data = &entry->f2; \
pes [1].color = index_color(1, num_nums, match_indices); \
printf ("%s ", #method + strlen ("binary_protocol_"));
#define BEGIN_PROTOCOL_ENTRY3(method,t1,f1,t2,f2,t3,f3) \
case PROTOCOL_ID(method): { \
PROTOCOL_STRUCT (method) *entry = data; \
const int pes_size G_GNUC_UNUSED = 3; \
PrintEntry pes [3] G_GNUC_UNUSED; \
pes [0].type = t1; \
pes [0].name = #f1; \
pes [0].data = &entry->f1; \
pes [0].color = index_color(0, num_nums, match_indices); \
pes [1].type = t2; \
pes [1].name = #f2; \
pes [1].data = &entry->f2; \
pes [1].color = index_color(1, num_nums, match_indices); \
pes [2].type = t3; \
pes [2].name = #f3; \
pes [2].data = &entry->f3; \
pes [2].color = index_color(2, num_nums, match_indices); \
printf ("%s ", #method + strlen ("binary_protocol_"));
#define BEGIN_PROTOCOL_ENTRY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
case PROTOCOL_ID(method): { \
PROTOCOL_STRUCT (method) *entry = data; \
const int pes_size G_GNUC_UNUSED = 4; \
PrintEntry pes [4] G_GNUC_UNUSED; \
pes [0].type = t1; \
pes [0].name = #f1; \
pes [0].data = &entry->f1; \
pes [0].color = index_color(0, num_nums, match_indices); \
pes [1].type = t2; \
pes [1].name = #f2; \
pes [1].data = &entry->f2; \
pes [1].color = index_color(1, num_nums, match_indices); \
pes [2].type = t3; \
pes [2].name = #f3; \
pes [2].data = &entry->f3; \
pes [2].color = index_color(2, num_nums, match_indices); \
pes [3].type = t4; \
pes [3].name = #f4; \
pes [3].data = &entry->f4; \
pes [3].color = index_color(3, num_nums, match_indices); \
printf ("%s ", #method + strlen ("binary_protocol_"));
#define BEGIN_PROTOCOL_ENTRY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
case PROTOCOL_ID(method): { \
PROTOCOL_STRUCT (method) *entry = data; \
const int pes_size G_GNUC_UNUSED = 5; \
PrintEntry pes [5] G_GNUC_UNUSED; \
pes [0].type = t1; \
pes [0].name = #f1; \
pes [0].data = &entry->f1; \
pes [0].color = index_color(0, num_nums, match_indices); \
pes [1].type = t2; \
pes [1].name = #f2; \
pes [1].data = &entry->f2; \
pes [1].color = index_color(1, num_nums, match_indices); \
pes [2].type = t3; \
pes [2].name = #f3; \
pes [2].data = &entry->f3; \
pes [2].color = index_color(2, num_nums, match_indices); \
pes [3].type = t4; \
pes [3].name = #f4; \
pes [3].data = &entry->f4; \
pes [3].color = index_color(3, num_nums, match_indices); \
pes [4].type = t5; \
pes [4].name = #f5; \
pes [4].data = &entry->f5; \
pes [4].color = index_color(4, num_nums, match_indices); \
printf ("%s ", #method + strlen ("binary_protocol_"));
#define BEGIN_PROTOCOL_ENTRY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
case PROTOCOL_ID(method): { \
PROTOCOL_STRUCT (method) *entry = data; \
const int pes_size G_GNUC_UNUSED = 6; \
PrintEntry pes [6] G_GNUC_UNUSED; \
pes [0].type = t1; \
pes [0].name = #f1; \
pes [0].data = &entry->f1; \
pes [0].color = index_color(0, num_nums, match_indices); \
pes [1].type = t2; \
pes [1].name = #f2; \
pes [1].data = &entry->f2; \
pes [1].color = index_color(1, num_nums, match_indices); \
pes [2].type = t3; \
pes [2].name = #f3; \
pes [2].data = &entry->f3; \
pes [2].color = index_color(2, num_nums, match_indices); \
pes [3].type = t4; \
pes [3].name = #f4; \
pes [3].data = &entry->f4; \
pes [3].color = index_color(3, num_nums, match_indices); \
pes [4].type = t5; \
pes [4].name = #f5; \
pes [4].data = &entry->f5; \
pes [4].color = index_color(4, num_nums, match_indices); \
pes [5].type = t6; \
pes [5].name = #f6; \
pes [5].data = &entry->f6; \
pes [5].color = index_color(5, num_nums, match_indices); \
printf ("%s ", #method + strlen ("binary_protocol_"));
#define BEGIN_PROTOCOL_ENTRY_HEAVY0(method) \
BEGIN_PROTOCOL_ENTRY0 (method)
#define BEGIN_PROTOCOL_ENTRY_HEAVY1(method,t1,f1) \
BEGIN_PROTOCOL_ENTRY1 (method,t1,f1)
#define BEGIN_PROTOCOL_ENTRY_HEAVY2(method,t1,f1,t2,f2) \
BEGIN_PROTOCOL_ENTRY2 (method,t1,f1,t2,f2)
#define BEGIN_PROTOCOL_ENTRY_HEAVY3(method,t1,f1,t2,f2,t3,f3) \
BEGIN_PROTOCOL_ENTRY3 (method,t1,f1,t2,f2,t3,f3)
#define BEGIN_PROTOCOL_ENTRY_HEAVY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
BEGIN_PROTOCOL_ENTRY4 (method,t1,f1,t2,f2,t3,f3,t4,f4)
#define BEGIN_PROTOCOL_ENTRY_HEAVY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
BEGIN_PROTOCOL_ENTRY5 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5)
#define BEGIN_PROTOCOL_ENTRY_HEAVY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
BEGIN_PROTOCOL_ENTRY6 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6)
#define FLUSH()
#define DEFAULT_PRINT() \
print_entry_content (pes_size, pes, color_output);
#define CUSTOM_PRINT(print) \
print;
#define IS_ALWAYS_MATCH(_)
#define MATCH_INDEX(_)
#define IS_VTABLE_MATCH(_)
#define END_PROTOCOL_ENTRY \
printf ("\n"); \
break; \
}
#define END_PROTOCOL_ENTRY_HEAVY \
END_PROTOCOL_ENTRY
#include <mono/metadata/sgen-protocol-def.h>
default: assert (0);
}
}
#undef TYPE_INT
#undef TYPE_LONGLONG
#undef TYPE_SIZE
#undef TYPE_POINTER
#define TYPE_INT int
#define TYPE_LONGLONG long long
#define TYPE_SIZE size_t
#define TYPE_POINTER gpointer
static gboolean
matches_interval (gpointer ptr, gpointer start, int size)
{
return ptr >= start && (char*)ptr < (char*)start + size;
}
/* Returns the index of the field where a match was found,
* BINARY_PROTOCOL_NO_MATCH for no match, or
* BINARY_PROTOCOL_MATCH for a match with no index.
*/
static int
match_index (gpointer ptr, int type, void *data)
{
switch (TYPE (type)) {
#define BEGIN_PROTOCOL_ENTRY0(method) \
case PROTOCOL_ID (method): {
#define BEGIN_PROTOCOL_ENTRY1(method,t1,f1) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY2(method,t1,f1,t2,f2) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY3(method,t1,f1,t2,f2,t3,f3) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY_HEAVY0(method) \
BEGIN_PROTOCOL_ENTRY0 (method)
#define BEGIN_PROTOCOL_ENTRY_HEAVY1(method,t1,f1) \
BEGIN_PROTOCOL_ENTRY1 (method,t1,f1)
#define BEGIN_PROTOCOL_ENTRY_HEAVY2(method,t1,f1,t2,f2) \
BEGIN_PROTOCOL_ENTRY2 (method,t1,f1,t2,f2)
#define BEGIN_PROTOCOL_ENTRY_HEAVY3(method,t1,f1,t2,f2,t3,f3) \
BEGIN_PROTOCOL_ENTRY3 (method,t1,f1,t2,f2,t3,f3)
#define BEGIN_PROTOCOL_ENTRY_HEAVY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
BEGIN_PROTOCOL_ENTRY4 (method,t1,f1,t2,f2,t3,f3,t4,f4)
#define BEGIN_PROTOCOL_ENTRY_HEAVY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
BEGIN_PROTOCOL_ENTRY5 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5)
#define BEGIN_PROTOCOL_ENTRY_HEAVY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
BEGIN_PROTOCOL_ENTRY6 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6)
#define FLUSH()
#define DEFAULT_PRINT()
#define CUSTOM_PRINT(_)
#define IS_ALWAYS_MATCH(_)
#define MATCH_INDEX(block) \
return (block);
#define IS_VTABLE_MATCH(_)
#define END_PROTOCOL_ENTRY \
break; \
}
#define END_PROTOCOL_ENTRY_HEAVY \
END_PROTOCOL_ENTRY
#include <mono/metadata/sgen-protocol-def.h>
default: assert (0);
}
}
static gboolean
is_vtable_match (gpointer ptr, int type, void *data)
{
switch (TYPE (type)) {
#define BEGIN_PROTOCOL_ENTRY0(method) \
case PROTOCOL_ID (method): {
#define BEGIN_PROTOCOL_ENTRY1(method,t1,f1) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY2(method,t1,f1,t2,f2) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY3(method,t1,f1,t2,f2,t3,f3) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
case PROTOCOL_ID (method): { \
PROTOCOL_STRUCT (method) *entry G_GNUC_UNUSED = data;
#define BEGIN_PROTOCOL_ENTRY_HEAVY0(method) \
BEGIN_PROTOCOL_ENTRY0 (method)
#define BEGIN_PROTOCOL_ENTRY_HEAVY1(method,t1,f1) \
BEGIN_PROTOCOL_ENTRY1 (method,t1,f1)
#define BEGIN_PROTOCOL_ENTRY_HEAVY2(method,t1,f1,t2,f2) \
BEGIN_PROTOCOL_ENTRY2 (method,t1,f1,t2,f2)
#define BEGIN_PROTOCOL_ENTRY_HEAVY3(method,t1,f1,t2,f2,t3,f3) \
BEGIN_PROTOCOL_ENTRY3 (method,t1,f1,t2,f2,t3,f3)
#define BEGIN_PROTOCOL_ENTRY_HEAVY4(method,t1,f1,t2,f2,t3,f3,t4,f4) \
BEGIN_PROTOCOL_ENTRY4 (method,t1,f1,t2,f2,t3,f3,t4,f4)
#define BEGIN_PROTOCOL_ENTRY_HEAVY5(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5) \
BEGIN_PROTOCOL_ENTRY5 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5)
#define BEGIN_PROTOCOL_ENTRY_HEAVY6(method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6) \
BEGIN_PROTOCOL_ENTRY6 (method,t1,f1,t2,f2,t3,f3,t4,f4,t5,f5,t6,f6)
#define FLUSH()
#define DEFAULT_PRINT()
#define CUSTOM_PRINT(_)
#define IS_ALWAYS_MATCH(_)
#define MATCH_INDEX(block) \
return (block);
#define IS_VTABLE_MATCH(_)
#define END_PROTOCOL_ENTRY \
break; \
}
#define END_PROTOCOL_ENTRY_HEAVY \
END_PROTOCOL_ENTRY
#include <mono/metadata/sgen-protocol-def.h>
default: assert (0);
}
}
#undef TYPE_INT
#undef TYPE_LONGLONG
#undef TYPE_SIZE
#undef TYPE_POINTER
int
main (int argc, char *argv[])
{
int type;
void *data = g_malloc0 (MAX_ENTRY_SIZE);
int num_args = argc - 1;
int num_nums = 0;
int num_vtables = 0;
int i;
long nums [num_args];
long vtables [num_args];
gboolean dump_all = FALSE;
gboolean pause_times = FALSE;
gboolean pause_times_stopped = FALSE;
gboolean pause_times_concurrent = FALSE;
gboolean pause_times_finish = FALSE;
gboolean color_output = FALSE;
long long pause_times_ts = 0;
const char *input_path = NULL;
int input_file;
EntryStream stream;
for (i = 0; i < num_args; ++i) {
char *arg = argv [i + 1];
char *next_arg = argv [i + 2];
if (!strcmp (arg, "--all")) {
dump_all = TRUE;
} else if (!strcmp (arg, "--pause-times")) {
pause_times = TRUE;
} else if (!strcmp (arg, "-v") || !strcmp (arg, "--vtable")) {
vtables [num_vtables++] = strtoul (next_arg, NULL, 16);
++i;
} else if (!strcmp (arg, "-c") || !strcmp (arg, "--color")) {
color_output = TRUE;
} else if (!strcmp (arg, "-i") || !strcmp (arg, "--input")) {
input_path = next_arg;
++i;
} else {
nums [num_nums++] = strtoul (arg, NULL, 16);
}
}
if (dump_all)
assert (!pause_times);
if (pause_times)
assert (!dump_all);
input_file = input_path ? open (input_path, O_RDONLY) : STDIN_FILENO;
init_stream (&stream, input_file);
while ((type = read_entry (&stream, data)) != SGEN_PROTOCOL_EOF) {
if (pause_times) {
switch (type) {
case PROTOCOL_ID (binary_protocol_world_stopping): {
PROTOCOL_STRUCT (binary_protocol_world_stopping) *entry = data;
assert (!pause_times_stopped);
pause_times_concurrent = FALSE;
pause_times_finish = FALSE;
pause_times_ts = entry->timestamp;
pause_times_stopped = TRUE;
break;
}
case PROTOCOL_ID (binary_protocol_concurrent_finish):
pause_times_finish = TRUE;
case PROTOCOL_ID (binary_protocol_concurrent_start):
case PROTOCOL_ID (binary_protocol_concurrent_update):
pause_times_concurrent = TRUE;
break;
case PROTOCOL_ID (binary_protocol_world_restarted): {
PROTOCOL_STRUCT (binary_protocol_world_restarted) *entry = data;
assert (pause_times_stopped);
printf ("pause-time %d %d %d %lld %lld\n",
entry->generation,
pause_times_concurrent,
pause_times_finish,
entry->timestamp - pause_times_ts,
pause_times_ts);
pause_times_stopped = FALSE;
break;
}
}
} else {
int match_indices [num_nums + 1];
gboolean match = is_always_match (type);
match_indices [num_nums] = num_nums == 0 ? match_index (NULL, type, data) : BINARY_PROTOCOL_NO_MATCH;
match = match_indices [num_nums] != BINARY_PROTOCOL_NO_MATCH;
for (i = 0; i < num_nums; ++i) {
match_indices [i] = match_index ((gpointer) nums [i], type, data);
match = match || match_indices [i] != BINARY_PROTOCOL_NO_MATCH;
}
if (!match) {
for (i = 0; i < num_vtables; ++i) {
if (is_vtable_match ((gpointer) vtables [i], type, data)) {
match = TRUE;
break;
}
}
}
if (dump_all)
printf (match ? "* " : " ");
if (match || dump_all)
print_entry (type, data, num_nums, match_indices, color_output);
}
}
close_stream (&stream);
if (input_path)
close (input_file);
g_free (data);
return 0;
}