////////////////////////////////////////////////////////////////////////////////////////// // A multi-platform support c++11 library with focus on asynchronous socket I/O for any // client application. ////////////////////////////////////////////////////////////////////////////////////////// /* The MIT License (MIT) Copyright (c) 2012-2023 HALX99 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. */ #if !defined(LUA_LIB) # define LUA_LIB #endif #include "yasio/yasio.hpp" #include "yasio/bindings/lyasio.hpp" using namespace yasio; namespace lyasio { enum { BUFFER_DEFAULT, BUFFER_RAW, BUFFER_FAST, }; template static void obstream_write_v(_Stream* obs, cxx17::string_view val, int length_field_bits) { // default: Use variant length of length field, just like .net BinaryWriter.Write(String), // see: // https://github.com/mono/mono/blob/master/mcs/class/referencesource/mscorlib/system/io/binarywriter.cs switch (length_field_bits) { case -1: default: return obs->write_v(val); case 32: return obs->write_v32(val); case 16: return obs->write_v16(val); case 8: return obs->write_v8(val); } }; template static cxx17::string_view ibstream_read_v(_Stream* ibs, int length_field_bits) { // default: Use variant length of length field, just like .net BinaryReader.ReadString, // see: // https://github.com/mono/mono/blob/master/mcs/class/referencesource/mscorlib/system/io/binaryreader.cs switch (length_field_bits) { case -1: default: return ibs->read_v(); case 32: return ibs->read_v32(); case 16: return ibs->read_v16(); case 8: return ibs->read_v8(); } }; } // namespace lyasio #if YASIO__HAS_CXX14 # include "yasio/bindings/yasio_sol.hpp" namespace lyasio { template static void register_obstream(sol::table& lib, const char* usertype) { lib.new_usertype<_Stream>( usertype, # if defined(YASIO_OBS_BUILTIN_STACK) "push32", &_Stream::push32, "pop32", sol::overload(static_cast(&_Stream::pop32), static_cast(&_Stream::pop32)), "push16", &_Stream::push16, "pop16", sol::overload(static_cast(&_Stream::pop16), static_cast(&_Stream::pop16)), "push8", &_Stream::push8, "pop8", sol::overload(static_cast(&_Stream::pop8), static_cast(&_Stream::pop8)), # else "push32", &_Stream::template push, "pop32", sol::overload(static_cast(&_Stream::template pop), static_cast(&_Stream::template pop)), "push16", &_Stream::template push, "pop16", sol::overload(static_cast(&_Stream::template pop), static_cast(&_Stream::template pop)), "push8", &_Stream::template push, "pop8", sol::overload(static_cast(&_Stream::template pop), static_cast(&_Stream::template pop)), # endif "write_ix", &_Stream::template write_ix, "write_bool", &_Stream::template write, "write_i8", &_Stream::template write, "write_i16", &_Stream::template write, "write_i32", &_Stream::template write, "write_i64", &_Stream::template write, "write_u8", &_Stream::template write, "write_u16", &_Stream::template write, "write_u32", &_Stream::template write, "write_u64", &_Stream::template write, # if defined(YASIO_ENABLE_HALF_FLOAT) "write_f16", &_Stream::template write, # endif "write_f", &_Stream::template write, "write_lf", &_Stream::template write, "write_v", [](_Stream* obs, cxx17::string_view sv, sol::variadic_args args) { int lfl = -1; if (args.size() > 0) lfl = static_cast(args[0]); return obstream_write_v<_Stream>(obs, sv, lfl); }, "write_bytes", static_cast(&_Stream::write_bytes), "length", &_Stream::length, "to_string", [](_Stream* obs) { return cxx17::string_view(obs->data(), obs->length()); }, "save", &_Stream::save); } template static void register_ibstream(sol::table& lib, const char* usertype) { lib.new_usertype<_Stream>( usertype, sol::constructors<_Stream(), _Stream(yasio::sbyte_buffer), _Stream(const _OStream*)>(), "load", &_Stream::load, "read_ix", &_Stream::template read_ix, "read_bool", &_Stream::template read, "read_i8", &_Stream::template read, "read_i16", &_Stream::template read, "read_i32", &_Stream::template read, "read_i64", &_Stream::template read, "read_u8", &_Stream::template read, "read_u16", &_Stream::template read, "read_u32", &_Stream::template read, "read_u64", &_Stream::template read, # if defined(YASIO_ENABLE_HALF_FLOAT) "read_f16", &_Stream::template read, # endif "read_f", &_Stream::template read, "read_lf", &_Stream::template read, "read_v", [](_Stream* ibs, sol::variadic_args args) { int lfl = -1; if (args.size() > 0) lfl = static_cast(args[0]); return ibstream_read_v<_Stream>(ibs, lfl); }, "read_bytes", static_cast(&_Stream::read_bytes), "seek", &_StreamView::seek, "length", &_StreamView::length, "to_string", [](_Stream* ibs) { return cxx17::string_view(ibs->data(), ibs->length()); }); } } // namespace lyasio extern "C" { YASIO_LUA_API int luaopen_yasio(lua_State* L) { sol::state_view state_view(L); # if !YASIO_LUA_ENABLE_GLOBAL auto yasio_lib = state_view.create_table(); # else auto yasio_lib = state_view.create_named_table("yasio"); # endif yasio_lib.new_usertype( "io_event", "kind", &io_event::kind, "status", &io_event::status, "passive", [](io_event* e) { return !!e->passive(); }, "packet", [](io_event* ev, sol::variadic_args args, sol::this_state s) { sol::state_view L(s); auto& pkt = ev->packet(); if (is_packet_empty(pkt)) return sol::make_object(L, sol::lua_nil); int buffer_type = (args.size() >= 1) ? static_cast(args[0]) : static_cast(lyasio::BUFFER_DEFAULT); switch (buffer_type) { case lyasio::BUFFER_RAW: return sol::make_object(L, cxx17::string_view{packet_data(pkt), packet_len(pkt)}); case lyasio::BUFFER_FAST: return sol::make_object(L, cxx14::make_unique(forward_packet((packet_t &&) pkt))); default: return sol::make_object(L, cxx14::make_unique(forward_packet((packet_t &&) pkt))); } }, "cindex", &io_event::cindex, "transport", &io_event::transport # if !defined(YASIO_MINIFY_EVENT) , "timestamp", &io_event::timestamp # endif ); yasio_lib.new_usertype( "io_service", "new", sol::initializers([](io_service& uninitialized_memory) { return new (&uninitialized_memory) io_service(); }, [](io_service& uninitialized_memory, int n) { return new (&uninitialized_memory) io_service(n); }, [](io_service& uninitialized_memory, sol::table channel_eps) { std::vector hosts; auto host = channel_eps["host"]; if (host.valid()) hosts.push_back(io_hostent(host.get(), channel_eps["port"])); else { for (auto item : channel_eps) { auto ep = item.second.as(); hosts.push_back(io_hostent(ep["host"].get(), ep["port"])); } } return new (&uninitialized_memory) io_service(!hosts.empty() ? &hosts.front() : nullptr, (std::max)(static_cast(hosts.size()), 1)); }), sol::meta_function::garbage_collect, sol::destructor([](io_service& memory_from_lua) { memory_from_lua.~io_service(); }), "start", [](io_service* service, sol::function cb) { service->set_option(YOPT_S_NO_DISPATCH, 1); // script doesn't support handle event at network thread service->start([=](event_ptr ev) { cb(std::move(ev)); }); }, "stop", &io_service::stop, "set_option", [](io_service* service, int opt, sol::variadic_args args) { switch (opt) { case YOPT_C_LOCAL_HOST: case YOPT_C_REMOTE_HOST: service->set_option(opt, static_cast(args[0]), args[1].as()); break; case YOPT_C_UNPACK_STRIP: case YOPT_C_LOCAL_PORT: case YOPT_C_REMOTE_PORT: case YOPT_C_KCP_CONV: case YOPT_C_UNPACK_NO_BSWAP: service->set_option(opt, static_cast(args[0]), static_cast(args[1])); break; case YOPT_C_ENABLE_MCAST: case YOPT_C_LOCAL_ENDPOINT: case YOPT_C_REMOTE_ENDPOINT: service->set_option(opt, static_cast(args[0]), args[1].as(), static_cast(args[2])); break; case YOPT_C_MOD_FLAGS: service->set_option(opt, static_cast(args[0]), static_cast(args[1]), static_cast(args[2])); break; case YOPT_S_TCP_KEEPALIVE: service->set_option(opt, static_cast(args[0]), static_cast(args[1]), static_cast(args[2])); break; case YOPT_C_UNPACK_PARAMS: service->set_option(opt, static_cast(args[0]), static_cast(args[1]), static_cast(args[2]), static_cast(args[3]), static_cast(args[4])); break; case YOPT_S_EVENT_CB: (void)0; { sol::function fn = args[0]; io_event_cb_t fnwrap = [=](event_ptr e) mutable -> void { fn(std::move(e)); }; service->set_option(opt, std::addressof(fnwrap)); } break; default: service->set_option(opt, static_cast(args[0])); } }, "dispatch", &io_service::dispatch, "open", &io_service::open, "is_open", sol::overload(static_cast(&io_service::is_open), static_cast(&io_service::is_open)), "close", sol::overload(static_cast(&io_service::close), static_cast(&io_service::close)), "write", sol::overload( [](io_service* service, transport_handle_t transport, cxx17::string_view s) { return service->write(transport, yasio::sbyte_buffer{s.data(), s.data() + s.length(), std::true_type{}}); }, [](io_service* service, transport_handle_t transport, yasio::obstream* obs) { return service->write(transport, std::move(obs->buffer())); }), "write_to", sol::overload( [](io_service* service, transport_handle_t transport, cxx17::string_view s, cxx17::string_view ip, u_short port) { return service->write_to(transport, yasio::sbyte_buffer{s.data(), s.data() + s.length(), std::true_type{}}, ip::endpoint{ip.data(), port}); }, [](io_service* service, transport_handle_t transport, yasio::obstream* obs, cxx17::string_view ip, u_short port) { return service->write_to(transport, std::move(obs->buffer()), ip::endpoint{ip.data(), port}); }), "native_ptr", [](io_service* service) { return (void*)service; }); // ##-- obstream lyasio::register_obstream(yasio_lib, "obstream"); lyasio::register_obstream(yasio_lib, "fast_obstream"); // ##-- yasio::ibstream lyasio::register_ibstream(yasio_lib, "ibstream"); lyasio::register_ibstream(yasio_lib, "fast_ibstream"); yasio_lib["highp_clock"] = &highp_clock; yasio_lib["highp_time"] = &highp_clock; yasio_lib["unwrap_ptr"] = [](lua_State* L) -> int { auto& pkt = *(packet_t*)lua_touserdata(L, 1); int offset = lua_isinteger(L, 2) ? static_cast(lua_tointeger(L, 2)) : 0; lua_pushlightuserdata(L, packet_data(pkt) + offset); return 1; }; yasio_lib["unwrap_len"] = [](lua_State* L) -> int { auto& pkt = *(packet_t*)lua_touserdata(L, 1); int offset = lua_isinteger(L, 2) ? static_cast(lua_tointeger(L, 2)) : 0; lua_pushinteger(L, packet_len(pkt) - offset); return 1; }; // ##-- yasio enums # define YASIO_EXPORT_ENUM(v) yasio_lib[#v] = v YASIO_EXPORT_ENUM(YCK_TCP_CLIENT); YASIO_EXPORT_ENUM(YCK_TCP_SERVER); YASIO_EXPORT_ENUM(YCK_UDP_CLIENT); YASIO_EXPORT_ENUM(YCK_UDP_SERVER); # if defined(YASIO_ENABLE_KCP) YASIO_EXPORT_ENUM(YCK_KCP_CLIENT); YASIO_EXPORT_ENUM(YCK_KCP_SERVER); # endif # if defined(YASIO_SSL_BACKEND) YASIO_EXPORT_ENUM(YCK_SSL_CLIENT); # endif YASIO_EXPORT_ENUM(YOPT_S_CONNECT_TIMEOUT); YASIO_EXPORT_ENUM(YOPT_S_DNS_CACHE_TIMEOUT); YASIO_EXPORT_ENUM(YOPT_S_DNS_QUERIES_TIMEOUT); YASIO_EXPORT_ENUM(YOPT_S_TCP_KEEPALIVE); YASIO_EXPORT_ENUM(YOPT_S_EVENT_CB); YASIO_EXPORT_ENUM(YOPT_C_UNPACK_PARAMS); YASIO_EXPORT_ENUM(YOPT_C_UNPACK_STRIP); YASIO_EXPORT_ENUM(YOPT_C_UNPACK_NO_BSWAP); YASIO_EXPORT_ENUM(YOPT_C_LFBFD_PARAMS); // alias for YOPT_C_UNPACK_PARAMS YASIO_EXPORT_ENUM(YOPT_C_LOCAL_HOST); YASIO_EXPORT_ENUM(YOPT_C_LOCAL_PORT); YASIO_EXPORT_ENUM(YOPT_C_LOCAL_ENDPOINT); YASIO_EXPORT_ENUM(YOPT_C_REMOTE_HOST); YASIO_EXPORT_ENUM(YOPT_C_REMOTE_PORT); YASIO_EXPORT_ENUM(YOPT_C_REMOTE_ENDPOINT); YASIO_EXPORT_ENUM(YOPT_C_ENABLE_MCAST); YASIO_EXPORT_ENUM(YOPT_C_DISABLE_MCAST); # if defined(YASIO_ENABLE_KCP) YASIO_EXPORT_ENUM(YOPT_C_KCP_CONV); YASIO_EXPORT_ENUM(YOPT_C_KCP_NODELAY); YASIO_EXPORT_ENUM(YOPT_C_KCP_WINDOW_SIZE); YASIO_EXPORT_ENUM(YOPT_C_KCP_MTU); YASIO_EXPORT_ENUM(YOPT_C_KCP_RTO_MIN); # endif YASIO_EXPORT_ENUM(YOPT_C_MOD_FLAGS); YASIO_EXPORT_ENUM(YCF_REUSEADDR); YASIO_EXPORT_ENUM(YCF_EXCLUSIVEADDRUSE); YASIO_EXPORT_ENUM(YEK_ON_OPEN); YASIO_EXPORT_ENUM(YEK_ON_CLOSE); YASIO_EXPORT_ENUM(YEK_ON_PACKET); YASIO_EXPORT_ENUM(YEK_CONNECT_RESPONSE); YASIO_EXPORT_ENUM(YEK_CONNECTION_LOST); YASIO_EXPORT_ENUM(YEK_PACKET); YASIO_EXPORT_ENUM(SEEK_CUR); YASIO_EXPORT_ENUM(SEEK_SET); YASIO_EXPORT_ENUM(SEEK_END); using namespace lyasio; YASIO_EXPORT_ENUM(BUFFER_DEFAULT); YASIO_EXPORT_ENUM(BUFFER_RAW); YASIO_EXPORT_ENUM(BUFFER_FAST); return yasio_lib.push(); /* return 'yasio' table */ } } /* extern "C" */ #else # include "kaguya/kaguya.hpp" /// customize the type conversion from/to lua namespace kaguya { // cxx17::string_view template <> struct lua_type_traits { typedef cxx17::string_view get_type; typedef cxx17::string_view push_type; static bool strictCheckType(lua_State* l, int index) { return lua_type(l, index) == LUA_TSTRING; } static bool checkType(lua_State* l, int index) { return lua_isstring(l, index) != 0; } static get_type get(lua_State* l, int index) { size_t size = 0; const char* buffer = lua_tolstring(l, index, &size); return cxx17::string_view(buffer, size); } static int push(lua_State* l, push_type s) { lua_pushlstring(l, s.data(), s.size()); return 1; } }; // yasio::sbyte_buffer template <> struct lua_type_traits { typedef yasio::sbyte_buffer get_type; typedef const yasio::sbyte_buffer& push_type; static bool strictCheckType(lua_State* l, int index) { return lua_type(l, index) == LUA_TSTRING; } static bool checkType(lua_State* l, int index) { return lua_isstring(l, index) != 0; } static get_type get(lua_State* l, int index) { size_t size = 0; const char* buffer = lua_tolstring(l, index, &size); return yasio::sbyte_buffer(buffer, buffer + size, std::true_type{}); } static int push(lua_State* l, push_type s) { lua_pushlstring(l, s.data(), s.size()); return 1; } }; // transport_handle_t template <> struct lua_type_traits { typedef yasio::inet::transport_handle_t get_type; typedef yasio::inet::transport_handle_t push_type; static bool strictCheckType(lua_State* l, int index) { return lua_type(l, index) == LUA_TLIGHTUSERDATA; } static bool checkType(lua_State* l, int index) { return lua_islightuserdata(l, index) != 0; } static get_type get(lua_State* l, int index) { return reinterpret_cast(lua_touserdata(l, index)); } static int push(lua_State* l, push_type s) { lua_pushlightuserdata(l, s); return 1; } }; template <> struct lua_type_traits> { typedef std::vector get_type; typedef std::vector push_type; static bool strictCheckType(lua_State* l, int index) { return lua_type(l, index) == LUA_TTABLE; } static bool checkType(lua_State* l, int index) { return lua_istable(l, index) != 0; } static get_type get(lua_State* l, int index) { lua_pushvalue(l, index); kaguya::LuaTable channel_eps(l, kaguya::StackTop{}); std::vector hosts; auto host = channel_eps["host"]; if (host) hosts.push_back(yasio::inet::io_hostent(host, channel_eps["port"])); else { channel_eps.foreach_table([&](int, kaguya::LuaTable ep) { hosts.push_back(yasio::inet::io_hostent(ep["host"], ep["port"])); }); } lua_pop(l, 1); return hosts; } static int push(lua_State* l, push_type s) { // don't need push yasio::inet::io_hostent to lua assert(false); return 1; } }; # if defined(YASIO_ENABLE_HALF_FLOAT) template <> struct lua_type_traits { typedef fp16_t get_type; typedef fp16_t push_type; static bool strictCheckType(lua_State* l, int index) { return lua_type(l, index) == LUA_TNUMBER; } static bool checkType(lua_State* l, int index) { return lua_isnumber(l, index) != 0; } static get_type get(lua_State* l, int index) { return fp16_t{static_cast(lua_tonumber(l, index))}; } static int push(lua_State* l, push_type value) { lua_pushnumber(l, static_cast(value)); return 1; } }; # endif }; // namespace kaguya namespace lyasio { # define kaguya_obstream_class(_Stream, _BaseStream) kaguya::UserdataMetatable<_Stream, _BaseStream>().setConstructors<_Stream(), _Stream(size_t)>() # define kaguya_obstream_base_class(_BaseStream) kaguya::UserdataMetatable<_BaseStream>().setConstructors<_BaseStream(_BaseStream::buffer_type*)>() # define kaguya_ibstream_view_class(_StreamView, _OStream) \ kaguya::UserdataMetatable<_StreamView>().setConstructors<_StreamView(), _StreamView(const void*, size_t), _StreamView(const _OStream*)>() # define kaguya_ibstream_class(_Stream, _StreamView, _OStream) \ kaguya::UserdataMetatable<_Stream, _StreamView>().setConstructors<_Stream(yasio::sbyte_buffer), _Stream(const _OStream*)>() template static void register_obstream(kaguya::LuaTable& lib, const char* usertype, const char* basetype, kaguya::UserdataMetatable<_Stream, _BaseStream>& userclass, kaguya::UserdataMetatable<_BaseStream>& baseclass) { lib[basetype].setClass( # if defined(YASIO_OBS_BUILTIN_STACK) userclass.addFunction("push32", &_BaseStream::push32) .addOverloadedFunctions("pop32", static_cast(&_BaseStream::pop32), static_cast(&_Stream::pop32)) .addFunction("push16", &_BaseStream::push16) .addOverloadedFunctions("pop16", static_cast(&_BaseStream::pop16), static_cast(&_BaseStream::pop16)) .addFunction("push8", &_BaseStream::push8) .addOverloadedFunctions("pop8", static_cast(&_BaseStream::pop8), static_cast(&_BaseStream::pop8)) # else baseclass.addFunction("push32", &_BaseStream::template push) .addOverloadedFunctions("pop32", static_cast(&_BaseStream::template pop), static_cast(&_BaseStream::template pop)) .addFunction("push16", &_BaseStream::template push) .addOverloadedFunctions("pop16", static_cast(&_BaseStream::template pop), static_cast(&_BaseStream::template pop)) .addFunction("push8", &_BaseStream::template push) .addOverloadedFunctions("pop8", static_cast(&_BaseStream::template pop), static_cast(&_BaseStream::template pop)) # endif .addFunction("write_ix", &_BaseStream::template write_ix) .addFunction("write_bool", &_BaseStream::template write) .addFunction("write_i8", &_BaseStream::template write) .addFunction("write_i16", &_BaseStream::template write) .addFunction("write_i32", &_BaseStream::template write) .addFunction("write_i64", &_BaseStream::template write) .addFunction("write_u8", &_BaseStream::template write) .addFunction("write_u16", &_BaseStream::template write) .addFunction("write_u32", &_BaseStream::template write) .addFunction("write_u64", &_BaseStream::template write) # if defined(YASIO_ENABLE_HALF_FLOAT) .addFunction("write_f16", &_BaseStream::template write) # endif .addFunction("write_f", &_BaseStream::template write) .addFunction("write_lf", &_BaseStream::template write) .addStaticFunction("write_v", [](_BaseStream* obs, cxx17::string_view sv, kaguya::VariadicArgType args) { int lfl = -1; if (args.size() > 0) lfl = static_cast(args[0]); return lyasio::obstream_write_v(obs, sv, lfl); }) .addFunction("write_bytes", static_cast(&_BaseStream::write_bytes)) .addFunction("length", &_BaseStream::length) .addStaticFunction("to_string", [](_BaseStream* obs) { return cxx17::string_view(obs->data(), obs->length()); })); // ##-- obstream lib[usertype].setClass(userclass); } template static void register_ibstream(kaguya::LuaTable& lib, const char* usertype, const char* basetype, kaguya::UserdataMetatable<_Stream, _StreamView>& userclass, kaguya::UserdataMetatable<_StreamView>& baseclass) { lib[basetype].setClass(baseclass.addFunction("read_ix", &_StreamView::template read_ix) .addFunction("read_bool", &_StreamView::template read) .addFunction("read_i8", &_StreamView::template read) .addFunction("read_i16", &_StreamView::template read) .addFunction("read_i32", &_StreamView::template read) .addFunction("read_i64", &_StreamView::template read) .addFunction("read_u8", &_StreamView::template read) .addFunction("read_u16", &_StreamView::template read) .addFunction("read_u32", &_StreamView::template read) .addFunction("read_u64", &_StreamView::template read) # if defined(YASIO_ENABLE_HALF_FLOAT) .addFunction("read_f16", &_Stream::template read) # endif .addFunction("read_f", &_StreamView::template read) .addFunction("read_lf", &_StreamView::template read) .addStaticFunction("read_v", [](_StreamView* ibs, kaguya::VariadicArgType args) { int length_field_bits = -1; if (args.size() > 0) length_field_bits = static_cast(args[0]); return lyasio::ibstream_read_v(ibs, length_field_bits); }) .addFunction("read_bytes", static_cast(&_StreamView::read_bytes)) .addFunction("seek", &_StreamView::seek) .addFunction("length", &_StreamView::length) .addStaticFunction("to_string", [](_StreamView* ibs) { return cxx17::string_view(ibs->data(), ibs->length()); })); // ##-- ibstream lib[usertype].setClass(userclass); } } // namespace lyasio extern "C" { YASIO_LUA_API int luaopen_yasio(lua_State* L) { kaguya::State state(L); auto yasio_lib = state.newTable(); # if YASIO_LUA_ENABLE_GLOBAL state["yasio"] = yasio_lib; # endif yasio_lib["io_event"].setClass(kaguya::UserdataMetatable() .addFunction("kind", &io_event::kind) .addFunction("status", &io_event::status) .addStaticFunction("passive", [](io_event* ev) { return !!ev->passive(); }) .addStaticFunction("default_packet", [](io_event* ev) { auto& pkt = ev->packet(); if (is_packet_empty(pkt)) return std::unique_ptr{}; return cxx14::make_unique(forward_packet((packet_t &&) pkt)); }) .addStaticFunction("raw_packet", [](io_event* ev) { auto& pkt = ev->packet(); if (is_packet_empty(pkt)) return cxx17::string_view{""}; return cxx17::string_view{packet_data(pkt), packet_len(pkt)}; }) .addStaticFunction("fast_packet", [](io_event* ev) { auto& pkt = ev->packet(); if (is_packet_empty(pkt)) return std::unique_ptr{}; return cxx14::make_unique(forward_packet((packet_t &&) pkt)); }) .addFunction("cindex", &io_event::cindex) .addFunction("transport", &io_event::transport) .addFunction("timestamp", &io_event::timestamp)); # if !YASIO_LUA_ENABLE_GLOBAL state["yasio"] = yasio_lib; # endif bool succeed = state.dostring(R"( local yasio = yasio; yasio.io_event.packet = function(self, buffer_type) if buffer_type == yasio.BUFFER_RAW then return yasio.io_event.raw_packet(self) elseif buffer_type == yasio.BUFFER_FAST then return yasio.io_event.fast_packet(self) else return yasio.io_event.default_packet(self) end end )"); assert(succeed); # if !YASIO_LUA_ENABLE_GLOBAL state["yasio"] = kaguya::NilValue{}; # endif yasio_lib["io_service"].setClass( kaguya::UserdataMetatable() .setConstructors&)>() .addStaticFunction("start", [](io_service* service, kaguya::LuaFunction cb) { io_event_cb_t fnwrap = [=](event_ptr e) mutable -> void { cb(e.get()); }; service->set_option(YOPT_S_NO_DISPATCH, 1); // script doesn't support handle event at network thread service->start(std::move(fnwrap)); }) .addFunction("stop", &io_service::stop) .addFunction("dispatch", &io_service::dispatch) .addFunction("open", &io_service::open) .addOverloadedFunctions("is_open", static_cast(&io_service::is_open), static_cast(&io_service::is_open)) .addOverloadedFunctions("close", static_cast(&io_service::close), static_cast(&io_service::close)) .addOverloadedFunctions( "write", [](io_service* service, transport_handle_t transport, cxx17::string_view s) { return service->write(transport, yasio::sbyte_buffer(s.data(), s.data() + s.length())); }, [](io_service* service, transport_handle_t transport, yasio::obstream* obs) { return service->write(transport, std::move(obs->buffer())); }) .addOverloadedFunctions( "write_to", [](io_service* service, transport_handle_t transport, cxx17::string_view s, cxx17::string_view ip, u_short port) { return service->write_to(transport, yasio::sbyte_buffer(s.data(), s.data() + s.length()), ip::endpoint{ip.data(), port}); }, [](io_service* service, transport_handle_t transport, yasio::obstream* obs, cxx17::string_view ip, u_short port) { return service->write_to(transport, std::move(obs->buffer()), ip::endpoint{ip.data(), port}); }) .addStaticFunction("set_option", [](io_service* service, int opt, kaguya::VariadicArgType args) { switch (opt) { case YOPT_C_LOCAL_HOST: case YOPT_C_REMOTE_HOST: service->set_option(opt, static_cast(args[0]), static_cast(args[1])); break; case YOPT_C_UNPACK_STRIP: case YOPT_C_LOCAL_PORT: case YOPT_C_REMOTE_PORT: case YOPT_C_KCP_CONV: case YOPT_C_UNPACK_NO_BSWAP: service->set_option(opt, static_cast(args[0]), static_cast(args[1])); break; case YOPT_C_ENABLE_MCAST: case YOPT_C_LOCAL_ENDPOINT: case YOPT_C_REMOTE_ENDPOINT: service->set_option(opt, static_cast(args[0]), static_cast(args[1]), static_cast(args[2])); break; case YOPT_C_MOD_FLAGS: service->set_option(opt, static_cast(args[0]), static_cast(args[1]), static_cast(args[2])); break; case YOPT_S_TCP_KEEPALIVE: service->set_option(opt, static_cast(args[0]), static_cast(args[1]), static_cast(args[2])); break; case YOPT_C_UNPACK_PARAMS: service->set_option(opt, static_cast(args[0]), static_cast(args[1]), static_cast(args[2]), static_cast(args[3]), static_cast(args[4])); break; case YOPT_S_EVENT_CB: (void)0; { kaguya::LuaFunction fn = args[0]; io_event_cb_t fnwrap = [=](event_ptr e) mutable -> void { fn(e.get()); }; service->set_option(opt, std::addressof(fnwrap)); } break; default: service->set_option(opt, static_cast(args[0])); } }) .addStaticFunction("native_ptr", [](io_service* service) { return (void*)service; })); // ##-- obstream lyasio::register_obstream(yasio_lib, "obstream", "obstream_base", kaguya_obstream_class(obstream, obstream::super_type), kaguya_obstream_base_class(obstream::super_type)); lyasio::register_obstream(yasio_lib, "fast_obstream", "fast_obstream_base", kaguya_obstream_class(fast_obstream, fast_obstream::super_type), kaguya_obstream_base_class(fast_obstream::super_type)); // ##-- yasio::ibstream lyasio::register_ibstream(yasio_lib, "ibstream", "ibstream_view", kaguya_ibstream_class(ibstream, ibstream_view, obstream), kaguya_ibstream_view_class(ibstream_view, obstream)); lyasio::register_ibstream(yasio_lib, "fast_ibstream", "fast_ibstream_view", kaguya_ibstream_class(fast_ibstream, fast_ibstream_view, fast_obstream), kaguya_ibstream_view_class(fast_ibstream_view, fast_obstream)); yasio_lib.setField("highp_clock", &highp_clock); yasio_lib.setField("highp_time", &highp_clock); yasio_lib.setField("unwrap_ptr", [](lua_State* L) -> int { auto& pkt = *(packet_t*)lua_touserdata(L, 1); int offset = lua_isinteger(L, 2) ? static_cast(lua_tointeger(L, 2)) : 0; lua_pushlightuserdata(L, packet_data(pkt) + offset); return 1; }); yasio_lib.setField("unwrap_len", [](lua_State* L) -> int { auto& pkt = *(packet_t*)lua_touserdata(L, 1); int offset = lua_isinteger(L, 2) ? static_cast(lua_tointeger(L, 2)) : 0; lua_pushinteger(L, packet_len(pkt) - offset); return 1; }); // ##-- yasio enums # define YASIO_EXPORT_ENUM(v) yasio_lib[#v] = v YASIO_EXPORT_ENUM(YCK_TCP_CLIENT); YASIO_EXPORT_ENUM(YCK_TCP_SERVER); YASIO_EXPORT_ENUM(YCK_UDP_CLIENT); YASIO_EXPORT_ENUM(YCK_UDP_SERVER); # if defined(YASIO_ENABLE_KCP) YASIO_EXPORT_ENUM(YCK_KCP_CLIENT); YASIO_EXPORT_ENUM(YCK_KCP_SERVER); # endif # if YASIO_SSL_BACKEND YASIO_EXPORT_ENUM(YCK_SSL_CLIENT); # endif YASIO_EXPORT_ENUM(YOPT_S_CONNECT_TIMEOUT); YASIO_EXPORT_ENUM(YOPT_S_DNS_CACHE_TIMEOUT); YASIO_EXPORT_ENUM(YOPT_S_DNS_QUERIES_TIMEOUT); YASIO_EXPORT_ENUM(YOPT_S_TCP_KEEPALIVE); YASIO_EXPORT_ENUM(YOPT_S_EVENT_CB); YASIO_EXPORT_ENUM(YOPT_C_UNPACK_PARAMS); YASIO_EXPORT_ENUM(YOPT_C_UNPACK_STRIP); YASIO_EXPORT_ENUM(YOPT_C_UNPACK_NO_BSWAP); YASIO_EXPORT_ENUM(YOPT_C_LFBFD_PARAMS); // alias for YOPT_C_UNPACK_PARAMS YASIO_EXPORT_ENUM(YOPT_C_LOCAL_HOST); YASIO_EXPORT_ENUM(YOPT_C_LOCAL_PORT); YASIO_EXPORT_ENUM(YOPT_C_LOCAL_ENDPOINT); YASIO_EXPORT_ENUM(YOPT_C_REMOTE_HOST); YASIO_EXPORT_ENUM(YOPT_C_REMOTE_PORT); YASIO_EXPORT_ENUM(YOPT_C_REMOTE_ENDPOINT); YASIO_EXPORT_ENUM(YOPT_C_ENABLE_MCAST); YASIO_EXPORT_ENUM(YOPT_C_DISABLE_MCAST); YASIO_EXPORT_ENUM(YOPT_C_KCP_CONV); YASIO_EXPORT_ENUM(YOPT_C_MOD_FLAGS); YASIO_EXPORT_ENUM(YCF_REUSEADDR); YASIO_EXPORT_ENUM(YCF_EXCLUSIVEADDRUSE); YASIO_EXPORT_ENUM(YEK_ON_OPEN); YASIO_EXPORT_ENUM(YEK_ON_CLOSE); YASIO_EXPORT_ENUM(YEK_ON_PACKET); YASIO_EXPORT_ENUM(YEK_CONNECT_RESPONSE); YASIO_EXPORT_ENUM(YEK_CONNECTION_LOST); YASIO_EXPORT_ENUM(YEK_PACKET); YASIO_EXPORT_ENUM(SEEK_CUR); YASIO_EXPORT_ENUM(SEEK_SET); YASIO_EXPORT_ENUM(SEEK_END); using namespace lyasio; YASIO_EXPORT_ENUM(BUFFER_DEFAULT); YASIO_EXPORT_ENUM(BUFFER_RAW); YASIO_EXPORT_ENUM(BUFFER_FAST); return yasio_lib.push(); /* return 'yasio' table */ } } /* extern "C" */ #endif /* YASIO__HAS_CXX17 */ extern "C" { YASIO_LUA_API void luaregister_yasio(lua_State* L) { lua_getglobal(L, "package"); lua_getfield(L, -1, "preload"); lua_pushcfunction(L, luaopen_yasio); lua_setfield(L, -2, "yasio"); lua_pop(L, 2); } }