diff options
Diffstat (limited to 'third_party/mcpc/src')
33 files changed, 8080 insertions, 0 deletions
diff --git a/third_party/mcpc/src/.indent.pro b/third_party/mcpc/src/.indent.pro new file mode 100644 index 0000000..f7bb493 --- /dev/null +++ b/third_party/mcpc/src/.indent.pro @@ -0,0 +1 @@ +../.indent.pro
\ No newline at end of file diff --git a/third_party/mcpc/src/CMakeLists.txt b/third_party/mcpc/src/CMakeLists.txt new file mode 100644 index 0000000..6933543 --- /dev/null +++ b/third_party/mcpc/src/CMakeLists.txt @@ -0,0 +1,32 @@ +set (libmcpc_SOURCES) + +# add_compile_options(-std=c23) +# add_compile_options(-Dis_unix) + +add_compile_options (/Dis_win) +add_compile_options (/Dis_wincl) +add_compile_options (/nologo) +add_compile_options (/std:c17) +add_compile_options (/DMCPC_PTCH_C23KW1) +add_compile_options (/DMCPC_PTCH_C23CKD1) +add_compile_options (/DMCPC_PTCH_C23UCHAR1) + + +list (APPEND libmcpc_SOURCES alloc.c) +list (APPEND libmcpc_SOURCES log.c) +list (APPEND libmcpc_SOURCES errcode.c) +list (APPEND libmcpc_SOURCES tool.c) +list (APPEND libmcpc_SOURCES rsc.c) +list (APPEND libmcpc_SOURCES server.c) +list (APPEND libmcpc_SOURCES retbuf.c) +list (APPEND libmcpc_SOURCES ucbr.c) +list (APPEND libmcpc_SOURCES serlz.c) +list (APPEND libmcpc_SOURCES mjson.c) + +add_library (mcpc STATIC ${libmcpc_SOURCES}) + +target_include_directories ( + mcpc + PRIVATE ${PROJECT_SOURCE_DIR} +) +target_link_libraries (mcpc ws2_32) diff --git a/third_party/mcpc/src/Makefile b/third_party/mcpc/src/Makefile new file mode 100644 index 0000000..b9c5c4c --- /dev/null +++ b/third_party/mcpc/src/Makefile @@ -0,0 +1,94 @@ +include ../mk/step0.mk + +ifneq (,$(is_linux)) + include ../mk/ccdefs-linux.mk +endif +ifneq (,$(is_cygwin)) + include ../mk/ccdefs-cygwin.mk +endif +ifneq (,$(is_mingw)) + include ../mk/ccdefs-mingw.mk +endif +ifneq (,$(is_macos)) + include ../mk/ccdefs-macos.mk +endif +ifneq (,$(is_win)) + include ../mk/ccdefs-win.mk +endif + +# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # + +include ../mk/dbgrel.mk + +# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # + +default: link + +clean: + $(if $(is_cmd) , del *.pdb *.ilk *.o *.out *.a *.so *.dSYM *.obj *.exe *.lib *.dll *.exp ) + $(if $(is_sh) , rm -rf *.pdb *.ilk *.o *.out *.a *.so *.dSYM *.obj *.exe *.lib *.dll *.exp ) + +comp: libmcpc_COMP + +link: libmcpc_LINKS +link: libmcpc_LINKD + +# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # +# libmcpc + +libmcpc_srcs = +libmcpc_srcs += alloc.c +libmcpc_srcs += log.c +libmcpc_srcs += errcode.c +libmcpc_srcs += anydata.c +libmcpc_srcs += tool.c +libmcpc_srcs += rsc.c +libmcpc_srcs += prmpt.c +libmcpc_srcs += server.c +libmcpc_srcs += retbuf.c +libmcpc_srcs += ucbr.c +libmcpc_srcs += complt.c +libmcpc_srcs += serlz.c +libmcpc_srcs += mjson.c + +libmcpc_objs = +ifneq (,$(is_gcclike)) + libmcpc_objs = $(patsubst %.c,%.o,$(libmcpc_srcs)) +endif +ifneq (,$(is_cllike)) + libmcpc_objs = $(patsubst %.c,%.obj,$(libmcpc_srcs)) +endif + +libmcpc_cflags = +ifeq (,$(is_win)) + libmcpc_cflags += -fPIC +endif +libmcpc_clcflags = + +libmcpc_ldflags = +ifneq (,$(is_unixar)) + libmcpc_ldflags += ../src/libmcpc.a +endif +ifneq (,$(is_winlib)) + libmcpc_ldflags += ws2_32.lib +endif +ifneq (,$(is_winld)) + libmcpc_ldflags += -lws2_32 +endif + +libmcpc_arflags = +libmcpc_arflags += rcs + +libmcpc_COMP: + $(if $(is_gcclike) , $(CC) $(CFLAGS) $(libmcpc_cflags) $(libmcpc_srcs) -c ) + $(if $(is_cllike) , $(CC) $(CFLAGS) $(libmcpc_clcflags) $(libmcpc_srcs) /c ) + +libmcpc_LINKS: libmcpc_COMP + $(if $(is_gcclike) , $(AR) $(libmcpc_arflags) libmcpc.a $(libmcpc_objs) ) + $(if $(is_cllike) , lib /nologo $(libmcpc_objs) /out:mcpc.lib $(libmcpc_ldflags) ) + +libmcpc_LINKD: libmcpc_COMP + $(if $(is_gcclike) , $(CC) $(LDFLAGS) -o libmcpc.so $(libmcpc_objs) -shared $(libmcpc_ldflags) ) + $(if $(is_cllike) , cl /nologo $(libmcpc_objs) /link /dll /out:mcpc.dll $(LDFLAGS) $(libmcpc_ldflags) ) + + diff --git a/third_party/mcpc/src/alloc.c b/third_party/mcpc/src/alloc.c new file mode 100644 index 0000000..4771274 --- /dev/null +++ b/third_party/mcpc/src/alloc.c @@ -0,0 +1,57 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "alloc.h" +#include <string.h> + +// #define MCPC_CALLOC 1 + +void * +mcpc_alloc (const size_t nbytes) +{ + void *p = +#ifdef MCPC_CALLOC + calloc (nbytes, 1) +#else + malloc (nbytes) +#endif + ; + return p; +} + +void * +mcpc_realloc (void *ptr, const size_t size) +{ + void *ret = realloc (ptr, size); + +#ifdef MCPC_CALLOC + memset (ret, 0, size); +#endif + + return ret; +} + +void * +mcpc_calloc (const size_t nbytes) +{ + void *p = calloc (nbytes, 1); + return p; +} diff --git a/third_party/mcpc/src/alloc.h b/third_party/mcpc/src/alloc.h new file mode 100644 index 0000000..1742350 --- /dev/null +++ b/third_party/mcpc/src/alloc.h @@ -0,0 +1,39 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h__alloc +#define h__alloc + +#if is_win || is_mingw +#include <malloc.h> +#else +#include <alloca.h> +#endif + +#include <stdlib.h> + +void *mcpc_alloc (const size_t nbytes); + +void *mcpc_realloc (void* ptr, const size_t size); + +void *mcpc_calloc (const size_t size); + +#endif diff --git a/third_party/mcpc/src/anydata.c b/third_party/mcpc/src/anydata.c new file mode 100644 index 0000000..0c663cd --- /dev/null +++ b/third_party/mcpc/src/anydata.c @@ -0,0 +1,817 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "anydata.h" + +#include "ext_string.h" + +#include <mcpc/errcode.h> +#include "alloc.h" +#include "errcode.h" + +const mcpc_anydata_t *mcpc_anypool_getbymeta1 (const mcpc_anypool_t * anypool, const char8_t * meta1, size_t meta1_len); +const mcpc_anydata_t *mcpc_anypool_getbymeta1nt (const mcpc_anypool_t * anypool, const char8_t * meta1nt); + +constexpr static size_t mcpc_anypool_gamt = 8; + +/* = = = = = = = = = = = = = = = = anydata = = = = = = = = = = = = = = = = */ + +mcpc_anydata_t * +mcpc_anydata_new () +{ + mcpc_anydata_t *hp_data = mcpc_calloc (sizeof (mcpc_anydata_t)); + return hp_data; +} + +mcpc_anydata_t * +mcpc_anydata_new_typ (mcpc_anytype_t typ) +{ + mcpc_anydata_t *hp_data = mcpc_calloc (sizeof (mcpc_anydata_t)); + hp_data->data_typ = typ; + return hp_data; +} + +mcpc_anydata_t * +mcpc_anydata_new_meta1_meta2 (const char8_t *meta1, const size_t meta1_len, const char8_t *meta2, const size_t meta2_len) +{ + mcpc_assert (meta1_len <= (size_t) UINT8_MAX, MCPC_EC_ANYDATA_LONGMETA1); + mcpc_assert (meta2_len <= (size_t) UINT16_MAX, MCPC_EC_ANYDATA_LONGMETA2); + + mcpc_anydata_t *hp_data = mcpc_calloc (sizeof (mcpc_anydata_t)); + + void *hp_meta1 = mcpc_alloc (meta1_len); + memcpy (hp_meta1, meta1, meta1_len); + hp_data->meta1 = hp_meta1; + hp_data->meta1_len = (u8_t) meta1_len; + + if (meta2_len > 0) + { + void *hp_meta2 = mcpc_alloc (meta2_len); + memcpy (hp_meta2, meta2, meta2_len); + hp_data->meta2 = hp_meta2; + hp_data->meta2_len = (u8_t) meta2_len; + } + + return hp_data; +} + +mcpc_anydata_t * +mcpc_anydata_new_meta1nt_meta2nt (const char8_t *meta1nt, const char8_t *meta2nt) +{ + return mcpc_anydata_new_meta1_meta2 (meta1nt, u8strlen (meta1nt), meta2nt, u8strlen (meta2nt)); +} + +mcpc_anydata_t * +mcpc_anydata_new_typ_meta1 (mcpc_anytype_t typ, const char8_t *meta1, const size_t meta1_len) +{ + mcpc_anydata_t *hp_data = mcpc_anydata_new_meta1_meta2 (meta1, meta1_len, nullptr, 0); + hp_data->data_typ = typ; + return hp_data; +} + +mcpc_anydata_t * +mcpc_anydata_new_typ_meta1_meta2 (mcpc_anytype_t typ, const char8_t *meta1, const size_t meta1_len, const char8_t *meta2, + const size_t meta2_len) +{ + mcpc_anydata_t *hp_data = mcpc_anydata_new_meta1_meta2 (meta1, meta1_len, meta2, meta2_len); + hp_data->data_typ = typ; + return hp_data; +} + +mcpc_anydata_t * +mcpc_anydata_new_typ_meta1nt_meta2nt (mcpc_anytype_t typ, const char8_t *meta1nt, const char8_t *meta2nt) +{ + return mcpc_anydata_new_typ_meta1_meta2 (typ, meta1nt, u8strlen (meta1nt), meta2nt, u8strlen (meta2nt)); +} + +void +mcpc_anydata_mbfree (mcpc_anydata_t *data) +{ + bug_if (data == nullptr); + + if (data->meta1 != nullptr) + { + free ((void *) data->meta1); + data->meta1_len = 0; + } + + if (data->meta2 != nullptr) + { + free ((void *) data->meta2); + data->meta2_len = 0; + } + + if (data->chd_pool != nullptr) + { + mcpc_anypool_free (data->chd_pool); + } + + if (mcpc_anytype_is_arrlike (data->data_typ)) + { + free ((void *) data->data_arrlk); + data->data_len = 0; + } +} + +void +mcpc_anydata_free (mcpc_anydata_t *data) +{ + mcpc_anydata_mbfree (data); + free (data); +} + +void +mcpc_anydata_copy (mcpc_anydata_t *dest_data, const mcpc_anydata_t *src_data) +{ + bug_if (dest_data == nullptr); + bug_if (src_data == nullptr); + + // make sure inited + memset (dest_data, 0, sizeof (mcpc_anydata_t)); + + if (src_data->meta1_len > 0 && src_data->meta1 != nullptr) + { + void *hp_meta1 = mcpc_alloc (src_data->meta1_len); + memcpy (hp_meta1, src_data->meta1, src_data->meta1_len); + dest_data->meta1 = hp_meta1; + dest_data->meta1_len = src_data->meta1_len; + } + + if (src_data->meta2_len > 0 && src_data->meta2 != nullptr) + { + void *hp_meta2 = mcpc_alloc (src_data->meta2_len); + memcpy (hp_meta2, src_data->meta2, src_data->meta2_len); + dest_data->meta2 = hp_meta2; + dest_data->meta2_len = src_data->meta2_len; + } + + if (false); + else if (mcpc_anytype_is_arrlike (src_data->data_typ) && src_data->data_len > 0) + { + dest_data->data_arrlk = mcpc_alloc (src_data->data_len); + memcpy (dest_data->data_arrlk, src_data->data_arrlk, src_data->data_len); + } + else if (mcpc_anytype_is_intlike (src_data->data_typ)) + dest_data->data_intlk = src_data->data_intlk; + + if (src_data->chd_pool != nullptr) + { + dest_data->chd_pool = mcpc_anypool_new (); + mcpc_anypool_copy (dest_data->chd_pool, src_data->chd_pool); + } + + dest_data->data_typ = src_data->data_typ; + dest_data->data_len = src_data->data_len; + dest_data->flags = src_data->flags; + dest_data->data_mime = src_data->data_mime; + +} + +void +mcpc_anydata_add_child (mcpc_anydata_t *data, mcpc_anydata_t *child) +{ + bug_if_nullptr (data); + bug_if_nullptr ((void *) child); + + if (data->chd_pool == nullptr) + { + data->chd_pool = mcpc_anypool_new (); + } + mcpc_anypool_addfre (data->chd_pool, child); +} + +void +mcpc_anydata_addcpy_child (mcpc_anydata_t *data, const mcpc_anydata_t *child) +{ + bug_if_nullptr (data); + bug_if_nullptr ((void *) child); + + if (data->chd_pool == nullptr) + { + data->chd_pool = mcpc_anypool_new (); + } + mcpc_anypool_addcpy (data->chd_pool, child); +} + +const mcpc_anydata_t * +mcpc_anydata_get_child (const mcpc_anydata_t *anydata, size_t idx) +{ + bug_if (anydata == nullptr); + return mcpc_anypool_get (anydata->chd_pool, idx); +} + +mcpc_errcode_t +mcpc_anydata_set_meta1 (mcpc_anydata_t *anydata, const char8_t *meta1, size_t meta1_len) +{ + bug_if_nullptr (anydata); + bug_if_nullptr ((void *) meta1); + + if (meta1_len == 0) + { + free ((void *) anydata->meta1); + anydata->meta1 = nullptr; + anydata->meta1_len = 0; + return MCPC_EC_0; + } + + if (meta1_len > (size_t) UINT8_MAX) + { + return MCPC_EC_ANYDATA_LONGMETA1; + } + + anydata->meta1 = mcpc_realloc ((void *) anydata->meta1, meta1_len); + memcpy ((void *) anydata->meta1, meta1, meta1_len); + anydata->meta1_len = (u8_t) meta1_len; + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_anydata_set_meta1nt (mcpc_anydata_t *anydata, const char8_t *meta1) +{ + size_t meta1_len = u8strlen (meta1); + return mcpc_anydata_set_meta1 (anydata, meta1, meta1_len); +} + +mcpc_errcode_t +mcpc_anydata_release_data (mcpc_anydata_t *anydata) +{ + if (!mcpc_anytype_is_arrlike (anydata->data_typ)) + return MCPC_EC_0; + + if (anydata->data_len == 0) + return MCPC_EC_0; + + free (anydata->data_arrlk); + anydata->data_len = 0; + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_anydata_set_intlk (mcpc_anydata_t *anydata, mcpc_anytype_t typ, const void *val) +{ + bug_if_nullptr (anydata); + + if (mcpc_anytype_is_arrlike (anydata->data_typ)) + mcpc_anydata_release_data (anydata); + + if (false); + else if (typ == MCPC_I8 || typ == MCPC_U8) + anydata->data_intlk = (uintmax_t) (*((u8_t *) val)); + else if (typ == MCPC_I16 || typ == MCPC_U16) + anydata->data_intlk = (uintmax_t) (*((u16_t *) val)); + else if (typ == MCPC_I32 || typ == MCPC_U32) + anydata->data_intlk = (uintmax_t) (*((u32_t *) val)); + else if (typ == MCPC_I64 || typ == MCPC_U64) + anydata->data_intlk = (uintmax_t) (*((u64_t *) val)); + else if (typ == MCPC_FUNC) + anydata->data_intlk = (uintmax_t) (uintptr_t) (val); + else + return MCPC_EC_BUG; + + anydata->data_typ = typ; + anydata->data_len = 1; + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_anydata_get_intlk (const mcpc_anydata_t *anydata, mcpc_anytype_t typ, void *ret) +{ + bug_if_nullptr (anydata); + bug_if_nullptr (ret); + + if (anydata->data_typ != typ) + return MCPC_EC_ANYDATA_NOTFOUND; + + if (false); + else if (typ == MCPC_I8) + { + i8_t *ret_typed = ret; + *ret_typed = (i8_t) anydata->data_intlk; + } + else if (typ == MCPC_U8) + { + u8_t *ret_typed = ret; + *ret_typed = (u8_t) anydata->data_intlk; + } + else if (typ == MCPC_I16) + { + i16_t *ret_typed = ret; + *ret_typed = (i16_t) anydata->data_intlk; + } + else if (typ == MCPC_U16) + { + u16_t *ret_typed = ret; + *ret_typed = (u16_t) anydata->data_intlk; + } + else if (typ == MCPC_I32) + { + i32_t *ret_typed = ret; + *ret_typed = (i32_t) anydata->data_intlk; + } + else if (typ == MCPC_U32) + { + u32_t *ret_typed = ret; + *ret_typed = (u32_t) anydata->data_intlk; + } + else if (typ == MCPC_I64) + { + i64_t *ret_typed = ret; + *ret_typed = (i64_t) anydata->data_intlk; + } + else if (typ == MCPC_U64) + { + u64_t *ret_typed = ret; + *ret_typed = (u64_t) anydata->data_intlk; + } + else if (typ == MCPC_FUNC) + { + uintptr_t *ret_typed = ret; + *ret_typed = (uintptr_t) anydata->data_intlk; + } + else + return MCPC_EC_BUG; + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_anydata_set_arrlk (mcpc_anydata_t *anydata, mcpc_anytype_t typ, const void *arr, size_t arr_len) +{ + bug_if_nullptr (anydata); + bug_if_nullptr (arr); + + size_t nbytes = 0; + + if (false); + else if (typ == MCPC_I8S || typ == MCPC_U8S) + nbytes = arr_len * sizeof (u8_t); + else if (typ == MCPC_I16S || typ == MCPC_U16S) + nbytes = arr_len * sizeof (u16_t); + else if (typ == MCPC_I32S || typ == MCPC_U32S) + nbytes = arr_len * sizeof (u32_t); + else if (typ == MCPC_I64S || typ == MCPC_U64S) + nbytes = arr_len * sizeof (u64_t); + else if (typ == MCPC_U8STR) + nbytes = arr_len * sizeof (char8_t); + else + return MCPC_EC_BUG; + + if (anydata->data_typ == typ) + { + anydata->data_arrlk = mcpc_realloc (anydata->data_arrlk, nbytes); + } + else + { + if (mcpc_anytype_is_arrlike (anydata->data_typ)) + mcpc_anydata_release_data (anydata); + + anydata->data_typ = typ; + anydata->data_arrlk = mcpc_alloc (nbytes); + } + memcpy (anydata->data_arrlk, arr, nbytes); + anydata->data_len = arr_len; + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_anydata_get_arrlk (const mcpc_anydata_t *anydata, mcpc_anytype_t typ, void *ret, size_t ret_cap, size_t *ret_len) +{ + // NOTE ret_len can be nullptr + + if (anydata->data_typ != typ) + return MCPC_EC_ANYDATA_NOTFOUND; + if (ret_cap == 0) + return MCPC_EC_ANYDATA_NOTFOUND; + + size_t nb_el = 0; + if (false); + else if (typ == MCPC_I8S || typ == MCPC_U8S) + nb_el = sizeof (u8_t); + else if (typ == MCPC_I16S || typ == MCPC_U16S) + nb_el = sizeof (u16_t); + else if (typ == MCPC_I32S || typ == MCPC_U32S) + nb_el = sizeof (u32_t); + else if (typ == MCPC_I64S || typ == MCPC_U64S) + nb_el = sizeof (u64_t); + else if (typ == MCPC_U8STR) + nb_el = sizeof (char8_t); + else + return MCPC_EC_BUG; + + size_t cplen = anydata->data_len; + if (cplen > ret_cap) + cplen = ret_cap; + + memcpy (ret, anydata->data_arrlk, cplen * nb_el); + if (ret_len != nullptr) + *ret_len = cplen; + + if (ret_cap > cplen) + { + char8_t *ret_typed = ret; + ret_typed[cplen] = 0; + return MCPC_EC_0; + } + else + { + return MCPC_EC_BUFCAP; + } +} + +mcpc_errcode_t +mcpc_anydata_set_i8 (mcpc_anydata_t *anydata, i8_t val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_I8, &val); +} + +mcpc_errcode_t +mcpc_anydata_set_u8 (mcpc_anydata_t *anydata, u8_t val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_U8, &val); +} + +mcpc_errcode_t +mcpc_anydata_set_i16 (mcpc_anydata_t *anydata, i16_t val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_I16, &val); +} + +mcpc_errcode_t +mcpc_anydata_set_u16 (mcpc_anydata_t *anydata, u16_t val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_U16, &val); +} + +mcpc_errcode_t +mcpc_anydata_set_i32 (mcpc_anydata_t *anydata, i32_t val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_I32, &val); +} + +mcpc_errcode_t +mcpc_anydata_set_u32 (mcpc_anydata_t *anydata, u32_t val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_U32, &val); +} + +mcpc_errcode_t +mcpc_anydata_set_i64 (mcpc_anydata_t *anydata, i64_t val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_I64, &val); +} + +mcpc_errcode_t +mcpc_anydata_set_u64 (mcpc_anydata_t *anydata, u64_t val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_U64, &val); +} + +mcpc_errcode_t +mcpc_anydata_set_func (mcpc_anydata_t *anydata, void *val) +{ + return mcpc_anydata_set_intlk (anydata, MCPC_FUNC, val); +} + +mcpc_errcode_t +mcpc_anydata_get_i8 (const mcpc_anydata_t *anydata, i8_t *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_I8, ret); +} + +mcpc_errcode_t +mcpc_anydata_get_u8 (const mcpc_anydata_t *anydata, u8_t *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_U8, ret); +} + +mcpc_errcode_t +mcpc_anydata_get_i16 (const mcpc_anydata_t *anydata, i16_t *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_I16, ret); +} + +mcpc_errcode_t +mcpc_anydata_get_u16 (const mcpc_anydata_t *anydata, u16_t *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_U16, ret); +} + +mcpc_errcode_t +mcpc_anydata_get_i32 (const mcpc_anydata_t *anydata, i32_t *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_I32, ret); +} + +mcpc_errcode_t +mcpc_anydata_get_u32 (const mcpc_anydata_t *anydata, u32_t *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_U32, ret); +} + +mcpc_errcode_t +mcpc_anydata_get_i64 (const mcpc_anydata_t *anydata, i64_t *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_I64, ret); +} + +mcpc_errcode_t +mcpc_anydata_get_u64 (const mcpc_anydata_t *anydata, u64_t *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_U64, ret); +} + +mcpc_errcode_t +mcpc_anydata_get_func (const mcpc_anydata_t *anydata, void *ret) +{ + return mcpc_anydata_get_intlk (anydata, MCPC_FUNC, ret); +} + +mcpc_errcode_t +mcpc_anydata_set_i8s (mcpc_anydata_t *anydata, const i8_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_I8S, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_set_u8s (mcpc_anydata_t *anydata, const u8_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_U8S, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_set_i16s (mcpc_anydata_t *anydata, const i16_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_I16S, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_set_u16s (mcpc_anydata_t *anydata, const u16_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_U16S, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_set_i32s (mcpc_anydata_t *anydata, const i32_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_I32S, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_set_u32s (mcpc_anydata_t *anydata, const u32_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_U32S, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_set_i64s (mcpc_anydata_t *anydata, const i64_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_I64S, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_set_u64s (mcpc_anydata_t *anydata, const u64_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_U64S, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_set_u8str (mcpc_anydata_t *anydata, const char8_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_arrlk (anydata, MCPC_U8STR, arr, arr_len); +} + +mcpc_errcode_t +mcpc_anydata_get_i8s (const mcpc_anydata_t *anydata, i8_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_I8S, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_get_u8s (const mcpc_anydata_t *anydata, u8_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_U8S, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_get_i16s (const mcpc_anydata_t *anydata, i16_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_I16S, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_get_u16s (const mcpc_anydata_t *anydata, u16_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_U16S, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_get_i32s (const mcpc_anydata_t *anydata, i32_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_I32S, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_get_u32s (const mcpc_anydata_t *anydata, u32_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_U32S, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_get_i64s (const mcpc_anydata_t *anydata, i64_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_I64S, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_get_u64s (const mcpc_anydata_t *anydata, u64_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_U64S, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_get_u8str (const mcpc_anydata_t *anydata, char8_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_arrlk (anydata, MCPC_U8STR, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_anydata_getref_u8str (const mcpc_anydata_t *anydata, const char8_t **ret, size_t *ret_len) +{ + bug_if_nullptr (anydata); + bug_if_nullptr (ret); + bug_if_nullptr (ret_len); + + *ret = (const char8_t *) anydata->data_arrlk; + *ret_len = anydata->data_len; + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_anydata_getref_meta1 (const mcpc_anydata_t *anydata, const char8_t **ret, size_t *ret_len) +{ + bug_if_nullptr (anydata); + bug_if_nullptr (ret); + bug_if_nullptr (ret_len); + + *ret = anydata->meta1; + *ret_len = anydata->meta1_len; + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_anydata_getref_meta2 (const mcpc_anydata_t *anydata, const char8_t **ret, size_t *ret_len) +{ + bug_if_nullptr (anydata); + bug_if_nullptr (ret); + bug_if_nullptr (ret_len); + + *ret = anydata->meta2; + *ret_len = anydata->meta2_len; + + return MCPC_EC_0; +} + +/* = = = = = = = = = = = = = = = = anypool = = = = = = = = = = = = = = = = */ + +mcpc_anypool_t * +mcpc_anypool_new () +{ + mcpc_anypool_t *hp_anypool = mcpc_calloc (sizeof (mcpc_anypool_t)); + return hp_anypool; +} + +void +mcpc_anypool_free (mcpc_anypool_t *anypool) +{ + bug_if (anypool == nullptr); + for (size_t i = 0; i < anypool->len; i++) + { + mcpc_anydata_t *cur_data = (mcpc_anydata_t *) anypool->head + i; + mcpc_anydata_mbfree (cur_data); + } + if (anypool->cap > 0 && anypool->head != nullptr) + { + free ((void *) anypool->head); + } + free (anypool); +} + + +void +mcpc_anypool_set_cap (mcpc_anypool_t *pool, size_t newcap) +{ + size_t newsz = newcap * sizeof (mcpc_anydata_t); + pool->head = mcpc_realloc ((void *) pool->head, newsz); // TODO use wrapper + pool->cap = newcap; +} + + +void +mcpc_anypool_copy (mcpc_anypool_t *dest_pool, const mcpc_anypool_t *src_pool) +{ + bug_if (dest_pool == nullptr); + bug_if (src_pool == nullptr); + + // make sure inited. this also means dest's content is insignificant + memset (dest_pool, 0, sizeof (mcpc_anypool_t)); + + mcpc_anypool_set_cap (dest_pool, src_pool->cap); + + for (size_t i = 0; i < src_pool->len; i++) // TODO ckdadd + mcpc_anydata_copy ((mcpc_anydata_t *) dest_pool->head + i, src_pool->head + i); + + dest_pool->len = src_pool->len; +} + +void +mcpc_anypool_grow (mcpc_anypool_t *pool) +{ + size_t newcap = pool->cap + mcpc_anypool_gamt; + mcpc_anypool_set_cap (pool, newcap); +} + +void +mcpc_anypool_addcpy (mcpc_anypool_t *anypool, const mcpc_anydata_t *tmp_data) +{ + bug_if (anypool == nullptr); + bug_if (tmp_data == nullptr); + + if (anypool->len + 1 > anypool->cap) + { + mcpc_anypool_grow (anypool); + } + bug_if (anypool->len + 1 > anypool->cap); + + mcpc_anydata_t *newdata = (mcpc_anydata_t *) (anypool->head + anypool->len); + mcpc_anydata_copy (newdata, tmp_data); + anypool->len += 1; +} + +void +mcpc_anypool_addfre (mcpc_anypool_t *anypool, mcpc_anydata_t *tmp_data) +{ + mcpc_anypool_addcpy (anypool, tmp_data); + mcpc_anydata_free (tmp_data); +} + +const mcpc_anydata_t * +mcpc_anypool_get (const mcpc_anypool_t *anypool, size_t idx) +{ + bug_if (anypool == nullptr); + if (idx >= anypool->len) + return nullptr; + return anypool->head + idx; +} + +const mcpc_anydata_t * +mcpc_anypool_getbymeta1 (const mcpc_anypool_t *anypool, const char8_t *meta1, size_t meta1_len) +{ + bug_if (anypool == nullptr); + + if (meta1 == nullptr || meta1_len == 0) + return nullptr; + + for (size_t i = 0; i < anypool->len; i++) // TODO: ckdadd + { + const mcpc_anydata_t *cur_data = anypool->head + i; + if (cur_data->meta1_len > 0) + { + size_t cmpsz = min_size (meta1_len, (size_t) cur_data->meta1_len); + if (0 == u8strncmp (cur_data->meta1, meta1, cmpsz)) + return cur_data; + } + } + + return nullptr; +} + +const mcpc_anydata_t * +mcpc_anypool_getbymeta1nt (const mcpc_anypool_t *pool, const char8_t *meta1nt) +{ + size_t meta1nt_len = u8strlen (meta1nt); + return mcpc_anypool_getbymeta1 (pool, meta1nt, meta1nt_len); +} diff --git a/third_party/mcpc/src/anydata.h b/third_party/mcpc/src/anydata.h new file mode 100644 index 0000000..1188e05 --- /dev/null +++ b/third_party/mcpc/src/anydata.h @@ -0,0 +1,225 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_anydata +#define h_anydata + +#include <stddef.h> + +#include "ext_stdint.h" + +#include <mcpc/_c23_keywords.h> +#include <mcpc/_c23_uchar.h> + +#include <mcpc/anydata.h> +#include <mcpc/errcode.h> + +// ---------------------------- anydata ---------------------------- + +#ifdef MCPC_C23GIVUP_FIXENUM +enum mcpc_anydata_flags +#else +enum mcpc_anydata_flags:u16_t +#endif +{ + ANYDATA_FL_REQUIRED = 1 << 0, + ANYDATA_FL_TOOL = 1 << 1, + ANYDATA_FL_TOOLPROP = 1 << 2, + ANYDATA_FL_RSC = 1 << 3, + ANYDATA_FL_PRMPT = 1 << 4, + ANYDATA_FL_PRMPTARG = 1 << 5, + ANYDATA_FL_USRCBRES_TOOLCALL = 1 << 6, + ANYDATA_FL_USRCBRES_TOOLCALL_TXTCTN = 1 << 7, + ANYDATA_FL_USRCBRES_TOOLCALL_IMGCTN = 1 << 8, + ANYDATA_FL_USRCBRES_PRMPTCALL = 1 << 9, + ANYDATA_FL_USRCBRES_PRMPTCALL_ASSIST = 1 << 10, + ANYDATA_FL_COMPLT = 1 << 11, + ANYDATA_FL_PRMPTARGHINT = 1 << 12, + ANYDATA_FL_PRESENT = 1 << 13, +}; +typedef enum mcpc_anydata_flags mcpc_anydata_flags_t; + +struct mcpc_anydata +{ + u8_t meta1_len; + mcpc_mime_t data_mime; + mcpc_anytype_t data_typ; + mcpc_anydata_flags_t flags; + u16_t meta2_len; + size_t data_len; + const char8_t *meta1; + const char8_t *meta2; + mcpc_anypool_t *chd_pool; + union + { + uintmax_t data_intlk; + uintptr_t *data_arrlk; + }; +}; + +mcpc_anydata_t *mcpc_anydata_new (); + +mcpc_anydata_t *mcpc_anydata_new_typ (mcpc_anytype_t typ); + +mcpc_anydata_t *mcpc_anydata_new_meta1_meta2 (const char8_t * meta1, + const size_t meta1_len, const char8_t * meta2, const size_t meta2_len); + +mcpc_anydata_t *mcpc_anydata_new_meta1nt_meta2nt (const char8_t * meta1nt, const char8_t * meta2nt); + +mcpc_anydata_t *mcpc_anydata_new_typ_meta1 (mcpc_anytype_t typ, const char8_t * meta1, const size_t meta1_len); + +mcpc_anydata_t *mcpc_anydata_new_typ_meta1_meta2 (mcpc_anytype_t typ, + const char8_t * meta1, + const size_t meta1_len, const char8_t * meta2, + const size_t meta2_len); + +mcpc_anydata_t *mcpc_anydata_new_typ_meta1nt_meta2nt (mcpc_anytype_t typ, + const char8_t * meta1nt, const char8_t * meta2nt); + +void mcpc_anydata_free (mcpc_anydata_t * data); + +void mcpc_anydata_copy (mcpc_anydata_t * dest_data, const mcpc_anydata_t * src_data); + +mcpc_errcode_t mcpc_anydata_set_meta1nt (mcpc_anydata_t * anydata, const char8_t * meta1); + +mcpc_errcode_t mcpc_anydata_set_i8 (mcpc_anydata_t * anydata, i8_t val); + +mcpc_errcode_t mcpc_anydata_set_u8 (mcpc_anydata_t * anydata, u8_t val); + +mcpc_errcode_t mcpc_anydata_set_i16 (mcpc_anydata_t * anydata, i16_t val); + +mcpc_errcode_t mcpc_anydata_set_u16 (mcpc_anydata_t * anydata, u16_t val); + +mcpc_errcode_t mcpc_anydata_set_i32 (mcpc_anydata_t * anydata, i32_t val); + +mcpc_errcode_t mcpc_anydata_set_u32 (mcpc_anydata_t * anydata, u32_t val); + +mcpc_errcode_t mcpc_anydata_set_i64 (mcpc_anydata_t * anydata, i64_t val); + +mcpc_errcode_t mcpc_anydata_set_u64 (mcpc_anydata_t * anydata, u64_t val); + +mcpc_errcode_t mcpc_anydata_set_func (mcpc_anydata_t * anydata, void *val); + +mcpc_errcode_t mcpc_anydata_get_i8 (const mcpc_anydata_t * anydata, i8_t * ret); + +mcpc_errcode_t mcpc_anydata_get_u8 (const mcpc_anydata_t * anydata, u8_t * ret); + +mcpc_errcode_t mcpc_anydata_get_i16 (const mcpc_anydata_t * anydata, i16_t * ret); + +mcpc_errcode_t mcpc_anydata_get_u16 (const mcpc_anydata_t * anydata, u16_t * ret); + +mcpc_errcode_t mcpc_anydata_get_i32 (const mcpc_anydata_t * anydata, i32_t * ret); + +mcpc_errcode_t mcpc_anydata_get_u32 (const mcpc_anydata_t * anydata, u32_t * ret); + +mcpc_errcode_t mcpc_anydata_get_i64 (const mcpc_anydata_t * anydata, i64_t * ret); + +mcpc_errcode_t mcpc_anydata_get_u64 (const mcpc_anydata_t * anydata, u64_t * ret); + +mcpc_errcode_t mcpc_anydata_get_func (const mcpc_anydata_t * anydata, void *ret); + +mcpc_errcode_t mcpc_anydata_set_i8s (mcpc_anydata_t * anydata, const i8_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_set_u8s (mcpc_anydata_t * anydata, const u8_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_set_i16s (mcpc_anydata_t * anydata, const i16_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_set_u16s (mcpc_anydata_t * anydata, const u16_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_set_i32s (mcpc_anydata_t * anydata, const i32_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_set_u32s (mcpc_anydata_t * anydata, const u32_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_set_i64s (mcpc_anydata_t * anydata, const i64_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_set_u64s (mcpc_anydata_t * anydata, const u64_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_set_u8str (mcpc_anydata_t * anydata, const char8_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_anydata_get_i8s (const mcpc_anydata_t * anydata, i8_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_get_u8s (const mcpc_anydata_t * anydata, u8_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_get_i16s (const mcpc_anydata_t * anydata, i16_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_get_u16s (const mcpc_anydata_t * anydata, u16_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_get_i32s (const mcpc_anydata_t * anydata, i32_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_get_u32s (const mcpc_anydata_t * anydata, u32_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_get_i64s (const mcpc_anydata_t * anydata, i64_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_get_u64s (const mcpc_anydata_t * anydata, u64_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_get_u8str (const mcpc_anydata_t * anydata, char8_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_getref_u8str (const mcpc_anydata_t * anydata, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_getref_meta1 (const mcpc_anydata_t * anydata, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_anydata_getref_meta2 (const mcpc_anydata_t * anydata, const char8_t ** ret, size_t *ret_len); + +const mcpc_anydata_t *mcpc_anydata_get_child (const mcpc_anydata_t * anydata, size_t idx); + +void mcpc_anydata_add_child (mcpc_anydata_t * data, mcpc_anydata_t * child); + +void mcpc_anydata_addcpy_child (mcpc_anydata_t * data, const mcpc_anydata_t * child); + +// ---------------------------- anypool ---------------------------- + +#ifdef MCPC_C23GIVUP_FIXENUM +enum mcpc_anypool_flags +#else +enum mcpc_anypool_flags:u16_t +#endif +{ + ANYPOOL_FL_USRCBRES_TOOLCALL = 1 << 0, + ANYPOOL_FL_USRCBRES_TOOLCALL_ERROR = 1 << 1, + ANYPOOL_FL_USRCBRES_PRMPTCALL = 1 << 2, +}; +typedef enum mcpc_anypool_flags mcpc_anypool_flags_t; + +struct mcpc_anypool +{ + mcpc_anypool_flags_t flags; + size_t cap; + size_t len; + const mcpc_anydata_t *head; +}; + +const mcpc_anydata_t *mcpc_anypool_get (const mcpc_anypool_t * anypool, size_t idx); + +const mcpc_anydata_t *mcpc_anypool_getbymeta1 (const mcpc_anypool_t * anypool, const char8_t * meta1, size_t meta1_len); + +const mcpc_anydata_t *mcpc_anypool_getbymeta1nt (const mcpc_anypool_t * pool, const char8_t * meta1nt); + +mcpc_anypool_t *mcpc_anypool_new (); + +void mcpc_anypool_free (mcpc_anypool_t * anypool); + +void mcpc_anypool_copy (mcpc_anypool_t * dest_pool, const mcpc_anypool_t * src_pool); + +void mcpc_anypool_addfre (mcpc_anypool_t * anypool, mcpc_anydata_t * tmp_data); + +void mcpc_anypool_addcpy (mcpc_anypool_t * anypool, const mcpc_anydata_t * tmp_data); + +#endif diff --git a/third_party/mcpc/src/complt.c b/third_party/mcpc/src/complt.c new file mode 100644 index 0000000..730d53b --- /dev/null +++ b/third_party/mcpc/src/complt.c @@ -0,0 +1,162 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "complt.h" + +#include "anydata.h" +#include "ext_string.h" + + +mcpc_complt_t * +mcpc_complt_new () +{ + mcpc_anydata_t *hp = mcpc_anydata_new (); + + hp->flags |= ANYDATA_FL_COMPLT; + + return hp; +} + +mcpc_errcode_t +mcpc_complt_add_compltcandi_u8str (mcpc_complt_t *complt, const char8_t *u8str, size_t u8str_len) +{ + bug_if_nullptr (complt); + bug_if_nullptr (u8str); + mcpc_anydata_t *chd = mcpc_anydata_new (); + bug_if_nz (mcpc_anydata_set_u8str (chd, u8str, u8str_len)); + if (complt->chd_pool == nullptr) + complt->chd_pool = mcpc_anypool_new (); + (mcpc_anypool_addfre (complt->chd_pool, chd)); + return MCPC_EC_0; +} + +// provide completions for specific prompt arg, completing its hint, (currently hint is anydata with merely u8str data) +mcpc_complt_t * +mcpc_complt_new_prmptarg (const mcpc_prmpt_t *prmpt, const char8_t *argname, size_t argname_len, const char8_t *argval, + size_t argval_len) +{ + mcpc_complt_t *complt = mcpc_complt_new (); + + const mcpc_prmptarg_t *prmptarg = mcpc_prmpt_getbyname_prmptarg (prmpt, argname, argname_len); + if (prmptarg == nullptr) + return complt; + + bool match_any = argval_len == 0; + + const mcpc_prmptarghint_t *cur_candi = nullptr; + while ((cur_candi = mcpc_prmptarg_get_next_prmptarghint (prmptarg))) + { + const char8_t *cur_val; + size_t cur_val_len; + bug_if_nz (mcpc_prmptarghint_getref_u8str (cur_candi, &cur_val, &cur_val_len)); + + if (match_any) + bug_if_nz (mcpc_complt_add_compltcandi_u8str (complt, cur_val, cur_val_len)); + else + { + size_t cmpsz = min_size (cur_val_len, argval_len); + if (0 == u8strncmp (argval, cur_val, cmpsz)) + { + bug_if_nz (mcpc_complt_add_compltcandi_u8str (complt, cur_val, cur_val_len)); + } + } + } + + return complt; +} + +// provide completions for specific rsc, completing uri +mcpc_complt_t * +mcpc_complt_new_rsc (const mcpc_rscpool_t *rscpool, const char8_t *uriparts, size_t uriparts_len) +{ + mcpc_complt_t *complt = mcpc_complt_new (); + + bool match_any = uriparts_len == 0; + + const mcpc_rsc_t *cur_rsc = nullptr; + while ((cur_rsc = mcpc_rscpool_get_next (rscpool))) + { + const char8_t *cur_uri; + size_t cur_uri_len; + bug_if_nz (mcpc_rsc_getref_uri (cur_rsc, &cur_uri, &cur_uri_len)); + + if (match_any) + bug_if_nz (mcpc_complt_add_compltcandi_u8str (complt, cur_uri, cur_uri_len)); + else if (uriparts_len <= cur_uri_len) + { + size_t cmpsz = uriparts_len; + if (0 == u8strncmp (cur_uri, uriparts, cmpsz)) + bug_if_nz (mcpc_complt_add_compltcandi_u8str (complt, cur_uri, cur_uri_len)); + } + } + + return complt; +} + +void +mcpc_complt_free (mcpc_complt_t *complt) +{ + mcpc_anydata_free (complt); +} + +mcpc_errcode_t +mcpc_complt_get_compltcandi_len (const mcpc_complt_t *complt, size_t *ret) +{ + bug_if_nullptr (complt); + if (complt->chd_pool == nullptr) + *ret = 0; + else + *ret = complt->chd_pool->len; + return MCPC_EC_0; +} + +const mcpc_compltcand_t * +mcpc_complt_get_next_compltcand (const mcpc_complt_t *complt) +{ + bug_if_nullptr (complt); + + static size_t idx = 0; + + const mcpc_anypool_t *pool = complt->chd_pool; + if (pool == nullptr) + return nullptr; + + const size_t pool_len = pool->len; + if (pool_len == 0) + return nullptr; + + if (idx == pool_len) // TODO MT-SAFE + { + idx = 0; + return nullptr; + } + + const mcpc_anydata_t *ret = pool->head + idx; + idx += 1; + + return ret; +} + +mcpc_errcode_t +mcpc_compltcand_getref_u8str (const mcpc_compltcand_t *compltcand, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_u8str (compltcand, ret, ret_len); +} diff --git a/third_party/mcpc/src/complt.h b/third_party/mcpc/src/complt.h new file mode 100644 index 0000000..c11dc3f --- /dev/null +++ b/third_party/mcpc/src/complt.h @@ -0,0 +1,60 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_complt +#define h_complt + +// not supposed to be used by lib users + +#include <mcpc/_c23_keywords.h> +#include <mcpc/_c23_uchar.h> + +#include "ext_string.h" + +#include <mcpc/anydata.h> + +#include "errcode.h" + +#include "prmpt.h" +#include "rsc.h" + + +typedef mcpc_anydata_t mcpc_complt_t; +typedef mcpc_anydata_t mcpc_compltcand_t; + +mcpc_complt_t *mcpc_complt_new_prmptarg (const mcpc_prmpt_t * prmpt, const char8_t * argname, size_t argname_len, + const char8_t * argval, size_t argval_len); + +mcpc_complt_t *mcpc_complt_new_rsc (const mcpc_rscpool_t * rscpool, const char8_t * uri, size_t uri_len); + +void mcpc_complt_free (mcpc_complt_t * complt); + +const mcpc_compltcand_t *mcpc_complt_get_next_compltcand (const mcpc_complt_t * complt); + +mcpc_errcode_t mcpc_complt_get_compltcandi_len (const mcpc_complt_t * complt, size_t *ret); + + +mcpc_errcode_t +mcpc_compltcand_getref_u8str (const mcpc_compltcand_t * compltcand, const char8_t ** ret, size_t *ret_len); + + + +#endif diff --git a/third_party/mcpc/src/errcode.c b/third_party/mcpc/src/errcode.c new file mode 100644 index 0000000..9fa5ddc --- /dev/null +++ b/third_party/mcpc/src/errcode.c @@ -0,0 +1,39 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "errcode.h" + +#include <stdlib.h> +// #include <stdio.h> + +mcpc_errcode_t mcpc_ec = MCPC_EC_0; + +void mcpc_assert(const bool cond, const mcpc_errcode_t ec) +{ + if (cond) + { + return; + } + mcpc_ec = ec; + abort(); + // fprintf(stderr,"!!!\n"); + // exit((int)ec); +} diff --git a/third_party/mcpc/src/errcode.h b/third_party/mcpc/src/errcode.h new file mode 100644 index 0000000..6cd1d88 --- /dev/null +++ b/third_party/mcpc/src/errcode.h @@ -0,0 +1,74 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_errcode +#define h_errcode + + +#include <mcpc/errcode.h> +#include <mcpc/_c23_keywords.h> + + + +// [[maybe_unused]] static mcpc_errcode_t mcpc_ec = MCPC_EC_CONNPOOL_NOTADD; +// useless: after abort(), p mcpc_ec is always default value +extern mcpc_errcode_t mcpc_ec; + +// [[noreturn]] ??? +void +mcpc_assert(const bool cond, const mcpc_errcode_t ec); + + +static inline void bug_if_nullptr(const void *ptr) +{ + if (ptr == nullptr) + { + mcpc_assert(false, MCPC_EC_BUG); + } +} + +static inline void bug_if_zero(const unsigned long long ptr) +{ + if (ptr == 0) + { + mcpc_assert(false, MCPC_EC_BUG); + } +} + +static inline void bug_if_nz(const unsigned long long ptr) +{ + if (ptr != 0) + { + mcpc_assert(false, MCPC_EC_BUG); + } +} + + +static inline void bug_if(bool f) +{ + if (f) + { + mcpc_assert(false, MCPC_EC_BUG); + } +} + + +#endif diff --git a/third_party/mcpc/src/ext_stdint.h b/third_party/mcpc/src/ext_stdint.h new file mode 100644 index 0000000..bc47e35 --- /dev/null +++ b/third_party/mcpc/src/ext_stdint.h @@ -0,0 +1,54 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_ext_stdint +#define h_ext_stdint + +#include <stdint.h> +#include <stddef.h> + +typedef int8_t i8_t; +typedef uint8_t u8_t; +typedef int16_t i16_t; +typedef uint16_t u16_t; +typedef int32_t i32_t; +typedef uint32_t u32_t; +typedef int64_t i64_t; +typedef uint64_t u64_t; + +static inline i32_t +min_i32 (i32_t a, i32_t b) +{ + if (a < b) + return a; + else + return b; +} + +static inline size_t +min_size (size_t a, size_t b) +{ + if (b < a) + return b; + return a; +} + +#endif diff --git a/third_party/mcpc/src/ext_string.h b/third_party/mcpc/src/ext_string.h new file mode 100644 index 0000000..87cc198 --- /dev/null +++ b/third_party/mcpc/src/ext_string.h @@ -0,0 +1,75 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h__string +#define h__string + +#include <string.h> +#include <stddef.h> +#include <string.h> +#include <ctype.h> + +#include <mcpc/_c23_keywords.h> +#include <mcpc/_c23_uchar.h> + +#define u8c1(a) u8##"" a +#define u8c2(a, b) u8##"" a b +#define u8c3(a, b, c) u8##"" a b c +#define u8c4(a, b, c, d) u8##"" a b c d +#define u8c5(a, b, c, d, e) u8##"" a b c d e +#define u8c6(a, b, c, d, e, f) u8##"" a b c d e f +#define u8c7(a, b, c, d, e, f, g) u8##"" a b c d e f g + +inline static bool +u8streq(const char8_t* a, const char8_t* b) +{ + const char* const aa = (const char* const)a; + const char* const bb = (const char* const)b; + return strcmp(aa, bb) == 0; +} + +inline static size_t +u8strlen(const char8_t* u8s) +{ + const char* s = (const char*)u8s; + return strlen(s); +} + +static inline int +u8strcmp (const char8_t *s1, const char8_t *s2) +{ + return strcmp ((const char *) s1, (const char *) s2); +} + +static inline int +u8strncmp (const char8_t *s1, const char8_t *s2, size_t n) +{ + return strncmp ((const char *) s1, (const char *) s2, n); +} + +static inline bool +beg_with_alph(const char8_t* const u8s) +{ + const char* const s = (const char* const)u8s; + return s[0] == '/' || isalpha((int)s[0]); +} + +#endif diff --git a/third_party/mcpc/src/log.c b/third_party/mcpc/src/log.c new file mode 100644 index 0000000..ebf9879 --- /dev/null +++ b/third_party/mcpc/src/log.c @@ -0,0 +1,46 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "log.h" + +LogLevel glv = Debug; + +void +lvlog (LogLevel lv, const char *meta, FILE *stream, const char *format, ...) +{ + if (lv > glv) + return; + + char logind = 'E'; + if (lv == Info) + logind = 'I'; + else if (lv == Debug) + logind = 'D'; + + va_list args; + va_start (args, format); + fprintf (stream, "[%c] %s ", logind, meta); + vfprintf (stream, format, args); + fprintf (stream, "\n"); + fflush (stream); + fclose (stream); + va_end (args); +} diff --git a/third_party/mcpc/src/log.h b/third_party/mcpc/src/log.h new file mode 100644 index 0000000..ae9eb72 --- /dev/null +++ b/third_party/mcpc/src/log.h @@ -0,0 +1,60 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_log +#define h_log + +#include <stdarg.h> +#include <stdio.h> + +typedef enum +{ + Error = 1, + Info, + Debug, +} LogLevel; + +extern LogLevel glv; // TODO: thread-safety + +#ifdef MCPC_DBG +#define tulog_d(...) \ + do \ + { \ + char meta[100]; \ + sprintf(meta, "%s,%d", __TU__, __LINE__); \ + lvlog(Debug, meta, stderr __VA_OPT__(, ) __VA_ARGS__); \ + } while (false) +#define tuflog_d(...) \ + do \ + { \ + FILE *logf = fopen("/tmp/mcpc.log", "a"); \ + char meta[100]; \ + sprintf(meta, "%s,%d", __TU__, __LINE__); \ + lvlog(Debug, meta, logf __VA_OPT__(, ) __VA_ARGS__); \ + } while (false) +#else +#define tulog_d(...) ; +#define tuflog_d(...) ; +#endif + +void lvlog(LogLevel lv, const char *meta, FILE *stream, const char *format, ...); + +#endif diff --git a/third_party/mcpc/src/mcpc.pc b/third_party/mcpc/src/mcpc.pc new file mode 100644 index 0000000..4dbd624 --- /dev/null +++ b/third_party/mcpc/src/mcpc.pc @@ -0,0 +1,10 @@ +prefix=/usr/local +libdir=${prefix}/lib +includedir=${prefix}/include + +Name: mcpc +Description: A library for Model Context Protocol(MCP) +URL: https://github.com/micl2e2/mcpc +Version: 0.1.0 +Libs: -L${libdir} -lmcpc +Cflags: -I${includedir} diff --git a/third_party/mcpc/src/mjson.c b/third_party/mcpc/src/mjson.c new file mode 100644 index 0000000..66123da --- /dev/null +++ b/third_party/mcpc/src/mjson.c @@ -0,0 +1,1372 @@ +// Copyright (c) 2018-2020 Cesanta Software Limited +// 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. + +#if is_wincl +#pragma warning(disable : 5040) +#endif + +#include <float.h> +#include <math.h> + +#include "mjson.h" + +#if is_win || is_mingw +#include <malloc.h> +#else +#include <alloca.h> +#endif + +#if defined(_MSC_VER) && _MSC_VER < 1700 +#define va_copy(x, y) (x) = (y) +#define isinf(x) !_finite(x) +#define isnan(x) _isnan(x) +#endif + +static double +mystrtod(const char* str, const char** end); + +static int +mjson_esc(int c, int esc) +{ + const char *p, *esc1 = "\b\f\n\r\t\\\"", *esc2 = "bfnrt\\\""; + for (p = esc ? esc1 : esc2; *p != '\0'; p++) { + if (*p == c) + return esc ? esc2[p - esc1] : esc1[p - esc2]; + } + return 0; +} + +static int +mjson_escape(int c) +{ + return mjson_esc(c, 1); +} + +static int +mjson_pass_string(const char* s, int len) +{ + int i; + for (i = 0; i < len; i++) { + if (s[i] == '\\' && i + 1 < len && mjson_escape(s[i + 1])) { + i++; + } else if (s[i] == '\0') { + return MJSON_ERROR_INVALID_INPUT; + } else if (s[i] == '"') { + return i; + } + } + return MJSON_ERROR_INVALID_INPUT; +} + +int +mjson(const char* s, int len, mjson_cb_t cb, void* ud) +{ + enum + { + S_VALUE, + S_KEY, + S_COLON, + S_COMMA_OR_EOO + } expecting = S_VALUE; + unsigned char nesting[MJSON_MAX_DEPTH]; + int i, depth = 0; +#define MJSONCALL(ev) \ + if (cb != NULL && cb(ev, s, start, i - start + 1, ud)) \ + return i + 1; + +// In the ascii table, the distance between `[` and `]` is 2. +// Ditto for `{` and `}`. Hence +2 in the code below. +#define MJSONEOO() \ + do { \ + if (c != nesting[depth - 1] + 2) \ + return MJSON_ERROR_INVALID_INPUT; \ + depth--; \ + if (depth == 0) { \ + MJSONCALL(tok); \ + return i + 1; \ + } \ + } while (0) + + for (i = 0; i < len; i++) { + int start = i; + unsigned char c = ((const unsigned char*)s)[i]; + int tok = c; + if (c == ' ' || c == '\t' || c == '\n' || c == '\r') + continue; + // printf("- %c [%.*s] %d %d\n", c, i, s, depth, expecting); + switch (expecting) { + case S_VALUE: + if (c == '{') { + if (depth >= (int)sizeof(nesting)) + return MJSON_ERROR_TOO_DEEP; + nesting[depth++] = c; + expecting = S_KEY; + break; + } else if (c == '[') { + if (depth >= (int)sizeof(nesting)) + return MJSON_ERROR_TOO_DEEP; + nesting[depth++] = c; + break; + } else if (c == ']' && depth > 0) { // Empty array + MJSONEOO(); + } else if (c == 't' && i + 3 < len && memcmp(&s[i], "true", 4) == 0) { + i += 3; + tok = MJSON_TOK_TRUE; + } else if (c == 'n' && i + 3 < len && memcmp(&s[i], "null", 4) == 0) { + i += 3; + tok = MJSON_TOK_NULL; + } else if (c == 'f' && i + 4 < len && memcmp(&s[i], "false", 5) == 0) { + i += 4; + tok = MJSON_TOK_FALSE; + } else if (c == '-' || ((c >= '0' && c <= '9'))) { + const char* end = NULL; + mystrtod(&s[i], &end); + if (end != NULL) + i += (int)(end - &s[i] - 1); + tok = MJSON_TOK_NUMBER; + } else if (c == '"') { + int n = mjson_pass_string(&s[i + 1], len - i - 1); + if (n < 0) + return n; + i += n + 1; + tok = MJSON_TOK_STRING; + } else { + return MJSON_ERROR_INVALID_INPUT; + } + if (depth == 0) { + MJSONCALL(tok); + return i + 1; + } + expecting = S_COMMA_OR_EOO; + break; + + case S_KEY: + if (c == '"') { + int n = mjson_pass_string(&s[i + 1], len - i - 1); + if (n < 0) + return n; + i += n + 1; + tok = MJSON_TOK_KEY; + expecting = S_COLON; + } else if (c == '}') { // Empty object + MJSONEOO(); + expecting = S_COMMA_OR_EOO; + } else { + return MJSON_ERROR_INVALID_INPUT; + } + break; + + case S_COLON: + if (c == ':') { + expecting = S_VALUE; + } else { + return MJSON_ERROR_INVALID_INPUT; + } + break; + + case S_COMMA_OR_EOO: + if (depth <= 0) + return MJSON_ERROR_INVALID_INPUT; + if (c == ',') { + expecting = (nesting[depth - 1] == '{') ? S_KEY : S_VALUE; + } else if (c == ']' || c == '}') { + MJSONEOO(); + } else { + return MJSON_ERROR_INVALID_INPUT; + } + break; + } + MJSONCALL(tok); + } + return MJSON_ERROR_INVALID_INPUT; +} + +struct msjon_get_data +{ + const char* path; // Lookup json path + int pos; // Current path position + int d1; // Current depth of traversal + int d2; // Expected depth of traversal + int i1; // Index in an array + int i2; // Expected index in an array + int obj; // If the value is array/object, offset where it starts + const char** tokptr; // Destination + int* toklen; // Destination length + int tok; // Returned token +}; + +#include <stdio.h> + +static int +plen1(const char* s) +{ + int i = 0, n = 0; + while (s[i] != '\0' && s[i] != '.' && s[i] != '[') + n++, i += s[i] == '\\' ? 2 : 1; + // printf("PLEN: s: [%s], [%.*s] => %d\n", s, i, s, n); + return n; +} + +static int +plen2(const char* s) +{ + int i = 0, n = 0; + while (s[i] != '\0' && s[i] != '.' && s[i] != '[') + n++, i += s[i] == '\\' ? 2 : 1; + // printf("PLEN: s: [%s], [%.*s] => %d\n", s, i, s, n); + return i; +} + +static int +kcmp(const char* a, const char* b, int n) +{ + int i = 0, j = 0, r = 0; + for (i = 0, j = 0; j < n; i++, j++) { + if (b[i] == '\\') + i++; + if ((r = a[j] - b[i]) != 0) + return r; + } + // printf("KCMP: a: [%.*s], b:[%.*s] ==> %d\n", n, a, i, b, r); + return r; +} + +static int +mjson_get_cb(int tok, const char* s, int off, int len, void* ud) +{ + struct msjon_get_data* d = (struct msjon_get_data*)ud; +#if 0 + printf("--> %2x %2d %2d %2d %2d\t %2d %2d\t'%s' '%s' '%s' '%s'\n", tok, d->d1, + d->d2, d->i1, d->i2, (int) off, (int) d->pos, s, d->path, s + off, + d->path + d->pos); +#endif + if (d->tok != MJSON_TOK_INVALID) + return 1; // Found + if (tok == '{' || tok == '[') { + if (d->d1 < d->d2) + d->obj = -1; + if (d->d1 == d->d2) + d->obj = off; + if (d->d1 == d->d2 && tok == '[' && d->path[d->pos] == '[') { + d->i1 = 0; + d->i2 = (int)mystrtod(&d->path[d->pos + 1], NULL); + if (d->i1 == d->i2) { + while (d->path[d->pos] && d->path[d->pos] != ']') + d->pos++; + if (d->path[d->pos] == ']') + d->pos++; + d->d2++; + } + } + d->d1++; + } else if (tok == '}' || tok == ']') { + if (tok == ']' && d->d1 == d->d2) + d->i1 = 0; + d->d1--; + // printf("X %s %d %d %d %d %d\n", d->path + d->pos, d->d1, d->d2, d->i1, + // d->i2, d->obj); + if (!d->path[d->pos] && d->d1 == d->d2 && d->obj != -1) { + d->tok = tok - 2; + if (d->tokptr) + *d->tokptr = s + d->obj; + if (d->toklen) + *d->toklen = off - d->obj + 1; + return 1; + } + } else if (tok == ',' && d->d1 == d->d2 && d->pos && + d->path[d->pos - 1] == ']') { + return 1; // Not found in the current elem array + } else if (tok == ',' && d->d1 == d->d2 + 1 && d->path[d->pos] == '[') { + // printf("GG '%s' '%s'\n", d->path, &d->path[d->pos]); + d->i1++; + if (d->i1 == d->i2) { + while (d->path[d->pos] && d->path[d->pos] != ']') + d->pos++; + if (d->path[d->pos] == ']') + d->pos++; + d->d2++; + } + } else if (tok == MJSON_TOK_KEY && d->d1 == d->d2 + 1 && + d->path[d->pos] == '.' && s[off] == '"' && + s[off + len - 1] == '"' && + plen1(&d->path[d->pos + 1]) == len - 2 && + kcmp(s + off + 1, &d->path[d->pos + 1], len - 2) == 0) { + d->d2++; + d->pos += plen2(&d->path[d->pos + 1]) + 1; + } else if (tok == MJSON_TOK_KEY && d->d1 == d->d2) { + return 1; // Exhausted path, not found + } else if (MJSON_TOK_IS_VALUE(tok)) { + // printf("T %d %d %d %d %d\n", tok, d->d1, d->d2, d->i1, d->i2); + if (d->d1 == d->d2 && d->i1 == d->i2 && !d->path[d->pos]) { + d->tok = tok; + if (d->tokptr) + *d->tokptr = s + off; + if (d->toklen) + *d->toklen = len; + return 1; + } + } + return 0; +} + +int +mjson_find(const char* s, int n, const char* jp, const char** tp, int* tl) +{ + struct msjon_get_data data = { jp, 1, 0, 0, 0, + 0, -1, tp, tl, MJSON_TOK_INVALID }; + if (jp[0] != '$') + return MJSON_TOK_INVALID; + if (mjson(s, n, mjson_get_cb, &data) < 0) + return MJSON_TOK_INVALID; + return data.tok; +} + +int32_t +mjson_find8(const char8_t* buf, + int32_t len, + const char8_t* jp, + const char8_t** tp, + int32_t* tl) +{ + return mjson_find( + (const char*)buf, (int)len, (const char*)jp, (const char**)tp, (int*)tl); +} + +int32_t +mjson_find88(const char* buf, + int32_t len, + const char8_t* jp, + const char8_t** tp, + int32_t* tl) +{ + return mjson_find( + (const char*)buf, (int)len, (const char*)jp, (const char**)tp, (int*)tl); +} + +int +mjson_get_number(const char* s, int len, const char* path, double* v) +{ + const char* p; + int tok, n; + if ((tok = mjson_find(s, len, path, &p, &n)) == MJSON_TOK_NUMBER) { + if (v != NULL) + *v = mystrtod(p, NULL); + } + return tok == MJSON_TOK_NUMBER ? 1 : 0; +} + +int +mjson_get_bool(const char* s, int len, const char* path, int* v) +{ + int tok = mjson_find(s, len, path, NULL, NULL); + if (tok == MJSON_TOK_TRUE && v != NULL) + *v = 1; + if (tok == MJSON_TOK_FALSE && v != NULL) + *v = 0; + return tok == MJSON_TOK_TRUE || tok == MJSON_TOK_FALSE ? 1 : 0; +} + +static unsigned char +unhex(unsigned char c) +{ + return (c >= '0' && c <= '9') ? (unsigned char)(c - '0') + : (c >= 'A' && c <= 'F') ? (unsigned char)(c - '7') + : (unsigned char)(c - 'W'); +} + +static unsigned char +mjson_unhex_nimble(const char* s) +{ + const unsigned char* u = (const unsigned char*)s; + return (unsigned char)(((unsigned char)(unhex(u[0]) << 4)) | unhex(u[1])); +} + +// n is at most? +static int +mjson_unescape(const char* s, int len, char* to, int n) +{ + int i, j; + for (i = 0, j = 0; i < len && j < n; i++, j++) { + if (s[i] == '\\' && i + 5 < len && s[i + 1] == 'u') { + // \uXXXX escape. We could process a simple one-byte chars + // \u00xx from the ASCII range. More complex chars would require + // dragging in a UTF8 library, which is too much for us + if (s[i + 2] != '0' || s[i + 3] != '0') + return -1; // Too much, give up + ((unsigned char*)to)[j] = mjson_unhex_nimble(s + i + 4); + i += 5; + } else if (s[i] == '\\' && i + 1 < len) { + int c = mjson_esc(s[i + 1], 0); + if (c == 0) + return -1; + to[j] = (char)(unsigned char)c; + i++; + } else { + to[j] = s[i]; + } + } + if (j >= n) + return -1; + if (n > 0) + to[j] = '\0'; + return j; +} + +int +mjson_get_string(const char* s, int len, const char* path, char* to, int n) +{ + const char* p; + int sz; + if (mjson_find(s, len, path, &p, &sz) != MJSON_TOK_STRING) + return -1; + return mjson_unescape(p + 1, sz - 2, to, n); +} + +int +mjson_get_hex(const char* s, int len, const char* x, char* to, int n) +{ + const char* p; + int i, j, sz; + if (mjson_find(s, len, x, &p, &sz) != MJSON_TOK_STRING) + return -1; + for (i = j = 0; i < sz - 3 && j < n; i += 2, j++) { + ((unsigned char*)to)[j] = mjson_unhex_nimble(p + i + 1); + } + if (j < n) + to[j] = '\0'; + return j; +} + +#if MJSON_ENABLE_BASE64 +static unsigned char +mjson_base64rev(int c) +{ + if (c >= 'A' && c <= 'Z') { + return (unsigned char)(c - 'A'); + } else if (c >= 'a' && c <= 'z') { + return (unsigned char)(c + 26 - 'a'); + } else if (c >= '0' && c <= '9') { + return (unsigned char)(c + 52 - '0'); + } else if (c == '+') { + return 62; + } else if (c == '/') { + return 63; + } else { + return 64; + } +} + +int +mjson_base64_dec(const char* src, int n, char* dst, int dlen) +{ + const char* end = src + n; + int len = 0; + while (src + 3 < end && len < dlen) { + unsigned char a = mjson_base64rev(src[0]), b = mjson_base64rev(src[1]), + c = mjson_base64rev(src[2]), d = mjson_base64rev(src[3]); + dst[len++] = (char)(unsigned char)((a << 2) | (b >> 4)); + if (src[2] != '=' && len < dlen) { + dst[len++] = (char)(unsigned char)((b << 4) | (c >> 2)); + if (src[3] != '=' && len < dlen) { + dst[len++] = (char)(unsigned char)((c << 6) | d); + } + } + src += 4; + } + if (len < dlen) + dst[len] = '\0'; + return len; +} + +int +mjson_get_base64(const char* s, int len, const char* path, char* to, int n) +{ + const char* p; + int sz; + if (mjson_find(s, len, path, &p, &sz) != MJSON_TOK_STRING) + return 0; + return mjson_base64_dec(p + 1, sz - 2, to, n); +} +#endif // MJSON_ENABLE_BASE64 + +#if MJSON_ENABLE_NEXT +struct nextdata +{ + int off, len, depth, t, vo, arrayindex; + int *koff, *klen, *voff, *vlen, *vtype; +}; + +static int +next_cb(int tok, const char* s, int off, int len, void* ud) +{ + struct nextdata* d = (struct nextdata*)ud; + // int i; + switch (tok) { + case '{': + case '[': + if (d->depth == 0 && tok == '[') + d->arrayindex = 0; + if (d->depth == 1 && off > d->off) { + d->vo = off; + d->t = tok == '{' ? MJSON_TOK_OBJECT : MJSON_TOK_ARRAY; + if (d->voff) + *d->voff = off; + if (d->vtype) + *d->vtype = d->t; + } + d->depth++; + break; + case '}': + case ']': + d->depth--; + if (d->depth == 1 && d->vo) { + d->len = off + len; + if (d->vlen) + *d->vlen = d->len - d->vo; + if (d->arrayindex >= 0) { + if (d->koff) + *d->koff = d->arrayindex; // koff holds array index + if (d->klen) + *d->klen = 0; // klen holds 0 + } + return 1; + } + if (d->depth == 1 && d->arrayindex >= 0) + d->arrayindex++; + break; + case ',': + case ':': + break; + case MJSON_TOK_KEY: + if (d->depth == 1 && d->off < off) { + if (d->koff) + *d->koff = off; // And report back to the user + if (d->klen) + *d->klen = len; // If we have to + } + break; + default: + if (d->depth != 1) + break; + // If we're iterating over the array + if (off > d->off) { + d->len = off + len; + if (d->vlen) + *d->vlen = len; // value length + if (d->voff) + *d->voff = off; // value offset + if (d->vtype) + *d->vtype = tok; // value type + if (d->arrayindex >= 0) { + if (d->koff) + *d->koff = d->arrayindex; // koff holds array index + if (d->klen) + *d->klen = 0; // klen holds 0 + } + return 1; + } + if (d->arrayindex >= 0) + d->arrayindex++; + break; + } + (void)s; + return 0; +} + +int +mjson_next(const char* s, + int n, + int off, + int* koff, + int* klen, + int* voff, + int* vlen, + int* vtype) +{ + struct nextdata d = { off, 0, 0, 0, 0, -1, koff, klen, voff, vlen, vtype }; + mjson(s, n, next_cb, &d); + return d.len; +} +#endif + +#if MJSON_ENABLE_PRINT +int +mjson_print_fixed_buf(const char* ptr, int len, void* fn_data) +{ + struct mjson_fixedbuf* fb = (struct mjson_fixedbuf*)fn_data; + int i, left = fb->size - fb->len; + /* int i, left = fb->size - 1 - fb->len; */ + if (left < len) + len = left; + for (i = 0; i < len; i++) + fb->ptr[fb->len + i] = ptr[i]; + fb->len += len; + /* fb->ptr[fb->len] = '\0'; */ + return len; +} + +// This function allocates memory in chunks of size MJSON_DYNBUF_CHUNK +// to decrease memory fragmentation, when many calls are executed to +// print e.g. a base64 string or a hex string. +int +mjson_print_dynamic_buf(const char* ptr, int len, void* fn_data) +{ + char *s, *buf = *(char**)fn_data; + size_t curlen = buf == NULL ? 0 : strlen(buf); + size_t new_size = curlen + (size_t)len + 1 + MJSON_DYNBUF_CHUNK; + new_size -= new_size % MJSON_DYNBUF_CHUNK; + + if ((s = (char*)MJSON_REALLOC(buf, new_size)) == NULL) { + return 0; + } else { + memcpy(s + curlen, ptr, (size_t)len); + s[curlen + (size_t)len] = '\0'; + *(char**)fn_data = s; + return len; + } +} + +int +mjson_snprintf(char* buf, size_t len, const char* fmt, ...) +{ + va_list ap; + struct mjson_fixedbuf fb = { buf, (int)len, 0 }; + va_start(ap, fmt); + mjson_vprintf(mjson_print_fixed_buf, &fb, fmt, &ap); + va_end(ap); + return fb.len; +} + +char* +mjson_aprintf(const char* fmt, ...) +{ + va_list ap; + char* result = NULL; + va_start(ap, fmt); + mjson_vprintf(mjson_print_dynamic_buf, &result, fmt, &ap); + va_end(ap); + return result; +} + +int +mjson_print_null(const char* ptr, int len, void* userdata) +{ + (void)ptr; + (void)userdata; + return len; +} + +int +mjson_print_buf(mjson_print_fn_t fn, void* fnd, const char* buf, int len) +{ + return fn(buf, len, fnd); +} + +int +mjson_print_long(mjson_print_fn_t fn, void* fnd, long val, int is_signed) +{ + unsigned long v = (unsigned long)val, s = 0, n, i; + char buf[20], t; + if (is_signed && val < 0) + buf[s++] = '-', v = (unsigned long)(-val); + // This loop prints a number in reverse order. I guess this is because we + // write numbers from right to left: least significant digit comes last. + // Maybe because we use Arabic numbers, and Arabs write RTL? + for (n = 0; v > 0; v /= 10) + buf[s + n++] = "0123456789"[v % 10]; + // Reverse a string + for (i = 0; i < n / 2; i++) + t = buf[s + i], buf[s + i] = buf[s + n - i - 1], buf[s + n - i - 1] = t; + if (val == 0) + buf[n++] = '0'; // Handle special case + return fn(buf, (int)(s + n), fnd); +} + +int +mjson_print_int(mjson_print_fn_t fn, void* fnd, int v, int s) +{ + return mjson_print_long(fn, fnd, s ? (long)v : (long)(unsigned)v, s); +} + +static int +addexp(char* buf, int e, int sign) +{ + int n = 0; + buf[n++] = 'e'; + buf[n++] = (char)sign; + if (e > 400) + return 0; + if (e < 10) + buf[n++] = '0'; + if (e >= 100) + buf[n++] = (char)(e / 100 + '0'), e -= 100 * (e / 100); + if (e >= 10) + buf[n++] = (char)(e / 10 + '0'), e -= 10 * (e / 10); + buf[n++] = (char)(e + '0'); + return n; +} + +int +mjson_print_dbl(mjson_print_fn_t fn, void* fnd, double d, int width) +{ + char buf[40]; + int i, s = 0, n = 0, e = 0; + double t, mul, saved; + if (d == 0.0) + return fn("0", 1, fnd); + if (isinf(d)) + return fn(d > 0 ? "inf" : "-inf", d > 0 ? 3 : 4, fnd); + if (isnan(d)) + return fn("nan", 3, fnd); + if (d < 0.0) + d = -d, buf[s++] = '-'; + + // Round + saved = d; + mul = 1.0; + while (d >= 10.0 && d / mul >= 10.0) + mul *= 10.0; + while (d <= 1.0 && d / mul <= 1.0) + mul /= 10.0; + for (i = 0, t = mul * 5; i < width; i++) + t /= 10.0; + d += t; + // Calculate exponent, and 'mul' for scientific representation + mul = 1.0; + while (d >= 10.0 && d / mul >= 10.0) + mul *= 10.0, e++; + while (d < 1.0 && d / mul < 1.0) + mul /= 10.0, e--; + // printf(" --> %g %d %g %g\n", saved, e, t, mul); + + if (e >= width) { + struct mjson_fixedbuf fb = { buf + s, (int)sizeof(buf) - s, 0 }; + n = mjson_print_dbl(mjson_print_fixed_buf, &fb, saved / mul, width); + // printf(" --> %.*g %d [%.*s]\n", 10, d / t, e, fb.len, fb.ptr); + n += addexp(buf + s + n, e, '+'); + return fn(buf, s + n, fnd); + } else if (e <= -width) { + struct mjson_fixedbuf fb = { buf + s, (int)sizeof(buf) - s, 0 }; + n = mjson_print_dbl(mjson_print_fixed_buf, &fb, saved / mul, width); + // printf(" --> %.*g %d [%.*s]\n", 10, d / mul, e, fb.len, fb.ptr); + n += addexp(buf + s + n, -e, '-'); + return fn(buf, s + n, fnd); + } else { + for (i = 0, t = mul; t >= 1.0 && s + n < (int)sizeof(buf); i++) { + int ch = (int)(d / t); + if (n > 0 || ch > 0) + buf[s + n++] = (char)(ch + '0'); + d -= ch * t; + t /= 10.0; + } + // printf(" --> [%g] -> %g %g (%d) [%.*s]\n", saved, d, t, n, s + n, buf); + if (n == 0) + buf[s++] = '0'; + while (t >= 1.0 && n + s < (int)sizeof(buf)) + buf[n++] = '0', t /= 10.0; + if (s + n < (int)sizeof(buf)) + buf[n + s++] = '.'; + // printf(" 1--> [%g] -> [%.*s]\n", saved, s + n, buf); + for (i = 0, t = 0.1; s + n < (int)sizeof(buf) && n < width; i++) { + int ch = (int)(d / t); + buf[s + n++] = (char)(ch + '0'); + d -= ch * t; + t /= 10.0; + } + } + while (n > 0 && buf[s + n - 1] == '0') + n--; // Trim trailing zeros + if (n > 0 && buf[s + n - 1] == '.') + n--; // Trim trailing dot + buf[s + n] = '\0'; + return fn(buf, s + n, fnd); +} + +int +mjson_print_str(mjson_print_fn_t fn, void* fnd, const char* s, int len) +{ + int i, n = fn("\"", 1, fnd); + for (i = 0; i < len; i++) { + char c = (char)(unsigned char)mjson_escape(s[i]); + if (c) { + n += fn("\\", 1, fnd); + n += fn(&c, 1, fnd); + } else { + n += fn(&s[i], 1, fnd); + } + } + return n + fn("\"", 1, fnd); +} + +#if MJSON_ENABLE_BASE64 +int +mjson_print_b64(mjson_print_fn_t fn, void* fnd, const unsigned char* s, int n) +{ + const char* t = + "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; + int i, len = fn("\"", 1, fnd); + for (i = 0; i < n; i += 3) { + int a = s[i], b = i + 1 < n ? s[i + 1] : 0, c = i + 2 < n ? s[i + 2] : 0; + char buf[4] = { t[a >> 2], t[(a & 3) << 4 | (b >> 4)], '=', '=' }; + if (i + 1 < n) + buf[2] = t[(b & 15) << 2 | (c >> 6)]; + if (i + 2 < n) + buf[3] = t[c & 63]; + len += fn(buf, sizeof(buf), fnd); + } + return len + fn("\"", 1, fnd); +} +#endif /* MJSON_ENABLE_BASE64 */ + +int +mjson_vprintf(mjson_print_fn_t fn, void* fnd, const char* fmt, va_list* ap) +{ + int i = 0, n = 0; + while (fmt[i] != '\0') { + if (fmt[i] == '%') { + char fc = fmt[++i]; + int is_long = 0; + if (fc == 'l') { + is_long = 1; + fc = fmt[i + 1]; + } + if (fc == 'Q') { + char* buf = va_arg(*ap, char*); + n += + mjson_print_str(fn, fnd, buf ? buf : "", buf ? (int)strlen(buf) : 0); + } else if (strncmp(&fmt[i], ".*Q", 3) == 0) { + int len = va_arg(*ap, int); + char* buf = va_arg(*ap, char*); + n += mjson_print_str(fn, fnd, buf, len); + i += 2; + } else if (fc == 'd' || fc == 'u') { + int is_signed = (fc == 'd'); + if (is_long) { + long val = va_arg(*ap, long); + n += mjson_print_long(fn, fnd, val, is_signed); + i++; + } else { + int val = va_arg(*ap, int); + n += mjson_print_int(fn, fnd, val, is_signed); + } + } else if (fc == 'B') { + const char* s = va_arg(*ap, int) ? "true" : "false"; + n += mjson_print_buf(fn, fnd, s, (int)strlen(s)); + } else if (fc == 's') { + char* buf = va_arg(*ap, char*); + n += mjson_print_buf(fn, fnd, buf, (int)strlen(buf)); + } else if (strncmp(&fmt[i], ".*s", 3) == 0) { + int len = va_arg(*ap, int); + char* buf = va_arg(*ap, char*); + n += mjson_print_buf(fn, fnd, buf, len); + i += 2; + } else if (fc == 'g') { + n += mjson_print_dbl(fn, fnd, va_arg(*ap, double), 6); + } else if (strncmp(&fmt[i], ".*g", 3) == 0) { + int width = va_arg(*ap, int); + n += mjson_print_dbl(fn, fnd, va_arg(*ap, double), width); + i += 2; +#if MJSON_ENABLE_BASE64 + } else if (fc == 'V') { + int len = va_arg(*ap, int); + const char* buf = va_arg(*ap, const char*); + n += mjson_print_b64(fn, fnd, (const unsigned char*)buf, len); +#endif + } else if (fc == 'H') { + const char* hex = "0123456789abcdef"; + int j, len = va_arg(*ap, int); + const unsigned char* p = va_arg(*ap, const unsigned char*); + n += fn("\"", 1, fnd); + for (j = 0; j < len; j++) { + n += fn(&hex[(p[j] >> 4) & 15], 1, fnd); + n += fn(&hex[p[j] & 15], 1, fnd); + } + n += fn("\"", 1, fnd); + } else if (fc == 'M') { + mjson_vprint_fn_t vfn = va_arg(*ap, mjson_vprint_fn_t); + n += vfn(fn, fnd, ap); + } + i++; + } else { + n += mjson_print_buf(fn, fnd, &fmt[i++], 1); + } + } + return n; +} + +int +mjson_printf(mjson_print_fn_t fn, void* fnd, const char* fmt, ...) +{ + va_list ap; + int len; + va_start(ap, fmt); + len = mjson_vprintf(fn, fnd, fmt, &ap); + va_end(ap); + return len; +} +#endif /* MJSON_ENABLE_PRINT */ + +static int +is_digit(int c) +{ + return c >= '0' && c <= '9'; +} + +/* NOTE: strtod() implementation by Yasuhiro Matsumoto. */ +static double +mystrtod(const char* str, const char** end) +{ + double d = 0.0; + int sign = 1, n = 0; + const char *p = str, *a = str; + + /* decimal part */ + if (*p == '-') { + sign = -1; + ++p; + } else if (*p == '+') { + ++p; + } + if (is_digit(*p)) { + d = (double)(*p++ - '0'); + while (*p && is_digit(*p)) { + d = d * 10.0 + (double)(*p - '0'); + ++p; + ++n; + } + a = p; + } else if (*p != '.') { + goto done; + } + d *= sign; + + /* fraction part */ + if (*p == '.') { + double f = 0.0; + double base = 0.1; + ++p; + + if (is_digit(*p)) { + while (*p && is_digit(*p)) { + f += base * (*p - '0'); + base /= 10.0; + ++p; + ++n; + } + } + d += f * sign; + a = p; + } + + /* exponential part */ + if ((*p == 'E') || (*p == 'e')) { + int i, e = 0, neg = 0; + p++; + if (*p == '-') + p++, neg++; + if (*p == '+') + p++; + while (is_digit(*p)) + e = e * 10 + *p++ - '0'; + if (neg) + e = -e; +#if 0 + if (d == 2.2250738585072011 && e == -308) { + d = 0.0; + a = p; + goto done; + } + if (d == 2.2250738585072012 && e <= -308) { + d *= 1.0e-308; + a = p; + goto done; + } +#endif + for (i = 0; i < e; i++) + d *= 10; + for (i = 0; i < -e; i++) + d /= 10; + a = p; + } else if (p > str && !is_digit(*(p - 1))) { + a = str; + goto done; + } + +done: + if (end) + *end = a; + return d; +} + +#if MJSON_ENABLE_MERGE +int +mjson_merge(const char* s, + int n, + const char* s2, + int n2, + mjson_print_fn_t fn, + void* userdata) +{ + int koff, klen, voff, vlen, t, t2, k, off = 0, len = 0, comma = 0; + if (n < 2) + return len; + len += fn("{", 1, userdata); + while ((off = mjson_next(s, n, off, &koff, &klen, &voff, &vlen, &t)) != 0) { + char* path = (char*)alloca((size_t)klen + 1); + const char* val; + memcpy(path, "$.", 2); + memcpy(path + 2, s + koff + 1, (size_t)(klen - 2)); + path[klen] = '\0'; + if ((t2 = mjson_find(s2, n2, path, &val, &k)) != MJSON_TOK_INVALID) { + if (t2 == MJSON_TOK_NULL) + continue; // null deletes the key + } else { + val = s + voff; // Key is not found in the update. Copy the old value. + } + if (comma) + len += fn(",", 1, userdata); + len += fn(s + koff, klen, userdata); + len += fn(":", 1, userdata); + if (t == MJSON_TOK_OBJECT && t2 == MJSON_TOK_OBJECT) { + len += mjson_merge(s + voff, vlen, val, k, fn, userdata); + } else { + if (t2 != MJSON_TOK_INVALID) + vlen = k; + len += fn(val, vlen, userdata); + } + comma = 1; + } + // Add missing keys + off = 0; + while ((off = mjson_next(s2, n2, off, &koff, &klen, &voff, &vlen, &t)) != 0) { + char* path = (char*)alloca((size_t)klen + 1); + const char* val; + if (t == MJSON_TOK_NULL) + continue; + memcpy(path, "$.", 2); + memcpy(path + 2, s2 + koff + 1, (size_t)(klen - 2)); + path[klen] = '\0'; + if (mjson_find(s, n, path, &val, &vlen) != MJSON_TOK_INVALID) + continue; + if (comma) + len += fn(",", 1, userdata); + len += fn(s2 + koff, klen, userdata); + len += fn(":", 1, userdata); + len += fn(s2 + voff, vlen, userdata); + comma = 1; + } + len += fn("}", 1, userdata); + return len; +} +#endif // MJSON_ENABLE_MERGE + +#if MJSON_ENABLE_PRETTY +struct prettydata +{ + int level; + int len; + int prev; + const char* pad; + int padlen; + mjson_print_fn_t fn; + void* userdata; +}; + +static int +pretty_cb(int ev, const char* s, int off, int len, void* ud) +{ + struct prettydata* d = (struct prettydata*)ud; + int i; + switch (ev) { + case '{': + case '[': + d->level++; + d->len += d->fn(s + off, len, d->userdata); + break; + case '}': + case ']': + d->level--; + if (d->prev != '[' && d->prev != '{' && d->padlen > 0) { + d->len += d->fn("\n", 1, d->userdata); + for (i = 0; i < d->level; i++) + d->len += d->fn(d->pad, d->padlen, d->userdata); + } + d->len += d->fn(s + off, len, d->userdata); + break; + case ',': + d->len += d->fn(s + off, len, d->userdata); + if (d->padlen > 0) { + d->len += d->fn("\n", 1, d->userdata); + for (i = 0; i < d->level; i++) + d->len += d->fn(d->pad, d->padlen, d->userdata); + } + break; + case ':': + d->len += d->fn(s + off, len, d->userdata); + if (d->padlen > 0) + d->len += d->fn(" ", 1, d->userdata); + break; + case MJSON_TOK_KEY: + if (d->prev == '{' && d->padlen > 0) { + d->len += d->fn("\n", 1, d->userdata); + for (i = 0; i < d->level; i++) + d->len += d->fn(d->pad, d->padlen, d->userdata); + } + d->len += d->fn(s + off, len, d->userdata); + break; + default: + if (d->prev == '[' && d->padlen > 0) { + d->len += d->fn("\n", 1, d->userdata); + for (i = 0; i < d->level; i++) + d->len += d->fn(d->pad, d->padlen, d->userdata); + } + d->len += d->fn(s + off, len, d->userdata); + break; + } + d->prev = ev; + return 0; +} + +int +mjson_pretty(const char* s, + int n, + const char* pad, + mjson_print_fn_t fn, + void* userdata) +{ + struct prettydata d = { 0, 0, 0, pad, (int)strlen(pad), fn, userdata }; + if (mjson(s, n, pretty_cb, &d) < 0) + return -1; + return d.len; +} +#endif // MJSON_ENABLE_PRETTY + +#if MJSON_ENABLE_RPC +struct jsonrpc_ctx jsonrpc_default_context; + +int +mjson_globmatch(const char* s1, int n1, const char* s2, int n2) +{ + int i = 0, j = 0, ni = 0, nj = 0; + while (i < n1 || j < n2) { + if (i < n1 && j < n2 && (s1[i] == '?' || s2[j] == s1[i])) { + i++, j++; + } else if (i < n1 && (s1[i] == '*' || s1[i] == '#')) { + ni = i, nj = j + 1, i++; + } else if (nj > 0 && nj <= n2 && (s1[i - 1] == '#' || s2[j] != '/')) { + i = ni, j = nj; + } else { + return 0; + } + } + return 1; +} + +void +jsonrpc_return_errorv(struct jsonrpc_request* r, + int code, + const char* message, + const char* data_fmt, + va_list* ap) +{ + mjson_printf(r->fn, r->fn_data, "{\"jsonrpc\":\"2.0\",\"id\":"); + if (r->id_len > 0 && r->id != NULL) { + mjson_printf(r->fn, r->fn_data, "%.*s", r->id_len, r->id); + } else { + mjson_printf(r->fn, r->fn_data, "null"); + } + mjson_printf(r->fn, + r->fn_data, + ",\"error\":{\"code\":%d,\"message\":%Q", + code, + message == NULL ? "" : message); + if (data_fmt != NULL) { + mjson_printf(r->fn, r->fn_data, ",\"data\":"); + mjson_vprintf(r->fn, r->fn_data, data_fmt, ap); + } + // LF NEWLINE NL + mjson_printf(r->fn, r->fn_data, "}}\n"); + /* mjson_printf(r->fn, r->fn_data, "}}"); */ +} + +void +jsonrpc_return_error(struct jsonrpc_request* r, + int code, + const char* message, + const char* data_fmt, + ...) +{ + va_list ap; + va_start(ap, data_fmt); + jsonrpc_return_errorv(r, code, message, data_fmt, &ap); + va_end(ap); +} + +void +jsonrpc_return_successv(struct jsonrpc_request* r, + const char* result_fmt, + va_list* ap) +{ + if (r->id_len == 0) + return; + mjson_printf(r->fn, r->fn_data, "{\"jsonrpc\":\"2.0\",\"id\":%.*s,\"result\":", r->id_len, r->id); + if (result_fmt != NULL) { + mjson_vprintf(r->fn, r->fn_data, result_fmt, ap); + } else { + mjson_printf(r->fn, r->fn_data, "%s", "null"); + } + // LF NEWLINE NL + mjson_printf(r->fn, r->fn_data, "}"); +} + + +void +jsonrpc_return_successv2(struct jsonrpc_request* r, bool add_endl, + const char* result_fmt, + va_list* ap) +{ + if (r->id_len == 0) + return; // TODO assert + mjson_printf(r->fn, r->fn_data, "{\"jsonrpc\":\"2.0\",\"id\":%.*s,\"result\":", r->id_len, r->id); + if (result_fmt != NULL) { + mjson_vprintf(r->fn, r->fn_data, result_fmt, ap); + } else { + mjson_printf(r->fn, r->fn_data, "%s", "null"); + } + // LF NEWLINE NL + mjson_printf (r->fn, r->fn_data, "}"); + if (add_endl) + { + mjson_printf(r->fn, r->fn_data, "\n"); + } +} + +void +jsonrpc_return_success2(struct jsonrpc_request* r, bool add_endl, const char* result_fmt, ...) +{ + va_list ap; + va_start(ap, result_fmt); + jsonrpc_return_successv2(r, add_endl, result_fmt, &ap); + va_end(ap); +} + + +void +jsonrpc_return_success(struct jsonrpc_request* r, const char* result_fmt, ...) +{ + va_list ap; + va_start(ap, result_fmt); + jsonrpc_return_successv(r, result_fmt, &ap); + va_end(ap); +} + +void +jsonrpc_ctx_process(struct jsonrpc_ctx* ctx, + const char* buf, + int len, + mjson_print_fn_t fn, + void* fn_data, + void* ud) +{ + const char *result = NULL, *error = NULL; + int result_sz = 0, error_sz = 0; + struct jsonrpc_method* m = NULL; + struct jsonrpc_request r = { + ctx, buf, len, 0, 0, 0, 0, 0, 0, fn, fn_data, ud + }; + + // Is is a response frame? + mjson_find(buf, len, "$.result", &result, &result_sz); + if (result == NULL) + mjson_find(buf, len, "$.error", &error, &error_sz); + if (result_sz > 0 || error_sz > 0) { + if (ctx->response_cb) + ctx->response_cb(buf, len, ctx->response_cb_data); + return; + } + + // Method must exist and must be a string + if (mjson_find(buf, len, "$.method", &r.method, &r.method_len) != + MJSON_TOK_STRING) { + mjson_printf(fn, + fn_data, + "{\"error\":{\"code\":-32700,\"message\":%.*Q}}\n", + len, + buf); + return; + } + + // id and params are optional + mjson_find(buf, len, "$.id", &r.id, &r.id_len); + mjson_find(buf, len, "$.params", &r.params, &r.params_len); + + for (m = ctx->methods; m != NULL; m = m->next) { + if (mjson_globmatch( + m->method, m->method_sz, r.method + 1, r.method_len - 2) > 0) { + if (r.params == NULL) + r.params = ""; + m->cb(&r); + break; + } + } + if (m == NULL) { + jsonrpc_return_error(&r, JSONRPC_ERROR_NOT_FOUND, "method not found", NULL); + } +} + +static int +jsonrpc_print_methods(mjson_print_fn_t fn, void* fn_data, va_list* ap) +{ + struct jsonrpc_ctx* ctx = va_arg(*ap, struct jsonrpc_ctx*); + struct jsonrpc_method* m; + int len = 0; + for (m = ctx->methods; m != NULL; m = m->next) { + if (m != ctx->methods) + len += mjson_print_buf(fn, fn_data, ",", 1); + len += mjson_print_str(fn, fn_data, m->method, (int)strlen(m->method)); + } + return len; +} + +void +jsonrpc_list(struct jsonrpc_request* r) +{ + jsonrpc_return_success(r, "[%M]", jsonrpc_print_methods, r->ctx); +} + +void +jsonrpc_ctx_init(struct jsonrpc_ctx* ctx, + mjson_print_fn_t response_cb, + void* response_cb_data) +{ + ctx->methods = NULL; + ctx->response_cb = response_cb; + ctx->response_cb_data = response_cb_data; +} + +void +jsonrpc_init(mjson_print_fn_t response_cb, void* userdata) +{ + struct jsonrpc_ctx* ctx = &jsonrpc_default_context; + jsonrpc_ctx_init(ctx, response_cb, userdata); + jsonrpc_ctx_export(ctx, MJSON_RPC_LIST_NAME, jsonrpc_list); +} +#endif // MJSON_ENABLE_RPC diff --git a/third_party/mcpc/src/mjson.h b/third_party/mcpc/src/mjson.h new file mode 100644 index 0000000..7a562f0 --- /dev/null +++ b/third_party/mcpc/src/mjson.h @@ -0,0 +1,296 @@ +// Copyright (c) 2018-2020 Cesanta Software Limited +// 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. + +#ifndef MJSON_H +#define MJSON_H + +#include <stdarg.h> +#include <stdlib.h> +#include <string.h> + +#include <mcpc/_c23_uchar.h> +#include <mcpc/_c23_keywords.h> +#include <stdint.h> + +#ifndef MJSON_ENABLE_PRINT +#define MJSON_ENABLE_PRINT 1 +#endif + +#ifndef MJSON_ENABLE_RPC +#define MJSON_ENABLE_RPC 1 +#endif + +#ifndef MJSON_ENABLE_BASE64 +#define MJSON_ENABLE_BASE64 1 +#endif + +#ifndef MJSON_ENABLE_MERGE +#define MJSON_ENABLE_MERGE 1 +#endif + +#ifndef MJSON_ENABLE_PRETTY +#define MJSON_ENABLE_PRETTY 1 +#endif + +#ifndef MJSON_ENABLE_NEXT +#define MJSON_ENABLE_NEXT 1 +#endif + +#ifndef MJSON_RPC_LIST_NAME +#define MJSON_RPC_LIST_NAME "rpc.list" +#endif + +#ifndef MJSON_DYNBUF_CHUNK +#define MJSON_DYNBUF_CHUNK 256 // Allocation granularity for print_dynamic_buf +#endif + +#ifndef MJSON_REALLOC +#define MJSON_REALLOC realloc +#endif + +#ifdef __cplusplus +extern "C" +{ +#endif + +#define MJSON_ERROR_INVALID_INPUT (-1) +#define MJSON_ERROR_TOO_DEEP (-2) +#define MJSON_TOK_INVALID 0 +#define MJSON_TOK_KEY 1 +#define MJSON_TOK_STRING 11 +#define MJSON_TOK_NUMBER 12 +#define MJSON_TOK_TRUE 13 +#define MJSON_TOK_FALSE 14 +#define MJSON_TOK_NULL 15 +#define MJSON_TOK_ARRAY 91 +#define MJSON_TOK_OBJECT 123 +#define MJSON_TOK_IS_VALUE(t) ((t) > 10 && (t) < 20) + + typedef int (*mjson_cb_t)(int event, + const char* buf, + int offset, + int len, + void* fn_data); + +#ifndef MJSON_MAX_DEPTH +#define MJSON_MAX_DEPTH 20 +#endif + + int mjson(const char* buf, int len, mjson_cb_t cb, void* ud); + int mjson_find(const char* buf, + int len, + const char* jp, + const char** tp, + int* tl); + int32_t mjson_find8(const char8_t* buf, + int32_t len, + const char8_t* jp, + const char8_t** tp, + int32_t* tl); + int32_t mjson_find88(const char* buf, + int32_t len, + const char8_t* jp, + const char8_t** tp, + int32_t* tl); + int32_t mjson_find88(const char* buf, + int32_t len, + const char8_t* jp, + const char8_t** tp, + int32_t* tl); + int mjson_get_number(const char* buf, int len, const char* path, double* v); + int mjson_get_bool(const char* buf, int len, const char* path, int* v); + int mjson_get_string(const char* buf, + int len, + const char* path, + char* to, + int n); + int mjson_get_hex(const char* buf, + int len, + const char* path, + char* to, + int n); + +#if MJSON_ENABLE_NEXT + int mjson_next(const char* buf, + int len, + int offset, + int* key_offset, + int* key_len, + int* val_offset, + int* val_len, + int* vale_type); +#endif + +#if MJSON_ENABLE_BASE64 + int mjson_get_base64(const char* buf, + int len, + const char* path, + char* dst, + int dst_len); + int mjson_base64_dec(const char* src, int src_len, char* dst, int dst_len); +#endif + +#if MJSON_ENABLE_PRINT + typedef int (*mjson_print_fn_t)(const char* buf, int len, void* fn_data); + typedef int (*mjson_vprint_fn_t)(mjson_print_fn_t fn, void*, va_list*); + + struct mjson_fixedbuf + { + char* ptr; + int size, len; + }; + + int mjson_printf(mjson_print_fn_t fn, void* fn_data, const char* fmt, ...); + int mjson_vprintf(mjson_print_fn_t fn, + void* fn_data, + const char* fmt, + va_list* ap); + int mjson_print_str(mjson_print_fn_t fn, + void* fn_data, + const char* buf, + int len); + int mjson_print_int(mjson_print_fn_t fn, + void* fn_data, + int value, + int is_signed); + int mjson_print_long(mjson_print_fn_t fn, + void* fn_data, + long value, + int is_signed); + int mjson_print_buf(mjson_print_fn_t fn, + void* fn_data, + const char* buf, + int len); + int mjson_print_dbl(mjson_print_fn_t fn, void* fn_data, double d, int width); + + int mjson_print_null(const char* ptr, int len, void* fn_data); + int mjson_print_fixed_buf(const char* ptr, int len, void* fn_data); + int mjson_print_dynamic_buf(const char* ptr, int len, void* fn_data); + + int mjson_snprintf(char* buf, size_t len, const char* fmt, ...); + char* mjson_aprintf(const char* fmt, ...); + +#if MJSON_ENABLE_PRETTY + int mjson_pretty(const char* s, + int n, + const char* pad, + mjson_print_fn_t fn, + void* fn_data); +#endif + +#if MJSON_ENABLE_MERGE + int mjson_merge(const char* s, + int n, + const char* s2, + int n2, + mjson_print_fn_t fn, + void* fn_data); +#endif + +#endif // MJSON_ENABLE_PRINT + +#if MJSON_ENABLE_RPC + + void jsonrpc_init(mjson_print_fn_t response_cb, void* fn_data); + int mjson_globmatch(const char* s1, int n1, const char* s2, int n2); + + struct jsonrpc_request + { + struct jsonrpc_ctx* ctx; + const char* frame; // Points to the whole frame + int frame_len; // Frame length + const char* params; // Points to the "params" in the request frame + int params_len; // Length of the "params" + const char* id; // Points to the "id" in the request frame + int id_len; // Length of the "id" + const char* method; // Points to the "method" in the request frame + int method_len; // Length of the "method" + mjson_print_fn_t fn; // Printer function + void* fn_data; // Printer function data + void* userdata; // Callback's user data as specified at export time + }; + + struct jsonrpc_method + { + const char* method; + int method_sz; + void (*cb)(struct jsonrpc_request*); + struct jsonrpc_method* next; + }; + + // Main RPC context, stores current request information and a list of + // exported RPC methods. + struct jsonrpc_ctx + { + struct jsonrpc_method* methods; + mjson_print_fn_t response_cb; + void* response_cb_data; + }; + +// Registers function fn under the given name within the given RPC context +#define jsonrpc_ctx_export(ctx, name, fn) \ + do { \ + static struct jsonrpc_method m = { (name), sizeof(name) - 1, (fn), 0 }; \ + m.next = (ctx)->methods; \ + (ctx)->methods = &m; \ + } while (0) + + void jsonrpc_ctx_init(struct jsonrpc_ctx* ctx, + mjson_print_fn_t response_cb, + void* response_cb_data); + void jsonrpc_return_error(struct jsonrpc_request* r, + int code, + const char* message, + const char* data_fmt, + ...); + void jsonrpc_return_success(struct jsonrpc_request* r, + const char* result_fmt, + ...); + void jsonrpc_return_success2(struct jsonrpc_request* r, bool add_endl, + const char* result_fmt, + ...); + void jsonrpc_ctx_process(struct jsonrpc_ctx* ctx, + const char* req, + int req_sz, + mjson_print_fn_t fn, + void* fn_data, + void* userdata); + + extern struct jsonrpc_ctx jsonrpc_default_context; + extern void jsonrpc_list(struct jsonrpc_request* r); + +#define jsonrpc_export(name, fn) \ + jsonrpc_ctx_export(&jsonrpc_default_context, (name), (fn)) + +#define jsonrpc_process(buf, len, fn, fnd, ud) \ + jsonrpc_ctx_process(&jsonrpc_default_context, (buf), (len), (fn), (fnd), (ud)) + +#define JSONRPC_ERROR_INVALID -32700 /* Invalid JSON was received */ +#define JSONRPC_ERROR_INVALID_REQUEST -32600 /* The JSON-RPC request is invalid */ +#define JSONRPC_ERROR_NOT_FOUND -32601 /* The method does not exist */ +#define JSONRPC_ERROR_BAD_PARAMS -32602 /* Invalid params passed */ +#define JSONRPC_ERROR_INTERNAL -32603 /* Internal JSON-RPC error */ + +#endif // MJSON_ENABLE_RPC +#ifdef __cplusplus +} +#endif +#endif // MJSON_H diff --git a/third_party/mcpc/src/prmpt.c b/third_party/mcpc/src/prmpt.c new file mode 100644 index 0000000..324e5dd --- /dev/null +++ b/third_party/mcpc/src/prmpt.c @@ -0,0 +1,378 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "prmpt.h" + +#include "ext_string.h" + +#include <stdio.h> +#include <stdarg.h> +#include <stdlib.h> +#include "errcode.h" +#include "alloc.h" +#include "anydata.h" + +/* = = = = = = = = = = = = = = = prmptarg = = = = = = = = = = = = = = = */ + +MCPC_API mcpc_prmptarg_t * +mcpc_prmptarg_new (const char8_t *name, const size_t name_len, const char8_t *desc, const size_t desc_len) +{ + mcpc_anydata_t *hp = mcpc_anydata_new_meta1_meta2 (name, name_len, desc, desc_len); + + hp->flags |= ANYDATA_FL_REQUIRED; + hp->flags |= ANYDATA_FL_PRMPTARG; + + hp->data_typ = MCPC_U8STR; + + return hp; +} + +MCPC_API mcpc_prmptarg_t * +mcpc_prmptarg_new2 (const char8_t *nament, const char8_t *descnt) +{ + return mcpc_prmptarg_new (nament, u8strlen (nament), descnt, u8strlen (descnt)); +} + +MCPC_API mcpc_prmptarg_t * +mcpc_prmptarg_new3 (const char8_t *nament) +{ + return mcpc_prmptarg_new (nament, u8strlen (nament), nullptr, 0); +} + +void +mcpc_prmptarg_free (mcpc_prmptarg_t *prmptarg) +{ + mcpc_anydata_free ((mcpc_anydata_t *) prmptarg); +} + +bool +mcpc_prmptarg_is_required (const mcpc_prmptarg_t *prmptarg) +{ + return prmptarg->flags & ANYDATA_FL_REQUIRED; +} + +mcpc_errcode_t +mcpc_prmptarg_getref_name (const mcpc_prmptarg_t *prmptarg, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta1 (prmptarg, ret, ret_len); +} + +mcpc_errcode_t +mcpc_prmptarg_getref_desc (const mcpc_prmptarg_t *prmptarg, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta2 (prmptarg, ret, ret_len); +} + +mcpc_errcode_t +mcpc_prmptarg_get_data_type (const mcpc_prmptarg_t *prmptarg, mcpc_anytype_t *ret) +{ + bug_if_nullptr (prmptarg); + bug_if_nullptr (ret); + *ret = ((const mcpc_anydata_t *) prmptarg)->data_typ; + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_prmptarg_set_required (mcpc_prmptarg_t *prmptarg, bool fl) +{ + bug_if_nullptr (prmptarg); + + if (fl) + { + prmptarg->flags |= ANYDATA_FL_REQUIRED; + } + else + { + prmptarg->flags &= (~ANYDATA_FL_REQUIRED); + } + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_prmptarg_set_nament (mcpc_prmptarg_t *prmptarg, const char8_t *name) +{ + return mcpc_anydata_set_meta1nt (prmptarg, name); +} + +mcpc_errcode_t +mcpc_prmptarg_set_u8str (mcpc_prmptarg_t *prmptarg, const char8_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_u8str (prmptarg, arr, arr_len); +} + +MCPC_API mcpc_errcode_t +mcpc_prmptarg_get_u8str (const mcpc_prmptarg_t *prmptarg, char8_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_u8str (prmptarg, ret, ret_cap, ret_len); +} + +MCPC_API mcpc_errcode_t +mcpc_prmptarg_getref_u8str (const mcpc_prmptarg_t *prmptarg, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_u8str (prmptarg, ret, ret_len); +} + +/* = = = = = = = = = = = = = = = = = prmpt = = = = = = = = = = = = = = = = = */ + +MCPC_API mcpc_prmpt_t * +mcpc_prmpt_new (const char8_t *name, const size_t name_len, const char8_t *desc, const size_t desc_len) +{ + mcpc_anydata_t *hp = mcpc_anydata_new_meta1_meta2 (name, name_len, desc, desc_len); + + hp->flags |= ANYDATA_FL_PRMPT; + + return hp; +} + +MCPC_API mcpc_prmpt_t * +mcpc_prmpt_new2 (const char8_t *nament, const char8_t *descnt) +{ + return mcpc_prmpt_new (nament, strlen ((const char *) nament), descnt, strlen ((const char *) descnt)); +} + +MCPC_API mcpc_prmpt_t * +mcpc_prmpt_new3 (const char8_t *nament) +{ + return mcpc_prmpt_new (nament, u8strlen (nament), nullptr, 0); +} + +void +mcpc_prmpt_free (mcpc_prmpt_t *prmpt) +{ + mcpc_anydata_free ((mcpc_anydata_t *) prmpt); +} + +mcpc_errcode_t +mcpc_prmpt_getref_name (const mcpc_prmpt_t *prmpt, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta1 (prmpt, ret, ret_len); +} + +mcpc_errcode_t +mcpc_prmpt_getref_desc (const mcpc_prmpt_t *prmpt, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta2 (prmpt, ret, ret_len); +} + +mcpc_errcode_t +mcpc_prmpt_get_callcb (const mcpc_prmpt_t *prmpt, mcpc_prmpt_callcb_t *ret) +{ + return mcpc_anydata_get_func (prmpt, ret); +} + +mcpc_errcode_t +mcpc_prmpt_set_callcb (mcpc_prmpt_t *prmpt, mcpc_prmpt_callcb_t cb) +{ + mcpc_anydata_set_func (prmpt, cb); + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_prmpt_addfre_prmptarg (mcpc_prmpt_t *prmpt, mcpc_prmptarg_t *prmptarg) +{ + mcpc_anydata_add_child (prmpt, prmptarg); + return MCPC_EC_0; + +} + +MCPC_API void +mcpc_prmpt_addcpy_prmptarg (mcpc_prmpt_t *prmpt, mcpc_prmptarg_t *prmptarg) +{ + mcpc_anydata_addcpy_child (prmpt, prmptarg); // TODO return EC +} + +MCPC_API const mcpc_prmptarg_t * +mcpc_prmpt_get_prmptarg (const mcpc_prmpt_t *prmpt, size_t idx) +{ + return mcpc_anydata_get_child (prmpt, idx); +} + +MCPC_API size_t +mcpc_prmpt_get_prmptarg_len (const mcpc_prmpt_t *prmpt) +{ + bug_if_nullptr (prmpt); + mcpc_anydata_t *prmpt_imp = (mcpc_anydata_t *) prmpt; + if (prmpt_imp->chd_pool == nullptr) + return 0; + return prmpt_imp->chd_pool->len; +} + +MCPC_API const mcpc_prmptarg_t * +mcpc_prmpt_get_next_prmptarg (const mcpc_prmpt_t *prmpt) +{ + bug_if_nullptr (prmpt); + + static size_t idx = 0; + + const mcpc_anypool_t *pool = prmpt->chd_pool; + if (pool == nullptr) + return nullptr; + + const size_t pool_len = pool->len; + if (pool_len == 0) + return nullptr; + + if (idx == pool_len) // TODO MT-SAFE + { + idx = 0; + return nullptr; + } + + const mcpc_anydata_t *ret = pool->head + idx; + idx += 1; + + return ret; +} + +MCPC_API const mcpc_prmptarg_t * +mcpc_prmpt_getbyname_prmptarg (const mcpc_prmpt_t *prmpt, const char8_t *parg_name, size_t parg_name_len) +{ + const mcpc_anydata_t *chd = mcpc_anypool_getbymeta1 (prmpt->chd_pool, parg_name, parg_name_len); + return (mcpc_prmptarg_t *) chd; +} + +MCPC_API const mcpc_prmptarg_t * +mcpc_prmpt_getbynament_prmptarg (const mcpc_prmpt_t *prmpt, const char8_t *parg_nament) +{ + const mcpc_anydata_t *chd = mcpc_anypool_getbymeta1nt (prmpt->chd_pool, parg_nament); + return (mcpc_prmptarg_t *) chd; +} + +/* = = = = = = = = = = = = = = = prmptpool = = = = = = = = = = = = = = = */ + +mcpc_prmptpool_t * +mcpc_prmptpool_new () +{ + return mcpc_anypool_new (); +} + +void +mcpc_prmptpool_free (mcpc_prmptpool_t *prmptpool) +{ + mcpc_anypool_free (prmptpool); +} + +void +mcpc_prmptpool_addcpy (mcpc_prmptpool_t *prmptpool, const mcpc_prmpt_t *prmpt) +{ + mcpc_anypool_addcpy (prmptpool, prmpt); +} + +void +mcpc_prmptpool_addfre (mcpc_prmptpool_t *prmptpool, mcpc_prmpt_t *prmpt) +{ + mcpc_anypool_addfre (prmptpool, prmpt); +} + +size_t +mcpc_prmptpool_get_len (const mcpc_prmptpool_t *prmptpool) +{ + bug_if_nullptr (prmptpool); + size_t ret = 0; + ret = ((const mcpc_anypool_t *) prmptpool)->len; + return ret; +} + +const mcpc_prmpt_t * +mcpc_prmptpool_get (const mcpc_prmptpool_t *prmptpool, size_t idx) +{ + return mcpc_anypool_get (prmptpool, idx); +} + +const mcpc_prmpt_t * +mcpc_prmptpool_getbyname (const mcpc_prmptpool_t *prmptpool, const char8_t *name, size_t name_len) +{ + return mcpc_anypool_getbymeta1 (prmptpool, name, name_len); +} + +/* -= = = = = = = = = = = = = = prmptarghint = = = = = = = = = = = = = = = */ + +mcpc_prmptarghint_t * +mcpc_prmptarghint_new () +{ + mcpc_anydata_t *anydata = mcpc_anydata_new_typ (MCPC_U8STR); + anydata->flags |= ANYDATA_FL_PRMPTARGHINT; + return anydata; +} + +mcpc_prmptarghint_t * +mcpc_prmptarghint_new_vprintf8 (const char8_t *fmt, va_list *vargs) +{ + char8_t *sbuf = mcpc_alloc (1024); + int printed = vsprintf ((char *) sbuf, (const char *) fmt, *vargs); + + mcpc_prmptarghint_t *prmptarghint = mcpc_prmptarghint_new (); + bug_if_nz (mcpc_anydata_set_u8str (prmptarghint, sbuf, (size_t) printed)); + + free (sbuf); + return prmptarghint; +} + +MCPC_API mcpc_errcode_t +mcpc_prmptarg_add_hint_printf8 (mcpc_prmptarg_t *prmptarg, const char8_t *fmt, ...) +{ + va_list vargs; + va_start (vargs, fmt); + + mcpc_prmptarghint_t *hint = mcpc_prmptarghint_new_vprintf8 (fmt, &vargs); + + if (prmptarg->chd_pool == nullptr) + prmptarg->chd_pool = mcpc_anypool_new (); + + mcpc_anypool_addfre (prmptarg->chd_pool, hint); + + va_end (vargs); + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_prmptarghint_getref_u8str (const mcpc_prmptarg_t *prmptarghint, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_u8str (prmptarghint, ret, ret_len); +} + +const mcpc_prmptarghint_t * +mcpc_prmptarg_get_next_prmptarghint (const mcpc_prmptarg_t *prmptarg) +{ + bug_if_nullptr (prmptarg); + + static size_t idx = 0; + + const mcpc_anypool_t *pool = prmptarg->chd_pool; + if (pool == nullptr) + return nullptr; + + const size_t pool_len = pool->len; + if (pool_len == 0) + return nullptr; + + if (idx == pool_len) // TODO MT-SAFE + { + idx = 0; + return nullptr; + } + + const mcpc_anydata_t *ret = pool->head + idx; + idx += 1; + + return ret; +} diff --git a/third_party/mcpc/src/prmpt.h b/third_party/mcpc/src/prmpt.h new file mode 100644 index 0000000..2199864 --- /dev/null +++ b/third_party/mcpc/src/prmpt.h @@ -0,0 +1,80 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_prmpt +#define h_prmpt + +#include <mcpc/prmpt.h> + +typedef mcpc_anypool_t mcpc_prmptpool_t; +typedef mcpc_anypool_t mcpc_prmptargpool_t; + +// ---------------------------- prmptarg ---------------------------- + +void mcpc_prmptarg_free (mcpc_prmptarg_t * prmptarg); + +bool mcpc_prmptarg_is_required (const mcpc_prmptarg_t * prmptarg); + +mcpc_errcode_t mcpc_prmptarg_getref_name (const mcpc_prmptarg_t * prmptarg, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_prmptarg_getref_desc (const mcpc_prmptarg_t * prmptarg, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_prmptarg_get_data_type (const mcpc_prmptarg_t * prmptarg, mcpc_anytype_t * ret); + +mcpc_errcode_t mcpc_prmptarg_set_required (mcpc_prmptarg_t * prmptarg, bool fl); + +mcpc_errcode_t mcpc_prmptarg_set_nament (mcpc_prmptarg_t * prmptarg, const char8_t * name); + +mcpc_errcode_t mcpc_prmptarg_set_u8str (mcpc_prmptarg_t * prmptarg, const char8_t * arr, size_t arr_len); + +// ------------------------------ prmpt ------------------------------ + +mcpc_errcode_t mcpc_prmpt_getref_name (const mcpc_prmpt_t * prmpt, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_prmpt_getref_desc (const mcpc_prmpt_t * prmpt, const char8_t ** ret, size_t *ret_len); + +void mcpc_prmpt_free (mcpc_prmpt_t * prmpt); + +// ---------------------------- prmptpool ---------------------------- + +mcpc_prmptpool_t *mcpc_prmptpool_new (); + +void mcpc_prmptpool_free (mcpc_prmptpool_t * prmptpool); + +void mcpc_prmptpool_addcpy (mcpc_prmptpool_t * prmptpool, const mcpc_prmpt_t * prmpt); + +void mcpc_prmptpool_addfre (mcpc_prmptpool_t * prmptpool, mcpc_prmpt_t * prmpt); + +size_t mcpc_prmptpool_get_len (const mcpc_prmptpool_t * prmptpool); + +const mcpc_prmpt_t *mcpc_prmptpool_get (const mcpc_prmptpool_t * prmptpool, size_t idx); + +const mcpc_prmpt_t *mcpc_prmptpool_getbyname (const mcpc_prmptpool_t * prmptpool, const char8_t * name, + size_t name_len); + +// -------------------------- prmptarghint -------------------------- + +const mcpc_prmptarghint_t *mcpc_prmptarg_get_next_prmptarghint (const mcpc_prmptarg_t * prmptarg); + +mcpc_errcode_t +mcpc_prmptarghint_getref_u8str (const mcpc_prmptarg_t * prmptarghint, const char8_t ** ret, size_t *ret_len); + +#endif diff --git a/third_party/mcpc/src/retbuf.c b/third_party/mcpc/src/retbuf.c new file mode 100644 index 0000000..0429f15 --- /dev/null +++ b/third_party/mcpc/src/retbuf.c @@ -0,0 +1,119 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "retbuf.h" + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include "mjson.h" + +retbuf_t +retbuf_new (size_t cap, size_t *len) +{ + retbuf_t x = { + .cap = cap,.buf0 = malloc (cap),.len = len + }; + return x; +} + +void +retbuf_free (retbuf_t rtbf) +{ + free (rtbf.buf0); +} + +uint8_t +rtbuf_printf_adv (retbuf_t rtbf, size_t advl, const char8_t *u8str, ...) +{ + va_list args; + int printed; + size_t remaining; + (void) advl; + if (rtbf.buf0 == nullptr || rtbf.len == nullptr || *rtbf.len >= rtbf.cap) + return 1; + va_start (args, u8str); + remaining = rtbf.cap - *rtbf.len; + printed = vsnprintf (rtbf.buf0 + *rtbf.len, remaining, (const char *) u8str, args); + va_end (args); + if (printed < 0 || (size_t) printed >= remaining) + { + *rtbf.len = rtbf.cap; + return 1; + } + *rtbf.len += (size_t) printed; + return 0; +} + +uint8_t +rtbuf_printf (retbuf_t rtbf, const char8_t *u8str, ...) +{ + va_list args; + int printed; + size_t remaining; + if (rtbf.buf0 == nullptr || rtbf.len == nullptr || *rtbf.len >= rtbf.cap) + return 1; + va_start (args, u8str); + remaining = rtbf.cap - *rtbf.len; + printed = vsnprintf (rtbf.buf0 + *rtbf.len, remaining, (const char *) u8str, args); + va_end (args); + if (printed < 0 || (size_t) printed >= remaining) + { + *rtbf.len = rtbf.cap; + return 1; + } + *rtbf.len += (size_t) printed; + return 0; +} + +uint8_t +rtbuf_printf_mjfmt (retbuf_t rtbf, const char8_t *fmt8, ...) +{ + va_list args; + va_start (args, fmt8); + + const char *fmt = (const char *) fmt8; + struct mjson_fixedbuf fb = { rtbf.buf0, (int)rtbf.cap, (int)*rtbf.len }; + int advl = mjson_vprintf (&mjson_print_fixed_buf, &fb, fmt, &args); + + if (advl < 0 || (size_t) advl > rtbf.cap - *rtbf.len) + *rtbf.len = rtbf.cap; + else + *rtbf.len += (size_t) advl; + + va_end (args); + return 0; +} + +// TODO change _printf_ use this one +u8_t +rtbuf_vprintf_mjfmt (retbuf_t rtbf, const char8_t *fmt8, va_list *vargs) +{ + const char *fmt = (const char *) fmt8; + struct mjson_fixedbuf fb = { rtbf.buf0, (int)rtbf.cap, (int)*rtbf.len }; + int advl = mjson_vprintf (&mjson_print_fixed_buf, &fb, fmt, vargs); // TODO: __va_list_tag (*)[1]’ + if (advl < 0 || (size_t) advl > rtbf.cap - *rtbf.len) + *rtbf.len = rtbf.cap; + else + *rtbf.len += (size_t) advl; + return 0; +} diff --git a/third_party/mcpc/src/retbuf.h b/third_party/mcpc/src/retbuf.h new file mode 100644 index 0000000..eadcce4 --- /dev/null +++ b/third_party/mcpc/src/retbuf.h @@ -0,0 +1,50 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_retbuf +#define h_retbuf + +#include <stdarg.h> + +#include <mcpc/_c23_uchar.h> + +#include "ext_stdint.h" + + +typedef struct retbuf +{ + const size_t cap; + char *const buf0; + size_t *len; +} retbuf_t; + +retbuf_t retbuf_new (size_t cap, size_t *len); + +void retbuf_free (retbuf_t rtbf); + +u8_t rtbuf_printf_adv (retbuf_t rtbf, size_t advl, const char8_t * u8str, ...); + +u8_t rtbuf_printf (retbuf_t rtbf, const char8_t * u8str, ...); + +u8_t rtbuf_printf_mjfmt (retbuf_t rtbf, const char8_t * u8str, ...); + + +#endif diff --git a/third_party/mcpc/src/rsc.c b/third_party/mcpc/src/rsc.c new file mode 100644 index 0000000..be5627b --- /dev/null +++ b/third_party/mcpc/src/rsc.c @@ -0,0 +1,177 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "rsc.h" + +#include <stdint.h> +#include <stdlib.h> + +#include "ext_string.h" + +#include "errcode.h" +#include "alloc.h" +#include "anydata.h" + +MCPC_API mcpc_rsc_t * +mcpc_rsc_new (const char8_t *uri, const size_t uri_len, const char8_t *name, const size_t name_len) +{ + mcpc_anydata_t *anydata = mcpc_anydata_new_typ_meta1 (MCPC_U8STR, name, name_len); + + anydata->flags |= ANYDATA_FL_RSC; + anydata->data_mime = MCPC_MIME_TEXT_PLAIN; + + if (MCPC_EC_0 != mcpc_anydata_set_u8str (anydata, uri, uri_len)) + goto badret; + + return (mcpc_rsc_t *) anydata; + +badret: + mcpc_anydata_free (anydata); + return nullptr; +} + +MCPC_API mcpc_rsc_t * +mcpc_rsc_new2 (const char8_t *uri_nt, const char8_t *nament) +{ + return mcpc_rsc_new (uri_nt, u8strlen (uri_nt), nament, u8strlen (nament)); +} + +MCPC_API mcpc_rsc_t * +mcpc_rsc_new3 (mcpc_mime_t mime, const char8_t *uri_nt, const char8_t *nament) +{ + mcpc_rsc_t *rsc = mcpc_rsc_new2 (uri_nt, nament); + rsc->data_mime = mime; + return rsc; +} + +MCPC_API mcpc_errcode_t +mcpc_rsc_get_uri (const mcpc_rsc_t *rsc, char8_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_u8str (rsc, ret, ret_cap, ret_len); +} + +MCPC_API mcpc_errcode_t +mcpc_rsc_getref_uri (const mcpc_rsc_t *rsc, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_u8str (rsc, ret, ret_len); +} + +MCPC_API mcpc_errcode_t +mcpc_rsc_getref_name (const mcpc_rsc_t *rsc, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta1 (rsc, ret, ret_len); +} + +MCPC_API mcpc_errcode_t +mcpc_rsc_get_mime (const mcpc_rsc_t *rsc, mcpc_mime_t *ret) +{ + bug_if_nullptr (rsc); + bug_if_nullptr (ret); + *ret = rsc->data_mime; + return MCPC_EC_0; +} + +MCPC_API void +mcpc_rsc_set_mime (mcpc_rsc_t *rsc, mcpc_mime_t mime) +{ + bug_if_nullptr (rsc); + rsc->data_mime = mime; +} + +MCPC_API void +mcpc_rsc_free (mcpc_rsc_t *rsc) +{ + mcpc_anydata_free (rsc); +} + +/* = = = = = = = = = = = = = = = = rscpool = = = = = = = = = = = = = = = = */ + +mcpc_rscpool_t * +mcpc_rscpool_new () +{ + return mcpc_anypool_new (); +} + +void +mcpc_rscpool_free (mcpc_rscpool_t *rscpool) +{ + mcpc_anypool_free (rscpool); +} + +void +mcpc_rscpool_addcpy (mcpc_rscpool_t *rscpool, const mcpc_rsc_t *rsc) +{ + mcpc_anypool_addcpy (rscpool, rsc); +} + +void +mcpc_rscpool_addfre (mcpc_rscpool_t *rscpool, mcpc_rsc_t *rsc) +{ + mcpc_anypool_addfre (rscpool, rsc); +} + +size_t +mcpc_rscpool_get_len (const mcpc_rscpool_t *rscpool) +{ + bug_if_nullptr (rscpool); + size_t ret = 0; + ret = ((const mcpc_anypool_t *) rscpool)->len; + return ret; +} + +const mcpc_rsc_t * +mcpc_rscpool_get (const mcpc_rscpool_t *rscpool, size_t idx) +{ + return mcpc_anypool_get (rscpool, idx); +} + +const mcpc_rsc_t * +mcpc_rscpool_getbyname (const mcpc_rscpool_t *rscpool, const char8_t *name, size_t name_len) +{ + return mcpc_anypool_getbymeta1 (rscpool, name, name_len); +} + +const mcpc_rsc_t * +mcpc_rscpool_get_next (const mcpc_rscpool_t *rscpool) +{ + bug_if_nullptr (rscpool); + + static size_t idx = 0; + + const mcpc_anypool_t *pool = rscpool; + if (pool == nullptr) + return nullptr; + + const size_t pool_len = pool->len; + if (pool_len == 0) + return nullptr; + + if (idx == pool_len) // TODO MT-SAFE + { + idx = 0; + return nullptr; + } + + const mcpc_anydata_t *ret = pool->head + idx; + idx += 1; + + return ret; +} diff --git a/third_party/mcpc/src/rsc.h b/third_party/mcpc/src/rsc.h new file mode 100644 index 0000000..d9f1477 --- /dev/null +++ b/third_party/mcpc/src/rsc.h @@ -0,0 +1,51 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_rsc +#define h_rsc + +#include <stddef.h> +#include <stdio.h> + +#include <mcpc/_c23_keywords.h> +#include <mcpc/_c23_uchar.h> + +#include <mcpc/rsc.h> + +typedef mcpc_anypool_t mcpc_rscpool_t; + +// ---------------------------- rscpool ---------------------------- + +mcpc_rscpool_t *mcpc_rscpool_new (); + +void mcpc_rscpool_free (mcpc_rscpool_t * rscpool); + +void mcpc_rscpool_add_cp (mcpc_rscpool_t * rscpool, const mcpc_rsc_t * tmp_rsc); + +void mcpc_rscpool_addfre (mcpc_rscpool_t * rscpool, mcpc_rsc_t * tmp_rsc); + +const mcpc_rsc_t *mcpc_rscpool_get_const (const mcpc_rscpool_t * rscpool, size_t idx); + +const mcpc_rsc_t *mcpc_rscpool_get_next (const mcpc_rscpool_t * rscpool); + +const mcpc_rsc_t *mcpc_rscpool_getbyname (const mcpc_rscpool_t * rscpool, const char8_t * name, size_t name_len); + +#endif diff --git a/third_party/mcpc/src/serlz.c b/third_party/mcpc/src/serlz.c new file mode 100644 index 0000000..0721233 --- /dev/null +++ b/third_party/mcpc/src/serlz.c @@ -0,0 +1,619 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#define __TU__ "serlz" + +#include "serlz.h" + +#include "ext_string.h" +#include "log.h" + + +/* = = = = = = = = = = = = = = = = = tool = = = = = = = = = = = = = = = = = */ + +u8_t +serlz_tool_inputschema_properties_prop (retbuf_t rtbf, const mcpc_toolprop_t *tprop) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + const char8_t *name, *desc = nullptr; + size_t name_len, desc_len = 0; + + bug_if_nz (mcpc_toolprop_getref_name (tprop, &name, &name_len)); + bug_if_nz (mcpc_toolprop_getref_desc (tprop, &desc, &desc_len)); + + mcpc_anytype_t typ = MCPC_NONE; + bug_if_nz (mcpc_toolprop_get_data_type (tprop, &typ)); + + if (false) + ; + else if (mcpc_anytype_is_intlike (typ)) + { + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%Q"), "type", "integer"); + } + else if (typ == MCPC_U8STR) + { + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%Q"), "type", "string"); + } + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "title", name_len, name); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "description", desc_len, desc); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +u8_t +serlz_tool_inputschema_properties (retbuf_t rtbf, const mcpc_toolproppool_t *tppool) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + if (tppool != nullptr) + { + size_t len_tppool = mcpc_toolproppool_get_len (tppool); + if (len_tppool > 0) + { + for (size_t i = 0; i < len_tppool; i++) + { + const mcpc_toolprop_t *tprop = mcpc_toolproppool_get (tppool, i); + size_t name_len = 0; + const char8_t *name = nullptr; + bug_if (mcpc_toolprop_getref_name (tprop, &name, &name_len)); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%.*Q:"), name_len, name); + serlz_tool_inputschema_properties_prop (rtbf, tprop); + if (i != len_tppool - 1) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + } + } + } + + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +u8_t +serlz_tool_inputschema (retbuf_t rtbf, const mcpc_tool_t *tool) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%Q"), "type", "object"); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "properties"); + serlz_tool_inputschema_properties (rtbf, mcpc_tool_get_toolproppool (tool)); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:["), "required"); + bool first = true; + size_t property_count = mcpc_tool_get_toolprop_len (tool); + for (size_t i = 0; i < property_count; ++i) + { + const mcpc_toolprop_t *property = mcpc_tool_get_toolprop (tool, i); + const char8_t *name = nullptr; + size_t name_len = 0; + if (!mcpc_toolprop_is_required (property)) + continue; + bug_if_nz (mcpc_toolprop_getref_name (property, &name, &name_len)); + if (!first) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%.*Q"), name_len, name); + first = false; + } + rtbuf_printf_mjfmt (rtbf, u8c1 ("]")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_tool (retbuf_t rtbf, const mcpc_tool_t *tool) +{ + const char8_t *name, *desc = nullptr; + size_t name_len, desc_len = 0; + bug_if_nz (mcpc_toolprop_getref_name (tool, &name, &name_len)); + bug_if_nz (mcpc_toolprop_getref_desc (tool, &desc, &desc_len)); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "name", name_len, name); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "description", desc_len, desc); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "inputSchema"); + serlz_tool_inputschema (rtbf, tool); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +u8_t +serlz_toolpool (retbuf_t rtbf, const mcpc_toolpool_t *toolpool) +{ + size_t len_tpool = mcpc_toolpool_get_len (toolpool); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "tools"); + rtbuf_printf_mjfmt (rtbf, u8c1 ("[")); + + // TODO TODO TODO use iterator + if (toolpool != nullptr && len_tpool > 0) + { + for (size_t i = 0; i < len_tpool; i++) + { + const mcpc_tool_t *cur_tool = mcpc_toolpool_get (toolpool, i); + serlz_tool (rtbf, cur_tool); + if (i != len_tpool - 1) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + } + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("]")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +/* = = = = = = = = = = = = = = = = = prmpt = = = = = = = = = = = = = = = = = */ + +u8_t +serlz_prmptarg (retbuf_t rtbf, const mcpc_prmptarg_t *prmptarg) +{ + const char8_t *name, *desc = nullptr; + size_t name_len, desc_len = 0; + bool required = false; + + bug_if_nz (mcpc_prmptarg_getref_name (prmptarg, &name, &name_len)); + bug_if_nz (mcpc_prmptarg_getref_desc (prmptarg, &desc, &desc_len)); + required = mcpc_prmptarg_is_required (prmptarg); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "name", name_len, name); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "description", desc_len, desc); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%B"), "required", required); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_prmpt (retbuf_t rtbf, const mcpc_prmpt_t *prmpt) +{ + const char8_t *name, *desc = nullptr; + size_t name_len, desc_len = 0; + bug_if_nz (mcpc_prmptarg_getref_name (prmpt, &name, &name_len)); + bug_if_nz (mcpc_prmptarg_getref_desc (prmpt, &desc, &desc_len)); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "name", name_len, name); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "description", desc_len, desc); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "arguments"); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("[")); + + const mcpc_prmptarg_t *cur_prmptarg = nullptr; + bool first = true; + while ((cur_prmptarg = mcpc_prmpt_get_next_prmptarg (prmpt))) + { + if (!first) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + serlz_prmptarg (rtbf, cur_prmptarg); + first = false; + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("]")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +u8_t +serlz_prmptpool (retbuf_t rtbf, const mcpc_prmptpool_t *prmptpool) +{ + size_t len_tpool = mcpc_prmptpool_get_len (prmptpool); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "prompts"); + rtbuf_printf_mjfmt (rtbf, u8c1 ("[")); + + if (prmptpool != nullptr && len_tpool > 0) + { + for (size_t i = 0; i < len_tpool; i++) + { + const mcpc_prmpt_t *cur_prmpt = mcpc_prmptpool_get (prmptpool, i); + serlz_prmpt (rtbf, cur_prmpt); + if (i != len_tpool - 1) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + } + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("]")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +/* = = = = = = = = = = = = = = = = = = rsc = = = = = = = = = = = = = = = = = = */ + +u8_t +serlz_rsc (retbuf_t rtbf, const mcpc_rsc_t *rsc) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + const char8_t *ref_uri = nullptr; + size_t ref_uri_len = 0; + const char8_t *ref_name = nullptr; + size_t ref_name_len = 0; + mcpc_mime_t mime = MCPC_MIME_NONE; + + mcpc_rsc_getref_uri (rsc, &ref_uri, &ref_uri_len); + mcpc_rsc_getref_name (rsc, &ref_name, &ref_name_len); + mcpc_rsc_get_mime (rsc, &mime); + + bool serlz_uri = ref_uri_len > 0 && ref_uri != nullptr; + bool serlz_name = ref_name_len > 0 && ref_name != nullptr; + bool serlz_mime = mime != MCPC_MIME_NONE; + + // uri is ID, hence first + if (serlz_uri) + { + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "uri", ref_uri_len, (const char *) ref_uri); + } + + if (serlz_name) + { + if (serlz_uri) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "name", ref_name_len, ref_name); + } + + if (serlz_mime) + { + if (serlz_name || serlz_uri) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "mimeType"); + if (false); + else if (mime == MCPC_MIME_TEXT_PLAIN) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("text/plain")); + else if (mime == MCPC_MIME_IMAGE_PNG) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("image/png")); + else if (mime == MCPC_MIME_IMAGE_JPEG) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("image/jpeg")); + else if (mime == MCPC_MIME_IMAGE_GIF) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("image/gif")); + else if (mime == MCPC_MIME_AUDIO_MPEG) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("audio/mpeg")); + else if (mime == MCPC_MIME_AUDIO_WAV) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("audio/wav")); + else if (mime == MCPC_MIME_VIDEO_MPEG) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("video/mpeg")); + else if (mime == MCPC_MIME_VIDEO_AVI) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("video/avi")); + else if (mime == MCPC_MIME_APPLI_OCTS) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("application/octet-stream")); + else if (mime == MCPC_MIME_APPLI_JSON) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("application/json")); + else if (mime == MCPC_MIME_APPLI_PDF) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), u8c1 ("application/pdf")); + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + + return 0; +} + +u8_t +serlz_rscpool (retbuf_t rtbf, const mcpc_rscpool_t *rscpool) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "resources"); + rtbuf_printf_mjfmt (rtbf, u8c1 ("[")); + + + const mcpc_rsc_t *cur_rsc = nullptr; + bool first = true; + while ((cur_rsc = mcpc_rscpool_get_next (rscpool))) + { + if (!first) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + serlz_rsc (rtbf, cur_rsc); + first = false; + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("]")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +/* = = = = = = = = = = = = = = = = = = = - = = = = = = = = = = = = = = = = = = = */ + +uint8_t +serlz_serverInfo (retbuf_t rtbf, const mcpc_server_t *sv) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "name", sv->name_len, sv->name); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%.*Q"), "version", sv->ver_len, sv->ver); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_capasprompts (retbuf_t rtbf, const mcpc_server_t *sv) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "listChanged"); + bool supp_lschg = mcpc_server_supp_prompts_lschg (sv); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%B"), supp_lschg); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_capastools (retbuf_t rtbf, const mcpc_server_t *sv) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "listChanged"); + bool supp_lschg = mcpc_server_supp_tools_lschg (sv); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%B"), supp_lschg); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_capasresources (retbuf_t rtbf, const mcpc_server_t *sv) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "subscribe"); + bool supp_subs = mcpc_server_supp_rscs_subs (sv); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%B"), supp_subs); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "listChanged"); + bool supp_lschg = mcpc_server_supp_rscs_lschg (sv); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%B"), supp_lschg); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_capas (retbuf_t rtbf, const mcpc_server_t *sv) +{ + bool need_comma = false; + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + if (mcpc_server_supp_prompts (sv)) + { + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "prompts"); + serlz_capasprompts (rtbf, sv); + need_comma = true; + } + + if (mcpc_server_supp_tools (sv)) + { + if (need_comma) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "tools"); + serlz_capastools (rtbf, sv); + need_comma = true; + } + + if (mcpc_server_supp_rscs (sv)) + { + if (need_comma) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "resources"); + serlz_capasresources (rtbf, sv); + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_initres (retbuf_t rtbf, const mcpc_server_t *sv) +{ + rtbuf_printf (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%Q"), "protocolVersion", "2024-11-05"); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "serverInfo"); + serlz_serverInfo (rtbf, sv); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "capabilities"); + serlz_capas (rtbf, sv); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +/* = = = = = = = = = = = = = = = = = ucbr = = = = = = = = = = = = = = = = = */ + +uint8_t +serlz_ucbrchd_tool_callres (retbuf_t rtbf, const mcpc_ucbrchd_t *ucbrchd) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%Q"), "type", "text"); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "text"); + + mcpc_anytype_t data_type = MCPC_NONE; + bug_if_nz (mcpc_ucbrchd_get_data_type (ucbrchd, &data_type)); + + if (false) + ; + else if (data_type == MCPC_U8STR) + { + const char8_t *s = nullptr; + size_t s_len = 0; + mcpc_ucbrchd_getref_u8str (ucbrchd, &s, &s_len); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%.*Q"), s_len, (const char *) s); + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_ucbrchd_prmpt_callres (retbuf_t rtbf, const mcpc_ucbrchd_t *ucbrchd) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "role"); + if (mcpc_ucbrchd_is_assist_prmpt (ucbrchd)) + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), "assistant"); + else + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q"), "user"); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "content"); + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%Q"), "type", "text"); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "text"); + + mcpc_anytype_t data_type = MCPC_NONE; + bug_if_nz (mcpc_ucbrchd_get_data_type (ucbrchd, &data_type)); + + if (false) + ; + else if (data_type == MCPC_U8STR) + { + const char8_t *s = nullptr; + size_t s_len = 0; + mcpc_ucbrchd_getref_u8str (ucbrchd, &s, &s_len); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%.*Q"), s_len, (const char *) s); + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +uint8_t +serlz_ucbrchd (retbuf_t rtbf, const mcpc_ucbrchd_t *ucbrchd) +{ + if (false); + else if (mcpc_ucbrchd_is_tool_callres (ucbrchd)) + { + return serlz_ucbrchd_tool_callres (rtbf, ucbrchd); + } + else if (mcpc_ucbrchd_is_prmpt_callres (ucbrchd)) + { + return serlz_ucbrchd_prmpt_callres (rtbf, ucbrchd); + } + return 0; +} + +u8_t +serlz_ucbr (retbuf_t rtbf, const mcpc_ucbr_t *ucbr) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("[")); + + size_t length = mcpc_ucbr_get_len (ucbr); + for (size_t i = 0; i < length; ++i) + { + const mcpc_ucbrchd_t *cur_ucbrchd = mcpc_ucbr_get (ucbr, i); + if (i != 0) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + serlz_ucbrchd (rtbf, cur_ucbrchd); + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("]")); + return 0; +} + +u8_t +serlz_tool_callres (retbuf_t rtbf, const mcpc_ucbr_t *ucbr) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + if (mcpc_ucbr_is_err (ucbr)) + { + // https://modelcontextprotocol.io/docs/concepts/tools#error-handling-2 + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%B"), "isError", true); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "content"); + serlz_ucbr (rtbf, ucbr); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +u8_t +serlz_prmpt_callres (retbuf_t rtbf, const mcpc_ucbr_t *ucbr) +{ + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%Q"), "description", "Prompt List"); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "messages"); + serlz_ucbr (rtbf, ucbr); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} + +/* = = = = = = = = = = = = = = = = complt = = = = = = = = = = = = = = = = */ + +u8_t +serlz_complt (retbuf_t rtbf, const mcpc_complt_t *complt) +{ + size_t total = 0; + + mcpc_complt_get_compltcandi_len (complt, &total); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "completion"); + rtbuf_printf_mjfmt (rtbf, u8c1 ("{")); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%lu"), "total", total); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:%B"), "hasMore", false); + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("%Q:"), "values"); + rtbuf_printf_mjfmt (rtbf, u8c1 ("[")); + + const mcpc_compltcand_t *cur_compltcand = nullptr; + bool first = true; + while ((cur_compltcand = mcpc_complt_get_next_compltcand (complt))) + { + if (!first) + rtbuf_printf_mjfmt (rtbf, u8c1 (",")); + const char8_t *sbuf = nullptr; + size_t sbuf_len = 0; + bug_if_nz (mcpc_compltcand_getref_u8str (cur_compltcand, &sbuf, &sbuf_len)); + bug_if_nullptr (sbuf); + bug_if_zero (sbuf_len); + + rtbuf_printf_mjfmt (rtbf, u8c1 ("%.*Q"), sbuf_len, sbuf); + first = false; + } + + rtbuf_printf_mjfmt (rtbf, u8c1 ("]")); + + + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + rtbuf_printf_mjfmt (rtbf, u8c1 ("}")); + return 0; +} diff --git a/third_party/mcpc/src/serlz.h b/third_party/mcpc/src/serlz.h new file mode 100644 index 0000000..9ee6618 --- /dev/null +++ b/third_party/mcpc/src/serlz.h @@ -0,0 +1,62 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_serialize +#define h_serialize + +#include "retbuf.h" +#include "tool.h" +#include "rsc.h" +#include "server.h" +#include "ucbr.h" +#include "complt.h" + +#define L_QUO u8c1("\"") +#define R_QUO u8c1("\"") +#define L_QUO_D u8c1("(") +#define R_QUO_D u8c1(")") + +u8_t serlz_tool (retbuf_t rtbf, const mcpc_tool_t * tool); + +u8_t serlz_toolpool (retbuf_t rtbf, const mcpc_toolpool_t * pool); + +u8_t serlz_prmptpool (retbuf_t rtbf, const mcpc_prmptpool_t * prmptpool); + +u8_t serlz_initres (retbuf_t rtbf, const mcpc_server_t * sv); + +u8_t serlz_rsc (retbuf_t rtbf, const mcpc_rsc_t * rsc); + +u8_t serlz_rscpool (retbuf_t rtbf, const mcpc_rscpool_t * rscpool); + +// deprecated +u8_t serlz_ucbrchd (retbuf_t rtbf, const mcpc_ucbrchd_t * ucbrchd); + +// deprecated +u8_t serlz_ucbr (retbuf_t rtbf, const mcpc_ucbr_t * ucbr); + +u8_t serlz_tool_callres (retbuf_t rtbf, const mcpc_ucbr_t * ucbr); + +u8_t serlz_prmpt_callres (retbuf_t rtbf, const mcpc_ucbr_t * ucbr); + +u8_t serlz_complt (retbuf_t rtbf, const mcpc_complt_t * complt); + + +#endif diff --git a/third_party/mcpc/src/server.c b/third_party/mcpc/src/server.c new file mode 100644 index 0000000..07f8e8b --- /dev/null +++ b/third_party/mcpc/src/server.c @@ -0,0 +1,1830 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#define __TU__ "server" + +#include "server.h" + +#include "ext_string.h" + +#include "alloc.h" +#include "anydata.h" +#include <mcpc/anydata.h> +#include "errcode.h" +#include "log.h" +#include "retbuf.h" +#include "rsc.h" +#include "tool.h" +#include "ucbr.h" +#include "serlz.h" + +#include <limits.h> +#include <math.h> + +#define BUFCAP_INIT_JSONREQ 1024 +#define CH_SPC 0x20 +#define CH_LF 0x0a +#define CH_LBRC 0x7b +#define CH_RBRC 0x7d +#define CH_QUO 0x22 +#define CH_BSLASH 0x5c + + +#if is_unix +#include <arpa/inet.h> +#include <errno.h> +#include <netinet/in.h> +#include <pthread.h> +#include <sys/socket.h> +#include <unistd.h> +#elif is_win +#include <winsock2.h> +typedef i32_t socklen_t; +#endif + +struct mjson_fixedbuf +_new_mjson_fixedbuf () +{ + struct mjson_fixedbuf fbuf = { nullptr, 0, 0 }; + char *buf = mcpc_alloc (65536); + fbuf.ptr = buf; + fbuf.size = 65536; + fbuf.len = 0; + return fbuf; +} + +bool +ensure_buf_suffi (struct mjson_fixedbuf *fbuf, size_t ipt_len) +{ + char *grown; + if (fbuf == nullptr || ipt_len == 0 || ipt_len > INT_MAX) + return false; + if (ipt_len <= (size_t) fbuf->size) + { + return true; + } + + grown = mcpc_realloc (fbuf->ptr, ipt_len); + if (grown == nullptr) + return false; + fbuf->ptr = grown; + fbuf->size = (int) ipt_len; + return true; +} + +void +_connpool_init (mcpc_connpool_t *ret) +{ + mcpc_connpool_t connpool = { 255, 0, nullptr }; + memcpy (ret, &connpool, sizeof (mcpc_connpool_t)); +} + +bool +_connpool_is_buggy (const mcpc_connpool_t *connpool) +{ + if (connpool == nullptr) + { + return true; + } + if (connpool->cap == 0) + { + return true; + } + if (connpool->len != 0) + { + u8_t nexist = 0; + mcpc_conn_t *cur_conn = connpool->head; + while (cur_conn != nullptr) + { + cur_conn = cur_conn->nex; + nexist += 1; // OR + // mcpc_assert(!ckd_add(&nexist, nexist, 1), MCPC_EC_BUG); + } + if (nexist != connpool->len) + { + return true; + } + } + return false; +} + +// extract a field's unquoted string +mcpc_errcode_t +_tulc_jstr_get_field_u8str (struct jsonrpc_request *r, const char8_t *jstr, size_t jstr_len, const char8_t *jpathnt, + char8_t **ret, size_t *ret_len, size_t ret_cap, bool not_empty) +{ + const char *jstr2 = (const char *) jstr; + int jstr_len2 = (int) jstr_len; + const char *jpathnt2 = (const char *) jpathnt; + const char8_t *ret_q = nullptr; + int32_t ret_q_len = 0; + int tmptok; + + tmptok = mjson_find (jstr2, jstr_len2, jpathnt2, (const char **) &ret_q, (int *) &ret_q_len); + if (tmptok != MJSON_TOK_STRING || ret_q == nullptr || ret_q_len == 0) + goto field_invalid; + + int getres = mjson_get_string ((const char *) ret_q, ret_q_len, "$", (char *) *ret, ret_cap); + if (getres == -1) + return MCPC_EC_BUFCAP; + if (not_empty && getres == 0) + goto field_invalid; + *ret_len = (size_t) getres; + + return MCPC_EC_0; + +field_invalid: + jsonrpc_return_error (r, JSONRPC_ERROR_BAD_PARAMS, "field invalid", "%Q", jpathnt2); + return MCPC_EC_JSON_FIELD_INVALID; +} + +// extract a field's unquoted string +mcpc_errcode_t +_tulc_jparams_get_field_u8str (struct jsonrpc_request *r, + const char8_t *jpathnt, char8_t **ret, size_t *ret_len, size_t ret_cap, bool not_empty) +{ + return _tulc_jstr_get_field_u8str (r, (const char8_t *) r->params, r->params_len, jpathnt, ret, ret_len, + ret_cap, not_empty); +} + +// extract a field's value, that is, everything +mcpc_errcode_t +_tulc_jparams_getref_field_obj (struct jsonrpc_request *r, const char8_t *jpathnt, const char8_t **ret, size_t *ret_len) +{ + const char *jpathnt2 = (const char *) jpathnt; + int tmptok; + + tmptok = mjson_find (r->params, r->params_len, jpathnt2, (const char **) ret, (int *) ret_len); + if (tmptok != MJSON_TOK_OBJECT || ret == nullptr || ret_len == 0) + goto field_invalid; + + return MCPC_EC_0; + +field_invalid: + jsonrpc_return_error (r, JSONRPC_ERROR_BAD_PARAMS, "field invalid", "%Q", jpathnt2); + return MCPC_EC_JSON_FIELD_INVALID; +} + + +// extract a field's value, that is, everything +mcpc_errcode_t +_tulc_jparams_getref_field_u8str_q (struct jsonrpc_request *r, const char8_t *jpathnt, const char8_t **ret, + size_t *ret_len) +{ + const char *jpathnt2 = (const char *) jpathnt; + int tmptok; + + tmptok = mjson_find (r->params, r->params_len, jpathnt2, (const char **) ret, (int *) ret_len); + if (tmptok != MJSON_TOK_STRING || ret == nullptr || ret_len == 0) + goto field_invalid; + + return MCPC_EC_0; + +field_invalid: + jsonrpc_return_error (r, JSONRPC_ERROR_BAD_PARAMS, "field invalid", "%Q", jpathnt2); + return MCPC_EC_JSON_FIELD_INVALID; +} + + +// extract a field's value, that is, everything +mcpc_errcode_t +_tulc_jstr_get_field_number (struct jsonrpc_request *r, const char8_t *jstr, size_t jstr_len, + const char8_t *jpathnt, double *ret) +{ + const char *jpathnt2 = (const char *) jpathnt; + double numval = 0; + + int ngot = mjson_get_number ((const char *) jstr, jstr_len, jpathnt2, &numval); + + if (ngot != 1) + goto field_invalid; + + *ret = numval; + + return MCPC_EC_0; + +field_invalid: + jsonrpc_return_error (r, JSONRPC_ERROR_BAD_PARAMS, "field invalid", "%Q", jpathnt2); + return MCPC_EC_JSON_FIELD_INVALID; +} + + +mcpc_errcode_t +mcpc_conn_free (mcpc_conn_t *conn) +{ + bug_if_nullptr (conn); + + if (conn->cname != nullptr) + { + free ((void *) conn->cname); + conn->cname = nullptr; + conn->cname_len = 0; + } + + if (conn->nex != nullptr) + { + mcpc_conn_free (conn->nex); + } + + free (conn); + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_connpool_free (mcpc_connpool_t *connpool) +{ + bug_if_nullptr (connpool); + mcpc_conn_free (connpool->head); + free (connpool); + return MCPC_EC_0; +} + +MCPC_API mcpc_server_t * +mcpc_server_new (mcpc_server_type_t typ, mcpc_toolpool_t *toolpool, u32_t vabeg, ...) +{ + mcpc_assert (typ != MCPC_SV_NONE, MCPC_EC_BUG); + mcpc_assert (toolpool != nullptr, MCPC_EC_BUG); + + mcpc_server_t *sv = (mcpc_server_t *) mcpc_alloc (sizeof (mcpc_server_t)); + + sv->typ = typ; + sv->toolpool = toolpool; + sv->prmptpool = mcpc_prmptpool_new (); + sv->rscpool = mcpc_rscpool_new (); + sv->rpcres = _new_mjson_fixedbuf (); + sv->connpool = mcpc_alloc (sizeof (mcpc_connpool_t)); + _connpool_init ((mcpc_connpool_t *) sv->connpool); + + if (false) + ; + else if (typ == MCPC_SV_IOSTRM) + { + va_list ap; + va_start (ap, vabeg); + FILE *strm0 = va_arg (ap, FILE *); + mcpc_assert (strm0 != nullptr, MCPC_EC_BUG); + FILE *strm1 = va_arg (ap, FILE *); + mcpc_assert (strm1 != nullptr, MCPC_EC_BUG); + mcpc_conn_t *conn0 = mcpc_conn_new (MCPC_CONN_STRM, strm0); + mcpc_assert (MCPC_EC_0 == mcpc_connpool_addref ((mcpc_connpool_t *) sv->connpool, conn0), MCPC_EC_BUG); + mcpc_conn_t *conn1 = mcpc_conn_new (MCPC_CONN_STRM, strm1); + mcpc_assert (MCPC_EC_0 == mcpc_connpool_addref ((mcpc_connpool_t *) (sv->connpool), conn1), MCPC_EC_BUG); + va_end (ap); + } + else if (typ == MCPC_SV_TCP) + { + } + else + { + mcpc_assert (false, MCPC_EC_BUG); + } + + ch8_t *default_name = u8c1 ("untitled"); + u8_t default_name_len = (u8_t) u8strlen (default_name); + sv->name = mcpc_alloc ((size_t) default_name_len); + memcpy ((void *) sv->name, default_name, default_name_len); + sv->name_len = default_name_len; + + ch8_t *default_ver = u8c1 ("0.1"); + u8_t default_ver_len = (u8_t) u8strlen (default_ver); + sv->ver = mcpc_alloc ((size_t) default_ver_len); + memcpy ((void *) sv->ver, default_ver, default_ver_len); + sv->ver_len = default_ver_len; + + sv->capa = 0; + + return sv; +} + +MCPC_API mcpc_server_t * +mcpc_server_new_iostrm (const FILE *const strm_in, const FILE *const strm_out) +{ + mcpc_assert (strm_in != nullptr, MCPC_EC_BUG); + mcpc_assert (strm_out != nullptr, MCPC_EC_BUG); + + mcpc_toolpool_t *toolpool = mcpc_toolpool_new (); + mcpc_server_t *sv = mcpc_server_new (MCPC_SV_IOSTRM, toolpool, true, strm_in, strm_out); + + return sv; +} + +MCPC_API mcpc_server_t * +mcpc_server_new_tcp () +{ + mcpc_toolpool_t *toolpool = mcpc_toolpool_new (); + mcpc_server_t *sv = mcpc_server_new (MCPC_SV_TCP, toolpool, false); + + return sv; +} + +MCPC_API mcpc_errcode_t +mcpc_server_set_name (mcpc_server_t *sv, const char8_t *name, size_t name_len) +{ + bug_if (name_len > (size_t) UINT8_MAX); + sv->name = mcpc_realloc ((void *) sv->name, name_len); + memcpy ((void *) sv->name, name, name_len); + sv->name_len = (u8_t) name_len; + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_set_nament (mcpc_server_t *sv, const char8_t *nament) +{ + return mcpc_server_set_name (sv, nament, u8strlen (nament)); +} + +MCPC_API mcpc_errcode_t +mcpc_server_capa_enable_prmt (mcpc_server_t *sv) +{ + sv->capa |= MCPC_CAPA_PRMT; + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_capa_enable_prmt_listchg (mcpc_server_t *sv) +{ + sv->capa |= MCPC_CAPA_PRMT_LISTCHG; + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_capa_enable_tool (mcpc_server_t *sv) +{ + sv->capa |= MCPC_CAPA_TOOL; + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_capa_enable_tool_listchg (mcpc_server_t *sv) +{ + sv->capa |= MCPC_CAPA_TOOL_LISTCHG; + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_capa_enable_rsc (mcpc_server_t *sv) +{ + sv->capa |= MCPC_CAPA_RSC; + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_capa_enable_rsc_listchg (mcpc_server_t *sv) +{ + sv->capa |= MCPC_CAPA_RSC_LISTCHG; + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_capa_enable_rsc_subscr (mcpc_server_t *sv) +{ + sv->capa |= MCPC_CAPA_RSC_SUBSRC; + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_capa_enable_complt (mcpc_server_t *sv) +{ + sv->capa |= MCPC_CAPA_COMPLT; + return MCPC_EC_0; +} + +static inline bool +_server_is_buggy (const mcpc_server_t *sv) +{ + if (sv == nullptr) + return true; + if (sv->typ == MCPC_SV_NONE) + return true; + if (sv->toolpool == nullptr) + return true; + + return false; +} + +MCPC_API mcpc_errcode_t +mcpc_server_add_tool (mcpc_server_t *sv, mcpc_tool_t *tool) +{ + mcpc_tcallcb_t callback = nullptr; + if (_server_is_buggy (sv)) + return MCPC_EC_BUG; + if (tool == nullptr) + return MCPC_EC_BUG; + if (mcpc_tool_get_call_cb (tool, &callback) != MCPC_EC_0 || callback == nullptr) + return MCPC_EC_BUG; + mcpc_toolpool_addfre ((mcpc_toolpool_t *) sv->toolpool, tool); + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_add_prmpt (mcpc_server_t *sv, mcpc_prmpt_t *prmpt) +{ + if (_server_is_buggy (sv)) + return MCPC_EC_BUG; + if (prmpt == nullptr) + return MCPC_EC_BUG; + mcpc_prmptpool_addfre ((mcpc_prmptpool_t *) sv->prmptpool, prmpt); + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_add_rsc (mcpc_server_t *sv, mcpc_rsc_t *rsc) +{ + if (_server_is_buggy (sv)) + return MCPC_EC_BUG; + if (rsc == nullptr) + return MCPC_EC_BUG; + mcpc_rscpool_addfre ((mcpc_rscpool_t *) sv->rscpool, rsc); + return MCPC_EC_0; +} + +bool +mcpc_server_supp_prompts (const mcpc_server_t *sv) +{ + return (sv->capa & MCPC_CAPA_PRMT); +} + +bool +mcpc_server_supp_prompts_lschg (const mcpc_server_t *sv) +{ + return (sv->capa & MCPC_CAPA_PRMT_LISTCHG) == MCPC_CAPA_PRMT_LISTCHG; +} + +bool +mcpc_server_supp_tools (const mcpc_server_t *sv) +{ + return (sv->capa & MCPC_CAPA_TOOL) == MCPC_CAPA_TOOL; +} + +bool +mcpc_server_supp_tools_lschg (const mcpc_server_t *sv) +{ + return (sv->capa & MCPC_CAPA_TOOL_LISTCHG) == MCPC_CAPA_TOOL_LISTCHG; +} + +bool +mcpc_server_supp_rscs (const mcpc_server_t *sv) +{ + return (sv->capa & MCPC_CAPA_RSC) == MCPC_CAPA_RSC; +} + +bool +mcpc_server_supp_rscs_lschg (const mcpc_server_t *sv) +{ + return (sv->capa & MCPC_CAPA_RSC_LISTCHG) == MCPC_CAPA_RSC_LISTCHG; +} + +bool +mcpc_server_supp_rscs_subs (const mcpc_server_t *sv) +{ + return (sv->capa & MCPC_CAPA_RSC_SUBSRC) == MCPC_CAPA_RSC_SUBSRC; +} + +bool +mcpc_server_supp_complt (const mcpc_server_t *sv) +{ + return (sv->capa & MCPC_CAPA_COMPLT) == MCPC_CAPA_COMPLT; +} + + +// ----------------------------- iostrm ----------------------------- + +mcpc_errcode_t +_wait_iostrm (mcpc_server_t *sv) +{ + mcpc_conn_t *conn0 = mcpc_connpool_getconn (sv->connpool, 0); + mcpc_assert (conn0->typ == MCPC_CONN_STRM, MCPC_EC_BUG); + mcpc_conn_t *conn1 = mcpc_connpool_getconn (sv->connpool, 1); + mcpc_assert (conn1->typ == MCPC_CONN_STRM, MCPC_EC_BUG); + + FILE *in_strm = mcpc_conn_getstrm (conn0); + mcpc_assert (in_strm != nullptr, MCPC_EC_BUG); + FILE *out_strm = mcpc_conn_getstrm (conn1); + mcpc_assert (out_strm != nullptr, MCPC_EC_BUG); + + size_t rbuf_cap = BUFCAP_INIT_JSONREQ; + size_t rbuf_len = 0; + char *rbuf = mcpc_alloc (rbuf_cap); + int ch; + if (rbuf == nullptr) + return MCPC_EC_BUFCAP; + + while ((ch = getc (in_strm)) != EOF) + { + if (ch == '\n') + { + if (rbuf_len > 0 && rbuf[rbuf_len - 1] == '\r') + rbuf_len -= 1; + if (rbuf_len > 0) + handle_inone (sv, rbuf, rbuf_len, (mcpc_sock_t) - 1); + rbuf_len = 0; + continue; + } + if (rbuf_len == rbuf_cap) + { + char *grown; + if (rbuf_cap > SIZE_MAX / 2) + { + free (rbuf); + return MCPC_EC_BUFCAP; + } + grown = mcpc_realloc (rbuf, rbuf_cap * 2); + if (grown == nullptr) + { + free (rbuf); + return MCPC_EC_BUFCAP; + } + rbuf = grown; + rbuf_cap *= 2; + } + rbuf[rbuf_len++] = (char) ch; + } + if (ferror (in_strm)) + { + free (rbuf); + return MCPC_EC_BUG; + } + if (rbuf_len > 0) + handle_inone (sv, rbuf, rbuf_len, (mcpc_sock_t) - 1); + free (rbuf); + return MCPC_EC_0; +} + +mcpc_errcode_t +_svimpl_iostrm (mcpc_server_t *sv) +{ + mcpc_assert (sv->connpool->len == 2, MCPC_EC_BUG); + jsonrpc_init (nullptr, nullptr); // no need to handle responses + + return _wait_iostrm (sv); +} + +mcpc_errcode_t +_start_iostrm (mcpc_server_t *sv) +{ + return _svimpl_iostrm (sv); +} + +// ------------------------------ tcp ------------------------------ + +typedef struct +{ + mcpc_sock_t csock; + struct sockaddr_in caddr; + mcpc_server_t *sv; +} accept_hdlr_args_t; + +mcpc_errcode_t +_new_tcp_listen (mcpc_sock_t *ssock) +{ + mcpc_sock_t tmpsock = (mcpc_sock_t) 0; + struct sockaddr_in saddr; + + tmpsock = socket (AF_INET, SOCK_STREAM, 0); +#if is_unix + if (tmpsock < 0) +#endif +#if is_win + if (tmpsock == INVALID_SOCKET) +#endif + { + return MCPC_EC_SVIMPL_SOCKET; + } + + int level = SOL_SOCKET; + int optname = SO_REUSEADDR; + int optval = 1; + mcpc_assert (0 == setsockopt (tmpsock, level, optname, (const char *) &optval, sizeof (optval)), MCPC_EC_BUG); + + saddr.sin_family = AF_INET; + saddr.sin_port = htons (PORT); +#if defined(is_unix) + inet_pton (AF_INET, "127.0.0.1", &saddr.sin_addr); +#elif defined(is_win) + saddr.sin_addr.s_addr = inet_addr ("127.0.0.1"); +#endif + + if (bind (tmpsock, (struct sockaddr *) &saddr, sizeof (saddr)) == -1) + { + return MCPC_EC_SVIMPL_BIND; + } + + if (listen (tmpsock, 16) == -1) + { + return MCPC_EC_SVIMPL_LISTEN; + } + + *ssock = tmpsock; + + return MCPC_EC_0; +} + +#ifdef is_unix +static void * +#endif +#ifdef is_win + DWORD WINAPI +#endif +_wait_conn (void *args) +{ + accept_hdlr_args_t *thread_args_ptr = (accept_hdlr_args_t *) args; + mcpc_sock_t csock = thread_args_ptr->csock; +#ifdef MCPC_DBG + struct sockaddr_in caddr = thread_args_ptr->caddr; +#endif + mcpc_server_t *sv = thread_args_ptr->sv; + + tulog_d ("New connection from client IP address %s and port %d", inet_ntoa (caddr.sin_addr), ntohs (caddr.sin_port)); + + char ch; + int n_brace = 0; + bool start_read = false; + bool enter_str = false; + size_t rbuf_cap = BUFCAP_INIT_JSONREQ; + char *rbuf = mcpc_alloc (rbuf_cap); + char *rbuf_p = rbuf; + i32_t last_nrecv = 0; + char last_read = 0; + char lastlast_read = 0; + + while (true) + { + last_nrecv = recv (csock, &ch, 1, 0); + if (false) + ; + else if (last_nrecv == 0) + { + break; + } + else if (last_nrecv < 0) + { + tulog_d ("recv err %s", strerror (errno)); + break; + } + + if (!enter_str && ch == CH_LBRC) + { + n_brace += 1; + if (!start_read) + start_read = true; + } + + if (!start_read) + { + continue; + } + + if (start_read) + { + *rbuf_p = ch; + rbuf_p += 1; + lastlast_read = last_read; + last_read = ch; + if (rbuf_p - rbuf >= (ptrdiff_t) rbuf_cap) + { + rbuf_cap += rbuf_cap; + rbuf = mcpc_realloc (rbuf, rbuf_cap); + } + } + + if (!enter_str && last_read == CH_RBRC) + { + n_brace -= 1; + if (n_brace == 0) + { + ptrdiff_t nread = rbuf_p - rbuf; + rbuf[nread - 1] = CH_RBRC; + rbuf[nread] = 0x00; + handle_inone (sv, rbuf, (size_t) nread, csock); + rbuf_p = rbuf; + start_read = false; + enter_str = false; + tulog_d ("one msg handled"); + } + } + if (last_read == CH_QUO && lastlast_read != CH_BSLASH) + { + enter_str = !enter_str; + } + } + + free (rbuf); + +#ifdef is_unix + close (csock); +#endif +#ifdef is_win + closesocket (csock); +#endif + + free (thread_args_ptr); + +#ifdef is_unix + tulog_d ("thread %lu exiting...", pthread_self ()); +#endif +#ifdef is_win + tulog_d ("thread %lu exiting...", GetCurrentThreadId ()); +#endif + +#ifdef is_unix + pthread_exit (nullptr); +#endif +#ifdef is_win + ExitThread (0); +#endif +} + +mcpc_errcode_t +_svimpl_tcp (mcpc_server_t *sv) +{ + +#ifdef is_win + WSADATA wsaData; + if (WSAStartup (MAKEWORD (2, 2), &wsaData) != 0) + { + return MCPC_EC_SVIMPL_SOCKET; + } +#endif + + mcpc_sock_t ssock; + struct sockaddr_in caddr; + socklen_t caddr_length = sizeof (caddr); + + mcpc_assert (MCPC_EC_0 == _new_tcp_listen (&ssock), MCPC_EC_BUG); + + tulog_d ("Server listening on port %d...", PORT); + + mcpc_sock_t csock; + while (true) + { + csock = accept (ssock, (struct sockaddr *) &caddr, &caddr_length); +#ifdef is_unix + if (csock < 0) +#endif +#ifdef is_win + if (csock == INVALID_SOCKET) +#endif + { + tulog_d ("accept failed"); + continue; + } + + accept_hdlr_args_t *args = (accept_hdlr_args_t *) mcpc_alloc (sizeof (accept_hdlr_args_t)); + args->csock = csock; + args->caddr = caddr; + args->sv = sv; + +#ifdef is_unix + pthread_t thread; + if (pthread_create (&thread, nullptr, _wait_conn, args) != 0) + { + tulog_d ("thread creation failed"); + free (args); + close (csock); + } +#endif +#ifdef is_win + HANDLE thread; + DWORD threadId; + thread = CreateThread (NULL, 0, _wait_conn, args, 0, &threadId); + if (thread == NULL) + { + tulog_d ("thread creation failed"); + free (args); + closesocket (csock); + } +#endif + } + +#ifdef is_win + WSACleanup (); +#endif + + return 0; +} + +mcpc_errcode_t +_start_tcp (mcpc_server_t *sv) +{ + return _svimpl_tcp (sv); +} + +// ------------------------------- - ------------------------------- + +MCPC_API mcpc_errcode_t +mcpc_server_start (mcpc_server_t *sv) +{ + if (_server_is_buggy (sv)) + { + return MCPC_EC_BUG; + } + if (false) + ; + else if (sv->typ == MCPC_SV_IOSTRM) + { + return _start_iostrm (sv); + } + else if (sv->typ == MCPC_SV_TCP) + { + return _start_tcp (sv); + } + else + { + return MCPC_EC_BUG; + } + return MCPC_EC_0; +} + +MCPC_API mcpc_errcode_t +mcpc_server_close (mcpc_server_t *sv) +{ + // return MCPC_EC_0; // NOTE: this does not trigger asan but valgrind + if (_server_is_buggy (sv)) + { + return MCPC_EC_BUG; + } + + if (sv->name != nullptr) + { + free ((void *) sv->name); + sv->name = nullptr; + sv->name_len = 0; + } + + if (sv->ver != nullptr) + { + free ((void *) sv->ver); + sv->ver = nullptr; + sv->ver_len = 0; + } + + free (sv->rpcres.ptr); + mcpc_toolpool_free ((mcpc_toolpool_t *) sv->toolpool); + mcpc_rscpool_free ((mcpc_rscpool_t *) sv->prmptpool); + mcpc_rscpool_free ((mcpc_rscpool_t *) sv->rscpool); + mcpc_connpool_free ((mcpc_connpool_t *) sv->connpool); + + free (sv); + + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_server_cleanup_rpcres (mcpc_server_t *sv) +{ + sv->rpcres.len = 0; + return MCPC_EC_0; +} + +bool +_connpool_is_addable (mcpc_connpool_t *connpool) +{ + if (connpool->len < connpool->cap) + { + return true; + } + return false; +} + +mcpc_conn_t * +mcpc_conn_new (mcpc_conn_type_t typ, void *back) +{ + mcpc_conn_t *conn = mcpc_alloc (sizeof (mcpc_conn_t)); + conn->typ = typ; + + if (false) + ; + else if (typ == MCPC_CONN_STRM) + { + FILE *strm = (FILE *) back; + conn->strm = strm; + } + else if (typ == MCPC_CONN_FD) + { + int fd = *((int *) back); + conn->fd = fd; + } + else + { + conn->strm = nullptr; + } + conn->cname = nullptr; + conn->cname_len = 0; + conn->client_init = MCPC_INIT_NONE; + conn->nex = nullptr; + + return conn; +} + +FILE * +mcpc_conn_getstrm (mcpc_conn_t *conn) +{ + if (conn->typ != MCPC_CONN_STRM) + { + return nullptr; + } + return conn->strm; +} + +int +mcpc_conn_getfd (mcpc_conn_t *conn) +{ + if (conn->typ != MCPC_CONN_FD) + { + return 0; + } + return conn->fd & ((1ULL << (sizeof (int) * 8)) - 1); +} + +mcpc_errcode_t +mcpc_connpool_addref (mcpc_connpool_t *connpool, mcpc_conn_t *conn) +{ + if (_connpool_is_buggy ((const mcpc_connpool_t *) connpool)) + { + return MCPC_EC_BUG; + } + + if (!_connpool_is_addable (connpool)) + { + return MCPC_EC_CONNPOOL_NOTADD; + } + + if (connpool->head == nullptr) + { + connpool->head = conn; + connpool->len = 1; + } + else + { + mcpc_conn_t *cur_conn = connpool->head; + while (cur_conn->nex != nullptr) + { + cur_conn = cur_conn->nex; + } + cur_conn->nex = conn; + connpool->len += 1; + // mcpc_assert(!ckd_add(&connpool->len, connpool->len, 1), MCPC_EC_BUG); + } + + return MCPC_EC_0; +} + +mcpc_conn_t * +mcpc_connpool_getconn (const mcpc_connpool_t *connpool, const mcpc_connpool_size_t idx) +{ + mcpc_assert (!_connpool_is_buggy (connpool), MCPC_EC_BUG); + mcpc_connpool_size_t cur_idx = 0; + mcpc_conn_t *cur_conn = connpool->head; + while (cur_conn != nullptr && cur_idx < idx) + { + cur_conn = cur_conn->nex; + cur_idx += 1; + // mcpc_assert(!ckd_add(&cur_idx, cur_idx, 1), MCPC_EC_BUG); + } + return cur_conn; +} + +mcpc_conn_t * +mcpc_connpool_find_by_sock (const mcpc_connpool_t *connpool, mcpc_sock_t sock) +{ + mcpc_assert (!_connpool_is_buggy (connpool), MCPC_EC_BUG); + + // iostrm + if (mcpc_sock_invalid (sock)) + { + return connpool->head; + } + + // tcp + mcpc_conn_t *cur_conn = connpool->head; + while (cur_conn != nullptr) + { + mcpc_sock_t cur_sock = cur_conn->fd; + if (cur_sock == sock) + { + return cur_conn; + } + cur_conn = cur_conn->nex; + } + + return nullptr; +} + +mcpc_errcode_t +mcpc_conn_set_cname (mcpc_conn_t *conn, const ch8_t *const cname, const u8_t cname_len) +{ + void *hp = mcpc_alloc (cname_len); + memcpy (hp, cname, cname_len); + conn->cname = hp; + conn->cname_len = cname_len; + return MCPC_EC_0; +} + +void +_handle_initbeg (mcpc_server_t *sv, struct jsonrpc_request *r, mcpc_sock_t csock) +{ + + mcpc_conn_t *conn = mcpc_connpool_find_by_sock (sv->connpool, csock); + if (conn == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "connection not found", NULL); + goto output_res; + } + if (conn->client_init == MCPC_INIT_SUCC) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "conn already initialized", NULL); + goto output_res; + } + + // - - - + + const char8_t *protover = nullptr; + size_t protover_len = 0; + if (MCPC_EC_0 != _tulc_jparams_getref_field_u8str_q (r, u8c1 ("$.protocolVersion"), &protover, &protover_len)) + return; + + // - - - client info + + char8_t *cinfo_name = nullptr; + size_t cinfo_name_len = 0; + char8_t *cinfo_ver = nullptr; + size_t cinfo_ver_len = 0; + + cinfo_name = alloca (128); + if (MCPC_EC_0 != + _tulc_jparams_get_field_u8str (r, u8c1 ("$.clientInfo.name"), &cinfo_name, &cinfo_name_len, 128, true)) + return; // TODO P5 add test + + cinfo_ver = alloca (128); + if (MCPC_EC_0 != + _tulc_jparams_get_field_u8str (r, u8c1 ("$.clientInfo.version"), &cinfo_ver, &cinfo_ver_len, 128, true)) + return; // TODO P5 add test + + mcpc_conn_set_cname (conn, cinfo_name, (u8_t) cinfo_name_len); + + // - - - capabilities + + const char8_t *ccapas = nullptr; + size_t ccapas_len = 0; + if (MCPC_EC_0 != _tulc_jparams_getref_field_obj (r, u8c1 ("$.capabilities"), &ccapas, &ccapas_len)) + return; // TODO P5 add test + + // TODO P6 capabilities negotiation + + char *buftpool = nullptr; + size_t buftpool_len = 0; + const size_t buftpool_cap = 65536; + buftpool = mcpc_alloc (buftpool_cap); + retbuf_t rtbf = { buftpool_cap, buftpool, &buftpool_len }; + mcpc_assert (!serlz_initres (rtbf, sv), MCPC_EC_BUG); + mcpc_assert (buftpool != nullptr, MCPC_EC_BUG); + mcpc_assert (buftpool_len > 0, MCPC_EC_BUG); + jsonrpc_return_success2 (r, mcpc_sock_invalid (csock), "%.*s", buftpool_len, buftpool); + free (buftpool); // TODO intn_free + + conn->client_init = MCPC_INIT_BEG; + +output_res: + return; +} + +void +_handle_initdone (mcpc_server_t *sv, struct jsonrpc_request *r, mcpc_sock_t csock, bool *noreply) +{ + + mcpc_conn_t *conn = mcpc_connpool_find_by_sock (sv->connpool, csock); + if (conn == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "connection not found", NULL); + goto output_res; + } + if (conn->client_init != MCPC_INIT_BEG) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "client not begin init", NULL); + goto output_res; + } + + conn->client_init = MCPC_INIT_SUCC; + *noreply = true; + +output_res: + return; +} + +void +_handle_tools_list (mcpc_server_t *sv, struct jsonrpc_request *r, mcpc_sock_t csock) +{ + + mcpc_conn_t *conn = mcpc_connpool_find_by_sock (sv->connpool, csock); + if (conn == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "connection not found", NULL); + goto output_res; + } + if (conn->client_init != MCPC_INIT_SUCC) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "client not initialized", NULL); + goto output_res; + } + + char *buftpool = nullptr; + size_t buftpool_len = 0; + const size_t buftpool_cap = 65536; + buftpool = mcpc_alloc (buftpool_cap); + retbuf_t rtbf = { buftpool_cap, buftpool, &buftpool_len }; + mcpc_assert (!serlz_toolpool (rtbf, sv->toolpool), MCPC_EC_BUG); + mcpc_assert (buftpool != nullptr, MCPC_EC_BUG); + mcpc_assert (buftpool_len > 0, MCPC_EC_BUG); + jsonrpc_return_success2 (r, mcpc_sock_invalid (csock), "%.*s", buftpool_len, buftpool); + free (buftpool); // TODO intn_free + +output_res: + return; +} + +void +_handle_prmpt_list (mcpc_server_t *sv, struct jsonrpc_request *r, mcpc_sock_t csock) +{ + if (!mcpc_server_supp_prompts (sv)) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "prompts is not supported", NULL); + goto output_res; + } + + mcpc_conn_t *conn = mcpc_connpool_find_by_sock (sv->connpool, csock); + if (conn == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "connection not found", NULL); + goto output_res; + } + if (conn->client_init != MCPC_INIT_SUCC) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "client not initialized", NULL); + goto output_res; + } + + char *buftpool = nullptr; + size_t buftpool_len = 0; + const size_t buftpool_cap = 1024; + buftpool = mcpc_alloc (buftpool_cap); + retbuf_t rtbf = { buftpool_cap, buftpool, &buftpool_len }; + mcpc_assert (!serlz_prmptpool (rtbf, sv->prmptpool), MCPC_EC_BUG); + mcpc_assert (buftpool != nullptr, MCPC_EC_BUG); + mcpc_assert (buftpool_len > 0, MCPC_EC_BUG); + jsonrpc_return_success2 (r, mcpc_sock_invalid (csock), "%.*s", buftpool_len, buftpool); + free (buftpool); // TODO intn_free + +output_res: + return; +} + +void +_handle_lsrsc (mcpc_server_t *sv, struct jsonrpc_request *r, mcpc_sock_t csock) +{ + + mcpc_conn_t *conn = mcpc_connpool_find_by_sock (sv->connpool, csock); + if (conn == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "connection not found", NULL); + goto output_res; + } + if (conn->client_init != MCPC_INIT_SUCC) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "client not initialized", NULL); + goto output_res; + } + + char *buftpool = nullptr; + size_t buftpool_len = 0; + const size_t buftpool_cap = 1024; + buftpool = mcpc_alloc (buftpool_cap); + retbuf_t rtbf = { buftpool_cap, buftpool, &buftpool_len }; + mcpc_assert (!serlz_rscpool (rtbf, sv->rscpool), MCPC_EC_BUG); + mcpc_assert (buftpool != nullptr, MCPC_EC_BUG); + mcpc_assert (buftpool_len > 0, MCPC_EC_BUG); + jsonrpc_return_success2 (r, mcpc_sock_invalid (csock), "%.*s", buftpool_len, buftpool); + free (buftpool); // TODO intn_free + +output_res: + return; +} + +void +_handle_tools_call (mcpc_server_t *sv, struct jsonrpc_request *r, mcpc_sock_t csock) +{ + mcpc_ucbr_t *ucbr = nullptr; + char *buftpool = nullptr; + mcpc_conn_t *conn = mcpc_connpool_find_by_sock (sv->connpool, csock); + if (conn == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "connection not found", NULL); + goto output_res; + } + if (conn->client_init != MCPC_INIT_SUCC) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "client not initialized", NULL); + goto output_res; + } + + char8_t *callname = nullptr; // TODO suff? + size_t callname_len = 0; + + callname = alloca (1024); // TODO insuffi? + if (MCPC_EC_0 != _tulc_jparams_get_field_u8str (r, u8c1 ("$.name"), &callname, &callname_len, 1024, true)) + return; + + // const mcpc_tool_t *tool = mcpc_toolpool_find_tool (sv->toolpool, callname, callname_len); + + const mcpc_tool_t *tool = (const mcpc_tool_t *) mcpc_toolpool_getbyname (sv->toolpool, callname, callname_len); + if (tool == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_BAD_PARAMS, "tool not found", NULL); + goto output_res; + } + + const char8_t *callargs = nullptr; + size_t callargs_len = 0; + if (MCPC_EC_0 != _tulc_jparams_getref_field_obj (r, u8c1 ("$.arguments"), &callargs, &callargs_len)) + goto output_res; + + size_t property_count = mcpc_tool_get_toolprop_len (tool); + for (size_t property_index = 0; property_index < property_count; ++property_index) + { + mcpc_toolprop_t *cur_tprop + = (mcpc_toolprop_t *) mcpc_tool_get_toolprop (tool, property_index); + const char8_t *cur_tp_name = nullptr; + size_t cur_tp_name_len = 0; + bug_if_nz (mcpc_toolprop_getref_name (cur_tprop, &cur_tp_name, &cur_tp_name_len)); + mcpc_anytype_t cur_tp_type = MCPC_NONE; + bug_if_nz (mcpc_toolprop_get_data_type (cur_tprop, &cur_tp_type)); + char8_t *cur_tp_jpath = alloca (cur_tp_name_len + 3); + const char *propval_q = nullptr; + int propval_q_len = 0; + int propval_type; + snprintf ((char *) cur_tp_jpath, cur_tp_name_len + 3, "$.%.*s", (int) cur_tp_name_len, cur_tp_name); + cur_tprop->flags &= ~ANYDATA_FL_PRESENT; + propval_type = mjson_find ((const char *) callargs, (int) callargs_len, + (const char *) cur_tp_jpath, &propval_q, &propval_q_len); + if (propval_type == MJSON_TOK_INVALID) + continue; + if (false) + ; + else if (cur_tp_type == MCPC_I32) + { + double propval = 0; + if (propval_type != MJSON_TOK_NUMBER + || mjson_get_number (propval_q, propval_q_len, "$", &propval) != 1 || !isfinite (propval) + || trunc (propval) != propval || propval < INT32_MIN || propval > INT32_MAX) + { + jsonrpc_return_error (r, JSONRPC_ERROR_BAD_PARAMS, "integer argument is invalid", "%.*Q", + cur_tp_name_len, cur_tp_name); + goto output_res; + } + mcpc_toolprop_set_i32 (cur_tprop, (i32_t) propval); + } + else if (cur_tp_type == MCPC_U8STR) + { + char8_t *propval; + int propval_len; + if (propval_type != MJSON_TOK_STRING) + { + jsonrpc_return_error (r, JSONRPC_ERROR_BAD_PARAMS, "string argument is invalid", "%.*Q", + cur_tp_name_len, cur_tp_name); + goto output_res; + } + propval = mcpc_alloc ((size_t) propval_q_len + 1); + if (propval == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_INTERNAL, "out of memory", NULL); + goto output_res; + } + propval_len = mjson_get_string (propval_q, propval_q_len, "$", (char *) propval, + (size_t) propval_q_len + 1); + if (propval_len < 0) + { + free (propval); + jsonrpc_return_error (r, JSONRPC_ERROR_BAD_PARAMS, "string argument is invalid", NULL); + goto output_res; + } + mcpc_toolprop_set_u8str (cur_tprop, propval, (size_t) propval_len); + free (propval); + } + else + { + jsonrpc_return_error (r, JSONRPC_ERROR_INTERNAL, "unsupported argument type", NULL); + goto output_res; + } + } + + ucbr = mcpc_ucbr_new (); + if (ucbr == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_INTERNAL, "out of memory", NULL); + goto output_res; + } + + mcpc_tcallcb_t callcb = nullptr; + if (mcpc_tool_get_call_cb (tool, &callcb) != MCPC_EC_0 || callcb == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_INTERNAL, "tool callback is unavailable", NULL); + goto output_res; + } + callcb (tool, ucbr); + + size_t buftpool_len = 0; + const size_t buftpool_cap = mcpc_ucbr_get_serlz_lenhint (ucbr); + if (buftpool_cap == 0) + { + jsonrpc_return_error (r, JSONRPC_ERROR_INTERNAL, "response is too large", NULL); + goto output_res; + } + buftpool = mcpc_alloc (buftpool_cap); + if (buftpool == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_INTERNAL, "out of memory", NULL); + goto output_res; + } + retbuf_t rtbf = { buftpool_cap, buftpool, &buftpool_len }; + + serlz_tool_callres (rtbf, ucbr); + if (buftpool_len == 0 || buftpool_len >= buftpool_cap) + { + jsonrpc_return_error (r, JSONRPC_ERROR_INTERNAL, "response serialization failed", NULL); + goto output_res; + } + + if (!ensure_buf_suffi (&sv->rpcres, buftpool_cap)) + { + jsonrpc_return_error (r, JSONRPC_ERROR_INTERNAL, "response buffer allocation failed", NULL); + goto output_res; + } + jsonrpc_return_success2 (r, mcpc_sock_invalid (csock), "%.*s", buftpool_len, buftpool); + +output_res: + if (ucbr != nullptr) + mcpc_ucbr_free (ucbr); + free (buftpool); + return; +} + +void +_handle_prmpt_call (mcpc_server_t *sv, struct jsonrpc_request *r, mcpc_sock_t csock) +{ + if (!mcpc_server_supp_prompts (sv)) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "prompts is not supported", NULL); + goto output_res; + } + + mcpc_conn_t *conn = mcpc_connpool_find_by_sock (sv->connpool, csock); + if (conn == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "connection not found", NULL); + goto output_res; + } + if (conn->client_init != MCPC_INIT_SUCC) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "client not initialized", NULL); + goto output_res; + } + + char8_t *callname = nullptr; + size_t callname_len = 0; + + callname = alloca (128); // TODO insuffi? + if (MCPC_EC_0 != _tulc_jparams_get_field_u8str (r, u8c1 ("$.name"), &callname, &callname_len, 1024, true)) + return; + + bug_if (callname_len == 0); + + const mcpc_prmpt_t *prmpt = (const mcpc_prmpt_t *) mcpc_prmptpool_getbyname (sv->prmptpool, callname, callname_len); + if (prmpt == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "prmopt not found", NULL); + goto output_res; + } + + + const char8_t *callargs = nullptr; + size_t callargs_len = 0; + callargs = alloca (1024); // TODO insuffi? + if (MCPC_EC_0 != _tulc_jparams_getref_field_obj (r, u8c1 ("$.arguments"), &callargs, &callargs_len)) + return; + + const mcpc_prmptarg_t *cur_parg = nullptr; + while ((cur_parg = mcpc_prmpt_get_next_prmptarg (prmpt))) + { + const char8_t *cur_tp_name = nullptr; + size_t cur_tp_name_len = 0; + bug_if_nz (mcpc_prmptarg_getref_name (cur_parg, &cur_tp_name, &cur_tp_name_len)); + mcpc_anytype_t cur_tp_type = MCPC_NONE; + bug_if_nz (mcpc_prmptarg_get_data_type (cur_parg, &cur_tp_type)); + char8_t *cur_tp_jpath = alloca (100); + sprintf ((char *) cur_tp_jpath, "$.%.*s", (int) cur_tp_name_len, cur_tp_name); + if (false) + ; + else if (cur_tp_type == MCPC_U8STR) + { + char8_t *propval = alloca (1024); // TODO might be insuff + size_t propval_len = 0; + + // copy is essential here + if (MCPC_EC_0 != + _tulc_jstr_get_field_u8str (r, callargs, callargs_len, cur_tp_jpath, &propval, &propval_len, 1024, true)) + return; + + mcpc_prmptarg_set_u8str ((mcpc_prmptarg_t *) cur_parg, propval, (size_t) propval_len); + } + } + + mcpc_ucbr_t *ucbr = mcpc_ucbr_new (); + + mcpc_prmpt_callcb_t callcb = nullptr; + mcpc_prmpt_get_callcb (prmpt, &callcb); + callcb (prmpt, ucbr); + + char *buftpool = nullptr; + size_t buftpool_len = 0; + const size_t buftpool_cap = 1024; + buftpool = mcpc_alloc (buftpool_cap); + retbuf_t rtbf = { buftpool_cap, buftpool, &buftpool_len }; + + serlz_prmpt_callres (rtbf, ucbr); + mcpc_assert (buftpool != nullptr, MCPC_EC_BUG); + mcpc_assert (buftpool_len > 0, MCPC_EC_BUG); + + jsonrpc_return_success2 (r, mcpc_sock_invalid (csock), "%.*s", buftpool_len, buftpool); + + mcpc_ucbr_free (ucbr); + free (buftpool); // TODO intn_free + +output_res: + return; +} + + +void +_handle_complt_complt (mcpc_server_t *sv, struct jsonrpc_request *r, mcpc_sock_t csock) +{ + if (!mcpc_server_supp_complt (sv)) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "completion is not supported", NULL); + goto output_res; + } + + mcpc_conn_t *conn = mcpc_connpool_find_by_sock (sv->connpool, csock); + if (conn == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "connection not found", NULL); + goto output_res; + } + if (conn->client_init != MCPC_INIT_SUCC) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "client not initialized", NULL); + goto output_res; + } + + + // ref.type + char8_t *compref_type = alloca (1024); // TODO suff? + size_t compref_type_len = 0; + if (MCPC_EC_0 != _tulc_jparams_get_field_u8str (r, u8c1 ("$.ref.type"), &compref_type, &compref_type_len, 1024, true)) + return; + + int askfor_what = -1; + + if (false); + else if (0 == u8strncmp (compref_type, u8c1 ("ref/prompt"), compref_type_len)) + askfor_what = 1; + else if (0 == u8strncmp (compref_type, u8c1 ("ref/resource"), compref_type_len)) + askfor_what = 2; + + if (askfor_what < 0) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "<ref.type> invalid", NULL); + return; + } + + char8_t *compref_name = nullptr; // TODO suff? + size_t compref_name_len = 0; + char8_t *compref_uri = nullptr; // TODO suff? + size_t compref_uri_len = 0; + char8_t *comparg_name = nullptr; // TODO suff? + size_t comparg_name_len = 0; + char8_t *comparg_val = nullptr; // TODO suff? + size_t comparg_val_len = 0; + + bool askfor_prmpt = askfor_what == 1; + bool askfor_resc = askfor_what == 2; + + mcpc_complt_t *complt = nullptr; + + if (false); + else if (askfor_prmpt) + { + // ref.name + compref_name = alloca (1024); // TODO insuffi? + if (MCPC_EC_0 != + _tulc_jparams_get_field_u8str (r, u8c1 ("$.ref.name"), &compref_name, &compref_name_len, 1024, true)) + return; + + // argument.name + comparg_name = alloca (1024); // TODO insuffi? + if (MCPC_EC_0 != + _tulc_jparams_get_field_u8str (r, u8c1 ("$.argument.name"), &comparg_name, &comparg_name_len, 1024, true)) + return; + + // argument.value + comparg_val = alloca (1024); // TODO insuffi? + if (MCPC_EC_0 != + _tulc_jparams_get_field_u8str (r, u8c1 ("$.argument.value"), &comparg_val, &comparg_val_len, 1024, false)) + return; + + const mcpc_prmpt_t *tgt_prmpt = mcpc_prmptpool_getbyname (sv->prmptpool, compref_name, compref_name_len); + if (tgt_prmpt == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "prompt not found", "%.*Q", compref_name_len, compref_name); + return; + } + + const mcpc_prmptarg_t *tgt_prmptarg = mcpc_prmpt_getbyname_prmptarg (tgt_prmpt, comparg_name, comparg_name_len); + if (tgt_prmptarg == nullptr) + { + jsonrpc_return_error (r, JSONRPC_ERROR_NOT_FOUND, "promptarg not found", "%.*Q", comparg_name_len, + comparg_name); + return; + } + + complt = mcpc_complt_new_prmptarg (tgt_prmpt, comparg_name, comparg_name_len, comparg_val, comparg_val_len); + } + else if (askfor_resc) + { + // ref.uri mycomm: rubbish design + compref_uri = alloca (1024); + if (MCPC_EC_0 != + _tulc_jparams_get_field_u8str (r, u8c1 ("$.ref.uri"), &compref_uri, &compref_uri_len, 1024, false)) + return; + complt = mcpc_complt_new_rsc (sv->rscpool, compref_uri, compref_uri_len); + } + + bug_if_nullptr (complt); + + char *buftpool = nullptr; + size_t buftpool_len = 0; + const size_t buftpool_cap = 1024; + buftpool = mcpc_alloc (buftpool_cap); + retbuf_t rtbf = { buftpool_cap, buftpool, &buftpool_len }; + + serlz_complt (rtbf, complt); + mcpc_complt_free (complt); + + // buftpool_len = + // sprintf (buftpool, " %.*s %.*s %.*s %.*s", (int) comparg_name_len, comparg_name, (int) comparg_val_len, comparg_val, + // (int) compref_type_len, compref_type, (int) compref_type_len, compref_type); + // strcpy (buftpool, "xxxxx"); + + mcpc_assert (buftpool != nullptr, MCPC_EC_BUG); + mcpc_assert (buftpool_len > 0, MCPC_EC_BUG); + + jsonrpc_return_success2 (r, mcpc_sock_invalid (csock), "%.*s", buftpool_len, buftpool); + + free (buftpool); // TODO intn_free + +output_res: + return; +} + +bool +is_metho_need_params (mcpc_metho_t metho) +{ + mcpc_assert (metho != MCPC_METHO_UNKNOWN, MCPC_EC_BUG); + if (metho == MCPC_METHO_INITBEG) + return true; + if (metho == MCPC_METHO_CALLTOOL) + return true; + if (metho == MCPC_METHO_PRMPT_CALL) + return true; + if (metho == MCPC_METHO_COMPLT_COMPLT) + return true; + return false; +} + +void +handle_inone (mcpc_server_t *sv, const char *const rbuf, size_t nread, mcpc_sock_t csock) +{ + bool noreply = false; + bool has_id = false; + char8_t *methodname = nullptr; + int32_t methodname_len = 0; + int parsed; + size_t offset; + tuflog_d ("rbuf:%.*s", nread, rbuf); + struct jsonrpc_request r = { 0 }; + sv->rpcres.len = 0; + r.ctx = &jsonrpc_default_context; + r.frame = rbuf; + r.frame_len = (int) nread; + r.fn = mjson_print_fixed_buf; + r.fn_data = (void *) &sv->rpcres; + + parsed = nread > INT_MAX ? MJSON_ERROR_INVALID_INPUT : mjson (rbuf, (int) nread, nullptr, nullptr); + if (parsed < 0) + { + jsonrpc_return_error (&r, JSONRPC_ERROR_INVALID, "parse error", NULL); + goto output_res; + } + for (offset = (size_t) parsed; offset < nread; ++offset) + if (rbuf[offset] != ' ' && rbuf[offset] != '\t' && rbuf[offset] != '\r') + { + jsonrpc_return_error (&r, JSONRPC_ERROR_INVALID, "parse error", NULL); + goto output_res; + } + offset = 0; + while (offset < nread && (rbuf[offset] == ' ' || rbuf[offset] == '\t' || rbuf[offset] == '\r')) + offset += 1; + if (offset == nread || rbuf[offset] != '{') + { + jsonrpc_return_error (&r, JSONRPC_ERROR_INVALID_REQUEST, "invalid request", NULL); + goto output_res; + } + + const char *jsonrpc_q = nullptr; + int jsonrpc_q_len = 0; + char jsonrpc[8]; + if (mjson_find (rbuf, (int) nread, "$.jsonrpc", &jsonrpc_q, &jsonrpc_q_len) != MJSON_TOK_STRING + || mjson_get_string (jsonrpc_q, jsonrpc_q_len, "$", jsonrpc, sizeof (jsonrpc)) != 3 + || strcmp (jsonrpc, "2.0") != 0) + { + jsonrpc_return_error (&r, JSONRPC_ERROR_INVALID_REQUEST, "jsonrpc must be 2.0", NULL); + goto output_res; + } + + int id_type = mjson_find (rbuf, (int) nread, "$.id", &r.id, &r.id_len); + has_id = id_type != MJSON_TOK_INVALID; + if (has_id && id_type != MJSON_TOK_STRING && id_type != MJSON_TOK_NUMBER) + { + r.id = nullptr; + r.id_len = 0; + jsonrpc_return_error (&r, JSONRPC_ERROR_INVALID_REQUEST, "id must be a string or number", NULL); + goto output_res; + } + + const char8_t *methodname_q = nullptr; + int32_t methodname_q_len = 0; + if (mjson_find (rbuf, (int) nread, "$.method", (const char **) &methodname_q, &methodname_q_len) != MJSON_TOK_STRING) + { + jsonrpc_return_error (&r, JSONRPC_ERROR_INVALID_REQUEST, "method must be a string", NULL); + goto output_res; + } + if (methodname_q == nullptr || methodname_q_len == 0) + { + jsonrpc_return_error (&r, JSONRPC_ERROR_INVALID_REQUEST, "method must not be empty", NULL); + goto output_res; + } + + methodname = mcpc_alloc ((size_t) methodname_q_len + 1); + if (methodname == nullptr) + { + jsonrpc_return_error (&r, JSONRPC_ERROR_INTERNAL, "out of memory", NULL); + goto output_res; + } + methodname_len = mjson_get_string ((const char *) methodname_q, methodname_q_len, "$", + (char *) methodname, (size_t) methodname_q_len + 1); + if (methodname_len <= 0) + { + jsonrpc_return_error (&r, JSONRPC_ERROR_INVALID_REQUEST, "method is invalid", NULL); + goto output_res; + } + methodname[methodname_len] = 0; + + mcpc_metho_t handle_what = MCPC_METHO_UNKNOWN; // nothing + + if (false) + ; + else if (u8streq (methodname, u8c1 ("initialize"))) + handle_what = MCPC_METHO_INITBEG; + else if (u8streq (methodname, u8c1 ("notifications/initialized"))) + handle_what = MCPC_METHO_INITDONE; + else if (u8streq (methodname, u8c1 ("tools/list"))) + handle_what = MCPC_METHO_LISTTOOL; + else if (u8streq (methodname, u8c1 ("tools/call"))) + handle_what = MCPC_METHO_CALLTOOL; + else if (u8streq (methodname, u8c1 ("resources/list"))) + handle_what = MCPC_METHO_LSRSC; + else if (u8streq (methodname, u8c1 ("prompts/list"))) + handle_what = MCPC_METHO_PRMPT_LS; + else if (u8streq (methodname, u8c1 ("prompts/get"))) + handle_what = MCPC_METHO_PRMPT_CALL; + else if (u8streq (methodname, u8c1 ("completion/complete"))) + handle_what = MCPC_METHO_COMPLT_COMPLT; + + if (handle_what == MCPC_METHO_UNKNOWN) + { + if (!has_id) + noreply = true; + else + jsonrpc_return_error (&r, JSONRPC_ERROR_NOT_FOUND, "method not found", NULL); + goto output_res; + } + + const char *params = nullptr; + int params_len = 0; + int params_type = mjson_find (rbuf, (int) nread, "$.params", ¶ms, ¶ms_len); + if (params_type != MJSON_TOK_INVALID && params_type != MJSON_TOK_OBJECT) + { + if (!has_id) + noreply = true; + else + jsonrpc_return_error (&r, JSONRPC_ERROR_BAD_PARAMS, "params must be an object", NULL); + goto output_res; + } + r.params = params_type == MJSON_TOK_OBJECT ? params : ""; + r.params_len = params_type == MJSON_TOK_OBJECT ? params_len : 0; + + if (is_metho_need_params (handle_what) && r.params_len < 1) + { + if (!has_id) + noreply = true; + else + jsonrpc_return_error (&r, JSONRPC_ERROR_BAD_PARAMS, "params are required", NULL); + goto output_res; + } + + if (!has_id && handle_what != MCPC_METHO_INITDONE) + { + noreply = true; + goto output_res; + } + + // Windows SOCKET is unsigned, while Unix file descriptors use a negative + // sentinel for stdio dispatch. +#if is_win + if (!mcpc_sock_invalid (csock)) +#elif defined(is_unix) + if (csock > 0) +#endif + { // TODO: more accurate? + mcpc_conn_t *conn0 = mcpc_conn_new (MCPC_CONN_FD, &csock); + mcpc_assert (MCPC_EC_0 == mcpc_connpool_addref ((mcpc_connpool_t *) sv->connpool, conn0), MCPC_EC_BUG); + } + + if (false) + ; + else if (handle_what == MCPC_METHO_INITBEG) + _handle_initbeg (sv, &r, csock); + else if (handle_what == MCPC_METHO_INITDONE) + _handle_initdone (sv, &r, csock, &noreply); + else if (handle_what == MCPC_METHO_LISTTOOL) + _handle_tools_list (sv, &r, csock); + else if (handle_what == MCPC_METHO_LSRSC) + _handle_lsrsc (sv, &r, csock); + else if (handle_what == MCPC_METHO_CALLTOOL) + _handle_tools_call (sv, &r, csock); + else if (handle_what == MCPC_METHO_PRMPT_LS) + _handle_prmpt_list (sv, &r, csock); + else if (handle_what == MCPC_METHO_PRMPT_CALL) + _handle_prmpt_call (sv, &r, csock); + else if (handle_what == MCPC_METHO_COMPLT_COMPLT) + _handle_complt_complt (sv, &r, csock); + +output_res: + free (methodname); + if (noreply) + return; + if (false) + ; + else if (sv->typ == MCPC_SV_IOSTRM) + { + tuflog_d ("sv->rpcres:%.*s", sv->rpcres.len, sv->rpcres.ptr); + mcpc_conn_t *outconn = mcpc_connpool_getconn (sv->connpool, 1); + FILE *outstrm = mcpc_conn_getstrm (outconn); + int pres = fprintf (outstrm, "%.*s", sv->rpcres.len, sv->rpcres.ptr); + mcpc_assert (pres >= 0, MCPC_EC_BUG); + fflush (outstrm); + sv->rpcres.len = 0; + } + else if (sv->typ == MCPC_SV_TCP) + { + mcpc_assert (0 < send (csock, sv->rpcres.ptr, sv->rpcres.len, 0), MCPC_EC_BUG); + sv->rpcres.len = 0; + } +} diff --git a/third_party/mcpc/src/server.h b/third_party/mcpc/src/server.h new file mode 100644 index 0000000..e2c4141 --- /dev/null +++ b/third_party/mcpc/src/server.h @@ -0,0 +1,196 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_server +#define h_server + +#include <mcpc/visb.h> +#include <mcpc/server.h> + +#include "ext_stdint.h" + +#include "errcode.h" +#include "mjson.h" +#include "rsc.h" +#include "tool.h" +#include "prmpt.h" + +#include <stdio.h> + +#if is_win +#include <winsock2.h> +typedef i32_t socklen_t; +#endif + +#define BUFFER_SIZE 1024 +#define PORT 8080 + +// ------------------------------ conn ------------------------------ + +enum mcpc_conn_type +{ + MCPC_CONN_FD, + MCPC_CONN_STRM, +}; +typedef enum mcpc_conn_type mcpc_conn_type_t; + +#if defined(is_unix) +typedef int mcpc_sock_t; +#elif is_win +typedef SOCKET mcpc_sock_t; +#endif + +static inline bool +mcpc_sock_invalid (mcpc_sock_t sock) +{ +#if defined(is_unix) + return sock < 0; +#endif +#if is_win + return sock == INVALID_SOCKET; +#endif +} + +enum mcpc_init_stat +{ + MCPC_INIT_NONE = 0, + MCPC_INIT_BEG, + MCPC_INIT_FAIL, + MCPC_INIT_SUCC, +}; +typedef enum mcpc_init_stat mcpc_init_stat_t; + +typedef struct mcpc_conn mcpc_conn_t; +struct mcpc_conn +{ + mcpc_conn_type_t typ; + union + { + mcpc_sock_t fd; + FILE *strm; + }; + const ch8_t *cname; + u8_t cname_len; + mcpc_init_stat_t client_init; + mcpc_conn_t *nex; +}; + +typedef uint8_t mcpc_connpool_size_t; + +typedef struct mcpc_connpool mcpc_connpool_t; +struct mcpc_connpool +{ + const mcpc_connpool_size_t cap; + mcpc_connpool_size_t len; + mcpc_conn_t *head; +}; + +// ----------------------------- server ----------------------------- + +enum mcpc_server_type +{ + MCPC_SV_NONE = 0, + MCPC_SV_IOSTRM, + MCPC_SV_STDIO, + MCPC_SV_TCP, +}; +typedef enum mcpc_server_type mcpc_server_type_t; + +struct mcpc_server +{ + mcpc_server_type_t typ; + const ch8_t *name; + u8_t name_len; + const ch8_t *ver; + u8_t ver_len; + u16_t capa; + const mcpc_toolpool_t *toolpool; + const mcpc_prmptpool_t *prmptpool; + const mcpc_rscpool_t *rscpool; + struct mjson_fixedbuf rpcres; + const mcpc_connpool_t *connpool; +}; +// implication: for iostrm server, the first conn will be used as client init status storing +// check mcpc_connpool_find_by_sock + +#define MCPC_CAPA_PRMT 0b0000000000000001 +#define MCPC_CAPA_PRMT_LISTCHG 0b0000000000000011 +#define MCPC_CAPA_TOOL 0b0000000000010000 +#define MCPC_CAPA_TOOL_LISTCHG 0b0000000000110000 +#define MCPC_CAPA_RSC 0b0000000100000000 +#define MCPC_CAPA_RSC_LISTCHG 0b0000001100000000 +#define MCPC_CAPA_RSC_SUBSRC 0b0000010100000000 +#define MCPC_CAPA_COMPLT 0b0001000000000000 + +mcpc_server_t *mcpc_server_new (mcpc_server_type_t typ, mcpc_toolpool_t * toolpool, u32_t vargs_beg, ...); + +mcpc_server_t *mcpc_server_new_stdio (); + +/* void _handle_tools_list(mcpc_server_t *sv, struct jsonrpc_request *r); */ + +/* void _handle_tools_call(mcpc_server_t *sv, struct jsonrpc_request *r); */ + +mcpc_errcode_t mcpc_server_cleanup_rpcres (mcpc_server_t * sv); + +void handle_inone (mcpc_server_t * sv, const char *const rbuf, size_t nread, mcpc_sock_t csock); + +bool mcpc_server_supp_prompts (const mcpc_server_t * sv); + +bool mcpc_server_supp_prompts_lschg (const mcpc_server_t * sv); + +bool mcpc_server_supp_tools (const mcpc_server_t * sv); + +bool mcpc_server_supp_tools_lschg (const mcpc_server_t * sv); + +bool mcpc_server_supp_rscs (const mcpc_server_t * sv); + +bool mcpc_server_supp_rscs_lschg (const mcpc_server_t * sv); + +bool mcpc_server_supp_rscs_subs (const mcpc_server_t * sv); + +enum mcpc_metho +{ + MCPC_METHO_UNKNOWN = 0, + MCPC_METHO_INITBEG, + MCPC_METHO_INITDONE, + MCPC_METHO_LISTTOOL, + MCPC_METHO_CALLTOOL, + MCPC_METHO_LSRSC, + MCPC_METHO_PRMPT_LS, + MCPC_METHO_PRMPT_CALL, + MCPC_METHO_COMPLT_COMPLT, +}; +typedef enum mcpc_metho mcpc_metho_t; + +// ---------------------------- connpool ---------------------------- + +mcpc_conn_t *mcpc_conn_new (mcpc_conn_type_t typ, void *back); + +FILE *mcpc_conn_getstrm (mcpc_conn_t * conn); + +int mcpc_conn_getfd (mcpc_conn_t * conn); + +// TODO private +mcpc_errcode_t mcpc_connpool_addref (mcpc_connpool_t * connpool, mcpc_conn_t * conn); + +mcpc_conn_t *mcpc_connpool_getconn (const mcpc_connpool_t * connpool, const mcpc_connpool_size_t idx); + +#endif diff --git a/third_party/mcpc/src/tool.c b/third_party/mcpc/src/tool.c new file mode 100644 index 0000000..7c746ec --- /dev/null +++ b/third_party/mcpc/src/tool.c @@ -0,0 +1,505 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "tool.h" + +#include "ext_string.h" + +#include <stdio.h> +#include <stdarg.h> +#include <stdlib.h> +#include "errcode.h" +#include "alloc.h" +#include "anydata.h" + + +/* = = = = = = = = = = = = = = = toolprop = = = = = = = = = = = = = = = */ + +MCPC_API mcpc_toolprop_t * +mcpc_toolprop_new (const char8_t *name, const size_t name_len, const char8_t *desc, const size_t desc_len, + mcpc_anytype_t typ) +{ + mcpc_anydata_t *hp = mcpc_anydata_new_typ_meta1_meta2 (typ, name, name_len, desc, desc_len); + + hp->flags |= ANYDATA_FL_REQUIRED; + hp->flags |= ANYDATA_FL_TOOLPROP; + + return hp; +} + +MCPC_API mcpc_toolprop_t * +mcpc_toolprop_new2 (const char8_t *nament, const char8_t *descnt, mcpc_anytype_t typ) +{ + return mcpc_toolprop_new (nament, strlen ((const char *) nament), descnt, strlen ((const char *) descnt), typ); +} + +MCPC_API bool +mcpc_toolprop_is_required (const mcpc_toolprop_t *toolprop) +{ + return toolprop != nullptr && (toolprop->flags & ANYDATA_FL_REQUIRED) != 0; +} + +MCPC_API mcpc_errcode_t +mcpc_toolprop_set_required (mcpc_toolprop_t *toolprop, bool required) +{ + if (toolprop == nullptr) + return MCPC_EC_BUG; + if (required) + toolprop->flags |= ANYDATA_FL_REQUIRED; + else + toolprop->flags &= ~ANYDATA_FL_REQUIRED; + return MCPC_EC_0; +} + +void +mcpc_toolprop_free (mcpc_toolprop_t *toolprop) +{ + mcpc_anydata_free ((mcpc_anydata_t *) toolprop); +} + +mcpc_errcode_t +mcpc_toolprop_getref_name (const mcpc_toolprop_t *toolprop, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta1 (toolprop, ret, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_getref_desc (const mcpc_toolprop_t *toolprop, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta2 (toolprop, ret, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_data_type (const mcpc_toolprop_t *toolprop, mcpc_anytype_t *ret) +{ + bug_if_nullptr (toolprop); + bug_if_nullptr (ret); + *ret = ((const mcpc_anydata_t *) toolprop)->data_typ; + return MCPC_EC_0; +} + +mcpc_errcode_t +mcpc_toolprop_set_nament (mcpc_toolprop_t *toolprop, const char8_t *name) +{ + return mcpc_anydata_set_meta1nt (toolprop, name); +} + +mcpc_errcode_t +mcpc_toolprop_set_i8 (mcpc_toolprop_t *toolprop, i8_t val) +{ + return mcpc_anydata_set_i8 (toolprop, val); +} + +mcpc_errcode_t +mcpc_toolprop_set_u8 (mcpc_toolprop_t *toolprop, u8_t val) +{ + return mcpc_anydata_set_u8 (toolprop, val); +} + +mcpc_errcode_t +mcpc_toolprop_set_i16 (mcpc_toolprop_t *toolprop, i16_t val) +{ + return mcpc_anydata_set_i16 (toolprop, val); +} + +mcpc_errcode_t +mcpc_toolprop_set_u16 (mcpc_toolprop_t *toolprop, u16_t val) +{ + return mcpc_anydata_set_u16 (toolprop, val); +} + +mcpc_errcode_t +mcpc_toolprop_set_i32 (mcpc_toolprop_t *toolprop, i32_t val) +{ + mcpc_errcode_t status = mcpc_anydata_set_i32 (toolprop, val); + if (status == MCPC_EC_0) + toolprop->flags |= ANYDATA_FL_PRESENT; + return status; +} + +mcpc_errcode_t +mcpc_toolprop_set_u32 (mcpc_toolprop_t *toolprop, u32_t val) +{ + return mcpc_anydata_set_u32 (toolprop, val); +} + +mcpc_errcode_t +mcpc_toolprop_set_i64 (mcpc_toolprop_t *toolprop, i64_t val) +{ + return mcpc_anydata_set_i64 (toolprop, val); +} + +mcpc_errcode_t +mcpc_toolprop_set_u64 (mcpc_toolprop_t *toolprop, u64_t val) +{ + return mcpc_anydata_set_u64 (toolprop, val); +} + +mcpc_errcode_t +mcpc_toolprop_set_func (mcpc_toolprop_t *toolprop, void *val) +{ + return mcpc_anydata_set_func (toolprop, val); +} + +mcpc_errcode_t +mcpc_toolprop_get_i8 (const mcpc_toolprop_t *toolprop, i8_t *ret) +{ + return mcpc_anydata_get_i8 (toolprop, ret); +} + +mcpc_errcode_t +mcpc_toolprop_get_u8 (const mcpc_toolprop_t *toolprop, u8_t *ret) +{ + return mcpc_anydata_get_u8 (toolprop, ret); +} + +mcpc_errcode_t +mcpc_toolprop_get_i16 (const mcpc_toolprop_t *toolprop, i16_t *ret) +{ + return mcpc_anydata_get_i16 (toolprop, ret); +} + +mcpc_errcode_t +mcpc_toolprop_get_u16 (const mcpc_toolprop_t *toolprop, u16_t *ret) +{ + return mcpc_anydata_get_u16 (toolprop, ret); +} + +mcpc_errcode_t +mcpc_toolprop_get_i32 (const mcpc_toolprop_t *toolprop, i32_t *ret) +{ + return mcpc_anydata_get_i32 (toolprop, ret); +} + +mcpc_errcode_t +mcpc_toolprop_get_u32 (const mcpc_toolprop_t *toolprop, u32_t *ret) +{ + return mcpc_anydata_get_u32 (toolprop, ret); +} + +mcpc_errcode_t +mcpc_toolprop_get_i64 (const mcpc_toolprop_t *toolprop, i64_t *ret) +{ + return mcpc_anydata_get_i64 (toolprop, ret); +} + +mcpc_errcode_t +mcpc_toolprop_get_u64 (const mcpc_toolprop_t *toolprop, u64_t *ret) +{ + return mcpc_anydata_get_u64 (toolprop, ret); +} + +mcpc_errcode_t +mcpc_toolprop_set_i8s (mcpc_toolprop_t *toolprop, const i8_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_i8s (toolprop, arr, arr_len); +} + +mcpc_errcode_t +mcpc_toolprop_set_u8s (mcpc_toolprop_t *toolprop, const u8_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_u8s (toolprop, arr, arr_len); +} + +mcpc_errcode_t +mcpc_toolprop_set_i16s (mcpc_toolprop_t *toolprop, const i16_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_i16s (toolprop, arr, arr_len); +} + +mcpc_errcode_t +mcpc_toolprop_set_u16s (mcpc_toolprop_t *toolprop, const u16_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_u16s (toolprop, arr, arr_len); +} + +mcpc_errcode_t +mcpc_toolprop_set_i32s (mcpc_toolprop_t *toolprop, const i32_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_i32s (toolprop, arr, arr_len); +} + +mcpc_errcode_t +mcpc_toolprop_set_u32s (mcpc_toolprop_t *toolprop, const u32_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_u32s (toolprop, arr, arr_len); +} + +mcpc_errcode_t +mcpc_toolprop_set_i64s (mcpc_toolprop_t *toolprop, const i64_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_i64s (toolprop, arr, arr_len); +} + +mcpc_errcode_t +mcpc_toolprop_set_u64s (mcpc_toolprop_t *toolprop, const u64_t *arr, size_t arr_len) +{ + return mcpc_anydata_set_u64s (toolprop, arr, arr_len); +} + +mcpc_errcode_t +mcpc_toolprop_set_u8str (mcpc_toolprop_t *toolprop, const char8_t *arr, size_t arr_len) +{ + mcpc_errcode_t status = mcpc_anydata_set_u8str (toolprop, arr, arr_len); + if (status == MCPC_EC_0) + toolprop->flags |= ANYDATA_FL_PRESENT; + return status; +} + +mcpc_errcode_t +mcpc_toolprop_get_i8s (const mcpc_toolprop_t *toolprop, i8_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_i8s (toolprop, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_u8s (const mcpc_toolprop_t *toolprop, u8_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_u8s (toolprop, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_i16s (const mcpc_toolprop_t *toolprop, i16_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_i16s (toolprop, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_u16s (const mcpc_toolprop_t *toolprop, u16_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_u16s (toolprop, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_i32s (const mcpc_toolprop_t *toolprop, i32_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_i32s (toolprop, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_u32s (const mcpc_toolprop_t *toolprop, u32_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_u32s (toolprop, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_i64s (const mcpc_toolprop_t *toolprop, i64_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_i64s (toolprop, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_u64s (const mcpc_toolprop_t *toolprop, u64_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_u64s (toolprop, ret, ret_cap, ret_len); +} + +mcpc_errcode_t +mcpc_toolprop_get_u8str (const mcpc_toolprop_t *toolprop, char8_t *ret, size_t ret_cap, size_t *ret_len) +{ + return mcpc_anydata_get_u8str (toolprop, ret, ret_cap, ret_len); +} + +/* = = = = = = = = = = = = = toolproppool = = = = = = = = = = = = = */ + +size_t +mcpc_toolproppool_get_len (const mcpc_toolproppool_t *toolproppool) +{ + bug_if_nullptr (toolproppool); + size_t ret = 0; + ret = ((const mcpc_anypool_t *) toolproppool)->len; + return ret; +} + +const mcpc_toolprop_t * +mcpc_toolproppool_get (const mcpc_toolproppool_t *toolproppool, size_t idx) +{ + bug_if_nullptr (toolproppool); + return ((mcpc_anypool_t *) toolproppool)->head + idx; +} + +/* = = = = = = = = = = = = = = = = = tool = = = = = = = = = = = = = = = = = */ + +MCPC_API mcpc_tool_t * +mcpc_tool_new (const char8_t *name, const size_t name_len, const char8_t *desc, const size_t desc_len) +{ + mcpc_anydata_t *hp = mcpc_anydata_new_meta1_meta2 (name, name_len, desc, desc_len); + + hp->flags |= ANYDATA_FL_TOOL; + + return hp; +} + +MCPC_API mcpc_tool_t * +mcpc_tool_new2 (const char8_t *nament, const char8_t *descnt) +{ + return mcpc_tool_new (nament, strlen ((const char *) nament), descnt, strlen ((const char *) descnt)); +} + +void +mcpc_tool_free (mcpc_tool_t *tool) +{ + mcpc_anydata_free ((mcpc_anydata_t *) tool); +} + +mcpc_errcode_t +mcpc_tool_getref_name (const mcpc_tool_t *tool, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta1 (tool, ret, ret_len); +} + +mcpc_errcode_t +mcpc_tool_getref_desc (const mcpc_tool_t *tool, const char8_t **ret, size_t *ret_len) +{ + return mcpc_anydata_getref_meta2 (tool, ret, ret_len); +} + +MCPC_API mcpc_errcode_t +mcpc_tool_get_call_cb (const mcpc_tool_t *tool, mcpc_tcallcb_t *ret) +{ + return mcpc_anydata_get_func (tool, ret); +} + +MCPC_API mcpc_errcode_t +mcpc_tool_set_call_cb (mcpc_tool_t *tool, mcpc_tcallcb_t cb) +{ + mcpc_anydata_set_func (tool, cb); + return MCPC_EC_0; +} + +MCPC_API void +mcpc_tool_addfre_toolprop (mcpc_tool_t *tool, mcpc_toolprop_t *toolprop) +{ + mcpc_anydata_add_child (tool, toolprop); // TODO return EC +} + +MCPC_API void +mcpc_tool_addcpy_toolprop (mcpc_tool_t *tool, mcpc_toolprop_t *toolprop) +{ + mcpc_anydata_addcpy_child (tool, toolprop); // TODO return EC +} + +MCPC_API const mcpc_toolprop_t * +mcpc_tool_get_toolprop (const mcpc_tool_t *tool, size_t idx) +{ + return mcpc_anydata_get_child (tool, idx); +} + +size_t +mcpc_tool_get_toolprop_len (const mcpc_tool_t *tool) +{ + bug_if_nullptr (tool); + mcpc_anydata_t *tool_imp = (mcpc_anydata_t *) tool; + if (tool_imp->chd_pool == nullptr) + return 0; + return tool_imp->chd_pool->len; +} + +const mcpc_toolprop_t * +mcpc_tool_getbynament_toolprop (const mcpc_tool_t *tool, const char8_t *tprop_nament) +{ + const mcpc_anydata_t *chd = mcpc_anypool_getbymeta1nt (tool->chd_pool, tprop_nament); + return (mcpc_toolprop_t *) chd; +} + +MCPC_API mcpc_errcode_t +mcpc_tool_get_tpropval_i32 (const mcpc_tool_t *tool, const char8_t *tprop_nament, i32_t *ret) +{ + bug_if_nullptr (tool); + bug_if_nullptr (tprop_nament); + bug_if_nullptr (ret); + const mcpc_toolprop_t *chd = mcpc_tool_getbynament_toolprop (tool, tprop_nament); + if (chd == nullptr) + return MCPC_EC_TOOLPROP_NOTFOUND; + if ((chd->flags & ANYDATA_FL_PRESENT) == 0) + return MCPC_EC_ANYDATA_NOTFOUND; + + return mcpc_toolprop_get_i32 (chd, ret); +} + +MCPC_API mcpc_errcode_t +mcpc_tool_get_tpropval_u8str (const mcpc_tool_t *tool, const char8_t *tprop_nament, char8_t *ret, + size_t ret_cap, size_t *ret_len) +{ + bug_if_nullptr (tool); + bug_if_nullptr (tprop_nament); + bug_if_nullptr (ret); + const mcpc_toolprop_t *chd = mcpc_tool_getbynament_toolprop (tool, tprop_nament); + if (chd == nullptr) + return MCPC_EC_TOOLPROP_NOTFOUND; + if ((chd->flags & ANYDATA_FL_PRESENT) == 0) + return MCPC_EC_ANYDATA_NOTFOUND; + + return mcpc_anydata_get_u8str (chd, ret, ret_cap, ret_len); +} + +const mcpc_toolproppool_t * +mcpc_tool_get_toolproppool (const mcpc_tool_t *tool) +{ + return tool->chd_pool; +} + +/* = = = = = = = = = = = = = = = toolpool = = = = = = = = = = = = = = = */ + +mcpc_toolpool_t * +mcpc_toolpool_new () +{ + return mcpc_anypool_new (); +} + +void +mcpc_toolpool_free (mcpc_toolpool_t *toolpool) +{ + mcpc_anypool_free (toolpool); +} + +void +mcpc_toolpool_addcpy (mcpc_toolpool_t *toolpool, const mcpc_tool_t *tool) +{ + mcpc_anypool_addcpy (toolpool, tool); +} + +void +mcpc_toolpool_addfre (mcpc_toolpool_t *toolpool, mcpc_tool_t *tool) +{ + mcpc_anypool_addfre (toolpool, tool); +} + +size_t +mcpc_toolpool_get_len (const mcpc_toolpool_t *toolpool) +{ + bug_if_nullptr (toolpool); + size_t ret = 0; + ret = ((const mcpc_anypool_t *) toolpool)->len; + return ret; +} + +const mcpc_tool_t * +mcpc_toolpool_get (const mcpc_toolpool_t *toolpool, size_t idx) +{ + return mcpc_anypool_get (toolpool, idx); +} + +const mcpc_tool_t * +mcpc_toolpool_getbyname (const mcpc_toolpool_t *toolpool, const char8_t *name, size_t name_len) +{ + return mcpc_anypool_getbymeta1 (toolpool, name, name_len); +} diff --git a/third_party/mcpc/src/tool.h b/third_party/mcpc/src/tool.h new file mode 100644 index 0000000..7da9bcd --- /dev/null +++ b/third_party/mcpc/src/tool.h @@ -0,0 +1,150 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_tool +#define h_tool + +#include <mcpc/tool.h> + +#include "ext_stdint.h" + +typedef mcpc_anypool_t mcpc_toolpool_t; +typedef mcpc_anypool_t mcpc_toolproppool_t; + +// ---------------------------- toolprop ---------------------------- + +void mcpc_toolprop_free (mcpc_toolprop_t * toolprop); + +mcpc_errcode_t mcpc_toolprop_getref_name (const mcpc_toolprop_t * toolprop, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_getref_desc (const mcpc_toolprop_t * toolprop, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_data_type (const mcpc_toolprop_t * toolprop, mcpc_anytype_t * ret); + +mcpc_errcode_t mcpc_toolprop_set_nament (mcpc_toolprop_t * toolprop, const char8_t * name); + +mcpc_errcode_t mcpc_toolprop_set_i8 (mcpc_toolprop_t * toolprop, i8_t val); + +mcpc_errcode_t mcpc_toolprop_set_u8 (mcpc_toolprop_t * toolprop, u8_t val); + +mcpc_errcode_t mcpc_toolprop_set_i16 (mcpc_toolprop_t * toolprop, i16_t val); + +mcpc_errcode_t mcpc_toolprop_set_u16 (mcpc_toolprop_t * toolprop, u16_t val); + +mcpc_errcode_t mcpc_toolprop_set_i32 (mcpc_toolprop_t * toolprop, i32_t val); + +mcpc_errcode_t mcpc_toolprop_set_u32 (mcpc_toolprop_t * toolprop, u32_t val); + +mcpc_errcode_t mcpc_toolprop_set_i64 (mcpc_toolprop_t * toolprop, i64_t val); + +mcpc_errcode_t mcpc_toolprop_set_u64 (mcpc_toolprop_t * toolprop, u64_t val); + +mcpc_errcode_t mcpc_toolprop_set_func (mcpc_toolprop_t * toolprop, void *val); + +mcpc_errcode_t mcpc_toolprop_get_i8 (const mcpc_toolprop_t * toolprop, i8_t * ret); + +mcpc_errcode_t mcpc_toolprop_get_u8 (const mcpc_toolprop_t * toolprop, u8_t * ret); + +mcpc_errcode_t mcpc_toolprop_get_i16 (const mcpc_toolprop_t * toolprop, i16_t * ret); + +mcpc_errcode_t mcpc_toolprop_get_u16 (const mcpc_toolprop_t * toolprop, u16_t * ret); + +mcpc_errcode_t mcpc_toolprop_get_i32 (const mcpc_toolprop_t * toolprop, i32_t * ret); + +mcpc_errcode_t mcpc_toolprop_get_u32 (const mcpc_toolprop_t * toolprop, u32_t * ret); + +mcpc_errcode_t mcpc_toolprop_get_i64 (const mcpc_toolprop_t * toolprop, i64_t * ret); + +mcpc_errcode_t mcpc_toolprop_get_u64 (const mcpc_toolprop_t * toolprop, u64_t * ret); + +mcpc_errcode_t mcpc_toolprop_get_func (const mcpc_toolprop_t * toolprop, void *ret); + +mcpc_errcode_t mcpc_toolprop_set_i8s (mcpc_toolprop_t * toolprop, const i8_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_set_u8s (mcpc_toolprop_t * toolprop, const u8_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_set_i16s (mcpc_toolprop_t * toolprop, const i16_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_set_u16s (mcpc_toolprop_t * toolprop, const u16_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_set_i32s (mcpc_toolprop_t * toolprop, const i32_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_set_u32s (mcpc_toolprop_t * toolprop, const u32_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_set_i64s (mcpc_toolprop_t * toolprop, const i64_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_set_u64s (mcpc_toolprop_t * toolprop, const u64_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_set_u8str (mcpc_toolprop_t * toolprop, const char8_t * arr, size_t arr_len); + +mcpc_errcode_t mcpc_toolprop_get_i8s (const mcpc_toolprop_t * toolprop, i8_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_u8s (const mcpc_toolprop_t * toolprop, u8_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_i16s (const mcpc_toolprop_t * toolprop, i16_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_u16s (const mcpc_toolprop_t * toolprop, u16_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_i32s (const mcpc_toolprop_t * toolprop, i32_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_u32s (const mcpc_toolprop_t * toolprop, u32_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_i64s (const mcpc_toolprop_t * toolprop, i64_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_u64s (const mcpc_toolprop_t * toolprop, u64_t * ret, size_t ret_cap, size_t *ret_len); + +mcpc_errcode_t mcpc_toolprop_get_u8str (const mcpc_toolprop_t * toolprop, char8_t * ret, size_t ret_cap, + size_t *ret_len); + +const mcpc_toolproppool_t *mcpc_tool_get_toolproppool (const mcpc_tool_t * tool); + +// -------------------------- toolproppool -------------------------- + +size_t mcpc_toolproppool_get_len (const mcpc_toolproppool_t * toolproppool); + +const mcpc_toolprop_t *mcpc_toolproppool_get (const mcpc_toolproppool_t * toolproppool, size_t idx); + +// ------------------------------ tool ------------------------------ + +mcpc_errcode_t mcpc_tool_getref_name (const mcpc_tool_t * tool, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_tool_getref_desc (const mcpc_tool_t * tool, const char8_t ** ret, size_t *ret_len); + +void mcpc_tool_free (mcpc_tool_t * tool); + +// ---------------------------- toolpool ---------------------------- + +mcpc_toolpool_t *mcpc_toolpool_new (); + +void mcpc_toolpool_free (mcpc_toolpool_t * toolpool); + +void mcpc_toolpool_addcpy (mcpc_toolpool_t * toolpool, const mcpc_tool_t * tool); + +void mcpc_toolpool_addfre (mcpc_toolpool_t * toolpool, mcpc_tool_t * tool); + +size_t mcpc_toolpool_get_len (const mcpc_toolpool_t * toolpool); + +const mcpc_tool_t *mcpc_toolpool_get (const mcpc_toolpool_t * toolpool, size_t idx); + +const mcpc_tool_t *mcpc_toolpool_getbyname (const mcpc_toolpool_t * toolpool, const char8_t * name, size_t name_len); + + +#endif diff --git a/third_party/mcpc/src/ucbr.c b/third_party/mcpc/src/ucbr.c new file mode 100644 index 0000000..5dfce26 --- /dev/null +++ b/third_party/mcpc/src/ucbr.c @@ -0,0 +1,288 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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 "ucbr.h" + +#include <errno.h> +#include <stdarg.h> +#include <stdio.h> + +#include "alloc.h" +#include "anydata.h" +#include "errcode.h" +#include "ext_string.h" + +/* = = = = = = = = = = = = = = = = ucbrchd = = = = = = = = = = = = = = = = */ + +mcpc_ucbrchd_t * +mcpc_ucbrchd_new_vprintf8 (const char8_t *fmt, va_list *vargs) +{ + size_t sbuf_cap = 1024; + size_t sbuf_len = 0; + char8_t *sbuf = mcpc_alloc (sbuf_cap); + int printed = 0; + while (true) + { + va_list vargs_cp; + va_copy (vargs_cp, *vargs); + printed + = vsnprintf ((char *)sbuf, sbuf_cap, (const char *)fmt, vargs_cp); + bug_if (printed < 0); + if ((size_t)printed >= sbuf_cap) + { + sbuf_cap += sbuf_cap; + sbuf = mcpc_realloc (sbuf, sbuf_cap); + } + else + { + sbuf_len = (size_t)printed; + break; + } + } + + mcpc_anydata_t *anydata = mcpc_anydata_new_typ (MCPC_U8STR); + + if (MCPC_EC_0 != mcpc_anydata_set_u8str (anydata, sbuf, sbuf_len)) + goto badret; + + free (sbuf); + return (mcpc_ucbrchd_t *)anydata; + +badret: + free (sbuf); + mcpc_anydata_free (anydata); + return nullptr; +} + +mcpc_errcode_t +mcpc_ucbrchd_getref_u8str (const mcpc_ucbrchd_t *ucbrchd, const char8_t **ret, + size_t *ret_len) +{ + return mcpc_anydata_getref_u8str (ucbrchd, ret, ret_len); +} + +mcpc_errcode_t +mcpc_ucbrchd_get_data_type (const mcpc_ucbrchd_t *ucbrchd, mcpc_anytype_t *ret) +{ + bug_if_nullptr (ucbrchd); + bug_if_nullptr (ret); + + *ret = ucbrchd->data_typ; + + return MCPC_EC_0; +} + +bool +mcpc_ucbrchd_is_tool_callres (const mcpc_ucbrchd_t *ucbrchd) +{ + return ucbrchd->flags & ANYDATA_FL_USRCBRES_TOOLCALL; +} + +bool +mcpc_ucbrchd_is_prmpt_callres (const mcpc_ucbrchd_t *ucbrchd) +{ + return ucbrchd->flags & ANYDATA_FL_USRCBRES_PRMPTCALL; +} + +bool +mcpc_ucbrchd_is_assist_prmpt (const mcpc_ucbrchd_t *ucbrchd) +{ + return ucbrchd->flags & ANYDATA_FL_USRCBRES_PRMPTCALL_ASSIST; +} + +/* = = = = = = = = = = = = = = = = = ucbr = = = = = = = = = = = = = = = = = */ + +mcpc_ucbr_t * +mcpc_ucbr_new () +{ + return mcpc_anypool_new (); +} + +void +mcpc_ucbr_free (mcpc_ucbr_t *ucbr) +{ + mcpc_anypool_free (ucbr); +} + +void +mcpc_ucbr_addcpy (mcpc_ucbr_t *ucbr, const mcpc_ucbrchd_t *ucbrchd) +{ + mcpc_anypool_addcpy (ucbr, ucbrchd); +} + +void +mcpc_ucbr_addfre (mcpc_ucbr_t *ucbr, mcpc_ucbrchd_t *ucbrchd) +{ + mcpc_anypool_addfre (ucbr, ucbrchd); +} + +bool +mcpc_ucbr_is_err (const mcpc_ucbr_t *ucbr) +{ + bug_if_nullptr (ucbr); + return ucbr->flags & ANYPOOL_FL_USRCBRES_TOOLCALL_ERROR; +} + +size_t +mcpc_ucbr_get_len (const mcpc_ucbr_t *ucbr) +{ + bug_if_nullptr (ucbr); + size_t ret = 0; + ret = ((const mcpc_anypool_t *)ucbr)->len; + return ret; +} + +const mcpc_ucbrchd_t * +mcpc_ucbr_get (const mcpc_ucbr_t *ucbr, size_t idx) +{ + return mcpc_anypool_get (ucbr, idx); +} + +const mcpc_ucbrchd_t * +mcpc_ucbr_getbyname (const mcpc_ucbr_t *ucbr, const char8_t *name, + size_t name_len) +{ + return mcpc_anypool_getbymeta1 (ucbr, name, name_len); +} + +const mcpc_ucbrchd_t * +mcpc_ucbr_get_next (const mcpc_ucbr_t *ucbr) +{ + bug_if_nullptr (ucbr); + + static size_t idx = 0; + + const mcpc_anypool_t *pool = ucbr; + if (pool == nullptr) + return nullptr; + + const size_t pool_len = pool->len; + if (pool_len == 0) + return nullptr; + + if (idx == pool_len) // TODO MT-SAFE + { + idx = 0; + return nullptr; + } + + const mcpc_anydata_t *ret = pool->head + idx; + idx += 1; + + return ret; +} + +size_t +mcpc_ucbr_get_serlz_lenhint (const mcpc_ucbr_t *ucbr) +{ + bug_if_nullptr (ucbr); + size_t ret = 0; + + size_t length = mcpc_ucbr_get_len (ucbr); + for (size_t i = 0; i < length; ++i) + { + const mcpc_ucbrchd_t *cur_chd = mcpc_ucbr_get (ucbr, i); + ret += cur_chd->data_len; + } + + /* JSON string escaping can expand one input byte to six output bytes. */ + if (ret > (SIZE_MAX - 1024) / 6) + return 0; + ret = ret * 6 + 1024; + + return ret; +} + +void +mcpc_ucbr_toolcallres_add_text_vprintf8 (mcpc_ucbr_t *ucbr, const char8_t *fmt, + va_list *vargs) +{ + bug_if_nullptr (ucbr); + bug_if_nullptr (fmt); + bug_if_nullptr (vargs); + + ucbr->flags |= ANYPOOL_FL_USRCBRES_TOOLCALL; + + mcpc_ucbrchd_t *newctn = mcpc_ucbrchd_new_vprintf8 (fmt, vargs); + newctn->flags |= ANYDATA_FL_USRCBRES_TOOLCALL; + newctn->flags |= ANYDATA_FL_USRCBRES_TOOLCALL_TXTCTN; + mcpc_ucbr_addfre (ucbr, newctn); +} + +MCPC_API void +mcpc_ucbr_toolcall_add_text_printf8 (mcpc_ucbr_t *ucbr, const char8_t *fmt, + ...) +{ + va_list vargs; + va_start (vargs, fmt); + + mcpc_ucbr_toolcallres_add_text_vprintf8 (ucbr, fmt, &vargs); + + va_end (vargs); +} + +MCPC_API void +mcpc_ucbr_toolcall_add_errmsg_printf8 (mcpc_ucbr_t *ucbr, const char8_t *fmt, + ...) +{ + va_list vargs; + va_start (vargs, fmt); + + ucbr->flags |= ANYPOOL_FL_USRCBRES_TOOLCALL_ERROR; + mcpc_ucbr_toolcallres_add_text_vprintf8 (ucbr, fmt, &vargs); + + va_end (vargs); +} + +MCPC_API void +mcpc_ucbr_prmptget_add_user_printf8 (mcpc_ucbr_t *ucbr, const char8_t *fmt, + ...) +{ + va_list vargs; + va_start (vargs, fmt); + + ucbr->flags |= ANYPOOL_FL_USRCBRES_PRMPTCALL; + + mcpc_ucbrchd_t *newctn = mcpc_ucbrchd_new_vprintf8 (fmt, &vargs); + newctn->flags |= ANYDATA_FL_USRCBRES_PRMPTCALL; + + mcpc_ucbr_addfre (ucbr, newctn); + + va_end (vargs); +} + +MCPC_API void +mcpc_ucbr_prmptget_add_assist_printf8 (mcpc_ucbr_t *ucbr, const char8_t *fmt, + ...) +{ + va_list vargs; + va_start (vargs, fmt); + + ucbr->flags |= ANYPOOL_FL_USRCBRES_PRMPTCALL; + + mcpc_ucbrchd_t *newctn = mcpc_ucbrchd_new_vprintf8 (fmt, &vargs); + newctn->flags |= ANYDATA_FL_USRCBRES_PRMPTCALL; + newctn->flags |= ANYDATA_FL_USRCBRES_PRMPTCALL_ASSIST; + + mcpc_ucbr_addfre (ucbr, newctn); + + va_end (vargs); +} diff --git a/third_party/mcpc/src/ucbr.h b/third_party/mcpc/src/ucbr.h new file mode 100644 index 0000000..796688d --- /dev/null +++ b/third_party/mcpc/src/ucbr.h @@ -0,0 +1,61 @@ +// This file is part of the mcpc library. +// +// Copyright (C) 2025 Michael Lee <micl2e2 AT proton.me> +// +// 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. + +#ifndef h_ucbr +#define h_ucbr + +#include <stddef.h> + +#include <mcpc/ucbr.h> + +typedef mcpc_anydata_t mcpc_ucbrchd_t; + +bool mcpc_ucbrchd_is_tool_callres (const mcpc_ucbrchd_t * ucbrchd); + +bool mcpc_ucbrchd_is_prmpt_callres (const mcpc_ucbrchd_t * ucbrchd); + +bool mcpc_ucbrchd_is_assist_prmpt (const mcpc_ucbrchd_t * ucbrchd); + +mcpc_errcode_t mcpc_ucbrchd_getref_u8str (const mcpc_ucbrchd_t * ucbrchd, const char8_t ** ret, size_t *ret_len); + +mcpc_errcode_t mcpc_ucbrchd_get_data_type (const mcpc_ucbrchd_t * ucbrchd, mcpc_anytype_t * ret); + +// ------------------------------ ucbr ------------------------------ + +mcpc_ucbr_t *mcpc_ucbr_new (); + +void mcpc_ucbr_free (mcpc_ucbr_t * ucbr); + +size_t mcpc_ucbr_get_len (const mcpc_ucbr_t * ucbr); + +const mcpc_ucbrchd_t *mcpc_ucbr_get (const mcpc_ucbr_t * ucbr, size_t idx); + +const mcpc_ucbrchd_t *mcpc_ucbr_get_next (const mcpc_ucbr_t * ucbr); + +size_t mcpc_ucbr_get_serlz_lenhint (const mcpc_ucbr_t * ucbr); + +void mcpc_ucbr_addfre (mcpc_ucbr_t * ucbr, mcpc_ucbrchd_t * ucbrchd); + +bool mcpc_ucbr_is_err (const mcpc_ucbr_t * ucbr); + + + +#endif diff --git a/third_party/mcpc/src/ver.txt b/third_party/mcpc/src/ver.txt new file mode 100644 index 0000000..6c6aa7c --- /dev/null +++ b/third_party/mcpc/src/ver.txt @@ -0,0 +1 @@ +0.1.0
\ No newline at end of file |