1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
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
|
