void erts_mtrace_init(char *receiver, char *nodename)
{
char hostname[MAXHOSTNAMELEN + 1];
char pid[21]; /* enough for a 64 bit number */
socket_desc = ERTS_SOCK_INVALID_SOCKET;
erts_mtrace_enabled = receiver != NULL;
if (erts_mtrace_enabled) {
unsigned a, b, c, d, p;
byte ip_addr[4];
Uint16 port;
erts_mtx_init(&mtrace_buf_mutex, "mtrace_buf", NIL,
ERTS_LOCK_FLAGS_PROPERTY_STATIC | ERTS_LOCK_FLAGS_CATEGORY_DEBUG);
erts_mtx_init(&mtrace_op_mutex, "mtrace_op", NIL,
ERTS_LOCK_FLAGS_PROPERTY_STATIC | ERTS_LOCK_FLAGS_CATEGORY_DEBUG);
socket_desc = erts_sock_open();
if (socket_desc == ERTS_SOCK_INVALID_SOCKET) {
disable_trace(1, "Failed to open socket", erts_sock_errno());
return;
}
if (5 != sscanf(receiver, "%u.%u.%u.%u:%u", &a, &b, &c, &d, &p)
|| a >= (1 << 8) || b >= (1 << 8)|| c >= (1 << 8) || d >= (1 << 8)
|| p >= (1 << 16)) {
disable_trace(1, "Invalid receiver address", 0);
return;
}
ip_addr[0] = (byte) a;
ip_addr[1] = (byte) b;
ip_addr[2] = (byte) c;
ip_addr[3] = (byte) d;
port = (Uint16) p;
if (!erts_sock_connect(socket_desc, ip_addr, 4, port)) {
disable_trace(1, "Failed to connect to receiver",
erts_sock_errno());
return;
}
tracep = trace_buffer;
endp = trace_buffer + TRACE_BUF_SZ;
/* gethostname requires that the len is max(hostname) + 1 */
if (erts_sock_gethostname(hostname, MAXHOSTNAMELEN + 1) != 0)
hostname[0] = '\0';
hostname[MAXHOSTNAMELEN] = '\0';
sys_get_pid(pid, sizeof(pid));
write_trace_header(nodename ? nodename : "", pid, hostname);
erts_mtrace_update_heap_size();
}
}
static int
send_trace_buffer(void)
{
ssize_t ssz;
size_t sz;
sz = tracep - trace_buffer;
tracep = trace_buffer;
do {
ssz = erts_sock_send(socket_desc, (void *) tracep, sz);
if (ssz < 0) {
int socket_errno = erts_sock_errno();
#ifdef EINTR
if (socket_errno == EINTR)
continue;
#endif
disable_trace(0, "Connection lost", socket_errno);
return 0;
}
if (ssz > sz) {
disable_trace(1, "Unexpected error", 0);
return 0;
}
tracep += ssz;
sz -= ssz;
} while (sz);
tracep = trace_buffer;
return 1;
}
