Буферизованный вывод libdtrace

Question

Я пытаюсь использовать dtrace через libdtrace (на Snow Leopard; 10.6.4). Я хочу поймать распечатку моего сценария dtrace в моей собственной программе. Один из способов сделать это - перевести вывод во временный файл и прочитать его оттуда. Тем не менее, libdtrace поддерживает функцию обратного вызова для непосредственного получения вывода, который я бы предпочел.

Я предполагал, что обратный вызов просто передаст мне отформатированные строки, которые я мог бы потреблять, но, похоже, это не так. Например, в приведенной ниже тестовой программе я ожидаю, что распечатка будет "process pid = 86138". Тем не менее, он всегда выводит "process pid = 1" (скрипт dtrace работает нормально, когда запускается с помощью "dtrace -n").

Что я делаю неправильно? Как я должен использовать данные, передаваемые в обработчик буфера? (В частности, меня интересуют данные от действий printf и tracemem).

#include <dtrace.h>
#include <stdio.h>
#include <stdlib.h>
#include <mach/mach.h>
#include <mach-o/loader.h>
#include <mach-o/dyld.h>
#include <mach-o/fat.h>
#include <sys/sysctl.h>
#include <signal.h>

static const char *g_prog =
  "pid86138::write:entry"
  "{"
  "  printf(\"process pid = %d\\n\", pid);"
  "}";

static int dcmdbuffered(const dtrace_bufdata_t *bufdata, void *arg) {
  if((bufdata->dtbda_recdesc != NULL) && (bufdata->dtbda_recdesc->dtrd_action == DTRACEACT_PRINTF))
    printf("BUF: %s\n", bufdata->dtbda_buffered);

  return DTRACE_HANDLE_OK;
}

static int chew(const dtrace_probedata_t *data, void *arg) {
  return DTRACE_CONSUME_THIS;
}

static int chewrec(const dtrace_probedata_t *data, const dtrace_recdesc_t *rec, void *arg) {
  if(rec == NULL)
    return (DTRACE_CONSUME_NEXT);
  return (DTRACE_CONSUME_THIS);
}

int main(int argc, char **argv) {
  int err, done = 0;
  dtrace_proginfo_t info;

  dtrace_hdl_t *g_dtp = dtrace_open(DTRACE_VERSION, DTRACE_O_ILP32, &err);
  dtrace_prog_t *prog = dtrace_program_strcompile(g_dtp, g_prog, DTRACE_PROBESPEC_NAME, 0, 0, NULL);
  dtrace_handle_buffered(g_dtp, dcmdbuffered, NULL);
  dtrace_program_exec(g_dtp, prog, &info);
  dtrace_setopt(g_dtp, "strsize", "4096");
  dtrace_setopt(g_dtp, "bufsize", "4m");
  dtrace_setopt(g_dtp, "arch", "x86_64");
  dtrace_go(g_dtp);

  while(dtrace_work(g_dtp, NULL, chew, chewrec, NULL) == DTRACE_WORKSTATUS_OKAY)
    dtrace_sleep(g_dtp);

  dtrace_stop(g_dtp);
  dtrace_close(g_dtp);
  return 0;
}

Answer 1

Буферизированный вывод, похоже, не работает в OSX. Действия, кажется, каким-то образом выполняются в контексте потребителя каким-то извращенным образом. ustack() не работает вообще, например. copyinstr() с другой стороны, кажется, функционирует должным образом.

Вы можете обойти буферизацию вывода, но при этом все равно получите тот же результат, используя pipe:

int fds [2];

if (pipe (fds) != 0)
    assert (0);

int flags = fcntl (fds [0], F_GETFL, 0);
assert (flags != -1);
fcntl (fds [0], F_SETFL, flags | O_NONBLOCK);

FILE *faux_stdout = fdopen (fds [1], "a");
assert (faux_stdout);

while(dtrace_work(g_dtp, faux_stdout, chew, chewrec, NULL) == DTRACE_WORKSTATUS_OKAY) {
    char buf [1024];
    for (;;) {
        ssize_t num_read = read (fds [0], buf, sizeof (buf));
        if (num_read <= 0)
            break;
        /* process your buffer here */
        fwrite (buf, 1, num_read, stdout);
    }
    dtrace_sleep(g_dtp);
}

Обработка ошибок оставлена ​​читателю в качестве упражнения.