mirror of
https://github.com/nodejs/node.git
synced 2024-12-01 16:10:02 +01:00
309 lines
9.1 KiB
C++
309 lines
9.1 KiB
C++
// Copyright Joyent, Inc. and other Node contributors.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
|
// copy of this software and associated documentation files (the
|
|
// "Software"), to deal in the Software without restriction, including
|
|
// without limitation the rights to use, copy, modify, merge, publish,
|
|
// distribute, sublicense, and/or sell copies of the Software, and to permit
|
|
// persons to whom the Software is furnished to do so, subject to the
|
|
// following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included
|
|
// in all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
|
|
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
|
|
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
|
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
|
// USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
#include <node_dtrace.h>
|
|
|
|
#ifdef HAVE_DTRACE
|
|
#include "node_provider.h"
|
|
#else
|
|
#define NODE_HTTP_SERVER_REQUEST(arg0, arg1)
|
|
#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 {
|
|
|
|
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, 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())
|
|
return Undefined();
|
|
|
|
HandleScope scope;
|
|
|
|
SLURP_CONNECTION(args[0], conn);
|
|
NODE_NET_SERVER_CONNECTION(&conn);
|
|
|
|
return Undefined();
|
|
}
|
|
|
|
Handle<Value> DTRACE_NET_STREAM_END(const Arguments& args) {
|
|
if (!NODE_NET_STREAM_END_ENABLED())
|
|
return Undefined();
|
|
|
|
HandleScope scope;
|
|
|
|
SLURP_CONNECTION(args[0], conn);
|
|
NODE_NET_STREAM_END(&conn);
|
|
|
|
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;
|
|
|
|
if (!NODE_HTTP_SERVER_REQUEST_ENABLED())
|
|
return Undefined();
|
|
|
|
HandleScope scope;
|
|
|
|
Local<Object> arg0 = Local<Object>::Cast(args[0]);
|
|
|
|
SLURP_STRING(arg0, url, &req.url);
|
|
SLURP_STRING(arg0, method, &req.method);
|
|
|
|
SLURP_CONNECTION(args[1], conn);
|
|
|
|
NODE_HTTP_SERVER_REQUEST(&req, &conn);
|
|
return Undefined();
|
|
}
|
|
|
|
Handle<Value> DTRACE_HTTP_SERVER_RESPONSE(const Arguments& args) {
|
|
if (!NODE_HTTP_SERVER_RESPONSE_ENABLED())
|
|
return Undefined();
|
|
|
|
HandleScope scope;
|
|
|
|
SLURP_CONNECTION(args[0], conn);
|
|
NODE_HTTP_SERVER_RESPONSE(&conn);
|
|
|
|
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;
|
|
Handle<Value> (*func)(const Arguments&);
|
|
Persistent<FunctionTemplate> templ;
|
|
} 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 }
|
|
};
|
|
|
|
for (int i = 0; tab[i].name != NULL; i++) {
|
|
tab[i].templ = Persistent<FunctionTemplate>::New(
|
|
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
|
|
}
|
|
|
|
}
|