1 // Copyright Joyent, Inc. and other Node contributors.
2 //
3 // Permission is hereby granted, free of charge, to any person obtaining a
4 // copy of this software and associated documentation files (the
5 // "Software"), to deal in the Software without restriction, including
6 // without limitation the rights to use, copy, modify, merge, publish,
7 // distribute, sublicense, and/or sell copies of the Software, and to permit
8 // persons to whom the Software is furnished to do so, subject to the
9 // following conditions:
10 //
11 // The above copyright notice and this permission notice shall be included
12 // in all copies or substantial portions of the Software.
13 //
14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
17 // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
18 // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
19 // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
20 // USE OR OTHER DEALINGS IN THE SOFTWARE.
21 
22 #ifndef SRC_TCP_WRAP_H_
23 #define SRC_TCP_WRAP_H_
24 
25 #if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
26 
27 #include "async_wrap.h"
28 #include "env.h"
29 #include "connection_wrap.h"
30 
31 namespace node {
32 
33 class TCPWrap : public ConnectionWrap<TCPWrap, uv_tcp_t> {
34  public:
35   enum SocketType {
36     SOCKET,
37     SERVER
38   };
39 
40   static v8::Local<v8::Object> Instantiate(Environment* env,
41                                            AsyncWrap* parent,
42                                            SocketType type);
43   static void Initialize(v8::Local<v8::Object> target,
44                          v8::Local<v8::Value> unused,
45                          v8::Local<v8::Context> context);
46 
47   SET_NO_MEMORY_INFO()
SET_SELF_SIZE(TCPWrap)48   SET_SELF_SIZE(TCPWrap)
49   std::string MemoryInfoName() const override {
50     switch (provider_type()) {
51       case ProviderType::PROVIDER_TCPWRAP:
52         return "TCPSocketWrap";
53       case ProviderType::PROVIDER_TCPSERVERWRAP:
54         return "TCPServerWrap";
55       default:
56         UNREACHABLE();
57     }
58   }
59 
60  private:
61   typedef uv_tcp_t HandleType;
62 
63   template <typename T,
64             int (*F)(const typename T::HandleType*, sockaddr*, int*)>
65   friend void GetSockOrPeerName(const v8::FunctionCallbackInfo<v8::Value>&);
66 
67   TCPWrap(Environment* env, v8::Local<v8::Object> object,
68           ProviderType provider);
69 
70   static void New(const v8::FunctionCallbackInfo<v8::Value>& args);
71   static void SetNoDelay(const v8::FunctionCallbackInfo<v8::Value>& args);
72   static void SetKeepAlive(const v8::FunctionCallbackInfo<v8::Value>& args);
73   static void Bind(const v8::FunctionCallbackInfo<v8::Value>& args);
74   static void Bind6(const v8::FunctionCallbackInfo<v8::Value>& args);
75   static void Listen(const v8::FunctionCallbackInfo<v8::Value>& args);
76   static void Connect(const v8::FunctionCallbackInfo<v8::Value>& args);
77   static void Connect6(const v8::FunctionCallbackInfo<v8::Value>& args);
78   template <typename T>
79   static void Connect(const v8::FunctionCallbackInfo<v8::Value>& args,
80       std::function<int(const char* ip_address, T* addr)> uv_ip_addr);
81   static void Open(const v8::FunctionCallbackInfo<v8::Value>& args);
82   template <typename T>
83   static void Bind(
84       const v8::FunctionCallbackInfo<v8::Value>& args,
85       int family,
86       std::function<int(const char* ip_address, int port, T* addr)> uv_ip_addr);
87 
88 #ifdef _WIN32
89   static void SetSimultaneousAccepts(
90       const v8::FunctionCallbackInfo<v8::Value>& args);
91 #endif
92 };
93 
94 
95 }  // namespace node
96 
97 #endif  // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
98 
99 #endif  // SRC_TCP_WRAP_H_
100