859 lines
20 KiB
C
859 lines
20 KiB
C
/*
|
|
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
|
|
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.
|
|
*/
|
|
|
|
#include "hcl-x.h"
|
|
#include "hcl-prv.h"
|
|
#include "hcl-tmr.h"
|
|
#include "hcl-xutl.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
|
|
#define HCL_SERVER_TOKEN_NAME_ALIGN 64
|
|
#define HCL_SERVER_WID_MAP_ALIGN 512
|
|
#define HCL_XPROTO_REPLY_BUF_SIZE 1300
|
|
|
|
#if defined(_WIN32)
|
|
# include <windows.h>
|
|
# include <tchar.h>
|
|
#elif defined(__OS2__)
|
|
# define INCL_DOSMODULEMGR
|
|
# define INCL_DOSPROCESS
|
|
# define INCL_DOSERRORS
|
|
# include <os2.h>
|
|
#elif defined(__DOS__)
|
|
# include <dos.h>
|
|
# include <time.h>
|
|
# include <signal.h>
|
|
#elif defined(macintosh)
|
|
# include <Timer.h>
|
|
#else
|
|
|
|
# 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
|
|
# if defined(HAVE_SYS_MMAN_H)
|
|
# include <sys/mman.h>
|
|
# endif
|
|
# if defined(HAVE_SYS_UIO_H)
|
|
# include <sys/uio.h>
|
|
# endif
|
|
# if defined(HAVE_SYS_EPOLL_H)
|
|
# include <sys/epoll.h>
|
|
# endif
|
|
|
|
# include <unistd.h>
|
|
# include <fcntl.h>
|
|
# include <sys/types.h>
|
|
# include <sys/socket.h>
|
|
# include <netinet/in.h>
|
|
# include <pthread.h>
|
|
# include <poll.h>
|
|
#endif
|
|
|
|
struct client_hcl_xtn_t
|
|
{
|
|
hcl_client_t* client;
|
|
};
|
|
typedef struct client_hcl_xtn_t client_hcl_xtn_t;
|
|
|
|
struct hcl_client_t
|
|
{
|
|
hcl_oow_t _instsize;
|
|
hcl_mmgr_t* _mmgr;
|
|
hcl_cmgr_t* _cmgr;
|
|
|
|
hcl_client_prim_t prim;
|
|
hcl_t* dummy_hcl;
|
|
|
|
hcl_errnum_t errnum;
|
|
struct
|
|
{
|
|
#if defined(HCL_OOCH_IS_BCH)
|
|
hcl_uch_t xerrmsg[HCL_ERRMSG_CAPA];
|
|
#else
|
|
hcl_bch_t xerrmsg[HCL_ERRMSG_CAPA * 2];
|
|
#endif
|
|
hcl_ooch_t buf[HCL_ERRMSG_CAPA];
|
|
hcl_oow_t len;
|
|
} errmsg;
|
|
int stopreq;
|
|
|
|
struct
|
|
{
|
|
hcl_bitmask_t trait;
|
|
hcl_bitmask_t logmask;
|
|
} cfg;
|
|
|
|
int sck;
|
|
hcl_xproto_t* proto;
|
|
|
|
int mux_pipe[2]; /* pipe to break the blocking multiplexer in the main server loop */
|
|
|
|
struct
|
|
{
|
|
hcl_bch_t buf[4096];
|
|
hcl_oow_t pos;
|
|
hcl_oow_t len;
|
|
} script;
|
|
};
|
|
|
|
|
|
/* ========================================================================= */
|
|
|
|
static void client_log_write_for_dummy (hcl_t* hcl, hcl_bitmask_t mask, const hcl_ooch_t* msg, hcl_oow_t len)
|
|
{
|
|
client_hcl_xtn_t* xtn = (client_hcl_xtn_t*)hcl_getxtn(hcl);
|
|
hcl_client_t* client;
|
|
|
|
client = xtn->client;
|
|
client->prim.log_write (client, mask, msg, len);
|
|
}
|
|
|
|
hcl_client_t* hcl_client_open (hcl_mmgr_t* mmgr, hcl_oow_t xtnsize, hcl_client_prim_t* prim, hcl_errnum_t* errnum)
|
|
{
|
|
hcl_client_t* client = HCL_NULL;
|
|
hcl_t* hcl = HCL_NULL;
|
|
client_hcl_xtn_t* xtn;
|
|
int pfd[2];
|
|
|
|
client = (hcl_client_t*)HCL_MMGR_ALLOC(mmgr, HCL_SIZEOF(*client) + xtnsize);
|
|
if (HCL_UNLIKELY(!client))
|
|
{
|
|
if (errnum) *errnum = HCL_ESYSMEM;
|
|
return HCL_NULL;
|
|
}
|
|
|
|
hcl = hcl_openstdwithmmgr(mmgr, HCL_SIZEOF(*xtn), errnum);
|
|
if (HCL_UNLIKELY(!hcl))
|
|
{
|
|
HCL_MMGR_FREE (mmgr, client);
|
|
return HCL_NULL;
|
|
}
|
|
|
|
if (hcl_sys_open_pipes(pfd, 1) <= -1)
|
|
{
|
|
if (errnum) *errnum = hcl->vmprim.syserrstrb(hcl, 0, errno, HCL_NULL, 0);
|
|
goto oops;
|
|
}
|
|
|
|
/* replace the vmprim.log_write function */
|
|
hcl->vmprim.log_write = client_log_write_for_dummy;
|
|
|
|
xtn = (client_hcl_xtn_t*)hcl_getxtn(hcl);
|
|
xtn->client = client;
|
|
|
|
HCL_MEMSET (client, 0, HCL_SIZEOF(*client) + xtnsize);
|
|
client->_instsize = HCL_SIZEOF(*client);
|
|
client->_mmgr = mmgr;
|
|
client->_cmgr = hcl_get_utf8_cmgr();
|
|
client->prim = *prim;
|
|
client->dummy_hcl = hcl;
|
|
client->sck = -1;
|
|
client->mux_pipe[0] = pfd[0];
|
|
client->mux_pipe[1] = pfd[1];
|
|
|
|
client->cfg.logmask = ~(hcl_bitmask_t)0;
|
|
|
|
/* the dummy hcl is used for this client to perform primitive operations
|
|
* such as getting system time or logging. so the heap size doesn't
|
|
* need to be changed from the tiny value set above. */
|
|
hcl_setoption (client->dummy_hcl, HCL_LOG_MASK, &client->cfg.logmask);
|
|
hcl_setcmgr (client->dummy_hcl, client->_cmgr);
|
|
|
|
return client;
|
|
|
|
oops:
|
|
/* NOTE: pipe should be closed if jump to here is made after pipe() above */
|
|
if (hcl) hcl_close (hcl);
|
|
if (client) HCL_MMGR_FREE (mmgr, client);
|
|
return HCL_NULL;
|
|
}
|
|
|
|
void hcl_client_close (hcl_client_t* client)
|
|
{
|
|
if (client->proto) hcl_xproto_close (client->proto);
|
|
if (client->sck >= 0) close (client->sck);
|
|
|
|
hcl_sys_close_pipes(client->mux_pipe);
|
|
hcl_close (client->dummy_hcl);
|
|
HCL_MMGR_FREE (client->_mmgr, client);
|
|
}
|
|
|
|
int hcl_client_setoption (hcl_client_t* client, hcl_client_option_t id, const void* value)
|
|
{
|
|
switch (id)
|
|
{
|
|
case HCL_CLIENT_TRAIT:
|
|
client->cfg.trait = *(const hcl_bitmask_t*)value;
|
|
return 0;
|
|
|
|
case HCL_CLIENT_LOG_MASK:
|
|
client->cfg.logmask = *(const hcl_bitmask_t*)value;
|
|
if (client->dummy_hcl)
|
|
{
|
|
/* setting this affects the dummy hcl immediately.
|
|
* existing hcl instances inside worker threads won't get
|
|
* affected. new hcl instances to be created later
|
|
* is supposed to use the new value */
|
|
hcl_setoption (client->dummy_hcl, HCL_LOG_MASK, value);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
hcl_client_seterrnum (client, HCL_EINVAL);
|
|
return -1;
|
|
}
|
|
|
|
int hcl_client_getoption (hcl_client_t* client, hcl_client_option_t id, void* value)
|
|
{
|
|
switch (id)
|
|
{
|
|
case HCL_CLIENT_TRAIT:
|
|
*(hcl_bitmask_t*)value = client->cfg.trait;
|
|
return 0;
|
|
|
|
case HCL_CLIENT_LOG_MASK:
|
|
*(hcl_bitmask_t*)value = client->cfg.logmask;
|
|
return 0;
|
|
};
|
|
|
|
hcl_client_seterrnum (client, HCL_EINVAL);
|
|
return -1;
|
|
}
|
|
|
|
|
|
void* hcl_client_getxtn (hcl_client_t* client)
|
|
{
|
|
return (void*)((hcl_uint8_t*)client + client->_instsize);
|
|
}
|
|
|
|
hcl_mmgr_t* hcl_client_getmmgr (hcl_client_t* client)
|
|
{
|
|
return client->_mmgr;
|
|
}
|
|
|
|
hcl_cmgr_t* hcl_client_getcmgr (hcl_client_t* client)
|
|
{
|
|
return client->_cmgr;
|
|
}
|
|
|
|
void hcl_client_setcmgr (hcl_client_t* client, hcl_cmgr_t* cmgr)
|
|
{
|
|
client->_cmgr = cmgr;
|
|
}
|
|
|
|
hcl_errnum_t hcl_client_geterrnum (hcl_client_t* client)
|
|
{
|
|
return client->errnum;
|
|
}
|
|
|
|
const hcl_ooch_t* hcl_client_geterrstr (hcl_client_t* client)
|
|
{
|
|
return hcl_errnum_to_errstr(client->errnum);
|
|
}
|
|
|
|
const hcl_ooch_t* hcl_client_geterrmsg (hcl_client_t* client)
|
|
{
|
|
if (client->errmsg.len <= 0) return hcl_errnum_to_errstr(client->errnum);
|
|
return client->errmsg.buf;
|
|
}
|
|
|
|
const hcl_bch_t* hcl_client_geterrbmsg (hcl_client_t* client)
|
|
{
|
|
#if defined(HCL_OOCH_IS_BCH)
|
|
return (client->errmsg.len <= 0)? hcl_errnum_to_errstr(client->errnum): client->errmsg.buf;
|
|
#else
|
|
const hcl_ooch_t* msg;
|
|
hcl_oow_t wcslen, mbslen;
|
|
|
|
msg = (client->errmsg.len <= 0)? hcl_errnum_to_errstr(client->errnum): client->errmsg.buf;
|
|
|
|
mbslen = HCL_COUNTOF(client->errmsg.xerrmsg);
|
|
hcl_conv_ucstr_to_bcstr_with_cmgr (msg, &wcslen, client->errmsg.xerrmsg, &mbslen, client->_cmgr);
|
|
|
|
return client->errmsg.xerrmsg;
|
|
#endif
|
|
}
|
|
|
|
const hcl_uch_t* hcl_client_geterrumsg (hcl_client_t* client)
|
|
{
|
|
#if defined(HCL_OOCH_IS_BCH)
|
|
const hcl_ooch_t* msg;
|
|
hcl_oow_t wcslen, mbslen;
|
|
|
|
msg = (client->errmsg.len <= 0)? hcl_errnum_to_errstr(client->errnum): client->errmsg.buf;
|
|
|
|
wcslen = HCL_COUNTOF(client->errmsg.xerrmsg);
|
|
hcl_conv_bcstr_to_ucstr_with_cmgr (msg, &mbslen, client->errmsg.xerrmsg, &wcslen, client->_cmgr, 1);
|
|
|
|
return client->errmsg.xerrmsg;
|
|
#else
|
|
return (client->errmsg.len == '\0')? hcl_errnum_to_errstr(client->errnum): client->errmsg.buf;
|
|
#endif
|
|
}
|
|
|
|
void hcl_client_seterrnum (hcl_client_t* client, hcl_errnum_t errnum)
|
|
{
|
|
/*if (client->shuterr) return; */
|
|
client->errnum = errnum;
|
|
client->errmsg.len = 0;
|
|
}
|
|
|
|
void hcl_client_seterrbfmt (hcl_client_t* client, hcl_errnum_t errnum, const hcl_bch_t* fmt, ...)
|
|
{
|
|
va_list ap;
|
|
|
|
va_start (ap, fmt);
|
|
hcl_seterrbfmtv (client->dummy_hcl, errnum, fmt, ap);
|
|
va_end (ap);
|
|
|
|
HCL_ASSERT (client->dummy_hcl, HCL_COUNTOF(client->errmsg.buf) == HCL_COUNTOF(client->dummy_hcl->errmsg.buf));
|
|
client->errnum = errnum;
|
|
hcl_copy_oochars (client->errmsg.buf, client->dummy_hcl->errmsg.buf, HCL_COUNTOF(client->errmsg.buf));
|
|
client->errmsg.len = client->dummy_hcl->errmsg.len;
|
|
}
|
|
|
|
void hcl_client_seterrufmt (hcl_client_t* client, hcl_errnum_t errnum, const hcl_uch_t* fmt, ...)
|
|
{
|
|
va_list ap;
|
|
|
|
va_start (ap, fmt);
|
|
hcl_seterrufmtv (client->dummy_hcl, errnum, fmt, ap);
|
|
va_end (ap);
|
|
|
|
HCL_ASSERT (client->dummy_hcl, HCL_COUNTOF(client->errmsg.buf) == HCL_COUNTOF(client->dummy_hcl->errmsg.buf));
|
|
client->errnum = errnum;
|
|
hcl_copy_oochars (client->errmsg.buf, client->dummy_hcl->errmsg.buf, HCL_COUNTOF(client->errmsg.buf));
|
|
client->errmsg.len = client->dummy_hcl->errmsg.len;
|
|
}
|
|
|
|
/* ========================================================================= */
|
|
|
|
void hcl_client_logbfmt (hcl_client_t* client, hcl_bitmask_t mask, const hcl_bch_t* fmt, ...)
|
|
{
|
|
va_list ap;
|
|
va_start (ap, fmt);
|
|
hcl_logbfmtv (client->dummy_hcl, mask, fmt, ap);
|
|
va_end (ap);
|
|
}
|
|
|
|
void hcl_client_logufmt (hcl_client_t* client, hcl_bitmask_t mask, const hcl_uch_t* fmt, ...)
|
|
{
|
|
va_list ap;
|
|
va_start (ap, fmt);
|
|
hcl_logufmtv (client->dummy_hcl, mask, fmt, ap);
|
|
va_end (ap);
|
|
}
|
|
|
|
/* ========================================================================= */
|
|
|
|
void* hcl_client_allocmem (hcl_client_t* client, hcl_oow_t size)
|
|
{
|
|
void* ptr;
|
|
|
|
ptr = HCL_MMGR_ALLOC(client->_mmgr, size);
|
|
if (!ptr) hcl_client_seterrnum (client, HCL_ESYSMEM);
|
|
return ptr;
|
|
}
|
|
|
|
void* hcl_client_callocmem (hcl_client_t* client, hcl_oow_t size)
|
|
{
|
|
void* ptr;
|
|
|
|
ptr = HCL_MMGR_ALLOC(client->_mmgr, size);
|
|
if (!ptr) hcl_client_seterrnum (client, HCL_ESYSMEM);
|
|
else HCL_MEMSET (ptr, 0, size);
|
|
return ptr;
|
|
}
|
|
|
|
void* hcl_client_reallocmem (hcl_client_t* client, void* ptr, hcl_oow_t size)
|
|
{
|
|
ptr = HCL_MMGR_REALLOC(client->_mmgr, ptr, size);
|
|
if (!ptr) hcl_client_seterrnum (client, HCL_ESYSMEM);
|
|
return ptr;
|
|
}
|
|
|
|
void hcl_client_freemem (hcl_client_t* client, void* ptr)
|
|
{
|
|
HCL_MMGR_FREE (client->_mmgr, ptr);
|
|
}
|
|
|
|
/* ========================================================================= */
|
|
|
|
struct proto_xtn_t
|
|
{
|
|
hcl_client_t* client;
|
|
};
|
|
typedef struct proto_xtn_t proto_xtn_t;
|
|
|
|
static int client_connect (hcl_client_t* client, const char* ipaddr, int reuse_addr)
|
|
{
|
|
hcl_sckaddr_t sckaddr;
|
|
hcl_scklen_t scklen;
|
|
int sckfam;
|
|
int sck = -1;
|
|
hcl_xproto_t* proto = HCL_NULL;
|
|
|
|
proto_xtn_t* proto_xtn;
|
|
hcl_xproto_cb_t proto_cb;
|
|
|
|
sckfam = hcl_bchars_to_sckaddr(ipaddr, strlen(ipaddr), &sckaddr, &scklen);
|
|
if (sckfam <= -1)
|
|
{
|
|
hcl_client_seterrbfmt (client, HCL_EINVAL, "cannot convert ip address - %hs", ipaddr);
|
|
goto oops;
|
|
}
|
|
|
|
sck = socket(sckfam, SOCK_STREAM, 0);
|
|
if (sck <= -1)
|
|
{
|
|
hcl_client_seterrbfmt (client, HCL_ESYSERR, "cannot create socket - %hs", strerror(errno));
|
|
goto oops;
|
|
}
|
|
|
|
hcl_sys_set_cloexec(sck, 1);
|
|
|
|
if (reuse_addr)
|
|
{
|
|
if (sckfam == AF_INET)
|
|
{
|
|
struct sockaddr_in anyaddr;
|
|
int opt = 1;
|
|
setsockopt(sck, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(opt));
|
|
memset (&anyaddr, 0, HCL_SIZEOF(anyaddr));
|
|
anyaddr.sin_family = sckfam;
|
|
if (bind(sck, (struct sockaddr *)&anyaddr, scklen) <= -1)
|
|
{
|
|
hcl_client_seterrbfmt (client, HCL_ESYSERR,
|
|
"cannot bind socket %d - %hs", sck, strerror(errno));
|
|
goto oops;
|
|
}
|
|
}
|
|
else if (sckfam == AF_INET6)
|
|
{
|
|
struct sockaddr_in6 anyaddr;
|
|
int opt = 1;
|
|
setsockopt(sck, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(opt));
|
|
memset (&anyaddr, 0, HCL_SIZEOF(anyaddr));
|
|
anyaddr.sin6_family = sckfam;
|
|
if (bind(sck, (struct sockaddr *)&anyaddr, scklen) <= -1)
|
|
{
|
|
hcl_client_seterrbfmt (client, HCL_ESYSERR,
|
|
"cannot bind socket %d - %hs", sck, strerror(errno));
|
|
goto oops;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* TODO: async connect? */
|
|
/* TODO: connect timeout */
|
|
if (connect(sck, (struct sockaddr*)&sckaddr, scklen) <= -1)
|
|
{
|
|
hcl_client_seterrbfmt (client, HCL_ESYSERR,
|
|
"cannot connect socket %d to %hs - %hs", sck, ipaddr, strerror(errno));
|
|
goto oops;
|
|
}
|
|
|
|
hcl_sys_set_nonblock(sck, 1); /* make it nonblocking after connection has been established */
|
|
|
|
memset (&proto, 0, HCL_SIZEOF(proto_cb));
|
|
proto_cb.on_packet = client->prim.on_packet;
|
|
|
|
proto = hcl_xproto_open(hcl_client_getmmgr(client), &proto_cb, HCL_SIZEOF(*proto_xtn));
|
|
if (HCL_UNLIKELY(!proto))
|
|
{
|
|
hcl_client_seterrbfmt (client, HCL_ESYSERR, "cannot open protocol to %s", ipaddr);
|
|
goto oops;
|
|
}
|
|
proto_xtn = hcl_xproto_getxtn(proto);
|
|
proto_xtn->client = client;
|
|
|
|
client->sck = sck;
|
|
client->proto = proto;
|
|
return 0;
|
|
|
|
oops:
|
|
if (proto) hcl_xproto_close (proto);
|
|
if (sck >= 0) close (sck);
|
|
return -1;
|
|
}
|
|
|
|
|
|
static void client_close (hcl_client_t* client)
|
|
{
|
|
if (client->proto)
|
|
{
|
|
hcl_xproto_close (client->proto);
|
|
client->proto = HCL_NULL;
|
|
}
|
|
|
|
if (client->sck >= 0)
|
|
{
|
|
close (client->sck);
|
|
client->sck = -1;
|
|
}
|
|
}
|
|
|
|
|
|
#if 0
|
|
static int client_send_code_to_server (hcl_client_t* client, const hcl_bch_t* ptr, hcl_oow_t len)
|
|
{
|
|
hcl_xpkt_hdr_t hdr;
|
|
struct iovec iov[2];
|
|
hcl_uint16_t seglen;
|
|
|
|
|
|
while (*sccur != '\0' && sccur - scptr < HCL_XPKT_MAX_PLD_LEN) sccur++;
|
|
|
|
seglen = sccur - scptr;
|
|
|
|
hdr.id = 1; /* TODO: */
|
|
hdr.type = HCL_XPKT_CODE | (((seglen >> 8) & 0x0F) << 4);
|
|
hdr.len = seglen & 0xFF;
|
|
|
|
iov[0].iov_base = &hdr;
|
|
iov[0].iov_len = HCL_SIZEOF(hdr);
|
|
iov[1].iov_base = scptr;
|
|
iov[1].iov_len = seglen;
|
|
|
|
hcl_sys_send_iov (client->sck, iov, 2); /* TODO: error check */
|
|
|
|
scptr = sccur;
|
|
}
|
|
|
|
|
|
static int client_send_code_to_server (hcl_client_t* client, hcl_iovec_t* iov, hcl_oow_t iovcnt)
|
|
{
|
|
hcl_oow_t urem, len;
|
|
hcl_oow_t index = 0, i;
|
|
int backup_index = -1, dcnt;
|
|
hcl_iovec_t backup;
|
|
|
|
|
|
len = 0;
|
|
for (i = 0; i < iovcnt; i++) len += iov[i].iov_len;
|
|
urem = len;
|
|
|
|
|
|
if (client->cwq)
|
|
{
|
|
goto enqueue;
|
|
}
|
|
|
|
do
|
|
{
|
|
dcnt = iovcnt - index;
|
|
|
|
memset (&msg, 0, HCL_SIZEOF(msg));
|
|
msg.msg_iov = (struct iovec*)&iov[index];
|
|
msg.msg_iovlen = &dcnt;
|
|
|
|
sendmsg(xxx, &msg, 0);
|
|
|
|
if (x <= -1) return -1;
|
|
else if (x == 0) goto enqueue;
|
|
|
|
urem -= dcnt;
|
|
while (index < iovcnt && (hio_oow_t)dcnt >= iov[index].iov_len)
|
|
dcnt -= iov[index++].iov_len;
|
|
|
|
if (index == iovcnt) break;
|
|
|
|
if (backup_index != index)
|
|
{
|
|
if (backup_index >= 0) iov[backup_index] = backup;
|
|
backup = iov[index];
|
|
backup_index = index;
|
|
}
|
|
|
|
iov[index].iov_ptr = (void*)((hio_uint8_t*)iov[index].iov_ptr + dcnt);
|
|
iov[index].iov_len -= dcnt;
|
|
}
|
|
while (1);
|
|
|
|
if (backup_index >= 0) iov[backup_index] = backup;
|
|
if (iovcnt <= 0)
|
|
{
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
while (1)
|
|
{
|
|
ssize_t nwritten;
|
|
struct msghdr msg;
|
|
|
|
memset (&msg, 0, HCL_SIZEOF(msg));
|
|
msg.msg_iov = (struct iovec*)&iov[index];
|
|
msg.msg_iovlen = count - index;
|
|
nwritten = sendmsg(sck, &msg, 0);
|
|
if (nwritten <= -1) return -1;
|
|
|
|
while (index < count && (size_t)nwritten >= iov[index].iov_len)
|
|
nwritten -= iov[index++].iov_len;
|
|
|
|
if (index == count) break;
|
|
|
|
iov[index].iov_base = (void*)((hcl_uint8_t*)iov[index].iov_base + nwritten);
|
|
iov[index].iov_len -= nwritten;
|
|
}
|
|
*/
|
|
|
|
return 0;
|
|
}
|
|
#endif
|
|
/* ========================================================================= */
|
|
|
|
static int on_control_event (hcl_client_t* client, struct pollfd* pfd)
|
|
{
|
|
char tmp[128];
|
|
while (read(client->mux_pipe[0], tmp, HCL_SIZEOF(tmp)) > 0) /* nothing */;
|
|
/* TODO: handle different command? */
|
|
}
|
|
|
|
|
|
|
|
static int on_server_event (hcl_client_t* client, struct pollfd* pfd, int shut_wr_after_req)
|
|
{
|
|
|
|
#if 0
|
|
if (pfd->revents & POLLERR)
|
|
{
|
|
hcl_client_seterrbfmt (client, HCL_ESYSERR, "error condition detected on %d", client->sck);
|
|
goto oops;
|
|
}
|
|
|
|
if (pfd->revents & POLLOUT)
|
|
{
|
|
hcl_xpkt_hdr_t hdr;
|
|
struct iovec iov[2];
|
|
hcl_uint16_t seglen;
|
|
|
|
while (*sccur != '\0' && sccur - scptr < HCL_XPKT_MAX_PLD_LEN) sccur++;
|
|
|
|
seglen = sccur - scptr;
|
|
|
|
hdr.id = 1; /* TODO: */
|
|
hdr.type = HCL_XPKT_CODE | (((seglen >> 8) & 0x0F) << 4);
|
|
hdr.len = seglen & 0xFF;
|
|
|
|
iov[0].iov_base = &hdr;
|
|
iov[0].iov_len = HCL_SIZEOF(hdr);
|
|
iov[1].iov_base = scptr;
|
|
iov[1].iov_len = seglen;
|
|
|
|
hcl_sys_send_iov (client->sck, iov, 2); /* TODO: error check */
|
|
|
|
scptr = sccur;
|
|
|
|
if (*sccur == '\0')
|
|
{
|
|
hdr.id = 1; /* TODO: */
|
|
hdr.type = HCL_XPKT_EXECUTE;
|
|
hdr.len = 0;
|
|
|
|
iov[0].iov_base = &hdr;
|
|
iov[0].iov_len = HCL_SIZEOF(hdr);
|
|
hcl_sys_send_iov (client->sck, iov, 1);
|
|
|
|
if (shut_wr_after_req)
|
|
{
|
|
shutdown (client->sck, SHUT_WR);
|
|
}
|
|
else
|
|
{
|
|
hdr.type = HCL_XPKT_DISCONNECT;
|
|
hdr.id = 1; /* TODO: */
|
|
hdr.len = 0;
|
|
|
|
iov[0].iov_base = &hdr;
|
|
iov[0].iov_len = HCL_SIZEOF(hdr);
|
|
hcl_sys_send_iov (client->sck, iov, 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (pfd->revents & POLLIN)
|
|
{
|
|
hcl_oow_t bcap;
|
|
hcl_uint8_t* bptr;
|
|
ssize_t x;
|
|
|
|
bptr = hcl_xproto_getbuf(client->proto, &bcap);;
|
|
x = recv(client->sck, bptr, bcap, 0);
|
|
if (x <= -1)
|
|
{
|
|
if (errno == EINTR) goto carry_on; /* didn't read read */
|
|
/*hcl_seterrwithsyserr (hcl, 0, errno); */
|
|
/* TODO: error info set... */
|
|
return -1;
|
|
}
|
|
if (x == 0) hcl_xproto_seteof(client->proto, 1);
|
|
hcl_xproto_advbuf (client->proto, x);
|
|
}
|
|
|
|
|
|
carry_on:
|
|
while (hcl_xproto_ready(client->proto))
|
|
{
|
|
int n;
|
|
|
|
if ((n = hcl_xproto_process(client->proto)) <= -1)
|
|
{
|
|
/* TODO: proper error message */
|
|
return -1;
|
|
}
|
|
if (n == 0)
|
|
{
|
|
/* TODO: chceck if there is remaining data in the buffer...?? */
|
|
printf ("NO MORE DATA. EXITING...\n");
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
if (hcl_xproto_geteof(client->proto)) break;
|
|
#endif
|
|
}
|
|
|
|
static int on_script_event (hcl_client_t* client, struct pollfd* pfd)
|
|
{
|
|
ssize_t n;
|
|
hcl_uint8_t buf[128];
|
|
|
|
n = read(pfd->fd, buf, HCL_SIZEOF(buf));
|
|
if (n <= -1)
|
|
{
|
|
}
|
|
else if (n == 0)
|
|
{
|
|
}
|
|
else
|
|
{
|
|
}
|
|
}
|
|
|
|
int hcl_client_start (hcl_client_t* client, const char* ipaddr, const char* script, int reuse_addr, int shut_wr_after_req)
|
|
{
|
|
const char* scptr, * sccur;
|
|
int cin = STDIN_FILENO;
|
|
int cout = STDOUT_FILENO;
|
|
int cerr = STDERR_FILENO;
|
|
|
|
/* TODO: cin, cout, cerr could be actual files or something other than the console.
|
|
the actual loop won't begin until all these file descriptors are ready */
|
|
|
|
if (client_connect(client, ipaddr, reuse_addr) <= -1) return -1; /* TODO: support time out or abort while connecting... */
|
|
|
|
scptr = sccur = script;
|
|
while (1)
|
|
{
|
|
ssize_t n, i;
|
|
struct pollfd pfd[5];
|
|
|
|
n = 0;
|
|
|
|
pfd[n].fd = client->mux_pipe[0];
|
|
pfd[n].events = POLLIN;
|
|
pfd[n++].revents = 0;
|
|
|
|
pfd[n].fd = client->sck;
|
|
pfd[n].events = POLLIN;
|
|
if (*sccur != '\0') pfd[n].events |= POLLOUT;
|
|
pfd[n++].revents = 0;
|
|
|
|
if (cin >= 0)
|
|
{
|
|
pfd[n].fd = cin;
|
|
pfd[n].events = POLLIN;
|
|
pfd[n++].revents = 0;
|
|
}
|
|
|
|
n = poll(pfd, n, 1000);
|
|
if (n <= -1)
|
|
{
|
|
hcl_client_seterrbfmt (client, HCL_ESYSERR, "poll error on %d - %hs", client->sck, strerror(n));
|
|
goto oops;
|
|
}
|
|
|
|
if (n == 0)
|
|
{
|
|
/* TODO: proper timeout handling */
|
|
continue;
|
|
}
|
|
|
|
|
|
for (i = 0; i < n; i++)
|
|
{
|
|
if (pfd[i].fd == client->mux_pipe[0])
|
|
{
|
|
on_control_event (client, &pfd[i]);
|
|
}
|
|
else if (pfd[i].fd == client->sck)
|
|
{
|
|
/* event from the server */
|
|
on_server_event (client, &pfd[i], shut_wr_after_req);
|
|
}
|
|
else if (pfd[i].fd == cin)
|
|
{
|
|
on_script_event (client, &pfd[i]);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
done:
|
|
/* TODO: we can check if the buffer has all been consumed. if not, there is trailing garbage.. */
|
|
/*{
|
|
struct linger linger;
|
|
linger.l_onoff = 1;
|
|
linger.l_linger = 0;
|
|
setsockopt (client->sck, SOL_SOCKET, SO_LINGER, (char *) &linger, sizeof(linger));
|
|
}*/
|
|
|
|
|
|
client_close (client);
|
|
return 0;
|
|
|
|
oops:
|
|
client_close (client);
|
|
return -1;
|
|
}
|
|
|
|
void hcl_client_stop (hcl_client_t* client)
|
|
{
|
|
/* TODO: */
|
|
/* TODO: break the cleint loop */
|
|
client->stopreq = 1;
|
|
write (client->mux_pipe[1], "Q", 1); /* don't care about failure */
|
|
}
|