2018-03-12 10:39:13 +00:00
|
|
|
/*
|
|
|
|
* $Id$
|
|
|
|
*
|
|
|
|
Copyright (c) 2016-2018 Chung, Hyung-Hwan. All rights reserved.
|
|
|
|
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
|
|
modification, are permitted provided that the following conditions
|
|
|
|
are met:
|
|
|
|
1. Redistributions of source code must retain the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
2. Redistributions in binary form must reproduce the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer in the
|
|
|
|
documentation and/or other materials provided with the distribution.
|
|
|
|
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR
|
|
|
|
IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
|
|
|
OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
|
|
|
IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
|
|
INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
2018-03-13 10:20:33 +00:00
|
|
|
NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
|
|
|
THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2024-05-14 12:14:39 +00:00
|
|
|
#include <hcl-x.h>
|
|
|
|
#include <hcl-json.h>
|
|
|
|
#include <hcl-opt.h>
|
|
|
|
#include <hcl-utl.h>
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <errno.h>
|
|
|
|
#include <locale.h>
|
2024-04-27 12:16:00 +00:00
|
|
|
#include <unistd.h>
|
|
|
|
#include <fcntl.h>
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
#if defined(HAVE_TIME_H)
|
|
|
|
# include <time.h>
|
|
|
|
#endif
|
|
|
|
#if defined(HAVE_SYS_TIME_H)
|
|
|
|
# include <sys/time.h>
|
|
|
|
#endif
|
|
|
|
#if defined(HAVE_SIGNAL_H)
|
|
|
|
# include <signal.h>
|
|
|
|
#endif
|
2018-03-16 16:20:40 +00:00
|
|
|
#if defined(HAVE_SYS_UIO_H)
|
|
|
|
# include <sys/uio.h>
|
|
|
|
#endif
|
2018-03-12 16:45:42 +00:00
|
|
|
|
2018-03-12 10:39:13 +00:00
|
|
|
|
|
|
|
/* ========================================================================= */
|
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
typedef struct server_xtn_t server_xtn_t;
|
2018-03-14 10:14:38 +00:00
|
|
|
struct server_xtn_t
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
int logfd;
|
2018-04-26 04:39:20 +00:00
|
|
|
hcl_bitmask_t logmask;
|
2018-03-13 10:20:33 +00:00
|
|
|
int logfd_istty;
|
2024-04-14 09:33:15 +00:00
|
|
|
|
2018-03-16 16:20:40 +00:00
|
|
|
struct
|
|
|
|
{
|
|
|
|
hcl_bch_t buf[4096];
|
|
|
|
hcl_oow_t len;
|
|
|
|
} logbuf;
|
2024-04-27 12:33:34 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* used by the json submodule */
|
|
|
|
int json_depth;
|
2018-03-13 10:20:33 +00:00
|
|
|
};
|
2018-03-12 16:45:42 +00:00
|
|
|
|
2024-04-27 12:16:00 +00:00
|
|
|
|
|
|
|
typedef server_xtn_t client_xtn_t;
|
2024-04-27 12:33:34 +00:00
|
|
|
typedef server_xtn_t json_xtn_t;
|
2024-04-27 12:16:00 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
/* ========================================================================= */
|
2018-03-12 16:45:42 +00:00
|
|
|
|
2018-03-14 10:14:38 +00:00
|
|
|
static void* sys_alloc (hcl_mmgr_t* mmgr, hcl_oow_t size)
|
|
|
|
{
|
|
|
|
return malloc(size);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void* sys_realloc (hcl_mmgr_t* mmgr, void* ptr, hcl_oow_t size)
|
|
|
|
{
|
|
|
|
return realloc(ptr, size);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void sys_free (hcl_mmgr_t* mmgr, void* ptr)
|
|
|
|
{
|
|
|
|
free (ptr);
|
|
|
|
}
|
|
|
|
|
|
|
|
static hcl_mmgr_t sys_mmgr =
|
|
|
|
{
|
|
|
|
sys_alloc,
|
|
|
|
sys_realloc,
|
|
|
|
sys_free,
|
|
|
|
HCL_NULL
|
|
|
|
};
|
|
|
|
/* ========================================================================= */
|
|
|
|
|
2018-03-16 16:20:40 +00:00
|
|
|
static int write_all (int fd, const hcl_bch_t* ptr, hcl_oow_t len)
|
2018-03-14 10:14:38 +00:00
|
|
|
{
|
|
|
|
while (len > 0)
|
|
|
|
{
|
|
|
|
hcl_ooi_t wr;
|
|
|
|
|
|
|
|
wr = write(fd, ptr, len);
|
|
|
|
|
|
|
|
if (wr <= -1)
|
|
|
|
{
|
|
|
|
#if defined(EAGAIN) && defined(EWOULDBLOCK) && (EAGAIN == EWOULDBLOCK)
|
|
|
|
if (errno == EAGAIN) continue;
|
|
|
|
#else
|
|
|
|
#if defined(EAGAIN)
|
|
|
|
if (errno == EAGAIN) continue;
|
|
|
|
#elif defined(EWOULDBLOCK)
|
|
|
|
if (errno == EWOULDBLOCK) continue;
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#if defined(EINTR)
|
|
|
|
/* TODO: would this interfere with non-blocking nature of this VM? */
|
|
|
|
if (errno == EINTR) continue;
|
|
|
|
#endif
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
ptr += wr;
|
|
|
|
len -= wr;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-03-16 16:20:40 +00:00
|
|
|
|
2024-04-27 12:16:00 +00:00
|
|
|
static int write_log (server_xtn_t* xtn, int fd, const hcl_bch_t* ptr, hcl_oow_t len)
|
2018-03-16 16:20:40 +00:00
|
|
|
{
|
|
|
|
while (len > 0)
|
|
|
|
{
|
|
|
|
if (xtn->logbuf.len > 0)
|
|
|
|
{
|
|
|
|
hcl_oow_t rcapa, cplen;
|
|
|
|
|
|
|
|
rcapa = HCL_COUNTOF(xtn->logbuf.buf) - xtn->logbuf.len;
|
|
|
|
cplen = (len >= rcapa)? rcapa: len;
|
|
|
|
|
|
|
|
memcpy (&xtn->logbuf.buf[xtn->logbuf.len], ptr, cplen);
|
|
|
|
xtn->logbuf.len += cplen;
|
|
|
|
ptr += cplen;
|
|
|
|
len -= cplen;
|
|
|
|
|
|
|
|
if (xtn->logbuf.len >= HCL_COUNTOF(xtn->logbuf.buf))
|
|
|
|
{
|
2018-03-19 09:50:56 +00:00
|
|
|
write_all(fd, xtn->logbuf.buf, xtn->logbuf.len);
|
2018-03-16 16:20:40 +00:00
|
|
|
xtn->logbuf.len = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
hcl_oow_t rcapa;
|
|
|
|
|
|
|
|
rcapa = HCL_COUNTOF(xtn->logbuf.buf);
|
|
|
|
if (len >= rcapa)
|
|
|
|
{
|
2018-03-19 09:22:12 +00:00
|
|
|
write_all (fd, ptr, rcapa);
|
2018-03-16 16:20:40 +00:00
|
|
|
ptr += rcapa;
|
|
|
|
len -= rcapa;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
memcpy (xtn->logbuf.buf, ptr, len);
|
|
|
|
xtn->logbuf.len += len;
|
|
|
|
ptr += len;
|
|
|
|
len -= len;
|
2024-04-14 09:33:15 +00:00
|
|
|
|
2018-03-16 16:20:40 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2024-04-27 12:16:00 +00:00
|
|
|
static void flush_log (server_xtn_t* xtn, int fd)
|
2018-03-16 16:20:40 +00:00
|
|
|
{
|
|
|
|
if (xtn->logbuf.len > 0)
|
|
|
|
{
|
2018-03-19 09:22:12 +00:00
|
|
|
write_all (fd, xtn->logbuf.buf, xtn->logbuf.len);
|
2018-03-16 16:20:40 +00:00
|
|
|
xtn->logbuf.len = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-04-27 12:16:00 +00:00
|
|
|
static void log_write (server_xtn_t* xtn, hcl_oow_t wid, hcl_bitmask_t mask, const hcl_ooch_t* msg, hcl_oow_t len)
|
2018-03-14 10:14:38 +00:00
|
|
|
{
|
|
|
|
hcl_bch_t buf[256];
|
|
|
|
hcl_oow_t ucslen, bcslen;
|
|
|
|
hcl_oow_t msgidx;
|
|
|
|
int n, logfd;
|
|
|
|
|
|
|
|
if (mask & HCL_LOG_STDERR)
|
|
|
|
{
|
|
|
|
/* the messages that go to STDERR don't get masked out */
|
|
|
|
logfd = 2;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (!(xtn->logmask & mask & ~HCL_LOG_ALL_LEVELS)) return; /* check log types */
|
|
|
|
if (!(xtn->logmask & mask & ~HCL_LOG_ALL_TYPES)) return; /* check log levels */
|
|
|
|
|
|
|
|
if (mask & HCL_LOG_STDOUT) logfd = 1;
|
|
|
|
else
|
|
|
|
{
|
|
|
|
logfd = xtn->logfd;
|
|
|
|
if (logfd <= -1) return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* TODO: beautify the log message.
|
|
|
|
* do classification based on mask. */
|
|
|
|
if (!(mask & (HCL_LOG_STDOUT | HCL_LOG_STDERR)))
|
|
|
|
{
|
|
|
|
time_t now;
|
|
|
|
char ts[32];
|
|
|
|
size_t tslen;
|
|
|
|
struct tm tm, *tmp;
|
|
|
|
|
|
|
|
now = time(NULL);
|
|
|
|
|
2018-10-14 10:28:28 +00:00
|
|
|
#if defined(__OS2__)
|
|
|
|
tmp = _localtime(&now, &tm);
|
|
|
|
#elif defined(HAVE_LOCALTIME_R)
|
|
|
|
tmp = localtime_r(&now, &tm);
|
|
|
|
#else
|
|
|
|
tmp = localtime(&now);
|
|
|
|
#endif
|
|
|
|
#if defined(HAVE_STRFTIME_SMALL_Z)
|
2018-03-14 10:14:38 +00:00
|
|
|
tslen = strftime(ts, sizeof(ts), "%Y-%m-%d %H:%M:%S %z ", tmp);
|
2018-10-14 10:28:28 +00:00
|
|
|
#else
|
2024-04-14 09:33:15 +00:00
|
|
|
tslen = strftime(ts, sizeof(ts), "%Y-%m-%d %H:%M:%S %Z ", tmp);
|
2018-10-14 10:28:28 +00:00
|
|
|
#endif
|
2024-04-14 09:33:15 +00:00
|
|
|
if (tslen == 0)
|
2018-03-14 10:14:38 +00:00
|
|
|
{
|
|
|
|
strcpy (ts, "0000-00-00 00:00:00 +0000");
|
2024-04-14 09:33:15 +00:00
|
|
|
tslen = 25;
|
2018-03-14 10:14:38 +00:00
|
|
|
}
|
|
|
|
|
2024-04-27 12:16:00 +00:00
|
|
|
write_log (xtn, logfd, ts, tslen);
|
2018-03-14 10:14:38 +00:00
|
|
|
|
2018-03-16 05:56:05 +00:00
|
|
|
if (wid != HCL_SERVER_WID_INVALID)
|
2018-03-14 10:14:38 +00:00
|
|
|
{
|
2018-03-16 05:56:05 +00:00
|
|
|
/* TODO: check if the underlying snprintf support %zd */
|
|
|
|
tslen = snprintf (ts, sizeof(ts), "[%zu] ", wid);
|
2024-04-27 12:16:00 +00:00
|
|
|
write_log (xtn, logfd, ts, tslen);
|
2018-03-14 10:14:38 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-03-19 09:50:56 +00:00
|
|
|
if (logfd == xtn->logfd && xtn->logfd_istty)
|
2018-03-14 10:14:38 +00:00
|
|
|
{
|
2024-04-27 12:16:00 +00:00
|
|
|
if (mask & HCL_LOG_FATAL) write_log (xtn, logfd, "\x1B[1;31m", 7);
|
|
|
|
else if (mask & HCL_LOG_ERROR) write_log (xtn, logfd, "\x1B[1;32m", 7);
|
|
|
|
else if (mask & HCL_LOG_WARN) write_log (xtn, logfd, "\x1B[1;33m", 7);
|
2018-03-14 10:14:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#if defined(HCL_OOCH_IS_UCH)
|
|
|
|
msgidx = 0;
|
|
|
|
while (len > 0)
|
|
|
|
{
|
|
|
|
ucslen = len;
|
|
|
|
bcslen = HCL_COUNTOF(buf);
|
|
|
|
|
2018-04-07 15:54:16 +00:00
|
|
|
n = hcl_conv_oochars_to_bchars_with_cmgr(&msg[msgidx], &ucslen, buf, &bcslen, hcl_get_utf8_cmgr());
|
2018-03-14 10:14:38 +00:00
|
|
|
if (n == 0 || n == -2)
|
|
|
|
{
|
2024-04-14 09:33:15 +00:00
|
|
|
/* n = 0:
|
|
|
|
* converted all successfully
|
|
|
|
* n == -2:
|
2018-03-14 10:14:38 +00:00
|
|
|
* buffer not sufficient. not all got converted yet.
|
|
|
|
* write what have been converted this round. */
|
|
|
|
|
|
|
|
/*HCL_ASSERT (hcl, ucslen > 0); */ /* if this fails, the buffer size must be increased */
|
2018-03-14 14:46:23 +00:00
|
|
|
/*assert (ucslen > 0);*/
|
2018-03-14 10:14:38 +00:00
|
|
|
|
|
|
|
/* attempt to write all converted characters */
|
2024-04-27 12:16:00 +00:00
|
|
|
if (write_log(xtn, logfd, buf, bcslen) <= -1) break;
|
2018-03-14 10:14:38 +00:00
|
|
|
|
|
|
|
if (n == 0) break;
|
|
|
|
else
|
|
|
|
{
|
|
|
|
msgidx += ucslen;
|
|
|
|
len -= ucslen;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (n <= -1)
|
|
|
|
{
|
|
|
|
/* conversion error */
|
2018-04-23 14:30:00 +00:00
|
|
|
if (bcslen <= 0) break;
|
2024-04-27 12:16:00 +00:00
|
|
|
if (write_log(xtn, logfd, buf, bcslen) <= -1) break;
|
2018-04-23 14:30:00 +00:00
|
|
|
msgidx += ucslen;
|
|
|
|
len -= ucslen;
|
2018-03-14 10:14:38 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
#else
|
2024-04-27 12:16:00 +00:00
|
|
|
write_log (xtn, logfd, msg, len);
|
2018-03-14 10:14:38 +00:00
|
|
|
#endif
|
|
|
|
|
2018-03-19 09:50:56 +00:00
|
|
|
if (logfd == xtn->logfd && xtn->logfd_istty)
|
2018-03-14 10:14:38 +00:00
|
|
|
{
|
2024-04-27 12:16:00 +00:00
|
|
|
if (mask & (HCL_LOG_FATAL | HCL_LOG_ERROR | HCL_LOG_WARN)) write_log (xtn, logfd, "\x1B[0m", 4);
|
2018-03-14 10:14:38 +00:00
|
|
|
}
|
2018-03-16 16:20:40 +00:00
|
|
|
|
2024-04-27 12:16:00 +00:00
|
|
|
flush_log (xtn, logfd);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* ========================================================================= */
|
|
|
|
|
|
|
|
static void server_log_write (hcl_server_t* server, hcl_oow_t wid, hcl_bitmask_t mask, const hcl_ooch_t* msg, hcl_oow_t len)
|
|
|
|
{
|
|
|
|
log_write ((server_xtn_t*)hcl_server_getxtn(server), wid, mask, msg, len);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void client_log_write (hcl_client_t* client, hcl_bitmask_t mask, const hcl_ooch_t* msg, hcl_oow_t len)
|
|
|
|
{
|
|
|
|
log_write ((client_xtn_t*)hcl_client_getxtn(client), HCL_SERVER_WID_INVALID, mask, msg, len);
|
2018-03-14 10:14:38 +00:00
|
|
|
}
|
|
|
|
|
2024-04-27 12:33:34 +00:00
|
|
|
static void json_log_write (hcl_json_t* json, hcl_bitmask_t mask, const hcl_ooch_t* msg, hcl_oow_t len)
|
|
|
|
{
|
|
|
|
log_write ((json_xtn_t*)hcl_json_getxtn(json), HCL_SERVER_WID_INVALID, mask, msg, len);
|
|
|
|
}
|
|
|
|
|
2018-03-14 10:14:38 +00:00
|
|
|
/* ========================================================================= */
|
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
static hcl_server_t* g_server = HCL_NULL;
|
2024-04-27 12:16:00 +00:00
|
|
|
static hcl_client_t* g_client = HCL_NULL;
|
2018-03-13 02:02:25 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
/* ========================================================================= */
|
2018-03-13 02:02:25 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
typedef void (*signal_handler_t) (int, siginfo_t*, void*);
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
static void handle_sigint (int sig, siginfo_t* siginfo, void* ctx)
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
if (g_server) hcl_server_stop (g_server);
|
2024-04-27 12:16:00 +00:00
|
|
|
if (g_client) hcl_client_stop (g_client);
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
static void set_signal (int sig, signal_handler_t handler)
|
2018-03-12 16:45:42 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
struct sigaction sa;
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
memset (&sa, 0, sizeof(sa));
|
|
|
|
/*sa.sa_handler = handler;*/
|
|
|
|
sa.sa_flags = SA_SIGINFO;
|
|
|
|
sa.sa_sigaction = handler;
|
|
|
|
sigemptyset (&sa.sa_mask);
|
2018-03-12 16:45:42 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
sigaction (sig, &sa, NULL);
|
2018-03-12 16:45:42 +00:00
|
|
|
}
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
static void set_signal_to_ignore (int sig)
|
2018-03-12 16:45:42 +00:00
|
|
|
{
|
2024-04-14 09:33:15 +00:00
|
|
|
struct sigaction sa;
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
memset (&sa, 0, sizeof(sa));
|
|
|
|
sa.sa_handler = SIG_IGN;
|
|
|
|
sa.sa_flags = 0;
|
|
|
|
sigemptyset (&sa.sa_mask);
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
sigaction (sig, &sa, NULL);
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
static void set_signal_to_default (int sig)
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
2024-04-14 09:33:15 +00:00
|
|
|
struct sigaction sa;
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
memset (&sa, 0, sizeof(sa));
|
|
|
|
sa.sa_handler = SIG_DFL;
|
|
|
|
sa.sa_flags = 0;
|
|
|
|
sigemptyset (&sa.sa_mask);
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
sigaction (sig, &sa, NULL);
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
/* ========================================================================= */
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2024-04-27 12:33:34 +00:00
|
|
|
static int handle_logopt (server_xtn_t* xtn, const hcl_bch_t* str)
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
hcl_bch_t* xstr = (hcl_bch_t*)str;
|
|
|
|
hcl_bch_t* cm, * flt;
|
2018-04-26 04:39:20 +00:00
|
|
|
hcl_bitmask_t logmask;
|
2018-03-12 16:45:42 +00:00
|
|
|
|
2018-04-07 15:54:16 +00:00
|
|
|
cm = hcl_find_bchar_in_bcstr(xstr, ',');
|
2024-04-14 09:33:15 +00:00
|
|
|
if (cm)
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
2024-04-14 09:33:15 +00:00
|
|
|
/* i duplicate this string for open() below as open() doesn't
|
2018-03-13 10:20:33 +00:00
|
|
|
* accept a length-bounded string */
|
2018-03-14 14:40:05 +00:00
|
|
|
xstr = strdup(str);
|
2024-04-14 09:33:15 +00:00
|
|
|
if (!xstr)
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
fprintf (stderr, "ERROR: out of memory in duplicating %s\n", str);
|
2018-03-12 10:39:13 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2018-04-07 15:54:16 +00:00
|
|
|
cm = hcl_find_bchar_in_bcstr(xstr, ',');
|
2018-03-13 10:20:33 +00:00
|
|
|
*cm = '\0';
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-17 11:57:02 +00:00
|
|
|
logmask = xtn->logmask;
|
2018-03-13 10:20:33 +00:00
|
|
|
do
|
|
|
|
{
|
|
|
|
flt = cm + 1;
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-04-07 15:54:16 +00:00
|
|
|
cm = hcl_find_bchar_in_bcstr(flt, ',');
|
2018-03-13 10:20:33 +00:00
|
|
|
if (cm) *cm = '\0';
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-04-07 15:54:16 +00:00
|
|
|
if (hcl_comp_bcstr(flt, "app") == 0) logmask |= HCL_LOG_APP;
|
|
|
|
else if (hcl_comp_bcstr(flt, "compiler") == 0) logmask |= HCL_LOG_COMPILER;
|
|
|
|
else if (hcl_comp_bcstr(flt, "vm") == 0) logmask |= HCL_LOG_VM;
|
|
|
|
else if (hcl_comp_bcstr(flt, "mnemonic") == 0) logmask |= HCL_LOG_MNEMONIC;
|
|
|
|
else if (hcl_comp_bcstr(flt, "gc") == 0) logmask |= HCL_LOG_GC;
|
|
|
|
else if (hcl_comp_bcstr(flt, "ic") == 0) logmask |= HCL_LOG_IC;
|
|
|
|
else if (hcl_comp_bcstr(flt, "primitive") == 0) logmask |= HCL_LOG_PRIMITIVE;
|
|
|
|
|
|
|
|
else if (hcl_comp_bcstr(flt, "fatal") == 0) logmask |= HCL_LOG_FATAL;
|
|
|
|
else if (hcl_comp_bcstr(flt, "error") == 0) logmask |= HCL_LOG_ERROR;
|
|
|
|
else if (hcl_comp_bcstr(flt, "warn") == 0) logmask |= HCL_LOG_WARN;
|
|
|
|
else if (hcl_comp_bcstr(flt, "info") == 0) logmask |= HCL_LOG_INFO;
|
|
|
|
else if (hcl_comp_bcstr(flt, "debug") == 0) logmask |= HCL_LOG_DEBUG;
|
|
|
|
|
|
|
|
else if (hcl_comp_bcstr(flt, "fatal+") == 0) logmask |= HCL_LOG_FATAL;
|
|
|
|
else if (hcl_comp_bcstr(flt, "error+") == 0) logmask |= HCL_LOG_FATAL | HCL_LOG_ERROR;
|
|
|
|
else if (hcl_comp_bcstr(flt, "warn+") == 0) logmask |= HCL_LOG_FATAL | HCL_LOG_ERROR | HCL_LOG_WARN;
|
|
|
|
else if (hcl_comp_bcstr(flt, "info+") == 0) logmask |= HCL_LOG_FATAL | HCL_LOG_ERROR | HCL_LOG_WARN | HCL_LOG_INFO;
|
|
|
|
else if (hcl_comp_bcstr(flt, "debug+") == 0) logmask |= HCL_LOG_FATAL | HCL_LOG_ERROR | HCL_LOG_WARN | HCL_LOG_INFO | HCL_LOG_DEBUG;
|
2018-03-13 02:02:25 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
else
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
fprintf (stderr, "ERROR: unknown log option value - %s\n", flt);
|
|
|
|
if (str != xstr) free (xstr);
|
|
|
|
return -1;
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
|
|
|
}
|
2018-03-13 10:20:33 +00:00
|
|
|
while (cm);
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
if (!(logmask & HCL_LOG_ALL_TYPES)) logmask |= HCL_LOG_ALL_TYPES; /* no types specified. force to all types */
|
|
|
|
if (!(logmask & HCL_LOG_ALL_LEVELS)) logmask |= HCL_LOG_ALL_LEVELS; /* no levels specified. force to all levels */
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
2018-03-13 10:20:33 +00:00
|
|
|
else
|
2018-03-13 02:02:25 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
logmask = HCL_LOG_ALL_LEVELS | HCL_LOG_ALL_TYPES;
|
2018-03-13 02:02:25 +00:00
|
|
|
}
|
|
|
|
|
2018-03-14 10:14:38 +00:00
|
|
|
xtn->logfd = open(xstr, O_CREAT | O_WRONLY | O_APPEND , 0644);
|
|
|
|
if (xtn->logfd == -1)
|
2018-03-13 02:02:25 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
fprintf (stderr, "ERROR: cannot open a log file %s\n", xstr);
|
|
|
|
if (str != xstr) free (xstr);
|
|
|
|
return -1;
|
|
|
|
}
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-17 11:57:02 +00:00
|
|
|
xtn->logmask = logmask;
|
2018-03-13 10:20:33 +00:00
|
|
|
#if defined(HAVE_ISATTY)
|
2018-03-14 10:14:38 +00:00
|
|
|
xtn->logfd_istty = isatty(xtn->logfd);
|
2018-03-13 10:20:33 +00:00
|
|
|
#endif
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
if (str != xstr) free (xstr);
|
|
|
|
return 0;
|
2018-03-12 16:45:42 +00:00
|
|
|
}
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2024-04-27 12:33:34 +00:00
|
|
|
static int server_handle_logopt (hcl_server_t* server, const hcl_bch_t* str)
|
|
|
|
{
|
|
|
|
return handle_logopt((server_xtn_t*)hcl_server_getxtn(server), str);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int client_handle_logopt (hcl_client_t* client, const hcl_bch_t* str)
|
|
|
|
{
|
|
|
|
return handle_logopt((client_xtn_t*)hcl_client_getxtn(client), str);
|
|
|
|
}
|
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
#if defined(HCL_BUILD_DEBUG)
|
2018-03-14 10:14:38 +00:00
|
|
|
static int handle_dbgopt (hcl_server_t* server, const char* str)
|
2018-03-12 16:45:42 +00:00
|
|
|
{
|
2018-03-13 10:20:33 +00:00
|
|
|
const hcl_bch_t* cm, * flt;
|
|
|
|
hcl_oow_t len;
|
2018-04-26 04:39:20 +00:00
|
|
|
hcl_bitmask_t trait;
|
2018-03-14 10:14:38 +00:00
|
|
|
|
|
|
|
hcl_server_getoption (server, HCL_SERVER_TRAIT, &trait);
|
2018-03-12 16:45:42 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
cm = str - 1;
|
|
|
|
do
|
|
|
|
{
|
|
|
|
flt = cm + 1;
|
2018-03-12 16:45:42 +00:00
|
|
|
|
2018-04-07 15:54:16 +00:00
|
|
|
cm = hcl_find_bchar_in_bcstr(flt, ',');
|
|
|
|
len = cm? (cm - flt): hcl_count_bcstr(flt);
|
|
|
|
if (hcl_comp_bchars_bcstr(flt, len, "gc") == 0) trait |= HCL_SERVER_TRAIT_DEBUG_GC;
|
|
|
|
else if (hcl_comp_bchars_bcstr(flt, len, "bigint") == 0) trait |= HCL_SERVER_TRAIT_DEBUG_BIGINT;
|
2018-03-13 10:20:33 +00:00
|
|
|
else
|
|
|
|
{
|
|
|
|
fprintf (stderr, "ERROR: unknown debug option value - %.*s\n", (int)len, flt);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
while (cm);
|
2018-03-12 16:45:42 +00:00
|
|
|
|
2018-03-14 10:14:38 +00:00
|
|
|
hcl_server_setoption (server, HCL_SERVER_TRAIT, &trait);
|
2018-03-13 10:20:33 +00:00
|
|
|
return 0;
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
2018-03-13 10:20:33 +00:00
|
|
|
#endif
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-17 05:44:40 +00:00
|
|
|
static int handle_incpath (hcl_server_t* server, const char* str)
|
|
|
|
{
|
|
|
|
#if defined(HCL_OOCH_IS_UCH)
|
|
|
|
hcl_ooch_t incpath[HCL_PATH_MAX + 1];
|
|
|
|
hcl_oow_t bcslen, ucslen;
|
|
|
|
|
|
|
|
ucslen = HCL_COUNTOF(incpath);
|
2019-03-19 13:29:59 +00:00
|
|
|
if (hcl_conv_bcstr_to_ucstr_with_cmgr(str, &bcslen, incpath, &ucslen, hcl_server_getcmgr(server), 1) <= -1) return -1;
|
2018-03-17 05:44:40 +00:00
|
|
|
return hcl_server_setoption(server, HCL_SERVER_SCRIPT_INCLUDE_PATH, incpath);
|
|
|
|
#else
|
|
|
|
return hcl_server_setoption(server, HCL_SERVER_SCRIPT_INCLUDE_PATH, str);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2018-03-12 16:45:42 +00:00
|
|
|
/* ========================================================================= */
|
|
|
|
|
2018-03-15 15:23:51 +00:00
|
|
|
#define MIN_WORKER_STACK_SIZE 512000ul
|
|
|
|
#define MIN_ACTOR_HEAP_SIZE 512000ul
|
2024-04-14 09:33:15 +00:00
|
|
|
|
2024-04-27 12:16:00 +00:00
|
|
|
static int server_main (const char* outer, int argc, char* argv[])
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
|
|
|
hcl_bci_t c;
|
|
|
|
static hcl_bopt_lng_t lopt[] =
|
|
|
|
{
|
2018-03-17 05:44:40 +00:00
|
|
|
{ ":log", 'l' },
|
|
|
|
{ ":worker-max-count", '\0' },
|
|
|
|
{ ":worker-stack-size", '\0' },
|
|
|
|
{ ":worker-idle-timeout", '\0' },
|
|
|
|
{ ":actor-heap-size", 'm' },
|
|
|
|
{ ":actor-max-runtime", '\0' },
|
|
|
|
{ ":script-include-path", '\0' },
|
2018-03-12 16:45:42 +00:00
|
|
|
#if defined(HCL_BUILD_DEBUG)
|
2018-03-12 10:39:13 +00:00
|
|
|
{ ":debug", '\0' }, /* NOTE: there is no short option for --debug */
|
2018-03-12 16:45:42 +00:00
|
|
|
#endif
|
2018-03-12 10:39:13 +00:00
|
|
|
{ HCL_NULL, '\0' }
|
|
|
|
};
|
|
|
|
static hcl_bopt_t opt =
|
|
|
|
{
|
|
|
|
"l:m:",
|
|
|
|
lopt
|
|
|
|
};
|
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
hcl_server_t* server;
|
|
|
|
server_xtn_t* xtn;
|
2018-03-14 10:14:38 +00:00
|
|
|
hcl_server_prim_t server_prim;
|
2018-03-12 16:45:42 +00:00
|
|
|
int n;
|
2018-03-12 10:39:13 +00:00
|
|
|
|
|
|
|
const char* logopt = HCL_NULL;
|
2018-03-14 10:14:38 +00:00
|
|
|
const char* dbgopt = HCL_NULL;
|
2018-03-17 05:44:40 +00:00
|
|
|
const char* incpath = HCL_NULL;
|
2018-03-16 14:57:34 +00:00
|
|
|
hcl_oow_t worker_max_count = 0;
|
2018-03-15 15:23:51 +00:00
|
|
|
hcl_oow_t worker_stack_size = MIN_ACTOR_HEAP_SIZE;
|
|
|
|
hcl_ntime_t worker_idle_timeout = { 0, 0 };
|
|
|
|
hcl_oow_t actor_heap_size = MIN_ACTOR_HEAP_SIZE;
|
|
|
|
hcl_ntime_t actor_max_runtime = { 0, 0 };
|
2018-03-12 10:39:13 +00:00
|
|
|
|
|
|
|
setlocale (LC_ALL, "");
|
|
|
|
|
|
|
|
if (argc < 2)
|
|
|
|
{
|
|
|
|
print_usage:
|
2024-04-27 12:16:00 +00:00
|
|
|
fprintf (stderr, "Usage: %s %s bind-address:port\n", outer, argv[0]);
|
2018-03-12 10:39:13 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2024-04-14 17:23:55 +00:00
|
|
|
while ((c = hcl_getbopt(argc, argv, &opt)) != HCL_BCI_EOF)
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
|
|
|
switch (c)
|
|
|
|
{
|
|
|
|
case 'l':
|
|
|
|
logopt = opt.arg;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case 'm':
|
2018-03-15 15:23:51 +00:00
|
|
|
actor_heap_size = strtoul(opt.arg, HCL_NULL, 0);
|
2021-02-09 14:54:54 +00:00
|
|
|
if (actor_heap_size > 0 && actor_heap_size <= MIN_ACTOR_HEAP_SIZE) actor_heap_size = MIN_ACTOR_HEAP_SIZE;
|
2018-03-12 10:39:13 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
case '\0':
|
2021-02-09 14:54:54 +00:00
|
|
|
if (hcl_comp_bcstr(opt.lngopt, "worker-max-count") == 0)
|
2018-03-16 14:57:34 +00:00
|
|
|
{
|
|
|
|
worker_max_count = strtoul(opt.arg, HCL_NULL, 0);
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
2018-04-07 15:54:16 +00:00
|
|
|
else if (hcl_comp_bcstr(opt.lngopt, "worker-stack-size") == 0)
|
2018-03-15 15:23:51 +00:00
|
|
|
{
|
|
|
|
worker_stack_size = strtoul(opt.arg, HCL_NULL, 0);
|
2018-03-20 10:36:00 +00:00
|
|
|
if (worker_stack_size <= MIN_WORKER_STACK_SIZE) worker_stack_size = MIN_WORKER_STACK_SIZE;
|
2018-03-15 15:23:51 +00:00
|
|
|
}
|
2018-04-07 15:54:16 +00:00
|
|
|
else if (hcl_comp_bcstr(opt.lngopt, "worker-idle-timeout") == 0)
|
2018-03-15 15:23:51 +00:00
|
|
|
{
|
|
|
|
worker_idle_timeout.sec = strtoul(opt.arg, HCL_NULL, 0);
|
|
|
|
}
|
2018-04-07 15:54:16 +00:00
|
|
|
else if (hcl_comp_bcstr(opt.lngopt, "actor-max-runtime") == 0)
|
2018-03-15 15:23:51 +00:00
|
|
|
{
|
|
|
|
actor_max_runtime.sec = strtoul(opt.arg, HCL_NULL, 0);
|
|
|
|
}
|
2018-04-07 15:54:16 +00:00
|
|
|
else if (hcl_comp_bcstr(opt.lngopt, "script-include-path") == 0)
|
2018-03-17 05:44:40 +00:00
|
|
|
{
|
|
|
|
incpath = opt.arg;
|
|
|
|
}
|
2018-03-12 10:39:13 +00:00
|
|
|
#if defined(HCL_BUILD_DEBUG)
|
2018-04-07 15:54:16 +00:00
|
|
|
else if (hcl_comp_bcstr(opt.lngopt, "debug") == 0)
|
2018-03-12 10:39:13 +00:00
|
|
|
{
|
2018-03-14 10:14:38 +00:00
|
|
|
dbgopt = opt.arg;
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
|
|
|
#endif
|
2018-03-16 14:57:34 +00:00
|
|
|
else goto print_usage;
|
|
|
|
break;
|
2018-03-12 10:39:13 +00:00
|
|
|
|
|
|
|
case ':':
|
|
|
|
if (opt.lngopt)
|
|
|
|
fprintf (stderr, "bad argument for '%s'\n", opt.lngopt);
|
|
|
|
else
|
|
|
|
fprintf (stderr, "bad argument for '%c'\n", opt.opt);
|
|
|
|
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
default:
|
|
|
|
goto print_usage;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (opt.ind >= argc) goto print_usage;
|
|
|
|
|
2018-03-14 10:14:38 +00:00
|
|
|
memset (&server_prim, 0, HCL_SIZEOF(server_prim));
|
2024-04-27 12:16:00 +00:00
|
|
|
server_prim.log_write = server_log_write;
|
2018-03-14 10:14:38 +00:00
|
|
|
|
|
|
|
server = hcl_server_open(&sys_mmgr, HCL_SIZEOF(server_xtn_t), &server_prim, HCL_NULL);
|
2018-03-12 10:39:13 +00:00
|
|
|
if (!server)
|
|
|
|
{
|
|
|
|
fprintf (stderr, "cannot open server\n");
|
|
|
|
return -1;
|
|
|
|
}
|
2018-03-13 10:20:33 +00:00
|
|
|
|
|
|
|
xtn = (server_xtn_t*)hcl_server_getxtn(server);
|
|
|
|
xtn->logfd = -1;
|
|
|
|
xtn->logfd_istty = 0;
|
|
|
|
|
2018-03-14 10:14:38 +00:00
|
|
|
if (logopt)
|
|
|
|
{
|
2024-04-27 12:33:34 +00:00
|
|
|
if (handle_logopt(xtn, logopt) <= -1) goto oops;
|
2018-03-14 10:14:38 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2018-03-17 11:57:02 +00:00
|
|
|
/* default logging mask when no logging option is set */
|
|
|
|
xtn->logmask = HCL_LOG_ALL_TYPES | HCL_LOG_ERROR | HCL_LOG_FATAL;
|
2018-03-14 10:14:38 +00:00
|
|
|
}
|
|
|
|
|
2018-03-17 05:44:40 +00:00
|
|
|
#if defined(HCL_BUILD_DEBUG)
|
2018-03-14 10:14:38 +00:00
|
|
|
if (dbgopt)
|
|
|
|
{
|
|
|
|
if (handle_dbgopt(server, dbgopt) <= -1) goto oops;
|
|
|
|
}
|
2018-03-17 05:44:40 +00:00
|
|
|
#endif
|
|
|
|
|
|
|
|
if (incpath)
|
|
|
|
{
|
|
|
|
if (handle_incpath(server, incpath) <= -1) goto oops;
|
|
|
|
}
|
2018-03-12 10:39:13 +00:00
|
|
|
|
2018-03-16 14:57:34 +00:00
|
|
|
hcl_server_setoption (server, HCL_SERVER_WORKER_MAX_COUNT, &worker_max_count);
|
2018-03-15 15:23:51 +00:00
|
|
|
hcl_server_setoption (server, HCL_SERVER_WORKER_STACK_SIZE, &worker_stack_size);
|
|
|
|
hcl_server_setoption (server, HCL_SERVER_WORKER_IDLE_TIMEOUT, &worker_idle_timeout);
|
|
|
|
hcl_server_setoption (server, HCL_SERVER_ACTOR_HEAP_SIZE, &actor_heap_size);
|
|
|
|
hcl_server_setoption (server, HCL_SERVER_ACTOR_MAX_RUNTIME, &actor_max_runtime);
|
2018-03-15 10:30:06 +00:00
|
|
|
|
2018-03-12 10:39:13 +00:00
|
|
|
g_server = server;
|
|
|
|
set_signal (SIGINT, handle_sigint);
|
2018-03-12 16:45:42 +00:00
|
|
|
set_signal_to_ignore (SIGPIPE);
|
2018-03-13 02:02:25 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
n = hcl_server_start(server, argv[opt.ind]);
|
2018-03-14 10:14:38 +00:00
|
|
|
|
2018-03-12 10:39:13 +00:00
|
|
|
set_signal_to_default (SIGINT);
|
2018-03-12 16:45:42 +00:00
|
|
|
set_signal_to_default (SIGPIPE);
|
2018-03-12 10:39:13 +00:00
|
|
|
g_server = NULL;
|
|
|
|
|
2018-03-14 10:14:38 +00:00
|
|
|
if (n <= -1)
|
|
|
|
{
|
|
|
|
hcl_server_logbfmt (server, HCL_LOG_APP | HCL_LOG_FATAL, "server error[%d] - %js\n", hcl_server_geterrnum(server), hcl_server_geterrmsg(server));
|
|
|
|
}
|
|
|
|
|
2018-03-21 10:38:10 +00:00
|
|
|
if (xtn->logfd >= 0)
|
|
|
|
{
|
|
|
|
close (xtn->logfd);
|
|
|
|
xtn->logfd = -1;
|
|
|
|
xtn->logfd_istty = 0;
|
|
|
|
}
|
2018-03-14 10:14:38 +00:00
|
|
|
|
2018-03-13 10:20:33 +00:00
|
|
|
hcl_server_close (server);
|
2018-03-12 16:45:42 +00:00
|
|
|
return n;
|
2018-03-13 10:20:33 +00:00
|
|
|
|
2018-03-12 16:45:42 +00:00
|
|
|
oops:
|
2018-03-13 10:20:33 +00:00
|
|
|
if (server) hcl_server_close (server);
|
2018-03-12 16:45:42 +00:00
|
|
|
return -1;
|
2018-03-12 10:39:13 +00:00
|
|
|
}
|
2024-04-27 05:36:35 +00:00
|
|
|
|
2024-04-27 12:16:00 +00:00
|
|
|
/* -------------------------------------------------------------- */
|
2024-05-14 11:52:25 +00:00
|
|
|
static int client_on_packet (hcl_client_t* client, hcl_xpkt_type_t type, const void* data, hcl_oow_t len)
|
2024-04-27 12:16:00 +00:00
|
|
|
{
|
|
|
|
if (type == HCL_XPKT_STDOUT)
|
|
|
|
{
|
|
|
|
if (len > 0) fprintf (stdout, "%.*s", (int)len, data);
|
|
|
|
}
|
|
|
|
else if (type == HCL_XPKT_STDERR)
|
|
|
|
{
|
|
|
|
if (len > 0) fprintf (stderr, "%.*s", (int)len, data);
|
|
|
|
}
|
|
|
|
else if (type == HCL_XPKT_ERROR)
|
|
|
|
{
|
|
|
|
/* error notification */
|
|
|
|
if (len > 0) fprintf (stderr, "ERROR: %.*s\n", (int)len, data);
|
|
|
|
}
|
2024-05-14 11:52:25 +00:00
|
|
|
else if (type == HCL_XPKT_RETVAL)
|
|
|
|
{
|
|
|
|
if (len > 0) fprintf (stderr, "RETURN VALUE: %.*s\n", (int)len, data);
|
|
|
|
hcl_client_stop (client);
|
|
|
|
}
|
2024-04-27 12:16:00 +00:00
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int client_main (const char* outer, int argc, char* argv[])
|
|
|
|
{
|
|
|
|
hcl_bci_t c;
|
|
|
|
static hcl_bopt_lng_t lopt[] =
|
|
|
|
{
|
|
|
|
{ ":log", 'l' },
|
|
|
|
{ "shutwr", '\0' },
|
|
|
|
{ HCL_NULL, '\0' }
|
|
|
|
};
|
|
|
|
static hcl_bopt_t opt =
|
|
|
|
{
|
|
|
|
"l:",
|
|
|
|
lopt
|
|
|
|
};
|
|
|
|
|
|
|
|
hcl_client_t* client;
|
|
|
|
client_xtn_t* xtn;
|
|
|
|
hcl_client_prim_t client_prim;
|
|
|
|
int n;
|
|
|
|
const char* logopt = HCL_NULL;
|
|
|
|
int shut_wr_after_req = 0;
|
|
|
|
|
|
|
|
setlocale (LC_ALL, "");
|
|
|
|
|
|
|
|
if (argc < 2)
|
|
|
|
{
|
|
|
|
print_usage:
|
|
|
|
fprintf (stderr, "Usage: %s %s [options] bind-address:port script-to-run\n", outer, argv[0]);
|
|
|
|
fprintf (stderr, "Options are:\n");
|
|
|
|
fprintf (stderr, " -l/--log log-options\n");
|
|
|
|
fprintf (stderr, " --shutwr\n");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
while ((c = hcl_getbopt(argc, argv, &opt)) != HCL_BCI_EOF)
|
|
|
|
{
|
|
|
|
switch (c)
|
|
|
|
{
|
|
|
|
case 'l':
|
|
|
|
logopt = opt.arg;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case '\0':
|
2024-05-11 05:16:22 +00:00
|
|
|
if (hcl_comp_bcstr(opt.lngopt, "shutwr") == 0)
|
2024-04-27 12:16:00 +00:00
|
|
|
{
|
|
|
|
shut_wr_after_req = 1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
goto print_usage;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ':':
|
|
|
|
if (opt.lngopt)
|
|
|
|
fprintf (stderr, "bad argument for '%s'\n", opt.lngopt);
|
|
|
|
else
|
|
|
|
fprintf (stderr, "bad argument for '%c'\n", opt.opt);
|
|
|
|
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
default:
|
|
|
|
goto print_usage;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* needs 2 fixed arguments */
|
|
|
|
if (opt.ind + 1 >= argc) goto print_usage;
|
|
|
|
|
|
|
|
memset (&client_prim, 0, HCL_SIZEOF(client_prim));
|
|
|
|
client_prim.log_write = client_log_write;
|
|
|
|
client_prim.on_packet = client_on_packet;
|
|
|
|
|
|
|
|
client = hcl_client_open(&sys_mmgr, HCL_SIZEOF(client_xtn_t), &client_prim, HCL_NULL);
|
|
|
|
if (!client)
|
|
|
|
{
|
|
|
|
fprintf (stderr, "cannot open client\n");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
xtn = (client_xtn_t*)hcl_client_getxtn(client);
|
|
|
|
xtn->logfd = -1;
|
|
|
|
xtn->logfd_istty = 0;
|
|
|
|
|
|
|
|
if (logopt)
|
|
|
|
{
|
2024-04-27 12:33:34 +00:00
|
|
|
if (handle_logopt(xtn, logopt) <= -1) goto oops;
|
2024-04-27 12:16:00 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* default logging mask when no logging option is set */
|
|
|
|
xtn->logmask = HCL_LOG_ALL_TYPES | HCL_LOG_ERROR | HCL_LOG_FATAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
g_client = client;
|
|
|
|
set_signal (SIGINT, handle_sigint);
|
|
|
|
set_signal_to_ignore (SIGPIPE);
|
|
|
|
|
2024-05-11 05:16:22 +00:00
|
|
|
n = hcl_client_start(client, argv[opt.ind], /*argv[opt.ind + 1],*/ shut_wr_after_req);
|
2024-04-27 12:16:00 +00:00
|
|
|
if (n <= -1)
|
|
|
|
{
|
|
|
|
fprintf (stderr, "ERROR: %s\n", hcl_client_geterrbmsg(client));
|
|
|
|
goto oops;
|
|
|
|
}
|
|
|
|
|
|
|
|
set_signal_to_default (SIGINT);
|
|
|
|
set_signal_to_default (SIGPIPE);
|
|
|
|
g_client = NULL;
|
|
|
|
|
|
|
|
if (xtn->logfd >= 0)
|
|
|
|
{
|
|
|
|
close (xtn->logfd);
|
|
|
|
xtn->logfd = -1;
|
|
|
|
xtn->logfd_istty = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
hcl_client_close (client);
|
|
|
|
return n;
|
|
|
|
|
|
|
|
oops:
|
|
|
|
if (client) hcl_client_close (client);
|
|
|
|
return -1;
|
|
|
|
}
|
2024-04-27 12:33:34 +00:00
|
|
|
|
|
|
|
/* -------------------------------------------------------------- */
|
|
|
|
|
|
|
|
static int json_inst_cb (hcl_json_t* json, hcl_json_inst_t it, const hcl_oocs_t* str)
|
|
|
|
{
|
|
|
|
json_xtn_t* json_xtn = (json_xtn_t*)hcl_json_getxtn(json);
|
|
|
|
|
|
|
|
switch (it)
|
|
|
|
{
|
|
|
|
case HCL_JSON_INST_START_ARRAY:
|
|
|
|
json_xtn->json_depth++;
|
|
|
|
hcl_json_logbfmt (json, HCL_LOG_INFO | HCL_LOG_APP, "[\n");
|
|
|
|
break;
|
|
|
|
case HCL_JSON_INST_END_ARRAY:
|
|
|
|
json_xtn->json_depth--;
|
|
|
|
hcl_json_logbfmt (json, HCL_LOG_INFO | HCL_LOG_APP, "]\n");
|
|
|
|
break;
|
|
|
|
case HCL_JSON_INST_START_DIC:
|
|
|
|
json_xtn->json_depth++;
|
|
|
|
hcl_json_logbfmt (json, HCL_LOG_INFO | HCL_LOG_APP, "{\n");
|
|
|
|
break;
|
|
|
|
case HCL_JSON_INST_END_DIC:
|
|
|
|
json_xtn->json_depth--;
|
|
|
|
hcl_json_logbfmt (json, HCL_LOG_INFO | HCL_LOG_APP, "}\n");
|
|
|
|
break;
|
|
|
|
|
|
|
|
case HCL_JSON_INST_KEY:
|
|
|
|
hcl_json_logbfmt (json, HCL_LOG_INFO | HCL_LOG_APP, "%.*js: ", str->len, str->ptr);
|
|
|
|
break;
|
|
|
|
|
|
|
|
case HCL_JSON_INST_CHARACTER:
|
|
|
|
case HCL_JSON_INST_STRING:
|
|
|
|
case HCL_JSON_INST_NUMBER:
|
|
|
|
case HCL_JSON_INST_TRUE:
|
|
|
|
case HCL_JSON_INST_FALSE:
|
|
|
|
case HCL_JSON_INST_NIL:
|
|
|
|
hcl_json_logbfmt (json, HCL_LOG_INFO | HCL_LOG_APP, "%.*js\n", str->len, str->ptr);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
2024-04-29 10:01:00 +00:00
|
|
|
|
2024-04-27 12:33:34 +00:00
|
|
|
int json_main (const char* outer, int argc, char* argv[])
|
|
|
|
{
|
|
|
|
hcl_json_t* json;
|
|
|
|
hcl_json_prim_t json_prim;
|
|
|
|
json_xtn_t* json_xtn;
|
|
|
|
hcl_oow_t xlen;
|
|
|
|
const char* p;
|
|
|
|
|
|
|
|
/* TODO: enhance this to accept parameters from command line */
|
|
|
|
|
|
|
|
memset (&json_prim, 0, HCL_SIZEOF(json_prim));
|
|
|
|
json_prim.log_write = json_log_write;
|
|
|
|
json_prim.instcb = json_inst_cb;
|
|
|
|
|
|
|
|
json = hcl_json_open (&sys_mmgr, HCL_SIZEOF(json_xtn_t), &json_prim, NULL);
|
|
|
|
|
|
|
|
json_xtn = (json_xtn_t*)hcl_json_getxtn(json);
|
|
|
|
json_xtn->logmask = HCL_LOG_ALL_LEVELS | HCL_LOG_ALL_TYPES;
|
|
|
|
|
|
|
|
p = "[ \"ab\\xab\\uC88B\\uC544\\uC6A9c\", \"kaden\", \"iron\", true, { \"null\": \"a\\1bc\", \"123\": \"AA20AA\", \"10\": -0.123, \"way\": '\\uC88A' } ]";
|
|
|
|
/*p = "{ \"result\": \"SUCCESS\", \"message\": \"1 clients\", \"sessions\": [] }";*/
|
|
|
|
|
|
|
|
if (hcl_json_feed(json, p, strlen(p), &xlen) <= -1)
|
|
|
|
{
|
|
|
|
hcl_json_logbfmt (json, HCL_LOG_FATAL | HCL_LOG_APP, "ERROR: unable to process - %js\n", hcl_json_geterrmsg(json));
|
|
|
|
}
|
|
|
|
else if (json_xtn->json_depth != 0)
|
|
|
|
{
|
|
|
|
hcl_json_logbfmt (json, HCL_LOG_FATAL | HCL_LOG_APP, "ERROR: incomplete input\n");
|
|
|
|
}
|
|
|
|
|
|
|
|
hcl_json_close (json);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2024-04-27 05:36:35 +00:00
|
|
|
/* -------------------------------------------------------------- */
|
|
|
|
|
|
|
|
static void print_main_usage (const char* argv0)
|
|
|
|
{
|
2024-04-27 12:33:34 +00:00
|
|
|
fprintf (stderr, "Usage: %s server|client|json\n", argv0);
|
2024-04-27 05:36:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
int main (int argc, char* argv[])
|
|
|
|
{
|
|
|
|
int n;
|
2024-04-27 12:16:00 +00:00
|
|
|
const char* argv0;
|
|
|
|
|
|
|
|
argv0 = hcl_get_base_name_from_bcstr_path(argv[0]);
|
2024-04-27 05:36:35 +00:00
|
|
|
|
|
|
|
if (argc < 2)
|
|
|
|
{
|
2024-04-27 12:16:00 +00:00
|
|
|
print_main_usage (argv0);
|
2024-04-27 05:36:35 +00:00
|
|
|
n = -1;
|
|
|
|
}
|
|
|
|
else if (strcmp(argv[1], "server") == 0)
|
|
|
|
{
|
2024-04-27 12:16:00 +00:00
|
|
|
n = server_main(argv0, argc -1, &argv[1]);
|
2024-04-27 05:36:35 +00:00
|
|
|
}
|
|
|
|
else if (strcmp(argv[1], "client") == 0)
|
|
|
|
{
|
|
|
|
n = client_main(argv[0], argc -1, &argv[1]);
|
|
|
|
}
|
2024-04-27 12:33:34 +00:00
|
|
|
else if (strcmp(argv[1], "json") == 0)
|
|
|
|
{
|
|
|
|
n = json_main(argv[0], argc -1, &argv[1]);
|
|
|
|
}
|
2024-04-27 05:36:35 +00:00
|
|
|
else
|
|
|
|
{
|
|
|
|
print_main_usage (argv[0]);
|
|
|
|
n = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return n;
|
|
|
|
}
|