diff options
author | Ryan Dahl <ry@tinyclouds.org> | 2011-02-10 05:50:26 +0300 |
---|---|---|
committer | Ryan Dahl <ry@tinyclouds.org> | 2011-02-10 05:50:26 +0300 |
commit | e9257b859d6bbeb68b47abf03741bc5378538b05 (patch) | |
tree | 5e48e386bb9b22708706b6965690c585ed4afe54 /src/node_dtrace.cc | |
parent | 778fb859c640f1cb67d11ce4550758b6bae31b43 (diff) |
New DTrace probes from CA team
Diffstat (limited to 'src/node_dtrace.cc')
-rw-r--r-- | src/node_dtrace.cc | 178 |
1 files changed, 176 insertions, 2 deletions
diff --git a/src/node_dtrace.cc b/src/node_dtrace.cc index 0f7726a6209..ca7495e2f75 100644 --- a/src/node_dtrace.cc +++ b/src/node_dtrace.cc @@ -7,10 +7,20 @@ #define NODE_HTTP_SERVER_REQUEST_ENABLED() (0) #define NODE_HTTP_SERVER_RESPONSE(arg0) #define NODE_HTTP_SERVER_RESPONSE_ENABLED() (0) +#define NODE_HTTP_CLIENT_REQUEST(arg0, arg1) +#define NODE_HTTP_CLIENT_REQUEST_ENABLED() (0) +#define NODE_HTTP_CLIENT_RESPONSE(arg0) +#define NODE_HTTP_CLIENT_RESPONSE_ENABLED() (0) #define NODE_NET_SERVER_CONNECTION(arg0) #define NODE_NET_SERVER_CONNECTION_ENABLED() (0) #define NODE_NET_STREAM_END(arg0) #define NODE_NET_STREAM_END_ENABLED() (0) +#define NODE_NET_SOCKET_READ(arg0, arg1) +#define NODE_NET_SOCKET_READ_ENABLED() (0) +#define NODE_NET_SOCKET_WRITE(arg0, arg1) +#define NODE_NET_SOCKET_WRITE_ENABLED() (0) +#define NODE_GC_START(arg0, arg1) +#define NODE_GC_DONE(arg0, arg1) #endif namespace node { @@ -18,19 +28,69 @@ namespace node { using namespace v8; #define SLURP_STRING(obj, member, valp) \ + if (!(obj)->IsObject()) { \ + return (ThrowException(Exception::Error(String::New("expected " \ + "object for " #obj " to contain string member " #member)))); \ + } \ String::Utf8Value _##member(obj->Get(String::New(#member))->ToString()); \ if ((*(const char **)valp = *_##member) == NULL) \ *(const char **)valp = "<unknown>"; #define SLURP_INT(obj, member, valp) \ + if (!(obj)->IsObject()) { \ + return (ThrowException(Exception::Error(String::New("expected " \ + "object for " #obj " to contain integer member " #member)))); \ + } \ *valp = obj->Get(String::New(#member))->ToInteger()->Value(); +#define SLURP_OBJECT(obj, member, valp) \ + if (!(obj)->IsObject()) { \ + return (ThrowException(Exception::Error(String::New("expected " \ + "object for " #obj " to contain object member " #member)))); \ + } \ + *valp = Local<Object>::Cast(obj->Get(String::New(#member))); + #define SLURP_CONNECTION(arg, conn) \ + if (!(arg)->IsObject()) { \ + return (ThrowException(Exception::Error(String::New("expected " \ + "argument " #arg " to be a connection object")))); \ + } \ node_dtrace_connection_t conn; \ Local<Object> _##conn = Local<Object>::Cast(arg); \ SLURP_INT(_##conn, fd, &conn.fd); \ SLURP_STRING(_##conn, remoteAddress, &conn.remote); \ - SLURP_INT(_##conn, remotePort, &conn.port); + SLURP_INT(_##conn, remotePort, &conn.port); \ + SLURP_INT(_##conn, bufferSize, &conn.buffered); + +#define SLURP_CONNECTION_HTTP_CLIENT(arg, conn) \ + if (!(arg)->IsObject()) { \ + return (ThrowException(Exception::Error(String::New("expected " \ + "argument " #arg " to be a connection object")))); \ + } \ + node_dtrace_connection_t conn; \ + Local<Object> _##conn = Local<Object>::Cast(arg); \ + SLURP_INT(_##conn, fd, &conn.fd); \ + SLURP_STRING(_##conn, host, &conn.remote); \ + SLURP_INT(_##conn, port, &conn.port); \ + SLURP_INT(_##conn, bufferSize, &conn.buffered); + +#define SLURP_CONNECTION_HTTP_CLIENT_RESPONSE(arg0, arg1, conn) \ + if (!(arg0)->IsObject()) { \ + return (ThrowException(Exception::Error(String::New("expected " \ + "argument " #arg0 " to be a connection object")))); \ + } \ + if (!(arg1)->IsObject()) { \ + return (ThrowException(Exception::Error(String::New("expected " \ + "argument " #arg1 " to be a connection object")))); \ + } \ + node_dtrace_connection_t conn; \ + Local<Object> _##conn = Local<Object>::Cast(arg0); \ + SLURP_INT(_##conn, fd, &conn.fd); \ + SLURP_INT(_##conn, bufferSize, &conn.buffered); \ + _##conn = Local<Object>::Cast(arg1); \ + SLURP_STRING(_##conn, host, &conn.remote); \ + SLURP_INT(_##conn, port, &conn.port); + Handle<Value> DTRACE_NET_SERVER_CONNECTION(const Arguments& args) { if (!NODE_NET_SERVER_CONNECTION_ENABLED()) @@ -56,6 +116,48 @@ Handle<Value> DTRACE_NET_STREAM_END(const Arguments& args) { return Undefined(); } +Handle<Value> DTRACE_NET_SOCKET_READ(const Arguments& args) { + if (!NODE_NET_SOCKET_READ_ENABLED()) + return Undefined(); + + HandleScope scope; + int nbytes; + + SLURP_CONNECTION(args[0], conn); + + if (!args[1]->IsNumber()) { + return (ThrowException(Exception::Error(String::New("expected " + "argument 1 to be number of bytes")))); + } + + nbytes = args[1]->Int32Value(); + + NODE_NET_SOCKET_READ(&conn, nbytes); + + return Undefined(); +} + +Handle<Value> DTRACE_NET_SOCKET_WRITE(const Arguments& args) { + if (!NODE_NET_SOCKET_WRITE_ENABLED()) + return Undefined(); + + HandleScope scope; + int nbytes; + + SLURP_CONNECTION(args[0], conn); + + if (!args[1]->IsNumber()) { + return (ThrowException(Exception::Error(String::New("expected " + "argument 1 to be number of bytes")))); + } + + nbytes = args[1]->Int32Value(); + + NODE_NET_SOCKET_WRITE(&conn, nbytes); + + return Undefined(); +} + Handle<Value> DTRACE_HTTP_SERVER_REQUEST(const Arguments& args) { node_dtrace_http_request_t req; @@ -65,7 +167,6 @@ Handle<Value> DTRACE_HTTP_SERVER_REQUEST(const Arguments& args) { HandleScope scope; Local<Object> arg0 = Local<Object>::Cast(args[0]); - Local<Object> arg1 = Local<Object>::Cast(args[1]); SLURP_STRING(arg0, url, &req.url); SLURP_STRING(arg0, method, &req.method); @@ -88,8 +189,72 @@ Handle<Value> DTRACE_HTTP_SERVER_RESPONSE(const Arguments& args) { return Undefined(); } +Handle<Value> DTRACE_HTTP_CLIENT_REQUEST(const Arguments& args) { + node_dtrace_http_request_t req; + char *header; + + if (!NODE_HTTP_CLIENT_REQUEST_ENABLED()) + return Undefined(); + + HandleScope scope; + + /* + * For the method and URL, we're going to dig them out of the header. This + * is not as efficient as it could be, but we would rather not force the + * caller here to retain their method and URL until the time at which + * DTRACE_HTTP_CLIENT_REQUEST can be called. + */ + Local<Object> arg0 = Local<Object>::Cast(args[0]); + SLURP_STRING(arg0, _header, &header); + + req.method = header; + + while (*header != '\0' && *header != ' ') + header++; + + if (*header != '\0') + *header++ = '\0'; + + req.url = header; + + while (*header != '\0' && *header != ' ') + header++; + + *header = '\0'; + + SLURP_CONNECTION_HTTP_CLIENT(args[1], conn); + NODE_HTTP_CLIENT_REQUEST(&req, &conn); + return Undefined(); +} + +Handle<Value> DTRACE_HTTP_CLIENT_RESPONSE(const Arguments& args) { + if (!NODE_HTTP_CLIENT_RESPONSE_ENABLED()) + return Undefined(); + + HandleScope scope; + + SLURP_CONNECTION_HTTP_CLIENT_RESPONSE(args[0], args[1], conn); + NODE_HTTP_CLIENT_RESPONSE(&conn); + + return Undefined(); +} + #define NODE_PROBE(name) #name, name +static int dtrace_gc_start(GCType type, GCCallbackFlags flags) { + NODE_GC_START(type, flags); + /* + * We avoid the tail-call elimination of the USDT probe (which screws up + * args) by forcing a return of 0. + */ + return 0; +} + +static int dtrace_gc_done(GCType type, GCCallbackFlags flags) { + NODE_GC_DONE(type, flags); + return 0; +} + void InitDTrace(Handle<Object> target) { static struct { const char *name; @@ -98,8 +263,12 @@ void InitDTrace(Handle<Object> target) { } tab[] = { { NODE_PROBE(DTRACE_NET_SERVER_CONNECTION) }, { NODE_PROBE(DTRACE_NET_STREAM_END) }, + { NODE_PROBE(DTRACE_NET_SOCKET_READ) }, + { NODE_PROBE(DTRACE_NET_SOCKET_WRITE) }, { NODE_PROBE(DTRACE_HTTP_SERVER_REQUEST) }, { NODE_PROBE(DTRACE_HTTP_SERVER_RESPONSE) }, + { NODE_PROBE(DTRACE_HTTP_CLIENT_REQUEST) }, + { NODE_PROBE(DTRACE_HTTP_CLIENT_RESPONSE) }, { NULL } }; @@ -108,6 +277,11 @@ void InitDTrace(Handle<Object> target) { FunctionTemplate::New(tab[i].func)); target->Set(String::NewSymbol(tab[i].name), tab[i].templ->GetFunction()); } + +#ifdef HAVE_DTRACE + v8::V8::AddGCPrologueCallback((GCPrologueCallback)dtrace_gc_start); + v8::V8::AddGCEpilogueCallback((GCEpilogueCallback)dtrace_gc_done); +#endif } } |