libdtrace緩衝輸出

我試圖通過libdtrace使用dtrace（在Snow Leopard上; 10.6.4）。我想在我自己的程序中捕獲我的dtrace腳本的打印輸出。一種方法是將輸出轉到臨時文件並從那裏讀取。但是，libdtrace支持一個回調函數來直接捕獲我喜歡的輸出。libdtrace緩衝輸出

我認爲回調只會傳遞格式化的字符串，我可以消耗，但這似乎並不是這種情況。例如，在下面的測試程序中，我期望打印輸出爲「process pid = 86138」。但是，它總是打印出「process pid = 1」（當使用'dtrace -n'運行時，dtrace腳本正常工作）。

我在做什麼錯？我應該如何使用傳遞給緩衝區處理程序的數據？（具體來說，我感興趣的是來自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; 
}

來源

2010-09-25 Sami

緩衝輸出似乎在OSX上被打破。這些行爲似乎以某種扭曲的方式在消費者的背景下以某種方式執行。例如，ustack()完全不起作用。另一方面，copyinstr()似乎功能正常。

可以規避輸出緩衝尚未通過管道仍然可以得到大致相同的結果：

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); 
}

錯誤處理作爲練習留給讀者。

來源

2012-10-17 12:15:46

libdtrace緩衝輸出

回答

相關問題