mirror of https://github.com/lukechilds/node.git
Fedor Indutny
11 years ago
62 changed files with 955 additions and 194 deletions
@ -0,0 +1,114 @@ |
|||
/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
|||
* |
|||
* 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 <assert.h> |
|||
#include <stdlib.h> |
|||
#include <stdio.h> |
|||
#include <string.h> |
|||
|
|||
#include "uv.h" |
|||
#include "internal.h" |
|||
|
|||
|
|||
static void uv__getnameinfo_work(struct uv__work* w) { |
|||
uv_getnameinfo_t* req; |
|||
int err; |
|||
socklen_t salen; |
|||
|
|||
req = container_of(w, uv_getnameinfo_t, work_req); |
|||
|
|||
if (req->storage.ss_family == AF_INET) |
|||
salen = sizeof(struct sockaddr_in); |
|||
else if (req->storage.ss_family == AF_INET6) |
|||
salen = sizeof(struct sockaddr_in6); |
|||
else |
|||
abort(); |
|||
|
|||
err = getnameinfo((struct sockaddr*) &req->storage, |
|||
salen, |
|||
req->host, |
|||
sizeof(req->host), |
|||
req->service, |
|||
sizeof(req->service), |
|||
req->flags); |
|||
req->retcode = uv__getaddrinfo_translate_error(err); |
|||
} |
|||
|
|||
static void uv__getnameinfo_done(struct uv__work* w, int status) { |
|||
uv_getnameinfo_t* req; |
|||
char* host; |
|||
char* service; |
|||
|
|||
req = container_of(w, uv_getnameinfo_t, work_req); |
|||
uv__req_unregister(req->loop, req); |
|||
host = service = NULL; |
|||
|
|||
if (status == -ECANCELED) { |
|||
assert(req->retcode == 0); |
|||
req->retcode = UV_EAI_CANCELED; |
|||
} else if (req->retcode == 0) { |
|||
host = req->host; |
|||
service = req->service; |
|||
} |
|||
|
|||
req->getnameinfo_cb(req, req->retcode, host, service); |
|||
} |
|||
|
|||
/*
|
|||
* Entry point for getnameinfo |
|||
* return 0 if a callback will be made |
|||
* return error code if validation fails |
|||
*/ |
|||
int uv_getnameinfo(uv_loop_t* loop, |
|||
uv_getnameinfo_t* req, |
|||
uv_getnameinfo_cb getnameinfo_cb, |
|||
const struct sockaddr* addr, |
|||
int flags) { |
|||
if (req == NULL || getnameinfo_cb == NULL || addr == NULL) |
|||
return UV_EINVAL; |
|||
|
|||
if (addr->sa_family == AF_INET) { |
|||
memcpy(&req->storage, |
|||
addr, |
|||
sizeof(struct sockaddr_in)); |
|||
} else if (addr->sa_family == AF_INET6) { |
|||
memcpy(&req->storage, |
|||
addr, |
|||
sizeof(struct sockaddr_in6)); |
|||
} else { |
|||
return UV_EINVAL; |
|||
} |
|||
|
|||
uv__req_init(loop, (uv_req_t*)req, UV_GETNAMEINFO); |
|||
|
|||
req->getnameinfo_cb = getnameinfo_cb; |
|||
req->flags = flags; |
|||
req->type = UV_GETNAMEINFO; |
|||
req->loop = loop; |
|||
req->retcode = 0; |
|||
|
|||
uv__work_submit(loop, |
|||
&req->work_req, |
|||
uv__getnameinfo_work, |
|||
uv__getnameinfo_done); |
|||
|
|||
return 0; |
|||
} |
|||
@ -0,0 +1,138 @@ |
|||
/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
|||
* |
|||
* 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 <assert.h> |
|||
#include <malloc.h> |
|||
#include <stdio.h> |
|||
|
|||
#include "uv.h" |
|||
#include "internal.h" |
|||
#include "req-inl.h" |
|||
|
|||
|
|||
/* getnameinfo worker thread implementation */ |
|||
static DWORD WINAPI getnameinfo_thread_proc(void* parameter) { |
|||
uv_getnameinfo_t* req = (uv_getnameinfo_t*)parameter; |
|||
uv_loop_t* loop = req->loop; |
|||
WCHAR host[NI_MAXHOST]; |
|||
WCHAR service[NI_MAXSERV]; |
|||
int ret = 0; |
|||
|
|||
assert(req != NULL); |
|||
|
|||
ret = GetNameInfoW((struct sockaddr*)&req->storage, |
|||
sizeof(req->storage), |
|||
host, |
|||
sizeof(host), |
|||
service, |
|||
sizeof(service), |
|||
req->flags); |
|||
req->retcode = uv__getaddrinfo_translate_error(ret); |
|||
|
|||
/* convert results to UTF-8 */ |
|||
WideCharToMultiByte(CP_UTF8, |
|||
0, |
|||
host, |
|||
-1, |
|||
req->host, |
|||
sizeof(req->host), |
|||
NULL, |
|||
NULL); |
|||
|
|||
WideCharToMultiByte(CP_UTF8, |
|||
0, |
|||
service, |
|||
-1, |
|||
req->service, |
|||
sizeof(req->service), |
|||
NULL, |
|||
NULL); |
|||
|
|||
/* post getnameinfo completed */ |
|||
POST_COMPLETION_FOR_REQ(loop, req); |
|||
|
|||
return 0; |
|||
} |
|||
|
|||
|
|||
/*
|
|||
* Called from uv_run when complete. |
|||
*/ |
|||
void uv_process_getnameinfo_req(uv_loop_t* loop, uv_getnameinfo_t* req) { |
|||
char* host; |
|||
char* service; |
|||
|
|||
if (req->retcode == 0) { |
|||
host = req->host; |
|||
service = req->service; |
|||
} else { |
|||
host = NULL; |
|||
service = NULL; |
|||
} |
|||
|
|||
uv__req_unregister(loop, req); |
|||
req->getnameinfo_cb(req, req->retcode, host, service); |
|||
} |
|||
|
|||
|
|||
/*
|
|||
* Entry point for getnameinfo |
|||
* return 0 if a callback will be made |
|||
* return error code if validation fails |
|||
*/ |
|||
int uv_getnameinfo(uv_loop_t* loop, |
|||
uv_getnameinfo_t* req, |
|||
uv_getnameinfo_cb getnameinfo_cb, |
|||
const struct sockaddr* addr, |
|||
int flags) { |
|||
if (req == NULL || getnameinfo_cb == NULL || addr == NULL) |
|||
return UV_EINVAL; |
|||
|
|||
if (addr->sa_family == AF_INET) { |
|||
memcpy(&req->storage, |
|||
addr, |
|||
sizeof(struct sockaddr_in)); |
|||
} else if (addr->sa_family == AF_INET6) { |
|||
memcpy(&req->storage, |
|||
addr, |
|||
sizeof(struct sockaddr_in6)); |
|||
} else { |
|||
return UV_EINVAL; |
|||
} |
|||
|
|||
uv_req_init(loop, (uv_req_t*)req); |
|||
|
|||
req->getnameinfo_cb = getnameinfo_cb; |
|||
req->flags = flags; |
|||
req->type = UV_GETNAMEINFO; |
|||
req->loop = loop; |
|||
|
|||
/* Ask thread to run. Treat this as a long operation. */ |
|||
if (QueueUserWorkItem(&getnameinfo_thread_proc, |
|||
req, |
|||
WT_EXECUTELONGFUNCTION) == 0) { |
|||
return uv_translate_sys_error(GetLastError()); |
|||
} |
|||
|
|||
uv__req_register(loop, req); |
|||
|
|||
return 0; |
|||
} |
|||
@ -0,0 +1,83 @@ |
|||
/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
|||
* |
|||
* 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 "uv.h" |
|||
#include "task.h" |
|||
#include <stdio.h> |
|||
#include <stdlib.h> |
|||
#include <string.h> |
|||
|
|||
|
|||
static const char* address_ip4 = "127.0.0.1"; |
|||
static const char* address_ip6 = "::1"; |
|||
static const int port = 80; |
|||
|
|||
static struct sockaddr_in addr4; |
|||
static struct sockaddr_in6 addr6; |
|||
static uv_getnameinfo_t req; |
|||
|
|||
static void getnameinfo_req(uv_getnameinfo_t* handle, |
|||
int status, |
|||
const char* hostname, |
|||
const char* service) { |
|||
ASSERT(handle != NULL); |
|||
ASSERT(status == 0); |
|||
ASSERT(hostname != NULL); |
|||
ASSERT(service != NULL); |
|||
} |
|||
|
|||
TEST_IMPL(getnameinfo_basic_ip4) { |
|||
int r; |
|||
|
|||
r = uv_ip4_addr(address_ip4, port, &addr4); |
|||
ASSERT(r == 0); |
|||
|
|||
r = uv_getnameinfo(uv_default_loop(), |
|||
&req, |
|||
&getnameinfo_req, |
|||
(const struct sockaddr*)&addr4, |
|||
0); |
|||
ASSERT(r == 0); |
|||
|
|||
uv_run(uv_default_loop(), UV_RUN_DEFAULT); |
|||
|
|||
MAKE_VALGRIND_HAPPY(); |
|||
return 0; |
|||
} |
|||
|
|||
TEST_IMPL(getnameinfo_basic_ip6) { |
|||
int r; |
|||
|
|||
r = uv_ip6_addr(address_ip6, port, &addr6); |
|||
ASSERT(r == 0); |
|||
|
|||
r = uv_getnameinfo(uv_default_loop(), |
|||
&req, |
|||
&getnameinfo_req, |
|||
(const struct sockaddr*)&addr6, |
|||
0); |
|||
ASSERT(r == 0); |
|||
|
|||
uv_run(uv_default_loop(), UV_RUN_DEFAULT); |
|||
|
|||
MAKE_VALGRIND_HAPPY(); |
|||
return 0; |
|||
} |
|||
@ -0,0 +1,89 @@ |
|||
/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
|||
* |
|||
* 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. |
|||
*/ |
|||
|
|||
#ifdef _WIN32 |
|||
|
|||
#include <errno.h> |
|||
|
|||
#include "uv.h" |
|||
#include "task.h" |
|||
|
|||
uv_os_sock_t sock; |
|||
uv_poll_t handle; |
|||
|
|||
static int close_cb_called = 0; |
|||
|
|||
|
|||
static void close_cb(uv_handle_t* h) { |
|||
close_cb_called++; |
|||
} |
|||
|
|||
|
|||
static void poll_cb(uv_poll_t* h, int status, int events) { |
|||
int r; |
|||
|
|||
ASSERT(status == 0); |
|||
ASSERT(h == &handle); |
|||
|
|||
r = uv_poll_start(&handle, UV_READABLE, poll_cb); |
|||
ASSERT(r == 0); |
|||
|
|||
closesocket(sock); |
|||
uv_close((uv_handle_t*) &handle, close_cb); |
|||
|
|||
} |
|||
|
|||
|
|||
TEST_IMPL(poll_closesocket) { |
|||
struct WSAData wsa_data; |
|||
int r; |
|||
unsigned long on; |
|||
struct sockaddr_in addr; |
|||
|
|||
r = WSAStartup(MAKEWORD(2, 2), &wsa_data); |
|||
ASSERT(r == 0); |
|||
|
|||
sock = socket(AF_INET, SOCK_STREAM, 0); |
|||
ASSERT(sock != INVALID_SOCKET); |
|||
on = 1; |
|||
r = ioctlsocket(sock, FIONBIO, &on); |
|||
ASSERT(r == 0); |
|||
|
|||
r = uv_ip4_addr("127.0.0.1", TEST_PORT, &addr); |
|||
ASSERT(r == 0); |
|||
|
|||
r = connect(sock, (const struct sockaddr*) &addr, sizeof addr); |
|||
ASSERT(r != 0); |
|||
ASSERT(WSAGetLastError() == WSAEWOULDBLOCK); |
|||
|
|||
r = uv_poll_init_socket(uv_default_loop(), &handle, sock); |
|||
ASSERT(r == 0); |
|||
r = uv_poll_start(&handle, UV_WRITABLE, poll_cb); |
|||
ASSERT(r == 0); |
|||
|
|||
uv_run(uv_default_loop(), UV_RUN_DEFAULT); |
|||
|
|||
ASSERT(close_cb_called == 1); |
|||
|
|||
MAKE_VALGRIND_HAPPY(); |
|||
return 0; |
|||
} |
|||
#endif |
|||
Loading…
Reference in new issue