Browse Source

net_uv: getsockname binding

v0.7.4-release
Henry Rawas 14 years ago
committed by Ryan Dahl
parent
commit
f3f3b12364
  1. 1
      Makefile
  2. 8
      lib/net_uv.js
  3. 60
      src/tcp_wrap.cc

1
Makefile

@ -151,6 +151,7 @@ UVTEST += simple/test-net-keepalive
UVTEST += simple/test-net-pingpong UVTEST += simple/test-net-pingpong
UVTEST += simple/test-net-reconnect UVTEST += simple/test-net-reconnect
UVTEST += simple/test-net-remote-address-port UVTEST += simple/test-net-remote-address-port
UVTEST += simple/test-net-server-bind
UVTEST += simple/test-net-server-max-connections UVTEST += simple/test-net-server-max-connections
UVTEST += simple/test-net-server-try-ports UVTEST += simple/test-net-server-try-ports
UVTEST += simple/test-net-stream UVTEST += simple/test-net-stream

8
lib/net_uv.js

@ -93,6 +93,11 @@ Socket.prototype.setKeepAlive = function(setting, msecs) {
}; };
Socket.prototype.address = function() {
return this._handle.getsockname();
};
Object.defineProperty(Socket.prototype, 'readyState', { Object.defineProperty(Socket.prototype, 'readyState', {
get: function() { get: function() {
if (this._connecting) { if (this._connecting) {
@ -559,6 +564,9 @@ Server.prototype.listen = function() {
} }
}; };
Server.prototype.address = function() {
return this._handle.getsockname();
};
function onconnection(clientHandle) { function onconnection(clientHandle) {
var handle = this; var handle = this;

60
src/tcp_wrap.cc

@ -1,6 +1,21 @@
#include <node.h> #include <node.h>
#include <node_buffer.h> #include <node_buffer.h>
// Temporary hack: libuv should provide uv_inet_pton and uv_inet_ntop.
#ifdef __MINGW32__
extern "C" {
# include <inet_net_pton.h>
# include <inet_ntop.h>
}
# define uv_inet_pton ares_inet_pton
# define uv_inet_ntop ares_inet_ntop
#else // __POSIX__
# include <arpa/inet.h>
# define uv_inet_pton inet_pton
# define uv_inet_ntop inet_ntop
#endif
#define SLAB_SIZE (1024 * 1024) #define SLAB_SIZE (1024 * 1024)
#define MIN(a, b) ((a) < (b) ? (a) : (b)) #define MIN(a, b) ((a) < (b) ? (a) : (b))
@ -57,6 +72,9 @@ static uv_tcp_t* handle_that_last_alloced;
static Persistent<String> slab_sym; static Persistent<String> slab_sym;
static Persistent<String> buffer_sym; static Persistent<String> buffer_sym;
static Persistent<String> family_symbol;
static Persistent<String> address_symbol;
static Persistent<String> port_symbol;
static Persistent<String> write_queue_size_sym; static Persistent<String> write_queue_size_sym;
class TCPWrap; class TCPWrap;
@ -113,6 +131,7 @@ class TCPWrap {
NODE_SET_PROTOTYPE_METHOD(t, "close", Close); NODE_SET_PROTOTYPE_METHOD(t, "close", Close);
NODE_SET_PROTOTYPE_METHOD(t, "bind6", Bind6); NODE_SET_PROTOTYPE_METHOD(t, "bind6", Bind6);
NODE_SET_PROTOTYPE_METHOD(t, "connect6", Connect6); NODE_SET_PROTOTYPE_METHOD(t, "connect6", Connect6);
NODE_SET_PROTOTYPE_METHOD(t, "getsockname", GetSockName);
constructor = Persistent<Function>::New(t->GetFunction()); constructor = Persistent<Function>::New(t->GetFunction());
@ -121,6 +140,10 @@ class TCPWrap {
write_queue_size_sym = write_queue_size_sym =
Persistent<String>::New(String::NewSymbol("writeQueueSize")); Persistent<String>::New(String::NewSymbol("writeQueueSize"));
family_symbol = NODE_PSYMBOL("family");
address_symbol = NODE_PSYMBOL("address");
port_symbol = NODE_PSYMBOL("port");
target->Set(String::NewSymbol("TCP"), constructor); target->Set(String::NewSymbol("TCP"), constructor);
} }
@ -172,6 +195,43 @@ class TCPWrap {
object_->Set(write_queue_size_sym, Integer::New(handle_.write_queue_size)); object_->Set(write_queue_size_sym, Integer::New(handle_.write_queue_size));
} }
static Handle<Value> GetSockName(const Arguments& args) {
HandleScope scope;
struct sockaddr address;
int family;
int port;
char ip[INET6_ADDRSTRLEN];
UNWRAP
int namelen = sizeof(address);
int r = uv_getsockname(&wrap->handle_, &address, &namelen);
Local<Object> sockname = Object::New();
if (r != 0) {
SetErrno(uv_last_error().code);
} else {
family = address.sa_family;
if (family == AF_INET) {
struct sockaddr_in* addrin = (struct sockaddr_in*)&address;
uv_inet_ntop(AF_INET, &(addrin->sin_addr), ip, INET6_ADDRSTRLEN);
port = ntohs(addrin->sin_port);
} else if (family == AF_INET6) {
struct sockaddr_in6* addrin6 = (struct sockaddr_in6*)&address;
uv_inet_ntop(AF_INET6, &(addrin6->sin6_addr), ip, INET6_ADDRSTRLEN);
port = ntohs(addrin6->sin6_port);
}
sockname->Set(port_symbol, Integer::New(port));
sockname->Set(family_symbol, Integer::New(family));
sockname->Set(address_symbol, String::New(ip));
}
return scope.Close(sockname);
}
static Handle<Value> Bind(const Arguments& args) { static Handle<Value> Bind(const Arguments& args) {
HandleScope scope; HandleScope scope;

Loading…
Cancel
Save