tildefriends/src/Socket.h

#ifndef INCLUDED_Socket
#define INCLUDED_Socket

#include <string>
#include <uv.h>
#include <v8.h>

typedef int promiseid_t;
class Task;
class TlsContext;
class TlsSession;

class Socket {
public:
	static void create(const v8::FunctionCallbackInfo<v8::Value>& args);

	void close();

	static int getCount() { return _count; }
	static int getOpenCount() { return _openCount; }

private:
	Socket(Task* task);
	~Socket();

	Task* _task;
	uv_tcp_t _socket;
	TlsSession* _tls = 0;
	promiseid_t _startTlsPromise = -1;
	promiseid_t _closePromise = -1;
	int _refCount = 1;
	bool _connected = false;
	std::string _peerName;

	enum Direction { kUndetermined, kAccept, kConnect };
	Direction _direction = kUndetermined;

	static int _count;
	static int _openCount;

	static TlsContext* _defaultTlsContext;

	v8::Persistent<v8::Object, v8::CopyablePersistentTraits<v8::Object> > _object;

	v8::Persistent<v8::Function, v8::CopyablePersistentTraits<v8::Function> > _onConnect;
	v8::Persistent<v8::Function, v8::CopyablePersistentTraits<v8::Function> > _onRead;
	v8::Persistent<v8::Function, v8::CopyablePersistentTraits<v8::Function> > _onError;

	static void startTls(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void stopTls(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void bind(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void connect(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void listen(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void accept(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void close(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void shutdown(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void read(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void onError(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void write(const v8::FunctionCallbackInfo<v8::Value>& args);
	static void getPeerName(v8::Local<v8::String> property, const v8::PropertyCallbackInfo<v8::Value>& info);
	static void getPeerCertificate(v8::Local<v8::String> property, const v8::PropertyCallbackInfo<v8::Value>& info);
	static void isConnected(v8::Local<v8::String> property, const v8::PropertyCallbackInfo<v8::Value>& info);

	static Socket* get(v8::Handle<v8::Value> socketObject);
	static void onClose(uv_handle_t* handle);
	static void onShutdown(uv_shutdown_t* request, int status);
	static void onResolvedForBind(uv_getaddrinfo_t* resolver, int status, struct addrinfo* result);
	static void onResolvedForConnect(uv_getaddrinfo_t* resolver, int status, struct addrinfo* result);
	static void onConnect(uv_connect_t* request, int status);
	static void onNewConnection(uv_stream_t* server, int status);

	static void allocateBuffer(uv_handle_t* handle, size_t suggestedSize, uv_buf_t* buffer);
	static void onRead(uv_stream_t* stream, ssize_t readSize, const uv_buf_t* buffer);
	static void onWrite(uv_write_t* request, int status);
	static void onRelease(const v8::WeakCallbackInfo<Socket>& data);

	void processTlsShutdown(promiseid_t promise);
	static void onTlsShutdown(uv_write_t* request, int status);
	void shutdownInternal(promiseid_t promise);

	bool processSomeOutgoingTls(promiseid_t promise, uv_write_cb callback);
	void processOutgoingTls();
	void reportTlsErrors();
	void reportError(const char* error);

	void ref();
	void release();
};

#endif